// 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