OLD | NEW |
| (Empty) |
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/browser/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 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"\\" + 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 } | |
OLD | NEW |