首页 / 网页编程 / PHP / php+html5基于websocket实现聊天室的方法

<?phperror_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,"
"）））;$new_key = base64_encode（sha1（$key."258EAFA5-E914-47DA-95CA-C5AB0DC85B11",true））;$new_message = "HTTP/1.1 101 Switching Protocols
";$new_message .= "Upgrade: websocket
";$new_message .= "Sec-WebSocket-Version: 13
";$new_message .= "Connection: Upgrade
";$new_message .= "Sec-WebSocket-Accept: " . $new_key . "

";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（"/
$/","/
$/","/
$/",）, "", $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."
";error_log（$str,3,$path）;echo iconv（"utf-8","gbk//IGNORE",$str）; }}?>