| /* |
| * Boilerplate code used for testing extent hooks via interception and |
| * passthrough. |
| */ |
| |
| static void *extent_alloc_hook(extent_hooks_t *extent_hooks, void *new_addr, |
| size_t size, size_t alignment, bool *zero, bool *commit, |
| unsigned arena_ind); |
| static bool extent_dalloc_hook(extent_hooks_t *extent_hooks, void *addr, |
| size_t size, bool committed, unsigned arena_ind); |
| static bool extent_commit_hook(extent_hooks_t *extent_hooks, void *addr, |
| size_t size, size_t offset, size_t length, unsigned arena_ind); |
| static bool extent_decommit_hook(extent_hooks_t *extent_hooks, void *addr, |
| size_t size, size_t offset, size_t length, unsigned arena_ind); |
| static bool extent_purge_lazy_hook(extent_hooks_t *extent_hooks, void *addr, |
| size_t size, size_t offset, size_t length, unsigned arena_ind); |
| static bool extent_purge_forced_hook(extent_hooks_t *extent_hooks, |
| void *addr, size_t size, size_t offset, size_t length, unsigned arena_ind); |
| static bool extent_split_hook(extent_hooks_t *extent_hooks, void *addr, |
| size_t size, size_t size_a, size_t size_b, bool committed, |
| unsigned arena_ind); |
| static bool extent_merge_hook(extent_hooks_t *extent_hooks, void *addr_a, |
| size_t size_a, void *addr_b, size_t size_b, bool committed, |
| unsigned arena_ind); |
| |
| static extent_hooks_t *default_hooks; |
| static extent_hooks_t hooks = { |
| extent_alloc_hook, |
| extent_dalloc_hook, |
| extent_commit_hook, |
| extent_decommit_hook, |
| extent_purge_lazy_hook, |
| extent_purge_forced_hook, |
| extent_split_hook, |
| extent_merge_hook |
| }; |
| |
| /* Control whether hook functions pass calls through to default hooks. */ |
| static bool try_alloc = true; |
| static bool try_dalloc = true; |
| static bool try_commit = true; |
| static bool try_decommit = true; |
| static bool try_purge_lazy = true; |
| static bool try_purge_forced = true; |
| static bool try_split = true; |
| static bool try_merge = true; |
| |
| /* Set to false prior to operations, then introspect after operations. */ |
| static bool called_alloc; |
| static bool called_dalloc; |
| static bool called_commit; |
| static bool called_decommit; |
| static bool called_purge_lazy; |
| static bool called_purge_forced; |
| static bool called_split; |
| static bool called_merge; |
| |
| /* Set to false prior to operations, then introspect after operations. */ |
| static bool did_alloc; |
| static bool did_dalloc; |
| static bool did_commit; |
| static bool did_decommit; |
| static bool did_purge_lazy; |
| static bool did_purge_forced; |
| static bool did_split; |
| static bool did_merge; |
| |
| #if 0 |
| # define TRACE_HOOK(fmt, ...) malloc_printf(fmt, __VA_ARGS__) |
| #else |
| # define TRACE_HOOK(fmt, ...) |
| #endif |
| |
| static void * |
| extent_alloc_hook(extent_hooks_t *extent_hooks, void *new_addr, size_t size, |
| size_t alignment, bool *zero, bool *commit, unsigned arena_ind) |
| { |
| void *ret; |
| |
| TRACE_HOOK("%s(extent_hooks=%p, new_addr=%p, size=%zu, alignment=%zu, " |
| "*zero=%s, *commit=%s, arena_ind=%u)\n", __func__, extent_hooks, |
| new_addr, size, alignment, *zero ? "true" : "false", *commit ? |
| "true" : "false", arena_ind); |
| assert_ptr_eq(extent_hooks, &hooks, |
| "extent_hooks should be same as pointer used to set hooks"); |
| assert_ptr_eq(extent_hooks->alloc, extent_alloc_hook, |
| "Wrong hook function"); |
| called_alloc = true; |
| if (!try_alloc) |
| return (NULL); |
| ret = default_hooks->alloc(default_hooks, new_addr, size, alignment, |
| zero, commit, 0); |
| did_alloc = (ret != NULL); |
| return (ret); |
| } |
| |
| static bool |
| extent_dalloc_hook(extent_hooks_t *extent_hooks, void *addr, size_t size, |
| bool committed, unsigned arena_ind) |
| { |
| bool err; |
| |
| TRACE_HOOK("%s(extent_hooks=%p, addr=%p, size=%zu, committed=%s, " |
| "arena_ind=%u)\n", __func__, extent_hooks, addr, size, committed ? |
| "true" : "false", arena_ind); |
| assert_ptr_eq(extent_hooks, &hooks, |
| "extent_hooks should be same as pointer used to set hooks"); |
| assert_ptr_eq(extent_hooks->dalloc, extent_dalloc_hook, |
| "Wrong hook function"); |
| called_dalloc = true; |
| if (!try_dalloc) |
| return (true); |
| err = default_hooks->dalloc(default_hooks, addr, size, committed, 0); |
| did_dalloc = !err; |
| return (err); |
| } |
| |
| static bool |
| extent_commit_hook(extent_hooks_t *extent_hooks, void *addr, size_t size, |
| size_t offset, size_t length, unsigned arena_ind) |
| { |
| bool err; |
| |
| TRACE_HOOK("%s(extent_hooks=%p, addr=%p, size=%zu, offset=%zu, " |
| "length=%zu, arena_ind=%u)\n", __func__, extent_hooks, addr, size, |
| offset, length, arena_ind); |
| assert_ptr_eq(extent_hooks, &hooks, |
| "extent_hooks should be same as pointer used to set hooks"); |
| assert_ptr_eq(extent_hooks->commit, extent_commit_hook, |
| "Wrong hook function"); |
| called_commit = true; |
| if (!try_commit) |
| return (true); |
| err = default_hooks->commit(default_hooks, addr, size, offset, length, |
| 0); |
| did_commit = !err; |
| return (err); |
| } |
| |
| static bool |
| extent_decommit_hook(extent_hooks_t *extent_hooks, void *addr, size_t size, |
| size_t offset, size_t length, unsigned arena_ind) |
| { |
| bool err; |
| |
| TRACE_HOOK("%s(extent_hooks=%p, addr=%p, size=%zu, offset=%zu, " |
| "length=%zu, arena_ind=%u)\n", __func__, extent_hooks, addr, size, |
| offset, length, arena_ind); |
| assert_ptr_eq(extent_hooks, &hooks, |
| "extent_hooks should be same as pointer used to set hooks"); |
| assert_ptr_eq(extent_hooks->decommit, extent_decommit_hook, |
| "Wrong hook function"); |
| called_decommit = true; |
| if (!try_decommit) |
| return (true); |
| err = default_hooks->decommit(default_hooks, addr, size, offset, length, |
| 0); |
| did_decommit = !err; |
| return (err); |
| } |
| |
| static bool |
| extent_purge_lazy_hook(extent_hooks_t *extent_hooks, void *addr, size_t size, |
| size_t offset, size_t length, unsigned arena_ind) |
| { |
| bool err; |
| |
| TRACE_HOOK("%s(extent_hooks=%p, addr=%p, size=%zu, offset=%zu, " |
| "length=%zu arena_ind=%u)\n", __func__, extent_hooks, addr, size, |
| offset, length, arena_ind); |
| assert_ptr_eq(extent_hooks, &hooks, |
| "extent_hooks should be same as pointer used to set hooks"); |
| assert_ptr_eq(extent_hooks->purge_lazy, extent_purge_lazy_hook, |
| "Wrong hook function"); |
| called_purge_lazy = true; |
| if (!try_purge_lazy) |
| return (true); |
| err = default_hooks->purge_lazy == NULL || |
| default_hooks->purge_lazy(default_hooks, addr, size, offset, length, |
| 0); |
| did_purge_lazy = !err; |
| return (err); |
| } |
| |
| static bool |
| extent_purge_forced_hook(extent_hooks_t *extent_hooks, void *addr, size_t size, |
| size_t offset, size_t length, unsigned arena_ind) |
| { |
| bool err; |
| |
| TRACE_HOOK("%s(extent_hooks=%p, addr=%p, size=%zu, offset=%zu, " |
| "length=%zu arena_ind=%u)\n", __func__, extent_hooks, addr, size, |
| offset, length, arena_ind); |
| assert_ptr_eq(extent_hooks, &hooks, |
| "extent_hooks should be same as pointer used to set hooks"); |
| assert_ptr_eq(extent_hooks->purge_forced, extent_purge_forced_hook, |
| "Wrong hook function"); |
| called_purge_forced = true; |
| if (!try_purge_forced) |
| return (true); |
| err = default_hooks->purge_forced == NULL || |
| default_hooks->purge_forced(default_hooks, addr, size, offset, |
| length, 0); |
| did_purge_forced = !err; |
| return (err); |
| } |
| |
| static bool |
| extent_split_hook(extent_hooks_t *extent_hooks, void *addr, size_t size, |
| size_t size_a, size_t size_b, bool committed, unsigned arena_ind) |
| { |
| bool err; |
| |
| TRACE_HOOK("%s(extent_hooks=%p, addr=%p, size=%zu, size_a=%zu, " |
| "size_b=%zu, committed=%s, arena_ind=%u)\n", __func__, extent_hooks, |
| addr, size, size_a, size_b, committed ? "true" : "false", |
| arena_ind); |
| assert_ptr_eq(extent_hooks, &hooks, |
| "extent_hooks should be same as pointer used to set hooks"); |
| assert_ptr_eq(extent_hooks->split, extent_split_hook, |
| "Wrong hook function"); |
| called_split = true; |
| if (!try_split) |
| return (true); |
| err = (default_hooks->split == NULL || |
| default_hooks->split(default_hooks, addr, size, size_a, size_b, |
| committed, 0)); |
| did_split = !err; |
| return (err); |
| } |
| |
| static bool |
| extent_merge_hook(extent_hooks_t *extent_hooks, void *addr_a, size_t size_a, |
| void *addr_b, size_t size_b, bool committed, unsigned arena_ind) |
| { |
| bool err; |
| |
| TRACE_HOOK("%s(extent_hooks=%p, addr_a=%p, size_a=%zu, addr_b=%p " |
| "size_b=%zu, committed=%s, arena_ind=%u)\n", __func__, extent_hooks, |
| addr_a, size_a, addr_b, size_b, committed ? "true" : "false", |
| arena_ind); |
| assert_ptr_eq(extent_hooks, &hooks, |
| "extent_hooks should be same as pointer used to set hooks"); |
| assert_ptr_eq(extent_hooks->merge, extent_merge_hook, |
| "Wrong hook function"); |
| called_merge = true; |
| if (!try_merge) |
| return (true); |
| err = (default_hooks->merge == NULL || |
| default_hooks->merge(default_hooks, addr_a, size_a, addr_b, size_b, |
| committed, 0)); |
| did_merge = !err; |
| return (err); |
| } |
| |
| static void |
| extent_hooks_prep(void) |
| { |
| size_t sz; |
| |
| sz = sizeof(default_hooks); |
| assert_d_eq(mallctl("arena.0.extent_hooks", (void *)&default_hooks, &sz, |
| NULL, 0), 0, "Unexpected mallctl() error"); |
| } |