| 1 | /* |
| 2 | * Copyright (C) 2023, Pierrick Bouvier <pierrick.bouvier@linaro.org> |
| 3 | * |
| 4 | * Demonstrates and tests usage of inline ops. |
| 5 | * |
| 6 | * License: GNU GPL, version 2 or later. |
| 7 | * See the COPYING file in the top-level directory. |
| 8 | */ |
| 9 | |
| 10 | #include <glib.h> |
| 11 | #include <stdint.h> |
| 12 | #include <stdio.h> |
| 13 | |
| 14 | #include <qemu-plugin.h> |
| 15 | |
| 16 | typedef struct { |
| 17 | uint64_t count_tb; |
| 18 | uint64_t count_tb_inline; |
| 19 | uint64_t count_insn; |
| 20 | uint64_t count_insn_inline; |
| 21 | uint64_t count_mem; |
| 22 | uint64_t count_mem_inline; |
| 23 | uint64_t tb_cond_num_trigger; |
| 24 | uint64_t tb_cond_track_count; |
| 25 | uint64_t insn_cond_num_trigger; |
| 26 | uint64_t insn_cond_track_count; |
| 27 | } CPUCount; |
| 28 | |
| 29 | static const uint64_t cond_trigger_limit = 100; |
| 30 | |
| 31 | typedef struct { |
| 32 | uint64_t data_insn; |
| 33 | uint64_t data_tb; |
| 34 | uint64_t data_mem; |
| 35 | } CPUData; |
| 36 | |
| 37 | static struct qemu_plugin_scoreboard *counts; |
| 38 | static qemu_plugin_u64 count_tb; |
| 39 | static qemu_plugin_u64 count_tb_inline; |
| 40 | static qemu_plugin_u64 count_insn; |
| 41 | static qemu_plugin_u64 count_insn_inline; |
| 42 | static qemu_plugin_u64 count_mem; |
| 43 | static qemu_plugin_u64 count_mem_inline; |
| 44 | static qemu_plugin_u64 tb_cond_num_trigger; |
| 45 | static qemu_plugin_u64 tb_cond_track_count; |
| 46 | static qemu_plugin_u64 insn_cond_num_trigger; |
| 47 | static qemu_plugin_u64 insn_cond_track_count; |
| 48 | static struct qemu_plugin_scoreboard *data; |
| 49 | static qemu_plugin_u64 data_insn; |
| 50 | static qemu_plugin_u64 data_tb; |
| 51 | static qemu_plugin_u64 data_mem; |
| 52 | |
| 53 | static uint64_t global_count_tb; |
| 54 | static uint64_t global_count_insn; |
| 55 | static uint64_t global_count_mem; |
| 56 | static unsigned int max_cpu_index; |
| 57 | static GMutex tb_lock; |
| 58 | static GMutex insn_lock; |
| 59 | static GMutex mem_lock; |
| 60 | |
| 61 | QEMU_PLUGIN_EXPORT int qemu_plugin_version = QEMU_PLUGIN_VERSION; |
| 62 | |
| 63 | static void stats_insn(void) |
| 64 | { |
| 65 | const uint64_t expected = global_count_insn; |
| 66 | const uint64_t per_vcpu = qemu_plugin_u64_sum(count_insn); |
| 67 | const uint64_t inl_per_vcpu = |
| 68 | qemu_plugin_u64_sum(count_insn_inline); |
| 69 | const uint64_t cond_num_trigger = |
| 70 | qemu_plugin_u64_sum(insn_cond_num_trigger); |
| 71 | const uint64_t cond_track_left = qemu_plugin_u64_sum(insn_cond_track_count); |
| 72 | const uint64_t conditional = |
| 73 | cond_num_trigger * cond_trigger_limit + cond_track_left; |
| 74 | g_autoptr(GString) stats = g_string_new(""); |
| 75 | g_string_append_printf(stats, "insn: %" PRIu64 "\n", expected); |
| 76 | g_string_append_printf(stats, "insn: %" PRIu64 " (per vcpu)\n", per_vcpu); |
| 77 | g_string_append_printf(stats, "insn: %" PRIu64 " (per vcpu inline)\n", inl_per_vcpu); |
| 78 | g_string_append_printf(stats, "insn: %" PRIu64 " (cond cb)\n", conditional); |
| 79 | qemu_plugin_outs(stats->str); |
| 80 | g_assert(expected > 0); |
| 81 | g_assert(per_vcpu == expected); |
| 82 | g_assert(inl_per_vcpu == expected); |
| 83 | g_assert(conditional == expected); |
| 84 | } |
| 85 | |
| 86 | static void stats_tb(void) |
| 87 | { |
| 88 | const uint64_t expected = global_count_tb; |
| 89 | const uint64_t per_vcpu = qemu_plugin_u64_sum(count_tb); |
| 90 | const uint64_t inl_per_vcpu = |
| 91 | qemu_plugin_u64_sum(count_tb_inline); |
| 92 | const uint64_t cond_num_trigger = qemu_plugin_u64_sum(tb_cond_num_trigger); |
| 93 | const uint64_t cond_track_left = qemu_plugin_u64_sum(tb_cond_track_count); |
| 94 | const uint64_t conditional = |
| 95 | cond_num_trigger * cond_trigger_limit + cond_track_left; |
| 96 | g_autoptr(GString) stats = g_string_new(""); |
| 97 | g_string_append_printf(stats, "tb: %" PRIu64 "\n", expected); |
| 98 | g_string_append_printf(stats, "tb: %" PRIu64 " (per vcpu)\n", per_vcpu); |
| 99 | g_string_append_printf(stats, "tb: %" PRIu64 " (per vcpu inline)\n", inl_per_vcpu); |
| 100 | g_string_append_printf(stats, "tb: %" PRIu64 " (conditional cb)\n", conditional); |
| 101 | qemu_plugin_outs(stats->str); |
| 102 | g_assert(expected > 0); |
| 103 | g_assert(per_vcpu == expected); |
| 104 | g_assert(inl_per_vcpu == expected); |
| 105 | g_assert(conditional == expected); |
| 106 | } |
| 107 | |
| 108 | static void stats_mem(void) |
| 109 | { |
| 110 | const uint64_t expected = global_count_mem; |
| 111 | const uint64_t per_vcpu = qemu_plugin_u64_sum(count_mem); |
| 112 | const uint64_t inl_per_vcpu = |
| 113 | qemu_plugin_u64_sum(count_mem_inline); |
| 114 | g_autoptr(GString) stats = g_string_new(""); |
| 115 | g_string_append_printf(stats, "mem: %" PRIu64 "\n", expected); |
| 116 | g_string_append_printf(stats, "mem: %" PRIu64 " (per vcpu)\n", per_vcpu); |
| 117 | g_string_append_printf(stats, "mem: %" PRIu64 " (per vcpu inline)\n", inl_per_vcpu); |
| 118 | qemu_plugin_outs(stats->str); |
| 119 | g_assert(expected > 0); |
| 120 | g_assert(per_vcpu == expected); |
| 121 | g_assert(inl_per_vcpu == expected); |
| 122 | } |
| 123 | |
| 124 | static void plugin_exit(void *udata) |
| 125 | { |
| 126 | const unsigned int num_cpus = qemu_plugin_num_vcpus(); |
| 127 | g_autoptr(GString) stats = g_string_new(""); |
| 128 | g_assert(num_cpus == max_cpu_index + 1); |
| 129 | |
| 130 | for (int i = 0; i < num_cpus ; ++i) { |
| 131 | const uint64_t tb = qemu_plugin_u64_get(count_tb, i); |
| 132 | const uint64_t tb_inline = qemu_plugin_u64_get(count_tb_inline, i); |
| 133 | const uint64_t insn = qemu_plugin_u64_get(count_insn, i); |
| 134 | const uint64_t insn_inline = qemu_plugin_u64_get(count_insn_inline, i); |
| 135 | const uint64_t mem = qemu_plugin_u64_get(count_mem, i); |
| 136 | const uint64_t mem_inline = qemu_plugin_u64_get(count_mem_inline, i); |
| 137 | const uint64_t tb_cond_trigger = |
| 138 | qemu_plugin_u64_get(tb_cond_num_trigger, i); |
| 139 | const uint64_t tb_cond_left = |
| 140 | qemu_plugin_u64_get(tb_cond_track_count, i); |
| 141 | const uint64_t insn_cond_trigger = |
| 142 | qemu_plugin_u64_get(insn_cond_num_trigger, i); |
| 143 | const uint64_t insn_cond_left = |
| 144 | qemu_plugin_u64_get(insn_cond_track_count, i); |
| 145 | g_string_printf(stats, "cpu %d: tb (%" PRIu64 ", %" PRIu64 |
| 146 | ", %" PRIu64 " * %" PRIu64 " + %" PRIu64 |
| 147 | ") | " |
| 148 | "insn (%" PRIu64 ", %" PRIu64 |
| 149 | ", %" PRIu64 " * %" PRIu64 " + %" PRIu64 |
| 150 | ") | " |
| 151 | "mem (%" PRIu64 ", %" PRIu64 ")" |
| 152 | "\n", |
| 153 | i, |
| 154 | tb, tb_inline, |
| 155 | tb_cond_trigger, cond_trigger_limit, tb_cond_left, |
| 156 | insn, insn_inline, |
| 157 | insn_cond_trigger, cond_trigger_limit, insn_cond_left, |
| 158 | mem, mem_inline); |
| 159 | qemu_plugin_outs(stats->str); |
| 160 | g_assert(tb == tb_inline); |
| 161 | g_assert(insn == insn_inline); |
| 162 | g_assert(mem == mem_inline); |
| 163 | g_assert(tb_cond_trigger == tb / cond_trigger_limit); |
| 164 | g_assert(tb_cond_left == tb % cond_trigger_limit); |
| 165 | g_assert(insn_cond_trigger == insn / cond_trigger_limit); |
| 166 | g_assert(insn_cond_left == insn % cond_trigger_limit); |
| 167 | } |
| 168 | |
| 169 | stats_tb(); |
| 170 | stats_insn(); |
| 171 | stats_mem(); |
| 172 | |
| 173 | qemu_plugin_scoreboard_free(counts); |
| 174 | qemu_plugin_scoreboard_free(data); |
| 175 | } |
| 176 | |
| 177 | static void vcpu_tb_exec(unsigned int cpu_index, void *udata) |
| 178 | { |
| 179 | qemu_plugin_u64_add(count_tb, cpu_index, 1); |
| 180 | g_assert(qemu_plugin_u64_get(data_tb, cpu_index) == (uintptr_t) udata); |
| 181 | g_mutex_lock(&tb_lock); |
| 182 | max_cpu_index = MAX(max_cpu_index, cpu_index); |
| 183 | global_count_tb++; |
| 184 | g_mutex_unlock(&tb_lock); |
| 185 | } |
| 186 | |
| 187 | static void vcpu_tb_cond_exec(unsigned int cpu_index, void *udata) |
| 188 | { |
| 189 | g_assert(qemu_plugin_u64_get(tb_cond_track_count, cpu_index) == |
| 190 | cond_trigger_limit); |
| 191 | g_assert(qemu_plugin_u64_get(data_tb, cpu_index) == (uintptr_t) udata); |
| 192 | qemu_plugin_u64_set(tb_cond_track_count, cpu_index, 0); |
| 193 | qemu_plugin_u64_add(tb_cond_num_trigger, cpu_index, 1); |
| 194 | } |
| 195 | |
| 196 | static void vcpu_insn_cond_exec(unsigned int cpu_index, void *udata) |
| 197 | { |
| 198 | g_assert(qemu_plugin_u64_get(insn_cond_track_count, cpu_index) == |
| 199 | cond_trigger_limit); |
| 200 | g_assert(qemu_plugin_u64_get(data_insn, cpu_index) == (uintptr_t) udata); |
| 201 | qemu_plugin_u64_set(insn_cond_track_count, cpu_index, 0); |
| 202 | qemu_plugin_u64_add(insn_cond_num_trigger, cpu_index, 1); |
| 203 | } |
| 204 | |
| 205 | static void vcpu_insn_exec(unsigned int cpu_index, void *udata) |
| 206 | { |
| 207 | qemu_plugin_u64_add(count_insn, cpu_index, 1); |
| 208 | g_assert(qemu_plugin_u64_get(data_insn, cpu_index) == (uintptr_t) udata); |
| 209 | g_mutex_lock(&insn_lock); |
| 210 | global_count_insn++; |
| 211 | g_mutex_unlock(&insn_lock); |
| 212 | } |
| 213 | |
| 214 | static void vcpu_mem_access(unsigned int cpu_index, |
| 215 | qemu_plugin_meminfo_t info, |
| 216 | uint64_t vaddr, |
| 217 | void *udata) |
| 218 | { |
| 219 | qemu_plugin_u64_add(count_mem, cpu_index, 1); |
| 220 | g_assert(qemu_plugin_u64_get(data_mem, cpu_index) == (uintptr_t) udata); |
| 221 | g_mutex_lock(&mem_lock); |
| 222 | global_count_mem++; |
| 223 | g_mutex_unlock(&mem_lock); |
| 224 | } |
| 225 | |
| 226 | static void vcpu_tb_trans(struct qemu_plugin_tb *tb, void *userdata) |
| 227 | { |
| 228 | void *tb_store = tb; |
| 229 | qemu_plugin_register_vcpu_tb_exec_inline_per_vcpu( |
| 230 | tb, QEMU_PLUGIN_INLINE_STORE_U64, data_tb, (uintptr_t) tb_store); |
| 231 | qemu_plugin_register_vcpu_tb_exec_cb( |
| 232 | tb, vcpu_tb_exec, QEMU_PLUGIN_CB_NO_REGS, tb_store); |
| 233 | qemu_plugin_register_vcpu_tb_exec_inline_per_vcpu( |
| 234 | tb, QEMU_PLUGIN_INLINE_ADD_U64, count_tb_inline, 1); |
| 235 | |
| 236 | qemu_plugin_register_vcpu_tb_exec_inline_per_vcpu( |
| 237 | tb, QEMU_PLUGIN_INLINE_ADD_U64, tb_cond_track_count, 1); |
| 238 | qemu_plugin_register_vcpu_tb_exec_cond_cb( |
| 239 | tb, vcpu_tb_cond_exec, QEMU_PLUGIN_CB_NO_REGS, |
| 240 | QEMU_PLUGIN_COND_EQ, tb_cond_track_count, cond_trigger_limit, tb_store); |
| 241 | |
| 242 | for (int idx = 0; idx < qemu_plugin_tb_n_insns(tb); ++idx) { |
| 243 | struct qemu_plugin_insn *insn = qemu_plugin_tb_get_insn(tb, idx); |
| 244 | void *insn_store = insn; |
| 245 | void *mem_store = (char *)insn_store + 0xff; |
| 246 | |
| 247 | qemu_plugin_register_vcpu_insn_exec_inline_per_vcpu( |
| 248 | insn, QEMU_PLUGIN_INLINE_STORE_U64, data_insn, |
| 249 | (uintptr_t) insn_store); |
| 250 | qemu_plugin_register_vcpu_insn_exec_cb( |
| 251 | insn, vcpu_insn_exec, QEMU_PLUGIN_CB_NO_REGS, insn_store); |
| 252 | qemu_plugin_register_vcpu_insn_exec_inline_per_vcpu( |
| 253 | insn, QEMU_PLUGIN_INLINE_ADD_U64, count_insn_inline, 1); |
| 254 | |
| 255 | qemu_plugin_register_vcpu_insn_exec_inline_per_vcpu( |
| 256 | insn, QEMU_PLUGIN_INLINE_ADD_U64, insn_cond_track_count, 1); |
| 257 | qemu_plugin_register_vcpu_insn_exec_cond_cb( |
| 258 | insn, vcpu_insn_cond_exec, QEMU_PLUGIN_CB_NO_REGS, |
| 259 | QEMU_PLUGIN_COND_EQ, insn_cond_track_count, cond_trigger_limit, |
| 260 | insn_store); |
| 261 | |
| 262 | qemu_plugin_register_vcpu_mem_inline_per_vcpu( |
| 263 | insn, QEMU_PLUGIN_MEM_RW, |
| 264 | QEMU_PLUGIN_INLINE_STORE_U64, |
| 265 | data_mem, (uintptr_t) mem_store); |
| 266 | qemu_plugin_register_vcpu_mem_cb(insn, &vcpu_mem_access, |
| 267 | QEMU_PLUGIN_CB_NO_REGS, |
| 268 | QEMU_PLUGIN_MEM_RW, mem_store); |
| 269 | qemu_plugin_register_vcpu_mem_inline_per_vcpu( |
| 270 | insn, QEMU_PLUGIN_MEM_RW, |
| 271 | QEMU_PLUGIN_INLINE_ADD_U64, |
| 272 | count_mem_inline, 1); |
| 273 | } |
| 274 | } |
| 275 | |
| 276 | QEMU_PLUGIN_EXPORT |
| 277 | int qemu_plugin_install(qemu_plugin_id_t id, const qemu_info_t *info, |
| 278 | int argc, char **argv) |
| 279 | { |
| 280 | counts = qemu_plugin_scoreboard_new(sizeof(CPUCount)); |
| 281 | count_tb = qemu_plugin_scoreboard_u64_in_struct( |
| 282 | counts, CPUCount, count_tb); |
| 283 | count_insn = qemu_plugin_scoreboard_u64_in_struct( |
| 284 | counts, CPUCount, count_insn); |
| 285 | count_mem = qemu_plugin_scoreboard_u64_in_struct( |
| 286 | counts, CPUCount, count_mem); |
| 287 | count_tb_inline = qemu_plugin_scoreboard_u64_in_struct( |
| 288 | counts, CPUCount, count_tb_inline); |
| 289 | count_insn_inline = qemu_plugin_scoreboard_u64_in_struct( |
| 290 | counts, CPUCount, count_insn_inline); |
| 291 | count_mem_inline = qemu_plugin_scoreboard_u64_in_struct( |
| 292 | counts, CPUCount, count_mem_inline); |
| 293 | tb_cond_num_trigger = qemu_plugin_scoreboard_u64_in_struct( |
| 294 | counts, CPUCount, tb_cond_num_trigger); |
| 295 | tb_cond_track_count = qemu_plugin_scoreboard_u64_in_struct( |
| 296 | counts, CPUCount, tb_cond_track_count); |
| 297 | insn_cond_num_trigger = qemu_plugin_scoreboard_u64_in_struct( |
| 298 | counts, CPUCount, insn_cond_num_trigger); |
| 299 | insn_cond_track_count = qemu_plugin_scoreboard_u64_in_struct( |
| 300 | counts, CPUCount, insn_cond_track_count); |
| 301 | data = qemu_plugin_scoreboard_new(sizeof(CPUData)); |
| 302 | data_insn = qemu_plugin_scoreboard_u64_in_struct(data, CPUData, data_insn); |
| 303 | data_tb = qemu_plugin_scoreboard_u64_in_struct(data, CPUData, data_tb); |
| 304 | data_mem = qemu_plugin_scoreboard_u64_in_struct(data, CPUData, data_mem); |
| 305 | |
| 306 | qemu_plugin_register_vcpu_tb_trans_cb(id, vcpu_tb_trans, NULL); |
| 307 | qemu_plugin_register_atexit_cb(id, plugin_exit, NULL); |
| 308 | |
| 309 | return 0; |
| 310 | } |