1 // Copyright 2012 The Chromium Authors
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 "base/base_paths.h"
6
7 #include <windows.h>
8
9 #include <KnownFolders.h>
10 #include <shlobj.h>
11
12 #include "base/environment.h"
13 #include "base/files/file_path.h"
14 #include "base/path_service.h"
15 #include "base/strings/string_util.h"
16 #include "base/strings/utf_string_conversions.h"
17 #include "base/win/current_module.h"
18 #include "base/win/scoped_co_mem.h"
19 #include "base/win/windows_version.h"
20
21 using base::FilePath;
22
23 namespace base {
24
PathProviderWin(int key,FilePath * result)25 bool PathProviderWin(int key, FilePath* result) {
26 // We need to go compute the value. It would be nice to support paths with
27 // names longer than MAX_PATH, but the system functions don't seem to be
28 // designed for it either, with the exception of GetTempPath (but other
29 // things will surely break if the temp path is too long, so we don't bother
30 // handling it.
31 wchar_t system_buffer[MAX_PATH];
32 system_buffer[0] = 0;
33
34 FilePath cur;
35 switch (key) {
36 case base::FILE_EXE:
37 if (GetModuleFileName(NULL, system_buffer, MAX_PATH) == 0)
38 return false;
39 cur = FilePath(system_buffer);
40 break;
41 case base::FILE_MODULE: {
42 // the resource containing module is assumed to be the one that
43 // this code lives in, whether that's a dll or exe
44 if (GetModuleFileName(CURRENT_MODULE(), system_buffer, MAX_PATH) == 0)
45 return false;
46 cur = FilePath(system_buffer);
47 break;
48 }
49 case base::DIR_WINDOWS:
50 GetWindowsDirectory(system_buffer, MAX_PATH);
51 cur = FilePath(system_buffer);
52 break;
53 case base::DIR_SYSTEM:
54 GetSystemDirectory(system_buffer, MAX_PATH);
55 cur = FilePath(system_buffer);
56 break;
57 case base::DIR_PROGRAM_FILESX86:
58 if (win::OSInfo::GetArchitecture() != win::OSInfo::X86_ARCHITECTURE) {
59 if (FAILED(SHGetFolderPath(NULL, CSIDL_PROGRAM_FILESX86, NULL,
60 SHGFP_TYPE_CURRENT, system_buffer)))
61 return false;
62 cur = FilePath(system_buffer);
63 break;
64 }
65 // Fall through to base::DIR_PROGRAM_FILES if we're on an X86 machine.
66 [[fallthrough]];
67 case base::DIR_PROGRAM_FILES:
68 if (FAILED(SHGetFolderPath(NULL, CSIDL_PROGRAM_FILES, NULL,
69 SHGFP_TYPE_CURRENT, system_buffer)))
70 return false;
71 cur = FilePath(system_buffer);
72 break;
73 case base::DIR_PROGRAM_FILES6432:
74 #if !defined(_WIN64)
75 if (base::win::OSInfo::GetInstance()->IsWowX86OnAMD64() ||
76 base::win::OSInfo::GetInstance()->IsWowX86OnARM64()) {
77 std::unique_ptr<base::Environment> env(base::Environment::Create());
78 std::string programfiles_w6432;
79 // 32-bit process running in WOW64 sets ProgramW6432 environment
80 // variable. See
81 // https://msdn.microsoft.com/library/windows/desktop/aa384274.aspx.
82 if (!env->GetVar("ProgramW6432", &programfiles_w6432))
83 return false;
84 // GetVar returns UTF8 - convert back to Wide.
85 cur = FilePath(UTF8ToWide(programfiles_w6432));
86 break;
87 }
88 #endif
89 if (FAILED(SHGetFolderPath(NULL, CSIDL_PROGRAM_FILES, NULL,
90 SHGFP_TYPE_CURRENT, system_buffer)))
91 return false;
92 cur = FilePath(system_buffer);
93 break;
94 case base::DIR_IE_INTERNET_CACHE:
95 if (FAILED(SHGetFolderPath(NULL, CSIDL_INTERNET_CACHE, NULL,
96 SHGFP_TYPE_CURRENT, system_buffer)))
97 return false;
98 cur = FilePath(system_buffer);
99 break;
100 case base::DIR_COMMON_START_MENU:
101 if (FAILED(SHGetFolderPath(NULL, CSIDL_COMMON_PROGRAMS, NULL,
102 SHGFP_TYPE_CURRENT, system_buffer)))
103 return false;
104 cur = FilePath(system_buffer);
105 break;
106 case base::DIR_START_MENU:
107 if (FAILED(SHGetFolderPath(NULL, CSIDL_PROGRAMS, NULL, SHGFP_TYPE_CURRENT,
108 system_buffer)))
109 return false;
110 cur = FilePath(system_buffer);
111 break;
112 case base::DIR_COMMON_STARTUP:
113 if (FAILED(SHGetFolderPath(nullptr, CSIDL_COMMON_STARTUP, nullptr,
114 SHGFP_TYPE_CURRENT, system_buffer)))
115 return false;
116 cur = FilePath(system_buffer);
117 break;
118 case base::DIR_USER_STARTUP:
119 if (FAILED(SHGetFolderPath(nullptr, CSIDL_STARTUP, nullptr,
120 SHGFP_TYPE_CURRENT, system_buffer)))
121 return false;
122 cur = FilePath(system_buffer);
123 break;
124 case base::DIR_ROAMING_APP_DATA:
125 if (FAILED(SHGetFolderPath(NULL, CSIDL_APPDATA, NULL, SHGFP_TYPE_CURRENT,
126 system_buffer)))
127 return false;
128 cur = FilePath(system_buffer);
129 break;
130 case base::DIR_COMMON_APP_DATA:
131 if (FAILED(SHGetFolderPath(NULL, CSIDL_COMMON_APPDATA, NULL,
132 SHGFP_TYPE_CURRENT, system_buffer)))
133 return false;
134 cur = FilePath(system_buffer);
135 break;
136 case base::DIR_LOCAL_APP_DATA:
137 if (FAILED(SHGetFolderPath(NULL, CSIDL_LOCAL_APPDATA, NULL,
138 SHGFP_TYPE_CURRENT, system_buffer)))
139 return false;
140 cur = FilePath(system_buffer);
141 break;
142 case base::DIR_SRC_TEST_DATA_ROOT: {
143 FilePath executableDir;
144 // On Windows, unit tests execute two levels deep from the source root.
145 // For example: chrome/{Debug|Release}/ui_tests.exe
146 PathService::Get(base::DIR_EXE, &executableDir);
147 cur = executableDir.DirName().DirName();
148 break;
149 }
150 case base::DIR_APP_SHORTCUTS: {
151 base::win::ScopedCoMem<wchar_t> path_buf;
152 if (FAILED(SHGetKnownFolderPath(FOLDERID_ApplicationShortcuts, 0, NULL,
153 &path_buf)))
154 return false;
155
156 cur = FilePath(path_buf.get());
157 break;
158 }
159 case base::DIR_USER_DESKTOP:
160 if (FAILED(SHGetFolderPath(NULL, CSIDL_DESKTOPDIRECTORY, NULL,
161 SHGFP_TYPE_CURRENT, system_buffer))) {
162 return false;
163 }
164 cur = FilePath(system_buffer);
165 break;
166 case base::DIR_COMMON_DESKTOP:
167 if (FAILED(SHGetFolderPath(NULL, CSIDL_COMMON_DESKTOPDIRECTORY, NULL,
168 SHGFP_TYPE_CURRENT, system_buffer))) {
169 return false;
170 }
171 cur = FilePath(system_buffer);
172 break;
173 case base::DIR_USER_QUICK_LAUNCH:
174 if (!PathService::Get(base::DIR_ROAMING_APP_DATA, &cur))
175 return false;
176 // According to various sources, appending
177 // "Microsoft\Internet Explorer\Quick Launch" to %appdata% is the only
178 // reliable way to get the quick launch folder across all versions of
179 // Windows.
180 // http://stackoverflow.com/questions/76080/how-do-you-reliably-get-the-quick-
181 // http://www.microsoft.com/technet/scriptcenter/resources/qanda/sept05/hey0901.mspx
182 cur = cur.Append(FILE_PATH_LITERAL("Microsoft"))
183 .Append(FILE_PATH_LITERAL("Internet Explorer"))
184 .Append(FILE_PATH_LITERAL("Quick Launch"));
185 break;
186 case base::DIR_TASKBAR_PINS: {
187 if (!PathService::Get(base::DIR_USER_QUICK_LAUNCH, &cur))
188 return false;
189 cur = cur.Append(FILE_PATH_LITERAL("User Pinned"))
190 .Append(FILE_PATH_LITERAL("TaskBar"));
191 break;
192 }
193 case base::DIR_IMPLICIT_APP_SHORTCUTS:
194 if (!PathService::Get(base::DIR_USER_QUICK_LAUNCH, &cur))
195 return false;
196 cur = cur.Append(FILE_PATH_LITERAL("User Pinned"))
197 .Append(FILE_PATH_LITERAL("ImplicitAppShortcuts"));
198 break;
199 case base::DIR_WINDOWS_FONTS:
200 if (FAILED(SHGetFolderPath(NULL, CSIDL_FONTS, NULL, SHGFP_TYPE_CURRENT,
201 system_buffer))) {
202 return false;
203 }
204 cur = FilePath(system_buffer);
205 break;
206 default:
207 return false;
208 }
209
210 *result = cur;
211 return true;
212 }
213
214 } // namespace base
215