IMP: Add driver creation tool, zxdmake. Also adding a FireWall system.

- A FW is used to block connections from unknown ends.
- The ZXD is a format used to describe a ZKA PE32+ driver.
- Some refactors have been done as well.

Signed-off-by: Amlal EL Mahrouss <amlalelmahrouss@icloud.com>

5 files changed, 147 insertions, 113 deletions

diff --git a/tools/drvsign.cxx b/tools/drvsign.cxx
deleted file mode 100644
index ec5a6945..00000000
--- a/tools/drvsign.cxx
+++ /dev/null
@@ -1,107 +0,0 @@
-/*
- * Created on Thu Aug 22 09:29:13 CEST 2024
- *
- * Copyright (c) 2024 ZKA Technologies
- */
-
-#include <cstdint>
-#include <iostream>
-#include <fstream>
-#include <string>
-#include <cstring>
-#include <sstream>
-#include <filesystem>
-
-#define kDriverSignedExt   ".zxd"
-#define kDriverExt		   ".sys"
-#define kSignedDriverMagic " ZXD"
-
-namespace ZXD
-{
-	struct ZXD_HEADER final
-	{
-		char d_binary_padding[512];
-		// doesn't change.
-		char d_binary_magic[5];
-		int	 d_binary_version;
-		// can change.
-		char		  d_binary_name[4096];
-		std::uint64_t d_binary_checksum;
-		std::uint64_t d_binary_size;
-	};
-
-	/***********************************************************************************/
-	/* This handles the detection of a MZ header. */
-	/***********************************************************************************/
-
-	bool drvsign_check_for_mz(std::string mz_blob) noexcept
-	{
-		return mz_blob[0] == 'M' &&
-			   mz_blob[1] == 'Z';
-	}
-} // namespace ZXD
-
-/// @brief This program converts a PE32+ driver, into a custom format, the ZXD.
-/// @note ZXD is a format for ZKA signed drivers.
-int main(int argc, char* argv[])
-{
-	for (size_t i = 1ul; i < argc; ++i)
-	{
-		if (strcmp(argv[i], "/?") == 0)
-		{
-			std::cout << "drvsign: ZKA ZXD Driver Tool.\n";
-			std::cout << "drvsign: © ZKA Technologies, all rights reserved.\n";
-
-			return 0;
-		}
-	}
-
-	if (!std::filesystem::exists(argv[1]) ||
-		!std::string(argv[1]).ends_with(kDriverExt))
-		return -1;
-
-	ZXD::ZXD_HEADER sig{0};
-
-	sig.d_binary_version = 1;
-
-	memcpy(sig.d_binary_magic, kSignedDriverMagic, strlen(kSignedDriverMagic));
-	memcpy(sig.d_binary_name, argv[1], strlen(argv[1]));
-
-	sig.d_binary_size = std::filesystem::file_size(argv[1]);
-
-	memset(sig.d_binary_padding, 0x00, 512);
-
-	sig.d_binary_checksum = 0;
-
-	std::string signed_path = argv[1];
-	signed_path.erase(signed_path.find(kDriverExt), strlen(kDriverExt));
-	signed_path += kDriverSignedExt;
-
-	std::ofstream of_drv(signed_path, std::ios::binary);
-	std::ifstream if_drv(argv[1], std::ios::binary);
-
-	std::stringstream ss;
-	ss << if_drv.rdbuf();
-
-	if (!ZXD::drvsign_check_for_mz(ss.str()))
-	{
-		std::filesystem::remove(signed_path);
-		std::cout << "drvsign: Couldn't sign current driver, Input driver isn't a valid executable.\n";
-
-		return 1;
-	}
-
-	for (auto ch : ss.str())
-	{
-		sig.d_binary_checksum |= ch;
-	}
-
-	sig.d_binary_checksum ^= sig.d_binary_size;
-
-	of_drv.write((char*)&sig, sizeof(ZXD::ZXD_HEADER));
-	of_drv.write(ss.str().c_str(), ss.str().size());
-
-	std::cout << "drvsign: Signing is done, quiting, here is the key: " << sig.d_binary_checksum << ".\n";
-
-	return 0;
-}
diff --git a/tools/drvsign.json b/tools/drvsign.json
deleted file mode 100644
index 33bd4ea8..00000000
--- a/tools/drvsign.json
+++ /dev/null
@@ -1,6 +0,0 @@
-{
-  "compiler_path": "g++",
-  "compiler_std": "c++20",
-  "sources_path": ["drvsign.cxx"],
-  "output_name": "drvsign"
-}
diff --git a/tools/src/zxdmake.cxx b/tools/src/zxdmake.cxx
new file mode 100644
index 00000000..8a3276ff
--- /dev/null
+++ b/tools/src/zxdmake.cxx
@@ -0,0 +1,72 @@
+/*
+ * Created on Thu Aug 22 09:29:13 CEST 2024
+ *
+ * Copyright (c) 2024 ZKA Technologies
+ */
+
+#include <zxd.hxx>
+
+/// @brief This program converts a PE32+ driver, into a custom format, the ZXD.
+/// @note ZXD is a format for ZKA signed drivers.
+int main(int argc, char* argv[])
+{
+	for (size_t i = 1ul; i < argc; ++i)
+	{
+		if (strcmp(argv[i], "/?") == 0)
+		{
+			std::cout << "zxdmake: ZKA ZXD Driver Tool.\n";
+			std::cout << "zxdmake: © ZKA Technologies, all rights reserved.\n";
+
+			return 0;
+		}
+	}
+
+	if (!std::filesystem::exists(argv[1]) ||
+		!std::string(argv[1]).ends_with(kDriverExt))
+		return -1;
+
+	ZXD::ZXD_HEADER zxd_hdr{0};
+
+	zxd_hdr.d_binary_version = 1;
+
+	memcpy(zxd_hdr.d_binary_magic, kSignedDriverMagic, strlen(kSignedDriverMagic));
+	memcpy(zxd_hdr.d_binary_name, argv[1], strlen(argv[1]));
+
+	zxd_hdr.d_binary_size = std::filesystem::file_size(argv[1]);
+
+	memset(zxd_hdr.d_binary_padding, 0x00, 512);
+
+	zxd_hdr.d_binary_checksum = 0;
+
+	std::string signed_path = argv[1];
+	signed_path.erase(signed_path.find(kDriverExt), strlen(kDriverExt));
+	signed_path += kDriverSignedExt;
+
+	std::ofstream of_drv(signed_path, std::ios::binary);
+	std::ifstream if_drv(argv[1], std::ios::binary);
+
+	std::stringstream ss;
+	ss << if_drv.rdbuf();
+
+	if (!ZXD::zxd_check_for_mz(ss.str()))
+	{
+		std::filesystem::remove(signed_path);
+		std::cout << "zxdmake: Couldn't sign current driver, Input driver isn't a valid executable.\n";
+
+		return 1;
+	}
+
+	for (auto ch : ss.str())
+	{
+		zxd_hdr.d_binary_checksum |= ch;
+	}
+
+	zxd_hdr.d_binary_checksum ^= zxd_hdr.d_binary_size;
+
+	of_drv.write((char*)&zxd_hdr, sizeof(ZXD::ZXD_HEADER));
+	of_drv.write(ss.str().c_str(), ss.str().size());
+
+	std::cout << "zxdmake: Signing is done, quiting, Checksum: " << zxd_hdr.d_binary_checksum << ".\n";
+
+	return 0;
+}
diff --git a/tools/zxd.hxx b/tools/zxd.hxx
new file mode 100644
index 00000000..5875b017
--- /dev/null
+++ b/tools/zxd.hxx
@@ -0,0 +1,56 @@
+/*
+ * Created on Thu Aug 22 09:29:13 CEST 2024
+ *
+ * Copyright (c) 2024 ZKA Technologies
+ */
+
+#pragma once
+
+#include <cstdint>
+#include <iostream>
+#include <fstream>
+#include <string>
+#include <cstring>
+#include <sstream>
+#include <filesystem>
+
+#define kDriverSignedExt   ".zxd"
+#define kDriverExt		   ".sys"
+#define kSignedDriverMagic " ZXD"
+
+#define cZXDPaddingSize (512)
+#define cZXDMagicSize	(5)
+
+namespace ZXD
+{
+	/// <summary>
+	/// ZXD header.
+	/// </summary>
+	struct ZXD_HEADER final
+	{
+		char		  d_binary_padding[cZXDPaddingSize];
+		// doesn't change.
+		char	      d_binary_magic[cZXDMagicSize];
+		std::int32_t  d_binary_version;
+		// can change.
+		char		  d_binary_name[4096];
+		std::uint64_t d_binary_checksum;
+		std::uint64_t d_binary_size;
+	};
+
+	/***********************************************************************************/
+	/* @brief These two handles the detection of a MZ header. */
+	/***********************************************************************************/
+
+	inline bool zxd_check_for_mz(const char* mz_blob) noexcept
+	{
+		return mz_blob[0] == 'M' &&
+			   mz_blob[1] == 'Z';
+	}
+
+	inline bool zxd_check_for_mz(std::string mz_blob) noexcept
+	{
+		return mz_blob[0] == 'M' &&
+			   mz_blob[1] == 'Z';
+	}
+} // namespace ZXD
\ No newline at end of file
diff --git a/tools/zxdmake.json b/tools/zxdmake.json
new file mode 100644
index 00000000..d24a5de9
--- /dev/null
+++ b/tools/zxdmake.json
@@ -0,0 +1,19 @@
+{
+  "compiler_path": "x86_64-w64-mingw32-g++.exe",
+  "compiler_std": "c++20",
+  "headers_path": [ "../", "./", "../../" ],
+  "sources_path": [ "src/zxdmake.cxx" ],
+  "output_name": "zxdmake.exe",
+  "compiler_flags": [
+    "-ffreestanding",
+    "-fno-rtti",
+    "-fno-exceptions",
+    "-Wl,--subsystem=17"
+  ],
+  "cpp_macros": [
+    "__DRVSIGN_AMD64__",
+    "cDSVersion=0x0100",
+    "cDSVersionHighest=0x0100",
+    "cDSVersionLowest=0x0100"
+  ]
+}