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