1 files changed, 93 insertions, 0 deletions
diff --git a/engine/core/abort.c b/engine/core/abort.c
new file mode 100644
index 0000000..5ee42f6
--- /dev/null
+++ b/engine/core/abort.c
@@ -0,0 +1,93 @@
+/*
+ * Rune Game Engine
+ * Copyright 2024 Danny Holman <dholman@gymli.org>
+ *
+ * This software is provided 'as-is', without any express or implied
+ * warranty.  In no event will the authors be held liable for any damages
+ * arising from the use of this software.
+ *
+ * Permission is granted to anyone to use this software for any purpose,
+ * including commercial applications, and to alter it and redistribute it
+ * freely, subject to the following restrictions:
+ *
+ * 1. The origin of this software must not be misrepresented; you must not
+ *    claim that you wrote the original software. If you use this software
+ *    in a product, an acknowledgment in the product documentation would be
+ *    appreciated but is not required.
+ * 2. Altered source versions must be plainly marked as such, and must not be
+ *    misrepresented as being the original software.
+ * 3. This notice may not be removed or altered from any source distribution.
+ */
+
+#include <rune/core/abort.h>
+#include <rune/core/alloc.h>
+#include <rune/core/init.h>
+#include <rune/core/logging.h>
+#include <rune/util/exits.h>
+#include <stdlib.h>
+
+#define MAX_TRACE_ITEMS 30
+
+#ifdef _WIN32
+
+#include <windows.h>
+#include <dbghelp.h>
+
+void _stack_trace(void) {
+        void* buffer[MAX_TRACE_ITEMS];
+        HANDLE process = GetCurrentProcess();
+
+        SymInitialize(process, NULL, TRUE);
+        int num_links = CaptureStackBackTrace(0, MAX_TRACE_ITEMS, buffer, NULL);
+        SYMBOL_INFO *symbol = (SYMBOL_INFO*)rune_alloc(sizeof(SYMBOL_INFO) + 256 * sizeof(char));
+        symbol->SizeOfStruct = sizeof(SYMBOL_INFO);
+        symbol->MaxNameLen = 255;
+
+        for (int i = 0; i < num_links; i++) {
+                SymFromAddr(process, (DWORD64)(buffer[i]), 0, symbol);
+                log_output(LOG_INFO, "#%d: %s", i, symbol->Name, symbol->Address);
+        }
+
+        rune_free(symbol);
+        SymCleanup(process);
+}
+
+#else
+
+#include <execinfo.h>
+
+void _stack_trace(void) {
+        void* buffer[MAX_TRACE_ITEMS];
+        int num_links = backtrace(buffer, MAX_TRACE_ITEMS);
+        char** symbuf = backtrace_symbols(buffer, num_links);
+
+        for (int i = 0; i < num_links; i++)
+                log_output(LOG_INFO, "#%d: %s", i, symbuf[i]);
+
+        free(symbuf);
+}
+
+#endif
+
+NORET void rune_abort(void) {
+        log_output(LOG_INFO, "Abort called, printing stack trace");
+        _stack_trace();
+        rune_exit();
+        exit(REXIT_FAIL);
+}
+
+#ifdef MSVC
+
+NORET void __security_error_handler(void) {
+        log_output(LOG_FATAL, "Stack smashing detected in engine code");
+        rune_abort();
+}
+
+#else
+
+NORET void __stack_chk_fail(void) {
+        log_output(LOG_FATAL, "Stack smashing detected in engine code");
+        rune_abort();
+}
+
+#endif