tools/testing/selftests/bpf/progs/test_sockmap_msg_pop_data.c


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

// SPDX-License-Identifier: GPL-2.0
#include "vmlinux.h"
#include <bpf/bpf_helpers.h>

struct {
	__uint(type, BPF_MAP_TYPE_SOCKMAP);
	__uint(max_entries, 1);
	__type(key, int);
	__type(value, int);
} sock_map SEC(".maps");

#define POP_START 0x48a3
#define POP_LEN   0xfffffffd

long pop_data_ret = 1;

SEC("sk_msg")
int prog_msg_pop_data(struct sk_msg_md *msg)
{
	if (msg->size <= POP_START)
		return SK_PASS;

	pop_data_ret = bpf_msg_pop_data(msg, POP_START, POP_LEN, 0);
	return SK_PASS;
}

char _license[] SEC("license") = "GPL";