git.oblomov.eu Git - wine/blob - 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     DWORD alg;
  91
  92     /* Test with a bogus one */
  93     SetLastError(0xdeadbeef);
  94     alg = CertOIDToAlgId("1.2.3");
  95     ok(!alg && GetLastError() == 0xdeadbeef,
  96      "Didn't expect last error (%08lx) to be set\n", GetLastError());
  97
  98     for (i = 0; i < sizeof(oidToAlgID) / sizeof(oidToAlgID[0]); i++)
  99     {
 100         alg = CertOIDToAlgId(oidToAlgID[i].oid);
 101         /* Not all Windows installations support all these, so make sure it's
 102          * at least not the wrong one.
 103          */
 104         ok(alg == 0 || alg == oidToAlgID[i].algID,
 105          "Expected %ld, got %ld\n", oidToAlgID[i].algID, alg);
 106     }
 107 }
 108
 109 static void testAlgIDToOID(void)
 110 {
 111     int i;
 112     LPCSTR oid;
 113
 114     /* Test with a bogus one */
 115     SetLastError(0xdeadbeef);
 116     oid = CertAlgIdToOID(ALG_CLASS_SIGNATURE | ALG_TYPE_ANY | 80);
 117     ok(!oid && GetLastError() == 0xdeadbeef,
 118      "Didn't expect last error (%08lx) to be set\n", GetLastError());
 119     for (i = 0; i < sizeof(algIDToOID) / sizeof(algIDToOID[0]); i++)
 120     {
 121         oid = CertAlgIdToOID(algIDToOID[i].algID);
 122         /* Allow failure, not every version of Windows supports every algo */
 123         if (oid)
 124             ok(!strcmp(oid, algIDToOID[i].oid),
 125              "Expected %s, got %s\n", algIDToOID[i].oid, oid);
 126     }
 127 }
 128
 129 static void test_findAttribute(void)
 130 {
 131     PCRYPT_ATTRIBUTE ret;
 132     BYTE blobbin[] = {0x02,0x01,0x01};
 133     CRYPT_ATTR_BLOB blobs[] = { { sizeof blobbin, blobbin }, };
 134     CRYPT_ATTRIBUTE attr = { "1.2.3", sizeof(blobs) / sizeof(blobs[0]), blobs };
 135
 136     /* returns NULL, last error not set */
 137     SetLastError(0xdeadbeef);
 138     ret = CertFindAttribute(NULL, 0, NULL);
 139     ok(ret == NULL, "Expected failure\n");
 140     ok(GetLastError() == 0xdeadbeef, "Last error was set to %08lx\n",
 141      GetLastError());
 142     /* crashes
 143     SetLastError(0xdeadbeef);
 144     ret = CertFindAttribute(NULL, 1, NULL);
 145      */
 146     /* returns NULL, last error is ERROR_INVALID_PARAMETER */
 147     SetLastError(0xdeadbeef);
 148     ret = CertFindAttribute(NULL, 1, &attr);
 149     ok(!ret && GetLastError() == ERROR_INVALID_PARAMETER,
 150      "Expected ERROR_INVALID_PARAMETER, got %ld (%08lx)\n", GetLastError(),
 151      GetLastError());
 152     /* returns NULL, last error not set */
 153     SetLastError(0xdeadbeef);
 154     ret = CertFindAttribute("bogus", 1, &attr);
 155     ok(ret == NULL, "Expected failure\n");
 156     ok(GetLastError() == 0xdeadbeef, "Last error was set to %08lx\n",
 157      GetLastError());
 158     /* returns NULL, last error not set */
 159     SetLastError(0xdeadbeef);
 160     ret = CertFindAttribute("1.2.4", 1, &attr);
 161     ok(ret == NULL, "Expected failure\n");
 162     ok(GetLastError() == 0xdeadbeef, "Last error was set to %08lx\n",
 163      GetLastError());
 164     /* succeeds, last error not set */
 165     SetLastError(0xdeadbeef);
 166     ret = CertFindAttribute("1.2.3", 1, &attr);
 167     ok(ret != NULL, "CertFindAttribute failed: %08lx\n", GetLastError());
 168 }
 169
 170 static void test_findExtension(void)
 171 {
 172     PCERT_EXTENSION ret;
 173     BYTE blobbin[] = {0x02,0x01,0x01};
 174     CERT_EXTENSION ext = { "1.2.3", TRUE, { sizeof blobbin, blobbin } };
 175
 176     /* returns NULL, last error not set */
 177     SetLastError(0xdeadbeef);
 178     ret = CertFindExtension(NULL, 0, NULL);
 179     ok(ret == NULL, "Expected failure\n");
 180     ok(GetLastError() == 0xdeadbeef, "Last error was set to %08lx\n",
 181      GetLastError());
 182     /* crashes
 183     SetLastError(0xdeadbeef);
 184     ret = CertFindExtension(NULL, 1, NULL);
 185      */
 186     /* returns NULL, last error is ERROR_INVALID_PARAMETER */
 187     SetLastError(0xdeadbeef);
 188     ret = CertFindExtension(NULL, 1, &ext);
 189     ok(!ret && GetLastError() == ERROR_INVALID_PARAMETER,
 190      "Expected ERROR_INVALID_PARAMETER, got %ld (%08lx)\n", GetLastError(),
 191      GetLastError());
 192     /* returns NULL, last error not set */
 193     SetLastError(0xdeadbeef);
 194     ret = CertFindExtension("bogus", 1, &ext);
 195     ok(ret == NULL, "Expected failure\n");
 196     ok(GetLastError() == 0xdeadbeef, "Last error was set to %08lx\n",
 197      GetLastError());
 198     /* returns NULL, last error not set */
 199     SetLastError(0xdeadbeef);
 200     ret = CertFindExtension("1.2.4", 1, &ext);
 201     ok(ret == NULL, "Expected failure\n");
 202     ok(GetLastError() == 0xdeadbeef, "Last error was set to %08lx\n",
 203      GetLastError());
 204     /* succeeds, last error not set */
 205     SetLastError(0xdeadbeef);
 206     ret = CertFindExtension("1.2.3", 1, &ext);
 207     ok(ret != NULL, "CertFindExtension failed: %08lx\n", GetLastError());
 208 }
 209
 210 static void test_findRDNAttr(void)
 211 {
 212     PCERT_RDN_ATTR ret;
 213     BYTE bin[] = { 0x16,0x09,'J','u','a','n',' ','L','a','n','g' };
 214     CERT_RDN_ATTR attrs[] = {
 215      { "1.2.3", CERT_RDN_IA5_STRING, { sizeof bin, bin } },
 216     };
 217     CERT_RDN rdns[] = {
 218      { sizeof(attrs) / sizeof(attrs[0]), attrs },
 219     };
 220     CERT_NAME_INFO nameInfo = { sizeof(rdns) / sizeof(rdns[0]), rdns };
 221
 222     /* crashes
 223     SetLastError(0xdeadbeef);
 224     ret = CertFindRDNAttr(NULL, NULL);
 225      */
 226     /* returns NULL, last error is ERROR_INVALID_PARAMETER */
 227     SetLastError(0xdeadbeef);
 228     ret = CertFindRDNAttr(NULL, &nameInfo);
 229     ok(!ret && GetLastError() == ERROR_INVALID_PARAMETER,
 230      "Expected ERROR_INVALID_PARAMETER, got %ld (%08lx)\n", GetLastError(),
 231      GetLastError());
 232     /* returns NULL, last error not set */
 233     SetLastError(0xdeadbeef);
 234     ret = CertFindRDNAttr("bogus", &nameInfo);
 235     ok(ret == NULL, "Expected failure\n");
 236     ok(GetLastError() == 0xdeadbeef, "Last error was set to %08lx\n",
 237      GetLastError());
 238     /* returns NULL, last error not set */
 239     SetLastError(0xdeadbeef);
 240     ret = CertFindRDNAttr("1.2.4", &nameInfo);
 241     ok(ret == NULL, "Expected failure\n");
 242     ok(GetLastError() == 0xdeadbeef, "Last error was set to %08lx\n",
 243      GetLastError());
 244     /* succeeds, last error not set */
 245     SetLastError(0xdeadbeef);
 246     ret = CertFindRDNAttr("1.2.3", &nameInfo);
 247     ok(ret != NULL, "CertFindRDNAttr failed: %08lx\n", GetLastError());
 248 }
 249
 250 static void test_verifyTimeValidity(void)
 251 {
 252     SYSTEMTIME sysTime;
 253     FILETIME fileTime;
 254     CERT_INFO info = { 0 };
 255     LONG ret;
 256
 257     GetSystemTime(&sysTime);
 258     SystemTimeToFileTime(&sysTime, &fileTime);
 259     /* crashes
 260     ret = CertVerifyTimeValidity(NULL, NULL);
 261     ret = CertVerifyTimeValidity(&fileTime, NULL);
 262      */
 263     /* Check with 0 NotBefore and NotAfter */
 264     ret = CertVerifyTimeValidity(&fileTime, &info);
 265     ok(ret == 1, "Expected 1, got %ld\n", ret);
 266     memcpy(&info.NotAfter, &fileTime, sizeof(info.NotAfter));
 267     /* Check with NotAfter equal to comparison time */
 268     ret = CertVerifyTimeValidity(&fileTime, &info);
 269     ok(ret == 0, "Expected 0, got %ld\n", ret);
 270     /* Check with NotBefore after comparison time */
 271     memcpy(&info.NotBefore, &fileTime, sizeof(info.NotBefore));
 272     info.NotBefore.dwLowDateTime += 5000;
 273     ret = CertVerifyTimeValidity(&fileTime, &info);
 274     ok(ret == -1, "Expected -1, got %ld\n", ret);
 275 }
 276
 277 static void test_cryptAllocate(void)
 278 {
 279     LPVOID buf;
 280
 281     buf = CryptMemAlloc(0);
 282     ok(buf != NULL, "CryptMemAlloc failed: %08lx\n", GetLastError());
 283     CryptMemFree(buf);
 284     buf = CryptMemRealloc(NULL, 0);
 285     ok(!buf, "Expected NULL\n");
 286     buf = CryptMemAlloc(0);
 287     buf = CryptMemRealloc(buf, 1);
 288     ok(buf != NULL, "CryptMemRealloc failed: %08lx\n", GetLastError());
 289     CryptMemFree(buf);
 290 }
 291
 292 typedef DWORD  (WINAPI *I_CryptAllocTlsFunc)(void);
 293 typedef LPVOID (WINAPI *I_CryptDetachTlsFunc)(DWORD dwTlsIndex);
 294 typedef LPVOID (WINAPI *I_CryptGetTlsFunc)(DWORD dwTlsIndex);
 295 typedef BOOL   (WINAPI *I_CryptSetTlsFunc)(DWORD dwTlsIndex, LPVOID lpTlsValue);
 296 typedef BOOL   (WINAPI *I_CryptFreeTlsFunc)(DWORD dwTlsIndex, DWORD unknown);
 297
 298 static I_CryptAllocTlsFunc pI_CryptAllocTls;
 299 static I_CryptDetachTlsFunc pI_CryptDetachTls;
 300 static I_CryptGetTlsFunc pI_CryptGetTls;
 301 static I_CryptSetTlsFunc pI_CryptSetTls;
 302 static I_CryptFreeTlsFunc pI_CryptFreeTls;
 303
 304 static void test_cryptTls(void)
 305 {
 306     HMODULE lib = LoadLibraryA("crypt32.dll");
 307
 308     if (lib)
 309     {
 310         DWORD index;
 311         BOOL ret;
 312
 313         pI_CryptAllocTls = (I_CryptAllocTlsFunc)GetProcAddress(lib,
 314          "I_CryptAllocTls");
 315         pI_CryptDetachTls = (I_CryptDetachTlsFunc)GetProcAddress(lib,
 316          "I_CryptDetachTls");
 317         pI_CryptGetTls = (I_CryptGetTlsFunc)GetProcAddress(lib,
 318          "I_CryptGetTls");
 319         pI_CryptSetTls = (I_CryptSetTlsFunc)GetProcAddress(lib,
 320          "I_CryptSetTls");
 321         pI_CryptFreeTls = (I_CryptFreeTlsFunc)GetProcAddress(lib,
 322          "I_CryptFreeTls");
 323
 324         /* One normal pass */
 325         index = pI_CryptAllocTls();
 326         ok(index, "I_CryptAllocTls failed: %08lx\n", GetLastError());
 327         if (index)
 328         {
 329             LPVOID ptr;
 330
 331             ptr = pI_CryptGetTls(index);
 332             ok(!ptr, "Expected NULL\n");
 333             ret = pI_CryptSetTls(index, (LPVOID)0xdeadbeef);
 334             ok(ret, "I_CryptSetTls failed: %08lx\n", GetLastError());
 335             ptr = pI_CryptGetTls(index);
 336             ok(ptr == (LPVOID)0xdeadbeef, "Expected 0xdeadbeef, got %p\n", ptr);
 337             /* This crashes
 338             ret = pI_CryptFreeTls(index, 1);
 339              */
 340             ret = pI_CryptFreeTls(index, 0);
 341             ok(ret, "I_CryptFreeTls failed: %08lx\n", GetLastError());
 342             ret = pI_CryptFreeTls(index, 0);
 343             /* Not sure if this fails because TlsFree should fail, so leave as
 344              * todo for now.
 345              */
 346             todo_wine ok(!ret && GetLastError() ==
 347              HRESULT_FROM_WIN32(ERROR_INVALID_PARAMETER),
 348              "Expected HRESULT_FROM_WIN32(ERROR_INVALID_PARAMETER), got %08lx\n",
 349              GetLastError());
 350         }
 351         /* Similar pass, check I_CryptDetachTls */
 352         index = pI_CryptAllocTls();
 353         ok(index, "I_CryptAllocTls failed: %08lx\n", GetLastError());
 354         if (index)
 355         {
 356             LPVOID ptr;
 357
 358             ptr = pI_CryptGetTls(index);
 359             ok(!ptr, "Expected NULL\n");
 360             ret = pI_CryptSetTls(index, (LPVOID)0xdeadbeef);
 361             ok(ret, "I_CryptSetTls failed: %08lx\n", GetLastError());
 362             ptr = pI_CryptGetTls(index);
 363             ok(ptr == (LPVOID)0xdeadbeef, "Expected 0xdeadbeef, got %p\n", ptr);
 364             ptr = pI_CryptDetachTls(index);
 365             ok(ptr == (LPVOID)0xdeadbeef, "Expected 0xdeadbeef, got %p\n", ptr);
 366             ptr = pI_CryptGetTls(index);
 367             ok(!ptr, "Expected NULL\n");
 368         }
 369         FreeLibrary(lib);
 370     }
 371 }
 372
 373 START_TEST(main)
 374 {
 375     testOIDToAlgID();
 376     testAlgIDToOID();
 377     test_findAttribute();
 378     test_findExtension();
 379     test_findRDNAttr();
 380     test_verifyTimeValidity();
 381     test_cryptAllocate();
 382     test_cryptTls();
 383 }