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