cachepc-linux

Fork of AMDESE/linux with modifications for CachePC side-channel attack
git clone https://git.sinitax.com/sinitax/cachepc-linux
Log | Files | Refs | README | LICENSE | sfeed.txt

bpf_iter_setsockopt_unix.c (1264B)

      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";