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