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