[cdc_stream] Add a CLI client to start/stop asset streaming sessions (#4)

Implements the cdc_stream client and adjusts asset streaming in
various places to work better outside of a GGP environment.

This CL tries to get quoting for SSH commands right. It also brings
back the ability to start a streaming session from
asset_stream_manager.

Also cleans up Bazel targets setup. Since the sln file is now in root,
it is no longer necessary to prepend ../ to relative filenames to
make clicking on errors work.
This commit is contained in:
Lutz Justen
2022-11-18 10:59:42 +01:00
committed by GitHub
parent ca84d3dd2e
commit 269fb2be45
38 changed files with 797 additions and 356 deletions

View File

@@ -6,8 +6,6 @@
Usage: Define 4 properties in your project file: Usage: Define 4 properties in your project file:
BazelTargets: Labels of Bazel targets to build, e.g. //common:status. BazelTargets: Labels of Bazel targets to build, e.g. //common:status.
BazelOutputFile: Output filename, e.g. cdc_rsync.exe BazelOutputFile: Output filename, e.g. cdc_rsync.exe
BazelSourcePathPrefix: Prefix for source paths, to translate paths relative to (/foo) to project-relative paths (../foo).
Must be escaped for sed (e.g. / -> \/), e.g. ..\/..\/..\/
Optionally, define: Optionally, define:
BazelIncludePaths: Include paths, used for Intellisense. BazelIncludePaths: Include paths, used for Intellisense.
Import NMakeCMakeProject.targets. --> Import NMakeCMakeProject.targets. -->
@@ -16,7 +14,6 @@
<Target Name="CheckProperties" BeforeTargets="PrepareForNMakeBuild"> <Target Name="CheckProperties" BeforeTargets="PrepareForNMakeBuild">
<Error Condition="'$(BazelTargets)' == ''" Text="Please define property BazelTargets" /> <Error Condition="'$(BazelTargets)' == ''" Text="Please define property BazelTargets" />
<Error Condition="'$(BazelOutputFile)' == ''" Text="Please define property BazelOutputFile" /> <Error Condition="'$(BazelOutputFile)' == ''" Text="Please define property BazelOutputFile" />
<Error Condition="'$(BazelSourcePathPrefix)' == ''" Text="Please define property BazelSourcePathPrefix" />
</Target> </Target>
<!-- Define Bazel properties. --> <!-- Define Bazel properties. -->
@@ -27,15 +24,10 @@
<BazelPlatformDir Condition="'$(Platform)'=='GGP'">k8</BazelPlatformDir> <BazelPlatformDir Condition="'$(Platform)'=='GGP'">k8</BazelPlatformDir>
<BazelCompilationMode Condition="'$(Configuration)'=='Debug'">dbg</BazelCompilationMode> <BazelCompilationMode Condition="'$(Configuration)'=='Debug'">dbg</BazelCompilationMode>
<BazelCompilationMode Condition="'$(Configuration)'=='Release'">opt</BazelCompilationMode> <BazelCompilationMode Condition="'$(Configuration)'=='Release'">opt</BazelCompilationMode>
<!-- The sed command converts repo-relative paths (/path/to/foo) to project-relative paths (../path/to/foo), so that errors etc. can be clicked in VS.
Matches foo:31:8: bar, foo(31): bar, foo(31,32): bar -->
<BazelSedCommand>| sed -r &quot;s/^([^:\(]+[:\(][[:digit:]]+(,[[:digit:]]+)?[:\)])/$(BazelSourcePathPrefix)\\1/&quot;</BazelSedCommand>
<!-- Clang prints errors to stderr, so pipe it into stdout. -->
<BazelSedCommand Condition="'$(Platform)'=='GGP'">2&gt;&amp;1 $(BazelSedCommand)</BazelSedCommand>
<BazelArgs>--config=$(BazelPlatform)</BazelArgs> <BazelArgs>--config=$(BazelPlatform)</BazelArgs>
<BazelArgs Condition="'$(Configuration)|$(Platform)'=='Release|GGP'">$(BazelArgs) --linkopt=-Wl,--strip-all</BazelArgs> <BazelArgs Condition="'$(Configuration)|$(Platform)'=='Release|GGP'">$(BazelArgs) --linkopt=-Wl,--strip-all</BazelArgs>
<!-- Prevent protobuf recompilation (protobuf is written to "host" by default for both x84 and ggp, causing recompiles). --> <!-- Prevent protobuf recompilation (protobuf is written to "host" by default for both x84 and ggp, causing recompiles). -->
<BazelArgs Condition="'$(Platform)'=='x64'">$(BazelArgs) --distinct_host_configuration=false</BazelArgs> <!-- <BazelArgs Condition="'$(Platform)'=='x64'">$(BazelArgs) - -distinct_host_configuration=false</BazelArgs> -->
<!-- Windows uses /LTCG (link-time code generation), but no /GL (global optimization), which is a requirement for that. --> <!-- Windows uses /LTCG (link-time code generation), but no /GL (global optimization), which is a requirement for that. -->
<BazelArgs Condition="'$(Configuration)|$(Platform)'=='Release|x64'">$(BazelArgs) --copt=/GL</BazelArgs> <BazelArgs Condition="'$(Configuration)|$(Platform)'=='Release|x64'">$(BazelArgs) --copt=/GL</BazelArgs>
<!-- Strip unused symbols on GGP. --> <!-- Strip unused symbols on GGP. -->
@@ -51,9 +43,9 @@
<!-- Define NMake properties. --> <!-- Define NMake properties. -->
<PropertyGroup> <PropertyGroup>
<NMakeIncludeSearchPath>$(SolutionDir)..\..\bazel-out\$(BazelPlatformDir)-$(BazelCompilationMode)\bin;$(BazelIncludePaths)</NMakeIncludeSearchPath> <NMakeIncludeSearchPath>$(SolutionDir)..\..\bazel-out\$(BazelPlatformDir)-$(BazelCompilationMode)\bin;$(BazelIncludePaths)</NMakeIncludeSearchPath>
<NMakeBuildCommandLine>$(RmBazelOutDir) bazel build --compilation_mode=$(BazelCompilationMode) $(BazelArgs) $(BazelTargets) $(BazelSedCommand) $(MakeRW)</NMakeBuildCommandLine> <NMakeBuildCommandLine>$(RmBazelOutDir) bazel build --compilation_mode=$(BazelCompilationMode) $(BazelArgs) $(BazelTargets) $(MakeRW)</NMakeBuildCommandLine>
<NMakeCleanCommandLine>$(RmBazelOutDir) bazel clean</NMakeCleanCommandLine> <NMakeCleanCommandLine>$(RmBazelOutDir) bazel clean</NMakeCleanCommandLine>
<NMakeReBuildCommandLine>$(RmBazelOutDir) bazel clean &amp;&amp; bazel build --compilation_mode=$(BazelCompilationMode) $(BazelArgs) $(BazelTargets) $(BazelSedCommand) $(MakeRW)</NMakeReBuildCommandLine> <NMakeReBuildCommandLine>$(RmBazelOutDir) bazel clean &amp;&amp; bazel build --compilation_mode=$(BazelCompilationMode) $(BazelArgs) $(BazelTargets) $(MakeRW)</NMakeReBuildCommandLine>
<NMakeOutput>$(OutDir)$(BazelOutputFile)</NMakeOutput> <NMakeOutput>$(OutDir)$(BazelOutputFile)</NMakeOutput>
</PropertyGroup> </PropertyGroup>
</Project> </Project>

View File

@@ -28,6 +28,7 @@
<ClCompile Include="$(MSBuildThisFileDirectory)asset_stream_manager\multi_session_test.cc" /> <ClCompile Include="$(MSBuildThisFileDirectory)asset_stream_manager\multi_session_test.cc" />
<ClCompile Include="$(MSBuildThisFileDirectory)asset_stream_manager\session.cc" /> <ClCompile Include="$(MSBuildThisFileDirectory)asset_stream_manager\session.cc" />
<ClCompile Include="$(MSBuildThisFileDirectory)asset_stream_manager\session_management_server.cc" /> <ClCompile Include="$(MSBuildThisFileDirectory)asset_stream_manager\session_management_server.cc" />
<ClCompile Include="$(MSBuildThisFileDirectory)asset_stream_manager\session_manager.cc" />
<ClCompile Include="$(MSBuildThisFileDirectory)asset_stream_manager\testing_asset_stream_server.cc" /> <ClCompile Include="$(MSBuildThisFileDirectory)asset_stream_manager\testing_asset_stream_server.cc" />
<ClCompile Include="$(MSBuildThisFileDirectory)cdc_fuse_fs\asset.cc" /> <ClCompile Include="$(MSBuildThisFileDirectory)cdc_fuse_fs\asset.cc" />
<ClCompile Include="$(MSBuildThisFileDirectory)cdc_fuse_fs\asset_stream_client.cc" /> <ClCompile Include="$(MSBuildThisFileDirectory)cdc_fuse_fs\asset_stream_client.cc" />
@@ -39,6 +40,8 @@
<ClCompile Include="$(MSBuildThisFileDirectory)cdc_fuse_fs\mock_libfuse.cc" /> <ClCompile Include="$(MSBuildThisFileDirectory)cdc_fuse_fs\mock_libfuse.cc" />
<ClCompile Include="$(MSBuildThisFileDirectory)cdc_indexer\indexer.cc" /> <ClCompile Include="$(MSBuildThisFileDirectory)cdc_indexer\indexer.cc" />
<ClCompile Include="$(MSBuildThisFileDirectory)cdc_indexer\main.cc" /> <ClCompile Include="$(MSBuildThisFileDirectory)cdc_indexer\main.cc" />
<ClCompile Include="$(MSBuildThisFileDirectory)cdc_stream\local_assets_stream_manager_client.cc" />
<ClCompile Include="$(MSBuildThisFileDirectory)cdc_stream\main.cc" />
<ClCompile Include="$(MSBuildThisFileDirectory)common\buffer.cc" /> <ClCompile Include="$(MSBuildThisFileDirectory)common\buffer.cc" />
<ClCompile Include="$(MSBuildThisFileDirectory)common\buffer_test.cc" /> <ClCompile Include="$(MSBuildThisFileDirectory)common\buffer_test.cc" />
<ClCompile Include="$(MSBuildThisFileDirectory)common\clock.cc" /> <ClCompile Include="$(MSBuildThisFileDirectory)common\clock.cc" />
@@ -150,6 +153,7 @@
<ClInclude Include="$(MSBuildThisFileDirectory)asset_stream_manager\session.h" /> <ClInclude Include="$(MSBuildThisFileDirectory)asset_stream_manager\session.h" />
<ClInclude Include="$(MSBuildThisFileDirectory)asset_stream_manager\session_config.h" /> <ClInclude Include="$(MSBuildThisFileDirectory)asset_stream_manager\session_config.h" />
<ClInclude Include="$(MSBuildThisFileDirectory)asset_stream_manager\session_management_server.h" /> <ClInclude Include="$(MSBuildThisFileDirectory)asset_stream_manager\session_management_server.h" />
<ClInclude Include="$(MSBuildThisFileDirectory)asset_stream_manager\session_manager.h" />
<ClInclude Include="$(MSBuildThisFileDirectory)asset_stream_manager\testing_asset_stream_server.h" /> <ClInclude Include="$(MSBuildThisFileDirectory)asset_stream_manager\testing_asset_stream_server.h" />
<ClInclude Include="$(MSBuildThisFileDirectory)cdc_fuse_fs\asset.h" /> <ClInclude Include="$(MSBuildThisFileDirectory)cdc_fuse_fs\asset.h" />
<ClInclude Include="$(MSBuildThisFileDirectory)cdc_fuse_fs\asset_stream_client.h" /> <ClInclude Include="$(MSBuildThisFileDirectory)cdc_fuse_fs\asset_stream_client.h" />
@@ -158,6 +162,7 @@
<ClInclude Include="$(MSBuildThisFileDirectory)cdc_fuse_fs\constants.h" /> <ClInclude Include="$(MSBuildThisFileDirectory)cdc_fuse_fs\constants.h" />
<ClInclude Include="$(MSBuildThisFileDirectory)cdc_fuse_fs\mock_libfuse.h" /> <ClInclude Include="$(MSBuildThisFileDirectory)cdc_fuse_fs\mock_libfuse.h" />
<ClInclude Include="$(MSBuildThisFileDirectory)cdc_indexer\indexer.h" /> <ClInclude Include="$(MSBuildThisFileDirectory)cdc_indexer\indexer.h" />
<ClInclude Include="$(MSBuildThisFileDirectory)cdc_stream\local_assets_stream_manager_client.h" />
<ClInclude Include="$(MSBuildThisFileDirectory)common\buffer.h" /> <ClInclude Include="$(MSBuildThisFileDirectory)common\buffer.h" />
<ClInclude Include="$(MSBuildThisFileDirectory)common\clock.h" /> <ClInclude Include="$(MSBuildThisFileDirectory)common\clock.h" />
<ClInclude Include="$(MSBuildThisFileDirectory)common\dir_iter.h" /> <ClInclude Include="$(MSBuildThisFileDirectory)common\dir_iter.h" />
@@ -237,6 +242,7 @@
<None Include="$(MSBuildThisFileDirectory)cdc_indexer\BUILD" /> <None Include="$(MSBuildThisFileDirectory)cdc_indexer\BUILD" />
<None Include="$(MSBuildThisFileDirectory)cdc_indexer\README.md" /> <None Include="$(MSBuildThisFileDirectory)cdc_indexer\README.md" />
<None Include="$(MSBuildThisFileDirectory)cdc_rsync\protos\BUILD" /> <None Include="$(MSBuildThisFileDirectory)cdc_rsync\protos\BUILD" />
<None Include="$(MSBuildThisFileDirectory)cdc_stream\BUILD" />
<None Include="$(MSBuildThisFileDirectory)common\BUILD" /> <None Include="$(MSBuildThisFileDirectory)common\BUILD" />
<None Include="$(MSBuildThisFileDirectory)data_store\BUILD" /> <None Include="$(MSBuildThisFileDirectory)data_store\BUILD" />
<None Include="$(MSBuildThisFileDirectory)cdc_rsync\base\BUILD" /> <None Include="$(MSBuildThisFileDirectory)cdc_rsync\base\BUILD" />

View File

@@ -9,6 +9,7 @@ cc_binary(
deps = [ deps = [
":asset_stream_config", ":asset_stream_config",
":session_management_server", ":session_management_server",
"//cdc_stream",
"//common:log", "//common:log",
"//common:path", "//common:path",
"//common:sdk_util", "//common:sdk_util",

View File

@@ -26,9 +26,6 @@
#include "common/status_macros.h" #include "common/status_macros.h"
#include "json/json.h" #include "json/json.h"
ABSL_DECLARE_FLAG(std::string, src_dir);
ABSL_DECLARE_FLAG(std::string, instance_ip);
ABSL_DECLARE_FLAG(uint16_t, instance_port);
ABSL_DECLARE_FLAG(int, verbosity); ABSL_DECLARE_FLAG(int, verbosity);
ABSL_DECLARE_FLAG(bool, debug); ABSL_DECLARE_FLAG(bool, debug);
ABSL_DECLARE_FLAG(bool, singlethreaded); ABSL_DECLARE_FLAG(bool, singlethreaded);
@@ -42,6 +39,14 @@ ABSL_DECLARE_FLAG(uint32_t, access_idle_timeout);
ABSL_DECLARE_FLAG(int, manifest_updater_threads); ABSL_DECLARE_FLAG(int, manifest_updater_threads);
ABSL_DECLARE_FLAG(int, file_change_wait_duration_ms); ABSL_DECLARE_FLAG(int, file_change_wait_duration_ms);
// Development flags.
ABSL_DECLARE_FLAG(std::string, dev_src_dir);
ABSL_DECLARE_FLAG(std::string, dev_user_host);
ABSL_DECLARE_FLAG(uint16_t, dev_ssh_port);
ABSL_DECLARE_FLAG(std::string, dev_ssh_command);
ABSL_DECLARE_FLAG(std::string, dev_scp_command);
ABSL_DECLARE_FLAG(std::string, dev_mount_dir);
// Declare AS20 flags, so that AS30 can be used on older SDKs simply by // Declare AS20 flags, so that AS30 can be used on older SDKs simply by
// replacing the binary. Note that the RETIRED_FLAGS macro can't be used // replacing the binary. Note that the RETIRED_FLAGS macro can't be used
// because the flags contain dashes. This code mimics the macro. // because the flags contain dashes. This code mimics the macro.
@@ -62,9 +67,6 @@ const auto RETIRED_FLAGS_REG_allow_edge =
namespace cdc_ft { namespace cdc_ft {
AssetStreamConfig::AssetStreamConfig() { AssetStreamConfig::AssetStreamConfig() {
src_dir_ = absl::GetFlag(FLAGS_src_dir);
instance_ip_ = absl::GetFlag(FLAGS_instance_ip);
instance_port_ = absl::GetFlag(FLAGS_instance_port);
session_cfg_.verbosity = absl::GetFlag(FLAGS_verbosity); session_cfg_.verbosity = absl::GetFlag(FLAGS_verbosity);
session_cfg_.fuse_debug = absl::GetFlag(FLAGS_debug); session_cfg_.fuse_debug = absl::GetFlag(FLAGS_debug);
session_cfg_.fuse_singlethreaded = absl::GetFlag(FLAGS_singlethreaded); session_cfg_.fuse_singlethreaded = absl::GetFlag(FLAGS_singlethreaded);
@@ -80,6 +82,13 @@ AssetStreamConfig::AssetStreamConfig() {
absl::GetFlag(FLAGS_manifest_updater_threads); absl::GetFlag(FLAGS_manifest_updater_threads);
session_cfg_.file_change_wait_duration_ms = session_cfg_.file_change_wait_duration_ms =
absl::GetFlag(FLAGS_file_change_wait_duration_ms); absl::GetFlag(FLAGS_file_change_wait_duration_ms);
dev_src_dir_ = absl::GetFlag(FLAGS_dev_src_dir);
dev_target_.user_host = absl::GetFlag(FLAGS_dev_user_host);
dev_target_.ssh_port = absl::GetFlag(FLAGS_dev_ssh_port);
dev_target_.ssh_command = absl::GetFlag(FLAGS_dev_ssh_command);
dev_target_.scp_command = absl::GetFlag(FLAGS_dev_scp_command);
dev_target_.mount_dir = absl::GetFlag(FLAGS_dev_mount_dir);
} }
AssetStreamConfig::~AssetStreamConfig() = default; AssetStreamConfig::~AssetStreamConfig() = default;
@@ -105,7 +114,6 @@ absl::Status AssetStreamConfig::LoadFromFile(const std::string& path) {
} \ } \
} while (0) } while (0)
ASSIGN_VAR(src_dir_, src_dir, String);
ASSIGN_VAR(session_cfg_.verbosity, verbosity, Int); ASSIGN_VAR(session_cfg_.verbosity, verbosity, Int);
ASSIGN_VAR(session_cfg_.fuse_debug, debug, Bool); ASSIGN_VAR(session_cfg_.fuse_debug, debug, Bool);
ASSIGN_VAR(session_cfg_.fuse_singlethreaded, singlethreaded, Bool); ASSIGN_VAR(session_cfg_.fuse_singlethreaded, singlethreaded, Bool);
@@ -144,7 +152,6 @@ absl::Status AssetStreamConfig::LoadFromFile(const std::string& path) {
std::string AssetStreamConfig::ToString() { std::string AssetStreamConfig::ToString() {
std::ostringstream ss; std::ostringstream ss;
ss << "src_dir = " << src_dir_ << std::endl;
ss << "verbosity = " << session_cfg_.verbosity ss << "verbosity = " << session_cfg_.verbosity
<< std::endl; << std::endl;
ss << "debug = " << session_cfg_.fuse_debug ss << "debug = " << session_cfg_.fuse_debug
@@ -166,6 +173,14 @@ std::string AssetStreamConfig::ToString() {
<< session_cfg_.manifest_updater_threads << std::endl; << session_cfg_.manifest_updater_threads << std::endl;
ss << "file_change_wait_duration_ms = " ss << "file_change_wait_duration_ms = "
<< session_cfg_.file_change_wait_duration_ms << std::endl; << session_cfg_.file_change_wait_duration_ms << std::endl;
ss << "dev_src_dir = " << dev_src_dir_ << std::endl;
ss << "dev_user_host = " << dev_target_.user_host << std::endl;
ss << "dev_ssh_port = " << dev_target_.ssh_port << std::endl;
ss << "dev_ssh_command = " << dev_target_.ssh_command
<< std::endl;
ss << "dev_scp_command = " << dev_target_.scp_command
<< std::endl;
ss << "dev_mount_dir = " << dev_target_.mount_dir << std::endl;
return ss.str(); return ss.str();
} }

View File

@@ -23,6 +23,7 @@
#include "absl/status/status.h" #include "absl/status/status.h"
#include "asset_stream_manager/session_config.h" #include "asset_stream_manager/session_config.h"
#include "session.h"
namespace cdc_ft { namespace cdc_ft {
@@ -38,7 +39,6 @@ class AssetStreamConfig {
// Loads a configuration from the JSON file at |path| and overrides any config // Loads a configuration from the JSON file at |path| and overrides any config
// values that are set in this file. Sample json file: // values that are set in this file. Sample json file:
// { // {
// "src_dir":"C:\\path\\to\\assets",
// "verbosity":3, // "verbosity":3,
// "debug":0, // "debug":0,
// "singlethreaded":0, // "singlethreaded":0,
@@ -67,34 +67,29 @@ class AssetStreamConfig {
// read from the JSON file. // read from the JSON file.
std::string GetFlagReadErrors(); std::string GetFlagReadErrors();
// Workstation directory to stream. Should usually be empty since mounts are
// triggered by the CLI or the partner portal via a gRPC call, but useful
// during development.
const std::string& src_dir() const { return src_dir_; }
// IP address of the instance to stream to. Should usually be empty since
// mounts are triggered by the CLI or the partner portal via a gRPC call, but
// useful during development.
const std::string& instance_ip() const { return instance_ip_; }
// IP address of the instance to stream to. Should usually be unset (0) since
// mounts are triggered by the CLI or the partner portal via a gRPC call, but
// useful during development.
const uint16_t instance_port() const { return instance_port_; }
// Session configuration. // Session configuration.
const SessionConfig session_cfg() const { return session_cfg_; } const SessionConfig& session_cfg() const { return session_cfg_; }
// Workstation directory to be streamed. Used for development purposes only
// to start a session right away when the service starts up. See dev CLI args.
const std::string& dev_src_dir() const { return dev_src_dir_; }
// Session target. Used for development purposes only to start a session right
// away when the service starts up. See dev CLI args.
const SessionTarget& dev_target() const { return dev_target_; }
// Whether to log to a file or to stdout. // Whether to log to a file or to stdout.
bool log_to_stdout() const { return log_to_stdout_; } bool log_to_stdout() const { return log_to_stdout_; }
private: private:
std::string src_dir_;
std::string instance_ip_;
uint16_t instance_port_ = 0;
SessionConfig session_cfg_; SessionConfig session_cfg_;
bool log_to_stdout_ = false; bool log_to_stdout_ = false;
// Configuration used for development. Allows users to specify a session
// via the service's command line.
std::string dev_src_dir_;
SessionTarget dev_target_;
// Use a set, so the flags are sorted alphabetically. // Use a set, so the flags are sorted alphabetically.
std::set<std::string> flags_read_from_file_; std::set<std::string> flags_read_from_file_;

View File

@@ -66,10 +66,9 @@
</ItemDefinitionGroup> </ItemDefinitionGroup>
<!-- Bazel setup --> <!-- Bazel setup -->
<PropertyGroup> <PropertyGroup>
<BazelTargets>//asset_stream_manager</BazelTargets> <BazelTargets>//asset_stream_manager //cdc_stream</BazelTargets>
<BazelOutputFile>asset_stream_manager.exe</BazelOutputFile> <BazelOutputFile>asset_stream_manager.exe</BazelOutputFile>
<BazelIncludePaths>..\;..\third_party\absl;..\third_party\jsoncpp\include;..\third_party\blake3\c;..\third_party\googletest\googletest\include;..\third_party\protobuf\src;..\third_party\grpc\include;..\bazel-out\x64_windows-dbg\bin;$(VC_IncludePath);$(WindowsSDK_IncludePath)</BazelIncludePaths> <BazelIncludePaths>..\;..\third_party\absl;..\third_party\jsoncpp\include;..\third_party\blake3\c;..\third_party\googletest\googletest\include;..\third_party\protobuf\src;..\third_party\grpc\include;..\bazel-out\x64_windows-dbg\bin;$(VC_IncludePath);$(WindowsSDK_IncludePath)</BazelIncludePaths>
<BazelSourcePathPrefix>..\/</BazelSourcePathPrefix>
</PropertyGroup> </PropertyGroup>
<Import Project="..\NMakeBazelProject.targets" /> <Import Project="..\NMakeBazelProject.targets" />
<!-- For some reason, msbuild doesn't include this file, so copy it explicitly. --> <!-- For some reason, msbuild doesn't include this file, so copy it explicitly. -->

View File

@@ -29,13 +29,10 @@ namespace {
constexpr char kFuseFilename[] = "cdc_fuse_fs"; constexpr char kFuseFilename[] = "cdc_fuse_fs";
constexpr char kLibFuseFilename[] = "libfuse.so"; constexpr char kLibFuseFilename[] = "libfuse.so";
constexpr char kFuseStdoutPrefix[] = "cdc_fuse_fs_stdout"; constexpr char kFuseStdoutPrefix[] = "cdc_fuse_fs_stdout";
constexpr char kRemoteToolsBinDir[] = "~/.cache/cdc_file_transfer/"; constexpr char kRemoteToolsBinDir[] = "~/.cache/cdc-file-transfer/bin/";
// Mount point for FUSE on the gamelet.
constexpr char kMountDir[] = "/mnt/workstation";
// Cache directory on the gamelet to store data chunks. // Cache directory on the gamelet to store data chunks.
constexpr char kCacheDir[] = "/var/cache/asset_streaming"; constexpr char kCacheDir[] = "~/.cache/cdc-file-transfer/chunks";
} // namespace } // namespace
@@ -56,23 +53,14 @@ absl::Status CdcFuseManager::Deploy() {
std::string exe_dir; std::string exe_dir;
RETURN_IF_ERROR(path::GetExeDir(&exe_dir), "Failed to get exe directory"); RETURN_IF_ERROR(path::GetExeDir(&exe_dir), "Failed to get exe directory");
std::string local_exe_path = path::Join(exe_dir, kFuseFilename); // Set the cwd to the exe dir and pass the filenames to scp. Otherwise, some
std::string local_lib_path = path::Join(exe_dir, kLibFuseFilename); // scp implementations can get confused and create the wrong remote filenames.
path::SetCwd(exe_dir);
#ifdef _DEBUG // Copy FUSE to the gamelet.
// Sync FUSE to the gamelet in debug. Debug builds are rather large, so
// there's a gain from using sync.
LOG_DEBUG("Syncing FUSE");
RETURN_IF_ERROR(
remote_util_->Sync({local_exe_path, local_lib_path}, kRemoteToolsBinDir),
"Failed to sync FUSE to gamelet");
LOG_DEBUG("Syncing FUSE succeeded");
#else
// Copy FUSE to the gamelet. This is usually faster in production since it
// doesn't have to deploy ggp__server first.
LOG_DEBUG("Copying FUSE"); LOG_DEBUG("Copying FUSE");
RETURN_IF_ERROR(remote_util_->Scp({local_exe_path, local_lib_path}, RETURN_IF_ERROR(remote_util_->Scp({kFuseFilename, kLibFuseFilename},
kRemoteToolsBinDir, true), kRemoteToolsBinDir, /*compress=*/false),
"Failed to copy FUSE to gamelet"); "Failed to copy FUSE to gamelet");
LOG_DEBUG("Copying FUSE succeeded"); LOG_DEBUG("Copying FUSE succeeded");
@@ -82,12 +70,12 @@ absl::Status CdcFuseManager::Deploy() {
RETURN_IF_ERROR(remote_util_->Chmod("a+x", remotePath), RETURN_IF_ERROR(remote_util_->Chmod("a+x", remotePath),
"Failed to set executable flag on FUSE"); "Failed to set executable flag on FUSE");
LOG_DEBUG("Making FUSE succeeded"); LOG_DEBUG("Making FUSE succeeded");
#endif
return absl::OkStatus(); return absl::OkStatus();
} }
absl::Status CdcFuseManager::Start(uint16_t local_port, uint16_t remote_port, absl::Status CdcFuseManager::Start(const std::string& mount_dir,
uint16_t local_port, uint16_t remote_port,
int verbosity, bool debug, int verbosity, bool debug,
bool singlethreaded, bool enable_stats, bool singlethreaded, bool enable_stats,
bool check, uint64_t cache_capacity, bool check, uint64_t cache_capacity,
@@ -115,15 +103,18 @@ absl::Status CdcFuseManager::Start(uint16_t local_port, uint16_t remote_port,
// Build the remote command. // Build the remote command.
std::string remotePath = path::JoinUnix(kRemoteToolsBinDir, kFuseFilename); std::string remotePath = path::JoinUnix(kRemoteToolsBinDir, kFuseFilename);
std::string remote_command = absl::StrFormat( std::string remote_command = absl::StrFormat(
"LD_LIBRARY_PATH=%s %s --instance='%s' " "mkdir -p %s; LD_LIBRARY_PATH=%s %s "
"--components='%s' --port=%i --cache_dir=%s " "--instance=%s "
"--components=%s --port=%i --cache_dir=%s "
"--verbosity=%i --cleanup_timeout=%i --access_idle_timeout=%i --stats=%i " "--verbosity=%i --cleanup_timeout=%i --access_idle_timeout=%i --stats=%i "
"--check=%i --cache_capacity=%u -- -o allow_root -o ro -o nonempty -o " "--check=%i --cache_capacity=%u -- -o allow_root -o ro -o nonempty -o "
"auto_unmount %s%s%s", "auto_unmount %s%s%s",
kRemoteToolsBinDir, remotePath, instance_, component_args, remote_port, kRemoteToolsBinDir, kRemoteToolsBinDir, remotePath,
kCacheDir, verbosity, cleanup_timeout_sec, access_idle_timeout_sec, RemoteUtil::QuoteForSsh(instance_),
enable_stats, check, cache_capacity, kMountDir, debug ? " -d" : "", RemoteUtil::QuoteForSsh(component_args), remote_port, kCacheDir,
singlethreaded ? " -s" : ""); verbosity, cleanup_timeout_sec, access_idle_timeout_sec, enable_stats,
check, cache_capacity, RemoteUtil::QuoteForSsh(mount_dir),
debug ? " -d" : "", singlethreaded ? " -s" : "");
bool needs_deploy = false; bool needs_deploy = false;
RETURN_IF_ERROR( RETURN_IF_ERROR(

View File

@@ -40,6 +40,7 @@ class CdcFuseManager {
// |remote_port| to the workstation's |local_port|. Deploys the binary if // |remote_port| to the workstation's |local_port|. Deploys the binary if
// necessary. // necessary.
// //
// |mount_dir| is the remote directory where to mount the FUSE.
// |verbosity| is the log verbosity used by the filesystem. // |verbosity| is the log verbosity used by the filesystem.
// |debug| puts the filesystem into debug mode if set to true. This also // |debug| puts the filesystem into debug mode if set to true. This also
// causes the process to run in the foreground, so that logs are piped through // causes the process to run in the foreground, so that logs are piped through
@@ -51,10 +52,10 @@ class CdcFuseManager {
// |cleanup_timeout_sec| defines the data provider cleanup timeout in seconds. // |cleanup_timeout_sec| defines the data provider cleanup timeout in seconds.
// |access_idle_timeout_sec| defines the number of seconds after which data // |access_idle_timeout_sec| defines the number of seconds after which data
// provider is considered to be access-idling. // provider is considered to be access-idling.
absl::Status Start(uint16_t local_port, uint16_t remote_port, int verbosity, absl::Status Start(const std::string& mount_dir, uint16_t local_port,
bool debug, bool singlethreaded, bool enable_stats, uint16_t remote_port, int verbosity, bool debug,
bool check, uint64_t cache_capacity, bool singlethreaded, bool enable_stats, bool check,
uint32_t cleanup_timeout_sec, uint64_t cache_capacity, uint32_t cleanup_timeout_sec,
uint32_t access_idle_timeout_sec); uint32_t access_idle_timeout_sec);
// Stops the CDC FUSE. // Stops the CDC FUSE.

View File

@@ -17,6 +17,7 @@
#include <iomanip> #include <iomanip>
#include "absl/strings/str_format.h" #include "absl/strings/str_format.h"
#include "absl/strings/str_replace.h"
#include "absl/strings/str_split.h" #include "absl/strings/str_split.h"
#include "asset_stream_manager/multi_session.h" #include "asset_stream_manager/multi_session.h"
#include "asset_stream_manager/session_manager.h" #include "asset_stream_manager/session_manager.h"
@@ -26,11 +27,21 @@
#include "common/process.h" #include "common/process.h"
#include "common/sdk_util.h" #include "common/sdk_util.h"
#include "common/status.h" #include "common/status.h"
#include "google/protobuf/text_format.h"
#include "manifest/manifest_updater.h" #include "manifest/manifest_updater.h"
using TextFormat = google::protobuf::TextFormat;
namespace cdc_ft { namespace cdc_ft {
namespace { namespace {
std::string RequestToString(const google::protobuf::Message& request) {
std::string str;
google::protobuf::TextFormat::PrintToString(request, &str);
if (!str.empty() && str.back() == '\n') str.pop_back();
return absl::StrReplaceAll(str, {{"\n", ", "}});
}
// Parses |instance_name| of the form // Parses |instance_name| of the form
// "organizations/{org-id}/projects/{proj-id}/pools/{pool-id}/gamelets/{gamelet-id}" // "organizations/{org-id}/projects/{proj-id}/pools/{pool-id}/gamelets/{gamelet-id}"
// into parts. The pool id is not returned. // into parts. The pool id is not returned.
@@ -102,48 +113,18 @@ LocalAssetsStreamManagerServiceImpl::~LocalAssetsStreamManagerServiceImpl() =
grpc::Status LocalAssetsStreamManagerServiceImpl::StartSession( grpc::Status LocalAssetsStreamManagerServiceImpl::StartSession(
grpc::ServerContext* /*context*/, const StartSessionRequest* request, grpc::ServerContext* /*context*/, const StartSessionRequest* request,
StartSessionResponse* /*response*/) { StartSessionResponse* /*response*/) {
LOG_INFO("RPC:StartSession(gamelet_name='%s', workstation_directory='%s'", LOG_INFO("RPC:StartSession(%s)", RequestToString(*request));
request->gamelet_name(), request->workstation_directory());
metrics::DeveloperLogEvent evt;
evt.as_manager_data = std::make_unique<metrics::AssetStreamingManagerData>();
evt.as_manager_data->session_start_data =
std::make_unique<metrics::SessionStartData>();
evt.as_manager_data->session_start_data->absl_status = absl::StatusCode::kOk;
evt.as_manager_data->session_start_data->status =
metrics::SessionStartStatus::kOk;
evt.as_manager_data->session_start_data->origin =
ConvertOrigin(request->origin());
// Parse instance/project/org id.
absl::Status status;
MultiSession* ms = nullptr; MultiSession* ms = nullptr;
std::string instance_id, project_id, organization_id, instance_ip; metrics::DeveloperLogEvent evt;
uint16_t instance_port = 0; std::string instance_id;
if (!ParseInstanceName(request->gamelet_name(), &instance_id, &project_id, absl::Status status = StartSessionInternal(request, &instance_id, &ms, &evt);
&organization_id)) {
status = absl::InvalidArgumentError(absl::StrFormat(
"Failed to parse instance name '%s'", request->gamelet_name()));
} else {
evt.project_id = project_id;
evt.organization_id = organization_id;
status = InitSsh(instance_id, project_id, organization_id, &instance_ip,
&instance_port);
if (status.ok()) {
status = session_manager_->StartSession(
instance_id, project_id, organization_id, instance_ip, instance_port,
request->workstation_directory(), &ms,
&evt.as_manager_data->session_start_data->status);
}
}
evt.as_manager_data->session_start_data->absl_status = status.code(); evt.as_manager_data->session_start_data->absl_status = status.code();
if (ms) { if (ms) {
evt.as_manager_data->session_start_data->concurrent_session_count = evt.as_manager_data->session_start_data->concurrent_session_count =
ms->GetSessionCount(); ms->GetSessionCount();
if (!instance_id.empty() && ms->HasSessionForInstance(instance_id)) { if (!instance_id.empty() && ms->HasSession(instance_id)) {
ms->RecordSessionEvent(std::move(evt), metrics::EventType::kSessionStart, ms->RecordSessionEvent(std::move(evt), metrics::EventType::kSessionStart,
instance_id); instance_id);
} else { } else {
@@ -166,9 +147,14 @@ grpc::Status LocalAssetsStreamManagerServiceImpl::StartSession(
grpc::Status LocalAssetsStreamManagerServiceImpl::StopSession( grpc::Status LocalAssetsStreamManagerServiceImpl::StopSession(
grpc::ServerContext* /*context*/, const StopSessionRequest* request, grpc::ServerContext* /*context*/, const StopSessionRequest* request,
StopSessionResponse* /*response*/) { StopSessionResponse* /*response*/) {
LOG_INFO("RPC:StopSession(gamelet_id='%s')", request->gamelet_id()); LOG_INFO("RPC:StopSession(%s)", RequestToString(*request));
absl::Status status = session_manager_->StopSession(request->gamelet_id()); std::string instance_id =
!request->gamelet_id().empty() // Stadia use case
? request->gamelet_id()
: absl::StrCat(request->user_host(), ":", request->mount_dir());
absl::Status status = session_manager_->StopSession(instance_id);
if (status.ok()) { if (status.ok()) {
LOG_INFO("StopSession() succeeded"); LOG_INFO("StopSession() succeeded");
} else { } else {
@@ -177,6 +163,86 @@ grpc::Status LocalAssetsStreamManagerServiceImpl::StopSession(
return ToGrpcStatus(status); return ToGrpcStatus(status);
} }
absl::Status LocalAssetsStreamManagerServiceImpl::StartSessionInternal(
const StartSessionRequest* request, std::string* instance_id,
MultiSession** ms, metrics::DeveloperLogEvent* evt) {
instance_id->clear();
*ms = nullptr;
evt->as_manager_data = std::make_unique<metrics::AssetStreamingManagerData>();
evt->as_manager_data->session_start_data =
std::make_unique<metrics::SessionStartData>();
evt->as_manager_data->session_start_data->absl_status = absl::StatusCode::kOk;
evt->as_manager_data->session_start_data->status =
metrics::SessionStartStatus::kOk;
evt->as_manager_data->session_start_data->origin =
ConvertOrigin(request->origin());
if (!(request->gamelet_name().empty() ^ request->user_host().empty())) {
return absl::InvalidArgumentError(
"Must set either gamelet_name or user_host.");
}
if (request->mount_dir().empty()) {
return absl::InvalidArgumentError("mount_dir cannot be empty.");
}
SessionTarget target;
if (!request->gamelet_name().empty()) {
ASSIGN_OR_RETURN(target,
GetTargetForStadia(*request, instance_id, &evt->project_id,
&evt->organization_id));
} else {
target = GetTarget(*request, instance_id);
}
return session_manager_->StartSession(
*instance_id, request->workstation_directory(), target, evt->project_id,
evt->organization_id, ms,
&evt->as_manager_data->session_start_data->status);
}
absl::StatusOr<SessionTarget>
LocalAssetsStreamManagerServiceImpl::GetTargetForStadia(
const StartSessionRequest& request, std::string* instance_id,
std::string* project_id, std::string* organization_id) {
SessionTarget target;
target.mount_dir = request.mount_dir();
target.ssh_command = request.ssh_command();
target.scp_command = request.scp_command();
// Parse instance/project/org id.
if (!ParseInstanceName(request.gamelet_name(), instance_id, project_id,
organization_id)) {
return absl::InvalidArgumentError(absl::StrFormat(
"Failed to parse instance name '%s'", request.gamelet_name()));
}
// Run 'ggp ssh init' to determine IP (host) and port.
std::string instance_ip;
uint16_t instance_port = 0;
RETURN_IF_ERROR(InitSsh(*instance_id, *project_id, *organization_id,
&instance_ip, &instance_port));
target.user_host = "cloudcast@" + instance_ip;
target.ssh_port = instance_port;
return target;
}
SessionTarget LocalAssetsStreamManagerServiceImpl::GetTarget(
const StartSessionRequest& request, std::string* instance_id) {
SessionTarget target;
target.user_host = request.user_host();
target.mount_dir = request.mount_dir();
target.ssh_command = request.ssh_command();
target.scp_command = request.scp_command();
target.ssh_port = request.port() > 0 && request.port() <= UINT16_MAX
? static_cast<uint16_t>(request.port())
: RemoteUtil::kDefaultSshPort;
*instance_id = absl::StrCat(target.user_host, ":", target.mount_dir);
return target;
}
metrics::RequestOrigin LocalAssetsStreamManagerServiceImpl::ConvertOrigin( metrics::RequestOrigin LocalAssetsStreamManagerServiceImpl::ConvertOrigin(
StartSessionRequestOrigin origin) const { StartSessionRequestOrigin origin) const {
switch (origin) { switch (origin) {

View File

@@ -19,6 +19,7 @@
#include "absl/status/status.h" #include "absl/status/status.h"
#include "absl/status/statusor.h" #include "absl/status/statusor.h"
#include "asset_stream_manager/session.h"
#include "asset_stream_manager/session_config.h" #include "asset_stream_manager/session_config.h"
#include "metrics/metrics.h" #include "metrics/metrics.h"
#include "proto/local_assets_stream_manager.grpc.pb.h" #include "proto/local_assets_stream_manager.grpc.pb.h"
@@ -66,6 +67,27 @@ class LocalAssetsStreamManagerServiceImpl final
StopSessionResponse* response) override; StopSessionResponse* response) override;
private: private:
// Internal implementation of StartSession(). Returns the unique session
// identifier |instance_id|, the created or retrieved MultiSession |ms| as
// well as the filled metrics event |evt|.
absl::Status LocalAssetsStreamManagerServiceImpl::StartSessionInternal(
const StartSessionRequest* request, std::string* instance_id,
MultiSession** ms, metrics::DeveloperLogEvent* evt);
// Stadia-specific: Returns a SessionTarget from a gamelet name and fills in
// the gamelet's |instance_id|, |project_id| and |organization_id|.
// Used if request.gamelet_name() is set.
// Fails if the gamelet name fails to parse or if ggp ssh init fails.
absl::StatusOr<SessionTarget> GetTargetForStadia(
const StartSessionRequest& request, std::string* instance_id,
std::string* project_id, std::string* organization_id);
// Returns a SessionTarget from the corresponding fields in |request|.
// |instance_id| is set to [user@]host:mount_dir.
// Used if request.gamelet_name() is not set.
SessionTarget GetTarget(const StartSessionRequest& request,
std::string* instance_id);
// Convert StartSessionRequest enum to metrics enum. // Convert StartSessionRequest enum to metrics enum.
metrics::RequestOrigin ConvertOrigin(StartSessionRequestOrigin origin) const; metrics::RequestOrigin ConvertOrigin(StartSessionRequestOrigin origin) const;

View File

@@ -20,6 +20,7 @@
#include "asset_stream_manager/local_assets_stream_manager_service_impl.h" #include "asset_stream_manager/local_assets_stream_manager_service_impl.h"
#include "asset_stream_manager/session_management_server.h" #include "asset_stream_manager/session_management_server.h"
#include "asset_stream_manager/session_manager.h" #include "asset_stream_manager/session_manager.h"
#include "common/grpc_status.h"
#include "common/log.h" #include "common/log.h"
#include "common/path.h" #include "common/path.h"
#include "common/process.h" #include "common/process.h"
@@ -49,15 +50,19 @@ absl::Status Run(const AssetStreamConfig& cfg) {
background_service.SetExitCallback( background_service.SetExitCallback(
[&sm_server]() { return sm_server.Shutdown(); }); [&sm_server]() { return sm_server.Shutdown(); });
RETURN_IF_ERROR(sm_server.Start(kSessionManagementPort)); if (!cfg.dev_src_dir().empty()) {
if (!cfg.src_dir().empty()) { localassetsstreammanager::StartSessionRequest request;
MultiSession* ms_unused; request.set_workstation_directory(cfg.dev_src_dir());
metrics::SessionStartStatus status_unused; request.set_user_host(cfg.dev_target().user_host);
RETURN_IF_ERROR(session_manager.StartSession( request.set_mount_dir(cfg.dev_target().mount_dir);
/*instance_id=*/cfg.instance_ip(), /*project_id=*/std::string(), request.set_port(cfg.dev_target().ssh_port);
/*organization_id=*/std::string(), cfg.instance_ip(), request.set_ssh_command(cfg.dev_target().ssh_command);
cfg.instance_port(), cfg.src_dir(), &ms_unused, &status_unused)); request.set_scp_command(cfg.dev_target().scp_command);
localassetsstreammanager::StartSessionResponse response;
RETURN_ABSL_IF_ERROR(
session_service.StartSession(nullptr, &request, &response));
} }
RETURN_IF_ERROR(sm_server.Start(kSessionManagementPort));
sm_server.RunUntilShutdown(); sm_server.RunUntilShutdown();
return absl::OkStatus(); return absl::OkStatus();
} }
@@ -96,18 +101,6 @@ const auto RETIRED_FLAGS_REG_allow_edge =
} // namespace } // namespace
} // namespace cdc_ft } // namespace cdc_ft
ABSL_FLAG(std::string, src_dir, "",
"Start a streaming session immediately from the given Windows path. "
"Used during development. Must have exactly one gamelet reserved or "
"specify the target gamelet with --instance.");
ABSL_FLAG(std::string, instance_ip, "",
"Connect to the instance with the given IP address for this session. "
"This flag is ignored unless --src_dir is set as well. Used "
"during development. ");
ABSL_FLAG(uint16_t, instance_port, 0,
"Connect to the instance through the given SSH port. "
"This flag is ignored unless --src_dir is set as well. Used "
"during development. ");
ABSL_FLAG(int, verbosity, 2, "Verbosity of the log output"); ABSL_FLAG(int, verbosity, 2, "Verbosity of the log output");
ABSL_FLAG(bool, debug, false, "Run FUSE filesystem in debug mode"); ABSL_FLAG(bool, debug, false, "Run FUSE filesystem in debug mode");
ABSL_FLAG(bool, singlethreaded, false, ABSL_FLAG(bool, singlethreaded, false,
@@ -132,6 +125,28 @@ ABSL_FLAG(uint32_t, access_idle_timeout, cdc_ft::DataProvider::kAccessIdleSec,
"Do not run instance cache cleanups for this many seconds after the " "Do not run instance cache cleanups for this many seconds after the "
"last file access"); "last file access");
// Development args.
ABSL_FLAG(std::string, dev_src_dir, "",
"Start a streaming session immediately from the given Windows path. "
"Used during development. Must also specify --dev_user_host and "
"--dev_mount_dir and possibly other --dev flags, depending on the "
"SSH setup");
ABSL_FLAG(std::string, dev_user_host, "",
"Username and host to stream to, of the form [user@]host. Used "
"during development. See --dev_src_dir for more info.");
ABSL_FLAG(uint16_t, dev_ssh_port, cdc_ft::RemoteUtil::kDefaultSshPort,
"SSH port to use for the connection to the host. Used during "
"development. See --dev_src_dir for more info.");
ABSL_FLAG(std::string, dev_ssh_command, "",
"Ssh command and extra flags to use for the connection to the host. "
"Used during development. See --dev_src_dir for more info.");
ABSL_FLAG(std::string, dev_scp_command, "",
"Scp command and extra flags to use for the connection to the host. "
"Used during development. See --dev_src_dir for more info.");
ABSL_FLAG(std::string, dev_mount_dir, "",
"Directory on the host to stream to. Used during development. See "
"--dev_src_dir for more info.");
int main(int argc, char* argv[]) { int main(int argc, char* argv[]) {
absl::ParseCommandLine(argc, argv); absl::ParseCommandLine(argc, argv);

View File

@@ -478,7 +478,7 @@ absl::Status MultiSession::Shutdown() {
while (!sessions_.empty()) { while (!sessions_.empty()) {
std::string instance_id = sessions_.begin()->first; std::string instance_id = sessions_.begin()->first;
RETURN_IF_ERROR(StopSession(instance_id), RETURN_IF_ERROR(StopSession(instance_id),
"Failed to stop session for instance id %s", instance_id); "Failed to stop session for instance id '%s'", instance_id);
sessions_.erase(instance_id); sessions_.erase(instance_id);
} }
@@ -499,10 +499,9 @@ absl::Status MultiSession::Status() {
} }
absl::Status MultiSession::StartSession(const std::string& instance_id, absl::Status MultiSession::StartSession(const std::string& instance_id,
const SessionTarget& target,
const std::string& project_id, const std::string& project_id,
const std::string& organization_id, const std::string& organization_id) {
const std::string& instance_ip,
uint16_t instance_port) {
absl::MutexLock lock(&sessions_mutex_); absl::MutexLock lock(&sessions_mutex_);
if (sessions_.find(instance_id) != sessions_.end()) { if (sessions_.find(instance_id) != sessions_.end()) {
@@ -523,9 +522,8 @@ absl::Status MultiSession::StartSession(const std::string& instance_id,
metrics_recorder_->GetMetricsService(), metrics_recorder_->GetMetricsService(),
metrics_recorder_->MultiSessionId(), project_id, organization_id); metrics_recorder_->MultiSessionId(), project_id, organization_id);
auto session = auto session = std::make_unique<Session>(
std::make_unique<Session>(instance_id, instance_ip, instance_port, cfg_, instance_id, target, cfg_, process_factory_, std::move(metrics_recorder));
process_factory_, std::move(metrics_recorder));
RETURN_IF_ERROR(session->Start(local_asset_stream_port_, RETURN_IF_ERROR(session->Start(local_asset_stream_port_,
kAssetStreamPortFirst, kAssetStreamPortLast)); kAssetStreamPortFirst, kAssetStreamPortLast));
@@ -552,7 +550,7 @@ absl::Status MultiSession::StopSession(const std::string& instance_id) {
return absl::OkStatus(); return absl::OkStatus();
} }
bool MultiSession::HasSessionForInstance(const std::string& instance_id) { bool MultiSession::HasSession(const std::string& instance_id) {
absl::ReaderMutexLock lock(&sessions_mutex_); absl::ReaderMutexLock lock(&sessions_mutex_);
return sessions_.find(instance_id) != sessions_.end(); return sessions_.find(instance_id) != sessions_.end();
} }
@@ -663,19 +661,19 @@ void MultiSession::OnContentSent(size_t byte_count, size_t chunck_count,
std::string instance_id) { std::string instance_id) {
if (instance_id.empty()) { if (instance_id.empty()) {
// |instance_id| is empty only in case when manifest wasn't acknowledged by // |instance_id| is empty only in case when manifest wasn't acknowledged by
// the gamelet yet (ConfigStreamServiceImpl::AckManifestIdReceived was not // the instance yet (ConfigStreamServiceImpl::AckManifestIdReceived was not
// invoked). This means MultiSession::StartSession is still waiting for // invoked). This means MultiSession::StartSession is still waiting for
// manifest acknowledge and |sessions_mutex_| is currently locked. In this // manifest acknowledge and |sessions_mutex_| is currently locked. In this
// case invoking MultiSession::FindSession and waiting for |sessions_mutex_| // case invoking MultiSession::FindSession and waiting for |sessions_mutex_|
// to get unlocked will block the current thread, which is also responsible // to get unlocked will block the current thread, which is also responsible
// for receiving a call at ConfigStreamServiceImpl::AckManifestIdReceived. // for receiving a call at ConfigStreamServiceImpl::AckManifestIdReceived.
// This causes a deadlock and leads to a DeadlineExceeded error. // This causes a deadlock and leads to a DeadlineExceeded error.
LOG_WARNING("Can not record session content for an empty instance_id."); LOG_WARNING("Cannot record session content for an empty instance_id.");
return; return;
} }
Session* session = FindSession(instance_id); Session* session = FindSession(instance_id);
if (session == nullptr) { if (session == nullptr) {
LOG_WARNING("Failed to find active session by instrance id: %s", LOG_WARNING("Failed to find active session by instance id '%s'",
instance_id); instance_id);
return; return;
} }

View File

@@ -36,6 +36,7 @@ namespace cdc_ft {
class ProcessFactory; class ProcessFactory;
class Session; class Session;
struct SessionTarget;
using ManifestUpdatedCb = std::function<void()>; using ManifestUpdatedCb = std::function<void()>;
// Updates the manifest and runs a file watcher in a background thread. // Updates the manifest and runs a file watcher in a background thread.
@@ -65,7 +66,7 @@ class MultiSessionRunner {
// Stops updating the manifest and |server_|. // Stops updating the manifest and |server_|.
absl::Status Shutdown() ABSL_LOCKS_EXCLUDED(mutex_); absl::Status Shutdown() ABSL_LOCKS_EXCLUDED(mutex_);
// Waits until a manifest is ready and the gamelet |instance_id| has // Waits until a manifest is ready and the session for |instance_id| has
// acknowledged the reception of the currently set manifest id. |fuse_timeout| // acknowledged the reception of the currently set manifest id. |fuse_timeout|
// is the timeout for waiting for the FUSE manifest ack. The time required to // is the timeout for waiting for the FUSE manifest ack. The time required to
// generate the manifest is not part of this timeout as this could take a // generate the manifest is not part of this timeout as this could take a
@@ -172,30 +173,29 @@ class MultiSession {
// Not thread-safe. // Not thread-safe.
absl::Status Status(); absl::Status Status();
// Starts a new streaming session to the instance with given |instance_id| and // Starts a new streaming session to the instance described by |target| and
// waits until the FUSE has received the initial manifest id. // waits until the FUSE has received the initial manifest id.
// Returns an error if a session for that instance already exists. // Returns an error if a session for that instance already exists.
// |instance_id| is the instance id of the target remote instance. // |instance_id| is a unique id for the remote instance and mount directory,
// |project_id| is id of the project that contains the instance. // e.g. user@host:mount_dir.
// |organization_id| is id of the organization that contains the instance. // |target| identifies the remote target and how to connect to it.
// |instance_ip| is the IP address of the instance. // |project_id| is the project that owns the instance. Stadia only.
// |instance_port| is the SSH port for connecting to the remote instance. // |organization_id| is organization that contains the instance. Stadia only.
// Thread-safe. // Thread-safe.
absl::Status StartSession(const std::string& instance_id, absl::Status StartSession(const std::string& instance_id,
const SessionTarget& target,
const std::string& project_id, const std::string& project_id,
const std::string& organization_id, const std::string& organization_id)
const std::string& instance_ip,
uint16_t instance_port)
ABSL_LOCKS_EXCLUDED(sessions_mutex_); ABSL_LOCKS_EXCLUDED(sessions_mutex_);
// Starts a new streaming session to the gamelet with given |instance_id|. // Stops the session for the given |instance_id|.
// Returns a NotFound error if a session for that instance does not exists. // Returns a NotFound error if a session for that instance does not exists.
// Thread-safe. // Thread-safe.
absl::Status StopSession(const std::string& instance_id) absl::Status StopSession(const std::string& instance_id)
ABSL_LOCKS_EXCLUDED(sessions_mutex_); ABSL_LOCKS_EXCLUDED(sessions_mutex_);
// Returns true if there is an existing session for |instance_id|. // Returns true if there is an existing session for |instance_id|.
bool HasSessionForInstance(const std::string& instance_id) bool HasSession(const std::string& instance_id)
ABSL_LOCKS_EXCLUDED(sessions_mutex_); ABSL_LOCKS_EXCLUDED(sessions_mutex_);
// Returns true if the FUSE process is up and running for an existing session // Returns true if the FUSE process is up and running for an existing session
@@ -226,7 +226,7 @@ class MultiSession {
void RecordMultiSessionEvent(metrics::DeveloperLogEvent event, void RecordMultiSessionEvent(metrics::DeveloperLogEvent event,
metrics::EventType code); metrics::EventType code);
// Record an event for a session associated with the |instance|. // Record an event for a session associated with the |instance_id|.
void RecordSessionEvent(metrics::DeveloperLogEvent event, void RecordSessionEvent(metrics::DeveloperLogEvent event,
metrics::EventType code, metrics::EventType code,
const std::string& instance_id); const std::string& instance_id);

View File

@@ -26,7 +26,7 @@ namespace cdc_ft {
namespace { namespace {
// Timeout for initial gamelet connection. // Timeout for initial gamelet connection.
constexpr double kInstanceConnectionTimeoutSec = 10.0f; constexpr double kInstanceConnectionTimeoutSec = 60.0f;
metrics::DeveloperLogEvent GetEventWithHeartBeatData(size_t bytes, metrics::DeveloperLogEvent GetEventWithHeartBeatData(size_t bytes,
size_t chunks) { size_t chunks) {
@@ -40,18 +40,24 @@ metrics::DeveloperLogEvent GetEventWithHeartBeatData(size_t bytes,
} // namespace } // namespace
Session::Session(std::string instance_id, std::string instance_ip, Session::Session(std::string instance_id, const SessionTarget& target,
uint16_t instance_port, SessionConfig cfg, SessionConfig cfg, ProcessFactory* process_factory,
ProcessFactory* process_factory,
std::unique_ptr<SessionMetricsRecorder> metrics_recorder) std::unique_ptr<SessionMetricsRecorder> metrics_recorder)
: instance_id_(std::move(instance_id)), : instance_id_(std::move(instance_id)),
mount_dir_(target.mount_dir),
cfg_(std::move(cfg)), cfg_(std::move(cfg)),
process_factory_(process_factory), process_factory_(process_factory),
remote_util_(cfg_.verbosity, cfg_.quiet, process_factory, remote_util_(cfg_.verbosity, cfg_.quiet, process_factory,
/*forward_output_to_logging=*/true), /*forward_output_to_logging=*/true),
metrics_recorder_(std::move(metrics_recorder)) { metrics_recorder_(std::move(metrics_recorder)) {
assert(metrics_recorder_); assert(metrics_recorder_);
remote_util_.SetUserHostAndPort(instance_ip, instance_port); remote_util_.SetUserHostAndPort(target.user_host, target.ssh_port);
if (!target.ssh_command.empty()) {
remote_util_.SetSshCommand(target.ssh_command);
}
if (!target.scp_command.empty()) {
remote_util_.SetScpCommand(target.scp_command);
}
} }
Session::~Session() { Session::~Session() {
@@ -80,9 +86,10 @@ absl::Status Session::Start(int local_port, int first_remote_port,
fuse_ = std::make_unique<CdcFuseManager>(instance_id_, process_factory_, fuse_ = std::make_unique<CdcFuseManager>(instance_id_, process_factory_,
&remote_util_); &remote_util_);
RETURN_IF_ERROR( RETURN_IF_ERROR(
fuse_->Start(local_port, remote_port, cfg_.verbosity, cfg_.fuse_debug, fuse_->Start(mount_dir_, local_port, remote_port, cfg_.verbosity,
cfg_.fuse_singlethreaded, cfg_.stats, cfg_.fuse_check, cfg_.fuse_debug, cfg_.fuse_singlethreaded, cfg_.stats,
cfg_.fuse_cache_capacity, cfg_.fuse_cleanup_timeout_sec, cfg_.fuse_check, cfg_.fuse_cache_capacity,
cfg_.fuse_cleanup_timeout_sec,
cfg_.fuse_access_idle_timeout_sec), cfg_.fuse_access_idle_timeout_sec),
"Failed to start instance component"); "Failed to start instance component");
return absl::OkStatus(); return absl::OkStatus();

View File

@@ -32,17 +32,29 @@ class CdcFuseManager;
class ProcessFactory; class ProcessFactory;
class Process; class Process;
// Manages the connection of a workstation to a single gamelet. // Defines a remote target and how to connect to it.
struct SessionTarget {
// SSH username and hostname of the remote target, formed as [user@]host.
std::string user_host;
// Port to use for SSH connections to the remote target.
uint16_t ssh_port;
// Ssh command to use to connect to the remote target.
std::string ssh_command;
// Scp command to use to copy files to the remote target.
std::string scp_command;
// Directory on the remote target where to mount the streamed directory.
std::string mount_dir;
};
// Manages the connection of a workstation to a single remote instance.
class Session { class Session {
public: public:
// |instance_id| is a unique id for the remote instance. // |instance_id| is a unique id for the remote instance.
// |instance_ip| is the IP address of the remote instance. // |target| identifies the remote target and how to connect to it.
// |instance_port| is the SSH tunnel port for connecting to the instance.
// |cfg| contains generic configuration parameters for the session. // |cfg| contains generic configuration parameters for the session.
// |process_factory| abstracts process creation. // |process_factory| abstracts process creation.
Session(std::string instance_id, std::string instance_ip, Session(std::string instance_id, const SessionTarget& target,
uint16_t instance_port, SessionConfig cfg, SessionConfig cfg, ProcessFactory* process_factory,
ProcessFactory* process_factory,
std::unique_ptr<SessionMetricsRecorder> metrics_recorder); std::unique_ptr<SessionMetricsRecorder> metrics_recorder);
~Session(); ~Session();
@@ -71,6 +83,7 @@ class Session {
private: private:
const std::string instance_id_; const std::string instance_id_;
const std::string mount_dir_;
const SessionConfig cfg_; const SessionConfig cfg_;
ProcessFactory* const process_factory_; ProcessFactory* const process_factory_;

View File

@@ -59,10 +59,10 @@ absl::Status SessionManager::Shutdown() {
} }
absl::Status SessionManager::StartSession( absl::Status SessionManager::StartSession(
const std::string& instance_id, const std::string& project_id, const std::string& instance_id, const std::string& src_dir,
const std::string& organization_id, const std::string& instance_ip, const SessionTarget& target, const std::string& project_id,
uint16_t instance_port, const std::string& src_dir, const std::string& organization_id, MultiSession** multi_session,
MultiSession** multi_session, metrics::SessionStartStatus* metrics_status) { metrics::SessionStartStatus* metrics_status) {
*multi_session = nullptr; *multi_session = nullptr;
*metrics_status = metrics::SessionStartStatus::kOk; *metrics_status = metrics::SessionStartStatus::kOk;
@@ -83,7 +83,7 @@ absl::Status SessionManager::StartSession(
// Early out if we are streaming the workstation dir to the given gamelet. // Early out if we are streaming the workstation dir to the given gamelet.
MultiSession* ms = GetMultiSession(src_dir); MultiSession* ms = GetMultiSession(src_dir);
*multi_session = ms; *multi_session = ms;
if (ms && ms->HasSessionForInstance(instance_id)) { if (ms && ms->HasSession(instance_id)) {
if (ms->IsSessionHealthy(instance_id)) { if (ms->IsSessionHealthy(instance_id)) {
LOG_INFO("Reusing existing session"); LOG_INFO("Reusing existing session");
return absl::OkStatus(); return absl::OkStatus();
@@ -95,8 +95,8 @@ absl::Status SessionManager::StartSession(
// We could also fall through, but this might restart the MultiSession. // We could also fall through, but this might restart the MultiSession.
status = ms->StopSession(instance_id); status = ms->StopSession(instance_id);
if (status.ok()) { if (status.ok()) {
status = ms->StartSession(instance_id, project_id, organization_id, status =
instance_ip, instance_port); ms->StartSession(instance_id, target, project_id, organization_id);
} }
if (!status.ok()) { if (!status.ok()) {
*metrics_status = metrics::SessionStartStatus::kRestartSessionError; *metrics_status = metrics::SessionStartStatus::kRestartSessionError;
@@ -129,8 +129,7 @@ absl::Status SessionManager::StartSession(
// Start the session. // Start the session.
LOG_INFO("Starting streaming session from path '%s' to instance '%s'", LOG_INFO("Starting streaming session from path '%s' to instance '%s'",
src_dir, instance_id); src_dir, instance_id);
status = ms->StartSession(instance_id, project_id, organization_id, status = ms->StartSession(instance_id, target, project_id, organization_id);
instance_ip, instance_port);
if (!status.ok()) { if (!status.ok()) {
*metrics_status = metrics::SessionStartStatus::kStartSessionError; *metrics_status = metrics::SessionStartStatus::kStartSessionError;
} }
@@ -164,15 +163,16 @@ absl::StatusOr<MultiSession*> SessionManager::GetOrCreateMultiSession(
return iter->second.get(); return iter->second.get();
} }
absl::Status SessionManager::StopSessionInternal(const std::string& instance) { absl::Status SessionManager::StopSessionInternal(
const std::string& instance_id) {
absl::Status status; absl::Status status;
for (const auto& [key, ms] : sessions_) { for (const auto& [key, ms] : sessions_) {
if (!ms->HasSessionForInstance(instance)) continue; if (!ms->HasSession(instance_id)) continue;
LOG_INFO("Stopping session streaming from '%s' to instance '%s'", LOG_INFO("Stopping session streaming from '%s' to instance '%s'",
ms->src_dir(), instance); ms->src_dir(), instance_id);
RETURN_IF_ERROR(ms->StopSession(instance), RETURN_IF_ERROR(ms->StopSession(instance_id),
"Failed to stop session for instance '%s'", instance); "Failed to stop session for instance '%s'", instance_id);
// Session was stopped. If the MultiSession is empty now, delete it. // Session was stopped. If the MultiSession is empty now, delete it.
if (ms->Empty()) { if (ms->Empty()) {
@@ -187,7 +187,7 @@ absl::Status SessionManager::StopSessionInternal(const std::string& instance) {
} }
return absl::NotFoundError( return absl::NotFoundError(
absl::StrFormat("No session for instance id '%s' found", instance)); absl::StrFormat("No session for instance '%s' found", instance_id));
} }
} // namespace cdc_ft } // namespace cdc_ft

View File

@@ -30,43 +30,46 @@ namespace cdc_ft {
class MultiSession; class MultiSession;
class ProcessFactory; class ProcessFactory;
struct SessionTarget;
// Implements a service to start and stop streaming sessions as a server. // Adds logic around MultiSession to start and stop streaming sessions. Makes
// The corresponding clients are implemented by the ggp CLI and SDK Proxy. // sure that some invariants are maintained, like no two streaming sessions
// The CLI triggers StartSession() from `ggp instance mount --local-dir` and // exist to the same target user@host:dir.
// StopSession() from `ggp instance unmount`. SDK Proxy invokes StartSession()
// when a user starts a new game from the partner portal and sets an `Asset
// streaming directory` in the `Advanced settings` in the `Play settings`
// dialog.
// This service is owned by SessionManagementServer.
class SessionManager { class SessionManager {
public: public:
SessionManager(SessionConfig cfg, ProcessFactory* process_factory, SessionManager(SessionConfig cfg, ProcessFactory* process_factory,
metrics::MetricsService* metrics_service); metrics::MetricsService* metrics_service);
~SessionManager(); ~SessionManager();
// Starts a session and populates |multi_session| and |metrics_status|. // Starts a new session or reuses an existing one.
// |instance_id| is a unique id for the remote instance and mount directory,
// e.g. user@host:mount_dir.
// |src_dir| is the local directory to stream.
// |target| identifies the remote target and how to connect to it.
// |project_id| is the project that owns the instance. Stadia only.
// |organization_id| is organization that contains the instance. Stadia only.
// Populates |multi_session| and |metrics_status| on success.
absl::Status StartSession(const std::string& instance_id, absl::Status StartSession(const std::string& instance_id,
const std::string& src_dir,
const SessionTarget& target,
const std::string& project_id, const std::string& project_id,
const std::string& organization_id, const std::string& organization_id,
const std::string& instance_ip,
uint16_t instance_port, const std::string& src_dir,
MultiSession** multi_session, MultiSession** multi_session,
metrics::SessionStartStatus* metrics_status) metrics::SessionStartStatus* metrics_status)
ABSL_LOCKS_EXCLUDED(sessions_mutex_); ABSL_LOCKS_EXCLUDED(sessions_mutex_);
// Stops the session for the given |instance|. Returns a NotFound error if no // Stops the session for the given |instance_id|.
// session exists. // Returns a NotFound error if no session exists.
absl::Status StopSession(const std::string& instance) absl::Status StopSession(const std::string& instance_id)
ABSL_LOCKS_EXCLUDED(sessions_mutex_); ABSL_LOCKS_EXCLUDED(sessions_mutex_);
// Shuts down all existing MultiSessions. // Shuts down all existing MultiSessions.
absl::Status Shutdown() ABSL_LOCKS_EXCLUDED(sessions_mutex_); absl::Status Shutdown() ABSL_LOCKS_EXCLUDED(sessions_mutex_);
private: private:
// Stops the session for the given |instance|. Returns a NotFound error if no // Stops the session for the given |instance_id|. Returns a NotFound error if
// session exists. // no session exists.
absl::Status StopSessionInternal(const std::string& instance) absl::Status StopSessionInternal(const std::string& instance_id)
ABSL_EXCLUSIVE_LOCKS_REQUIRED(sessions_mutex_); ABSL_EXCLUSIVE_LOCKS_REQUIRED(sessions_mutex_);
// Returns the MultiSession for the given workstation directory |src_dir| or // Returns the MultiSession for the given workstation directory |src_dir| or

View File

@@ -52,7 +52,6 @@
<BazelTargets>//cdc_fuse_fs</BazelTargets> <BazelTargets>//cdc_fuse_fs</BazelTargets>
<BazelOutputFile>cdc_fuse_fs</BazelOutputFile> <BazelOutputFile>cdc_fuse_fs</BazelOutputFile>
<BazelIncludePaths>..\;..\third_party\absl;..\third_party\blake3\c;..\third_party\googletest\googletest\include;..\third_party\protobuf\src;..\third_party\grpc\include;$(AdditionalIncludeDirectories)</BazelIncludePaths> <BazelIncludePaths>..\;..\third_party\absl;..\third_party\blake3\c;..\third_party\googletest\googletest\include;..\third_party\protobuf\src;..\third_party\grpc\include;$(AdditionalIncludeDirectories)</BazelIncludePaths>
<BazelSourcePathPrefix>..\/</BazelSourcePathPrefix>
</PropertyGroup> </PropertyGroup>
<Import Project="..\NMakeBazelProject.targets" /> <Import Project="..\NMakeBazelProject.targets" />
<Import Project="$(VCTargetsPath)\Microsoft.Cpp.targets" /> <Import Project="$(VCTargetsPath)\Microsoft.Cpp.targets" />

View File

@@ -69,16 +69,15 @@
<BazelTargets>//cdc_rsync</BazelTargets> <BazelTargets>//cdc_rsync</BazelTargets>
<BazelOutputFile>cdc_rsync.exe</BazelOutputFile> <BazelOutputFile>cdc_rsync.exe</BazelOutputFile>
<BazelIncludePaths>..\;..\third_party\absl;..\third_party\blake3\c;..\bazel-stadia-file-transfer\external\com_github_zstd\lib;..\third_party\googletest\googletest\include;..\third_party\protobuf\src;$(VC_IncludePath);$(WindowsSDK_IncludePath)</BazelIncludePaths> <BazelIncludePaths>..\;..\third_party\absl;..\third_party\blake3\c;..\bazel-stadia-file-transfer\external\com_github_zstd\lib;..\third_party\googletest\googletest\include;..\third_party\protobuf\src;$(VC_IncludePath);$(WindowsSDK_IncludePath)</BazelIncludePaths>
<BazelSourcePathPrefix>..\/</BazelSourcePathPrefix>
</PropertyGroup> </PropertyGroup>
<Import Project="..\NMakeBazelProject.targets" /> <Import Project="..\NMakeBazelProject.targets" />
<!-- For some reason, msbuild doesn't include this file, so copy it explicitly. --> <!-- For some reason, msbuild doesn't include this file, so copy it explicitly. -->
<!-- TODO: Reenable once we can cross-compile these. <!-- TODO: Reenable once we can cross-compile these.
<PropertyGroup> <PropertyGroup>
<GgpRsyncServerFile>$(SolutionDir)bazel-out\k8-$(BazelCompilationMode)\bin\cdc_rsync_server\cdc_rsync_server</GgpRsyncServerFile> <CdcRsyncServerFile>$(SolutionDir)bazel-out\k8-$(BazelCompilationMode)\bin\cdc_rsync_server\cdc_rsync_server</CdcRsyncServerFile>
</PropertyGroup> </PropertyGroup>
<Target Name="CopyServer" Inputs="$(GgpRsyncServerFile)" Outputs="$(OutDir)cdc_rsync_server" AfterTargets="Build"> <Target Name="CopyServer" Inputs="$(CdcRsyncServerFile)" Outputs="$(OutDir)cdc_rsync_server" AfterTargets="Build">
<Copy SourceFiles="$(GgpRsyncServerFile)" DestinationFiles="$(OutDir)cdc_rsync_server" /> <Copy SourceFiles="$(CdcRsyncServerFile)" DestinationFiles="$(OutDir)cdc_rsync_server" />
</Target> </Target>
--> -->
<Import Project="$(VCTargetsPath)\Microsoft.Cpp.targets" /> <Import Project="$(VCTargetsPath)\Microsoft.Cpp.targets" />

View File

@@ -46,8 +46,8 @@ constexpr int kExitCodeNotFound = 127;
constexpr int kForwardPortFirst = 44450; constexpr int kForwardPortFirst = 44450;
constexpr int kForwardPortLast = 44459; constexpr int kForwardPortLast = 44459;
constexpr char kGgpServerFilename[] = "cdc_rsync_server"; constexpr char kCdcServerFilename[] = "cdc_rsync_server";
constexpr char kRemoteToolsBinDir[] = "~/.cache/cdc_file_transfer/"; constexpr char kRemoteToolsBinDir[] = "~/.cache/cdc-file-transfer/bin/";
SetOptionsRequest::FilterRule::Type ToProtoType(PathFilter::Rule::Type type) { SetOptionsRequest::FilterRule::Type ToProtoType(PathFilter::Rule::Type type) {
switch (type) { switch (type) {
@@ -178,7 +178,7 @@ absl::Status CdcRsyncClient::StartServer() {
std::vector<GameletComponent> components; std::vector<GameletComponent> components;
status = GameletComponent::Get( status = GameletComponent::Get(
{path::Join(component_dir, kGgpServerFilename)}, &components); {path::Join(component_dir, kCdcServerFilename)}, &components);
if (!status.ok()) { if (!status.ok()) {
return MakeStatus( return MakeStatus(
"Required instance component not found. Make sure the file " "Required instance component not found. Make sure the file "
@@ -202,7 +202,7 @@ absl::Status CdcRsyncClient::StartServer() {
int port = *port_res; int port = *port_res;
std::string remote_server_path = std::string remote_server_path =
std::string(kRemoteToolsBinDir) + kGgpServerFilename; std::string(kRemoteToolsBinDir) + kCdcServerFilename;
// Test existence manually to prevent misleading bash output message // Test existence manually to prevent misleading bash output message
// "bash: .../cdc_rsync_server: No such file or directory". // "bash: .../cdc_rsync_server: No such file or directory".
// Also create the bin dir because otherwise scp below might fail. // Also create the bin dir because otherwise scp below might fail.
@@ -411,9 +411,9 @@ absl::Status CdcRsyncClient::DeployServer() {
// scp cdc_rsync_server to a temp location on the gamelet. // scp cdc_rsync_server to a temp location on the gamelet.
std::string remoteServerTmpPath = std::string remoteServerTmpPath =
absl::StrFormat("%s%s.%s", kRemoteToolsBinDir, kGgpServerFilename, absl::StrFormat("%s%s.%s", kRemoteToolsBinDir, kCdcServerFilename,
Util::GenerateUniqueId()); Util::GenerateUniqueId());
std::string localServerPath = path::Join(exe_dir, kGgpServerFilename); std::string localServerPath = path::Join(exe_dir, kCdcServerFilename);
status = remote_util_.Scp({localServerPath}, remoteServerTmpPath, status = remote_util_.Scp({localServerPath}, remoteServerTmpPath,
/*compress=*/true); /*compress=*/true);
if (!status.ok()) { if (!status.ok()) {
@@ -424,9 +424,9 @@ absl::Status CdcRsyncClient::DeployServer() {
// - Make the old cdc_rsync_server writable (if it exists). // - Make the old cdc_rsync_server writable (if it exists).
// - Make the new cdc_rsync_server executable. // - Make the new cdc_rsync_server executable.
// - Replace the old cdc_rsync_server by the new one. // - Replace the old cdc_rsync_server by the new one.
std::string old_path = RemoteUtil::EscapeForWindows( std::string old_path = RemoteUtil::QuoteForWindows(
std::string(kRemoteToolsBinDir) + kGgpServerFilename); std::string(kRemoteToolsBinDir) + kCdcServerFilename);
std::string new_path = RemoteUtil::EscapeForWindows(remoteServerTmpPath); std::string new_path = RemoteUtil::QuoteForWindows(remoteServerTmpPath);
std::string replace_cmd = absl::StrFormat( std::string replace_cmd = absl::StrFormat(
" ([ ! -f %s ] || chmod u+w %s) && chmod a+x %s && mv %s %s", old_path, " ([ ! -f %s ] || chmod u+w %s) && chmod a+x %s && mv %s %s", old_path,
old_path, new_path, new_path, old_path); old_path, new_path, new_path, old_path);

View File

@@ -73,10 +73,10 @@ Options:
-R, --relative Use relative path names -R, --relative Use relative path names
--existing Skip creating new files on instance --existing Skip creating new files on instance
--copy-dest dir Use files from dir as sync base if files are missing --copy-dest dir Use files from dir as sync base if files are missing
--ssh-command Path and arguments of SSH command to use, e.g. --ssh-command Path and arguments of ssh command to use, e.g.
C:\path\to\ssh.exe -F config -i id_rsa -oStrictHostKeyChecking=yes -oUserKnownHostsFile="""known_hosts""" C:\path\to\ssh.exe -F config -i id_rsa -oStrictHostKeyChecking=yes -oUserKnownHostsFile="""known_hosts"""
Can also be specified by the CDC_SSH_COMMAND environment variable. Can also be specified by the CDC_SSH_COMMAND environment variable.
--scp-command Path and arguments of SSH command to use, e.g. --scp-command Path and arguments of scp command to use, e.g.
C:\path\to\scp.exe -F config -i id_rsa -oStrictHostKeyChecking=yes -oUserKnownHostsFile="""known_hosts""" C:\path\to\scp.exe -F config -i id_rsa -oStrictHostKeyChecking=yes -oUserKnownHostsFile="""known_hosts"""
Can also be specified by the CDC_SCP_COMMAND environment variable. Can also be specified by the CDC_SCP_COMMAND environment variable.
-h --help Help for cdc_rsync -h --help Help for cdc_rsync

View File

@@ -50,7 +50,6 @@
<BazelTargets>//cdc_rsync_server:cdc_rsync_server</BazelTargets> <BazelTargets>//cdc_rsync_server:cdc_rsync_server</BazelTargets>
<BazelOutputFile>cdc_rsync_server</BazelOutputFile> <BazelOutputFile>cdc_rsync_server</BazelOutputFile>
<BazelIncludePaths>..\;..\third_party\absl;..\third_party\blake3\c;..\bazel-stadia-file-transfer\external\com_github_zstd\lib;..\third_party\googletest\googletest\include;..\third_party\protobuf\src</BazelIncludePaths> <BazelIncludePaths>..\;..\third_party\absl;..\third_party\blake3\c;..\bazel-stadia-file-transfer\external\com_github_zstd\lib;..\third_party\googletest\googletest\include;..\third_party\protobuf\src</BazelIncludePaths>
<BazelSourcePathPrefix>..\/</BazelSourcePathPrefix>
</PropertyGroup> </PropertyGroup>
<Import Project="..\NMakeBazelProject.targets" /> <Import Project="..\NMakeBazelProject.targets" />
<Import Project="$(VCTargetsPath)\Microsoft.Cpp.targets" /> <Import Project="$(VCTargetsPath)\Microsoft.Cpp.targets" />

3
cdc_stream/.gitignore vendored Normal file
View File

@@ -0,0 +1,3 @@
x64/*
*.log
*.user

26
cdc_stream/BUILD Normal file
View File

@@ -0,0 +1,26 @@
package(default_visibility = [
"//:__subpackages__",
])
cc_binary(
name = "cdc_stream",
srcs = ["main.cc"],
deps = [
":local_assets_stream_manager_client",
"//common:log",
"//common:path",
"@com_google_absl//absl/flags:parse",
"@com_google_absl//absl/status",
],
)
cc_library(
name = "local_assets_stream_manager_client",
srcs = ["local_assets_stream_manager_client.cc"],
hdrs = ["local_assets_stream_manager_client.h"],
deps = [
"//common:grpc_status",
"//proto:local_assets_stream_manager_grpc_proto",
"@com_google_absl//absl/status",
],
)

View File

@@ -0,0 +1,62 @@
// Copyright 2022 Google LLC
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
#include "cdc_stream/local_assets_stream_manager_client.h"
#include "absl/status/status.h"
#include "common/grpc_status.h"
namespace cdc_ft {
using StartSessionRequest = localassetsstreammanager::StartSessionRequest;
using StartSessionResponse = localassetsstreammanager::StartSessionResponse;
using StopSessionRequest = localassetsstreammanager::StopSessionRequest;
using StopSessionResponse = localassetsstreammanager::StopSessionResponse;
LocalAssetsStreamManagerClient::LocalAssetsStreamManagerClient(
std::shared_ptr<grpc::Channel> channel) {
stub_ = LocalAssetsStreamManager::NewStub(std::move(channel));
}
LocalAssetsStreamManagerClient::~LocalAssetsStreamManagerClient() = default;
absl::Status LocalAssetsStreamManagerClient::StartSession(
const std::string& src_dir, const std::string& user_host, uint16_t ssh_port,
const std::string& mount_dir, const std::string& ssh_command,
const std::string& scp_command) {
StartSessionRequest request;
request.set_workstation_directory(src_dir);
request.set_user_host(user_host);
request.set_port(ssh_port);
request.set_mount_dir(mount_dir);
request.set_ssh_command(ssh_command);
request.set_scp_command(scp_command);
grpc::ClientContext context;
StartSessionResponse response;
return ToAbslStatus(stub_->StartSession(&context, request, &response));
}
absl::Status LocalAssetsStreamManagerClient::StopSession(
const std::string& user_host, const std::string& mount_dir) {
StopSessionRequest request;
request.set_user_host(user_host);
request.set_mount_dir(mount_dir);
grpc::ClientContext context;
StopSessionResponse response;
return ToAbslStatus(stub_->StopSession(&context, request, &response));
}
} // namespace cdc_ft

View File

@@ -0,0 +1,63 @@
/*
* Copyright 2022 Google LLC
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#ifndef ASSET_STREAM_MANAGER_LOCAL_ASSETS_STREAM_MANAGER_CLIENT_H_
#define ASSET_STREAM_MANAGER_LOCAL_ASSETS_STREAM_MANAGER_CLIENT_H_
#include <memory>
#include "absl/status/status.h"
#include "grpcpp/channel.h"
#include "proto/local_assets_stream_manager.grpc.pb.h"
namespace grpc_impl {
class Channel;
}
namespace cdc_ft {
// gRpc client for starting/stopping asset streaming sessions.
class LocalAssetsStreamManagerClient {
public:
// |channel| is a grpc channel to use.
explicit LocalAssetsStreamManagerClient(
std::shared_ptr<grpc::Channel> channel);
~LocalAssetsStreamManagerClient();
// Starts streaming the Windows directory |src_dir| to the Linux target
// |user_host_dir|, which must be formatted as [user@]host:dir, e.g.
// jdoe@jdoe.corp.foo.com:~/assets
// Starting a second session to the same target will stop the first one.
absl::Status StartSession(const std::string& src_dir,
const std::string& user_host, uint16_t ssh_port,
const std::string& mount_dir,
const std::string& ssh_command,
const std::string& scp_command);
// Stops the streaming session to the Linux target |user_host_dir|, which must
// be formatted as [user@]host:dir, e.g. jdoe@jdoe.corp.foo.com:~/assets
absl::Status StopSession(const std::string& user_host_dir,
const std::string& mount_dir);
private:
using LocalAssetsStreamManager =
localassetsstreammanager::LocalAssetsStreamManager;
std::unique_ptr<LocalAssetsStreamManager::Stub> stub_;
};
} // namespace cdc_ft
#endif // ASSET_STREAM_MANAGER_LOCAL_ASSETS_STREAM_MANAGER_CLIENT_H_

158
cdc_stream/main.cc Normal file
View File

@@ -0,0 +1,158 @@
// Copyright 2022 Google LLC
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
#include <memory>
#include <vector>
#include "absl/flags/flag.h"
#include "absl/flags/parse.h"
#include "absl/flags/usage.h"
#include "absl/status/status.h"
#include "absl/strings/str_split.h"
#include "cdc_stream/local_assets_stream_manager_client.h"
#include "common/log.h"
#include "common/path.h"
#include "grpcpp/channel.h"
#include "grpcpp/security/credentials.h"
ABSL_FLAG(uint16_t, service_port, 44432,
"Port to use while connecting to the local asset stream service.");
ABSL_FLAG(
uint16_t, ssh_port, 22,
"Port to use while connecting to the remote instance being streamed to.");
ABSL_FLAG(std::string, ssh_command, "",
"Path and arguments of ssh command to use, e.g. "
"\"C:\\path\\to\\ssh.exe -F config_file\". Can also be specified by "
"the CDC_SSH_COMMAND environment variable.");
ABSL_FLAG(std::string, scp_command, "",
"Path and arguments of scp command to use, e.g. "
"\"C:\\path\\to\\scp.exe -F config_file\". Can also be specified by "
"the CDC_SCP_COMMAND environment variable.");
ABSL_FLAG(int, verbosity, 2,
"Verbosity of the log output. Increase to make logs more verbose.");
namespace {
constexpr char kHelpText[] = R"!(Stream files from a Windows to a Linux device
Usage: cdc_stream [flags] start windows_dir [user@]host:linux_dir
Streams the Windows directory windows_dir to directory linux_dir on the
Linux host using SSH username user.
cdc_stream [flags] stop [user@]host:linux_dir
Stops the streaming session to the given Linux target.
Type cdc_stream --helpfull for available flags.)!";
// Splits |user_host_dir| = [user@]host:dir up into [user@]host and dir.
// Does not touch Windows drives, e.g. C:\foo.
bool ParseUserHostDir(const std::string& user_host_dir, std::string* user_host,
std::string* dir) {
std::vector<std::string> parts =
absl::StrSplit(user_host_dir, absl::MaxSplits(':', 1));
if (parts.size() < 2 ||
(parts[0].size() == 1 && toupper(parts[0][0]) >= 'A' &&
toupper(parts[0][0]) <= 'Z')) {
LOG_ERROR(
"Failed to parse '%s'. Make sure it is of the form "
"[user@]host:linux_dir.",
user_host_dir);
return false;
}
*user_host = parts[0];
*dir = parts[1];
return true;
}
std::string GetFlagFromEnvOrArgs(const char* env,
const absl::Flag<std::string>& flag) {
std::string value = absl::GetFlag(flag);
if (value.empty()) {
cdc_ft::path::GetEnv(env, &value).IgnoreError();
}
return value;
}
} // namespace
int main(int argc, char* argv[]) {
absl::SetProgramUsageMessage(kHelpText);
std::vector<char*> args = absl::ParseCommandLine(argc, argv);
uint16_t service_port = absl::GetFlag(FLAGS_service_port);
int verbosity = absl::GetFlag(FLAGS_verbosity);
std::string ssh_command =
GetFlagFromEnvOrArgs("CDC_SSH_COMMAND", FLAGS_ssh_command);
std::string scp_command =
GetFlagFromEnvOrArgs("CDC_SCP_COMMAND", FLAGS_scp_command);
uint16_t ssh_port = absl::GetFlag(FLAGS_ssh_port);
cdc_ft::LogLevel level = cdc_ft::Log::VerbosityToLogLevel(verbosity);
cdc_ft::Log::Initialize(std::make_unique<cdc_ft::ConsoleLog>(level));
if (args.size() < 2) {
LOG_INFO(kHelpText);
return 0;
}
std::string command = args[1];
if (command != "start" && command != "stop") {
LOG_ERROR("Unknown command '%s'. Must be 'start' or 'stop'.", command);
return 1;
}
// Start a gRpc client.
std::string client_address = absl::StrFormat("localhost:%u", service_port);
std::shared_ptr<grpc::Channel> grpc_channel = grpc::CreateCustomChannel(
client_address, grpc::InsecureChannelCredentials(),
grpc::ChannelArguments());
cdc_ft::LocalAssetsStreamManagerClient client(grpc_channel);
absl::Status status;
if (command == "start") {
if (args.size() < 4) {
LOG_ERROR(
"Command 'start' needs 2 arguments: the Windows directory to stream"
" and the Linux target [user@]host:dir.");
return 1;
}
std::string src_dir = args[2];
std::string user_host, mount_dir;
if (!ParseUserHostDir(args[3], &user_host, &mount_dir)) return 1;
status = client.StartSession(src_dir, user_host, ssh_port, mount_dir,
ssh_command, scp_command);
} else /* if (command == "stop") */ {
if (args.size() < 3) {
LOG_ERROR(
"Command 'stop' needs 1 argument: the Linux target "
"[user@]host:linux_dir.");
return 1;
}
std::string user_host, mount_dir;
if (!ParseUserHostDir(args[2], &user_host, &mount_dir)) return 1;
status = client.StopSession(user_host, mount_dir);
}
if (!status.ok()) {
LOG_ERROR("Error: %s", status.ToString());
}
cdc_ft::Log::Shutdown();
return static_cast<int>(status.code());
}

View File

@@ -326,6 +326,10 @@ std::string GetDrivePrefix(const std::string& path) {
std::string GetCwd() { return std::filesystem::current_path().u8string(); } std::string GetCwd() { return std::filesystem::current_path().u8string(); }
void SetCwd(const std::string& path) {
std::filesystem::current_path(std::filesystem::u8path(path));
}
std::string GetFullPath(const std::string& path) { std::string GetFullPath(const std::string& path) {
if (path.empty()) { if (path.empty()) {
return std::string(); return std::string();

View File

@@ -130,6 +130,9 @@ std::string GetDrivePrefix(const std::string& path);
// Gets the current working directory. // Gets the current working directory.
std::string GetCwd(); std::string GetCwd();
// Sets the current working directory.
void SetCwd(const std::string& path);
// Expands a relative path to an absolute path (relative to the current working // Expands a relative path to an absolute path (relative to the current working
// directory). Also canonicalizes the path, removing any . and .. elements. // directory). Also canonicalizes the path, removing any . and .. elements.
// Note that if the path does not exist or contains invalid characters, it may // Note that if the path does not exist or contains invalid characters, it may

View File

@@ -16,6 +16,7 @@
#include <regex> #include <regex>
#include "absl/strings/match.h"
#include "absl/strings/str_cat.h" #include "absl/strings/str_cat.h"
#include "absl/strings/str_format.h" #include "absl/strings/str_format.h"
#include "common/path.h" #include "common/path.h"
@@ -46,6 +47,7 @@ void RemoteUtil::SetUserHostAndPort(std::string user_host, int port) {
user_host_ = std::move(user_host); user_host_ = std::move(user_host);
ssh_port_ = port; ssh_port_ = port;
} }
void RemoteUtil::SetScpCommand(std::string scp_command) { void RemoteUtil::SetScpCommand(std::string scp_command) {
scp_command_ = std::move(scp_command); scp_command_ = std::move(scp_command);
} }
@@ -56,7 +58,7 @@ void RemoteUtil::SetSshCommand(std::string ssh_command) {
absl::Status RemoteUtil::Scp(std::vector<std::string> source_filepaths, absl::Status RemoteUtil::Scp(std::vector<std::string> source_filepaths,
const std::string& dest, bool compress) { const std::string& dest, bool compress) {
absl::Status status = CheckHostPort(); absl::Status status = CheckUserHostPort();
if (!status.ok()) { if (!status.ok()) {
return status; return status;
} }
@@ -64,7 +66,11 @@ absl::Status RemoteUtil::Scp(std::vector<std::string> source_filepaths,
std::string source_args; std::string source_args;
for (const std::string& sourceFilePath : source_filepaths) { for (const std::string& sourceFilePath : source_filepaths) {
// Workaround for scp thinking that C is a host in C:\path\to\foo. // Workaround for scp thinking that C is a host in C:\path\to\foo.
source_args += QuoteArgument("//./" + sourceFilePath) + " "; if (absl::StrContains(path::GetDrivePrefix(sourceFilePath), ":")) {
source_args += QuoteForWindows("//./" + sourceFilePath) + " ";
} else {
source_args += QuoteForWindows(sourceFilePath) + " ";
}
} }
// -p preserves timestamps. This enables timestamp-based up-to-date checks. // -p preserves timestamps. This enables timestamp-based up-to-date checks.
@@ -73,66 +79,27 @@ absl::Status RemoteUtil::Scp(std::vector<std::string> source_filepaths,
"%s " "%s "
"%s %s -p -T " "%s %s -p -T "
"-P %i %s " "-P %i %s "
"%s", "%s:%s",
scp_command_, quiet_ || verbosity_ < 2 ? "-q" : "", compress ? "-C" : "", QuoteForWindows(scp_command_), quiet_ || verbosity_ < 2 ? "-q" : "",
ssh_port_, source_args, QuoteArgument(user_host_ + ":" + dest)); compress ? "-C" : "", ssh_port_, source_args, QuoteForWindows(user_host_),
QuoteForWindows(dest));
start_info.name = "scp"; start_info.name = "scp";
start_info.forward_output_to_log = forward_output_to_log_; start_info.forward_output_to_log = forward_output_to_log_;
return process_factory_->Run(start_info); return process_factory_->Run(start_info);
} }
absl::Status RemoteUtil::Sync(std::vector<std::string> source_filepaths,
const std::string& dest) {
absl::Status status = CheckHostPort();
if (!status.ok()) {
return status;
}
std::string source_args;
for (const std::string& sourceFilePath : source_filepaths) {
source_args += QuoteArgument(sourceFilePath) + " ";
}
ProcessStartInfo start_info;
start_info.command = absl::StrFormat(
"cdc_rsync --ip=%s --port=%i -z "
"%s %s%s",
QuoteArgument(user_host_), ssh_port_,
quiet_ || verbosity_ < 2 ? "-q " : " ", source_args, QuoteArgument(dest));
start_info.name = "cdc_rsync";
start_info.forward_output_to_log = forward_output_to_log_;
return process_factory_->Run(start_info);
}
absl::Status RemoteUtil::Chmod(const std::string& mode, absl::Status RemoteUtil::Chmod(const std::string& mode,
const std::string& remote_path, bool quiet) { const std::string& remote_path, bool quiet) {
std::string remote_command = std::string remote_command =
absl::StrFormat("chmod %s %s %s", QuoteArgument(mode), absl::StrFormat("chmod %s %s %s", QuoteForSsh(mode),
EscapeForWindows(remote_path), quiet ? "-f" : ""); QuoteForSsh(remote_path), quiet ? "-f" : "");
return Run(remote_command, "chmod"); return Run(remote_command, "chmod");
} }
absl::Status RemoteUtil::Rm(const std::string& remote_path, bool force) {
std::string remote_command = absl::StrFormat("rm %s %s", force ? "-f" : "",
EscapeForWindows(remote_path));
return Run(remote_command, "rm");
}
absl::Status RemoteUtil::Mv(const std::string& old_remote_path,
const std::string& new_remote_path) {
std::string remote_command =
absl::StrFormat("mv %s %s", EscapeForWindows(old_remote_path),
EscapeForWindows(new_remote_path));
return Run(remote_command, "mv");
}
absl::Status RemoteUtil::Run(std::string remote_command, std::string name) { absl::Status RemoteUtil::Run(std::string remote_command, std::string name) {
absl::Status status = CheckHostPort(); absl::Status status = CheckUserHostPort();
if (!status.ok()) { if (!status.ok()) {
return status; return status;
} }
@@ -177,33 +144,61 @@ ProcessStartInfo RemoteUtil::BuildProcessStartInfoForSshInternal(
"-oServerAliveCountMax=6 " // Number of lost msgs before ssh terminates "-oServerAliveCountMax=6 " // Number of lost msgs before ssh terminates
"-oServerAliveInterval=5 " // Time interval between alive msgs "-oServerAliveInterval=5 " // Time interval between alive msgs
"%s %s -p %i %s", "%s %s -p %i %s",
ssh_command_, quiet_ || verbosity_ < 2 ? "-q" : "", forward_arg, QuoteForWindows(ssh_command_), quiet_ || verbosity_ < 2 ? "-q" : "",
QuoteArgument(user_host_), ssh_port_, remote_command_arg); forward_arg, QuoteForWindows(user_host_), ssh_port_, remote_command_arg);
start_info.forward_output_to_log = forward_output_to_log_; start_info.forward_output_to_log = forward_output_to_log_;
return start_info; return start_info;
} }
std::string RemoteUtil::EscapeForWindows(const std::string& argument) { std::string RemoteUtil::QuoteForWindows(const std::string& argument) {
std::string str = // Escape certain backslashes (see doc of this function).
std::string escaped =
std::regex_replace(argument, std::regex(R"(\\*(?="|$))"), "$&$&"); std::regex_replace(argument, std::regex(R"(\\*(?="|$))"), "$&$&");
return std::regex_replace(str, std::regex(R"(")"), R"(\")"); // Escape " -> \".
escaped = std::regex_replace(escaped, std::regex(R"(")"), R"(\")");
// Quote.
return absl::StrCat("\"", escaped, "\"");
} }
std::string RemoteUtil::QuoteArgument(const std::string& argument) { std::string RemoteUtil::QuoteForSsh(const std::string& argument) {
return absl::StrCat("\"", EscapeForWindows(argument), "\""); // Escape \ ->: \\.
std::string escaped =
std::regex_replace(argument, std::regex(R"(\\)"), R"(\\)");
// Escape " -> \".
escaped = std::regex_replace(escaped, std::regex(R"(")"), R"(\")");
// Quote, but handle special case for ~.
if (escaped.empty() || escaped[0] != '~') {
return QuoteForWindows(absl::StrCat("\"", escaped, "\""));
}
// Simple special cases. Quote() isn't required, but called for consistency.
if (escaped == "~" || escaped == "~/") {
return QuoteForWindows(escaped);
}
// Check whether the username contains only valid characters.
// E.g. ~user name/foo -> Quote(~user name/foo)
size_t slash_pos = escaped.find('/');
size_t username_end_pos =
slash_pos == std::string::npos ? escaped.size() : slash_pos;
if (username_end_pos > 1 &&
!std::regex_match(escaped.substr(1, username_end_pos - 1),
std::regex("^[a-z][-a-z0-9]*"))) {
return QuoteForWindows(absl::StrCat("\"", escaped, "\""));
}
if (slash_pos == std::string::npos) {
// E.g. ~username -> Quote(~username)
return QuoteForWindows(escaped);
}
// E.g. or ~username/foo -> Quote(~username/"foo")
return QuoteForWindows(absl::StrCat(escaped.substr(0, slash_pos + 1), "\"",
escaped.substr(slash_pos + 1), "\""));
} }
std::string RemoteUtil::QuoteArgumentForSsh(const std::string& argument) { absl::Status RemoteUtil::CheckUserHostPort() {
return absl::StrFormat(
"'%s'", std::regex_replace(argument, std::regex("'"), "'\\''"));
}
std::string RemoteUtil::QuoteAndEscapeArgumentForSsh(
const std::string& argument) {
return EscapeForWindows(QuoteArgumentForSsh(argument));
}
absl::Status RemoteUtil::CheckHostPort() {
if (user_host_.empty() || ssh_port_ == 0) { if (user_host_.empty() || ssh_port_ == 0) {
return MakeStatus("IP or port not set"); return MakeStatus("IP or port not set");
} }

View File

@@ -61,25 +61,11 @@ class RemoteUtil {
absl::Status Scp(std::vector<std::string> source_filepaths, absl::Status Scp(std::vector<std::string> source_filepaths,
const std::string& dest, bool compress); const std::string& dest, bool compress);
// Syncs |source_filepaths| to the remote folder |dest| on the gamelet using
// cdc_rsync. Must call SetUserHostAndPort before calling this method.
absl::Status Sync(std::vector<std::string> source_filepaths,
const std::string& dest);
// Calls 'chmod |mode| |remote_path|' on the gamelet. // Calls 'chmod |mode| |remote_path|' on the gamelet.
// Must call SetUserHostAndPort before calling this method. // Must call SetUserHostAndPort before calling this method.
absl::Status Chmod(const std::string& mode, const std::string& remote_path, absl::Status Chmod(const std::string& mode, const std::string& remote_path,
bool quiet = false); bool quiet = false);
// Calls 'rm [-f] |remote_path|' on the gamelet.
// Must call SetUserHostAndPort before calling this method.
absl::Status Rm(const std::string& remote_path, bool force);
// Calls `mv |old_remote_path| |new_remote_path| on the gamelet.
// Must call SetUserHostAndPort before calling this method.
absl::Status Mv(const std::string& old_remote_path,
const std::string& new_remote_path);
// Runs |remote_command| on the gamelet. The command must be properly escaped. // Runs |remote_command| on the gamelet. The command must be properly escaped.
// |name| is the name of the command displayed in the logs. // |name| is the name of the command displayed in the logs.
// Must call SetUserHostAndPort before calling this method. // Must call SetUserHostAndPort before calling this method.
@@ -107,28 +93,32 @@ class RemoteUtil {
// Returns whether output is suppressed. // Returns whether output is suppressed.
bool Quiet() const { return quiet_; } bool Quiet() const { return quiet_; }
// Escapes command line argument for the Microsoft command line parser in
// preparation for quoting. Double quotes are backslash-escaped. One or more
// backslashes are backslash-escaped if they are followed by a double quote,
// or if they occur at the end of the string, e.g.
// foo\bar -> foo\bar, foo\ -> foo\\, foo\\"bar -> foo\\\\\"bar.
static std::string EscapeForWindows(const std::string& argument);
// Quotes and escapes a command line argument following the convention // Quotes and escapes a command line argument following the convention
// understood by the Microsoft command line parser. // understood by the Microsoft command line parser.
static std::string QuoteArgument(const std::string& argument); // Double quotes are backslash-escaped. One or more backslashes are backslash-
// escaped if they are followed by a double quote, or if they occur at the end
// Quotes and escapes a command line argument for usage in SSH. // of the string, e.g.
static std::string QuoteArgumentForSsh(const std::string& argument); // foo -> "foo"
// foo\bar -> "foo\bar"
// foo\ -> "foo\\"
// foo\\"bar -> "foo\\\\\"bar".
static std::string QuoteForWindows(const std::string& argument);
// Quotes and escapes a command line arguments for use in SSH command. The // Quotes and escapes a command line arguments for use in SSH command. The
// argument is first escaped and quoted for Linux using single quotes and then // argument is first escaped and quoted for Linux using double quotes and then
// it is escaped to be used by the Microsoft command line parser. // it is escaped to be used by the Microsoft command line parser.
static std::string QuoteAndEscapeArgumentForSsh(const std::string& argument); // Properly supports path starting with ~ and ~username.
// foo -> "\"foo\""
// foo\bar -> "\"foo\bar\""
// foo\ -> "\"foo\\\\\""
// foo\"bar -> "\"foo\\\\\\\"bar\"".
// ~/foo -> "~/\"foo\""
// ~user/foo -> "~user/\"foo\""
static std::string QuoteForSsh(const std::string& argument);
private: private:
// Verifies that both || and |ssh_port_| are set. // Verifies that both |user_host_| and |ssh_port_| are set.
absl::Status CheckHostPort(); absl::Status CheckUserHostPort();
// Common code for BuildProcessStartInfoForSsh*. // Common code for BuildProcessStartInfoForSsh*.
ProcessStartInfo BuildProcessStartInfoForSshInternal( ProcessStartInfo BuildProcessStartInfoForSshInternal(

View File

@@ -97,35 +97,40 @@ TEST_F(RemoteUtilTest, BuildProcessStartInfoForSshWithCustomCommand) {
ExpectContains(si.command, {kCustomSshCmd}); ExpectContains(si.command, {kCustomSshCmd});
} }
TEST_F(RemoteUtilTest, EscapeForWindows) { TEST_F(RemoteUtilTest, QuoteForWindows) {
EXPECT_EQ("foo", RemoteUtil::EscapeForWindows("foo")); EXPECT_EQ(RemoteUtil::QuoteForWindows("foo"), "\"foo\"");
EXPECT_EQ("foo bar", RemoteUtil::EscapeForWindows("foo bar")); EXPECT_EQ(RemoteUtil::QuoteForWindows("foo bar"), "\"foo bar\"");
EXPECT_EQ("foo\\bar", RemoteUtil::EscapeForWindows("foo\\bar")); EXPECT_EQ(RemoteUtil::QuoteForWindows("foo\\bar"), "\"foo\\bar\"");
EXPECT_EQ("\\\\foo", RemoteUtil::EscapeForWindows("\\\\foo")); EXPECT_EQ(RemoteUtil::QuoteForWindows("\\\\foo"), "\"\\\\foo\"");
EXPECT_EQ("foo\\\\", RemoteUtil::EscapeForWindows("foo\\")); EXPECT_EQ(RemoteUtil::QuoteForWindows("foo\\"), "\"foo\\\\\"");
EXPECT_EQ("foo\\\\\\\\", RemoteUtil::EscapeForWindows("foo\\\\")); EXPECT_EQ(RemoteUtil::QuoteForWindows("foo\\\\"), "\"foo\\\\\\\\\"");
EXPECT_EQ("foo\\\"", RemoteUtil::EscapeForWindows("foo\"")); EXPECT_EQ(RemoteUtil::QuoteForWindows("foo\""), "\"foo\\\"\"");
EXPECT_EQ("foo\\\"bar", RemoteUtil::EscapeForWindows("foo\"bar")); EXPECT_EQ(RemoteUtil::QuoteForWindows("foo\"bar"), "\"foo\\\"bar\"");
EXPECT_EQ("foo\\\\\\\"bar", RemoteUtil::EscapeForWindows("foo\\\"bar")); EXPECT_EQ(RemoteUtil::QuoteForWindows("foo\\\"bar"), "\"foo\\\\\\\"bar\"");
EXPECT_EQ("foo\\\\\\\\\\\"bar", RemoteUtil::EscapeForWindows("foo\\\\\"bar")); EXPECT_EQ(RemoteUtil::QuoteForWindows("foo\\\\\"bar"),
EXPECT_EQ("\\\"foo\\\"", RemoteUtil::EscapeForWindows("\"foo\"")); "\"foo\\\\\\\\\\\"bar\"");
EXPECT_EQ("\\\" \\file.txt", RemoteUtil::EscapeForWindows("\" \\file.txt")); EXPECT_EQ(RemoteUtil::QuoteForWindows("\"foo\""), "\"\\\"foo\\\"\"");
EXPECT_EQ(RemoteUtil::QuoteForWindows("\" \\file.txt"),
"\"\\\" \\file.txt\"");
} }
TEST_F(RemoteUtilTest, QuoteArgument) { TEST_F(RemoteUtilTest, QuoteForSsh) {
EXPECT_EQ("\"foo\"", RemoteUtil::QuoteArgument("foo")); EXPECT_EQ(RemoteUtil::QuoteForSsh("foo"), "\"\\\"foo\\\"\"");
EXPECT_EQ("\"foo bar\"", RemoteUtil::QuoteArgument("foo bar")); EXPECT_EQ(RemoteUtil::QuoteForSsh("foo\\bar"), "\"\\\"foo\\\\bar\\\"\"");
EXPECT_EQ("\"foo\\bar\"", RemoteUtil::QuoteArgument("foo\\bar")); EXPECT_EQ(RemoteUtil::QuoteForSsh("foo\\"), "\"\\\"foo\\\\\\\\\\\"\"");
EXPECT_EQ("\"\\\\foo\"", RemoteUtil::QuoteArgument("\\\\foo")); EXPECT_EQ(RemoteUtil::QuoteForSsh("foo\\\"bar"),
EXPECT_EQ("\"foo\\\\\"", RemoteUtil::QuoteArgument("foo\\")); "\"\\\"foo\\\\\\\\\\\\\\\"bar\\\"\"");
EXPECT_EQ("\"foo\\\\\\\\\"", RemoteUtil::QuoteArgument("foo\\\\")); EXPECT_EQ(RemoteUtil::QuoteForSsh("~"), "\"~\"");
EXPECT_EQ("\"foo\\\"\"", RemoteUtil::QuoteArgument("foo\"")); EXPECT_EQ(RemoteUtil::QuoteForSsh("~username"), "\"~username\"");
EXPECT_EQ("\"foo\\\"bar\"", RemoteUtil::QuoteArgument("foo\"bar")); EXPECT_EQ(RemoteUtil::QuoteForSsh("~/foo"), "\"~/\\\"foo\\\"\"");
EXPECT_EQ("\"foo\\\\\\\"bar\"", RemoteUtil::QuoteArgument("foo\\\"bar")); EXPECT_EQ(RemoteUtil::QuoteForSsh("~username/foo"),
EXPECT_EQ("\"foo\\\\\\\\\\\"bar\"", "\"~username/\\\"foo\\\"\"");
RemoteUtil::QuoteArgument("foo\\\\\"bar")); EXPECT_EQ(RemoteUtil::QuoteForSsh("~invalid user name"),
EXPECT_EQ("\"\\\"foo\\\"\"", RemoteUtil::QuoteArgument("\"foo\"")); "\"\\\"~invalid user name\\\"\"");
EXPECT_EQ("\"\\\" \\file.txt\"", RemoteUtil::QuoteArgument("\" \\file.txt")); EXPECT_EQ(RemoteUtil::QuoteForSsh("~invalid user name/foo"),
"\"\\\"~invalid user name/foo\\\"\"");
EXPECT_EQ(RemoteUtil::QuoteForSsh("~user-name69/foo"),
"\"~user-name69/\\\"foo\\\"\""); // Nice!
} }
} // namespace } // namespace

View File

@@ -14,14 +14,12 @@
#include "data_store/data_provider.h" #include "data_store/data_provider.h"
#include <chrono>
#include <numeric> #include <numeric>
#include <thread> #include <thread>
#include "common/path.h" #include "common/path.h"
#include "common/status_test_macros.h" #include "common/status_test_macros.h"
#include "common/testing_clock.h" #include "common/testing_clock.h"
#include "common/util.h"
#include "data_store/disk_data_store.h" #include "data_store/disk_data_store.h"
#include "data_store/mem_data_store.h" #include "data_store/mem_data_store.h"
#include "gtest/gtest.h" #include "gtest/gtest.h"

View File

@@ -25,35 +25,50 @@ service LocalAssetsStreamManager {
rpc StopSession(StopSessionRequest) returns (StopSessionResponse) {} rpc StopSession(StopSessionRequest) returns (StopSessionResponse) {}
} }
// NextID: 7 // NextID: 12
message StartSessionRequest { message StartSessionRequest {
// ID of assets streaming target gamelet. gamelet_id will continue to be set
// alongside gamelet_name for backwards compatibility, but new code should
// not read from the gamelet_id field.
string gamelet_id = 1;
// The resource name of the assets streaming target gamelet, in the form // The resource name of the assets streaming target gamelet, in the form
// "organizations/{org-id}/projects/{proj-id}/pools/{pool-id}/gamelets/{gamelet-id}". // "organizations/{org-id}/projects/{proj-id}/pools/{pool-id}/gamelets/{gamelet-id}".
// If gamelet_name is specified, it will take precedence over gamelet_id. // Only used by Stadia. Should set either this or user_host.
string gamelet_name = 5; string gamelet_name = 5;
// Path in the local workstation to stream assets from. // Directory in the local workstation to stream assets from.
string workstation_directory = 2; string workstation_directory = 2;
// The user's email. // Caller of the StartSession request.
string account = 3; // Only used by Stadia. May be left unspecified.
// The OnePlatForm Url of the publishing API.
string url = 4;
// Caller of the SartSession request.
enum Origin { enum Origin {
ORIGIN_UNKNOWN = 0; ORIGIN_UNKNOWN = 0;
ORIGIN_CLI = 1; ORIGIN_CLI = 1;
ORIGIN_PARTNER_PORTAL = 2; ORIGIN_PARTNER_PORTAL = 2;
} }
Origin origin = 6; Origin origin = 6;
// Username and host, in the form [user@]host.
string user_host = 7;
// Remote directory where to mount the streamed directory.
string mount_dir = 8;
// SSH port to use while connecting to the remote instance.
// Optional, falls back to port 22 (default SSH port).
int32 port = 9;
// SSH command to connect to the remote instance.
// Optional, falls back to searching ssh.
string ssh_command = 10;
// SCP command to copy files to the remote instance.
// Optional, falls back to searching scp.
string scp_command = 11;
reserved 1, 3, 4;
} }
message StartSessionResponse {} message StartSessionResponse {}
// NextID: 4
message StopSessionRequest { message StopSessionRequest {
// ID of assets streaming target gamelet.
// Only used by Stadia. Should set either this or user_host_dir.
string gamelet_id = 1; string gamelet_id = 1;
// Username and host, in the form [user@]host.
string user_host = 2;
// Remote directory where the streamed directory is mounted.
string mount_dir = 3;
} }
message StopSessionResponse {} message StopSessionResponse {}

View File

@@ -28,6 +28,7 @@ cc_binary(
"//asset_stream_manager:multi_session", "//asset_stream_manager:multi_session",
"//cdc_fuse_fs:asset", "//cdc_fuse_fs:asset",
"//cdc_fuse_fs:cdc_fuse_fs_lib_mocked", "//cdc_fuse_fs:cdc_fuse_fs_lib_mocked",
"//cdc_fuse_fs:mock_config_stream_client",
"//common:test_main", "//common:test_main",
"//common:testing_clock", "//common:testing_clock",
"//data_store:data_provider", "//data_store:data_provider",

View File

@@ -62,7 +62,6 @@
<BazelTargets>//tests_asset_streaming_30</BazelTargets> <BazelTargets>//tests_asset_streaming_30</BazelTargets>
<BazelOutputFile>tests_asset_streaming_30.exe</BazelOutputFile> <BazelOutputFile>tests_asset_streaming_30.exe</BazelOutputFile>
<BazelIncludePaths>..\;..\third_party\absl;..\third_party\jsoncpp\include;..\third_party\blake3\c;..\third_party\googletest\googletest\include;..\third_party\protobuf\src;..\third_party\grpc\include;..\bazel-out\x64_windows-dbg\bin;$(VC_IncludePath);$(WindowsSDK_IncludePath)</BazelIncludePaths> <BazelIncludePaths>..\;..\third_party\absl;..\third_party\jsoncpp\include;..\third_party\blake3\c;..\third_party\googletest\googletest\include;..\third_party\protobuf\src;..\third_party\grpc\include;..\bazel-out\x64_windows-dbg\bin;$(VC_IncludePath);$(WindowsSDK_IncludePath)</BazelIncludePaths>
<BazelSourcePathPrefix>..\/</BazelSourcePathPrefix>
</PropertyGroup> </PropertyGroup>
<Import Project="..\NMakeBazelProject.targets" /> <Import Project="..\NMakeBazelProject.targets" />
<Import Project="$(VCTargetsPath)\Microsoft.Cpp.targets" /> <Import Project="$(VCTargetsPath)\Microsoft.Cpp.targets" />

View File

@@ -63,7 +63,6 @@
<BazelTargets>//tests_cdc_rsync</BazelTargets> <BazelTargets>//tests_cdc_rsync</BazelTargets>
<BazelOutputFile>tests_cdc_rsync.exe</BazelOutputFile> <BazelOutputFile>tests_cdc_rsync.exe</BazelOutputFile>
<BazelIncludePaths>..\;..\third_party\absl;..\third_party\jsoncpp\include;..\bazel-stadia-file-transfer\external\com_github_zstd\lib;..\third_party\blake3\c;..\third_party\googletest\googletest\include;..\third_party\protobuf\src;..\third_party\grpc\include;..\bazel-out\x64_windows-dbg\bin;$(VC_IncludePath);$(WindowsSDK_IncludePath)</BazelIncludePaths> <BazelIncludePaths>..\;..\third_party\absl;..\third_party\jsoncpp\include;..\bazel-stadia-file-transfer\external\com_github_zstd\lib;..\third_party\blake3\c;..\third_party\googletest\googletest\include;..\third_party\protobuf\src;..\third_party\grpc\include;..\bazel-out\x64_windows-dbg\bin;$(VC_IncludePath);$(WindowsSDK_IncludePath)</BazelIncludePaths>
<BazelSourcePathPrefix>..\/</BazelSourcePathPrefix>
</PropertyGroup> </PropertyGroup>
<Import Project="..\NMakeBazelProject.targets" /> <Import Project="..\NMakeBazelProject.targets" />
<Import Project="$(VCTargetsPath)\Microsoft.Cpp.targets" /> <Import Project="$(VCTargetsPath)\Microsoft.Cpp.targets" />

View File

@@ -62,7 +62,6 @@
<BazelTargets>//tests_common</BazelTargets> <BazelTargets>//tests_common</BazelTargets>
<BazelOutputFile>tests_common.exe</BazelOutputFile> <BazelOutputFile>tests_common.exe</BazelOutputFile>
<BazelIncludePaths>..\;..\third_party\absl;..\third_party\jsoncpp\include;..\third_party\blake3\c;..\third_party\googletest\googletest\include;..\third_party\protobuf\src;..\third_party\grpc\include;..\bazel-out\x64_windows-dbg\bin;$(VC_IncludePath);$(WindowsSDK_IncludePath)</BazelIncludePaths> <BazelIncludePaths>..\;..\third_party\absl;..\third_party\jsoncpp\include;..\third_party\blake3\c;..\third_party\googletest\googletest\include;..\third_party\protobuf\src;..\third_party\grpc\include;..\bazel-out\x64_windows-dbg\bin;$(VC_IncludePath);$(WindowsSDK_IncludePath)</BazelIncludePaths>
<BazelSourcePathPrefix>..\/</BazelSourcePathPrefix>
</PropertyGroup> </PropertyGroup>
<Import Project="..\NMakeBazelProject.targets" /> <Import Project="..\NMakeBazelProject.targets" />
<Import Project="$(VCTargetsPath)\Microsoft.Cpp.targets" /> <Import Project="$(VCTargetsPath)\Microsoft.Cpp.targets" />