From 2c9718ce3ff77128e964aeeac220c941bdd1a300 Mon Sep 17 00:00:00 2001
From: momo5502 <mauriceheumann@gmail.com>
Date: Thu, 24 Oct 2024 20:34:18 +0200
Subject: [PATCH] Prepare test binary

---
 src/CMakeLists.txt             |   1 +
 src/test-sample/CMakeLists.txt |  11 ++++
 src/test-sample/test.cpp       | 107 +++++++++++++++++++++++++++++++++
 3 files changed, 119 insertions(+)
 create mode 100644 src/test-sample/CMakeLists.txt
 create mode 100644 src/test-sample/test.cpp

diff --git a/src/CMakeLists.txt b/src/CMakeLists.txt
index 240f460c..e16cc98e 100644
--- a/src/CMakeLists.txt
+++ b/src/CMakeLists.txt
@@ -6,3 +6,4 @@ add_subdirectory(analyzer)
 add_subdirectory(fuzzing-engine)
 add_subdirectory(fuzzer)
 add_subdirectory(bad-sample)
+add_subdirectory(test-sample)
diff --git a/src/test-sample/CMakeLists.txt b/src/test-sample/CMakeLists.txt
new file mode 100644
index 00000000..6a131eff
--- /dev/null
+++ b/src/test-sample/CMakeLists.txt
@@ -0,0 +1,11 @@
+file(GLOB_RECURSE SRC_FILES CONFIGURE_DEPENDS
+  *.cpp
+  *.hpp
+  *.rc
+)
+
+list(SORT SRC_FILES)
+
+add_executable(test-sample ${SRC_FILES})
+
+momo_assign_source_group(${SRC_FILES})
diff --git a/src/test-sample/test.cpp b/src/test-sample/test.cpp
new file mode 100644
index 00000000..f38350f9
--- /dev/null
+++ b/src/test-sample/test.cpp
@@ -0,0 +1,107 @@
+#include <cstdint>
+#include <cstring>
+#include <string>
+#include <fstream>
+#include <thread>
+#include <atomic>
+#include <vector>
+#include <Windows.h>
+
+// getenv is broken right now :(
+std::string read_env(const char* env)
+{
+	char buffer[0x1000] = {};
+	if (!GetEnvironmentVariableA(env, buffer, sizeof(buffer)))
+	{
+		return {};
+	}
+
+	return buffer;
+}
+
+bool test_threads()
+{
+	constexpr auto thread_count = 5ULL;
+
+	std::atomic<uint64_t> counter{0};
+
+	std::vector<std::thread> threads{};
+	threads.reserve(thread_count);
+
+	for (auto i = 0ULL; i < thread_count; ++i)
+	{
+		threads.emplace_back([&counter]
+		{
+			++counter;
+			std::this_thread::yield();
+			++counter;
+			std::this_thread::sleep_for(std::chrono::milliseconds(100));
+			++counter;
+		});
+	}
+
+	for (auto& t : threads)
+	{
+		t.join();
+	}
+
+	return counter == (thread_count * 3ULL);
+}
+
+bool test_env()
+{
+	const auto computername = read_env("COMPUTERNAME");
+
+	SetEnvironmentVariableA("BLUB", "LUL");
+
+	const auto blub = read_env("BLUB");
+
+	return !computername.empty() && blub == "LUL";
+}
+
+bool test_io()
+{
+	const auto* filename = "a.txt";
+
+	FILE* fp{};
+	(void)fopen_s(&fp, filename, "wb");
+
+	if (!fp)
+	{
+		puts("Bad file");
+		return false;
+	}
+
+	const std::string text = "Blub";
+
+	(void)fwrite(text.data(), 1, text.size(), fp);
+	(void)fclose(fp);
+
+	std::ifstream t(filename);
+	t.seekg(0, std::ios::end);
+	const size_t size = t.tellg();
+	std::string buffer(size, ' ');
+	t.seekg(0);
+	t.read(buffer.data(), static_cast<std::streamsize>(size));
+
+	return text == buffer;
+}
+
+#define RUN_TEST(func, name)         \
+{                                    \
+	printf("Running test '" name "': "); \
+	const auto res = func();         \
+	valid &= res;                    \
+	puts(res ? "Sucess" : "Fail");   \
+}
+
+int main(int /*argc*/, const char* /*argv*/[])
+{
+	bool valid = true;
+
+	RUN_TEST(test_io, "I/O")
+	RUN_TEST(test_threads, "Threads")
+	RUN_TEST(test_env, "Environment")
+
+	return valid ? 0 : 1;
+}