php +html5 websocket 聊天室

1. [代码]php服务器端的    
<?php
error_reporting(E_ALL);
ob_implicit_flush();
 
$sk=new Sock('127.0.0.1',8000);
$sk->run();
class Sock{
    public $sockets;
    public $users;
    public $master;
     
    public function __construct($address, $port){
        $this->master=$this->WebSocket($address, $port);
        $this->sockets=array('s'=>$this->master);
    }
     
     
    function run(){
        while(true){
            $changes=$this->sockets;
            socket_select($changes,$write=NULL,$except=NULL,NULL);
            foreach($changes as $sock){
                if($sock==$this->master){
                    $client=socket_accept($this->master);
                    //$key=uniqid();
                    $this->sockets[]=$client;
                    $this->users[]=array(
                        'socket'=>$client,
                        'shou'=>false
                    );
                }else{
                    $len=socket_recv($sock,$buffer,2048,0);
                    $k=$this->search($sock);
                    if($len<7){
                        $name=$this->users[$k]['ming'];
                        $this->close($sock);
                        $this->send2($name,$k);
                        continue;
                    }
                    if(!$this->users[$k]['shou']){
                        $this->woshou($k,$buffer);
                    }else{
                        $buffer = $this->uncode($buffer);
                        $this->send($k,$buffer);
                    }
                }
            }
             
        }
         
    }
     
    function close($sock){
        $k=array_search($sock, $this->sockets);
        socket_close($sock);
        unset($this->sockets[$k]);
        unset($this->users[$k]);
        $this->e("key:$k close");
    }
     
    function search($sock){
        foreach ($this->users as $k=>$v){
            if($sock==$v['socket'])
            return $k;
        }
        return false;
    }
     
    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->e('Server Started : '.date('Y-m-d H:i:s'));
        $this->e('Listening on   : '.$address.' port '.$port);
        return $server;
    }
     
     
    function woshou($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]['shou']=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++;  http://www.enterdesk.com/special/shouhui/
            }  
        }  
        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 send($k,$msg){
        /*$this->send1($k,$this->code($msg),'all');*/
        parse_str($msg,$g);
        $this->e($msg);
        $ar=array();
        if($g['type']=='add'){
            $this->users[$k]['ming']=$g['ming'];
            $ar['add']=true;
            $ar['nrong']='欢迎'.$g['ming'].'加入!';
            $ar['users']=$this->getusers();
            $key='all';
        }else if($g['type']=='ltiao'){
            $ar['nrong']=$g['nr'];
            $key=$g['key'];
        }
        $msg=json_encode($ar);
        $this->e($msg);
        $msg = $this->code($msg);
        $this->send1($k,$msg,$key);
        //socket_write($this->users[$k]['socket'],$msg,strlen($msg));
    }
     
    function getusers(){
        $ar=array();
        foreach($this->users as $k=>$v){
            $ar[$k]=$v['ming'];
        }
        return $ar;
    }
     
    function send1($k,$str,$key='all'){
        if($key=='all'){
            foreach($this->users as $v){
                socket_write($v['socket'],$str,strlen($str));
            }
        }else{
            if($k!=$key)
            socket_write($this->users[$k]['socket'],$str,strlen($str));
            socket_write($this->users[$key]['socket'],$str,strlen($str));
        }
    }
     
    function send2($ming,$k){
        $ar['remove']=true;
        $ar['removekey']=$k;
        $ar['nrong']=$ming.'退出聊天室';
        $str = $this->code(json_encode($ar));
        $this->send1(false,$str,'all');
    }
     
    function e($str){
        $path=dirname(__FILE__).'/log.txt';
        $str=$str."\n";
        error_log($str,3,$path);
        echo iconv('utf-8','gbk//IGNORE',$str);
    }
}
?>

你可能感兴趣的:(websocket)