From 88d253a41bd85f01ae22118237e286b0907888df Mon Sep 17 00:00:00 2001
From: Pavel Samolysov <samolisov@gmail.com>
Date: Fri, 27 Sep 2019 11:20:17 +0300
Subject: [PATCH] Adopt TVM for Windows

The following changes updates tVM for Windows and allow building and
using tVM on this OS:
 - DLL Exports are defined in the include/tvm/Visibility.hpp, core
   classes from tVM are marked as 'TVM_API'

 - Exceptions are enabled (the '/EHsc' parameter has been added)

 - Two string manipulation methods: and are defined.

Signed-off-by: Pavel Samolysov <samolisov@gmail.com>
---
 CMakeLists.txt                     |  4 ++++
 tvm-run/main.cpp                   |  5 +++++
 tvm/CMakeLists.txt                 |  5 +++++
 tvm/include/tvm/BytecodeParser.hpp |  5 +++--
 tvm/include/tvm/Compiler.hpp       | 15 +++++++++++++++
 tvm/include/tvm/Module.hpp         |  3 ++-
 tvm/include/tvm/VirtualMachine.hpp |  7 ++++---
 tvm/include/tvm/Visibility.hpp     | 20 ++++++++++++++++++++
 tvm/include/tvm/instructions.hpp   |  1 +
 9 files changed, 59 insertions(+), 6 deletions(-)
 create mode 100644 tvm/include/tvm/Visibility.hpp

diff --git a/CMakeLists.txt b/CMakeLists.txt
index 7e2fd11..8a6e036 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -7,5 +7,9 @@ project(tVM
 
 set(CMAKE_CXX_STANDARD 14)
 
+if (WIN32)
+	set(CMAKE_CXX_FLAGS "/EHsc")
+endif()
+
 add_subdirectory(tvm)
 add_subdirectory(tvm-run)
diff --git a/tvm-run/main.cpp b/tvm-run/main.cpp
index e37e0b5..7619023 100644
--- a/tvm-run/main.cpp
+++ b/tvm-run/main.cpp
@@ -5,6 +5,11 @@
 #include "tvm/ExecutionContext.hpp"
 #include "tvm/BytecodeParser.hpp"
 
+#ifdef _MSC_VER
+#define strncasecmp _strnicmp
+#define strcasecmp _stricmp
+#endif
+
 /// tvm-run's usage string. Printed when run with -help.
 static const char* usage =
     "Usage: tvm-run [<option>...] [--] <module> [<arg>...]\n"
diff --git a/tvm/CMakeLists.txt b/tvm/CMakeLists.txt
index 24d0d64..6fee7a8 100644
--- a/tvm/CMakeLists.txt
+++ b/tvm/CMakeLists.txt
@@ -10,3 +10,8 @@ target_include_directories(tvm
 	PUBLIC
 		include/
 )
+
+target_compile_definitions(tvm
+	PRIVATE
+		TVM_DLL_EXPORTS
+)
diff --git a/tvm/include/tvm/BytecodeParser.hpp b/tvm/include/tvm/BytecodeParser.hpp
index 59f7eca..f6d69b3 100644
--- a/tvm/include/tvm/BytecodeParser.hpp
+++ b/tvm/include/tvm/BytecodeParser.hpp
@@ -6,10 +6,11 @@
 #include <vector>
 
 #include "instructions.hpp"
+#include "tvm/Visibility.hpp"
 
 namespace tvm {
 
-struct ByteCodeParserException : public std::runtime_error {
+struct TVM_API ByteCodeParserException : public std::runtime_error {
   using std::runtime_error::runtime_error;
 };
 
@@ -17,7 +18,7 @@ struct ByteCodeParserException : public std::runtime_error {
  * @brief class BytecodeParser parses tVM instructions in .bc files
  *
  */
-class BytecodeParser {
+class TVM_API BytecodeParser {
 public:
   explicit BytecodeParser(bool verbose = false) noexcept : _verbose(verbose) {}
 
diff --git a/tvm/include/tvm/Compiler.hpp b/tvm/include/tvm/Compiler.hpp
index bff2c70..a4ab77a 100644
--- a/tvm/include/tvm/Compiler.hpp
+++ b/tvm/include/tvm/Compiler.hpp
@@ -3,7 +3,11 @@
 
 #include "instructions.hpp"
 
+#if defined(_WIN32)
+#include <windows.h>
+#else
 #include <sys/mman.h>
+#endif /* _WIN32 */
 #include <vector>
 #include <memory>
 #include <cstring>
@@ -26,14 +30,25 @@ struct CodeCache {
    */
   CodeCache(size_t size)
       : size_(size) {
+
+#if defined(_WIN32)
+    start_ = VirtualAlloc(nullptr, size_, MEM_COMMIT, PAGE_EXECUTE_READWRITE);
+    if (start_ == nullptr)
+      throw CompilationException{"Failed to allocate code cache memory."};
+#else
     start_ = mmap(0, size_ , PROT_READ | PROT_WRITE | PROT_EXEC,
                   MAP_PRIVATE | MAP_ANON, -1, 0);
     if (start_ == MAP_FAILED)
       throw CompilationException{"Failed to allocate code cache memory."};
+#endif /* _WIN32 */
   }
 
   ~CodeCache() noexcept {
+#if defined(_WIN32)
+    VirtualFree(start_, 0, MEM_RELEASE);
+#else
     munmap(start_, size_);
+#endif /* _WIN32 */
   }
 
   /**
diff --git a/tvm/include/tvm/Module.hpp b/tvm/include/tvm/Module.hpp
index 355e2a2..83ae4ea 100644
--- a/tvm/include/tvm/Module.hpp
+++ b/tvm/include/tvm/Module.hpp
@@ -2,6 +2,7 @@
 #define TVM_MODULE_HPP_
 
 #include "tvm/instructions.hpp"
+#include "tvm/Visibility.hpp"
 
 #include <cstdint>
 #include <iostream>
@@ -11,7 +12,7 @@
 namespace tvm {
 
 /// An interpreter module.
-struct Module {
+struct TVM_API Module {
   std::vector<Instruction> instructions;
 };
 
diff --git a/tvm/include/tvm/VirtualMachine.hpp b/tvm/include/tvm/VirtualMachine.hpp
index 72e3e66..3c62dcf 100644
--- a/tvm/include/tvm/VirtualMachine.hpp
+++ b/tvm/include/tvm/VirtualMachine.hpp
@@ -5,6 +5,7 @@
 #include "tvm/OperandStack.hpp"
 #include "tvm/instructions.hpp"
 #include "tvm/Compiler.hpp"
+#include "tvm/Visibility.hpp"
 
 #include <cstring>
 #include <ostream>
@@ -17,7 +18,7 @@ namespace tvm {
 class ExecutionContext;
 class VirtualMachine;
 
-struct Config {
+struct TVM_API Config {
   bool debug = false;              //< Enable debug code
   bool verbose = false;            //< Enable verbose printing and tracing
   bool jit = false;
@@ -30,13 +31,13 @@ inline std::ostream &operator<<(std::ostream &out, const Config &cfg) {
   return out;
 }
 
-struct InterpreterException : public std::runtime_error {
+struct TVM_API InterpreterException : public std::runtime_error {
   using std::runtime_error::runtime_error;
 };
 
 typedef StackElement (* JittedCode)();
 
-class VirtualMachine {
+class TVM_API VirtualMachine {
  public:
   VirtualMachine(const Config &cfg);
 
diff --git a/tvm/include/tvm/Visibility.hpp b/tvm/include/tvm/Visibility.hpp
new file mode 100644
index 0000000..d705637
--- /dev/null
+++ b/tvm/include/tvm/Visibility.hpp
@@ -0,0 +1,20 @@
+// https://gcc.gnu.org/wiki/Visibility
+// Generic helper definitions for shared library support
+#if defined __CYGWIN__ || defined _WIN32
+#define TVM_DLL_IMPORT __declspec(dllimport)
+#define TVM_DLL_EXPORT __declspec(dllexport)
+#else
+#if __GNUC__ >= 4
+#define TVM_DLL_IMPORT __attribute__((visibility("default")))
+#define TVM_DLL_EXPORT __attribute__((visibility("default")))
+#else
+#define TVM_DLL_IMPORT
+#define TVM_DLL_EXPORT
+#endif
+#endif
+
+#ifdef TVM_DLL_EXPORTS
+#define TVM_API TVM_DLL_EXPORT
+#else
+#define TVM_API TVM_DLL_IMPORT
+#endif
diff --git a/tvm/include/tvm/instructions.hpp b/tvm/include/tvm/instructions.hpp
index 7ef8fba..d181414 100644
--- a/tvm/include/tvm/instructions.hpp
+++ b/tvm/include/tvm/instructions.hpp
@@ -3,6 +3,7 @@
 
 #include <cstdint>
 #include <ostream>
+#include <string>
 #include <unordered_map>
 
 namespace tvm {