test: add handling for Wasm/WASI (#39519)

This change adds support for WASI in stdlib tests. Some tests that expect a crash to happen had to be disabled, since there's currently no way to observe such crash from a WASI host.

Author: MaxDesiatov

stdlib/private/StdlibUnittest/RaceTest.swift

@@ -43,6 +43,8 @@ import SwiftPrivateThreadExtras
 import Darwin
 #elseif canImport(Glibc)
 import Glibc
+#elseif os(WASI)
+import WASILibc
 #elseif os(Windows)
 import CRT
 import WinSDK
@@ -335,6 +337,36 @@ public func evaluateObservationsAllEqual<T : Equatable>(_ observations: [T])
   return .pass
 }
 
+// WebAssembly/WASI doesn't support multi-threading yet
+#if os(WASI)
+public func runRaceTest<RT : RaceTestWithPerTrialData>(
+  _: RT.Type,
+  trials: Int,
+  timeoutInSeconds: Int? = nil,
+  threads: Int? = nil
+) {}
+public func runRaceTest<RT : RaceTestWithPerTrialData>(
+  _ test: RT.Type,
+  operations: Int,
+  timeoutInSeconds: Int? = nil,
+  threads: Int? = nil
+) {}
+public func consumeCPU(units amountOfWork: Int) {}
+public func runRaceTest(
+  trials: Int,
+  timeoutInSeconds: Int? = nil,
+  threads: Int? = nil,
+  invoking body: @escaping () -> Void
+) {}
+
+public func runRaceTest(
+  operations: Int,
+  timeoutInSeconds: Int? = nil,
+  threads: Int? = nil,
+  invoking body: @escaping () -> Void
+) {}
+#else
+
 struct _RaceTestAggregatedEvaluations : CustomStringConvertible {
   var passCount: Int = 0
   var passInterestingCount = [String: Int]()
@@ -756,3 +788,4 @@ public func runRaceTest(
     timeoutInSeconds: timeoutInSeconds, threads: threads)
 }
 
+#endif // os(WASI)

stdlib/private/StdlibUnittest/StdlibCoreExtras.swift

@@ -16,6 +16,8 @@ import SwiftPrivateLibcExtras
 import Darwin
 #elseif canImport(Glibc)
 import Glibc
+#elseif os(WASI)
+import WASILibc
 #elseif os(Windows)
 import CRT
 #endif

stdlib/private/StdlibUnittest/SymbolLookup.swift

@@ -14,6 +14,8 @@
   import Darwin
 #elseif canImport(Glibc)
   import Glibc
+#elseif os(WASI)
+  import WASILibc
 #elseif os(Windows)
   import CRT
   import WinSDK
@@ -23,7 +25,7 @@
 
 #if canImport(Darwin) || os(OpenBSD)
   let RTLD_DEFAULT = UnsafeMutableRawPointer(bitPattern: -2)
-#elseif os(Linux)
+#elseif os(Linux) || os(WASI)
   let RTLD_DEFAULT = UnsafeMutableRawPointer(bitPattern: 0)
 #elseif os(Android)
   #if arch(arm) || arch(i386)

stdlib/private/SwiftPrivateLibcExtras/Subprocess.swift

@@ -15,6 +15,8 @@ import SwiftPrivate
 import Darwin
 #elseif canImport(Glibc)
 import Glibc
+#elseif os(WASI)
+import WASILibc
 #elseif os(Windows)
 import CRT
 import WinSDK

stdlib/private/SwiftPrivateLibcExtras/SwiftPrivateLibcExtras.swift

@@ -15,6 +15,8 @@ import SwiftPrivate
 import Darwin
 #elseif canImport(Glibc)
 import Glibc
+#elseif os(WASI)
+import WASILibc
 #elseif os(Windows)
 import CRT
 #endif

stdlib/private/SwiftPrivateThreadExtras/SwiftPrivateThreadExtras.swift

@@ -19,6 +19,8 @@
 import Darwin
 #elseif canImport(Glibc)
 import Glibc
+#elseif os(WASI)
+import WASILibc
 #elseif os(Windows)
 import CRT
 import WinSDK
@@ -98,6 +100,9 @@ public func _stdlib_thread_create_block<Argument, Result>(
   } else {
     return (0, ThreadHandle(bitPattern: threadID))
   }
+#elseif os(WASI)
+  // WASI environment has a only single thread
+  return (0, nil)
 #else
   var threadID = _make_pthread_t()
   let result = pthread_create(&threadID, nil,
@@ -129,6 +134,9 @@ public func _stdlib_thread_join<Result>(
     }
   }
   return (CInt(result), value)
+#elseif os(WASI)
+   // WASI environment has a only single thread
+   return (0, nil)
 #else
   var threadResultRawPtr: UnsafeMutableRawPointer?
   let result = pthread_join(thread, &threadResultRawPtr)

stdlib/private/SwiftPrivateThreadExtras/ThreadBarriers.swift

@@ -14,6 +14,8 @@
 import Darwin
 #elseif canImport(Glibc)
 import Glibc
+#elseif os(WASI)
+import WASILibc
 #elseif os(Windows)
 import CRT
 import WinSDK
@@ -36,6 +38,8 @@ public struct _stdlib_thread_barrier_t {
 #elseif os(Cygwin) || os(FreeBSD) || os(OpenBSD)
   var mutex: UnsafeMutablePointer<pthread_mutex_t?>?
   var cond: UnsafeMutablePointer<pthread_cond_t?>?
+#elseif os(WASI)
+  // pthread is currently not available on WASI
 #else
   var mutex: UnsafeMutablePointer<pthread_mutex_t>?
   var cond: UnsafeMutablePointer<pthread_cond_t>?
@@ -67,6 +71,8 @@ public func _stdlib_thread_barrier_init(
 
   barrier.pointee.cond = UnsafeMutablePointer.allocate(capacity: 1)
   InitializeConditionVariable(barrier.pointee.cond!)
+#elseif os(WASI)
+  // WASI environment has only a single thread
 #else
   barrier.pointee.mutex = UnsafeMutablePointer.allocate(capacity: 1)
   barrier.pointee.cond = UnsafeMutablePointer.allocate(capacity: 1)
@@ -82,7 +88,7 @@ public func _stdlib_thread_barrier_init(
   return 0
 }
 
-#if !os(Windows)
+#if !os(Windows) && !os(WASI)
 private func _stdlib_thread_barrier_mutex_and_cond_init(_ barrier: UnsafeMutablePointer<_stdlib_thread_barrier_t>) -> CInt {
   guard pthread_mutex_init(barrier.pointee.mutex!, nil) == 0 else {
     return -1
@@ -101,17 +107,22 @@ public func _stdlib_thread_barrier_destroy(
 #if os(Windows)
   // Condition Variables do not need to be explicitly destroyed
   // Mutexes do not need to be explicitly destroyed
+#elseif os(WASI)
+  // WASI environment has only a single thread
 #else
   guard pthread_cond_destroy(barrier.pointee.cond!) == 0 &&
     pthread_mutex_destroy(barrier.pointee.mutex!) == 0 else {
     fatalError("_stdlib_thread_barrier_destroy() failed")
   }
 #endif
+
+#if !os(WASI)
   barrier.pointee.cond!.deinitialize(count: 1)
   barrier.pointee.cond!.deallocate()
 
   barrier.pointee.mutex!.deinitialize(count: 1)
   barrier.pointee.mutex!.deallocate()
+#endif
 
   return
 }
@@ -121,6 +132,8 @@ public func _stdlib_thread_barrier_wait(
 ) -> CInt {
 #if os(Windows)
   AcquireSRWLockExclusive(barrier.pointee.mutex!)
+#elseif os(WASI)
+  // WASI environment has only a single thread
 #else
   if pthread_mutex_lock(barrier.pointee.mutex!) != 0 {
     return -1
@@ -135,6 +148,8 @@ public func _stdlib_thread_barrier_wait(
       return -1
     }
     ReleaseSRWLockExclusive(barrier.pointee.mutex!)
+#elseif os(WASI)
+  // WASI environment has a only single thread
 #else
     if pthread_cond_wait(barrier.pointee.cond!, barrier.pointee.mutex!) != 0 {
       return -1
@@ -152,6 +167,8 @@ public func _stdlib_thread_barrier_wait(
 #if os(Windows)
     WakeAllConditionVariable(barrier.pointee.cond!)
     ReleaseSRWLockExclusive(barrier.pointee.mutex!)
+#elseif os(WASI)
+  // WASI environment has a only single thread
 #else
     if pthread_cond_broadcast(barrier.pointee.cond!) != 0 {
       return -1

stdlib/public/Concurrency/Actor.cpp

@@ -299,6 +299,8 @@ static bool isExecutingOnMainThread() {
   }
 
   return __initialPthread == GetCurrentThread();
+#elif defined(__wasi__)
+  return true;
 #else
   return pthread_main_np() == 1;
 #endif

stdlib/public/Concurrency/AsyncLet.cpp

@@ -26,7 +26,7 @@
 #include "TaskPrivate.h"
 #include "Debug.h"
 
-#if !defined(_WIN32)
+#if !defined(_WIN32) && !defined(__wasi__)
 #include <dlfcn.h>
 #endif
 

stdlib/public/Concurrency/Task.cpp

@@ -32,7 +32,7 @@
 #include <dispatch/dispatch.h>
 #endif
 
-#if !defined(_WIN32)
+#if !defined(_WIN32) && !defined(__wasi__)
 #include <dlfcn.h>
 #endif