Windows Port

This allows IndexStoreDB to build on Windows.

Author: gmittert

lib/Database/Database.cpp

@@ -23,6 +23,15 @@
 #include "llvm/Support/Mutex.h"
 #include "llvm/Support/Path.h"
 #include "llvm/Support/raw_ostream.h"
+#if defined(_WIN32)
+#include "Windows.h"
+#endif
+
+#if defined(_WIN32)
+typedef HANDLE indexstorePid_t;
+#else
+typedef pid_t indexstorePid_t;
+#endif
 
 // Dispatch on Linux doesn't have QOS_* macros.
 #if !__has_include(<sys/qos.h>)
@@ -97,7 +106,11 @@ Database::Implementation::create(StringRef path, bool readonly, Optional<size_t>
   SmallString<128> savedPathBuf = versionPath;
   llvm::sys::path::append(savedPathBuf, "saved");
   SmallString<128> processPathBuf = versionPath;
+#if defined(WIN32)
+  llvm::raw_svector_ostream(processPathBuf) << "/p" << GetCurrentProcess();
+#else
   llvm::raw_svector_ostream(processPathBuf) << "/p" << getpid();
+#endif
 
   bool existingDB = true;
 
@@ -269,10 +282,16 @@ void Database::Implementation::increaseMapSize() {
   LOG_INFO_FUNC(High, "increased lmdb map size to: " << MapSize);
 }
 
-static bool isProcessStillExecuting(pid_t PID) {
+static bool isProcessStillExecuting(indexstorePid_t PID) {
+#if defined(_WIN32)
+  DWORD dwExitCode;
+  bool result = GetExitCodeProcess(PID, &dwExitCode);
+  return result && (dwExitCode == STILL_ACTIVE);
+#else
   if (getsid(PID) == -1 && errno == ESRCH)
     return false;
   return true;
+#endif
 }
 
 // This runs in a background priority queue.
@@ -283,7 +302,11 @@ static void cleanupDiscardedDBsImpl(StringRef versionedPath) {
   // A directory is dead if it has been marked with the suffix "-dead" or if it
   // has the name "p<PID>" where process PID is no longer running.
 
-  pid_t currPID = getpid();
+#if defined(WIN32)
+  indexstorePid_t currPID = GetCurrentProcess();
+#else
+  indexstorePid_t currPID = getpid();
+#endif
 
   std::error_code EC;
   directory_iterator Begin(versionedPath, EC);
@@ -292,7 +315,7 @@ static void cleanupDiscardedDBsImpl(StringRef versionedPath) {
     auto &Item = *Begin;
     StringRef currPath = Item.path();
 
-    auto shouldRemove = [](StringRef fullpath, pid_t currPID) -> bool {
+    auto shouldRemove = [](StringRef fullpath, indexstorePid_t currPID) -> bool {
       StringRef path = llvm::sys::path::filename(fullpath);
       if (path.endswith(DeadProcessDBSuffix))
         return true;
@@ -301,9 +324,9 @@ static void cleanupDiscardedDBsImpl(StringRef versionedPath) {
       size_t pathPID;
       if (path.substr(1).getAsInteger(10, pathPID))
         return false;
-      if (pathPID == currPID)
+      if ((indexstorePid_t)pathPID == currPID)
         return false;
-      return !isProcessStillExecuting(pathPID);
+      return !isProcessStillExecuting((indexstorePid_t)pathPID);
     };
 
     if (shouldRemove(currPath, currPID)) {

lib/Index/IndexStoreLibraryProvider.cpp

@@ -13,7 +13,12 @@
 #include "IndexStoreDB/Index/IndexStoreLibraryProvider.h"
 #include "indexstore/IndexStoreCXX.h"
 #include "llvm/ADT/StringRef.h"
+#include "llvm/Support/ConvertUTF.h"
+#if defined(_WIN32)
+#include <Windows.h>
+#else
 #include <dlfcn.h>
+#endif
 
 using namespace IndexStoreDB;
 using namespace index;
@@ -28,11 +33,28 @@ IndexStoreLibraryRef GlobalIndexStoreLibraryProvider::getLibraryForStorePath(Str
 
   // Note: we're using dlsym with RTLD_DEFAULT because we cannot #incldue indexstore.h and indexstore_functions.h
   std::string ignored;
-  return loadIndexStoreLibraryFromDLHandle(RTLD_DEFAULT, ignored);
+#if defined(_WIN32)
+  void* defaultHandle = GetModuleHandleW(NULL);
+#else
+  void* defaultHandle = RTLD_DEFAULT;
+#endif
+  return loadIndexStoreLibraryFromDLHandle(defaultHandle, ignored);
 }
 
 IndexStoreLibraryRef index::loadIndexStoreLibrary(std::string dylibPath,
                                                   std::string &error) {
+#if defined(_WIN32)
+  llvm::SmallVector<llvm::UTF16, 30> u16Path;
+  if (!convertUTF8ToUTF16String(dylibPath, u16Path)) {
+    error += "Failed to convert path: " + dylibPath + " to UTF-16";
+    return nullptr;
+  }
+  HMODULE dlHandle = LoadLibraryW((LPCWSTR)u16Path.data());
+  if (dlHandle == NULL) {
+    error += "Failed to load " + dylibPath + ". Error: " + std::to_string(GetLastError());
+    return nullptr;
+  }
+#else
   auto flags = RTLD_LAZY | RTLD_LOCAL;
 #ifdef RTLD_FIRST
   flags |= RTLD_FIRST;
@@ -44,6 +66,7 @@ IndexStoreLibraryRef index::loadIndexStoreLibrary(std::string dylibPath,
     error += dlerror();
     return nullptr;
   }
+#endif
 
   // Intentionally leak the dlhandle; we have no reason to dlclose it and it may be unsafe.
   (void)dlHandle;
@@ -54,12 +77,21 @@ IndexStoreLibraryRef index::loadIndexStoreLibrary(std::string dylibPath,
 static IndexStoreLibraryRef loadIndexStoreLibraryFromDLHandle(void *dlHandle, std::string &error) {
   indexstore_functions_t api;
 
+#if defined(_WIN32)
+#define INDEXSTORE_FUNCTION(func, required) \
+  api.func = (decltype(indexstore_functions_t::func))GetProcAddress((HMODULE)dlHandle, "indexstore_" #func); \
+  if (!api.func && required) { \
+    error = "indexstore library missing required function indexstore_" #func; \
+    return nullptr; \
+  }
+#else
 #define INDEXSTORE_FUNCTION(func, required) \
   api.func = (decltype(indexstore_functions_t::func))dlsym(dlHandle, "indexstore_" #func); \
   if (!api.func && required) { \
     error = "indexstore library missing required function indexstore_" #func; \
     return nullptr; \
   }
+#endif
 
 #include "indexstore_functions.def"
 

lib/Support/Path.cpp

@@ -16,16 +16,36 @@
 #include "llvm/Support/Path.h"
 #include "llvm/Support/FileSystem.h"
 #include "llvm/Support/Mutex.h"
+#if defined(_WIN32)
+#include <Windows.h>
+#else
 #include <unistd.h>
+#endif
 #include <limits.h>
 #include <stdlib.h>
 
+#if defined(_WIN32)
+#define PATH_MAX MAX_PATH
+#endif
+
 using namespace IndexStoreDB;
 
 static StringRef getCanonicalPath(const char *Path, char *PathBuf) {
+#if defined(_WIN32)
+  llvm::SmallString<128> buffer;
+  if (llvm::sys::fs::real_path(Path, buffer, false)) {
+    return StringRef();
+  }
+  if (!PathBuf) {
+    PathBuf = (char *)malloc(sizeof(char) * buffer.size());
+  }
+  std::copy(buffer.begin(), buffer.end(), PathBuf);
+  return PathBuf;
+#else
   if (const char *path = realpath(Path, PathBuf))
     return path;
   return StringRef();
+#endif
 }
 
 namespace {