PHP 简单实现webSocket

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

1）客户端实现

<html>
* h" C. y4 e8 B6 _. Y9 \
<head> b" x0 x$ Q; B" A0 _
<meta charset="UTF-8">
+ b: }/ o: W9 L- P7 M0 `" W
<title>Web sockets test</title>
, D% P6 x* I- G% v3 v
<script src="jquery-min.js" type="text/javascript"></script>
1 e3 {7 _ I- A2 d( [& a
<script type="text/javascript">
# |5 h* j1 a+ H0 F. f& c
var ws;
# e: R6 Z5 }1 }
function ToggleConnectionClicked() {
, I" q2 L7 v! p, ]
try {
0 V; k" y! q3 `
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器
6 ?% N2 N! ?% ]1 |+ e
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};
m3 O5 f% O( X" k3 Y4 S
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);};
8 ~( E( w- q( x! b
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);};& G& g; L7 J* {8 o/ W# a
ws.onerror = function(event){alert("WebSocket异常！");};0 k: K0 C0 I- s
} catch (ex) {
6 v, _( }5 {' V, V0 P- x% K
alert(ex.message); ) l( o! ~6 J* | E' D. |% o
}9 ]5 b. H: S. [7 U/ J9 i
};8 x7 q- h1 K0 w" ?7 `9 \% [- D
( S/ i( y# x) q1 y* n& F
function SendData() {
/ b1 A& |" ?+ P1 P
try{ |8 u( |* ~1 P" A" [/ R
var content = document.getElementById("content").value;
2 e. B* k, J& [1 k0 \" [( Y
if(content){' d' ~4 w( x1 [
ws.send(content);' j3 v: U2 } ~1 d
}& a/ A. Z0 h! n, _* m' i% Z/ {
0 |9 X; E, u0 |
}catch(ex){ l2 M" v0 {6 J
alert(ex.message);" F7 f0 p9 T5 Y) p% t8 }
}
0 g- p7 e% X! }7 F4 p7 M9 U
};
+ h* G- F; Y! m; f! b
J9 S' N3 Z- w/ o: T! j% d
function seestate(){! n( v3 |' T8 R
alert(ws.readyState);' T( `9 W+ ^. V) l4 h
}6 P) J# S$ c6 Q2 C6 j% s
, E- G8 ?/ o ~9 g1 E8 s
</script>
7 ~/ R$ x: d) w( @5 M! ?0 J% w
</head>' d, K8 R* I5 Y1 n r) H/ ~
<body>/ M& f+ b5 G3 T: |0 k0 E7 |6 M
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br />
) H( v+ M+ ~! O2 Z! `. @" [9 D
<textarea id="content" ></textarea>
$ V% s8 G( M0 U) p7 I y3 T, k
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br />
9 o" _' e9 B% \6 m! k
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br />
& Z) x1 q: Q/ Q* x/ {8 W
3 j/ j1 B) u: N( J& ?: T
</body>
1 Q; L( Y1 t9 j5 t7 E) c& e
</html>
* U$ H% Y. b' |, W0 v5 C; E4 ~

复制代码

2）服务器端实现

class WS {$ H @! b3 K G. C& T
var $master; // 连接 server 的 client1 T! ~2 c- ]- ^! I! I- C; M
var $sockets = array(); // 不同状态的 socket 管理$ M1 {7 ~" v( v) }) x4 s8 h$ d" ?
var $handshake = false; // 判断是否握手
; O' N8 v/ b$ s5 n
: U, r% }9 H* b3 @% I
function __construct($address, $port){
6 c- p+ B+ M k4 `* ^ Y6 M9 H
// 建立一个 socket 套接字* T* n. ?0 Q" n3 J& T
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP)
9 S# ?, l/ k5 X( n* T
or die("socket_create() failed");
. |; C# T; ^6 a/ y ^" I
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1)
4 ?$ W! K- C' E1 e4 Q* Q; t
or die("socket_option() failed");9 o7 B. b* J2 c8 ~0 t
socket_bind($this->master, $address, $port) 9 |' Q5 o. M B. w
or die("socket_bind() failed");, n" o; I _7 r/ g* j0 a5 G& c
socket_listen($this->master, 2) 4 l; G$ n" w. i S
or die("socket_listen() failed");
3 C1 L3 z6 C7 u' s. T4 j
. w' Z$ y/ E5 I$ R: m
$this->sockets[] = $this->master;
* O; P, I; Y) ?% a& F
* @3 R. L" v7 t9 g& X3 A3 A3 I
// debug, c& E8 F" V$ s$ g. O
echo("Master socket : ".$this->master."\n");/ f) W/ R$ f; [- P6 L
0 q" J$ M" d" s7 U: L% L7 O4 J9 q
while(true) {1 z3 t: [4 e- z. U2 t% I! v/ L- W5 r
//自动选择来消息的 socket 如果是握手自动选择主机! O: H/ H Y7 T* z% W; n
$write = NULL;/ h* _ J7 c t+ r. ]+ ]
$except = NULL;+ H' j9 H; p; T( t
socket_select($this->sockets, $write, $except, NULL);- s: R4 c' F v6 {. i% g# ]
$ G& D4 {& p T$ l- Y! B1 v$ G
foreach ($this->sockets as $socket) {& [: f7 R' g/ G0 p; g9 a# ~
//连接主机的 client 0 x3 i$ z- e I
if ($socket == $this->master){
4 u+ J5 @& P8 f6 A s
$client = socket_accept($this->master);; x1 W( S/ ^' c
if ($client < 0) {
3 d6 x$ B% Q' V, o$ q8 V: I0 k9 |$ b
// debug
8 ]1 S* @. V; p, O/ | w4 J
echo "socket_accept() failed";
# Y7 F. C( _- U! p8 W; i
continue;
$ a7 L! q9 i9 u3 p! ?" ^
} else {7 K* y$ @8 k( Y9 G& |( {! `
//connect($client);
+ K# h" n/ @4 X* @/ W) c
array_push($this->sockets, $client);7 F" o+ k N# T, [
echo "connect client\n";3 F; z7 \/ }" |) @
}
/ a3 V% P X b8 j
} else {. }6 ]6 V+ j4 w Z
$bytes = @socket_recv($socket,$buffer,2048,0);
# }& z6 L) A6 c2 L
print_r($buffer);
* N* W9 N& j/ y3 u' i6 z
if($bytes == 0) return;
% m+ z5 ^( t' a- e, @
if (!$this->handshake) {! x1 n/ h1 P$ j3 U" v, B. v: j
// 如果没有握手，先握手回应 Y7 V- {* [- e9 t2 o1 a n4 i' d
$this->doHandShake($socket, $buffer);- ~, w, k% C% C) h K
echo "shakeHands\n";
+ I; s& e6 l- E1 E% x, V
} else {* \9 @ l; g# c/ V# Q: J
/ p8 w) y6 }' v/ L. I$ [
// 如果已经握手，直接接受数据，并处理 [* G6 z9 a/ V" g
$buffer = $this->decode($buffer);$ E- R; r( n7 I/ {2 ?& s1 f
//process($socket, $buffer); 2 Y! @7 l/ h# ^: C: x2 r
echo "send file\n";
3 j* x( e- b- z% n6 y+ t9 \& R1 S
}/ |+ s+ c+ m5 T/ t, h) W5 M: I
}; z, t1 |% [5 i
}
" B2 R1 k# y: P; |* Z" t4 o/ M
}
; }' t) C' F! t4 ^/ ^
}' v% i6 w% o, |' s: @0 U) `
5 N( s4 A+ ^% ~3 U( S' f
function dohandshake($socket, $req)
7 O/ b5 Z6 ?% L& s. n5 y
{- J' x3 j9 Z+ |
// 获取加密key
$ v# Y6 Q9 K* P- o0 c+ W" m6 b" j
$acceptKey = $this->encry($req);
& o; \# }$ z! N, {* x/ o! {3 n
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" .& R1 L, S7 K/ ?* `. I5 Y5 h. O9 V
"Upgrade: websocket\r\n" .
"Connection: Upgrade\r\n" ." q* e3 k( D4 `+ P( L3 L- z+ b ^! G
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" .$ K8 T* h7 X9 j& G
"\r\n";
3 t# J+ b7 I. f ~6 a y& `( i. w
0 q: v5 N( ^2 }, p9 e r w* c
echo "dohandshake ".$upgrade.chr(0); ! ?5 H+ S$ X( {# j2 T5 C
// 写入socket' Q) w9 A" V9 Z# t/ q3 X
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0)));
1 ]# `5 v) W# a( i
// 标记握手已经成功，下次接受数据采用数据帧格式
. l7 g5 _9 w! ?8 f$ g# f
$this->handshake = true;; J6 ?4 }0 l3 f+ ]) G0 b
}
) C3 R* r9 p1 u
+ ~9 e; V. T( ]- |5 Z
+ z; c. @( q* C' l0 O
function encry($req)6 X$ T( W3 |* y! m2 |! c* s6 Y$ i
{
' ]# n3 }7 f) t8 \" S* [
$key = $this->getKey($req);1 l" x" W! C) G9 z2 c. v
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";
4 M0 K3 l, P( z+ N& |) Z
0 Q2 |3 B( Q; n$ Z
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));+ r6 R3 F( T: w
}
6 h6 v" Y5 Z7 \) ~0 I" F4 q
( d1 @; ]: W3 x1 J, B" l
function getKey($req)
( ~- @! x8 f4 P- v( Q; Q
{
1 x8 i' y( \" ]9 y8 N0 \
$key = null;: x& B6 u& a/ G" Q' u0 P
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) {
5 X$ ]+ t2 A* A- T0 c2 v+ g
$key = $match[1];
6 I# J2 S* `% l- H
}
( e9 |+ F* v$ }$ `7 R. K k
return $key;
' g* o' ~9 |: s* \
}
( X/ v* n* U# C: ?1 T
$ d+ T) h6 E) I
// 解析数据帧: X* x& _' m' p6 @8 ^! Q
function decode($buffer) 6 L, E4 |0 v, m
{
$ o8 f! H) v B3 U9 H4 p) ]
$len = $masks = $data = $decoded = null;$ ?3 E# c' c3 t$ O6 }5 c
$len = ord($buffer[1]) & 127;
( i+ k1 J7 e a1 F9 Y; o
; I$ R! `. X3 D7 B7 i0 J! J8 Z
if ($len === 126) {& f; l5 [! u% G1 t# K0 U# D
$masks = substr($buffer, 4, 4);
A7 r1 ]: h( g# L
$data = substr($buffer, 8);
& ~4 F; D: i' D% ~) o
} else if ($len === 127) {
2 }& i% P( N+ O/ c7 A
$masks = substr($buffer, 10, 4);
9 J6 C; {1 x$ j
$data = substr($buffer, 14);, K" M8 t( w" N( P+ f
} else {3 A6 g o5 F% _' H, y/ E" [
$masks = substr($buffer, 2, 4);0 {' B& M7 T/ u7 w
$data = substr($buffer, 6);1 F! [( j8 f. } _% \. i& I
}
6 T) r1 f8 ^; t
for ($index = 0; $index < strlen($data); $index++) {
% x* f2 ^- v+ O3 G( S
$decoded .= $data[$index] ^ $masks[$index % 4]; e- ^: [; j) S. w% ]2 f. s
}0 r, c0 b' Y) o$ l
return $decoded;/ \3 X1 Q$ T2 K/ K+ [
}
P) p- t d I! [# y/ o8 ]
& L/ a% a# g1 l/ q7 f
// 返回帧信息处理
( |4 N6 N9 H$ [6 p5 k1 ]
function frame($s) 7 O! @( R) f V1 h
{
( s9 E( P. T0 F- `$ i9 i5 Z
$a = str_split($s, 125);
4 n, Q' y+ ^6 k4 G8 H* Z
if (count($a) == 1) {0 p% f/ \0 E+ J2 _( R+ U. ~! X
return "\x81" . chr(strlen($a[0])) . $a[0];) |' S9 h& J( m, T* W6 M A
}( B& }% V4 ?, O/ |
$ns = "";
4 n- w& @- S# ~9 M1 P" `
foreach ($a as $o) {
* U* y- i. }6 P. f o9 d* N' T
$ns .= "\x81" . chr(strlen($o)) . $o;
+ V/ |0 _& r5 K7 k
}
( E+ C2 ], o2 G+ H9 _ |6 @* C
return $ns;6 [9 K t4 C( v+ r- h1 N
}- i) {( M; |6 z- z
. |7 ?+ ^5 x$ S3 N4 j. c
// 返回数据
$ }% `8 ?9 {) @) f l$ Y) s) X# K
function send($client, $msg), g, @6 V( w5 Q- G/ x7 b
{
4 z; @8 o0 H/ G0 G; S& [. n
$msg = $this->frame($msg);
8 M9 r/ @, \: T
socket_write($client, $msg, strlen($msg));) i" t* f3 G* P5 t/ l) {8 D% a9 v
}+ Y* K g# r& S
}) r6 o! @1 `& U7 h: F
" R0 \+ c; z% M% n/ V
测试 $ws = new WS("127.0.0.1",2000);; g7 K7 G5 o8 O8 u2 o* L
# n4 J; P1 K: K

复制代码

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

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