10b57cec5SDimitry Andric //===- FuzzerExtraCounters.cpp - Extra coverage counters ------------------===// 20b57cec5SDimitry Andric // 30b57cec5SDimitry Andric // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 40b57cec5SDimitry Andric // See https://llvm.org/LICENSE.txt for license information. 50b57cec5SDimitry Andric // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 60b57cec5SDimitry Andric // 70b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 80b57cec5SDimitry Andric // Extra coverage counters defined by user code. 90b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 100b57cec5SDimitry Andric 115ffd83dbSDimitry Andric #include "FuzzerPlatform.h" 125ffd83dbSDimitry Andric #include <cstdint> 130b57cec5SDimitry Andric 140b57cec5SDimitry Andric #if LIBFUZZER_LINUX || LIBFUZZER_NETBSD || LIBFUZZER_FREEBSD || \ 15*e8d8bef9SDimitry Andric LIBFUZZER_FUCHSIA || LIBFUZZER_EMSCRIPTEN 160b57cec5SDimitry Andric __attribute__((weak)) extern uint8_t __start___libfuzzer_extra_counters; 170b57cec5SDimitry Andric __attribute__((weak)) extern uint8_t __stop___libfuzzer_extra_counters; 180b57cec5SDimitry Andric 190b57cec5SDimitry Andric namespace fuzzer { ExtraCountersBegin()200b57cec5SDimitry Andricuint8_t *ExtraCountersBegin() { return &__start___libfuzzer_extra_counters; } ExtraCountersEnd()210b57cec5SDimitry Andricuint8_t *ExtraCountersEnd() { return &__stop___libfuzzer_extra_counters; } 220b57cec5SDimitry Andric ATTRIBUTE_NO_SANITIZE_ALL ClearExtraCounters()230b57cec5SDimitry Andricvoid ClearExtraCounters() { // hand-written memset, don't asan-ify. 240b57cec5SDimitry Andric uintptr_t *Beg = reinterpret_cast<uintptr_t*>(ExtraCountersBegin()); 250b57cec5SDimitry Andric uintptr_t *End = reinterpret_cast<uintptr_t*>(ExtraCountersEnd()); 260b57cec5SDimitry Andric for (; Beg < End; Beg++) { 270b57cec5SDimitry Andric *Beg = 0; 280b57cec5SDimitry Andric __asm__ __volatile__("" : : : "memory"); 290b57cec5SDimitry Andric } 300b57cec5SDimitry Andric } 310b57cec5SDimitry Andric 320b57cec5SDimitry Andric } // namespace fuzzer 330b57cec5SDimitry Andric 340b57cec5SDimitry Andric #endif 35