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

test_usdt.c (2041B)

      1// SPDX-License-Identifier: GPL-2.0
      2/* Copyright (c) 2022 Meta Platforms, Inc. and affiliates. */
      3
      4#include "vmlinux.h"
      5#include <bpf/bpf_helpers.h>
      6#include <bpf/usdt.bpf.h>
      7
      8int my_pid;
      9
     10int usdt0_called;
     11u64 usdt0_cookie;
     12int usdt0_arg_cnt;
     13int usdt0_arg_ret;
     14
     15SEC("usdt")
     16int usdt0(struct pt_regs *ctx)
     17{
     18	long tmp;
     19
     20	if (my_pid != (bpf_get_current_pid_tgid() >> 32))
     21		return 0;
     22
     23	__sync_fetch_and_add(&usdt0_called, 1);
     24
     25	usdt0_cookie = bpf_usdt_cookie(ctx);
     26	usdt0_arg_cnt = bpf_usdt_arg_cnt(ctx);
     27	/* should return -ENOENT for any arg_num */
     28	usdt0_arg_ret = bpf_usdt_arg(ctx, bpf_get_prandom_u32(), &tmp);
     29	return 0;
     30}
     31
     32int usdt3_called;
     33u64 usdt3_cookie;
     34int usdt3_arg_cnt;
     35int usdt3_arg_rets[3];
     36u64 usdt3_args[3];
     37
     38SEC("usdt//proc/self/exe:test:usdt3")
     39int usdt3(struct pt_regs *ctx)
     40{
     41	long tmp;
     42
     43	if (my_pid != (bpf_get_current_pid_tgid() >> 32))
     44		return 0;
     45
     46	__sync_fetch_and_add(&usdt3_called, 1);
     47
     48	usdt3_cookie = bpf_usdt_cookie(ctx);
     49	usdt3_arg_cnt = bpf_usdt_arg_cnt(ctx);
     50
     51	usdt3_arg_rets[0] = bpf_usdt_arg(ctx, 0, &tmp);
     52	usdt3_args[0] = (int)tmp;
     53
     54	usdt3_arg_rets[1] = bpf_usdt_arg(ctx, 1, &tmp);
     55	usdt3_args[1] = (long)tmp;
     56
     57	usdt3_arg_rets[2] = bpf_usdt_arg(ctx, 2, &tmp);
     58	usdt3_args[2] = (uintptr_t)tmp;
     59
     60	return 0;
     61}
     62
     63int usdt12_called;
     64u64 usdt12_cookie;
     65int usdt12_arg_cnt;
     66u64 usdt12_args[12];
     67
     68SEC("usdt//proc/self/exe:test:usdt12")
     69int BPF_USDT(usdt12, int a1, int a2, long a3, long a4, unsigned a5,
     70		     long a6, __u64 a7, uintptr_t a8, int a9, short a10,
     71		     short a11, signed char a12)
     72{
     73	if (my_pid != (bpf_get_current_pid_tgid() >> 32))
     74		return 0;
     75
     76	__sync_fetch_and_add(&usdt12_called, 1);
     77
     78	usdt12_cookie = bpf_usdt_cookie(ctx);
     79	usdt12_arg_cnt = bpf_usdt_arg_cnt(ctx);
     80
     81	usdt12_args[0] = a1;
     82	usdt12_args[1] = a2;
     83	usdt12_args[2] = a3;
     84	usdt12_args[3] = a4;
     85	usdt12_args[4] = a5;
     86	usdt12_args[5] = a6;
     87	usdt12_args[6] = a7;
     88	usdt12_args[7] = a8;
     89	usdt12_args[8] = a9;
     90	usdt12_args[9] = a10;
     91	usdt12_args[10] = a11;
     92	usdt12_args[11] = a12;
     93	return 0;
     94}
     95
     96char _license[] SEC("license") = "GPL";