hackedteam/core-symbian

#include "aes.h"
#include <cryptosymmetric.h>         // For AES Stuff
#include <cryptopadding.h>             // For AES Stuff

HBufC8* AES::DecryptL(const TDesC8& encryptedData, const TDesC8& iv, const TDesC8& key)
    {
    // takes in consideration only the first 16bytes of the key
    TPtrC8 keyPtr = key.Left(key.Length());
    if (key.Length() > 16)
        keyPtr.Set( key.Left(16) );
    CBlockTransformation* aesDecr = CAESDecryptor::NewL(keyPtr);
    CModeCBCDecryptor* cbcMode = CModeCBCDecryptor::NewL(aesDecr, iv);
    CPadding* padding = CPaddingNone::NewL(aesDecr->BlockSize());
    CBufferedDecryptor* bufDecryptor = CBufferedDecryptor::NewLC(cbcMode, padding);
    
    // Data Length must be multiple of BlockSize() or the ProcessFinalL() will raise a Panic()
    ASSERT( encryptedData.Length() % aesDecr->BlockSize() == 0);
    if (encryptedData.Length() % aesDecr->BlockSize() != 0)
        {
        CleanupStack::PopAndDestroy(bufDecryptor);
        return HBufC8::NewL(0);
        }

    TInt maxOutputSize = bufDecryptor->MaxFinalOutputLength(encryptedData.Size());
    HBufC8* plain;
    TRAPD(err, plain = HBufC8::NewL(maxOutputSize));
    if(err != KErrNone)
        {
        CleanupStack::PopAndDestroy(bufDecryptor);
        plain = HBufC8::NewL(0);
        return plain;
        }
    CleanupStack::PushL(plain);
    TPtr8 plainPtr = plain->Des();
    bufDecryptor->ProcessFinalL(encryptedData, plainPtr);
    CleanupStack::Pop(plain);

    CleanupStack::PopAndDestroy(bufDecryptor);
    return plain;
    }

HBufC8* AES::EncryptL(const TDesC8& plainData, const TDesC8& iv, const TDesC8& key)
    {
    // takes in consideration only the first 16bytes of the key
    TPtrC8 keyPtr = key.Left(key.Length());
    if (key.Length() > 16)
        keyPtr.Set( key.Left(16) );
    CBlockTransformation* aesEncr = CAESEncryptor::NewL(keyPtr);
    CModeCBCEncryptor* cbcMode = CModeCBCEncryptor::NewL(aesEncr, iv);
    CPadding* padding = CPaddingNone::NewL(aesEncr->BlockSize());
    CBufferedEncryptor* bufEncryptor = CBufferedEncryptor::NewLC(cbcMode, padding);

    // Data Length must be multiple of BlockSize() or the ProcessFinalL() will raise a Panic()
    // Add Padding...
    TUint32 paddedLen = plainData.Length() + aesEncr->BlockSize() - 1;
    paddedLen = paddedLen - (paddedLen % aesEncr->BlockSize());
    
    
    RBuf8 paddedPlainData;
    paddedPlainData.CleanupClosePushL();
    TInt err=paddedPlainData.Create(paddedLen);
    if(err == KErrNoMemory)  //added jo'
        {
        CleanupStack::PopAndDestroy(&paddedPlainData);
        CleanupStack::PopAndDestroy(bufEncryptor);
        HBufC8* emptyBuf = HBufC8::NewL(0);
        return emptyBuf;
        }
    paddedPlainData.Copy(plainData);
    paddedPlainData.SetMax();
    ASSERT( paddedPlainData.Length() % aesEncr->BlockSize() == 0);

    TInt maxOutputSize = bufEncryptor->MaxFinalOutputLength(paddedPlainData.Size());
    HBufC8* encryptedData;
    TRAPD(error, (encryptedData = HBufC8::NewL(maxOutputSize)));
    if(error == KErrNoMemory)
        {
        CleanupStack::PopAndDestroy(&paddedPlainData);
        CleanupStack::PopAndDestroy(bufEncryptor);
        HBufC8* emptyBuf = HBufC8::NewL(0);
        return emptyBuf;
        }
    CleanupStack::PushL(encryptedData);
    TPtr8 encryptedPtr = encryptedData->Des();
    bufEncryptor->ProcessFinalL(paddedPlainData, encryptedPtr);
    CleanupStack::Pop(encryptedData);

    CleanupStack::PopAndDestroy(&paddedPlainData);
    CleanupStack::PopAndDestroy(bufEncryptor);

    return encryptedData;
    }

HBufC8* AES::DecryptPkcs5L(const TDesC8& encryptedData, const TDesC8& iv, const TDesC8& key)
    {
    // takes in consideration only the first 16bytes of the key
        TPtrC8 keyPtr = key.Left(key.Length());
        if (key.Length() > 16)
            keyPtr.Set( key.Left(16) );
        CBlockTransformation* aesDecr = CAESDecryptor::NewL(keyPtr);
        CModeCBCDecryptor* cbcMode = CModeCBCDecryptor::NewL(aesDecr, iv);
        CPadding* padding = CPaddingNone::NewL(aesDecr->BlockSize());
        CBufferedDecryptor* bufDecryptor = CBufferedDecryptor::NewLC(cbcMode, padding);
        
        // Data Length must be multiple of BlockSize() or the ProcessFinalL() will raise a Panic()
        ASSERT( encryptedData.Length() % aesDecr->BlockSize() == 0);
        if (encryptedData.Length() % aesDecr->BlockSize() != 0)
            {
            CleanupStack::PopAndDestroy(bufDecryptor);
            return HBufC8::NewL(0);
            }

        HBufC8* plain = HBufC8::NewLC(bufDecryptor->MaxFinalOutputLength(encryptedData.Length()));
        TPtr8 plainPtr = plain->Des();
        bufDecryptor->ProcessFinalL(encryptedData, plainPtr);
        
        // remove PKCS5 padding
        TUint8 paddedLen;
        Mem::Copy(&paddedLen,plain->Right(1).Ptr(),1);
        plain->Des().SetLength(plain->Size()-paddedLen);
        
        CleanupStack::Pop(plain);
        CleanupStack::PopAndDestroy(bufDecryptor);
        return plain;
    }

HBufC8* AES::EncryptPkcs5L(const TDesC8& plainData, const TDesC8& iv, const TDesC8& key)
    {
    // takes in consideration only the first 16bytes of the key
    TPtrC8 keyPtr = key.Left(key.Length());
    if (key.Length() > 16)
        keyPtr.Set( key.Left(16) );
    CBlockTransformation* aesEncr = CAESEncryptor::NewL(keyPtr);
    CModeCBCEncryptor* cbcMode = CModeCBCEncryptor::NewL(aesEncr, iv);
    CPadding* padding = CPaddingNone::NewL(aesEncr->BlockSize());
    CBufferedEncryptor* bufEncryptor = CBufferedEncryptor::NewLC(cbcMode, padding);

    // Data Length must be multiple of BlockSize() or the ProcessFinalL() will raise a Panic()
    // Add Padding PKCS5
    TInt8 paddedLen = 16 - (plainData.Length() % 16);
    RBuf8 paddedPlainData;
    paddedPlainData.CleanupClosePushL();
    TInt err = paddedPlainData.Create(paddedLen + plainData.Length());
    if(err == KErrNoMemory)  
        {
        CleanupStack::PopAndDestroy(&paddedPlainData);
        CleanupStack::PopAndDestroy(bufEncryptor);
        HBufC8* emptyBuf = HBufC8::NewL(0);
        return emptyBuf;
        }
    paddedPlainData.Copy(plainData);
    paddedPlainData.AppendFill(paddedLen,paddedLen);

    TInt maxOutputSize = bufEncryptor->MaxFinalOutputLength(paddedPlainData.Size());
    HBufC8* encryptedData;
    TRAPD(error, (encryptedData = HBufC8::NewL(maxOutputSize)));
    if(error == KErrNoMemory)
        {
        CleanupStack::PopAndDestroy(&paddedPlainData);
        CleanupStack::PopAndDestroy(bufEncryptor);
        HBufC8* emptyBuf = HBufC8::NewL(0);
        return emptyBuf;
        }
    CleanupStack::PushL(encryptedData);
    TPtr8 encryptedPtr = encryptedData->Des();
    bufEncryptor->ProcessFinalL(paddedPlainData, encryptedPtr);
    CleanupStack::Pop(encryptedData);
    CleanupStack::PopAndDestroy(&paddedPlainData);
    CleanupStack::PopAndDestroy(bufEncryptor);

    return encryptedData;
    }