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
GetCurrentFirefoxMajorVersionFromRegistry()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
GetFirefoxInstallPathFromRegistry()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
GetProfilesINI()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