summaryrefslogtreecommitdiffstats
path: root/src/shared
diff options
context:
space:
mode:
Diffstat (limited to 'src/shared')
-rw-r--r--src/shared/CMakeLists.txt5
-rw-r--r--src/shared/sockhelper.c2
2 files changed, 7 insertions, 0 deletions
diff --git a/src/shared/CMakeLists.txt b/src/shared/CMakeLists.txt
index a1bd49a..ce129af 100644
--- a/src/shared/CMakeLists.txt
+++ b/src/shared/CMakeLists.txt
@@ -11,6 +11,11 @@ find_package(Libatomic REQUIRED)
# add compile option to get POLLRDHUP support for signals
add_definitions(-D_GNU_SOURCE)
+if(DNBD3_SERVER_AFL)
+ message(STATUS "Building dnbd3-shared with AFL support")
+ add_definitions(-DDNBD3_SERVER_AFL)
+endif(DNBD3_SERVER_AFL)
+
set(DNBD3_SHARED_SOURCE_FILES ${CMAKE_CURRENT_SOURCE_DIR}/crc32.c
${CMAKE_CURRENT_SOURCE_DIR}/fdsignal.c
${CMAKE_CURRENT_SOURCE_DIR}/log.c
diff --git a/src/shared/sockhelper.c b/src/shared/sockhelper.c
index 0aad1a3..08d73fc 100644
--- a/src/shared/sockhelper.c
+++ b/src/shared/sockhelper.c
@@ -13,6 +13,8 @@
#define MAXLISTEN 20
+#include <dnbd3/afl.h>
+
struct _poll_list {
int count;
struct pollfd entry[MAXLISTEN];