1 /*

     2 * Copyright (c) 2003-2009 Nokia Corporation and/or its subsidiary(-ies).

     3 * All rights reserved.

     4 * This component and the accompanying materials are made available

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

     6 * which accompanies this distribution, and is available

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

     8 *

     9 * Initial Contributors:

    10 * Nokia Corporation - initial contribution.

    11 *

    12 * Contributors:

    13 *

    14 * Description:

    15 *

    16 */

    17 

    18 

    19 // INCLUDE FILES

    20 #include <e32std.h>

    21 #include <f32file.h>

    22 #include <flogger.h>

    23 #include <x509cert.h>

    24 #include <symmetric.h>

    25 #include <asymmetric.h>

    26 

    27 #ifdef RD_MULTIPLE_DRIVE

    28 #include <driveinfo.h>

    29 #endif

    30 

    31 #include <zipfile.h>

    32 #include <aknnotewrappers.h>

    33 

    34 #include "DrmKeyStorage.h"

    35 

    36 // EXTERNAL DATA STRUCTURES

    37 

    38 // EXTERNAL FUNCTION PROTOTYPES

    39 

    40 // CONSTANTS

    41 

    42 // MACROS

    43 

    44 // LOCAL CONSTANTS AND MACROS

    45 

    46 // Test logging macros

    47 

    48 #ifdef RD_MULTIPLE_DRIVE

    49 _LIT(KPrivateRightsDir, "%c:\\private\\101F51F2\\PKI");

    50 _LIT(KFullLogDir, "%c:\\logs\\drm\\");

    51 #else

    52 _LIT(KPrivateRightsDir, "c:\\private\\101F51F2\\PKI");

    53 _LIT(KFullLogDir, "c:\\logs\\drm\\");

    54 #endif

    55 

    56 

    57 #ifdef __WINS__

    58 _LIT(KInputDir, "c:\\data\\drm\\keys\\");

    59 _LIT(KInputFilePattern, "c:\\data\\drm\\keys\\SigningCert*");

    60 #else

    61 #ifdef RD_MULTIPLE_DRIVE

    62 _LIT(KInputDir, "%c:\\drm\\keys\\");

    63 _LIT(KInputFilePattern, "%c:\\drm\\keys\\SigningCert*");

    64 #else

    65 _LIT(KInputDir, "e:\\drm\\keys\\");

    66 _LIT(KInputFilePattern, "e:\\drm\\keys\\SigningCert*");

    67 #endif

    68 #endif

    69 

    70 _LIT(KLogDir, "drm");

    71 _LIT(KLogName, "MtDrmKeyStorage.log");

    72 _LIT(KDeviceKeyFileName, "DevicePrivateKey.der");

    73 _LIT(KDeviceCertFileName, "DeviceCert.der");

    74 

    75 #define TEST_STEP(string) \

    76     GLog.WriteFormat(_L("Next Test: %S"), &string);

    77 

    78 #define CHECK(condition) \

    79     if (!condition) GLog.WriteFormat(_L("FAIL: line %d"), __LINE__);

    80 

    81 // MODULE DATA STRUCTURES

    82 

    83 RFs GFs;

    84 RFileLogger GLog;

    85 

    86 // STATIC TEST CONTENT AND RIGHTS OBJECTS

    87 

    88 // LOCAL FUNCTION PROTOTYPES

    89 

    90 // ==================== LOCAL FUNCTIONS ====================

    91 

    92 LOCAL_C void ReadFileL(HBufC8*& aContent, const TDesC& aName)

    93     {

    94     TInt size = 0;

    95     RFile file;

    96 

    97     User::LeaveIfError(file.Open(GFs, aName, EFileRead));

    98     User::LeaveIfError(file.Size(size));

    99     aContent = HBufC8::NewLC(size);

   100     TPtr8 ptr(aContent->Des());

   101     User::LeaveIfError(file.Read(ptr, size));

   102     CleanupStack::Pop(); //aContent

   103     }

   104 

   105 

   106 // ==================== TEST FUNCTIONS =====================

   107 

   108 LOCAL_C TUint MDrmKeyStorage_ImportDataL()

   109     {

   110     MDrmKeyStorage* storage = NULL;

   111     HBufC8* privateKey = NULL;

   112     HBufC8* cert = NULL;

   113     RArray<TPtrC8> certChain;

   114     RPointerArray<HBufC8> buffers;

   115     TFileName fileName;

   116     TInt i;

   117     CDir* dir;

   118     TEntry entry;

   119     TUint fileCount = 0;

   120 

   121     storage = DrmKeyStorageNewL();

   122 

   123 #ifndef RD_MULTIPLE_DRIVE

   124 

   125     GFs.SetSessionPath(KInputDir);

   126 

   127 #else //RD_MULTIPLE_DRIVE

   128 

   129     TFileName tempPath;

   130     TInt driveNumber( -1 );

   131     TChar driveLetter;

   132     DriveInfo::GetDefaultDrive( DriveInfo::EDefaultMassStorage, driveNumber );

   133     GFs.DriveToChar( driveNumber, driveLetter );

   134 

   135     tempPath.Format( KInputDir, (TUint)driveLetter );

   136 

   137     GFs.SetSessionPath(tempPath);

   138 

   139 #endif

   140 

   141     ReadFileL(privateKey, KDeviceKeyFileName);

   142     ReadFileL(cert, KDeviceCertFileName);

   143     buffers.Append(cert);

   144 

   145 #ifndef RD_MULTIPLE_DRIVE

   146 

   147     GFs.GetDir(KInputFilePattern, KEntryAttNormal, ESortByName, dir);

   148 

   149 #else //RD_MULTIPLE_DRIVE

   150 

   151     tempPath.Format( KInputFilePattern, (TUint)driveLetter );

   152 

   153     GFs.GetDir(tempPath, KEntryAttNormal, ESortByName, dir);

   154 

   155 #endif

   156 

   157     for (i = 0; i < dir->Count(); i++)

   158         {

   159         ReadFileL(cert, (*dir)[i].iName);

   160         buffers.Append(cert);

   161         }

   162     for (i = 0; i < buffers.Count(); i++)

   163         {

   164         certChain.Append(*(buffers[i]));

   165         }

   166     storage->ImportDataL(*privateKey, certChain);

   167     delete storage;

   168     delete privateKey;

   169     fileCount = buffers.Count();

   170     certChain.Close();

   171     buffers.ResetAndDestroy();

   172     buffers.Close();

   173     return fileCount;

   174     }

   175 

   176 

   177 TUint KeyStorage()

   178     {

   179     TUint result = 0;

   180     result = GFs.Connect();

   181     if( result != KErrNone )

   182         {

   183         return result;

   184         }

   185 

   186 #ifndef RD_MULTIPLE_DRIVE

   187 

   188     GFs.MkDirAll(KFullLogDir);

   189     GFs.MkDirAll(KPrivateRightsDir);

   190     GFs.MkDirAll(KInputDir);

   191 

   192 #else //RD_MULTIPLE_DRIVE

   193 

   194     TFileName tempPath;

   195     TInt driveNumber( -1 );

   196     TChar driveLetter;

   197     DriveInfo::GetDefaultDrive( DriveInfo::EDefaultSystem, driveNumber );

   198     GFs.DriveToChar( driveNumber, driveLetter );

   199 

   200     tempPath.Format( KFullLogDir, (TUint)driveLetter );

   201     GFs.MkDirAll(tempPath);

   202 

   203     tempPath.Format( KPrivateRightsDir, (TUint)driveLetter );

   204     GFs.MkDirAll(tempPath);

   205 

   206     DriveInfo::GetDefaultDrive( DriveInfo::EDefaultMassStorage, driveNumber );

   207     GFs.DriveToChar( driveNumber, driveLetter );

   208 

   209     tempPath.Format( KInputDir, (TUint)driveLetter );

   210     GFs.MkDirAll(tempPath);

   211 

   212 #endif

   213 

   214 

   215     result = GLog.Connect();

   216     if( result != KErrNone )

   217         {

   218         GFs.Close();

   219         return result;

   220         }

   221     GLog.CreateLog(KLogDir, KLogName, EFileLoggingModeOverwrite);

   222     GLog.Write(_L("Start %D"));

   223 

   224     TRAPD(err,result = MDrmKeyStorage_ImportDataL());

   225 

   226     GLog.WriteFormat(_L("Result: %d"), err);

   227     CHECK(err == KErrNone);

   228     GLog.CloseLog();

   229     GFs.Close();

   230     return result;

   231     }

   232 

   233 TUint KeyStorageFromZipL(const TDesC& aFileName, RFs& aFs)

   234     {

   235     RFile f;

   236     CZipFile* file;

   237     CZipFileMemberIterator* iter;

   238     CZipFileMember* member;

   239     MDrmKeyStorage* storage = NULL;

   240     TInt err = KErrNone;

   241     HBufC8* privateKey = NULL;

   242     HBufC8* cert = NULL;

   243     RArray<TPtrC8> certChain;

   244     TInt numCerts;

   245     TInt i;

   246     RZipFileMemberReaderStream* data;

   247     TPtr8 ptr(NULL, 0);

   248     TFileName fileName;

   249     RPointerArray<HBufC8> buffers;

   250 

   251     storage = DrmKeyStorageNewL();

   252 

   253     err = f.Open( aFs, aFileName, EFileShareReadersOrWriters | EFileRead );

   254     if ( err == KErrInUse )

   255         {

   256         err = f.Open( aFs, aFileName, EFileShareAny | EFileRead );

   257         if ( err == KErrInUse )

   258             {

   259             err = f.Open( aFs, aFileName, EFileShareReadersOnly| EFileRead );

   260             }

   261         }

   262     User::LeaveIfError( err );

   263 

   264     CleanupClosePushL<RFile>( f );

   265     file = CZipFile::NewL( aFs, f );

   266     CleanupStack::PushL( file );

   267     iter = file->GetMembersL();

   268     CleanupStack::PushL( iter );

   269     member = iter->NextL();

   270     numCerts = 0;

   271     while ( member != NULL )

   272         {

   273         if (member->Name()->Left(11).CompareF(_L("SigningCert")) == 0)

   274             {

   275             numCerts++;

   276             }

   277         delete member;

   278         member = iter->NextL();

   279         }

   280 

   281     member = file->CaseInsensitiveMemberL(_L("DeviceCert.der"));

   282     cert = HBufC8::NewL(member->UncompressedSize());

   283     buffers.Append(cert);

   284     file->GetInputStreamL(member, data);

   285     ptr.Set(cert->Des());

   286     data->Read(ptr, member->UncompressedSize());

   287     certChain.Append(ptr);

   288     delete data;

   289     delete member;

   290     for (i = 0; i < numCerts; i++)

   291         {

   292         fileName.Copy(_L("SigningCert"));

   293         if (i < 10)

   294             {

   295             fileName.Append(_L("0"));

   296             }

   297         fileName.AppendNum(i);

   298         fileName.Append(_L(".der"));

   299         member = file->CaseInsensitiveMemberL(fileName);

   300         cert = HBufC8::NewL(member->UncompressedSize());

   301         buffers.Append(cert);

   302         file->GetInputStreamL(member, data);

   303         ptr.Set(cert->Des());

   304         data->Read(ptr, member->UncompressedSize());

   305         certChain.Append(ptr);

   306         delete data;

   307         delete member;

   308         }

   309     member = file->CaseInsensitiveMemberL(_L("DevicePrivateKey.der"));

   310     privateKey = HBufC8::NewL(member->UncompressedSize());

   311     file->GetInputStreamL(member, data);

   312     ptr.Set(privateKey->Des());

   313     data->Read(ptr, member->UncompressedSize());

   314     delete data;

   315     delete member;

   316 

   317     storage->ImportDataL(*privateKey, certChain);

   318     delete privateKey;

   319     certChain.Close();

   320     buffers.ResetAndDestroy();

   321 

   322     CleanupStack::PopAndDestroy(3);

   323     delete storage;

   324 

   325     numCerts++;

   326     fileName.SetLength(0);

   327     fileName.AppendNum(numCerts);

   328     fileName.Append(_L(" keys imported"));

   329     CAknInformationNote* informationNote = new (ELeave) CAknInformationNote;

   330     informationNote->ExecuteLD(fileName);

   331 

   332     return KErrNone;

   333     }

   334