|
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 |