php+html5基于websocket实现聊天室的方法

<?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++;

}

}

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);

}

}

?>