Update keygen to use correct NSS slot on ChromeOS multiprofile.

chrome/browser/ui/crypto_module_delegate_nss.cc

 // Copyright (c) 2012 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/ui/crypto_module_password_dialog.h"
+#include "chrome/browser/ui/crypto_module_delegate_nss.h"
 
 #include "base/basictypes.h"
 #include "base/bind.h"
-#include "base/strings/utf_string_conversions.h"
 #include "base/synchronization/waitable_event.h"
+#include "chrome/browser/net/nss_context.h"
 #include "content/public/browser/browser_thread.h"
-#include "crypto/crypto_module_blocking_password_delegate.h"
-#include "grit/generated_resources.h"
-#include "ui/base/l10n/l10n_util.h"
-#include "url/gurl.h"
+#include "crypto/nss_crypto_module_delegate.h"
 
 using content::BrowserThread;
 
 namespace chrome {
 
 namespace {
 
-class CryptoModuleBlockingDialogDelegate
-    : public crypto::CryptoModuleBlockingPasswordDelegate {
+class ChromeNSSCryptoModuleDelegate
+    : public crypto::NSSCryptoModuleDelegate {
  public:
-  CryptoModuleBlockingDialogDelegate(CryptoModulePasswordReason reason,
-                                     const std::string& server)
+  ChromeNSSCryptoModuleDelegate(content::ResourceContext* context,
+                                CryptoModulePasswordReason reason,
+                                const std::string& server)
       : event_(false, false),
+        context_(context),
         reason_(reason),
         server_(server),
-        cancelled_(false) {
+        cancelled_(false) {}
+
+  virtual ~ChromeNSSCryptoModuleDelegate() {}
+
+  // crypto::NSSCryptoModuleDelegate implementation.
+  virtual void Initialize(const base::Closure& callback) OVERRIDE {
+    DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
+
+    content::ResourceContext* context = context_;
+    context_ = NULL;
+    OnPrivateNSSKeySlotForResourceContextReady(
+        context,
+        base::Bind(&ChromeNSSCryptoModuleDelegate::DidGetSlot,
+                   // Caller is responsible for keeping us alive until we run
+                   // the callback.
+                   base::Unretained(this),
+                   callback));
   }
 
-  virtual ~CryptoModuleBlockingDialogDelegate() {
-    // Make sure we clear the password in memory.
-    password_.replace(0, password_.size(), password_.size(), 0);
+  // TODO(mattm): allow choosing which slot to generate and store the key.
+  virtual crypto::ScopedPK11Slot RequestSlot() OVERRIDE {
+    return slot_.Pass();
   }
 
-  // crypto::CryptoModuleBlockingDialogDelegate implementation.
+  // crypto::CryptoModuleBlockingPasswordDelegate implementation.
   virtual std::string RequestPassword(const std::string& slot_name,
                                       bool retry,
                                       bool* cancelled) OVERRIDE {
     DCHECK(!BrowserThread::CurrentlyOn(BrowserThread::UI));
     DCHECK(!event_.IsSignaled());
     event_.Reset();
 
     if (BrowserThread::PostTask(
             BrowserThread::UI, FROM_HERE,
-            base::Bind(&CryptoModuleBlockingDialogDelegate::ShowDialog,
+            base::Bind(&ChromeNSSCryptoModuleDelegate::ShowDialog,
                        // We block on event_ until the task completes, so
                        // there's no need to ref-count.
                        base::Unretained(this),
                        slot_name,
                        retry))) {
       event_.Wait();
     }
     *cancelled = cancelled_;
     return password_;
   }
 
  private:
   void ShowDialog(const std::string& slot_name,
                   bool retry) {
     DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
     ShowCryptoModulePasswordDialog(
         slot_name,
         retry,
         reason_,
         server_,
         NULL,  // TODO(mattm): Supply parent window.
-        base::Bind(&CryptoModuleBlockingDialogDelegate::GotPassword,
+        base::Bind(&ChromeNSSCryptoModuleDelegate::GotPassword,
                    // We block on event_ until the task completes, so
                    // there's no need to ref-count.
                    base::Unretained(this)));
   }
 
   void GotPassword(const char* password) {
     if (password)
       password_ = password;
     else
       cancelled_ = true;
     event_.Signal();
   }
 
+  void DidGetSlot(const base::Closure& callback, crypto::ScopedPK11Slot slot) {
+    slot_ = slot.Pass();
+    callback.Run();
+  }
+
   base::WaitableEvent event_;
+  content::ResourceContext* context_;
   CryptoModulePasswordReason reason_;
   std::string server_;
   std::string password_;
+  crypto::ScopedPK11Slot slot_;
   bool cancelled_;
 
-  DISALLOW_COPY_AND_ASSIGN(CryptoModuleBlockingDialogDelegate);
+  DISALLOW_COPY_AND_ASSIGN(ChromeNSSCryptoModuleDelegate);
 };
 
 }  // namespace
 
-crypto::CryptoModuleBlockingPasswordDelegate*
-    NewCryptoModuleBlockingDialogDelegate(CryptoModulePasswordReason reason,
-                                          const std::string& server) {
-  return new CryptoModuleBlockingDialogDelegate(reason, server);
+crypto::NSSCryptoModuleDelegate* CreateNSSCryptoModuleDelegate(
+    content::ResourceContext* context,
+    CryptoModulePasswordReason reason,
+    const std::string& server) {
+  return new ChromeNSSCryptoModuleDelegate(context, reason, server);
 }
 
+
 }  // namespace chrome