wintrust/tests: Fix some tests on win9x.
[wine] / dlls / wintrust / tests / 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\n");
210     ok(GetLastError() == 0xdeadbeef,
211        "Expected no change in last error, got %d\n", GetLastError());
212
213     /* Try to release a second time */
214     SetLastError(0xdeadbeef);
215     ret = pCryptCATAdminReleaseContext(hca, 0);
216     ok(!ret, "Expected failure\n");
217     ok(GetLastError() == ERROR_INVALID_PARAMETER,
218        "Expected ERROR_INVALID_PARAMETER, got %d\n", GetLastError());
219
220     /* NULL context handle and dummy GUID */
221     SetLastError(0xdeadbeef);
222     ret = pCryptCATAdminAcquireContext(NULL, &dummy, 0);
223     ok(!ret, "Expected failure\n");
224     ok(GetLastError() == ERROR_INVALID_PARAMETER,
225        "Expected ERROR_INVALID_PARAMETER, got %d\n", GetLastError());
226
227     /* Correct context handle and dummy GUID
228      *
229      * The tests run in the past unfortunately made sure that some directories were created.
230      *
231      * FIXME:
232      * We don't want to mess too much with these for now so we should delete only the ones
233      * that shouldn't be there like the deadbeef ones. We first have to figure out if it's
234      * save to remove files and directories from CatRoot/CatRoot2.
235      */
236
237     ret = pCryptCATAdminAcquireContext(&hca, &dummy, 0);
238     ok(ret, "Expected success\n");
239     ok(hca != NULL, "Expected a context handle, got NULL\n");
240
241     attrs = GetFileAttributes(catroot);
242     ok(attrs != INVALID_FILE_ATTRIBUTES, "Expected the CatRoot directory to exist\n");
243
244     /* Windows creates the GUID directory in capitals */
245     lstrcpyA(dummydir, catroot);
246     lstrcatA(dummydir, "\\{DEADBEEF-DEAD-BEEF-DEAD-BEEFDEADBEEF}");
247     attrs = GetFileAttributes(dummydir);
248     ok(attrs != INVALID_FILE_ATTRIBUTES,
249        "Expected CatRoot\\{DEADBEEF-DEAD-BEEF-DEAD-BEEFDEADBEEF} directory to exist\n");
250
251     /* Only present on XP or higher. */
252     attrs = GetFileAttributes(catroot2);
253     if (attrs != INVALID_FILE_ATTRIBUTES)
254     {
255         lstrcpyA(dummydir, catroot2);
256         lstrcatA(dummydir, "\\{DEADBEEF-DEAD-BEEF-DEAD-BEEFDEADBEEF}");
257         attrs = GetFileAttributes(dummydir);
258         ok(attrs != INVALID_FILE_ATTRIBUTES,
259             "Expected CatRoot2\\{DEADBEEF-DEAD-BEEF-DEAD-BEEFDEADBEEF} directory to exist\n");
260     }
261
262     ret = pCryptCATAdminReleaseContext(hca, 0);
263     ok(ret, "Expected success\n");
264
265     /* Correct context handle and GUID */
266     ret = pCryptCATAdminAcquireContext(&hca, &unknown, 0);
267     ok(ret, "Expected success\n");
268     ok(hca != NULL, "Expected a context handle, got NULL\n");
269
270     ret = pCryptCATAdminReleaseContext(hca, 0);
271     ok(ret, "Expected success\n");
272
273     /* Flags not equal to 0 */
274     ret = pCryptCATAdminAcquireContext(&hca, &unknown, 1);
275     ok(ret, "Expected success\n");
276     ok(hca != NULL, "Expected a context handle, got NULL\n");
277
278     ret = pCryptCATAdminReleaseContext(hca, 0);
279     ok(ret, "Expected success\n");
280 }
281
282 /* TODO: Check whether SHA-1 is the algorithm that's always used */
283 static void test_calchash(void)
284 {
285     BOOL ret;
286     HANDLE file;
287     DWORD hashsize = 0;
288     BYTE* hash;
289     BYTE expectedhash[20] = {0x3a,0xa1,0x19,0x08,0xec,0xa6,0x0d,0x2e,0x7e,0xcc,0x7a,0xca,0xf5,0xb8,0x2e,0x62,0x6a,0xda,0xf0,0x19};
290     CHAR temp[MAX_PATH];
291     DWORD written;
292
293     /* All NULL */
294     SetLastError(0xdeadbeef);
295     ret = pCryptCATAdminCalcHashFromFileHandle(NULL, NULL, NULL, 0);
296     ok(!ret, "Expected failure\n");
297     ok(GetLastError() == ERROR_INVALID_PARAMETER,
298        "Expected ERROR_INVALID_PARAMETER, got %d\n", GetLastError());
299
300     /* NULL filehandle, rest is legal */
301     SetLastError(0xdeadbeef);
302     ret = pCryptCATAdminCalcHashFromFileHandle(NULL, &hashsize, NULL, 0);
303     ok(!ret, "Expected failure\n");
304     ok(GetLastError() == ERROR_INVALID_PARAMETER,
305        "Expected ERROR_INVALID_PARAMETER, got %d\n", GetLastError());
306
307     /* Correct filehandle, rest is NULL */
308     file = CreateFileA(selfname, GENERIC_READ, 0, NULL, OPEN_EXISTING, 0, NULL);
309     SetLastError(0xdeadbeef);
310     ret = pCryptCATAdminCalcHashFromFileHandle(file, NULL, NULL, 0);
311     ok(!ret, "Expected failure\n");
312     ok(GetLastError() == ERROR_INVALID_PARAMETER,
313        "Expected ERROR_INVALID_PARAMETER, got %d\n", GetLastError());
314     CloseHandle(file);
315
316     /* All OK, but dwFlags set to 1 */
317     file = CreateFileA(selfname, GENERIC_READ, 0, NULL, OPEN_EXISTING, 0, NULL);
318     SetLastError(0xdeadbeef);
319     ret = pCryptCATAdminCalcHashFromFileHandle(file, &hashsize, NULL, 1);
320     ok(!ret, "Expected failure\n");
321     ok(GetLastError() == ERROR_INVALID_PARAMETER,
322        "Expected ERROR_INVALID_PARAMETER, got %d\n", GetLastError());
323     CloseHandle(file);
324
325     /* All OK, requesting the size of the hash */
326     file = CreateFileA(selfname, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, 0, NULL);
327     ok(file != INVALID_HANDLE_VALUE, "CreateFile failed %u\n", GetLastError());
328     SetLastError(0xdeadbeef);
329     ret = pCryptCATAdminCalcHashFromFileHandle(file, &hashsize, NULL, 0);
330     ok(ret, "Expected success %u\n", GetLastError());
331     ok(hashsize == 20," Expected a hash size of 20, got %d\n", hashsize);
332     ok(GetLastError() == ERROR_INSUFFICIENT_BUFFER,
333        "Expected ERROR_INSUFFICIENT_BUFFER, got %d\n", GetLastError());
334     CloseHandle(file);
335
336     /* All OK, retrieve the hash
337      * Double the hash buffer to see what happens to the size parameter
338      */
339     file = CreateFileA(selfname, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, 0, NULL);
340     hashsize *= 2;
341     hash = HeapAlloc(GetProcessHeap(), 0, hashsize);
342     SetLastError(0xdeadbeef);
343     ret = pCryptCATAdminCalcHashFromFileHandle(file, &hashsize, hash, 0);
344     ok(ret, "Expected success %u\n", GetLastError());
345     ok(hashsize == 20," Expected a hash size of 20, got %d\n", hashsize);
346     ok(GetLastError() == ERROR_SUCCESS,
347        "Expected ERROR_SUCCESS, got %d\n", GetLastError());
348     CloseHandle(file);
349     HeapFree(GetProcessHeap(), 0, hash);
350
351     /* Do the same test with a file created and filled by ourselves (and we thus
352      * have a known hash for).
353      */
354     GetTempFileNameA(CURR_DIR, "hsh", 0, temp); 
355     file = CreateFileA(temp, GENERIC_WRITE, 0, NULL, OPEN_EXISTING, 0, NULL);
356     WriteFile(file, "Text in this file is needed to create a know hash", 49, &written, NULL);
357     CloseHandle(file);
358
359     /* All OK, first request the size and then retrieve the hash */
360     file = CreateFileA(temp, GENERIC_READ, 0, NULL, OPEN_EXISTING, 0, NULL);
361     hashsize = 0;
362     pCryptCATAdminCalcHashFromFileHandle(file, &hashsize, NULL, 0);
363     hash = HeapAlloc(GetProcessHeap(), 0, hashsize);
364     SetLastError(0xdeadbeef);
365     ret = pCryptCATAdminCalcHashFromFileHandle(file, &hashsize, hash, 0);
366     ok(ret, "Expected success\n");
367     ok(GetLastError() == ERROR_SUCCESS,
368        "Expected ERROR_SUCCESS, got %d\n", GetLastError());
369     ok(hashsize == sizeof(expectedhash) &&
370        !memcmp(hash, expectedhash, sizeof(expectedhash)),
371        "Hashes didn't match\n");
372     CloseHandle(file);
373
374     HeapFree(GetProcessHeap(), 0, hash);
375     DeleteFileA(temp);
376 }
377
378 static DWORD error_area;
379 static DWORD local_error;
380
381 static void WINAPI cdf_callback(DWORD area, DWORD error, WCHAR* line)
382 {
383     ok(error_area != -2, "Didn't expect cdf_callback() to be called (%08x, %08x)\n",
384        area, error);
385
386     error_area = area;
387     local_error = error;
388 }
389
390 static void test_CryptCATCDF_params(void)
391 {
392     static WCHAR nonexistent[] = {'d','e','a','d','b','e','e','f','.','c','d','f',0};
393     CRYPTCATCDF *catcdf;
394     BOOL ret;
395
396     if (!pCryptCATCDFOpen)
397     {
398         win_skip("CryptCATCDFOpen is not available\n");
399         return;
400     }
401
402     SetLastError(0xdeadbeef);
403     catcdf = pCryptCATCDFOpen(NULL, NULL);
404     ok(catcdf == NULL, "CryptCATCDFOpen succeeded\n");
405     todo_wine
406     ok(GetLastError() == ERROR_INVALID_PARAMETER, "Expected ERROR_INVALID_PARAMETER, got %d\n", GetLastError());
407
408     SetLastError(0xdeadbeef);
409     catcdf = pCryptCATCDFOpen(NULL, cdf_callback);
410     ok(catcdf == NULL, "CryptCATCDFOpen succeeded\n");
411     todo_wine
412     ok(GetLastError() == ERROR_INVALID_PARAMETER, "Expected ERROR_INVALID_PARAMETER, got %d\n", GetLastError());
413
414     /* File doesn't exist */
415     SetLastError(0xdeadbeef);
416     catcdf = pCryptCATCDFOpen(nonexistent, cdf_callback);
417     ok(catcdf == NULL, "CryptCATCDFOpen succeeded\n");
418     todo_wine
419     ok(GetLastError() == ERROR_FILE_NOT_FOUND, "Expected ERROR_FILE_NOT_FOUND, got %d\n", GetLastError());
420
421     SetLastError(0xdeadbeef);
422     ret = pCryptCATCDFClose(NULL);
423     ok(!ret, "Expected failure\n");
424     todo_wine
425     ok(GetLastError() == ERROR_INVALID_PARAMETER, "Expected ERROR_INVALID_PARAMETER, got %d\n", GetLastError());
426
427     catcdf = NULL;
428     SetLastError(0xdeadbeef);
429     ret = pCryptCATCDFClose(catcdf);
430     ok(!ret, "Expected failure\n");
431     todo_wine
432     ok(GetLastError() == ERROR_INVALID_PARAMETER, "Expected ERROR_INVALID_PARAMETER, got %d\n", GetLastError());
433 }
434
435 /* FIXME: Once Wine can create catalog files we should use the created catalog file in this test */
436 static void test_CryptCATAdminAddRemoveCatalog(void)
437 {
438     static WCHAR basenameW[] = {'w','i','n','e','t','e','s','t','.','c','a','t',0};
439     static CHAR basename[] = "winetest.cat";
440     HCATADMIN hcatadmin;
441     HCATINFO hcatinfo;
442     CATALOG_INFO info;
443     WCHAR tmpfileW[MAX_PATH];
444     char tmpfile[MAX_PATH];
445     char catfile[MAX_PATH], catfilepath[MAX_PATH], *p;
446     WCHAR catfileW[MAX_PATH];
447     HANDLE file;
448     DWORD error, written;
449     BOOL ret;
450     DWORD attrs;
451
452     if (!pCryptCATAdminRemoveCatalog)
453     {
454         /* NT4 and W2K do have CryptCATAdminAddCatalog !! */
455         win_skip("CryptCATAdminRemoveCatalog is not available\n");
456         return;
457     }
458
459     if (!GetTempFileNameA(CURR_DIR, "cat", 0, tmpfile)) return;
460     DeleteFileA(tmpfile);
461     file = CreateFileA(tmpfile, GENERIC_WRITE, 0, NULL, CREATE_NEW, 0, NULL);
462     ok(file != INVALID_HANDLE_VALUE, "CreateFileA failed %u\n", GetLastError());
463     CloseHandle(file);
464
465     ret = pCryptCATAdminAcquireContext(&hcatadmin, &dummy, 0);
466     ok(ret, "CryptCATAdminAcquireContext failed %u\n", GetLastError());
467
468     SetLastError(0xdeadbeef);
469     hcatinfo = pCryptCATAdminAddCatalog(NULL, NULL, NULL, 0);
470     error = GetLastError();
471     ok(hcatinfo == NULL, "CryptCATAdminAddCatalog succeeded\n");
472     ok(error == ERROR_INVALID_PARAMETER, "got %u expected ERROR_INVALID_PARAMETER\n", GetLastError());
473
474     SetLastError(0xdeadbeef);
475     hcatinfo = pCryptCATAdminAddCatalog(hcatadmin, NULL, NULL, 0);
476     error = GetLastError();
477     ok(hcatinfo == NULL, "CryptCATAdminAddCatalog succeeded\n");
478     ok(error == ERROR_INVALID_PARAMETER, "got %u expected ERROR_INVALID_PARAMETER\n", GetLastError());
479
480     MultiByteToWideChar(0, 0, tmpfile, -1, tmpfileW, MAX_PATH);
481
482     SetLastError(0xdeadbeef);
483     hcatinfo = pCryptCATAdminAddCatalog(hcatadmin, tmpfileW, basenameW, 0);
484     error = GetLastError();
485     todo_wine {
486     ok(hcatinfo == NULL, "CryptCATAdminAddCatalog succeeded\n");
487     ok(error == ERROR_BAD_FORMAT, "got %u expected ERROR_BAD_FORMAT\n", GetLastError());
488     }
489
490     SetLastError(0xdeadbeef);
491     hcatinfo = pCryptCATAdminAddCatalog(hcatadmin, tmpfileW, basenameW, 1);
492     error = GetLastError();
493     ok(hcatinfo == NULL, "CryptCATAdminAddCatalog succeeded\n");
494     ok(error == ERROR_INVALID_PARAMETER, "got %u expected ERROR_INVALID_PARAMETER\n", GetLastError());
495
496     SetLastError(0xdeadbeef);
497     hcatinfo = pCryptCATAdminAddCatalog(hcatadmin, tmpfileW, NULL, 0);
498     error = GetLastError();
499     ok(hcatinfo == NULL, "CryptCATAdminAddCatalog succeeded\n");
500     todo_wine ok(error == ERROR_BAD_FORMAT, "got %u expected ERROR_BAD_FORMAT\n", GetLastError());
501
502     DeleteFileA(tmpfile);
503     file = CreateFileA(tmpfile, GENERIC_WRITE, 0, NULL, CREATE_NEW, 0, NULL);
504     ok(file != INVALID_HANDLE_VALUE, "CreateFileA failed %u\n", GetLastError());
505     WriteFile(file, test_catalog, sizeof(test_catalog), &written, NULL);
506     CloseHandle(file);
507
508     /* Unique name will be created */
509     hcatinfo = pCryptCATAdminAddCatalog(hcatadmin, tmpfileW, NULL, 0);
510     todo_wine ok(hcatinfo != NULL, "CryptCATAdminAddCatalog failed %u\n", GetLastError());
511
512     info.cbStruct = sizeof(info);
513     info.wszCatalogFile[0] = 0;
514     ret = pCryptCATCatalogInfoFromContext(hcatinfo, &info, 0);
515     todo_wine
516     {
517     ok(ret, "CryptCATCatalogInfoFromContext failed %u\n", GetLastError());
518     ok(info.wszCatalogFile[0] != 0, "Expected a filename\n");
519     }
520     WideCharToMultiByte(CP_ACP, 0, info.wszCatalogFile, -1, catfile, MAX_PATH, 0, 0);
521     if ((p = strrchr(catfile, '\\'))) p++;
522     memset(catfileW, 0, sizeof(catfileW));
523     MultiByteToWideChar(0, 0, p, -1, catfileW, MAX_PATH);
524
525     /* Set the file attributes so we can check what happens with them during the 'copy' */
526     attrs = FILE_ATTRIBUTE_READONLY;
527     ret = SetFileAttributesA(tmpfile, attrs);
528     ok(ret, "SetFileAttributesA failed : %u\n", GetLastError());
529
530     /* winetest.cat will be created */
531     hcatinfo = pCryptCATAdminAddCatalog(hcatadmin, tmpfileW, basenameW, 0);
532     ok(hcatinfo != NULL, "CryptCATAdminAddCatalog failed %u\n", GetLastError());
533
534     lstrcpyA(catfilepath, catroot);
535     lstrcatA(catfilepath, "\\{DEADBEEF-DEAD-BEEF-DEAD-BEEFDEADBEEF}\\winetest.cat");
536     attrs = GetFileAttributes(catfilepath);
537     ok(attrs != INVALID_FILE_ATTRIBUTES, "Expected %s to exist\n", catfilepath);
538     todo_wine
539     ok(attrs == FILE_ATTRIBUTE_SYSTEM ||
540        attrs == (FILE_ATTRIBUTE_NOT_CONTENT_INDEXED | FILE_ATTRIBUTE_SYSTEM), /* Vista */
541        "File has wrong attributes : %08x\n", attrs);
542
543     info.cbStruct = sizeof(info);
544     info.wszCatalogFile[0] = 0;
545     ret = pCryptCATCatalogInfoFromContext(hcatinfo, &info, 0);
546     ok(ret, "CryptCATCatalogInfoFromContext failed %u\n", GetLastError());
547     ok(info.wszCatalogFile[0] != 0, "Expected a filename\n");
548     WideCharToMultiByte(CP_ACP, 0, info.wszCatalogFile, -1, catfile, MAX_PATH, 0, 0);
549     if ((p = strrchr(catfile, '\\'))) p++;
550     ok(!lstrcmpA(basename, p), "Expected %s, got %s\n", basename, p);
551
552     ret = pCryptCATAdminReleaseCatalogContext(hcatadmin, hcatinfo, 0);
553     ok(ret, "CryptCATAdminReleaseCatalogContext failed %u\n", GetLastError());
554
555     /* Remove the catalog file with the unique name */
556     ret = pCryptCATAdminRemoveCatalog(hcatadmin, catfileW, 0);
557     ok(ret, "CryptCATAdminRemoveCatalog failed %u\n", GetLastError());
558
559     /* Remove the winetest.cat catalog file, first with the full path. This should not succeed
560      * according to MSDN */
561     ret = pCryptCATAdminRemoveCatalog(hcatadmin, info.wszCatalogFile, 0);
562     ok(ret, "CryptCATAdminRemoveCatalog failed %u\n", GetLastError());
563     /* The call succeeded with the full path but the file is not removed */
564     attrs = GetFileAttributes(catfilepath);
565     ok(attrs != INVALID_FILE_ATTRIBUTES, "Expected %s to exist\n", catfilepath);
566     /* Given only the filename the file is removed */
567     ret = pCryptCATAdminRemoveCatalog(hcatadmin, basenameW, 0);
568     ok(ret, "CryptCATAdminRemoveCatalog failed %u\n", GetLastError());
569     attrs = GetFileAttributes(catfilepath);
570     ok(attrs == INVALID_FILE_ATTRIBUTES, "Expected %s to be removed\n", catfilepath);
571
572     ret = pCryptCATAdminReleaseContext(hcatadmin, 0);
573     ok(ret, "CryptCATAdminReleaseContext failed %u\n", GetLastError());
574
575     /* Set the attributes so we can delete the file */
576     attrs = FILE_ATTRIBUTE_NORMAL;
577     ret = SetFileAttributesA(tmpfile, attrs);
578     DeleteFileA(tmpfile);
579 }
580
581 static void test_catalog_properties(CHAR *catfile, int attributes, int members)
582 {
583     static const GUID subject = {0xde351a42,0x8e59,0x11d0,{0x8c,0x47,0x00,0xc0,0x4f,0xc2,0x95,0xee}};
584
585     HANDLE hcat;
586     CRYPTCATMEMBER *m;
587     CRYPTCATATTRIBUTE *attr;
588     char catalog[MAX_PATH];
589     WCHAR catalogW[MAX_PATH];
590     DWORD attrs;
591     BOOL ret;
592     int attrcount = 0, membercount = 0;
593
594     /* FIXME: Wine can't create catalog files out of catalog definition files yet. Remove this piece
595      * once wine is fixed
596      */
597     attrs = GetFileAttributesA(catfile);
598     if (attrs == INVALID_FILE_ATTRIBUTES)
599     {
600         HANDLE file;
601         DWORD written;
602
603         trace("Creating the catalog file\n");
604         if (!GetTempFileNameA(CURR_DIR, "cat", 0, catalog)) return;
605         file = CreateFileA(catalog, GENERIC_WRITE, 0, NULL, OPEN_EXISTING, 0, NULL);
606         ok(file != INVALID_HANDLE_VALUE, "CreateFileA failed %u\n", GetLastError());
607         WriteFile(file, test_catalog, sizeof(test_catalog), &written, NULL);
608         CloseHandle(file);
609
610         attributes = 2;
611         members = 1;
612         MultiByteToWideChar(CP_ACP, 0, catalog, -1, catalogW, MAX_PATH);
613     }
614     else
615     {
616         MultiByteToWideChar(CP_ACP, 0, catfile, -1, catalogW, MAX_PATH);
617         catalog[0] = 0;
618     }
619
620     hcat = pCryptCATOpen(NULL, 0, 0, 0, 0);
621     ok(hcat == INVALID_HANDLE_VALUE, "CryptCATOpen succeeded\n");
622
623     hcat = pCryptCATOpen(catalogW, 0, 0, 0, 0);
624     if (hcat == INVALID_HANDLE_VALUE && members == 0)
625     {
626         win_skip("CryptCATOpen on W2K can't handle catalog files with no members\n");
627         return;
628     }
629     ok(hcat != INVALID_HANDLE_VALUE, "CryptCATOpen failed %u\n", GetLastError());
630
631     m = pCryptCATEnumerateMember(NULL, NULL);
632     ok(m == NULL, "CryptCATEnumerateMember succeeded\n");
633
634     m = NULL;
635     while ((m = pCryptCATEnumerateMember(hcat, m)))
636     {
637         ok(m->cbStruct == sizeof(CRYPTCATMEMBER), "unexpected size %u\n", m->cbStruct);
638         todo_wine ok(!lstrcmpW(m->pwszReferenceTag, hashmeW), "unexpected tag\n");
639         ok(!memcmp(&m->gSubjectType, &subject, sizeof(subject)), "guid differs\n");
640         ok(!m->fdwMemberFlags, "got %x expected 0\n", m->fdwMemberFlags);
641         ok(m->dwCertVersion == 0x200, "got %x expected 0x200\n", m->dwCertVersion);
642         ok(!m->dwReserved, "got %x expected 0\n", m->dwReserved);
643         ok(m->hReserved == NULL, "got %p expected NULL\n", m->hReserved);
644
645         attr = pCryptCATEnumerateAttr(hcat, m, NULL);
646         ok(attr == NULL, "CryptCATEnumerateAttr succeeded\n");
647
648         membercount++;
649     }
650     ok(membercount == members, "Expected %d members, got %d\n", members, membercount);
651
652     attr = pCryptCATEnumerateAttr(NULL, NULL, NULL);
653     ok(attr == NULL, "CryptCATEnumerateAttr succeeded\n");
654
655     attr = pCryptCATEnumerateAttr(hcat, NULL, NULL);
656     ok(attr == NULL, "CryptCATEnumerateAttr succeeded\n");
657
658     attr = NULL;
659     while ((attr = pCryptCATEnumerateCatAttr(hcat, attr)))
660     {
661         ok(!lstrcmpW(attr->pwszReferenceTag, attr1W) ||
662            !lstrcmpW(attr->pwszReferenceTag, attr2W),
663            "Expected 'attr1' or 'attr2'\n");
664
665         attrcount++;
666     }
667     todo_wine
668     ok(attrcount == attributes, "Expected %d catalog attributes, got %d\n", attributes, attrcount);
669
670     ret = pCryptCATClose(hcat);
671     ok(ret, "CryptCATClose failed\n");
672     if (catalog[0]) DeleteFileA( catalog );
673 }
674
675 static void test_create_catalog_file(void)
676 {
677     static CHAR  catfileA[] = "winetest.cat";
678     static CHAR  cdffileA[] = "winetest.cdf";
679     static WCHAR cdffileW[] = {'w','i','n','e','t','e','s','t','.','c','d','f',0};
680     CRYPTCATCDF *catcdf;
681     CRYPTCATATTRIBUTE *catattr;
682     CRYPTCATMEMBER *catmember;
683     WCHAR  *catmembertag;
684     DWORD written, attrs;
685     HANDLE file;
686     BOOL ret;
687     int attrcount, membercount;
688
689     if (!pCryptCATCDFOpen)
690     {
691         win_skip("CryptCATCDFOpen is not available\n");
692         return;
693     }
694
695     /* Create the cdf file */
696     file = CreateFileA(cdffileA, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
697     ok(file != INVALID_HANDLE_VALUE, "CreateFileA failed %u\n", GetLastError());
698     WriteFile(file, test_cdf, sizeof(test_cdf) - 1, &written, NULL);
699     CloseHandle(file);
700
701     /* Don't enumerate attributes and members */
702     trace("No attribs and members\n");
703     SetLastError(0xdeadbeef);
704     catcdf = pCryptCATCDFOpen(cdffileW, NULL);
705     todo_wine
706     {
707     ok(catcdf != NULL, "CryptCATCDFOpen failed\n");
708     ok(GetLastError() == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %d\n", GetLastError());
709     }
710
711     ret = pCryptCATCDFClose(catcdf);
712     todo_wine
713     {
714     ok(ret, "Expected success\n");
715     ok(GetLastError() == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %d\n", GetLastError());
716     }
717
718     attrs = GetFileAttributesA(catfileA);
719     todo_wine
720     ok(attrs != INVALID_FILE_ATTRIBUTES, "Expected the catalog file to exist\n");
721
722     test_catalog_properties(catfileA, 0, 0);
723     DeleteFileA(catfileA);
724
725     /* Only enumerate the attributes */
726     trace("Only attributes\n");
727     attrcount = membercount = 0;
728     catcdf = pCryptCATCDFOpen(cdffileW, NULL);
729
730     catattr = NULL;
731     while ((catattr = pCryptCATCDFEnumCatAttributes(catcdf, catattr, NULL)))
732     {
733         ok(!lstrcmpW(catattr->pwszReferenceTag, attr1W) ||
734            !lstrcmpW(catattr->pwszReferenceTag, attr2W),
735            "Expected 'attr1' or 'attr2'\n");
736
737         attrcount++;
738     }
739     todo_wine
740     ok(attrcount == 2, "Expected 2 attributes, got %d\n", attrcount);
741
742     pCryptCATCDFClose(catcdf);
743     /* Even though the resulting catalog file shows the attributes, they will not be enumerated */
744     test_catalog_properties(catfileA, 0, 0);
745     DeleteFileA(catfileA);
746
747     /* Only enumerate the members */
748     trace("Only members\n");
749     attrcount = membercount = 0;
750     catcdf = pCryptCATCDFOpen(cdffileW, NULL);
751
752     catmember = NULL;
753     catmembertag = NULL;
754     while ((catmembertag = pCryptCATCDFEnumMembersByCDFTagEx(catcdf, catmembertag, NULL, &catmember, FALSE, NULL)))
755     {
756         ok(!lstrcmpW(catmembertag, hashmeW), "Expected 'hashme'\n");
757         membercount++;
758     }
759     todo_wine
760     ok(membercount == 1, "Expected 1 member, got %d\n", membercount);
761
762     pCryptCATCDFClose(catcdf);
763     test_catalog_properties(catfileA, 0, 1);
764     DeleteFileA(catfileA);
765
766     /* Enumerate members and attributes */
767     trace("Attributes and members\n");
768     attrcount = membercount = 0;
769     catcdf = pCryptCATCDFOpen(cdffileW, NULL);
770
771     catattr = NULL;
772     while ((catattr = pCryptCATCDFEnumCatAttributes(catcdf, catattr, NULL)))
773         attrcount++;
774     todo_wine
775     ok(attrcount == 2, "Expected 2 attributes, got %d\n", attrcount);
776
777     catmember = NULL;
778     catmembertag = NULL;
779     while ((catmembertag = pCryptCATCDFEnumMembersByCDFTagEx(catcdf, catmembertag, NULL, &catmember, FALSE, NULL)))
780         membercount++;
781     todo_wine
782     ok(membercount == 1, "Expected 1 member, got %d\n", membercount);
783
784     pCryptCATCDFClose(catcdf);
785     test_catalog_properties(catfileA, 2, 1);
786     DeleteFileA(catfileA);
787
788     DeleteFileA(cdffileA);
789 }
790
791 static void create_cdf_file(const CHAR *filename, const CHAR *contents)
792 {
793     HANDLE file;
794     DWORD written;
795
796     file = CreateFileA(filename, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
797     ok(file != INVALID_HANDLE_VALUE, "CreateFileA failed %u\n", GetLastError());
798     WriteFile(file, contents, lstrlenA(contents), &written, NULL);
799     CloseHandle(file);
800 }
801
802 #define CHECK_EXPECT(a, b) \
803     do { \
804         ok(a == error_area, "Expected %08x, got %08x\n", a, error_area); \
805         ok(b == local_error, "Expected %08x, got %08x\n", b, local_error); \
806     } while (0)
807
808 /* Clear the variables (can't use 0) */
809 #define CLEAR_EXPECT \
810     error_area = local_error = -1
811
812 /* Set both variables so the callback routine can check if a call to it was unexpected */
813 #define SET_UNEXPECTED \
814     error_area = local_error = -2
815
816 static void test_cdf_parsing(void)
817 {
818     static CHAR  catfileA[] = "tempfile.cat";
819     static CHAR  cdffileA[] = "tempfile.cdf";
820     static WCHAR cdffileW[] = {'t','e','m','p','f','i','l','e','.','c','d','f',0};
821     CHAR cdf_contents[4096];
822     CRYPTCATCDF *catcdf;
823     CRYPTCATATTRIBUTE *catattr;
824     CRYPTCATMEMBER *catmember;
825     WCHAR  *catmembertag;
826
827     if (!pCryptCATCDFOpen)
828     {
829         win_skip("CryptCATCDFOpen is not available\n");
830         return;
831     }
832
833     /* Empty file */
834     DeleteFileA(cdffileA);
835     create_cdf_file(cdffileA, "");
836
837     CLEAR_EXPECT;
838     SetLastError(0xdeadbeef);
839     catcdf = pCryptCATCDFOpen(cdffileW, cdf_callback);
840     todo_wine
841     CHECK_EXPECT(CRYPTCAT_E_AREA_HEADER, CRYPTCAT_E_CDF_TAGNOTFOUND);
842     ok(catcdf == NULL, "CryptCATCDFOpen succeeded\n");
843     todo_wine
844     ok(GetLastError() == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %d\n", GetLastError());
845     DeleteFileA(cdffileA);
846     ok(!DeleteFileA(catfileA), "Didn't expect a catalog file to be created\n");
847
848     /* Just the header */
849     lstrcpyA(cdf_contents, "[CatalogHeader]\r\n");
850     create_cdf_file(cdffileA, cdf_contents);
851
852     SET_UNEXPECTED;
853     SetLastError(0xdeadbeef);
854     catcdf = pCryptCATCDFOpen(cdffileW, cdf_callback);
855     ok(catcdf == NULL, "CryptCATCDFOpen succeeded\n");
856     todo_wine
857     ok(GetLastError() == ERROR_SHARING_VIOLATION ||
858         broken(GetLastError() == ERROR_SUCCESS),    /* win9x */
859         "Expected ERROR_SHARING_VIOLATION, got %d\n", GetLastError());
860     DeleteFileA(cdffileA);
861
862     /* Header and member only */
863     lstrcpyA(cdf_contents, "[CatalogHeader]\r\n");
864     lstrcatA(cdf_contents, "[CatalogFiles]\r\n");
865     lstrcatA(cdf_contents, "hashme=.\\tempfile.cdf\r\n");
866     create_cdf_file(cdffileA, cdf_contents);
867
868     SET_UNEXPECTED;
869     SetLastError(0xdeadbeef);
870     catcdf = pCryptCATCDFOpen(cdffileW, cdf_callback);
871     ok(catcdf == NULL, "CryptCATCDFOpen succeeded\n");
872     todo_wine
873     ok(GetLastError() == ERROR_SHARING_VIOLATION ||
874         broken(GetLastError() == ERROR_SUCCESS),    /* win9x */
875         "Expected ERROR_SHARING_VIOLATION, got %d\n", GetLastError());
876     DeleteFileA(cdffileA);
877     ok(!DeleteFileA(catfileA), "Didn't expect a catalog file to be created\n");
878
879     /* Header and Name (no value) */
880     lstrcpyA(cdf_contents, "[CatalogHeader]\r\n");
881     lstrcatA(cdf_contents, "Name=\r\n");
882     create_cdf_file(cdffileA, cdf_contents);
883
884     SET_UNEXPECTED;
885     SetLastError(0xdeadbeef);
886     catcdf = pCryptCATCDFOpen(cdffileW, cdf_callback);
887     ok(catcdf == NULL, "CryptCATCDFOpen succeeded\n");
888     todo_wine
889     ok(GetLastError() == ERROR_SHARING_VIOLATION ||
890         broken(GetLastError() == ERROR_SUCCESS),    /* win9x */
891         "Expected ERROR_SHARING_VIOLATION, got %d\n", GetLastError());
892     DeleteFileA(cdffileA);
893     ok(!DeleteFileA(catfileA), "Didn't expect a catalog file to be created\n");
894
895     /* Header and Name */
896     lstrcpyA(cdf_contents, "[CatalogHeader]\r\n");
897     lstrcatA(cdf_contents, "Name=tempfile.cat\r\n");
898     create_cdf_file(cdffileA, cdf_contents);
899
900     SET_UNEXPECTED;
901     SetLastError(0xdeadbeef);
902     catcdf = pCryptCATCDFOpen(cdffileW, cdf_callback);
903     todo_wine
904     {
905     ok(catcdf != NULL, "CryptCATCDFOpen failed\n");
906     ok(GetLastError() == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %d\n", GetLastError());
907     }
908     pCryptCATCDFClose(catcdf);
909     DeleteFileA(cdffileA);
910     todo_wine
911     ok(DeleteFileA(catfileA), "Expected a catalog file to be created\n");
912
913     /* Header and nonexistent member */
914     lstrcpyA(cdf_contents, "[CatalogHeader]\r\n");
915     lstrcatA(cdf_contents, "Name=tempfile.cat\r\n");
916     lstrcatA(cdf_contents, "[CatalogFiles]\r\n");
917     lstrcatA(cdf_contents, "hashme=.\\deadbeef.cdf\r\n");
918     create_cdf_file(cdffileA, cdf_contents);
919
920     SET_UNEXPECTED;
921     SetLastError(0xdeadbeef);
922     catcdf = pCryptCATCDFOpen(cdffileW, cdf_callback);
923     todo_wine
924     {
925     ok(catcdf != NULL, "CryptCATCDFOpen failed\n");
926     ok(GetLastError() == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %d\n", GetLastError());
927     }
928     /* Loop through the members */
929     CLEAR_EXPECT;
930     catmember = NULL;
931     catmembertag = NULL;
932     while ((catmembertag = pCryptCATCDFEnumMembersByCDFTagEx(catcdf, catmembertag, cdf_callback, &catmember, FALSE, NULL))) ;
933     todo_wine
934     CHECK_EXPECT(CRYPTCAT_E_AREA_MEMBER, CRYPTCAT_E_CDF_MEMBER_FILENOTFOUND);
935     pCryptCATCDFClose(catcdf);
936     DeleteFileA(cdffileA);
937     todo_wine
938     ok(DeleteFileA(catfileA), "Expected a catalog file to be created\n");
939
940     /* Header, correct member but no explicit newline */
941     lstrcpyA(cdf_contents, "[CatalogHeader]\r\n");
942     lstrcatA(cdf_contents, "Name=tempfile.cat\r\n");
943     lstrcatA(cdf_contents, "[CatalogFiles]\r\n");
944     lstrcatA(cdf_contents, "hashme=.\\tempfile.cdf\r");
945     create_cdf_file(cdffileA, cdf_contents);
946
947     SET_UNEXPECTED;
948     SetLastError(0xdeadbeef);
949     catcdf = pCryptCATCDFOpen(cdffileW, cdf_callback);
950     todo_wine
951     {
952     ok(catcdf != NULL, "CryptCATCDFOpen failed\n");
953     ok(GetLastError() == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %d\n", GetLastError());
954     }
955     /* Loop through the members */
956     CLEAR_EXPECT;
957     catmember = NULL;
958     catmembertag = NULL;
959     while ((catmembertag = pCryptCATCDFEnumMembersByCDFTagEx(catcdf, catmembertag, cdf_callback, &catmember, FALSE, NULL))) ;
960     todo_wine
961     CHECK_EXPECT(CRYPTCAT_E_AREA_MEMBER, CRYPTCAT_E_CDF_MEMBER_FILE_PATH);
962     pCryptCATCDFClose(catcdf);
963     DeleteFileA(cdffileA);
964     todo_wine
965     ok(DeleteFileA(catfileA), "Expected a catalog file to be created\n");
966
967     /* Header and 2 duplicate members */
968     lstrcpyA(cdf_contents, "[CatalogHeader]\r\n");
969     lstrcatA(cdf_contents, "Name=tempfile.cat\r\n");
970     lstrcatA(cdf_contents, "[CatalogFiles]\r\n");
971     lstrcatA(cdf_contents, "hashme=.\\tempfile.cdf\r\n");
972     lstrcatA(cdf_contents, "hashme=.\\tempfile.cdf\r\n");
973     create_cdf_file(cdffileA, cdf_contents);
974
975     SET_UNEXPECTED;
976     SetLastError(0xdeadbeef);
977     catcdf = pCryptCATCDFOpen(cdffileW, cdf_callback);
978     todo_wine
979     {
980     ok(catcdf != NULL, "CryptCATCDFOpen failed\n");
981     ok(GetLastError() == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %d\n", GetLastError());
982     }
983     /* Loop through the members */
984     SET_UNEXPECTED;
985     catmember = NULL;
986     catmembertag = NULL;
987     while ((catmembertag = pCryptCATCDFEnumMembersByCDFTagEx(catcdf, catmembertag, cdf_callback, &catmember, FALSE, NULL))) ;
988     pCryptCATCDFClose(catcdf);
989     test_catalog_properties(catfileA, 0, 1);
990     DeleteFileA(cdffileA);
991     todo_wine
992     ok(DeleteFileA(catfileA), "Expected a catalog file to be created\n");
993
994     /* Wrong attribute */
995     lstrcpyA(cdf_contents, "[CatalogHeader]\r\n");
996     lstrcatA(cdf_contents, "Name=tempfile.cat\r\n");
997     lstrcatA(cdf_contents, "CATATTR1=0x10010001:attr1\r\n");
998     create_cdf_file(cdffileA, cdf_contents);
999
1000     SET_UNEXPECTED;
1001     SetLastError(0xdeadbeef);
1002     catcdf = pCryptCATCDFOpen(cdffileW, cdf_callback);
1003     todo_wine
1004     {
1005     ok(catcdf != NULL, "CryptCATCDFOpen failed\n");
1006     ok(GetLastError() == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %d\n", GetLastError());
1007     }
1008     /* Loop through the attributes */
1009     CLEAR_EXPECT;
1010     catattr = NULL;
1011     while ((catattr = pCryptCATCDFEnumCatAttributes(catcdf, catattr, cdf_callback))) ;
1012     todo_wine
1013     CHECK_EXPECT(CRYPTCAT_E_AREA_ATTRIBUTE, CRYPTCAT_E_CDF_ATTR_TOOFEWVALUES);
1014     pCryptCATCDFClose(catcdf);
1015     DeleteFileA(cdffileA);
1016     todo_wine
1017     ok(DeleteFileA(catfileA), "Expected a catalog file to be created\n");
1018
1019     /* Two identical attributes */
1020     lstrcpyA(cdf_contents, "[CatalogHeader]\r\n");
1021     lstrcatA(cdf_contents, "Name=tempfile.cat\r\n");
1022     lstrcatA(cdf_contents, "CATATTR1=0x10010001:attr1:value1\r\n");
1023     lstrcatA(cdf_contents, "CATATTR1=0x10010001:attr1:value1\r\n");
1024     lstrcatA(cdf_contents, "[CatalogFiles]\r\n");
1025     lstrcatA(cdf_contents, "hashme=.\\tempfile.cdf\r\n");
1026     create_cdf_file(cdffileA, cdf_contents);
1027
1028     SET_UNEXPECTED;
1029     SetLastError(0xdeadbeef);
1030     catcdf = pCryptCATCDFOpen(cdffileW, cdf_callback);
1031     todo_wine
1032     {
1033     ok(catcdf != NULL, "CryptCATCDFOpen failed\n");
1034     ok(GetLastError() == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %d\n", GetLastError());
1035     }
1036     /* Loop through the members */
1037     SET_UNEXPECTED;
1038     catmember = NULL;
1039     catmembertag = NULL;
1040     while ((catmembertag = pCryptCATCDFEnumMembersByCDFTagEx(catcdf, catmembertag, cdf_callback, &catmember, FALSE, NULL))) ;
1041     /* Loop through the attributes */
1042     SET_UNEXPECTED;
1043     catattr = NULL;
1044     while ((catattr = pCryptCATCDFEnumCatAttributes(catcdf, catattr, cdf_callback))) ;
1045     pCryptCATCDFClose(catcdf);
1046     test_catalog_properties(catfileA, 1, 1);
1047     DeleteFileA(cdffileA);
1048     todo_wine
1049     ok(DeleteFileA(catfileA), "Expected a catalog file to be created\n");
1050
1051     /* Two different attribute values with the same tag */
1052     lstrcpyA(cdf_contents, "[CatalogHeader]\r\n");
1053     lstrcatA(cdf_contents, "Name=tempfile.cat\r\n");
1054     lstrcatA(cdf_contents, "CATATTR1=0x10010001:attr1:value1\r\n");
1055     lstrcatA(cdf_contents, "CATATTR1=0x10010001:attr2:value2\r\n");
1056     lstrcatA(cdf_contents, "[CatalogFiles]\r\n");
1057     lstrcatA(cdf_contents, "hashme=.\\tempfile.cdf\r\n");
1058     create_cdf_file(cdffileA, cdf_contents);
1059
1060     SET_UNEXPECTED;
1061     SetLastError(0xdeadbeef);
1062     catcdf = pCryptCATCDFOpen(cdffileW, cdf_callback);
1063     todo_wine
1064     {
1065     ok(catcdf != NULL, "CryptCATCDFOpen failed\n");
1066     ok(GetLastError() == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %d\n", GetLastError());
1067     }
1068     /* Loop through the members */
1069     SET_UNEXPECTED;
1070     catmember = NULL;
1071     catmembertag = NULL;
1072     while ((catmembertag = pCryptCATCDFEnumMembersByCDFTagEx(catcdf, catmembertag, cdf_callback, &catmember, FALSE, NULL))) ;
1073     /* Loop through the attributes */
1074     SET_UNEXPECTED;
1075     catattr = NULL;
1076     while ((catattr = pCryptCATCDFEnumCatAttributes(catcdf, catattr, cdf_callback))) ;
1077     pCryptCATCDFClose(catcdf);
1078     test_catalog_properties(catfileA, 1, 1);
1079     DeleteFileA(cdffileA);
1080     todo_wine
1081     ok(DeleteFileA(catfileA), "Expected a catalog file to be created\n");
1082 }
1083
1084 START_TEST(crypt)
1085 {
1086     int myARGC;
1087     char** myARGV;
1088     char sysdir[MAX_PATH];
1089
1090     InitFunctionPtrs();
1091
1092     if (!pCryptCATAdminAcquireContext)
1093     {
1094         win_skip("CryptCATAdmin functions are not available\n");
1095         return;
1096     }
1097
1098     GetSystemDirectoryA(sysdir, MAX_PATH);
1099     lstrcpyA(catroot, sysdir);
1100     lstrcatA(catroot, "\\CatRoot");
1101     lstrcpyA(catroot2, sysdir);
1102     lstrcatA(catroot2, "\\CatRoot2");
1103
1104     myARGC = winetest_get_mainargs(&myARGV);
1105     strcpy(selfname, myARGV[0]);
1106
1107     GetCurrentDirectoryA(MAX_PATH, CURR_DIR);
1108    
1109     test_context();
1110     test_calchash();
1111     /* Parameter checking only */
1112     test_CryptCATCDF_params();
1113     /* Test the parsing of a cdf file */
1114     test_cdf_parsing();
1115     /* Create a catalog file out of our own catalog definition file */
1116     test_create_catalog_file();
1117     test_CryptCATAdminAddRemoveCatalog();
1118 }