1 /*
2 * Copyright (C) 2021 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 requied 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 #define LOG_TAG "BpfCompatTest"
19
20 #include <fstream>
21
22 #include <gtest/gtest.h>
23
24 #include "android-modules-utils/sdk_level.h"
25
26 #include "libbpf_android.h"
27
28 using namespace android::bpf;
29
doBpfStructSizeTest(const char * elfPath)30 void doBpfStructSizeTest(const char *elfPath) {
31 std::ifstream elfFile(elfPath, std::ios::in | std::ios::binary);
32 ASSERT_TRUE(elfFile.is_open());
33
34 if (android::modules::sdklevel::IsAtLeastU()) {
35 EXPECT_EQ(120, readSectionUint("size_of_bpf_map_def", elfFile, 0));
36 EXPECT_EQ(92, readSectionUint("size_of_bpf_prog_def", elfFile, 0));
37 } else if (android::modules::sdklevel::IsAtLeastT()) {
38 EXPECT_EQ(116, readSectionUint("size_of_bpf_map_def", elfFile, 0));
39 EXPECT_EQ(92, readSectionUint("size_of_bpf_prog_def", elfFile, 0));
40 } else {
41 EXPECT_EQ(48, readSectionUint("size_of_bpf_map_def", elfFile, 0));
42 EXPECT_EQ(28, readSectionUint("size_of_bpf_prog_def", elfFile, 0));
43 }
44 }
45
TEST(BpfTest,bpfStructSizeTestPreT)46 TEST(BpfTest, bpfStructSizeTestPreT) {
47 if (android::modules::sdklevel::IsAtLeastT()) GTEST_SKIP() << "T+ device.";
48 doBpfStructSizeTest("/system/etc/bpf/netd.o");
49 doBpfStructSizeTest("/system/etc/bpf/clatd.o");
50 }
51
TEST(BpfTest,bpfStructSizeTest)52 TEST(BpfTest, bpfStructSizeTest) {
53 if (android::modules::sdklevel::IsAtLeastU()) {
54 doBpfStructSizeTest("/system/etc/bpf/gpuMem.o");
55 doBpfStructSizeTest("/system/etc/bpf/timeInState.o");
56 } else {
57 doBpfStructSizeTest("/system/etc/bpf/gpu_mem.o");
58 doBpfStructSizeTest("/system/etc/bpf/time_in_state.o");
59 }
60 }
61
main(int argc,char ** argv)62 int main(int argc, char **argv) {
63 testing::InitGoogleTest(&argc, argv);
64 return RUN_ALL_TESTS();
65 }
66