fix unsafe use of os_unfair_lock

Author: morganchen12

FirebaseAuth/Sources/Swift/Auth/AuthComponent.swift

@@ -33,7 +33,7 @@ class AuthComponent: NSObject, Library, ComponentLifecycleMaintainer {
   private var instances: [String: Auth] = [:]
 
   /// Lock to manage access to the instances array to avoid race conditions.
-  private var instancesLock: os_unfair_lock = .init()
+  private let instancesLock = OSAllocatedUnfairLock()
 
   // MARK: - Initializers
 
@@ -58,10 +58,10 @@ class AuthComponent: NSObject, Library, ComponentLifecycleMaintainer {
   // MARK: - AuthProvider conformance
 
   @discardableResult func auth() -> Auth {
-    os_unfair_lock_lock(&instancesLock)
+    instancesLock.lock()
 
     // Unlock before the function returns.
-    defer { os_unfair_lock_unlock(&instancesLock) }
+    defer { instancesLock.unlock() }
 
     if let instance = instances[app.name] {
       return instance

FirebaseFunctions/Sources/Functions.swift

@@ -56,7 +56,7 @@ enum FunctionsConstants {
   private static var instances: [String: [Functions]] = [:]
 
   /// Lock to manage access to the instances array to avoid race conditions.
-  private static var instancesLock: os_unfair_lock = .init()
+  private static let instancesLock = OSAllocatedUnfairLock()
 
   /// The custom domain to use for all functions references (optional).
   let customDomain: String?
@@ -304,10 +304,10 @@ enum FunctionsConstants {
     guard let app else {
       fatalError("`FirebaseApp.configure()` needs to be called before using Functions.")
     }
-    os_unfair_lock_lock(&instancesLock)
+    instancesLock.lock()
 
     // Unlock before the function returns.
-    defer { os_unfair_lock_unlock(&instancesLock) }
+    defer { instancesLock.unlock() }
 
     if let associatedInstances = instances[app.name] {
       for instance in associatedInstances {

FirebaseStorage/Sources/Storage.swift

@@ -249,13 +249,13 @@ import FirebaseCore
     private var instances: [String: Storage] = [:]
 
     /// Lock to manage access to the instances array to avoid race conditions.
-    private var instancesLock: os_unfair_lock = .init()
+    private let instancesLock = OSAllocatedUnfairLock
 
     private init() {}
 
     func storage(app: FirebaseApp, bucket: String) -> Storage {
-      os_unfair_lock_lock(&instancesLock)
-      defer { os_unfair_lock_unlock(&instancesLock) }
+      instancesLock.lock()
+      defer { instancesLock.unlock() }
 
       if let instance = instances[bucket] {
         return instance

FirebaseVertexAI/Sources/VertexAI.swift

@@ -124,20 +124,17 @@ public class VertexAI {
 
   let apiConfig: APIConfig
 
+  /// Lock to manage access to the `instances` array to avoid race conditions.
+  private static let instancesLock = OSAllocatedUnfairLock()
+
   #if compiler(>=6)
     /// A map of active  `VertexAI` instances keyed by the `FirebaseApp` name and the `location`, in
     /// the format `appName:location`.
     private nonisolated(unsafe) static var instances: [InstanceKey: VertexAI] = [:]
-
-    /// Lock to manage access to the `instances` array to avoid race conditions.
-    private nonisolated(unsafe) static var instancesLock: os_unfair_lock = .init()
   #else
     /// A map of active  `VertexAI` instances keyed by the `FirebaseApp` name and the `location`, in
     /// the format `appName:location`.
     private static var instances: [InstanceKey: VertexAI] = [:]
-
-    /// Lock to manage access to the `instances` array to avoid race conditions.
-    private static var instancesLock: os_unfair_lock = .init()
   #endif
 
   let location: String?
@@ -149,10 +146,10 @@ public class VertexAI {
       fatalError("No instance of the default Firebase app was found.")
     }
 
-    os_unfair_lock_lock(&instancesLock)
+    instancesLock.lock()
 
     // Unlock before the function returns.
-    defer { os_unfair_lock_unlock(&instancesLock) }
+    defer { instancesLock.unlock() }
 
     let instanceKey = InstanceKey(appName: app.name, location: location, apiConfig: apiConfig)
     if let instance = instances[instanceKey] {