mirror of
https://github.com/nestriness/cdc-file-transfer.git
synced 2026-01-30 14:45:37 +02:00
Releasing the former Stadia file transfer tools
The tools allow efficient and fast synchronization of large directory trees from a Windows workstation to a Linux target machine. cdc_rsync* support efficient copy of files by using content-defined chunking (CDC) to identify chunks within files that can be reused. asset_stream_manager + cdc_fuse_fs support efficient streaming of a local directory to a remote virtual file system based on FUSE. It also employs CDC to identify and reuse unchanged data chunks.
This commit is contained in:
132
cdc_rsync/cdc_rsync_client.h
Normal file
132
cdc_rsync/cdc_rsync_client.h
Normal file
@@ -0,0 +1,132 @@
|
||||
/*
|
||||
* 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 CDC_RSYNC_CDC_RSYNC_CLIENT_H_
|
||||
#define CDC_RSYNC_CDC_RSYNC_CLIENT_H_
|
||||
|
||||
#include <string>
|
||||
#include <vector>
|
||||
|
||||
#include "absl/status/status.h"
|
||||
#include "cdc_rsync/base/message_pump.h"
|
||||
#include "cdc_rsync/cdc_rsync.h"
|
||||
#include "cdc_rsync/client_socket.h"
|
||||
#include "cdc_rsync/progress_tracker.h"
|
||||
#include "common/path_filter.h"
|
||||
#include "common/port_manager.h"
|
||||
#include "common/remote_util.h"
|
||||
|
||||
namespace cdc_ft {
|
||||
|
||||
class Process;
|
||||
class ZstdStream;
|
||||
|
||||
class GgpRsyncClient {
|
||||
public:
|
||||
GgpRsyncClient(const Options& options, PathFilter filter,
|
||||
std::string sources_dir, std::vector<std::string> sources,
|
||||
std::string destination);
|
||||
|
||||
~GgpRsyncClient();
|
||||
|
||||
// Deploys the server if necessary, starts it and runs the rsync procedure.
|
||||
absl::Status Run();
|
||||
|
||||
private:
|
||||
// Starts the server process. If the method returns a status with tag
|
||||
// |kTagDeployServer|, Run() calls DeployServer() and tries again.
|
||||
absl::Status StartServer();
|
||||
|
||||
// Stops the server process.
|
||||
absl::Status StopServer();
|
||||
|
||||
// Handler for stdout and stderr data emitted by the server.
|
||||
absl::Status HandleServerOutput(const char* data);
|
||||
|
||||
// Runs the rsync procedure.
|
||||
absl::Status Sync();
|
||||
|
||||
// Copies all gamelet components to the gamelet.
|
||||
absl::Status DeployServer();
|
||||
|
||||
// Sends relevant options to the server.
|
||||
absl::Status SendOptions();
|
||||
|
||||
// Finds all source files and sends the file infos to the server.
|
||||
absl::Status FindAndSendAllSourceFiles();
|
||||
|
||||
// Receives the stats from the file diffs (e.g. number of missing, changed
|
||||
// etc. files) from the server.
|
||||
absl::Status ReceiveFileStats();
|
||||
|
||||
// Receives paths of deleted files and prints them out.
|
||||
absl::Status ReceiveDeletedFiles();
|
||||
|
||||
// Receives file indices from the server. Used for missing and changed files.
|
||||
absl::Status ReceiveFileIndices(const char* file_type,
|
||||
std::vector<uint32_t>* file_indices);
|
||||
|
||||
// Copies missing files to the server.
|
||||
absl::Status SendMissingFiles();
|
||||
|
||||
// Core rsync algorithm. Receives signatures of changed files from server,
|
||||
// calculates the diffs and sends them to the server.
|
||||
absl::Status ReceiveSignaturesAndSendDelta();
|
||||
|
||||
// Start the zstd compression stream. Used before file copy and diff.
|
||||
absl::Status StartCompressionStream();
|
||||
|
||||
// Stops the zstd compression stream.
|
||||
absl::Status StopCompressionStream();
|
||||
|
||||
Options options_;
|
||||
PathFilter path_filter_;
|
||||
const std::string sources_dir_;
|
||||
std::vector<std::string> sources_;
|
||||
const std::string destination_;
|
||||
|
||||
WinProcessFactory process_factory_;
|
||||
RemoteUtil remote_util_;
|
||||
PortManager port_manager_;
|
||||
ClientSocket socket_;
|
||||
MessagePump message_pump_{&socket_, MessagePump::PacketReceivedDelegate()};
|
||||
ConsoleProgressPrinter printer_;
|
||||
ProgressTracker progress_;
|
||||
std::unique_ptr<ZstdStream> compression_stream_;
|
||||
|
||||
std::unique_ptr<Process> server_process_;
|
||||
std::string server_output_; // Written in a background thread. Do not access
|
||||
std::string server_error_; // while the server process is active.
|
||||
int server_exit_code_ = 0;
|
||||
std::atomic_bool is_server_listening_{false};
|
||||
bool is_server_error_ = false;
|
||||
|
||||
// All source files found on the client.
|
||||
std::vector<ClientFileInfo> files_;
|
||||
|
||||
// All source dirs found on the client.
|
||||
std::vector<ClientDirInfo> dirs_;
|
||||
|
||||
// Indices (into files_) of files that are missing on the server.
|
||||
std::vector<uint32_t> missing_file_indices_;
|
||||
|
||||
// Indices (into files_) of files that exist, but are different on the server.
|
||||
std::vector<uint32_t> changed_file_indices_;
|
||||
};
|
||||
|
||||
} // namespace cdc_ft
|
||||
|
||||
#endif // CDC_RSYNC_CDC_RSYNC_CLIENT_H_
|
||||
Reference in New Issue
Block a user