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
|
/*
* Copyright (C) 2018 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.
*/
#include <algorithm>
#include <iterator>
#include <limits>
#include <sstream>
#include <thread>
#include <vector>
#include <glog/logging.h>
#include <gflags/gflags.h>
#include <unistd.h>
#include <host/commands/kernel_log_monitor/kernel_log_server.h>
#include "common/libs/fs/shared_fd.h"
#include "common/libs/strings/str_split.h"
#include "host/libs/config/cuttlefish_config.h"
#include "host/libs/adb_connection_maintainer/adb_connection_maintainer.h"
DEFINE_string(addresses, "", "Comma-separated list of addresses to "
"'adb connect' to");
DEFINE_int32(adbd_events_fd, -1, "A file descriptor. If set it will wait for "
"AdbdStarted boot event from the kernel log "
"monitor before trying to connect adb");
namespace {
void LaunchConnectionMaintainerThread(const std::string& address) {
std::thread(cvd::EstablishAndMaintainConnection, address).detach();
}
std::vector<std::string> ParseAddressList(std::string ports) {
std::replace(ports.begin(), ports.end(), ',', ' ');
std::istringstream port_stream{ports};
return {std::istream_iterator<std::string>{port_stream},
std::istream_iterator<std::string>{}};
}
[[noreturn]] void SleepForever() {
while (true) {
sleep(std::numeric_limits<unsigned int>::max());
}
}
void WaitForAdbdToBeStarted(int events_fd) {
auto evt_shared_fd = cvd::SharedFD::Dup(events_fd);
close(events_fd);
while (evt_shared_fd->IsOpen()) {
monitor::BootEvent event;
auto bytes_read = evt_shared_fd->Read(&event, sizeof(event));
if (bytes_read != sizeof(event)) {
LOG(ERROR) << "Fail to read a complete event, read " << bytes_read
<< " bytes only instead of the expected " << sizeof(event);
// The file descriptor can't be trusted anymore, stop waiting and try to
// connect
return;
}
if (event == monitor::BootEvent::AdbdStarted) {
LOG(INFO) << "Adbd has started in the guest, connecting adb";
return;
}
}
}
} // namespace
int main(int argc, char* argv[]) {
gflags::ParseCommandLineFlags(&argc, &argv, true);
CHECK(!FLAGS_addresses.empty()) << "Must specify --addresses flag";
if (FLAGS_adbd_events_fd >= 0) {
WaitForAdbdToBeStarted(FLAGS_adbd_events_fd);
}
for (auto address : ParseAddressList(FLAGS_addresses)) {
LaunchConnectionMaintainerThread(address);
}
SleepForever();
}
|
