1// SPDX-License-Identifier: GPL-2.0
2/* Copyright Amazon.com Inc. or its affiliates. */
3#include "bpf_iter.h"
4#include "bpf_tracing_net.h"
5#include <bpf/bpf_helpers.h>
6#include <limits.h>
7
8#define AUTOBIND_LEN 6
9char sun_path[AUTOBIND_LEN];
10
11#define NR_CASES 5
12int sndbuf_setsockopt[NR_CASES] = {-1, 0, 8192, INT_MAX / 2, INT_MAX};
13int sndbuf_getsockopt[NR_CASES] = {-1, -1, -1, -1, -1};
14int sndbuf_getsockopt_expected[NR_CASES];
15
16static inline int cmpname(struct unix_sock *unix_sk)
17{
18	int i;
19
20	for (i = 0; i < AUTOBIND_LEN; i++) {
21		if (unix_sk->addr->name->sun_path[i] != sun_path[i])
22			return -1;
23	}
24
25	return 0;
26}
27
28SEC("iter/unix")
29int change_sndbuf(struct bpf_iter__unix *ctx)
30{
31	struct unix_sock *unix_sk = ctx->unix_sk;
32	int i, err;
33
34	if (!unix_sk || !unix_sk->addr)
35		return 0;
36
37	if (unix_sk->addr->name->sun_path[0])
38		return 0;
39
40	if (cmpname(unix_sk))
41		return 0;
42
43	for (i = 0; i < NR_CASES; i++) {
44		err = bpf_setsockopt(unix_sk, SOL_SOCKET, SO_SNDBUF,
45				     &sndbuf_setsockopt[i],
46				     sizeof(sndbuf_setsockopt[i]));
47		if (err)
48			break;
49
50		err = bpf_getsockopt(unix_sk, SOL_SOCKET, SO_SNDBUF,
51				     &sndbuf_getsockopt[i],
52				     sizeof(sndbuf_getsockopt[i]));
53		if (err)
54			break;
55	}
56
57	return 0;
58}
59
60char _license[] SEC("license") = "GPL";
61