1 // SPDX-License-Identifier: GPL-2.0 2 /* 3 * User Events Dyn Events Test Program 4 * 5 * Copyright (c) 2021 Beau Belgrave <beaub@linux.microsoft.com> 6 */ 7 8 #include <errno.h> 9 #include <linux/user_events.h> 10 #include <stdio.h> 11 #include <stdlib.h> 12 #include <fcntl.h> 13 #include <sys/ioctl.h> 14 #include <sys/stat.h> 15 #include <unistd.h> 16 17 #include "../kselftest_harness.h" 18 19 const char *dyn_file = "/sys/kernel/tracing/dynamic_events"; 20 const char *clear = "!u:__test_event"; 21 22 static int Append(const char *value) 23 { 24 int fd = open(dyn_file, O_RDWR | O_APPEND); 25 int ret = write(fd, value, strlen(value)); 26 27 close(fd); 28 return ret; 29 } 30 31 #define CLEAR() \ 32 do { \ 33 int ret = Append(clear); \ 34 if (ret == -1) \ 35 ASSERT_EQ(ENOENT, errno); \ 36 } while (0) 37 38 #define TEST_PARSE(x) \ 39 do { \ 40 ASSERT_NE(-1, Append(x)); \ 41 CLEAR(); \ 42 } while (0) 43 44 #define TEST_NPARSE(x) ASSERT_EQ(-1, Append(x)) 45 46 FIXTURE(user) { 47 }; 48 49 FIXTURE_SETUP(user) { 50 CLEAR(); 51 } 52 53 FIXTURE_TEARDOWN(user) { 54 CLEAR(); 55 } 56 57 TEST_F(user, basic_types) { 58 /* All should work */ 59 TEST_PARSE("u:__test_event u64 a"); 60 TEST_PARSE("u:__test_event u32 a"); 61 TEST_PARSE("u:__test_event u16 a"); 62 TEST_PARSE("u:__test_event u8 a"); 63 TEST_PARSE("u:__test_event char a"); 64 TEST_PARSE("u:__test_event unsigned char a"); 65 TEST_PARSE("u:__test_event int a"); 66 TEST_PARSE("u:__test_event unsigned int a"); 67 TEST_PARSE("u:__test_event short a"); 68 TEST_PARSE("u:__test_event unsigned short a"); 69 TEST_PARSE("u:__test_event char[20] a"); 70 TEST_PARSE("u:__test_event unsigned char[20] a"); 71 TEST_PARSE("u:__test_event char[0x14] a"); 72 TEST_PARSE("u:__test_event unsigned char[0x14] a"); 73 /* Bad size format should fail */ 74 TEST_NPARSE("u:__test_event char[aa] a"); 75 /* Large size should fail */ 76 TEST_NPARSE("u:__test_event char[9999] a"); 77 /* Long size string should fail */ 78 TEST_NPARSE("u:__test_event char[0x0000000000001] a"); 79 } 80 81 TEST_F(user, loc_types) { 82 /* All should work */ 83 TEST_PARSE("u:__test_event __data_loc char[] a"); 84 TEST_PARSE("u:__test_event __data_loc unsigned char[] a"); 85 TEST_PARSE("u:__test_event __rel_loc char[] a"); 86 TEST_PARSE("u:__test_event __rel_loc unsigned char[] a"); 87 } 88 89 TEST_F(user, size_types) { 90 /* Should work */ 91 TEST_PARSE("u:__test_event struct custom a 20"); 92 /* Size not specified on struct should fail */ 93 TEST_NPARSE("u:__test_event struct custom a"); 94 /* Size specified on non-struct should fail */ 95 TEST_NPARSE("u:__test_event char a 20"); 96 } 97 98 TEST_F(user, flags) { 99 /* Should work */ 100 TEST_PARSE("u:__test_event:BPF_ITER u32 a"); 101 /* Forward compat */ 102 TEST_PARSE("u:__test_event:BPF_ITER,FLAG_FUTURE u32 a"); 103 } 104 105 TEST_F(user, matching) { 106 /* Register */ 107 ASSERT_NE(-1, Append("u:__test_event struct custom a 20")); 108 /* Should not match */ 109 TEST_NPARSE("!u:__test_event struct custom b"); 110 /* Should match */ 111 TEST_PARSE("!u:__test_event struct custom a"); 112 /* Multi field reg */ 113 ASSERT_NE(-1, Append("u:__test_event u32 a; u32 b")); 114 /* Non matching cases */ 115 TEST_NPARSE("!u:__test_event u32 a"); 116 TEST_NPARSE("!u:__test_event u32 b"); 117 TEST_NPARSE("!u:__test_event u32 a; u32 "); 118 TEST_NPARSE("!u:__test_event u32 a; u32 a"); 119 /* Matching case */ 120 TEST_PARSE("!u:__test_event u32 a; u32 b"); 121 /* Register */ 122 ASSERT_NE(-1, Append("u:__test_event u32 a; u32 b")); 123 /* Ensure trailing semi-colon case */ 124 TEST_PARSE("!u:__test_event u32 a; u32 b;"); 125 } 126 127 int main(int argc, char **argv) 128 { 129 return test_harness_run(argc, argv); 130 } 131