WebSocket 使用PHP实现的类

前面的文章是借鉴的 　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　

<?php
/**
 * @Author: FatKee
 * @Date:   2019-01-09 12:42:28
 * @Last Modified by:   FatKee
 * @Last Modified time: 2019-01-10 16:37:13
 */


class WebSocket{
    //socket池
    private $sockets = [];
    //用户池
    private $users = [];
    //主sock
    private $master = NULL;
    //当前错误
    private $error = '';
    public function __construct($host ,$port){
        $this->master = socket_create(AF_INET,SOCK_STREAM,SOL_TCP  );
        if($this->master === false){
            $this->error = 'socket create error';
            exit;
        }
        if(socket_bind($this->master,$host,$port) === false){
            $this->error = 'socket bind error';
             exit;
        }

        if(socket_listen($this->master) === false){
            $this->error = 'socket listen  error';
             exit;
        }
        $this->sockets[] = $this->master;
        $this->e('start:'.$host.'port:' . $port);


    }
    public function run(){
        while(true){
            $sockets = $this->sockets;
            $write = NULL;
            $except = NULL;
            socket_select($sockets,$write,$except,NULL);

            foreach($sockets as $sock){
                if($sock == $this->master){
                    //等待新用户
                    $k = uniqid();
                    $this->e('user:'. $k . ' 加入用户池');
                    $new_sock = socket_accept($this->master);


                    $this->users[$k] = array('socket' => $new_sock, 'is_hand' => false );
                    $this->sockets[]  = $new_sock;

                }else{
                    //接收消息的或者关闭链接
                    $k = $this->search($sock);
                    if($this->users[$k]['is_hand'] === false){
                        //进行握手
                        $this->handshake($sock);
                        $this->users[$k]['is_hand'] = true;
                        $this->e('user:'. $k . ' 握手成功');
                          continue;
                    }
                    //读取
                    $byte=socket_recv($sock, $buff,20480, 0);
                    if($byte<7){
                        $this->e('user:'. $k . ' 退出');
                        $this->close($k);
                        continue;
                    }

                    //$this->e('user:'. $k . ' 发送消息');
                    //转发消息给别人
                    $this->e($k.'群发:'.$this->decode($buff));
                    $this->qunfa($sock,$this->decode($buff));
                }

            }
        }
    }

    private function qunfa($sock,$str){
        foreach ($this->users as  $v) {
            if($v['socket'] != $sock)
            $this->send($v['socket'],$str);
        }
    }
    //寻找socket的key
    private function search($sock){
        foreach($this->users as $k=>$v){
            if($v['socket'] == $sock){
                return $k;
            }
        }
    }

    private function handshake($sock){
        $c = socket_read($sock,1024);
        preg_match('/Sec-WebSocket-Key:(.*)/um',$c,$str);
        if(isset($str[1])){
        $new_key = base64_encode(sha1(trim($str[1])."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($sock,$new_message,strlen($new_message));
        }
    }
    //关闭链接
    private function close($k){
        socket_close($this->users[$k]['socket']);
        unset($this->users[$k]);
        $this->sockets = [];
        $this->sockets[] = $this->master;
        foreach($this->users as $v){
            $this->sockets[] = $v['socket'];
        }
    }

    public function e($str){

        echo iconv('UTF-8','GB2312',$str) . "\r\n";
    }

    //解析数据
    private function decode($buff){
        $arr['fin'] = ord($buff[0]) >> 7;
        $arr['mask'] = ord($buff[1])>>7;
        $arr['len'] = bindec(substr(decbin(ord($buff[1])),1,7));
        if($arr['len'] == 126){
            $arr['len'] = ord($buff[2]) + ord($buff[3]) ;
            $maskkey = substr($buff,4,4);
            $data = substr($buff,8);
        }
        if($arr['len'] == 127){
            $arr['len'] = ord($buff[2]) + ord($buff[3]) + ord($buff[4])+ ord($buff[5]);
            $maskkey = substr($buff,10,4);
            $data = substr($buff,14);
        }
        if($arr['len']<=125){
            $maskkey = substr($buff,2,4);
            $data = substr($buff,6);
        }
        $a = "";
        for($i = 0 ; $i<strlen($data);$i++){
            $a .= $data[$i] ^ $maskkey[$i%4];
        }
        return $a;
    }
    private function send($sock,$str){
        $bf ="\x81".chr(strlen($str)).$str;
        socket_write($sock,$bf,strlen($bf));
    }
}

$socket = new WebSocket('192.168.2.188',1234);

$socket->run();