From 95a38f3ac5f960159b1cceadf1e034e6b5a2648c Mon Sep 17 00:00:00 2001
From: Juan Lang <juan.lang@gmail.com>
Date: Mon, 22 Dec 2008 19:04:10 -0800
Subject: [PATCH] cryptui: Implement importing stores in CryptUIWizImport.

---
 dlls/cryptui/main.c          | 103 +++++++++++++++++++++++++++++++++++
 dlls/cryptui/tests/cryptui.c |   3 -
 2 files changed, 103 insertions(+), 3 deletions(-)

diff --git a/dlls/cryptui/main.c b/dlls/cryptui/main.c
index 901de338e52..4ca73645304 100644
--- a/dlls/cryptui/main.c
+++ b/dlls/cryptui/main.c
@@ -3496,6 +3496,34 @@ static BOOL import_crl(PCCRL_CONTEXT crl, HCERTSTORE hDestCertStore)
     return ret;
 }
 
+static BOOL import_ctl(PCCTL_CONTEXT ctl, HCERTSTORE hDestCertStore)
+{
+    HCERTSTORE store;
+    BOOL ret;
+
+    if (!ctl)
+    {
+        SetLastError(E_INVALIDARG);
+        return FALSE;
+    }
+    if (hDestCertStore) store = hDestCertStore;
+    else
+    {
+        static const WCHAR trust[] = { 'T','r','u','s','t',0 };
+
+        if (!(store = CertOpenStore(CERT_STORE_PROV_SYSTEM_W, 0, 0,
+         CERT_SYSTEM_STORE_CURRENT_USER, trust)))
+        {
+            WARN("unable to open certificate store\n");
+            return FALSE;
+        }
+    }
+    ret = CertAddCTLContextToStore(store, ctl,
+     CERT_STORE_ADD_REPLACE_EXISTING_INHERIT_PROPERTIES, NULL);
+    if (!hDestCertStore) CertCloseStore(store, 0);
+    return ret;
+}
+
 /* Checks type, a type such as CERT_QUERY_CONTENT_CERT returned by
  * CryptQueryObject, against the allowed types.  Returns TRUE if the
  * type is allowed, FALSE otherwise.
@@ -3561,6 +3589,77 @@ static void import_warn_type_mismatch(DWORD dwFlags, HWND hwnd, LPCWSTR szTitle)
     }
 }
 
+static BOOL check_store_context_type(DWORD dwFlags, HCERTSTORE store)
+{
+    BOOL ret;
+
+    if (dwFlags &
+     (CRYPTUI_WIZ_IMPORT_ALLOW_CERT | CRYPTUI_WIZ_IMPORT_ALLOW_CRL |
+     CRYPTUI_WIZ_IMPORT_ALLOW_CTL))
+    {
+        PCCERT_CONTEXT cert;
+        PCCRL_CONTEXT crl;
+        PCCTL_CONTEXT ctl;
+
+        ret = TRUE;
+        if ((cert = CertEnumCertificatesInStore(store, NULL)))
+        {
+            CertFreeCertificateContext(cert);
+            if (!(dwFlags & CRYPTUI_WIZ_IMPORT_ALLOW_CERT))
+                ret = FALSE;
+        }
+        if (ret && (crl = CertEnumCRLsInStore(store, NULL)))
+        {
+            CertFreeCRLContext(crl);
+            if (!(dwFlags & CRYPTUI_WIZ_IMPORT_ALLOW_CRL))
+                ret = FALSE;
+        }
+        if (ret && (ctl = CertEnumCTLsInStore(store, NULL)))
+        {
+            CertFreeCTLContext(ctl);
+            if (!(dwFlags & CRYPTUI_WIZ_IMPORT_ALLOW_CTL))
+                ret = FALSE;
+        }
+    }
+    else
+        ret = TRUE;
+    if (!ret)
+        SetLastError(E_INVALIDARG);
+    return ret;
+}
+
+static BOOL import_store(DWORD dwFlags, HWND hwnd, LPCWSTR szTitle,
+ HCERTSTORE source, HCERTSTORE dest)
+{
+    BOOL ret;
+
+    if ((ret = check_store_context_type(dwFlags, source)))
+    {
+        PCCERT_CONTEXT cert = NULL;
+        PCCRL_CONTEXT crl = NULL;
+        PCCTL_CONTEXT ctl = NULL;
+
+        do {
+            cert = CertEnumCertificatesInStore(source, cert);
+            if (cert)
+                ret = import_cert(cert, dest);
+        } while (ret && cert);
+        do {
+            crl = CertEnumCRLsInStore(source, crl);
+            if (crl)
+                ret = import_crl(crl, dest);
+        } while (ret && crl);
+        do {
+            ctl = CertEnumCTLsInStore(source, ctl);
+            if (ctl)
+                ret = import_ctl(ctl, dest);
+        } while (ret && ctl);
+    }
+    else
+        import_warn_type_mismatch(dwFlags, hwnd, szTitle);
+    return ret;
+}
+
 BOOL WINAPI CryptUIWizImport(DWORD dwFlags, HWND hwndParent, LPCWSTR pwszWizardTitle,
                              PCCRYPTUI_WIZ_IMPORT_SRC_INFO pImportSrc, HCERTSTORE hDestCertStore)
 {
@@ -3605,6 +3704,10 @@ BOOL WINAPI CryptUIWizImport(DWORD dwFlags, HWND hwndParent, LPCWSTR pwszWizardT
         else
             import_warn_type_mismatch(dwFlags, hwndParent, pwszWizardTitle);
         break;
+    case CRYPTUI_WIZ_IMPORT_SUBJECT_CERT_STORE:
+        ret = import_store(dwFlags, hwndParent, pwszWizardTitle,
+         pImportSrc->u.hCertStore, hDestCertStore);
+        break;
     default:
         FIXME("source type not implemented: %u\n", pImportSrc->dwSubjectChoice);
         SetLastError(E_INVALIDARG);
diff --git a/dlls/cryptui/tests/cryptui.c b/dlls/cryptui/tests/cryptui.c
index 643f3270f22..2cac5211fc9 100644
--- a/dlls/cryptui/tests/cryptui.c
+++ b/dlls/cryptui/tests/cryptui.c
@@ -537,7 +537,6 @@ static void test_crypt_ui_wiz_import(void)
      CRYPTUI_WIZ_IMPORT_NO_CHANGE_DEST_STORE |
      CRYPTUI_WIZ_IMPORT_ALLOW_CERT | CRYPTUI_WIZ_IMPORT_ALLOW_CRL, 0, NULL,
      &info, store);
-    todo_wine
     ok(ret, "CryptUIWizImport failed: %08x\n", GetLastError());
     if (ret)
     {
@@ -574,7 +573,6 @@ static void test_crypt_ui_wiz_import(void)
     SetLastError(0xdeadbeef);
     ret = CryptUIWizImport(CRYPTUI_WIZ_NO_UI | CRYPTUI_WIZ_IMPORT_ALLOW_CERT,
      0, NULL, &info, store);
-    todo_wine
     ok(ret, "CryptUIWizImport failed: %08x\n", GetLastError());
     if (ret)
     {
@@ -615,7 +613,6 @@ static void test_crypt_ui_wiz_import(void)
     SetLastError(0xdeadbeef);
     ret = CryptUIWizImport(CRYPTUI_WIZ_NO_UI | CRYPTUI_WIZ_IMPORT_ALLOW_CRL,
      0, NULL, &info, store);
-    todo_wine
     ok(ret, "CryptUIWizImport failed: %08x\n", GetLastError());
     if (ret)
     {