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. 18static const HKEY kFireFoxRegistryPaths[] = { 19 HKEY_CURRENT_USER, 20 HKEY_LOCAL_MACHINE 21}; 22 23int 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 43FilePath 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 65FilePath 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