123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214 |
- // Copyright (c) 2012 The Chromium Authors. All rights reserved.
- // Use of this source code is governed by a BSD-style license that can be
- // found in the LICENSE file.
- #include <windows.h>
- #include <KnownFolders.h>
- #include <shlobj.h>
- #include "base/base_paths.h"
- #include "base/environment.h"
- #include "base/files/file_path.h"
- #include "base/path_service.h"
- #include "base/strings/string_util.h"
- #include "base/strings/utf_string_conversions.h"
- #include "base/win/current_module.h"
- #include "base/win/scoped_co_mem.h"
- #include "base/win/windows_version.h"
- using base::FilePath;
- namespace base {
- bool PathProviderWin(int key, FilePath* result) {
- // We need to go compute the value. It would be nice to support paths with
- // names longer than MAX_PATH, but the system functions don't seem to be
- // designed for it either, with the exception of GetTempPath (but other
- // things will surely break if the temp path is too long, so we don't bother
- // handling it.
- wchar_t system_buffer[MAX_PATH];
- system_buffer[0] = 0;
- FilePath cur;
- switch (key) {
- case base::FILE_EXE:
- if (GetModuleFileName(NULL, system_buffer, MAX_PATH) == 0)
- return false;
- cur = FilePath(system_buffer);
- break;
- case base::FILE_MODULE: {
- // the resource containing module is assumed to be the one that
- // this code lives in, whether that's a dll or exe
- if (GetModuleFileName(CURRENT_MODULE(), system_buffer, MAX_PATH) == 0)
- return false;
- cur = FilePath(system_buffer);
- break;
- }
- case base::DIR_WINDOWS:
- GetWindowsDirectory(system_buffer, MAX_PATH);
- cur = FilePath(system_buffer);
- break;
- case base::DIR_SYSTEM:
- GetSystemDirectory(system_buffer, MAX_PATH);
- cur = FilePath(system_buffer);
- break;
- case base::DIR_PROGRAM_FILESX86:
- if (win::OSInfo::GetArchitecture() != win::OSInfo::X86_ARCHITECTURE) {
- if (FAILED(SHGetFolderPath(NULL, CSIDL_PROGRAM_FILESX86, NULL,
- SHGFP_TYPE_CURRENT, system_buffer)))
- return false;
- cur = FilePath(system_buffer);
- break;
- }
- // Fall through to base::DIR_PROGRAM_FILES if we're on an X86 machine.
- [[fallthrough]];
- case base::DIR_PROGRAM_FILES:
- if (FAILED(SHGetFolderPath(NULL, CSIDL_PROGRAM_FILES, NULL,
- SHGFP_TYPE_CURRENT, system_buffer)))
- return false;
- cur = FilePath(system_buffer);
- break;
- case base::DIR_PROGRAM_FILES6432:
- #if !defined(_WIN64)
- if (base::win::OSInfo::GetInstance()->IsWowX86OnAMD64()) {
- std::unique_ptr<base::Environment> env(base::Environment::Create());
- std::string programfiles_w6432;
- // 32-bit process running in WOW64 sets ProgramW6432 environment
- // variable. See
- // https://msdn.microsoft.com/library/windows/desktop/aa384274.aspx.
- if (!env->GetVar("ProgramW6432", &programfiles_w6432))
- return false;
- // GetVar returns UTF8 - convert back to Wide.
- cur = FilePath(UTF8ToWide(programfiles_w6432));
- break;
- }
- #endif
- if (FAILED(SHGetFolderPath(NULL, CSIDL_PROGRAM_FILES, NULL,
- SHGFP_TYPE_CURRENT, system_buffer)))
- return false;
- cur = FilePath(system_buffer);
- break;
- case base::DIR_IE_INTERNET_CACHE:
- if (FAILED(SHGetFolderPath(NULL, CSIDL_INTERNET_CACHE, NULL,
- SHGFP_TYPE_CURRENT, system_buffer)))
- return false;
- cur = FilePath(system_buffer);
- break;
- case base::DIR_COMMON_START_MENU:
- if (FAILED(SHGetFolderPath(NULL, CSIDL_COMMON_PROGRAMS, NULL,
- SHGFP_TYPE_CURRENT, system_buffer)))
- return false;
- cur = FilePath(system_buffer);
- break;
- case base::DIR_START_MENU:
- if (FAILED(SHGetFolderPath(NULL, CSIDL_PROGRAMS, NULL, SHGFP_TYPE_CURRENT,
- system_buffer)))
- return false;
- cur = FilePath(system_buffer);
- break;
- case base::DIR_COMMON_STARTUP:
- if (FAILED(SHGetFolderPath(nullptr, CSIDL_COMMON_STARTUP, nullptr,
- SHGFP_TYPE_CURRENT, system_buffer)))
- return false;
- cur = FilePath(system_buffer);
- break;
- case base::DIR_USER_STARTUP:
- if (FAILED(SHGetFolderPath(nullptr, CSIDL_STARTUP, nullptr,
- SHGFP_TYPE_CURRENT, system_buffer)))
- return false;
- cur = FilePath(system_buffer);
- break;
- case base::DIR_ROAMING_APP_DATA:
- if (FAILED(SHGetFolderPath(NULL, CSIDL_APPDATA, NULL, SHGFP_TYPE_CURRENT,
- system_buffer)))
- return false;
- cur = FilePath(system_buffer);
- break;
- case base::DIR_COMMON_APP_DATA:
- if (FAILED(SHGetFolderPath(NULL, CSIDL_COMMON_APPDATA, NULL,
- SHGFP_TYPE_CURRENT, system_buffer)))
- return false;
- cur = FilePath(system_buffer);
- break;
- case base::DIR_LOCAL_APP_DATA:
- if (FAILED(SHGetFolderPath(NULL, CSIDL_LOCAL_APPDATA, NULL,
- SHGFP_TYPE_CURRENT, system_buffer)))
- return false;
- cur = FilePath(system_buffer);
- break;
- case base::DIR_SRC_TEST_DATA_ROOT: {
- FilePath executableDir;
- // On Windows, unit tests execute two levels deep from the source root.
- // For example: chrome/{Debug|Release}/ui_tests.exe
- PathService::Get(base::DIR_EXE, &executableDir);
- cur = executableDir.DirName().DirName();
- break;
- }
- case base::DIR_APP_SHORTCUTS: {
- if (win::GetVersion() < win::Version::WIN8)
- return false;
- base::win::ScopedCoMem<wchar_t> path_buf;
- if (FAILED(SHGetKnownFolderPath(FOLDERID_ApplicationShortcuts, 0, NULL,
- &path_buf)))
- return false;
- cur = FilePath(path_buf.get());
- break;
- }
- case base::DIR_USER_DESKTOP:
- if (FAILED(SHGetFolderPath(NULL, CSIDL_DESKTOPDIRECTORY, NULL,
- SHGFP_TYPE_CURRENT, system_buffer))) {
- return false;
- }
- cur = FilePath(system_buffer);
- break;
- case base::DIR_COMMON_DESKTOP:
- if (FAILED(SHGetFolderPath(NULL, CSIDL_COMMON_DESKTOPDIRECTORY, NULL,
- SHGFP_TYPE_CURRENT, system_buffer))) {
- return false;
- }
- cur = FilePath(system_buffer);
- break;
- case base::DIR_USER_QUICK_LAUNCH:
- if (!PathService::Get(base::DIR_ROAMING_APP_DATA, &cur))
- return false;
- // According to various sources, appending
- // "Microsoft\Internet Explorer\Quick Launch" to %appdata% is the only
- // reliable way to get the quick launch folder across all versions of
- // Windows.
- // http://stackoverflow.com/questions/76080/how-do-you-reliably-get-the-quick-
- // http://www.microsoft.com/technet/scriptcenter/resources/qanda/sept05/hey0901.mspx
- cur = cur.Append(FILE_PATH_LITERAL("Microsoft"))
- .Append(FILE_PATH_LITERAL("Internet Explorer"))
- .Append(FILE_PATH_LITERAL("Quick Launch"));
- break;
- case base::DIR_TASKBAR_PINS: {
- if (!PathService::Get(base::DIR_USER_QUICK_LAUNCH, &cur))
- return false;
- cur = cur.Append(FILE_PATH_LITERAL("User Pinned"))
- .Append(FILE_PATH_LITERAL("TaskBar"));
- break;
- }
- case base::DIR_IMPLICIT_APP_SHORTCUTS:
- if (!PathService::Get(base::DIR_USER_QUICK_LAUNCH, &cur))
- return false;
- cur = cur.Append(FILE_PATH_LITERAL("User Pinned"))
- .Append(FILE_PATH_LITERAL("ImplicitAppShortcuts"));
- break;
- case base::DIR_WINDOWS_FONTS:
- if (FAILED(SHGetFolderPath(NULL, CSIDL_FONTS, NULL, SHGFP_TYPE_CURRENT,
- system_buffer))) {
- return false;
- }
- cur = FilePath(system_buffer);
- break;
- default:
- return false;
- }
- *result = cur;
- return true;
- }
- } // namespace base
|