论坛 › php › php 原生websocket

123213 发表于 2015-11-23 10:22:02

php 原生websocket

class websocket
{
    public $log;
    public $event;
    public $signets;
    public $users;
    public $master;
    public $obj;
    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->obj = $config['obj'];
      $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, 0, 2);
      if (hexdec($head{1}) === 8) {
            $data = false;
      } else if (hexdec($head{1}) === 1) {
            $mask[] = hexdec(substr($msg, 4, 2));
            $mask[] = hexdec(substr($msg, 6, 2));
            $mask[] = hexdec(substr($msg, 8, 2));
            $mask[] = hexdec(substr($msg, 10, 2));
            $s = 12;
            $e = strlen($msg) - 2;
            $n = 0;
            for ($i = $s; $i <= $e; $i += 2) {
                $data .= chr($mask[$n % 4] ^ hexdec(substr($msg, $i, 2)));
                $n++;
            }
      }
      return $data;
    }
    function code($msg)
    {
      $msg = preg_replace(array('/\r$/', '/\n$/', '/\r\n$/',), '', $msg);
      $frame = array();
      $frame = '81';
      $len = strlen($msg);
      $frame = $len < 16 ? '0' . dechex($len) : dechex($len);
      $frame = $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(array($this->obj, $this->event), $this, $type, $event);
    }
    function log($t)
    {//控制台输出
      if ($this->log) {
            $t = $t . "\r\n";
            fwrite(STDOUT, $t);
      }
    }
}

server.php


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
<?php
include('websocket.class.php');

$config=array(
'address'=>'192.168.1.12',
'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;
    break;
default:
    $show='( ⊙o⊙?)不懂,你可以尝试说:hello,name,time,再见,天王盖地虎.';
}
$websocket->write($sign,'Robot:'.$show);
}
?>




在终端 php server.php


index.html


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
<!doctype html>
<html>
<head>
<meta charset="utf-8">
<title>websocket_TEST</title>
</head>
<body>

<a href="index2.html">index2</a>
<textarea class="log" style="width: 100%; height: 500px;">

</textarea>
<input type="button" value="连接" onClick="link()">
<input type="button" value="断开" onClick="dis()">
<input type="text" id="text2">
<input type="button" value="发送" onClick="send2()">
<script src="http://jt.hiooy.com/hsshop/js/jquery.js?v=59"></script>
<script>
function link(){
var url='ws://192.168.1.12: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