^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 1) // SPDX-License-Identifier: GPL-2.0
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 2) /*
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 3) * perf_hooks.c
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 4) *
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 5) * Copyright (C) 2016 Wang Nan <wangnan0@huawei.com>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 6) * Copyright (C) 2016 Huawei Inc.
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 7) */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 8)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 9) #include <errno.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 10) #include <stdlib.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 11) #include <string.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 12) #include <setjmp.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 13) #include <linux/err.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 14) #include <linux/kernel.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 15) #include "util/debug.h"
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 16) #include "util/perf-hooks.h"
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 17)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 18) static sigjmp_buf jmpbuf;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 19) static const struct perf_hook_desc *current_perf_hook;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 20)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 21) void perf_hooks__invoke(const struct perf_hook_desc *desc)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 22) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 23) if (!(desc && desc->p_hook_func && *desc->p_hook_func))
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 24) return;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 25)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 26) if (sigsetjmp(jmpbuf, 1)) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 27) pr_warning("Fatal error (SEGFAULT) in perf hook '%s'\n",
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 28) desc->hook_name);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 29) *(current_perf_hook->p_hook_func) = NULL;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 30) } else {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 31) current_perf_hook = desc;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 32) (**desc->p_hook_func)(desc->hook_ctx);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 33) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 34) current_perf_hook = NULL;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 35) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 36)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 37) void perf_hooks__recover(void)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 38) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 39) if (current_perf_hook)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 40) siglongjmp(jmpbuf, 1);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 41) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 42)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 43) #define PERF_HOOK(name) \
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 44) perf_hook_func_t __perf_hook_func_##name = NULL; \
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 45) struct perf_hook_desc __perf_hook_desc_##name = \
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 46) {.hook_name = #name, \
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 47) .p_hook_func = &__perf_hook_func_##name, \
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 48) .hook_ctx = NULL};
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 49) #include "perf-hooks-list.h"
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 50) #undef PERF_HOOK
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 51)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 52) #define PERF_HOOK(name) \
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 53) &__perf_hook_desc_##name,
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 54)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 55) static struct perf_hook_desc *perf_hooks[] = {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 56) #include "perf-hooks-list.h"
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 57) };
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 58) #undef PERF_HOOK
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 59)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 60) int perf_hooks__set_hook(const char *hook_name,
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 61) perf_hook_func_t hook_func,
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 62) void *hook_ctx)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 63) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 64) unsigned int i;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 65)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 66) for (i = 0; i < ARRAY_SIZE(perf_hooks); i++) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 67) if (strcmp(hook_name, perf_hooks[i]->hook_name) != 0)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 68) continue;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 69)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 70) if (*(perf_hooks[i]->p_hook_func))
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 71) pr_warning("Overwrite existing hook: %s\n", hook_name);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 72) *(perf_hooks[i]->p_hook_func) = hook_func;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 73) perf_hooks[i]->hook_ctx = hook_ctx;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 74) return 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 75) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 76) return -ENOENT;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 77) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 78)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 79) perf_hook_func_t perf_hooks__get_hook(const char *hook_name)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 80) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 81) unsigned int i;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 82)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 83) for (i = 0; i < ARRAY_SIZE(perf_hooks); i++) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 84) if (strcmp(hook_name, perf_hooks[i]->hook_name) != 0)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 85) continue;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 86)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 87) return *(perf_hooks[i]->p_hook_func);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 88) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 89) return ERR_PTR(-ENOENT);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 90) }