tools/testing/selftests/sched_ext/create_dsq.c

   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
  13 static 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
  27 static 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
  43 static 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)