bpf/prog_tests/skeleton.c

0001 // SPDX-License-Identifier: GPL-2.0
0002 /* Copyright (c) 2019 Facebook */
0003
0004 #include <test_progs.h>
0005
0006 struct s {
0007     int a;
0008     long long b;
0009 } __attribute__((packed));
0010
0011 #include "test_skeleton.skel.h"
0012
0013 void test_skeleton(void)
0014 {
0015     int duration = 0, err;
0016     struct test_skeleton* skel;
0017     struct test_skeleton__bss *bss;
0018     struct test_skeleton__data *data;
0019     struct test_skeleton__data_dyn *data_dyn;
0020     struct test_skeleton__rodata *rodata;
0021     struct test_skeleton__rodata_dyn *rodata_dyn;
0022     struct test_skeleton__kconfig *kcfg;
0023     const void *elf_bytes;
0024     size_t elf_bytes_sz = 0;
0025     int i;
0026
0027     skel = test_skeleton__open();
0028     if (CHECK(!skel, "skel_open", "failed to open skeleton\n"))
0029         return;
0030
0031     if (CHECK(skel->kconfig, "skel_kconfig", "kconfig is mmaped()!\n"))
0032         goto cleanup;
0033
0034     bss = skel->bss;
0035     data = skel->data;
0036     data_dyn = skel->data_dyn;
0037     rodata = skel->rodata;
0038     rodata_dyn = skel->rodata_dyn;
0039
0040     ASSERT_STREQ(bpf_map__name(skel->maps.rodata_dyn), ".rodata.dyn", "rodata_dyn_name");
0041     ASSERT_STREQ(bpf_map__name(skel->maps.data_dyn), ".data.dyn", "data_dyn_name");
0042
0043     /* validate values are pre-initialized correctly */
0044     CHECK(data->in1 != -1, "in1", "got %d != exp %d\n", data->in1, -1);
0045     CHECK(data->out1 != -1, "out1", "got %d != exp %d\n", data->out1, -1);
0046     CHECK(data->in2 != -1, "in2", "got %lld != exp %lld\n", data->in2, -1LL);
0047     CHECK(data->out2 != -1, "out2", "got %lld != exp %lld\n", data->out2, -1LL);
0048
0049     CHECK(bss->in3 != 0, "in3", "got %d != exp %d\n", bss->in3, 0);
0050     CHECK(bss->out3 != 0, "out3", "got %d != exp %d\n", bss->out3, 0);
0051     CHECK(bss->in4 != 0, "in4", "got %lld != exp %lld\n", bss->in4, 0LL);
0052     CHECK(bss->out4 != 0, "out4", "got %lld != exp %lld\n", bss->out4, 0LL);
0053
0054     CHECK(rodata->in.in6 != 0, "in6", "got %d != exp %d\n", rodata->in.in6, 0);
0055     CHECK(bss->out6 != 0, "out6", "got %d != exp %d\n", bss->out6, 0);
0056
0057     ASSERT_EQ(rodata_dyn->in_dynarr_sz, 0, "in_dynarr_sz");
0058     for (i = 0; i < 4; i++)
0059         ASSERT_EQ(rodata_dyn->in_dynarr[i], -(i + 1), "in_dynarr");
0060     for (i = 0; i < 4; i++)
0061         ASSERT_EQ(data_dyn->out_dynarr[i], i + 1, "out_dynarr");
0062
0063     /* validate we can pre-setup global variables, even in .bss */
0064     data->in1 = 10;
0065     data->in2 = 11;
0066     bss->in3 = 12;
0067     bss->in4 = 13;
0068     rodata->in.in6 = 14;
0069
0070     rodata_dyn->in_dynarr_sz = 4;
0071     for (i = 0; i < 4; i++)
0072         rodata_dyn->in_dynarr[i] = i + 10;
0073
0074     err = test_skeleton__load(skel);
0075     if (CHECK(err, "skel_load", "failed to load skeleton: %d\n", err))
0076         goto cleanup;
0077
0078     /* validate pre-setup values are still there */
0079     CHECK(data->in1 != 10, "in1", "got %d != exp %d\n", data->in1, 10);
0080     CHECK(data->in2 != 11, "in2", "got %lld != exp %lld\n", data->in2, 11LL);
0081     CHECK(bss->in3 != 12, "in3", "got %d != exp %d\n", bss->in3, 12);
0082     CHECK(bss->in4 != 13, "in4", "got %lld != exp %lld\n", bss->in4, 13LL);
0083     CHECK(rodata->in.in6 != 14, "in6", "got %d != exp %d\n", rodata->in.in6, 14);
0084
0085     ASSERT_EQ(rodata_dyn->in_dynarr_sz, 4, "in_dynarr_sz");
0086     for (i = 0; i < 4; i++)
0087         ASSERT_EQ(rodata_dyn->in_dynarr[i], i + 10, "in_dynarr");
0088
0089     /* now set new values and attach to get them into outX variables */
0090     data->in1 = 1;
0091     data->in2 = 2;
0092     bss->in3 = 3;
0093     bss->in4 = 4;
0094     bss->in5.a = 5;
0095     bss->in5.b = 6;
0096     kcfg = skel->kconfig;
0097
0098     skel->data_read_mostly->read_mostly_var = 123;
0099
0100     err = test_skeleton__attach(skel);
0101     if (CHECK(err, "skel_attach", "skeleton attach failed: %d\n", err))
0102         goto cleanup;
0103
0104     /* trigger tracepoint */
0105     usleep(1);
0106
0107     CHECK(data->out1 != 1, "res1", "got %d != exp %d\n", data->out1, 1);
0108     CHECK(data->out2 != 2, "res2", "got %lld != exp %d\n", data->out2, 2);
0109     CHECK(bss->out3 != 3, "res3", "got %d != exp %d\n", (int)bss->out3, 3);
0110     CHECK(bss->out4 != 4, "res4", "got %lld != exp %d\n", bss->out4, 4);
0111     CHECK(bss->out5.a != 5, "res5", "got %d != exp %d\n", bss->out5.a, 5);
0112     CHECK(bss->out5.b != 6, "res6", "got %lld != exp %d\n", bss->out5.b, 6);
0113     CHECK(bss->out6 != 14, "res7", "got %d != exp %d\n", bss->out6, 14);
0114
0115     CHECK(bss->bpf_syscall != kcfg->CONFIG_BPF_SYSCALL, "ext1",
0116           "got %d != exp %d\n", bss->bpf_syscall, kcfg->CONFIG_BPF_SYSCALL);
0117     CHECK(bss->kern_ver != kcfg->LINUX_KERNEL_VERSION, "ext2",
0118           "got %d != exp %d\n", bss->kern_ver, kcfg->LINUX_KERNEL_VERSION);
0119
0120     for (i = 0; i < 4; i++)
0121         ASSERT_EQ(data_dyn->out_dynarr[i], i + 10, "out_dynarr");
0122
0123     ASSERT_EQ(skel->bss->out_mostly_var, 123, "out_mostly_var");
0124
0125     ASSERT_EQ(bss->huge_arr[ARRAY_SIZE(bss->huge_arr) - 1], 123, "huge_arr");
0126
0127     elf_bytes = test_skeleton__elf_bytes(&elf_bytes_sz);
0128     ASSERT_OK_PTR(elf_bytes, "elf_bytes");
0129     ASSERT_GE(elf_bytes_sz, 0, "elf_bytes_sz");
0130
0131 cleanup:
0132     test_skeleton__destroy(skel);
0133 }