Linux: Fix NSSDecryptor to import Firefox passwords from the user database correctly.

chrome/browser/importer/firefox_profile_lock_unittest.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 "testing/gtest/include/gtest/gtest.h"
 
 #include "base/file_util.h"
 #include "base/path_service.h"
 #include "base/process_util.h"
 #include "base/string_util.h"
+#include "build/build_config.h"
 #include "chrome/browser/importer/firefox_profile_lock.h"
 #include "chrome/common/chrome_paths.h"
+#include "chrome/test/file_test_utils.h"
 
 class FirefoxProfileLockTest : public testing::Test {
  public:
  protected:
   virtual void SetUp() {
     ASSERT_TRUE(PathService::Get(base::DIR_TEMP, &test_path_));
     std::wstring dir_name(L"FirefoxProfileLockTest");
     dir_name.append(StringPrintf(L"-%d", base::GetCurrentProcId()));
     test_path_ = test_path_.Append(FilePath::FromWStringHack(dir_name));
     file_util::Delete(test_path_, true);
     file_util::CreateDirectory(test_path_);
   }
 
   virtual void TearDown() {
     ASSERT_TRUE(file_util::Delete(test_path_, true));
     ASSERT_FALSE(file_util::PathExists(test_path_));
   }
 
   FilePath test_path_;
 };
 
 TEST_F(FirefoxProfileLockTest, LockTest) {
   FirefoxProfileLock lock1(test_path_.ToWStringHack());
   ASSERT_TRUE(lock1.HasAcquired());
   lock1.Unlock();
   ASSERT_FALSE(lock1.HasAcquired());
   lock1.Lock();
   ASSERT_TRUE(lock1.HasAcquired());
 }
+
+// Tests basic functionality and verifies that the lock file is deleted after
+// use.
+TEST_F(FirefoxProfileLockTest, ProfileLock) {
+  std::wstring test_path;
+  file_util::CreateNewTempDirectory(L"firefox_profile", &test_path);
+  FilePath lock_file_path = FilePath::FromWStringHack(test_path);
+  FileAutoDeleter deleter(lock_file_path);
+  lock_file_path = lock_file_path.Append(FirefoxProfileLock::kLockFileName);
+
+  scoped_ptr<FirefoxProfileLock> lock;
+  EXPECT_EQ(static_cast<FirefoxProfileLock*>(NULL), lock.get());
+  EXPECT_FALSE(file_util::PathExists(lock_file_path));
+  lock.reset(new FirefoxProfileLock(test_path));
+  EXPECT_TRUE(lock->HasAcquired());
+  EXPECT_TRUE(file_util::PathExists(lock_file_path));
+  lock->Unlock();
+  EXPECT_FALSE(lock->HasAcquired());
+
+  // In the posix code, we don't delete the file when releasing the lock.
+#if !defined(OS_POSIX)
+  EXPECT_FALSE(file_util::PathExists(lock_file_path));
+#endif  // !defined(OS_POSIX)
+  lock->Lock();
+  EXPECT_TRUE(lock->HasAcquired());
+  EXPECT_TRUE(file_util::PathExists(lock_file_path));
+  lock->Lock();
+  EXPECT_TRUE(lock->HasAcquired());
+  lock->Unlock();
+  EXPECT_FALSE(lock->HasAcquired());
+  // In the posix code, we don't delete the file when releasing the lock.
+#if !defined(OS_POSIX)
+  EXPECT_FALSE(file_util::PathExists(lock_file_path));
+#endif  // !defined(OS_POSIX)
+}
+
+// If for some reason the lock file is left behind by the previous owner, we
+// should still be able to lock it, at least in the Windows implementation.
+TEST_F(FirefoxProfileLockTest, ProfileLockOrphaned) {
+  std::wstring test_path;
+  file_util::CreateNewTempDirectory(L"firefox_profile", &test_path);
+  FilePath lock_file_path = FilePath::FromWStringHack(test_path);
+  FileAutoDeleter deleter(lock_file_path);
+  lock_file_path = lock_file_path.Append(FirefoxProfileLock::kLockFileName);
+
+  // Create the orphaned lock file.
+  FILE* lock_file = file_util::OpenFile(lock_file_path, "w");
+  ASSERT_TRUE(lock_file);
+  file_util::CloseFile(lock_file);
+  EXPECT_TRUE(file_util::PathExists(lock_file_path));
+
+  scoped_ptr<FirefoxProfileLock> lock;
+  EXPECT_EQ(static_cast<FirefoxProfileLock*>(NULL), lock.get());
+  lock.reset(new FirefoxProfileLock(test_path));
+  EXPECT_TRUE(lock->HasAcquired());
+  lock->Unlock();
+  EXPECT_FALSE(lock->HasAcquired());
+}
+
+// This is broken on POSIX since the same process is allowed to reacquire a
+// lock.
+#if !defined(OS_POSIX)
+// Tests two locks contending for the same lock file.
+TEST_F(FirefoxProfileLockTest, ProfileLockContention) {
+  std::wstring test_path;
+  file_util::CreateNewTempDirectory(L"firefox_profile", &test_path);
+  FileAutoDeleter deleter(FilePath::FromWStringHack(test_path));
+
+  scoped_ptr<FirefoxProfileLock> lock1;
+  EXPECT_EQ(static_cast<FirefoxProfileLock*>(NULL), lock1.get());
+  lock1.reset(new FirefoxProfileLock(test_path));
+  EXPECT_TRUE(lock1->HasAcquired());
+
+  scoped_ptr<FirefoxProfileLock> lock2;
+  EXPECT_EQ(static_cast<FirefoxProfileLock*>(NULL), lock2.get());
+  lock2.reset(new FirefoxProfileLock(test_path));
+  EXPECT_FALSE(lock2->HasAcquired());
+
+  lock1->Unlock();
+  EXPECT_FALSE(lock1->HasAcquired());
+
+  lock2->Lock();
+  EXPECT_TRUE(lock2->HasAcquired());
+  lock2->Unlock();
+  EXPECT_FALSE(lock2->HasAcquired());
+}
+#endif