metadata.c (3525B)
1// SPDX-License-Identifier: GPL-2.0-only 2 3/* 4 * Copyright 2020 Google LLC. 5 */ 6 7#include <test_progs.h> 8#include <cgroup_helpers.h> 9#include <network_helpers.h> 10 11#include "metadata_unused.skel.h" 12#include "metadata_used.skel.h" 13 14static int duration; 15 16static int prog_holds_map(int prog_fd, int map_fd) 17{ 18 struct bpf_prog_info prog_info = {}; 19 struct bpf_prog_info map_info = {}; 20 __u32 prog_info_len; 21 __u32 map_info_len; 22 __u32 *map_ids; 23 int nr_maps; 24 int ret; 25 int i; 26 27 map_info_len = sizeof(map_info); 28 ret = bpf_obj_get_info_by_fd(map_fd, &map_info, &map_info_len); 29 if (ret) 30 return -errno; 31 32 prog_info_len = sizeof(prog_info); 33 ret = bpf_obj_get_info_by_fd(prog_fd, &prog_info, &prog_info_len); 34 if (ret) 35 return -errno; 36 37 map_ids = calloc(prog_info.nr_map_ids, sizeof(__u32)); 38 if (!map_ids) 39 return -ENOMEM; 40 41 nr_maps = prog_info.nr_map_ids; 42 memset(&prog_info, 0, sizeof(prog_info)); 43 prog_info.nr_map_ids = nr_maps; 44 prog_info.map_ids = ptr_to_u64(map_ids); 45 prog_info_len = sizeof(prog_info); 46 47 ret = bpf_obj_get_info_by_fd(prog_fd, &prog_info, &prog_info_len); 48 if (ret) { 49 ret = -errno; 50 goto free_map_ids; 51 } 52 53 ret = -ENOENT; 54 for (i = 0; i < prog_info.nr_map_ids; i++) { 55 if (map_ids[i] == map_info.id) { 56 ret = 0; 57 break; 58 } 59 } 60 61free_map_ids: 62 free(map_ids); 63 return ret; 64} 65 66static void test_metadata_unused(void) 67{ 68 struct metadata_unused *obj; 69 int err; 70 71 obj = metadata_unused__open_and_load(); 72 if (CHECK(!obj, "skel-load", "errno %d", errno)) 73 return; 74 75 err = prog_holds_map(bpf_program__fd(obj->progs.prog), 76 bpf_map__fd(obj->maps.rodata)); 77 if (CHECK(err, "prog-holds-rodata", "errno: %d", err)) 78 return; 79 80 /* Assert that we can access the metadata in skel and the values are 81 * what we expect. 82 */ 83 if (CHECK(strncmp(obj->rodata->bpf_metadata_a, "foo", 84 sizeof(obj->rodata->bpf_metadata_a)), 85 "bpf_metadata_a", "expected \"foo\", value differ")) 86 goto close_bpf_object; 87 if (CHECK(obj->rodata->bpf_metadata_b != 1, "bpf_metadata_b", 88 "expected 1, got %d", obj->rodata->bpf_metadata_b)) 89 goto close_bpf_object; 90 91 /* Assert that binding metadata map to prog again succeeds. */ 92 err = bpf_prog_bind_map(bpf_program__fd(obj->progs.prog), 93 bpf_map__fd(obj->maps.rodata), NULL); 94 CHECK(err, "rebind_map", "errno %d, expected 0", errno); 95 96close_bpf_object: 97 metadata_unused__destroy(obj); 98} 99 100static void test_metadata_used(void) 101{ 102 struct metadata_used *obj; 103 int err; 104 105 obj = metadata_used__open_and_load(); 106 if (CHECK(!obj, "skel-load", "errno %d", errno)) 107 return; 108 109 err = prog_holds_map(bpf_program__fd(obj->progs.prog), 110 bpf_map__fd(obj->maps.rodata)); 111 if (CHECK(err, "prog-holds-rodata", "errno: %d", err)) 112 return; 113 114 /* Assert that we can access the metadata in skel and the values are 115 * what we expect. 116 */ 117 if (CHECK(strncmp(obj->rodata->bpf_metadata_a, "bar", 118 sizeof(obj->rodata->bpf_metadata_a)), 119 "metadata_a", "expected \"bar\", value differ")) 120 goto close_bpf_object; 121 if (CHECK(obj->rodata->bpf_metadata_b != 2, "metadata_b", 122 "expected 2, got %d", obj->rodata->bpf_metadata_b)) 123 goto close_bpf_object; 124 125 /* Assert that binding metadata map to prog again succeeds. */ 126 err = bpf_prog_bind_map(bpf_program__fd(obj->progs.prog), 127 bpf_map__fd(obj->maps.rodata), NULL); 128 CHECK(err, "rebind_map", "errno %d, expected 0", errno); 129 130close_bpf_object: 131 metadata_used__destroy(obj); 132} 133 134void test_metadata(void) 135{ 136 if (test__start_subtest("unused")) 137 test_metadata_unused(); 138 139 if (test__start_subtest("used")) 140 test_metadata_used(); 141}