Refactor client cert private key handling.

chrome/browser/chromeos/net/client_cert_store_chromeos.cc

 // Copyright 2013 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/chromeos/net/client_cert_store_chromeos.h"
 
 #include <cert.h>
 #include <algorithm>
+#include <iterator>
 #include <utility>
 
 #include "base/bind.h"
 #include "base/bind_helpers.h"
 #include "base/callback.h"
 #include "base/location.h"
+#include "base/memory/ptr_util.h"
 #include "base/task_runner_util.h"
 #include "base/threading/worker_pool.h"
 #include "chrome/browser/chromeos/certificate_provider/certificate_provider.h"
 #include "crypto/nss_crypto_module_delegate.h"
+#include "net/ssl/client_key_store.h"
 #include "net/ssl/ssl_cert_request_info.h"
+#include "net/ssl/ssl_platform_key_nss.h"
+#include "net/ssl/ssl_private_key.h"
 
 namespace chromeos {
 
 namespace {
 
+scoped_refptr<net::SSLPrivateKey> FetchClientCertPrivateKeyCros(
+    const net::X509Certificate* certificate) {
+  scoped_refptr<net::SSLPrivateKey> result =
+      net::FetchClientCertPrivateKey(certificate, nullptr);
+  if (result)
+    return result;
+
+  return net::ClientKeyStore::GetInstance()->FetchClientCertPrivateKey(
+      *certificate);
+}
+
+class ClientCertIdentityCros : public net::ClientCertIdentity {
+ public:
+  explicit ClientCertIdentityCros(scoped_refptr<net::X509Certificate> cert)
+      : net::ClientCertIdentity(std::move(cert)) {}
+  ~ClientCertIdentityCros() override = default;
+
+  void AcquirePrivateKey(
+      const base::Callback<void(scoped_refptr<net::SSLPrivateKey>)>&
+          private_key_callback) override;
+};
+
+void ClientCertIdentityCros::AcquirePrivateKey(
+    const base::Callback<void(scoped_refptr<net::SSLPrivateKey>)>&
+        private_key_callback) {
+  if (base::PostTaskAndReplyWithResult(
+          base::WorkerPool::GetTaskRunner(true /* task_is_slow */).get(),
+          FROM_HERE,
+          base::Bind(&FetchClientCertPrivateKeyCros,
+                     base::RetainedRef(certificate())),
+          private_key_callback)) {
+    return;
+  }
+  // If the task could not be posted, behave as if there was no key.
+  private_key_callback.Run(nullptr);
+}
+
 class CertNotAllowedPredicate {
  public:
   explicit CertNotAllowedPredicate(
       const ClientCertStoreChromeOS::CertFilter* filter)
       : filter_(filter) {}
-  bool operator()(const scoped_refptr<net::X509Certificate>& cert) const {
-    return !filter_->IsCertAllowed(cert);
+  bool operator()(const std::unique_ptr<net::ClientCertIdentity>& cert) const {
+    return !filter_->IsCertAllowed(cert->certificate());
   }
 
  private:
   const ClientCertStoreChromeOS::CertFilter* const filter_;
 };
 
 }  // namespace
 
 ClientCertStoreChromeOS::ClientCertStoreChromeOS(
     std::unique_ptr<CertificateProvider> cert_provider,
@@ skipping 25 matching lines @@
   }
 
   if (cert_filter_->Init(get_additional_certs_and_continue))
     get_additional_certs_and_continue.Run();
 }
 
 void ClientCertStoreChromeOS::GotAdditionalCerts(
     const net::SSLCertRequestInfo* request,
     const ClientCertListCallback& callback,
     const net::CertificateList& additional_certs) {
-  std::unique_ptr<crypto::CryptoModuleBlockingPasswordDelegate>
-      password_delegate;
-  if (!password_delegate_factory_.is_null()) {
-    password_delegate.reset(
-        password_delegate_factory_.Run(request->host_and_port));
-  }
+  scoped_refptr<crypto::CryptoModuleBlockingPasswordDelegate> password_delegate;
+  if (!password_delegate_factory_.is_null())
+    password_delegate = password_delegate_factory_.Run(request->host_and_port);
   if (base::PostTaskAndReplyWithResult(
           base::WorkerPool::GetTaskRunner(true /* task_is_slow */).get(),
           FROM_HERE,
           base::Bind(&ClientCertStoreChromeOS::GetAndFilterCertsOnWorkerThread,
-                     base::Unretained(this), base::Passed(&password_delegate),
-                     request, additional_certs),
+                     base::Unretained(this), password_delegate, request,
+                     std::move(additional_certs)),
           callback)) {
     return;
   }
   // If the task could not be posted, behave as if there were no certificates.
-  callback.Run(net::CertificateList());
+  callback.Run(net::ClientCertIdentityList());
 }
 
-net::CertificateList ClientCertStoreChromeOS::GetAndFilterCertsOnWorkerThread(
-    std::unique_ptr<crypto::CryptoModuleBlockingPasswordDelegate>
+net::ClientCertIdentityList
+ClientCertStoreChromeOS::GetAndFilterCertsOnWorkerThread(
+    scoped_refptr<crypto::CryptoModuleBlockingPasswordDelegate>
         password_delegate,
     const net::SSLCertRequestInfo* request,
-    const net::CertificateList& additional_certs) {
-  net::CertificateList unfiltered_certs;
+    net::CertificateList additional_certs) {
+  net::ClientCertIdentityList unfiltered_certs;
   net::ClientCertStoreNSS::GetPlatformCertsOnWorkerThread(
       std::move(password_delegate), &unfiltered_certs);
 
   unfiltered_certs.erase(
       std::remove_if(unfiltered_certs.begin(), unfiltered_certs.end(),
                      CertNotAllowedPredicate(cert_filter_.get())),
       unfiltered_certs.end());
 
-  unfiltered_certs.insert(unfiltered_certs.end(), additional_certs.begin(),
-                          additional_certs.end());
-
-  net::CertificateList selected_certs;
-  net::ClientCertStoreNSS::FilterCertsOnWorkerThread(unfiltered_certs, *request,
-                                                     &selected_certs);
-  return selected_certs;
+  for (const scoped_refptr<net::X509Certificate>& cert : additional_certs)
+    unfiltered_certs.push_back(base::MakeUnique<ClientCertIdentityCros>(cert));
+  /*unfiltered_certs.insert(unfiltered_certs.end(),
+                          std::make_move_iterator(additional_certs.begin()),
+                          std::make_move_iterator(additional_certs.end()));
+*/
+  net::ClientCertStoreNSS::FilterCertsOnWorkerThread(&unfiltered_certs,
+                                                     *request);
+  // XXX rename unfiltered_certs ?
+  return unfiltered_certs;
 }
 
 }  // namespace chromeos