前言
adb shell是Android开发者常用的一个工具，它可以让我们在电脑上通过USB或网络连接到Android设备，并执行一些命令或操作。但是，有时候我们可能不想让任何人都能随意使用adb shell，而是需要一些授权或验证的机制，以保护我们的设备和数据。本文将介绍如何在基于rockchip rk3568 android11的系统上实现自定义的adb shell授权功能。
1.实现原理
要实现自定义的adb shell授权功能，我们需要修改adb daemon的代码()，让它在接收到shell服务请求时，先检查是否已经通过了授权，如果没有，就返回一个错误信息，并提示用户输入一个随机生成的授权码。用户可以通过adb shell auth 命令来输入授权码，如果正确，就可以正常使用adb shell，否则就会被拒绝。
system/core/adb/client/commandline.cpp 这里是adb client端的,改这里没有 最开始走弯路.
system/core/adb/daemon/file_sync_service.cpp 这里是adb push/pull 逻辑,后面再研究
1.1 ADB 授权流程
- 开始
- 生成随机授权码
- 检查授权状态
- 如果已经授权，执行 ADB 命令，然后结束
- 如果未授权，检查命令是否以 “auth” 开头
- 如果命令以 “auth” 开头，检查授权码是否与预期授权码匹配
- 如果授权码匹配，设置已授权标志为真，返回 “authorization ok”，执行 ADB 命令，然后结束
- 如果授权码不匹配，返回 “authorization code:预期授权码”，然后结束
- 如果命令不以 “auth” 开头，返回 “authorization code:预期授权码”，然后结束
1.2 ADB 命令执行流程
- 接收 ADB 命令。
- 如果命令以 “shell” 开头，执行 ShellService。
- 在 ShellService 中，如果已经授权，执行 ADB 命令并返回执行结果。
- 如果未授权，检查命令是否以 “auth” 开头。
- 如果命令以 “auth” 开头，检查授权码是否与预期的授权码匹配。
- 如果授权码匹配，设置授权标志为真，并返回 “authorization ok”。
- 如果授权码不匹配，返回 “authorization code:预期授权码”。
- 如果命令不以 “auth” 开头，返回 “authorization code:预期授权码”。
- 如果命令不以 “shell” 开头，根据命令的不同执行不同的服务，如 “exec:”, “sync:”, “reverse:”, “reconnect”, “spin” 等，并返回执行结果。
2.修改adb daemon源码
需要修改system/core/adb/daemon/services.cpp文件中的ShellService函数，增加一个CheckAuthorization函数来检查授权状态和处理授权码输入。CheckAuthorization函数的实现如下：
bool authorized = false;
std::string GenerateRandomCode() {
static std::string expected_code; // 在启动时生成随机码。
if (!expected_code.empty()) {
return expected_code;
}
std::random_device rd;
std::mt19937 gen(rd());
std::uniform_int_distribution<> dis(0, 35);
for (int i = 0; i < 6; ++i) {
int random = dis(gen);
if (random < 10) {
expected_code += '0' + random;
} else {
expected_code += 'a' + random - 10;
}
}
return expected_code;
}
unique_fd CheckAuthorization(std::string_view command, SubprocessProtocol protocol) {
std::string expected_code = GenerateRandomCode();
if (command.starts_with("auth")) {
std::string auth_code = std::string(command.substr(strlen("auth")));
auth_code = android::base::Trim(auth_code); // remove leading/trailing whitespace
if (auth_code == expected_code) {
authorized = true;
std::string message = "authorization ok";
return ReportError(protocol, message);
} else {
std::string message = android::base::StringPrintf("authorization code:%s", expected_code.c_str());
return ReportError(protocol, message);
}
}
if (!authorized) {
std::string message = android::base::StringPrintf("authorization code:%s", expected_code.c_str());
return ReportError(protocol, message);
}
return unique_fd{
}; // Return an invalid fd if authorized
}
在ShellService函数中，在解析服务参数和命令之后，调用CheckAuthorization函数，并根据返回值判断是否继续执行shell服务：
unique_fd ShellService(std::string_view args, const atransport* transport) {
size_t delimiter_index = args.find(':');
if (delimiter_index == std::string::npos) {
LOG(ERROR) << "Invalid shell service arguments: " << args;
return unique_fd{
};
}
// TODO: android::base::Split(const std::string_view&, ...)
std::string service_args(args.substr(0, delimiter_index));
std::string command(args.substr(delimiter_index + 1));
// Defaults:
// PTY for interactive, raw for non-interactive.
// No protocol.
// $TERM set to "dumb".
SubprocessType type(command.empty() ? SubprocessType::kPty : SubprocessType::kRaw);
SubprocessProtocol protocol = SubprocessProtocol::kNone;
std::string terminal_type = "dumb";
for (const auto& arg : android::base::Split(service_args, ",")) {
if (arg == "raw") {
type = SubprocessType::kRaw;
} else if (arg == "pty") {
type = SubprocessType::kPty;
} else if (arg == "v2") {
protocol = SubprocessProtocol::kV2;
} else if (arg.starts_with("TERM=")) {
terminal_type = arg.substr(strlen("TERM="));
} else if (!arg.empty()) {
// This is not an error to allow for future expansion.
LOG(WARNING) << "Ignoring unknown shell service argument: " << arg;
}
}
unique_fd auth_fd = CheckAuthorization(command, protocol);
if (auth_fd.get() != -1) {
return auth_fd;
}
return StartSubprocess(command, terminal_type.c_str(), type, protocol);
}
还需要修改system/core/adb/Android.bp文件，去掉-Wexit-time-destructors的编译选项，因为我们使用了一个静态变量来存储授权码，它会在程序退出时触发析构函数，导致编译报错：
cc_defaults {
name: "adb_defaults",
cflags: [
"-Wall",
"-Wextra",
"-Werror",
- "-Wexit-time-destructors",
+ "-Wno-exit-time-destructors",
"-Wno-unused-parameter",
"-Wno-missing-field-initializers",
"-Wthread-safety",
修改完代码后，我们重新编译并刷入系统，然后连接设备并尝试使用adb shell。我们可以看到，如果没有输入正确的授权码，就会收到一个错误信息，并提示我们输入授权码。如果输入正确的授权码，就可以正常使用adb shell。测试结果如下：
总结
本文介绍了如何在基于rockchip rk3568 android11的系统上实现自定义的adb shell授权功能，主要是通过修改adb daemon的代码，增加一个CheckAuthorization函数来检查授权状态和处理授权码输入。这样可以提高设备的安全性，防止未经授权的人员使用adb shell。
这只是一个简单的示例，实际应用中可能需要更复杂的逻辑和更安全的机制。希望本文对你有所帮助。