Quelle  core_reloc_raw.c   Sprache: C

// SPDX-License-Identifier: GPL-2.0

/* Test cases that can't load programs using libbpf and need direct
 * BPF syscall access
 */

#include <sys/syscall.h>
#include <bpf/libbpf.h>
#include <bpf/btf.h>

#include "test_progs.h"
#include "test_btf.h"
#include "bpf/libbpf_internal.h"

static char log[16 * 1024];

/* Check that verifier rejects BPF program containing relocation
 * pointing to non-existent BTF type.
 */
static void test_bad_local_id(void)
{
 struct test_btf {
  struct btf_header hdr;
  __u32 types[15];
  char strings[128];
 } raw_btf = {
  .hdr = {
   .magic = BTF_MAGIC,
   .version = BTF_VERSION,
   .hdr_len = sizeof(struct btf_header),
   .type_off = 0,
   .type_len = sizeof(raw_btf.types),
   .str_off = offsetof(struct test_btf, strings) -
       offsetof(struct test_btf, types),
   .str_len = sizeof(raw_btf.strings),
  },
  .types = {
   BTF_PTR_ENC(0),     /* [1] void*  */
   BTF_TYPE_INT_ENC(1, BTF_INT_SIGNED, 0, 32, 4), /* [2] int    */
   BTF_FUNC_PROTO_ENC(2, 1),   /* [3] int (*)(void*) */
   BTF_FUNC_PROTO_ARG_ENC(8, 1),
   BTF_FUNC_ENC(8, 3)   /* [4] FUNC 'foo' type_id=2   */
  },
  .strings = "\0int\0 0\0foo\0"
 };
 __u32 log_level = 1 | 2 | 4;
 LIBBPF_OPTS(bpf_btf_load_opts, opts,
      .log_buf = log,
      .log_size = sizeof(log),
      .log_level = log_level,
 );
 struct bpf_insn insns[] = {
  BPF_ALU64_IMM(BPF_MOV, BPF_REG_0, 0),
  BPF_EXIT_INSN(),
 };
 struct bpf_func_info funcs[] = {
  {
   .insn_off = 0,
   .type_id = 4,
  }
 };
 struct bpf_core_relo relos[] = {
  {
   .insn_off = 0,  /* patch first instruction (r0 = 0) */
   .type_id = 100500, /* !!! this type id does not exist */
   .access_str_off = 6, /* offset of "0" */
   .kind = BPF_CORE_TYPE_ID_LOCAL,
  }
 };
 union bpf_attr attr;
 int saved_errno;
 int prog_fd = -1;
 int btf_fd = -1;

 btf_fd = bpf_btf_load(&raw_btf, sizeof(raw_btf), &opts);
 saved_errno = errno;
 if (btf_fd < 0 || env.verbosity > VERBOSE_NORMAL) {
  printf("-------- BTF load log start --------\n");
  printf("%s", log);
  printf("-------- BTF load log end ----------\n");
 }
 if (btf_fd < 0) {
  PRINT_FAIL("bpf_btf_load() failed, errno=%d\n", saved_errno);
  return;
 }

 log[0] = 0;
 memset(&attr, 0, sizeof(attr));
 attr.prog_btf_fd = btf_fd;
 attr.prog_type = BPF_TRACE_RAW_TP;
 attr.license = (__u64)"GPL";
 attr.insns = (__u64)&insns;
 attr.insn_cnt = sizeof(insns) / sizeof(*insns);
 attr.log_buf = (__u64)log;
 attr.log_size = sizeof(log);
 attr.log_level = log_level;
 attr.func_info = (__u64)funcs;
 attr.func_info_cnt = sizeof(funcs) / sizeof(*funcs);
 attr.func_info_rec_size = sizeof(*funcs);
 attr.core_relos = (__u64)relos;
 attr.core_relo_cnt = sizeof(relos) / sizeof(*relos);
 attr.core_relo_rec_size = sizeof(*relos);
 prog_fd = sys_bpf_prog_load(&attr, sizeof(attr), 1);
 saved_errno = errno;
 if (prog_fd < 0 || env.verbosity > VERBOSE_NORMAL) {
  printf("-------- program load log start --------\n");
  printf("%s", log);
  printf("-------- program load log end ----------\n");
 }
 if (prog_fd >= 0) {
  PRINT_FAIL("sys_bpf_prog_load() expected to fail\n");
  goto out;
 }
 ASSERT_HAS_SUBSTR(log, "relo #0: bad type id 100500", "program load log");

out:
 close(prog_fd);
 close(btf_fd);
}

void test_core_reloc_raw(void)
{
 if (test__start_subtest("bad_local_id"))
  test_bad_local_id();
}