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 <scx/common.h> 8 #include <sys/wait.h> 9 #include <unistd.h> 10 #include "create_dsq.bpf.skel.h" 11 #include "scx_test.h" 12 setup(void ** ctx)13static enum scx_test_status setup(void **ctx) 14 { 15 struct create_dsq *skel; 16 17 skel = create_dsq__open_and_load(); 18 if (!skel) { 19 SCX_ERR("Failed to open and load skel"); 20 return SCX_TEST_FAIL; 21 } 22 *ctx = skel; 23 24 return SCX_TEST_PASS; 25 } 26 run(void * ctx)27static enum scx_test_status run(void *ctx) 28 { 29 struct create_dsq *skel = ctx; 30 struct bpf_link *link; 31 32 link = bpf_map__attach_struct_ops(skel->maps.create_dsq_ops); 33 if (!link) { 34 SCX_ERR("Failed to attach scheduler"); 35 return SCX_TEST_FAIL; 36 } 37 38 bpf_link__destroy(link); 39 40 return SCX_TEST_PASS; 41 } 42 cleanup(void * ctx)43static void cleanup(void *ctx) 44 { 45 struct create_dsq *skel = ctx; 46 47 create_dsq__destroy(skel); 48 } 49 50 struct scx_test create_dsq = { 51 .name = "create_dsq", 52 .description = "Create and destroy a dsq in a loop", 53 .setup = setup, 54 .run = run, 55 .cleanup = cleanup, 56 }; 57 REGISTER_SCX_TEST(&create_dsq) 58