git.oblomov.eu Git - wine/blob - 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_ReadSerializedStoreFromFile(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     /* Try it first as a PKCS content info */
 303     if ((dwExpectedContentTypeFlags & CERT_QUERY_CONTENT_FLAG_PKCS7_SIGNED) ||
 304      (dwExpectedContentTypeFlags & CERT_QUERY_CONTENT_FLAG_PKCS7_UNSIGNED))
 305     {
 306         msg = CryptMsgOpenToDecode(encodingType, 0, 0, 0, NULL, NULL);
 307         if (msg)
 308         {
 309             ret = CryptMsgUpdate(msg, blob->pbData, blob->cbData, TRUE);
 310             if (ret)
 311             {
 312                 DWORD type, len = sizeof(type);
 313
 314                 ret = CryptMsgGetParam(msg, CMSG_TYPE_PARAM, 0, &type, &len);
 315                 if (ret)
 316                 {
 317                     if ((dwExpectedContentTypeFlags &
 318                      CERT_QUERY_CONTENT_FLAG_PKCS7_SIGNED))
 319                     {
 320                         if (type != CMSG_SIGNED)
 321                         {
 322                             SetLastError(ERROR_INVALID_DATA);
 323                             ret = FALSE;
 324                         }
 325                         else if (pdwContentType)
 326                             *pdwContentType = CERT_QUERY_CONTENT_PKCS7_SIGNED;
 327                     }
 328                     else if ((dwExpectedContentTypeFlags &
 329                      CERT_QUERY_CONTENT_FLAG_PKCS7_UNSIGNED))
 330                     {
 331                         if (type != CMSG_DATA)
 332                         {
 333                             SetLastError(ERROR_INVALID_DATA);
 334                             ret = FALSE;
 335                         }
 336                         else if (pdwContentType)
 337                             *pdwContentType = CERT_QUERY_CONTENT_PKCS7_UNSIGNED;
 338                     }
 339                 }
 340             }
 341             if (!ret)
 342             {
 343                 CryptMsgClose(msg);
 344                 msg = NULL;
 345             }
 346         }
 347     }
 348     /* Failing that, try explicitly typed messages */
 349     if (!ret &&
 350      (dwExpectedContentTypeFlags & CERT_QUERY_CONTENT_FLAG_PKCS7_SIGNED))
 351     {
 352         msg = CryptMsgOpenToDecode(encodingType, 0, CMSG_SIGNED, 0, NULL, NULL);
 353         if (msg)
 354         {
 355             ret = CryptMsgUpdate(msg, blob->pbData, blob->cbData, TRUE);
 356             if (!ret)
 357             {
 358                 CryptMsgClose(msg);
 359                 msg = NULL;
 360             }
 361         }
 362         if (msg && pdwContentType)
 363             *pdwContentType = CERT_QUERY_CONTENT_PKCS7_SIGNED;
 364     }
 365     if (!ret &&
 366      (dwExpectedContentTypeFlags & CERT_QUERY_CONTENT_FLAG_PKCS7_UNSIGNED))
 367     {
 368         msg = CryptMsgOpenToDecode(encodingType, 0, CMSG_DATA, 0, NULL, NULL);
 369         if (msg)
 370         {
 371             ret = CryptMsgUpdate(msg, blob->pbData, blob->cbData, TRUE);
 372             if (!ret)
 373             {
 374                 CryptMsgClose(msg);
 375                 msg = NULL;
 376             }
 377         }
 378         if (msg && pdwContentType)
 379             *pdwContentType = CERT_QUERY_CONTENT_PKCS7_UNSIGNED;
 380     }
 381     if (pdwMsgAndCertEncodingType)
 382         *pdwMsgAndCertEncodingType = encodingType;
 383     if (msg)
 384     {
 385         if (phMsg)
 386             *phMsg = msg;
 387         if (phCertStore)
 388             *phCertStore = CertOpenStore(CERT_STORE_PROV_MSG, encodingType, 0,
 389              0, msg);
 390     }
 391     if (blob == &fileBlob)
 392         CryptMemFree(blob->pbData);
 393     TRACE("returning %d\n", ret);
 394     return ret;
 395 }
 396
 397 static BOOL CRYPT_QueryEmbeddedMessageObject(DWORD dwObjectType,
 398  const void *pvObject, DWORD dwExpectedContentTypeFlags,
 399  DWORD *pdwMsgAndCertEncodingType, DWORD *pdwContentType,
 400  HCERTSTORE *phCertStore, HCRYPTMSG *phMsg)
 401 {
 402     HANDLE file;
 403     BOOL ret = FALSE;
 404
 405     if (dwObjectType != CERT_QUERY_OBJECT_FILE)
 406     {
 407         FIXME("don't know what to do for type %d embedded signed messages\n",
 408          dwObjectType);
 409         SetLastError(E_INVALIDARG);
 410         return FALSE;
 411     }
 412     file = CreateFileW((LPCWSTR)pvObject, GENERIC_READ, FILE_SHARE_READ,
 413      NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
 414     if (file != INVALID_HANDLE_VALUE)
 415     {
 416         DWORD len = 0;
 417
 418         ret = ImageGetCertificateData(file, 0, NULL, &len);
 419         if (GetLastError() == ERROR_INSUFFICIENT_BUFFER)
 420         {
 421             WIN_CERTIFICATE *winCert = HeapAlloc(GetProcessHeap(), 0, len);
 422
 423             if (winCert)
 424             {
 425                 ret = ImageGetCertificateData(file, 0, winCert, &len);
 426                 if (ret)
 427                 {
 428                     CERT_BLOB blob = { winCert->dwLength,
 429                      winCert->bCertificate };
 430
 431                     ret = CRYPT_QueryMessageObject(CERT_QUERY_OBJECT_BLOB,
 432                      &blob, CERT_QUERY_CONTENT_FLAG_PKCS7_SIGNED,
 433                      pdwMsgAndCertEncodingType, NULL, phCertStore, phMsg);
 434                     if (ret && pdwContentType)
 435                         *pdwContentType = CERT_QUERY_CONTENT_FLAG_PKCS7_SIGNED;
 436                 }
 437                 HeapFree(GetProcessHeap(), 0, winCert);
 438             }
 439         }
 440         CloseHandle(file);
 441     }
 442     TRACE("returning %d\n", ret);
 443     return ret;
 444 }
 445
 446 BOOL WINAPI CryptQueryObject(DWORD dwObjectType, const void *pvObject,
 447  DWORD dwExpectedContentTypeFlags, DWORD dwExpectedFormatTypeFlags,
 448  DWORD dwFlags, DWORD *pdwMsgAndCertEncodingType, DWORD *pdwContentType,
 449  DWORD *pdwFormatType, HCERTSTORE *phCertStore, HCRYPTMSG *phMsg,
 450  const void **ppvContext)
 451 {
 452     static const DWORD unimplementedTypes =
 453      CERT_QUERY_CONTENT_FLAG_PKCS10 | CERT_QUERY_CONTENT_FLAG_PFX |
 454      CERT_QUERY_CONTENT_FLAG_CERT_PAIR;
 455     BOOL ret = TRUE;
 456
 457     TRACE("(%08x, %p, %08x, %08x, %08x, %p, %p, %p, %p, %p, %p)\n",
 458      dwObjectType, pvObject, dwExpectedContentTypeFlags,
 459      dwExpectedFormatTypeFlags, dwFlags, pdwMsgAndCertEncodingType,
 460      pdwContentType, pdwFormatType, phCertStore, phMsg, ppvContext);
 461
 462     if (dwExpectedContentTypeFlags & unimplementedTypes)
 463         WARN("unimplemented for types %08x\n",
 464          dwExpectedContentTypeFlags & unimplementedTypes);
 465     if (!(dwExpectedFormatTypeFlags & CERT_QUERY_FORMAT_FLAG_BINARY))
 466     {
 467         FIXME("unimplemented for anything but binary\n");
 468         SetLastError(ERROR_CALL_NOT_IMPLEMENTED);
 469         return FALSE;
 470     }
 471     if (pdwFormatType)
 472         *pdwFormatType = CERT_QUERY_FORMAT_BINARY;
 473
 474     if (phCertStore)
 475         *phCertStore = NULL;
 476     if (phMsg)
 477         *phMsg = NULL;
 478     if (ppvContext)
 479         *ppvContext = NULL;
 480
 481     ret = FALSE;
 482     if ((dwExpectedContentTypeFlags & CERT_QUERY_CONTENT_FLAG_CERT) ||
 483      (dwExpectedContentTypeFlags & CERT_QUERY_CONTENT_FLAG_CRL) ||
 484      (dwExpectedContentTypeFlags & CERT_QUERY_CONTENT_FLAG_CTL))
 485     {
 486         ret = CRYPT_QueryContextObject(dwObjectType, pvObject,
 487          dwExpectedContentTypeFlags, pdwMsgAndCertEncodingType, pdwContentType,
 488          phCertStore, ppvContext);
 489     }
 490     if (!ret &&
 491      (dwExpectedContentTypeFlags & CERT_QUERY_CONTENT_FLAG_SERIALIZED_STORE))
 492     {
 493         ret = CRYPT_QuerySerializedStoreObject(dwObjectType, pvObject,
 494          pdwMsgAndCertEncodingType, pdwContentType, phCertStore, phMsg);
 495     }
 496     if (!ret &&
 497      ((dwExpectedContentTypeFlags & CERT_QUERY_CONTENT_FLAG_SERIALIZED_CERT) ||
 498      (dwExpectedContentTypeFlags & CERT_QUERY_CONTENT_FLAG_SERIALIZED_CRL) ||
 499      (dwExpectedContentTypeFlags & CERT_QUERY_CONTENT_FLAG_SERIALIZED_CTL)))
 500     {
 501         ret = CRYPT_QuerySerializedContextObject(dwObjectType, pvObject,
 502          dwExpectedContentTypeFlags, pdwMsgAndCertEncodingType, pdwContentType,
 503          phCertStore, ppvContext);
 504     }
 505     if (!ret &&
 506      ((dwExpectedContentTypeFlags & CERT_QUERY_CONTENT_FLAG_PKCS7_SIGNED) ||
 507      (dwExpectedContentTypeFlags & CERT_QUERY_CONTENT_FLAG_PKCS7_UNSIGNED)))
 508     {
 509         ret = CRYPT_QueryMessageObject(dwObjectType, pvObject,
 510          dwExpectedContentTypeFlags, pdwMsgAndCertEncodingType, pdwContentType,
 511          phCertStore, phMsg);
 512     }
 513     if (!ret &&
 514      (dwExpectedContentTypeFlags & CERT_QUERY_CONTENT_FLAG_PKCS7_SIGNED_EMBED))
 515     {
 516         ret = CRYPT_QueryEmbeddedMessageObject(dwObjectType, pvObject,
 517          dwExpectedContentTypeFlags, pdwMsgAndCertEncodingType, pdwContentType,
 518          phCertStore, phMsg);
 519     }
 520     TRACE("returning %d\n", ret);
 521     return ret;
 522 }