SOCKS5 proxy will work correctly when dealing with socket sending dat…

…a first
zhuhaow · Dec 9, 2016 · bc6ff80 · bc6ff80
1 parent 0bf5b26
commit bc6ff80
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -2,6 +2,11 @@
 All notable changes to this project will be documented in this file.
 I will do my best to guarantee that this project adheres to [Semantic Versioning](http://semver.org/) after 1.0.0, but please do read change log before updating.
 
+## 0.10.1
+
+### Fixed
+- SOCKS5 proxy will work correctly when dealing with socket sending data first.
+
 ## 0.10.0
 
 ### Changed

diff --git a/src/Socket/ProxySocket/SOCKS5ProxySocket.swift b/src/Socket/ProxySocket/SOCKS5ProxySocket.swift
@@ -76,6 +76,17 @@ public class SOCKS5ProxySocket: ProxySocket {
         socket.readDataTo(length: 2)
     }
 
+    public override func write(data: Data) {
+        switch internalStatus {
+        case .waitingToForward:
+            internalStatus = .forwarding
+        default:
+            break
+        }
+
+        super.write(data: data)
+    }
+
     // swiftlint:disable function_body_length
     // swiftlint:disable cyclomatic_complexity
     /**