tools/testing/selftests/bpf/prog_tests/linked_funcs.c

   1 // SPDX-License-Identifier: GPL-2.0
   2 /* Copyright (c) 2021 Facebook */
   3
   4 #include <test_progs.h>
   5 #include <sys/syscall.h>
   6 #include "linked_funcs.skel.h"
   7
   8 void test_linked_funcs(void)
   9 {
  10         int err;
  11         struct linked_funcs *skel;
  12
  13         skel = linked_funcs__open();
  14         if (!ASSERT_OK_PTR(skel, "skel_open"))
  15                 return;
  16
  17         /* handler1 and handler2 are marked as SEC("?raw_tp/sys_enter") and
  18          * are set to not autoload by default
  19          */
  20         bpf_program__set_autoload(skel->progs.handler1, true);
  21         bpf_program__set_autoload(skel->progs.handler2, true);
  22
  23         skel->rodata->my_tid = sys_gettid();
  24         skel->bss->syscall_id = SYS_getpgid;
  25
  26         err = linked_funcs__load(skel);
  27         if (!ASSERT_OK(err, "skel_load"))
  28                 goto cleanup;
  29
  30         err = linked_funcs__attach(skel);
  31         if (!ASSERT_OK(err, "skel_attach"))
  32                 goto cleanup;
  33
  34         /* trigger */
  35         syscall(SYS_getpgid);
  36
  37         ASSERT_EQ(skel->bss->output_val1, 2000 + 2000, "output_val1");
  38         ASSERT_EQ(skel->bss->output_ctx1, SYS_getpgid, "output_ctx1");
  39         ASSERT_EQ(skel->bss->output_weak1, 42, "output_weak1");
  40
  41         ASSERT_EQ(skel->bss->output_val2, 2 * 1000 + 2 * (2 * 1000), "output_val2");
  42         ASSERT_EQ(skel->bss->output_ctx2, SYS_getpgid, "output_ctx2");
  43         /* output_weak2 should never be updated */
  44         ASSERT_EQ(skel->bss->output_weak2, 0, "output_weak2");
  45
  46 cleanup:
  47         linked_funcs__destroy(skel);
  48 }