1// SPDX-License-Identifier: GPL-2.0
  2/* 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}