Websocket ---PHP

//封装类,嗯,网上找的,链接地址在哪也忘了


class WebSocketClient {

    private $_host;
    private $_port;
    private $_path;
    private $_origin;
    private $_Socket = null;
    private $_connected = false;

    public function __construct() {
        
    }

    public function __destruct() {
        $this->disconnect();
    }

    public function sendData2($data, $type = 'text', $masked = true) {
        if ($this->_connected === false) {
            trigger_error("Not connected", E_USER_WARNING);
            return false;
        }

        if (!is_string($data)) {
            trigger_error("Not a string data was given.", E_USER_WARNING);
            return false;
        }
        if (strlen($data) == 0) {
            return false;
        }
        $res = @fwrite($this->_Socket, $this->_hybi10Encode($data, $type, $masked));

        if ($res === 0 || $res === false) {
            return false;
        }
        $buffer = ' ';
        while (true) {
            $buffer = fread($this->_Socket, 512);
            preg_replace('# #', '', $buffer);
            break;
        }
        return true;
    }

    public function sendData($data2, $type = 'text', $masked = true) {

        $data = json_encode(array("TYPE" => 1, "ID" => "SERVER"));
        if ($this->_connected === false) {
            trigger_error("Not connected", E_USER_WARNING);
            return false;
        }

        if (!is_string($data)) {
            trigger_error("Not a string data was given.", E_USER_WARNING);
            return false;
        }
        if (strlen($data) == 0) {
            return false;
        }
        $res = @fwrite($this->_Socket, $this->_hybi10Encode($data, $type, $masked));

        if ($res === 0 || $res === false) {
            return false;
        }
        $buffer = ' ';
        while (true) {
            $buffer = fread($this->_Socket, 512);
            preg_replace('# #', '', $buffer);
            //创建SERVER用户,且在链接时期发送消息
            if (!empty($buffer)) {
                $this->sendData2($data2);
                break;
            }
        }
        return true;
    }

    public function connect($host, $port, $path, $origin = false) {
        $this->_host = $host;
        $this->_port = $port;
        $this->_path = $path;
        $this->_origin = $origin;

        $key = base64_encode($this->_generateRandomString(16, false, true));
        $header = "GET " . $path . " HTTP/1.1\r\n";
        $header.= "Host: " . $host . ":" . $port . "\r\n";
        $header.= "Upgrade: websocket\r\n";
        $header.= "Connection: Upgrade\r\n";
        //$header.= "Sec-WebSocket-Extensions: permessage-deflate; client_max_window_bits\r\n";
        $header.= "Sec-WebSocket-Key: " . $key . "\r\n";

        if ($origin !== false) {
            $header.= "Sec-WebSocket-Origin: " . $origin . "\r\n";
        }
        $header.= "Sec-WebSocket-Version: 13\r\n\r\n";

        $this->_Socket = fsockopen($host, $port, $errno, $errstr, 2);
        socket_set_timeout($this->_Socket, 2, 10000);
        //socket_write($this->_Socket, $header);
        $res = @fwrite($this->_Socket, $header);
        if ($res === false) {
            echo "fwrite false \n";
        }

        $response = @fread($this->_Socket, 1500);
        //$response = socket_read($this->_Socket);
        preg_match('#Sec-WebSocket-Accept:\s(.*)$#mU', $response, $matches);
        if ($matches) {
            $keyAccept = trim($matches[1]);
            $expectedResonse = base64_encode(pack('H*', sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11')));
            $this->_connected = ($keyAccept === $expectedResonse) ? true : false;
        }
        return $this->_connected;
    }

    public function checkConnection() {
        $this->_connected = false;

        // send ping:
        $data = 'ping?';
        @fwrite($this->_Socket, $this->_hybi10Encode($data, 'ping', true));
        $response = @fread($this->_Socket, 300);
        if (empty($response)) {
            return false;
        }
        $response = $this->_hybi10Decode($response);
        if (!is_array($response)) {
            return false;
        }
        if (!isset($response['type']) || $response['type'] !== 'pong') {
            return false;
        }
        $this->_connected = true;
        return true;
    }

    public function disconnect() {
        $this->_connected = false;
        is_resource($this->_Socket) and fclose($this->_Socket);
    }

    public function reconnect() {
        sleep(10);
        $this->_connected = false;
        fclose($this->_Socket);
        $this->connect($this->_host, $this->_port, $this->_path, $this->_origin);
    }

    private function _generateRandomString($length = 10, $addSpaces = true, $addNumbers = true) {
        $characters = 'abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ!"ยง$%&/()=[]{}';
        $useChars = array();
        // select some random chars:    
        for ($i = 0; $i < $length; $i++) {
            $useChars[] = $characters[mt_rand(0, strlen($characters) - 1)];
        }
        // add spaces and numbers:
        if ($addSpaces === true) {
            array_push($useChars, ' ', ' ', ' ', ' ', ' ', ' ');
        }
        if ($addNumbers === true) {
            array_push($useChars, rand(0, 9), rand(0, 9), rand(0, 9));
        }
        shuffle($useChars);
        $randomString = trim(implode('', $useChars));
        $randomString = substr($randomString, 0, $length);
        return $randomString;
    }

    private function _hybi10Encode($payload, $type = 'text', $masked = true) {
        $frameHead = array();
        $frame = '';
        $payloadLength = strlen($payload);

        switch ($type) {
            case 'text':
                // first byte indicates FIN, Text-Frame (10000001):
                $frameHead[0] = 129;
                break;

            case 'close':
                // first byte indicates FIN, Close Frame(10001000):
                $frameHead[0] = 136;
                break;

            case 'ping':
                // first byte indicates FIN, Ping frame (10001001):
                $frameHead[0] = 137;
                break;

            case 'pong':
                // first byte indicates FIN, Pong frame (10001010):
                $frameHead[0] = 138;
                break;
        }

        // set mask and payload length (using 1, 3 or 9 bytes)
        if ($payloadLength > 65535) {
            $payloadLengthBin = str_split(sprintf('%064b', $payloadLength), 8);
            $frameHead[1] = ($masked === true) ? 255 : 127;
            for ($i = 0; $i < 8; $i++) {
                $frameHead[$i + 2] = bindec($payloadLengthBin[$i]);
            }
            // most significant bit MUST be 0 (close connection if frame too big)
            if ($frameHead[2] > 127) {
                $this->close(1004);
                return false;
            }
        } elseif ($payloadLength > 125) {
            $payloadLengthBin = str_split(sprintf('%016b', $payloadLength), 8);
            $frameHead[1] = ($masked === true) ? 254 : 126;
            $frameHead[2] = bindec($payloadLengthBin[0]);
            $frameHead[3] = bindec($payloadLengthBin[1]);
        } else {
            $frameHead[1] = ($masked === true) ? $payloadLength + 128 : $payloadLength;
        }

        // convert frame-head to string:
        foreach (array_keys($frameHead) as $i) {
            $frameHead[$i] = chr($frameHead[$i]);
        }
        if ($masked === true) {
            // generate a random mask:
            $mask = array();
            for ($i = 0; $i < 4; $i++) {
                $mask[$i] = chr(rand(0, 255));
            }

            $frameHead = array_merge($frameHead, $mask);
        }
        $frame = implode('', $frameHead);

        // append payload to frame:
        $framePayload = array();
        for ($i = 0; $i < $payloadLength; $i++) {
            $frame .= ($masked === true) ? $payload[$i] ^ $mask[$i % 4] : $payload[$i];
        }

        return $frame;
    }

    private function _hybi10Decode($data) {
        $payloadLength = '';
        $mask = '';
        $unmaskedPayload = '';
        $decodedData = array();

        // estimate frame type:
        $firstByteBinary = sprintf('%08b', ord($data[0]));
        $secondByteBinary = sprintf('%08b', ord($data[1]));
        $opcode = bindec(substr($firstByteBinary, 4, 4));
        $isMasked = ($secondByteBinary[0] == '1') ? true : false;
        $payloadLength = ord($data[1]) & 127;

        switch ($opcode) {
            // text frame:
            case 1:
                $decodedData['type'] = 'text';
                break;

            case 2:
                $decodedData['type'] = 'binary';
                break;

            // connection close frame:
            case 8:
                $decodedData['type'] = 'close';
                break;

            // ping frame:
            case 9:
                $decodedData['type'] = 'ping';
                break;

            // pong frame:
            case 10:
                $decodedData['type'] = 'pong';
                break;

            default:
                return false;
                break;
        }

        if ($payloadLength === 126) {
            $mask = substr($data, 4, 4);
            $payloadOffset = 8;
            $dataLength = bindec(sprintf('%08b', ord($data[2])) . sprintf('%08b', ord($data[3]))) + $payloadOffset;
        } elseif ($payloadLength === 127) {
            $mask = substr($data, 10, 4);
            $payloadOffset = 14;
            $tmp = '';
            for ($i = 0; $i < 8; $i++) {
                $tmp .= sprintf('%08b', ord($data[$i + 2]));
            }
            $dataLength = bindec($tmp) + $payloadOffset;
            unset($tmp);
        } else {
            $mask = substr($data, 2, 4);
            $payloadOffset = 6;
            $dataLength = $payloadLength + $payloadOffset;
        }

        if ($isMasked === true) {
            for ($i = $payloadOffset; $i < $dataLength; $i++) {
                $j = $i - $payloadOffset;
                if (isset($data[$i])) {
                    $unmaskedPayload .= $data[$i] ^ $mask[$j % 4];
                }
            }
            $decodedData['payload'] = $unmaskedPayload;
        } else {
            $payloadOffset = $payloadOffset - 4;
            $decodedData['payload'] = substr($data, $payloadOffset);
        }

        return $decodedData;
    }

}

<font size = "+1" color = 'red'>

//这里差多不就是类的引用了

</font>

class WebsocketController extends Controller {

    function __construct() {
        parent::__construct();
    }

    /**
     * 发送信息
     * 业务逻辑
     */
    function web_socket($msg) {
        set_time_limit(0); //等待程序运行完
        $data = $this->check_socket();
        if ($data['state'] == 1) {
            /**
             * 数据传输
             */
            $res = $this->socket_send($data['socket'], $msg);
//            $this->close_socket($data['socket']); //关闭
            $data = $res;
        } else {
            $this->data_log(date("Y-m-d H:i:s") . "  {$data['msg']}");
        }
        return $data;
        // echo json_encode($data);
    }

    /**
     * 发送消息
     */
    function socket_send($socket, $msg) {
        $res['state'] = 1;
        if (!socket_write($socket, $msg, strlen($msg))) {
            $res['state'] = 0;
            $res['msg'] = "消息发送失败";
            $this->data_log(date("Y-m-d H:i:s") . "  " . $res['msg']);
        } else {
//            $this->data_log($msg);
            $res['msg'] = "发送信息成功!发送的内容为: {$msg} ";
        }
//        sleep(1);
        return $res;
    }

    /**
     * 检测socket可用性
     */
    function check_socket($ip = "", $port = "8087") {
        $socket = socket_create(AF_INET, SOCK_STREAM, SOL_TCP); //创建一个socket,返回套接字
        $data["socket"] = $socket;
        $data['state'] = 1;
        if ($socket < 0) {
            $data['state'] = 0;
            $data['msg'] = "无法创建链接";
        } else {
            $result = socket_connect($socket, $ip, $port); //尝试连接  result bool
            if ($result < 0) {
                $data['state'] = 0;
                $data['msg'] = "连接不上该端口 $ip";
            }
        }
        return $data;
    }

    /**
     * 关闭socket链接
     */
    function close_socket($socket) {
        socket_close($socket);
    }

    public function index($msg = '1') {
        $client = new WebSocketClient();
        $client->connect("IP_URL", 8087, '/');
        $payload = json_encode($msg);

        $rs = $client->sendData($payload);
        /**
         * 发送成功,添加记录
         */
        if ($rs) {
            $add = array(
                "addtime"=>NOW_TIME,
                "uptime"=>NOW_TIME,
                "msg"=>$payload,
            );
            M("send_websocket")->add($add);
        }
        return $rs;
//        if ($rs !== true) {
//            echo "sendData error...\n";
//        } else {
//            echo "ok\n";
//        }
    }

}

猜你喜欢

转载自blog.csdn.net/Purgatory001/article/details/81075842
今日推荐