| /* |
| * QEMU Plugin Core code |
| * |
| * This is the core code that deals with injecting instrumentation into the code |
| * |
| * Copyright (C) 2017, Emilio G. Cota <cota@braap.org> |
| * Copyright (C) 2019, Linaro |
| * |
| * License: GNU GPL, version 2 or later. |
| * See the COPYING file in the top-level directory. |
| * |
| * SPDX-License-Identifier: GPL-2.0-or-later |
| */ |
| #include "qemu/osdep.h" |
| #include "qemu/error-report.h" |
| #include "qemu/config-file.h" |
| #include "qapi/error.h" |
| #include "qemu/lockable.h" |
| #include "qemu/option.h" |
| #include "qemu/plugin.h" |
| #include "qemu/queue.h" |
| #include "qemu/rcu_queue.h" |
| #include "qemu/xxhash.h" |
| #include "qemu/rcu.h" |
| #include "hw/core/cpu.h" |
| |
| #include "exec/exec-all.h" |
| #include "exec/tb-flush.h" |
| #include "tcg/tcg.h" |
| #include "tcg/tcg-op.h" |
| #include "plugin.h" |
| |
| struct qemu_plugin_cb { |
| struct qemu_plugin_ctx *ctx; |
| union qemu_plugin_cb_sig f; |
| void *udata; |
| QLIST_ENTRY(qemu_plugin_cb) entry; |
| }; |
| |
| struct qemu_plugin_state plugin; |
| |
| struct qemu_plugin_ctx *plugin_id_to_ctx_locked(qemu_plugin_id_t id) |
| { |
| struct qemu_plugin_ctx *ctx; |
| qemu_plugin_id_t *id_p; |
| |
| id_p = g_hash_table_lookup(plugin.id_ht, &id); |
| ctx = container_of(id_p, struct qemu_plugin_ctx, id); |
| if (ctx == NULL) { |
| error_report("plugin: invalid plugin id %" PRIu64, id); |
| abort(); |
| } |
| return ctx; |
| } |
| |
| static void plugin_cpu_update__async(CPUState *cpu, run_on_cpu_data data) |
| { |
| bitmap_copy(cpu->plugin_state->event_mask, |
| &data.host_ulong, QEMU_PLUGIN_EV_MAX); |
| tcg_flush_jmp_cache(cpu); |
| } |
| |
| static void plugin_cpu_update__locked(gpointer k, gpointer v, gpointer udata) |
| { |
| CPUState *cpu = container_of(k, CPUState, cpu_index); |
| run_on_cpu_data mask = RUN_ON_CPU_HOST_ULONG(*plugin.mask); |
| |
| if (DEVICE(cpu)->realized) { |
| async_run_on_cpu(cpu, plugin_cpu_update__async, mask); |
| } else { |
| plugin_cpu_update__async(cpu, mask); |
| } |
| } |
| |
| void plugin_unregister_cb__locked(struct qemu_plugin_ctx *ctx, |
| enum qemu_plugin_event ev) |
| { |
| struct qemu_plugin_cb *cb = ctx->callbacks[ev]; |
| |
| if (cb == NULL) { |
| return; |
| } |
| QLIST_REMOVE_RCU(cb, entry); |
| g_free(cb); |
| ctx->callbacks[ev] = NULL; |
| if (QLIST_EMPTY_RCU(&plugin.cb_lists[ev])) { |
| clear_bit(ev, plugin.mask); |
| g_hash_table_foreach(plugin.cpu_ht, plugin_cpu_update__locked, NULL); |
| } |
| } |
| |
| /* |
| * Disable CFI checks. |
| * The callback function has been loaded from an external library so we do not |
| * have type information |
| */ |
| QEMU_DISABLE_CFI |
| static void plugin_vcpu_cb__simple(CPUState *cpu, enum qemu_plugin_event ev) |
| { |
| struct qemu_plugin_cb *cb, *next; |
| |
| switch (ev) { |
| case QEMU_PLUGIN_EV_VCPU_INIT: |
| case QEMU_PLUGIN_EV_VCPU_EXIT: |
| case QEMU_PLUGIN_EV_VCPU_IDLE: |
| case QEMU_PLUGIN_EV_VCPU_RESUME: |
| /* iterate safely; plugins might uninstall themselves at any time */ |
| QLIST_FOREACH_SAFE_RCU(cb, &plugin.cb_lists[ev], entry, next) { |
| qemu_plugin_vcpu_simple_cb_t func = cb->f.vcpu_simple; |
| |
| func(cb->ctx->id, cpu->cpu_index); |
| } |
| break; |
| default: |
| g_assert_not_reached(); |
| } |
| } |
| |
| /* |
| * Disable CFI checks. |
| * The callback function has been loaded from an external library so we do not |
| * have type information |
| */ |
| QEMU_DISABLE_CFI |
| static void plugin_cb__simple(enum qemu_plugin_event ev) |
| { |
| struct qemu_plugin_cb *cb, *next; |
| |
| switch (ev) { |
| case QEMU_PLUGIN_EV_FLUSH: |
| QLIST_FOREACH_SAFE_RCU(cb, &plugin.cb_lists[ev], entry, next) { |
| qemu_plugin_simple_cb_t func = cb->f.simple; |
| |
| func(cb->ctx->id); |
| } |
| break; |
| default: |
| g_assert_not_reached(); |
| } |
| } |
| |
| /* |
| * Disable CFI checks. |
| * The callback function has been loaded from an external library so we do not |
| * have type information |
| */ |
| QEMU_DISABLE_CFI |
| static void plugin_cb__udata(enum qemu_plugin_event ev) |
| { |
| struct qemu_plugin_cb *cb, *next; |
| |
| switch (ev) { |
| case QEMU_PLUGIN_EV_ATEXIT: |
| QLIST_FOREACH_SAFE_RCU(cb, &plugin.cb_lists[ev], entry, next) { |
| qemu_plugin_udata_cb_t func = cb->f.udata; |
| |
| func(cb->ctx->id, cb->udata); |
| } |
| break; |
| default: |
| g_assert_not_reached(); |
| } |
| } |
| |
| static void |
| do_plugin_register_cb(qemu_plugin_id_t id, enum qemu_plugin_event ev, |
| void *func, void *udata) |
| { |
| struct qemu_plugin_ctx *ctx; |
| |
| QEMU_LOCK_GUARD(&plugin.lock); |
| ctx = plugin_id_to_ctx_locked(id); |
| /* if the plugin is on its way out, ignore this request */ |
| if (unlikely(ctx->uninstalling)) { |
| return; |
| } |
| if (func) { |
| struct qemu_plugin_cb *cb = ctx->callbacks[ev]; |
| |
| if (cb) { |
| cb->f.generic = func; |
| cb->udata = udata; |
| } else { |
| cb = g_new(struct qemu_plugin_cb, 1); |
| cb->ctx = ctx; |
| cb->f.generic = func; |
| cb->udata = udata; |
| ctx->callbacks[ev] = cb; |
| QLIST_INSERT_HEAD_RCU(&plugin.cb_lists[ev], cb, entry); |
| if (!test_bit(ev, plugin.mask)) { |
| set_bit(ev, plugin.mask); |
| g_hash_table_foreach(plugin.cpu_ht, plugin_cpu_update__locked, |
| NULL); |
| } |
| } |
| } else { |
| plugin_unregister_cb__locked(ctx, ev); |
| } |
| } |
| |
| void plugin_register_cb(qemu_plugin_id_t id, enum qemu_plugin_event ev, |
| void *func) |
| { |
| do_plugin_register_cb(id, ev, func, NULL); |
| } |
| |
| void |
| plugin_register_cb_udata(qemu_plugin_id_t id, enum qemu_plugin_event ev, |
| void *func, void *udata) |
| { |
| do_plugin_register_cb(id, ev, func, udata); |
| } |
| |
| CPUPluginState *qemu_plugin_create_vcpu_state(void) |
| { |
| return g_new0(CPUPluginState, 1); |
| } |
| |
| static void plugin_grow_scoreboards__locked(CPUState *cpu) |
| { |
| if (cpu->cpu_index < plugin.scoreboard_alloc_size) { |
| return; |
| } |
| |
| bool need_realloc = FALSE; |
| while (cpu->cpu_index >= plugin.scoreboard_alloc_size) { |
| plugin.scoreboard_alloc_size *= 2; |
| need_realloc = TRUE; |
| } |
| |
| |
| if (!need_realloc || QLIST_EMPTY(&plugin.scoreboards)) { |
| /* nothing to do, we just updated sizes for future scoreboards */ |
| return; |
| } |
| |
| /* cpus must be stopped, as tb might still use an existing scoreboard. */ |
| start_exclusive(); |
| struct qemu_plugin_scoreboard *score; |
| QLIST_FOREACH(score, &plugin.scoreboards, entry) { |
| g_array_set_size(score->data, plugin.scoreboard_alloc_size); |
| } |
| /* force all tb to be flushed, as scoreboard pointers were changed. */ |
| tb_flush(cpu); |
| end_exclusive(); |
| } |
| |
| void qemu_plugin_vcpu_init_hook(CPUState *cpu) |
| { |
| bool success; |
| |
| qemu_rec_mutex_lock(&plugin.lock); |
| plugin.num_vcpus = MAX(plugin.num_vcpus, cpu->cpu_index + 1); |
| plugin_cpu_update__locked(&cpu->cpu_index, NULL, NULL); |
| success = g_hash_table_insert(plugin.cpu_ht, &cpu->cpu_index, |
| &cpu->cpu_index); |
| g_assert(success); |
| plugin_grow_scoreboards__locked(cpu); |
| qemu_rec_mutex_unlock(&plugin.lock); |
| |
| plugin_vcpu_cb__simple(cpu, QEMU_PLUGIN_EV_VCPU_INIT); |
| } |
| |
| void qemu_plugin_vcpu_exit_hook(CPUState *cpu) |
| { |
| bool success; |
| |
| plugin_vcpu_cb__simple(cpu, QEMU_PLUGIN_EV_VCPU_EXIT); |
| |
| qemu_rec_mutex_lock(&plugin.lock); |
| success = g_hash_table_remove(plugin.cpu_ht, &cpu->cpu_index); |
| g_assert(success); |
| qemu_rec_mutex_unlock(&plugin.lock); |
| } |
| |
| struct plugin_for_each_args { |
| struct qemu_plugin_ctx *ctx; |
| qemu_plugin_vcpu_simple_cb_t cb; |
| }; |
| |
| static void plugin_vcpu_for_each(gpointer k, gpointer v, gpointer udata) |
| { |
| struct plugin_for_each_args *args = udata; |
| int cpu_index = *(int *)k; |
| |
| args->cb(args->ctx->id, cpu_index); |
| } |
| |
| void qemu_plugin_vcpu_for_each(qemu_plugin_id_t id, |
| qemu_plugin_vcpu_simple_cb_t cb) |
| { |
| struct plugin_for_each_args args; |
| |
| if (cb == NULL) { |
| return; |
| } |
| qemu_rec_mutex_lock(&plugin.lock); |
| args.ctx = plugin_id_to_ctx_locked(id); |
| args.cb = cb; |
| g_hash_table_foreach(plugin.cpu_ht, plugin_vcpu_for_each, &args); |
| qemu_rec_mutex_unlock(&plugin.lock); |
| } |
| |
| /* Allocate and return a callback record */ |
| static struct qemu_plugin_dyn_cb *plugin_get_dyn_cb(GArray **arr) |
| { |
| GArray *cbs = *arr; |
| |
| if (!cbs) { |
| cbs = g_array_sized_new(false, true, |
| sizeof(struct qemu_plugin_dyn_cb), 1); |
| *arr = cbs; |
| } |
| |
| g_array_set_size(cbs, cbs->len + 1); |
| return &g_array_index(cbs, struct qemu_plugin_dyn_cb, cbs->len - 1); |
| } |
| |
| static enum plugin_dyn_cb_type op_to_cb_type(enum qemu_plugin_op op) |
| { |
| switch (op) { |
| case QEMU_PLUGIN_INLINE_ADD_U64: |
| return PLUGIN_CB_INLINE_ADD_U64; |
| case QEMU_PLUGIN_INLINE_STORE_U64: |
| return PLUGIN_CB_INLINE_STORE_U64; |
| default: |
| g_assert_not_reached(); |
| } |
| } |
| |
| void plugin_register_inline_op_on_entry(GArray **arr, |
| enum qemu_plugin_mem_rw rw, |
| enum qemu_plugin_op op, |
| qemu_plugin_u64 entry, |
| uint64_t imm) |
| { |
| struct qemu_plugin_dyn_cb *dyn_cb; |
| |
| struct qemu_plugin_inline_cb inline_cb = { .rw = rw, |
| .entry = entry, |
| .imm = imm }; |
| dyn_cb = plugin_get_dyn_cb(arr); |
| dyn_cb->type = op_to_cb_type(op); |
| dyn_cb->inline_insn = inline_cb; |
| } |
| |
| void plugin_register_dyn_cb__udata(GArray **arr, |
| qemu_plugin_vcpu_udata_cb_t cb, |
| enum qemu_plugin_cb_flags flags, |
| void *udata) |
| { |
| static TCGHelperInfo info[3] = { |
| [QEMU_PLUGIN_CB_NO_REGS].flags = TCG_CALL_NO_RWG, |
| [QEMU_PLUGIN_CB_R_REGS].flags = TCG_CALL_NO_WG, |
| /* |
| * Match qemu_plugin_vcpu_udata_cb_t: |
| * void (*)(uint32_t, void *) |
| */ |
| [0 ... 2].typemask = (dh_typemask(void, 0) | |
| dh_typemask(i32, 1) | |
| dh_typemask(ptr, 2)) |
| }; |
| assert((unsigned)flags < ARRAY_SIZE(info)); |
| |
| struct qemu_plugin_dyn_cb *dyn_cb = plugin_get_dyn_cb(arr); |
| struct qemu_plugin_regular_cb regular_cb = { .f.vcpu_udata = cb, |
| .userp = udata, |
| .info = &info[flags] }; |
| dyn_cb->type = PLUGIN_CB_REGULAR; |
| dyn_cb->regular = regular_cb; |
| } |
| |
| void plugin_register_dyn_cond_cb__udata(GArray **arr, |
| qemu_plugin_vcpu_udata_cb_t cb, |
| enum qemu_plugin_cb_flags flags, |
| enum qemu_plugin_cond cond, |
| qemu_plugin_u64 entry, |
| uint64_t imm, |
| void *udata) |
| { |
| static TCGHelperInfo info[3] = { |
| [QEMU_PLUGIN_CB_NO_REGS].flags = TCG_CALL_NO_RWG, |
| [QEMU_PLUGIN_CB_R_REGS].flags = TCG_CALL_NO_WG, |
| /* |
| * Match qemu_plugin_vcpu_udata_cb_t: |
| * void (*)(uint32_t, void *) |
| */ |
| [0 ... 2].typemask = (dh_typemask(void, 0) | |
| dh_typemask(i32, 1) | |
| dh_typemask(ptr, 2)) |
| }; |
| assert((unsigned)flags < ARRAY_SIZE(info)); |
| |
| struct qemu_plugin_dyn_cb *dyn_cb = plugin_get_dyn_cb(arr); |
| struct qemu_plugin_conditional_cb cond_cb = { .userp = udata, |
| .f.vcpu_udata = cb, |
| .cond = cond, |
| .entry = entry, |
| .imm = imm, |
| .info = &info[flags] }; |
| dyn_cb->type = PLUGIN_CB_COND; |
| dyn_cb->cond = cond_cb; |
| } |
| |
| void plugin_register_vcpu_mem_cb(GArray **arr, |
| void *cb, |
| enum qemu_plugin_cb_flags flags, |
| enum qemu_plugin_mem_rw rw, |
| void *udata) |
| { |
| /* |
| * Expect that the underlying type for enum qemu_plugin_meminfo_t |
| * is either int32_t or uint32_t, aka int or unsigned int. |
| */ |
| QEMU_BUILD_BUG_ON( |
| !__builtin_types_compatible_p(qemu_plugin_meminfo_t, uint32_t) && |
| !__builtin_types_compatible_p(qemu_plugin_meminfo_t, int32_t)); |
| |
| static TCGHelperInfo info[3] = { |
| [QEMU_PLUGIN_CB_NO_REGS].flags = TCG_CALL_NO_RWG, |
| [QEMU_PLUGIN_CB_R_REGS].flags = TCG_CALL_NO_WG, |
| /* |
| * Match qemu_plugin_vcpu_mem_cb_t: |
| * void (*)(uint32_t, qemu_plugin_meminfo_t, uint64_t, void *) |
| */ |
| [0 ... 2].typemask = |
| (dh_typemask(void, 0) | |
| dh_typemask(i32, 1) | |
| (__builtin_types_compatible_p(qemu_plugin_meminfo_t, uint32_t) |
| ? dh_typemask(i32, 2) : dh_typemask(s32, 2)) | |
| dh_typemask(i64, 3) | |
| dh_typemask(ptr, 4)) |
| }; |
| assert((unsigned)flags < ARRAY_SIZE(info)); |
| |
| struct qemu_plugin_dyn_cb *dyn_cb = plugin_get_dyn_cb(arr); |
| struct qemu_plugin_regular_cb regular_cb = { .userp = udata, |
| .rw = rw, |
| .f.vcpu_mem = cb, |
| .info = &info[flags] }; |
| dyn_cb->type = PLUGIN_CB_MEM_REGULAR; |
| dyn_cb->regular = regular_cb; |
| } |
| |
| /* |
| * Disable CFI checks. |
| * The callback function has been loaded from an external library so we do not |
| * have type information |
| */ |
| QEMU_DISABLE_CFI |
| void qemu_plugin_tb_trans_cb(CPUState *cpu, struct qemu_plugin_tb *tb) |
| { |
| struct qemu_plugin_cb *cb, *next; |
| enum qemu_plugin_event ev = QEMU_PLUGIN_EV_VCPU_TB_TRANS; |
| |
| /* no plugin_state->event_mask check here; caller should have checked */ |
| |
| QLIST_FOREACH_SAFE_RCU(cb, &plugin.cb_lists[ev], entry, next) { |
| qemu_plugin_vcpu_tb_trans_cb_t func = cb->f.vcpu_tb_trans; |
| |
| func(cb->ctx->id, tb); |
| } |
| } |
| |
| /* |
| * Disable CFI checks. |
| * The callback function has been loaded from an external library so we do not |
| * have type information |
| */ |
| QEMU_DISABLE_CFI |
| void |
| qemu_plugin_vcpu_syscall(CPUState *cpu, int64_t num, uint64_t a1, uint64_t a2, |
| uint64_t a3, uint64_t a4, uint64_t a5, |
| uint64_t a6, uint64_t a7, uint64_t a8) |
| { |
| struct qemu_plugin_cb *cb, *next; |
| enum qemu_plugin_event ev = QEMU_PLUGIN_EV_VCPU_SYSCALL; |
| |
| if (!test_bit(ev, cpu->plugin_state->event_mask)) { |
| return; |
| } |
| |
| QLIST_FOREACH_SAFE_RCU(cb, &plugin.cb_lists[ev], entry, next) { |
| qemu_plugin_vcpu_syscall_cb_t func = cb->f.vcpu_syscall; |
| |
| func(cb->ctx->id, cpu->cpu_index, num, a1, a2, a3, a4, a5, a6, a7, a8); |
| } |
| } |
| |
| /* |
| * Disable CFI checks. |
| * The callback function has been loaded from an external library so we do not |
| * have type information |
| */ |
| QEMU_DISABLE_CFI |
| void qemu_plugin_vcpu_syscall_ret(CPUState *cpu, int64_t num, int64_t ret) |
| { |
| struct qemu_plugin_cb *cb, *next; |
| enum qemu_plugin_event ev = QEMU_PLUGIN_EV_VCPU_SYSCALL_RET; |
| |
| if (!test_bit(ev, cpu->plugin_state->event_mask)) { |
| return; |
| } |
| |
| QLIST_FOREACH_SAFE_RCU(cb, &plugin.cb_lists[ev], entry, next) { |
| qemu_plugin_vcpu_syscall_ret_cb_t func = cb->f.vcpu_syscall_ret; |
| |
| func(cb->ctx->id, cpu->cpu_index, num, ret); |
| } |
| } |
| |
| void qemu_plugin_vcpu_idle_cb(CPUState *cpu) |
| { |
| /* idle and resume cb may be called before init, ignore in this case */ |
| if (cpu->cpu_index < plugin.num_vcpus) { |
| plugin_vcpu_cb__simple(cpu, QEMU_PLUGIN_EV_VCPU_IDLE); |
| } |
| } |
| |
| void qemu_plugin_vcpu_resume_cb(CPUState *cpu) |
| { |
| if (cpu->cpu_index < plugin.num_vcpus) { |
| plugin_vcpu_cb__simple(cpu, QEMU_PLUGIN_EV_VCPU_RESUME); |
| } |
| } |
| |
| void qemu_plugin_register_vcpu_idle_cb(qemu_plugin_id_t id, |
| qemu_plugin_vcpu_simple_cb_t cb) |
| { |
| plugin_register_cb(id, QEMU_PLUGIN_EV_VCPU_IDLE, cb); |
| } |
| |
| void qemu_plugin_register_vcpu_resume_cb(qemu_plugin_id_t id, |
| qemu_plugin_vcpu_simple_cb_t cb) |
| { |
| plugin_register_cb(id, QEMU_PLUGIN_EV_VCPU_RESUME, cb); |
| } |
| |
| void qemu_plugin_register_flush_cb(qemu_plugin_id_t id, |
| qemu_plugin_simple_cb_t cb) |
| { |
| plugin_register_cb(id, QEMU_PLUGIN_EV_FLUSH, cb); |
| } |
| |
| static bool free_dyn_cb_arr(void *p, uint32_t h, void *userp) |
| { |
| g_array_free((GArray *) p, true); |
| return true; |
| } |
| |
| void qemu_plugin_flush_cb(void) |
| { |
| qht_iter_remove(&plugin.dyn_cb_arr_ht, free_dyn_cb_arr, NULL); |
| qht_reset(&plugin.dyn_cb_arr_ht); |
| |
| plugin_cb__simple(QEMU_PLUGIN_EV_FLUSH); |
| } |
| |
| void exec_inline_op(enum plugin_dyn_cb_type type, |
| struct qemu_plugin_inline_cb *cb, |
| int cpu_index) |
| { |
| char *ptr = cb->entry.score->data->data; |
| size_t elem_size = g_array_get_element_size( |
| cb->entry.score->data); |
| size_t offset = cb->entry.offset; |
| uint64_t *val = (uint64_t *)(ptr + offset + cpu_index * elem_size); |
| |
| switch (type) { |
| case PLUGIN_CB_INLINE_ADD_U64: |
| *val += cb->imm; |
| break; |
| case PLUGIN_CB_INLINE_STORE_U64: |
| *val = cb->imm; |
| break; |
| default: |
| g_assert_not_reached(); |
| } |
| } |
| |
| void qemu_plugin_vcpu_mem_cb(CPUState *cpu, uint64_t vaddr, |
| MemOpIdx oi, enum qemu_plugin_mem_rw rw) |
| { |
| GArray *arr = cpu->neg.plugin_mem_cbs; |
| size_t i; |
| |
| if (arr == NULL) { |
| return; |
| } |
| for (i = 0; i < arr->len; i++) { |
| struct qemu_plugin_dyn_cb *cb = |
| &g_array_index(arr, struct qemu_plugin_dyn_cb, i); |
| |
| switch (cb->type) { |
| case PLUGIN_CB_MEM_REGULAR: |
| if (rw && cb->regular.rw) { |
| cb->regular.f.vcpu_mem(cpu->cpu_index, |
| make_plugin_meminfo(oi, rw), |
| vaddr, cb->regular.userp); |
| } |
| break; |
| case PLUGIN_CB_INLINE_ADD_U64: |
| case PLUGIN_CB_INLINE_STORE_U64: |
| if (rw && cb->inline_insn.rw) { |
| exec_inline_op(cb->type, &cb->inline_insn, cpu->cpu_index); |
| } |
| break; |
| default: |
| g_assert_not_reached(); |
| } |
| } |
| } |
| |
| void qemu_plugin_atexit_cb(void) |
| { |
| plugin_cb__udata(QEMU_PLUGIN_EV_ATEXIT); |
| } |
| |
| void qemu_plugin_register_atexit_cb(qemu_plugin_id_t id, |
| qemu_plugin_udata_cb_t cb, |
| void *udata) |
| { |
| plugin_register_cb_udata(id, QEMU_PLUGIN_EV_ATEXIT, cb, udata); |
| } |
| |
| /* |
| * Handle exit from linux-user. Unlike the normal atexit() mechanism |
| * we need to handle the clean-up manually as it's possible threads |
| * are still running. We need to remove all callbacks from code |
| * generation, flush the current translations and then we can safely |
| * trigger the exit callbacks. |
| */ |
| |
| void qemu_plugin_user_exit(void) |
| { |
| enum qemu_plugin_event ev; |
| CPUState *cpu; |
| |
| /* |
| * Locking order: we must acquire locks in an order that is consistent |
| * with the one in fork_start(). That is: |
| * - start_exclusive(), which acquires qemu_cpu_list_lock, |
| * must be called before acquiring plugin.lock. |
| * - tb_flush(), which acquires mmap_lock(), must be called |
| * while plugin.lock is not held. |
| */ |
| start_exclusive(); |
| |
| qemu_rec_mutex_lock(&plugin.lock); |
| /* un-register all callbacks except the final AT_EXIT one */ |
| for (ev = 0; ev < QEMU_PLUGIN_EV_MAX; ev++) { |
| if (ev != QEMU_PLUGIN_EV_ATEXIT) { |
| struct qemu_plugin_cb *cb, *next; |
| |
| QLIST_FOREACH_SAFE_RCU(cb, &plugin.cb_lists[ev], entry, next) { |
| plugin_unregister_cb__locked(cb->ctx, ev); |
| } |
| } |
| } |
| CPU_FOREACH(cpu) { |
| qemu_plugin_disable_mem_helpers(cpu); |
| } |
| qemu_rec_mutex_unlock(&plugin.lock); |
| |
| tb_flush(current_cpu); |
| end_exclusive(); |
| |
| /* now it's safe to handle the exit case */ |
| qemu_plugin_atexit_cb(); |
| } |
| |
| /* |
| * Helpers for *-user to ensure locks are sane across fork() events. |
| */ |
| |
| void qemu_plugin_user_prefork_lock(void) |
| { |
| qemu_rec_mutex_lock(&plugin.lock); |
| } |
| |
| void qemu_plugin_user_postfork(bool is_child) |
| { |
| if (is_child) { |
| /* should we just reset via plugin_init? */ |
| qemu_rec_mutex_init(&plugin.lock); |
| } else { |
| qemu_rec_mutex_unlock(&plugin.lock); |
| } |
| } |
| |
| static bool plugin_dyn_cb_arr_cmp(const void *ap, const void *bp) |
| { |
| return ap == bp; |
| } |
| |
| static void __attribute__((__constructor__)) plugin_init(void) |
| { |
| int i; |
| |
| for (i = 0; i < QEMU_PLUGIN_EV_MAX; i++) { |
| QLIST_INIT(&plugin.cb_lists[i]); |
| } |
| qemu_rec_mutex_init(&plugin.lock); |
| plugin.id_ht = g_hash_table_new(g_int64_hash, g_int64_equal); |
| plugin.cpu_ht = g_hash_table_new(g_int_hash, g_int_equal); |
| QLIST_INIT(&plugin.scoreboards); |
| plugin.scoreboard_alloc_size = 16; /* avoid frequent reallocation */ |
| QTAILQ_INIT(&plugin.ctxs); |
| qht_init(&plugin.dyn_cb_arr_ht, plugin_dyn_cb_arr_cmp, 16, |
| QHT_MODE_AUTO_RESIZE); |
| atexit(qemu_plugin_atexit_cb); |
| } |
| |
| int plugin_num_vcpus(void) |
| { |
| return plugin.num_vcpus; |
| } |
| |
| struct qemu_plugin_scoreboard *plugin_scoreboard_new(size_t element_size) |
| { |
| struct qemu_plugin_scoreboard *score = |
| g_malloc0(sizeof(struct qemu_plugin_scoreboard)); |
| score->data = g_array_new(FALSE, TRUE, element_size); |
| g_array_set_size(score->data, plugin.scoreboard_alloc_size); |
| |
| qemu_rec_mutex_lock(&plugin.lock); |
| QLIST_INSERT_HEAD(&plugin.scoreboards, score, entry); |
| qemu_rec_mutex_unlock(&plugin.lock); |
| |
| return score; |
| } |
| |
| void plugin_scoreboard_free(struct qemu_plugin_scoreboard *score) |
| { |
| qemu_rec_mutex_lock(&plugin.lock); |
| QLIST_REMOVE(score, entry); |
| qemu_rec_mutex_unlock(&plugin.lock); |
| |
| g_array_free(score->data, TRUE); |
| g_free(score); |
| } |