1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
|
// Copyright 2019 The Chromium Authors
// 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/directory_helpers.h"
#include <windows.h>
#include <shlobj.h>
#include "base/check_is_test.h"
#include "base/logging.h"
#include "base/win/scoped_handle.h"
namespace remoting {
static base::FilePath* g_upload_directory_for_testing = nullptr;
// 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> GetFileUploadDirectory() {
if (g_upload_directory_for_testing) {
CHECK_IS_TEST();
return *g_upload_directory_for_testing;
}
// 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};
}
void SetFileUploadDirectoryForTesting(base::FilePath dir) {
if (g_upload_directory_for_testing) {
delete g_upload_directory_for_testing;
}
g_upload_directory_for_testing = new base::FilePath(dir);
}
} // namespace remoting
|
