[SourceKit] Make the SourceKit plugin infrastructure buildable on Linux and Windows

Author: ahoppen

lib/Basic/LoadDynamicLibrary.cpp

@@ -17,6 +17,7 @@
 #define NOMINMAX
 #include "llvm/Support/ConvertUTF.h"
 #include "llvm/Support/Windows/WindowsSupport.h"
+#include "swift/Basic/LLVM.h"
 #include <windows.h>
 #else
 #include <dlfcn.h>

tools/SourceKit/tools/sourcekitd/include/sourcekitd/Internal.h

@@ -48,7 +48,7 @@ using SourceKit::SourceKitCancellationToken;
 static const unsigned ProtocolMajorVersion = 1;
 static const unsigned ProtocolMinorVersion = 0;
 
-enum class CustomBufferKind {
+enum class CustomBufferKind : size_t {
   TokenAnnotationsArray,
   DeclarationsArray,
   DocSupportAnnotationArray,
@@ -259,6 +259,8 @@ struct PluginInitParams {
 
 void loadPlugins(llvm::ArrayRef<std::string> registeredPlugins,
                  PluginInitParams &pluginParams);
+
+VariantFunctions *getPluginVariantFunctions(size_t BufKind);
 }
 
 #endif

tools/SourceKit/tools/sourcekitd/lib/API/sourcekitdAPI-Common.cpp

@@ -896,7 +896,7 @@ void YAMLRequestParser::initError(StringRef Desc, llvm::yaml::Node *Node,
 sourcekitd_variant_functions_t sourcekitd_variant_functions_create() {
   auto *vfuncs = new VariantFunctions();
   // Zero-initialize.
-  bzero(vfuncs, sizeof(VariantFunctions));
+  memset(vfuncs, 0, sizeof(VariantFunctions));
   return vfuncs;
 }
 
@@ -1133,3 +1133,30 @@ PluginInitParams::PluginInitParams(
   };
   this->opaqueIDEInspectionInstance = opaqueIDEInspectionInstance;
 }
+
+// MARK: Plugin variant functions
+
+// Note: only modified during plugin loading.
+static std::vector<VariantFunctions *> PluginVariantFunctions;
+
+VariantFunctions *sourcekitd::getPluginVariantFunctions(size_t BufKind) {
+  size_t index = BufKind - (size_t)CustomBufferKind::CustomBufferKind_End;
+  if (index >= PluginVariantFunctions.size() ||
+      PluginVariantFunctions[index] == nullptr) {
+    llvm::report_fatal_error(
+        "unknown custom buffer kind; possible plugin loading failure");
+  }
+  return PluginVariantFunctions[index];
+}
+
+void sourcekitd::pluginRegisterCustomBufferKind(
+    uint64_t kind, sourcekitd_variant_functions_t funcs) {
+  auto index = kind - (uint64_t)CustomBufferKind::CustomBufferKind_End;
+  if (index < PluginVariantFunctions.size()) {
+    assert(PluginVariantFunctions[index] == nullptr &&
+           "overwriting existing buffer");
+  } else {
+    PluginVariantFunctions.resize(index + 1);
+  }
+  PluginVariantFunctions[index] = static_cast<VariantFunctions *>(funcs);
+}