天天看点

mysql实现websocket_socket编程之websocket实现

class SocketService

{

public $host="tcp://0.0.0.0:8000";

private $address;

private $port;

private $_sockets;

public $clients;

public $maxid=1000;

public function __construct($address = '', $port='')

{

if(!empty($address)){

$this->address = $address;

}

if(!empty($port)) {

$this->port = $port;

}

}

public function onConnect($client_id){

echo "Client client_id:{$client_id}\n";

}

public function onMessage($client_id,$msg){

//发给所有的 foreach($this->clients as $kk=>$cc){

if($kk>0){

$this->send($cc, $msg);

}

}

}

public function onClose($client_id){

echo "$client_idclose\n";

}

public function service(){

//获取tcp协议号码。 $tcp = getprotobyname("tcp");

$sock = stream_socket_server($this->host, $errno, $errstr);;

if(!$sock)

{

throw new Exception("failed to create socket: ".socket_strerror($sock)."\n");

}

stream_set_blocking($sock,0);

$this->_sockets = $sock;

echo "listen on$this->address$this->host...\n";

}

public function run(){

$this->service();

$this->clients[] = $this->_sockets;

while (true){

$changes = $this->clients;

//$write = NULL; //$except = NULL; stream_select($changes, $write, $except, NULL);

foreach ($changes as $key => $_sock){

if($this->_sockets == $_sock){ //判断是不是新接入的socket if(($newClient = stream_socket_accept($_sock)) === false){

unset($this->clients[$key]);

continue;

}

$line = trim(stream_socket_recvfrom($newClient, 1024));

//握手 $this->handshaking($newClient, $line);

$this->maxid++;

$this->clients[$this->maxid] = $newClient;

$this->onConnect($this->maxid);

} else {

[email protected]_socket_recvfrom($_sock, 2048);

//客户端主动关闭 if(strlen($res) < 9) {

stream_socket_shutdown($this->clients[$key],STREAM_SHUT_RDWR);

unset($this->clients[$key]);

$this->onClose($key);

}else{

//解密 $msg = $this->decode($res);

$this->onMessage($key,$msg);

}

}

}

}

}

public function handshaking($newClient, $line){

$headers = array();

$lines = preg_split("/\r\n/", $line);

foreach($lines as $line)

{

$line = chop($line);

if(preg_match('/\A(\S+): (.*)\z/', $line, $matches))

{

$headers[$matches[1]] = $matches[2];

}

}

$secKey = $headers['Sec-WebSocket-Key'];

$secAccept = base64_encode(pack('H*', sha1($secKey . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11')));

$upgrade = "HTTP/1.1 101 Web Socket Protocol Handshake\r\n" .

"Upgrade: websocket\r\n" .

"Connection: Upgrade\r\n" .

"WebSocket-Origin:$this->address\r\n" .

"WebSocket-Location: ws://$this->address:$this->port/websocket/websocket\r\n".

"Sec-WebSocket-Accept:$secAccept\r\n\r\n";

return stream_socket_sendto($newClient, $upgrade);

}

public function send($newClinet, $msg){

$msg = $this->encode($msg);

stream_socket_sendto($newClinet, $msg);

}

public function decode($buffer){

$len = $masks = $data = $decoded = null;

$len = ord($buffer[1]) & 127;

if ($len === 126) {

$masks = substr($buffer, 4, 4);

$data = substr($buffer, 8);

} else if ($len === 127) {

$masks = substr($buffer, 10, 4);

$data = substr($buffer, 14);

} else {

$masks = substr($buffer, 2, 4);

$data = substr($buffer, 6);

}

for ($index = 0; $index < strlen($data); $index++) {

$decoded .= $data[$index] ^ $masks[$index % 4];

}

return $decoded;

}

public function encode($buffer) {

$first_byte="\x81";

$len=strlen($buffer);

if ($len <= 125) {

$encode_buffer = $first_byte . chr($len) . $buffer;

} else {

if ($len <= 65535) {

$encode_buffer = $first_byte . chr(126) . pack("n", $len) . $buffer;

} else {

$encode_buffer = $first_byte . chr(127) . pack("xxxxN", $len) . $buffer;

}

}

return $encode_buffer;

}

public function close(){

return socket_close($this->_sockets);

}

}

$sock = new SocketService('127.0.0.1','9000');

$sock->run();