PHP 简单实现webSocket

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

1）客户端实现

<html>9 T) _8 ^; p. Y- ?$ t
<head>
) T* t. @1 t# y+ A* r5 W+ M. B7 s
<meta charset="UTF-8">9 J5 ^, x7 c; t; j1 r7 `
<title>Web sockets test</title>2 W, h7 P7 v4 f' X; h9 N! ]. V3 Z
<script src="jquery-min.js" type="text/javascript"></script>
' g \& O) V7 k
<script type="text/javascript">
1 ?4 _2 B7 X b9 _2 ^
var ws;- m1 E- s; @ i4 B
function ToggleConnectionClicked() {
$ {$ _4 t& @/ ~ A! y2 w
try {
- n: q/ W/ q+ l, @- S
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器
6 g# l6 P2 z4 }
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};
4 P6 t: u' T! P. Y8 D) m/ O5 _* J
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);};
- Y5 b! I2 N1 ^ s0 \
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);};' ?0 k/ a$ N$ K4 e: K: O$ c
ws.onerror = function(event){alert("WebSocket异常！");};' x/ F4 \% q6 ?! V
} catch (ex) {
& X. `$ b% ?7 u. E$ k
alert(ex.message);
5 E( |7 N5 R) I# Q8 K2 |
}
. c" o4 {; h) c7 O5 L, b/ p$ I2 x
};
X, S7 ~- X( N, A6 J1 o
* c3 R" d% m2 E$ ]* @' f, h+ U/ s# {: c
function SendData() {# S# r2 \6 t8 r8 V1 t N$ E
try{6 |0 D7 ?/ u: D+ h6 f/ |) l
var content = document.getElementById("content").value;. G1 O$ b. J, }* H3 Q0 R% I
if(content){
- w/ n& E5 g+ G) q0 w
ws.send(content);
1 v; q5 f9 C. |7 D' A
}1 t( Z0 g! ^, ^2 Y, g, @* h1 m
% \, e! b% Q" [% r0 Z! v
}catch(ex){
, P9 Y5 y. u$ L! q. @
alert(ex.message);2 U+ J% c( S3 o6 l
}# D/ S6 ~0 w4 a# ~% K
};7 S/ c% T+ q7 l* n. w
4 X. V# e. o4 [5 b# b' ^. S% I
function seestate(){
5 y9 R8 D- {/ x) P
alert(ws.readyState);
' \/ g! \" Q Q; l5 O
}7 I$ W+ \& g, Q1 w
' q$ C) |: ~, M/ D- K9 i; J* M( K$ y
</script>- K- _# r' K5 x
</head>: V) V: h1 c7 @; h- B
<body>
' R1 Z9 C) _; W9 x; e
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br />3 X0 I' J4 I, T
<textarea id="content" ></textarea>
, d$ ]+ t8 ?$ P4 o
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br />
1 e' G( b7 y1 E0 ]# K" I
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br />/ a( h: \% I) z6 ] O$ m
& `( a% U: W( P# C2 C$ K2 b3 V, [
</body>! b! P3 n. N( u$ m
</html>
8 t. H6 T I1 f ~4 y( k: t

复制代码

2）服务器端实现

class WS {; Y1 j1 S1 ]9 U- F
var $master; // 连接 server 的 client; A9 E. {+ h: K0 `, C
var $sockets = array(); // 不同状态的 socket 管理
' h+ Z8 l) `1 N0 } ?; y" c; b( G
var $handshake = false; // 判断是否握手( g2 X0 M" b7 M- S( F
- z7 g0 q. |& P
function __construct($address, $port){/ t- g' k1 | u8 y9 z: o- i
// 建立一个 socket 套接字0 i4 F% H( H& E
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP) / P) I- M9 x9 G1 N n( x( ~3 j. M
or die("socket_create() failed");
: M4 E1 Y( t7 [' w. \) b
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1)
# H4 W# e/ z, d3 `
or die("socket_option() failed");! P; f3 O5 T1 i0 j& c& e* m ^* p
socket_bind($this->master, $address, $port) 2 x0 d! c O5 g" C& Y
or die("socket_bind() failed");! I. Y+ O% Y) S% N- J
socket_listen($this->master, 2) $ {! V! K( {$ ?4 }3 J
or die("socket_listen() failed");. b# V3 i( l1 i( g$ l
1 R) c1 H) y0 G1 ]
$this->sockets[] = $this->master;
0 [" I, t$ C! p3 \# R6 S W9 p
% k3 ]; y) y3 B8 _5 ~3 i8 Y
// debug0 q* R3 W3 j# Q" O( O- p! A# ^. ]
echo("Master socket : ".$this->master."\n");
8 W8 K2 W6 B9 ]- j. N
: N& H8 \7 i: D* a
while(true) {
1 d1 c3 {8 w+ S! e8 W) U
//自动选择来消息的 socket 如果是握手自动选择主机
7 B- {( ~3 i3 u5 D9 V: ^: Q
$write = NULL;
- Z& T% w2 V( a1 K7 `
$except = NULL;8 U A* w! u7 r S4 R
socket_select($this->sockets, $write, $except, NULL);
9 s7 D- B$ t6 P: \
$ s3 q7 W, V" y! ?
foreach ($this->sockets as $socket) {
' D) ^. `) ~& Q8 I* u
//连接主机的 client
. q. T* s% c$ b7 `, T
if ($socket == $this->master){
* s2 t8 O; m' `
$client = socket_accept($this->master);
3 p9 A5 ^# |- v
if ($client < 0) {
r1 P( ^2 m( s- h0 m
// debug6 J1 f2 j) u/ M! o0 n$ y. f4 g
echo "socket_accept() failed";$ p; g* f/ o1 _5 }1 h& r) O
continue;
3 ?( ^: D6 A8 R1 a- l# B
} else {
7 ]! ]1 p2 M( I+ o! i6 n( }
//connect($client);9 F0 G: W4 X5 a
array_push($this->sockets, $client);
% s% \8 U) p' V4 @* A1 `+ Y6 D6 x
echo "connect client\n";" _9 ~8 [7 Q" H$ q# X; i; e6 [# H
}
. L, Q( c. x# p; \- ?; B* V
} else {1 U/ ^& ]7 T1 b* P
$bytes = @socket_recv($socket,$buffer,2048,0);$ P" Y+ `" n( R) |+ m! y
print_r($buffer);9 e, K9 k: o+ Q! V0 P2 @% V' w
if($bytes == 0) return;
* G* I2 |$ W& I
if (!$this->handshake) {
4 [2 ]6 H. S! @; {
// 如果没有握手，先握手回应
6 ?7 a' m0 w, T. F
$this->doHandShake($socket, $buffer);% Y% R. I. C+ z" M+ e
echo "shakeHands\n";
6 p; l3 A( K( J V8 w5 Z6 A3 Y8 R
} else {8 f) M1 P2 g8 Z
b) i" ~% W$ _
// 如果已经握手，直接接受数据，并处理
2 c* \7 S( w. ?+ w% T+ b7 G' p
$buffer = $this->decode($buffer);
4 O* q' X7 x) y: I
//process($socket, $buffer); ~5 {* A$ g5 Z' c
echo "send file\n";3 b6 ^; ^* L# ^1 i3 ^2 g
}. n- @5 A/ N+ U, W9 I* U x) N$ `
}% h3 p. d1 `, A9 F3 `
}
; o2 q! E) K8 i
}) r* f2 m- O6 B/ p8 D
}5 `/ ?6 J) D( l. u- @/ C) e
0 @( M H$ k4 r8 l8 H
function dohandshake($socket, $req): z) y& m. i6 \. }- L1 L
{
( b) N* ` g l' ?
// 获取加密key; z( C H$ _3 D& k" f! }% o; X7 {* {
$acceptKey = $this->encry($req);
5 A( S" u$ ]( a$ Y& P+ q
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" .: C0 w8 f7 Q. B
"Upgrade: websocket\r\n" .1 |& T4 O5 J( O& W
"Connection: Upgrade\r\n" .
* M$ Q% x: T( e% \' d/ l
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" .; M! R( ^ K" o; Z# Y* U$ _
"\r\n";; \5 v- {( e2 T( d# G
( e, k2 R, D, W5 j; e* v
echo "dohandshake ".$upgrade.chr(0); 3 P! u! e: i$ l% w. G Q
// 写入socket1 c6 ^% i6 a: L% e+ H
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0)));2 s( T0 W3 x: d, B, ~) E& n
// 标记握手已经成功，下次接受数据采用数据帧格式4 s% r$ T+ K7 p; J
$this->handshake = true;
2 R* [2 @ [$ q+ s5 S
}
( o0 ]% A( U4 ]' I8 z; ]1 u
. G1 U W' A% u3 J" `
* H7 L; C2 q2 F: h2 E
function encry($req)
3 A; v0 l; g6 S
{
: K9 Z7 A, M ^) d- L- D6 e
$key = $this->getKey($req);
% U! I/ r- S, L! b& F2 E! U
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";
6 ^1 _' t$ ]+ f
) O) i! w) h+ W) i: k; e$ h
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));, ~4 a- A( {/ _& z# m, l. v) h- T
}
; ]! ^% m% `5 L* R8 e% M: X/ f
. D4 f4 t- X2 g2 w+ B- j y* {
function getKey($req)
: p2 z3 n9 ]2 j7 p Q
{
! v8 R6 E4 n/ p$ U
$key = null;: s+ r# x' g! t
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) {
6 P0 Z! t0 L% c. m' {5 }
$key = $match[1]; 8 @4 N" T. \- L$ N0 _3 f
}7 v% I; E0 n" c1 j4 g8 W7 ^
return $key;
. Q' M9 Y6 g( F
}( `8 x" ]! O% e1 t3 S+ ?6 l: y3 C! W
3 }( H2 N; o7 Q. X
// 解析数据帧
% c$ H+ B+ a! J) D3 E, d. P# V4 u
function decode($buffer) % ^8 D0 e; q4 i4 f" a U1 g
{- T" v3 f4 b; s9 w1 W! W# ?
$len = $masks = $data = $decoded = null;9 y8 B5 [# s2 i6 s; s. o
$len = ord($buffer[1]) & 127;
3 w# L! l" q2 W! \# f4 N) X" n
7 ^/ [( w. s7 ~
if ($len === 126) {
% B. k8 N, C# Y1 g
$masks = substr($buffer, 4, 4);
5 r6 K# h$ ?+ ~- S* |+ L5 {
$data = substr($buffer, 8);
0 `+ p, n! @) j/ o8 J, H6 ]3 p% F
} else if ($len === 127) {" W$ w4 O7 K3 Y8 @+ H* o
$masks = substr($buffer, 10, 4);
3 A# B+ }- D$ K7 q V3 Y: Y9 |
$data = substr($buffer, 14);+ B4 T$ z3 A4 J( b x' B4 T3 i
} else {; m. V3 E) B3 ~0 i( w( Z" n
$masks = substr($buffer, 2, 4);1 C' \2 o0 a; Q3 x
$data = substr($buffer, 6);4 e4 W2 n. ~- w7 i: p! ]% f$ @
}
3 u# p4 O" B! U& j5 c/ I
for ($index = 0; $index < strlen($data); $index++) {
6 g& W% e1 x8 B ?! k$ |
$decoded .= $data[$index] ^ $masks[$index % 4];& _3 h6 T% u5 Y8 x/ I" Q2 _1 a, L
}# w; D& K- d2 C; W: h& ]
return $decoded;, u' ~( n$ F' z8 D
}
& A- _7 \/ U/ E5 o0 f$ a
, j" A; u4 ^5 p7 l- A1 V8 H" o* D3 V
// 返回帧信息处理
( a0 V' B) w4 n+ t
function frame($s) 4 F) s- w! ?5 M/ t
{
2 _& i4 z1 T0 u: L
$a = str_split($s, 125);
, c5 c1 A( e, v
if (count($a) == 1) {
, S/ {) s$ i2 t( q ]
return "\x81" . chr(strlen($a[0])) . $a[0];
_1 a. G$ Q- Q% a* u" K
}5 f0 m6 ]" N5 ]7 C
$ns = "";0 w, \, m4 N+ F% K
foreach ($a as $o) {
, | n N* M, B" s( W
$ns .= "\x81" . chr(strlen($o)) . $o;- Q& I# Z! R2 |2 x2 I) u' Q( e
}
" e* _" d: |& U, ^; p1 b
return $ns;6 T; T* u1 v& q4 M& A; a
}6 I% @& @& N* u0 ~ e6 d: G$ N
6 f/ p5 K5 [* n" g0 ^/ |# k
// 返回数据3 }7 Z/ @: j$ n( k7 N/ m$ J0 f$ O
function send($client, $msg)
. s1 D( x* s( p9 m0 U
{
+ f( O! o4 \1 O) F( r) J
$msg = $this->frame($msg);" u( j9 M5 K0 \ X4 l/ f5 B d5 ]
socket_write($client, $msg, strlen($msg));
9 e8 c p" Q/ d
}
4 Y# m. X7 [2 P. Q: D
}
; J4 E' l# U6 Z4 |) Z0 j+ S4 \
: `7 V# r! S* Q% L2 d
测试 $ws = new WS("127.0.0.1",2000);
0 Y8 N. |- J/ x7 j5 F; |
* ^# e4 M: M! y: R. j( O- E

复制代码

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

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