Ben Davis
122 lines
3.9 KiB
Swift
122 lines
3.9 KiB
Swift
//
|
|
// TCPConnection.swift
|
|
// BitTorrent
|
|
//
|
|
// Created by Ben Davis on 06/07/2017.
|
|
// Copyright © 2017 Ben Davis. All rights reserved.
|
|
//
|
|
|
|
import Foundation
|
|
import CocoaAsyncSocket
|
|
|
|
protocol TCPConnectionProtocol: class {
|
|
weak var delegate: TCPConnectionDelegate? { set get }
|
|
|
|
var connectedHost: String? { get }
|
|
var connectedPort: UInt16? { get }
|
|
|
|
var connected: Bool { get }
|
|
func connect(to host: String, onPort port: UInt16) throws
|
|
func disconnect()
|
|
|
|
func readData(withTimeout timeout: TimeInterval, tag: Int)
|
|
func write(_ data: Data, withTimeout timeout: TimeInterval, tag: Int)
|
|
func write(_ data: Data, withTimeout timeout: TimeInterval, completion: (()->Void)?)
|
|
}
|
|
|
|
protocol TCPConnectionDelegate: class {
|
|
func tcpConnection(_ sender: TCPConnectionProtocol, didConnectToHost host: String, port: UInt16)
|
|
func tcpConnection(_ sender: TCPConnectionProtocol, didRead data: Data, withTag tag: Int)
|
|
func tcpConnection(_ sender: TCPConnectionProtocol, didWriteDataWithTag tag: Int)
|
|
func tcpConnection(_ sender: TCPConnectionProtocol, disconnectedWithError error: Error?)
|
|
}
|
|
|
|
/// This class is a thin wrapper around the socket library to protect against changes
|
|
/// in its interface, and to allow me to replace CocoaAsyncSocket with a swift framework
|
|
/// one day.
|
|
class TCPConnection: NSObject, TCPConnectionProtocol {
|
|
|
|
weak var delegate: TCPConnectionDelegate?
|
|
|
|
private let socket: GCDAsyncSocket
|
|
|
|
private var currentTag: Int = 1000 // This class will use tags above 1000 incrementally
|
|
private var completionBlocks: [Int: ()->Void] = [:]
|
|
|
|
var connectedHost: String? {
|
|
return socket.connectedHost
|
|
}
|
|
|
|
var connectedPort: UInt16? {
|
|
guard connectedHost != nil else {
|
|
return nil
|
|
}
|
|
return socket.connectedPort
|
|
}
|
|
|
|
var connected: Bool {
|
|
return socket.isConnected
|
|
}
|
|
|
|
init(socket: GCDAsyncSocket = GCDAsyncSocket()) {
|
|
self.socket = socket
|
|
super.init()
|
|
socket.delegateQueue = .main
|
|
socket.synchronouslySetDelegate(self)
|
|
}
|
|
|
|
func connect(to host: String, onPort port: UInt16) throws {
|
|
try socket.connect(toHost: host, onPort: port, withTimeout: 15)
|
|
}
|
|
|
|
func disconnect() {
|
|
socket.delegate = nil
|
|
socket.disconnect()
|
|
}
|
|
|
|
func readData(withTimeout timeout: TimeInterval, tag: Int) {
|
|
socket.readData(withTimeout: timeout, tag: tag)
|
|
}
|
|
|
|
func write(_ data: Data, withTimeout timeout: TimeInterval, tag: Int) {
|
|
write(data, withTimeout: timeout, tag: tag, completion: nil)
|
|
}
|
|
|
|
func write(_ data: Data, withTimeout timeout: TimeInterval, completion: (() -> Void)?) {
|
|
write(data, withTimeout: timeout, tag: nil, completion: completion)
|
|
}
|
|
|
|
func write(_ data: Data, withTimeout timeout: TimeInterval, tag: Int? = nil, completion: (()->Void)? = nil) {
|
|
let tag = tag ?? nextTag()
|
|
completionBlocks[tag] = completion
|
|
socket.write(data, withTimeout: timeout, tag: tag)
|
|
}
|
|
|
|
func nextTag() -> Int {
|
|
let result = currentTag
|
|
currentTag += 1
|
|
return result
|
|
}
|
|
}
|
|
|
|
extension TCPConnection: GCDAsyncSocketDelegate {
|
|
|
|
func socket(_ sock: GCDAsyncSocket, didConnectToHost host: String, port: UInt16) {
|
|
delegate?.tcpConnection(self, didConnectToHost: host, port: port)
|
|
}
|
|
|
|
func socket(_ sock: GCDAsyncSocket, didRead data: Data, withTag tag: Int) {
|
|
delegate?.tcpConnection(self, didRead: data, withTag: tag)
|
|
}
|
|
|
|
func socket(_ sock: GCDAsyncSocket, didWriteDataWithTag tag: Int) {
|
|
completionBlocks[tag]?()
|
|
completionBlocks[tag] = nil
|
|
delegate?.tcpConnection(self, didWriteDataWithTag: tag)
|
|
}
|
|
|
|
func socketDidDisconnect(_ sock: GCDAsyncSocket, withError err: Error?) {
|
|
delegate?.tcpConnection(self, disconnectedWithError: err)
|
|
}
|
|
}
|