1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253 |
- // Copyright 2019 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 "remoting/host/file_transfer/get_desktop_directory.h"
- #include <windows.h>
- #include <shlobj.h>
- #include "base/logging.h"
- #include "base/win/scoped_handle.h"
- namespace remoting {
- // We can't use PathService on Windows because it doesn't play nicely with
- // impersonation. Even if we disable PathService's own cache, the Windows API
- // itself does process-wide caching that can cause trouble on an impersonating
- // thread. As such, we have to call the relevant API directly and be explicit
- // about wanting impersonation handling.
- protocol::FileTransferResult<base::FilePath> GetDesktopDirectory() {
- // SHGetFolderPath on Windows 7 doesn't seem to like the pseudo handle
- // returned by GetCurrentThreadToken(), so call OpenThreadToken to get a real
- // handle.
- HANDLE user_token = nullptr;
- if (!OpenThreadToken(GetCurrentThread(),
- TOKEN_QUERY | TOKEN_IMPERSONATE | TOKEN_DUPLICATE, TRUE,
- &user_token)) {
- PLOG(ERROR) << "Failed to open thread token";
- return protocol::MakeFileTransferError(
- FROM_HERE, protocol::FileTransfer_Error_Type_UNEXPECTED_ERROR,
- GetLastError());
- }
- base::win::ScopedHandle scoped_user_token(user_token);
- wchar_t buffer[MAX_PATH];
- buffer[0] = 0;
- // While passing NULL for the third parameter would normally get the directory
- // for the current user, there are process-wide caches that can cause trouble
- // when impersonation is in play, so specify the token explicitly.
- HRESULT hr =
- SHGetFolderPath(NULL, CSIDL_DESKTOPDIRECTORY, scoped_user_token.Get(),
- SHGFP_TYPE_CURRENT, buffer);
- if (FAILED(hr)) {
- LOG(ERROR) << "Failed to get desktop directory: " << hr;
- return protocol::MakeFileTransferError(
- FROM_HERE, protocol::FileTransfer_Error_Type_UNEXPECTED_ERROR, hr);
- }
- return {kSuccessTag, buffer};
- }
- } // namespace remoting
|