websocket服务端和客户端代码
March 8, 2018

服务端:
<?php
/*
创建类websocket($config);
$config结构:
$config=array(
'address'=>'192.168.0.200',//绑定地址
'port'=>'8000',//绑定端口
'event'=>'WSevent',//回调函数的函数名
'log'=>true,//命令行显示记录
);

回调函数返回数据格式
function WSevent($type,$event)

$type字符串 事件类型有以下三种
in 客户端进入
out 客户端断开
msg 客户端消息到达
均为小写

$event 数组
$event['k']内置用户列表的userid;
$event['sign']客户标示
$event['msg']收到的消息 $type='msg'时才有该信息

方法:
run()运行
search(标示)遍历取得该标示的id
close(标示)断开连接
write(标示,信息)推送信息
idwrite(id,信息)推送信息

属性:
$users 客户列表
结构:
$users=array(
[用户id]=>array('socket'=>[标示],'hand'=[是否握手-布尔值]),
[用户id]=>arr.....
)
*/

class websocket{

public $log;
public $event;
public $signets;
public $users;  
public $master; 
public function __construct($config){
    if (substr(php_sapi_name(), 0, 3) !== 'cli') {
        die("请通过命令行模式运行!");
    }
    error_reporting(E_ALL);
    set_time_limit(0);
    ob_implicit_flush();
    $this->event = $config['event'];
    $this->log = $config['log'];
    $this->master=$this->WebSocket($config['address'], $config['port']);
    $this->sockets=array('s'=>$this->master);
}
function WebSocket($address,$port){
    $server = socket_create(AF_INET, SOCK_STREAM, SOL_TCP);
    socket_set_option($server, SOL_SOCKET, SO_REUSEADDR, 1);
    socket_bind($server, $address, $port);
    socket_listen($server);
    $this->log('开始监听: '.$address.' : '.$port);
    return $server;
}

function run(){

while(true){
  $changes=$this->sockets;
  @socket_select($changes,$write=NULL,$except=NULL,NULL);
  foreach($changes as $sign){
    if($sign==$this->master){
      $client=socket_accept($this->master);
      $this->sockets[]=$client;
      $user = array(
        'socket'=>$client,
        'hand'=>false,
      );
      $this->users[] = $user;
      $k=$this->search($client);
      $eventreturn = array('k'=>$k,'sign'=>$sign);
      $this->eventoutput('in',$eventreturn);
    }else{
      $len=socket_recv($sign,$buffer,2048,0);
      $k=$this->search($sign);
      $user=$this->users[$k];
      if($len<7){
        $this->close($sign);
        $eventreturn = array('k'=>$k,'sign'=>$sign);
        $this->eventoutput('out',$eventreturn);
        continue;
      }
      if(!$this->users[$k]['hand']){//没有握手进行握手
        $this->handshake($k,$buffer);
      }else{
        $buffer = $this->uncode($buffer);
        $eventreturn = array('k'=>$k,'sign'=>$sign,'msg'=>$buffer);
        $this->eventoutput('msg',$this->users);
      }
    }
  }
}

}
function search($sign){//通过标示遍历获取id

foreach ($this->users as $k=>$v){
  if($sign==$v['socket'])
  return $k;
}
return false;

}
function close($sign){//通过标示断开连接

$k=array_search($sign, $this->sockets);
socket_close($sign);
unset($this->sockets[$k]);
unset($this->users[$k]);

}
function handshake($k,$buffer){

$buf  = substr($buffer,strpos($buffer,'Sec-WebSocket-Key:')+18);
$key  = trim(substr($buf,0,strpos($buf,"\r\n")));
$new_key = base64_encode(sha1($key."258EAFA5-E914-47DA-95CA-C5AB0DC85B11",true));
$new_message = "HTTP/1.1 101 Switching Protocols\r\n";
$new_message .= "Upgrade: websocket\r\n";
$new_message .= "Sec-WebSocket-Version: 13\r\n";
$new_message .= "Connection: Upgrade\r\n";
$new_message .= "Sec-WebSocket-Accept: " . $new_key . "\r\n\r\n";
socket_write($this->users[$k]['socket'],$new_message,strlen($new_message));
$this->users[$k]['hand']=true;
return true;

}
function uncode($str){

$mask = array();  
$data = '';  
$msg = unpack('H*',$str);  
$head = substr($msg[1],0,2);  
if (hexdec($head{1}) === 8) {  
  $data = false;  
}else if (hexdec($head{1}) === 1){  
  $mask[] = hexdec(substr($msg[1],4,2));
  $mask[] = hexdec(substr($msg[1],6,2));
  $mask[] = hexdec(substr($msg[1],8,2));
  $mask[] = hexdec(substr($msg[1],10,2));
  $s = 12;  
  $e = strlen($msg[1])-2;  
  $n = 0;  
  for ($i=$s; $i<= $e; $i+= 2) {  
    $data .= chr($mask[$n%4]^hexdec(substr($msg[1],$i,2)));  
    $n++;  
  }  
}  
return $data;

}

function code($msg){
  $msg = preg_replace(array('/\r$/','/\n$/','/\r\n$/',), '', $msg);
  $frame = array();  
  $frame[0] = '81';  
  $len = strlen($msg);  
  $frame[1] = $len<16?'0'.dechex($len):dechex($len);
  $frame[2] = $this->ord_hex($msg);
  $data = implode('',$frame);
  return pack("H*", $data);
}
function ord_hex($data)  {  
  $msg = '';  
  $l = strlen($data);  
  for ($i= 0; $i<$l; $i++) {  
    $msg .= dechex(ord($data{$i}));  
  }  
  return $msg;  
}

function idwrite($id,$t){//通过id推送信息
  if(!$this->users[$id]['socket']){return false;}//没有这个标示
  $t=$this->code($t);
  return socket_write($this->users[$id]['socket'],$t,strlen($t));
}
function write($k,$t){//通过标示推送信息
  $t=$this->code($t);
  return socket_write($k,$t,strlen($t));
}
function eventoutput($type,$event){//事件回调
  call_user_func($this->event,$type,$event);
}
function log($t){//控制台输出
  if($this->log){
    $t=$t."\r\n";
    fwrite(STDOUT, iconv('utf-8','gbk//IGNORE',$t));
  }
}

}

客户端:
<?php
// ini_set('display_errors', 1);
// error_reporting(E_ALL);

/**

  • Very basic websocket client.
  • Supporting draft hybi-10.
  • @author Simon Samtleben <web@lemmingzshadow.net>
  • @version 2011-10-18

*/

class WebSocketClient
{

private $_host;
private $_port;
private $_path;
private $_origin;
private $_Socket = null;
private $_connected = false;

public function __construct() { }

public function __destruct()
{
    $this->disconnect();
}

public function sendData($data, $type = 'text', $masked = true)
{
    if($this->_connected === false)
    {
        trigger_error("Not connected", E_USER_WARNING);
        return false;
    }
    
    if( !is_string($data)) {
        trigger_error("Not a string data was given.", E_USER_WARNING);
        return false;        
    }
    if (strlen($data) == 0)
    {
        return false;
    }
    $res = @fwrite($this->_Socket, $this->_hybi10Encode($data, $type, $masked));
    
    if($res === 0 || $res === false)
    {
        return false;
    }    
    /*    
    $buffer = ' ';
    $str = '';
    while($buffer !== '')
    {            
        $buffer = fread($this->_Socket, 512);
        $str .=$buffer;
    }
    return $str;*/
}

public function connect($host, $port, $path, $origin = false)
{
    $this->_host = $host;
    $this->_port = $port;
    $this->_path = $path;
    $this->_origin = $origin;
    
    $key = base64_encode($this->_generateRandomString(16, false, false));                
    $header = "GET " . $path . " HTTP/1.1\r\n";
    $header.= "Host: ".$host.":".$port. "\r\n";
    $header.= "Upgrade: websocket\r\n";
    $header.= "Connection: Upgrade\r\n";
    //$header.= "Sec-WebSocket-Extensions: permessage-deflate; client_max_window_bits\r\n";
    $header.= "Sec-WebSocket-Key: " . $key . "\r\n";

    if($origin !== false)
    {
        $header.= "Sec-WebSocket-Origin: " . $origin . "\r\n";
    }
    $header.= "Sec-WebSocket-Version: 13\r\n\r\n";
    
    $this->_Socket = fsockopen($host, $port, $errno, $errstr, 2);
    socket_set_timeout($this->_Socket, 2, 10000);
    //socket_write($this->_Socket, $header);
    $res = @fwrite($this->_Socket, $header);
    if( $res === false ){
        echo "fwrite false \n";
    }
    
    $response = @fread($this->_Socket, 1500);
    //$response = socket_read($this->_Socket);
    preg_match('#Sec-WebSocket-Accept:\s(.*)$#mU', $response, $matches);
    if ($matches) {
        $keyAccept = trim($matches[1]);
        $expectedResonse = base64_encode(pack('H*', sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11')));
        $this->_connected = ($keyAccept === $expectedResonse) ? true : false;
    }
    return $this->_connected;
}

public function checkConnection()
{
    $this->_connected = false;
    
    // send ping:
    $data = 'ping?';
    @fwrite($this->_Socket, $this->_hybi10Encode($data, 'ping', true));
    $response = @fread($this->_Socket, 300);
    if(empty($response))
    {            
        return false;
    }
    $response = $this->_hybi10Decode($response);
    if(!is_array($response))
    {            
        return false;
    }
    if(!isset($response['type']) || $response['type'] !== 'pong')
    {            
        return false;
    }
    $this->_connected = true;
    return true;
}


public function disconnect()
{
    $this->_connected = false;
    is_resource($this->_Socket) and fclose($this->_Socket);
}

public function reconnect()
{
    sleep(10);
    $this->_connected = false;
    fclose($this->_Socket);
    $this->connect($this->_host, $this->_port, $this->_path, $this->_origin);        
}

private function _generateRandomString($length = 10, $addSpaces = true, $addNumbers = true)
{  
    $characters = 'abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ!"ยง$%&/()=[]{}';
    $useChars = array();
    // select some random chars:    
    for($i = 0; $i < $length; $i++)
    {
        $useChars[] = $characters[mt_rand(0, strlen($characters)-1)];
    }
    // add spaces and numbers:
    if($addSpaces === true)
    {
        array_push($useChars, ' ', ' ', ' ', ' ', ' ', ' ');
    }
    if($addNumbers === true)
    {
        array_push($useChars, rand(0,9), rand(0,9), rand(0,9));
    }
    shuffle($useChars);
    $randomString = trim(implode('', $useChars));
    $randomString = substr($randomString, 0, $length);
    return $randomString;
}

private function _hybi10Encode($payload, $type = 'text', $masked = true)
{
    $frameHead = array();
    $frame = '';
    $payloadLength = strlen($payload);
    
    switch($type)
    {        
        case 'text':
            // first byte indicates FIN, Text-Frame (10000001):
            $frameHead[0] = 129;                
        break;            
    
        case 'close':
            // first byte indicates FIN, Close Frame(10001000):
            $frameHead[0] = 136;
        break;
    
        case 'ping':
            // first byte indicates FIN, Ping frame (10001001):
            $frameHead[0] = 137;
        break;
    
        case 'pong':
            // first byte indicates FIN, Pong frame (10001010):
            $frameHead[0] = 138;
        break;
    }
    
    // set mask and payload length (using 1, 3 or 9 bytes) 
    if($payloadLength > 65535)
    {
        $payloadLengthBin = str_split(sprintf('%064b', $payloadLength), 8);
        $frameHead[1] = ($masked === true) ? 255 : 127;
        for($i = 0; $i < 8; $i++)
        {
            $frameHead[$i+2] = bindec($payloadLengthBin[$i]);
        }
        // most significant bit MUST be 0 (close connection if frame too big)
        if($frameHead[2] > 127)
        {
            $this->close(1004);
            return false;
        }
    }
    elseif($payloadLength > 125)
    {
        $payloadLengthBin = str_split(sprintf('%016b', $payloadLength), 8);
        $frameHead[1] = ($masked === true) ? 254 : 126;
        $frameHead[2] = bindec($payloadLengthBin[0]);
        $frameHead[3] = bindec($payloadLengthBin[1]);
    }
    else
    {
        $frameHead[1] = ($masked === true) ? $payloadLength + 128 : $payloadLength;
    }

    // convert frame-head to string:
    foreach(array_keys($frameHead) as $i)
    {
        $frameHead[$i] = chr($frameHead[$i]);
    }
    if($masked === true)
    {
        // generate a random mask:
        $mask = array();
        for($i = 0; $i < 4; $i++)
        {
            $mask[$i] = chr(rand(0, 255));
        }
        
        $frameHead = array_merge($frameHead, $mask);            
    }                        
    $frame = implode('', $frameHead);

    // append payload to frame:
    $framePayload = array();    
    for($i = 0; $i < $payloadLength; $i++)
    {        
        $frame .= ($masked === true) ? $payload[$i] ^ $mask[$i % 4] : $payload[$i];
    }

    return $frame;
}

private function _hybi10Decode($data)
{
    $payloadLength = '';
    $mask = '';
    $unmaskedPayload = '';
    $decodedData = array();
    
    // estimate frame type:
    $firstByteBinary = sprintf('%08b', ord($data[0]));        
    $secondByteBinary = sprintf('%08b', ord($data[1]));
    $opcode = bindec(substr($firstByteBinary, 4, 4));
    $isMasked = ($secondByteBinary[0] == '1') ? true : false;
    $payloadLength = ord($data[1]) & 127;        
    
    switch($opcode)
    {
        // text frame:
        case 1:
            $decodedData['type'] = 'text';                
        break;
    
        case 2:
            $decodedData['type'] = 'binary';
        break;
        
        // connection close frame:
        case 8:
            $decodedData['type'] = 'close';
        break;
        
        // ping frame:
        case 9:
            $decodedData['type'] = 'ping';                
        break;
        
        // pong frame:
        case 10:
            $decodedData['type'] = 'pong';
        break;
        
        default:
            return false;
        break;
    }
    
    if($payloadLength === 126)
    {
       $mask = substr($data, 4, 4);
       $payloadOffset = 8;
       $dataLength = bindec(sprintf('%08b', ord($data[2])) . sprintf('%08b', ord($data[3]))) + $payloadOffset;
    }
    elseif($payloadLength === 127)
    {
        $mask = substr($data, 10, 4);
        $payloadOffset = 14;
        $tmp = '';
        for($i = 0; $i < 8; $i++)
        {
            $tmp .= sprintf('%08b', ord($data[$i+2]));
        }
        $dataLength = bindec($tmp) + $payloadOffset;
        unset($tmp);
    }
    else
    {
        $mask = substr($data, 2, 4);    
        $payloadOffset = 6;
        $dataLength = $payloadLength + $payloadOffset;
    }    
    
    if($isMasked === true)
    {
        for($i = $payloadOffset; $i < $dataLength; $i++)
        {
            $j = $i - $payloadOffset;
            if(isset($data[$i]))
            {
                $unmaskedPayload .= $data[$i] ^ $mask[$j % 4];
            }
        }
        $decodedData['payload'] = $unmaskedPayload;
    }
    else
    {
        $payloadOffset = $payloadOffset - 4;
        $decodedData['payload'] = substr($data, $payloadOffset);
    }
    
    return $decodedData;
}

}

Leave a Reply

如果您是第一次在本站留言,需要审核后才会显示内容!