1 /*
2 * Copyright (C) 2022 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 *
16 */
17
18 #include <gtest/gtest.h>
19 #include <vintf/VintfObject.h>
20
21 #include <fstream>
22 #include <string>
23
24 #include "bpf/KernelUtils.h"
25
26 namespace android {
27 namespace net {
28
29 namespace {
30
31 using ::android::vintf::RuntimeInfo;
32 using ::android::vintf::VintfObject;
33
34 class KernelConfigVerifier final {
35 public:
KernelConfigVerifier()36 KernelConfigVerifier() : mRuntimeInfo(VintfObject::GetRuntimeInfo()) {}
37
hasOption(const std::string & option) const38 bool hasOption(const std::string& option) const {
39 const auto& configMap = mRuntimeInfo->kernelConfigs();
40 auto it = configMap.find(option);
41 if (it != configMap.cend()) {
42 return it->second == "y";
43 }
44 return false;
45 }
46
47 private:
48 std::shared_ptr<const RuntimeInfo> mRuntimeInfo;
49 };
50
isGsiImage()51 bool isGsiImage() {
52 std::ifstream ifs("/system/system_ext/etc/init/init.gsi.rc");
53 return ifs.good();
54 }
55
56 } // namespace
57
58 /**
59 * If this test fails, enable the following kernel modules in your kernel config:
60 * CONFIG_NET_CLS_MATCHALL=y
61 * CONFIG_NET_ACT_POLICE=y
62 * CONFIG_NET_ACT_BPF=y
63 * CONFIG_BPF_JIT=y
64 */
TEST(KernelTest,TestRateLimitingSupport)65 TEST(KernelTest, TestRateLimitingSupport) {
66 if (isGsiImage()) {
67 // skip test on gsi images
68 GTEST_SKIP() << "GSI Image";
69 }
70 KernelConfigVerifier configVerifier;
71 ASSERT_TRUE(configVerifier.hasOption("CONFIG_NET_CLS_MATCHALL"));
72 ASSERT_TRUE(configVerifier.hasOption("CONFIG_NET_ACT_POLICE"));
73 ASSERT_TRUE(configVerifier.hasOption("CONFIG_NET_ACT_BPF"));
74 ASSERT_TRUE(configVerifier.hasOption("CONFIG_BPF_JIT"));
75 }
76
TEST(KernelTest,TestBpfJitAlwaysOn)77 TEST(KernelTest, TestBpfJitAlwaysOn) {
78 // 32-bit arm & x86 kernels aren't capable of JIT-ing all of our BPF code,
79 if (bpf::isKernel32Bit()) GTEST_SKIP() << "Exempt on 32-bit kernel.";
80 KernelConfigVerifier configVerifier;
81 ASSERT_TRUE(configVerifier.hasOption("CONFIG_BPF_JIT_ALWAYS_ON"));
82 }
83
84 /* Android 14/U should only launch on 64-bit kernels
85 * T launches on 5.10/5.15
86 * U launches on 5.15/6.1
87 * So >=5.16 implies isKernel64Bit()
88 */
TEST(KernelTest,TestKernel64Bit)89 TEST(KernelTest, TestKernel64Bit) {
90 if (!bpf::isAtLeastKernelVersion(5, 16, 0)) GTEST_SKIP() << "Exempt on < 5.16 kernel.";
91 ASSERT_TRUE(bpf::isKernel64Bit());
92 }
93
94 } // namespace net
95 } // namespace android
96