1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
|
/*
* Copyright (C) 2017 The Android Open Source Project
*
* 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.
*/
#pragma once
#include <atomic>
#include <string>
#include <unordered_map>
#include <unordered_set>
#include <vector>
#include <fruit/fruit.h>
#include "common/libs/fs/shared_fd.h"
#include "common/libs/utils/json.h"
#include "common/libs/utils/result.h"
#include "host/commands/run_cvd/launch/webrtc_recorder.h"
#include "host/commands/run_cvd/server_loop.h"
#include "host/libs/command_util/runner/defs.h"
#include "host/libs/command_util/util.h"
#include "host/libs/config/command_source.h"
#include "host/libs/config/cuttlefish_config.h"
#include "host/libs/config/feature.h"
#include "host/libs/config/inject.h"
#include "host/libs/process_monitor/process_monitor.h"
namespace cuttlefish {
namespace run_cvd_impl {
class ServerLoopImpl : public ServerLoop,
public SetupFeature,
public LateInjected {
public:
INJECT(ServerLoopImpl(const CuttlefishConfig& config,
const CuttlefishConfig::InstanceSpecific& instance,
AutoSecureEnvFiles::Type& secure_env_files,
WebRtcRecorder& webrtc_recorder));
Result<void> LateInject(fruit::Injector<>& injector) override;
// ServerLoop
Result<void> Run() override;
// SetupFeature
std::string Name() const override { return "ServerLoop"; }
enum class DeviceStatus : int {
kUnknown = 0,
kActive = 1,
kSuspended = 2,
};
private:
bool Enabled() const override { return true; }
std::unordered_set<SetupFeature*> Dependencies() const override {
return {&secure_env_files_};
}
Result<void> ResultSetup() override;
Result<void> HandleExtended(const LauncherActionInfo& action_info,
ProcessMonitor& process_monitor);
Result<void> HandleSuspend(ProcessMonitor& process_monitor);
Result<void> HandleResume(ProcessMonitor& process_monitor);
Result<void> HandleSnapshotTake(const run_cvd::SnapshotTake& snapshot_take);
Result<void> HandleStartScreenRecording();
Result<void> HandleStopScreenRecording();
void HandleActionWithNoData(const LauncherAction action,
const SharedFD& client,
ProcessMonitor& process_monitor);
void DeleteFifos();
bool PowerwashFiles();
void RestartRunCvd(int notification_fd);
static bool CreateQcowOverlay(const std::string& crosvm_path,
const std::string& backing_file,
const std::string& output_overlay_path);
Result<void> SuspendGuest();
Result<void> ResumeGuest();
static std::unordered_map<std::string, std::string>
InitializeVmToControlSockPath(const CuttlefishConfig::InstanceSpecific&);
Result<std::string> VmControlSocket() const;
Result<void> TakeGuestSnapshot(const std::string&, const std::string&);
Result<void> TakeCrosvmGuestSnapshot(const Json::Value&);
const CuttlefishConfig& config_;
const CuttlefishConfig::InstanceSpecific& instance_;
/*
* This is needed to get the run_cvd side socket pair connected to
* secure_env. The socket pairs are used to send suspend/resume to
* secure_env, and get the responses.
*/
AutoSecureEnvFiles::Type& secure_env_files_;
WebRtcRecorder& webrtc_recorder_;
std::vector<CommandSource*> command_sources_;
SharedFD server_;
// mapping from the name of vm_manager to control_sock path
std::unordered_map<std::string, std::string> vm_name_to_control_sock_;
std::atomic<DeviceStatus> device_status_;
};
} // namespace run_cvd_impl
} // namespace cuttlefish
|
