Implement a few utility functions, with tests.
[wine] / dlls / crypt32 / tests / main.c
1 /*
2  * Miscellaneous crypt32 tests
3  *
4  * Copyright 2005 Juan Lang
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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
19  */
20
21 #include <stdio.h>
22 #include <stdarg.h>
23 #include <windef.h>
24 #include <winbase.h>
25 #include <winerror.h>
26 #include <wincrypt.h>
27
28 #include "wine/test.h"
29
30 struct OIDToAlgID
31 {
32     LPCSTR oid;
33     DWORD algID;
34 };
35
36 static const struct OIDToAlgID oidToAlgID[] = {
37  { szOID_RSA_RSA, CALG_RSA_KEYX },
38  { szOID_RSA_MD2RSA, CALG_MD2 },
39  { szOID_RSA_MD4RSA, CALG_MD4 },
40  { szOID_RSA_MD5RSA, CALG_MD5 },
41  { szOID_RSA_SHA1RSA, CALG_SHA },
42  { szOID_RSA_DH, CALG_DH_SF },
43  { szOID_RSA_SMIMEalgESDH, CALG_DH_EPHEM },
44  { szOID_RSA_SMIMEalgCMS3DESwrap, CALG_3DES },
45  { szOID_RSA_SMIMEalgCMSRC2wrap, CALG_RC2 },
46  { szOID_RSA_MD2, CALG_MD2 },
47  { szOID_RSA_MD4, CALG_MD4 },
48  { szOID_RSA_MD5, CALG_MD5 },
49  { szOID_RSA_RC2CBC, CALG_RC2 },
50  { szOID_RSA_RC4, CALG_RC4 },
51  { szOID_RSA_DES_EDE3_CBC, CALG_3DES },
52  { szOID_ANSI_X942_DH, CALG_DH_SF },
53  { szOID_X957_DSA, CALG_DSS_SIGN },
54  { szOID_X957_SHA1DSA, CALG_SHA },
55  { szOID_OIWSEC_md4RSA, CALG_MD4 },
56  { szOID_OIWSEC_md5RSA, CALG_MD5 },
57  { szOID_OIWSEC_md4RSA2, CALG_MD4 },
58  { szOID_OIWSEC_desCBC, CALG_DES },
59  { szOID_OIWSEC_dsa, CALG_DSS_SIGN },
60  { szOID_OIWSEC_shaDSA, CALG_SHA },
61  { szOID_OIWSEC_shaRSA, CALG_SHA },
62  { szOID_OIWSEC_sha, CALG_SHA },
63  { szOID_OIWSEC_rsaXchg, CALG_RSA_KEYX },
64  { szOID_OIWSEC_sha1, CALG_SHA },
65  { szOID_OIWSEC_dsaSHA1, CALG_SHA },
66  { szOID_OIWSEC_sha1RSASign, CALG_SHA },
67  { szOID_OIWDIR_md2RSA, CALG_MD2 },
68  { szOID_INFOSEC_mosaicUpdatedSig, CALG_SHA },
69  { szOID_INFOSEC_mosaicKMandUpdSig, CALG_DSS_SIGN },
70 };
71
72 static const struct OIDToAlgID algIDToOID[] = {
73  { szOID_RSA_RSA, CALG_RSA_KEYX },
74  { szOID_RSA_SMIMEalgESDH, CALG_DH_EPHEM },
75  { szOID_RSA_MD2, CALG_MD2 },
76  { szOID_RSA_MD4, CALG_MD4 },
77  { szOID_RSA_MD5, CALG_MD5 },
78  { szOID_RSA_RC2CBC, CALG_RC2 },
79  { szOID_RSA_RC4, CALG_RC4 },
80  { szOID_RSA_DES_EDE3_CBC, CALG_3DES },
81  { szOID_ANSI_X942_DH, CALG_DH_SF },
82  { szOID_X957_DSA, CALG_DSS_SIGN },
83  { szOID_OIWSEC_desCBC, CALG_DES },
84  { szOID_OIWSEC_sha1, CALG_SHA },
85 };
86
87 static void testOIDToAlgID(void)
88 {
89     int i;
90
91     for (i = 0; i < sizeof(oidToAlgID) / sizeof(oidToAlgID[0]); i++)
92     {
93         DWORD alg = CertOIDToAlgId(oidToAlgID[i].oid);
94
95         ok(alg == oidToAlgID[i].algID,
96          "Expected %ld, got %ld\n", oidToAlgID[i].algID, alg);
97     }
98 }
99
100 static void testAlgIDToOID(void)
101 {
102     int i;
103
104     for (i = 0; i < sizeof(algIDToOID) / sizeof(algIDToOID[0]); i++)
105     {
106         LPCSTR oid = CertAlgIdToOID(algIDToOID[i].algID);
107
108         if ((!oid || !algIDToOID[i].oid) && oid != algIDToOID[i].oid &&
109          algIDToOID[i].algID)
110             printf("Expected %s, got %s\n", algIDToOID[i].oid, oid);
111         else if (oid && algIDToOID[i].oid && strcmp(oid, algIDToOID[i].oid))
112             printf("Expected %s, got %s\n", algIDToOID[i].oid, oid);
113     }
114 }
115
116 static void test_findAttribute(void)
117 {
118     PCRYPT_ATTRIBUTE ret;
119     CRYPT_ATTR_BLOB blobs[] = {
120      { 3, "\x02\x01\x01" },
121     };
122     CRYPT_ATTRIBUTE attr = { "1.2.3", sizeof(blobs) / sizeof(blobs[0]), blobs };
123
124     /* returns NULL, last error not set */
125     SetLastError(0xdeadbeef);
126     ret = CertFindAttribute(NULL, 0, NULL);
127     ok(ret == NULL, "Expected failure\n");
128     ok(GetLastError() == 0xdeadbeef, "Last error was set to %08lx\n",
129      GetLastError());
130     /* crashes
131     SetLastError(0xdeadbeef);
132     ret = CertFindAttribute(NULL, 1, NULL);
133      */
134     /* returns NULL, last error is ERROR_INVALID_PARAMETER */
135     SetLastError(0xdeadbeef);
136     ret = CertFindAttribute(NULL, 1, &attr);
137     ok(!ret && GetLastError() == ERROR_INVALID_PARAMETER,
138      "Expected ERROR_INVALID_PARAMETER, got %ld (%08lx)\n", GetLastError(),
139      GetLastError());
140     /* returns NULL, last error not set */
141     SetLastError(0xdeadbeef);
142     ret = CertFindAttribute("bogus", 1, &attr);
143     ok(ret == NULL, "Expected failure\n");
144     ok(GetLastError() == 0xdeadbeef, "Last error was set to %08lx\n",
145      GetLastError());
146     /* returns NULL, last error not set */
147     SetLastError(0xdeadbeef);
148     ret = CertFindAttribute("1.2.4", 1, &attr);
149     ok(ret == NULL, "Expected failure\n");
150     ok(GetLastError() == 0xdeadbeef, "Last error was set to %08lx\n",
151      GetLastError());
152     /* succeeds, last error not set */
153     SetLastError(0xdeadbeef);
154     ret = CertFindAttribute("1.2.3", 1, &attr);
155     ok(ret != NULL, "CertFindAttribute failed: %08lx\n", GetLastError());
156 }
157
158 static void test_findExtension(void)
159 {
160     PCERT_EXTENSION ret;
161     CERT_EXTENSION ext = { "1.2.3", TRUE, { 3, "\x02\x01\x01" } };
162
163     /* returns NULL, last error not set */
164     SetLastError(0xdeadbeef);
165     ret = CertFindExtension(NULL, 0, NULL);
166     ok(ret == NULL, "Expected failure\n");
167     ok(GetLastError() == 0xdeadbeef, "Last error was set to %08lx\n",
168      GetLastError());
169     /* crashes
170     SetLastError(0xdeadbeef);
171     ret = CertFindExtension(NULL, 1, NULL);
172      */
173     /* returns NULL, last error is ERROR_INVALID_PARAMETER */
174     SetLastError(0xdeadbeef);
175     ret = CertFindExtension(NULL, 1, &ext);
176     ok(!ret && GetLastError() == ERROR_INVALID_PARAMETER,
177      "Expected ERROR_INVALID_PARAMETER, got %ld (%08lx)\n", GetLastError(),
178      GetLastError());
179     /* returns NULL, last error not set */
180     SetLastError(0xdeadbeef);
181     ret = CertFindExtension("bogus", 1, &ext);
182     ok(ret == NULL, "Expected failure\n");
183     ok(GetLastError() == 0xdeadbeef, "Last error was set to %08lx\n",
184      GetLastError());
185     /* returns NULL, last error not set */
186     SetLastError(0xdeadbeef);
187     ret = CertFindExtension("1.2.4", 1, &ext);
188     ok(ret == NULL, "Expected failure\n");
189     ok(GetLastError() == 0xdeadbeef, "Last error was set to %08lx\n",
190      GetLastError());
191     /* succeeds, last error not set */
192     SetLastError(0xdeadbeef);
193     ret = CertFindExtension("1.2.3", 1, &ext);
194     ok(ret != NULL, "CertFindExtension failed: %08lx\n", GetLastError());
195 }
196
197 static void test_findRDNAttr(void)
198 {
199     PCERT_RDN_ATTR ret;
200     CERT_RDN_ATTR attrs[] = {
201      { "1.2.3", CERT_RDN_IA5_STRING, { 11, "\x16\x09Juan Lang" } },
202     };
203     CERT_RDN rdns[] = {
204      { sizeof(attrs) / sizeof(attrs[0]), attrs },
205     };
206     CERT_NAME_INFO nameInfo = { sizeof(rdns) / sizeof(rdns[0]), rdns };
207
208     /* crashes
209     SetLastError(0xdeadbeef);
210     ret = CertFindRDNAttr(NULL, NULL);
211      */
212     /* returns NULL, last error is ERROR_INVALID_PARAMETER */
213     SetLastError(0xdeadbeef);
214     ret = CertFindRDNAttr(NULL, &nameInfo);
215     ok(!ret && GetLastError() == ERROR_INVALID_PARAMETER,
216      "Expected ERROR_INVALID_PARAMETER, got %ld (%08lx)\n", GetLastError(),
217      GetLastError());
218     /* returns NULL, last error not set */
219     SetLastError(0xdeadbeef);
220     ret = CertFindRDNAttr("bogus", &nameInfo);
221     ok(ret == NULL, "Expected failure\n");
222     ok(GetLastError() == 0xdeadbeef, "Last error was set to %08lx\n",
223      GetLastError());
224     /* returns NULL, last error not set */
225     SetLastError(0xdeadbeef);
226     ret = CertFindRDNAttr("1.2.4", &nameInfo);
227     ok(ret == NULL, "Expected failure\n");
228     ok(GetLastError() == 0xdeadbeef, "Last error was set to %08lx\n",
229      GetLastError());
230     /* succeeds, last error not set */
231     SetLastError(0xdeadbeef);
232     ret = CertFindRDNAttr("1.2.3", &nameInfo);
233     ok(ret != NULL, "CertFindRDNAttr failed: %08lx\n", GetLastError());
234 }
235
236 static void test_verifyTimeValidity(void)
237 {
238     SYSTEMTIME sysTime;
239     FILETIME fileTime;
240     CERT_INFO info = { 0 };
241     LONG ret;
242
243     GetSystemTime(&sysTime);
244     SystemTimeToFileTime(&sysTime, &fileTime);
245     /* crashes
246     ret = CertVerifyTimeValidity(NULL, NULL);
247     ret = CertVerifyTimeValidity(&fileTime, NULL);
248      */
249     /* Check with 0 NotBefore and NotAfter */
250     ret = CertVerifyTimeValidity(&fileTime, &info);
251     ok(ret == 1, "Expected 1, got %ld\n", ret);
252     memcpy(&info.NotAfter, &fileTime, sizeof(info.NotAfter));
253     /* Check with NotAfter equal to comparison time */
254     ret = CertVerifyTimeValidity(&fileTime, &info);
255     ok(ret == 0, "Expected 0, got %ld\n", ret);
256     /* Check with NotBefore after comparison time */
257     memcpy(&info.NotBefore, &fileTime, sizeof(info.NotBefore));
258     info.NotBefore.dwLowDateTime += 5000;
259     ret = CertVerifyTimeValidity(&fileTime, &info);
260     ok(ret == -1, "Expected -1, got %ld\n", ret);
261 }
262
263 START_TEST(main)
264 {
265     testOIDToAlgID();
266     testAlgIDToOID();
267     test_findAttribute();
268     test_findExtension();
269     test_findRDNAttr();
270     test_verifyTimeValidity();
271 }