neoarz/SideStore
1
0
Fork 0
mirror of https://github.com/SideStore/SideStore.git synced 2026-02-20 04:03:26 +01:00
Code Issues Packages Projects Releases Wiki Activity

Stops wired connection listening socket when entering background

Browse Source
This commit is contained in:
Riley Testut
2020-01-21 17:11:16 -08:00
parent e7930b95d0
commit b6ac0b5f06
1 changed files with 29 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

38
AltStore/Server/ServerManager.swift
View File

@@ -23,9 +23,9 @@ class ServerManager: NSObject
private let dispatchQueue = DispatchQueue(label: "io.altstore.ServerManager") private let dispatchQueue = DispatchQueue(label: "io.altstore.ServerManager")
private lazy var connectionListener = self.makeListener() private var connectionListener: NWListener?
private var incomingConnections = [NWConnection]() private var incomingConnections: [NWConnection]?
private let incomingConnectionsSemaphore = DispatchSemaphore(value: 0) private var incomingConnectionsSemaphore: DispatchSemaphore?
private override init() private override init()
{ {
@@ -45,7 +45,7 @@ extension ServerManager
self.serviceBrowser.searchForServices(ofType: ALTServerServiceType, inDomain: "") self.serviceBrowser.searchForServices(ofType: ALTServerServiceType, inDomain: "")
self.connectionListener.start(queue: self.dispatchQueue) self.startListeningForWiredConnections()
} }
func stopDiscovering() func stopDiscovering()
@@ -56,6 +56,8 @@ extension ServerManager
self.discoveredServers.removeAll() self.discoveredServers.removeAll()
self.services.removeAll() self.services.removeAll()
self.serviceBrowser.stop() self.serviceBrowser.stop()
self.stopListeningForWiredConnection()
} }
func connect(to server: Server, completion: @escaping (Result<ServerConnection, Error>) -> Void) func connect(to server: Server, completion: @escaping (Result<ServerConnection, Error>) -> Void)
@@ -92,16 +94,16 @@ extension ServerManager
connection.start(queue: self.dispatchQueue) connection.start(queue: self.dispatchQueue)
} }
if server.isWiredConnection if let incomingConnectionsSemaphore = self.incomingConnectionsSemaphore, server.isWiredConnection
{ {
print("Waiting for new wired connection...") print("Waiting for new wired connection...")
let notificationCenter = CFNotificationCenterGetDarwinNotifyCenter() let notificationCenter = CFNotificationCenterGetDarwinNotifyCenter()
CFNotificationCenterPostNotification(notificationCenter, .wiredServerConnectionStartRequest, nil, nil, true) CFNotificationCenterPostNotification(notificationCenter, .wiredServerConnectionStartRequest, nil, nil, true)
_ = self.incomingConnectionsSemaphore.wait(timeout: .now() + 10.0) _ = incomingConnectionsSemaphore.wait(timeout: .now() + 10.0)
if let connection = self.incomingConnections.popLast() if let connection = self.incomingConnections?.popLast()
{ {
start(connection) start(connection)
} }
@@ -135,8 +137,8 @@ private extension ServerManager
{ {
let listener = try! NWListener(using: .tcp, on: NWEndpoint.Port(rawValue: ALTDeviceListeningSocket)!) let listener = try! NWListener(using: .tcp, on: NWEndpoint.Port(rawValue: ALTDeviceListeningSocket)!)
listener.newConnectionHandler = { [weak self] (connection) in listener.newConnectionHandler = { [weak self] (connection) in
self?.incomingConnections.append(connection) self?.incomingConnections?.append(connection)
self?.incomingConnectionsSemaphore.signal() self?.incomingConnectionsSemaphore?.signal()
} }
listener.stateUpdateHandler = { (state) in listener.stateUpdateHandler = { (state) in
switch state switch state
@@ -151,6 +153,24 @@ private extension ServerManager
return listener return listener
} }
func startListeningForWiredConnections()
{
self.incomingConnections = []
self.incomingConnectionsSemaphore = DispatchSemaphore(value: 0)
self.connectionListener = self.makeListener()
self.connectionListener?.start(queue: self.dispatchQueue)
}
func stopListeningForWiredConnection()
{
self.connectionListener?.cancel()
self.connectionListener = nil
self.incomingConnections = nil
self.incomingConnectionsSemaphore = nil
}
} }
extension ServerManager: NetServiceBrowserDelegate extension ServerManager: NetServiceBrowserDelegate