143 lines · c
1// SPDX-License-Identifier: GPL-2.02/* Copyright (c) 2019 Facebook */3 4#include <test_progs.h>5#include <sys/mman.h>6 7struct s {8 int a;9 long long b;10} __attribute__((packed));11 12#include "test_skeleton.skel.h"13 14void test_skeleton(void)15{16 int duration = 0, err;17 struct test_skeleton* skel;18 struct test_skeleton__bss *bss;19 struct test_skeleton__data *data;20 struct test_skeleton__data_dyn *data_dyn;21 struct test_skeleton__rodata *rodata;22 struct test_skeleton__rodata_dyn *rodata_dyn;23 struct test_skeleton__kconfig *kcfg;24 const void *elf_bytes;25 size_t elf_bytes_sz = 0;26 void *m;27 int i, fd;28 29 skel = test_skeleton__open();30 if (CHECK(!skel, "skel_open", "failed to open skeleton\n"))31 return;32 33 if (CHECK(skel->kconfig, "skel_kconfig", "kconfig is mmaped()!\n"))34 goto cleanup;35 36 bss = skel->bss;37 data = skel->data;38 data_dyn = skel->data_dyn;39 rodata = skel->rodata;40 rodata_dyn = skel->rodata_dyn;41 42 ASSERT_STREQ(bpf_map__name(skel->maps.rodata_dyn), ".rodata.dyn", "rodata_dyn_name");43 ASSERT_STREQ(bpf_map__name(skel->maps.data_dyn), ".data.dyn", "data_dyn_name");44 45 /* validate values are pre-initialized correctly */46 CHECK(data->in1 != -1, "in1", "got %d != exp %d\n", data->in1, -1);47 CHECK(data->out1 != -1, "out1", "got %d != exp %d\n", data->out1, -1);48 CHECK(data->in2 != -1, "in2", "got %lld != exp %lld\n", data->in2, -1LL);49 CHECK(data->out2 != -1, "out2", "got %lld != exp %lld\n", data->out2, -1LL);50 51 CHECK(bss->in3 != 0, "in3", "got %d != exp %d\n", bss->in3, 0);52 CHECK(bss->out3 != 0, "out3", "got %d != exp %d\n", bss->out3, 0);53 CHECK(bss->in4 != 0, "in4", "got %lld != exp %lld\n", bss->in4, 0LL);54 CHECK(bss->out4 != 0, "out4", "got %lld != exp %lld\n", bss->out4, 0LL);55 56 CHECK(rodata->in.in6 != 0, "in6", "got %d != exp %d\n", rodata->in.in6, 0);57 CHECK(bss->out6 != 0, "out6", "got %d != exp %d\n", bss->out6, 0);58 59 ASSERT_EQ(rodata_dyn->in_dynarr_sz, 0, "in_dynarr_sz");60 for (i = 0; i < 4; i++)61 ASSERT_EQ(rodata_dyn->in_dynarr[i], -(i + 1), "in_dynarr");62 for (i = 0; i < 4; i++)63 ASSERT_EQ(data_dyn->out_dynarr[i], i + 1, "out_dynarr");64 65 /* validate we can pre-setup global variables, even in .bss */66 data->in1 = 10;67 data->in2 = 11;68 bss->in3 = 12;69 bss->in4 = 13;70 rodata->in.in6 = 14;71 72 rodata_dyn->in_dynarr_sz = 4;73 for (i = 0; i < 4; i++)74 rodata_dyn->in_dynarr[i] = i + 10;75 76 err = test_skeleton__load(skel);77 if (CHECK(err, "skel_load", "failed to load skeleton: %d\n", err))78 goto cleanup;79 80 /* validate pre-setup values are still there */81 CHECK(data->in1 != 10, "in1", "got %d != exp %d\n", data->in1, 10);82 CHECK(data->in2 != 11, "in2", "got %lld != exp %lld\n", data->in2, 11LL);83 CHECK(bss->in3 != 12, "in3", "got %d != exp %d\n", bss->in3, 12);84 CHECK(bss->in4 != 13, "in4", "got %lld != exp %lld\n", bss->in4, 13LL);85 CHECK(rodata->in.in6 != 14, "in6", "got %d != exp %d\n", rodata->in.in6, 14);86 87 ASSERT_EQ(rodata_dyn->in_dynarr_sz, 4, "in_dynarr_sz");88 for (i = 0; i < 4; i++)89 ASSERT_EQ(rodata_dyn->in_dynarr[i], i + 10, "in_dynarr");90 91 /* now set new values and attach to get them into outX variables */92 data->in1 = 1;93 data->in2 = 2;94 bss->in3 = 3;95 bss->in4 = 4;96 bss->in5.a = 5;97 bss->in5.b = 6;98 kcfg = skel->kconfig;99 100 skel->data_read_mostly->read_mostly_var = 123;101 102 err = test_skeleton__attach(skel);103 if (CHECK(err, "skel_attach", "skeleton attach failed: %d\n", err))104 goto cleanup;105 106 /* trigger tracepoint */107 usleep(1);108 109 CHECK(data->out1 != 1, "res1", "got %d != exp %d\n", data->out1, 1);110 CHECK(data->out2 != 2, "res2", "got %lld != exp %d\n", data->out2, 2);111 CHECK(bss->out3 != 3, "res3", "got %d != exp %d\n", (int)bss->out3, 3);112 CHECK(bss->out4 != 4, "res4", "got %lld != exp %d\n", bss->out4, 4);113 CHECK(bss->out5.a != 5, "res5", "got %d != exp %d\n", bss->out5.a, 5);114 CHECK(bss->out5.b != 6, "res6", "got %lld != exp %d\n", bss->out5.b, 6);115 CHECK(bss->out6 != 14, "res7", "got %d != exp %d\n", bss->out6, 14);116 117 CHECK(bss->bpf_syscall != kcfg->CONFIG_BPF_SYSCALL, "ext1",118 "got %d != exp %d\n", bss->bpf_syscall, kcfg->CONFIG_BPF_SYSCALL);119 CHECK(bss->kern_ver != kcfg->LINUX_KERNEL_VERSION, "ext2",120 "got %d != exp %d\n", bss->kern_ver, kcfg->LINUX_KERNEL_VERSION);121 122 for (i = 0; i < 4; i++)123 ASSERT_EQ(data_dyn->out_dynarr[i], i + 10, "out_dynarr");124 125 ASSERT_EQ(skel->bss->out_mostly_var, 123, "out_mostly_var");126 127 ASSERT_EQ(bss->huge_arr[ARRAY_SIZE(bss->huge_arr) - 1], 123, "huge_arr");128 129 fd = bpf_map__fd(skel->maps.data_non_mmapable);130 m = mmap(NULL, getpagesize(), PROT_READ, MAP_SHARED, fd, 0);131 if (!ASSERT_EQ(m, MAP_FAILED, "unexpected_mmap_success"))132 munmap(m, getpagesize());133 134 ASSERT_EQ(bpf_map__map_flags(skel->maps.data_non_mmapable), 0, "non_mmap_flags");135 136 elf_bytes = test_skeleton__elf_bytes(&elf_bytes_sz);137 ASSERT_OK_PTR(elf_bytes, "elf_bytes");138 ASSERT_GE(elf_bytes_sz, 0, "elf_bytes_sz");139 140cleanup:141 test_skeleton__destroy(skel);142}143