Fix connecting to IPv6 address via SOCSK5

Author: clue

src/Server.php

@@ -320,9 +320,15 @@ public function handleSocks5(ConnectionInterface $stream, $auth=null, StreamRead
 
     public function connectTarget(ConnectionInterface $stream, array $target)
     {
+        $uri = $target[0];
+        if (strpos($uri, ':') !== false) {
+            $uri = '[' . $uri . ']';
+        }
+        $uri = $uri . ':' . $target[1];
+
         $stream->emit('target', $target);
         $that = $this;
-        $connecting = $this->connector->connect($target[0] . ':' . $target[1]);
+        $connecting = $this->connector->connect($uri);
 
         $stream->on('close', function () use ($connecting) {
             $connecting->cancel();

tests/ServerTest.php

@@ -152,7 +152,7 @@ public function testHandleSocksConnectionWillEndOnInvalidData()
         $connection->emit('data', array('asdasdasdasdasd'));
     }
 
-    public function testHandleSocksConnectionWillEstablishOutgoingConnection()
+    public function testHandleSocks4ConnectionWillEstablishOutgoingConnection()
     {
         $connection = $this->getMockBuilder('React\Socket\Connection')->disableOriginalConstructor()->setMethods(array('pause', 'end'))->getMock();
 
@@ -165,6 +165,32 @@ public function testHandleSocksConnectionWillEstablishOutgoingConnection()
         $connection->emit('data', array("\x04\x01" . "\x00\x50" . pack('N', ip2long('127.0.0.1')) . "\x00"));
     }
 
+    public function testHandleSocks5ConnectionWillEstablishOutgoingConnection()
+    {
+        $connection = $this->getMockBuilder('React\Socket\Connection')->disableOriginalConstructor()->setMethods(array('pause', 'end', 'write'))->getMock();
+
+        $promise = new Promise(function () { });
+
+        $this->connector->expects($this->once())->method('connect')->with('127.0.0.1:80')->willReturn($promise);
+
+        $this->server->onConnection($connection);
+
+        $connection->emit('data', array("\x05\x01\x00" . "\x05\x01\x00\x01" . pack('N', ip2long('127.0.0.1')) . "\x00\x50"));
+    }
+
+    public function testHandleSocks5ConnectionWithIpv6WillEstablishOutgoingConnection()
+    {
+        $connection = $this->getMockBuilder('React\Socket\Connection')->disableOriginalConstructor()->setMethods(array('pause', 'end', 'write'))->getMock();
+
+        $promise = new Promise(function () { });
+
+        $this->connector->expects($this->once())->method('connect')->with('[::1]:80')->willReturn($promise);
+
+        $this->server->onConnection($connection);
+
+        $connection->emit('data', array("\x05\x01\x00" . "\x05\x01\x00\x04" . inet_pton('::1') . "\x00\x50"));
+    }
+
     public function testHandleSocksConnectionWillCancelOutputConnectionIfIncomingCloses()
     {
         $connection = $this->getMockBuilder('React\Socket\Connection')->disableOriginalConstructor()->setMethods(array('pause', 'end'))->getMock();