From d662291b3d53cd09659b4885b83a7545b9a43aa2 Mon Sep 17 00:00:00 2001
From: Chris Conlon <chris@wolfssl.com>
Date: Fri, 7 Dec 2018 10:27:51 -0700
Subject: [PATCH] add define to use test/benchmark.h without path prefix

---
 wolfcrypt/benchmark/benchmark.c | 6 +++++-
 wolfcrypt/test/test.c           | 6 +++++-
 2 files changed, 10 insertions(+), 2 deletions(-)

diff --git a/wolfcrypt/benchmark/benchmark.c b/wolfcrypt/benchmark/benchmark.c
index 630ea9ef9..a20069ad4 100644
--- a/wolfcrypt/benchmark/benchmark.c
+++ b/wolfcrypt/benchmark/benchmark.c
@@ -548,7 +548,11 @@ static const char* bench_desc_words[][9] = {
     #pragma warning(disable: 4996)
 #endif
 
-#include "wolfcrypt/benchmark/benchmark.h"
+#ifdef USE_FLAT_BENCHMARK_H
+    #include "benchmark.h"
+#else
+    #include "wolfcrypt/benchmark/benchmark.h"
+#endif
 
 #ifdef WOLFSSL_CURRTIME_REMAP
     #define current_time WOLFSSL_CURRTIME_REMAP
diff --git a/wolfcrypt/test/test.c b/wolfcrypt/test/test.c
index f86ce1fe5..a54ff0c72 100644
--- a/wolfcrypt/test/test.c
+++ b/wolfcrypt/test/test.c
@@ -216,7 +216,11 @@
     #define printf wolfssl_pb_print
 #endif
 
-#include "wolfcrypt/test/test.h"
+#ifdef USE_FLAT_TEST_H
+    #include "test.h"
+#else
+    #include "wolfcrypt/test/test.h"
+#endif
 
 #if defined(WOLFSSL_CERT_GEN) && defined(WOLFSSL_MULTI_ATTRIB)
 static void initDefaultName(void);