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 std::unique_ptr<base::Environment> env(base::Environment::Create());
76 std::string programfiles_w6432;
77 // 32-bit process running in WOW64 sets ProgramW6432 environment
78 // variable. See
79 // https://msdn.microsoft.com/library/windows/desktop/aa384274.aspx.
80 if (!env->GetVar("ProgramW6432", &programfiles_w6432))
81 return false;
82 // GetVar returns UTF8 - convert back to Wide.
83 cur = FilePath(UTF8ToWide(programfiles_w6432));
84 break;
85 }
86 #endif
87 if (FAILED(SHGetFolderPath(NULL, CSIDL_PROGRAM_FILES, NULL,
88 SHGFP_TYPE_CURRENT, system_buffer)))
89 return false;
90 cur = FilePath(system_buffer);
91 break;
92 case base::DIR_IE_INTERNET_CACHE:
93 if (FAILED(SHGetFolderPath(NULL, CSIDL_INTERNET_CACHE, NULL,
94 SHGFP_TYPE_CURRENT, system_buffer)))
95 return false;
96 cur = FilePath(system_buffer);
97 break;
98 case base::DIR_COMMON_START_MENU:
99 if (FAILED(SHGetFolderPath(NULL, CSIDL_COMMON_PROGRAMS, NULL,
100 SHGFP_TYPE_CURRENT, system_buffer)))
101 return false;
102 cur = FilePath(system_buffer);
103 break;
104 case base::DIR_START_MENU:
105 if (FAILED(SHGetFolderPath(NULL, CSIDL_PROGRAMS, NULL, SHGFP_TYPE_CURRENT,
106 system_buffer)))
107 return false;
108 cur = FilePath(system_buffer);
109 break;
110 case base::DIR_COMMON_STARTUP:
111 if (FAILED(SHGetFolderPath(nullptr, CSIDL_COMMON_STARTUP, nullptr,
112 SHGFP_TYPE_CURRENT, system_buffer)))
113 return false;
114 cur = FilePath(system_buffer);
115 break;
116 case base::DIR_USER_STARTUP:
117 if (FAILED(SHGetFolderPath(nullptr, CSIDL_STARTUP, nullptr,
118 SHGFP_TYPE_CURRENT, system_buffer)))
119 return false;
120 cur = FilePath(system_buffer);
121 break;
122 case base::DIR_ROAMING_APP_DATA:
123 if (FAILED(SHGetFolderPath(NULL, CSIDL_APPDATA, NULL, SHGFP_TYPE_CURRENT,
124 system_buffer)))
125 return false;
126 cur = FilePath(system_buffer);
127 break;
128 case base::DIR_COMMON_APP_DATA:
129 if (FAILED(SHGetFolderPath(NULL, CSIDL_COMMON_APPDATA, NULL,
130 SHGFP_TYPE_CURRENT, system_buffer)))
131 return false;
132 cur = FilePath(system_buffer);
133 break;
134 case base::DIR_LOCAL_APP_DATA:
135 if (FAILED(SHGetFolderPath(NULL, CSIDL_LOCAL_APPDATA, NULL,
136 SHGFP_TYPE_CURRENT, system_buffer)))
137 return false;
138 cur = FilePath(system_buffer);
139 break;
140 case base::DIR_SRC_TEST_DATA_ROOT: {
141 FilePath executableDir;
142 // On Windows, unit tests execute two levels deep from the source root.
143 // For example: chrome/{Debug|Release}/ui_tests.exe
144 PathService::Get(base::DIR_EXE, &executableDir);
145 cur = executableDir.DirName().DirName();
146 break;
147 }
148 case base::DIR_APP_SHORTCUTS: {
149 base::win::ScopedCoMem<wchar_t> path_buf;
150 if (FAILED(SHGetKnownFolderPath(FOLDERID_ApplicationShortcuts, 0, NULL,
151 &path_buf)))
152 return false;
153
154 cur = FilePath(path_buf.get());
155 break;
156 }
157 case base::DIR_USER_DESKTOP:
158 if (FAILED(SHGetFolderPath(NULL, CSIDL_DESKTOPDIRECTORY, NULL,
159 SHGFP_TYPE_CURRENT, system_buffer))) {
160 return false;
161 }
162 cur = FilePath(system_buffer);
163 break;
164 case base::DIR_COMMON_DESKTOP:
165 if (FAILED(SHGetFolderPath(NULL, CSIDL_COMMON_DESKTOPDIRECTORY, NULL,
166 SHGFP_TYPE_CURRENT, system_buffer))) {
167 return false;
168 }
169 cur = FilePath(system_buffer);
170 break;
171 case base::DIR_USER_QUICK_LAUNCH:
172 if (!PathService::Get(base::DIR_ROAMING_APP_DATA, &cur))
173 return false;
174 // According to various sources, appending
175 // "Microsoft\Internet Explorer\Quick Launch" to %appdata% is the only
176 // reliable way to get the quick launch folder across all versions of
177 // Windows.
178 // http://stackoverflow.com/questions/76080/how-do-you-reliably-get-the-quick-
179 // http://www.microsoft.com/technet/scriptcenter/resources/qanda/sept05/hey0901.mspx
180 cur = cur.Append(FILE_PATH_LITERAL("Microsoft"))
181 .Append(FILE_PATH_LITERAL("Internet Explorer"))
182 .Append(FILE_PATH_LITERAL("Quick Launch"));
183 break;
184 case base::DIR_TASKBAR_PINS: {
185 if (!PathService::Get(base::DIR_USER_QUICK_LAUNCH, &cur))
186 return false;
187 cur = cur.Append(FILE_PATH_LITERAL("User Pinned"))
188 .Append(FILE_PATH_LITERAL("TaskBar"));
189 break;
190 }
191 case base::DIR_IMPLICIT_APP_SHORTCUTS:
192 if (!PathService::Get(base::DIR_USER_QUICK_LAUNCH, &cur))
193 return false;
194 cur = cur.Append(FILE_PATH_LITERAL("User Pinned"))
195 .Append(FILE_PATH_LITERAL("ImplicitAppShortcuts"));
196 break;
197 case base::DIR_WINDOWS_FONTS:
198 if (FAILED(SHGetFolderPath(NULL, CSIDL_FONTS, NULL, SHGFP_TYPE_CURRENT,
199 system_buffer))) {
200 return false;
201 }
202 cur = FilePath(system_buffer);
203 break;
204 default:
205 return false;
206 }
207
208 *result = cur;
209 return true;
210 }
211
212 } // namespace base
213