1/* SPDX-License-Identifier: GPL-2.0 */
 2/*
 3 * Copyright (c) 2024 Meta Platforms, Inc. and affiliates.
 4 * Copyright (c) 2024 David Vernet <dvernet@meta.com>
 5 */
 6#include <bpf/bpf.h>
 7#include <sched.h>
 8#include <scx/common.h>
 9#include <sys/wait.h>
10#include <unistd.h>
11#include "prog_run.bpf.skel.h"
12#include "scx_test.h"
13
14static enum scx_test_status setup(void **ctx)
15{
16	struct prog_run *skel;
17
18	skel = prog_run__open_and_load();
19	if (!skel) {
20		SCX_ERR("Failed to open and load skel");
21		return SCX_TEST_FAIL;
22	}
23	*ctx = skel;
24
25	return SCX_TEST_PASS;
26}
27
28static enum scx_test_status run(void *ctx)
29{
30	struct prog_run *skel = ctx;
31	struct bpf_link *link;
32	int prog_fd, err = 0;
33
34	prog_fd = bpf_program__fd(skel->progs.prog_run_syscall);
35	if (prog_fd < 0) {
36		SCX_ERR("Failed to get BPF_PROG_RUN prog");
37		return SCX_TEST_FAIL;
38	}
39
40	LIBBPF_OPTS(bpf_test_run_opts, topts);
41
42	link = bpf_map__attach_struct_ops(skel->maps.prog_run_ops);
43	if (!link) {
44		SCX_ERR("Failed to attach scheduler");
45		close(prog_fd);
46		return SCX_TEST_FAIL;
47	}
48
49	err = bpf_prog_test_run_opts(prog_fd, &topts);
50	SCX_EQ(err, 0);
51
52	/* Assumes uei.kind is written last */
53	while (skel->data->uei.kind == EXIT_KIND(SCX_EXIT_NONE))
54		sched_yield();
55
56	SCX_EQ(skel->data->uei.kind, EXIT_KIND(SCX_EXIT_UNREG_BPF));
57	SCX_EQ(skel->data->uei.exit_code, 0xdeadbeef);
58	close(prog_fd);
59	bpf_link__destroy(link);
60
61	return SCX_TEST_PASS;
62}
63
64static void cleanup(void *ctx)
65{
66	struct prog_run *skel = ctx;
67
68	prog_run__destroy(skel);
69}
70
71struct scx_test prog_run = {
72	.name = "prog_run",
73	.description = "Verify we can call into a scheduler with BPF_PROG_RUN, and invoke kfuncs",
74	.setup = setup,
75	.run = run,
76	.cleanup = cleanup,
77};
78REGISTER_SCX_TEST(&prog_run)