255 lines · c
1// SPDX-License-Identifier: GPL-2.02/*3 * User Events Perf Events Test Program4 *5 * Copyright (c) 2021 Beau Belgrave <beaub@linux.microsoft.com>6 */7 8#include <errno.h>9#include <linux/user_events.h>10#include <linux/perf_event.h>11#include <stdio.h>12#include <stdlib.h>13#include <fcntl.h>14#include <sys/ioctl.h>15#include <sys/stat.h>16#include <unistd.h>17#include <asm/unistd.h>18 19#include "../kselftest_harness.h"20#include "user_events_selftests.h"21 22const char *data_file = "/sys/kernel/tracing/user_events_data";23const char *id_file = "/sys/kernel/tracing/events/user_events/__test_event/id";24const char *fmt_file = "/sys/kernel/tracing/events/user_events/__test_event/format";25 26struct event {27 __u32 index;28 __u32 field1;29 __u32 field2;30};31 32static long perf_event_open(struct perf_event_attr *pe, pid_t pid,33 int cpu, int group_fd, unsigned long flags)34{35 return syscall(__NR_perf_event_open, pe, pid, cpu, group_fd, flags);36}37 38static int get_id(void)39{40 FILE *fp = fopen(id_file, "r");41 int ret, id = 0;42 43 if (!fp)44 return -1;45 46 ret = fscanf(fp, "%d", &id);47 fclose(fp);48 49 if (ret != 1)50 return -1;51 52 return id;53}54 55static int get_offset(void)56{57 FILE *fp = fopen(fmt_file, "r");58 int ret, c, last = 0, offset = 0;59 60 if (!fp)61 return -1;62 63 /* Read until empty line */64 while (true) {65 c = getc(fp);66 67 if (c == EOF)68 break;69 70 if (last == '\n' && c == '\n')71 break;72 73 last = c;74 }75 76 ret = fscanf(fp, "\tfield:u32 field1;\toffset:%d;", &offset);77 fclose(fp);78 79 if (ret != 1)80 return -1;81 82 return offset;83}84 85static int clear(int *check)86{87 struct user_unreg unreg = {0};88 89 unreg.size = sizeof(unreg);90 unreg.disable_bit = 31;91 unreg.disable_addr = (__u64)check;92 93 int fd = open(data_file, O_RDWR);94 95 if (fd == -1)96 return -1;97 98 if (ioctl(fd, DIAG_IOCSUNREG, &unreg) == -1)99 if (errno != ENOENT)100 return -1;101 102 if (ioctl(fd, DIAG_IOCSDEL, "__test_event") == -1)103 if (errno != ENOENT)104 return -1;105 106 close(fd);107 108 return 0;109}110 111FIXTURE(user) {112 int data_fd;113 int check;114 bool umount;115};116 117FIXTURE_SETUP(user) {118 USER_EVENT_FIXTURE_SETUP(return, self->umount);119 120 self->data_fd = open(data_file, O_RDWR);121 ASSERT_NE(-1, self->data_fd);122}123 124FIXTURE_TEARDOWN(user) {125 USER_EVENT_FIXTURE_TEARDOWN(self->umount);126 127 close(self->data_fd);128 129 if (clear(&self->check) != 0)130 printf("WARNING: Clear didn't work!\n");131}132 133TEST_F(user, perf_write) {134 struct perf_event_attr pe = {0};135 struct user_reg reg = {0};136 struct event event;137 struct perf_event_mmap_page *perf_page;138 int page_size = sysconf(_SC_PAGESIZE);139 int id, fd, offset;140 __u32 *val;141 142 reg.size = sizeof(reg);143 reg.name_args = (__u64)"__test_event u32 field1; u32 field2";144 reg.enable_bit = 31;145 reg.enable_addr = (__u64)&self->check;146 reg.enable_size = sizeof(self->check);147 148 /* Register should work */149 ASSERT_EQ(0, ioctl(self->data_fd, DIAG_IOCSREG, ®));150 ASSERT_EQ(0, reg.write_index);151 ASSERT_EQ(0, self->check);152 153 /* Id should be there */154 id = get_id();155 ASSERT_NE(-1, id);156 offset = get_offset();157 ASSERT_NE(-1, offset);158 159 pe.type = PERF_TYPE_TRACEPOINT;160 pe.size = sizeof(pe);161 pe.config = id;162 pe.sample_type = PERF_SAMPLE_RAW;163 pe.sample_period = 1;164 pe.wakeup_events = 1;165 166 /* Tracepoint attach should work */167 fd = perf_event_open(&pe, 0, -1, -1, 0);168 ASSERT_NE(-1, fd);169 170 perf_page = mmap(NULL, page_size * 2, PROT_READ, MAP_SHARED, fd, 0);171 ASSERT_NE(MAP_FAILED, perf_page);172 173 /* Status should be updated */174 ASSERT_EQ(1 << reg.enable_bit, self->check);175 176 event.index = reg.write_index;177 event.field1 = 0xc001;178 event.field2 = 0xc01a;179 180 /* Ensure write shows up at correct offset */181 ASSERT_NE(-1, write(self->data_fd, &event, sizeof(event)));182 val = (void *)(((char *)perf_page) + perf_page->data_offset);183 ASSERT_EQ(PERF_RECORD_SAMPLE, *val);184 /* Skip over header and size, move to offset */185 val += 3;186 val = (void *)((char *)val) + offset;187 /* Ensure correct */188 ASSERT_EQ(event.field1, *val++);189 ASSERT_EQ(event.field2, *val++);190 191 munmap(perf_page, page_size * 2);192 close(fd);193 194 /* Status should be updated */195 ASSERT_EQ(0, self->check);196}197 198TEST_F(user, perf_empty_events) {199 struct perf_event_attr pe = {0};200 struct user_reg reg = {0};201 struct perf_event_mmap_page *perf_page;202 int page_size = sysconf(_SC_PAGESIZE);203 int id, fd;204 __u32 *val;205 206 reg.size = sizeof(reg);207 reg.name_args = (__u64)"__test_event";208 reg.enable_bit = 31;209 reg.enable_addr = (__u64)&self->check;210 reg.enable_size = sizeof(self->check);211 212 /* Register should work */213 ASSERT_EQ(0, ioctl(self->data_fd, DIAG_IOCSREG, ®));214 ASSERT_EQ(0, reg.write_index);215 ASSERT_EQ(0, self->check);216 217 /* Id should be there */218 id = get_id();219 ASSERT_NE(-1, id);220 221 pe.type = PERF_TYPE_TRACEPOINT;222 pe.size = sizeof(pe);223 pe.config = id;224 pe.sample_type = PERF_SAMPLE_RAW;225 pe.sample_period = 1;226 pe.wakeup_events = 1;227 228 /* Tracepoint attach should work */229 fd = perf_event_open(&pe, 0, -1, -1, 0);230 ASSERT_NE(-1, fd);231 232 perf_page = mmap(NULL, page_size * 2, PROT_READ, MAP_SHARED, fd, 0);233 ASSERT_NE(MAP_FAILED, perf_page);234 235 /* Status should be updated */236 ASSERT_EQ(1 << reg.enable_bit, self->check);237 238 /* Ensure write shows up at correct offset */239 ASSERT_NE(-1, write(self->data_fd, ®.write_index,240 sizeof(reg.write_index)));241 val = (void *)(((char *)perf_page) + perf_page->data_offset);242 ASSERT_EQ(PERF_RECORD_SAMPLE, *val);243 244 munmap(perf_page, page_size * 2);245 close(fd);246 247 /* Status should be updated */248 ASSERT_EQ(0, self->check);249}250 251int main(int argc, char **argv)252{253 return test_harness_run(argc, argv);254}255