aboutsummaryrefslogtreecommitdiff
path: root/Hotline/Shared/NetSocket.swift
diff options
context:
space:
mode:
authorRuben Beltran del Rio <git@r.bdr.sh>2025-11-27 23:38:04 +0100
committerRuben Beltran del Rio <git@r.bdr.sh>2025-11-27 23:38:04 +0100
commit213710bf5bd6413c747bf126db50816ef5de5a6e (patch)
tree912f8cf87955a08077c92fea8ad934f50b7ab975 /Hotline/Shared/NetSocket.swift
parentf466b21dc02f78c984ba6748e703f6780a7a0db4 (diff)
parent6a95b53616a4abfa306ddce43151cf4fefbd20ed (diff)
Merge remote-tracking branch 'upstream/main'
Diffstat (limited to 'Hotline/Shared/NetSocket.swift')
-rw-r--r--Hotline/Shared/NetSocket.swift272
1 files changed, 0 insertions, 272 deletions
diff --git a/Hotline/Shared/NetSocket.swift b/Hotline/Shared/NetSocket.swift
deleted file mode 100644
index 59c26af..0000000
--- a/Hotline/Shared/NetSocket.swift
+++ /dev/null
@@ -1,272 +0,0 @@
-// NetSocket.swift
-// A simple delegate based buffered read/write TCP socket.
-// Created by Dustin Mierau
-
-import Foundation
-
-protocol NetSocketDelegate: AnyObject {
- @MainActor func netsocketConnected(socket: NetSocket)
- @MainActor func netsocketDisconnected(socket: NetSocket, error: Error?)
- @MainActor func netsocketReceived(socket: NetSocket, bytes: [UInt8])
- func netsocketSent(socket: NetSocket, count: Int)
-}
-
-extension NetSocketDelegate {
- func netsocketConnected(socket: NetSocket) {}
- func netsocketDisconnected(socket: NetSocket, error: Error?) {}
- func netsocketReceived(socket: NetSocket, bytes: [UInt8]) {}
- func netsocketSent(socket: NetSocket, count: Int) {}
-}
-
-enum NetSocketStatus {
- case disconnected
- case connecting
- case connected
-}
-
-final class NetSocket: NSObject, StreamDelegate {
- weak var delegate: NetSocketDelegate? = nil
-
- private var output: OutputStream? = nil
- private var input: InputStream? = nil
-
- private var outputBuffer: [UInt8] = []
- private var inputBuffer: [UInt8] = []
-
- private var readBuffer: [UInt8] = Array(repeating: 0, count: 4 * 1024)
-
- public func peek() -> [UInt8] { self.inputBuffer }
- public var available: Int { self.inputBuffer.count }
-
- private var status: NetSocketStatus = .disconnected
-
- @MainActor public func has(_ length: Int) -> Bool {
- return (self.available >= length)
- }
-
- override init() {}
-
- @MainActor public func connect(host: String, port: Int) {
- self.close()
-
- var outputStream: OutputStream? = nil
- var inputStream: InputStream? = nil
-
- self.status = .connecting
-
- Stream.getStreamsToHost(
- withName: host, port: port, inputStream: &inputStream, outputStream: &outputStream)
-
- self.input = inputStream
- self.output = outputStream
-
- inputStream?.delegate = self
- outputStream?.delegate = self
-
- inputStream?.schedule(in: .current, forMode: .default)
- outputStream?.schedule(in: .current, forMode: .default)
-
- inputStream?.open()
- outputStream?.open()
- }
-
- @MainActor public func close(_ err: Error? = nil) {
- print("NetSocket: Closed")
-
- let disconnected = (self.status != .disconnected)
-
- self.status = .disconnected
-
- self.input?.delegate = nil
- self.output?.delegate = nil
- self.input?.close()
- self.output?.close()
- self.input?.remove(from: .current, forMode: .default)
- self.output?.remove(from: .current, forMode: .default)
- self.input = nil
- self.output = nil
- self.inputBuffer = []
- self.outputBuffer = []
-
- if disconnected {
- self.delegate?.netsocketDisconnected(socket: self, error: err)
- }
- }
-
- @MainActor public func write(_ data: Data) {
- guard let output = self.output else {
- return
- }
-
- self.outputBuffer.append(contentsOf: data)
-
- if output.hasSpaceAvailable {
- self.writeBufferToStream()
- }
- }
-
- @MainActor public func write(_ data: [UInt8]) {
- guard let output = self.output else {
- return
- }
-
- self.outputBuffer.append(contentsOf: data)
-
- if output.hasSpaceAvailable {
- self.writeBufferToStream()
- }
- }
-
- @MainActor public func read(count: Int) -> [UInt8] {
- guard self.inputBuffer.count > 0, count > 0 else {
- return []
- }
-
- let amountToRead = min(count, self.inputBuffer.count)
- let dataRead: [UInt8] = Array(self.inputBuffer[0..<amountToRead])
- self.inputBuffer.removeFirst(amountToRead)
-
- return dataRead
- }
-
- @MainActor public func read(count: Int) -> Data {
- guard self.inputBuffer.count > 0, count > 0 else {
- return Data()
- }
-
- let amountToRead = min(count, self.inputBuffer.count)
-
- let dataRead: Data = Data(self.inputBuffer[0..<amountToRead])
- self.inputBuffer.removeFirst(amountToRead)
-
- return dataRead
- }
-
- @MainActor public func readAll() -> [UInt8] {
- guard self.inputBuffer.count > 0 else {
- return []
- }
-
- let dataRead: [UInt8] = Array(self.inputBuffer)
- self.inputBuffer = []
-
- return dataRead
- }
-
- @MainActor public func readAll() -> Data {
- guard self.inputBuffer.count > 0 else {
- return Data()
- }
-
- let dataRead: Data = Data(self.inputBuffer)
- self.inputBuffer = []
-
- return dataRead
- }
-
- @MainActor private func writeBufferToStream() {
- guard let output = self.output, self.outputBuffer.count > 0 else {
- return
- }
-
- let bytesWritten = output.write(self.outputBuffer, maxLength: self.outputBuffer.count)
- print("NetSocket => \(bytesWritten) bytes")
- if bytesWritten > 0 {
- self.outputBuffer.removeFirst(bytesWritten)
- self.delegate?.netsocketSent(socket: self, count: bytesWritten)
- } else if bytesWritten == -1 {
- self.close(output.streamError)
- }
- }
-
- @MainActor private func readStreamToBuffer() {
- guard let input = self.input else {
- return
- }
-
- let bytesRead = input.read(&self.readBuffer, maxLength: 4 * 1024)
- print("NetSocket <= \(bytesRead) bytes")
- if bytesRead > 0 {
- self.inputBuffer.append(contentsOf: self.readBuffer[0..<bytesRead])
- self.delegate?.netsocketReceived(socket: self, bytes: self.inputBuffer)
- } else if bytesRead == -1 {
- self.close(input.streamError)
- }
- }
-
- // MARK: -
-
- @MainActor func stream(_ aStream: Stream, handle eventCode: Stream.Event) {
- guard let input = self.input, let output = self.output else {
- return
- }
-
- switch eventCode {
- case .openCompleted:
- if aStream == input {
- self.setupStreamOptions()
- }
- if input.streamStatus == .open && output.streamStatus == .open {
- if self.status == .connecting {
- print("NetSocket: Connected")
- self.status = .connected
- self.delegate?.netsocketConnected(socket: self)
- }
- }
- break
- case .hasBytesAvailable:
- print("NetSocket: Data available")
- self.readStreamToBuffer()
- case .hasSpaceAvailable:
- print("NetSocket: Space available")
- self.writeBufferToStream()
- case .endEncountered:
- print("NetSocket: End encounterd")
- self.close()
- case .errorOccurred:
- print(
- "NetSocket: Error encounterd", input.streamError.debugDescription,
- output.streamError.debugDescription)
- let err = input.streamError ?? output.streamError
- self.close(err)
- default:
- break
- }
- }
-
- // MARK: -
-
- private func setupStreamOptions() {
- if let input = self.input {
- let socketData: Data =
- CFReadStreamCopyProperty(input as CFReadStream, CFStreamPropertyKey.socketNativeHandle)
- as! Data
- var socketHandle: CFSocketNativeHandle = 0
- (socketData as NSData).getBytes(
- &socketHandle, length: MemoryLayout.size(ofValue: socketHandle))
-
- var value: Int = 0
- let size = UInt32(MemoryLayout.size(ofValue: value))
-
- value = 1
- if setsockopt(socketHandle, IPPROTO_TCP, TCP_NODELAY, &value, size) != 0 {
- print("NetSocket: failed to set TCP_NODELAY")
- }
- // Enable keepalive
- value = 1
- if setsockopt(socketHandle, SOL_SOCKET, SO_KEEPALIVE, &value, size) != 0 {
- print("NetSocket: failed to set SO_KEEPALIVE")
- }
- // // Number of keepalives before close (including first keepalive packet)
- // value = 5
- // if setsockopt(socketHandle, IPPROTO_TCP, TCP_KEEPCNT, &value, size) != 0 {
- // print("NetSocket: failed to set TCP_KEEPCNT");
- // }
- // // Idle time used when SO_KEEPALIVE is enabled. Sets how long connection must be idle before keepalive is sent.
- // value = 60
- // if setsockopt(socketHandle, IPPROTO_TCP, TCP_KEEPALIVE, &value, size) != 0 {
- // print("NetSocket: failed to set TCP_KEEPALIVE")
- // }
- }
- }
-}