Basic implentation of the client auth UI

chrome/browser/ssl/ssl_client_auth_handler.cc

+// Copyright (c) 2009 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "chrome/browser/ssl/ssl_client_auth_handler.h"
+
+#if defined (OS_WIN)
+#include <cryptuiapi.h>
+#pragma comment(lib, "cryptui.lib")
+#endif
+
+#include "app/l10n_util.h"
+#include "base/message_loop.h"
+#include "base/string_util.h"
+#include "chrome/browser/browser_list.h"
+#include "chrome/browser/browser.h"
+#include "chrome/browser/browser_window.h"
+#include "grit/generated_resources.h"
+#include "net/url_request/url_request.h"
+
+SSLClientAuthHandler::SSLClientAuthHandler(
+    URLRequest* request,
+    net::SSLCertRequestInfo* cert_request_info,
+    MessageLoop* io_loop,
+    MessageLoop* ui_loop)
+    : request_(request),
+      cert_request_info_(cert_request_info),
+      io_loop_(io_loop),
+      ui_loop_(ui_loop) {
+  // Keep us alive until a cert is selected.
+  AddRef();
+}
+
+SSLClientAuthHandler::~SSLClientAuthHandler() {
+}
+
+void SSLClientAuthHandler::OnRequestCancelled() {
+  request_ = NULL;
+}
+
+void SSLClientAuthHandler::SelectCertificate() {
+  // Let's move the request to the UI thread.
+  ui_loop_->PostTask(FROM_HERE, NewRunnableMethod(this,
+      &SSLClientAuthHandler::AskUserForCertificate));
+}
+
+void SSLClientAuthHandler::AskUserForCertificate() {
+  net::X509Certificate* cert = NULL;
+#if defined(OS_WIN)
+  // TODO(jcampan): replace this with our own cert selection dialog.
+  // CryptUIDlgSelectCertificateFromStore is blocking (but still processes
+  // Windows messages), which is scary.
+  HCERTSTORE client_certs = CertOpenStore(CERT_STORE_PROV_MEMORY, 0, NULL,
+                                          0, NULL);
+  BOOL ok;
+  for (size_t i = 0; i < cert_request_info_->client_certs.size(); ++i) {
+    PCCERT_CONTEXT cc = cert_request_info_->client_certs[i]->os_cert_handle();
+    ok = CertAddCertificateContextToStore(client_certs, cc,
+                                          CERT_STORE_ADD_ALWAYS, NULL);
+    DCHECK(ok);
+  }
+
+  HWND browser_hwnd = NULL;
+  Browser* browser = BrowserList::GetLastActive();
+  if (browser)
+    browser_hwnd = browser->window()->GetNativeHandle();
+
+  std::wstring title = l10n_util::GetString(IDS_CLIENT_CERT_DIALOG_TITLE);
+  std::wstring text = l10n_util::GetStringF(
+      IDS_CLIENT_CERT_DIALOG_TEXT,
+      ASCIIToWide(cert_request_info_->host_and_port));
+  PCCERT_CONTEXT cert_context = CryptUIDlgSelectCertificateFromStore(
+      client_certs, browser_hwnd, title.c_str(), text.c_str(), 0, 0, NULL);
+
+  if (cert_context) {
+    cert = net::X509Certificate::CreateFromHandle(
+        cert_context,
+        net::X509Certificate::SOURCE_LONE_CERT_IMPORT);
+  }
+
+  ok = CertCloseStore(client_certs, CERT_CLOSE_STORE_CHECK_FLAG);
+  DCHECK(ok);
+#else
+  NOTIMPLEMENTED();
+#endif
+
+  // Notify the IO thread that we have selected a cert.
+  io_loop_->PostTask(FROM_HERE, NewRunnableMethod(this,
+      &SSLClientAuthHandler::CertificateSelected, cert));
+}
+
+void SSLClientAuthHandler::CertificateSelected(net::X509Certificate* cert) {
+  // request_ could have been NULLed if the request was cancelled while the user
+  // was choosing a cert.
+  if (request_)
+    request_->ContinueWithCertificate(cert);
+
+  // We are done.
+  Release();
+}