websocket的php测试demo

<?php
class WS {
    var $master;
    var $sockets = array();
    var $debug = false;
    var $handshake = false;

    function __construct($address, $port){
        $this->master=socket_create(AF_INET, SOCK_STREAM, SOL_TCP)     or die("socket_create() failed");
        socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1)  or die("socket_option() failed");
        socket_bind($this->master, $address, $port)                    or die("socket_bind() failed");
        socket_listen($this->master,20)                                or die("socket_listen() failed");

        $this->sockets[] = $this->master;
        $this->say("Server Started : ".date('Y-m-d H:i:s'));
        $this->say("Listening on   : ".$address." port ".$port);
        $this->say("Master socket  : ".$this->master."\n");

        while(true){
            $socketArr = $this->sockets;
            $write = NULL;
            $except = NULL;
            socket_select($socketArr, $write, $except, NULL);  //自动选择来消息的socket 如果是握手 自动选择主机
            foreach ($socketArr as $socket){
                if ($socket == $this->master){  //主机
                    $client = socket_accept($this->master);
                    if ($client < 0){
                        $this->log("socket_accept() failed");
                        continue;
                    } else{
                        $this->connect($client);
                    }
                } else {
                    $this->log("^^^^");
                    $bytes = @socket_recv($socket,$buffer,2048,0);
                    $this->log("^^^^");
                    if ($bytes == 0){
                        $this->disConnect($socket);
                    }
                    else{
                        if (!$this->handshake){
                            $this->doHandShake($socket, $buffer);
                        }
                        else{
                            $buffer = $this->decode($buffer);
                            $this->send($socket, $buffer);
                        }
                    }
                }
            }
        }
    }

    function send($client, $msg){
        $this->log("> " . $msg);
        $msg = $this->frame($msg);
        socket_write($client, $msg, strlen($msg));
        $this->log("! " . strlen($msg));
    }
    function connect($socket){
        array_push($this->sockets, $socket);
        $this->say("\n" . $socket . " CONNECTED!");
        $this->say(date("Y-n-d H:i:s"));
    }
    function disConnect($socket){
        $index = array_search($socket, $this->sockets);
        socket_close($socket);
        $this->say($socket . " DISCONNECTED!");
        if ($index >= 0){
            array_splice($this->sockets, $index, 1);
        }
    }
    function doHandShake($socket, $buffer){
        $this->log("\nRequesting handshake...");
        $this->log($buffer);
        list($resource, $host, $origin, $key) = $this->getHeaders($buffer);
        $this->log("Handshaking...");
        $upgrade  = "HTTP/1.1 101 Switching Protocol\r\n" .
                    "Upgrade: websocket\r\n" .
                    "Connection: Upgrade\r\n" .
                    "Sec-WebSocket-Accept: " . $this->calcKey($key) . "\r\n\r\n";  //必须以两个回车结尾
        $this->log($upgrade);
        $sent = socket_write($socket, $upgrade, strlen($upgrade));
        $this->handshake=true;
        $this->log("Done handshaking...");
        return true;
    }

    function getHeaders($req){
        $r = $h = $o = $key = null;
        if (preg_match("/GET (.*) HTTP/"              ,$req,$match)) { $r = $match[1]; }
        if (preg_match("/Host: (.*)\r\n/"             ,$req,$match)) { $h = $match[1]; }
        if (preg_match("/Origin: (.*)\r\n/"           ,$req,$match)) { $o = $match[1]; }
        if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/",$req,$match)) { $key = $match[1]; }
        return array($r, $h, $o, $key);
    }

    function calcKey($key){
        //基于websocket version 13
        $accept = base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));
        return $accept;
    }

    function decode($buffer) {
        $len = $masks = $data = $decoded = null;
        $len = ord($buffer[1]) & 127;

        if ($len === 126) {
            $masks = substr($buffer, 4, 4);
            $data = substr($buffer, 8);
        }
        else if ($len === 127) {
            $masks = substr($buffer, 10, 4);
            $data = substr($buffer, 14);
        }
        else {
            $masks = substr($buffer, 2, 4);
            $data = substr($buffer, 6);
        }
        for ($index = 0; $index < strlen($data); $index++) {
            $decoded .= $data[$index] ^ $masks[$index % 4];
        }
        return $decoded;
    }

    function frame($s){
        $a = str_split($s, 125);
        if (count($a) == 1){
            return "\x81" . chr(strlen($a[0])) . $a[0];
        }
        $ns = "";
        foreach ($a as $o){
            $ns .= "\x81" . chr(strlen($o)) . $o;
        }
        return $ns;
    }

    function say($msg = ""){
        echo $msg . "\n";
    }
    function log($msg = ""){
        if ($this->debug){
            echo $msg . "\n";
        }
    }
}

new WS('localhost', 4000);

<script type="text/javascript">
var ws = new WebSocket("ws://localhost:4000");
ws.onopen = function(){
    console.log("握手成功");
}
ws.onmessage = function(e){
    console.log("message:" + e.data);
}
ws.onerror = function(){
    console.log("error");
}
</script>

demo2:

<!doctype html>
<html>
<head>
    <meta charset="utf-8">
    <title>websocket_TEST</title>
</head>
<body>
<textarea class="log" style="width: 100%; height: 500px;">
=======websocket======
</textarea>
<input type="button" value="连接" onClick="link()">
<input type="button" value="断开" onClick="dis()">
<input type="text" id="text">
<input type="button" value="发送" onClick="send()">
<script type="text/javascript" src="http://apps.bdimg.com/libs/jquery/2.1.4/jquery.min.js"></script>
<script>
    function link(){
        var url='ws://devapp.yaolan.com:8000';
        socket=new WebSocket(url);
        socket.onopen=function(){log('连接成功')}
        socket.onmessage=function(msg){log('获得消息:'+msg.data);console.log(msg);}
        socket.onclose=function(){log('断开连接')}
    }
    function dis(){
        socket.close();
        socket=null;
    }
    function log(var1){
        $('.log').append(var1+"\r\n");
    }
    function send(){
        socket.send($('#text').attr('value'));
    }
    function send2(){
        var json = JSON.stringify({'type':'php','msg':$('#text2').attr('value')})
        socket.send(json);
    }
</script>
</body>
</html>

<?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',$eventreturn);
                    }
                }
            }
        }
    }
    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));
        }
    }
}

<?
include 'websocket.class.php';

$config=array(
    'address'=>'devapp.yaolan.com',
    'port'=>'8000',
    'event'=>'WSevent',//回调函数的函数名
    'log'=>true,
);
$websocket = new websocket($config);
$websocket->run();
function WSevent($type,$event){
    global $websocket;
    if('in'==$type){
        $websocket->log('客户进入id:'.$event['k']);
    }elseif('out'==$type){
        $websocket->log('客户退出id:'.$event['k']);
    }elseif('msg'==$type){
        $websocket->log($event['k'].'消息:'.$event['msg']);
        roboot($event['sign'],$event['msg']);
    }
}

function roboot($sign,$t){
    global $websocket;
    switch ($t)
    {
        case 'hello':
            $show='hello,GIt @ OSC';
            break;
        case 'name':
            $show='Robot';
            break;
        case 'time':
            $show='当前时间:'.date('Y-m-d H:i:s');
            break;
        case '再见':
            $show='( ^_^ )/~~拜拜';
            $websocket->write($sign,'Robot:'.$show);
            $websocket->close($sign);
            return;
            break;
        case '天王盖地虎':
            $array = array('小鸡炖蘑菇','宝塔震河妖','粒粒皆辛苦');
            $show = $array[rand(0,2)];
            break;
        default:
            $show='( ⊙o⊙?)不懂,你可以尝试说:hello,name,time,再见,天王盖地虎.';
    }
    $websocket->write($sign,'Robot:'.$show);
}
?>