From ddd3550a438fef9fd2cfc39a4e84683d3eb9117d Mon Sep 17 00:00:00 2001
From: walkor <walkor@workerman.net>
Date: Sun, 5 Apr 2015 23:03:09 +0800
Subject: [PATCH] ok

---
 Applications/PassWall/start.php | 137 +++++++++++++++++++++++++++++++-
 1 file changed, 136 insertions(+), 1 deletion(-)

diff --git a/Applications/PassWall/start.php b/Applications/PassWall/start.php
index af20367..5169fac 100644
--- a/Applications/PassWall/start.php
+++ b/Applications/PassWall/start.php
@@ -2,14 +2,149 @@
 use \Workerman\Worker;
 use \Workerman\WebServer;
 use \Workerman\Connection\TcpConnection;
+use \Workerman\Connection\AsyncTcpConnection;
 
 // 自动加载类
 require_once __DIR__ . '/../../Workerman/Autoloader.php';
 
+define('STAGE_INIT', 0);
+define('STAGE_ADDR', 1);
+define('STAGE_UDP_ASSOC', 2);
+define('STAGE_DNS', 3);
+define('STAGE_CONNECTING', 4);
+define('STAGE_STREAM', 5);
+define('STAGE_DESTROYED', -1);
 
 
+define('CMD_CONNECT', 1);
+define('CMD_BIND', 2);
+define('CMD_UDP_ASSOCIATE', 3);
+
+define('ADDRTYPE_IPV4', 1);
+define('ADDRTYPE_IPV6', 4);
+define('ADDRTYPE_HOST', 3);
+
+
+$worker = new Worker('tcp://0.0.0.0:1080');
+$worker->onConnect = function($connection)
+{
+    $connection->stage = STAGE_INIT;
+};
+$worker->onMessage = function($connection, $buffer)
+{
+    switch($connection->stage)
+    {
+        case STAGE_INIT:
+            $connection->send("\x05\x00");
+            $connection->stage = STAGE_ADDR;
+            return;
+        case STAGE_ADDR:
+            $cmd = ord($buffer[1]);
+            if($cmd != CMD_CONNECT)
+            {
+               echo "bad cmd $cmd\n";
+               $connection->close();
+               return;
+            }
+            $header_data = parse_socket5_header($buffer);
+            if(!$header_data)
+            {
+                $connection->close();
+                return;
+            }
+            //$connection->send("\x05\x00\x00\x01\x00\x00\x00\x00\x10\x10");
+            //$connection->stage = STAGE_DNS;
+            $connection->stage = STAGE_CONNECTING;
+            $remote_connection = new AsyncTcpConnection('tcp://'.$header_data[1].':'.$header_data[2]);
+            $remote_connection->onConnect = function($remote_connection)use($connection)
+            {
+                $connection->state = STAGE_STREAM;
+                $connection->send("\x05\x00\x00\x01\x00\x00\x00\x00\x10\x10");
+            };
+            $remote_connection->onMessage = function($remote_connection, $buffer)use($connection)
+            {
+                $connection->send($buffer);
+            };
+            $remote_connection->onClose = function($remote_connection)use($connection)
+            {
+                $connection->close();
+            };
+            $remote_connection->onError = function($remote_connection, $code, $type)use($connection)
+            {
+                if($connection->stage == STAGE_CONNECTING)
+                {
+                    $connection->send("\x05\x03\x00\x01\x00\x00\x00\x00\x10\x10");
+                }
+                $connection->close();
+            };
+            $connection->onMessage = function($connection, $data)use($remote_connection)
+            {
+                $remote_connection->send($data);
+            };
+            $connection->onClose = function($connection)use($remote_connection)
+            {
+                $remote_connection->close();
+            };
+            $connection->onError = function($connection)use($remote_connection)
+            {
+                echo "connection err\n";
+                $connection->close();
+                $remote_connection->close();
+            };
+            
+    }
+};
+
+
+function parse_socket5_header($buffer)
+{
+    $addr_type = ord($buffer[3]);
+    switch($addr_type)
+    {
+        case ADDRTYPE_IPV4:
+            if(strlen($buffer) < 10)
+            {
+                echo bin2hex($buffer)."\n";
+                echo "buffer too short\n";
+                return false;
+            }
+            $dest_addr = ord($buffer[4]).'.'.ord($buffer[5]).'.'.ord($buffer[6]).'.'.ord($buffer[7]);
+            $port_data = unpack('n', substr($buffer, -2));
+            $dest_port = $port_data[1];
+            $header_length = 10;
+            break;
+        case ADDRTYPE_HOST:
+            $addrlen = ord($buffer[4]);
+echo $addrlen."<<<<<<<<<\n";
+            if(strlen($buffer) < $addrlen + 5)
+            {
+                echo $buffer."\n";
+                echo bin2hex($buffer)."\n";
+                echo "buffer too short\n";
+                return false;
+            }
+            $dest_addr = substr($buffer, 5, $addrlen);
+            $port_data = unpack('n', substr($buffer, -2));
+            $dest_port = $port_data[1];
+            $header_length = $addrlen + 7;
+            break;
+       case ADDRTYPE_IPV6:
+            if(strlen($buffer) < 22)
+            {
+                echo "buffer too short\n";
+                return false;
+            }
+            echo "todo ipv6\n";
+            return false;
+       default:
+            echo "unsupported addrtype $addr_type\n";
+            return false;
+    }
+    return array($addr_type, $dest_addr, $dest_port, $header_length);
+}
+
 // 如果不是在根目录启动，则运行runAll方法
 if(!defined('GLOBAL_START'))
 {
     Worker::runAll();
-}
\ No newline at end of file
+}