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/files/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 } 83