php + websocket 聊天室

2013 年 6 月 30 日4,3100

html5的websocket实现了双向通信，折腾了几天弄了个聊天室，分享给大家
源码：http://http://www.zjjv.com///sg.php?fp=0&tag=websocket
demo：http://http://www.zjjv.com///ui/sk.html

代码片段(1)

[全屏查看所有代码]

1.[代码]php服务器端的

跳至

[1]

[全屏预览]

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




	}




}




?>