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

fentry_test.c (1630B)

      1// SPDX-License-Identifier: GPL-2.0
      2/* Copyright (c) 2019 Facebook */
      3#include <linux/bpf.h>
      4#include <bpf/bpf_helpers.h>
      5#include <bpf/bpf_tracing.h>
      6
      7char _license[] SEC("license") = "GPL";
      8
      9__u64 test1_result = 0;
     10SEC("fentry/bpf_fentry_test1")
     11int BPF_PROG(test1, int a)
     12{
     13	test1_result = a == 1;
     14	return 0;
     15}
     16
     17__u64 test2_result = 0;
     18SEC("fentry/bpf_fentry_test2")
     19int BPF_PROG(test2, int a, __u64 b)
     20{
     21	test2_result = a == 2 && b == 3;
     22	return 0;
     23}
     24
     25__u64 test3_result = 0;
     26SEC("fentry/bpf_fentry_test3")
     27int BPF_PROG(test3, char a, int b, __u64 c)
     28{
     29	test3_result = a == 4 && b == 5 && c == 6;
     30	return 0;
     31}
     32
     33__u64 test4_result = 0;
     34SEC("fentry/bpf_fentry_test4")
     35int BPF_PROG(test4, void *a, char b, int c, __u64 d)
     36{
     37	test4_result = a == (void *)7 && b == 8 && c == 9 && d == 10;
     38	return 0;
     39}
     40
     41__u64 test5_result = 0;
     42SEC("fentry/bpf_fentry_test5")
     43int BPF_PROG(test5, __u64 a, void *b, short c, int d, __u64 e)
     44{
     45	test5_result = a == 11 && b == (void *)12 && c == 13 && d == 14 &&
     46		e == 15;
     47	return 0;
     48}
     49
     50__u64 test6_result = 0;
     51SEC("fentry/bpf_fentry_test6")
     52int BPF_PROG(test6, __u64 a, void *b, short c, int d, void * e, __u64 f)
     53{
     54	test6_result = a == 16 && b == (void *)17 && c == 18 && d == 19 &&
     55		e == (void *)20 && f == 21;
     56	return 0;
     57}
     58
     59struct bpf_fentry_test_t {
     60	struct bpf_fentry_test_t *a;
     61};
     62
     63__u64 test7_result = 0;
     64SEC("fentry/bpf_fentry_test7")
     65int BPF_PROG(test7, struct bpf_fentry_test_t *arg)
     66{
     67	if (!arg)
     68		test7_result = 1;
     69	return 0;
     70}
     71
     72__u64 test8_result = 0;
     73SEC("fentry/bpf_fentry_test8")
     74int BPF_PROG(test8, struct bpf_fentry_test_t *arg)
     75{
     76	if (arg->a == 0)
     77		test8_result = 1;
     78	return 0;
     79}