112 lines
3.7 KiB
Swift
112 lines
3.7 KiB
Swift
import Foundation
|
|
import Network
|
|
|
|
protocol SocketDelegate: AnyObject {
|
|
func didChangeState(socket: String, state: SocketState)
|
|
func didReceiveMessage(socket: String, message: String)
|
|
}
|
|
|
|
public class Socket: NSObject {
|
|
|
|
var id: String
|
|
|
|
var connection: NWConnection?
|
|
|
|
weak var delegate: SocketDelegate?
|
|
|
|
var dataBuffer = Data()
|
|
|
|
public init(id: String) {
|
|
self.id = id
|
|
}
|
|
|
|
public func connect(host: String, port: Int, useTLS: Bool, acceptInvalidCertificates: Bool) {
|
|
let parameters = NWParameters.tcp
|
|
if useTLS {
|
|
let tls = NWProtocolTLS.Options()
|
|
|
|
if (acceptInvalidCertificates) {
|
|
sec_protocol_options_set_verify_block(tls.securityProtocolOptions, { (sec_protocol_metadata, sec_trust, sec_protocol_verify_complete) in
|
|
sec_protocol_verify_complete(true)
|
|
}, DispatchQueue.global())
|
|
}
|
|
|
|
parameters.defaultProtocolStack.applicationProtocols.insert(tls, at: 0)
|
|
}
|
|
|
|
let connection = NWConnection(host: NWEndpoint.Host(host), port: NWEndpoint.Port(String(port))!, using: parameters)
|
|
connection.stateUpdateHandler = self.stateDidChange(to:)
|
|
self.receive(on: connection)
|
|
connection.start(queue: DispatchQueue(label: "Igloo"))
|
|
self.connection = connection
|
|
}
|
|
|
|
public func write(_ message: String) {
|
|
if let data = message.data(using: .utf8) {
|
|
connection?.send(content: data, completion: .idempotent)
|
|
}
|
|
}
|
|
|
|
public func disconnect() {
|
|
connection?.cancel()
|
|
}
|
|
|
|
func stateDidChange(to state: NWConnection.State) {
|
|
switch state {
|
|
case .setup:
|
|
print("connection setup")
|
|
|
|
case .waiting(let error):
|
|
print("connection waiting: \(error)")
|
|
|
|
case .preparing:
|
|
self.delegate?.didChangeState(socket: self.id, state: .connecting)
|
|
|
|
case .ready:
|
|
self.delegate?.didChangeState(socket: self.id, state: .connected)
|
|
|
|
case .failed(let error):
|
|
print("connection failed: \(error)")
|
|
self.delegate?.didChangeState(socket: self.id, state: .disconnected)
|
|
self.disconnect()
|
|
|
|
case .cancelled:
|
|
self.delegate?.didChangeState(socket: self.id, state: .disconnected)
|
|
|
|
default:
|
|
print("other")
|
|
}
|
|
}
|
|
|
|
func receive(on connection: NWConnection) {
|
|
connection.receive(minimumIncompleteLength: 1, maximumLength: 65536) { (data, contentContext, isComplete, error) in
|
|
if let error = error {
|
|
print("Receive error: \(error)")
|
|
self.delegate?.didChangeState(socket: self.id, state: .disconnected)
|
|
return
|
|
}
|
|
|
|
if let data = data, !data.isEmpty {
|
|
self.dataBuffer.append(data) // Append the data to the buffer
|
|
|
|
// Process buffer if enough data is available (protocol-specific logic)
|
|
if let message = String(data: self.dataBuffer, encoding: .utf8) {
|
|
self.delegate?.didReceiveMessage(socket: self.id, message: message)
|
|
self.dataBuffer.removeAll() // Clear buffer after processing
|
|
}
|
|
}
|
|
|
|
if connection.state == .ready {
|
|
self.receive(on: connection)
|
|
}
|
|
|
|
if isComplete {
|
|
print("Connection completed.")
|
|
self.delegate?.didChangeState(socket: self.id, state: .disconnected)
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
}
|