From b328b830264408b467a5c904a474e7112c5d678c Mon Sep 17 00:00:00 2001 From: Aaron Patterson Date: Fri, 25 Sep 2020 14:55:12 -0700 Subject: make ext/coverage ASAN friendly --- ext/coverage/coverage.c | 8 ++++++++ 1 file changed, 8 insertions(+) (limited to 'ext/coverage') diff --git a/ext/coverage/coverage.c b/ext/coverage/coverage.c index 5b29ea2aea..be61595163 100644 --- a/ext/coverage/coverage.c +++ b/ext/coverage/coverage.c @@ -11,6 +11,7 @@ #include "gc.h" #include "internal/hash.h" #include "internal/thread.h" +#include "internal/sanitizers.h" #include "ruby.h" #include "vm_core.h" @@ -150,6 +151,9 @@ method_coverage_i(void *vstart, void *vend, size_t stride, void *data) VALUE ncoverages = *(VALUE*)data, v; for (v = (VALUE)vstart; v != (VALUE)vend; v += stride) { + void *poisoned = asan_poisoned_object_p(v); + asan_unpoison_object(v, false); + if (RB_TYPE_P(v, T_IMEMO) && imemo_type(v) == imemo_ment) { const rb_method_entry_t *me = (rb_method_entry_t *) v; VALUE path, first_lineno, first_column, last_lineno, last_column; @@ -189,6 +193,10 @@ method_coverage_i(void *vstart, void *vend, size_t stride, void *data) rb_hash_aset(methods, key, rcount); } } + + if (poisoned) { + asan_poison_object(v); + } } return 0; } -- cgit v1.2.3