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,unsigned mapSz,unsigned progSz)30 void doBpfStructSizeTest(const char *elfPath, unsigned mapSz, unsigned progSz) {
31 std::ifstream elfFile(elfPath, std::ios::in | std::ios::binary);
32 ASSERT_TRUE(elfFile.is_open());
33
34 EXPECT_EQ(mapSz, readSectionUint("size_of_bpf_map_def", elfFile, 0));
35 EXPECT_EQ(progSz, readSectionUint("size_of_bpf_prog_def", elfFile, 0));
36 }
37
TEST(BpfTest,bpfStructSizeTest)38 TEST(BpfTest, bpfStructSizeTest) {
39 if (android::modules::sdklevel::IsAtLeastV()) {
40 // Due to V+ using mainline netbpfload, there is no longer a need to
41 // enforce consistency between platform and mainline bpf .o files.
42 GTEST_SKIP() << "V+ device.";
43 } else if (android::modules::sdklevel::IsAtLeastU()) {
44 doBpfStructSizeTest("/system/etc/bpf/gpuMem.o", 120, 92);
45 doBpfStructSizeTest("/system/etc/bpf/timeInState.o", 120, 92);
46 } else if (android::modules::sdklevel::IsAtLeastT()) {
47 doBpfStructSizeTest("/system/etc/bpf/gpu_mem.o", 116, 92);
48 doBpfStructSizeTest("/system/etc/bpf/time_in_state.o", 116, 92);
49 } else if (android::modules::sdklevel::IsAtLeastS()) {
50 // These files were moved to mainline in Android T
51 doBpfStructSizeTest("/system/etc/bpf/netd.o", 48, 28);
52 doBpfStructSizeTest("/system/etc/bpf/clatd.o", 48, 28);
53 } else {
54 // There is no mainline bpf code before S.
55 GTEST_SKIP() << "R- device.";
56 }
57 }
58
main(int argc,char ** argv)59 int main(int argc, char **argv) {
60 testing::InitGoogleTest(&argc, argv);
61 return RUN_ALL_TESTS();
62 }
63