61 lines · c
1// SPDX-License-Identifier: GPL-2.02/* 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 69char sun_path[AUTOBIND_LEN];10 11#define NR_CASES 512int 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