1/* SPDX-License-Identifier: GPL-2.0 */
 2/*
 3 * Copyright (c) 2023 Meta Platforms, Inc. and affiliates.
 4 * Copyright (c) 2023 David Vernet <dvernet@meta.com>
 5 * Copyright (c) 2023 Tejun Heo <tj@kernel.org>
 6 */
 7#include <bpf/bpf.h>
 8#include <scx/common.h>
 9#include <sys/wait.h>
10#include <unistd.h>
11#include "select_cpu_dispatch_dbl_dsp.bpf.skel.h"
12#include "scx_test.h"
13
14static enum scx_test_status setup(void **ctx)
15{
16	struct select_cpu_dispatch_dbl_dsp *skel;
17
18	skel = select_cpu_dispatch_dbl_dsp__open_and_load();
19	SCX_FAIL_IF(!skel, "Failed to open and load skel");
20	*ctx = skel;
21
22	return SCX_TEST_PASS;
23}
24
25static enum scx_test_status run(void *ctx)
26{
27	struct select_cpu_dispatch_dbl_dsp *skel = ctx;
28	struct bpf_link *link;
29
30	link = bpf_map__attach_struct_ops(skel->maps.select_cpu_dispatch_dbl_dsp_ops);
31	SCX_FAIL_IF(!link, "Failed to attach scheduler");
32
33	sleep(1);
34
35	SCX_EQ(skel->data->uei.kind, EXIT_KIND(SCX_EXIT_ERROR));
36	bpf_link__destroy(link);
37
38	return SCX_TEST_PASS;
39}
40
41static void cleanup(void *ctx)
42{
43	struct select_cpu_dispatch_dbl_dsp *skel = ctx;
44
45	select_cpu_dispatch_dbl_dsp__destroy(skel);
46}
47
48struct scx_test select_cpu_dispatch_dbl_dsp = {
49	.name = "select_cpu_dispatch_dbl_dsp",
50	.description = "Verify graceful failure if we dispatch twice to a "
51		       "DSQ in ops.select_cpu()",
52	.setup = setup,
53	.run = run,
54	.cleanup = cleanup,
55};
56REGISTER_SCX_TEST(&select_cpu_dispatch_dbl_dsp)