Linux: Import passwords from Firefox.

chrome/browser/importer/firefox_importer_utils.cc

-// Copyright (c) 2006-2008 The Chromium Authors. All rights reserved.
+// 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/importer/firefox_importer_utils.h"
 
 #include <algorithm>
 
 #if defined(OS_WIN)
 #include <shlobj.h>
 
 #include "app/win_util.h"
 #include "base/registry.h"
 #endif
 #include "base/logging.h"
 #include "base/string_util.h"
 #include "base/sys_string_conversions.h"
 #include "chrome/browser/search_engines/template_url.h"
 #include "chrome/browser/search_engines/template_url_model.h"
 #include "chrome/browser/search_engines/template_url_parser.h"
 #include "googleurl/src/gurl.h"
 #include "net/base/base64.h"
 
 
 namespace {
 
-#if defined(OS_WIN)
-// NOTE: Keep these in order since we need test all those paths according
-// to priority. For example. One machine has multiple users. One non-admin
-// user installs Firefox 2, which causes there is a Firefox2 entry under HKCU.
-// One admin user installs Firefox 3, which causes there is a Firefox 3 entry
-// under HKLM. So when the non-admin user log in, we should deal with Firefox 2
-// related data instead of Firefox 3.
-static const HKEY kFireFoxRegistryPaths[] = {
-  HKEY_CURRENT_USER,
-  HKEY_LOCAL_MACHINE
-};
-#endif
-
 // FirefoxURLParameterFilter is used to remove parameter mentioning Firefox from
 // the search URL when importing search engines.
 class FirefoxURLParameterFilter : public TemplateURLParser::ParameterFilter {
  public:
   FirefoxURLParameterFilter() { }
   ~FirefoxURLParameterFilter() { }
 
   // TemplateURLParser::ParameterFilter method.
   virtual bool KeepParameter(const std::string& key,
                              const std::string& value) {
@@ skipping 27 matching lines @@
   void set_func(SetDllDirectoryFunc func) { func_ = func; }
 
  private:
   SetDllDirectoryFunc func_;
 };
 #endif
 
 }  // namespace
 
 #if defined(OS_WIN)
+// NOTE: Keep these in order since we need test all those paths according
+// to priority. For example. One machine has multiple users. One non-admin
+// user installs Firefox 2, which causes there is a Firefox2 entry under HKCU.
+// One admin user installs Firefox 3, which causes there is a Firefox 3 entry
+// under HKLM. So when the non-admin user log in, we should deal with Firefox 2
+// related data instead of Firefox 3.
+static const HKEY kFireFoxRegistryPaths[] = {
+  HKEY_CURRENT_USER,
+  HKEY_LOCAL_MACHINE
+};
+
 int GetCurrentFirefoxMajorVersionFromRegistry() {
   TCHAR ver_buffer[128];
   DWORD ver_buffer_length = sizeof(ver_buffer);
   int highest_version = 0;
   // When installing Firefox with admin account, the product keys will be
   // written under HKLM\Mozilla. Otherwise it the keys will be written under
   // HKCU\Mozilla.
   for (int i = 0; i < arraysize(kFireFoxRegistryPaths); ++i) {
     bool result = ReadFromRegistry(kFireFoxRegistryPaths[i],
         L"Software\\Mozilla\\Mozilla Firefox",
@@ skipping 350 matching lines @@
 const wchar_t NSSDecryptor::kNSS3Library[] = L"nss3.dll";
 const wchar_t NSSDecryptor::kSoftokn3Library[] = L"softokn3.dll";
 const wchar_t NSSDecryptor::kPLDS4Library[] = L"plds4.dll";
 const wchar_t NSSDecryptor::kNSPR4Library[] = L"nspr4.dll";
 
 NSSDecryptor::NSSDecryptor()
     : NSS_Init(NULL), NSS_Shutdown(NULL), PK11_GetInternalKeySlot(NULL),
       PK11_CheckUserPassword(NULL), PK11_FreeSlot(NULL),
       PK11_Authenticate(NULL), PK11SDR_Decrypt(NULL), SECITEM_FreeItem(NULL),
       PL_ArenaFinish(NULL), PR_Cleanup(NULL),
-#if defined(OS_WIN)
       nss3_dll_(NULL), softokn3_dll_(NULL),
-#endif
       is_nss_initialized_(false) {
 }
 
 NSSDecryptor::~NSSDecryptor() {
   Free();
 }
 
 bool NSSDecryptor::Init(const std::wstring& dll_path,
                         const std::wstring& db_path) {
 #if defined(OS_WIN)
@@ skipping 37 matching lines @@
     // harmless for NSS 3.11.
     path = dll_path;
     file_util::AppendToPath(&path, kSoftokn3Library);
     softokn3_dll_ = LoadLibraryEx(path.c_str(), NULL,
                                   LOAD_WITH_ALTERED_SEARCH_PATH);
     if (softokn3_dll_ == NULL) {
       Free();
       return false;
     }
   }
+  HMODULE plds4_dll = GetModuleHandle(kPLDS4Library);
+  HMODULE nspr4_dll = GetModuleHandle(kNSPR4Library);
+#elif defined(OS_LINUX)
+  nss3_dll_ = base::LoadNativeLibrary(FilePath("libnss3.so"));
+  if (nss3_dll_ == NULL)
+    return false;
+  base::NativeLibrary plds4_dll = base::LoadNativeLibrary(
+      FilePath("libplds4.so"));
+  base::NativeLibrary nspr4_dll = base::LoadNativeLibrary(
+      FilePath("libnspr4.so"));
+#else
+  // TODO(port): Check on MAC
+  base::NativeLibrary plds4_dll, nspr4_dll;
+  NOTIMPLEMENTED();
+  return false;
+#endif
 
   // NSPR DLLs are already loaded now.
-  HMODULE plds4_dll = GetModuleHandle(kPLDS4Library);
-  HMODULE nspr4_dll = GetModuleHandle(kNSPR4Library);
   if (plds4_dll == NULL || nspr4_dll == NULL) {
     Free();
     return false;
   }
 
   // Gets the function address.
-  NSS_Init = (NSSInitFunc)GetProcAddress(nss3_dll_, "NSS_Init");
-  NSS_Shutdown = (NSSShutdownFunc)GetProcAddress(nss3_dll_, "NSS_Shutdown");
+  NSS_Init = (NSSInitFunc)
+      base::GetFunctionPointerFromNativeLibrary(nss3_dll_, "NSS_Init");
+  NSS_Shutdown = (NSSShutdownFunc)
+      base::GetFunctionPointerFromNativeLibrary(nss3_dll_, "NSS_Shutdown");
   PK11_GetInternalKeySlot = (PK11GetInternalKeySlotFunc)
-      GetProcAddress(nss3_dll_, "PK11_GetInternalKeySlot");
-  PK11_FreeSlot = (PK11FreeSlotFunc)GetProcAddress(nss3_dll_, "PK11_FreeSlot");
+      base::GetFunctionPointerFromNativeLibrary(nss3_dll_,
+                                                "PK11_GetInternalKeySlot");
+  PK11_FreeSlot = (PK11FreeSlotFunc)
+      base::GetFunctionPointerFromNativeLibrary(nss3_dll_, "PK11_FreeSlot");
   PK11_Authenticate = (PK11AuthenticateFunc)
-      GetProcAddress(nss3_dll_, "PK11_Authenticate");
+      base::GetFunctionPointerFromNativeLibrary(nss3_dll_, "PK11_Authenticate");
   PK11SDR_Decrypt = (PK11SDRDecryptFunc)
-      GetProcAddress(nss3_dll_, "PK11SDR_Decrypt");
+      base::GetFunctionPointerFromNativeLibrary(nss3_dll_, "PK11SDR_Decrypt");
   SECITEM_FreeItem = (SECITEMFreeItemFunc)
-      GetProcAddress(nss3_dll_, "SECITEM_FreeItem");
+      base::GetFunctionPointerFromNativeLibrary(nss3_dll_, "SECITEM_FreeItem");
   PL_ArenaFinish = (PLArenaFinishFunc)
-      GetProcAddress(plds4_dll, "PL_ArenaFinish");
-  PR_Cleanup = (PRCleanupFunc)GetProcAddress(nspr4_dll, "PR_Cleanup");
+      base::GetFunctionPointerFromNativeLibrary(plds4_dll, "PL_ArenaFinish");
+  PR_Cleanup = (PRCleanupFunc)
+      base::GetFunctionPointerFromNativeLibrary(nspr4_dll, "PR_Cleanup");
 
   if (NSS_Init == NULL || NSS_Shutdown == NULL ||
       PK11_GetInternalKeySlot == NULL || PK11_FreeSlot == NULL ||
       PK11_Authenticate == NULL || PK11SDR_Decrypt == NULL ||
       SECITEM_FreeItem == NULL || PL_ArenaFinish == NULL ||
       PR_Cleanup == NULL) {
     Free();
     return false;
   }
 
   SECStatus result = NSS_Init(base::SysWideToNativeMB(db_path).c_str());
   if (result != SECSuccess) {
     Free();
     return false;
   }
 
   is_nss_initialized_ = true;
   return true;
-#else
-  // TODO(port): Load NSS.
-  NOTIMPLEMENTED();
-  return false;
-#endif
 }
 
 void NSSDecryptor::Free() {
   if (is_nss_initialized_) {
     NSS_Shutdown();
     PL_ArenaFinish();
     PR_Cleanup();
     is_nss_initialized_ = false;
   }
-#if defined(OS_WIN)
+#if defined(OS_WIN) || defined(OS_LINUX)
   if (softokn3_dll_ != NULL)
-    FreeLibrary(softokn3_dll_);
-  softokn3_dll_ = NULL;
+    base::UnloadNativeLibrary(softokn3_dll_);
   if (nss3_dll_ != NULL)
-    FreeLibrary(nss3_dll_);
-  nss3_dll_ = NULL;
+    base::UnloadNativeLibrary(nss3_dll_);
 #endif
   NSS_Init = NULL;
   NSS_Shutdown = NULL;
   PK11_GetInternalKeySlot = NULL;
   PK11_FreeSlot = NULL;
   PK11_Authenticate = NULL;
   PK11SDR_Decrypt = NULL;
   SECITEM_FreeItem = NULL;
   PL_ArenaFinish = NULL;
   PR_Cleanup = NULL;
+  nss3_dll_ = NULL;
+  softokn3_dll_ = NULL;
 }
 
 // This method is based on some Firefox code in
 //   security/manager/ssl/src/nsSDR.cpp
 // The license block is:
 
 /* ***** BEGIN LICENSE BLOCK *****
 * Version: MPL 1.1/GPL 2.0/LGPL 2.1
 *
 * The contents of this file are subject to the Mozilla Public License Version
@@ skipping 23 matching lines @@
 * under the terms of either the GPL or the LGPL, and not to allow others to
 * use your version of this file under the terms of the MPL, indicate your
 * decision by deleting the provisions above and replace them with the notice
 * and other provisions required by the GPL or the LGPL. If you do not delete
 * the provisions above, a recipient may use your version of this file under
 * the terms of any one of the MPL, the GPL or the LGPL.
 *
 * ***** END LICENSE BLOCK ***** */
 
 std::wstring NSSDecryptor::Decrypt(const std::string& crypt) const {
-#if defined(OS_WIN)
-  // Do nothing if NSS is not loaded.
-  if (!nss3_dll_)
-    return std::wstring();
-#else
+#if !defined(OS_WIN) && !defined(OS_LINUX)
   // TODO(port): Load nss3.
   NOTIMPLEMENTED();
   return std::wstring();
 #endif
 
-  std::string plain;
+  // Do nothing if NSS is not loaded.
+  if (!nss3_dll_)
+    return std::wstring();
 
   // The old style password is encoded in base64. They are identified
   // by a leading '~'. Otherwise, we should decrypt the text.
+  std::string plain;
   if (crypt[0] != '~') {
     std::string decoded_data;
     net::Base64Decode(crypt, &decoded_data);
     PK11SlotInfo* slot = NULL;
     slot = PK11_GetInternalKeySlot();
     SECStatus result = PK11_Authenticate(slot, PR_TRUE, NULL);
     if (result != SECSuccess) {
       PK11_FreeSlot(slot);
       return std::wstring();
     }
@@ skipping 138 matching lines @@
       }
       // Version 3 has an extra line for further use.
       if (version == 3) {
         ++begin;
       }
 
       forms->push_back(form);
     }
   }
 }