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., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, 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 #include <winreg.h>
  28
  29 #include "wine/test.h"
  30
  31 HMODULE hCrypt;
  32
  33 static void test_findAttribute(void)
  34 {
  35     PCRYPT_ATTRIBUTE ret;
  36     BYTE blobbin[] = {0x02,0x01,0x01};
  37     static CHAR oid[] = "1.2.3";
  38     CRYPT_ATTR_BLOB blobs[] = { { sizeof blobbin, blobbin }, };
  39     CRYPT_ATTRIBUTE attr = { oid, sizeof(blobs) / sizeof(blobs[0]), blobs };
  40
  41     /* returns NULL, last error not set */
  42     SetLastError(0xdeadbeef);
  43     ret = CertFindAttribute(NULL, 0, NULL);
  44     ok(ret == NULL, "Expected failure\n");
  45     ok(GetLastError() == 0xdeadbeef, "Last error was set to %08x\n",
  46      GetLastError());
  47     if (0)
  48     {
  49         /* crashes */
  50         SetLastError(0xdeadbeef);
  51         ret = CertFindAttribute(NULL, 1, NULL);
  52         /* returns NULL, last error is ERROR_INVALID_PARAMETER
  53          * crashes on Vista
  54          */
  55         SetLastError(0xdeadbeef);
  56         ret = CertFindAttribute(NULL, 1, &attr);
  57         ok(!ret && GetLastError() == ERROR_INVALID_PARAMETER,
  58          "Expected ERROR_INVALID_PARAMETER, got %d (%08x)\n", GetLastError(),
  59          GetLastError());
  60     }
  61     /* returns NULL, last error not set */
  62     SetLastError(0xdeadbeef);
  63     ret = CertFindAttribute("bogus", 1, &attr);
  64     ok(ret == NULL, "Expected failure\n");
  65     ok(GetLastError() == 0xdeadbeef, "Last error was set to %08x\n",
  66      GetLastError());
  67     /* returns NULL, last error not set */
  68     SetLastError(0xdeadbeef);
  69     ret = CertFindAttribute("1.2.4", 1, &attr);
  70     ok(ret == NULL, "Expected failure\n");
  71     ok(GetLastError() == 0xdeadbeef, "Last error was set to %08x\n",
  72      GetLastError());
  73     /* succeeds, last error not set */
  74     SetLastError(0xdeadbeef);
  75     ret = CertFindAttribute("1.2.3", 1, &attr);
  76     ok(ret != NULL, "CertFindAttribute failed: %08x\n", GetLastError());
  77 }
  78
  79 static void test_findExtension(void)
  80 {
  81     PCERT_EXTENSION ret;
  82     static CHAR oid[] = "1.2.3";
  83     BYTE blobbin[] = {0x02,0x01,0x01};
  84     CERT_EXTENSION ext = { oid, TRUE, { sizeof blobbin, blobbin } };
  85
  86     /* returns NULL, last error not set */
  87     SetLastError(0xdeadbeef);
  88     ret = CertFindExtension(NULL, 0, NULL);
  89     ok(ret == NULL, "Expected failure\n");
  90     ok(GetLastError() == 0xdeadbeef, "Last error was set to %08x\n",
  91      GetLastError());
  92     if (0)
  93     {
  94         /* crashes */
  95         SetLastError(0xdeadbeef);
  96         ret = CertFindExtension(NULL, 1, NULL);
  97         /* returns NULL, last error is ERROR_INVALID_PARAMETER
  98          * crashes on Vista
  99          */
 100         SetLastError(0xdeadbeef);
 101         ret = CertFindExtension(NULL, 1, &ext);
 102         ok(!ret && GetLastError() == ERROR_INVALID_PARAMETER,
 103          "Expected ERROR_INVALID_PARAMETER, got %d (%08x)\n", GetLastError(),
 104          GetLastError());
 105     }
 106     /* returns NULL, last error not set */
 107     SetLastError(0xdeadbeef);
 108     ret = CertFindExtension("bogus", 1, &ext);
 109     ok(ret == NULL, "Expected failure\n");
 110     ok(GetLastError() == 0xdeadbeef, "Last error was set to %08x\n",
 111      GetLastError());
 112     /* returns NULL, last error not set */
 113     SetLastError(0xdeadbeef);
 114     ret = CertFindExtension("1.2.4", 1, &ext);
 115     ok(ret == NULL, "Expected failure\n");
 116     ok(GetLastError() == 0xdeadbeef, "Last error was set to %08x\n",
 117      GetLastError());
 118     /* succeeds, last error not set */
 119     SetLastError(0xdeadbeef);
 120     ret = CertFindExtension("1.2.3", 1, &ext);
 121     ok(ret != NULL, "CertFindExtension failed: %08x\n", GetLastError());
 122 }
 123
 124 static void test_findRDNAttr(void)
 125 {
 126     PCERT_RDN_ATTR ret;
 127     static CHAR oid[] = "1.2.3";
 128     BYTE bin[] = { 0x16,0x09,'J','u','a','n',' ','L','a','n','g' };
 129     CERT_RDN_ATTR attrs[] = {
 130      { oid, CERT_RDN_IA5_STRING, { sizeof bin, bin } },
 131     };
 132     CERT_RDN rdns[] = {
 133      { sizeof(attrs) / sizeof(attrs[0]), attrs },
 134     };
 135     CERT_NAME_INFO nameInfo = { sizeof(rdns) / sizeof(rdns[0]), rdns };
 136
 137     if (0)
 138     {
 139         /* crashes */
 140         SetLastError(0xdeadbeef);
 141         ret = CertFindRDNAttr(NULL, NULL);
 142         /* returns NULL, last error is ERROR_INVALID_PARAMETER
 143          * crashes on Vista
 144          */
 145         SetLastError(0xdeadbeef);
 146         ret = CertFindRDNAttr(NULL, &nameInfo);
 147         ok(!ret && GetLastError() == ERROR_INVALID_PARAMETER,
 148          "Expected ERROR_INVALID_PARAMETER, got %d (%08x)\n", GetLastError(),
 149          GetLastError());
 150     }
 151     /* returns NULL, last error not set */
 152     SetLastError(0xdeadbeef);
 153     ret = CertFindRDNAttr("bogus", &nameInfo);
 154     ok(ret == NULL, "Expected failure\n");
 155     ok(GetLastError() == 0xdeadbeef, "Last error was set to %08x\n",
 156      GetLastError());
 157     /* returns NULL, last error not set */
 158     SetLastError(0xdeadbeef);
 159     ret = CertFindRDNAttr("1.2.4", &nameInfo);
 160     ok(ret == NULL, "Expected failure\n");
 161     ok(GetLastError() == 0xdeadbeef, "Last error was set to %08x\n",
 162      GetLastError());
 163     /* succeeds, last error not set */
 164     SetLastError(0xdeadbeef);
 165     ret = CertFindRDNAttr("1.2.3", &nameInfo);
 166     ok(ret != NULL, "CertFindRDNAttr failed: %08x\n", GetLastError());
 167 }
 168
 169 static void test_verifyTimeValidity(void)
 170 {
 171     SYSTEMTIME sysTime;
 172     FILETIME fileTime;
 173     CERT_INFO info = { 0 };
 174     LONG ret;
 175
 176     GetSystemTime(&sysTime);
 177     SystemTimeToFileTime(&sysTime, &fileTime);
 178     /* crashes
 179     ret = CertVerifyTimeValidity(NULL, NULL);
 180     ret = CertVerifyTimeValidity(&fileTime, NULL);
 181      */
 182     /* Check with 0 NotBefore and NotAfter */
 183     ret = CertVerifyTimeValidity(&fileTime, &info);
 184     ok(ret == 1, "Expected 1, got %d\n", ret);
 185     memcpy(&info.NotAfter, &fileTime, sizeof(info.NotAfter));
 186     /* Check with NotAfter equal to comparison time */
 187     ret = CertVerifyTimeValidity(&fileTime, &info);
 188     ok(ret == 0, "Expected 0, got %d\n", ret);
 189     /* Check with NotBefore after comparison time */
 190     memcpy(&info.NotBefore, &fileTime, sizeof(info.NotBefore));
 191     info.NotBefore.dwLowDateTime += 5000;
 192     ret = CertVerifyTimeValidity(&fileTime, &info);
 193     ok(ret == -1, "Expected -1, got %d\n", ret);
 194 }
 195
 196 static void test_cryptAllocate(void)
 197 {
 198     LPVOID buf;
 199
 200     buf = CryptMemAlloc(0);
 201     ok(buf != NULL, "CryptMemAlloc failed: %08x\n", GetLastError());
 202     CryptMemFree(buf);
 203     /* CryptMemRealloc(NULL, 0) fails pre-Vista */
 204     buf = CryptMemAlloc(0);
 205     buf = CryptMemRealloc(buf, 1);
 206     ok(buf != NULL, "CryptMemRealloc failed: %08x\n", GetLastError());
 207     CryptMemFree(buf);
 208 }
 209
 210 typedef DWORD  (WINAPI *I_CryptAllocTlsFunc)(void);
 211 typedef LPVOID (WINAPI *I_CryptDetachTlsFunc)(DWORD dwTlsIndex);
 212 typedef LPVOID (WINAPI *I_CryptGetTlsFunc)(DWORD dwTlsIndex);
 213 typedef BOOL   (WINAPI *I_CryptSetTlsFunc)(DWORD dwTlsIndex, LPVOID lpTlsValue);
 214 typedef BOOL   (WINAPI *I_CryptFreeTlsFunc)(DWORD dwTlsIndex, DWORD unknown);
 215
 216 static I_CryptAllocTlsFunc pI_CryptAllocTls;
 217 static I_CryptDetachTlsFunc pI_CryptDetachTls;
 218 static I_CryptGetTlsFunc pI_CryptGetTls;
 219 static I_CryptSetTlsFunc pI_CryptSetTls;
 220 static I_CryptFreeTlsFunc pI_CryptFreeTls;
 221
 222 static void test_cryptTls(void)
 223 {
 224     DWORD index;
 225     BOOL ret;
 226
 227     pI_CryptAllocTls = (I_CryptAllocTlsFunc)GetProcAddress(hCrypt,
 228      "I_CryptAllocTls");
 229     pI_CryptDetachTls = (I_CryptDetachTlsFunc)GetProcAddress(hCrypt,
 230      "I_CryptDetachTls");
 231     pI_CryptGetTls = (I_CryptGetTlsFunc)GetProcAddress(hCrypt,
 232      "I_CryptGetTls");
 233     pI_CryptSetTls = (I_CryptSetTlsFunc)GetProcAddress(hCrypt,
 234      "I_CryptSetTls");
 235     pI_CryptFreeTls = (I_CryptFreeTlsFunc)GetProcAddress(hCrypt,
 236      "I_CryptFreeTls");
 237
 238     /* One normal pass */
 239     index = pI_CryptAllocTls();
 240     ok(index, "I_CryptAllocTls failed: %08x\n", GetLastError());
 241     if (index)
 242     {
 243         LPVOID ptr;
 244
 245         ptr = pI_CryptGetTls(index);
 246         ok(!ptr, "Expected NULL\n");
 247         ret = pI_CryptSetTls(index, (LPVOID)0xdeadbeef);
 248         ok(ret, "I_CryptSetTls failed: %08x\n", GetLastError());
 249         ptr = pI_CryptGetTls(index);
 250         ok(ptr == (LPVOID)0xdeadbeef, "Expected 0xdeadbeef, got %p\n", ptr);
 251         /* This crashes
 252         ret = pI_CryptFreeTls(index, 1);
 253          */
 254         ret = pI_CryptFreeTls(index, 0);
 255         ok(ret, "I_CryptFreeTls failed: %08x\n", GetLastError());
 256         ret = pI_CryptFreeTls(index, 0);
 257         /* Not sure if this fails because TlsFree should fail, so leave as
 258          * todo for now.
 259          */
 260         todo_wine ok(!ret && GetLastError() == E_INVALIDARG,
 261          "Expected E_INVALIDARG, got %08x\n", GetLastError());
 262     }
 263     /* Similar pass, check I_CryptDetachTls */
 264     index = pI_CryptAllocTls();
 265     ok(index, "I_CryptAllocTls failed: %08x\n", GetLastError());
 266     if (index)
 267     {
 268         LPVOID ptr;
 269
 270         ptr = pI_CryptGetTls(index);
 271         ok(!ptr, "Expected NULL\n");
 272         ret = pI_CryptSetTls(index, (LPVOID)0xdeadbeef);
 273         ok(ret, "I_CryptSetTls failed: %08x\n", GetLastError());
 274         ptr = pI_CryptGetTls(index);
 275         ok(ptr == (LPVOID)0xdeadbeef, "Expected 0xdeadbeef, got %p\n", ptr);
 276         ptr = pI_CryptDetachTls(index);
 277         ok(ptr == (LPVOID)0xdeadbeef, "Expected 0xdeadbeef, got %p\n", ptr);
 278         ptr = pI_CryptGetTls(index);
 279         ok(!ptr, "Expected NULL\n");
 280     }
 281 }
 282
 283 typedef BOOL (WINAPI *I_CryptReadTrustedPublisherDWORDValueFromRegistryFunc)
 284  (LPCWSTR, DWORD *);
 285
 286 static void test_readTrustedPublisherDWORD(void)
 287 {
 288     I_CryptReadTrustedPublisherDWORDValueFromRegistryFunc pReadDWORD;
 289
 290     pReadDWORD =
 291      (I_CryptReadTrustedPublisherDWORDValueFromRegistryFunc)GetProcAddress(
 292      hCrypt, "I_CryptReadTrustedPublisherDWORDValueFromRegistry");
 293     if (pReadDWORD)
 294     {
 295         static const WCHAR safer[] = {
 296          'S','o','f','t','w','a','r','e','\\',
 297          'P','o','l','i','c','i','e','s','\\',
 298          'M','i','c','r','o','s','o','f','t','\\','S','y','s','t','e','m',
 299          'C','e','r','t','i','f','i','c','a','t','e','s','\\',
 300          'T','r','u','s','t','e','d','P','u','b','l','i','s','h','e','r',
 301          '\\','S','a','f','e','r',0 };
 302         static const WCHAR authenticodeFlags[] = { 'A','u','t','h','e','n',
 303          't','i','c','o','d','e','F','l','a','g','s',0 };
 304         BOOL ret, exists = FALSE;
 305         DWORD size, readFlags = 0, returnedFlags;
 306         HKEY key;
 307         LONG rc;
 308
 309         rc = RegOpenKeyW(HKEY_LOCAL_MACHINE, safer, &key);
 310         if (rc == ERROR_SUCCESS)
 311         {
 312             size = sizeof(readFlags);
 313             rc = RegQueryValueExW(key, authenticodeFlags, NULL, NULL,
 314              (LPBYTE)&readFlags, &size);
 315             if (rc == ERROR_SUCCESS)
 316                 exists = TRUE;
 317         }
 318         returnedFlags = 0xdeadbeef;
 319         ret = pReadDWORD(authenticodeFlags, &returnedFlags);
 320         ok(ret == exists, "Unexpected return value\n");
 321         ok(readFlags == returnedFlags,
 322          "Expected flags %08x, got %08x\n", readFlags, returnedFlags);
 323     }
 324 }
 325
 326 typedef HCRYPTPROV (WINAPI *I_CryptGetDefaultCryptProvFunc)(DWORD w);
 327
 328 static void test_getDefaultCryptProv(void)
 329 {
 330     I_CryptGetDefaultCryptProvFunc pI_CryptGetDefaultCryptProv;
 331     HCRYPTPROV prov;
 332
 333     pI_CryptGetDefaultCryptProv = (I_CryptGetDefaultCryptProvFunc)
 334      GetProcAddress(hCrypt, "I_CryptGetDefaultCryptProv");
 335     if (!pI_CryptGetDefaultCryptProv) return;
 336
 337     prov = pI_CryptGetDefaultCryptProv(0xdeadbeef);
 338     ok(prov == 0 && GetLastError() == E_INVALIDARG,
 339      "Expected E_INVALIDARG, got %08x\n", GetLastError());
 340     prov = pI_CryptGetDefaultCryptProv(PROV_RSA_FULL);
 341     ok(prov == 0 && GetLastError() == E_INVALIDARG,
 342      "Expected E_INVALIDARG, got %08x\n", GetLastError());
 343     prov = pI_CryptGetDefaultCryptProv(1);
 344     ok(prov == 0 && GetLastError() == E_INVALIDARG,
 345      "Expected E_INVALIDARG, got %08x\n", GetLastError());
 346     prov = pI_CryptGetDefaultCryptProv(0);
 347     ok(prov != 0, "I_CryptGetDefaultCryptProv failed: %08x\n", GetLastError());
 348     CryptReleaseContext(prov, 0);
 349 }
 350
 351 typedef int (WINAPI *I_CryptInstallOssGlobal)(DWORD,DWORD,DWORD);
 352
 353 static void test_CryptInstallOssGlobal(void)
 354 {
 355     int ret,i;
 356     I_CryptInstallOssGlobal pI_CryptInstallOssGlobal;
 357
 358     pI_CryptInstallOssGlobal= (I_CryptInstallOssGlobal)GetProcAddress(hCrypt,"I_CryptInstallOssGlobal");
 359     /* passing in some random values to I_CryptInstallOssGlobal, it always returns 9 the first time, then 10, 11 etc.*/
 360     for(i=0;i<30;i++)
 361     {
 362       ret =  pI_CryptInstallOssGlobal(rand(),rand(),rand());
 363       ok((9+i) == ret ||
 364          ret == 0, /* Vista */
 365          "Expected %d or 0, got %d\n",(9+i),ret);
 366     }
 367 }
 368
 369 START_TEST(main)
 370 {
 371     hCrypt = GetModuleHandleA("crypt32.dll");
 372
 373     test_findAttribute();
 374     test_findExtension();
 375     test_findRDNAttr();
 376     test_verifyTimeValidity();
 377     test_cryptAllocate();
 378     test_cryptTls();
 379     test_readTrustedPublisherDWORD();
 380     test_getDefaultCryptProv();
 381     test_CryptInstallOssGlobal();
 382 }