155 lines
4.2 KiB
PHP
155 lines
4.2 KiB
PHP
<?php
|
||
/**
|
||
* This file is part of workerman.
|
||
*
|
||
* Licensed under The MIT License
|
||
* For full copyright and license information, please see the MIT-LICENSE.txt
|
||
* Redistributions of files must retain the above copyright notice.
|
||
*
|
||
* @author walkor<walkor@workerman.net>
|
||
* @copyright walkor<walkor@workerman.net>
|
||
* @link http://www.workerman.net/
|
||
* @license http://www.opensource.org/licenses/mit-license.php MIT License
|
||
*/
|
||
use Workerman\Protocols\Http;
|
||
/**
|
||
* 批量请求
|
||
* @param array $request_buffer_array ['ip:port'=>req_buf, 'ip:port'=>req_buf, ...]
|
||
* @return multitype:unknown string
|
||
*/
|
||
function multiRequest($request_buffer_array)
|
||
{
|
||
\Statistics\Lib\Cache::$lastSuccessIpArray = array();
|
||
$client_array = $sock_to_ip = $ip_list = array();
|
||
foreach($request_buffer_array as $address => $buffer)
|
||
{
|
||
list($ip, $port) = explode(':', $address);
|
||
$ip_list[$ip] = $ip;
|
||
$client = stream_socket_client("tcp://$address", $errno, $errmsg, 1);
|
||
if(!$client)
|
||
{
|
||
continue;
|
||
}
|
||
$client_array[$address] = $client;
|
||
stream_set_timeout($client_array[$address], 0, 100000);
|
||
fwrite($client_array[$address], $buffer);
|
||
stream_set_blocking($client_array[$address], 0);
|
||
$sock_to_address[(int)$client] = $address;
|
||
}
|
||
$read = $client_array;
|
||
$write = $except = $read_buffer = array();
|
||
$time_start = microtime(true);
|
||
$timeout = 0.99;
|
||
// 轮询处理数据
|
||
while(count($read) > 0)
|
||
{
|
||
if(@stream_select($read, $write, $except, 0, 200000))
|
||
{
|
||
foreach($read as $socket)
|
||
{
|
||
$address = $sock_to_address[(int)$socket];
|
||
$buf = fread($socket, 8192);
|
||
if(!$buf)
|
||
{
|
||
if(feof($socket))
|
||
{
|
||
unset($client_array[$address]);
|
||
}
|
||
continue;
|
||
}
|
||
if(!isset($read_buffer[$address]))
|
||
{
|
||
$read_buffer[$address] = $buf;
|
||
}
|
||
else
|
||
{
|
||
$read_buffer[$address] .= $buf;
|
||
}
|
||
// 数据接收完毕
|
||
if(($len = strlen($read_buffer[$address])) && $read_buffer[$address][$len-1] === "\n")
|
||
{
|
||
unset($client_array[$address]);
|
||
}
|
||
}
|
||
}
|
||
// 超时了
|
||
if(microtime(true) - $time_start > $timeout)
|
||
{
|
||
break;
|
||
}
|
||
$read = $client_array;
|
||
}
|
||
|
||
foreach($read_buffer as $address => $buf)
|
||
{
|
||
list($ip, $port) = explode(':', $address);
|
||
\Statistics\Lib\Cache::$lastSuccessIpArray[$ip] = $ip;
|
||
}
|
||
|
||
\Statistics\Lib\Cache::$lastFailedIpArray = array_diff($ip_list, \Statistics\Lib\Cache::$lastSuccessIpArray);
|
||
|
||
ksort($read_buffer);
|
||
|
||
return $read_buffer;
|
||
}
|
||
|
||
/**
|
||
* 检查是否登录
|
||
*/
|
||
function check_auth()
|
||
{
|
||
// 如果配置中管理员用户名密码为空则说明不用验证
|
||
if(Statistics\Config::$adminName == '' && Statistics\Config::$adminPassword == '')
|
||
{
|
||
return true;
|
||
}
|
||
// 进入验证流程
|
||
_session_start();
|
||
if(!isset($_SESSION['admin']))
|
||
{
|
||
if(!isset($_POST['admin_name']) || !isset($_POST['admin_password']))
|
||
{
|
||
include ST_ROOT . '/Views/login.tpl.php';
|
||
_exit();
|
||
}
|
||
else
|
||
{
|
||
$admin_name = $_POST['admin_name'];
|
||
$admin_password = $_POST['admin_password'];
|
||
if($admin_name != Statistics\Config::$adminName || $admin_password != Statistics\Config::$adminPassword)
|
||
{
|
||
$msg = "用户名或者密码不正确";
|
||
include ST_ROOT . '/Views/login.tpl.php';
|
||
_exit();
|
||
}
|
||
$_SESSION['admin'] = $admin_name;
|
||
}
|
||
}
|
||
return true;
|
||
}
|
||
|
||
/**
|
||
* 启动session，兼容fpm
|
||
*/
|
||
function _session_start()
|
||
{
|
||
if(PHP_SAPI === 'cli')
|
||
{
|
||
return Http::sessionStart();
|
||
}
|
||
return session_start();
|
||
}
|
||
|
||
/**
|
||
* 退出
|
||
* @param string $str
|
||
*/
|
||
function _exit($str = '')
|
||
{
|
||
if(PHP_SAPI === 'cli')
|
||
{
|
||
return Http::end($str);
|
||
}
|
||
return exit($str);
|
||
}
|