PHP 简单实现webSocket

admin · 发表于 2018-10-27 12:35:15

1）客户端实现

<html>6 b+ J$ @! @& p! j) y) p; \5 |
<head>
* f% b! z! x0 q1 L& x: S
<meta charset="UTF-8">
8 n# b* P9 A. N. K- }5 T1 j) ]0 E
<title>Web sockets test</title>
1 j& s9 a5 m; M9 W$ j( Q
<script src="jquery-min.js" type="text/javascript"></script>& v. i# [( m6 u4 L* h( {9 ?
<script type="text/javascript">! j' S a; m7 O8 q2 d
var ws;
( K9 O/ C i& u" A
function ToggleConnectionClicked() { , n0 L# Z% h4 |3 V* U
try {. ~" t, ~! \0 ?& p7 G
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器
+ u8 F: J, Z/ G( {, K& j
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};3 _& @" w; Y* `, H0 @% Y, E
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);};
1 K% C! P/ f: v4 r
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);};
. R7 ?4 H+ Y% B; t+ u* r. ^
ws.onerror = function(event){alert("WebSocket异常！");};. e8 F5 \/ N2 b9 N; p' e5 @9 o5 n. M
} catch (ex) {+ X' d: n) ~- D
alert(ex.message);
9 ?- P: m; W. |5 }( p" {! z
}7 A$ D9 c# T% {
};, q1 |# H! Z" n5 }. {
7 Q/ p: S2 i' k
function SendData() {5 }0 m% y5 \# d4 F+ {
try{9 E3 h( _: H3 r# N
var content = document.getElementById("content").value;$ }/ a6 _- } Z3 g/ R
if(content){
% K5 q( w( g% y
ws.send(content);+ ?8 k) w! i9 t" z6 ^* ~
}
* P+ j8 g( r2 ^' _/ \+ F
$ q6 R; Q" H9 T! W% V
}catch(ex){
5 @& B+ W. w' s) w, P
alert(ex.message);8 o+ o, y$ G2 i8 q* B8 v; H
}
7 m. N% c H5 L6 c4 u
};
# J2 m- F9 j! K4 q' H& w
( k: O+ w' y( Y9 k
function seestate(){
1 Y0 C; p8 Y* ]) b
alert(ws.readyState); r6 B' U5 ^( T+ ~! J% V
}
( m- Q- |2 d+ O0 K" A
</script>0 b. U t5 c0 y, _( F" {
</head># _6 V: k, z N# c3 J
<body>; Z# w4 u3 b; e2 G" ^
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br />
6 d) }$ M7 X7 T& {1 c' C4 V$ T
<textarea id="content" ></textarea>- h- d8 d/ T& l
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br />: N1 e& }8 S4 p/ O; @) g7 B" u
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br />
. Q) v6 b: @/ b" w M& ^
</body>& g0 ^# b5 J0 {" g6 [9 v6 _0 T" b
</html>
4 X. T8 o9 Z; n2 H! f

复制代码

2）服务器端实现

class WS {' a7 h7 Y; }: z
var $master; // 连接 server 的 client0 }. Y' y$ X' y: o n6 z: f
var $sockets = array(); // 不同状态的 socket 管理2 h% {' F. x' P. T5 }
var $handshake = false; // 判断是否握手
@2 Q, t1 u4 L
5 f# o4 @: o8 ]$ D2 S* i$ T+ s4 K
function __construct($address, $port){9 @( E. O/ @6 h- T
// 建立一个 socket 套接字
; c9 m: W/ R( A; Z) J6 _4 R
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP) ! q- `+ q t" u7 Q" y
or die("socket_create() failed");
5 ~8 e, u. m6 X2 g% Y
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1)
7 e# c. y: W7 w8 Z: f5 I% M
or die("socket_option() failed");
$ u) @! M. N' S5 Q- q# ^
socket_bind($this->master, $address, $port) 8 ? e& R! q1 \3 B) Y E# Y
or die("socket_bind() failed");
# L, F" {8 N' h: F3 ~1 a
socket_listen($this->master, 2)
4 c5 U. R) u4 a- o3 z3 @6 E% j a
or die("socket_listen() failed");$ y3 c* v) O+ O. a3 f- p% \
" l8 U5 o0 ?4 b- c
$this->sockets[] = $this->master;0 a3 r0 T! f: g+ ~
4 q E- ?, {4 U+ h9 i
// debug$ j: W/ B8 |9 C* ~$ K& n9 M
echo("Master socket : ".$this->master."\n");
, F5 o! T! _3 b% d8 h2 N7 M
& u/ ~4 l9 {" x0 U; n. o
while(true) {
U# k! d3 `0 U7 a0 C+ @/ i& U" _
//自动选择来消息的 socket 如果是握手自动选择主机
8 K8 }, ]- J& \( i, o
$write = NULL;
0 u5 S" d! h2 M! g
$except = NULL;
, J$ k' q1 ^: J) d, _: n
socket_select($this->sockets, $write, $except, NULL);. D# J1 z3 N5 {# \$ M8 U
$ D2 Y, B3 P7 _
foreach ($this->sockets as $socket) {. D5 @9 v- e8 H0 w8 A* D
//连接主机的 client , M% c/ {: }+ q* ? x* j0 g
if ($socket == $this->master){1 S. L" ]. |! A9 N: ^9 V: Q$ V
$client = socket_accept($this->master);
. ~. H% q; h& Q9 y( [5 j
if ($client < 0) {! z& }" V+ d1 d8 `. w
// debug
& Z; Q7 R- S$ ~) @' T5 t
echo "socket_accept() failed";
9 d7 f# N3 G; o Z
continue;
3 ]' F* k) }% L8 u' G& A
} else {
4 X5 R( w$ [# ~
//connect($client);
( I1 ` y8 H5 }7 |
array_push($this->sockets, $client);: q7 ~6 E% F# P+ n- F) \% s% {
echo "connect client\n";
* D C: B: w0 T% v! ^3 _6 R) t
}
! g: U1 |' c+ @- k! K7 ~
} else {
' A5 w" Q- y0 o. j! Z% }
$bytes = @socket_recv($socket,$buffer,2048,0);4 n0 c j" }& o
print_r($buffer);
& ?* P Q9 |& y' D' r! O
if($bytes == 0) return;7 ?) B& h8 K: f; K w4 l4 q) H2 e6 r
if (!$this->handshake) {
& b& V* l& V% b0 ^! d
// 如果没有握手，先握手回应
. M3 j# a. s# x' H: d1 P- a4 Y1 ]# U
$this->doHandShake($socket, $buffer);( u, Q2 v# G3 B; A4 V
echo "shakeHands\n";! [1 t# K) a: ^" I) A- e
} else {0 H+ A' v0 e$ m0 ]4 F# Q0 Q
% v# f7 \6 r0 \- Q1 m/ G2 F
// 如果已经握手，直接接受数据，并处理4 D: n. X% R r0 `8 D
$buffer = $this->decode($buffer);; S/ C3 R5 j+ [1 s2 ^% a- R. H
//process($socket, $buffer);
T D5 F9 }1 j w; o3 m$ k
echo "send file\n";% E; j7 S7 `, |3 ~
}0 Q2 D5 F$ o/ r0 h! c
}
* U9 E1 Q% ~" k3 o7 Q
}/ B' p. H8 ~2 c
}
9 \6 U& T* A0 q7 ~# r8 Z
}2 G2 c+ ~3 G* D4 k7 T
8 b9 D$ i0 Z0 J
function dohandshake($socket, $req) r! v- a! U+ p P& @# D
{
2 f% ]9 J% T: M
// 获取加密key
- @# ]8 m% J F* O! Y
$acceptKey = $this->encry($req);) s% d1 e! Z! I* [7 ]8 Z
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" .
. V! }+ `8 V% I7 N8 S' W7 }
"Upgrade: websocket\r\n" .( _# C. I* k- n( z3 {3 v8 r
"Connection: Upgrade\r\n" .
4 _( @8 L! d! h
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" .
* i2 u8 h8 J+ b
"\r\n";: O, b+ A4 C. \! s( q$ o6 a+ j
, c* s. F5 t D8 K4 s
echo "dohandshake ".$upgrade.chr(0); 9 X8 M. F8 T: ^# U
// 写入socket
3 a0 ]! z5 C, \% I3 V
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0)));- g8 C$ v7 J1 i' q5 G4 M
// 标记握手已经成功，下次接受数据采用数据帧格式5 G: u3 i; o+ v
$this->handshake = true;
?/ M5 I2 M8 I3 C g5 N% A
}
4 ] k/ h5 x' S2 D% S( y* v
- @ y$ {; j& I, J" N
+ Z2 P/ h2 i- k3 s
function encry($req)
2 A1 f- [8 _) t- O$ \$ {/ i/ J3 v4 ^
{5 x4 C4 }- t' P0 u5 a* ^$ L
$key = $this->getKey($req);3 r' P& r5 }" M2 I# r1 I, t8 E7 G' ?
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";, S3 V. I- {9 z9 I7 V- O; O0 y
# D3 X) p8 _& X( v" v+ l. K
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));
1 @9 r/ f9 F6 J9 N" |$ ?
}7 Q5 R4 }" O2 R3 m+ t5 C& n
& f1 c+ A# h9 h9 w V
function getKey($req)
9 ?) n6 Z4 `$ K3 T" t( z* ^0 f
{! h O6 \/ |5 `3 i1 d# ^
$key = null;9 F; A- f0 z; H+ g& q
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) { , e0 Q% z* d2 t$ _6 Q4 q8 ]
$key = $match[1]; 7 Q/ \2 O0 D$ E- K
}
- r$ z M+ N" m2 n, i! e+ v, G. n7 k
return $key;1 O% o4 h7 \) f
}! P& N/ ?9 B6 i
1 u2 ]( A4 h8 B% \4 H
// 解析数据帧
* ~, H' U+ A1 r" ~4 z7 r# x
function decode($buffer) 4 K% @+ _6 K& Q2 H6 p
{
2 t1 a6 F9 v, H5 T: ^& J8 i. B
$len = $masks = $data = $decoded = null;
/ M$ z: h1 _5 B7 t/ u5 }8 M- M
$len = ord($buffer[1]) & 127;
- M0 q9 y: f7 d
# s; ^: [1 l$ h# h# W
if ($len === 126) {; z+ J( q: W p- M2 f2 C
$masks = substr($buffer, 4, 4);) `, V! X0 I2 H) x5 L0 i
$data = substr($buffer, 8);, p9 p/ `; y& A5 u
} else if ($len === 127) {
J+ q5 W7 \+ G% g; ~) V
$masks = substr($buffer, 10, 4);
' }: M8 E& n- d) [# W' X
$data = substr($buffer, 14);
& ?8 s1 Z+ h) }4 I, v
} else {
6 s$ M; m; g. k D: W
$masks = substr($buffer, 2, 4);, x! U- `: S1 B/ c# g2 @2 d
$data = substr($buffer, 6); c0 i2 _2 S& a+ ?" T
}5 }+ H+ z8 C3 J, k, ]: v
for ($index = 0; $index < strlen($data); $index++) {3 n- Q& p- Z* C! C
$decoded .= $data[$index] ^ $masks[$index % 4];* T6 {- A% q9 A/ f
}. o( v: Z, }0 t& Q6 a
return $decoded;
& y6 @ T( w$ k, \9 [
}
5 C4 R6 o2 l* D0 Z+ X/ H4 A5 K
1 Y) U, h7 J9 o% O
// 返回帧信息处理/ g: W7 ^! u; O1 F7 Q. I8 `* G
function frame($s)
4 S7 n w" G: s9 g$ G
{
9 L! `6 | J* ~! R8 R0 K4 S& c
$a = str_split($s, 125);
: s! H' G9 a3 |. ^7 u# |) ? M! v
if (count($a) == 1) {
! D) m& T5 ?1 C6 x8 X
return "\x81" . chr(strlen($a[0])) . $a[0];6 Q! o7 i/ X& G; t Z; N
}, @8 q5 W1 l/ M5 F
$ns = "";
& R% H. X0 w+ y8 j
foreach ($a as $o) {
; b5 `) c4 v6 a* F; G
$ns .= "\x81" . chr(strlen($o)) . $o;* I8 u* X4 K3 Y& d6 r: }8 I# [
}
$ x7 S) `3 l: r
return $ns;
" W0 \5 t2 v) N! G
}9 w$ m3 D/ p8 l* t, M
N' \" g7 h0 \% e- o3 I9 @
// 返回数据
4 U! V" R, {: G
function send($client, $msg)8 R: b; w7 I5 @, y b+ c
{( K* Y" K/ m/ q' g4 D) v( x' X
$msg = $this->frame($msg);4 p* a$ s+ v$ G
socket_write($client, $msg, strlen($msg));
( h8 k0 y9 \ L8 R; R
}2 n1 R. G& J6 i D5 Y' W
}
7 {3 d5 `/ f% A2 W
, {" k$ B% f# q$ X/ e( q8 \' k
测试 $ws = new WS("127.0.0.1",2000);, K$ s3 N' ]1 \6 |
' [; G* Z$ \$ O$ H

复制代码

帐号		自动登录	找回密码
密码			立即注册

[php学习资料] PHP 简单实现webSocket