WebSocket服务器搭建笔记

本文作者：Tianheng Ni  本文分类：前端开发  浏览：297
阅读时间：4718字, 约5-8分钟

网站好冷清啊，我来活跃一下气氛吧～

网上看了一堆资料，踩了一堆坑，终于把websocket服务器搭起来了。

下面和大家分享一下搭建的艰苦经历。

首先，代码部分：

ws.class.php:

<?php
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("Please run with cli!");
        }
        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('Listening: '.$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',$eventreturn);
          }
        }
      }
    }
  }
  function search($sign){
    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){
      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));
      }
    }
}

wss.php:

<?php
include 'ws.class.php';
 
$config=array(
  'address'=>'127.0.0.1',
  'port'=>'32000',//随便填写端口，只要没有被占用即可。注意后面nginx配置文件中也要改成127.0.0.1:你的端口
  'event'=>'WSevent',//回调函数的函数名
  'log'=>true,
);
$websocket = new websocket($config);
$websocket->run();
function WSevent($type,$event){
  global $websocket;
    if('in'==$type){
      $websocket->log('User join id:'.$event['k']);
    }elseif('out'==$type){
      $websocket->log('User exit id:'.$event['k']);
    }elseif('msg'==$type){
      $websocket->log($event['k'].'Message:'.$event['msg']);
      roboot($event['sign'],$event['msg']);
    }
}
 
function roboot($sign,$t){
  global $websocket;
  switch ($t)
  {
  case 'hello':
    $show='hello';
    break;  
  case '用户socket.send()的消息':
    $show='这个消息的回复';
    break;
  default:
    $show='Unknown message.';
  }
  $websocket->write($sign,$show);
}
?>

然后，在终端中运行这个程序。

可能会有很多警告，不过不要在意。

当然，如果要一直运行的话，可以通过systemd创建守护进程，这个不详细说明了。

最后，打开nginx（主）配置文件。

在您想搭建ws的那个server中，添加这么一段：

    location /ws1 {
      proxy_pass http://127.0.0.1:32001;
      proxy_http_version 1.1;
      proxy_set_header Upgrade $http_upgrade;
      proxy_set_header Connection "upgrade";
    }

保存，重启nginx服务。

然后ws服务器就搭好啦。

如果您这台服务器已经配置了SSL，那么就可以通过 wss://你的域名/ws 链接访问。（提示：https网页只能连接wss协议链接哦）

不然的话，只能通过 ws://你的域名/ws 访问。

然后如何在html中使用websocket这个网上教程一大堆，大家自己去看吧/

一个测试网页：https://apis.nth.ink/ws/

关于作者Tianheng Ni

• 卑微站长23564～ 苣蒻OIer，电脑爱好者，喜欢C++编程/折腾网站
• Email: eric_ni2008@163.com
• 注册于: 2020-04-05 07:11:36