PHP 简单实现webSocket

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

1）客户端实现

<html>
8 H9 k5 C2 S# i) b& G" H" W; O
<head># P, c9 J9 o+ g0 g4 k
<meta charset="UTF-8">
9 w' H; |1 k% x( {) W' `& n6 l) c3 D
<title>Web sockets test</title>7 \, V9 n5 ?1 G0 Z. i, w. v9 e
<script src="jquery-min.js" type="text/javascript"></script>
# h3 j, [( I. @! l* d' C& t
<script type="text/javascript">; ?/ ?" }; F% p8 i
var ws;
& K. a, q( C w
function ToggleConnectionClicked() {
. L F% v( e( _$ h! P9 b; ~( c
try {) U+ r5 R3 I, N2 p
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器 ^7 Y6 u! o, e+ E
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};, z, Q: v. w7 F7 H4 |1 a
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);};
# I$ D0 W6 u% m
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);}; ]. h0 e5 p+ H2 l* B* _% I$ \
ws.onerror = function(event){alert("WebSocket异常！");};; h" b8 U! }% W* `6 y @
} catch (ex) {
' i% i+ Q3 E% h; t6 @
alert(ex.message); 7 b* K- j2 C' \0 p; o9 T
}
4 v* @! g H! J" M4 ^
};
0 g" S. o- f6 p1 k! p4 C8 n
' O& A1 d) j* S J/ s1 n2 u% D, a
function SendData() {
0 v$ N3 i! p/ i6 v8 _. z
try{/ C) c% h% C; O. X& a, u
var content = document.getElementById("content").value;" }/ S; p Q; S% ]7 R9 w/ a
if(content){. K6 n. W# q& N& g; z( I
ws.send(content);
% R- m# }4 m, |) r, ^. T7 J8 G
}
, X/ j! E- ]- H2 W/ l% R- [, v7 p
" [+ e: J4 C& T9 Q5 t4 c4 _
}catch(ex){4 V' H4 r7 G) P9 y$ {
alert(ex.message);
/ n, @+ M% c5 A) D2 G& h3 j
}, g7 H9 s6 G6 M" C
};
+ N$ X. _) _& v. n
( A! H. D6 T8 ~7 g+ R7 A
function seestate(){
. x, T; C, g6 k% x( I$ d
alert(ws.readyState);
; h" E. M( ~8 w$ n6 g0 {+ w2 M2 E! ]" S
}
4 s" _6 r7 v. B2 S$ d& H
" Q8 i; X* }& u* u. R
</script>
* F5 I% J2 b# e7 k
</head>
# C, Y) V `; g1 [) D% t
<body>! C9 t( e* T# I3 l
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br />
' w- Q# F, k+ `, H+ i
<textarea id="content" ></textarea>
* u2 S9 D3 U: ^+ S2 h' ]' `4 j
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br /> [/ z0 L. j$ q# k8 @
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br />
- t6 t. l3 B& a+ ]% P& m3 W) }
) F; {5 t! g& a6 H) A
</body>
4 F; V4 a' z" h) m- s* P& }6 M
</html>( b3 E" T) V. p' [! I

复制代码

2）服务器端实现

class WS {2 P, D' E! z) t# ?% A
var $master; // 连接 server 的 client2 K0 w6 H% v, N9 G' Y
var $sockets = array(); // 不同状态的 socket 管理) }" ?+ [& V1 D/ O/ r0 `9 f, Y
var $handshake = false; // 判断是否握手
. Y% g2 \! ^# |# s
# o6 _0 E4 o- _- |% V! D9 K
function __construct($address, $port){
/ u6 R; ~" o4 M7 A0 o
// 建立一个 socket 套接字
) t- s% F) O4 B1 i
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP) 0 [8 U3 r$ Z; v; z; \) H
or die("socket_create() failed");' ~% b: |6 w: \: n, O
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1) ' D$ ^/ A" F+ `. {% {4 n
or die("socket_option() failed");
2 {- L% o, a$ b3 B! |2 W5 u# t. F
socket_bind($this->master, $address, $port)
1 K/ V, a' ?6 S1 u
or die("socket_bind() failed");" q1 [& t# t9 n Z, E+ e* _7 Z
socket_listen($this->master, 2) 1 O" j; e9 T& }% b; b
or die("socket_listen() failed");" ?. y4 r; L+ b: z4 ?" ^: E$ A& U
7 ]! m2 b6 Z) ]
$this->sockets[] = $this->master;! E5 j* M4 A+ V6 k k' U
: n8 S x1 U2 i" [6 [2 [% c/ w: S
// debug
# |; ? m+ R( c, p4 N
echo("Master socket : ".$this->master."\n");' D2 {; _0 I$ M, K6 v) M" s; @
/ }: c% E6 X* O" E
while(true) { ~ |' Z; B* S# R# F4 n, [
//自动选择来消息的 socket 如果是握手自动选择主机, A% n$ Y" n: l+ x
$write = NULL;% k2 K! R# C/ z- ?% G1 n
$except = NULL;
6 c- L. [2 T# Z* d
socket_select($this->sockets, $write, $except, NULL);
& q3 G2 }6 ?- ? {2 w
- O) j# a3 h) r" U
foreach ($this->sockets as $socket) {
2 E$ f) Z0 A( l u& S( O6 s
//连接主机的 client 2 ?. r: i/ ^; O
if ($socket == $this->master){2 _8 g2 _! F# h
$client = socket_accept($this->master);
! r' |0 N2 d! d: n/ ~& E; r
if ($client < 0) {
& U, ^2 a/ W: p4 D6 T- z) o
// debug' E; ? R7 U1 ]0 }5 [% t
echo "socket_accept() failed";
3 b: E7 ]+ U; q) T& u Q
continue;
- W2 Y, ]! |1 B G1 g* e1 y L
} else {
8 I+ w* D* L0 U0 ]5 i% |6 r: |
//connect($client);
/ I; @6 p2 ?4 k I
array_push($this->sockets, $client);( O" b- i2 [# z( k: z: m
echo "connect client\n";
: [- J6 b3 ]1 u/ f8 X
}
# i& ]$ y: o) U) b8 Y' w
} else {6 z! I" \ _9 ?7 |3 t9 C
$bytes = @socket_recv($socket,$buffer,2048,0);2 _& E, ]% J: L8 K8 A
print_r($buffer);
; ~5 M. m4 ?* @/ I. s, X W
if($bytes == 0) return; j0 W: [" ~0 p. E5 M, ~
if (!$this->handshake) {' M$ W/ d- ?3 L: z$ D) M4 s
// 如果没有握手，先握手回应6 U" @8 V' e+ w+ Q0 W$ p
$this->doHandShake($socket, $buffer);
4 o. ~% T: B$ a1 {
echo "shakeHands\n";
6 i" q! W% S$ Q7 L' i3 \# Q1 R
} else {! c* z5 l% w" @! Y9 j7 j- {6 B% R
: \. t9 X" b/ \7 b
// 如果已经握手，直接接受数据，并处理
' d" a! X+ _* Q5 V& D+ ^: A
$buffer = $this->decode($buffer);
6 ~2 ^( v/ w& F4 ^" K6 ~1 L
//process($socket, $buffer); & C: x9 j( Y# N, b' N2 [: u% `
echo "send file\n";
- L" C& G* w% J* g9 Q/ P" s8 e
} n! T' |3 f9 ]/ j* J0 k7 U6 C
}
5 M# K6 W1 d t" }# v, g2 d( _
}; w; M N: \( |3 l) y m
}( W( T7 K% P P0 t- G7 S
}+ z$ K- W+ Q7 x. \. d; ~8 a$ H5 B
$ Y: F: \5 A7 a# ]
function dohandshake($socket, $req)
9 {& J( L, y0 @2 H+ F
{, m) r L Z; P1 x. V" x: W
// 获取加密key% i- C$ i% T2 U
$acceptKey = $this->encry($req);
7 E U/ T' v o1 t# X( b3 h
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" .+ d c2 s. i' V0 X) q* d
"Upgrade: websocket\r\n" .
, s) x0 Z0 ]6 b! @- q
"Connection: Upgrade\r\n" .) j% H X0 S7 P
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" .
" r2 m: k1 t* N
"\r\n";
' E. F5 Z2 V1 d7 w: p. T0 a" }
$ W9 K% r5 [/ e% f5 C
echo "dohandshake ".$upgrade.chr(0); 1 P3 a7 C8 D5 I+ B
// 写入socket- ^* |8 ~! E3 f3 P
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0)));
4 J* T" c" K/ ~! w5 o' T
// 标记握手已经成功，下次接受数据采用数据帧格式+ _8 K0 d3 G9 m- `
$this->handshake = true;) y( N3 z2 w9 W/ T6 d
}/ r: r7 V8 ^8 [8 y$ f
) z- _0 n2 W+ R: i# i! X* c; A
0 L0 F9 u$ {- T/ i- I9 r4 ^0 W
function encry($req)
1 v( @. D3 u; a6 V2 ^
{9 \5 X$ k0 p3 D: _2 P" m% o
$key = $this->getKey($req);
) p8 `, P' u' l; L# X) Z; K
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";" \; e* q5 w' ^2 N T' y
5 [( Y4 E2 Q# h8 R8 x! h7 P
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));
+ z+ g8 b0 c4 W$ f5 g; F, }8 r) n; H- E
}
+ y8 V( q8 N. X) {: n1 W
& t1 U1 ?* @; i* R. `; M
function getKey($req) ) V$ h$ F$ |/ |1 e5 i
{) j( }5 @9 R2 Z6 W
$key = null;
5 m) h7 o# }2 Z: v8 n# ? R
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) {
. [6 c4 O, `2 d% ], x/ K U
$key = $match[1];
3 u0 X1 r' J6 d
}& u. b* Z0 y' g5 s, p% O3 R; d- |
return $key;2 e, a8 b3 G# {6 s& z9 Z
}( ~4 z r& ^$ h8 v( l! S+ O/ B1 Y- K0 P
+ A/ D* n% {7 N! r$ c; R4 }2 w
// 解析数据帧
& o; l& I) W9 l
function decode($buffer)
0 U7 ?+ [& W( d) a5 C
{6 t/ H4 `" I: U
$len = $masks = $data = $decoded = null;/ Z* i3 c" _* x) o! I* q
$len = ord($buffer[1]) & 127;' D4 g D) v* G2 {
8 K$ O* m# P$ }- }( U
if ($len === 126) {
; i6 y! F5 `$ _' [
$masks = substr($buffer, 4, 4);
+ s# D' u" E) ^9 a
$data = substr($buffer, 8);; X, a, e% C8 N8 z3 d5 v& O
} else if ($len === 127) {& J. K. H% r8 I9 }$ K
$masks = substr($buffer, 10, 4);; S: O' c! K9 p
$data = substr($buffer, 14);* d, M, n! W/ I$ k
} else {
4 f2 b% N+ e2 ]5 w% {( v
$masks = substr($buffer, 2, 4);) D7 Y6 F4 b. }
$data = substr($buffer, 6);
3 h4 v% |3 U/ ?; y! v$ z& p
}) Q, j7 n9 y H. C K' F
for ($index = 0; $index < strlen($data); $index++) {7 J: q% A( I1 [, f' d) z8 G$ l
$decoded .= $data[$index] ^ $masks[$index % 4];% q9 }& b a4 G5 j- E$ \! ]7 S
}
4 P+ z+ d4 h+ F' F5 |# Z
return $decoded;) d# C9 `% L" X0 _7 G
}
" X; |& L' y6 J s$ N
+ L/ H9 p* u9 g9 s
// 返回帧信息处理
; f* ?: K7 r5 c# c$ H
function frame($s)
9 f Z2 O" {; _' {
{4 i* F& I7 i# P; k
$a = str_split($s, 125);
! Z: J& ? g% x+ E, q6 L6 S- q5 g
if (count($a) == 1) {+ O( k) d7 W: E* F
return "\x81" . chr(strlen($a[0])) . $a[0];
( _0 m9 |2 }, Y! z0 u
}: t& R3 S4 V4 j" |9 a7 _3 i
$ns = "";
- }+ R g9 @6 E1 q9 B# H
foreach ($a as $o) {4 q0 v/ c' x, o6 D( T, h
$ns .= "\x81" . chr(strlen($o)) . $o;
o# a1 S+ R* V/ p! ^7 n
}
* Y* d' d: _$ [
return $ns;
2 ^. J* x! M2 X
}
+ a# a% n' Y8 ~3 N: R# ~
* H3 B5 i0 ?/ `0 Y% @" g
// 返回数据: b1 u7 ~' R. f
function send($client, $msg)
; v) o1 B: ~$ d1 C* f2 m
{
# y4 x3 k- L9 c# A) v
$msg = $this->frame($msg);
5 Z- J5 _& b! x& Z
socket_write($client, $msg, strlen($msg));
" A6 D6 B: L9 O+ h) a- }
}
" C, t# G! G; @; ~; m
}4 Y; ?% N& k4 A
1 U) X7 E# ]; z; r" ~% E
测试 $ws = new WS("127.0.0.1",2000);, q# C% o$ m7 H0 e- c
: ? N2 T- ]3 a( B

复制代码

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

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