N行代码实现一个简单的代理服务器
<?php
/**
* Web代理服务器(支持http/https)
* @author zhjx922
*/
class WebProxyServer
{
private $_client = [];
private $_server;
/**
* 日志打印
* @author zhjx922
* @param string $message
*/
protected function log($message)
{
echo $message . PHP_EOL;
}
/**
* 获取代理ip
* @author zhjx922
*/
protected function getLocalIp()
{
//获取代理IP
$ipList = swoole_get_local_ip();
foreach($ipList as $interface => $ip) {
$this->log("{$interface}:{$ip}");
}
}
/**
* 初始化
* @author zhjx922
*/
protected function init()
{
$this->getLocalIp();
$this->_server = new swoole_server("0.0.0.0", 8889);
$this->_server->set([
'buffer_output_size' => 64 * 1024 *1024, //必须为数字
]);
}
/**
* 跑起来
* @author zhjx922
*/
public function run()
{
$this->init();
$this->_server->on('connect', function ($server, $fd){
$this->log("Server connection open: {$fd}");
});
$this->_server->on('receive', function ($server, $fd, $reactor_id, $buffer){
//判断是否为新连接
if(!isset($this->_client[$fd])) {
//判断代理模式
list($method, $url) = explode(' ', $buffer, 3);
$url = parse_url($url);
//ipv6为啥外面还有个方括号？
if(strpos($url['host'], ']')) {
$url['host'] = str_replace(['[', ']'], '', $url['host']);
}
//解析host+port
$host = $url['host'];
$port = isset($url['port']) ? $url['port'] : 80;
//ipv4/v6处理
$tcpMode = strpos($url['host'], ':') !== false ? SWOOLE_SOCK_TCP6 : SWOOLE_SOCK_TCP;
$this->_client[$fd] = new swoole_client($tcpMode, SWOOLE_SOCK_ASYNC);
if($method == 'CONNECT')
{
$this->_client[$fd]->on("connect", function (swoole_client $cli) use ($fd) {
$this->log("隧道模式-连接成功!");
//告诉客户端准备就绪，可以继续发包
$this->_server->send($fd, "HTTP/1.1 200 Connection Established\r\n\r\n");
});
} else {
$this->_client[$fd]->on("connect", function(swoole_client $cli) use ($buffer) {
$this->log("正常模式-连接成功!");
//直接转发数据
$cli->send($buffer);
});
}
$this->_client[$fd]->on("receive", function(swoole_client $cli, $buffer) use ($fd){
//将收到的数据转发到客户端
if($this->_server->exist($fd)) {
$this->_server->send($fd, $buffer);
}
});
$this->_client[$fd]->on("error", function(swoole_client $cli) use ($fd){
$this->log("Client {$fd} error");
});
$this->_client[$fd]->on("close", function(swoole_client $cli) use ($fd){
$this->log("Client {$fd} connection close");
});
$this->_client[$fd]->connect($host, $port);
} else {
//已连接，正常转发数据
if($this->_client[$fd]->isConnected()) {
$this->_client[$fd]->send($buffer);
}
}
});
$this->_server->on('close', function ($server, $fd) {
$this->log("Server connection close: {$fd}");
unset($this->_client[$fd]);
});
$this->_server->start();
}
}
$server = new WebProxyServer();
$server->run();