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