Orange Pi5 kernel

Deprecated Linux kernel 5.10.110 for OrangePi 5/5B/5+ boards

3 Commits   0 Branches   0 Tags
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300   1) // SPDX-License-Identifier: (GPL-2.0-only OR BSD-2-Clause)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300   2) /* Copyright (C) 2017-2018 Netronome Systems, Inc. */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300   3) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300   4) #define _GNU_SOURCE
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300   5) #include <ctype.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300   6) #include <errno.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300   7) #include <fcntl.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300   8) #include <ftw.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300   9) #include <libgen.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  10) #include <mntent.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  11) #include <stdbool.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  12) #include <stdio.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  13) #include <stdlib.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  14) #include <string.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  15) #include <unistd.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  16) #include <linux/limits.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  17) #include <linux/magic.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  18) #include <net/if.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  19) #include <sys/mount.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  20) #include <sys/resource.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  21) #include <sys/stat.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  22) #include <sys/vfs.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  23) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  24) #include <bpf/bpf.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  25) #include <bpf/libbpf.h> /* libbpf_num_possible_cpus */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  26) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  27) #include "main.h"
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  28) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  29) #ifndef BPF_FS_MAGIC
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  30) #define BPF_FS_MAGIC		0xcafe4a11
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  31) #endif
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  32) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  33) const char * const attach_type_name[__MAX_BPF_ATTACH_TYPE] = {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  34) 	[BPF_CGROUP_INET_INGRESS]	= "ingress",
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  35) 	[BPF_CGROUP_INET_EGRESS]	= "egress",
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  36) 	[BPF_CGROUP_INET_SOCK_CREATE]	= "sock_create",
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  37) 	[BPF_CGROUP_INET_SOCK_RELEASE]	= "sock_release",
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  38) 	[BPF_CGROUP_SOCK_OPS]		= "sock_ops",
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  39) 	[BPF_CGROUP_DEVICE]		= "device",
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  40) 	[BPF_CGROUP_INET4_BIND]		= "bind4",
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  41) 	[BPF_CGROUP_INET6_BIND]		= "bind6",
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  42) 	[BPF_CGROUP_INET4_CONNECT]	= "connect4",
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  43) 	[BPF_CGROUP_INET6_CONNECT]	= "connect6",
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  44) 	[BPF_CGROUP_INET4_POST_BIND]	= "post_bind4",
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  45) 	[BPF_CGROUP_INET6_POST_BIND]	= "post_bind6",
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  46) 	[BPF_CGROUP_INET4_GETPEERNAME]	= "getpeername4",
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  47) 	[BPF_CGROUP_INET6_GETPEERNAME]	= "getpeername6",
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  48) 	[BPF_CGROUP_INET4_GETSOCKNAME]	= "getsockname4",
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  49) 	[BPF_CGROUP_INET6_GETSOCKNAME]	= "getsockname6",
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  50) 	[BPF_CGROUP_UDP4_SENDMSG]	= "sendmsg4",
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  51) 	[BPF_CGROUP_UDP6_SENDMSG]	= "sendmsg6",
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  52) 	[BPF_CGROUP_SYSCTL]		= "sysctl",
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  53) 	[BPF_CGROUP_UDP4_RECVMSG]	= "recvmsg4",
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  54) 	[BPF_CGROUP_UDP6_RECVMSG]	= "recvmsg6",
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  55) 	[BPF_CGROUP_GETSOCKOPT]		= "getsockopt",
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  56) 	[BPF_CGROUP_SETSOCKOPT]		= "setsockopt",
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  57) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  58) 	[BPF_SK_SKB_STREAM_PARSER]	= "sk_skb_stream_parser",
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  59) 	[BPF_SK_SKB_STREAM_VERDICT]	= "sk_skb_stream_verdict",
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  60) 	[BPF_SK_MSG_VERDICT]		= "sk_msg_verdict",
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  61) 	[BPF_LIRC_MODE2]		= "lirc_mode2",
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  62) 	[BPF_FLOW_DISSECTOR]		= "flow_dissector",
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  63) 	[BPF_TRACE_RAW_TP]		= "raw_tp",
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  64) 	[BPF_TRACE_FENTRY]		= "fentry",
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  65) 	[BPF_TRACE_FEXIT]		= "fexit",
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  66) 	[BPF_MODIFY_RETURN]		= "mod_ret",
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  67) 	[BPF_LSM_MAC]			= "lsm_mac",
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  68) 	[BPF_SK_LOOKUP]			= "sk_lookup",
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  69) };
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  70) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  71) void p_err(const char *fmt, ...)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  72) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  73) 	va_list ap;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  74) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  75) 	va_start(ap, fmt);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  76) 	if (json_output) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  77) 		jsonw_start_object(json_wtr);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  78) 		jsonw_name(json_wtr, "error");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  79) 		jsonw_vprintf_enquote(json_wtr, fmt, ap);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  80) 		jsonw_end_object(json_wtr);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  81) 	} else {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  82) 		fprintf(stderr, "Error: ");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  83) 		vfprintf(stderr, fmt, ap);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  84) 		fprintf(stderr, "\n");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  85) 	}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  86) 	va_end(ap);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  87) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  88) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  89) void p_info(const char *fmt, ...)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  90) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  91) 	va_list ap;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  92) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  93) 	if (json_output)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  94) 		return;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  95) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  96) 	va_start(ap, fmt);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  97) 	vfprintf(stderr, fmt, ap);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  98) 	fprintf(stderr, "\n");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  99) 	va_end(ap);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 100) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 101) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 102) static bool is_bpffs(char *path)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 103) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 104) 	struct statfs st_fs;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 105) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 106) 	if (statfs(path, &st_fs) < 0)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 107) 		return false;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 108) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 109) 	return (unsigned long)st_fs.f_type == BPF_FS_MAGIC;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 110) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 111) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 112) void set_max_rlimit(void)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 113) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 114) 	struct rlimit rinf = { RLIM_INFINITY, RLIM_INFINITY };
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 115) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 116) 	setrlimit(RLIMIT_MEMLOCK, &rinf);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 117) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 118) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 119) static int
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 120) mnt_fs(const char *target, const char *type, char *buff, size_t bufflen)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 121) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 122) 	bool bind_done = false;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 123) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 124) 	while (mount("", target, "none", MS_PRIVATE | MS_REC, NULL)) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 125) 		if (errno != EINVAL || bind_done) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 126) 			snprintf(buff, bufflen,
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 127) 				 "mount --make-private %s failed: %s",
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 128) 				 target, strerror(errno));
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 129) 			return -1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 130) 		}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 131) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 132) 		if (mount(target, target, "none", MS_BIND, NULL)) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 133) 			snprintf(buff, bufflen,
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 134) 				 "mount --bind %s %s failed: %s",
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 135) 				 target, target, strerror(errno));
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 136) 			return -1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 137) 		}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 138) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 139) 		bind_done = true;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 140) 	}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 141) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 142) 	if (mount(type, target, type, 0, "mode=0700")) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 143) 		snprintf(buff, bufflen, "mount -t %s %s %s failed: %s",
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 144) 			 type, type, target, strerror(errno));
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 145) 		return -1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 146) 	}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 147) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 148) 	return 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 149) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 150) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 151) int mount_tracefs(const char *target)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 152) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 153) 	char err_str[ERR_MAX_LEN];
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 154) 	int err;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 155) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 156) 	err = mnt_fs(target, "tracefs", err_str, ERR_MAX_LEN);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 157) 	if (err) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 158) 		err_str[ERR_MAX_LEN - 1] = '\0';
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 159) 		p_err("can't mount tracefs: %s", err_str);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 160) 	}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 161) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 162) 	return err;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 163) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 164) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 165) int open_obj_pinned(const char *path, bool quiet)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 166) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 167) 	char *pname;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 168) 	int fd = -1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 169) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 170) 	pname = strdup(path);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 171) 	if (!pname) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 172) 		if (!quiet)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 173) 			p_err("mem alloc failed");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 174) 		goto out_ret;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 175) 	}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 176) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 177) 	fd = bpf_obj_get(pname);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 178) 	if (fd < 0) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 179) 		if (!quiet)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 180) 			p_err("bpf obj get (%s): %s", pname,
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 181) 			      errno == EACCES && !is_bpffs(dirname(pname)) ?
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 182) 			    "directory not in bpf file system (bpffs)" :
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 183) 			    strerror(errno));
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 184) 		goto out_free;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 185) 	}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 186) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 187) out_free:
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 188) 	free(pname);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 189) out_ret:
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 190) 	return fd;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 191) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 192) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 193) int open_obj_pinned_any(const char *path, enum bpf_obj_type exp_type)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 194) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 195) 	enum bpf_obj_type type;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 196) 	int fd;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 197) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 198) 	fd = open_obj_pinned(path, false);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 199) 	if (fd < 0)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 200) 		return -1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 201) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 202) 	type = get_fd_type(fd);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 203) 	if (type < 0) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 204) 		close(fd);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 205) 		return type;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 206) 	}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 207) 	if (type != exp_type) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 208) 		p_err("incorrect object type: %s", get_fd_type_name(type));
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 209) 		close(fd);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 210) 		return -1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 211) 	}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 212) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 213) 	return fd;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 214) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 215) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 216) int mount_bpffs_for_pin(const char *name)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 217) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 218) 	char err_str[ERR_MAX_LEN];
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 219) 	char *file;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 220) 	char *dir;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 221) 	int err = 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 222) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 223) 	file = malloc(strlen(name) + 1);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 224) 	if (!file) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 225) 		p_err("mem alloc failed");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 226) 		return -1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 227) 	}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 228) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 229) 	strcpy(file, name);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 230) 	dir = dirname(file);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 231) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 232) 	if (is_bpffs(dir))
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 233) 		/* nothing to do if already mounted */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 234) 		goto out_free;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 235) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 236) 	if (block_mount) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 237) 		p_err("no BPF file system found, not mounting it due to --nomount option");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 238) 		err = -1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 239) 		goto out_free;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 240) 	}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 241) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 242) 	err = mnt_fs(dir, "bpf", err_str, ERR_MAX_LEN);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 243) 	if (err) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 244) 		err_str[ERR_MAX_LEN - 1] = '\0';
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 245) 		p_err("can't mount BPF file system to pin the object (%s): %s",
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 246) 		      name, err_str);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 247) 	}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 248) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 249) out_free:
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 250) 	free(file);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 251) 	return err;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 252) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 253) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 254) int do_pin_fd(int fd, const char *name)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 255) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 256) 	int err;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 257) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 258) 	err = mount_bpffs_for_pin(name);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 259) 	if (err)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 260) 		return err;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 261) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 262) 	err = bpf_obj_pin(fd, name);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 263) 	if (err)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 264) 		p_err("can't pin the object (%s): %s", name, strerror(errno));
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 265) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 266) 	return err;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 267) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 268) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 269) int do_pin_any(int argc, char **argv, int (*get_fd)(int *, char ***))
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 270) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 271) 	int err;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 272) 	int fd;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 273) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 274) 	fd = get_fd(&argc, &argv);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 275) 	if (fd < 0)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 276) 		return fd;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 277) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 278) 	err = do_pin_fd(fd, *argv);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 279) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 280) 	close(fd);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 281) 	return err;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 282) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 283) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 284) const char *get_fd_type_name(enum bpf_obj_type type)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 285) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 286) 	static const char * const names[] = {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 287) 		[BPF_OBJ_UNKNOWN]	= "unknown",
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 288) 		[BPF_OBJ_PROG]		= "prog",
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 289) 		[BPF_OBJ_MAP]		= "map",
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 290) 	};
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 291) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 292) 	if (type < 0 || type >= ARRAY_SIZE(names) || !names[type])
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 293) 		return names[BPF_OBJ_UNKNOWN];
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 294) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 295) 	return names[type];
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 296) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 297) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 298) int get_fd_type(int fd)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 299) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 300) 	char path[PATH_MAX];
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 301) 	char buf[512];
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 302) 	ssize_t n;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 303) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 304) 	snprintf(path, sizeof(path), "/proc/self/fd/%d", fd);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 305) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 306) 	n = readlink(path, buf, sizeof(buf));
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 307) 	if (n < 0) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 308) 		p_err("can't read link type: %s", strerror(errno));
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 309) 		return -1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 310) 	}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 311) 	if (n == sizeof(path)) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 312) 		p_err("can't read link type: path too long!");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 313) 		return -1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 314) 	}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 315) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 316) 	if (strstr(buf, "bpf-map"))
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 317) 		return BPF_OBJ_MAP;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 318) 	else if (strstr(buf, "bpf-prog"))
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 319) 		return BPF_OBJ_PROG;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 320) 	else if (strstr(buf, "bpf-link"))
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 321) 		return BPF_OBJ_LINK;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 322) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 323) 	return BPF_OBJ_UNKNOWN;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 324) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 325) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 326) char *get_fdinfo(int fd, const char *key)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 327) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 328) 	char path[PATH_MAX];
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 329) 	char *line = NULL;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 330) 	size_t line_n = 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 331) 	ssize_t n;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 332) 	FILE *fdi;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 333) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 334) 	snprintf(path, sizeof(path), "/proc/self/fdinfo/%d", fd);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 335) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 336) 	fdi = fopen(path, "r");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 337) 	if (!fdi)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 338) 		return NULL;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 339) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 340) 	while ((n = getline(&line, &line_n, fdi)) > 0) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 341) 		char *value;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 342) 		int len;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 343) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 344) 		if (!strstr(line, key))
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 345) 			continue;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 346) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 347) 		fclose(fdi);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 348) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 349) 		value = strchr(line, '\t');
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 350) 		if (!value || !value[1]) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 351) 			free(line);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 352) 			return NULL;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 353) 		}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 354) 		value++;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 355) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 356) 		len = strlen(value);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 357) 		memmove(line, value, len);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 358) 		line[len - 1] = '\0';
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 359) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 360) 		return line;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 361) 	}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 362) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 363) 	free(line);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 364) 	fclose(fdi);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 365) 	return NULL;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 366) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 367) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 368) void print_data_json(uint8_t *data, size_t len)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 369) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 370) 	unsigned int i;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 371) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 372) 	jsonw_start_array(json_wtr);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 373) 	for (i = 0; i < len; i++)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 374) 		jsonw_printf(json_wtr, "%d", data[i]);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 375) 	jsonw_end_array(json_wtr);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 376) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 377) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 378) void print_hex_data_json(uint8_t *data, size_t len)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 379) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 380) 	unsigned int i;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 381) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 382) 	jsonw_start_array(json_wtr);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 383) 	for (i = 0; i < len; i++)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 384) 		jsonw_printf(json_wtr, "\"0x%02hhx\"", data[i]);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 385) 	jsonw_end_array(json_wtr);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 386) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 387) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 388) /* extra params for nftw cb */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 389) static struct pinned_obj_table *build_fn_table;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 390) static enum bpf_obj_type build_fn_type;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 391) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 392) static int do_build_table_cb(const char *fpath, const struct stat *sb,
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 393) 			     int typeflag, struct FTW *ftwbuf)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 394) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 395) 	struct bpf_prog_info pinned_info;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 396) 	__u32 len = sizeof(pinned_info);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 397) 	struct pinned_obj *obj_node;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 398) 	enum bpf_obj_type objtype;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 399) 	int fd, err = 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 400) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 401) 	if (typeflag != FTW_F)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 402) 		goto out_ret;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 403) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 404) 	fd = open_obj_pinned(fpath, true);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 405) 	if (fd < 0)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 406) 		goto out_ret;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 407) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 408) 	objtype = get_fd_type(fd);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 409) 	if (objtype != build_fn_type)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 410) 		goto out_close;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 411) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 412) 	memset(&pinned_info, 0, sizeof(pinned_info));
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 413) 	if (bpf_obj_get_info_by_fd(fd, &pinned_info, &len))
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 414) 		goto out_close;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 415) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 416) 	obj_node = calloc(1, sizeof(*obj_node));
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 417) 	if (!obj_node) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 418) 		err = -1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 419) 		goto out_close;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 420) 	}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 421) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 422) 	obj_node->id = pinned_info.id;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 423) 	obj_node->path = strdup(fpath);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 424) 	if (!obj_node->path) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 425) 		err = -1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 426) 		free(obj_node);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 427) 		goto out_close;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 428) 	}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 429) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 430) 	hash_add(build_fn_table->table, &obj_node->hash, obj_node->id);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 431) out_close:
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 432) 	close(fd);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 433) out_ret:
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 434) 	return err;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 435) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 436) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 437) int build_pinned_obj_table(struct pinned_obj_table *tab,
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 438) 			   enum bpf_obj_type type)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 439) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 440) 	struct mntent *mntent = NULL;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 441) 	FILE *mntfile = NULL;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 442) 	int flags = FTW_PHYS;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 443) 	int nopenfd = 16;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 444) 	int err = 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 445) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 446) 	mntfile = setmntent("/proc/mounts", "r");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 447) 	if (!mntfile)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 448) 		return -1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 449) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 450) 	build_fn_table = tab;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 451) 	build_fn_type = type;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 452) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 453) 	while ((mntent = getmntent(mntfile))) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 454) 		char *path = mntent->mnt_dir;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 455) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 456) 		if (strncmp(mntent->mnt_type, "bpf", 3) != 0)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 457) 			continue;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 458) 		err = nftw(path, do_build_table_cb, nopenfd, flags);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 459) 		if (err)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 460) 			break;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 461) 	}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 462) 	fclose(mntfile);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 463) 	return err;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 464) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 465) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 466) void delete_pinned_obj_table(struct pinned_obj_table *tab)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 467) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 468) 	struct pinned_obj *obj;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 469) 	struct hlist_node *tmp;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 470) 	unsigned int bkt;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 471) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 472) 	hash_for_each_safe(tab->table, bkt, tmp, obj, hash) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 473) 		hash_del(&obj->hash);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 474) 		free(obj->path);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 475) 		free(obj);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 476) 	}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 477) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 478) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 479) unsigned int get_page_size(void)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 480) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 481) 	static int result;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 482) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 483) 	if (!result)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 484) 		result = getpagesize();
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 485) 	return result;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 486) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 487) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 488) unsigned int get_possible_cpus(void)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 489) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 490) 	int cpus = libbpf_num_possible_cpus();
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 491) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 492) 	if (cpus < 0) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 493) 		p_err("Can't get # of possible cpus: %s", strerror(-cpus));
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 494) 		exit(-1);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 495) 	}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 496) 	return cpus;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 497) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 498) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 499) static char *
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 500) ifindex_to_name_ns(__u32 ifindex, __u32 ns_dev, __u32 ns_ino, char *buf)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 501) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 502) 	struct stat st;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 503) 	int err;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 504) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 505) 	err = stat("/proc/self/ns/net", &st);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 506) 	if (err) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 507) 		p_err("Can't stat /proc/self: %s", strerror(errno));
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 508) 		return NULL;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 509) 	}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 510) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 511) 	if (st.st_dev != ns_dev || st.st_ino != ns_ino)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 512) 		return NULL;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 513) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 514) 	return if_indextoname(ifindex, buf);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 515) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 516) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 517) static int read_sysfs_hex_int(char *path)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 518) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 519) 	char vendor_id_buf[8];
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 520) 	int len;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 521) 	int fd;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 522) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 523) 	fd = open(path, O_RDONLY);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 524) 	if (fd < 0) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 525) 		p_err("Can't open %s: %s", path, strerror(errno));
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 526) 		return -1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 527) 	}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 528) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 529) 	len = read(fd, vendor_id_buf, sizeof(vendor_id_buf));
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 530) 	close(fd);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 531) 	if (len < 0) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 532) 		p_err("Can't read %s: %s", path, strerror(errno));
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 533) 		return -1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 534) 	}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 535) 	if (len >= (int)sizeof(vendor_id_buf)) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 536) 		p_err("Value in %s too long", path);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 537) 		return -1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 538) 	}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 539) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 540) 	vendor_id_buf[len] = 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 541) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 542) 	return strtol(vendor_id_buf, NULL, 0);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 543) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 544) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 545) static int read_sysfs_netdev_hex_int(char *devname, const char *entry_name)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 546) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 547) 	char full_path[64];
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 548) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 549) 	snprintf(full_path, sizeof(full_path), "/sys/class/net/%s/device/%s",
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 550) 		 devname, entry_name);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 551) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 552) 	return read_sysfs_hex_int(full_path);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 553) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 554) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 555) const char *
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 556) ifindex_to_bfd_params(__u32 ifindex, __u64 ns_dev, __u64 ns_ino,
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 557) 		      const char **opt)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 558) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 559) 	char devname[IF_NAMESIZE];
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 560) 	int vendor_id;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 561) 	int device_id;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 562) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 563) 	if (!ifindex_to_name_ns(ifindex, ns_dev, ns_ino, devname)) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 564) 		p_err("Can't get net device name for ifindex %d: %s", ifindex,
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 565) 		      strerror(errno));
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 566) 		return NULL;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 567) 	}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 568) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 569) 	vendor_id = read_sysfs_netdev_hex_int(devname, "vendor");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 570) 	if (vendor_id < 0) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 571) 		p_err("Can't get device vendor id for %s", devname);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 572) 		return NULL;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 573) 	}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 574) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 575) 	switch (vendor_id) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 576) 	case 0x19ee:
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 577) 		device_id = read_sysfs_netdev_hex_int(devname, "device");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 578) 		if (device_id != 0x4000 &&
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 579) 		    device_id != 0x6000 &&
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 580) 		    device_id != 0x6003)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 581) 			p_info("Unknown NFP device ID, assuming it is NFP-6xxx arch");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 582) 		*opt = "ctx4";
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 583) 		return "NFP-6xxx";
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 584) 	default:
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 585) 		p_err("Can't get bfd arch name for device vendor id 0x%04x",
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 586) 		      vendor_id);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 587) 		return NULL;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 588) 	}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 589) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 590) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 591) void print_dev_plain(__u32 ifindex, __u64 ns_dev, __u64 ns_inode)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 592) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 593) 	char name[IF_NAMESIZE];
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 594) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 595) 	if (!ifindex)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 596) 		return;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 597) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 598) 	printf("  offloaded_to ");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 599) 	if (ifindex_to_name_ns(ifindex, ns_dev, ns_inode, name))
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 600) 		printf("%s", name);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 601) 	else
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 602) 		printf("ifindex %u ns_dev %llu ns_ino %llu",
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 603) 		       ifindex, ns_dev, ns_inode);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 604) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 605) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 606) void print_dev_json(__u32 ifindex, __u64 ns_dev, __u64 ns_inode)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 607) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 608) 	char name[IF_NAMESIZE];
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 609) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 610) 	if (!ifindex)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 611) 		return;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 612) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 613) 	jsonw_name(json_wtr, "dev");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 614) 	jsonw_start_object(json_wtr);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 615) 	jsonw_uint_field(json_wtr, "ifindex", ifindex);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 616) 	jsonw_uint_field(json_wtr, "ns_dev", ns_dev);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 617) 	jsonw_uint_field(json_wtr, "ns_inode", ns_inode);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 618) 	if (ifindex_to_name_ns(ifindex, ns_dev, ns_inode, name))
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 619) 		jsonw_string_field(json_wtr, "ifname", name);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 620) 	jsonw_end_object(json_wtr);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 621) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 622) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 623) int parse_u32_arg(int *argc, char ***argv, __u32 *val, const char *what)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 624) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 625) 	char *endptr;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 626) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 627) 	NEXT_ARGP();
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 628) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 629) 	if (*val) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 630) 		p_err("%s already specified", what);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 631) 		return -1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 632) 	}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 633) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 634) 	*val = strtoul(**argv, &endptr, 0);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 635) 	if (*endptr) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 636) 		p_err("can't parse %s as %s", **argv, what);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 637) 		return -1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 638) 	}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 639) 	NEXT_ARGP();
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 640) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 641) 	return 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 642) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 643) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 644) int __printf(2, 0)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 645) print_all_levels(__maybe_unused enum libbpf_print_level level,
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 646) 		 const char *format, va_list args)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 647) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 648) 	return vfprintf(stderr, format, args);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 649) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 650) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 651) static int prog_fd_by_nametag(void *nametag, int **fds, bool tag)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 652) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 653) 	unsigned int id = 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 654) 	int fd, nb_fds = 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 655) 	void *tmp;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 656) 	int err;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 657) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 658) 	while (true) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 659) 		struct bpf_prog_info info = {};
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 660) 		__u32 len = sizeof(info);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 661) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 662) 		err = bpf_prog_get_next_id(id, &id);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 663) 		if (err) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 664) 			if (errno != ENOENT) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 665) 				p_err("%s", strerror(errno));
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 666) 				goto err_close_fds;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 667) 			}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 668) 			return nb_fds;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 669) 		}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 670) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 671) 		fd = bpf_prog_get_fd_by_id(id);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 672) 		if (fd < 0) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 673) 			p_err("can't get prog by id (%u): %s",
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 674) 			      id, strerror(errno));
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 675) 			goto err_close_fds;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 676) 		}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 677) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 678) 		err = bpf_obj_get_info_by_fd(fd, &info, &len);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 679) 		if (err) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 680) 			p_err("can't get prog info (%u): %s",
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 681) 			      id, strerror(errno));
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 682) 			goto err_close_fd;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 683) 		}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 684) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 685) 		if ((tag && memcmp(nametag, info.tag, BPF_TAG_SIZE)) ||
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 686) 		    (!tag && strncmp(nametag, info.name, BPF_OBJ_NAME_LEN))) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 687) 			close(fd);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 688) 			continue;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 689) 		}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 690) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 691) 		if (nb_fds > 0) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 692) 			tmp = realloc(*fds, (nb_fds + 1) * sizeof(int));
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 693) 			if (!tmp) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 694) 				p_err("failed to realloc");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 695) 				goto err_close_fd;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 696) 			}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 697) 			*fds = tmp;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 698) 		}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 699) 		(*fds)[nb_fds++] = fd;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 700) 	}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 701) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 702) err_close_fd:
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 703) 	close(fd);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 704) err_close_fds:
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 705) 	while (--nb_fds >= 0)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 706) 		close((*fds)[nb_fds]);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 707) 	return -1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 708) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 709) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 710) int prog_parse_fds(int *argc, char ***argv, int **fds)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 711) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 712) 	if (is_prefix(**argv, "id")) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 713) 		unsigned int id;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 714) 		char *endptr;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 715) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 716) 		NEXT_ARGP();
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 717) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 718) 		id = strtoul(**argv, &endptr, 0);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 719) 		if (*endptr) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 720) 			p_err("can't parse %s as ID", **argv);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 721) 			return -1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 722) 		}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 723) 		NEXT_ARGP();
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 724) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 725) 		(*fds)[0] = bpf_prog_get_fd_by_id(id);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 726) 		if ((*fds)[0] < 0) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 727) 			p_err("get by id (%u): %s", id, strerror(errno));
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 728) 			return -1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 729) 		}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 730) 		return 1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 731) 	} else if (is_prefix(**argv, "tag")) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 732) 		unsigned char tag[BPF_TAG_SIZE];
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 733) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 734) 		NEXT_ARGP();
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 735) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 736) 		if (sscanf(**argv, BPF_TAG_FMT, tag, tag + 1, tag + 2,
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 737) 			   tag + 3, tag + 4, tag + 5, tag + 6, tag + 7)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 738) 		    != BPF_TAG_SIZE) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 739) 			p_err("can't parse tag");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 740) 			return -1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 741) 		}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 742) 		NEXT_ARGP();
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 743) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 744) 		return prog_fd_by_nametag(tag, fds, true);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 745) 	} else if (is_prefix(**argv, "name")) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 746) 		char *name;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 747) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 748) 		NEXT_ARGP();
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 749) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 750) 		name = **argv;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 751) 		if (strlen(name) > BPF_OBJ_NAME_LEN - 1) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 752) 			p_err("can't parse name");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 753) 			return -1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 754) 		}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 755) 		NEXT_ARGP();
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 756) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 757) 		return prog_fd_by_nametag(name, fds, false);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 758) 	} else if (is_prefix(**argv, "pinned")) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 759) 		char *path;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 760) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 761) 		NEXT_ARGP();
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 762) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 763) 		path = **argv;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 764) 		NEXT_ARGP();
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 765) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 766) 		(*fds)[0] = open_obj_pinned_any(path, BPF_OBJ_PROG);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 767) 		if ((*fds)[0] < 0)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 768) 			return -1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 769) 		return 1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 770) 	}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 771) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 772) 	p_err("expected 'id', 'tag', 'name' or 'pinned', got: '%s'?", **argv);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 773) 	return -1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 774) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 775) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 776) int prog_parse_fd(int *argc, char ***argv)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 777) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 778) 	int *fds = NULL;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 779) 	int nb_fds, fd;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 780) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 781) 	fds = malloc(sizeof(int));
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 782) 	if (!fds) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 783) 		p_err("mem alloc failed");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 784) 		return -1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 785) 	}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 786) 	nb_fds = prog_parse_fds(argc, argv, &fds);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 787) 	if (nb_fds != 1) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 788) 		if (nb_fds > 1) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 789) 			p_err("several programs match this handle");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 790) 			while (nb_fds--)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 791) 				close(fds[nb_fds]);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 792) 		}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 793) 		fd = -1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 794) 		goto exit_free;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 795) 	}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 796) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 797) 	fd = fds[0];
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 798) exit_free:
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 799) 	free(fds);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 800) 	return fd;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 801) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 802) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 803) static int map_fd_by_name(char *name, int **fds)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 804) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 805) 	unsigned int id = 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 806) 	int fd, nb_fds = 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 807) 	void *tmp;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 808) 	int err;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 809) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 810) 	while (true) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 811) 		struct bpf_map_info info = {};
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 812) 		__u32 len = sizeof(info);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 813) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 814) 		err = bpf_map_get_next_id(id, &id);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 815) 		if (err) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 816) 			if (errno != ENOENT) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 817) 				p_err("%s", strerror(errno));
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 818) 				goto err_close_fds;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 819) 			}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 820) 			return nb_fds;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 821) 		}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 822) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 823) 		fd = bpf_map_get_fd_by_id(id);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 824) 		if (fd < 0) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 825) 			p_err("can't get map by id (%u): %s",
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 826) 			      id, strerror(errno));
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 827) 			goto err_close_fds;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 828) 		}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 829) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 830) 		err = bpf_obj_get_info_by_fd(fd, &info, &len);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 831) 		if (err) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 832) 			p_err("can't get map info (%u): %s",
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 833) 			      id, strerror(errno));
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 834) 			goto err_close_fd;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 835) 		}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 836) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 837) 		if (strncmp(name, info.name, BPF_OBJ_NAME_LEN)) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 838) 			close(fd);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 839) 			continue;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 840) 		}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 841) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 842) 		if (nb_fds > 0) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 843) 			tmp = realloc(*fds, (nb_fds + 1) * sizeof(int));
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 844) 			if (!tmp) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 845) 				p_err("failed to realloc");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 846) 				goto err_close_fd;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 847) 			}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 848) 			*fds = tmp;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 849) 		}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 850) 		(*fds)[nb_fds++] = fd;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 851) 	}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 852) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 853) err_close_fd:
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 854) 	close(fd);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 855) err_close_fds:
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 856) 	while (--nb_fds >= 0)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 857) 		close((*fds)[nb_fds]);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 858) 	return -1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 859) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 860) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 861) int map_parse_fds(int *argc, char ***argv, int **fds)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 862) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 863) 	if (is_prefix(**argv, "id")) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 864) 		unsigned int id;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 865) 		char *endptr;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 866) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 867) 		NEXT_ARGP();
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 868) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 869) 		id = strtoul(**argv, &endptr, 0);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 870) 		if (*endptr) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 871) 			p_err("can't parse %s as ID", **argv);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 872) 			return -1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 873) 		}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 874) 		NEXT_ARGP();
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 875) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 876) 		(*fds)[0] = bpf_map_get_fd_by_id(id);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 877) 		if ((*fds)[0] < 0) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 878) 			p_err("get map by id (%u): %s", id, strerror(errno));
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 879) 			return -1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 880) 		}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 881) 		return 1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 882) 	} else if (is_prefix(**argv, "name")) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 883) 		char *name;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 884) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 885) 		NEXT_ARGP();
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 886) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 887) 		name = **argv;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 888) 		if (strlen(name) > BPF_OBJ_NAME_LEN - 1) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 889) 			p_err("can't parse name");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 890) 			return -1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 891) 		}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 892) 		NEXT_ARGP();
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 893) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 894) 		return map_fd_by_name(name, fds);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 895) 	} else if (is_prefix(**argv, "pinned")) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 896) 		char *path;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 897) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 898) 		NEXT_ARGP();
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 899) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 900) 		path = **argv;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 901) 		NEXT_ARGP();
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 902) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 903) 		(*fds)[0] = open_obj_pinned_any(path, BPF_OBJ_MAP);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 904) 		if ((*fds)[0] < 0)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 905) 			return -1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 906) 		return 1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 907) 	}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 908) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 909) 	p_err("expected 'id', 'name' or 'pinned', got: '%s'?", **argv);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 910) 	return -1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 911) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 912) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 913) int map_parse_fd(int *argc, char ***argv)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 914) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 915) 	int *fds = NULL;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 916) 	int nb_fds, fd;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 917) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 918) 	fds = malloc(sizeof(int));
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 919) 	if (!fds) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 920) 		p_err("mem alloc failed");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 921) 		return -1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 922) 	}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 923) 	nb_fds = map_parse_fds(argc, argv, &fds);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 924) 	if (nb_fds != 1) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 925) 		if (nb_fds > 1) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 926) 			p_err("several maps match this handle");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 927) 			while (nb_fds--)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 928) 				close(fds[nb_fds]);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 929) 		}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 930) 		fd = -1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 931) 		goto exit_free;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 932) 	}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 933) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 934) 	fd = fds[0];
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 935) exit_free:
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 936) 	free(fds);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 937) 	return fd;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 938) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 939) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 940) int map_parse_fd_and_info(int *argc, char ***argv, void *info, __u32 *info_len)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 941) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 942) 	int err;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 943) 	int fd;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 944) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 945) 	fd = map_parse_fd(argc, argv);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 946) 	if (fd < 0)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 947) 		return -1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 948) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 949) 	err = bpf_obj_get_info_by_fd(fd, info, info_len);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 950) 	if (err) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 951) 		p_err("can't get map info: %s", strerror(errno));
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 952) 		close(fd);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 953) 		return err;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 954) 	}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 955) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 956) 	return fd;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 957) }