mirrors/bitcoin
1
0
Fork 0
mirror of https://github.com/bitcoin/bitcoin.git synced 2024-11-20 02:25:40 +01:00
Code Issues Projects Releases Wiki Activity

test: only declare a main() when fuzzing with AFL

Browse Source 
libFuzzer will provide a main(). This also fixes a weak linking
issue when fuzzing with libFuzzer on macOS.
This commit is contained in:
fanquake 2020-01-27 14:51:33 +08:00
parent fe48ac8580
commit b35567fe0b
No known key found for this signature in database
GPG Key ID: 2EEB9F5CC09526C1
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/test/fuzz/fuzz.cpp
View File

@ -12,6 +12,7 @@
const std::function<void(const std::string&)> G_TEST_LOG_FUN{}; const std::function<void(const std::string&)> G_TEST_LOG_FUN{};
#if defined(__AFL_COMPILER)
static bool read_stdin(std::vector<uint8_t>& data) static bool read_stdin(std::vector<uint8_t>& data)
{ {
uint8_t buffer[1024]; uint8_t buffer[1024];
@ -23,6 +24,7 @@ static bool read_stdin(std::vector<uint8_t>& data)
} }
return length == 0; return length == 0;
} }
#endif
// Default initialization: Override using a non-weak initialize(). // Default initialization: Override using a non-weak initialize().
__attribute__((weak)) void initialize() __attribute__((weak)) void initialize()
@ -44,9 +46,9 @@ extern "C" int LLVMFuzzerInitialize(int* argc, char*** argv)
return 0; return 0;
} }
// Declare main(...) "weak" to allow for libFuzzer linking. libFuzzer provides // Generally, the fuzzer will provide main(), except for AFL
// the main(...) function. #if defined(__AFL_COMPILER)
__attribute__((weak)) int main(int argc, char** argv) int main(int argc, char** argv)
{ {
initialize(); initialize();
#ifdef __AFL_INIT #ifdef __AFL_INIT
@ -74,3 +76,4 @@ __attribute__((weak)) int main(int argc, char** argv)
#endif #endif
return 0; return 0;
} }
#endif