blob: 79734771bd29a00a590a5cb1c69fe671bc510149 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
|
/* SPDX-License-Identifier: GPL-2.0 */
#include <linux/bpf.h>
#include <bpf/bpf_helpers.h>
#include <xdp/xdp_helpers.h>
#include "xsk_def_xdp_prog.h"
#define DEFAULT_QUEUE_IDS 64
struct {
__uint(type, BPF_MAP_TYPE_XSKMAP);
__uint(key_size, sizeof(int));
__uint(value_size, sizeof(int));
__uint(max_entries, DEFAULT_QUEUE_IDS);
} xsks_map SEC(".maps");
struct {
__uint(priority, 20);
__uint(XDP_PASS, 1);
} XDP_RUN_CONFIG(xsk_def_prog);
/* Program refcount, in order to work properly,
* must be declared before any other global variables
* and initialized with '1'.
*/
volatile int refcnt = 1;
/* This is the program for 5.3 kernels and older. */
SEC("xdp")
int xsk_def_prog(struct xdp_md *ctx)
{
int index = ctx->rx_queue_index;
/* Make sure refcount is referenced by the program */
if (!refcnt)
return XDP_PASS;
/* A set entry here means that the corresponding queue_id
* has an active AF_XDP socket bound to it.
*/
if (bpf_map_lookup_elem(&xsks_map, &index))
return bpf_redirect_map(&xsks_map, index, 0);
return XDP_PASS;
}
char _license[] SEC("license") = "GPL";
__uint(xsk_prog_version, XSK_PROG_VERSION) SEC(XDP_METADATA_SECTION);
|
