PHP 简单实现webSocket

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

1）客户端实现

<html>9 ~' C/ d/ g3 n# F. x! X2 s' E
<head>0 t6 F B7 U( I& D" g/ ^8 H
<meta charset="UTF-8">1 x' p% z7 O. K
<title>Web sockets test</title>
; j5 T" k& ?% h7 W" l* Z' j M( ^
<script src="jquery-min.js" type="text/javascript"></script>% N4 O& j, Q: t! u/ v% e# Q
<script type="text/javascript">; H* o7 A9 B% m# L3 A8 C
var ws;4 d( i$ {7 D! ]. }7 d
function ToggleConnectionClicked() { % G2 g% H& p% R, m
try {2 B3 s1 Y! `: l2 _* j& g
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器 , j L) f- Z1 e: e' i- b
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};% U1 L: ?* E% m# ~
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);};- \( m4 v3 u4 U- r4 I0 t2 P3 {
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);};
" c6 J" L+ N& `& z# F$ _6 v" Q4 b
ws.onerror = function(event){alert("WebSocket异常！");}; q. a6 g: G' P, l
} catch (ex) {4 l* M3 C; C' F
alert(ex.message); 9 ]/ _, w# w4 m: _
}
/ }) D' P" B' }
};
( Q; ?* Q8 q; L7 E$ M L d+ o
" V7 I6 n# |3 O) E4 m3 u
function SendData() {
: k) D s, Y9 E5 S
try{
% i/ i7 ~3 g& y8 ?) f
var content = document.getElementById("content").value;2 Y& F# W8 R3 d
if(content){
% I4 v5 S3 m6 Y- _
ws.send(content);
, p6 G6 G, s/ Y8 y& Q, \
}5 d- O% n c+ W" J2 y" n3 r
3 }/ I' b$ a* p+ s: h
}catch(ex){3 U9 K6 ]1 U5 j# i! p
alert(ex.message);
+ I. t4 P( @1 p2 y+ S5 ^8 d
}
0 I# q; t4 z3 d) m, r3 R% o R% I
};- [- |: J4 H4 d9 {* o" L. H
# z" E1 [& A4 {0 T. ~+ F
function seestate(){" {. o+ q/ ~! Q4 B" r( u
alert(ws.readyState);+ p) n1 N2 Q5 Y0 V
}
5 g6 K) V9 ^+ x# k
7 `/ R/ @; ^/ |( H
</script>. t* L! h& \+ B. o2 r, { T
</head>) q- g" ]& H9 [6 s5 z# q! }9 H
<body>
7 [+ c6 @3 Q3 d) b+ L
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br />
4 \; P; w# b0 z+ Z& O
<textarea id="content" ></textarea>
8 Z# T' K+ B. [% y
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br />
' X& l: q J+ d W+ _0 C
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br />8 V& ]- V# K1 B' s
' I4 O% }! F6 D& s/ F2 [6 f+ L
</body>
! ^- {4 Q8 _( o" ~4 s! s6 @0 f
</html>4 p4 {6 F$ }' w( a! t

复制代码

2）服务器端实现

class WS {8 G* S7 @/ Z- m
var $master; // 连接 server 的 client
- v( D+ t8 o% I4 ^, l
var $sockets = array(); // 不同状态的 socket 管理
6 w( A: N: c' L0 }/ {1 z8 w7 a
var $handshake = false; // 判断是否握手
) ^, C$ n9 y% j0 _0 v# ` ]! c
, N' H3 W1 B3 K6 g" A' q
function __construct($address, $port){0 e7 e) Q7 ]& D& d5 H; O
// 建立一个 socket 套接字& Y+ _' c1 t4 N
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP)
% j" T2 U4 ^' L
or die("socket_create() failed");
9 n# Z2 v/ ?- g4 Q
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1) - F( b7 [. d' q5 J! v: c3 n; ^7 h- y
or die("socket_option() failed");7 E( t U$ e4 e5 h% @- G* _
socket_bind($this->master, $address, $port)
) u) D7 b: D3 z8 y4 i
or die("socket_bind() failed");& E$ v4 U K6 a
socket_listen($this->master, 2) 9 \# M! I& M/ |! M' K* F' ?
or die("socket_listen() failed");
/ G' n7 v" o5 V/ H, ~; \0 n
; m; y# d7 h9 T# \
$this->sockets[] = $this->master;# p0 B* w! o+ O2 z8 d
9 @& {! m; G$ F
// debug
- a8 m8 y, p4 b; |2 i2 a
echo("Master socket : ".$this->master."\n");4 u1 i% @8 E# V. q+ j" Q
6 q$ W5 t( M% t3 |- V X
while(true) {, Y# ` e/ g5 Z7 z8 T
//自动选择来消息的 socket 如果是握手自动选择主机
& P& Q! I$ f: @& W$ y1 J6 }4 m
$write = NULL;
* N- _" U. ~" p7 G" h" ^
$except = NULL;
* j; e) l3 |" j
socket_select($this->sockets, $write, $except, NULL);
2 }9 L$ \5 H2 I$ a
" R; s* a- }1 T4 }3 C
foreach ($this->sockets as $socket) {2 V2 A' g9 a, K; S0 x G
//连接主机的 client ' G; |, N' I0 s! T
if ($socket == $this->master){
/ q) H% M l% Y+ g
$client = socket_accept($this->master);5 R1 u& S$ s5 p) L8 V
if ($client < 0) {
# V. N8 R; z) B
// debug5 X( m9 w% L. K. `7 J- G) j& f3 d
echo "socket_accept() failed";+ \. \9 s! `$ w( C4 t+ v
continue;/ n) }6 q5 \- ~, U' Z$ ]5 i3 }
} else {; `! Z8 _6 F; n* w
//connect($client);
2 G- o- ~2 X1 A* d$ [
array_push($this->sockets, $client);
- b; ]1 L# R% j5 q) O$ N
echo "connect client\n";: r3 q" z/ v6 @+ Q% @+ [
}
/ |2 X3 S$ `2 b Z
} else {
! u s2 D3 t1 h* j1 Z) N
$bytes = @socket_recv($socket,$buffer,2048,0);/ Q$ S' a5 {8 v& Z, q
print_r($buffer);/ f2 b/ b( e9 }% K" o/ b* p) T }
if($bytes == 0) return;7 V4 t }7 G" h& d3 i- G
if (!$this->handshake) {% r2 }( I) }8 N0 Y
// 如果没有握手，先握手回应
" l) |* y2 l4 [, J! l; M8 I/ f; @9 f6 d
$this->doHandShake($socket, $buffer);- X7 _6 l. N* @% q+ M
echo "shakeHands\n";+ ?# \' n5 B' [) T) }# L
} else {
( x; Y) K+ ?* E- c. k/ ]7 G0 H" d" u
6 f' M3 ^4 q1 ?2 Z* M8 e& v" z
// 如果已经握手，直接接受数据，并处理0 j1 X) U9 G! C% f
$buffer = $this->decode($buffer);
0 J) u6 K8 y% H- S% |: j0 K
//process($socket, $buffer); 6 r; Z: t: Y+ c; A0 b
echo "send file\n";2 x7 x& D9 ^; y4 L2 F# c2 j% Y8 q
}4 E9 ?) D, O/ Z
}8 t5 P( g! {% s# g
}
}$ r4 c. B- y# A
}+ l# F5 A. A/ o; R1 c$ c; u1 H. a1 R
$ A0 s0 M( ]; o A% x8 O
function dohandshake($socket, $req), C3 j* f+ ]! h9 D2 _9 W# x( Y
{
1 o6 w5 w) G" f3 R: S: l8 \
// 获取加密key+ n+ l1 Q7 J$ G, ^) F. O! _# X" m
$acceptKey = $this->encry($req);
% }* S0 z# x E5 _! A; ^
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" .! E* g4 e0 y# E4 @9 R
"Upgrade: websocket\r\n" . a( z+ F( C d3 a9 E5 {- p7 X' }
"Connection: Upgrade\r\n" .
+ W; u: R: E0 W
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" .
0 V, K {. Q2 }5 y/ k) }9 j
"\r\n";
7 f1 z, ?5 g% h+ v+ {' a* a/ X
4 p0 \) q4 m+ k# p* s, u! g; X7 u( ?
echo "dohandshake ".$upgrade.chr(0); # _ v- D0 t. t0 O! g/ I; b0 N
// 写入socket3 Y% S9 @$ z- q
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0)));8 P* ^4 Y8 b: D4 T; W& n# a z( r
// 标记握手已经成功，下次接受数据采用数据帧格式
; s0 ?" p8 I) [ l* B
$this->handshake = true;
7 T$ l# g; O, f( W
}6 k2 @; a0 W% c/ @
. j% M+ q( J- T$ M+ o
- [* C, c8 K( m
function encry($req)- f1 F I$ {: v6 O2 }9 m! e, w
{
6 w8 i' S* L3 Q3 l: a: H
$key = $this->getKey($req);, d# T- e2 r2 M
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";9 E0 V5 J7 x( k+ u& C5 y2 z/ e
* f* O) F) \- y7 v+ x* I/ E* `
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));$ N b E+ ^5 W. e4 i! d
}
1 |) p$ r$ [& }2 }% S
1 A) w) j, a1 l1 o
function getKey($req) + p2 K7 y9 A0 l3 [) d
{
, v; L' \, i+ a- n O
$key = null;
5 C, o2 D1 L0 N T6 Y+ W6 W
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) {
/ `3 H+ `) \1 C4 x
$key = $match[1];
0 G4 q6 S4 F+ A% e( }3 O T2 l/ @
}8 W3 y0 y! F+ Q3 S5 c
return $key;
/ D! e8 d6 T' t2 y. x' t& A1 L
}
' t$ s/ a( P; w& A/ l. }+ g, a* v0 i8 U
( {4 Q% K$ R% K% |3 S; p
// 解析数据帧
: D2 y6 D: d7 q
function decode($buffer)
, q/ ]% \ h* m# B, _
{
& y9 v, a P: Y6 c( E1 f# U
$len = $masks = $data = $decoded = null;* Y$ _$ ^/ f: V
$len = ord($buffer[1]) & 127;
9 n' [9 ?2 ^3 Y8 R( N
5 c2 n5 p( r/ [ K' \5 \
if ($len === 126) {
8 Q$ ?4 o/ z: M& n6 t& I
$masks = substr($buffer, 4, 4);( q- O( ?7 B; Z' i0 o/ r
$data = substr($buffer, 8);, w6 e5 M, i; F6 X5 S ?
} else if ($len === 127) {" `$ J2 x7 V& ^/ J# G( z
$masks = substr($buffer, 10, 4);" y$ g# p5 ?- ` ^
$data = substr($buffer, 14);( O2 k; ^7 l7 Z7 |, g
} else {: D, q& a7 k" @. v1 m
$masks = substr($buffer, 2, 4);
( A. }# [5 ^. E
$data = substr($buffer, 6);
% A2 m6 Y2 C. U0 ]
}) z% T/ [: X0 r }1 k+ S# f
for ($index = 0; $index < strlen($data); $index++) {
: U( i" [' v: C$ M% o; f7 {* l" s, ~
$decoded .= $data[$index] ^ $masks[$index % 4];
* G( r) z- X1 Y& h
}8 f/ o: L6 D, I4 I; T
return $decoded;" n# W; h2 O9 I! y
}
; J' d; s- k0 e4 ~$ Z
% u: r" J. X1 M* t5 A1 s/ |1 s
// 返回帧信息处理/ u* m% ~; t; F t/ l9 h" W
function frame($s) / }. [. j o" g. l' H, g
{$ g d, E# v- j8 A% ^) r
$a = str_split($s, 125);& X1 T% e8 Y) g
if (count($a) == 1) {
J1 b$ r* S) j3 K4 J4 n
return "\x81" . chr(strlen($a[0])) . $a[0];( I7 y" _& p6 w+ Y2 [
}
$ J) ?$ f- r1 R3 J2 s
$ns = "";3 I, F6 D, ~8 ^
foreach ($a as $o) {
; J6 c7 w( A' c: y% b: {! z9 B
$ns .= "\x81" . chr(strlen($o)) . $o;
! a m3 ]& H+ u6 z: B
}
$ v2 ~: T; V2 h9 L
return $ns;
! N; y1 f2 h' L! M8 Z' {
}9 T# `% m5 n" d
6 }; F" f- v0 P; P: `
// 返回数据6 [( v/ J5 n! j$ P
function send($client, $msg)
3 z/ O8 Y$ {+ Y0 k9 U& r
{ k9 e9 P$ {) |* R! N4 f4 ?- b% r
$msg = $this->frame($msg);
$ K- [; W0 d1 S7 n& X& N' \
socket_write($client, $msg, strlen($msg));% ^, g; m8 h* T: r" t6 Y
}9 W$ j4 o4 o! M6 {# v% ^2 B+ O
}3 f* W. J. C1 A( n" s
) o6 v; g) V* \+ p t1 b& b, W) \
测试 $ws = new WS("127.0.0.1",2000);/ [" B( o( R: [2 W3 O" F6 @
! x+ I( v7 I& ~+ n

复制代码

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

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