PHP 简单实现webSocket

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

1）客户端实现

<html>
0 Y+ K1 n, R$ X
<head>
3 Q1 B! L3 Y: G. `2 w
<meta charset="UTF-8">
. b/ Y8 p% _8 n1 y
<title>Web sockets test</title>& [+ F0 ]$ T& W' [8 O) k# f
<script src="jquery-min.js" type="text/javascript"></script>0 ^8 |( e8 N8 `: I* H5 k _
<script type="text/javascript">
( R/ Q( G: p( ?# x6 E/ v
var ws;' R Y: P+ A* n6 J( c) n6 e
function ToggleConnectionClicked() { 7 G( O- A2 E/ s5 f
try {
2 ^/ q% i) O6 t5 m" J' f7 I
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器 . m$ E: Y" b: t8 q3 t: N. C' v
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};4 k4 P2 [$ ?. e8 z; B
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);};5 m9 M: g' }+ K" E8 I% b
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);};
9 P( Z/ N; p, n0 R
ws.onerror = function(event){alert("WebSocket异常！");};5 i* `2 r8 K! S% l4 q; U
} catch (ex) {) u. L3 h/ K# } E6 N
alert(ex.message);
& T" G# Z& a/ }4 U6 [* O) W
}2 c2 Z: w0 p5 Q! t& u
};+ H, Q3 y6 |8 k5 C6 J# A
3 B" i4 s6 o2 r4 k* w6 n
function SendData() {
0 ^- M3 ?3 o5 n9 y& _# M: K, I+ L
try{
+ N2 q3 E4 |# W. V/ {
var content = document.getElementById("content").value;
, f2 \3 j/ I: h/ Z t) }4 t# c
if(content){% A5 o9 Y7 s; }* Q1 y$ k2 a/ i
ws.send(content);$ |; O( C0 E: k! ?2 I9 N' J
}
* U. \" S7 t0 [0 p
0 i- p0 s+ m# `% K& e: d
}catch(ex){
! V' f9 E- o5 T2 |7 c6 U6 J
alert(ex.message);" s+ g( T8 y0 o. |
}2 H- e/ A+ S1 ]! M
};" k' A Q7 V. i, h$ n* e1 }
1 J3 f9 M1 b" E- K M* Z0 k
function seestate(){" c7 \2 d" x8 c7 ~2 H4 {6 [$ e8 ]
alert(ws.readyState);
- G/ a6 E' p4 N* `" w1 q: G# k
}& ]7 J/ t3 b( Z4 W2 d/ s
8 s3 D) {4 ]0 v6 _
</script>
2 R- m r5 t. l, S u9 M
</head>
3 B# F0 K0 y' O5 S1 J% a( i
<body>' B% g T; z' s G" @+ i' a
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br />' u5 D Z' W/ \6 g2 X" F# O
<textarea id="content" ></textarea>; g8 B/ L% l5 Q9 T0 N- u1 `4 @
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br />
( k9 Q$ z$ Q3 A" [5 u, D
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br />, R* L: `; M5 @& o! v
* }) M; V& |; _+ p
</body>
, k- W# }; x% [7 H9 V
</html>
: ]0 V0 f/ m# R+ {( j/ Z, G

复制代码

2）服务器端实现

class WS {& n7 \3 T6 \0 E
var $master; // 连接 server 的 client
- ^' w0 D/ e, U0 t/ A; D' h
var $sockets = array(); // 不同状态的 socket 管理
1 b6 t2 H; x8 g, _& n: d
var $handshake = false; // 判断是否握手
) s5 H0 Z- l( [2 H* r
* O8 }! W* N k. X- ]0 `2 s
function __construct($address, $port){
6 c& N4 k+ {, i
// 建立一个 socket 套接字0 i0 W3 N/ f2 l N' S" _
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP)
& S& @8 \8 @* m5 @; n% C4 X
or die("socket_create() failed");' Y8 A+ Z7 G" A5 o& d1 d# H1 M
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1) " {* w+ Z. w6 w, m
or die("socket_option() failed");
0 W3 B. m5 p4 }. Q
socket_bind($this->master, $address, $port) - k& s* k/ r: `/ L& Q3 G! o" u
or die("socket_bind() failed");
1 G5 ~/ [( [8 `: D
socket_listen($this->master, 2)
% ^# C. J% Y) ?
or die("socket_listen() failed");7 i7 G: P$ I# I& M
) I4 n6 t* J- G
$this->sockets[] = $this->master;0 j8 \, n' i6 E n) \! f
2 z6 [) i% N% T) B2 j0 ]: l: \+ r
// debug+ h: p4 ^. i/ U$ X
echo("Master socket : ".$this->master."\n");
2 m4 S. Z! e. W) [1 r
( g2 i9 ]0 Q' W: g7 O* B
while(true) {, `: ^5 w0 j. G
//自动选择来消息的 socket 如果是握手自动选择主机* L. ]6 N/ D% c) ?
$write = NULL;
; }) T7 p+ T: E$ t% V% K
$except = NULL;
9 p8 }, R: d. z# q' b N4 ^
socket_select($this->sockets, $write, $except, NULL);
! L7 B6 {3 C$ a6 o o/ \
* w; U& d: a) b/ W4 w
foreach ($this->sockets as $socket) {
@. e& x2 z4 [1 ?7 D3 _ @, p
//连接主机的 client + w- J$ t+ _0 q5 |
if ($socket == $this->master){
& W$ U: H8 }8 ^ m+ f
$client = socket_accept($this->master);6 i$ T) [1 s5 t& Q# ^
if ($client < 0) {8 p+ q0 g* }. R* |" R' s
// debug6 Q" }: l( F8 I5 \/ U/ e5 D5 `
echo "socket_accept() failed";
4 c* Z4 B. W1 f- @4 x* n. w( }
continue;1 W& [5 v3 M- L/ Q, H
} else {
" m8 j; b) M- u3 r% I
//connect($client);, u4 G( `9 \2 k4 C
array_push($this->sockets, $client);
$ H% X; B' i1 t
echo "connect client\n";' S* J3 p# l. ^( B6 F! Q
}
5 Y) ~* R# ] H& p' B9 n
} else {0 c7 `9 @1 m9 c$ b7 R, X
$bytes = @socket_recv($socket,$buffer,2048,0);
7 H$ O' k) a& h/ x- D+ Y8 K( N
print_r($buffer);
0 s2 Y- v+ L! g4 u6 a
if($bytes == 0) return;
$ t; M" @& a" ]/ Q
if (!$this->handshake) {* E2 C) W5 P8 v* d! J% P0 y. u) F
// 如果没有握手，先握手回应
$ y9 G# P* g) b3 T- R
$this->doHandShake($socket, $buffer);( u; l6 k! ]6 o) a: C2 M2 L
echo "shakeHands\n";' X9 `! q+ f$ ~/ G
} else {
- F& E) v$ v8 D1 W: ?! P1 p! l
" @3 z" L/ A; t+ T
// 如果已经握手，直接接受数据，并处理
4 f6 G9 ~1 K+ M
$buffer = $this->decode($buffer);
1 q/ F1 N' ^4 ^" \
//process($socket, $buffer);
" m( X$ L7 u2 G4 L# v9 T
echo "send file\n";" \% A3 Z0 c2 D8 f# m
}4 |* g0 F5 T( K3 `
}& R# c$ x, g0 y3 `1 Q$ R
}1 r) Q7 F1 _6 K& T$ W! A
}: B6 g3 [ b. S* i5 f7 u
}3 E( O/ ~* ]" I+ Y( D
: K* z: c# H8 e) d
function dohandshake($socket, $req)
{ A, y# N. p( Y: ^* ?8 j
// 获取加密key: _6 n) J# w. Z) L1 k3 W
$acceptKey = $this->encry($req);
! \. ~$ I8 g' u0 M* l
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" .
E; R; E- [" M& J# K
"Upgrade: websocket\r\n" .
7 c$ E* m- m2 j0 o2 k" ]
"Connection: Upgrade\r\n" .* }. o, h+ ]& u) a+ p; X! |8 x
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" .
2 f L( p& a9 h+ E. W. n* ?
"\r\n";
) Z+ r0 }+ b3 D5 \3 q: N
" C8 ~- t {' m
echo "dohandshake ".$upgrade.chr(0);
1 D3 b3 [0 @/ a8 J
// 写入socket
, x; m4 Y. o/ ~ m" p X) d! |
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0)));" J1 v1 N- f# n% `
// 标记握手已经成功，下次接受数据采用数据帧格式
( I8 N# M- e$ x: U5 r6 L G' X
$this->handshake = true;
: @# s7 G: ^, x, V
}
+ i1 k' o, j+ C- N$ Q E
1 l0 I1 Y' c+ D S S1 M4 ~( m
3 t z8 Q! v' v0 E0 t! O
function encry($req)4 ]; d' F* k8 y& P$ c
{
2 S/ \0 b5 b5 ?# }! Y( g8 c" T
$key = $this->getKey($req);) B; ^- S8 q* C! M* ~, W8 r9 I4 }! H# W
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";; w3 J. O4 K- \2 @$ _* R
' c; _1 u( |) Z, p! o% B# ]
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));# o/ _; v' w% J# F) R: K! l' ]
}
! p0 c; e" A3 k- q1 q i3 {4 w* m
! D* j% v, M# N" `. }
function getKey($req)
5 B; w+ z7 S4 N1 e% }
{; {$ d4 v+ P$ t1 K
$key = null;
( C) j: P4 O; w* V
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) {
' V5 d! G' k* ^1 W
$key = $match[1]; 2 N }1 P, Y' S# k& D) _2 _
}8 g7 r$ H" J$ o X+ V
return $key;. Y; v# C: q# |6 n Z1 T& |
}2 Z% u$ _6 J! G0 }+ V. i
$ d3 | [4 a' E7 ?# q
// 解析数据帧
/ M: o |7 k' V3 }0 T
function decode($buffer)
8 O- `6 U% n/ P6 V3 h3 i
{1 k4 _" P, z8 O# p' T+ q# F) c
$len = $masks = $data = $decoded = null;- n1 b" h/ H. z% r
$len = ord($buffer[1]) & 127;# p( Y( d& P' \' O7 P
+ O4 U1 c+ z% e+ q+ o
if ($len === 126) {* `: _8 \* _) P& y+ s* B5 A
$masks = substr($buffer, 4, 4);
, c, N N$ p) L' r# C
$data = substr($buffer, 8);# J* ?0 @3 C L+ x& `" N6 `
} else if ($len === 127) { {. v# v: d! T, w5 C. o' c
$masks = substr($buffer, 10, 4);. j* l( D8 R7 M2 C) O, Z
$data = substr($buffer, 14);
" f) d5 i N5 \' ?, q
} else {
0 c' c H; i8 s: |) A
$masks = substr($buffer, 2, 4);: l( P z$ V! m% j" O$ d1 w3 m
$data = substr($buffer, 6);
. {/ a# C! z( m' D Z) u
}# J0 D! R$ _+ I! n
for ($index = 0; $index < strlen($data); $index++) {3 q( c# L* W* @( M
$decoded .= $data[$index] ^ $masks[$index % 4];: Z2 J0 o+ ?* V
}
: B Z( J6 B( Y% H% ~+ o
return $decoded;# X. s8 Z/ }4 n$ \) D0 S
}
& [- F2 p6 d. Q, F/ R" V
j6 Q, Y1 _+ X( O. I/ b
// 返回帧信息处理3 j Y9 |% ^* F. T. b+ [8 b& {
function frame($s) 5 {# A2 @! Q9 ^. z. w$ F, x4 n* u' m
{: L/ J6 E8 r- z# k* m$ z! M
$a = str_split($s, 125);0 g& k( x) l. }, t; w* x; R3 @$ M
if (count($a) == 1) {4 }. A% W% j( C- E+ X* W! H
return "\x81" . chr(strlen($a[0])) . $a[0];: ~" z. ~% ~' M1 k2 u
}! _3 u1 u3 F: K) M8 G& D( o
$ns = "";5 S9 O7 ]+ z7 [2 R6 U
foreach ($a as $o) {
8 k* N; P! [; q0 ?5 t1 ?; i
$ns .= "\x81" . chr(strlen($o)) . $o;! R/ z2 X( F5 G6 X$ @
}
7 Y0 |4 j2 x( D0 L5 m6 G- Q
return $ns;; N4 {% i/ Y# p0 j( ^ E s
}
: F2 G& E' ~ n; e
/ R7 t# e1 W0 d* G* P. y
// 返回数据9 I U2 H- S# u& [8 S" i
function send($client, $msg)+ k/ O5 M6 @) y$ }/ z0 f7 h
{# B8 c+ a- z V# q. g
$msg = $this->frame($msg);6 v/ N0 z: P. ~5 S* T. i
socket_write($client, $msg, strlen($msg));& k. H# M% J* a' S
}9 D9 {7 M5 w ]$ ]+ @+ S% _, D
}
, S- p& ~6 d1 l# B. N( g
0 d! {5 U* d1 n$ p5 t" l
测试 $ws = new WS("127.0.0.1",2000);
6 X. e3 S$ w4 a- t8 K" ]

复制代码

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

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