ChromeOS: Fix crash if login profile triggers client auth.

net/cert/nss_profile_filter_chromeos_unittest.cc

Index: net/cert/nss_profile_filter_chromeos_unittest.cc
diff --git a/net/cert/nss_profile_filter_chromeos_unittest.cc b/net/cert/nss_profile_filter_chromeos_unittest.cc
new file mode 100644
index 0000000000000000000000000000000000000000..8f3806186139fb97608dcd718cc9b9653e9b7558
--- /dev/null
+++ b/net/cert/nss_profile_filter_chromeos_unittest.cc
@@ -0,0 +1,176 @@
+// Copyright 2014 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 "net/cert/nss_profile_filter_chromeos.h"
+
+#include <cert.h>
+#include <pk11pub.h>
+#include <secmod.h>
+
+#include "crypto/nss_util.h"
+#include "crypto/nss_util_internal.h"
+#include "crypto/scoped_nss_types.h"
+#include "net/base/test_data_directory.h"
+#include "net/test/cert_test_util.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace net {
+
+namespace {
+
+crypto::ScopedPK11Slot GetRootCertsSlot() {
+  crypto::AutoSECMODListReadLock auto_lock;
+  SECMODModuleList* head = SECMOD_GetDefaultModuleList();
+  for (SECMODModuleList* item = head; item != NULL; item = item->next) {
+    int slot_count = item->module->loaded ? item->module->slotCount : 0;
+    for (int i = 0; i < slot_count; i++) {
+      PK11SlotInfo* slot = item->module->slots[i];
+      if (!PK11_IsPresent(slot))
+        continue;
+      if (PK11_HasRootCerts(slot))
+        return crypto::ScopedPK11Slot(PK11_ReferenceSlot(slot));
+    }
+  }
+  return crypto::ScopedPK11Slot();
+}
+
+CertificateList ListCertsInSlot(PK11SlotInfo* slot) {
+  CertificateList result;
+  CERTCertList* cert_list = PK11_ListCertsInSlot(slot);
+  for (CERTCertListNode* node = CERT_LIST_HEAD(cert_list);
+       !CERT_LIST_END(node, cert_list);
+       node = CERT_LIST_NEXT(node)) {
+    result.push_back(X509Certificate::CreateFromHandle(
+        node->cert, X509Certificate::OSCertHandles()));
+  }
+  CERT_DestroyCertList(cert_list);
+
+  // Sort the result so that test comparisons can be deterministic.
+  std::sort(result.begin(), result.end(), X509Certificate::LessThan());
+  return result;
+}
+
+}
+
+class NSSProfileFilterChromeOSTest : public testing::Test {
+ public:
+  NSSProfileFilterChromeOSTest() : user_1_("user1"), user_2_("user2") {}
+
+  virtual void SetUp() OVERRIDE {
+    // Initialize nss_util slots.
+    ASSERT_TRUE(user_1_.constructed_successfully());
+    ASSERT_TRUE(user_2_.constructed_successfully());
+    user_1_.FinishInit();
+    user_2_.FinishInit();
+
+    // TODO(mattm): more accurately test public/private slot filtering somehow.
+    // (The slots used to initialize a profile filter should be separate slots
+    // in separate modules, while ScopedTestNSSChromeOSUser uses the same slot
+    // for both.)
+    crypto::ScopedPK11Slot private_slot_1(crypto::GetPrivateSlotForChromeOSUser(
+        user_1_.username_hash(),
+        base::Callback<void(crypto::ScopedPK11Slot)>()));
+    ASSERT_TRUE(private_slot_1.get());
+    profile_filter_1_.Init(
+        crypto::GetPublicSlotForChromeOSUser(user_1_.username_hash()),
+        private_slot_1.Pass());
+
+    crypto::ScopedPK11Slot private_slot_2(crypto::GetPrivateSlotForChromeOSUser(
+        user_2_.username_hash(),
+        base::Callback<void(crypto::ScopedPK11Slot)>()));
+    ASSERT_TRUE(private_slot_2.get());
+    profile_filter_2_.Init(
+        crypto::GetPublicSlotForChromeOSUser(user_2_.username_hash()),
+        private_slot_2.Pass());
+
+    certs_ = CreateCertificateListFromFile(GetTestCertsDirectory(),
+                                           "root_ca_cert.pem",
+                                           X509Certificate::FORMAT_AUTO);
+    ASSERT_EQ(1U, certs_.size());
+  }
+
+ protected:
+  CertificateList certs_;
+  crypto::ScopedTestNSSChromeOSUser user_1_;
+  crypto::ScopedTestNSSChromeOSUser user_2_;
+  NSSProfileFilterChromeOS no_slots_profile_filter_;
+  NSSProfileFilterChromeOS profile_filter_1_;
+  NSSProfileFilterChromeOS profile_filter_2_;
+};
+
+TEST_F(NSSProfileFilterChromeOSTest, TempCertAllowed) {
+  EXPECT_EQ(NULL, certs_[0]->os_cert_handle()->slot);
+  EXPECT_TRUE(no_slots_profile_filter_.IsCertAllowed(certs_[0]));
+  EXPECT_TRUE(profile_filter_1_.IsCertAllowed(certs_[0]));
+  EXPECT_TRUE(profile_filter_2_.IsCertAllowed(certs_[0]));
+}
+
+TEST_F(NSSProfileFilterChromeOSTest, InternalSlotAllowed) {
+  crypto::ScopedPK11Slot internal_slot(PK11_GetInternalSlot());
+  ASSERT_TRUE(internal_slot.get());
+  EXPECT_TRUE(no_slots_profile_filter_.IsModuleAllowed(internal_slot.get()));
+  EXPECT_TRUE(profile_filter_1_.IsModuleAllowed(internal_slot.get()));
+  EXPECT_TRUE(profile_filter_2_.IsModuleAllowed(internal_slot.get()));
+
+  crypto::ScopedPK11Slot internal_key_slot(PK11_GetInternalKeySlot());
+  ASSERT_TRUE(internal_key_slot.get());
+  EXPECT_TRUE(
+      no_slots_profile_filter_.IsModuleAllowed(internal_key_slot.get()));
+  EXPECT_TRUE(profile_filter_1_.IsModuleAllowed(internal_key_slot.get()));
+  EXPECT_TRUE(profile_filter_2_.IsModuleAllowed(internal_key_slot.get()));
+}
+
+TEST_F(NSSProfileFilterChromeOSTest, RootCertsAllowed) {
+  crypto::ScopedPK11Slot root_certs_slot(GetRootCertsSlot());
+  ASSERT_TRUE(root_certs_slot.get());
+  EXPECT_TRUE(no_slots_profile_filter_.IsModuleAllowed(root_certs_slot.get()));
+  EXPECT_TRUE(profile_filter_1_.IsModuleAllowed(root_certs_slot.get()));
+  EXPECT_TRUE(profile_filter_2_.IsModuleAllowed(root_certs_slot.get()));
+
+  CertificateList root_certs(ListCertsInSlot(root_certs_slot.get()));
+  ASSERT_FALSE(root_certs.empty());
+  EXPECT_TRUE(no_slots_profile_filter_.IsCertAllowed(root_certs[0]));
+  EXPECT_TRUE(profile_filter_1_.IsCertAllowed(root_certs[0]));
+  EXPECT_TRUE(profile_filter_2_.IsCertAllowed(root_certs[0]));
+}
+
+TEST_F(NSSProfileFilterChromeOSTest, SoftwareSlots) {
+  crypto::ScopedPK11Slot slot_1(
+      crypto::GetPublicSlotForChromeOSUser(user_1_.username_hash()));
+  ASSERT_TRUE(slot_1);
+  crypto::ScopedPK11Slot slot_2(
+      crypto::GetPublicSlotForChromeOSUser(user_2_.username_hash()));
+  ASSERT_TRUE(slot_2);
+
+  scoped_refptr<X509Certificate> cert_1 = certs_[0];
+  CertificateList certs_2 = CreateCertificateListFromFile(
+      GetTestCertsDirectory(), "ok_cert.pem", X509Certificate::FORMAT_AUTO);
+  ASSERT_EQ(1U, certs_2.size());
+  scoped_refptr<X509Certificate> cert_2 = certs_2[0];
+
+  ASSERT_EQ(SECSuccess,
+            PK11_ImportCert(slot_1.get(),
+                            cert_1->os_cert_handle(),
+                            CK_INVALID_HANDLE,
+                            "cert1",
+                            PR_FALSE /* includeTrust (unused) */));
+
+  ASSERT_EQ(SECSuccess,
+            PK11_ImportCert(slot_2.get(),
+                            cert_2->os_cert_handle(),
+                            CK_INVALID_HANDLE,
+                            "cert2",
+                            PR_FALSE /* includeTrust (unused) */));
+
+  EXPECT_FALSE(no_slots_profile_filter_.IsCertAllowed(cert_1));
+  EXPECT_FALSE(no_slots_profile_filter_.IsCertAllowed(cert_2));
+
+  EXPECT_TRUE(profile_filter_1_.IsCertAllowed(cert_1));
+  EXPECT_FALSE(profile_filter_1_.IsCertAllowed(cert_2));
+
+  EXPECT_FALSE(profile_filter_2_.IsCertAllowed(cert_1));
+  EXPECT_TRUE(profile_filter_2_.IsCertAllowed(cert_2));
+}
+
+}  // namespace net