tools/testing/selftests/bpf/progs/connect4_dropper.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
28

// SPDX-License-Identifier: GPL-2.0

#include <string.h>

#include <linux/stddef.h>
#include <linux/bpf.h>

#include <sys/socket.h>

#include <bpf/bpf_helpers.h>
#include <bpf/bpf_endian.h>

#define VERDICT_REJECT	0
#define VERDICT_PROCEED	1

int port;

SEC("cgroup/connect4")
int connect_v4_dropper(struct bpf_sock_addr *ctx)
{
	if (ctx->type != SOCK_STREAM)
		return VERDICT_PROCEED;
	if (ctx->user_port == bpf_htons(port))
		return VERDICT_REJECT;
	return VERDICT_PROCEED;
}

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