Android系统 adb shell auth授权使用_adb授权

前言

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
}
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

在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);
}
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

还需要修改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",
1
2
3
4
5
6
7
8
9
10
11

修改完代码后，我们重新编译并刷入系统，然后连接设备并尝试使用adb shell。我们可以看到，如果没有输入正确的授权码，就会收到一个错误信息，并提示我们输入授权码。如果输入正确的授权码，就可以正常使用adb shell。测试结果如下：

总结

本文介绍了如何在基于rockchip rk3568 android11的系统上实现自定义的adb shell授权功能，主要是通过修改adb daemon的代码，增加一个CheckAuthorization函数来检查授权状态和处理授权码输入。这样可以提高设备的安全性，防止未经授权的人员使用adb shell。

这只是一个简单的示例，实际应用中可能需要更复杂的逻辑和更安全的机制。希望本文对你有所帮助。