dlls/crypt32/object.c

   1 /*
   2  * crypt32 Crypt*Object functions
   3  *
   4  * Copyright 2007 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 #include <stdarg.h>
  21 #include "windef.h"
  22 #include "winbase.h"
  23 #include "wincrypt.h"
  24 #include "imagehlp.h"
  25 #include "crypt32_private.h"
  26 #include "wine/debug.h"
  27
  28 WINE_DEFAULT_DEBUG_CHANNEL(crypt);
  29
  30 static BOOL CRYPT_ReadBlobFromFile(LPCWSTR fileName, PCERT_BLOB blob)
  31 {
  32     BOOL ret = FALSE;
  33     HANDLE file;
  34
  35     TRACE("%s\n", debugstr_w(fileName));
  36
  37     file = CreateFileW(fileName, GENERIC_READ, FILE_SHARE_READ, NULL,
  38      OPEN_EXISTING, 0, NULL);
  39     if (file != INVALID_HANDLE_VALUE)
  40     {
  41         ret = TRUE;
  42         blob->cbData = GetFileSize(file, NULL);
  43         if (blob->cbData)
  44         {
  45             blob->pbData = CryptMemAlloc(blob->cbData);
  46             if (blob->pbData)
  47             {
  48                 DWORD read;
  49
  50                 ret = ReadFile(file, blob->pbData, blob->cbData, &read, NULL);
  51             }
  52         }
  53         CloseHandle(file);
  54     }
  55     TRACE("returning %d\n", ret);
  56     return ret;
  57 }
  58
  59 static BOOL CRYPT_QueryContextObject(DWORD dwObjectType, const void *pvObject,
  60  DWORD dwExpectedContentTypeFlags, DWORD *pdwMsgAndCertEncodingType,
  61  DWORD *pdwContentType, HCERTSTORE *phCertStore, const void **ppvContext)
  62 {
  63     CERT_BLOB fileBlob;
  64     const CERT_BLOB *blob;
  65     HCERTSTORE store;
  66     DWORD contentType;
  67     BOOL ret;
  68
  69     switch (dwObjectType)
  70     {
  71     case CERT_QUERY_OBJECT_FILE:
  72         /* Cert, CRL, and CTL contexts can't be "embedded" in a file, so
  73          * just read the file directly
  74          */
  75         ret = CRYPT_ReadBlobFromFile((LPCWSTR)pvObject, &fileBlob);
  76         blob = &fileBlob;
  77         break;
  78     case CERT_QUERY_OBJECT_BLOB:
  79         blob = (const CERT_BLOB *)pvObject;
  80         ret = TRUE;
  81         break;
  82     default:
  83         SetLastError(E_INVALIDARG); /* FIXME: is this the correct error? */
  84         ret = FALSE;
  85     }
  86     if (!ret)
  87         return FALSE;
  88
  89     store = CertOpenStore(CERT_STORE_PROV_MEMORY, 0, 0,
  90      CERT_STORE_CREATE_NEW_FLAG, NULL);
  91     ret = FALSE;
  92     if (dwExpectedContentTypeFlags & CERT_QUERY_CONTENT_FLAG_CERT)
  93     {
  94         ret = pCertInterface->addEncodedToStore(store, X509_ASN_ENCODING,
  95          blob->pbData, blob->cbData, CERT_STORE_ADD_ALWAYS, ppvContext);
  96         if (ret)
  97             contentType = CERT_QUERY_CONTENT_CERT;
  98     }
  99     if (!ret && (dwExpectedContentTypeFlags & CERT_QUERY_CONTENT_FLAG_CRL))
 100     {
 101         ret = pCRLInterface->addEncodedToStore(store, X509_ASN_ENCODING,
 102          blob->pbData, blob->cbData, CERT_STORE_ADD_ALWAYS, ppvContext);
 103         if (ret)
 104             contentType = CERT_QUERY_CONTENT_CRL;
 105     }
 106     if (!ret && (dwExpectedContentTypeFlags & CERT_QUERY_CONTENT_FLAG_CTL))
 107     {
 108         ret = pCTLInterface->addEncodedToStore(store, X509_ASN_ENCODING,
 109          blob->pbData, blob->cbData, CERT_STORE_ADD_ALWAYS, ppvContext);
 110         if (ret)
 111             contentType = CERT_QUERY_CONTENT_CTL;
 112     }
 113     if (ret)
 114     {
 115         if (pdwMsgAndCertEncodingType)
 116             *pdwMsgAndCertEncodingType = X509_ASN_ENCODING;
 117         if (pdwContentType)
 118             *pdwContentType = contentType;
 119         if (phCertStore)
 120             *phCertStore = CertDuplicateStore(store);
 121     }
 122     CertCloseStore(store, 0);
 123     if (blob == &fileBlob)
 124         CryptMemFree(blob->pbData);
 125     TRACE("returning %d\n", ret);
 126     return ret;
 127 }
 128
 129 static BOOL CRYPT_QuerySerializedContextObject(DWORD dwObjectType,
 130  const void *pvObject, DWORD dwExpectedContentTypeFlags,
 131  DWORD *pdwMsgAndCertEncodingType, DWORD *pdwContentType,
 132  HCERTSTORE *phCertStore, const void **ppvContext)
 133 {
 134     CERT_BLOB fileBlob;
 135     const CERT_BLOB *blob;
 136     const WINE_CONTEXT_INTERFACE *contextInterface = NULL;
 137     const void *context;
 138     DWORD contextType;
 139     BOOL ret;
 140
 141     switch (dwObjectType)
 142     {
 143     case CERT_QUERY_OBJECT_FILE:
 144         /* Cert, CRL, and CTL contexts can't be "embedded" in a file, so
 145          * just read the file directly
 146          */
 147         ret = CRYPT_ReadBlobFromFile((LPCWSTR)pvObject, &fileBlob);
 148         blob = &fileBlob;
 149         break;
 150     case CERT_QUERY_OBJECT_BLOB:
 151         blob = (const CERT_BLOB *)pvObject;
 152         ret = TRUE;
 153         break;
 154     default:
 155         SetLastError(E_INVALIDARG); /* FIXME: is this the correct error? */
 156         ret = FALSE;
 157     }
 158     if (!ret)
 159         return FALSE;
 160
 161     context = CRYPT_ReadSerializedElement(blob->pbData, blob->cbData,
 162      CERT_STORE_ALL_CONTEXT_FLAG, &contextType);
 163     if (context)
 164     {
 165         DWORD contentType, certStoreOffset;
 166
 167         ret = TRUE;
 168         switch (contextType)
 169         {
 170         case CERT_STORE_CERTIFICATE_CONTEXT:
 171             contextInterface = pCertInterface;
 172             contentType = CERT_QUERY_CONTENT_SERIALIZED_CERT;
 173             certStoreOffset = offsetof(CERT_CONTEXT, hCertStore);
 174             if (!(dwExpectedContentTypeFlags &
 175              CERT_QUERY_CONTENT_FLAG_SERIALIZED_CERT))
 176             {
 177                 SetLastError(ERROR_INVALID_DATA);
 178                 ret = FALSE;
 179                 goto end;
 180             }
 181             break;
 182         case CERT_STORE_CRL_CONTEXT:
 183             contextInterface = pCRLInterface;
 184             contentType = CERT_QUERY_CONTENT_SERIALIZED_CRL;
 185             certStoreOffset = offsetof(CRL_CONTEXT, hCertStore);
 186             if (!(dwExpectedContentTypeFlags &
 187              CERT_QUERY_CONTENT_FLAG_SERIALIZED_CRL))
 188             {
 189                 SetLastError(ERROR_INVALID_DATA);
 190                 ret = FALSE;
 191                 goto end;
 192             }
 193             break;
 194         case CERT_STORE_CTL_CONTEXT:
 195             contextInterface = pCTLInterface;
 196             contentType = CERT_QUERY_CONTENT_SERIALIZED_CTL;
 197             certStoreOffset = offsetof(CTL_CONTEXT, hCertStore);
 198             if (!(dwExpectedContentTypeFlags &
 199              CERT_QUERY_CONTENT_FLAG_SERIALIZED_CTL))
 200             {
 201                 SetLastError(ERROR_INVALID_DATA);
 202                 ret = FALSE;
 203                 goto end;
 204             }
 205             break;
 206         default:
 207             SetLastError(ERROR_INVALID_DATA);
 208             ret = FALSE;
 209             goto end;
 210         }
 211         if (pdwMsgAndCertEncodingType)
 212             *pdwMsgAndCertEncodingType = X509_ASN_ENCODING;
 213         if (pdwContentType)
 214             *pdwContentType = contentType;
 215         if (phCertStore)
 216             *phCertStore = CertDuplicateStore(
 217              *(HCERTSTORE *)((const BYTE *)context + certStoreOffset));
 218         if (ppvContext)
 219             *ppvContext = contextInterface->duplicate(context);
 220     }
 221
 222 end:
 223     if (contextInterface && context)
 224         contextInterface->free(context);
 225     if (blob == &fileBlob)
 226         CryptMemFree(blob->pbData);
 227     TRACE("returning %d\n", ret);
 228     return ret;
 229 }
 230
 231 static BOOL CRYPT_QuerySerializedStoreObject(DWORD dwObjectType,
 232  const void *pvObject, DWORD *pdwMsgAndCertEncodingType, DWORD *pdwContentType,
 233  HCERTSTORE *phCertStore, HCRYPTMSG *phMsg)
 234 {
 235     LPCWSTR fileName = (LPCWSTR)pvObject;
 236     HANDLE file;
 237     BOOL ret = FALSE;
 238
 239     if (dwObjectType != CERT_QUERY_OBJECT_FILE)
 240     {
 241         FIXME("unimplemented for non-file type %d\n", dwObjectType);
 242         SetLastError(E_INVALIDARG); /* FIXME: is this the correct error? */
 243         return FALSE;
 244     }
 245     TRACE("%s\n", debugstr_w(fileName));
 246     file = CreateFileW(fileName, GENERIC_READ, FILE_SHARE_READ, NULL,
 247      OPEN_EXISTING, 0, NULL);
 248     if (file != INVALID_HANDLE_VALUE)
 249     {
 250         HCERTSTORE store = CertOpenStore(CERT_STORE_PROV_MEMORY, 0, 0,
 251          CERT_STORE_CREATE_NEW_FLAG, NULL);
 252
 253         ret = CRYPT_ReadSerializedFile(file, store);
 254         if (ret)
 255         {
 256             if (pdwMsgAndCertEncodingType)
 257                 *pdwMsgAndCertEncodingType = X509_ASN_ENCODING;
 258             if (pdwContentType)
 259                 *pdwContentType = CERT_QUERY_CONTENT_SERIALIZED_STORE;
 260             if (phCertStore)
 261                 *phCertStore = CertDuplicateStore(store);
 262         }
 263         CertCloseStore(store, 0);
 264         CloseHandle(file);
 265     }
 266     TRACE("returning %d\n", ret);
 267     return ret;
 268 }
 269
 270 /* Used to decode non-embedded messages */
 271 static BOOL CRYPT_QueryMessageObject(DWORD dwObjectType, const void *pvObject,
 272  DWORD dwExpectedContentTypeFlags, DWORD *pdwMsgAndCertEncodingType,
 273  DWORD *pdwContentType, HCERTSTORE *phCertStore, HCRYPTMSG *phMsg)
 274 {
 275     CERT_BLOB fileBlob;
 276     const CERT_BLOB *blob;
 277     BOOL ret;
 278     HCRYPTMSG msg = NULL;
 279     DWORD encodingType = X509_ASN_ENCODING | PKCS_7_ASN_ENCODING;
 280
 281     switch (dwObjectType)
 282     {
 283     case CERT_QUERY_OBJECT_FILE:
 284         /* This isn't an embedded PKCS7 message, so just read the file
 285          * directly
 286          */
 287         ret = CRYPT_ReadBlobFromFile((LPCWSTR)pvObject, &fileBlob);
 288         blob = &fileBlob;
 289         break;
 290     case CERT_QUERY_OBJECT_BLOB:
 291         blob = (const CERT_BLOB *)pvObject;
 292         ret = TRUE;
 293         break;
 294     default:
 295         SetLastError(E_INVALIDARG); /* FIXME: is this the correct error? */
 296         ret = FALSE;
 297     }
 298     if (!ret)
 299         return FALSE;
 300
 301     ret = FALSE;
 302     if (dwExpectedContentTypeFlags & CERT_QUERY_CONTENT_FLAG_PKCS7_SIGNED)
 303     {
 304         msg = CryptMsgOpenToDecode(encodingType, 0, CMSG_SIGNED, 0, NULL, NULL);
 305         if (msg)
 306         {
 307             ret = CryptMsgUpdate(msg, blob->pbData, blob->cbData, TRUE);
 308             if (!ret)
 309             {
 310                 CryptMsgClose(msg);
 311                 msg = NULL;
 312             }
 313         }
 314         if (msg && pdwContentType)
 315             *pdwContentType = CERT_QUERY_CONTENT_PKCS7_SIGNED;
 316     }
 317     if (!ret &&
 318      (dwExpectedContentTypeFlags & CERT_QUERY_CONTENT_FLAG_PKCS7_UNSIGNED))
 319     {
 320         msg = CryptMsgOpenToDecode(encodingType, 0, CMSG_DATA, 0, NULL, NULL);
 321         if (msg)
 322         {
 323             ret = CryptMsgUpdate(msg, blob->pbData, blob->cbData, TRUE);
 324             if (!ret)
 325             {
 326                 CryptMsgClose(msg);
 327                 msg = NULL;
 328             }
 329         }
 330         if (msg && pdwContentType)
 331             *pdwContentType = CERT_QUERY_CONTENT_PKCS7_UNSIGNED;
 332     }
 333     if (pdwMsgAndCertEncodingType)
 334         *pdwMsgAndCertEncodingType = encodingType;
 335     if (msg)
 336     {
 337         if (phMsg)
 338             *phMsg = msg;
 339         if (phCertStore)
 340             *phCertStore = CertOpenStore(CERT_STORE_PROV_MSG, encodingType, 0,
 341              0, msg);
 342     }
 343     if (blob == &fileBlob)
 344         CryptMemFree(blob->pbData);
 345     TRACE("returning %d\n", ret);
 346     return ret;
 347 }
 348
 349 static BOOL CRYPT_QueryEmbeddedMessageObject(DWORD dwObjectType,
 350  const void *pvObject, DWORD dwExpectedContentTypeFlags,
 351  DWORD *pdwMsgAndCertEncodingType, DWORD *pdwContentType,
 352  HCERTSTORE *phCertStore, HCRYPTMSG *phMsg)
 353 {
 354     HANDLE file;
 355     BOOL ret = FALSE;
 356
 357     if (dwObjectType != CERT_QUERY_OBJECT_FILE)
 358     {
 359         FIXME("don't know what to do for type %d embedded signed messages\n",
 360          dwObjectType);
 361         SetLastError(E_INVALIDARG);
 362         return FALSE;
 363     }
 364     file = CreateFileW((LPCWSTR)pvObject, GENERIC_READ, FILE_SHARE_READ,
 365      NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
 366     if (file != INVALID_HANDLE_VALUE)
 367     {
 368         CERT_BLOB blob;
 369
 370         ret = ImageGetCertificateData(file, 0, NULL, &blob.cbData);
 371         if (ret)
 372         {
 373             blob.pbData = HeapAlloc(GetProcessHeap(), 0, blob.cbData);
 374             if (blob.pbData)
 375             {
 376                 ret = ImageGetCertificateData(file, 0,
 377                  (WIN_CERTIFICATE *)blob.pbData, &blob.cbData);
 378                 if (ret)
 379                 {
 380                     ret = CRYPT_QueryMessageObject(CERT_QUERY_OBJECT_BLOB,
 381                      &blob, CERT_QUERY_CONTENT_FLAG_PKCS7_SIGNED,
 382                      pdwMsgAndCertEncodingType, NULL, phCertStore, phMsg);
 383                     if (ret && pdwContentType)
 384                         *pdwContentType = CERT_QUERY_CONTENT_FLAG_PKCS7_SIGNED;
 385                 }
 386                 HeapFree(GetProcessHeap(), 0, blob.pbData);
 387             }
 388         }
 389         CloseHandle(file);
 390     }
 391     TRACE("returning %d\n", ret);
 392     return ret;
 393 }
 394
 395 BOOL WINAPI CryptQueryObject(DWORD dwObjectType, const void *pvObject,
 396  DWORD dwExpectedContentTypeFlags, DWORD dwExpectedFormatTypeFlags,
 397  DWORD dwFlags, DWORD *pdwMsgAndCertEncodingType, DWORD *pdwContentType,
 398  DWORD *pdwFormatType, HCERTSTORE *phCertStore, HCRYPTMSG *phMsg,
 399  const void **ppvContext)
 400 {
 401     static const DWORD unimplementedTypes =
 402      CERT_QUERY_CONTENT_FLAG_PKCS10 | CERT_QUERY_CONTENT_FLAG_PFX |
 403      CERT_QUERY_CONTENT_FLAG_CERT_PAIR;
 404     BOOL ret = TRUE;
 405
 406     TRACE("(%08x, %p, %08x, %08x, %08x, %p, %p, %p, %p, %p, %p)\n",
 407      dwObjectType, pvObject, dwExpectedContentTypeFlags,
 408      dwExpectedFormatTypeFlags, dwFlags, pdwMsgAndCertEncodingType,
 409      pdwContentType, pdwFormatType, phCertStore, phMsg, ppvContext);
 410
 411     if (dwExpectedContentTypeFlags & unimplementedTypes)
 412         WARN("unimplemented for types %08x\n",
 413          dwExpectedContentTypeFlags & unimplementedTypes);
 414     if (!(dwExpectedFormatTypeFlags & CERT_QUERY_FORMAT_FLAG_BINARY))
 415     {
 416         FIXME("unimplemented for anything but binary\n");
 417         SetLastError(ERROR_CALL_NOT_IMPLEMENTED);
 418         return FALSE;
 419     }
 420     if (pdwFormatType)
 421         *pdwFormatType = CERT_QUERY_FORMAT_BINARY;
 422
 423     if (phCertStore)
 424         *phCertStore = NULL;
 425     if (phMsg)
 426         *phMsg = NULL;
 427     if (ppvContext)
 428         *ppvContext = NULL;
 429
 430     ret = FALSE;
 431     if ((dwExpectedContentTypeFlags & CERT_QUERY_CONTENT_FLAG_CERT) ||
 432      (dwExpectedContentTypeFlags & CERT_QUERY_CONTENT_FLAG_CRL) ||
 433      (dwExpectedContentTypeFlags & CERT_QUERY_CONTENT_FLAG_CTL))
 434     {
 435         ret = CRYPT_QueryContextObject(dwObjectType, pvObject,
 436          dwExpectedContentTypeFlags, pdwMsgAndCertEncodingType, pdwContentType,
 437          phCertStore, ppvContext);
 438     }
 439     if (!ret &&
 440      (dwExpectedContentTypeFlags & CERT_QUERY_CONTENT_FLAG_SERIALIZED_STORE))
 441     {
 442         ret = CRYPT_QuerySerializedStoreObject(dwObjectType, pvObject,
 443          pdwMsgAndCertEncodingType, pdwContentType, phCertStore, phMsg);
 444     }
 445     if (!ret &&
 446      ((dwExpectedContentTypeFlags & CERT_QUERY_CONTENT_FLAG_SERIALIZED_CERT) ||
 447      (dwExpectedContentTypeFlags & CERT_QUERY_CONTENT_FLAG_SERIALIZED_CRL) ||
 448      (dwExpectedContentTypeFlags & CERT_QUERY_CONTENT_FLAG_SERIALIZED_CTL)))
 449     {
 450         ret = CRYPT_QuerySerializedContextObject(dwObjectType, pvObject,
 451          dwExpectedContentTypeFlags, pdwMsgAndCertEncodingType, pdwContentType,
 452          phCertStore, ppvContext);
 453     }
 454     if (!ret &&
 455      ((dwExpectedContentTypeFlags & CERT_QUERY_CONTENT_FLAG_PKCS7_SIGNED) ||
 456      (dwExpectedContentTypeFlags & CERT_QUERY_CONTENT_FLAG_PKCS7_UNSIGNED)))
 457     {
 458         ret = CRYPT_QueryMessageObject(dwObjectType, pvObject,
 459          dwExpectedContentTypeFlags, pdwMsgAndCertEncodingType, pdwContentType,
 460          phCertStore, phMsg);
 461     }
 462     if (!ret &&
 463      (dwExpectedContentTypeFlags & CERT_QUERY_CONTENT_FLAG_PKCS7_SIGNED_EMBED))
 464     {
 465         ret = CRYPT_QueryEmbeddedMessageObject(dwObjectType, pvObject,
 466          dwExpectedContentTypeFlags, pdwMsgAndCertEncodingType, pdwContentType,
 467          phCertStore, phMsg);
 468     }
 469     TRACE("returning %d\n");
 470     return ret;
 471 }