swift-mirror/stdlib/public/EmbeddedPlatform/EmbeddedPlatformSingleThreaded.swift

//===----------- EmbeddedPlatformSingleThreaded.swift ---------------------===//
//
// This source file is part of the Swift.org open source project
//
// Copyright (c) 2026 Apple Inc. and the Swift project authors
// Licensed under Apache License v2.0 with Runtime Library Exception
//
// See https://swift.org/LICENSE.txt for license information
// See https://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
//
//===----------------------------------------------------------------------===//

@implementation @c
public func _swift_lockStandardOutput() {
}

@implementation @c
public func _swift_unlockStandardOutput() {
}

fileprivate struct SingleThreadedMutex {
  var checked: Bool
  var lockCount: UInt
}

@implementation @c
public func _swift_mutex_init(
  _ mutex: UnsafeMutableRawPointer,
  _ flags: SwiftMutexFlags
) {
  let storage = mutex.assumingMemoryBound(to: SingleThreadedMutex.self)
  storage.pointee = SingleThreadedMutex(
    checked: flags.contains(.checked),
    lockCount: 0)
}

@implementation @c
public func _swift_mutex_destroy(_ mutex: UnsafeMutableRawPointer) {
  let storage = mutex.assumingMemoryBound(to: SingleThreadedMutex.self)
  if storage.pointee.checked && storage.pointee.lockCount != 0 {
    fatalError("destroying a locked mutex")
  }

  storage.pointee = SingleThreadedMutex(checked: false, lockCount: 0)
}

@implementation @c
public func _swift_mutex_lock(_ mutex: UnsafeMutableRawPointer) {
  let storage = mutex.assumingMemoryBound(to: SingleThreadedMutex.self)
  if storage.pointee.checked {
    if storage.pointee.lockCount != 0 {
      fatalError("locking an already locked mutex")
    }
    storage.pointee.lockCount += 1
  }
}

@implementation @c
public func _swift_mutex_unlock(_ mutex: UnsafeMutableRawPointer) {
  let storage = mutex.assumingMemoryBound(to: SingleThreadedMutex.self)
  if storage.pointee.checked {
    if storage.pointee.lockCount == 0 {
      fatalError("unlocking an unlocked mutex")
    }
    storage.pointee.lockCount -= 1
  }
}

@implementation @c
public func _swift_mutex_tryLock(_ mutex: UnsafeMutableRawPointer) -> Int {
  let storage = mutex.assumingMemoryBound(to: SingleThreadedMutex.self)
  if storage.pointee.checked {
    if storage.pointee.lockCount != 0 {
      return 0
    }
    storage.pointee.lockCount += 1
  }
  return 1
}

fileprivate struct SingleThreadedRecursiveMutex {
  var checked: Bool
  var lockCount: UInt
}

@implementation @c
public func _swift_mutexRecursive_init(
  _ mutex: UnsafeMutableRawPointer,
  _ flags: SwiftMutexFlags
) {
  let storage = mutex.assumingMemoryBound(to: SingleThreadedRecursiveMutex.self)
  storage.pointee = SingleThreadedRecursiveMutex(
    checked: flags.contains(.checked),
    lockCount: 0)
}

@implementation @c
public func _swift_mutexRecursive_destroy(_ mutex: UnsafeMutableRawPointer) {
  let storage = mutex.assumingMemoryBound(to: SingleThreadedRecursiveMutex.self)
  if storage.pointee.checked && storage.pointee.lockCount != 0 {
    fatalError("destroying a locked mutex")
  }

  storage.pointee = SingleThreadedRecursiveMutex(checked: false, lockCount: 0)
}

@implementation @c
public func _swift_mutexRecursive_lock(_ mutex: UnsafeMutableRawPointer) {
  let storage = mutex.assumingMemoryBound(to: SingleThreadedRecursiveMutex.self)
  if storage.pointee.checked {
    storage.pointee.lockCount += 1
  }
}

@implementation @c
public func _swift_mutexRecursive_unlock(_ mutex: UnsafeMutableRawPointer) {
  let storage = mutex.assumingMemoryBound(to: SingleThreadedRecursiveMutex.self)
  if storage.pointee.checked {
    if storage.pointee.lockCount == 0 {
      fatalError("unlocking an unlocked mutex")
    }
    storage.pointee.lockCount -= 1
  }
}

fileprivate struct SingleThreadedTLS {
  static let keyCount = Int(SWIFT_TLS_KEY_COUNT)

  // Debuggers inspect this variable to find the currently executing task.
  @used
  @_expose(wasm)
  @_silgen_name("_swift_concurrency_debug_global_tls_array")
  static var values: [8 of UnsafeMutableRawPointer?] = [nil, nil, nil, nil, nil, nil, nil, nil]
}

// SWIFT_CONCURRENCY_CURRENT_TASK_STORAGE_KIND_GLOBAL_TLS_ARRAY, see
// ConcurrencyDebug.h.
@used
@_expose(wasm)
@_silgen_name("_swift_concurrency_debug_current_task_storage_kind")
public let _swift_concurrency_debug_current_task_storage_kind: UInt32 = 5

@implementation @c
public func _swift_tls_init(
  _ key: Int,
  _ destructor: (@convention(c) (UnsafeMutableRawPointer?) -> Void)?
) {
}

@implementation @c
public func _swift_tls_get(_ key: Int) -> UnsafeMutableRawPointer? {
  precondition(key >= 0 && key < SingleThreadedTLS.keyCount)
  return SingleThreadedTLS.values[key]
}

@implementation @c
public func _swift_tls_set(_ key: Int, _ value: UnsafeMutableRawPointer?) {
  precondition(key >= 0 && key < SingleThreadedTLS.keyCount)
  SingleThreadedTLS.values[key] = value
}

@implementation @c
public func _swift_thread_isMain() -> Int {
  1
}