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     buf = CryptMemRealloc(NULL, 0);
 204     ok(!buf, "Expected NULL\n");
 205     buf = CryptMemAlloc(0);
 206     buf = CryptMemRealloc(buf, 1);
 207     ok(buf != NULL, "CryptMemRealloc failed: %08x\n", GetLastError());
 208     CryptMemFree(buf);
 209 }
 210
 211 typedef DWORD  (WINAPI *I_CryptAllocTlsFunc)(void);
 212 typedef LPVOID (WINAPI *I_CryptDetachTlsFunc)(DWORD dwTlsIndex);
 213 typedef LPVOID (WINAPI *I_CryptGetTlsFunc)(DWORD dwTlsIndex);
 214 typedef BOOL   (WINAPI *I_CryptSetTlsFunc)(DWORD dwTlsIndex, LPVOID lpTlsValue);
 215 typedef BOOL   (WINAPI *I_CryptFreeTlsFunc)(DWORD dwTlsIndex, DWORD unknown);
 216
 217 static I_CryptAllocTlsFunc pI_CryptAllocTls;
 218 static I_CryptDetachTlsFunc pI_CryptDetachTls;
 219 static I_CryptGetTlsFunc pI_CryptGetTls;
 220 static I_CryptSetTlsFunc pI_CryptSetTls;
 221 static I_CryptFreeTlsFunc pI_CryptFreeTls;
 222
 223 static void test_cryptTls(void)
 224 {
 225     DWORD index;
 226     BOOL ret;
 227
 228     if (!hCrypt) return;
 229
 230     pI_CryptAllocTls = (I_CryptAllocTlsFunc)GetProcAddress(hCrypt,
 231      "I_CryptAllocTls");
 232     pI_CryptDetachTls = (I_CryptDetachTlsFunc)GetProcAddress(hCrypt,
 233      "I_CryptDetachTls");
 234     pI_CryptGetTls = (I_CryptGetTlsFunc)GetProcAddress(hCrypt,
 235      "I_CryptGetTls");
 236     pI_CryptSetTls = (I_CryptSetTlsFunc)GetProcAddress(hCrypt,
 237      "I_CryptSetTls");
 238     pI_CryptFreeTls = (I_CryptFreeTlsFunc)GetProcAddress(hCrypt,
 239      "I_CryptFreeTls");
 240
 241     /* One normal pass */
 242     index = pI_CryptAllocTls();
 243     ok(index, "I_CryptAllocTls failed: %08x\n", GetLastError());
 244     if (index)
 245     {
 246         LPVOID ptr;
 247
 248         ptr = pI_CryptGetTls(index);
 249         ok(!ptr, "Expected NULL\n");
 250         ret = pI_CryptSetTls(index, (LPVOID)0xdeadbeef);
 251         ok(ret, "I_CryptSetTls failed: %08x\n", GetLastError());
 252         ptr = pI_CryptGetTls(index);
 253         ok(ptr == (LPVOID)0xdeadbeef, "Expected 0xdeadbeef, got %p\n", ptr);
 254         /* This crashes
 255         ret = pI_CryptFreeTls(index, 1);
 256          */
 257         ret = pI_CryptFreeTls(index, 0);
 258         ok(ret, "I_CryptFreeTls failed: %08x\n", GetLastError());
 259         ret = pI_CryptFreeTls(index, 0);
 260         /* Not sure if this fails because TlsFree should fail, so leave as
 261          * todo for now.
 262          */
 263         todo_wine ok(!ret && GetLastError() == E_INVALIDARG,
 264          "Expected E_INVALIDARG, got %08x\n", GetLastError());
 265     }
 266     /* Similar pass, check I_CryptDetachTls */
 267     index = pI_CryptAllocTls();
 268     ok(index, "I_CryptAllocTls failed: %08x\n", GetLastError());
 269     if (index)
 270     {
 271         LPVOID ptr;
 272
 273         ptr = pI_CryptGetTls(index);
 274         ok(!ptr, "Expected NULL\n");
 275         ret = pI_CryptSetTls(index, (LPVOID)0xdeadbeef);
 276         ok(ret, "I_CryptSetTls failed: %08x\n", GetLastError());
 277         ptr = pI_CryptGetTls(index);
 278         ok(ptr == (LPVOID)0xdeadbeef, "Expected 0xdeadbeef, got %p\n", ptr);
 279         ptr = pI_CryptDetachTls(index);
 280         ok(ptr == (LPVOID)0xdeadbeef, "Expected 0xdeadbeef, got %p\n", ptr);
 281         ptr = pI_CryptGetTls(index);
 282         ok(!ptr, "Expected NULL\n");
 283     }
 284 }
 285
 286 typedef BOOL (WINAPI *I_CryptReadTrustedPublisherDWORDValueFromRegistryFunc)
 287  (LPCWSTR, DWORD *);
 288
 289 static void test_readTrustedPublisherDWORD(void)
 290 {
 291     I_CryptReadTrustedPublisherDWORDValueFromRegistryFunc pReadDWORD;
 292
 293     if (!hCrypt) return;
 294
 295     pReadDWORD =
 296      (I_CryptReadTrustedPublisherDWORDValueFromRegistryFunc)GetProcAddress(
 297      hCrypt, "I_CryptReadTrustedPublisherDWORDValueFromRegistry");
 298     if (pReadDWORD)
 299     {
 300         static const WCHAR safer[] = {
 301          'S','o','f','t','w','a','r','e','\\',
 302          'P','o','l','i','c','i','e','s','\\',
 303          'M','i','c','r','o','s','o','f','t','\\','S','y','s','t','e','m',
 304          'C','e','r','t','i','f','i','c','a','t','e','s','\\',
 305          'T','r','u','s','t','e','d','P','u','b','l','i','s','h','e','r',
 306          '\\','S','a','f','e','r',0 };
 307         static const WCHAR authenticodeFlags[] = { 'A','u','t','h','e','n',
 308          't','i','c','o','d','e','F','l','a','g','s',0 };
 309         BOOL ret, exists = FALSE;
 310         DWORD size, readFlags = 0, returnedFlags;
 311         HKEY key;
 312         LONG rc;
 313
 314         rc = RegOpenKeyW(HKEY_LOCAL_MACHINE, safer, &key);
 315         if (rc == ERROR_SUCCESS)
 316         {
 317             size = sizeof(readFlags);
 318             rc = RegQueryValueExW(key, authenticodeFlags, NULL, NULL,
 319              (LPBYTE)&readFlags, &size);
 320             if (rc == ERROR_SUCCESS)
 321                 exists = TRUE;
 322         }
 323         returnedFlags = 0xdeadbeef;
 324         ret = pReadDWORD(authenticodeFlags, &returnedFlags);
 325         ok(ret == exists, "Unexpected return value\n");
 326         ok(readFlags == returnedFlags,
 327          "Expected flags %08x, got %08x\n", readFlags, returnedFlags);
 328     }
 329 }
 330
 331 typedef HCRYPTPROV (WINAPI *I_CryptGetDefaultCryptProvFunc)(DWORD w);
 332
 333 static void test_getDefaultCryptProv(void)
 334 {
 335     I_CryptGetDefaultCryptProvFunc pI_CryptGetDefaultCryptProv;
 336     HCRYPTPROV prov;
 337
 338     if (!hCrypt) return;
 339
 340     pI_CryptGetDefaultCryptProv = (I_CryptGetDefaultCryptProvFunc)
 341      GetProcAddress(hCrypt, "I_CryptGetDefaultCryptProv");
 342     if (!pI_CryptGetDefaultCryptProv) return;
 343
 344     prov = pI_CryptGetDefaultCryptProv(0xdeadbeef);
 345     ok(prov == 0 && GetLastError() == E_INVALIDARG,
 346      "Expected E_INVALIDARG, got %08x\n", GetLastError());
 347     prov = pI_CryptGetDefaultCryptProv(PROV_RSA_FULL);
 348     ok(prov == 0 && GetLastError() == E_INVALIDARG,
 349      "Expected E_INVALIDARG, got %08x\n", GetLastError());
 350     prov = pI_CryptGetDefaultCryptProv(1);
 351     ok(prov == 0 && GetLastError() == E_INVALIDARG,
 352      "Expected E_INVALIDARG, got %08x\n", GetLastError());
 353     prov = pI_CryptGetDefaultCryptProv(0);
 354     ok(prov != 0, "I_CryptGetDefaultCryptProv failed: %08x\n", GetLastError());
 355     CryptReleaseContext(prov, 0);
 356 }
 357
 358 typedef int (WINAPI *I_CryptInstallOssGlobal)(DWORD,DWORD,DWORD);
 359
 360 static void test_CryptInstallOssGlobal(void)
 361 {
 362     int ret,i;
 363     I_CryptInstallOssGlobal pI_CryptInstallOssGlobal;
 364
 365     if (!hCrypt) return;
 366
 367     pI_CryptInstallOssGlobal= (I_CryptInstallOssGlobal)GetProcAddress(hCrypt,"I_CryptInstallOssGlobal");
 368     /* passing in some random values to I_CryptInstallOssGlobal, it always returns 9 the first time, then 10, 11 etc.*/
 369     for(i=0;i<30;i++)
 370     {
 371       ret =  pI_CryptInstallOssGlobal(rand(),rand(),rand());
 372       ok((9+i) == ret,"Expected %d, got %d\n",(9+i),ret);
 373     }
 374 }
 375
 376 START_TEST(main)
 377 {
 378     hCrypt = LoadLibraryA("crypt32.dll");
 379     test_findAttribute();
 380     test_findExtension();
 381     test_findRDNAttr();
 382     test_verifyTimeValidity();
 383     test_cryptAllocate();
 384     test_cryptTls();
 385     test_readTrustedPublisherDWORD();
 386     test_getDefaultCryptProv();
 387     test_CryptInstallOssGlobal();
 388 }