1*9ef64711SSamuel Wu // SPDX-License-Identifier: GPL-2.0 2*9ef64711SSamuel Wu /* Copyright 2026 Google LLC */ 3*9ef64711SSamuel Wu 4*9ef64711SSamuel Wu #include <test_progs.h> 5*9ef64711SSamuel Wu #include <bpf/btf.h> 6*9ef64711SSamuel Wu #include <fcntl.h> 7*9ef64711SSamuel Wu #include "test_wakeup_source.skel.h" 8*9ef64711SSamuel Wu #include "wakeup_source_fail.skel.h" 9*9ef64711SSamuel Wu #include "progs/wakeup_source.h" 10*9ef64711SSamuel Wu 11*9ef64711SSamuel Wu static int lock_ws(const char *name) 12*9ef64711SSamuel Wu { 13*9ef64711SSamuel Wu int fd; 14*9ef64711SSamuel Wu ssize_t bytes; 15*9ef64711SSamuel Wu 16*9ef64711SSamuel Wu fd = open("/sys/power/wake_lock", O_WRONLY); 17*9ef64711SSamuel Wu if (!ASSERT_OK_FD(fd, "open /sys/power/wake_lock")) 18*9ef64711SSamuel Wu return -1; 19*9ef64711SSamuel Wu 20*9ef64711SSamuel Wu bytes = write(fd, name, strlen(name)); 21*9ef64711SSamuel Wu close(fd); 22*9ef64711SSamuel Wu if (!ASSERT_EQ(bytes, strlen(name), "write to wake_lock")) 23*9ef64711SSamuel Wu return -1; 24*9ef64711SSamuel Wu 25*9ef64711SSamuel Wu return 0; 26*9ef64711SSamuel Wu } 27*9ef64711SSamuel Wu 28*9ef64711SSamuel Wu static void unlock_ws(const char *name) 29*9ef64711SSamuel Wu { 30*9ef64711SSamuel Wu int fd; 31*9ef64711SSamuel Wu 32*9ef64711SSamuel Wu fd = open("/sys/power/wake_unlock", O_WRONLY); 33*9ef64711SSamuel Wu if (fd < 0) 34*9ef64711SSamuel Wu return; 35*9ef64711SSamuel Wu 36*9ef64711SSamuel Wu write(fd, name, strlen(name)); 37*9ef64711SSamuel Wu close(fd); 38*9ef64711SSamuel Wu } 39*9ef64711SSamuel Wu 40*9ef64711SSamuel Wu struct rb_ctx { 41*9ef64711SSamuel Wu const char *name; 42*9ef64711SSamuel Wu bool found; 43*9ef64711SSamuel Wu long long active_time_ns; 44*9ef64711SSamuel Wu long long total_time_ns; 45*9ef64711SSamuel Wu }; 46*9ef64711SSamuel Wu 47*9ef64711SSamuel Wu static int process_sample(void *ctx, void *data, size_t len) 48*9ef64711SSamuel Wu { 49*9ef64711SSamuel Wu struct rb_ctx *rb_ctx = ctx; 50*9ef64711SSamuel Wu struct wakeup_event_t *e = data; 51*9ef64711SSamuel Wu 52*9ef64711SSamuel Wu if (strcmp(e->name, rb_ctx->name) == 0) { 53*9ef64711SSamuel Wu rb_ctx->found = true; 54*9ef64711SSamuel Wu rb_ctx->active_time_ns = e->active_time_ns; 55*9ef64711SSamuel Wu rb_ctx->total_time_ns = e->total_time_ns; 56*9ef64711SSamuel Wu } 57*9ef64711SSamuel Wu return 0; 58*9ef64711SSamuel Wu } 59*9ef64711SSamuel Wu 60*9ef64711SSamuel Wu void test_wakeup_source(void) 61*9ef64711SSamuel Wu { 62*9ef64711SSamuel Wu struct btf *btf; 63*9ef64711SSamuel Wu int id; 64*9ef64711SSamuel Wu 65*9ef64711SSamuel Wu btf = btf__load_vmlinux_btf(); 66*9ef64711SSamuel Wu if (!ASSERT_OK_PTR(btf, "btf_vmlinux")) 67*9ef64711SSamuel Wu return; 68*9ef64711SSamuel Wu 69*9ef64711SSamuel Wu id = btf__find_by_name_kind(btf, "bpf_wakeup_sources_get_head", BTF_KIND_FUNC); 70*9ef64711SSamuel Wu btf__free(btf); 71*9ef64711SSamuel Wu 72*9ef64711SSamuel Wu if (id < 0) { 73*9ef64711SSamuel Wu printf("%s:SKIP:bpf_wakeup_sources_get_head kfunc not found in BTF\n", __func__); 74*9ef64711SSamuel Wu test__skip(); 75*9ef64711SSamuel Wu return; 76*9ef64711SSamuel Wu } 77*9ef64711SSamuel Wu 78*9ef64711SSamuel Wu if (test__start_subtest("iterate_and_verify_times")) { 79*9ef64711SSamuel Wu struct test_wakeup_source *skel; 80*9ef64711SSamuel Wu struct ring_buffer *rb = NULL; 81*9ef64711SSamuel Wu struct rb_ctx rb_ctx = { 82*9ef64711SSamuel Wu .name = "bpf_selftest_ws_times", 83*9ef64711SSamuel Wu .found = false, 84*9ef64711SSamuel Wu }; 85*9ef64711SSamuel Wu int err; 86*9ef64711SSamuel Wu 87*9ef64711SSamuel Wu skel = test_wakeup_source__open_and_load(); 88*9ef64711SSamuel Wu if (!ASSERT_OK_PTR(skel, "skel_open_and_load")) 89*9ef64711SSamuel Wu return; 90*9ef64711SSamuel Wu 91*9ef64711SSamuel Wu rb = ring_buffer__new(bpf_map__fd(skel->maps.rb), process_sample, &rb_ctx, NULL); 92*9ef64711SSamuel Wu if (!ASSERT_OK_PTR(rb, "ring_buffer__new")) 93*9ef64711SSamuel Wu goto destroy; 94*9ef64711SSamuel Wu 95*9ef64711SSamuel Wu /* Create a temporary wakeup source */ 96*9ef64711SSamuel Wu if (!ASSERT_OK(lock_ws(rb_ctx.name), "lock_ws")) 97*9ef64711SSamuel Wu goto unlock; 98*9ef64711SSamuel Wu 99*9ef64711SSamuel Wu err = bpf_prog_test_run_opts(bpf_program__fd( 100*9ef64711SSamuel Wu skel->progs.iterate_wakeupsources), NULL); 101*9ef64711SSamuel Wu ASSERT_OK(err, "bpf_prog_test_run"); 102*9ef64711SSamuel Wu 103*9ef64711SSamuel Wu ring_buffer__consume(rb); 104*9ef64711SSamuel Wu 105*9ef64711SSamuel Wu ASSERT_TRUE(rb_ctx.found, "found_test_ws_in_rb"); 106*9ef64711SSamuel Wu ASSERT_GT(rb_ctx.active_time_ns, 0, "active_time_gt_0"); 107*9ef64711SSamuel Wu ASSERT_GT(rb_ctx.total_time_ns, 0, "total_time_gt_0"); 108*9ef64711SSamuel Wu 109*9ef64711SSamuel Wu unlock: 110*9ef64711SSamuel Wu unlock_ws(rb_ctx.name); 111*9ef64711SSamuel Wu destroy: 112*9ef64711SSamuel Wu if (rb) 113*9ef64711SSamuel Wu ring_buffer__free(rb); 114*9ef64711SSamuel Wu test_wakeup_source__destroy(skel); 115*9ef64711SSamuel Wu } 116*9ef64711SSamuel Wu 117*9ef64711SSamuel Wu RUN_TESTS(wakeup_source_fail); 118*9ef64711SSamuel Wu } 119