Replace NEKit dependency with reduced subset

2020-03-24 21:12:58 +01:00
parent 2473e77519
commit cbec3981bb
103 changed files with 24996 additions and 264 deletions
--- a/GlassVPN/zhuhaow-NEKit/Socket/AdapterSocket/SOCKS5Adapter.swift
+++ b/GlassVPN/zhuhaow-NEKit/Socket/AdapterSocket/SOCKS5Adapter.swift
@@ -0,0 +1,112 @@
+import Foundation
+
+public class SOCKS5Adapter: AdapterSocket {
+    enum SOCKS5AdapterStatus {
+        case invalid,
+        connecting,
+        readingMethodResponse,
+        readingResponseFirstPart,
+        readingResponseSecondPart,
+        forwarding
+    }
+    public let serverHost: String
+    public let serverPort: Int
+
+    var internalStatus: SOCKS5AdapterStatus = .invalid
+
+    let helloData = Data(bytes: UnsafePointer<UInt8>(([0x05, 0x01, 0x00] as [UInt8])), count: 3)
+
+    public enum ReadTag: Int {
+        case methodResponse = -20000, connectResponseFirstPart, connectResponseSecondPart
+    }
+
+    public enum WriteTag: Int {
+        case open = -21000, connectIPv4, connectIPv6, connectDomainLength, connectPort
+    }
+
+    public init(serverHost: String, serverPort: Int) {
+        self.serverHost = serverHost
+        self.serverPort = serverPort
+        super.init()
+    }
+
+    public override func openSocketWith(session: ConnectSession) {
+        super.openSocketWith(session: session)
+
+        guard !isCancelled else {
+            return
+        }
+
+        do {
+            internalStatus = .connecting
+            try socket.connectTo(host: serverHost, port: serverPort, enableTLS: false, tlsSettings: nil)
+        } catch {}
+    }
+
+    public override func didConnectWith(socket: RawTCPSocketProtocol) {
+        super.didConnectWith(socket: socket)
+
+        write(data: helloData)
+        internalStatus = .readingMethodResponse
+        socket.readDataTo(length: 2)
+    }
+
+    public override func didRead(data: Data, from socket: RawTCPSocketProtocol) {
+        super.didRead(data: data, from: socket)
+
+        switch internalStatus {
+        case .readingMethodResponse:
+            var response: [UInt8]
+            if session.isIPv4() {
+                response = [0x05, 0x01, 0x00, 0x01]
+                let address = IPAddress(fromString: session.host)!
+                response += [UInt8](address.dataInNetworkOrder)
+            } else if session.isIPv6() {
+                response = [0x05, 0x01, 0x00, 0x04]
+                let address = IPAddress(fromString: session.host)!
+                response += [UInt8](address.dataInNetworkOrder)
+            } else {
+                response = [0x05, 0x01, 0x00, 0x03]
+                response.append(UInt8(session.host.utf8.count))
+                response += [UInt8](session.host.utf8)
+            }
+
+            let portBytes: [UInt8] = Utils.toByteArray(UInt16(session.port)).reversed()
+            response.append(contentsOf: portBytes)
+            write(data: Data(response))
+
+            internalStatus = .readingResponseFirstPart
+            socket.readDataTo(length: 5)
+        case .readingResponseFirstPart:
+            var readLength = 0
+            switch data[3] {
+            case 1:
+                readLength = 3 + 2
+            case 3:
+                readLength = Int(data[4]) + 2
+            case 4:
+                readLength = 15 + 2
+            default:
+                break
+            }
+            internalStatus = .readingResponseSecondPart
+            socket.readDataTo(length: readLength)
+        case .readingResponseSecondPart:
+            internalStatus = .forwarding
+            observer?.signal(.readyForForward(self))
+            delegate?.didBecomeReadyToForwardWith(socket: self)
+        case .forwarding:
+            delegate?.didRead(data: data, from: self)
+        default:
+            return
+        }
+    }
+
+    override open func didWrite(data: Data?, by socket: RawTCPSocketProtocol) {
+        super.didWrite(data: data, by: socket)
+
+        if internalStatus == .forwarding {
+            delegate?.didWrite(data: data, by: self)
+        }
+    }
+}