Fuzing: move yubihsm_fuzz.h

Yubico · Oct 10, 2023 · cdb46fb · cdb46fb
1 parent 51735d3
commit cdb46fb
Show file tree

Hide file tree

Showing 5 changed files with 27 additions and 3 deletions.
diff --git a/lib/fuzz/fuzz_send_plain_msg.cc b/lib/fuzz/fuzz_send_plain_msg.cc
@@ -8,7 +8,7 @@ uint8_t *backend_data;
 size_t backend_data_len;
 }
 
-#include "../src/fuzz/fuzzer.h"
+#include "yubihsm_fuzz.h"
 
 yh_connector *connector;
 

diff --git a/lib/fuzz/yubihsm_fuzz.cc b/lib/fuzz/yubihsm_fuzz.cc
@@ -32,7 +32,7 @@ extern "C" {
 #include "../aes_cmac/aes_cmac.h"
 }
 
-#include "../src/fuzz/fuzzer.h"
+#include "yubihsm_fuzz.h"
 
 static void process_msg(Msg *msg, Msg *response);
 static bool compute_mac(Scp_ctx *s, uint8_t *key, Msg *msg, size_t raw_msg_len,

diff --git a/lib/fuzz/yubihsm_fuzz.h b/lib/fuzz/yubihsm_fuzz.h
@@ -0,0 +1,21 @@
+#ifndef _FUZZER_H
+#define _FUZZER_H
+
+#ifdef __cplusplus
+extern "C" {
+#endif
+
+#include <stddef.h>
+#include <yubihsm.h>
+
+extern yh_session *fuzz_session;
+#define FUZZ_BACKEND_PASSWORD "fuzzfuzz"
+
+extern uint8_t *backend_data;
+extern size_t backend_data_len;
+
+#ifdef __cplusplus
+}
+#endif
+
+#endif
diff --git a/pkcs11/CMakeLists.txt b/pkcs11/CMakeLists.txt
@@ -100,6 +100,9 @@ install(FILES pkcs11y.h DESTINATION "${YUBIHSM_INSTALL_INC_DIR}/pkcs11")
 add_subdirectory (tests)
 
 if (FUZZING)
+    include_directories(
+        ../lib/fuzz
+    )
     add_executable(fuzz_get_attribute_value
         # harness sources
         fuzz/fuzz_get_attribute_value.cc

diff --git a/pkcs11/fuzz/fuzz_get_attribute_value.cc b/pkcs11/fuzz/fuzz_get_attribute_value.cc
@@ -4,7 +4,7 @@
 #include <openssl/ec.h>
 #include <openssl/x509.h>
 
-#include "../src/fuzz/fuzzer.h"
+#include "yubihsm_fuzz.h"
 
 extern "C" {
 #include "pkcs11.h"