forked from Qortal/Brooklyn
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
106 lines
2.1 KiB
106 lines
2.1 KiB
// SPDX-License-Identifier: GPL-2.0-only |
|
/* Copyright (c) 2016 Facebook |
|
*/ |
|
#include <linux/unistd.h> |
|
#include <linux/bpf.h> |
|
|
|
#include <stdio.h> |
|
#include <stdint.h> |
|
#include <unistd.h> |
|
#include <string.h> |
|
#include <errno.h> |
|
#include <fcntl.h> |
|
|
|
#include <bpf/bpf.h> |
|
|
|
static void usage(void) |
|
{ |
|
printf("Usage: test_cgrp2_array_pin [...]\n"); |
|
printf(" -F <file> File to pin an BPF cgroup array\n"); |
|
printf(" -U <file> Update an already pinned BPF cgroup array\n"); |
|
printf(" -v <value> Full path of the cgroup2\n"); |
|
printf(" -h Display this help\n"); |
|
} |
|
|
|
int main(int argc, char **argv) |
|
{ |
|
const char *pinned_file = NULL, *cg2 = NULL; |
|
int create_array = 1; |
|
int array_key = 0; |
|
int array_fd = -1; |
|
int cg2_fd = -1; |
|
int ret = -1; |
|
int opt; |
|
|
|
while ((opt = getopt(argc, argv, "F:U:v:")) != -1) { |
|
switch (opt) { |
|
/* General args */ |
|
case 'F': |
|
pinned_file = optarg; |
|
break; |
|
case 'U': |
|
pinned_file = optarg; |
|
create_array = 0; |
|
break; |
|
case 'v': |
|
cg2 = optarg; |
|
break; |
|
default: |
|
usage(); |
|
goto out; |
|
} |
|
} |
|
|
|
if (!cg2 || !pinned_file) { |
|
usage(); |
|
goto out; |
|
} |
|
|
|
cg2_fd = open(cg2, O_RDONLY); |
|
if (cg2_fd < 0) { |
|
fprintf(stderr, "open(%s,...): %s(%d)\n", |
|
cg2, strerror(errno), errno); |
|
goto out; |
|
} |
|
|
|
if (create_array) { |
|
array_fd = bpf_create_map(BPF_MAP_TYPE_CGROUP_ARRAY, |
|
sizeof(uint32_t), sizeof(uint32_t), |
|
1, 0); |
|
if (array_fd < 0) { |
|
fprintf(stderr, |
|
"bpf_create_map(BPF_MAP_TYPE_CGROUP_ARRAY,...): %s(%d)\n", |
|
strerror(errno), errno); |
|
goto out; |
|
} |
|
} else { |
|
array_fd = bpf_obj_get(pinned_file); |
|
if (array_fd < 0) { |
|
fprintf(stderr, "bpf_obj_get(%s): %s(%d)\n", |
|
pinned_file, strerror(errno), errno); |
|
goto out; |
|
} |
|
} |
|
|
|
ret = bpf_map_update_elem(array_fd, &array_key, &cg2_fd, 0); |
|
if (ret) { |
|
perror("bpf_map_update_elem"); |
|
goto out; |
|
} |
|
|
|
if (create_array) { |
|
ret = bpf_obj_pin(array_fd, pinned_file); |
|
if (ret) { |
|
fprintf(stderr, "bpf_obj_pin(..., %s): %s(%d)\n", |
|
pinned_file, strerror(errno), errno); |
|
goto out; |
|
} |
|
} |
|
|
|
out: |
|
if (array_fd != -1) |
|
close(array_fd); |
|
if (cg2_fd != -1) |
|
close(cg2_fd); |
|
return ret; |
|
}
|
|
|