Add WASI as a supported target for sanitizers

We are adding support for sanitizers on WASI targets. This commit
adds driver support by unblocking hard-coded checks and forwarding
sanitizer flags to clang linker driver.

Author: kateinoigakukun

Sources/SwiftDriver/Driver/Driver.swift

@@ -2751,7 +2751,7 @@ extension Driver {
     }
 
     // Check that we're one of the known supported targets for sanitizers.
-    if !(targetTriple.isWindows || targetTriple.isDarwin || targetTriple.os == .linux) {
+    if !(targetTriple.isWindows || targetTriple.isDarwin || targetTriple.os == .linux || targetTriple.os == .wasi) {
       diagnosticEngine.emit(
         .error_unsupported_opt_for_target(
           arg: "-sanitize=",

Sources/SwiftDriver/Jobs/WebAssemblyToolchain+LinkerSupport.swift

@@ -165,8 +165,12 @@ extension WebAssemblyToolchain {
 
       // Delegate to Clang for sanitizers. It will figure out the correct linker
       // options.
-      guard sanitizers.isEmpty else {
-        throw Error.sanitizersUnsupportedForTarget(targetTriple.triple)
+      if linkerOutputType == .executable && !sanitizers.isEmpty {
+        let sanitizerNames = sanitizers
+          .map { $0.rawValue }
+          .sorted() // Sort so we get a stable, testable order
+          .joined(separator: ",")
+        commandLine.appendFlag("-fsanitize=\(sanitizerNames)")
       }
 
       if parsedOptions.hasArgument(.profileGenerate) {

Sources/SwiftDriver/Toolchains/WebAssemblyToolchain.swift

@@ -146,7 +146,12 @@ public final class WebAssemblyToolchain: Toolchain {
     targetTriple: Triple,
     isShared: Bool
   ) throws -> String {
-    throw Error.sanitizersUnsupportedForTarget(targetTriple.triple)
+    switch sanitizer {
+    case .address:
+      return "libclang_rt.\(sanitizer.libraryName)-\(targetTriple.archName).a"
+    default:
+      throw Error.sanitizersUnsupportedForTarget(targetTriple.triple)
+    }
   }
 
   public func platformSpecificInterpreterEnvironmentVariables(env: [String : String],

Tests/SwiftDriverTests/SwiftDriverTests.swift

@@ -2877,6 +2877,47 @@ final class SwiftDriverTests: XCTestCase {
       XCTAssertTrue(linkCmd.contains(.flag("-fsanitize=address")))
     }
   #endif
+
+    func checkWASITarget(target: String, clangOSDir: String) throws {
+      try withTemporaryDirectory { resourceDir in
+        var env = ProcessEnv.vars
+        env["SWIFT_DRIVER_SWIFT_AUTOLINK_EXTRACT_EXEC"] = "/garbage/swift-autolink-extract"
+
+        let asanRuntimeLibPath = resourceDir.appending(components: [
+          "clang", "lib", clangOSDir, "libclang_rt.asan-wasm32.a"
+        ])
+        try localFileSystem.writeFileContents(asanRuntimeLibPath) {
+          $0.send("garbage")
+        }
+        try localFileSystem.writeFileContents(resourceDir.appending(components: "wasi", "static-executable-args.lnk")) {
+          $0.send("garbage")
+        }
+
+        var driver = try Driver(
+          args: commonArgs + [
+            "-target", target, "-sanitize=address",
+            "-resource-dir", resourceDir.pathString
+          ],
+          env: env
+        )
+        let plannedJobs = try driver.planBuild()
+
+        XCTAssertEqual(plannedJobs.count, 4)
+
+        let compileJob = plannedJobs[0]
+        let compileCmd = compileJob.commandLine
+        XCTAssertTrue(compileCmd.contains(.flag("-sanitize=address")))
+
+        let linkJob = plannedJobs[3]
+        let linkCmd = linkJob.commandLine
+        XCTAssertTrue(linkCmd.contains(.flag("-fsanitize=address")))
+      }
+    }
+    do {
+      try checkWASITarget(target: "wasm32-unknown-wasi", clangOSDir: "wasi")
+      try checkWASITarget(target: "wasm32-unknown-wasip1", clangOSDir: "wasip1")
+      try checkWASITarget(target: "wasm32-unknown-wasip1-threads", clangOSDir: "wasip1")
+    }
   }
 
   func testSanitizerCoverageArgs() throws {