chrome/common/importer/firefox_importer_utils_win.cc

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "chrome/common/importer/firefox_importer_utils.h"
   6
   7 #include <shlobj.h>
   8
   9 #include "base/file_util.h"
  10 #include "base/path_service.h"
  11 #include "base/strings/string16.h"
  12 #include "base/win/registry.h"
  13
  14 // NOTE: Keep these in order since we need test all those paths according
  15 // to priority. For example. One machine has multiple users. One non-admin
  16 // user installs Firefox 2, which causes there is a Firefox2 entry under HKCU.
  17 // One admin user installs Firefox 3, which causes there is a Firefox 3 entry
  18 // under HKLM. So when the non-admin user log in, we should deal with Firefox 2
  19 // related data instead of Firefox 3.
  20 static const HKEY kFireFoxRegistryPaths[] = {
  21   HKEY_CURRENT_USER,
  22   HKEY_LOCAL_MACHINE
  23 };
  24
  25 static const wchar_t* kFirefoxPath = L"Software\\Mozilla\\Mozilla Firefox";
  26 static const wchar_t* kCurrentVersion = L"CurrentVersion";
  27
  28 int GetCurrentFirefoxMajorVersionFromRegistry() {
  29   TCHAR ver_buffer[128];
  30   DWORD ver_buffer_length = sizeof(ver_buffer);
  31   int highest_version = 0;
  32   // When installing Firefox with admin account, the product keys will be
  33   // written under HKLM\Mozilla. Otherwise it the keys will be written under
  34   // HKCU\Mozilla.
  35   for (int i = 0; i < arraysize(kFireFoxRegistryPaths); ++i) {
  36     base::win::RegKey reg_key(kFireFoxRegistryPaths[i], kFirefoxPath,
  37                               KEY_READ);
  38
  39     LONG result = reg_key.ReadValue(kCurrentVersion, ver_buffer,
  40                                     &ver_buffer_length, NULL);
  41     if (result != ERROR_SUCCESS)
  42       continue;
  43     highest_version = std::max(highest_version, _wtoi(ver_buffer));
  44   }
  45   return highest_version;
  46 }
  47
  48 base::FilePath GetFirefoxInstallPathFromRegistry() {
  49   // Detects the path that Firefox is installed in.
  50   base::string16 registry_path = kFirefoxPath;
  51   wchar_t buffer[MAX_PATH];
  52   DWORD buffer_length = sizeof(buffer);
  53   base::win::RegKey reg_key(HKEY_LOCAL_MACHINE, registry_path.c_str(),
  54                             KEY_READ);
  55   LONG result = reg_key.ReadValue(kCurrentVersion, buffer,
  56                                   &buffer_length, NULL);
  57   if (result != ERROR_SUCCESS)
  58     return base::FilePath();
  59
  60   registry_path += L"\\" + base::string16(buffer) + L"\\Main";
  61   buffer_length = sizeof(buffer);
  62   base::win::RegKey reg_key_directory(HKEY_LOCAL_MACHINE,
  63                                       registry_path.c_str(), KEY_READ);
  64   result = reg_key_directory.ReadValue(L"Install Directory", buffer,
  65                                        &buffer_length, NULL);
  66
  67   return (result != ERROR_SUCCESS) ? base::FilePath() : base::FilePath(buffer);
  68 }
  69
  70 base::FilePath GetProfilesINI() {
  71   base::FilePath ini_file;
  72   // The default location of the profile folder containing user data is
  73   // under the "Application Data" folder in Windows XP, Vista, and 7.
  74   if (!PathService::Get(base::DIR_APP_DATA, &ini_file))
  75     return base::FilePath();
  76
  77   ini_file = ini_file.AppendASCII("Mozilla");
  78   ini_file = ini_file.AppendASCII("Firefox");
  79   ini_file = ini_file.AppendASCII("profiles.ini");
  80
  81   return base::PathExists(ini_file) ? ini_file : base::FilePath();
  82 }