PHP 简单实现webSocket

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

1）客户端实现

<html>
, ?1 h- K# X% _- ?& t# c- F
<head>
4 P5 Q- |% l; |
<meta charset="UTF-8">
( L. G! M, M/ W9 J, n
<title>Web sockets test</title>
3 r, p% O8 }5 ^$ _( W
<script src="jquery-min.js" type="text/javascript"></script>" }9 x1 d z" b" ?2 b
<script type="text/javascript">9 Y ?. X! ~! b a7 l! |( [
var ws;. o0 ~! [( z- h/ `5 L1 G. c" \
function ToggleConnectionClicked() { / j# T- x1 }9 x
try {/ o( H/ z" q" C7 D, h. g7 U- k+ N
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器
6 f- A. H& x+ h# z+ J
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};# i3 ~8 f. u# u, o3 N
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);}; j5 A5 X6 y# z7 ^/ V* B8 A( q: B* b
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);};
0 F* j- b9 X; V$ q. c
ws.onerror = function(event){alert("WebSocket异常！");};
6 I" g9 W8 r8 ~% T3 w( r5 m
} catch (ex) {5 i/ h( l6 }3 c# F
alert(ex.message);
8 }3 A2 D1 k# ^3 `8 @4 ]
}! b! |4 Y) d% C; H
};
: E1 j+ L+ n8 ^3 v$ }0 r
, u+ K8 @# _& z" E6 B+ t. b3 k
function SendData() {
' S1 v4 F3 ]2 x2 P
try{7 U- j$ k& x' s& |- _6 o' r
var content = document.getElementById("content").value;$ r" b# G6 N8 w- L$ w: d4 \( \
if(content){8 E# f, E* ~7 O" i r
ws.send(content);
' W- @) R( p% E" J% E3 X* L
}) U2 e. P+ m, f' Z: v
4 ]$ y' ~# w2 v- a1 z
}catch(ex){
5 L: ]- T5 a0 A B" E# f
alert(ex.message);
+ n! e" W$ M( S. B) |- d
}! | d4 T( i9 h* d/ \# n" V
};
( f s {2 _, c e% Q4 _4 o3 m
( Y/ u# h, ?& r0 n# P+ D( [# e' N2 }
function seestate(){
$ t8 O, I7 J7 P5 l9 g% N
alert(ws.readyState);* m. j3 y8 S8 H+ c. H* w
}. B6 ~1 p) ^' R6 |
2 o# z- Z$ y) I5 i9 c( C5 M4 c1 p
</script>
' B, H( Z+ q' L& s% X7 f3 t0 W
</head>) a5 @, X& \; Z% z6 e! C) G5 L
<body>. [# y, E! T0 K' w3 ]! T4 ]
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br />
- o. R9 {9 t! O# |9 j: R* b! r7 c# D
<textarea id="content" ></textarea>0 X6 t8 A5 d8 y
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br />
3 W9 B( h0 x: [9 h6 U7 p* U
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br />
3 j! r) ~; L% Y6 P, M) R1 a- n& f
* ~7 F$ W. K. i5 e/ e' ~ z @
</body>) n9 q0 \8 x* g" z% ^
</html>
; g Y3 f# L d& t- I! ~

复制代码

2）服务器端实现

class WS {
& B( a$ t6 K/ m0 ]2 k4 s }+ Z b
var $master; // 连接 server 的 client
: ^5 ^8 L2 ?( g3 o! Y* j4 ?
var $sockets = array(); // 不同状态的 socket 管理- \) F/ N: Q4 J6 ~
var $handshake = false; // 判断是否握手9 t2 \0 p7 c, o: J% J- T$ u, _
/ L3 ], t9 Y* {; p- I
function __construct($address, $port){0 @: O+ x% g( L9 J8 h
// 建立一个 socket 套接字
* {* y- x1 j: F. \& }$ c, |5 E
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP)
1 |& Z2 E- g" Z! a, r
or die("socket_create() failed");" C/ X2 F3 U, T7 V5 X. }( ~& E" ~. Q
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1) 1 y+ Z, A3 r- P, \4 M
or die("socket_option() failed");
$ N& D; v4 c7 R1 t
socket_bind($this->master, $address, $port) ! c O' O+ W2 ?7 L# y7 U5 D7 q
or die("socket_bind() failed");
; w% O9 T7 c0 H, Q& M: \! Q
socket_listen($this->master, 2)
; N# t+ t+ }" H0 b
or die("socket_listen() failed");
i$ a; {( t( }, u) Y- g
" J% z* p6 [# d( e6 T
$this->sockets[] = $this->master;) e; z4 r- y8 e; Y& Z
" J( U+ D9 ~8 A
// debug
% X; @3 e0 h5 W
echo("Master socket : ".$this->master."\n");. |* R. M, o" i) ?, j5 D b6 I
2 s" w3 ]3 m6 k# \1 o# ^
while(true) {- |. a6 N2 I+ E# g, @
//自动选择来消息的 socket 如果是握手自动选择主机" z+ a( m; _4 m$ B G* v" {* ]+ y* K
$write = NULL;
2 z; i$ E1 b* [- L$ H; E9 Z
$except = NULL;
+ E) L7 U! L! u
socket_select($this->sockets, $write, $except, NULL);$ S1 ~) v }% |- ?" B" U# k' Q
7 Y% R4 ~2 S/ e8 H! K5 v, x9 R7 |. j
foreach ($this->sockets as $socket) {. h2 K/ W! k5 p: H
//连接主机的 client 7 {9 g$ f; h6 w2 m9 s; H5 y$ g. g1 s
if ($socket == $this->master){- y3 E: N4 }8 n( J/ y6 g6 J. E
$client = socket_accept($this->master);2 _' r; v7 h5 Y7 w$ w
if ($client < 0) {
( G5 H+ C q9 i, N; P* z. u
// debug# [+ k$ V. h7 W1 ?8 q- I9 d4 J* B
echo "socket_accept() failed";8 D* r A. Y* u0 A @* Z( o" _) _! }
continue;: \; Y; U R2 j
} else {1 t; L/ v& d& p5 I2 X7 d' \! p8 X! y7 r
//connect($client);2 {5 m3 c, w$ E: J4 x. I5 V$ T7 Q9 O
array_push($this->sockets, $client);
! h2 [, M3 U! |/ q" P7 C% |+ x7 T
echo "connect client\n";
S: |6 W. p' F( T% F, u9 b0 M9 ~
}
! U+ o2 h$ w9 I! e K6 o. U+ P
} else { G/ u1 c# K B4 m
$bytes = @socket_recv($socket,$buffer,2048,0);( @& }' \, j7 n3 U
print_r($buffer);9 `+ U: y8 c. V t, F& j
if($bytes == 0) return;8 Y. a" u: U' V2 M9 T( j
if (!$this->handshake) {" P. U+ S$ O* Z- z0 P w
// 如果没有握手，先握手回应, ]6 Q- U H O* E/ o! r/ v: w; _
$this->doHandShake($socket, $buffer);
7 q2 I* x2 t' l8 b
echo "shakeHands\n";4 t3 h4 z b6 t( T
} else {, {4 `/ j# m$ P0 g* X# t7 ~' n
6 _. X7 x7 }; O/ P* k
// 如果已经握手，直接接受数据，并处理
# y; D C5 { W6 B& p
$buffer = $this->decode($buffer);
" m- [( w2 B* o/ I
//process($socket, $buffer);
9 o( x5 H8 [& C* I; f
echo "send file\n";
% ?& ^5 B, q, `9 i$ J* u
}
8 K4 Q7 U7 w+ B% a+ ^2 \, p$ u
}( p! `6 @% n. { g
}
5 {; i% T: r1 W, A- Y$ H
}
4 H0 V: ?# p- y4 s! C2 o; t7 y# R5 x
}
9 i# O2 ~, M8 I3 s% w. G, T
+ I. ^* r r! A: n- ^+ g; t8 d, I
function dohandshake($socket, $req)
6 ~/ I# Q) X! y
{
& p5 R, ]6 W/ K Q) r6 ~
// 获取加密key
, R4 S d2 ^/ j, z. Z! b3 s3 j
$acceptKey = $this->encry($req);
' l! W* b* z9 ~1 K
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" .
2 B' _1 C4 ~! x9 M% p% c1 u
"Upgrade: websocket\r\n" .
- C7 g; o, X3 E3 r8 V# x2 @' `
"Connection: Upgrade\r\n" .0 u9 @5 ~8 `$ H) k9 d
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" .
; A7 s$ R* F* N" x9 r
"\r\n";0 Z9 I8 z6 Q2 E% q( B: _
$ i, z' A/ O3 _7 F/ L
echo "dohandshake ".$upgrade.chr(0);
2 l$ P4 W0 N9 c: [8 C
// 写入socket
7 P* o U3 f8 I. ]) p/ w# u6 K
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0)));
! n! |/ `5 s5 i* ]" A
// 标记握手已经成功，下次接受数据采用数据帧格式( P, d$ E s: Y7 B* D1 s; F7 v" [
$this->handshake = true;/ e! c" b, m# Z5 V* f. m2 ~: K
}2 J' J' W4 c+ p( b
& M- [; c. H3 K/ }
8 K3 q+ ~+ K! i b: \- ~ ?
function encry($req)
0 @% \2 Y a0 \& X8 x9 h$ c: H
{- ]5 [; v* q4 p
$key = $this->getKey($req);; ], x' Z, ?0 E5 i1 N/ |2 g
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";1 P, k. Y4 r1 x
4 Q. t1 j5 s1 R. {) r9 ~/ [: R" d
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));
5 o, G" R! I D2 z7 ~
}& a0 T8 M* q* e) U
" \, S- m* _$ t# Y9 {2 w# y; d3 X
function getKey($req)
+ g0 N9 {! Y7 v& b3 d
{/ X, ?9 {0 u' d; C. z0 D
$key = null;
4 I& R! a- @4 e
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) {
0 Z. _; q4 T1 H8 B
$key = $match[1]; 6 I% d" Z4 S, B, e! z# y
}- t K+ q* I: B* ~5 z5 d+ Z a
return $key;7 X) m0 l; \5 ?7 k9 y
}# t9 b2 D5 U3 s4 P
) w3 i1 b* E/ |6 o" p3 \$ D
// 解析数据帧+ T" O7 z( a3 `/ o! O0 M
function decode($buffer)
. @# P7 C3 L7 \9 R0 x$ y
{
" c3 U* ]5 u# n9 t: Q* v
$len = $masks = $data = $decoded = null;: O/ p S/ ?7 S2 D( Z& L3 G) ~: w* ^
$len = ord($buffer[1]) & 127;
' ~9 A9 D6 r& w$ p/ M( r
% u) _) | n2 N% A8 ]- K+ [0 w
if ($len === 126) {
, _7 O/ C- m7 m1 O" N, S
$masks = substr($buffer, 4, 4);
7 v8 ^& N0 |" X \: x @4 k
$data = substr($buffer, 8);
: I# Z' u* n) p; ~9 a9 Z
} else if ($len === 127) {' ]5 a2 X3 S' c' e+ H
$masks = substr($buffer, 10, 4);5 g8 X' S3 r9 x- l* S+ ~+ V
$data = substr($buffer, 14);
0 x5 d( K# V( y
} else {6 w* f6 ^& t2 H! S0 a- a
$masks = substr($buffer, 2, 4);1 T8 F# }+ _; J* o- r
$data = substr($buffer, 6);- a4 n8 I! o! c5 U" e6 [4 j" X
}
) M( _+ \6 H7 `" g3 W) X4 o
for ($index = 0; $index < strlen($data); $index++) {: e r5 O% F( u9 c0 w: c( r: o
$decoded .= $data[$index] ^ $masks[$index % 4];- [9 L# Y, B- ]/ ]% _; A
}* a3 i! }/ l6 f, y. T
return $decoded;: t& L- o4 r; y/ P" I8 m* T/ N
}7 g0 i6 d$ l, `) o2 {( R1 g' {
0 U* n! V" Y: ]0 k- P4 X# u- [
// 返回帧信息处理
+ k3 b E- F" Z, A: ]/ O
function frame($s)
' b6 ]; X& S- s
{
" {0 y/ w$ r' V
$a = str_split($s, 125);
( s1 w' j- C0 u4 @
if (count($a) == 1) {5 ?- f0 ]# k) m0 L" o) d
return "\x81" . chr(strlen($a[0])) . $a[0];$ H+ e8 d- B/ r9 y+ [/ X, ~
}
9 F- A/ n0 ^! G4 g' t% J$ Z
$ns = "";8 f7 Q% `2 x/ d2 w3 j# b8 T! T+ E
foreach ($a as $o) {
, ?# J$ O8 \6 [ X* z; @
$ns .= "\x81" . chr(strlen($o)) . $o;, H: M) u5 b" ~- ~
}
( k& {7 M" F0 h3 l+ W1 ^# q3 c
return $ns;
. _2 \3 q3 l8 D" T" m: K: c
}
3 ?: G3 W% j5 W: |7 D% _
/ O0 _* x1 G1 c7 o3 l
// 返回数据. b& Z5 g9 p8 g
function send($client, $msg): }5 c. f- X: T0 v e! ^' j
{
+ Z9 E0 k6 R3 @; i5 c: b
$msg = $this->frame($msg);6 P5 d4 Y9 A- l5 |
socket_write($client, $msg, strlen($msg));
/ Y+ `/ R3 f$ p
}
. X7 f$ i3 p9 I. t
}
. B% r4 K) q1 r6 z2 ?% a
7 `2 q; W( v; e( v
测试 $ws = new WS("127.0.0.1",2000);! p" t- L: r# z: q) ?& ?: Z
; G; X7 L+ U( x7 D

复制代码

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

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