From 9e2e94ecb8140b159e1ba4d148d2e6dc57b5fc92 Mon Sep 17 00:00:00 2001
From: Simon Rettberg
Date: Fri, 24 Oct 2025 08:52:25 +0200
Subject: Fix AFL build

---
 src/shared/CMakeLists.txt | 5 +++++
 src/shared/sockhelper.c   | 2 ++
 2 files changed, 7 insertions(+)

(limited to 'src/shared')

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];
-- 
cgit v1.2.3-55-g7522