[SHELL32]
[reactos.git] / rostests / winetests / wintrust / crypt.c
1 /* Unit test suite for wintrust crypt functions
2 *
3 * Copyright 2007 Paul Vriens
4 * Copyright 2008 Hans Leidekker for CodeWeavers
5 *
6 * This library is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
10 *
11 * This library is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
15 *
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with this library; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
19 *
20 */
21
22 #include <stdarg.h>
23 #include <stdio.h>
24
25 #include "windows.h"
26 #include "mscat.h"
27
28 #include "wine/test.h"
29
30 static char selfname[MAX_PATH];
31 static CHAR CURR_DIR[MAX_PATH];
32 static CHAR catroot[MAX_PATH];
33 static CHAR catroot2[MAX_PATH];
34
35 static const WCHAR hashmeW[] = {'h','a','s','h','m','e',0};
36 static const WCHAR attr1W[] = {'a','t','t','r','1',0};
37 static const WCHAR attr2W[] = {'a','t','t','r','2',0};
38
39 /*
40 * Minimalistic catalog file. To reconstruct, save text below as winetest.cdf,
41 * convert to DOS line endings and run 'makecat /cat winetest.cdf'
42 */
43
44 /*
45 [CatalogHeader]
46 Name=winetest.cat
47 ResultDir=.\
48 PublicVersion=0x00000001
49 EncodingType=
50 CATATTR1=0x10010001:attr1:value1
51 CATATTR2=0x10010001:attr2:value2
52
53 [CatalogFiles]
54 hashme=.\winetest.cdf
55 */
56
57 static const CHAR test_cdf[] =
58 "[CatalogHeader]\r\n"
59 "Name=winetest.cat\r\n"
60 "ResultDir=.\\\r\n"
61 "PublicVersion=0x00000001\r\n"
62 "EncodingType=\r\n"
63 "CATATTR1=0x10010001:attr1:value1\r\n"
64 "CATATTR2=0x10010001:attr2:value2\r\n"
65 "\r\n"
66 "[CatalogFiles]\r\n"
67 "hashme=.\\winetest.cdf\r\n";
68
69 static const BYTE test_catalog[] = {
70 0x30, 0x82, 0x01, 0xbc, 0x06, 0x09, 0x2a, 0x86, 0x48, 0x86, 0xf7, 0x0d, 0x01, 0x07, 0x02, 0xa0,
71 0x82, 0x01, 0xad, 0x30, 0x82, 0x01, 0xa9, 0x02, 0x01, 0x01, 0x31, 0x00, 0x30, 0x82, 0x01, 0x9e,
72 0x06, 0x09, 0x2b, 0x06, 0x01, 0x04, 0x01, 0x82, 0x37, 0x0a, 0x01, 0xa0, 0x82, 0x01, 0x8f, 0x30,
73 0x82, 0x01, 0x8b, 0x30, 0x0c, 0x06, 0x0a, 0x2b, 0x06, 0x01, 0x04, 0x01, 0x82, 0x37, 0x0c, 0x01,
74 0x01, 0x04, 0x10, 0xfa, 0x55, 0x2c, 0xc2, 0xf6, 0xcc, 0xdd, 0x11, 0x2a, 0x9c, 0x00, 0x14, 0x22,
75 0xec, 0x8f, 0x3b, 0x17, 0x0d, 0x30, 0x38, 0x31, 0x32, 0x31, 0x38, 0x31, 0x31, 0x32, 0x36, 0x34,
76 0x38, 0x5a, 0x30, 0x0e, 0x06, 0x0a, 0x2b, 0x06, 0x01, 0x04, 0x01, 0x82, 0x37, 0x0c, 0x01, 0x02,
77 0x05, 0x00, 0x30, 0x81, 0xdd, 0x30, 0x81, 0xda, 0x04, 0x0e, 0x68, 0x00, 0x61, 0x00, 0x73, 0x00,
78 0x68, 0x00, 0x6d, 0x00, 0x65, 0x00, 0x00, 0x00, 0x31, 0x81, 0xc7, 0x30, 0x61, 0x06, 0x0a, 0x2b,
79 0x06, 0x01, 0x04, 0x01, 0x82, 0x37, 0x02, 0x01, 0x04, 0x31, 0x53, 0x30, 0x51, 0x30, 0x2c, 0x06,
80 0x0a, 0x2b, 0x06, 0x01, 0x04, 0x01, 0x82, 0x37, 0x02, 0x01, 0x19, 0xa2, 0x1e, 0x80, 0x1c, 0x00,
81 0x3c, 0x00, 0x3c, 0x00, 0x3c, 0x00, 0x4f, 0x00, 0x62, 0x00, 0x73, 0x00, 0x6f, 0x00, 0x6c, 0x00,
82 0x65, 0x00, 0x74, 0x00, 0x65, 0x00, 0x3e, 0x00, 0x3e, 0x00, 0x3e, 0x30, 0x21, 0x30, 0x09, 0x06,
83 0x05, 0x2b, 0x0e, 0x03, 0x02, 0x1a, 0x05, 0x00, 0x04, 0x14, 0xed, 0xd6, 0x9c, 0x9c, 0xb2, 0xfc,
84 0xaa, 0x03, 0xe8, 0xd3, 0x20, 0xf6, 0xab, 0x28, 0xc3, 0xff, 0xbd, 0x07, 0x36, 0xf5, 0x30, 0x62,
85 0x06, 0x0a, 0x2b, 0x06, 0x01, 0x04, 0x01, 0x82, 0x37, 0x0c, 0x02, 0x02, 0x31, 0x54, 0x30, 0x52,
86 0x1e, 0x4c, 0x00, 0x7b, 0x00, 0x44, 0x00, 0x45, 0x00, 0x33, 0x00, 0x35, 0x00, 0x31, 0x00, 0x41,
87 0x00, 0x34, 0x00, 0x32, 0x00, 0x2d, 0x00, 0x38, 0x00, 0x45, 0x00, 0x35, 0x00, 0x39, 0x00, 0x2d,
88 0x00, 0x31, 0x00, 0x31, 0x00, 0x44, 0x00, 0x30, 0x00, 0x2d, 0x00, 0x38, 0x00, 0x43, 0x00, 0x34,
89 0x00, 0x37, 0x00, 0x2d, 0x00, 0x30, 0x00, 0x30, 0x00, 0x43, 0x00, 0x30, 0x00, 0x34, 0x00, 0x46,
90 0x00, 0x43, 0x00, 0x32, 0x00, 0x39, 0x00, 0x35, 0x00, 0x45, 0x00, 0x45, 0x00, 0x7d, 0x02, 0x02,
91 0x02, 0x00, 0xa0, 0x6a, 0x30, 0x68, 0x30, 0x32, 0x06, 0x0a, 0x2b, 0x06, 0x01, 0x04, 0x01, 0x82,
92 0x37, 0x0c, 0x02, 0x01, 0x04, 0x24, 0x30, 0x22, 0x1e, 0x0a, 0x00, 0x61, 0x00, 0x74, 0x00, 0x74,
93 0x00, 0x72, 0x00, 0x32, 0x02, 0x04, 0x10, 0x01, 0x00, 0x01, 0x04, 0x0e, 0x76, 0x00, 0x61, 0x00,
94 0x6c, 0x00, 0x75, 0x00, 0x65, 0x00, 0x32, 0x00, 0x00, 0x00, 0x30, 0x32, 0x06, 0x0a, 0x2b, 0x06,
95 0x01, 0x04, 0x01, 0x82, 0x37, 0x0c, 0x02, 0x01, 0x04, 0x24, 0x30, 0x22, 0x1e, 0x0a, 0x00, 0x61,
96 0x00, 0x74, 0x00, 0x74, 0x00, 0x72, 0x00, 0x31, 0x02, 0x04, 0x10, 0x01, 0x00, 0x01, 0x04, 0x0e,
97 0x76, 0x00, 0x61, 0x00, 0x6c, 0x00, 0x75, 0x00, 0x65, 0x00, 0x31, 0x00, 0x00, 0x00, 0x31, 0x00,
98 };
99
100 static BOOL (WINAPI * pCryptCATAdminAcquireContext)(HCATADMIN*, const GUID*, DWORD);
101 static BOOL (WINAPI * pCryptCATAdminReleaseContext)(HCATADMIN, DWORD);
102 static BOOL (WINAPI * pCryptCATAdminCalcHashFromFileHandle)(HANDLE hFile, DWORD*, BYTE*, DWORD);
103 static HCATINFO (WINAPI * pCryptCATAdminAddCatalog)(HCATADMIN, PWSTR, PWSTR, DWORD);
104 static BOOL (WINAPI * pCryptCATAdminRemoveCatalog)(HCATADMIN, LPCWSTR, DWORD);
105 static BOOL (WINAPI * pCryptCATAdminReleaseCatalogContext)(HCATADMIN, HCATINFO, DWORD);
106 static HANDLE (WINAPI * pCryptCATOpen)(LPWSTR, DWORD, HCRYPTPROV, DWORD, DWORD);
107 static BOOL (WINAPI * pCryptCATCatalogInfoFromContext)(HCATINFO, CATALOG_INFO *, DWORD);
108 static BOOL (WINAPI * pCryptCATCDFClose)(CRYPTCATCDF *);
109 static CRYPTCATATTRIBUTE * (WINAPI * pCryptCATCDFEnumCatAttributes)(CRYPTCATCDF *, CRYPTCATATTRIBUTE *,
110 PFN_CDF_PARSE_ERROR_CALLBACK);
111 static LPWSTR (WINAPI * pCryptCATCDFEnumMembersByCDFTagEx)(CRYPTCATCDF *, LPWSTR, PFN_CDF_PARSE_ERROR_CALLBACK,
112 CRYPTCATMEMBER **, BOOL, LPVOID);
113 static CRYPTCATCDF * (WINAPI * pCryptCATCDFOpen)(LPWSTR, PFN_CDF_PARSE_ERROR_CALLBACK);
114 static CRYPTCATATTRIBUTE * (WINAPI * pCryptCATEnumerateCatAttr)(HANDLE, CRYPTCATATTRIBUTE *);
115 static CRYPTCATMEMBER * (WINAPI * pCryptCATEnumerateMember)(HANDLE, CRYPTCATMEMBER *);
116 static CRYPTCATATTRIBUTE * (WINAPI * pCryptCATEnumerateAttr)(HANDLE, CRYPTCATMEMBER *, CRYPTCATATTRIBUTE *);
117 static BOOL (WINAPI * pCryptCATClose)(HANDLE);
118
119 static void InitFunctionPtrs(void)
120 {
121 HMODULE hWintrust = GetModuleHandleA("wintrust.dll");
122
123 #define WINTRUST_GET_PROC(func) \
124 p ## func = (void*)GetProcAddress(hWintrust, #func); \
125 if(!p ## func) { \
126 trace("GetProcAddress(%s) failed\n", #func); \
127 }
128
129 WINTRUST_GET_PROC(CryptCATAdminAcquireContext)
130 WINTRUST_GET_PROC(CryptCATAdminReleaseContext)
131 WINTRUST_GET_PROC(CryptCATAdminCalcHashFromFileHandle)
132 WINTRUST_GET_PROC(CryptCATAdminAddCatalog)
133 WINTRUST_GET_PROC(CryptCATAdminRemoveCatalog)
134 WINTRUST_GET_PROC(CryptCATAdminReleaseCatalogContext)
135 WINTRUST_GET_PROC(CryptCATOpen)
136 WINTRUST_GET_PROC(CryptCATCatalogInfoFromContext)
137 WINTRUST_GET_PROC(CryptCATCDFClose)
138 WINTRUST_GET_PROC(CryptCATCDFEnumCatAttributes)
139 WINTRUST_GET_PROC(CryptCATCDFEnumMembersByCDFTagEx)
140 WINTRUST_GET_PROC(CryptCATCDFOpen)
141 WINTRUST_GET_PROC(CryptCATEnumerateCatAttr)
142 WINTRUST_GET_PROC(CryptCATEnumerateMember)
143 WINTRUST_GET_PROC(CryptCATEnumerateAttr)
144 WINTRUST_GET_PROC(CryptCATClose)
145
146 #undef WINTRUST_GET_PROC
147 }
148
149 static GUID dummy = {0xdeadbeef,0xdead,0xbeef,{0xde,0xad,0xbe,0xef,0xde,0xad,0xbe,0xef}};
150
151 static void test_context(void)
152 {
153 BOOL ret;
154 HCATADMIN hca;
155 static GUID unknown = { 0xC689AABA, 0x8E78, 0x11D0, { 0x8C,0x47,0x00,0xC0,0x4F,0xC2,0x95,0xEE }}; /* WINTRUST.DLL */
156 CHAR dummydir[MAX_PATH];
157 DWORD attrs;
158
159 /* When CryptCATAdminAcquireContext is successful it will create
160 * several directories if they don't exist:
161 *
162 * ...\system32\CatRoot\{GUID}, this directory holds the .cat files
163 * ...\system32\CatRoot2\{GUID} (WinXP and up), here we find the catalog database for that GUID
164 *
165 * Windows Vista uses lowercase catroot and catroot2.
166 *
167 * When passed a NULL GUID it will create the following directories although on
168 * WinXP and up these directories are already present when Windows is installed:
169 *
170 * ...\system32\CatRoot\{127D0A1D-4EF2-11D1-8608-00C04FC295EE}
171 * ...\system32\CatRoot2\{127D0A1D-4EF2-11D1-8608-00C04FC295EE} (WinXP up)
172 *
173 * TODO: Find out what this GUID is/does.
174 *
175 * On WinXP and up there is also a TimeStamp file in some of directories that
176 * seem to indicate the last change to the catalog database for that GUID.
177 *
178 * On Windows 2000 some files are created/updated:
179 *
180 * ...\system32\CatRoot\SYSMAST.cbk
181 * ...\system32\CatRoot\SYSMAST.cbd
182 * ...\system32\CatRoot\{GUID}\CATMAST.cbk
183 * ...\system32\CatRoot\{GUID}\CATMAST.cbd
184 *
185 */
186
187 /* All NULL */
188 SetLastError(0xdeadbeef);
189 ret = pCryptCATAdminAcquireContext(NULL, NULL, 0);
190 ok(!ret, "Expected failure\n");
191 ok(GetLastError() == ERROR_INVALID_PARAMETER,
192 "Expected ERROR_INVALID_PARAMETER, got %d\n", GetLastError());
193
194 /* NULL GUID */
195 ret = pCryptCATAdminAcquireContext(&hca, NULL, 0);
196 ok(ret, "Expected success\n");
197 ok(hca != NULL, "Expected a context handle, got NULL\n");
198
199 /* All NULL */
200 SetLastError(0xdeadbeef);
201 ret = pCryptCATAdminReleaseContext(NULL, 0);
202 ok(!ret, "Expected failure\n");
203 ok(GetLastError() == ERROR_INVALID_PARAMETER,
204 "Expected ERROR_INVALID_PARAMETER, got %d\n", GetLastError());
205
206 /* Proper release */
207 SetLastError(0xdeadbeef);
208 ret = pCryptCATAdminReleaseContext(hca, 0);
209 ok(ret, "Expected success, got FALSE with %d\n", GetLastError());
210
211 /* Try to release a second time */
212 SetLastError(0xdeadbeef);
213 ret = pCryptCATAdminReleaseContext(hca, 0);
214 ok(!ret, "Expected failure\n");
215 ok(GetLastError() == ERROR_INVALID_PARAMETER,
216 "Expected ERROR_INVALID_PARAMETER, got %d\n", GetLastError());
217
218 /* NULL context handle and dummy GUID */
219 SetLastError(0xdeadbeef);
220 ret = pCryptCATAdminAcquireContext(NULL, &dummy, 0);
221 ok(!ret, "Expected failure\n");
222 ok(GetLastError() == ERROR_INVALID_PARAMETER,
223 "Expected ERROR_INVALID_PARAMETER, got %d\n", GetLastError());
224
225 /* Correct context handle and dummy GUID
226 *
227 * The tests run in the past unfortunately made sure that some directories were created.
228 *
229 * FIXME:
230 * We don't want to mess too much with these for now so we should delete only the ones
231 * that shouldn't be there like the deadbeef ones. We first have to figure out if it's
232 * save to remove files and directories from CatRoot/CatRoot2.
233 */
234
235 ret = pCryptCATAdminAcquireContext(&hca, &dummy, 0);
236 ok(ret || GetLastError() == ERROR_ACCESS_DENIED, "CryptCATAdminAcquireContext failed %u\n", GetLastError());
237 if (!ret && GetLastError() == ERROR_ACCESS_DENIED)
238 {
239 win_skip("Not running as administrator\n");
240 return;
241 }
242 ok(hca != NULL, "Expected a context handle, got NULL\n");
243
244 attrs = GetFileAttributes(catroot);
245 ok(attrs != INVALID_FILE_ATTRIBUTES, "Expected the CatRoot directory to exist\n");
246
247 /* Windows creates the GUID directory in capitals */
248 lstrcpyA(dummydir, catroot);
249 lstrcatA(dummydir, "\\{DEADBEEF-DEAD-BEEF-DEAD-BEEFDEADBEEF}");
250 attrs = GetFileAttributes(dummydir);
251 ok(attrs != INVALID_FILE_ATTRIBUTES,
252 "Expected CatRoot\\{DEADBEEF-DEAD-BEEF-DEAD-BEEFDEADBEEF} directory to exist\n");
253
254 /* Only present on XP or higher. */
255 attrs = GetFileAttributes(catroot2);
256 if (attrs != INVALID_FILE_ATTRIBUTES)
257 {
258 lstrcpyA(dummydir, catroot2);
259 lstrcatA(dummydir, "\\{DEADBEEF-DEAD-BEEF-DEAD-BEEFDEADBEEF}");
260 attrs = GetFileAttributes(dummydir);
261 ok(attrs != INVALID_FILE_ATTRIBUTES,
262 "Expected CatRoot2\\{DEADBEEF-DEAD-BEEF-DEAD-BEEFDEADBEEF} directory to exist\n");
263 }
264
265 ret = pCryptCATAdminReleaseContext(hca, 0);
266 ok(ret, "Expected success\n");
267
268 /* Correct context handle and GUID */
269 ret = pCryptCATAdminAcquireContext(&hca, &unknown, 0);
270 ok(ret, "Expected success\n");
271 ok(hca != NULL, "Expected a context handle, got NULL\n");
272
273 ret = pCryptCATAdminReleaseContext(hca, 0);
274 ok(ret, "Expected success\n");
275
276 /* Flags not equal to 0 */
277 ret = pCryptCATAdminAcquireContext(&hca, &unknown, 1);
278 ok(ret, "Expected success\n");
279 ok(hca != NULL, "Expected a context handle, got NULL\n");
280
281 ret = pCryptCATAdminReleaseContext(hca, 0);
282 ok(ret, "Expected success\n");
283 }
284
285 /* TODO: Check whether SHA-1 is the algorithm that's always used */
286 static void test_calchash(void)
287 {
288 BOOL ret;
289 HANDLE file;
290 DWORD hashsize = 0;
291 BYTE* hash;
292 BYTE expectedhash[20] = {0x3a,0xa1,0x19,0x08,0xec,0xa6,0x0d,0x2e,0x7e,0xcc,0x7a,0xca,0xf5,0xb8,0x2e,0x62,0x6a,0xda,0xf0,0x19};
293 CHAR temp[MAX_PATH];
294 DWORD written;
295
296 /* All NULL */
297 SetLastError(0xdeadbeef);
298 ret = pCryptCATAdminCalcHashFromFileHandle(NULL, NULL, NULL, 0);
299 ok(!ret, "Expected failure\n");
300 ok(GetLastError() == ERROR_INVALID_PARAMETER,
301 "Expected ERROR_INVALID_PARAMETER, got %d\n", GetLastError());
302
303 /* NULL filehandle, rest is legal */
304 SetLastError(0xdeadbeef);
305 ret = pCryptCATAdminCalcHashFromFileHandle(NULL, &hashsize, NULL, 0);
306 ok(!ret, "Expected failure\n");
307 ok(GetLastError() == ERROR_INVALID_PARAMETER,
308 "Expected ERROR_INVALID_PARAMETER, got %d\n", GetLastError());
309
310 /* Correct filehandle, rest is NULL */
311 file = CreateFileA(selfname, GENERIC_READ, 0, NULL, OPEN_EXISTING, 0, NULL);
312 SetLastError(0xdeadbeef);
313 ret = pCryptCATAdminCalcHashFromFileHandle(file, NULL, NULL, 0);
314 ok(!ret, "Expected failure\n");
315 ok(GetLastError() == ERROR_INVALID_PARAMETER,
316 "Expected ERROR_INVALID_PARAMETER, got %d\n", GetLastError());
317 CloseHandle(file);
318
319 /* All OK, but dwFlags set to 1 */
320 file = CreateFileA(selfname, GENERIC_READ, 0, NULL, OPEN_EXISTING, 0, NULL);
321 SetLastError(0xdeadbeef);
322 ret = pCryptCATAdminCalcHashFromFileHandle(file, &hashsize, NULL, 1);
323 ok(!ret, "Expected failure\n");
324 ok(GetLastError() == ERROR_INVALID_PARAMETER,
325 "Expected ERROR_INVALID_PARAMETER, got %d\n", GetLastError());
326 CloseHandle(file);
327
328 /* All OK, requesting the size of the hash */
329 file = CreateFileA(selfname, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, 0, NULL);
330 ok(file != INVALID_HANDLE_VALUE, "CreateFile failed %u\n", GetLastError());
331 SetLastError(0xdeadbeef);
332 ret = pCryptCATAdminCalcHashFromFileHandle(file, &hashsize, NULL, 0);
333 ok(ret, "Expected success %u\n", GetLastError());
334 ok(hashsize == 20," Expected a hash size of 20, got %d\n", hashsize);
335 ok(GetLastError() == ERROR_INSUFFICIENT_BUFFER,
336 "Expected ERROR_INSUFFICIENT_BUFFER, got %d\n", GetLastError());
337 CloseHandle(file);
338
339 /* All OK, retrieve the hash
340 * Double the hash buffer to see what happens to the size parameter
341 */
342 file = CreateFileA(selfname, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, 0, NULL);
343 hashsize *= 2;
344 hash = HeapAlloc(GetProcessHeap(), 0, hashsize);
345 SetLastError(0xdeadbeef);
346 ret = pCryptCATAdminCalcHashFromFileHandle(file, &hashsize, hash, 0);
347 ok(ret, "Expected success %u\n", GetLastError());
348 ok(hashsize == 20," Expected a hash size of 20, got %d\n", hashsize);
349 ok(GetLastError() == ERROR_SUCCESS,
350 "Expected ERROR_SUCCESS, got %d\n", GetLastError());
351 CloseHandle(file);
352 HeapFree(GetProcessHeap(), 0, hash);
353
354 /* Do the same test with a file created and filled by ourselves (and we thus
355 * have a known hash for).
356 */
357 GetTempFileNameA(CURR_DIR, "hsh", 0, temp);
358 file = CreateFileA(temp, GENERIC_WRITE, 0, NULL, OPEN_EXISTING, 0, NULL);
359 WriteFile(file, "Text in this file is needed to create a know hash", 49, &written, NULL);
360 CloseHandle(file);
361
362 /* All OK, first request the size and then retrieve the hash */
363 file = CreateFileA(temp, GENERIC_READ, 0, NULL, OPEN_EXISTING, 0, NULL);
364 hashsize = 0;
365 pCryptCATAdminCalcHashFromFileHandle(file, &hashsize, NULL, 0);
366 hash = HeapAlloc(GetProcessHeap(), 0, hashsize);
367 SetLastError(0xdeadbeef);
368 ret = pCryptCATAdminCalcHashFromFileHandle(file, &hashsize, hash, 0);
369 ok(ret, "Expected success\n");
370 ok(GetLastError() == ERROR_SUCCESS,
371 "Expected ERROR_SUCCESS, got %d\n", GetLastError());
372 ok(hashsize == sizeof(expectedhash) &&
373 !memcmp(hash, expectedhash, sizeof(expectedhash)),
374 "Hashes didn't match\n");
375 CloseHandle(file);
376
377 HeapFree(GetProcessHeap(), 0, hash);
378 DeleteFileA(temp);
379 }
380
381 static void test_CryptCATOpen(void)
382 {
383 HANDLE hcat;
384 char empty[MAX_PATH];
385 WCHAR emptyW[MAX_PATH];
386 HANDLE file;
387 BOOL ret;
388
389 SetLastError(0xdeadbeef);
390 hcat = pCryptCATOpen(NULL, 0, 0, 0, 0);
391 ok(hcat == INVALID_HANDLE_VALUE, "CryptCATOpen succeeded\n");
392 ok(GetLastError() == ERROR_INVALID_PARAMETER,
393 "Expected ERROR_INVALID_PARAMETER, got %08x\n", GetLastError());
394
395 if (!GetTempFileNameA(CURR_DIR, "cat", 0, empty)) return;
396
397 file = CreateFileA(empty, GENERIC_WRITE, 0, NULL, OPEN_EXISTING, 0, NULL);
398 ok(file != INVALID_HANDLE_VALUE, "CreateFileA failed %u\n", GetLastError());
399 CloseHandle(file);
400 MultiByteToWideChar(CP_ACP, 0, empty, -1, emptyW, MAX_PATH);
401
402 hcat = pCryptCATOpen(emptyW, 0, 0, 0, 0);
403 todo_wine
404 ok(hcat != INVALID_HANDLE_VALUE, "Expected a correct handle\n");
405
406 ret = pCryptCATClose(hcat);
407 todo_wine
408 ok(ret, "CryptCATClose failed\n");
409 DeleteFileA(empty);
410 }
411
412 static DWORD error_area;
413 static DWORD local_error;
414
415 static void WINAPI cdf_callback(DWORD area, DWORD error, WCHAR* line)
416 {
417 ok(error_area != -2, "Didn't expect cdf_callback() to be called (%08x, %08x)\n",
418 area, error);
419
420 error_area = area;
421 local_error = error;
422 }
423
424 static void test_CryptCATCDF_params(void)
425 {
426 static WCHAR nonexistent[] = {'d','e','a','d','b','e','e','f','.','c','d','f',0};
427 CRYPTCATCDF *catcdf;
428 BOOL ret;
429
430 if (!pCryptCATCDFOpen)
431 {
432 win_skip("CryptCATCDFOpen is not available\n");
433 return;
434 }
435
436 SetLastError(0xdeadbeef);
437 catcdf = pCryptCATCDFOpen(NULL, NULL);
438 ok(catcdf == NULL, "CryptCATCDFOpen succeeded\n");
439 todo_wine
440 ok(GetLastError() == ERROR_INVALID_PARAMETER, "Expected ERROR_INVALID_PARAMETER, got %d\n", GetLastError());
441
442 SetLastError(0xdeadbeef);
443 catcdf = pCryptCATCDFOpen(NULL, cdf_callback);
444 ok(catcdf == NULL, "CryptCATCDFOpen succeeded\n");
445 todo_wine
446 ok(GetLastError() == ERROR_INVALID_PARAMETER, "Expected ERROR_INVALID_PARAMETER, got %d\n", GetLastError());
447
448 /* File doesn't exist */
449 SetLastError(0xdeadbeef);
450 catcdf = pCryptCATCDFOpen(nonexistent, cdf_callback);
451 ok(catcdf == NULL, "CryptCATCDFOpen succeeded\n");
452 todo_wine
453 ok(GetLastError() == ERROR_FILE_NOT_FOUND, "Expected ERROR_FILE_NOT_FOUND, got %d\n", GetLastError());
454
455 SetLastError(0xdeadbeef);
456 ret = pCryptCATCDFClose(NULL);
457 ok(!ret, "Expected failure\n");
458 todo_wine
459 ok(GetLastError() == ERROR_INVALID_PARAMETER, "Expected ERROR_INVALID_PARAMETER, got %d\n", GetLastError());
460
461 catcdf = NULL;
462 SetLastError(0xdeadbeef);
463 ret = pCryptCATCDFClose(catcdf);
464 ok(!ret, "Expected failure\n");
465 todo_wine
466 ok(GetLastError() == ERROR_INVALID_PARAMETER, "Expected ERROR_INVALID_PARAMETER, got %d\n", GetLastError());
467 }
468
469 /* FIXME: Once Wine can create catalog files we should use the created catalog file in this test */
470 static void test_CryptCATAdminAddRemoveCatalog(void)
471 {
472 static WCHAR basenameW[] = {'w','i','n','e','t','e','s','t','.','c','a','t',0};
473 static CHAR basename[] = "winetest.cat";
474 HCATADMIN hcatadmin;
475 HCATINFO hcatinfo;
476 CATALOG_INFO info;
477 WCHAR tmpfileW[MAX_PATH];
478 char tmpfile[MAX_PATH];
479 char catfile[MAX_PATH], catfilepath[MAX_PATH], *p;
480 WCHAR catfileW[MAX_PATH];
481 HANDLE file;
482 DWORD error, written;
483 BOOL ret;
484 DWORD attrs;
485
486 if (!pCryptCATAdminRemoveCatalog)
487 {
488 /* NT4 and W2K do have CryptCATAdminAddCatalog !! */
489 win_skip("CryptCATAdminRemoveCatalog is not available\n");
490 return;
491 }
492
493 if (!GetTempFileNameA(CURR_DIR, "cat", 0, tmpfile)) return;
494 DeleteFileA(tmpfile);
495 file = CreateFileA(tmpfile, GENERIC_WRITE, 0, NULL, CREATE_NEW, 0, NULL);
496 ok(file != INVALID_HANDLE_VALUE, "CreateFileA failed %u\n", GetLastError());
497 CloseHandle(file);
498
499 ret = pCryptCATAdminAcquireContext(&hcatadmin, &dummy, 0);
500 ok(ret || GetLastError() == ERROR_ACCESS_DENIED, "CryptCATAdminAcquireContext failed %u\n", GetLastError());
501 if (!ret && GetLastError() == ERROR_ACCESS_DENIED)
502 {
503 win_skip("Not running as administrator\n");
504 return;
505 }
506
507 SetLastError(0xdeadbeef);
508 hcatinfo = pCryptCATAdminAddCatalog(NULL, NULL, NULL, 0);
509 error = GetLastError();
510 ok(hcatinfo == NULL, "CryptCATAdminAddCatalog succeeded\n");
511 ok(error == ERROR_INVALID_PARAMETER, "got %u expected ERROR_INVALID_PARAMETER\n", GetLastError());
512
513 SetLastError(0xdeadbeef);
514 hcatinfo = pCryptCATAdminAddCatalog(hcatadmin, NULL, NULL, 0);
515 error = GetLastError();
516 ok(hcatinfo == NULL, "CryptCATAdminAddCatalog succeeded\n");
517 ok(error == ERROR_INVALID_PARAMETER, "got %u expected ERROR_INVALID_PARAMETER\n", GetLastError());
518
519 MultiByteToWideChar(0, 0, tmpfile, -1, tmpfileW, MAX_PATH);
520
521 SetLastError(0xdeadbeef);
522 hcatinfo = pCryptCATAdminAddCatalog(hcatadmin, tmpfileW, basenameW, 0);
523 error = GetLastError();
524 todo_wine {
525 ok(hcatinfo == NULL, "CryptCATAdminAddCatalog succeeded\n");
526 ok(error == ERROR_BAD_FORMAT, "got %u expected ERROR_BAD_FORMAT\n", GetLastError());
527 }
528 if (hcatinfo != NULL)
529 pCryptCATAdminReleaseCatalogContext(hcatadmin, hcatinfo, 0);
530
531 SetLastError(0xdeadbeef);
532 hcatinfo = pCryptCATAdminAddCatalog(hcatadmin, tmpfileW, basenameW, 1);
533 error = GetLastError();
534 ok(hcatinfo == NULL, "CryptCATAdminAddCatalog succeeded\n");
535 ok(error == ERROR_INVALID_PARAMETER, "got %u expected ERROR_INVALID_PARAMETER\n", GetLastError());
536
537 SetLastError(0xdeadbeef);
538 hcatinfo = pCryptCATAdminAddCatalog(hcatadmin, tmpfileW, NULL, 0);
539 error = GetLastError();
540 ok(hcatinfo == NULL, "CryptCATAdminAddCatalog succeeded\n");
541 todo_wine ok(error == ERROR_BAD_FORMAT, "got %u expected ERROR_BAD_FORMAT\n", GetLastError());
542
543 DeleteFileA(tmpfile);
544 file = CreateFileA(tmpfile, GENERIC_WRITE, 0, NULL, CREATE_NEW, 0, NULL);
545 ok(file != INVALID_HANDLE_VALUE, "CreateFileA failed %u\n", GetLastError());
546 WriteFile(file, test_catalog, sizeof(test_catalog), &written, NULL);
547 CloseHandle(file);
548
549 /* Unique name will be created */
550 hcatinfo = pCryptCATAdminAddCatalog(hcatadmin, tmpfileW, NULL, 0);
551 todo_wine ok(hcatinfo != NULL, "CryptCATAdminAddCatalog failed %u\n", GetLastError());
552
553 info.cbStruct = sizeof(info);
554 info.wszCatalogFile[0] = 0;
555 ret = pCryptCATCatalogInfoFromContext(hcatinfo, &info, 0);
556 todo_wine
557 {
558 ok(ret, "CryptCATCatalogInfoFromContext failed %u\n", GetLastError());
559 ok(info.wszCatalogFile[0] != 0, "Expected a filename\n");
560 }
561 WideCharToMultiByte(CP_ACP, 0, info.wszCatalogFile, -1, catfile, MAX_PATH, 0, 0);
562 if ((p = strrchr(catfile, '\\'))) p++;
563 memset(catfileW, 0, sizeof(catfileW));
564 MultiByteToWideChar(0, 0, p, -1, catfileW, MAX_PATH);
565
566 /* Set the file attributes so we can check what happens with them during the 'copy' */
567 attrs = FILE_ATTRIBUTE_READONLY;
568 ret = SetFileAttributesA(tmpfile, attrs);
569 ok(ret, "SetFileAttributesA failed : %u\n", GetLastError());
570
571 /* winetest.cat will be created */
572 hcatinfo = pCryptCATAdminAddCatalog(hcatadmin, tmpfileW, basenameW, 0);
573 ok(hcatinfo != NULL, "CryptCATAdminAddCatalog failed %u\n", GetLastError());
574
575 lstrcpyA(catfilepath, catroot);
576 lstrcatA(catfilepath, "\\{DEADBEEF-DEAD-BEEF-DEAD-BEEFDEADBEEF}\\winetest.cat");
577 attrs = GetFileAttributes(catfilepath);
578 ok(attrs != INVALID_FILE_ATTRIBUTES, "Expected %s to exist\n", catfilepath);
579 todo_wine
580 ok(attrs == FILE_ATTRIBUTE_SYSTEM ||
581 attrs == (FILE_ATTRIBUTE_NOT_CONTENT_INDEXED | FILE_ATTRIBUTE_SYSTEM), /* Vista */
582 "File has wrong attributes : %08x\n", attrs);
583
584 info.cbStruct = sizeof(info);
585 info.wszCatalogFile[0] = 0;
586 ret = pCryptCATCatalogInfoFromContext(hcatinfo, &info, 0);
587 ok(ret, "CryptCATCatalogInfoFromContext failed %u\n", GetLastError());
588 ok(info.wszCatalogFile[0] != 0, "Expected a filename\n");
589 WideCharToMultiByte(CP_ACP, 0, info.wszCatalogFile, -1, catfile, MAX_PATH, 0, 0);
590 if ((p = strrchr(catfile, '\\'))) p++;
591 ok(!lstrcmpA(basename, p), "Expected %s, got %s\n", basename, p);
592
593 ret = pCryptCATAdminReleaseCatalogContext(hcatadmin, hcatinfo, 0);
594 ok(ret, "CryptCATAdminReleaseCatalogContext failed %u\n", GetLastError());
595
596 /* Remove the catalog file with the unique name */
597 ret = pCryptCATAdminRemoveCatalog(hcatadmin, catfileW, 0);
598 ok(ret, "CryptCATAdminRemoveCatalog failed %u\n", GetLastError());
599
600 /* Remove the winetest.cat catalog file, first with the full path. This should not succeed
601 * according to MSDN */
602 ret = pCryptCATAdminRemoveCatalog(hcatadmin, info.wszCatalogFile, 0);
603 ok(ret, "CryptCATAdminRemoveCatalog failed %u\n", GetLastError());
604 /* The call succeeded with the full path but the file is not removed */
605 attrs = GetFileAttributes(catfilepath);
606 ok(attrs != INVALID_FILE_ATTRIBUTES, "Expected %s to exist\n", catfilepath);
607 /* Given only the filename the file is removed */
608 ret = pCryptCATAdminRemoveCatalog(hcatadmin, basenameW, 0);
609 ok(ret, "CryptCATAdminRemoveCatalog failed %u\n", GetLastError());
610 attrs = GetFileAttributes(catfilepath);
611 ok(attrs == INVALID_FILE_ATTRIBUTES, "Expected %s to be removed\n", catfilepath);
612
613 ret = pCryptCATAdminReleaseContext(hcatadmin, 0);
614 ok(ret, "CryptCATAdminReleaseContext failed %u\n", GetLastError());
615
616 /* Set the attributes so we can delete the file */
617 attrs = FILE_ATTRIBUTE_NORMAL;
618 ret = SetFileAttributesA(tmpfile, attrs);
619 DeleteFileA(tmpfile);
620 }
621
622 static void test_catalog_properties(CHAR *catfile, int attributes, int members)
623 {
624 static const GUID subject = {0xde351a42,0x8e59,0x11d0,{0x8c,0x47,0x00,0xc0,0x4f,0xc2,0x95,0xee}};
625
626 HANDLE hcat;
627 CRYPTCATMEMBER *m;
628 CRYPTCATATTRIBUTE *attr;
629 char catalog[MAX_PATH];
630 WCHAR catalogW[MAX_PATH];
631 DWORD attrs;
632 BOOL ret;
633 int attrcount = 0, membercount = 0;
634
635 /* FIXME: Wine can't create catalog files out of catalog definition files yet. Remove this piece
636 * once wine is fixed
637 */
638 attrs = GetFileAttributesA(catfile);
639 if (attrs == INVALID_FILE_ATTRIBUTES)
640 {
641 HANDLE file;
642 DWORD written;
643
644 trace("Creating the catalog file\n");
645 if (!GetTempFileNameA(CURR_DIR, "cat", 0, catalog)) return;
646 file = CreateFileA(catalog, GENERIC_WRITE, 0, NULL, OPEN_EXISTING, 0, NULL);
647 ok(file != INVALID_HANDLE_VALUE, "CreateFileA failed %u\n", GetLastError());
648 WriteFile(file, test_catalog, sizeof(test_catalog), &written, NULL);
649 CloseHandle(file);
650
651 attributes = 2;
652 members = 1;
653 MultiByteToWideChar(CP_ACP, 0, catalog, -1, catalogW, MAX_PATH);
654 }
655 else
656 {
657 MultiByteToWideChar(CP_ACP, 0, catfile, -1, catalogW, MAX_PATH);
658 catalog[0] = 0;
659 }
660
661 hcat = pCryptCATOpen(catalogW, 0, 0, 0, 0);
662 if (hcat == INVALID_HANDLE_VALUE && members == 0)
663 {
664 win_skip("CryptCATOpen on W2K can't handle catalog files with no members\n");
665 return;
666 }
667 ok(hcat != INVALID_HANDLE_VALUE, "CryptCATOpen failed %u\n", GetLastError());
668
669 m = pCryptCATEnumerateMember(NULL, NULL);
670 ok(m == NULL, "CryptCATEnumerateMember succeeded\n");
671
672 m = NULL;
673 while ((m = pCryptCATEnumerateMember(hcat, m)))
674 {
675 ok(m->cbStruct == sizeof(CRYPTCATMEMBER), "unexpected size %u\n", m->cbStruct);
676 todo_wine ok(!lstrcmpW(m->pwszReferenceTag, hashmeW), "unexpected tag\n");
677 ok(!memcmp(&m->gSubjectType, &subject, sizeof(subject)), "guid differs\n");
678 ok(!m->fdwMemberFlags, "got %x expected 0\n", m->fdwMemberFlags);
679 ok(m->dwCertVersion == 0x200, "got %x expected 0x200\n", m->dwCertVersion);
680 ok(!m->dwReserved, "got %x expected 0\n", m->dwReserved);
681 ok(m->hReserved == NULL, "got %p expected NULL\n", m->hReserved);
682
683 attr = pCryptCATEnumerateAttr(hcat, m, NULL);
684 ok(attr == NULL, "CryptCATEnumerateAttr succeeded\n");
685
686 membercount++;
687 }
688 ok(membercount == members, "Expected %d members, got %d\n", members, membercount);
689
690 attr = pCryptCATEnumerateAttr(NULL, NULL, NULL);
691 ok(attr == NULL, "CryptCATEnumerateAttr succeeded\n");
692
693 attr = pCryptCATEnumerateAttr(hcat, NULL, NULL);
694 ok(attr == NULL, "CryptCATEnumerateAttr succeeded\n");
695
696 attr = NULL;
697 while ((attr = pCryptCATEnumerateCatAttr(hcat, attr)))
698 {
699 ok(!lstrcmpW(attr->pwszReferenceTag, attr1W) ||
700 !lstrcmpW(attr->pwszReferenceTag, attr2W),
701 "Expected 'attr1' or 'attr2'\n");
702
703 attrcount++;
704 }
705 todo_wine
706 ok(attrcount == attributes, "Expected %d catalog attributes, got %d\n", attributes, attrcount);
707
708 ret = pCryptCATClose(hcat);
709 ok(ret, "CryptCATClose failed\n");
710 if (catalog[0]) DeleteFileA( catalog );
711 }
712
713 static void test_create_catalog_file(void)
714 {
715 static CHAR catfileA[] = "winetest.cat";
716 static CHAR cdffileA[] = "winetest.cdf";
717 static WCHAR cdffileW[] = {'w','i','n','e','t','e','s','t','.','c','d','f',0};
718 CRYPTCATCDF *catcdf;
719 CRYPTCATATTRIBUTE *catattr;
720 CRYPTCATMEMBER *catmember;
721 WCHAR *catmembertag;
722 DWORD written, attrs;
723 HANDLE file;
724 BOOL ret;
725 int attrcount, membercount;
726
727 if (!pCryptCATCDFOpen)
728 {
729 win_skip("CryptCATCDFOpen is not available\n");
730 return;
731 }
732
733 /* Create the cdf file */
734 file = CreateFileA(cdffileA, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
735 ok(file != INVALID_HANDLE_VALUE, "CreateFileA failed %u\n", GetLastError());
736 WriteFile(file, test_cdf, sizeof(test_cdf) - 1, &written, NULL);
737 CloseHandle(file);
738
739 /* Don't enumerate attributes and members */
740 trace("No attribs and members\n");
741 SetLastError(0xdeadbeef);
742 catcdf = pCryptCATCDFOpen(cdffileW, NULL);
743 todo_wine
744 {
745 ok(catcdf != NULL, "CryptCATCDFOpen failed\n");
746 ok(GetLastError() == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %d\n", GetLastError());
747 }
748
749 ret = pCryptCATCDFClose(catcdf);
750 todo_wine
751 {
752 ok(ret, "Expected success\n");
753 ok(GetLastError() == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %d\n", GetLastError());
754 }
755
756 attrs = GetFileAttributesA(catfileA);
757 todo_wine
758 ok(attrs != INVALID_FILE_ATTRIBUTES, "Expected the catalog file to exist\n");
759
760 test_catalog_properties(catfileA, 0, 0);
761 DeleteFileA(catfileA);
762
763 /* Only enumerate the attributes */
764 trace("Only attributes\n");
765 attrcount = membercount = 0;
766 catcdf = pCryptCATCDFOpen(cdffileW, NULL);
767
768 catattr = NULL;
769 while ((catattr = pCryptCATCDFEnumCatAttributes(catcdf, catattr, NULL)))
770 {
771 ok(!lstrcmpW(catattr->pwszReferenceTag, attr1W) ||
772 !lstrcmpW(catattr->pwszReferenceTag, attr2W),
773 "Expected 'attr1' or 'attr2'\n");
774
775 attrcount++;
776 }
777 todo_wine
778 ok(attrcount == 2, "Expected 2 attributes, got %d\n", attrcount);
779
780 pCryptCATCDFClose(catcdf);
781 /* Even though the resulting catalog file shows the attributes, they will not be enumerated */
782 test_catalog_properties(catfileA, 0, 0);
783 DeleteFileA(catfileA);
784
785 /* Only enumerate the members */
786 trace("Only members\n");
787 attrcount = membercount = 0;
788 catcdf = pCryptCATCDFOpen(cdffileW, NULL);
789
790 catmember = NULL;
791 catmembertag = NULL;
792 while ((catmembertag = pCryptCATCDFEnumMembersByCDFTagEx(catcdf, catmembertag, NULL, &catmember, FALSE, NULL)))
793 {
794 ok(!lstrcmpW(catmembertag, hashmeW), "Expected 'hashme'\n");
795 membercount++;
796 }
797 todo_wine
798 ok(membercount == 1, "Expected 1 member, got %d\n", membercount);
799
800 pCryptCATCDFClose(catcdf);
801 test_catalog_properties(catfileA, 0, 1);
802 DeleteFileA(catfileA);
803
804 /* Enumerate members and attributes */
805 trace("Attributes and members\n");
806 attrcount = membercount = 0;
807 catcdf = pCryptCATCDFOpen(cdffileW, NULL);
808
809 catattr = NULL;
810 while ((catattr = pCryptCATCDFEnumCatAttributes(catcdf, catattr, NULL)))
811 attrcount++;
812 todo_wine
813 ok(attrcount == 2, "Expected 2 attributes, got %d\n", attrcount);
814
815 catmember = NULL;
816 catmembertag = NULL;
817 while ((catmembertag = pCryptCATCDFEnumMembersByCDFTagEx(catcdf, catmembertag, NULL, &catmember, FALSE, NULL)))
818 membercount++;
819 todo_wine
820 ok(membercount == 1, "Expected 1 member, got %d\n", membercount);
821
822 pCryptCATCDFClose(catcdf);
823 test_catalog_properties(catfileA, 2, 1);
824 DeleteFileA(catfileA);
825
826 DeleteFileA(cdffileA);
827 }
828
829 static void create_cdf_file(const CHAR *filename, const CHAR *contents)
830 {
831 HANDLE file;
832 DWORD written;
833
834 file = CreateFileA(filename, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
835 ok(file != INVALID_HANDLE_VALUE, "CreateFileA failed %u\n", GetLastError());
836 WriteFile(file, contents, lstrlenA(contents), &written, NULL);
837 CloseHandle(file);
838 }
839
840 #define CHECK_EXPECT(a, b) \
841 do { \
842 ok(a == error_area, "Expected %08x, got %08x\n", a, error_area); \
843 ok(b == local_error, "Expected %08x, got %08x\n", b, local_error); \
844 } while (0)
845
846 /* Clear the variables (can't use 0) */
847 #define CLEAR_EXPECT \
848 error_area = local_error = -1
849
850 /* Set both variables so the callback routine can check if a call to it was unexpected */
851 #define SET_UNEXPECTED \
852 error_area = local_error = -2
853
854 static void test_cdf_parsing(void)
855 {
856 static CHAR catfileA[] = "tempfile.cat";
857 static CHAR cdffileA[] = "tempfile.cdf";
858 static WCHAR cdffileW[] = {'t','e','m','p','f','i','l','e','.','c','d','f',0};
859 CHAR cdf_contents[4096];
860 CRYPTCATCDF *catcdf;
861 CRYPTCATATTRIBUTE *catattr;
862 CRYPTCATMEMBER *catmember;
863 WCHAR *catmembertag;
864
865 if (!pCryptCATCDFOpen)
866 {
867 win_skip("CryptCATCDFOpen is not available\n");
868 return;
869 }
870
871 /* Empty file */
872 DeleteFileA(cdffileA);
873 create_cdf_file(cdffileA, "");
874
875 CLEAR_EXPECT;
876 SetLastError(0xdeadbeef);
877 catcdf = pCryptCATCDFOpen(cdffileW, cdf_callback);
878 todo_wine
879 CHECK_EXPECT(CRYPTCAT_E_AREA_HEADER, CRYPTCAT_E_CDF_TAGNOTFOUND);
880 ok(catcdf == NULL, "CryptCATCDFOpen succeeded\n");
881 todo_wine
882 ok(GetLastError() == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %d\n", GetLastError());
883 DeleteFileA(cdffileA);
884 ok(!DeleteFileA(catfileA), "Didn't expect a catalog file to be created\n");
885
886 /* Just the header */
887 lstrcpyA(cdf_contents, "[CatalogHeader]\r\n");
888 create_cdf_file(cdffileA, cdf_contents);
889
890 SET_UNEXPECTED;
891 SetLastError(0xdeadbeef);
892 catcdf = pCryptCATCDFOpen(cdffileW, cdf_callback);
893 ok(catcdf == NULL, "CryptCATCDFOpen succeeded\n");
894 todo_wine
895 ok(GetLastError() == ERROR_SHARING_VIOLATION ||
896 broken(GetLastError() == ERROR_SUCCESS), /* win9x */
897 "Expected ERROR_SHARING_VIOLATION, got %d\n", GetLastError());
898 DeleteFileA(cdffileA);
899
900 /* Header and member only */
901 lstrcpyA(cdf_contents, "[CatalogHeader]\r\n");
902 lstrcatA(cdf_contents, "[CatalogFiles]\r\n");
903 lstrcatA(cdf_contents, "hashme=.\\tempfile.cdf\r\n");
904 create_cdf_file(cdffileA, cdf_contents);
905
906 SET_UNEXPECTED;
907 SetLastError(0xdeadbeef);
908 catcdf = pCryptCATCDFOpen(cdffileW, cdf_callback);
909 ok(catcdf == NULL, "CryptCATCDFOpen succeeded\n");
910 todo_wine
911 ok(GetLastError() == ERROR_SHARING_VIOLATION ||
912 broken(GetLastError() == ERROR_SUCCESS), /* win9x */
913 "Expected ERROR_SHARING_VIOLATION, got %d\n", GetLastError());
914 DeleteFileA(cdffileA);
915 ok(!DeleteFileA(catfileA), "Didn't expect a catalog file to be created\n");
916
917 /* Header and Name (no value) */
918 lstrcpyA(cdf_contents, "[CatalogHeader]\r\n");
919 lstrcatA(cdf_contents, "Name=\r\n");
920 create_cdf_file(cdffileA, cdf_contents);
921
922 SET_UNEXPECTED;
923 SetLastError(0xdeadbeef);
924 catcdf = pCryptCATCDFOpen(cdffileW, cdf_callback);
925 ok(catcdf == NULL, "CryptCATCDFOpen succeeded\n");
926 todo_wine
927 ok(GetLastError() == ERROR_SHARING_VIOLATION ||
928 broken(GetLastError() == ERROR_SUCCESS), /* win9x */
929 "Expected ERROR_SHARING_VIOLATION, got %d\n", GetLastError());
930 DeleteFileA(cdffileA);
931 ok(!DeleteFileA(catfileA), "Didn't expect a catalog file to be created\n");
932
933 /* Header and Name */
934 lstrcpyA(cdf_contents, "[CatalogHeader]\r\n");
935 lstrcatA(cdf_contents, "Name=tempfile.cat\r\n");
936 create_cdf_file(cdffileA, cdf_contents);
937
938 SET_UNEXPECTED;
939 SetLastError(0xdeadbeef);
940 catcdf = pCryptCATCDFOpen(cdffileW, cdf_callback);
941 todo_wine
942 {
943 ok(catcdf != NULL, "CryptCATCDFOpen failed\n");
944 ok(GetLastError() == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %d\n", GetLastError());
945 }
946 pCryptCATCDFClose(catcdf);
947 DeleteFileA(cdffileA);
948 todo_wine
949 ok(DeleteFileA(catfileA), "Expected a catalog file to be created\n");
950
951 /* Header and nonexistent member */
952 lstrcpyA(cdf_contents, "[CatalogHeader]\r\n");
953 lstrcatA(cdf_contents, "Name=tempfile.cat\r\n");
954 lstrcatA(cdf_contents, "[CatalogFiles]\r\n");
955 lstrcatA(cdf_contents, "hashme=.\\deadbeef.cdf\r\n");
956 create_cdf_file(cdffileA, cdf_contents);
957
958 SET_UNEXPECTED;
959 SetLastError(0xdeadbeef);
960 catcdf = pCryptCATCDFOpen(cdffileW, cdf_callback);
961 todo_wine
962 {
963 ok(catcdf != NULL, "CryptCATCDFOpen failed\n");
964 ok(GetLastError() == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %d\n", GetLastError());
965 }
966 /* Loop through the members */
967 CLEAR_EXPECT;
968 catmember = NULL;
969 catmembertag = NULL;
970 while ((catmembertag = pCryptCATCDFEnumMembersByCDFTagEx(catcdf, catmembertag, cdf_callback, &catmember, FALSE, NULL))) ;
971 todo_wine
972 CHECK_EXPECT(CRYPTCAT_E_AREA_MEMBER, CRYPTCAT_E_CDF_MEMBER_FILENOTFOUND);
973 pCryptCATCDFClose(catcdf);
974 DeleteFileA(cdffileA);
975 todo_wine
976 ok(DeleteFileA(catfileA), "Expected a catalog file to be created\n");
977
978 /* Header, correct member but no explicit newline */
979 lstrcpyA(cdf_contents, "[CatalogHeader]\r\n");
980 lstrcatA(cdf_contents, "Name=tempfile.cat\r\n");
981 lstrcatA(cdf_contents, "[CatalogFiles]\r\n");
982 lstrcatA(cdf_contents, "hashme=.\\tempfile.cdf\r");
983 create_cdf_file(cdffileA, cdf_contents);
984
985 SET_UNEXPECTED;
986 SetLastError(0xdeadbeef);
987 catcdf = pCryptCATCDFOpen(cdffileW, cdf_callback);
988 todo_wine
989 {
990 ok(catcdf != NULL, "CryptCATCDFOpen failed\n");
991 ok(GetLastError() == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %d\n", GetLastError());
992 }
993 /* Loop through the members */
994 CLEAR_EXPECT;
995 catmember = NULL;
996 catmembertag = NULL;
997 while ((catmembertag = pCryptCATCDFEnumMembersByCDFTagEx(catcdf, catmembertag, cdf_callback, &catmember, FALSE, NULL))) ;
998 todo_wine
999 CHECK_EXPECT(CRYPTCAT_E_AREA_MEMBER, CRYPTCAT_E_CDF_MEMBER_FILE_PATH);
1000 pCryptCATCDFClose(catcdf);
1001 DeleteFileA(cdffileA);
1002 todo_wine
1003 ok(DeleteFileA(catfileA), "Expected a catalog file to be created\n");
1004
1005 /* Header and 2 duplicate members */
1006 lstrcpyA(cdf_contents, "[CatalogHeader]\r\n");
1007 lstrcatA(cdf_contents, "Name=tempfile.cat\r\n");
1008 lstrcatA(cdf_contents, "[CatalogFiles]\r\n");
1009 lstrcatA(cdf_contents, "hashme=.\\tempfile.cdf\r\n");
1010 lstrcatA(cdf_contents, "hashme=.\\tempfile.cdf\r\n");
1011 create_cdf_file(cdffileA, cdf_contents);
1012
1013 SET_UNEXPECTED;
1014 SetLastError(0xdeadbeef);
1015 catcdf = pCryptCATCDFOpen(cdffileW, cdf_callback);
1016 todo_wine
1017 {
1018 ok(catcdf != NULL, "CryptCATCDFOpen failed\n");
1019 ok(GetLastError() == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %d\n", GetLastError());
1020 }
1021 /* Loop through the members */
1022 SET_UNEXPECTED;
1023 catmember = NULL;
1024 catmembertag = NULL;
1025 while ((catmembertag = pCryptCATCDFEnumMembersByCDFTagEx(catcdf, catmembertag, cdf_callback, &catmember, FALSE, NULL))) ;
1026 pCryptCATCDFClose(catcdf);
1027 test_catalog_properties(catfileA, 0, 1);
1028 DeleteFileA(cdffileA);
1029 todo_wine
1030 ok(DeleteFileA(catfileA), "Expected a catalog file to be created\n");
1031
1032 /* Wrong attribute */
1033 lstrcpyA(cdf_contents, "[CatalogHeader]\r\n");
1034 lstrcatA(cdf_contents, "Name=tempfile.cat\r\n");
1035 lstrcatA(cdf_contents, "CATATTR1=0x10010001:attr1\r\n");
1036 create_cdf_file(cdffileA, cdf_contents);
1037
1038 SET_UNEXPECTED;
1039 SetLastError(0xdeadbeef);
1040 catcdf = pCryptCATCDFOpen(cdffileW, cdf_callback);
1041 todo_wine
1042 {
1043 ok(catcdf != NULL, "CryptCATCDFOpen failed\n");
1044 ok(GetLastError() == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %d\n", GetLastError());
1045 }
1046 /* Loop through the attributes */
1047 CLEAR_EXPECT;
1048 catattr = NULL;
1049 while ((catattr = pCryptCATCDFEnumCatAttributes(catcdf, catattr, cdf_callback))) ;
1050 todo_wine
1051 CHECK_EXPECT(CRYPTCAT_E_AREA_ATTRIBUTE, CRYPTCAT_E_CDF_ATTR_TOOFEWVALUES);
1052 pCryptCATCDFClose(catcdf);
1053 DeleteFileA(cdffileA);
1054 todo_wine
1055 ok(DeleteFileA(catfileA), "Expected a catalog file to be created\n");
1056
1057 /* Two identical attributes */
1058 lstrcpyA(cdf_contents, "[CatalogHeader]\r\n");
1059 lstrcatA(cdf_contents, "Name=tempfile.cat\r\n");
1060 lstrcatA(cdf_contents, "CATATTR1=0x10010001:attr1:value1\r\n");
1061 lstrcatA(cdf_contents, "CATATTR1=0x10010001:attr1:value1\r\n");
1062 lstrcatA(cdf_contents, "[CatalogFiles]\r\n");
1063 lstrcatA(cdf_contents, "hashme=.\\tempfile.cdf\r\n");
1064 create_cdf_file(cdffileA, cdf_contents);
1065
1066 SET_UNEXPECTED;
1067 SetLastError(0xdeadbeef);
1068 catcdf = pCryptCATCDFOpen(cdffileW, cdf_callback);
1069 todo_wine
1070 {
1071 ok(catcdf != NULL, "CryptCATCDFOpen failed\n");
1072 ok(GetLastError() == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %d\n", GetLastError());
1073 }
1074 /* Loop through the members */
1075 SET_UNEXPECTED;
1076 catmember = NULL;
1077 catmembertag = NULL;
1078 while ((catmembertag = pCryptCATCDFEnumMembersByCDFTagEx(catcdf, catmembertag, cdf_callback, &catmember, FALSE, NULL))) ;
1079 /* Loop through the attributes */
1080 SET_UNEXPECTED;
1081 catattr = NULL;
1082 while ((catattr = pCryptCATCDFEnumCatAttributes(catcdf, catattr, cdf_callback))) ;
1083 pCryptCATCDFClose(catcdf);
1084 test_catalog_properties(catfileA, 1, 1);
1085 DeleteFileA(cdffileA);
1086 todo_wine
1087 ok(DeleteFileA(catfileA), "Expected a catalog file to be created\n");
1088
1089 /* Two different attribute values with the same tag */
1090 lstrcpyA(cdf_contents, "[CatalogHeader]\r\n");
1091 lstrcatA(cdf_contents, "Name=tempfile.cat\r\n");
1092 lstrcatA(cdf_contents, "CATATTR1=0x10010001:attr1:value1\r\n");
1093 lstrcatA(cdf_contents, "CATATTR1=0x10010001:attr2:value2\r\n");
1094 lstrcatA(cdf_contents, "[CatalogFiles]\r\n");
1095 lstrcatA(cdf_contents, "hashme=.\\tempfile.cdf\r\n");
1096 create_cdf_file(cdffileA, cdf_contents);
1097
1098 SET_UNEXPECTED;
1099 SetLastError(0xdeadbeef);
1100 catcdf = pCryptCATCDFOpen(cdffileW, cdf_callback);
1101 todo_wine
1102 {
1103 ok(catcdf != NULL, "CryptCATCDFOpen failed\n");
1104 ok(GetLastError() == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %d\n", GetLastError());
1105 }
1106 /* Loop through the members */
1107 SET_UNEXPECTED;
1108 catmember = NULL;
1109 catmembertag = NULL;
1110 while ((catmembertag = pCryptCATCDFEnumMembersByCDFTagEx(catcdf, catmembertag, cdf_callback, &catmember, FALSE, NULL))) ;
1111 /* Loop through the attributes */
1112 SET_UNEXPECTED;
1113 catattr = NULL;
1114 while ((catattr = pCryptCATCDFEnumCatAttributes(catcdf, catattr, cdf_callback))) ;
1115 pCryptCATCDFClose(catcdf);
1116 test_catalog_properties(catfileA, 1, 1);
1117 DeleteFileA(cdffileA);
1118 todo_wine
1119 ok(DeleteFileA(catfileA), "Expected a catalog file to be created\n");
1120 }
1121
1122 START_TEST(crypt)
1123 {
1124 char** myARGV;
1125 char sysdir[MAX_PATH];
1126
1127 InitFunctionPtrs();
1128
1129 if (!pCryptCATAdminAcquireContext)
1130 {
1131 win_skip("CryptCATAdmin functions are not available\n");
1132 return;
1133 }
1134
1135 GetSystemDirectoryA(sysdir, MAX_PATH);
1136 lstrcpyA(catroot, sysdir);
1137 lstrcatA(catroot, "\\CatRoot");
1138 lstrcpyA(catroot2, sysdir);
1139 lstrcatA(catroot2, "\\CatRoot2");
1140
1141 winetest_get_mainargs(&myARGV);
1142 strcpy(selfname, myARGV[0]);
1143
1144 GetCurrentDirectoryA(MAX_PATH, CURR_DIR);
1145
1146 test_context();
1147 test_calchash();
1148 test_CryptCATOpen();
1149 /* Parameter checking only */
1150 test_CryptCATCDF_params();
1151 /* Test the parsing of a cdf file */
1152 test_cdf_parsing();
1153 /* Create a catalog file out of our own catalog definition file */
1154 test_create_catalog_file();
1155 test_CryptCATAdminAddRemoveCatalog();
1156 }