Merge pull request #8726 from redsun82/swift-prebuilt-fetching

AlexDenisov · web-flow · commit df2cc181a0de · 2022-04-13T16:58:36.000+02:00
Swift: fetch prebuilt swift and link against it
diff --git a/misc/bazel/workspace.bzl b/misc/bazel/workspace.bzl
@@ -1,7 +1,31 @@
 load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive")
 load("@bazel_tools//tools/build_defs/repo:utils.bzl", "maybe")
 
-def codeql_workspace():
+_swift_prebuilt_version = "swift-5.6-RELEASE.42271.54"
+_swift_sha_map = {
+    "linux": "92b26fbbc45f812b0581385cc81470ab636fed9cff460052aa2173de765ddef4",
+    "macos-x86_64": "b24f609c7868c1709e1e552e3b394664014e6a7f1e097522db80deb359bbcd2f",
+}
+
+_swift_arch_map = {
+    "linux": "linux",
+    "macos-x86_64": "darwin_x86_64",
+}
+
+def codeql_workspace(repository_name = "codeql"):
+    for repo_arch, arch in _swift_arch_map.items():
+        sha256 = _swift_sha_map[repo_arch]
+
+        http_archive(
+            name = "swift_prebuilt_%s" % arch,
+            url = "https://github.com/dsp-testing/codeql-swift-artifacts/releases/download/%s/swift-prebuilt-%s.zip" % (
+                _swift_prebuilt_version,
+                repo_arch,
+            ),
+            build_file = "@%s//swift/extractor:BUILD.swift-prebuilt.bazel" % repository_name,
+            sha256 = sha256,
+        )
+
     maybe(
         repo_rule = http_archive,
         name = "rules_pkg",
diff --git a/swift/extractor/BUILD.bazel b/swift/extractor/BUILD.bazel
@@ -1,3 +1,11 @@
+alias(
+    name = "swift-llvm-support",
+    actual = select({
+        "@bazel_tools//src/conditions:%s" % arch: "@swift_prebuilt_%s//:swift-llvm-support" % arch
+        for arch in ("linux", "darwin_x86_64", "darwin_arm64")
+    }),
+)
+
 cc_binary(
     name = "extractor",
     srcs = ["main.cpp"],
@@ -7,4 +15,6 @@ cc_binary(
         "//conditions:default": ["@platforms//:incompatible"],
     }),
     visibility = ["//swift:__pkg__"],
+    deps = [":swift-llvm-support"],
+    features = ["-universal_binaries"],
 )
diff --git a/swift/extractor/BUILD.swift-prebuilt.bazel b/swift/extractor/BUILD.swift-prebuilt.bazel
@@ -0,0 +1,28 @@
+cc_library(
+    name = "swift-llvm-support",
+    srcs = [
+        "libCodeQLSwiftFrontendTool.a",
+    ] + select({
+        "@platforms//os:linux": [
+            "libCodeQLSwiftFrontendTool.so",
+        ],
+        "@platforms//os:macos": [
+            "libCodeQLSwiftFrontendTool.dylib",
+        ],
+    }),
+    hdrs = glob(["include/**/*"]),
+    linkopts = [
+        "-lm",
+        "-lz",
+    ] + select({
+        "@platforms//os:linux": [
+            "-luuid",
+            "-lrt",
+            "-lpthread",
+            "-ldl",
+        ],
+        "//conditions:default": [],
+    }),
+    strip_include_prefix = "include",
+    visibility = ["//visibility:public"],
+)
diff --git a/swift/extractor/main.cpp b/swift/extractor/main.cpp
@@ -2,7 +2,10 @@
 #include <iomanip>
 #include <stdlib.h>
 
-int main() {
+#include <swift/Basic/LLVMInitialize.h>
+
+int main(int argc, char** argv) {
+  PROGRAM_START(argc, argv);
   if (auto trapDir = getenv("CODEQL_EXTRACTOR_SWIFT_TRAP_DIR")) {
     std::string file = trapDir;
     file += "/my_first.trap";
