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