From fef80b544b74e8f0a58a3c6cb4828caf3a03ff06 Mon Sep 17 00:00:00 2001
From: Simon Rozman <simon@rozman.si>
Date: Sat, 7 Nov 2020 20:58:15 +0100
Subject: [PATCH] Add ARM and ARM64 platforms

Signed-off-by: Simon Rozman <simon@rozman.si>
---
 MemoryModule.c | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/MemoryModule.c b/MemoryModule.c
index 9f95a70..0c5d209 100644
--- a/MemoryModule.c
+++ b/MemoryModule.c
@@ -54,10 +54,16 @@
 #define IMAGE_SIZEOF_BASE_RELOCATION (sizeof(IMAGE_BASE_RELOCATION))
 #endif
 
-#ifdef _WIN64
-#define HOST_MACHINE IMAGE_FILE_MACHINE_AMD64
-#else
+#if defined(_M_IX86)
 #define HOST_MACHINE IMAGE_FILE_MACHINE_I386
+#elif defined(_M_AMD64)
+#define HOST_MACHINE IMAGE_FILE_MACHINE_AMD64
+#elif defined(_M_ARM)
+#define HOST_MACHINE IMAGE_FILE_MACHINE_ARMNT
+#elif defined(_M_ARM64)
+#define HOST_MACHINE IMAGE_FILE_MACHINE_ARM64
+#else
+#error Unsupported architecture
 #endif
 
 #include "MemoryModule.h"