stdlib/msvc: Runtime with MSVC library

This patch is for libswiftCore.lib, linking with the library set of Visual Studio 2015. Clang with the option -fms-extension is used to build.

Author: tinysun212

include/swift/Runtime/HeapObject.h

@@ -106,8 +106,8 @@ struct TwoWordPair {
   // in registers, so cram the result into an unsigned long long.
   // Use an enum class with implicit conversions so we don't dirty C callers
   // too much.
-#if __arm__ || __i386__ || defined(__CYGWIN__)
-#if defined(__CYGWIN__)
+#if __arm__ || __i386__ || defined(__CYGWIN__) || defined(_MSC_VER)
+#if defined(__CYGWIN__) || defined(_MSC_VER)
   enum class Return : unsigned __int128 {};
 #else
   enum class Return : unsigned long long {};

stdlib/public/runtime/Errors.cpp

@@ -18,17 +18,24 @@
 #include <stdlib.h>
 #include <string.h>
 #include <stdint.h>
+#if defined(_MSC_VER)
+#include <io.h>
+#else
 #include <unistd.h>
+#endif
+#if !defined(_MSC_VER)
+#include <pthread.h>
+#endif
 #include <stdarg.h>
 #include "swift/Runtime/Debug.h"
 #include "swift/Runtime/Mutex.h"
 #include "swift/Basic/Demangle.h"
 #include "swift/Basic/LLVM.h"
 #include "llvm/ADT/StringRef.h"
-
+#if !defined(_MSC_VER)
 #include <cxxabi.h>
-
-#if !defined(__CYGWIN__) && !defined(__ANDROID__)
+#endif
+#if !defined(__CYGWIN__) && !defined(__ANDROID__) && !defined(_MSC_VER)
 
 // execinfo.h is not available on Android. Checks in this file ensure that
 // fatalError behaves as expected, but without stack traces.
@@ -50,7 +57,7 @@ enum: uint32_t {
 
 using namespace swift;
 
-#if !defined(__CYGWIN__) && !defined(__ANDROID__)
+#if !defined(__CYGWIN__) && !defined(__ANDROID__) && !defined(_MSC_VER)
 
 static bool getSymbolNameAddr(llvm::StringRef libraryName, Dl_info dlinfo,
                               std::string &symbolName, uintptr_t &addrOut) {
@@ -187,11 +194,16 @@ reportOnCrash(uint32_t flags, const char *message)
 static void
 reportNow(uint32_t flags, const char *message)
 {
+#if defined(_MSC_VER)
+#define STDERR_FILENO 2
+  _write(STDERR_FILENO, message, strlen(message));
+#else
   write(STDERR_FILENO, message, strlen(message));
+#endif
 #ifdef __APPLE__
   asl_log(NULL, NULL, ASL_LEVEL_ERR, "%s", message);
 #endif
-#if !defined(__CYGWIN__) && !defined(__ANDROID__)
+#if !defined(__CYGWIN__) && !defined(__ANDROID__) && !defined(_MSC_VER)
   if (flags & FatalErrorFlags::ReportBacktrace) {
     fputs("Current stack trace:\n", stderr);
     constexpr unsigned maxSupportedStackDepth = 128;
@@ -222,7 +234,13 @@ swift::fatalError(uint32_t flags, const char *format, ...)
   va_start(args, format);
 
   char *log;
+#if defined(_MSC_VER)
+  int len = _vscprintf(format, args) + 1;
+  log = reinterpret_cast<char *>(malloc(len));
+  vsprintf(log, format, args);
+#else
   vasprintf(&log, format, args);
+#endif
 
   swift_reportError(flags, log);
   abort();

stdlib/public/runtime/Metadata.cpp

@@ -28,8 +28,14 @@
 #include <condition_variable>
 #include <new>
 #include <cctype>
+#if defined(_MSC_VER)
+// Avoid defining macro max(), min() which conflict with std::max(), std::min()
+#define NOMINMAX
+#include <windows.h>
+#else
 #include <sys/mman.h>
 #include <unistd.h>
+#endif
 #include "llvm/ADT/DenseMap.h"
 #include "llvm/ADT/Hashing.h"
 #include "ErrorObject.h"
@@ -56,19 +62,45 @@
 using namespace swift;
 using namespace metadataimpl;
 
+// allocate memory up to a nearby page boundary
+static void *swift_allocPage(size_t size) {
+#if defined(_MSC_VER)
+  auto mem = VirtualAlloc(
+      nullptr, size, MEM_TOP_DOWN | MEM_RESERVE | MEM_COMMIT, PAGE_READWRITE);
+#else
+  auto mem = mmap(nullptr, size, PROT_READ | PROT_WRITE, MAP_ANON | MAP_PRIVATE,
+                  VM_TAG_FOR_SWIFT_METADATA, 0);
+  if (mem == MAP_FAILED)
+    mem = nullptr;
+#endif
+  return mem;
+}
+
+// free memory allocated by swift_allocPage()
+// On success, swift_freePage() returns 0, on failure -1
+static int swift_freePage(void *addr, size_t size) {
+#if defined(_MSC_VER)
+  return VirtualFree(addr, 0, MEM_RELEASE) == 0 ? -1 : 0;
+#else
+  return munmap(addr, size);
+#endif
+}
+
 void *MetadataAllocator::alloc(size_t size) {
 #if defined(__APPLE__)
   const uintptr_t PageSizeMask = vm_page_mask;
+#elif defined(_MSC_VER)
+  SYSTEM_INFO SystemInfo;
+  GetSystemInfo(&SystemInfo);
+  const uintptr_t PageSizeMask = SystemInfo.dwPageSize - 1;
 #else
   static const uintptr_t PageSizeMask = sysconf(_SC_PAGESIZE) - 1;
 #endif
   // If the requested size is a page or larger, map page(s) for it
   // specifically.
   if (LLVM_UNLIKELY(size > PageSizeMask)) {
-    auto mem = mmap(nullptr, (size + PageSizeMask) & ~PageSizeMask,
-                    PROT_READ|PROT_WRITE, MAP_ANON|MAP_PRIVATE,
-                    VM_TAG_FOR_SWIFT_METADATA, 0);
-    if (mem == MAP_FAILED)
+    void *mem = swift_allocPage((size + PageSizeMask) & ~PageSizeMask);
+    if (!mem)
       crash("unable to allocate memory for metadata cache");
     return mem;
   }
@@ -83,13 +115,9 @@ void *MetadataAllocator::alloc(size_t size) {
     if (LLVM_UNLIKELY(((uintptr_t)next & ~PageSizeMask)
                         != (((uintptr_t)end & ~PageSizeMask)))) {
       // Allocate a new page if we haven't already.
-      allocation = mmap(nullptr, PageSizeMask + 1,
-                        PROT_READ|PROT_WRITE,
-                        MAP_ANON|MAP_PRIVATE,
-                        VM_TAG_FOR_SWIFT_METADATA,
-                        /*offset*/ 0);
+      allocation = swift_allocPage(PageSizeMask + 1);
 
-      if (allocation == MAP_FAILED)
+      if (!allocation)
         crash("unable to allocate memory for metadata cache");
 
       next = (char*) allocation;
@@ -107,7 +135,7 @@ void *MetadataAllocator::alloc(size_t size) {
     // This potentially causes us to perform multiple mmaps under contention,
     // but it keeps the fast path pristine.
     if (allocation) {
-      munmap(allocation, PageSizeMask + 1);
+      swift_freePage(allocation, PageSizeMask + 1);
     }
   }
 }

stdlib/public/runtime/MetadataLookup.cpp

@@ -34,6 +34,12 @@
 #include <link.h>
 #endif
 
+#if defined(_MSC_VER)
+#include <windows.h>
+#else
+#include <dlfcn.h>
+#endif
+
 using namespace swift;
 using namespace Demangle;
 
@@ -47,7 +53,7 @@ using namespace Demangle;
 #define SWIFT_TYPE_METADATA_SECTION "__swift2_types"
 #elif defined(__ELF__)
 #define SWIFT_TYPE_METADATA_SECTION ".swift2_type_metadata_start"
-#elif defined(__CYGWIN__)
+#elif defined(__CYGWIN__) || defined(_MSC_VER)
 #define SWIFT_TYPE_METADATA_SECTION ".sw2tymd"
 #endif
 

stdlib/public/runtime/Private.h

@@ -172,7 +172,7 @@ namespace swift {
       dest = newValue;
   }
 
-#if defined(__CYGWIN__)
+#if defined(__CYGWIN__) || defined(_MSC_VER)
   struct dl_phdr_info {
     void *dlpi_addr;
     const char *dlpi_name;
@@ -183,6 +183,8 @@ namespace swift {
                              void *data);
   uint8_t *_swift_getSectionDataPE(void *handle, const char *sectionName,
                                    unsigned long *sectionSize);
+#endif
+#if defined(__CYGWIN__)
   void _swift_once_f(uintptr_t *predicate, void *context,
                      void (*function)(void *));
 #endif

stdlib/public/runtime/ProtocolConformance.cpp

@@ -29,7 +29,11 @@
 #include <link.h>
 #endif
 
+#if defined(_MSC_VER)
+#include <windows.h>
+#else
 #include <dlfcn.h>
+#endif
 
 using namespace swift;
 
@@ -144,7 +148,7 @@ const {
 #define SWIFT_PROTOCOL_CONFORMANCES_SECTION "__swift2_proto"
 #elif defined(__ELF__)
 #define SWIFT_PROTOCOL_CONFORMANCES_SECTION ".swift2_protocol_conformances_start"
-#elif defined(__CYGWIN__)
+#elif defined(__CYGWIN__) || defined(_MSC_VER)
 #define SWIFT_PROTOCOL_CONFORMANCES_SECTION ".sw2prtc"
 #endif
 
@@ -396,17 +400,25 @@ void swift::_swift_initializeCallbacksToInspectDylib(
   // rdar://problem/19045112
   dl_iterate_phdr(_addImageProtocolConformances, &inspectArgs);
 }
-#elif defined(__CYGWIN__)
+#elif defined(__CYGWIN__) || defined(_MSC_VER)
 static int _addImageProtocolConformances(struct dl_phdr_info *info,
                                           size_t size, void *data) {
   InspectArgs *inspectArgs = (InspectArgs *)data;
   // inspectArgs contains addImage*Block function and the section name
+#if defined(_MSC_VER)
+  HMODULE handle;
 
+  if (!info->dlpi_name || info->dlpi_name[0] == '\0')
+    handle = GetModuleHandle(nullptr);
+  else
+    handle = GetModuleHandle(info->dlpi_name);
+#else
   void *handle;
-  if (!info->dlpi_name || info->dlpi_name[0] == '\0') {
+  if (!info->dlpi_name || info->dlpi_name[0] == '\0')
     handle = dlopen(nullptr, RTLD_LAZY);
-  } else
+  else
     handle = dlopen(info->dlpi_name, RTLD_LAZY | RTLD_NOLOAD);
+#endif
 
   unsigned long conformancesSize;
   const uint8_t *conformances =
@@ -416,7 +428,11 @@ static int _addImageProtocolConformances(struct dl_phdr_info *info,
   if (conformances)
     inspectArgs->fnAddImageBlock(conformances, conformancesSize);
 
+#if defined(_MSC_VER)
+  FreeLibrary(handle);
+#else
   dlclose(handle);
+#endif
   return 0;
 }
 

stdlib/public/runtime/Reflection.mm

@@ -24,6 +24,9 @@
 #include <cstring>
 #include <new>
 #include <string>
+#if !defined(_MSC_VER)
+#include <dlfcn.h>
+#endif
 
 #if SWIFT_OBJC_INTEROP
 #include "swift/Runtime/ObjCBridge.h"

stdlib/public/runtime/SwiftObject.mm

@@ -34,7 +34,9 @@
 #include "../SwiftShims/RuntimeShims.h"
 #include "Private.h"
 #include "swift/Runtime/Debug.h"
+#if !defined(_MSC_VER)
 #include <dlfcn.h>
+#endif
 #include <stdio.h>
 #include <stdlib.h>
 #include <unordered_map>

stdlib/public/stubs/LibcShims.cpp

@@ -12,11 +12,16 @@
 
 #include <random>
 #include <type_traits>
+#if defined(_MSC_VER)
+#include <io.h>
+#else
 #include <unistd.h>
+#endif
 #include <stdlib.h>
 #include <stdio.h>
 #include <string.h>
 #include "../SwiftShims/LibcShims.h"
+#include "llvm/Support/DataTypes.h"
 
 using namespace swift;
 
@@ -28,7 +33,11 @@ void swift::_swift_stdlib_free(void *ptr) {
 }
 
 int swift::_swift_stdlib_putchar_unlocked(int c) {
+#if defined(_MSC_VER)
+  return _putc_nolock(c, stdout);
+#else
   return putchar_unlocked(c);
+#endif
 }
 
 __swift_size_t swift::_swift_stdlib_fwrite_stdout(const void *ptr,
@@ -70,6 +79,11 @@ size_t swift::_swift_stdlib_malloc_size(const void *ptr) {
 size_t swift::_swift_stdlib_malloc_size(const void *ptr) {
   return malloc_usable_size(const_cast<void *>(ptr));
 }
+#elif defined(_MSC_VER)
+#include <malloc.h>
+size_t _swift_stdlib_malloc_size(const void *ptr) {
+  return _msize(const_cast<void *>(ptr));
+}
 #elif defined(__FreeBSD__)
 #include <malloc_np.h>
 size_t swift::_swift_stdlib_malloc_size(const void *ptr) {