/*

* Copyright (c) 2007 Nokia Corporation and/or its subsidiary(-ies).

* All rights reserved.

* This component and the accompanying materials are made available

* under the terms of "Eclipse Public License v1.0"

* which accompanies this distribution, and is available

* at the URL "http://www.eclipse.org/legal/epl-v10.html".

*

* Initial Contributors:

* Nokia Corporation - initial contribution.

*

* Contributors:

*

* Description:  AES encryption utility for drm engine use

*

*/

// INCLUDE FILES

#include <e32std.h>

#include <symmetric.h>

#include <rijndael.h>

#include <DRMTypes.h>

#include "drmaescrypto.h"

// ============================ MEMBER FUNCTIONS ===============================

// -----------------------------------------------------------------------------

// DrmAesCrypto::DrmAesEncryptL

// Encrypt data using a given key and insert initial vector on beginning

// -----------------------------------------------------------------------------

//

EXPORT_C HBufC8* DrmAesCrypto::DrmAesEncryptL(

    const TDesC8& aKey,

    const TDesC8& aIV,

    const TBool aAddPadding,

    const TDesC8& aData )

    {

    HBufC8* retBuf( NULL );

    CModeCBCEncryptor* cbc( NULL );

    TInt lastBlockStart( 0 );

    TPtr8 data( NULL, 0, 0 );

    if( aIV.Length() != KDCFKeySize || aKey.Length() != KDCFKeySize )

        {

        User::Leave( KErrArgument );

        }

    cbc = CModeCBCEncryptor::NewL( CAESEncryptor::NewLC( aKey ), aIV );

    CleanupStack::Pop(); // CAESEncryptor, owned by cbc.

    CleanupStack::PushL( cbc );

    retBuf = HBufC8::NewLC( aData.Size() + aIV.Size() + KDCFKeySize );

    data.Set( retBuf->Des() );

    data.Copy( aIV );

    data.Append( aData );

    lastBlockStart = data.Length() - ( data.Length() % KDCFKeySize );

    // Loop through the data, excluding aIV

    for ( TInt i = KDCFKeySize; i  < lastBlockStart; i+= KDCFKeySize )

        {

        data.Set( retBuf->Des().MidTPtr( i, KDCFKeySize ) );

        cbc->Transform( data );

        }

    if ( aAddPadding )

        {

        TInt dataLength = retBuf->Length();

        TUint8 padding( static_cast< TUint8 >

            ( lastBlockStart + KDCFKeySize - dataLength ) );

        __ASSERT_DEBUG( lastBlockStart + KDCFKeySize - dataLength <= KDCFKeySize,

            User::Invariant() );

        retBuf->Des().SetLength( lastBlockStart + KDCFKeySize );

        for ( TInt i = dataLength; i < retBuf->Length(); ++i )

            {

            retBuf->Des()[ i ] = padding;

            }

        data.Set( retBuf->Des().MidTPtr( lastBlockStart, KDCFKeySize ) );

        cbc->Transform( data );

        }

    CleanupStack::Pop( retBuf );

    CleanupStack::PopAndDestroy( cbc );

    return retBuf;

    }

//  End of File