php + h5 實現socket推送技術

在socket出現以前已經有ajax定時請求、長輪詢等方案，但都不能知足需求，socket就應用而生了。

socket基本函數socket

總結下經常使用的socket函數

服務端： socket_create 建立socket設置基本參數 

　　　　 socket_bind 綁定ip和端口號   

　　　　 socket_listen 監聽

　　　　 socket_accept 客戶端的鏈接

　　　　 socket_read 讀取客戶端的數據

　　　　 socket_write 給單獨客戶端發送數據 

　　　　 socket_close 關閉鏈接

客戶端：socket_create 建立socket設置基本參數 

　　　　 socket_connect 鏈接socket

　　　　 socket_write 給服務端發送數據

　　　　 socket_read 讀取服務端數據

　　　　 socket_close 關閉鏈接

H5websocket很少說了，上連接

OK，開始貼代碼~

----------------------------------------------------------分割線

服務端代碼：

<?php
class WS {
    var $master;
    var $sockets = array();
    var $debug = false;//true爲調試模式，輸出log日誌
    var $handshake = array();

    function __construct($address, $port){
        $this->master=socket_create(AF_INET, SOCK_STREAM, SOL_TCP)     or die("socket_create() failed");
        socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1)  or die("socket_option() failed");
        socket_bind($this->master, $address, $port)                    or die("socket_bind() failed");
        socket_listen($this->master,20)                                or die("socket_listen() failed");
        
        $this->sockets[] = $this->master;
        $this->say("Server Started : ".date('Y-m-d H:i:s'));
        $this->say("Listening on   : ".$address." port ".$port);
        $this->say("Master socket  : ".$this->master."\n");
        
        while(true){
            $socketArr = $this->sockets;
            $write = NULL;
            $except = NULL;
            socket_select($socketArr, $write, $except, NULL);  //自動選擇來消息的socket 若是是握手 自動選擇主機
            foreach ($socketArr as $socket){
                if ($socket == $this->master){  //主機
                    $client = socket_accept($this->master);
                    if ($client < 0){
                        $this->log("socket_accept() failed");
                        continue;
                    } else{
                        $this->connect($client);
                    }
                } else {
                    $bytes = @socket_recv($socket,$buffer,2048,0);
                    if ($bytes == 0){
                        $this->disConnect($socket);
                    }
                    else{
                        $key = array_search($socket, $this->sockets);
                        if (empty($this->handshake) || !isset($this->handshake[$key]) || !$this->handshake[$key]){
                            $this->doHandShake($socket, $buffer, $key);
                        }
                        else{
                            $buffer = $this->decode($buffer);
                            echo $buffer.PHP_EOL;
                            $key = array_search($socket, $this->sockets);
                            $arr = $this->sockets;
                            array_shift($arr);
                            foreach ($arr as $s){
                                $this->send($s, $buffer);
                            }
                        }
                    }
                }
            }
        }
    }
    
    function send($client, $msg){
        $msg = $this->frame($msg);
        socket_write($client, $msg, strlen($msg));
    }
    function connect($socket){
        array_push($this->sockets, $socket);
        $this->say("\n" . $socket . " CONNECTED!");
        $this->say(date("Y-n-d H:i:s"));
    }
    function disConnect($socket){
        $index = array_search($socket, $this->sockets);
        socket_close($socket);
        $this->say($socket . " DISCONNECTED!");
        if ($index >= 0){
            echo 'unset index is:'.PHP_EOL;
            unset($this->sockets[$index]);
        }
    }
    function doHandShake($socket, $buffer, $handKey){
        $this->log("\nRequesting handshake...");
        $this->log($buffer);
        list($resource, $host, $origin, $key) = $this->getHeaders($buffer);
        $this->log("Handshaking...");
        $upgrade  = "HTTP/1.1 101 Switching Protocol\r\n" .
                    "Upgrade: websocket\r\n" .
                    "Connection: Upgrade\r\n" .
                    "Sec-WebSocket-Accept: " . $this->calcKey($key) . "\r\n\r\n";  //必須以兩個回車結尾
        $this->log($upgrade);
        $sent = socket_write($socket, $upgrade, strlen($upgrade));
        $this->handshake[$handKey]=true;
        $this->log("Done handshaking...");
        return true;
    }

    function getHeaders($req){
        $r = $h = $o = $key = null;
        if (preg_match("/GET (.*) HTTP/"              ,$req,$match)) { $r = $match[1]; }
        if (preg_match("/Host: (.*)\r\n/"             ,$req,$match)) { $h = $match[1]; }
        if (preg_match("/Origin: (.*)\r\n/"           ,$req,$match)) { $o = $match[1]; }
        if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/",$req,$match)) { $key = $match[1]; }
        return array($r, $h, $o, $key);
    }

    function calcKey($key){
        //基於websocket version 13
        $accept = base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));
        return $accept;
    }

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

    function frame($s){
        $a = str_split($s, 125);
        if (count($a) == 1){
            return "\x81" . chr(strlen($a[0])) . $a[0];
        }
        $ns = "";
        foreach ($a as $o){
            $ns .= "\x81" . chr(strlen($o)) . $o;
        }
        return $ns;
    }

    
    function say($msg = ""){
        echo $msg . "\n";
    }
    function log($msg = ""){
        if ($this->debug){
            echo $msg . "\n";
        } 
    }
}
    

new WS('localhost', 4000);

客戶端代碼（H5）：

<html>
    <head>
        <title>demo</title>
        <script src="https://cdn.bootcss.com/jquery/1.9.1/jquery.min.js"></script>
    </head>
    <body>
    <input type="text" id="content">
    <input type="button" value="send" id="send">
        <script type="text/javascript">
            var ws = new WebSocket("ws://localhost:4000");
            ws.onopen = function(){
                console.log("握手成功");
            }
            ws.onmessage = function(e){
                console.log("message:" + e.data);
            }
            ws.onerror = function(){
                console.log("error");
            }
            $("#send").click(function(){
                content = $("#content").val();
                console.log(content);
                ws.send(content);
            })
        </script>
    </body>
</html>

而後執行php demo.php 開啓socket（從運維那偷學一招，linux下執行nohup php demo.php &能夠在後臺執行），瀏覽器打開多個index.html，就能創建通信了。

代碼解析：

1.屬性$sockets數組保存每一個accept鏈接（不知道這麼描述對不對）; 

2.屬性$handshake數組保存鏈接是否在鏈接狀態;