git.oblomov.eu Git - wine/blob - dlls/crypt32/tests/store.c

   1 /*
   2  * crypt32 cert store function tests
   3  *
   4  * Copyright 2005-2006 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 <assert.h>
  22 #include <stdio.h>
  23 #include <stdarg.h>
  24 #include <windef.h>
  25 #include <winbase.h>
  26 #include <winreg.h>
  27 #include <winerror.h>
  28 #include <wincrypt.h>
  29
  30 #include "wine/test.h"
  31
  32 /* The following aren't defined in wincrypt.h, as they're "reserved" */
  33 #define CERT_CERT_PROP_ID 32
  34 #define CERT_CRL_PROP_ID  33
  35 #define CERT_CTL_PROP_ID  34
  36
  37 struct CertPropIDHeader
  38 {
  39     DWORD propID;
  40     DWORD unknown1;
  41     DWORD cb;
  42 };
  43
  44 static const BYTE emptyCert[] = { 0x30, 0x00 };
  45 static const BYTE bigCert[] = { 0x30, 0x7a, 0x02, 0x01, 0x01, 0x30, 0x02, 0x06,
  46  0x00, 0x30, 0x15, 0x31, 0x13, 0x30, 0x11, 0x06, 0x03, 0x55, 0x04, 0x03, 0x13,
  47  0x0a, 0x4a, 0x75, 0x61, 0x6e, 0x20, 0x4c, 0x61, 0x6e, 0x67, 0x00, 0x30, 0x22,
  48  0x18, 0x0f, 0x31, 0x36, 0x30, 0x31, 0x30, 0x31, 0x30, 0x31, 0x30, 0x30, 0x30,
  49  0x30, 0x30, 0x30, 0x5a, 0x18, 0x0f, 0x31, 0x36, 0x30, 0x31, 0x30, 0x31, 0x30,
  50  0x31, 0x30, 0x30, 0x30, 0x30, 0x30, 0x30, 0x5a, 0x30, 0x15, 0x31, 0x13, 0x30,
  51  0x11, 0x06, 0x03, 0x55, 0x04, 0x03, 0x13, 0x0a, 0x4a, 0x75, 0x61, 0x6e, 0x20,
  52  0x4c, 0x61, 0x6e, 0x67, 0x00, 0x30, 0x07, 0x30, 0x02, 0x06, 0x00, 0x03, 0x01,
  53  0x00, 0xa3, 0x16, 0x30, 0x14, 0x30, 0x12, 0x06, 0x03, 0x55, 0x1d, 0x13, 0x01,
  54  0x01, 0xff, 0x04, 0x08, 0x30, 0x06, 0x01, 0x01, 0xff, 0x02, 0x01, 0x01 };
  55 static const BYTE signedBigCert[] = {
  56  0x30, 0x81, 0x93, 0x30, 0x7a, 0x02, 0x01, 0x01, 0x30, 0x02, 0x06, 0x00, 0x30,
  57  0x15, 0x31, 0x13, 0x30, 0x11, 0x06, 0x03, 0x55, 0x04, 0x03, 0x13, 0x0a, 0x4a,
  58  0x75, 0x61, 0x6e, 0x20, 0x4c, 0x61, 0x6e, 0x67, 0x00, 0x30, 0x22, 0x18, 0x0f,
  59  0x31, 0x36, 0x30, 0x31, 0x30, 0x31, 0x30, 0x31, 0x30, 0x30, 0x30, 0x30, 0x30,
  60  0x30, 0x5a, 0x18, 0x0f, 0x31, 0x36, 0x30, 0x31, 0x30, 0x31, 0x30, 0x31, 0x30,
  61  0x30, 0x30, 0x30, 0x30, 0x30, 0x5a, 0x30, 0x15, 0x31, 0x13, 0x30, 0x11, 0x06,
  62  0x03, 0x55, 0x04, 0x03, 0x13, 0x0a, 0x4a, 0x75, 0x61, 0x6e, 0x20, 0x4c, 0x61,
  63  0x6e, 0x67, 0x00, 0x30, 0x07, 0x30, 0x02, 0x06, 0x00, 0x03, 0x01, 0x00, 0xa3,
  64  0x16, 0x30, 0x14, 0x30, 0x12, 0x06, 0x03, 0x55, 0x1d, 0x13, 0x01, 0x01, 0xff,
  65  0x04, 0x08, 0x30, 0x06, 0x01, 0x01, 0xff, 0x02, 0x01, 0x01, 0x30, 0x02, 0x06,
  66  0x00, 0x03, 0x11, 0x00, 0x0f, 0x0e, 0x0d, 0x0c, 0x0b, 0x0a, 0x09, 0x08, 0x07,
  67  0x06, 0x05, 0x04, 0x03, 0x02, 0x01, 0x00 };
  68 static const BYTE serializedCert[] = { 0x20, 0x00, 0x00, 0x00,
  69  0x01, 0x00, 0x00, 0x00, 0x7c, 0x00, 0x00, 0x00, 0x30, 0x7a, 0x02, 0x01, 0x01,
  70  0x30, 0x02, 0x06, 0x00, 0x30, 0x15, 0x31, 0x13, 0x30, 0x11, 0x06, 0x03, 0x55,
  71  0x04, 0x03, 0x13, 0x0a, 0x4a, 0x75, 0x61, 0x6e, 0x20, 0x4c, 0x61, 0x6e, 0x67,
  72  0x00, 0x30, 0x22, 0x18, 0x0f, 0x31, 0x36, 0x30, 0x31, 0x30, 0x31, 0x30, 0x31,
  73  0x30, 0x30, 0x30, 0x30, 0x30, 0x30, 0x5a, 0x18, 0x0f, 0x31, 0x36, 0x30, 0x31,
  74  0x30, 0x31, 0x30, 0x31, 0x30, 0x30, 0x30, 0x30, 0x30, 0x30, 0x5a, 0x30, 0x15,
  75  0x31, 0x13, 0x30, 0x11, 0x06, 0x03, 0x55, 0x04, 0x03, 0x13, 0x0a, 0x4a, 0x75,
  76  0x61, 0x6e, 0x20, 0x4c, 0x61, 0x6e, 0x67, 0x00, 0x30, 0x07, 0x30, 0x02, 0x06,
  77  0x00, 0x03, 0x01, 0x00, 0xa3, 0x16, 0x30, 0x14, 0x30, 0x12, 0x06, 0x03, 0x55,
  78  0x1d, 0x13, 0x01, 0x01, 0xff, 0x04, 0x08, 0x30, 0x06, 0x01, 0x01, 0xff, 0x02,
  79  0x01, 0x01 };
  80 static const BYTE signedCRL[] = { 0x30, 0x45, 0x30, 0x2c, 0x30, 0x02, 0x06,
  81  0x00, 0x30, 0x15, 0x31, 0x13, 0x30, 0x11, 0x06, 0x03, 0x55, 0x04, 0x03, 0x13,
  82  0x0a, 0x4a, 0x75, 0x61, 0x6e, 0x20, 0x4c, 0x61, 0x6e, 0x67, 0x00, 0x18, 0x0f,
  83  0x31, 0x36, 0x30, 0x31, 0x30, 0x31, 0x30, 0x31, 0x30, 0x30, 0x30, 0x30, 0x30,
  84  0x30, 0x5a, 0x30, 0x02, 0x06, 0x00, 0x03, 0x11, 0x00, 0x0f, 0x0e, 0x0d, 0x0c,
  85  0x0b, 0x0a, 0x09, 0x08, 0x07, 0x06, 0x05, 0x04, 0x03, 0x02, 0x01, 0x00 };
  86 static const BYTE bigCert2[] = { 0x30, 0x7a, 0x02, 0x01, 0x01, 0x30, 0x02, 0x06,
  87  0x00, 0x30, 0x15, 0x31, 0x13, 0x30, 0x11, 0x06, 0x03, 0x55, 0x04, 0x03, 0x13,
  88  0x0a, 0x41, 0x6c, 0x65, 0x78, 0x20, 0x4c, 0x61, 0x6e, 0x67, 0x00, 0x30, 0x22,
  89  0x18, 0x0f, 0x31, 0x36, 0x30, 0x31, 0x30, 0x31, 0x30, 0x31, 0x30, 0x30, 0x30,
  90  0x30, 0x30, 0x30, 0x5a, 0x18, 0x0f, 0x31, 0x36, 0x30, 0x31, 0x30, 0x31, 0x30,
  91  0x31, 0x30, 0x30, 0x30, 0x30, 0x30, 0x30, 0x5a, 0x30, 0x15, 0x31, 0x13, 0x30,
  92  0x11, 0x06, 0x03, 0x55, 0x04, 0x03, 0x13, 0x0a, 0x41, 0x6c, 0x65, 0x78, 0x20,
  93  0x4c, 0x61, 0x6e, 0x67, 0x00, 0x30, 0x07, 0x30, 0x02, 0x06, 0x00, 0x03, 0x01,
  94  0x00, 0xa3, 0x16, 0x30, 0x14, 0x30, 0x12, 0x06, 0x03, 0x55, 0x1d, 0x13, 0x01,
  95  0x01, 0xff, 0x04, 0x08, 0x30, 0x06, 0x01, 0x01, 0xff, 0x02, 0x01, 0x01 };
  96
  97
  98 static BOOL (WINAPI *pCertAddStoreToCollection)(HCERTSTORE,HCERTSTORE,DWORD,DWORD);
  99 static BOOL (WINAPI *pCertControlStore)(HCERTSTORE,DWORD,DWORD,void const*);
 100 static PCCRL_CONTEXT (WINAPI *pCertEnumCRLsInStore)(HCERTSTORE,PCCRL_CONTEXT);
 101 static BOOL (WINAPI *pCertEnumSystemStore)(DWORD,void*,void*,PFN_CERT_ENUM_SYSTEM_STORE);
 102 static BOOL (WINAPI *pCertGetStoreProperty)(HCERTSTORE,DWORD,void*,DWORD*);
 103 static void (WINAPI *pCertRemoveStoreFromCollection)(HCERTSTORE,HCERTSTORE);
 104 static BOOL (WINAPI *pCertSetStoreProperty)(HCERTSTORE,DWORD,DWORD,const void*);
 105 static BOOL (WINAPI *pCertAddCertificateLinkToStore)(HCERTSTORE,PCCERT_CONTEXT,DWORD,PCCERT_CONTEXT*);
 106
 107 static void testMemStore(void)
 108 {
 109     HCERTSTORE store1, store2;
 110     PCCERT_CONTEXT context;
 111     BOOL ret;
 112     DWORD GLE;
 113
 114     /* NULL provider */
 115     store1 = CertOpenStore(0, 0, 0, 0, NULL);
 116     ok(!store1 && GetLastError() == ERROR_FILE_NOT_FOUND,
 117      "Expected ERROR_FILE_NOT_FOUND, got %d\n", GetLastError());
 118     /* weird flags */
 119     store1 = CertOpenStore(CERT_STORE_PROV_MEMORY, 0, 0,
 120      CERT_STORE_DELETE_FLAG, NULL);
 121     ok(!store1 && GetLastError() == ERROR_CALL_NOT_IMPLEMENTED,
 122      "Expected ERROR_CALL_NOT_IMPLEMENTED, got %d\n", GetLastError());
 123
 124     /* normal */
 125     store1 = CertOpenStore(CERT_STORE_PROV_MEMORY, 0, 0,
 126      CERT_STORE_CREATE_NEW_FLAG, NULL);
 127     ok(store1 != NULL, "CertOpenStore failed: %d\n", GetLastError());
 128     /* open existing doesn't */
 129     store2 = CertOpenStore(CERT_STORE_PROV_MEMORY, 0, 0,
 130      CERT_STORE_OPEN_EXISTING_FLAG, NULL);
 131     ok(store2 != NULL, "CertOpenStore failed: %d\n", GetLastError());
 132     ok(store1 != store2, "Expected different stores\n");
 133
 134     /* add a bogus (empty) cert */
 135     context = NULL;
 136     ret = CertAddEncodedCertificateToStore(store1, X509_ASN_ENCODING, emptyCert,
 137      sizeof(emptyCert), CERT_STORE_ADD_ALWAYS, &context);
 138     /* Windows returns CRYPT_E_ASN1_EOD or OSS_DATA_ERROR, but accept
 139      * CRYPT_E_ASN1_CORRUPT as well (because matching errors is tough in this
 140      * case)
 141      */
 142     GLE = GetLastError();
 143     ok(!ret && (GLE == CRYPT_E_ASN1_EOD || GLE == CRYPT_E_ASN1_CORRUPT ||
 144      GLE == OSS_DATA_ERROR),
 145      "Expected CRYPT_E_ASN1_EOD or CRYPT_E_ASN1_CORRUPT or OSS_DATA_ERROR, got %08x\n",
 146      GLE);
 147     /* add a "signed" cert--the signature isn't a real signature, so this adds
 148      * without any check of the signature's validity
 149      */
 150     ret = CertAddEncodedCertificateToStore(store1, X509_ASN_ENCODING,
 151      signedBigCert, sizeof(signedBigCert), CERT_STORE_ADD_ALWAYS, &context);
 152     ok(ret, "CertAddEncodedCertificateToStore failed: %08x\n", GetLastError());
 153     ok(context != NULL, "Expected a valid cert context\n");
 154     if (context)
 155     {
 156         ok(context->cbCertEncoded == sizeof(signedBigCert),
 157          "Wrong cert size %d\n", context->cbCertEncoded);
 158         ok(!memcmp(context->pbCertEncoded, signedBigCert,
 159          sizeof(signedBigCert)), "Unexpected encoded cert in context\n");
 160         /* remove it, the rest of the tests will work on an unsigned cert */
 161         ret = CertDeleteCertificateFromStore(context);
 162         ok(ret, "CertDeleteCertificateFromStore failed: %08x\n",
 163          GetLastError());
 164     }
 165     /* try adding a "signed" CRL as a cert */
 166     ret = CertAddEncodedCertificateToStore(store1, X509_ASN_ENCODING,
 167      signedCRL, sizeof(signedCRL), CERT_STORE_ADD_ALWAYS, &context);
 168     GLE = GetLastError();
 169     ok(!ret && (GLE == CRYPT_E_ASN1_BADTAG || GLE == CRYPT_E_ASN1_CORRUPT ||
 170      GLE == OSS_DATA_ERROR),
 171      "Expected CRYPT_E_ASN1_BADTAG or CRYPT_E_ASN1_CORRUPT or OSS_DATA_ERROR, got %08x\n",
 172      GLE);
 173     /* add a cert to store1 */
 174     ret = CertAddEncodedCertificateToStore(store1, X509_ASN_ENCODING, bigCert,
 175      sizeof(bigCert), CERT_STORE_ADD_ALWAYS, &context);
 176     ok(ret, "CertAddEncodedCertificateToStore failed: %08x\n", GetLastError());
 177     ok(context != NULL, "Expected a valid cert context\n");
 178     if (context)
 179     {
 180         DWORD size;
 181         BYTE *buf;
 182
 183         ok(context->cbCertEncoded == sizeof(bigCert),
 184          "Wrong cert size %d\n", context->cbCertEncoded);
 185         ok(!memcmp(context->pbCertEncoded, bigCert, sizeof(bigCert)),
 186          "Unexpected encoded cert in context\n");
 187         ok(context->hCertStore == store1, "Unexpected store\n");
 188
 189         /* check serializing this element */
 190         /* These crash
 191         ret = CertSerializeCertificateStoreElement(NULL, 0, NULL, NULL);
 192         ret = CertSerializeCertificateStoreElement(context, 0, NULL, NULL);
 193         ret = CertSerializeCertificateStoreElement(NULL, 0, NULL, &size);
 194          */
 195         /* apparently flags are ignored */
 196         ret = CertSerializeCertificateStoreElement(context, 1, NULL, &size);
 197         ok(ret, "CertSerializeCertificateStoreElement failed: %08x\n",
 198          GetLastError());
 199         buf = HeapAlloc(GetProcessHeap(), 0, size);
 200         if (buf)
 201         {
 202             ret = CertSerializeCertificateStoreElement(context, 0, buf, &size);
 203             ok(size == sizeof(serializedCert), "Wrong size %d\n", size);
 204             ok(!memcmp(serializedCert, buf, size),
 205              "Unexpected serialized cert\n");
 206             HeapFree(GetProcessHeap(), 0, buf);
 207         }
 208
 209         ret = CertFreeCertificateContext(context);
 210         ok(ret, "CertFreeCertificateContext failed: %08x\n", GetLastError());
 211     }
 212     /* verify the cert's in store1 */
 213     context = CertEnumCertificatesInStore(store1, NULL);
 214     ok(context != NULL, "Expected a valid context\n");
 215     context = CertEnumCertificatesInStore(store1, context);
 216     ok(!context && GetLastError() == CRYPT_E_NOT_FOUND,
 217      "Expected CRYPT_E_NOT_FOUND, got %08x\n", GetLastError());
 218     /* verify store2 (the "open existing" mem store) is still empty */
 219     context = CertEnumCertificatesInStore(store2, NULL);
 220     ok(!context, "Expected an empty store\n");
 221     /* delete the cert from store1, and check it's empty */
 222     context = CertEnumCertificatesInStore(store1, NULL);
 223     if (context)
 224     {
 225         /* Deleting a bitwise copy crashes with an access to an uninitialized
 226          * pointer, so a cert context has some special data out there in memory
 227          * someplace
 228         CERT_CONTEXT copy;
 229         memcpy(&copy, context, sizeof(copy));
 230         ret = CertDeleteCertificateFromStore(&copy);
 231          */
 232         PCCERT_CONTEXT copy = CertDuplicateCertificateContext(context);
 233
 234         ok(copy != NULL, "CertDuplicateCertificateContext failed: %08x\n",
 235          GetLastError());
 236         ret = CertDeleteCertificateFromStore(context);
 237         ok(ret, "CertDeleteCertificateFromStore failed: %08x\n",
 238          GetLastError());
 239         /* try deleting a copy */
 240         ret = CertDeleteCertificateFromStore(copy);
 241         ok(ret, "CertDeleteCertificateFromStore failed: %08x\n",
 242          GetLastError());
 243         /* check that the store is empty */
 244         context = CertEnumCertificatesInStore(store1, NULL);
 245         ok(!context, "Expected an empty store\n");
 246     }
 247
 248     /* close an empty store */
 249     ret = CertCloseStore(NULL, 0);
 250     ok(ret, "CertCloseStore failed: %d\n", GetLastError());
 251     ret = CertCloseStore(store1, 0);
 252     ok(ret, "CertCloseStore failed: %d\n", GetLastError());
 253     ret = CertCloseStore(store2, 0);
 254     ok(ret, "CertCloseStore failed: %d\n", GetLastError());
 255
 256     /* This seems nonsensical, but you can open a read-only mem store, only
 257      * it isn't read-only
 258      */
 259     store1 = CertOpenStore(CERT_STORE_PROV_MEMORY, 0, 0,
 260      CERT_STORE_READONLY_FLAG, NULL);
 261     ok(store1 != NULL, "CertOpenStore failed: %d\n", GetLastError());
 262     /* yep, this succeeds */
 263     ret = CertAddEncodedCertificateToStore(store1, X509_ASN_ENCODING, bigCert,
 264      sizeof(bigCert), CERT_STORE_ADD_ALWAYS, &context);
 265     ok(ret, "CertAddEncodedCertificateToStore failed: %08x\n", GetLastError());
 266     ok(context != NULL, "Expected a valid cert context\n");
 267     if (context)
 268     {
 269         ok(context->cbCertEncoded == sizeof(bigCert),
 270          "Wrong cert size %d\n", context->cbCertEncoded);
 271         ok(!memcmp(context->pbCertEncoded, bigCert, sizeof(bigCert)),
 272          "Unexpected encoded cert in context\n");
 273         ok(context->hCertStore == store1, "Unexpected store\n");
 274         ret = CertDeleteCertificateFromStore(context);
 275         ok(ret, "CertDeleteCertificateFromStore failed: %08x\n",
 276          GetLastError());
 277     }
 278     CertCloseStore(store1, 0);
 279 }
 280
 281 static void compareStore(HCERTSTORE store, LPCSTR name, const BYTE *pb,
 282  DWORD cb, BOOL todo)
 283 {
 284     BOOL ret;
 285     CRYPT_DATA_BLOB blob = { 0, NULL };
 286
 287     ret = CertSaveStore(store, X509_ASN_ENCODING, CERT_STORE_SAVE_AS_STORE,
 288      CERT_STORE_SAVE_TO_MEMORY, &blob, 0);
 289     ok(ret, "CertSaveStore failed: %08x\n", GetLastError());
 290     if (todo)
 291         todo_wine
 292         ok(blob.cbData == cb, "%s: expected size %d, got %d\n", name, cb,
 293          blob.cbData);
 294     else
 295         ok(blob.cbData == cb, "%s: expected size %d, got %d\n", name, cb,
 296          blob.cbData);
 297     blob.pbData = HeapAlloc(GetProcessHeap(), 0, blob.cbData);
 298     if (blob.pbData)
 299     {
 300         ret = CertSaveStore(store, X509_ASN_ENCODING, CERT_STORE_SAVE_AS_STORE,
 301          CERT_STORE_SAVE_TO_MEMORY, &blob, 0);
 302         ok(ret, "CertSaveStore failed: %08x\n", GetLastError());
 303         if (todo)
 304             todo_wine
 305             ok(!memcmp(pb, blob.pbData, cb), "%s: unexpected value\n", name);
 306         else
 307             ok(!memcmp(pb, blob.pbData, cb), "%s: unexpected value\n", name);
 308         HeapFree(GetProcessHeap(), 0, blob.pbData);
 309     }
 310 }
 311
 312 static const BYTE serializedStoreWithCert[] = {
 313  0x00,0x00,0x00,0x00,0x43,0x45,0x52,0x54,0x20,0x00,0x00,0x00,0x01,0x00,0x00,
 314  0x00,0x7c,0x00,0x00,0x00,0x30,0x7a,0x02,0x01,0x01,0x30,0x02,0x06,0x00,0x30,
 315  0x15,0x31,0x13,0x30,0x11,0x06,0x03,0x55,0x04,0x03,0x13,0x0a,0x4a,0x75,0x61,
 316  0x6e,0x20,0x4c,0x61,0x6e,0x67,0x00,0x30,0x22,0x18,0x0f,0x31,0x36,0x30,0x31,
 317  0x30,0x31,0x30,0x31,0x30,0x30,0x30,0x30,0x30,0x30,0x5a,0x18,0x0f,0x31,0x36,
 318  0x30,0x31,0x30,0x31,0x30,0x31,0x30,0x30,0x30,0x30,0x30,0x30,0x5a,0x30,0x15,
 319  0x31,0x13,0x30,0x11,0x06,0x03,0x55,0x04,0x03,0x13,0x0a,0x4a,0x75,0x61,0x6e,
 320  0x20,0x4c,0x61,0x6e,0x67,0x00,0x30,0x07,0x30,0x02,0x06,0x00,0x03,0x01,0x00,
 321  0xa3,0x16,0x30,0x14,0x30,0x12,0x06,0x03,0x55,0x1d,0x13,0x01,0x01,0xff,0x04,
 322  0x08,0x30,0x06,0x01,0x01,0xff,0x02,0x01,0x01,0x00,0x00,0x00,0x00,0x00,0x00,
 323  0x00,0x00,0x00,0x00,0x00,0x00 };
 324
 325 static void testCollectionStore(void)
 326 {
 327     HCERTSTORE store1, store2, collection, collection2;
 328     PCCERT_CONTEXT context;
 329     BOOL ret;
 330     static const WCHAR szPrefix[] = { 'c','e','r',0 };
 331     static const WCHAR szDot[] = { '.',0 };
 332     WCHAR filename[MAX_PATH];
 333     HANDLE file;
 334
 335     if (!pCertAddStoreToCollection)
 336     {
 337         win_skip("CertAddStoreToCollection() is not available\n");
 338         return;
 339     }
 340
 341     collection = CertOpenStore(CERT_STORE_PROV_COLLECTION, 0, 0,
 342      CERT_STORE_CREATE_NEW_FLAG, NULL);
 343
 344     /* Try adding a cert to any empty collection */
 345     ret = CertAddEncodedCertificateToStore(collection, X509_ASN_ENCODING,
 346      bigCert, sizeof(bigCert), CERT_STORE_ADD_ALWAYS, NULL);
 347     ok(!ret && GetLastError() == E_ACCESSDENIED,
 348      "Expected E_ACCESSDENIED, got %08x\n", GetLastError());
 349
 350     /* Create and add a cert to a memory store */
 351     store1 = CertOpenStore(CERT_STORE_PROV_MEMORY, 0, 0,
 352      CERT_STORE_CREATE_NEW_FLAG, NULL);
 353     ret = CertAddEncodedCertificateToStore(store1, X509_ASN_ENCODING,
 354      bigCert, sizeof(bigCert), CERT_STORE_ADD_ALWAYS, NULL);
 355     ok(ret, "CertAddEncodedCertificateToStore failed: %08x\n", GetLastError());
 356     /* Add the memory store to the collection, without allowing adding */
 357     ret = pCertAddStoreToCollection(collection, store1, 0, 0);
 358     ok(ret, "CertAddStoreToCollection failed: %08x\n", GetLastError());
 359     /* Verify the cert is in the collection */
 360     context = CertEnumCertificatesInStore(collection, NULL);
 361     ok(context != NULL, "Expected a valid context\n");
 362     if (context)
 363     {
 364         ok(context->hCertStore == collection, "Unexpected store\n");
 365         CertFreeCertificateContext(context);
 366     }
 367     /* Check that adding to the collection isn't allowed */
 368     ret = CertAddEncodedCertificateToStore(collection, X509_ASN_ENCODING,
 369      bigCert2, sizeof(bigCert2), CERT_STORE_ADD_ALWAYS, NULL);
 370     ok(!ret && GetLastError() == E_ACCESSDENIED,
 371      "Expected E_ACCESSDENIED, got %08x\n", GetLastError());
 372
 373     /* Create a new memory store */
 374     store2 = CertOpenStore(CERT_STORE_PROV_MEMORY, 0, 0,
 375      CERT_STORE_CREATE_NEW_FLAG, NULL);
 376     /* Try adding a store to a non-collection store */
 377     ret = pCertAddStoreToCollection(store1, store2,
 378      CERT_PHYSICAL_STORE_ADD_ENABLE_FLAG, 0);
 379     ok(!ret && GetLastError() == E_INVALIDARG,
 380      "Expected E_INVALIDARG, got %08x\n", GetLastError());
 381     /* Try adding some bogus stores */
 382     /* This crashes in Windows
 383     ret = pCertAddStoreToCollection(0, store2,
 384      CERT_PHYSICAL_STORE_ADD_ENABLE_FLAG, 0);
 385      */
 386     /* This "succeeds"... */
 387     ret = pCertAddStoreToCollection(collection, 0,
 388      CERT_PHYSICAL_STORE_ADD_ENABLE_FLAG, 0);
 389     ok(ret, "CertAddStoreToCollection failed: %08x\n", GetLastError());
 390     /* while this crashes.
 391     ret = pCertAddStoreToCollection(collection, 1,
 392      CERT_PHYSICAL_STORE_ADD_ENABLE_FLAG, 0);
 393      */
 394
 395     /* Add it to the collection, this time allowing adding */
 396     ret = pCertAddStoreToCollection(collection, store2,
 397      CERT_PHYSICAL_STORE_ADD_ENABLE_FLAG, 0);
 398     ok(ret, "CertAddStoreToCollection failed: %08x\n", GetLastError());
 399     /* Check that adding to the collection is allowed */
 400     ret = CertAddEncodedCertificateToStore(collection, X509_ASN_ENCODING,
 401      bigCert2, sizeof(bigCert2), CERT_STORE_ADD_ALWAYS, NULL);
 402     ok(ret, "CertAddEncodedCertificateToStore failed: %08x\n", GetLastError());
 403     /* Now check that it was actually added to store2 */
 404     context = CertEnumCertificatesInStore(store2, NULL);
 405     ok(context != NULL, "Expected a valid context\n");
 406     if (context)
 407     {
 408         ok(context->hCertStore == store2, "Unexpected store\n");
 409         CertFreeCertificateContext(context);
 410     }
 411     /* Check that the collection has both bigCert and bigCert2.  bigCert comes
 412      * first because store1 was added first.
 413      */
 414     context = CertEnumCertificatesInStore(collection, NULL);
 415     ok(context != NULL, "Expected a valid context\n");
 416     if (context)
 417     {
 418         ok(context->hCertStore == collection, "Unexpected store\n");
 419         ok(context->cbCertEncoded == sizeof(bigCert),
 420          "Wrong size %d\n", context->cbCertEncoded);
 421         ok(!memcmp(context->pbCertEncoded, bigCert, context->cbCertEncoded),
 422          "Unexpected cert\n");
 423         context = CertEnumCertificatesInStore(collection, context);
 424         ok(context != NULL, "Expected a valid context\n");
 425         if (context)
 426         {
 427             ok(context->hCertStore == collection, "Unexpected store\n");
 428             ok(context->cbCertEncoded == sizeof(bigCert2),
 429              "Wrong size %d\n", context->cbCertEncoded);
 430             ok(!memcmp(context->pbCertEncoded, bigCert2,
 431              context->cbCertEncoded), "Unexpected cert\n");
 432             context = CertEnumCertificatesInStore(collection, context);
 433             ok(!context, "Unexpected cert\n");
 434         }
 435     }
 436     /* close store2, and check that the collection is unmodified */
 437     CertCloseStore(store2, 0);
 438     context = CertEnumCertificatesInStore(collection, NULL);
 439     ok(context != NULL, "Expected a valid context\n");
 440     if (context)
 441     {
 442         ok(context->hCertStore == collection, "Unexpected store\n");
 443         ok(context->cbCertEncoded == sizeof(bigCert),
 444          "Wrong size %d\n", context->cbCertEncoded);
 445         ok(!memcmp(context->pbCertEncoded, bigCert, context->cbCertEncoded),
 446          "Unexpected cert\n");
 447         context = CertEnumCertificatesInStore(collection, context);
 448         ok(context != NULL, "Expected a valid context\n");
 449         if (context)
 450         {
 451             ok(context->hCertStore == collection, "Unexpected store\n");
 452             ok(context->cbCertEncoded == sizeof(bigCert2),
 453              "Wrong size %d\n", context->cbCertEncoded);
 454             ok(!memcmp(context->pbCertEncoded, bigCert2,
 455              context->cbCertEncoded), "Unexpected cert\n");
 456             context = CertEnumCertificatesInStore(collection, context);
 457             ok(!context, "Unexpected cert\n");
 458         }
 459     }
 460
 461     /* Adding a collection to a collection is legal */
 462     collection2 = CertOpenStore(CERT_STORE_PROV_COLLECTION, 0, 0,
 463      CERT_STORE_CREATE_NEW_FLAG, NULL);
 464     ret = pCertAddStoreToCollection(collection2, collection,
 465      CERT_PHYSICAL_STORE_ADD_ENABLE_FLAG, 0);
 466     ok(ret, "CertAddStoreToCollection failed: %08x\n", GetLastError());
 467     /* check the contents of collection2 */
 468     context = CertEnumCertificatesInStore(collection2, NULL);
 469     ok(context != NULL, "Expected a valid context\n");
 470     if (context)
 471     {
 472         ok(context->hCertStore == collection2, "Unexpected store\n");
 473         ok(context->cbCertEncoded == sizeof(bigCert),
 474          "Wrong size %d\n", context->cbCertEncoded);
 475         ok(!memcmp(context->pbCertEncoded, bigCert, context->cbCertEncoded),
 476          "Unexpected cert\n");
 477         context = CertEnumCertificatesInStore(collection2, context);
 478         ok(context != NULL, "Expected a valid context\n");
 479         if (context)
 480         {
 481             ok(context->hCertStore == collection2, "Unexpected store\n");
 482             ok(context->cbCertEncoded == sizeof(bigCert2),
 483              "Wrong size %d\n", context->cbCertEncoded);
 484             ok(!memcmp(context->pbCertEncoded, bigCert2,
 485              context->cbCertEncoded), "Unexpected cert\n");
 486             context = CertEnumCertificatesInStore(collection2, context);
 487             ok(!context, "Unexpected cert\n");
 488         }
 489     }
 490
 491     /* I'd like to test closing the collection in the middle of enumeration,
 492      * but my tests have been inconsistent.  The first time calling
 493      * CertEnumCertificatesInStore on a closed collection succeeded, while the
 494      * second crashed.  So anything appears to be fair game.
 495      * I'd also like to test removing a store from a collection in the middle
 496      * of an enumeration, but my tests in Windows have been inconclusive.
 497      * In one scenario it worked.  In another scenario, about a third of the
 498      * time this leads to "random" crashes elsewhere in the code.  This
 499      * probably means this is not allowed.
 500      */
 501
 502     CertCloseStore(store1, 0);
 503     CertCloseStore(collection, 0);
 504     CertCloseStore(collection2, 0);
 505
 506     /* Add the same cert to two memory stores, then put them in a collection */
 507     store1 = CertOpenStore(CERT_STORE_PROV_MEMORY, 0, 0,
 508      CERT_STORE_CREATE_NEW_FLAG, NULL);
 509     ok(store1 != 0, "CertOpenStore failed: %08x\n", GetLastError());
 510     store2 = CertOpenStore(CERT_STORE_PROV_MEMORY, 0, 0,
 511      CERT_STORE_CREATE_NEW_FLAG, NULL);
 512     ok(store2 != 0, "CertOpenStore failed: %08x\n", GetLastError());
 513
 514     ret = CertAddEncodedCertificateToStore(store1, X509_ASN_ENCODING,
 515      bigCert, sizeof(bigCert), CERT_STORE_ADD_ALWAYS, NULL);
 516     ok(ret, "CertAddEncodedCertificateToStore failed: %08x\n", GetLastError());
 517     ret = CertAddEncodedCertificateToStore(store2, X509_ASN_ENCODING,
 518      bigCert, sizeof(bigCert), CERT_STORE_ADD_ALWAYS, NULL);
 519     ok(ret, "CertAddEncodedCertificateToStore failed: %08x\n", GetLastError());
 520     collection = CertOpenStore(CERT_STORE_PROV_COLLECTION, 0, 0,
 521      CERT_STORE_CREATE_NEW_FLAG, NULL);
 522     ok(collection != 0, "CertOpenStore failed: %08x\n", GetLastError());
 523
 524     ret = pCertAddStoreToCollection(collection, store1,
 525      CERT_PHYSICAL_STORE_ADD_ENABLE_FLAG, 0);
 526     ok(ret, "CertAddStoreToCollection failed: %08x\n", GetLastError());
 527     ret = pCertAddStoreToCollection(collection, store2,
 528      CERT_PHYSICAL_STORE_ADD_ENABLE_FLAG, 0);
 529     ok(ret, "CertAddStoreToCollection failed: %08x\n", GetLastError());
 530
 531     /* Check that the collection has two copies of the same cert */
 532     context = CertEnumCertificatesInStore(collection, NULL);
 533     ok(context != NULL, "Expected a valid context\n");
 534     if (context)
 535     {
 536         ok(context->hCertStore == collection, "Unexpected store\n");
 537         ok(context->cbCertEncoded == sizeof(bigCert),
 538          "Wrong size %d\n", context->cbCertEncoded);
 539         ok(!memcmp(context->pbCertEncoded, bigCert, context->cbCertEncoded),
 540          "Unexpected cert\n");
 541         context = CertEnumCertificatesInStore(collection, context);
 542         ok(context != NULL, "Expected a valid context\n");
 543         if (context)
 544         {
 545             ok(context->hCertStore == collection, "Unexpected store\n");
 546             ok(context->cbCertEncoded == sizeof(bigCert),
 547              "Wrong size %d\n", context->cbCertEncoded);
 548             ok(!memcmp(context->pbCertEncoded, bigCert, context->cbCertEncoded),
 549              "Unexpected cert\n");
 550             context = CertEnumCertificatesInStore(collection, context);
 551             ok(context == NULL, "Unexpected cert\n");
 552         }
 553     }
 554
 555     /* The following would check whether I can delete an identical cert, rather
 556      * than one enumerated from the store.  It crashes, so that means I must
 557      * only call CertDeleteCertificateFromStore with contexts enumerated from
 558      * the store.
 559     context = CertCreateCertificateContext(X509_ASN_ENCODING, bigCert,
 560      sizeof(bigCert));
 561     ok(context != NULL, "CertCreateCertificateContext failed: %08x\n",
 562      GetLastError());
 563     if (context)
 564     {
 565         ret = CertDeleteCertificateFromStore(collection, context);
 566         printf("ret is %d, GetLastError is %08x\n", ret, GetLastError());
 567         CertFreeCertificateContext(context);
 568     }
 569      */
 570
 571     /* Now check deleting from the collection. */
 572     context = CertEnumCertificatesInStore(collection, NULL);
 573     ok(context != NULL, "Expected a valid context\n");
 574     if (context)
 575     {
 576         CertDeleteCertificateFromStore(context);
 577         /* store1 should now be empty */
 578         context = CertEnumCertificatesInStore(store1, NULL);
 579         ok(!context, "Unexpected cert\n");
 580         /* and there should be one certificate in the collection */
 581         context = CertEnumCertificatesInStore(collection, NULL);
 582         ok(context != NULL, "Expected a valid cert\n");
 583         if (context)
 584         {
 585             ok(context->hCertStore == collection, "Unexpected store\n");
 586             ok(context->cbCertEncoded == sizeof(bigCert),
 587              "Wrong size %d\n", context->cbCertEncoded);
 588             ok(!memcmp(context->pbCertEncoded, bigCert, context->cbCertEncoded),
 589              "Unexpected cert\n");
 590         }
 591         context = CertEnumCertificatesInStore(collection, context);
 592         ok(context == NULL, "Unexpected cert\n");
 593     }
 594
 595     if (!pCertRemoveStoreFromCollection)
 596     {
 597         win_skip("CertRemoveStoreFromCollection() is not available\n");
 598     }
 599     else
 600     {
 601         /* Finally, test removing stores from the collection.  No return
 602          *  value, so it's a bit funny to test.
 603          */
 604         /* This crashes
 605          * pCertRemoveStoreFromCollection(NULL, NULL);
 606          */
 607         /* This "succeeds," no crash, no last error set */
 608         SetLastError(0xdeadbeef);
 609         pCertRemoveStoreFromCollection(store2, collection);
 610         ok(GetLastError() == 0xdeadbeef,
 611            "Didn't expect an error to be set: %08x\n", GetLastError());
 612
 613         /* After removing store2, the collection should be empty */
 614         SetLastError(0xdeadbeef);
 615         pCertRemoveStoreFromCollection(collection, store2);
 616         ok(GetLastError() == 0xdeadbeef,
 617            "Didn't expect an error to be set: %08x\n", GetLastError());
 618         context = CertEnumCertificatesInStore(collection, NULL);
 619         ok(!context, "Unexpected cert\n");
 620     }
 621
 622     CertCloseStore(collection, 0);
 623     CertCloseStore(store2, 0);
 624     CertCloseStore(store1, 0);
 625
 626     /* Test adding certificates to and deleting certificates from collections.
 627      */
 628     store1 = CertOpenSystemStoreA(0, "My");
 629     collection = CertOpenStore(CERT_STORE_PROV_COLLECTION, 0, 0,
 630      CERT_STORE_CREATE_NEW_FLAG, NULL);
 631
 632     ret = CertAddEncodedCertificateToStore(store1, X509_ASN_ENCODING,
 633      bigCert, sizeof(bigCert), CERT_STORE_ADD_ALWAYS, &context);
 634     ok(ret, "CertAddEncodedCertificateToStore failed: %08x\n", GetLastError());
 635     CertDeleteCertificateFromStore(context);
 636
 637     pCertAddStoreToCollection(collection, store1,
 638      CERT_PHYSICAL_STORE_ADD_ENABLE_FLAG, 0);
 639
 640     ret = CertAddEncodedCertificateToStore(collection, X509_ASN_ENCODING,
 641      bigCert, sizeof(bigCert), CERT_STORE_ADD_ALWAYS, &context);
 642     ok(ret, "CertAddEncodedCertificateToStore failed: %08x\n", GetLastError());
 643     CertDeleteCertificateFromStore(context);
 644
 645     CertCloseStore(collection, 0);
 646     CertCloseStore(store1, 0);
 647
 648     /* Test whether a collection store can be committed */
 649     if (!pCertControlStore)
 650     {
 651         win_skip("CertControlStore() is not available\n");
 652         return;
 653     }
 654     collection = CertOpenStore(CERT_STORE_PROV_COLLECTION, 0, 0,
 655      CERT_STORE_CREATE_NEW_FLAG, NULL);
 656
 657     SetLastError(0xdeadbeef);
 658     ret = pCertControlStore(collection, 0, CERT_STORE_CTRL_COMMIT, NULL);
 659     ok(ret, "CertControlStore failed: %08x\n", GetLastError());
 660
 661     /* Adding a mem store that can't be committed prevents a successful commit.
 662      */
 663     store1 = CertOpenStore(CERT_STORE_PROV_MEMORY, 0, 0,
 664      CERT_STORE_CREATE_NEW_FLAG, NULL);
 665     pCertAddStoreToCollection(collection, store1, 0, 0);
 666     SetLastError(0xdeadbeef);
 667     ret = pCertControlStore(collection, 0, CERT_STORE_CTRL_COMMIT, NULL);
 668     ok(!ret && GetLastError() == ERROR_CALL_NOT_IMPLEMENTED,
 669      "expected ERROR_CALL_NOT_IMPLEMENTED, got %d\n", GetLastError());
 670     pCertRemoveStoreFromCollection(collection, store1);
 671     CertCloseStore(store1, 0);
 672
 673     /* Test adding a cert to a collection with a file store, committing the
 674      * change to the collection, and comparing the resulting file.
 675      */
 676     if (!GetTempFileNameW(szDot, szPrefix, 0, filename))
 677         return;
 678
 679     DeleteFileW(filename);
 680     file = CreateFileW(filename, GENERIC_READ | GENERIC_WRITE, 0, NULL,
 681      CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
 682     if (file == INVALID_HANDLE_VALUE)
 683         return;
 684
 685     store1 = CertOpenStore(CERT_STORE_PROV_FILE, 0, 0,
 686      CERT_FILE_STORE_COMMIT_ENABLE_FLAG, file);
 687     ok(store1 != NULL, "CertOpenStore failed: %08x\n", GetLastError());
 688     CloseHandle(file);
 689     pCertAddStoreToCollection(collection, store1,
 690      CERT_PHYSICAL_STORE_ADD_ENABLE_FLAG, 0);
 691     CertCloseStore(store1, 0);
 692
 693     ret = CertAddEncodedCertificateToStore(collection, X509_ASN_ENCODING,
 694      bigCert, sizeof(bigCert), CERT_STORE_ADD_ALWAYS, NULL);
 695     ok(ret, "CertAddEncodedCertificateToStore failed: %08x\n",
 696      GetLastError());
 697     ret = pCertControlStore(collection, 0, CERT_STORE_CTRL_COMMIT, NULL);
 698     ok(ret, "CertControlStore failed: %d\n", ret);
 699     compareStore(collection, "serialized store with cert",
 700      serializedStoreWithCert, sizeof(serializedStoreWithCert), FALSE);
 701     CertCloseStore(collection, 0);
 702
 703     DeleteFileW(filename);
 704 }
 705
 706 /* Looks for the property with ID propID in the buffer buf.  Returns a pointer
 707  * to its header if found, NULL if not.
 708  */
 709 static const struct CertPropIDHeader *findPropID(const BYTE *buf, DWORD size,
 710  DWORD propID)
 711 {
 712     const struct CertPropIDHeader *ret = NULL;
 713     BOOL failed = FALSE;
 714
 715     while (size && !ret && !failed)
 716     {
 717         if (size < sizeof(struct CertPropIDHeader))
 718             failed = TRUE;
 719         else
 720         {
 721             const struct CertPropIDHeader *hdr =
 722              (const struct CertPropIDHeader *)buf;
 723
 724             size -= sizeof(struct CertPropIDHeader);
 725             buf += sizeof(struct CertPropIDHeader);
 726             if (size < hdr->cb)
 727                 failed = TRUE;
 728             else if (hdr->propID == propID)
 729                 ret = hdr;
 730             else
 731             {
 732                 buf += hdr->cb;
 733                 size -= hdr->cb;
 734             }
 735         }
 736     }
 737     return ret;
 738 }
 739
 740 typedef DWORD (WINAPI *SHDeleteKeyAFunc)(HKEY, LPCSTR);
 741
 742 static void testRegStore(void)
 743 {
 744     static const char tempKey[] = "Software\\Wine\\CryptTemp";
 745     HCERTSTORE store;
 746     LONG rc;
 747     HKEY key = NULL;
 748     DWORD disp, GLE;
 749
 750     store = CertOpenStore(CERT_STORE_PROV_REG, 0, 0, 0, NULL);
 751     GLE = GetLastError();
 752     ok(!store && (GLE == ERROR_INVALID_HANDLE || GLE == ERROR_BADKEY),
 753      "Expected ERROR_INVALID_HANDLE or ERROR_BADKEY, got %d\n", GLE);
 754     store = CertOpenStore(CERT_STORE_PROV_REG, 0, 0, 0, key);
 755     GLE = GetLastError();
 756     ok(!store && (GLE == ERROR_INVALID_HANDLE || GLE == ERROR_BADKEY),
 757      "Expected ERROR_INVALID_HANDLE or ERROR_BADKEY, got %d\n", GLE);
 758
 759     /* Opening up any old key works.. */
 760     key = HKEY_CURRENT_USER;
 761     store = CertOpenStore(CERT_STORE_PROV_REG, 0, 0, 0, key);
 762     /* Not sure if this is a bug in DuplicateHandle, marking todo_wine for now
 763      */
 764     todo_wine ok(store != 0, "CertOpenStore failed: %08x\n", GetLastError());
 765     CertCloseStore(store, 0);
 766
 767     /* It looks like the remainder pretty much needs CertControlStore() */
 768     if (!pCertControlStore)
 769     {
 770         win_skip("CertControlStore() is not available\n");
 771         return;
 772     }
 773
 774     rc = RegCreateKeyExA(HKEY_CURRENT_USER, tempKey, 0, NULL, 0, KEY_ALL_ACCESS,
 775      NULL, &key, NULL);
 776     ok(!rc, "RegCreateKeyExA failed: %d\n", rc);
 777     if (key)
 778     {
 779         BOOL ret;
 780         BYTE hash[20];
 781         DWORD size, i;
 782         static const char certificates[] = "Certificates\\";
 783         char subKeyName[sizeof(certificates) + 20 * 2 + 1], *ptr;
 784         HKEY subKey;
 785         PCCERT_CONTEXT context;
 786
 787         store = CertOpenStore(CERT_STORE_PROV_REG, 0, 0, 0, key);
 788         ok(store != 0, "CertOpenStore failed: %08x\n", GetLastError());
 789         /* Add a certificate.  It isn't persisted right away, since it's only
 790          * added to the cache..
 791          */
 792         ret = CertAddEncodedCertificateToStore(store, X509_ASN_ENCODING,
 793          bigCert2, sizeof(bigCert2), CERT_STORE_ADD_ALWAYS, NULL);
 794         ok(ret, "CertAddEncodedCertificateToStore failed: %08x\n",
 795          GetLastError());
 796         /* so flush the cache to force a commit.. */
 797         ret = pCertControlStore(store, 0, CERT_STORE_CTRL_COMMIT, NULL);
 798         ok(ret, "CertControlStore failed: %08x\n", GetLastError());
 799         /* and check that the expected subkey was written. */
 800         size = sizeof(hash);
 801         ret = CryptHashCertificate(0, 0, 0, bigCert2, sizeof(bigCert2),
 802          hash, &size);
 803         ok(ret, "CryptHashCertificate failed: %d\n", GetLastError());
 804         strcpy(subKeyName, certificates);
 805         for (i = 0, ptr = subKeyName + sizeof(certificates) - 1; i < size;
 806          i++, ptr += 2)
 807             sprintf(ptr, "%02X", hash[i]);
 808         rc = RegCreateKeyExA(key, subKeyName, 0, NULL, 0, KEY_ALL_ACCESS, NULL,
 809          &subKey, NULL);
 810         ok(!rc, "RegCreateKeyExA failed: %d\n", rc);
 811         if (subKey)
 812         {
 813             LPBYTE buf;
 814
 815             size = 0;
 816             RegQueryValueExA(subKey, "Blob", NULL, NULL, NULL, &size);
 817             buf = HeapAlloc(GetProcessHeap(), 0, size);
 818             if (buf)
 819             {
 820                 rc = RegQueryValueExA(subKey, "Blob", NULL, NULL, buf, &size);
 821                 ok(!rc, "RegQueryValueExA failed: %d\n", rc);
 822                 if (!rc)
 823                 {
 824                     const struct CertPropIDHeader *hdr;
 825
 826                     /* Both the hash and the cert should be present */
 827                     hdr = findPropID(buf, size, CERT_CERT_PROP_ID);
 828                     ok(hdr != NULL, "Expected to find a cert property\n");
 829                     if (hdr)
 830                     {
 831                         ok(hdr->cb == sizeof(bigCert2),
 832                            "Wrong size %d of cert property\n", hdr->cb);
 833                         ok(!memcmp((const BYTE *)hdr + sizeof(*hdr), bigCert2,
 834                          hdr->cb), "Unexpected cert in cert property\n");
 835                     }
 836                     hdr = findPropID(buf, size, CERT_HASH_PROP_ID);
 837                     ok(hdr != NULL, "Expected to find a hash property\n");
 838                     if (hdr)
 839                     {
 840                         ok(hdr->cb == sizeof(hash),
 841                            "Wrong size %d of hash property\n", hdr->cb);
 842                         ok(!memcmp((const BYTE *)hdr + sizeof(*hdr), hash,
 843                          hdr->cb), "Unexpected hash in cert property\n");
 844                     }
 845                 }
 846                 HeapFree(GetProcessHeap(), 0, buf);
 847             }
 848             RegCloseKey(subKey);
 849         }
 850
 851         /* Remove the existing context */
 852         context = CertEnumCertificatesInStore(store, NULL);
 853         ok(context != NULL, "Expected a cert context\n");
 854         if (context)
 855             CertDeleteCertificateFromStore(context);
 856         ret = pCertControlStore(store, 0, CERT_STORE_CTRL_COMMIT, NULL);
 857         ok(ret, "CertControlStore failed: %08x\n", GetLastError());
 858
 859         /* Add a serialized cert with a bogus hash directly to the registry */
 860         memset(hash, 0, sizeof(hash));
 861         strcpy(subKeyName, certificates);
 862         for (i = 0, ptr = subKeyName + sizeof(certificates) - 1;
 863          i < sizeof(hash); i++, ptr += 2)
 864             sprintf(ptr, "%02X", hash[i]);
 865         rc = RegCreateKeyExA(key, subKeyName, 0, NULL, 0, KEY_ALL_ACCESS, NULL,
 866          &subKey, NULL);
 867         ok(!rc, "RegCreateKeyExA failed: %d\n", rc);
 868         if (subKey)
 869         {
 870             BYTE buf[sizeof(struct CertPropIDHeader) * 2 + sizeof(hash) +
 871              sizeof(bigCert)], *ptr;
 872             DWORD certCount = 0;
 873             struct CertPropIDHeader *hdr;
 874
 875             hdr = (struct CertPropIDHeader *)buf;
 876             hdr->propID = CERT_HASH_PROP_ID;
 877             hdr->unknown1 = 1;
 878             hdr->cb = sizeof(hash);
 879             ptr = buf + sizeof(*hdr);
 880             memcpy(ptr, hash, sizeof(hash));
 881             ptr += sizeof(hash);
 882             hdr = (struct CertPropIDHeader *)ptr;
 883             hdr->propID = CERT_CERT_PROP_ID;
 884             hdr->unknown1 = 1;
 885             hdr->cb = sizeof(bigCert);
 886             ptr += sizeof(*hdr);
 887             memcpy(ptr, bigCert, sizeof(bigCert));
 888
 889             rc = RegSetValueExA(subKey, "Blob", 0, REG_BINARY, buf,
 890              sizeof(buf));
 891             ok(!rc, "RegSetValueExA failed: %d\n", rc);
 892
 893             ret = pCertControlStore(store, 0, CERT_STORE_CTRL_RESYNC, NULL);
 894             ok(ret, "CertControlStore failed: %08x\n", GetLastError());
 895
 896             /* Make sure the bogus hash cert gets loaded. */
 897             certCount = 0;
 898             context = NULL;
 899             do {
 900                 context = CertEnumCertificatesInStore(store, context);
 901                 if (context)
 902                     certCount++;
 903             } while (context != NULL);
 904             ok(certCount == 1, "Expected 1 certificates, got %d\n", certCount);
 905
 906             RegCloseKey(subKey);
 907         }
 908
 909         /* Add another serialized cert directly to the registry, this time
 910          * under the correct key name (named with the correct hash value).
 911          */
 912         size = sizeof(hash);
 913         ret = CryptHashCertificate(0, 0, 0, bigCert2,
 914          sizeof(bigCert2), hash, &size);
 915         ok(ret, "CryptHashCertificate failed: %d\n", GetLastError());
 916         strcpy(subKeyName, certificates);
 917         for (i = 0, ptr = subKeyName + sizeof(certificates) - 1;
 918          i < sizeof(hash); i++, ptr += 2)
 919             sprintf(ptr, "%02X", hash[i]);
 920         rc = RegCreateKeyExA(key, subKeyName, 0, NULL, 0, KEY_ALL_ACCESS, NULL,
 921          &subKey, NULL);
 922         ok(!rc, "RegCreateKeyExA failed: %d\n", rc);
 923         if (subKey)
 924         {
 925             BYTE buf[sizeof(struct CertPropIDHeader) * 2 + sizeof(hash) +
 926              sizeof(bigCert2)], *ptr;
 927             DWORD certCount = 0;
 928             PCCERT_CONTEXT context;
 929             struct CertPropIDHeader *hdr;
 930
 931             /* First try with a bogus hash... */
 932             hdr = (struct CertPropIDHeader *)buf;
 933             hdr->propID = CERT_HASH_PROP_ID;
 934             hdr->unknown1 = 1;
 935             hdr->cb = sizeof(hash);
 936             ptr = buf + sizeof(*hdr);
 937             memset(ptr, 0, sizeof(hash));
 938             ptr += sizeof(hash);
 939             hdr = (struct CertPropIDHeader *)ptr;
 940             hdr->propID = CERT_CERT_PROP_ID;
 941             hdr->unknown1 = 1;
 942             hdr->cb = sizeof(bigCert2);
 943             ptr += sizeof(*hdr);
 944             memcpy(ptr, bigCert2, sizeof(bigCert2));
 945
 946             rc = RegSetValueExA(subKey, "Blob", 0, REG_BINARY, buf,
 947              sizeof(buf));
 948             ok(!rc, "RegSetValueExA failed: %d\n", rc);
 949
 950             ret = pCertControlStore(store, 0, CERT_STORE_CTRL_RESYNC, NULL);
 951             ok(ret, "CertControlStore failed: %08x\n", GetLastError());
 952
 953             /* and make sure just one cert still gets loaded. */
 954             certCount = 0;
 955             context = NULL;
 956             do {
 957                 context = CertEnumCertificatesInStore(store, context);
 958                 if (context)
 959                     certCount++;
 960             } while (context != NULL);
 961             ok(certCount == 1 ||
 962                broken(certCount == 2), /* win9x */
 963                "Expected 1 certificates, got %d\n", certCount);
 964
 965             /* Try again with the correct hash... */
 966             ptr = buf + sizeof(*hdr);
 967             memcpy(ptr, hash, sizeof(hash));
 968
 969             rc = RegSetValueExA(subKey, "Blob", 0, REG_BINARY, buf,
 970              sizeof(buf));
 971             ok(!rc, "RegSetValueExA failed: %d\n", rc);
 972
 973             ret = pCertControlStore(store, 0, CERT_STORE_CTRL_RESYNC, NULL);
 974             ok(ret, "CertControlStore failed: %08x\n", GetLastError());
 975
 976             /* and make sure two certs get loaded. */
 977             certCount = 0;
 978             context = NULL;
 979             do {
 980                 context = CertEnumCertificatesInStore(store, context);
 981                 if (context)
 982                     certCount++;
 983             } while (context != NULL);
 984             ok(certCount == 2, "Expected 2 certificates, got %d\n", certCount);
 985
 986             RegCloseKey(subKey);
 987         }
 988         CertCloseStore(store, 0);
 989         /* Is delete allowed on a reg store? */
 990         store = CertOpenStore(CERT_STORE_PROV_REG, 0, 0,
 991          CERT_STORE_DELETE_FLAG, key);
 992         ok(store == NULL, "Expected NULL return from CERT_STORE_DELETE_FLAG\n");
 993         ok(GetLastError() == 0, "CertOpenStore failed: %08x\n",
 994          GetLastError());
 995
 996         RegCloseKey(key);
 997     }
 998     /* The CertOpenStore with CERT_STORE_DELETE_FLAG above will delete the
 999      * contents of the key, but not the key itself.
1000      */
1001     rc = RegCreateKeyExA(HKEY_CURRENT_USER, tempKey, 0, NULL, 0, KEY_ALL_ACCESS,
1002      NULL, &key, &disp);
1003     ok(!rc, "RegCreateKeyExA failed: %d\n", rc);
1004     ok(disp == REG_OPENED_EXISTING_KEY,
1005      "Expected REG_OPENED_EXISTING_KEY, got %d\n", disp);
1006     if (!rc)
1007     {
1008         RegCloseKey(key);
1009         rc = RegDeleteKeyA(HKEY_CURRENT_USER, tempKey);
1010         if (rc)
1011         {
1012             HMODULE shlwapi = LoadLibraryA("shlwapi");
1013
1014             /* Use shlwapi's SHDeleteKeyA to _really_ blow away the key,
1015              * otherwise subsequent tests will fail.
1016              */
1017             if (shlwapi)
1018             {
1019                 SHDeleteKeyAFunc pSHDeleteKeyA =
1020                  (SHDeleteKeyAFunc)GetProcAddress(shlwapi, "SHDeleteKeyA");
1021
1022                 if (pSHDeleteKeyA)
1023                     pSHDeleteKeyA(HKEY_CURRENT_USER, tempKey);
1024                 FreeLibrary(shlwapi);
1025             }
1026         }
1027     }
1028 }
1029
1030 static const char MyA[] = { 'M','y',0,0 };
1031 static const WCHAR MyW[] = { 'M','y',0 };
1032 static const WCHAR BogusW[] = { 'B','o','g','u','s',0 };
1033 static const WCHAR BogusPathW[] = { 'S','o','f','t','w','a','r','e','\\',
1034  'M','i','c','r','o','s','o','f','t','\\','S','y','s','t','e','m','C','e','r',
1035  't','i','f','i','c','a','t','e','s','\\','B','o','g','u','s',0 };
1036
1037 static void testSystemRegStore(void)
1038 {
1039     HCERTSTORE store, memStore;
1040
1041     /* Check with a UNICODE name */
1042     store = CertOpenStore(CERT_STORE_PROV_SYSTEM_REGISTRY, 0, 0,
1043      CERT_SYSTEM_STORE_CURRENT_USER | CERT_STORE_OPEN_EXISTING_FLAG, MyW);
1044     /* Not all OSes support CERT_STORE_PROV_SYSTEM_REGISTRY, so don't continue
1045      * testing if they don't.
1046      */
1047     if (!store)
1048         return;
1049
1050     /* Check that it isn't a collection store */
1051     memStore = CertOpenStore(CERT_STORE_PROV_MEMORY, 0, 0,
1052      CERT_STORE_CREATE_NEW_FLAG, NULL);
1053     if (memStore)
1054     {
1055         if (pCertAddStoreToCollection)
1056         {
1057             BOOL ret = pCertAddStoreToCollection(store, memStore, 0, 0);
1058             ok(!ret && GetLastError() == E_INVALIDARG,
1059                "Expected E_INVALIDARG, got %08x\n", GetLastError());
1060         }
1061         CertCloseStore(memStore, 0);
1062     }
1063     CertCloseStore(store, 0);
1064
1065     /* Check opening a bogus store */
1066     store = CertOpenStore(CERT_STORE_PROV_SYSTEM_REGISTRY, 0, 0,
1067      CERT_SYSTEM_STORE_CURRENT_USER | CERT_STORE_OPEN_EXISTING_FLAG, BogusW);
1068     ok(!store && GetLastError() == ERROR_FILE_NOT_FOUND,
1069      "Expected ERROR_FILE_NOT_FOUND, got %08x\n", GetLastError());
1070     store = CertOpenStore(CERT_STORE_PROV_SYSTEM_REGISTRY, 0, 0,
1071      CERT_SYSTEM_STORE_CURRENT_USER, BogusW);
1072     ok(store != 0, "CertOpenStore failed: %08x\n", GetLastError());
1073     if (store)
1074         CertCloseStore(store, 0);
1075     /* Now check whether deleting is allowed */
1076     store = CertOpenStore(CERT_STORE_PROV_SYSTEM_REGISTRY, 0, 0,
1077      CERT_SYSTEM_STORE_CURRENT_USER | CERT_STORE_DELETE_FLAG, BogusW);
1078     RegDeleteKeyW(HKEY_CURRENT_USER, BogusPathW);
1079
1080     store = CertOpenStore(CERT_STORE_PROV_SYSTEM_REGISTRY, 0, 0, 0, NULL);
1081     ok(!store && GetLastError() == E_INVALIDARG,
1082      "Expected E_INVALIDARG, got %08x\n", GetLastError());
1083     store = CertOpenStore(CERT_STORE_PROV_SYSTEM_REGISTRY, 0, 0,
1084      CERT_SYSTEM_STORE_LOCAL_MACHINE | CERT_SYSTEM_STORE_CURRENT_USER, MyA);
1085     ok(!store && GetLastError() == E_INVALIDARG,
1086      "Expected E_INVALIDARG, got %08x\n", GetLastError());
1087     store = CertOpenStore(CERT_STORE_PROV_SYSTEM_REGISTRY, 0, 0,
1088      CERT_SYSTEM_STORE_LOCAL_MACHINE | CERT_SYSTEM_STORE_CURRENT_USER, MyW);
1089     ok(!store && GetLastError() == E_INVALIDARG,
1090      "Expected E_INVALIDARG, got %08x\n", GetLastError());
1091     /* The name is expected to be UNICODE, check with an ASCII name */
1092     store = CertOpenStore(CERT_STORE_PROV_SYSTEM_REGISTRY, 0, 0,
1093      CERT_SYSTEM_STORE_CURRENT_USER | CERT_STORE_OPEN_EXISTING_FLAG, MyA);
1094     ok(!store && GetLastError() == ERROR_FILE_NOT_FOUND,
1095      "Expected ERROR_FILE_NOT_FOUND, got %08x\n", GetLastError());
1096 }
1097
1098 static void testSystemStore(void)
1099 {
1100     static const WCHAR baskslashW[] = { '\\',0 };
1101     HCERTSTORE store;
1102     WCHAR keyName[MAX_PATH];
1103     HKEY key;
1104     LONG rc;
1105
1106     store = CertOpenStore(CERT_STORE_PROV_SYSTEM, 0, 0, 0, NULL);
1107     ok(!store && GetLastError() == ERROR_FILE_NOT_FOUND,
1108      "Expected ERROR_FILE_NOT_FOUND, got %08x\n", GetLastError());
1109     store = CertOpenStore(CERT_STORE_PROV_SYSTEM, 0, 0,
1110      CERT_SYSTEM_STORE_LOCAL_MACHINE | CERT_SYSTEM_STORE_CURRENT_USER, MyA);
1111     ok(!store && GetLastError() == ERROR_FILE_NOT_FOUND,
1112      "Expected ERROR_FILE_NOT_FOUND, got %08x\n", GetLastError());
1113     store = CertOpenStore(CERT_STORE_PROV_SYSTEM, 0, 0,
1114      CERT_SYSTEM_STORE_LOCAL_MACHINE | CERT_SYSTEM_STORE_CURRENT_USER, MyW);
1115     ok(!store && GetLastError() == ERROR_FILE_NOT_FOUND,
1116      "Expected ERROR_FILE_NOT_FOUND, got %08x\n", GetLastError());
1117     /* The name is expected to be UNICODE, first check with an ASCII name */
1118     store = CertOpenStore(CERT_STORE_PROV_SYSTEM, 0, 0,
1119      CERT_SYSTEM_STORE_CURRENT_USER | CERT_STORE_OPEN_EXISTING_FLAG, MyA);
1120     ok(!store && GetLastError() == ERROR_FILE_NOT_FOUND,
1121      "Expected ERROR_FILE_NOT_FOUND, got %08x\n", GetLastError());
1122     /* Create the expected key */
1123     lstrcpyW(keyName, CERT_LOCAL_MACHINE_SYSTEM_STORE_REGPATH);
1124     lstrcatW(keyName, baskslashW);
1125     lstrcatW(keyName, MyW);
1126     rc = RegCreateKeyExW(HKEY_CURRENT_USER, keyName, 0, NULL, 0, KEY_READ,
1127      NULL, &key, NULL);
1128     ok(!rc, "RegCreateKeyEx failed: %d\n", rc);
1129     if (!rc)
1130         RegCloseKey(key);
1131     /* Check opening with a UNICODE name, specifying the create new flag */
1132     store = CertOpenStore(CERT_STORE_PROV_SYSTEM, 0, 0,
1133      CERT_SYSTEM_STORE_CURRENT_USER | CERT_STORE_CREATE_NEW_FLAG, MyW);
1134     ok(!store && GetLastError() == ERROR_FILE_EXISTS,
1135      "Expected ERROR_FILE_EXISTS, got %08x\n", GetLastError());
1136     /* Now check opening with a UNICODE name, this time opening existing */
1137     store = CertOpenStore(CERT_STORE_PROV_SYSTEM, 0, 0,
1138      CERT_SYSTEM_STORE_CURRENT_USER | CERT_STORE_OPEN_EXISTING_FLAG, MyW);
1139     ok(store != 0, "CertOpenStore failed: %08x\n", GetLastError());
1140     if (store)
1141     {
1142         HCERTSTORE memStore = CertOpenStore(CERT_STORE_PROV_MEMORY, 0, 0,
1143          CERT_STORE_CREATE_NEW_FLAG, NULL);
1144
1145         /* Check that it's a collection store */
1146         if (memStore)
1147         {
1148             if (pCertAddStoreToCollection)
1149             {
1150                 BOOL ret = pCertAddStoreToCollection(store, memStore, 0, 0);
1151                 /* FIXME: this'll fail on NT4, but what error will it give? */
1152                 ok(ret, "CertAddStoreToCollection failed: %08x\n", GetLastError());
1153             }
1154             CertCloseStore(memStore, 0);
1155         }
1156         CertCloseStore(store, 0);
1157     }
1158
1159     /* Check opening a bogus store */
1160     store = CertOpenStore(CERT_STORE_PROV_SYSTEM, 0, 0,
1161      CERT_SYSTEM_STORE_CURRENT_USER | CERT_STORE_OPEN_EXISTING_FLAG, BogusW);
1162     ok((!store ||
1163      broken(store != 0)) && /* win9x */
1164      GetLastError() == ERROR_FILE_NOT_FOUND,
1165      "Expected ERROR_FILE_NOT_FOUND, got %08x\n", GetLastError());
1166     store = CertOpenStore(CERT_STORE_PROV_SYSTEM, 0, 0,
1167      CERT_SYSTEM_STORE_CURRENT_USER, BogusW);
1168     ok(store != 0, "CertOpenStore failed: %08x\n", GetLastError());
1169     if (store)
1170         CertCloseStore(store, 0);
1171     /* Now check whether deleting is allowed */
1172     store = CertOpenStore(CERT_STORE_PROV_SYSTEM, 0, 0,
1173      CERT_SYSTEM_STORE_CURRENT_USER | CERT_STORE_DELETE_FLAG, BogusW);
1174     RegDeleteKeyW(HKEY_CURRENT_USER, BogusPathW);
1175 }
1176
1177 static const BYTE serializedStoreWithCertAndCRL[] = {
1178  0x00,0x00,0x00,0x00,0x43,0x45,0x52,0x54,0x20,0x00,0x00,0x00,0x01,0x00,0x00,
1179  0x00,0x7c,0x00,0x00,0x00,0x30,0x7a,0x02,0x01,0x01,0x30,0x02,0x06,0x00,0x30,
1180  0x15,0x31,0x13,0x30,0x11,0x06,0x03,0x55,0x04,0x03,0x13,0x0a,0x4a,0x75,0x61,
1181  0x6e,0x20,0x4c,0x61,0x6e,0x67,0x00,0x30,0x22,0x18,0x0f,0x31,0x36,0x30,0x31,
1182  0x30,0x31,0x30,0x31,0x30,0x30,0x30,0x30,0x30,0x30,0x5a,0x18,0x0f,0x31,0x36,
1183  0x30,0x31,0x30,0x31,0x30,0x31,0x30,0x30,0x30,0x30,0x30,0x30,0x5a,0x30,0x15,
1184  0x31,0x13,0x30,0x11,0x06,0x03,0x55,0x04,0x03,0x13,0x0a,0x4a,0x75,0x61,0x6e,
1185  0x20,0x4c,0x61,0x6e,0x67,0x00,0x30,0x07,0x30,0x02,0x06,0x00,0x03,0x01,0x00,
1186  0xa3,0x16,0x30,0x14,0x30,0x12,0x06,0x03,0x55,0x1d,0x13,0x01,0x01,0xff,0x04,
1187  0x08,0x30,0x06,0x01,0x01,0xff,0x02,0x01,0x01,0x21,0x00,0x00,0x00,0x01,0x00,
1188  0x00,0x00,0x47,0x00,0x00,0x00,0x30,0x45,0x30,0x2c,0x30,0x02,0x06,0x00,0x30,
1189  0x15,0x31,0x13,0x30,0x11,0x06,0x03,0x55,0x04,0x03,0x13,0x0a,0x4a,0x75,0x61,
1190  0x6e,0x20,0x4c,0x61,0x6e,0x67,0x00,0x18,0x0f,0x31,0x36,0x30,0x31,0x30,0x31,
1191  0x30,0x31,0x30,0x30,0x30,0x30,0x30,0x30,0x5a,0x30,0x02,0x06,0x00,0x03,0x11,
1192  0x00,0x0f,0x0e,0x0d,0x0c,0x0b,0x0a,0x09,0x08,0x07,0x06,0x05,0x04,0x03,0x02,
1193  0x01,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00 };
1194
1195 static void testFileStore(void)
1196 {
1197     static const WCHAR szPrefix[] = { 'c','e','r',0 };
1198     static const WCHAR szDot[] = { '.',0 };
1199     WCHAR filename[MAX_PATH];
1200     HCERTSTORE store;
1201     BOOL ret;
1202     PCCERT_CONTEXT cert;
1203     HANDLE file;
1204
1205     if (!pCertControlStore)
1206     {
1207         win_skip("CertControlStore() is not available\n");
1208         return;
1209     }
1210
1211     store = CertOpenStore(CERT_STORE_PROV_FILE, 0, 0, 0, NULL);
1212     ok(!store && GetLastError() == ERROR_INVALID_HANDLE,
1213      "Expected ERROR_INVALID_HANDLE, got %08x\n", GetLastError());
1214
1215     if (!GetTempFileNameW(szDot, szPrefix, 0, filename))
1216        return;
1217
1218     DeleteFileW(filename);
1219     file = CreateFileW(filename, GENERIC_READ | GENERIC_WRITE, 0, NULL,
1220      CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
1221     if (file == INVALID_HANDLE_VALUE)
1222         return;
1223
1224     store = CertOpenStore(CERT_STORE_PROV_FILE, 0, 0, CERT_STORE_DELETE_FLAG,
1225      file);
1226     ok(!store && GetLastError() == E_INVALIDARG,
1227      "Expected E_INVALIDARG, got %08x\n", GetLastError());
1228     store = CertOpenStore(CERT_STORE_PROV_FILE, 0, 0,
1229      CERT_FILE_STORE_COMMIT_ENABLE_FLAG | CERT_STORE_READONLY_FLAG, file);
1230     ok(!store && GetLastError() == E_INVALIDARG,
1231      "Expected E_INVALIDARG, got %08x\n", GetLastError());
1232
1233     /* A "read-only" file store.. */
1234     store = CertOpenStore(CERT_STORE_PROV_FILE, 0, 0,
1235      CERT_STORE_OPEN_EXISTING_FLAG | CERT_STORE_READONLY_FLAG, file);
1236     ok(store != NULL, "CertOpenStore failed: %08x\n", GetLastError());
1237     if (store)
1238     {
1239         DWORD size;
1240
1241         ret = CertAddEncodedCertificateToStore(store, X509_ASN_ENCODING,
1242          bigCert, sizeof(bigCert), CERT_STORE_ADD_ALWAYS, NULL);
1243         /* apparently allows adding certificates.. */
1244         ok(ret, "CertAddEncodedCertificateToStore failed: %d\n", ret);
1245         /* but not commits.. */
1246         ret = pCertControlStore(store, 0, CERT_STORE_CTRL_COMMIT, NULL);
1247         ok(!ret && GetLastError() == ERROR_CALL_NOT_IMPLEMENTED,
1248          "Expected ERROR_CALL_NOT_IMPLEMENTED, got %08x\n", GetLastError());
1249         /* It still has certs in memory.. */
1250         cert = CertEnumCertificatesInStore(store, NULL);
1251         ok(cert != NULL, "CertEnumCertificatesInStore failed: %08x\n",
1252          GetLastError());
1253         CertFreeCertificateContext(cert);
1254         /* but the file size is still 0. */
1255         size = GetFileSize(file, NULL);
1256         ok(size == 0, "Expected size 0, got %d\n", size);
1257         CertCloseStore(store, 0);
1258     }
1259
1260     /* The create new flag is allowed.. */
1261     store = CertOpenStore(CERT_STORE_PROV_FILE, 0, 0,
1262      CERT_STORE_CREATE_NEW_FLAG, file);
1263     ok(store != NULL, "CertOpenStore failed: %08x\n", GetLastError());
1264     if (store)
1265     {
1266         /* but without the commit enable flag, commits don't happen. */
1267         ret = CertAddEncodedCertificateToStore(store, X509_ASN_ENCODING,
1268          bigCert, sizeof(bigCert), CERT_STORE_ADD_ALWAYS, NULL);
1269         ok(ret, "CertAddEncodedCertificateToStore failed: %d\n", ret);
1270         ret = pCertControlStore(store, 0, CERT_STORE_CTRL_COMMIT, NULL);
1271         ok(!ret && GetLastError() == ERROR_CALL_NOT_IMPLEMENTED,
1272          "Expected ERROR_CALL_NOT_IMPLEMENTED, got %08x\n", GetLastError());
1273         CertCloseStore(store, 0);
1274     }
1275     /* as is the open existing flag. */
1276     store = CertOpenStore(CERT_STORE_PROV_FILE, 0, 0,
1277      CERT_STORE_OPEN_EXISTING_FLAG, file);
1278     ok(store != NULL, "CertOpenStore failed: %08x\n", GetLastError());
1279     if (store)
1280     {
1281         /* but without the commit enable flag, commits don't happen. */
1282         ret = CertAddEncodedCertificateToStore(store, X509_ASN_ENCODING,
1283          bigCert, sizeof(bigCert), CERT_STORE_ADD_ALWAYS, NULL);
1284         ok(ret, "CertAddEncodedCertificateToStore failed: %d\n", ret);
1285         ret = pCertControlStore(store, 0, CERT_STORE_CTRL_COMMIT, NULL);
1286         ok(!ret && GetLastError() == ERROR_CALL_NOT_IMPLEMENTED,
1287          "Expected ERROR_CALL_NOT_IMPLEMENTED, got %08x\n", GetLastError());
1288         CertCloseStore(store, 0);
1289     }
1290     store = CertOpenStore(CERT_STORE_PROV_FILE, 0, 0,
1291      CERT_FILE_STORE_COMMIT_ENABLE_FLAG, file);
1292     ok(store != NULL, "CertOpenStore failed: %08x\n", GetLastError());
1293     if (store)
1294     {
1295         CloseHandle(file);
1296         ret = CertAddEncodedCertificateToStore(store, X509_ASN_ENCODING,
1297          bigCert, sizeof(bigCert), CERT_STORE_ADD_ALWAYS, NULL);
1298         ok(ret, "CertAddEncodedCertificateToStore failed: %08x\n",
1299          GetLastError());
1300         /* with commits enabled, commit is allowed */
1301         ret = pCertControlStore(store, 0, CERT_STORE_CTRL_COMMIT, NULL);
1302         ok(ret, "CertControlStore failed: %d\n", ret);
1303         compareStore(store, "serialized store with cert",
1304          serializedStoreWithCert, sizeof(serializedStoreWithCert), FALSE);
1305         CertCloseStore(store, 0);
1306     }
1307     file = CreateFileW(filename, GENERIC_READ | GENERIC_WRITE, 0, NULL,
1308      OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
1309     if (file == INVALID_HANDLE_VALUE)
1310         return;
1311     store = CertOpenStore(CERT_STORE_PROV_FILE, 0, 0,
1312      CERT_FILE_STORE_COMMIT_ENABLE_FLAG, file);
1313     ok(store != NULL, "CertOpenStore failed: %08x\n", GetLastError());
1314     if (store)
1315     {
1316         CloseHandle(file);
1317         ret = CertAddEncodedCRLToStore(store, X509_ASN_ENCODING, signedCRL,
1318          sizeof(signedCRL), CERT_STORE_ADD_ALWAYS, NULL);
1319         ok(ret, "CertAddEncodedCRLToStore failed: %08x\n", GetLastError());
1320         compareStore(store, "serialized store with cert and CRL",
1321          serializedStoreWithCertAndCRL, sizeof(serializedStoreWithCertAndCRL),
1322          FALSE);
1323         CertCloseStore(store, 0);
1324     }
1325
1326     DeleteFileW(filename);
1327 }
1328
1329 static BOOL initFileFromData(LPCWSTR filename, const BYTE *pb, DWORD cb)
1330 {
1331     HANDLE file = CreateFileW(filename, GENERIC_READ | GENERIC_WRITE, 0, NULL,
1332      CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
1333     BOOL ret;
1334
1335     if (file != INVALID_HANDLE_VALUE)
1336     {
1337         DWORD written;
1338
1339         ret = WriteFile(file, pb, cb, &written, NULL);
1340         CloseHandle(file);
1341     }
1342     else
1343         ret = FALSE;
1344     return ret;
1345 }
1346
1347 static const BYTE base64SPC[] =
1348 "MIICJQYJKoZIhvcNAQcCoIICFjCCAhICAQExADALBgkqhkiG9w0BBwGgggH6MIIB"
1349 "9jCCAV+gAwIBAgIQnP8+EF4opr9OxH7h4uBPWTANBgkqhkiG9w0BAQQFADAUMRIw"
1350 "EAYDVQQDEwlKdWFuIExhbmcwHhcNMDgxMjEyMTcxMDE0WhcNMzkxMjMxMjM1OTU5"
1351 "WjAUMRIwEAYDVQQDEwlKdWFuIExhbmcwgZ8wDQYJKoZIhvcNAQEBBQADgY0AMIGJ"
1352 "AoGBALCgNjyNvOic0FOfjxvi43HbM+D5joDkhiGSXe+gbZlf8f16k07kkObFEunz"
1353 "mdB5coscmA7gyqiWNN4ZUyr2cA3lCbnpGPA/0IblyyOcuGIFmmCzeZaVa5ZG6xZP"
1354 "K7L7o+73Qo6jXVbGhBGnMZ7Q9sAn6s2933olnStnejnqwV0NAgMBAAGjSTBHMEUG"
1355 "A1UdAQQ+MDyAEFKbKEdXYyx+CWKcV6vxM6ShFjAUMRIwEAYDVQQDEwlKdWFuIExh"
1356 "bmeCEJz/PhBeKKa/TsR+4eLgT1kwDQYJKoZIhvcNAQEEBQADgYEALpkgLgW3mEaK"
1357 "idPQ3iPJYLG0Ub1wraqEl9bd42hrhzIdcDzlQgxnm8/5cHYVxIF/C20x/HJplb1R"
1358 "G6U1ipFe/q8byWD/9JpiBKMGPi9YlUTgXHfS9d4S/QWO1h9Z7KeipBYhoslQpHXu"
1359 "y9bUr8Adqi6SzgHpCnMu53dxgxUD1r4xAA==";
1360 /* Same as base64SPC, but as a wide-char string */
1361 static const WCHAR utf16Base64SPC[] = {
1362 'M','I','I','C','J','Q','Y','J','K','o','Z','I','h','v','c','N','A',
1363 'Q','c','C','o','I','I','C','F','j','C','C','A','h','I','C','A','Q',
1364 'E','x','A','D','A','L','B','g','k','q','h','k','i','G','9','w','0',
1365 'B','B','w','G','g','g','g','H','6','M','I','I','B','9','j','C','C',
1366 'A','V','+','g','A','w','I','B','A','g','I','Q','n','P','8','+','E',
1367 'F','4','o','p','r','9','O','x','H','7','h','4','u','B','P','W','T',
1368 'A','N','B','g','k','q','h','k','i','G','9','w','0','B','A','Q','Q',
1369 'F','A','D','A','U','M','R','I','w','E','A','Y','D','V','Q','Q','D',
1370 'E','w','l','K','d','W','F','u','I','E','x','h','b','m','c','w','H',
1371 'h','c','N','M','D','g','x','M','j','E','y','M','T','c','x','M','D',
1372 'E','0','W','h','c','N','M','z','k','x','M','j','M','x','M','j','M',
1373 '1','O','T','U','5','W','j','A','U','M','R','I','w','E','A','Y','D',
1374 'V','Q','Q','D','E','w','l','K','d','W','F','u','I','E','x','h','b',
1375 'm','c','w','g','Z','8','w','D','Q','Y','J','K','o','Z','I','h','v',
1376 'c','N','A','Q','E','B','B','Q','A','D','g','Y','0','A','M','I','G',
1377 'J','A','o','G','B','A','L','C','g','N','j','y','N','v','O','i','c',
1378 '0','F','O','f','j','x','v','i','4','3','H','b','M','+','D','5','j',
1379 'o','D','k','h','i','G','S','X','e','+','g','b','Z','l','f','8','f',
1380 '1','6','k','0','7','k','k','O','b','F','E','u','n','z','m','d','B',
1381 '5','c','o','s','c','m','A','7','g','y','q','i','W','N','N','4','Z',
1382 'U','y','r','2','c','A','3','l','C','b','n','p','G','P','A','/','0',
1383 'I','b','l','y','y','O','c','u','G','I','F','m','m','C','z','e','Z',
1384 'a','V','a','5','Z','G','6','x','Z','P','K','7','L','7','o','+','7',
1385 '3','Q','o','6','j','X','V','b','G','h','B','G','n','M','Z','7','Q',
1386 '9','s','A','n','6','s','2','9','3','3','o','l','n','S','t','n','e',
1387 'j','n','q','w','V','0','N','A','g','M','B','A','A','G','j','S','T',
1388 'B','H','M','E','U','G','A','1','U','d','A','Q','Q','+','M','D','y',
1389 'A','E','F','K','b','K','E','d','X','Y','y','x','+','C','W','K','c',
1390 'V','6','v','x','M','6','S','h','F','j','A','U','M','R','I','w','E',
1391 'A','Y','D','V','Q','Q','D','E','w','l','K','d','W','F','u','I','E',
1392 'x','h','b','m','e','C','E','J','z','/','P','h','B','e','K','K','a',
1393 '/','T','s','R','+','4','e','L','g','T','1','k','w','D','Q','Y','J',
1394 'K','o','Z','I','h','v','c','N','A','Q','E','E','B','Q','A','D','g',
1395 'Y','E','A','L','p','k','g','L','g','W','3','m','E','a','K','i','d',
1396 'P','Q','3','i','P','J','Y','L','G','0','U','b','1','w','r','a','q',
1397 'E','l','9','b','d','4','2','h','r','h','z','I','d','c','D','z','l',
1398 'Q','g','x','n','m','8','/','5','c','H','Y','V','x','I','F','/','C',
1399 '2','0','x','/','H','J','p','l','b','1','R','G','6','U','1','i','p',
1400 'F','e','/','q','8','b','y','W','D','/','9','J','p','i','B','K','M',
1401 'G','P','i','9','Y','l','U','T','g','X','H','f','S','9','d','4','S',
1402 '/','Q','W','O','1','h','9','Z','7','K','e','i','p','B','Y','h','o',
1403 's','l','Q','p','H','X','u','y','9','b','U','r','8','A','d','q','i',
1404 '6','S','z','g','H','p','C','n','M','u','5','3','d','x','g','x','U',
1405 'D','1','r','4','x','A','A','=','=',0 };
1406
1407 static void testFileNameStore(void)
1408 {
1409     static const WCHAR szPrefix[] = { 'c','e','r',0 };
1410     static const WCHAR spcPrefix[] = { 's','p','c',0 };
1411     static const WCHAR szDot[] = { '.',0 };
1412     WCHAR filename[MAX_PATH];
1413     HCERTSTORE store;
1414     BOOL ret;
1415     DWORD GLE;
1416
1417     if (0)
1418     {
1419         /* Crashes on NT4 */
1420         store = CertOpenStore(CERT_STORE_PROV_FILENAME_W, 0, 0, 0, NULL);
1421         GLE = GetLastError();
1422         ok(!store && (GLE == ERROR_PATH_NOT_FOUND || GLE == ERROR_INVALID_PARAMETER),
1423          "Expected ERROR_PATH_NOT_FOUND or ERROR_INVALID_PARAMETER, got %08x\n",
1424          GLE);
1425     }
1426
1427     if (!GetTempFileNameW(szDot, szPrefix, 0, filename))
1428        return;
1429     DeleteFileW(filename);
1430
1431     /* The two flags are mutually exclusive */
1432     store = CertOpenStore(CERT_STORE_PROV_FILENAME_W, 0, 0,
1433      CERT_FILE_STORE_COMMIT_ENABLE_FLAG | CERT_STORE_READONLY_FLAG, filename);
1434     ok(!store && GetLastError() == E_INVALIDARG,
1435      "Expected E_INVALIDARG, got %08x\n", GetLastError());
1436
1437     /* In all of the following tests, the encoding type seems to be ignored */
1438     if (initFileFromData(filename, bigCert, sizeof(bigCert)))
1439     {
1440         PCCERT_CONTEXT cert;
1441         PCCRL_CONTEXT crl;
1442
1443         store = CertOpenStore(CERT_STORE_PROV_FILENAME_W, 0, 0,
1444          CERT_STORE_READONLY_FLAG, filename);
1445         ok(store != NULL, "CertOpenStore failed: %08x\n", GetLastError());
1446
1447         cert = CertEnumCertificatesInStore(store, NULL);
1448         ok(cert != NULL, "CertEnumCertificatesInStore failed: %08x\n",
1449          GetLastError());
1450         cert = CertEnumCertificatesInStore(store, cert);
1451         ok(!cert, "Expected only one cert\n");
1452         if (pCertEnumCRLsInStore)
1453         {
1454             crl = pCertEnumCRLsInStore(store, NULL);
1455             ok(!crl, "Expected no CRLs\n");
1456         }
1457
1458         CertCloseStore(store, 0);
1459         DeleteFileW(filename);
1460     }
1461     if (initFileFromData(filename, serializedStoreWithCert,
1462      sizeof(serializedStoreWithCert)))
1463     {
1464         PCCERT_CONTEXT cert;
1465         PCCRL_CONTEXT crl;
1466
1467         store = CertOpenStore(CERT_STORE_PROV_FILENAME_W, 0, 0,
1468          CERT_STORE_READONLY_FLAG, filename);
1469         ok(store != NULL, "CertOpenStore failed: %08x\n", GetLastError());
1470
1471         cert = CertEnumCertificatesInStore(store, NULL);
1472         ok(cert != NULL, "CertEnumCertificatesInStore failed: %08x\n",
1473          GetLastError());
1474         cert = CertEnumCertificatesInStore(store, cert);
1475         ok(!cert, "Expected only one cert\n");
1476         if (pCertEnumCRLsInStore)
1477         {
1478             crl = pCertEnumCRLsInStore(store, NULL);
1479             ok(!crl, "Expected no CRLs\n");
1480         }
1481
1482         CertCloseStore(store, 0);
1483         DeleteFileW(filename);
1484     }
1485     if (initFileFromData(filename, serializedStoreWithCertAndCRL,
1486      sizeof(serializedStoreWithCertAndCRL)))
1487     {
1488         PCCERT_CONTEXT cert;
1489         PCCRL_CONTEXT crl;
1490
1491         store = CertOpenStore(CERT_STORE_PROV_FILENAME_W, 0, 0,
1492          CERT_STORE_READONLY_FLAG, filename);
1493         ok(store != NULL, "CertOpenStore failed: %08x\n", GetLastError());
1494
1495         cert = CertEnumCertificatesInStore(store, NULL);
1496         ok(cert != NULL, "CertEnumCertificatesInStore failed: %08x\n",
1497          GetLastError());
1498         cert = CertEnumCertificatesInStore(store, cert);
1499         ok(!cert, "Expected only one cert\n");
1500         if (pCertEnumCRLsInStore)
1501         {
1502             crl = pCertEnumCRLsInStore(store, NULL);
1503             ok(crl != NULL, "CertEnumCRLsInStore failed: %08x\n", GetLastError());
1504             crl = pCertEnumCRLsInStore(store, crl);
1505             ok(!crl, "Expected only one CRL\n");
1506         }
1507
1508         CertCloseStore(store, 0);
1509         /* Don't delete it this time, the next test uses it */
1510     }
1511     /* Now that the file exists, we can open it read-only */
1512     store = CertOpenStore(CERT_STORE_PROV_FILENAME_W, 0, 0,
1513      CERT_STORE_READONLY_FLAG, filename);
1514     ok(store != NULL, "CertOpenStore failed: %08x\n", GetLastError());
1515     CertCloseStore(store, 0);
1516     DeleteFileW(filename);
1517
1518     store = CertOpenStore(CERT_STORE_PROV_FILENAME_W, 0, 0,
1519      CERT_FILE_STORE_COMMIT_ENABLE_FLAG | CERT_STORE_CREATE_NEW_FLAG, filename);
1520     ok(store != NULL, "CertOpenStore failed: %08x\n", GetLastError());
1521     if (store)
1522     {
1523         ret = CertAddEncodedCertificateToStore(store, X509_ASN_ENCODING,
1524          bigCert, sizeof(bigCert), CERT_STORE_ADD_ALWAYS, NULL);
1525         ok(ret, "CertAddEncodedCertificateToStore failed: %08x\n",
1526          GetLastError());
1527         compareStore(store, "serialized store with cert",
1528          serializedStoreWithCert, sizeof(serializedStoreWithCert), FALSE);
1529         CertCloseStore(store, 0);
1530     }
1531     store = CertOpenStore(CERT_STORE_PROV_FILENAME_W, 0, 0,
1532      CERT_FILE_STORE_COMMIT_ENABLE_FLAG, filename);
1533     ok(store != NULL, "CertOpenStore failed: %08x\n", GetLastError());
1534     if (store)
1535     {
1536         ret = CertAddEncodedCRLToStore(store, X509_ASN_ENCODING,
1537          signedCRL, sizeof(signedCRL), CERT_STORE_ADD_ALWAYS, NULL);
1538         ok(ret, "CertAddEncodedCRLToStore failed: %08x\n", GetLastError());
1539         compareStore(store, "serialized store with cert and CRL",
1540          serializedStoreWithCertAndCRL, sizeof(serializedStoreWithCertAndCRL),
1541          FALSE);
1542         CertCloseStore(store, 0);
1543     }
1544     DeleteFileW(filename);
1545
1546     if (!GetTempFileNameW(szDot, spcPrefix, 0, filename))
1547        return;
1548     DeleteFileW(filename);
1549
1550     if (initFileFromData(filename, base64SPC, sizeof(base64SPC)))
1551     {
1552         PCCERT_CONTEXT cert;
1553         PCCRL_CONTEXT crl;
1554
1555         store = CertOpenStore(CERT_STORE_PROV_FILENAME_W, 0, 0,
1556          CERT_STORE_READONLY_FLAG, filename);
1557         ok(store != NULL, "CertOpenStore failed: %08x\n", GetLastError());
1558
1559         cert = CertEnumCertificatesInStore(store, NULL);
1560         ok(cert != NULL, "CertEnumCertificatesInStore failed: %08x\n",
1561          GetLastError());
1562         cert = CertEnumCertificatesInStore(store, cert);
1563         ok(!cert, "Expected only one cert\n");
1564         if (pCertEnumCRLsInStore)
1565         {
1566             crl = pCertEnumCRLsInStore(store, NULL);
1567             ok(!crl, "Expected no CRLs\n");
1568         }
1569
1570         CertCloseStore(store, 0);
1571         DeleteFileW(filename);
1572     }
1573     if (initFileFromData(filename, (BYTE *)utf16Base64SPC,
1574      sizeof(utf16Base64SPC)))
1575     {
1576         PCCERT_CONTEXT cert;
1577         PCCRL_CONTEXT crl;
1578
1579         store = CertOpenStore(CERT_STORE_PROV_FILENAME_W, 0, 0,
1580          CERT_STORE_READONLY_FLAG, filename);
1581         ok(store != NULL, "CertOpenStore failed: %08x\n", GetLastError());
1582
1583         cert = CertEnumCertificatesInStore(store, NULL);
1584         ok(cert != NULL, "CertEnumCertificatesInStore failed: %08x\n",
1585          GetLastError());
1586         cert = CertEnumCertificatesInStore(store, cert);
1587         ok(!cert, "Expected only one cert\n");
1588         if (pCertEnumCRLsInStore)
1589         {
1590             crl = pCertEnumCRLsInStore(store, NULL);
1591             ok(!crl, "Expected no CRLs\n");
1592         }
1593
1594         CertCloseStore(store, 0);
1595         DeleteFileW(filename);
1596     }
1597 }
1598
1599 static const BYTE signedContent[] = {
1600 0x30,0x81,0xb2,0x06,0x09,0x2a,0x86,0x48,0x86,0xf7,0x0d,0x01,0x07,0x02,0xa0,
1601 0x81,0xa4,0x30,0x81,0xa1,0x02,0x01,0x01,0x31,0x0e,0x30,0x0c,0x06,0x08,0x2a,
1602 0x86,0x48,0x86,0xf7,0x0d,0x02,0x05,0x05,0x00,0x30,0x13,0x06,0x09,0x2a,0x86,
1603 0x48,0x86,0xf7,0x0d,0x01,0x07,0x01,0xa0,0x06,0x04,0x04,0x01,0x02,0x03,0x04,
1604 0x31,0x77,0x30,0x75,0x02,0x01,0x01,0x30,0x1a,0x30,0x15,0x31,0x13,0x30,0x11,
1605 0x06,0x03,0x55,0x04,0x03,0x13,0x0a,0x4a,0x75,0x61,0x6e,0x20,0x4c,0x61,0x6e,
1606 0x67,0x00,0x02,0x01,0x01,0x30,0x0c,0x06,0x08,0x2a,0x86,0x48,0x86,0xf7,0x0d,
1607 0x02,0x05,0x05,0x00,0x30,0x04,0x06,0x00,0x05,0x00,0x04,0x40,0x81,0xa6,0x70,
1608 0xb3,0xef,0x59,0xd1,0x66,0xd1,0x9b,0xc0,0x9a,0xb6,0x9a,0x5e,0x6d,0x6f,0x6d,
1609 0x0d,0x59,0xa9,0xaa,0x6e,0xe9,0x2c,0xa0,0x1e,0xee,0xc2,0x60,0xbc,0x59,0xbe,
1610 0x3f,0x63,0x06,0x8d,0xc9,0x11,0x1d,0x23,0x64,0x92,0xef,0x2e,0xfc,0x57,0x29,
1611 0xa4,0xaf,0xe0,0xee,0x93,0x19,0x39,0x51,0xe4,0x44,0xb8,0x0b,0x28,0xf4,0xa8,
1612 0x0d };
1613 static const BYTE signedWithCertAndCrlBareContent[] = {
1614 0x30,0x82,0x01,0x4f,0x02,0x01,0x01,0x31,0x0e,0x30,0x0c,0x06,0x08,0x2a,0x86,
1615 0x48,0x86,0xf7,0x0d,0x02,0x05,0x05,0x00,0x30,0x13,0x06,0x09,0x2a,0x86,0x48,
1616 0x86,0xf7,0x0d,0x01,0x07,0x01,0xa0,0x06,0x04,0x04,0x01,0x02,0x03,0x04,0xa0,
1617 0x7c,0x30,0x7a,0x02,0x01,0x01,0x30,0x02,0x06,0x00,0x30,0x15,0x31,0x13,0x30,
1618 0x11,0x06,0x03,0x55,0x04,0x03,0x13,0x0a,0x4a,0x75,0x61,0x6e,0x20,0x4c,0x61,
1619 0x6e,0x67,0x00,0x30,0x22,0x18,0x0f,0x31,0x36,0x30,0x31,0x30,0x31,0x30,0x31,
1620 0x30,0x30,0x30,0x30,0x30,0x30,0x5a,0x18,0x0f,0x31,0x36,0x30,0x31,0x30,0x31,
1621 0x30,0x31,0x30,0x30,0x30,0x30,0x30,0x30,0x5a,0x30,0x15,0x31,0x13,0x30,0x11,
1622 0x06,0x03,0x55,0x04,0x03,0x13,0x0a,0x4a,0x75,0x61,0x6e,0x20,0x4c,0x61,0x6e,
1623 0x67,0x00,0x30,0x07,0x30,0x02,0x06,0x00,0x03,0x01,0x00,0xa3,0x16,0x30,0x14,
1624 0x30,0x12,0x06,0x03,0x55,0x1d,0x13,0x01,0x01,0xff,0x04,0x08,0x30,0x06,0x01,
1625 0x01,0xff,0x02,0x01,0x01,0xa1,0x2e,0x30,0x2c,0x30,0x02,0x06,0x00,0x30,0x15,
1626 0x31,0x13,0x30,0x11,0x06,0x03,0x55,0x04,0x03,0x13,0x0a,0x4a,0x75,0x61,0x6e,
1627 0x20,0x4c,0x61,0x6e,0x67,0x00,0x18,0x0f,0x31,0x36,0x30,0x31,0x30,0x31,0x30,
1628 0x31,0x30,0x30,0x30,0x30,0x30,0x30,0x5a,0x31,0x77,0x30,0x75,0x02,0x01,0x01,
1629 0x30,0x1a,0x30,0x15,0x31,0x13,0x30,0x11,0x06,0x03,0x55,0x04,0x03,0x13,0x0a,
1630 0x4a,0x75,0x61,0x6e,0x20,0x4c,0x61,0x6e,0x67,0x00,0x02,0x01,0x01,0x30,0x0c,
1631 0x06,0x08,0x2a,0x86,0x48,0x86,0xf7,0x0d,0x02,0x05,0x05,0x00,0x30,0x04,0x06,
1632 0x00,0x05,0x00,0x04,0x40,0x81,0xa6,0x70,0xb3,0xef,0x59,0xd1,0x66,0xd1,0x9b,
1633 0xc0,0x9a,0xb6,0x9a,0x5e,0x6d,0x6f,0x6d,0x0d,0x59,0xa9,0xaa,0x6e,0xe9,0x2c,
1634 0xa0,0x1e,0xee,0xc2,0x60,0xbc,0x59,0xbe,0x3f,0x63,0x06,0x8d,0xc9,0x11,0x1d,
1635 0x23,0x64,0x92,0xef,0x2e,0xfc,0x57,0x29,0xa4,0xaf,0xe0,0xee,0x93,0x19,0x39,
1636 0x51,0xe4,0x44,0xb8,0x0b,0x28,0xf4,0xa8,0x0d };
1637 static const BYTE hashContent[] = {
1638 0x30,0x47,0x06,0x09,0x2a,0x86,0x48,0x86,0xf7,0x0d,0x01,0x07,0x05,0xa0,0x3a,
1639 0x30,0x38,0x02,0x01,0x00,0x30,0x0c,0x06,0x08,0x2a,0x86,0x48,0x86,0xf7,0x0d,
1640 0x02,0x05,0x05,0x00,0x30,0x13,0x06,0x09,0x2a,0x86,0x48,0x86,0xf7,0x0d,0x01,
1641 0x07,0x01,0xa0,0x06,0x04,0x04,0x01,0x02,0x03,0x04,0x04,0x10,0x08,0xd6,0xc0,
1642 0x5a,0x21,0x51,0x2a,0x79,0xa1,0xdf,0xeb,0x9d,0x2a,0x8f,0x26,0x2f };
1643 static const BYTE hashBareContent[] = {
1644 0x30,0x38,0x02,0x01,0x00,0x30,0x0c,0x06,0x08,0x2a,0x86,0x48,0x86,0xf7,0x0d,
1645 0x02,0x05,0x05,0x00,0x30,0x13,0x06,0x09,0x2a,0x86,0x48,0x86,0xf7,0x0d,0x01,
1646 0x07,0x01,0xa0,0x06,0x04,0x04,0x01,0x02,0x03,0x04,0x04,0x10,0x08,0xd6,0xc0,
1647 0x5a,0x21,0x51,0x2a,0x79,0xa1,0xdf,0xeb,0x9d,0x2a,0x8f,0x26,0x2f };
1648
1649 static void testMessageStore(void)
1650 {
1651     HCERTSTORE store;
1652     HCRYPTMSG msg = CryptMsgOpenToDecode(PKCS_7_ASN_ENCODING, 0, 0, 0, NULL,
1653      NULL);
1654     CRYPT_DATA_BLOB blob = { sizeof(signedWithCertAndCrlBareContent),
1655      (LPBYTE)signedWithCertAndCrlBareContent };
1656     DWORD count, size;
1657     BOOL ret;
1658
1659     /* Crashes
1660     store = CertOpenStore(CERT_STORE_PROV_MSG, 0, 0, 0, NULL);
1661      */
1662     SetLastError(0xdeadbeef);
1663     store = CertOpenStore(CERT_STORE_PROV_MSG, 0, 0, 0, msg);
1664     ok(!store && GetLastError() == CRYPT_E_INVALID_MSG_TYPE,
1665      "Expected CRYPT_E_INVALID_MSG_TYPE, got %08x\n", GetLastError());
1666     CryptMsgUpdate(msg, signedContent, sizeof(signedContent), TRUE);
1667     store = CertOpenStore(CERT_STORE_PROV_MSG, 0, 0, 0, msg);
1668     ok(store != NULL, "CertOpenStore failed: %08x\n", GetLastError());
1669     if (store)
1670     {
1671         PCCERT_CONTEXT cert = NULL;
1672         PCCRL_CONTEXT crl = NULL;
1673
1674         count = 0;
1675         do {
1676             cert = CertEnumCertificatesInStore(store, cert);
1677             if (cert)
1678                 count++;
1679         } while (cert);
1680         ok(count == 0, "Expected 0 certificates, got %d\n", count);
1681
1682         if (pCertEnumCRLsInStore)
1683         {
1684             count = 0;
1685             do {
1686                 crl = pCertEnumCRLsInStore(store, crl);
1687                 if (crl)
1688                     count++;
1689             } while (crl);
1690             ok(count == 0, "Expected 0 CRLs, got %d\n", count);
1691         }
1692
1693         /* Can add certs to a message store */
1694         ret = CertAddEncodedCertificateToStore(store, X509_ASN_ENCODING,
1695          bigCert, sizeof(bigCert), CERT_STORE_ADD_ALWAYS, NULL);
1696         ok(ret, "CertAddEncodedCertificateToStore failed: %08x\n",
1697          GetLastError());
1698         count = 0;
1699         do {
1700             cert = CertEnumCertificatesInStore(store, cert);
1701             if (cert)
1702                 count++;
1703         } while (cert);
1704         ok(count == 1, "Expected 1 certificate, got %d\n", count);
1705
1706         CertCloseStore(store, 0);
1707     }
1708     /* but the added certs weren't actually added to the message */
1709     size = sizeof(count);
1710     ret = CryptMsgGetParam(msg, CMSG_CERT_COUNT_PARAM, 0, &count, &size);
1711     ok(ret, "CryptMsgGetParam failed: %08x\n", GetLastError());
1712     ok(count == 0, "Expected 0 certificates, got %d\n", count);
1713     CryptMsgClose(msg);
1714
1715     /* Crashes
1716     store = CertOpenStore(CERT_STORE_PROV_PKCS7, 0, 0, 0, NULL);
1717      */
1718     store = CertOpenStore(CERT_STORE_PROV_PKCS7, 0, 0, 0, &blob);
1719     ok(store != NULL, "CertOpenStore failed: %08x\n", GetLastError());
1720     if (store)
1721     {
1722         DWORD count = 0;
1723         PCCERT_CONTEXT cert = NULL;
1724         PCCRL_CONTEXT crl = NULL;
1725
1726         do {
1727             cert = CertEnumCertificatesInStore(store, cert);
1728             if (cert)
1729                 count++;
1730         } while (cert);
1731         ok(count == 1, "Expected 1 certificate, got %d\n", count);
1732
1733         if (pCertEnumCRLsInStore)
1734         {
1735             count = 0;
1736             do {
1737                 crl = pCertEnumCRLsInStore(store, crl);
1738                 if (crl)
1739                     count++;
1740             } while (crl);
1741             ok(count == 1, "Expected 1 CRL, got %d\n", count);
1742         }
1743         CertCloseStore(store, 0);
1744     }
1745     /* Encoding appears to be ignored */
1746     store = CertOpenStore(CERT_STORE_PROV_PKCS7, X509_ASN_ENCODING, 0, 0,
1747      &blob);
1748     ok(store != NULL, "CertOpenStore failed: %08x\n", GetLastError());
1749     if (store)
1750         CertCloseStore(store, 0);
1751     /* Messages other than signed messages aren't allowed */
1752     blob.cbData = sizeof(hashContent);
1753     blob.pbData = (LPBYTE)hashContent;
1754     SetLastError(0xdeadbeef);
1755     store = CertOpenStore(CERT_STORE_PROV_PKCS7, 0, 0, 0, &blob);
1756     ok(!store && GetLastError() == CRYPT_E_INVALID_MSG_TYPE,
1757      "Expected CRYPT_E_INVALID_MSG_TYPE, got %08x\n", GetLastError());
1758     blob.cbData = sizeof(hashBareContent);
1759     blob.pbData = (LPBYTE)hashBareContent;
1760     SetLastError(0xdeadbeef);
1761     store = CertOpenStore(CERT_STORE_PROV_PKCS7, 0, 0, 0, &blob);
1762     ok(!store &&
1763      (GetLastError() == CRYPT_E_ASN1_BADTAG ||
1764       GetLastError() == OSS_DATA_ERROR), /* win9x */
1765      "Expected CRYPT_E_ASN1_BADTAG, got %08x\n", GetLastError());
1766 }
1767
1768 static void testSerializedStore(void)
1769 {
1770     HCERTSTORE store;
1771     CRYPT_DATA_BLOB blob;
1772
1773     if (0)
1774     {
1775         /* Crash */
1776         store = CertOpenStore(CERT_STORE_PROV_SERIALIZED, 0, 0, 0, NULL);
1777         store = CertOpenStore(CERT_STORE_PROV_SERIALIZED, 0, 0,
1778          CERT_STORE_DELETE_FLAG, NULL);
1779     }
1780     blob.cbData = sizeof(serializedStoreWithCert);
1781     blob.pbData = (BYTE *)serializedStoreWithCert;
1782     store = CertOpenStore(CERT_STORE_PROV_SERIALIZED, 0, 0,
1783      CERT_STORE_DELETE_FLAG, &blob);
1784     ok(!store && GetLastError() == ERROR_CALL_NOT_IMPLEMENTED,
1785      "Expected ERROR_CALL_NOT_IMPLEMENTED, got %08x\n", GetLastError());
1786     store = CertOpenStore(CERT_STORE_PROV_SERIALIZED, 0, 0, 0, &blob);
1787     ok(store != NULL, "CertOpenStore failed: %08x\n", GetLastError());
1788     if (store)
1789     {
1790         PCCERT_CONTEXT cert;
1791         PCCRL_CONTEXT crl;
1792
1793         cert = CertEnumCertificatesInStore(store, NULL);
1794         ok(cert != NULL, "CertEnumCertificatesInStore failed: %08x\n",
1795          GetLastError());
1796         cert = CertEnumCertificatesInStore(store, cert);
1797         ok(!cert, "Expected only one cert\n");
1798         if (pCertEnumCRLsInStore)
1799         {
1800             crl = pCertEnumCRLsInStore(store, NULL);
1801             ok(!crl, "Expected no CRLs\n");
1802         }
1803         CertCloseStore(store, 0);
1804     }
1805     blob.cbData = sizeof(serializedStoreWithCertAndCRL);
1806     blob.pbData = (BYTE *)serializedStoreWithCertAndCRL;
1807     store = CertOpenStore(CERT_STORE_PROV_SERIALIZED, 0, 0, 0, &blob);
1808     ok(store != NULL, "CertOpenStore failed: %08x\n", GetLastError());
1809     if (store)
1810     {
1811         PCCERT_CONTEXT cert;
1812         PCCRL_CONTEXT crl;
1813
1814         cert = CertEnumCertificatesInStore(store, NULL);
1815         ok(cert != NULL, "CertEnumCertificatesInStore failed: %08x\n",
1816          GetLastError());
1817         cert = CertEnumCertificatesInStore(store, cert);
1818         ok(!cert, "Expected only one cert\n");
1819         if (pCertEnumCRLsInStore)
1820         {
1821             crl = pCertEnumCRLsInStore(store, NULL);
1822             ok(crl != NULL, "CertEnumCRLsInStore failed: %08x\n",
1823              GetLastError());
1824             crl = pCertEnumCRLsInStore(store, crl);
1825             ok(!crl, "Expected only one CRL\n");
1826         }
1827         CertCloseStore(store, 0);
1828     }
1829 }
1830
1831 static void testCertOpenSystemStore(void)
1832 {
1833     HCERTSTORE store;
1834
1835     store = CertOpenSystemStoreW(0, NULL);
1836     ok(!store && GetLastError() == E_INVALIDARG,
1837      "Expected E_INVALIDARG, got %08x\n", GetLastError());
1838     /* This succeeds, and on WinXP at least, the Bogus key is created under
1839      * HKCU (but not under HKLM, even when run as an administrator.)
1840      */
1841     store = CertOpenSystemStoreW(0, BogusW);
1842     ok(store != 0, "CertOpenSystemStore failed: %08x\n", GetLastError());
1843     if (store)
1844         CertCloseStore(store, 0);
1845     /* Delete it so other tests succeed next time around */
1846     store = CertOpenStore(CERT_STORE_PROV_SYSTEM, 0, 0,
1847      CERT_SYSTEM_STORE_CURRENT_USER | CERT_STORE_DELETE_FLAG, BogusW);
1848     RegDeleteKeyW(HKEY_CURRENT_USER, BogusPathW);
1849 }
1850
1851 struct EnumSystemStoreInfo
1852 {
1853     BOOL  goOn;
1854     DWORD storeCount;
1855 };
1856
1857 static BOOL CALLBACK enumSystemStoreCB(const void *systemStore, DWORD dwFlags,
1858  PCERT_SYSTEM_STORE_INFO pStoreInfo, void *pvReserved, void *pvArg)
1859 {
1860     struct EnumSystemStoreInfo *info = pvArg;
1861
1862     info->storeCount++;
1863     return info->goOn;
1864 }
1865
1866 static void testCertEnumSystemStore(void)
1867 {
1868     BOOL ret;
1869     struct EnumSystemStoreInfo info = { FALSE, 0 };
1870
1871     if (!pCertEnumSystemStore)
1872     {
1873         win_skip("CertEnumSystemStore() is not available\n");
1874         return;
1875     }
1876
1877     SetLastError(0xdeadbeef);
1878     ret = pCertEnumSystemStore(0, NULL, NULL, NULL);
1879     ok(!ret && GetLastError() == ERROR_FILE_NOT_FOUND,
1880      "Expected ERROR_FILE_NOT_FOUND, got %08x\n", GetLastError());
1881     /* Crashes
1882     ret = pCertEnumSystemStore(CERT_SYSTEM_STORE_LOCAL_MACHINE, NULL, NULL,
1883      NULL);
1884      */
1885
1886     SetLastError(0xdeadbeef);
1887     ret = pCertEnumSystemStore(CERT_SYSTEM_STORE_LOCAL_MACHINE, NULL, &info,
1888      enumSystemStoreCB);
1889     /* Callback returning FALSE stops enumeration */
1890     ok(!ret, "Expected CertEnumSystemStore to stop\n");
1891     ok(info.storeCount == 0 || info.storeCount == 1,
1892      "Expected 0 or 1 stores\n");
1893
1894     info.goOn = TRUE;
1895     info.storeCount = 0;
1896     ret = pCertEnumSystemStore(CERT_SYSTEM_STORE_LOCAL_MACHINE, NULL, &info,
1897      enumSystemStoreCB);
1898     ok(ret, "CertEnumSystemStore failed: %08x\n", GetLastError());
1899     /* There should always be at least My, Root, and CA stores */
1900     ok(info.storeCount == 0 || info.storeCount >= 3,
1901      "Expected at least 3 stores\n");
1902 }
1903
1904 static void testStoreProperty(void)
1905 {
1906     HCERTSTORE store;
1907     BOOL ret;
1908     DWORD propID, size = 0, state;
1909     CRYPT_DATA_BLOB blob;
1910
1911     if (!pCertGetStoreProperty || !pCertSetStoreProperty)
1912     {
1913         win_skip("CertGet/SetStoreProperty() is not available\n");
1914         return;
1915     }
1916
1917     /* Crash
1918     ret = pCertGetStoreProperty(NULL, 0, NULL, NULL);
1919     ret = pCertGetStoreProperty(NULL, 0, NULL, &size);
1920     ret = pCertGetStoreProperty(store, 0, NULL, NULL);
1921      */
1922
1923     store = CertOpenStore(CERT_STORE_PROV_MEMORY, 0, 0,
1924      CERT_STORE_CREATE_NEW_FLAG, NULL);
1925     /* Check a missing prop ID */
1926     SetLastError(0xdeadbeef);
1927     ret = pCertGetStoreProperty(store, 0, NULL, &size);
1928     ok(!ret && GetLastError() == CRYPT_E_NOT_FOUND,
1929      "Expected CRYPT_E_NOT_FOUND, got %08x\n", GetLastError());
1930     /* Contrary to MSDN, CERT_ACCESS_STATE_PROP_ID is supported for stores.. */
1931     size = sizeof(state);
1932     ret = pCertGetStoreProperty(store, CERT_ACCESS_STATE_PROP_ID, &state, &size);
1933     ok(ret, "CertGetStoreProperty failed for CERT_ACCESS_STATE_PROP_ID: %08x\n",
1934      GetLastError());
1935     ok(!state, "Expected a non-persisted store\n");
1936     /* and CERT_STORE_LOCALIZED_NAME_PROP_ID isn't supported by default. */
1937     size = 0;
1938     ret = pCertGetStoreProperty(store, CERT_STORE_LOCALIZED_NAME_PROP_ID, NULL,
1939      &size);
1940     ok(!ret && GetLastError() == CRYPT_E_NOT_FOUND,
1941      "Expected CRYPT_E_NOT_FOUND, got %08x\n", GetLastError());
1942     /* Delete an arbitrary property on a store */
1943     ret = pCertSetStoreProperty(store, CERT_FIRST_USER_PROP_ID, 0, NULL);
1944     ok(ret, "CertSetStoreProperty failed: %08x\n", GetLastError());
1945     /* Set an arbitrary property on a store */
1946     blob.pbData = (LPBYTE)&state;
1947     blob.cbData = sizeof(state);
1948     ret = pCertSetStoreProperty(store, CERT_FIRST_USER_PROP_ID, 0, &blob);
1949     ok(ret, "CertSetStoreProperty failed: %08x\n", GetLastError());
1950     /* Get an arbitrary property that's been set */
1951     ret = pCertGetStoreProperty(store, CERT_FIRST_USER_PROP_ID, NULL, &size);
1952     ok(ret, "CertGetStoreProperty failed: %08x\n", GetLastError());
1953     ok(size == sizeof(state), "Unexpected data size %d\n", size);
1954     ret = pCertGetStoreProperty(store, CERT_FIRST_USER_PROP_ID, &propID, &size);
1955     ok(ret, "CertGetStoreProperty failed: %08x\n", GetLastError());
1956     ok(propID == state, "CertGetStoreProperty got the wrong value\n");
1957     /* Delete it again */
1958     ret = pCertSetStoreProperty(store, CERT_FIRST_USER_PROP_ID, 0, NULL);
1959     ok(ret, "CertSetStoreProperty failed: %08x\n", GetLastError());
1960     /* And check that it's missing */
1961     SetLastError(0xdeadbeef);
1962     ret = pCertGetStoreProperty(store, CERT_FIRST_USER_PROP_ID, NULL, &size);
1963     ok(!ret && GetLastError() == CRYPT_E_NOT_FOUND,
1964      "Expected CRYPT_E_NOT_FOUND, got %08x\n", GetLastError());
1965     CertCloseStore(store, 0);
1966
1967     /* Recheck on the My store.. */
1968     store = CertOpenSystemStoreW(0, MyW);
1969     size = sizeof(state);
1970     ret = pCertGetStoreProperty(store, CERT_ACCESS_STATE_PROP_ID, &state, &size);
1971     ok(ret, "CertGetStoreProperty failed for CERT_ACCESS_STATE_PROP_ID: %08x\n",
1972      GetLastError());
1973     ok(state, "Expected a persisted store\n");
1974     SetLastError(0xdeadbeef);
1975     size = 0;
1976     ret = pCertGetStoreProperty(store, CERT_STORE_LOCALIZED_NAME_PROP_ID, NULL,
1977      &size);
1978     ok(!ret && GetLastError() == CRYPT_E_NOT_FOUND,
1979      "Expected CRYPT_E_NOT_FOUND, got %08x\n", GetLastError());
1980     CertCloseStore(store, 0);
1981 }
1982
1983 static void testAddSerialized(void)
1984 {
1985     BOOL ret;
1986     HCERTSTORE store;
1987     BYTE buf[sizeof(struct CertPropIDHeader) * 2 + 20 + sizeof(bigCert)] =
1988      { 0 };
1989     BYTE hash[20];
1990     struct CertPropIDHeader *hdr;
1991     PCCERT_CONTEXT context;
1992
1993     ret = CertAddSerializedElementToStore(0, NULL, 0, 0, 0, 0, NULL, NULL);
1994     ok(!ret && GetLastError() == ERROR_END_OF_MEDIA,
1995      "Expected ERROR_END_OF_MEDIA, got %08x\n", GetLastError());
1996
1997     store = CertOpenStore(CERT_STORE_PROV_MEMORY, 0, 0,
1998      CERT_STORE_CREATE_NEW_FLAG, NULL);
1999     ok(store != 0, "CertOpenStore failed: %08x\n", GetLastError());
2000
2001     ret = CertAddSerializedElementToStore(store, NULL, 0, 0, 0, 0, NULL, NULL);
2002     ok(!ret && GetLastError() == ERROR_END_OF_MEDIA,
2003      "Expected ERROR_END_OF_MEDIA, got %08x\n", GetLastError());
2004
2005     /* Test with an empty property */
2006     hdr = (struct CertPropIDHeader *)buf;
2007     hdr->propID = CERT_CERT_PROP_ID;
2008     hdr->unknown1 = 1;
2009     hdr->cb = 0;
2010     ret = CertAddSerializedElementToStore(store, buf, sizeof(buf), 0, 0, 0,
2011      NULL, NULL);
2012     ok(!ret && GetLastError() == E_INVALIDARG,
2013      "Expected E_INVALIDARG, got %08x\n", GetLastError());
2014     /* Test with a bad size in property header */
2015     hdr->cb = sizeof(bigCert) - 1;
2016     memcpy(buf + sizeof(struct CertPropIDHeader), bigCert, sizeof(bigCert));
2017     ret = CertAddSerializedElementToStore(store, buf, sizeof(buf), 0, 0, 0,
2018      NULL, NULL);
2019     ok(!ret && GetLastError() == E_INVALIDARG,
2020      "Expected E_INVALIDARG, got %08x\n", GetLastError());
2021     ret = CertAddSerializedElementToStore(store, buf,
2022      sizeof(struct CertPropIDHeader) + sizeof(bigCert), 0, 0, 0, NULL,
2023      NULL);
2024     ok(!ret && GetLastError() == E_INVALIDARG,
2025      "Expected E_INVALIDARG, got %08x\n", GetLastError());
2026     ret = CertAddSerializedElementToStore(store, buf,
2027      sizeof(struct CertPropIDHeader) + sizeof(bigCert), CERT_STORE_ADD_NEW,
2028      0, 0, NULL, NULL);
2029     ok(!ret && GetLastError() == E_INVALIDARG,
2030      "Expected E_INVALIDARG, got %08x\n", GetLastError());
2031     /* Kosher size in property header, but no context type */
2032     hdr->cb = sizeof(bigCert);
2033     ret = CertAddSerializedElementToStore(store, buf, sizeof(buf), 0, 0, 0,
2034      NULL, NULL);
2035     ok(!ret && GetLastError() == E_INVALIDARG,
2036      "Expected E_INVALIDARG, got %08x\n", GetLastError());
2037     ret = CertAddSerializedElementToStore(store, buf,
2038      sizeof(struct CertPropIDHeader) + sizeof(bigCert), 0, 0, 0, NULL,
2039      NULL);
2040     ok(!ret && GetLastError() == E_INVALIDARG,
2041      "Expected E_INVALIDARG, got %08x\n", GetLastError());
2042     ret = CertAddSerializedElementToStore(store, buf,
2043      sizeof(struct CertPropIDHeader) + sizeof(bigCert), CERT_STORE_ADD_NEW,
2044      0, 0, NULL, NULL);
2045     ok(!ret && GetLastError() == E_INVALIDARG,
2046      "Expected E_INVALIDARG, got %08x\n", GetLastError());
2047     /* With a bad context type */
2048     ret = CertAddSerializedElementToStore(store, buf, sizeof(buf), 0, 0,
2049      CERT_STORE_CRL_CONTEXT_FLAG, NULL, NULL);
2050     ok(!ret && GetLastError() == E_INVALIDARG,
2051      "Expected E_INVALIDARG, got %08x\n", GetLastError());
2052     ret = CertAddSerializedElementToStore(store, buf,
2053      sizeof(struct CertPropIDHeader) + sizeof(bigCert), 0, 0,
2054      CERT_STORE_CRL_CONTEXT_FLAG, NULL, NULL);
2055     ok(!ret && GetLastError() == E_INVALIDARG,
2056      "Expected E_INVALIDARG, got %08x\n", GetLastError());
2057     ret = CertAddSerializedElementToStore(store, buf,
2058      sizeof(struct CertPropIDHeader) + sizeof(bigCert), CERT_STORE_ADD_NEW,
2059      0, CERT_STORE_CRL_CONTEXT_FLAG, NULL, NULL);
2060     ok(!ret && GetLastError() == E_INVALIDARG,
2061      "Expected E_INVALIDARG, got %08x\n", GetLastError());
2062     /* Bad unknown field, good type */
2063     hdr->unknown1 = 2;
2064     ret = CertAddSerializedElementToStore(store, buf, sizeof(buf), 0, 0,
2065      CERT_STORE_CERTIFICATE_CONTEXT_FLAG, NULL, NULL);
2066     ok(!ret && GetLastError() == ERROR_FILE_NOT_FOUND,
2067      "Expected ERROR_FILE_NOT_FOUND got %08x\n", GetLastError());
2068     ret = CertAddSerializedElementToStore(store, buf,
2069      sizeof(struct CertPropIDHeader) + sizeof(bigCert), 0, 0,
2070      CERT_STORE_CERTIFICATE_CONTEXT_FLAG, NULL, NULL);
2071     ok(!ret && GetLastError() == ERROR_FILE_NOT_FOUND,
2072      "Expected ERROR_FILE_NOT_FOUND got %08x\n", GetLastError());
2073     ret = CertAddSerializedElementToStore(store, buf,
2074      sizeof(struct CertPropIDHeader) + sizeof(bigCert), CERT_STORE_ADD_NEW,
2075      0, CERT_STORE_CERTIFICATE_CONTEXT_FLAG, NULL, NULL);
2076     ok(!ret && GetLastError() == ERROR_FILE_NOT_FOUND,
2077      "Expected ERROR_FILE_NOT_FOUND got %08x\n", GetLastError());
2078     /* Most everything okay, but bad add disposition */
2079     hdr->unknown1 = 1;
2080     /* This crashes
2081     ret = CertAddSerializedElementToStore(store, buf, sizeof(buf), 0, 0,
2082      CERT_STORE_CERTIFICATE_CONTEXT_FLAG, NULL, NULL);
2083      * as does this
2084     ret = CertAddSerializedElementToStore(store, buf,
2085      sizeof(struct CertPropIDHeader) + sizeof(bigCert), 0, 0,
2086      CERT_STORE_CERTIFICATE_CONTEXT_FLAG, NULL, NULL);
2087      */
2088     /* Everything okay, but buffer's too big */
2089     ret = CertAddSerializedElementToStore(store, buf, sizeof(buf),
2090      CERT_STORE_ADD_NEW, 0, CERT_STORE_CERTIFICATE_CONTEXT_FLAG, NULL, NULL);
2091     ok(ret, "CertAddSerializedElementToStore failed: %08x\n", GetLastError());
2092     /* Everything okay, check it's not re-added */
2093     ret = CertAddSerializedElementToStore(store, buf,
2094      sizeof(struct CertPropIDHeader) + sizeof(bigCert), CERT_STORE_ADD_NEW,
2095      0, CERT_STORE_CERTIFICATE_CONTEXT_FLAG, NULL, NULL);
2096     ok(!ret && GetLastError() == CRYPT_E_EXISTS,
2097      "Expected CRYPT_E_EXISTS, got %08x\n", GetLastError());
2098
2099     context = CertEnumCertificatesInStore(store, NULL);
2100     ok(context != NULL, "Expected a cert\n");
2101     if (context)
2102         CertDeleteCertificateFromStore(context);
2103
2104     /* Try adding with a bogus hash.  Oddly enough, it succeeds, and the hash,
2105      * when queried, is the real hash rather than the bogus hash.
2106      */
2107     hdr = (struct CertPropIDHeader *)(buf + sizeof(struct CertPropIDHeader) +
2108      sizeof(bigCert));
2109     hdr->propID = CERT_HASH_PROP_ID;
2110     hdr->unknown1 = 1;
2111     hdr->cb = sizeof(hash);
2112     memset(hash, 0xc, sizeof(hash));
2113     memcpy((LPBYTE)hdr + sizeof(struct CertPropIDHeader), hash, sizeof(hash));
2114     ret = CertAddSerializedElementToStore(store, buf, sizeof(buf),
2115      CERT_STORE_ADD_NEW, 0, CERT_STORE_CERTIFICATE_CONTEXT_FLAG, NULL,
2116      (const void **)&context);
2117     ok(ret, "CertAddSerializedElementToStore failed: %08x\n", GetLastError());
2118     if (context)
2119     {
2120         BYTE hashVal[20], realHash[20];
2121         DWORD size = sizeof(hashVal);
2122
2123         ret = CryptHashCertificate(0, 0, 0, bigCert, sizeof(bigCert),
2124          realHash, &size);
2125         ok(ret, "CryptHashCertificate failed: %08x\n", GetLastError());
2126         ret = CertGetCertificateContextProperty(context, CERT_HASH_PROP_ID,
2127          hashVal, &size);
2128         ok(ret, "CertGetCertificateContextProperty failed: %08x\n",
2129          GetLastError());
2130         ok(!memcmp(hashVal, realHash, size), "Unexpected hash\n");
2131         CertFreeCertificateContext(context);
2132     }
2133
2134     CertCloseStore(store, 0);
2135 }
2136
2137 static const BYTE serializedCertWithFriendlyName[] = {
2138 0x0b,0x00,0x00,0x00,0x01,0x00,0x00,0x00,0x12,0x00,0x00,0x00,0x57,0x00,0x69,
2139 0x00,0x6e,0x00,0x65,0x00,0x54,0x00,0x65,0x00,0x73,0x00,0x74,0x00,0x00,0x00,
2140 0x20,0x00,0x00,0x00,0x01,0x00,0x00,0x00,0x7c,0x00,0x00,0x00,0x30,0x7a,0x02,
2141 0x01,0x01,0x30,0x02,0x06,0x00,0x30,0x15,0x31,0x13,0x30,0x11,0x06,0x03,0x55,
2142 0x04,0x03,0x13,0x0a,0x4a,0x75,0x61,0x6e,0x20,0x4c,0x61,0x6e,0x67,0x00,0x30,
2143 0x22,0x18,0x0f,0x31,0x36,0x30,0x31,0x30,0x31,0x30,0x31,0x30,0x30,0x30,0x30,
2144 0x30,0x30,0x5a,0x18,0x0f,0x31,0x36,0x30,0x31,0x30,0x31,0x30,0x31,0x30,0x30,
2145 0x30,0x30,0x30,0x30,0x5a,0x30,0x15,0x31,0x13,0x30,0x11,0x06,0x03,0x55,0x04,
2146 0x03,0x13,0x0a,0x4a,0x75,0x61,0x6e,0x20,0x4c,0x61,0x6e,0x67,0x00,0x30,0x07,
2147 0x30,0x02,0x06,0x00,0x03,0x01,0x00,0xa3,0x16,0x30,0x14,0x30,0x12,0x06,0x03,
2148 0x55,0x1d,0x13,0x01,0x01,0xff,0x04,0x08,0x30,0x06,0x01,0x01,0xff,0x02,0x01,
2149 0x01 };
2150 static const BYTE serializedStoreWithCertWithFriendlyName[] = {
2151 0x00,0x00,0x00,0x00,0x43,0x45,0x52,0x54,0x0b,0x00,0x00,0x00,0x01,0x00,0x00,
2152 0x00,0x12,0x00,0x00,0x00,0x57,0x00,0x69,0x00,0x6e,0x00,0x65,0x00,0x54,0x00,
2153 0x65,0x00,0x73,0x00,0x74,0x00,0x00,0x00,0x20,0x00,0x00,0x00,0x01,0x00,0x00,
2154 0x00,0x7c,0x00,0x00,0x00,0x30,0x7a,0x02,0x01,0x01,0x30,0x02,0x06,0x00,0x30,
2155 0x15,0x31,0x13,0x30,0x11,0x06,0x03,0x55,0x04,0x03,0x13,0x0a,0x4a,0x75,0x61,
2156 0x6e,0x20,0x4c,0x61,0x6e,0x67,0x00,0x30,0x22,0x18,0x0f,0x31,0x36,0x30,0x31,
2157 0x30,0x31,0x30,0x31,0x30,0x30,0x30,0x30,0x30,0x30,0x5a,0x18,0x0f,0x31,0x36,
2158 0x30,0x31,0x30,0x31,0x30,0x31,0x30,0x30,0x30,0x30,0x30,0x30,0x5a,0x30,0x15,
2159 0x31,0x13,0x30,0x11,0x06,0x03,0x55,0x04,0x03,0x13,0x0a,0x4a,0x75,0x61,0x6e,
2160 0x20,0x4c,0x61,0x6e,0x67,0x00,0x30,0x07,0x30,0x02,0x06,0x00,0x03,0x01,0x00,
2161 0xa3,0x16,0x30,0x14,0x30,0x12,0x06,0x03,0x55,0x1d,0x13,0x01,0x01,0xff,0x04,
2162 0x08,0x30,0x06,0x01,0x01,0xff,0x02,0x01,0x01,0x00,0x00,0x00,0x00,0x00,0x00,
2163 0x00,0x00,0x00,0x00,0x00,0x00 };
2164 static const BYTE serializedStoreWithCertAndHash[] = {
2165 0x00,0x00,0x00,0x00,0x43,0x45,0x52,0x54,0x03,0x00,0x00,0x00,0x01,0x00,0x00,
2166 0x00,0x14,0x00,0x00,0x00,0x6e,0x30,0x90,0x71,0x5f,0xd9,0x23,0x56,0xeb,0xae,
2167 0x25,0x40,0xe6,0x22,0xda,0x19,0x26,0x02,0xa6,0x08,0x20,0x00,0x00,0x00,0x01,
2168 0x00,0x00,0x00,0x7c,0x00,0x00,0x00,0x30,0x7a,0x02,0x01,0x01,0x30,0x02,0x06,
2169 0x00,0x30,0x15,0x31,0x13,0x30,0x11,0x06,0x03,0x55,0x04,0x03,0x13,0x0a,0x4a,
2170 0x75,0x61,0x6e,0x20,0x4c,0x61,0x6e,0x67,0x00,0x30,0x22,0x18,0x0f,0x31,0x36,
2171 0x30,0x31,0x30,0x31,0x30,0x31,0x30,0x30,0x30,0x30,0x30,0x30,0x5a,0x18,0x0f,
2172 0x31,0x36,0x30,0x31,0x30,0x31,0x30,0x31,0x30,0x30,0x30,0x30,0x30,0x30,0x5a,
2173 0x30,0x15,0x31,0x13,0x30,0x11,0x06,0x03,0x55,0x04,0x03,0x13,0x0a,0x4a,0x75,
2174 0x61,0x6e,0x20,0x4c,0x61,0x6e,0x67,0x00,0x30,0x07,0x30,0x02,0x06,0x00,0x03,
2175 0x01,0x00,0xa3,0x16,0x30,0x14,0x30,0x12,0x06,0x03,0x55,0x1d,0x13,0x01,0x01,
2176 0xff,0x04,0x08,0x30,0x06,0x01,0x01,0xff,0x02,0x01,0x01,0x00,0x00,0x00,0x00,
2177 0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00 };
2178
2179 static void testAddCertificateLink(void)
2180 {
2181     BOOL ret;
2182     HCERTSTORE store1, store2;
2183     PCCERT_CONTEXT source, linked;
2184     DWORD size;
2185     LPBYTE buf;
2186     CERT_NAME_BLOB blob;
2187     static const WCHAR szPrefix[] = { 'c','e','r',0 };
2188     static const WCHAR szDot[] = { '.',0 };
2189     static const WCHAR WineTestW[] = { 'W','i','n','e','T','e','s','t',0 };
2190     WCHAR filename1[MAX_PATH], filename2[MAX_PATH];
2191     HANDLE file;
2192
2193     if (!pCertAddCertificateLinkToStore)
2194     {
2195         win_skip("CertAddCertificateLinkToStore not found\n");
2196         return;
2197     }
2198
2199     if (0)
2200     {
2201         /* Crashes, i.e. the store is dereferenced without checking. */
2202         ret = pCertAddCertificateLinkToStore(NULL, NULL, 0, NULL);
2203     }
2204
2205     /* Adding a certificate link to a store requires a valid add disposition */
2206     store1 = CertOpenStore(CERT_STORE_PROV_MEMORY, 0, 0,
2207      CERT_STORE_CREATE_NEW_FLAG, NULL);
2208     SetLastError(0xdeadbeef);
2209     ret = pCertAddCertificateLinkToStore(store1, NULL, 0, NULL);
2210     ok(!ret && GetLastError() == E_INVALIDARG,
2211      "expected E_INVALIDARG, got %08x\n", GetLastError());
2212     source = CertCreateCertificateContext(X509_ASN_ENCODING, bigCert,
2213      sizeof(bigCert));
2214     SetLastError(0xdeadbeef);
2215     ret = pCertAddCertificateLinkToStore(store1, source, 0, NULL);
2216     ok(!ret && GetLastError() == E_INVALIDARG,
2217      "expected E_INVALIDARG, got %08x\n", GetLastError());
2218     ret = pCertAddCertificateLinkToStore(store1, source, CERT_STORE_ADD_ALWAYS,
2219      NULL);
2220     ok(ret, "CertAddCertificateLinkToStore failed: %08x\n", GetLastError());
2221     if (0)
2222     {
2223         /* Crashes, i.e. the source certificate is dereferenced without
2224          * checking when a valid add disposition is given.
2225          */
2226         ret = pCertAddCertificateLinkToStore(store1, NULL, CERT_STORE_ADD_ALWAYS,
2227          NULL);
2228     }
2229     CertCloseStore(store1, 0);
2230
2231     store1 = CertOpenStore(CERT_STORE_PROV_MEMORY, 0, 0,
2232      CERT_STORE_CREATE_NEW_FLAG, NULL);
2233     ret = pCertAddCertificateLinkToStore(store1, source, CERT_STORE_ADD_ALWAYS,
2234      &linked);
2235     ok(ret, "CertAddCertificateLinkToStore failed: %08x\n", GetLastError());
2236     if (ret)
2237     {
2238         ok(linked->hCertStore == store1, "unexpected store\n");
2239         ret = CertSerializeCertificateStoreElement(linked, 0, NULL, &size);
2240         ok(ret, "CertSerializeCertificateStoreElement failed: %08x\n",
2241          GetLastError());
2242         buf = HeapAlloc(GetProcessHeap(), 0, size);
2243         if (buf)
2244         {
2245             ret = CertSerializeCertificateStoreElement(linked, 0, buf, &size);
2246             /* The serialized linked certificate is identical to the serialized
2247              * original certificate.
2248              */
2249             ok(size == sizeof(serializedCert), "Wrong size %d\n", size);
2250             ok(!memcmp(serializedCert, buf, size),
2251              "Unexpected serialized cert\n");
2252             HeapFree(GetProcessHeap(), 0, buf);
2253         }
2254         /* Set a friendly name on the source certificate... */
2255         blob.pbData = (LPBYTE)WineTestW;
2256         blob.cbData = sizeof(WineTestW);
2257         ret = CertSetCertificateContextProperty(source,
2258          CERT_FRIENDLY_NAME_PROP_ID, 0, &blob);
2259         ok(ret, "CertSetCertificateContextProperty failed: %08x\n",
2260          GetLastError());
2261         /* and the linked certificate has the same friendly name. */
2262         ret = CertGetCertificateContextProperty(linked,
2263          CERT_FRIENDLY_NAME_PROP_ID, NULL, &size);
2264         ok(ret, "CertGetCertificateContextProperty failed: %08x\n",
2265          GetLastError());
2266         buf = HeapAlloc(GetProcessHeap(), 0, size);
2267         if (buf)
2268         {
2269             ret = CertGetCertificateContextProperty(linked,
2270              CERT_FRIENDLY_NAME_PROP_ID, buf, &size);
2271             ok(!lstrcmpW((LPCWSTR)buf, WineTestW),
2272              "unexpected friendly name\n");
2273             HeapFree(GetProcessHeap(), 0, buf);
2274         }
2275         CertFreeCertificateContext(linked);
2276     }
2277     CertFreeCertificateContext(source);
2278     CertCloseStore(store1, 0);
2279
2280     /* Test adding a cert to a file store, committing the change to the store,
2281      * and creating a link to the resulting cert.
2282      */
2283     if (!GetTempFileNameW(szDot, szPrefix, 0, filename1))
2284        return;
2285
2286     DeleteFileW(filename1);
2287     file = CreateFileW(filename1, GENERIC_READ | GENERIC_WRITE, 0, NULL,
2288      CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
2289     if (file == INVALID_HANDLE_VALUE)
2290         return;
2291
2292     store1 = CertOpenStore(CERT_STORE_PROV_FILE, 0, 0,
2293      CERT_FILE_STORE_COMMIT_ENABLE_FLAG, file);
2294     ok(store1 != NULL, "CertOpenStore failed: %08x\n", GetLastError());
2295     CloseHandle(file);
2296
2297     ret = CertAddEncodedCertificateToStore(store1, X509_ASN_ENCODING,
2298      bigCert, sizeof(bigCert), CERT_STORE_ADD_ALWAYS, &source);
2299     ok(ret, "CertAddEncodedCertificateToStore failed: %08x\n",
2300      GetLastError());
2301
2302     /* Test adding a link to a memory store. */
2303     store2 = CertOpenStore(CERT_STORE_PROV_MEMORY, 0, 0,
2304      CERT_STORE_CREATE_NEW_FLAG, NULL);
2305     ret = pCertAddCertificateLinkToStore(store2, source, CERT_STORE_ADD_ALWAYS,
2306      &linked);
2307     ok(ret, "CertAddCertificateLinkToStore failed: %08x\n", GetLastError());
2308     if (ret)
2309     {
2310         ok(linked->hCertStore == store2, "unexpected store\n");
2311         ret = CertSerializeCertificateStoreElement(linked, 0, NULL, &size);
2312         ok(ret, "CertSerializeCertificateStoreElement failed: %08x\n",
2313          GetLastError());
2314         buf = HeapAlloc(GetProcessHeap(), 0, size);
2315         if (buf)
2316         {
2317             ret = CertSerializeCertificateStoreElement(linked, 0, buf, &size);
2318             /* The serialized linked certificate is identical to the serialized
2319              * original certificate.
2320              */
2321             ok(size == sizeof(serializedCert), "Wrong size %d\n", size);
2322             ok(!memcmp(serializedCert, buf, size),
2323              "Unexpected serialized cert\n");
2324             HeapFree(GetProcessHeap(), 0, buf);
2325         }
2326         /* Set a friendly name on the source certificate... */
2327         blob.pbData = (LPBYTE)WineTestW;
2328         blob.cbData = sizeof(WineTestW);
2329         ret = CertSetCertificateContextProperty(source,
2330          CERT_FRIENDLY_NAME_PROP_ID, 0, &blob);
2331         ok(ret, "CertSetCertificateContextProperty failed: %08x\n",
2332          GetLastError());
2333         /* and the linked certificate has the same friendly name. */
2334         ret = CertGetCertificateContextProperty(linked,
2335          CERT_FRIENDLY_NAME_PROP_ID, NULL, &size);
2336         ok(ret, "CertGetCertificateContextProperty failed: %08x\n",
2337          GetLastError());
2338         buf = HeapAlloc(GetProcessHeap(), 0, size);
2339         if (buf)
2340         {
2341             ret = CertGetCertificateContextProperty(linked,
2342              CERT_FRIENDLY_NAME_PROP_ID, buf, &size);
2343             ok(!lstrcmpW((LPCWSTR)buf, WineTestW),
2344              "unexpected friendly name\n");
2345             HeapFree(GetProcessHeap(), 0, buf);
2346         }
2347         CertFreeCertificateContext(linked);
2348     }
2349     CertCloseStore(store2, 0);
2350
2351     if (!GetTempFileNameW(szDot, szPrefix, 0, filename2))
2352        return;
2353
2354     DeleteFileW(filename2);
2355     file = CreateFileW(filename2, GENERIC_READ | GENERIC_WRITE, 0, NULL,
2356      CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
2357     if (file == INVALID_HANDLE_VALUE)
2358         return;
2359
2360     store2 = CertOpenStore(CERT_STORE_PROV_FILE, 0, 0,
2361      CERT_FILE_STORE_COMMIT_ENABLE_FLAG, file);
2362     ok(store2 != NULL, "CertOpenStore failed: %08x\n", GetLastError());
2363     CloseHandle(file);
2364     /* Test adding a link to a file store. */
2365     ret = pCertAddCertificateLinkToStore(store2, source, CERT_STORE_ADD_ALWAYS,
2366      &linked);
2367     ok(ret, "CertAddCertificateLinkToStore failed: %08x\n", GetLastError());
2368     if (ret)
2369     {
2370         ok(linked->hCertStore == store2, "unexpected store\n");
2371         ret = CertSerializeCertificateStoreElement(linked, 0, NULL, &size);
2372         ok(ret, "CertSerializeCertificateStoreElement failed: %08x\n",
2373          GetLastError());
2374         buf = HeapAlloc(GetProcessHeap(), 0, size);
2375         if (buf)
2376         {
2377             ret = CertSerializeCertificateStoreElement(linked, 0, buf, &size);
2378             ok(ret, "CertSerializeCertificateStoreElement failed: %08x\n",
2379              GetLastError());
2380             /* The serialized linked certificate now contains the friendly
2381              * name property.
2382              */
2383             ok(size == sizeof(serializedCertWithFriendlyName),
2384              "Wrong size %d\n", size);
2385             ok(!memcmp(serializedCertWithFriendlyName, buf, size),
2386              "Unexpected serialized cert\n");
2387             HeapFree(GetProcessHeap(), 0, buf);
2388         }
2389         CertFreeCertificateContext(linked);
2390         compareStore(store2, "file store -> file store",
2391          serializedStoreWithCertWithFriendlyName,
2392          sizeof(serializedStoreWithCertWithFriendlyName), FALSE);
2393     }
2394     CertCloseStore(store2, 0);
2395     DeleteFileW(filename2);
2396
2397     CertFreeCertificateContext(source);
2398
2399     CertCloseStore(store1, 0);
2400     DeleteFileW(filename1);
2401
2402     /* Test adding a link to a system store (which is a collection store.) */
2403     store1 = CertOpenSystemStoreA(0, "My");
2404     source = CertCreateCertificateContext(X509_ASN_ENCODING, bigCert,
2405      sizeof(bigCert));
2406     SetLastError(0xdeadbeef);
2407     ret = pCertAddCertificateLinkToStore(store1, source, CERT_STORE_ADD_ALWAYS,
2408      &linked);
2409     ok(!ret && GetLastError() == E_INVALIDARG,
2410      "expected E_INVALIDARG, got %08x\n", GetLastError());
2411     CertFreeCertificateContext(source);
2412
2413     /* Test adding a link to a file store, where the linked certificate is
2414      * in a system store.
2415      */
2416     ret = CertAddEncodedCertificateToStore(store1, X509_ASN_ENCODING,
2417      bigCert, sizeof(bigCert), CERT_STORE_ADD_ALWAYS, &source);
2418     ok(ret, "CertAddEncodedCertificateToStore failed: %08x\n",
2419      GetLastError());
2420     if (!GetTempFileNameW(szDot, szPrefix, 0, filename1))
2421        return;
2422
2423     DeleteFileW(filename1);
2424     file = CreateFileW(filename1, GENERIC_READ | GENERIC_WRITE, 0, NULL,
2425      CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
2426     if (file == INVALID_HANDLE_VALUE)
2427         return;
2428
2429     store2 = CertOpenStore(CERT_STORE_PROV_FILE, 0, 0,
2430      CERT_FILE_STORE_COMMIT_ENABLE_FLAG, file);
2431     ok(store2 != NULL, "CertOpenStore failed: %08x\n", GetLastError());
2432     CloseHandle(file);
2433
2434     ret = pCertAddCertificateLinkToStore(store2, source, CERT_STORE_ADD_ALWAYS,
2435      &linked);
2436     ok(ret, "CertAddCertificateLinkToStore failed: %08x\n", GetLastError());
2437     if (ret)
2438     {
2439         ok(linked->hCertStore == store2, "unexpected store\n");
2440         ret = pCertControlStore(store2, 0, CERT_STORE_CTRL_COMMIT, NULL);
2441         ok(ret, "CertControlStore failed: %d\n", ret);
2442         compareStore(store2, "file store -> system store",
2443          serializedStoreWithCertAndHash,
2444          sizeof(serializedStoreWithCertAndHash), TRUE);
2445         CertFreeCertificateContext(linked);
2446     }
2447
2448     CertCloseStore(store2, 0);
2449     DeleteFileW(filename1);
2450
2451     /* Test adding a link to a registry store, where the linked certificate is
2452      * in a system store.
2453      */
2454     store2 = CertOpenStore(CERT_STORE_PROV_SYSTEM_REGISTRY, 0, 0,
2455      CERT_SYSTEM_STORE_CURRENT_USER, WineTestW);
2456     ok(store2 != NULL, "CertOpenStore failed: %08x\n", GetLastError());
2457     ret = pCertAddCertificateLinkToStore(store2, source, CERT_STORE_ADD_ALWAYS,
2458      &linked);
2459     ok(ret, "CertAddCertificateLinkToStore failed: %08x\n", GetLastError());
2460     if (ret)
2461     {
2462         ok(linked->hCertStore == store2, "unexpected store\n");
2463         CertDeleteCertificateFromStore(linked);
2464     }
2465     CertCloseStore(store2, 0);
2466
2467     CertFreeCertificateContext(source);
2468     CertCloseStore(store1, 0);
2469 }
2470
2471 static DWORD countCertsInStore(HCERTSTORE store)
2472 {
2473     PCCERT_CONTEXT cert = NULL;
2474     DWORD certs = 0;
2475
2476     do {
2477         cert = CertEnumCertificatesInStore(store, cert);
2478         if (cert)
2479             certs++;
2480     } while (cert);
2481     return certs;
2482 }
2483
2484 static DWORD countCRLsInStore(HCERTSTORE store)
2485 {
2486     PCCRL_CONTEXT crl = NULL;
2487     DWORD crls = 0;
2488
2489     do {
2490         crl = pCertEnumCRLsInStore(store, crl);
2491         if (crl)
2492             crls++;
2493     } while (crl);
2494     return crls;
2495 }
2496
2497 static void test_I_UpdateStore(void)
2498 {
2499     HMODULE lib = GetModuleHandleA("crypt32");
2500     BOOL (WINAPI *pI_CertUpdatestore)(HCERTSTORE, HCERTSTORE, DWORD, DWORD) =
2501      (void *)GetProcAddress(lib, "I_CertUpdateStore");
2502     BOOL ret;
2503     HCERTSTORE store1, store2;
2504     PCCERT_CONTEXT cert;
2505     DWORD certs;
2506
2507     if (!pI_CertUpdatestore)
2508     {
2509         win_skip("No I_CertUpdateStore\n");
2510         return;
2511     }
2512     store1 = CertOpenStore(CERT_STORE_PROV_MEMORY, 0, 0,
2513      CERT_STORE_CREATE_NEW_FLAG, NULL);
2514     store2 = CertOpenStore(CERT_STORE_PROV_MEMORY, 0, 0,
2515      CERT_STORE_CREATE_NEW_FLAG, NULL);
2516
2517     /* Crash
2518     ret = pI_CertUpdatestore(NULL, NULL, 0, 0);
2519     ret = pI_CertUpdatestore(store1, NULL, 0, 0);
2520     ret = pI_CertUpdatestore(NULL, store2, 0, 0);
2521      */
2522     ret = pI_CertUpdatestore(store1, store2, 0, 0);
2523     ok(ret, "I_CertUpdateStore failed: %08x\n", GetLastError());
2524
2525     CertAddEncodedCertificateToStore(store2, X509_ASN_ENCODING, bigCert,
2526      sizeof(bigCert), CERT_STORE_ADD_ALWAYS, &cert);
2527     /* I_CertUpdateStore adds the contexts from store2 to store1 */
2528     ret = pI_CertUpdatestore(store1, store2, 0, 0);
2529     ok(ret, "I_CertUpdateStore failed: %08x\n", GetLastError());
2530     certs = countCertsInStore(store1);
2531     ok(certs == 1, "Expected 1 cert, got %d\n", certs);
2532     /* Calling it a second time has no effect */
2533     ret = pI_CertUpdatestore(store1, store2, 0, 0);
2534     ok(ret, "I_CertUpdateStore failed: %08x\n", GetLastError());
2535     certs = countCertsInStore(store1);
2536     ok(certs == 1, "Expected 1 cert, got %d\n", certs);
2537
2538     /* The last parameters to I_CertUpdateStore appear to be ignored */
2539     ret = pI_CertUpdatestore(store1, store2, 1, 0);
2540     ok(ret, "I_CertUpdateStore failed: %08x\n", GetLastError());
2541     ret = pI_CertUpdatestore(store1, store2, 0, 1);
2542     ok(ret, "I_CertUpdateStore failed: %08x\n", GetLastError());
2543
2544     CertAddEncodedCRLToStore(store2, X509_ASN_ENCODING, signedCRL,
2545      sizeof(signedCRL), CERT_STORE_ADD_ALWAYS, NULL);
2546
2547     /* I_CertUpdateStore also adds the CRLs from store2 to store1 */
2548     ret = pI_CertUpdatestore(store1, store2, 0, 0);
2549     ok(ret, "I_CertUpdateStore failed: %08x\n", GetLastError());
2550     certs = countCertsInStore(store1);
2551     ok(certs == 1, "Expected 1 cert, got %d\n", certs);
2552     if (pCertEnumCRLsInStore)
2553     {
2554         certs = countCRLsInStore(store1);
2555         ok(certs == 1, "Expected 1 CRL, got %d\n", certs);
2556     }
2557
2558     CertDeleteCertificateFromStore(cert);
2559     /* If a context is deleted from store2, I_CertUpdateStore deletes it
2560      * from store1
2561      */
2562     ret = pI_CertUpdatestore(store1, store2, 0, 0);
2563     ok(ret, "I_CertUpdateStore failed: %08x\n", GetLastError());
2564     certs = countCertsInStore(store1);
2565     ok(certs == 0, "Expected 0 certs, got %d\n", certs);
2566
2567     CertCloseStore(store1, 0);
2568     CertCloseStore(store2, 0);
2569 }
2570
2571 START_TEST(store)
2572 {
2573     HMODULE hdll;
2574
2575     hdll = GetModuleHandleA("Crypt32.dll");
2576     pCertAddStoreToCollection = (void*)GetProcAddress(hdll, "CertAddStoreToCollection");
2577     pCertControlStore = (void*)GetProcAddress(hdll, "CertControlStore");
2578     pCertEnumCRLsInStore = (void*)GetProcAddress(hdll, "CertEnumCRLsInStore");
2579     pCertEnumSystemStore = (void*)GetProcAddress(hdll, "CertEnumSystemStore");
2580     pCertGetStoreProperty = (void*)GetProcAddress(hdll, "CertGetStoreProperty");
2581     pCertRemoveStoreFromCollection = (void*)GetProcAddress(hdll, "CertRemoveStoreFromCollection");
2582     pCertSetStoreProperty = (void*)GetProcAddress(hdll, "CertSetStoreProperty");
2583     pCertAddCertificateLinkToStore = (void*)GetProcAddress(hdll, "CertAddCertificateLinkToStore");
2584
2585     /* various combinations of CertOpenStore */
2586     testMemStore();
2587     testCollectionStore();
2588     testRegStore();
2589     testSystemRegStore();
2590     testSystemStore();
2591     testFileStore();
2592     testFileNameStore();
2593     testMessageStore();
2594     testSerializedStore();
2595
2596     testCertOpenSystemStore();
2597     testCertEnumSystemStore();
2598     testStoreProperty();
2599
2600     testAddSerialized();
2601     testAddCertificateLink();
2602
2603     test_I_UpdateStore();
2604 }