include/msvcrt: Define more CPU control word flags.
[wine] / dlls / wintrust / tests / softpub.c
1 /*
2  * wintrust softpub functions tests
3  *
4  * Copyright 2007,2010 Juan Lang
5  * Copyright 2010 Andrey Turkin
6  *
7  * This library is free software; you can redistribute it and/or
8  * modify it under the terms of the GNU Lesser General Public
9  * License as published by the Free Software Foundation; either
10  * version 2.1 of the License, or (at your option) any later version.
11  *
12  * This library is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
15  * Lesser General Public License for more details.
16  *
17  * You should have received a copy of the GNU Lesser General Public
18  * License along with this library; if not, write to the Free Software
19  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
20  */
21 #include <assert.h>
22 #include <stdio.h>
23 #include <stdarg.h>
24 #include <windef.h>
25 #include <winbase.h>
26 #include <winerror.h>
27 #include <wintrust.h>
28 #include <softpub.h>
29 #include <mssip.h>
30 #include <winuser.h>
31 #include "winnls.h"
32
33 #include "wine/test.h"
34
35 /* Just in case we're being built with borked headers, redefine function
36  * pointers to have the correct calling convention.
37  */
38 typedef void   *(WINAPI *SAFE_MEM_ALLOC)(DWORD);
39 typedef void    (WINAPI *SAFE_MEM_FREE)(void *);
40 typedef BOOL    (WINAPI *SAFE_ADD_STORE)(CRYPT_PROVIDER_DATA *,
41  HCERTSTORE);
42 typedef BOOL    (WINAPI *SAFE_ADD_SGNR)(CRYPT_PROVIDER_DATA *,
43  BOOL, DWORD, struct _CRYPT_PROVIDER_SGNR *);
44 typedef BOOL    (WINAPI *SAFE_ADD_CERT)(CRYPT_PROVIDER_DATA *,
45  DWORD, BOOL, DWORD, PCCERT_CONTEXT);
46 typedef BOOL    (WINAPI *SAFE_ADD_PRIVDATA)(CRYPT_PROVIDER_DATA *,
47  CRYPT_PROVIDER_PRIVDATA *);
48 typedef HRESULT (WINAPI *SAFE_PROVIDER_INIT_CALL)(CRYPT_PROVIDER_DATA *);
49 typedef HRESULT (WINAPI *SAFE_PROVIDER_OBJTRUST_CALL)(CRYPT_PROVIDER_DATA *);
50 typedef HRESULT (WINAPI *SAFE_PROVIDER_SIGTRUST_CALL)(CRYPT_PROVIDER_DATA *);
51 typedef HRESULT (WINAPI *SAFE_PROVIDER_CERTTRUST_CALL)(CRYPT_PROVIDER_DATA *);
52 typedef HRESULT (WINAPI *SAFE_PROVIDER_FINALPOLICY_CALL)(CRYPT_PROVIDER_DATA *);
53 typedef HRESULT (WINAPI *SAFE_PROVIDER_TESTFINALPOLICY_CALL)(
54  CRYPT_PROVIDER_DATA *);
55 typedef HRESULT (WINAPI *SAFE_PROVIDER_CLEANUP_CALL)(CRYPT_PROVIDER_DATA *);
56 typedef BOOL    (WINAPI *SAFE_PROVIDER_CERTCHKPOLICY_CALL)(
57  CRYPT_PROVIDER_DATA *, DWORD, BOOL, DWORD);
58
59 typedef struct _SAFE_PROVIDER_FUNCTIONS
60 {
61     DWORD                              cbStruct;
62     SAFE_MEM_ALLOC                     pfnAlloc;
63     SAFE_MEM_FREE                      pfnFree;
64     SAFE_ADD_STORE                     pfnAddStore2Chain;
65     SAFE_ADD_SGNR                      pfnAddSgnr2Chain;
66     SAFE_ADD_CERT                      pfnAddCert2Chain;
67     SAFE_ADD_PRIVDATA                  pfnAddPrivData2Chain;
68     SAFE_PROVIDER_INIT_CALL            pfnInitialize;
69     SAFE_PROVIDER_OBJTRUST_CALL        pfnObjectTrust;
70     SAFE_PROVIDER_SIGTRUST_CALL        pfnSignatureTrust;
71     SAFE_PROVIDER_CERTTRUST_CALL       pfnCertificateTrust;
72     SAFE_PROVIDER_FINALPOLICY_CALL     pfnFinalPolicy;
73     SAFE_PROVIDER_CERTCHKPOLICY_CALL   pfnCertCheckPolicy;
74     SAFE_PROVIDER_TESTFINALPOLICY_CALL pfnTestFinalPolicy;
75     struct _CRYPT_PROVUI_FUNCS        *psUIpfns;
76     SAFE_PROVIDER_CLEANUP_CALL         pfnCleanupPolicy;
77 } SAFE_PROVIDER_FUNCTIONS;
78
79 static BOOL (WINAPI * pWTHelperGetKnownUsages)(DWORD action, PCCRYPT_OID_INFO **usages);
80 static BOOL (WINAPI * CryptSIPCreateIndirectData_p)(SIP_SUBJECTINFO *, DWORD *, SIP_INDIRECT_DATA *);
81 static VOID (WINAPI * CertFreeCertificateChain_p)(PCCERT_CHAIN_CONTEXT);
82
83 static void InitFunctionPtrs(void)
84 {
85     HMODULE hWintrust = GetModuleHandleA("wintrust.dll");
86     HMODULE hCrypt32 = GetModuleHandleA("crypt32.dll");
87
88 #define WINTRUST_GET_PROC(func) \
89     p ## func = (void*)GetProcAddress(hWintrust, #func); \
90     if(!p ## func) { \
91       trace("GetProcAddress(%s) failed\n", #func); \
92     }
93
94     WINTRUST_GET_PROC(WTHelperGetKnownUsages)
95
96 #undef WINTRUST_GET_PROC
97
98 #define CRYPT32_GET_PROC(func) \
99     func ## _p = (void*)GetProcAddress(hCrypt32, #func); \
100     if(!func ## _p) { \
101       trace("GetProcAddress(%s) failed\n", #func); \
102     }
103
104     CRYPT32_GET_PROC(CryptSIPCreateIndirectData)
105     CRYPT32_GET_PROC(CertFreeCertificateChain)
106
107 #undef CRYPT32_GET_PROC
108 }
109
110 static const BYTE v1CertWithPubKey[] = {
111 0x30,0x81,0x95,0x02,0x01,0x01,0x30,0x02,0x06,0x00,0x30,0x15,0x31,0x13,0x30,
112 0x11,0x06,0x03,0x55,0x04,0x03,0x13,0x0a,0x4a,0x75,0x61,0x6e,0x20,0x4c,0x61,
113 0x6e,0x67,0x00,0x30,0x22,0x18,0x0f,0x31,0x36,0x30,0x31,0x30,0x31,0x30,0x31,
114 0x30,0x30,0x30,0x30,0x30,0x30,0x5a,0x18,0x0f,0x31,0x36,0x30,0x31,0x30,0x31,
115 0x30,0x31,0x30,0x30,0x30,0x30,0x30,0x30,0x5a,0x30,0x15,0x31,0x13,0x30,0x11,
116 0x06,0x03,0x55,0x04,0x03,0x13,0x0a,0x4a,0x75,0x61,0x6e,0x20,0x4c,0x61,0x6e,
117 0x67,0x00,0x30,0x22,0x30,0x0d,0x06,0x09,0x2a,0x86,0x48,0x86,0xf7,0x0d,0x01,
118 0x01,0x01,0x05,0x00,0x03,0x11,0x00,0x00,0x01,0x02,0x03,0x04,0x05,0x06,0x07,
119 0x08,0x09,0x0a,0x0b,0x0c,0x0d,0x0e,0x0f,0xa3,0x16,0x30,0x14,0x30,0x12,0x06,
120 0x03,0x55,0x1d,0x13,0x01,0x01,0xff,0x04,0x08,0x30,0x06,0x01,0x01,0xff,0x02,
121 0x01,0x01 };
122
123 static void test_utils(SAFE_PROVIDER_FUNCTIONS *funcs)
124 {
125     CRYPT_PROVIDER_DATA data = { 0 };
126     HCERTSTORE store;
127     CRYPT_PROVIDER_SGNR sgnr = { 0 };
128     BOOL ret;
129
130     /* Crash
131     ret = funcs->pfnAddStore2Chain(NULL, NULL);
132     ret = funcs->pfnAddStore2Chain(&data, NULL);
133      */
134     store = CertOpenStore(CERT_STORE_PROV_MEMORY, X509_ASN_ENCODING, 0,
135      CERT_STORE_CREATE_NEW_FLAG, NULL);
136     if (store)
137     {
138         ret = funcs->pfnAddStore2Chain(&data, store);
139         ok(ret, "pfnAddStore2Chain failed: %08x\n", GetLastError());
140         ok(data.chStores == 1, "Expected 1 store, got %d\n", data.chStores);
141         ok(data.pahStores != NULL, "Expected pahStores to be allocated\n");
142         if (data.pahStores)
143         {
144             ok(data.pahStores[0] == store, "Unexpected store\n");
145             CertCloseStore(data.pahStores[0], 0);
146             funcs->pfnFree(data.pahStores);
147             data.pahStores = NULL;
148             data.chStores = 0;
149             CertCloseStore(store, 0);
150             store = NULL;
151         }
152     }
153     else
154         skip("CertOpenStore failed: %08x\n", GetLastError());
155
156     /* Crash
157     ret = funcs->pfnAddSgnr2Chain(NULL, FALSE, 0, NULL);
158     ret = funcs->pfnAddSgnr2Chain(&data, FALSE, 0, NULL);
159      */
160     ret = funcs->pfnAddSgnr2Chain(&data, FALSE, 0, &sgnr);
161     ok(ret, "pfnAddSgnr2Chain failed: %08x\n", GetLastError());
162     ok(data.csSigners == 1, "Expected 1 signer, got %d\n", data.csSigners);
163     ok(data.pasSigners != NULL, "Expected pasSigners to be allocated\n");
164     if (data.pasSigners)
165     {
166         PCCERT_CONTEXT cert;
167
168         ok(!memcmp(&data.pasSigners[0], &sgnr, sizeof(sgnr)),
169          "Unexpected data in signer\n");
170         /* Adds into the location specified by the index */
171         sgnr.cbStruct = sizeof(CRYPT_PROVIDER_SGNR);
172         sgnr.sftVerifyAsOf.dwLowDateTime = 0xdeadbeef;
173         ret = funcs->pfnAddSgnr2Chain(&data, FALSE, 1, &sgnr);
174         ok(ret, "pfnAddSgnr2Chain failed: %08x\n", GetLastError());
175         ok(data.csSigners == 2, "Expected 2 signers, got %d\n", data.csSigners);
176         ok(!memcmp(&data.pasSigners[1], &sgnr, sizeof(sgnr)),
177          "Unexpected data in signer\n");
178         /* This also adds, but the data aren't copied */
179         sgnr.cbStruct = sizeof(DWORD);
180         ret = funcs->pfnAddSgnr2Chain(&data, FALSE, 0, &sgnr);
181         ok(ret, "pfnAddSgnr2Chain failed: %08x\n", GetLastError());
182         ok(data.csSigners == 3, "Expected 3 signers, got %d\n", data.csSigners);
183         ok(data.pasSigners[0].cbStruct == 0, "Unexpected data size %d\n",
184          data.pasSigners[0].cbStruct);
185         ok(data.pasSigners[0].sftVerifyAsOf.dwLowDateTime == 0,
186          "Unexpected verify time %d\n",
187          data.pasSigners[0].sftVerifyAsOf.dwLowDateTime);
188         /* But too large a thing isn't added */
189         sgnr.cbStruct = sizeof(sgnr) + sizeof(DWORD);
190         SetLastError(0xdeadbeef);
191         ret = funcs->pfnAddSgnr2Chain(&data, FALSE, 0, &sgnr);
192         ok(!ret && GetLastError() == ERROR_INVALID_PARAMETER,
193          "Expected ERROR_INVALID_PARAMETER, got %d\n", GetLastError());
194
195         /* Crash
196         ret = funcs->pfnAddCert2Chain(NULL, 0, FALSE, 0, NULL);
197         ret = funcs->pfnAddCert2Chain(&data, 0, FALSE, 0, NULL);
198          */
199         cert = CertCreateCertificateContext(X509_ASN_ENCODING, v1CertWithPubKey,
200          sizeof(v1CertWithPubKey));
201         if (cert)
202         {
203             /* Notes on behavior that are hard to test:
204              * 1. If pasSigners is invalid, pfnAddCert2Chain crashes
205              * 2. An invalid signer index isn't checked.
206              */
207             ret = funcs->pfnAddCert2Chain(&data, 0, FALSE, 0, cert);
208             ok(ret, "pfnAddCert2Chain failed: %08x\n", GetLastError());
209             ok(data.pasSigners[0].csCertChain == 1, "Expected 1 cert, got %d\n",
210              data.pasSigners[0].csCertChain);
211             ok(data.pasSigners[0].pasCertChain != NULL,
212              "Expected pasCertChain to be allocated\n");
213             if (data.pasSigners[0].pasCertChain)
214             {
215                 ok(data.pasSigners[0].pasCertChain[0].pCert == cert,
216                  "Unexpected cert\n");
217                 CertFreeCertificateContext(
218                  data.pasSigners[0].pasCertChain[0].pCert);
219             }
220             CertFreeCertificateContext(cert);
221         }
222         else
223             skip("CertCreateCertificateContext failed: %08x\n", GetLastError());
224     }
225 }
226
227 static void testInitialize(SAFE_PROVIDER_FUNCTIONS *funcs, GUID *actionID)
228 {
229     HRESULT ret;
230     CRYPT_PROVIDER_DATA data = { 0 };
231     WINTRUST_DATA wintrust_data = { 0 };
232
233     if (!funcs->pfnInitialize)
234     {
235         skip("missing pfnInitialize\n");
236         return;
237     }
238
239     /* Crashes
240     ret = funcs->pfnInitialize(NULL);
241      */
242     memset(&data, 0, sizeof(data));
243     ret = funcs->pfnInitialize(&data);
244     ok(ret == S_FALSE, "Expected S_FALSE, got %08x\n", ret);
245     data.padwTrustStepErrors =
246      funcs->pfnAlloc(TRUSTERROR_MAX_STEPS * sizeof(DWORD));
247     /* Without wintrust data set, crashes when padwTrustStepErrors is set */
248     data.pWintrustData = &wintrust_data;
249     if (data.padwTrustStepErrors)
250     {
251         /* Apparently, cdwTrustStepErrors does not need to be set. */
252         ret = funcs->pfnInitialize(&data);
253         ok(ret == S_OK, "Expected S_OK, got %08x\n", ret);
254         data.cdwTrustStepErrors = 1;
255         ret = funcs->pfnInitialize(&data);
256         ok(ret == S_OK, "Expected S_OK, got %08x\n", ret);
257         memset(data.padwTrustStepErrors, 0xba,
258          TRUSTERROR_MAX_STEPS * sizeof(DWORD));
259         ret = funcs->pfnInitialize(&data);
260         ok(ret == S_FALSE, "Expected S_FALSE, got %08x\n", ret);
261         data.padwTrustStepErrors[TRUSTERROR_STEP_FINAL_WVTINIT] = 0;
262         ret = funcs->pfnInitialize(&data);
263         ok(ret == S_OK, "Expected S_OK, got %08x\n", ret);
264         funcs->pfnFree(data.padwTrustStepErrors);
265     }
266 }
267
268 static void getNotepadPath(WCHAR *notepadPathW, DWORD size)
269 {
270     static const CHAR notepad[] = "\\notepad.exe";
271     CHAR notepadPath[MAX_PATH];
272
273     /* Workaround missing W-functions for win9x */
274     GetWindowsDirectoryA(notepadPath, MAX_PATH);
275     lstrcatA(notepadPath, notepad);
276     MultiByteToWideChar(0, 0, notepadPath, -1, notepadPathW, size);
277 }
278
279 static void testObjTrust(SAFE_PROVIDER_FUNCTIONS *funcs, GUID *actionID)
280 {
281     HRESULT ret;
282     CRYPT_PROVIDER_DATA data = { 0 };
283     WINTRUST_DATA wintrust_data = { 0 };
284     WINTRUST_CERT_INFO certInfo = { sizeof(WINTRUST_CERT_INFO), 0 };
285     WINTRUST_FILE_INFO fileInfo = { sizeof(WINTRUST_FILE_INFO), 0 };
286
287     if (!funcs->pfnObjectTrust)
288     {
289         skip("missing pfnObjectTrust\n");
290         return;
291     }
292
293     /* Crashes
294     ret = funcs->pfnObjectTrust(NULL);
295      */
296     data.pWintrustData = &wintrust_data;
297     data.padwTrustStepErrors =
298      funcs->pfnAlloc(TRUSTERROR_MAX_STEPS * sizeof(DWORD));
299     if (data.padwTrustStepErrors)
300     {
301         WCHAR notepadPathW[MAX_PATH];
302         PROVDATA_SIP provDataSIP = { 0 };
303         static const GUID unknown = { 0xC689AAB8, 0x8E78, 0x11D0, { 0x8C,0x47,
304          0x00,0xC0,0x4F,0xC2,0x95,0xEE } };
305         static GUID bogusGuid = { 0xdeadbeef, 0xbaad, 0xf00d, { 0x00,0x00,0x00,
306          0x00,0x00,0x00,0x00,0x00 } };
307
308         ret = funcs->pfnObjectTrust(&data);
309         ok(ret == S_FALSE, "Expected S_FALSE, got %08x\n", ret);
310         ok(data.padwTrustStepErrors[TRUSTERROR_STEP_FINAL_OBJPROV] ==
311          ERROR_INVALID_PARAMETER,
312          "Expected ERROR_INVALID_PARAMETER, got %08x\n",
313          data.padwTrustStepErrors[TRUSTERROR_STEP_FINAL_OBJPROV]);
314         U(wintrust_data).pCert = &certInfo;
315         wintrust_data.dwUnionChoice = WTD_CHOICE_CERT;
316         ret = funcs->pfnObjectTrust(&data);
317         ok(ret == S_OK, "Expected S_OK, got %08x\n", ret);
318         certInfo.psCertContext = (PCERT_CONTEXT)CertCreateCertificateContext(
319          X509_ASN_ENCODING, v1CertWithPubKey, sizeof(v1CertWithPubKey));
320         ret = funcs->pfnObjectTrust(&data);
321         ok(ret == S_OK, "Expected S_OK, got %08x\n", ret);
322         CertFreeCertificateContext(certInfo.psCertContext);
323         certInfo.psCertContext = NULL;
324         wintrust_data.dwUnionChoice = WTD_CHOICE_FILE;
325         U(wintrust_data).pFile = NULL;
326         ret = funcs->pfnObjectTrust(&data);
327         ok(ret == S_FALSE, "Expected S_FALSE, got %08x\n", ret);
328         ok(data.padwTrustStepErrors[TRUSTERROR_STEP_FINAL_OBJPROV] ==
329          ERROR_INVALID_PARAMETER,
330          "Expected ERROR_INVALID_PARAMETER, got %08x\n",
331          data.padwTrustStepErrors[TRUSTERROR_STEP_FINAL_OBJPROV]);
332         U(wintrust_data).pFile = &fileInfo;
333         /* Crashes
334         ret = funcs->pfnObjectTrust(&data);
335          */
336         getNotepadPath(notepadPathW, MAX_PATH);
337         fileInfo.pcwszFilePath = notepadPathW;
338         /* pfnObjectTrust now crashes unless both pPDSip and psPfns are set */
339         U(data).pPDSip = &provDataSIP;
340         data.psPfns = (CRYPT_PROVIDER_FUNCTIONS *)funcs;
341         ret = funcs->pfnObjectTrust(&data);
342         ok(ret == S_FALSE, "Expected S_FALSE, got %08x\n", ret);
343         ok(data.padwTrustStepErrors[TRUSTERROR_STEP_FINAL_OBJPROV] ==
344          TRUST_E_NOSIGNATURE ||
345          data.padwTrustStepErrors[TRUSTERROR_STEP_FINAL_OBJPROV] ==
346          TRUST_E_SUBJECT_FORM_UNKNOWN,
347          "Expected TRUST_E_NOSIGNATURE or TRUST_E_SUBJECT_FORM_UNKNOWN, got %08x\n",
348          data.padwTrustStepErrors[TRUSTERROR_STEP_FINAL_OBJPROV]);
349         if (data.padwTrustStepErrors[TRUSTERROR_STEP_FINAL_OBJPROV] ==
350          TRUST_E_NOSIGNATURE)
351         {
352             ok(!memcmp(&provDataSIP.gSubject, &unknown, sizeof(unknown)),
353              "Unexpected subject GUID\n");
354             ok(provDataSIP.pSip != NULL, "Expected a SIP\n");
355             ok(provDataSIP.psSipSubjectInfo != NULL,
356              "Expected a subject info\n");
357         }
358         /* Specifying the GUID results in that GUID being the subject GUID */
359         fileInfo.pgKnownSubject = &bogusGuid;
360         ret = funcs->pfnObjectTrust(&data);
361         ok(ret == S_FALSE, "Expected S_FALSE, got %08x\n", ret);
362         ok(data.padwTrustStepErrors[TRUSTERROR_STEP_FINAL_OBJPROV] ==
363          TRUST_E_NOSIGNATURE ||
364          data.padwTrustStepErrors[TRUSTERROR_STEP_FINAL_OBJPROV] ==
365          TRUST_E_SUBJECT_FORM_UNKNOWN ||
366          data.padwTrustStepErrors[TRUSTERROR_STEP_FINAL_OBJPROV] ==
367          TRUST_E_PROVIDER_UNKNOWN,
368          "Expected TRUST_E_NOSIGNATURE or TRUST_E_SUBJECT_FORM_UNKNOWN or TRUST_E_PROVIDER_UNKNOWN, got %08x\n",
369          data.padwTrustStepErrors[TRUSTERROR_STEP_FINAL_OBJPROV]);
370         if (data.padwTrustStepErrors[TRUSTERROR_STEP_FINAL_OBJPROV] ==
371          TRUST_E_NOSIGNATURE)
372         {
373             ok(!memcmp(&provDataSIP.gSubject, &bogusGuid, sizeof(bogusGuid)),
374              "unexpected subject GUID\n");
375         }
376         /* Specifying a bogus GUID pointer crashes */
377         if (0)
378         {
379             fileInfo.pgKnownSubject = (GUID *)0xdeadbeef;
380             ret = funcs->pfnObjectTrust(&data);
381         }
382         funcs->pfnFree(data.padwTrustStepErrors);
383     }
384 }
385
386 static const BYTE selfSignedCert[] = {
387  0x30, 0x82, 0x01, 0x1f, 0x30, 0x81, 0xce, 0xa0, 0x03, 0x02, 0x01, 0x02, 0x02,
388  0x10, 0xeb, 0x0d, 0x57, 0x2a, 0x9c, 0x09, 0xba, 0xa4, 0x4a, 0xb7, 0x25, 0x49,
389  0xd9, 0x3e, 0xb5, 0x73, 0x30, 0x09, 0x06, 0x05, 0x2b, 0x0e, 0x03, 0x02, 0x1d,
390  0x05, 0x00, 0x30, 0x15, 0x31, 0x13, 0x30, 0x11, 0x06, 0x03, 0x55, 0x04, 0x03,
391  0x13, 0x0a, 0x4a, 0x75, 0x61, 0x6e, 0x20, 0x4c, 0x61, 0x6e, 0x67, 0x00, 0x30,
392  0x1e, 0x17, 0x0d, 0x30, 0x36, 0x30, 0x36, 0x32, 0x39, 0x30, 0x35, 0x30, 0x30,
393  0x34, 0x36, 0x5a, 0x17, 0x0d, 0x30, 0x37, 0x30, 0x36, 0x32, 0x39, 0x31, 0x31,
394  0x30, 0x30, 0x34, 0x36, 0x5a, 0x30, 0x15, 0x31, 0x13, 0x30, 0x11, 0x06, 0x03,
395  0x55, 0x04, 0x03, 0x13, 0x0a, 0x4a, 0x75, 0x61, 0x6e, 0x20, 0x4c, 0x61, 0x6e,
396  0x67, 0x00, 0x30, 0x5c, 0x30, 0x0d, 0x06, 0x09, 0x2a, 0x86, 0x48, 0x86, 0xf7,
397  0x0d, 0x01, 0x01, 0x01, 0x05, 0x00, 0x03, 0x4b, 0x00, 0x30, 0x48, 0x02, 0x41,
398  0x00, 0xe2, 0x54, 0x3a, 0xa7, 0x83, 0xb1, 0x27, 0x14, 0x3e, 0x59, 0xbb, 0xb4,
399  0x53, 0xe6, 0x1f, 0xe7, 0x5d, 0xf1, 0x21, 0x68, 0xad, 0x85, 0x53, 0xdb, 0x6b,
400  0x1e, 0xeb, 0x65, 0x97, 0x03, 0x86, 0x60, 0xde, 0xf3, 0x6c, 0x38, 0x75, 0xe0,
401  0x4c, 0x61, 0xbb, 0xbc, 0x62, 0x17, 0xa9, 0xcd, 0x79, 0x3f, 0x21, 0x4e, 0x96,
402  0xcb, 0x0e, 0xdc, 0x61, 0x94, 0x30, 0x18, 0x10, 0x6b, 0xd0, 0x1c, 0x10, 0x79,
403  0x02, 0x03, 0x01, 0x00, 0x01, 0x30, 0x09, 0x06, 0x05, 0x2b, 0x0e, 0x03, 0x02,
404  0x1d, 0x05, 0x00, 0x03, 0x41, 0x00, 0x25, 0x90, 0x53, 0x34, 0xd9, 0x56, 0x41,
405  0x5e, 0xdb, 0x7e, 0x01, 0x36, 0xec, 0x27, 0x61, 0x5e, 0xb7, 0x4d, 0x90, 0x66,
406  0xa2, 0xe1, 0x9d, 0x58, 0x76, 0xd4, 0x9c, 0xba, 0x2c, 0x84, 0xc6, 0x83, 0x7a,
407  0x22, 0x0d, 0x03, 0x69, 0x32, 0x1a, 0x6d, 0xcb, 0x0c, 0x15, 0xb3, 0x6b, 0xc7,
408  0x0a, 0x8c, 0xb4, 0x5c, 0x34, 0x78, 0xe0, 0x3c, 0x9c, 0xe9, 0xf3, 0x30, 0x9f,
409  0xa8, 0x76, 0x57, 0x92, 0x36 };
410
411 static void testCertTrust(SAFE_PROVIDER_FUNCTIONS *funcs, GUID *actionID)
412 {
413     CRYPT_PROVIDER_DATA data = { 0 };
414     CRYPT_PROVIDER_SGNR sgnr = { sizeof(sgnr), { 0 } };
415     HRESULT ret;
416
417     if (!CertFreeCertificateChain_p)
418     {
419         win_skip("CertFreeCertificateChain not found\n");
420         return;
421     }
422
423     data.padwTrustStepErrors =
424      funcs->pfnAlloc(TRUSTERROR_MAX_STEPS * sizeof(DWORD));
425     if (!data.padwTrustStepErrors)
426     {
427         skip("pfnAlloc failed\n");
428         return;
429     }
430     ret = funcs->pfnCertificateTrust(&data);
431     ok(ret == S_FALSE, "Expected S_FALSE, got %08x\n", ret);
432     ok(data.padwTrustStepErrors[TRUSTERROR_STEP_FINAL_CERTPROV] ==
433      TRUST_E_NOSIGNATURE, "Expected TRUST_E_NOSIGNATURE, got %08x\n",
434      data.padwTrustStepErrors[TRUSTERROR_STEP_FINAL_CERTPROV]);
435     ret = funcs->pfnAddSgnr2Chain(&data, FALSE, 0, &sgnr);
436     if (ret)
437     {
438         PCCERT_CONTEXT cert;
439
440         /* An empty signer "succeeds," even though there's no cert */
441         ret = funcs->pfnCertificateTrust(&data);
442         ok(ret == S_OK, "Expected S_OK, got %08x\n", ret);
443         cert = CertCreateCertificateContext(X509_ASN_ENCODING, selfSignedCert,
444          sizeof(selfSignedCert));
445         if (cert)
446         {
447             WINTRUST_DATA wintrust_data = { 0 };
448
449             ret = funcs->pfnAddCert2Chain(&data, 0, FALSE, 0, cert);
450             /* If pWintrustData isn't set, crashes attempting to access
451              * pWintrustData->fdwRevocationChecks
452              */
453             data.pWintrustData = &wintrust_data;
454             /* If psPfns isn't set, crashes attempting to access
455              * psPfns->pfnCertCheckPolicy
456              */
457             data.psPfns = (CRYPT_PROVIDER_FUNCTIONS *)funcs;
458             ret = funcs->pfnCertificateTrust(&data);
459             ok(ret == S_OK, "Expected S_OK, got %08x\n", ret);
460             ok(data.csSigners == 1, "Unexpected number of signers %d\n",
461              data.csSigners);
462             ok(data.pasSigners[0].pChainContext != NULL,
463              "Expected a certificate chain\n");
464             ok(data.pasSigners[0].csCertChain == 1,
465              "Unexpected number of chain elements %d\n",
466              data.pasSigners[0].csCertChain);
467             /* pasSigners and pasSigners[0].pasCertChain are guaranteed to be
468              * initialized, see tests for pfnAddSgnr2Chain and pfnAddCert2Chain
469              */
470             ok(!data.pasSigners[0].pasCertChain[0].fTrustedRoot,
471              "Didn't expect cert to be trusted\n");
472             ok(data.pasSigners[0].pasCertChain[0].fSelfSigned,
473              "Expected cert to be self-signed\n");
474             ok(data.pasSigners[0].pasCertChain[0].dwConfidence ==
475              (CERT_CONFIDENCE_SIG | CERT_CONFIDENCE_TIMENEST),
476              "Expected CERT_CONFIDENCE_SIG | CERT_CONFIDENCE_TIMENEST, got %08x\n",
477              data.pasSigners[0].pasCertChain[0].dwConfidence);
478             CertFreeCertificateContext(
479              data.pasSigners[0].pasCertChain[0].pCert);
480             CertFreeCertificateChain_p(data.pasSigners[0].pChainContext);
481             CertFreeCertificateContext(cert);
482         }
483     }
484 }
485
486 static void test_provider_funcs(void)
487 {
488     static GUID generic_verify_v2 = WINTRUST_ACTION_GENERIC_VERIFY_V2;
489     SAFE_PROVIDER_FUNCTIONS funcs = { sizeof(SAFE_PROVIDER_FUNCTIONS), 0 };
490     BOOL ret;
491
492     ret = WintrustLoadFunctionPointers(&generic_verify_v2,
493      (CRYPT_PROVIDER_FUNCTIONS *)&funcs);
494     if (!ret)
495         skip("WintrustLoadFunctionPointers failed\n");
496     else
497     {
498         test_utils(&funcs);
499         testInitialize(&funcs, &generic_verify_v2);
500         testObjTrust(&funcs, &generic_verify_v2);
501         testCertTrust(&funcs, &generic_verify_v2);
502     }
503 }
504
505 /* minimal PE file image */
506 #define VA_START 0x400000
507 #define FILE_PE_START 0x50
508 #define NUM_SECTIONS 3
509 #define FILE_TEXT 0x200
510 #define RVA_TEXT 0x1000
511 #define RVA_BSS 0x2000
512 #define FILE_IDATA 0x400
513 #define RVA_IDATA 0x3000
514 #define FILE_TOTAL 0x600
515 #define RVA_TOTAL 0x4000
516 #include <pshpack1.h>
517 struct Imports {
518     IMAGE_IMPORT_DESCRIPTOR descriptors[2];
519     IMAGE_THUNK_DATA32 original_thunks[2];
520     IMAGE_THUNK_DATA32 thunks[2];
521     struct __IMPORT_BY_NAME {
522         WORD hint;
523         char funcname[0x20];
524     } ibn;
525     char dllname[0x10];
526 };
527 #define EXIT_PROCESS (VA_START+RVA_IDATA+FIELD_OFFSET(struct Imports, thunks[0]))
528
529 static struct _PeImage {
530     IMAGE_DOS_HEADER dos_header;
531     char __alignment1[FILE_PE_START - sizeof(IMAGE_DOS_HEADER)];
532     IMAGE_NT_HEADERS32 nt_headers;
533     IMAGE_SECTION_HEADER sections[NUM_SECTIONS];
534     char __alignment2[FILE_TEXT - FILE_PE_START - sizeof(IMAGE_NT_HEADERS32) -
535         NUM_SECTIONS * sizeof(IMAGE_SECTION_HEADER)];
536     unsigned char text_section[FILE_IDATA-FILE_TEXT];
537     struct Imports idata_section;
538     char __alignment3[FILE_TOTAL-FILE_IDATA-sizeof(struct Imports)];
539 } bin = {
540     /* dos header */
541     {IMAGE_DOS_SIGNATURE, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, {}, 0, 0, {}, FILE_PE_START},
542     /* alignment before PE header */
543     {},
544     /* nt headers */
545     {IMAGE_NT_SIGNATURE,
546         /* basic headers - 3 sections, no symbols, EXE file */
547         {IMAGE_FILE_MACHINE_I386, NUM_SECTIONS, 0, 0, 0, sizeof(IMAGE_OPTIONAL_HEADER32),
548             IMAGE_FILE_32BIT_MACHINE | IMAGE_FILE_EXECUTABLE_IMAGE},
549         /* optional header */
550         {IMAGE_NT_OPTIONAL_HDR32_MAGIC, 4, 0, FILE_IDATA-FILE_TEXT,
551             FILE_TOTAL-FILE_IDATA + FILE_IDATA-FILE_TEXT, 0x400,
552             RVA_TEXT, RVA_TEXT, RVA_BSS, VA_START, 0x1000, 0x200, 4, 0, 1, 0, 4, 0, 0,
553             RVA_TOTAL, FILE_TEXT, 0, IMAGE_SUBSYSTEM_WINDOWS_GUI, 0,
554             0x200000, 0x1000, 0x100000, 0x1000, 0, 0x10,
555             {{0, 0},
556              {RVA_IDATA, sizeof(struct Imports)}
557             }
558         }
559     },
560     /* sections */
561     {
562         {".text", {0x100}, RVA_TEXT, FILE_IDATA-FILE_TEXT, FILE_TEXT,
563             0, 0, 0, 0, IMAGE_SCN_CNT_CODE | IMAGE_SCN_MEM_EXECUTE | IMAGE_SCN_MEM_READ},
564         {".bss", {0x400}, RVA_BSS, 0, 0, 0, 0, 0, 0,
565             IMAGE_SCN_CNT_UNINITIALIZED_DATA | IMAGE_SCN_MEM_READ | IMAGE_SCN_MEM_WRITE},
566         {".idata", {sizeof(struct Imports)}, RVA_IDATA, FILE_TOTAL-FILE_IDATA, FILE_IDATA, 0,
567             0, 0, 0, IMAGE_SCN_CNT_INITIALIZED_DATA | IMAGE_SCN_MEM_READ | IMAGE_SCN_MEM_WRITE}
568     },
569     /* alignment before first section */
570     {},
571     /* .text section */
572     {
573         0x31, 0xC0, /* xor eax, eax */
574         0xFF, 0x25, EXIT_PROCESS&0xFF, (EXIT_PROCESS>>8)&0xFF, (EXIT_PROCESS>>16)&0xFF,
575             (EXIT_PROCESS>>24)&0xFF, /* jmp ExitProcess */
576         0
577     },
578     /* .idata section */
579     {
580         {
581             {{RVA_IDATA + FIELD_OFFSET(struct Imports, original_thunks)}, 0, 0,
582             RVA_IDATA + FIELD_OFFSET(struct Imports, dllname),
583             RVA_IDATA + FIELD_OFFSET(struct Imports, thunks)
584             },
585             {{0}, 0, 0, 0, 0}
586         },
587         {{{RVA_IDATA+FIELD_OFFSET(struct Imports, ibn)}}, {{0}}},
588         {{{RVA_IDATA+FIELD_OFFSET(struct Imports, ibn)}}, {{0}}},
589         {0,"ExitProcess"},
590         "KERNEL32.DLL"
591     },
592     /* final alignment */
593     {}
594 };
595 #include <poppack.h>
596
597 /* Creates a test file and returns a handle to it.  The file's path is returned
598  * in temp_file, which must be at least MAX_PATH characters in length.
599  */
600 static HANDLE create_temp_file(char *temp_file)
601 {
602     HANDLE file = INVALID_HANDLE_VALUE;
603     char temp_path[MAX_PATH];
604
605     if (GetTempPathA(sizeof(temp_path), temp_path))
606     {
607         if (GetTempFileNameA(temp_path, "img", 0, temp_file))
608             file = CreateFileA(temp_file, GENERIC_READ | GENERIC_WRITE, 0, NULL,
609              CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
610     }
611     return file;
612 }
613
614 static void test_sip_create_indirect_data(void)
615 {
616     static GUID unknown = { 0xC689AAB8, 0x8E78, 0x11D0, { 0x8C,0x47,
617      0x00,0xC0,0x4F,0xC2,0x95,0xEE } };
618     static char oid_sha1[] = szOID_OIWSEC_sha1;
619     BOOL ret;
620     SIP_SUBJECTINFO subjinfo = { 0 };
621     char temp_file[MAX_PATH];
622     HANDLE file;
623     DWORD count;
624
625     if (!CryptSIPCreateIndirectData_p)
626     {
627         skip("Missing CryptSIPCreateIndirectData\n");
628         return;
629     }
630     SetLastError(0xdeadbeef);
631     ret = CryptSIPCreateIndirectData_p(NULL, NULL, NULL);
632     ok(!ret && GetLastError() == ERROR_INVALID_PARAMETER,
633        "expected ERROR_INVALID_PARAMETER, got %d\n", GetLastError());
634     SetLastError(0xdeadbeef);
635     ret = CryptSIPCreateIndirectData_p(&subjinfo, NULL, NULL);
636     ok(!ret && GetLastError() == ERROR_INVALID_PARAMETER,
637        "expected ERROR_INVALID_PARAMETER, got %d\n", GetLastError());
638     subjinfo.cbSize = sizeof(subjinfo);
639     SetLastError(0xdeadbeef);
640     ret = CryptSIPCreateIndirectData_p(&subjinfo, NULL, NULL);
641     ok(!ret && GetLastError() == ERROR_INVALID_PARAMETER,
642        "expected ERROR_INVALID_PARAMETER, got %d\n", GetLastError());
643     file = create_temp_file(temp_file);
644     if (file == INVALID_HANDLE_VALUE)
645     {
646         skip("couldn't create temp file\n");
647         return;
648     }
649     WriteFile(file, &bin, sizeof(bin), &count, NULL);
650     FlushFileBuffers(file);
651
652     subjinfo.hFile = file;
653     SetLastError(0xdeadbeef);
654     ret = CryptSIPCreateIndirectData_p(&subjinfo, NULL, NULL);
655     ok(!ret && GetLastError() == ERROR_INVALID_PARAMETER,
656        "expected ERROR_INVALID_PARAMETER, got %d\n", GetLastError());
657     subjinfo.pgSubjectType = &unknown;
658     SetLastError(0xdeadbeef);
659     ret = CryptSIPCreateIndirectData_p(&subjinfo, NULL, NULL);
660     ok(!ret && GetLastError() == ERROR_INVALID_PARAMETER,
661        "expected ERROR_INVALID_PARAMETER, got %d\n", GetLastError());
662     count = 0xdeadbeef;
663     SetLastError(0xdeadbeef);
664     ret = CryptSIPCreateIndirectData_p(&subjinfo, &count, NULL);
665     todo_wine
666     ok(!ret && (GetLastError() == NTE_BAD_ALGID ||
667                 GetLastError() == ERROR_INVALID_PARAMETER /* Win7 */),
668        "expected NTE_BAD_ALGID or ERROR_INVALID_PARAMETER, got %08x\n",
669        GetLastError());
670     ok(count == 0xdeadbeef, "expected count to be unmodified, got %d\n", count);
671     subjinfo.DigestAlgorithm.pszObjId = oid_sha1;
672     count = 0xdeadbeef;
673     ret = CryptSIPCreateIndirectData_p(&subjinfo, &count, NULL);
674     todo_wine
675     ok(ret, "CryptSIPCreateIndirectData failed: %d\n", GetLastError());
676     ok(count, "expected a positive count\n");
677     if (ret)
678     {
679         SIP_INDIRECT_DATA *indirect = HeapAlloc(GetProcessHeap(), 0, count);
680
681         count = 256;
682         ret = CryptSIPCreateIndirectData_p(&subjinfo, &count, indirect);
683         ok(ret, "CryptSIPCreateIndirectData failed: %d\n", GetLastError());
684         /* If the count is larger than needed, it's unmodified */
685         ok(count == 256, "unexpected count %d\n", count);
686         ok(!strcmp(indirect->Data.pszObjId, SPC_PE_IMAGE_DATA_OBJID),
687            "unexpected data oid %s\n",
688            indirect->Data.pszObjId);
689         ok(!strcmp(indirect->DigestAlgorithm.pszObjId, oid_sha1),
690            "unexpected digest algorithm oid %s\n",
691            indirect->DigestAlgorithm.pszObjId);
692         ok(indirect->Digest.cbData == 20, "unexpected hash size %d\n",
693            indirect->Digest.cbData);
694         if (indirect->Digest.cbData == 20)
695         {
696             const BYTE hash[20] = {
697                 0x8a,0xd5,0x45,0x53,0x3d,0x67,0xdf,0x2f,0x78,0xe0,
698                 0x55,0x0a,0xe0,0xd9,0x7a,0x28,0x3e,0xbf,0x45,0x2b };
699
700             ok(!memcmp(indirect->Digest.pbData, hash, 20),
701                "unexpected value\n");
702         }
703
704         HeapFree(GetProcessHeap(), 0, indirect);
705     }
706     CloseHandle(file);
707     DeleteFileA(temp_file);
708 }
709
710 static void test_wintrust(void)
711 {
712     static GUID generic_action_v2 = WINTRUST_ACTION_GENERIC_VERIFY_V2;
713     WINTRUST_DATA wtd;
714     WINTRUST_FILE_INFO file;
715     LONG r;
716     HRESULT hr;
717     WCHAR notepadPathW[MAX_PATH];
718
719     memset(&wtd, 0, sizeof(wtd));
720     wtd.cbStruct = sizeof(wtd);
721     wtd.dwUIChoice = WTD_UI_NONE;
722     wtd.fdwRevocationChecks = WTD_REVOKE_WHOLECHAIN;
723     wtd.dwUnionChoice = WTD_CHOICE_FILE;
724     U(wtd).pFile = &file;
725     wtd.dwStateAction = WTD_STATEACTION_VERIFY;
726     memset(&file, 0, sizeof(file));
727     file.cbStruct = sizeof(file);
728     getNotepadPath(notepadPathW, MAX_PATH);
729     file.pcwszFilePath = notepadPathW;
730     r = WinVerifyTrust(INVALID_HANDLE_VALUE, &generic_action_v2, &wtd);
731     ok(r == TRUST_E_NOSIGNATURE || r == CRYPT_E_FILE_ERROR,
732      "expected TRUST_E_NOSIGNATURE or CRYPT_E_FILE_ERROR, got %08x\n", r);
733     hr = WinVerifyTrustEx(INVALID_HANDLE_VALUE, &generic_action_v2, &wtd);
734     ok(hr == TRUST_E_NOSIGNATURE || r == CRYPT_E_FILE_ERROR,
735      "expected TRUST_E_NOSIGNATURE or CRYPT_E_FILE_ERROR, got %08x\n", hr);
736 }
737
738 static void test_get_known_usages(void)
739 {
740     BOOL ret;
741     PCCRYPT_OID_INFO *usages;
742
743     if (!pWTHelperGetKnownUsages)
744     {
745         skip("missing WTHelperGetKnownUsages\n");
746         return;
747     }
748     SetLastError(0xdeadbeef);
749     ret = pWTHelperGetKnownUsages(0, NULL);
750     ok(!ret && GetLastError() == ERROR_INVALID_PARAMETER,
751      "expected ERROR_INVALID_PARAMETER, got %d\n", GetLastError());
752     SetLastError(0xdeadbeef);
753     ret = pWTHelperGetKnownUsages(1, NULL);
754     ok(!ret && GetLastError() == ERROR_INVALID_PARAMETER,
755      "expected ERROR_INVALID_PARAMETER, got %d\n", GetLastError());
756     SetLastError(0xdeadbeef);
757     ret = pWTHelperGetKnownUsages(0, &usages);
758     ok(!ret && GetLastError() == ERROR_INVALID_PARAMETER,
759      "expected ERROR_INVALID_PARAMETER, got %d\n", GetLastError());
760     /* A value of 1 for the first parameter seems to imply the value is
761      * allocated
762      */
763     SetLastError(0xdeadbeef);
764     usages = NULL;
765     ret = pWTHelperGetKnownUsages(1, &usages);
766     ok(ret, "WTHelperGetKnownUsages failed: %d\n", GetLastError());
767     ok(usages != NULL, "expected a pointer\n");
768     if (ret && usages)
769     {
770         PCCRYPT_OID_INFO *ptr;
771
772         /* The returned usages are an array of PCCRYPT_OID_INFOs, terminated with a
773          * NULL pointer.
774          */
775         for (ptr = usages; *ptr; ptr++)
776         {
777             ok((*ptr)->cbSize == sizeof(CRYPT_OID_INFO) ||
778              (*ptr)->cbSize == (sizeof(CRYPT_OID_INFO) + 2 * sizeof(LPCWSTR)), /* Vista */
779              "unexpected size %d\n", (*ptr)->cbSize);
780             /* Each returned usage is in the CRYPT_ENHKEY_USAGE_OID_GROUP_ID group */
781             ok((*ptr)->dwGroupId == CRYPT_ENHKEY_USAGE_OID_GROUP_ID,
782              "expected group CRYPT_ENHKEY_USAGE_OID_GROUP_ID, got %d\n",
783              (*ptr)->dwGroupId);
784         }
785     }
786     /* A value of 2 for the second parameter seems to imply the value is freed
787      */
788     SetLastError(0xdeadbeef);
789     ret = pWTHelperGetKnownUsages(2, &usages);
790     ok(ret, "WTHelperGetKnownUsages failed: %d\n", GetLastError());
791     ok(usages == NULL, "expected pointer to be cleared\n");
792     SetLastError(0xdeadbeef);
793     usages = NULL;
794     ret = pWTHelperGetKnownUsages(2, &usages);
795     ok(ret, "WTHelperGetKnownUsages failed: %d\n", GetLastError());
796     SetLastError(0xdeadbeef);
797     ret = pWTHelperGetKnownUsages(2, NULL);
798     ok(!ret && GetLastError() == ERROR_INVALID_PARAMETER,
799      "expected ERROR_INVALID_PARAMETER, got %d\n", GetLastError());
800 }
801
802 START_TEST(softpub)
803 {
804     InitFunctionPtrs();
805     test_provider_funcs();
806     test_sip_create_indirect_data();
807     test_wintrust();
808     test_get_known_usages();
809 }