1 // SPDX-License-Identifier: GPL-2.0 2 /* Copyright Amazon.com Inc. or its affiliates. */ 3 #include <vmlinux.h> 4 #include "bpf_tracing_net.h" 5 #include <bpf/bpf_helpers.h> 6 #include <limits.h> 7 8 #define AUTOBIND_LEN 6 9 char sun_path[AUTOBIND_LEN]; 10 11 #define NR_CASES 5 12 int sndbuf_setsockopt[NR_CASES] = {-1, 0, 8192, INT_MAX / 2, INT_MAX}; 13 int sndbuf_getsockopt[NR_CASES] = {-1, -1, -1, -1, -1}; 14 int sndbuf_getsockopt_expected[NR_CASES]; 15 16 static 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 28 SEC("iter/unix") 29 int 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 60 char _license[] SEC("license") = "GPL"; 61