125763b3cSThomas Gleixner // SPDX-License-Identifier: GPL-2.0-only
2a3f74617SMartin KaFai Lau /* Copyright (c) 2016 Facebook
3a3f74617SMartin KaFai Lau */
4a3f74617SMartin KaFai Lau #include <linux/unistd.h>
5a3f74617SMartin KaFai Lau #include <linux/bpf.h>
6a3f74617SMartin KaFai Lau
7a3f74617SMartin KaFai Lau #include <stdio.h>
8a3f74617SMartin KaFai Lau #include <stdint.h>
9a3f74617SMartin KaFai Lau #include <unistd.h>
10a3f74617SMartin KaFai Lau #include <string.h>
11a3f74617SMartin KaFai Lau #include <errno.h>
12a3f74617SMartin KaFai Lau #include <fcntl.h>
13a3f74617SMartin KaFai Lau
142bf3e2efSJakub Kicinski #include <bpf/bpf.h>
15a3f74617SMartin KaFai Lau
usage(void)16a3f74617SMartin KaFai Lau static void usage(void)
17a3f74617SMartin KaFai Lau {
18a3f74617SMartin KaFai Lau printf("Usage: test_cgrp2_array_pin [...]\n");
19a3f74617SMartin KaFai Lau printf(" -F <file> File to pin an BPF cgroup array\n");
20a3f74617SMartin KaFai Lau printf(" -U <file> Update an already pinned BPF cgroup array\n");
21a3f74617SMartin KaFai Lau printf(" -v <value> Full path of the cgroup2\n");
22a3f74617SMartin KaFai Lau printf(" -h Display this help\n");
23a3f74617SMartin KaFai Lau }
24a3f74617SMartin KaFai Lau
main(int argc,char ** argv)25a3f74617SMartin KaFai Lau int main(int argc, char **argv)
26a3f74617SMartin KaFai Lau {
27a3f74617SMartin KaFai Lau const char *pinned_file = NULL, *cg2 = NULL;
28a3f74617SMartin KaFai Lau int create_array = 1;
29a3f74617SMartin KaFai Lau int array_key = 0;
30a3f74617SMartin KaFai Lau int array_fd = -1;
31a3f74617SMartin KaFai Lau int cg2_fd = -1;
32a3f74617SMartin KaFai Lau int ret = -1;
33a3f74617SMartin KaFai Lau int opt;
34a3f74617SMartin KaFai Lau
35a3f74617SMartin KaFai Lau while ((opt = getopt(argc, argv, "F:U:v:")) != -1) {
36a3f74617SMartin KaFai Lau switch (opt) {
37a3f74617SMartin KaFai Lau /* General args */
38a3f74617SMartin KaFai Lau case 'F':
39a3f74617SMartin KaFai Lau pinned_file = optarg;
40a3f74617SMartin KaFai Lau break;
41a3f74617SMartin KaFai Lau case 'U':
42a3f74617SMartin KaFai Lau pinned_file = optarg;
43a3f74617SMartin KaFai Lau create_array = 0;
44a3f74617SMartin KaFai Lau break;
45a3f74617SMartin KaFai Lau case 'v':
46a3f74617SMartin KaFai Lau cg2 = optarg;
47a3f74617SMartin KaFai Lau break;
48a3f74617SMartin KaFai Lau default:
49a3f74617SMartin KaFai Lau usage();
50a3f74617SMartin KaFai Lau goto out;
51a3f74617SMartin KaFai Lau }
52a3f74617SMartin KaFai Lau }
53a3f74617SMartin KaFai Lau
54a3f74617SMartin KaFai Lau if (!cg2 || !pinned_file) {
55a3f74617SMartin KaFai Lau usage();
56a3f74617SMartin KaFai Lau goto out;
57a3f74617SMartin KaFai Lau }
58a3f74617SMartin KaFai Lau
59a3f74617SMartin KaFai Lau cg2_fd = open(cg2, O_RDONLY);
60a3f74617SMartin KaFai Lau if (cg2_fd < 0) {
61a3f74617SMartin KaFai Lau fprintf(stderr, "open(%s,...): %s(%d)\n",
62a3f74617SMartin KaFai Lau cg2, strerror(errno), errno);
63a3f74617SMartin KaFai Lau goto out;
64a3f74617SMartin KaFai Lau }
65a3f74617SMartin KaFai Lau
66a3f74617SMartin KaFai Lau if (create_array) {
67*c58f9815SAndrii Nakryiko array_fd = bpf_map_create(BPF_MAP_TYPE_CGROUP_ARRAY, NULL,
68a3f74617SMartin KaFai Lau sizeof(uint32_t), sizeof(uint32_t),
69*c58f9815SAndrii Nakryiko 1, NULL);
70a3f74617SMartin KaFai Lau if (array_fd < 0) {
71a3f74617SMartin KaFai Lau fprintf(stderr,
72a3f74617SMartin KaFai Lau "bpf_create_map(BPF_MAP_TYPE_CGROUP_ARRAY,...): %s(%d)\n",
73a3f74617SMartin KaFai Lau strerror(errno), errno);
74a3f74617SMartin KaFai Lau goto out;
75a3f74617SMartin KaFai Lau }
76a3f74617SMartin KaFai Lau } else {
77a3f74617SMartin KaFai Lau array_fd = bpf_obj_get(pinned_file);
78a3f74617SMartin KaFai Lau if (array_fd < 0) {
79a3f74617SMartin KaFai Lau fprintf(stderr, "bpf_obj_get(%s): %s(%d)\n",
80a3f74617SMartin KaFai Lau pinned_file, strerror(errno), errno);
81a3f74617SMartin KaFai Lau goto out;
82a3f74617SMartin KaFai Lau }
83a3f74617SMartin KaFai Lau }
84a3f74617SMartin KaFai Lau
85d40fc181SJoe Stringer ret = bpf_map_update_elem(array_fd, &array_key, &cg2_fd, 0);
86a3f74617SMartin KaFai Lau if (ret) {
87d40fc181SJoe Stringer perror("bpf_map_update_elem");
88a3f74617SMartin KaFai Lau goto out;
89a3f74617SMartin KaFai Lau }
90a3f74617SMartin KaFai Lau
91a3f74617SMartin KaFai Lau if (create_array) {
92a3f74617SMartin KaFai Lau ret = bpf_obj_pin(array_fd, pinned_file);
93a3f74617SMartin KaFai Lau if (ret) {
94a3f74617SMartin KaFai Lau fprintf(stderr, "bpf_obj_pin(..., %s): %s(%d)\n",
95a3f74617SMartin KaFai Lau pinned_file, strerror(errno), errno);
96a3f74617SMartin KaFai Lau goto out;
97a3f74617SMartin KaFai Lau }
98a3f74617SMartin KaFai Lau }
99a3f74617SMartin KaFai Lau
100a3f74617SMartin KaFai Lau out:
101a3f74617SMartin KaFai Lau if (array_fd != -1)
102a3f74617SMartin KaFai Lau close(array_fd);
103a3f74617SMartin KaFai Lau if (cg2_fd != -1)
104a3f74617SMartin KaFai Lau close(cg2_fd);
105a3f74617SMartin KaFai Lau return ret;
106a3f74617SMartin KaFai Lau }
107