113 lines
3.1 KiB
Diff
113 lines
3.1 KiB
Diff
From 5e488bdb1ebb14cd16e2df640e90e7eb39ff9efa Mon Sep 17 00:00:00 2001
|
|
From: JofDiamonds <kwb0523@163.com>
|
|
Date: Sat, 20 May 2023 16:38:45 +0800
|
|
Subject: [PATCH] adapt libbpf-0.8.1
|
|
|
|
---
|
|
bpf/bwm_prio_kern.c | 14 +++++++-------
|
|
bwmcli.c | 41 +++++++++++++++++++++++++++++------------
|
|
2 files changed, 36 insertions(+), 19 deletions(-)
|
|
|
|
diff --git a/bpf/bwm_prio_kern.c b/bpf/bwm_prio_kern.c
|
|
index 1c59322..68127f9 100644
|
|
--- a/bpf/bwm_prio_kern.c
|
|
+++ b/bpf/bwm_prio_kern.c
|
|
@@ -20,13 +20,13 @@
|
|
|
|
#define DEFAULT_CGP_PRIO 0
|
|
|
|
-struct bpf_map_def cgrp_prio SEC("maps") = {
|
|
- .type = BPF_MAP_TYPE_ARRAY,
|
|
- .key_size = sizeof(unsigned int),
|
|
- .value_size = sizeof(unsigned int),
|
|
- .max_entries = 1,
|
|
- .map_flags = 0,
|
|
-};
|
|
+struct {
|
|
+ __uint(type, BPF_MAP_TYPE_ARRAY);
|
|
+ __type(key, unsigned int);
|
|
+ __type(value, unsigned int);
|
|
+ __uint(max_entries, 1);
|
|
+ __uint(map_flags, 0);
|
|
+} cgrp_prio SEC(".maps");
|
|
|
|
SEC("cgroup_skb/egress")
|
|
int _bwm_out_cg(struct __sk_buff *skb)
|
|
diff --git a/bwmcli.c b/bwmcli.c
|
|
index f087ea6..c9fe698 100644
|
|
--- a/bwmcli.c
|
|
+++ b/bwmcli.c
|
|
@@ -212,40 +212,57 @@ static int BreakArgs(char *s, char *arg1, char *arg2, unsigned long arg1Len, uns
|
|
return EXIT_OK;
|
|
}
|
|
|
|
-static int ProgLoad(char *prog, struct bpf_object ** obj, int * bpfprogFd)
|
|
+static int ProgLoad(char *prog, struct bpf_object **obj, int * bpfprogFd)
|
|
{
|
|
- struct bpf_prog_load_attr progLoadAttr = {
|
|
- .prog_type = BPF_PROG_TYPE_CGROUP_SKB,
|
|
- .file = prog,
|
|
- .expected_attach_type = BPF_CGROUP_INET_EGRESS,
|
|
- .ifindex = 0,
|
|
- .log_level = 0,
|
|
- };
|
|
int mapFd;
|
|
struct bpf_map *map = NULL;
|
|
+ struct bpf_program *program = NULL;
|
|
|
|
if (access(prog, O_RDONLY) < 0) {
|
|
BWM_LOG_ERR("Error accessing file %s\n", prog);
|
|
return -1;
|
|
}
|
|
- if (bpf_prog_load_xattr(&progLoadAttr, &(*obj), bpfprogFd)) {
|
|
- BWM_LOG_ERR("ERROR: bpf_prog_load_xattr failed for: %s. errno:%d\n", prog, errno);
|
|
+
|
|
+ *obj = bpf_object__open(prog);
|
|
+ if (libbpf_get_error(*obj)) {
|
|
+ BWM_LOG_ERR("ERROR: bpf_object__open failed for: %s. errno:%d\n", prog, errno);
|
|
return -1;
|
|
}
|
|
|
|
+ program = bpf_object__next_program(*obj, NULL);
|
|
+ if (!program) {
|
|
+ BWM_LOG_ERR("ERROR: bpf_object__next_program failed for: %s. errno:%d\n", prog, errno);
|
|
+ goto err;
|
|
+ }
|
|
+
|
|
+ bpf_program__set_type(program, BPF_PROG_TYPE_CGROUP_SKB);
|
|
+
|
|
+ if (bpf_object__load(*obj)) {
|
|
+ BWM_LOG_ERR("ERROR: bpf_object__load failed for: %s. errno:%d\n", prog, errno);
|
|
+ goto err;
|
|
+ }
|
|
+
|
|
+ *bpfprogFd = bpf_program__fd(program);
|
|
+ if (*bpfprogFd < 0) {
|
|
+ BWM_LOG_ERR("Failed to get program fd. errno:%d\n", errno);
|
|
+ goto err;
|
|
+ }
|
|
+
|
|
map = bpf_object__find_map_by_name(*obj, "cgrp_prio");
|
|
if (!map) {
|
|
BWM_LOG_ERR("Failed to load map cgrp_prio from bpf prog. errno:%d\n", errno);
|
|
- return -1;
|
|
+ goto err;
|
|
}
|
|
|
|
mapFd = bpf_map__fd(map);
|
|
if (mapFd < 0) {
|
|
BWM_LOG_ERR("Map not found: %d. errno:%d\n", mapFd, errno);
|
|
- return -1;
|
|
+ goto err;
|
|
}
|
|
|
|
return mapFd;
|
|
+err:
|
|
+ return -1;
|
|
}
|
|
|
|
static int GetMapFdByProgId(__u32 progId)
|
|
--
|
|
2.33.0
|
|
|