From 0cbfbe129934976359460fdbe69fb97632d81d24 Mon Sep 17 00:00:00 2001
From: Toni Uhlig <matzeton@googlemail.com>
Date: Fri, 15 Sep 2023 11:21:31 +0200
Subject: Added C++ (`ksocket/ksocket.hpp`) Socket wrapper classes.

 * another flatbuffers example (WiP!)
 * Makefile improvements

Signed-off-by: Toni Uhlig <matzeton@googlemail.com>
---
 ksocket/ksocket.h | 13 +++++++++++--
 1 file changed, 11 insertions(+), 2 deletions(-)

(limited to 'ksocket/ksocket.h')

diff --git a/ksocket/ksocket.h b/ksocket/ksocket.h
index 0a6717e..e44035b 100644
--- a/ksocket/ksocket.h
+++ b/ksocket/ksocket.h
@@ -1,5 +1,12 @@
-#pragma once
-#include "wsk.h"
+#ifndef KSOCKET_H
+#define KSOCKET_H 1
+
+#ifdef BUILD_USERMODE
+#error                                                                         \
+    "This file should only be included if building for kernel mode! Include <ksocket/ksocket.hpp> wrapper instead."
+#endif
+
+#include <ksocket/wsk.h>
 #include <ntddk.h>
 
 #define STATUS_UNSUPPORTED_WINDOWS_VERSION                                     \
@@ -97,3 +104,5 @@ KsRecvFrom(_In_ PKSOCKET Socket, _In_ PVOID Buffer, _Inout_ PULONG Length,
 #ifdef __cplusplus
 }
 #endif
+
+#endif
-- 
cgit v1.2.3