omadrm/drmengine/utils/src/drmaescrypto.cpp
changeset 0 95b198f216e5
child 18 8a03a285ab14
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/omadrm/drmengine/utils/src/drmaescrypto.cpp	Thu Dec 17 08:52:27 2009 +0200
@@ -0,0 +1,98 @@
+/*
+* 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