PHP 简单实现webSocket

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

1）客户端实现

<html>
: \8 f6 L/ t: F" H: P9 s- b$ ?
<head>
4 i' I7 z2 W% ?1 ^ u$ n* I
<meta charset="UTF-8">
: \( z( y7 T+ `9 j- z) |: Z
<title>Web sockets test</title>
- n f% r! I% q' W
<script src="jquery-min.js" type="text/javascript"></script>0 n$ A+ P% }- [
<script type="text/javascript">1 J. `6 o" w) ]# a6 M
var ws;3 X! r5 _( ?/ ^6 F' U
function ToggleConnectionClicked() {
& l: L0 }* X; d. ]* h6 m, F
try {
& i' ~# ^8 }( W& W3 O6 {, i: O0 A1 X
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器
* M, M* c* h0 E0 q A
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};
5 y$ v2 N% q7 @% o! O
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);};/ Q" w8 P O' m$ v7 G# O4 j
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);};$ i* P6 Z3 [; p0 [2 b4 \
ws.onerror = function(event){alert("WebSocket异常！");};
" o# U8 J" U& e
} catch (ex) {
2 O; Q- E! X( G/ z
alert(ex.message); 9 e) H" P! l2 G q
}
! b* b/ G* ^8 t
};
- q# `/ }( E" q- u
( V9 O' `7 Q) A4 D/ L; m
function SendData() {
- v, z5 x! e( E* [
try{& q3 \2 M. f/ E/ y
var content = document.getElementById("content").value;8 v" L" D4 Q: h/ K$ x
if(content){* {/ b: s: h J* a
ws.send(content);
1 n- _4 i4 d+ z5 y k* N2 F" R3 V
}
& ]& c1 N) |- C4 {( C# n% ~
8 [9 _: r4 S, s' o# |4 d! g- o* Y
}catch(ex){
: M8 S$ G& e1 s
alert(ex.message);: b% L* ?& N& S$ u0 Z m, [9 Y
}
5 u! w7 V8 `9 Z% s
};
0 b6 [0 x" L- q
) w7 A& U" z9 H6 D$ J
function seestate(){. l8 S& a9 C5 I
alert(ws.readyState);
. ?$ q6 i# J* v0 J. x$ D8 c7 A
}
( f a, I8 N6 D) i2 d2 _
: G! l H9 U9 @8 F6 d3 e
</script>, ~7 T% b# Q8 ?% B5 W
</head>
- ?( Z; e- v) r0 V) ?: n5 w' A" i
<body>6 y' c5 \& n0 M2 r
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br />
8 e/ P+ e: _7 Q6 ] H
<textarea id="content" ></textarea>% P- O+ X1 r+ Q+ R
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br />
I: T! _/ W# F$ ~
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br />
# I- G) o8 O" K# q# j" e
, w6 a1 K6 v& k& g" B4 u
</body>
( M7 @: q+ ~( c8 o
</html>
! X; ^7 [: o# Z9 l

复制代码

2）服务器端实现

class WS {
2 f2 g5 v9 g7 o1 p/ \2 M
var $master; // 连接 server 的 client
/ H0 q( |1 e( X5 G0 i& h; l" E+ ~
var $sockets = array(); // 不同状态的 socket 管理
, K+ F1 ?: A5 M' d, n0 b# m7 S
var $handshake = false; // 判断是否握手
2 a. R5 ?1 j3 a
) a$ |" B& f/ { m) u5 F+ |
function __construct($address, $port){
1 ], k# d. d0 F5 \1 T: _9 G$ t
// 建立一个 socket 套接字
G2 K/ [' X& }* j$ R
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP)
/ c+ _& Q; N7 [0 [( C5 @
or die("socket_create() failed");
* x v4 m3 Q1 O3 k. |! P( U
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1)
, Z& A9 J0 ~9 r
or die("socket_option() failed");3 p& S# I0 R: w7 ]$ o
socket_bind($this->master, $address, $port)
5 u _4 D- a: H# P$ n: Z; H
or die("socket_bind() failed");
& k- ?; u& @5 K0 M2 M _6 P! j
socket_listen($this->master, 2)
+ I& F6 j& ^0 t# a# s/ h6 `( e; l2 b0 P
or die("socket_listen() failed");- P6 D" d# I0 Q o
" M( l# g+ l* Y3 a0 w9 q
$this->sockets[] = $this->master;
7 V- k6 d) L8 V& c2 S X( x; m
. e- m" e( ?, F5 \) O
// debug
p( L i8 R. Q, L; L
echo("Master socket : ".$this->master."\n");8 Z0 {) |' t: x! G2 z( w5 h0 _/ l
' }( e0 E6 u9 P% [$ Q4 m9 \! V' e
while(true) {
, Z& @) _) C2 t7 D, ^
//自动选择来消息的 socket 如果是握手自动选择主机) n: F0 J' B: e
$write = NULL;! j, M1 A+ G5 `
$except = NULL;
0 g: W2 v) L0 ^6 \6 `& M. | X
socket_select($this->sockets, $write, $except, NULL);
: X* z2 k- ~+ o9 @' x
7 Y H/ J2 Z2 e8 |% _# B
foreach ($this->sockets as $socket) {
. D% G$ k5 i- u3 K
//连接主机的 client
: P. @8 v/ m) |6 U5 G4 k% q! X9 C
if ($socket == $this->master){& ?+ m) K g: C
$client = socket_accept($this->master);2 r+ ~/ R' p. }" v
if ($client < 0) {
" U4 z( U6 x1 @7 q; P: E( f* d; O
// debug
& \& |; n2 n; E0 C6 }5 y: R% f7 @
echo "socket_accept() failed"; x! r& ]' E: o1 m, V0 l
continue;3 w0 U! l7 R1 T6 p5 @
} else {7 L( `% g7 X7 J# `, D _5 r
//connect($client);
& j) c% a) C) P# f9 T
array_push($this->sockets, $client);
) J( m' X" R1 N* q9 B5 z5 x" e
echo "connect client\n";+ a% j% `+ V# G4 g. b
}
3 {1 \/ M8 y# a& X
} else {
* N8 ]) w. {, l8 U4 y9 b
$bytes = @socket_recv($socket,$buffer,2048,0); f E; P- E( Z6 l4 c( o+ H2 f/ G
print_r($buffer);$ x0 D6 q" e! M- P& m
if($bytes == 0) return;, R+ x+ u8 l) Y) Y4 ^9 y
if (!$this->handshake) {
% B1 C7 ^% c8 j6 |. B3 B( `
// 如果没有握手，先握手回应3 \. T1 v) M5 S* m8 n, D
$this->doHandShake($socket, $buffer);
H( B6 Y, Q% ]* h, x! q! j C8 c
echo "shakeHands\n";
4 X- b1 E4 G# r7 Z6 Y$ ^
} else {
' ~% ~+ W0 H/ l0 n4 x7 _5 c& A5 o- T
& @ r8 v0 K" M4 @- f9 H
// 如果已经握手，直接接受数据，并处理" z9 D5 h3 \% U; D& p7 i! G1 q N) H' p
$buffer = $this->decode($buffer);7 c* N1 M, ?, q# N& {, q: q
//process($socket, $buffer); ' p4 h/ U# I' Q2 o9 }2 I: t3 x
echo "send file\n";) q$ ~! Z4 x7 l: W6 q
}
, X. U3 A0 E* |* U6 h
}
( {- P" [% X1 s+ p& b3 L
}7 W i/ Q" D& D" m
}% V+ }) \8 y2 M$ R( Y
}
! |! U" B$ t3 a( ^9 b( @
0 O4 Z' E$ L) ~$ _5 _& {! j
function dohandshake($socket, $req)5 X& M$ W! Z& l1 |- K
{" [" R: Y1 \, w% O* y6 M
// 获取加密key6 I4 E4 G7 F: a' o, h
$acceptKey = $this->encry($req);" x' A- V# C0 T6 s4 r j1 |! J0 c
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" .
9 ~% r1 e$ W. J* X
"Upgrade: websocket\r\n" .+ X5 Q, o' G( v; t8 b' Y9 b3 Z
"Connection: Upgrade\r\n" .
! o2 Y% m+ i9 h+ ~4 x* P' p7 n
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" .
) g+ ~8 K& l- l
"\r\n";
+ |7 Y6 o q! t3 Z2 ^' g
* x9 m1 [" S% y) U! n( `( @
echo "dohandshake ".$upgrade.chr(0); ! k! o6 J. J5 x( B- a1 S
// 写入socket
* S6 i1 J, Y4 g# L
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0)));
1 }* ]: G2 E# }
// 标记握手已经成功，下次接受数据采用数据帧格式4 h9 }: V7 h( R4 t# K% ?
$this->handshake = true;+ @3 K# p! o9 F8 n; ~. [5 Q
}( C5 u% s4 d1 Y2 s( n: g1 x: F
9 s3 c5 Z5 s2 D3 o5 V4 M
8 I8 D' L5 m _# q
function encry($req)
, V" R4 ]. k: M2 }
{
- {+ i1 d: ]. M! |! E
$key = $this->getKey($req);
' ?) ^; A; u" F! t/ F1 R
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";
L4 P" p1 R9 g: x5 t
; @. K. e" s5 `: ]# Z4 X$ \6 }# ~
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));# l) ^) ]6 q% f( I
}
# [% ]' p3 J9 h) A4 h0 F& `
7 f4 j _# P) s3 I
function getKey($req)
7 F2 Y7 Y& J! B
{0 V& v/ M0 Y7 x; l
$key = null;& Q! Z6 ~5 r7 q$ }* f8 J2 _) R6 u3 ~
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) {
, }/ w0 p' g) }6 X, d" v. f7 O% v
$key = $match[1];
* x5 C+ q' J/ E' X) P6 ~
}. V. F s* |' w, }' {
return $key;
4 D0 L$ _. c5 N/ u7 \& g v( p, @
}) J+ M/ i1 ?1 _* ]
& e) k" h" r' e8 K& m
// 解析数据帧0 d y8 u8 L& @' Z2 z
function decode($buffer) 2 Q8 e' M. q1 V; A' b
{- h Y: l( P9 @# P D2 I" Z$ p+ @
$len = $masks = $data = $decoded = null;9 f ~: O1 i6 V
$len = ord($buffer[1]) & 127;& D" O7 C# ^) Z& z* u. s# [
+ N) `7 \# z& e/ X
if ($len === 126) {% T9 V4 V4 @' j) D
$masks = substr($buffer, 4, 4);
/ y' n0 p# y6 F) H( ^
$data = substr($buffer, 8); a" t" |: H1 r$ x
} else if ($len === 127) {
, `. I: C9 z. }! Q9 G# p# ~1 z" M
$masks = substr($buffer, 10, 4);$ ~. m* y2 Z: y6 `: I
$data = substr($buffer, 14);
( Z% F0 o; B6 `& y" J5 ~( K
} else {$ m# `5 u4 W& R! ~
$masks = substr($buffer, 2, 4);
. n6 K5 k, @* X6 p. V
$data = substr($buffer, 6);
) A6 `3 K4 p1 v0 b) L" d! z) V
}
2 O; h. c# x* d: g* {. r6 b' F6 m
for ($index = 0; $index < strlen($data); $index++) {- i$ E. ?+ M. V! Y- ]6 ^
$decoded .= $data[$index] ^ $masks[$index % 4];
8 Y8 J# ^! U' d! w. h8 J
}
7 ^$ ^+ g7 O" h0 g( P& N9 u. v9 \
return $decoded;
$ {* g# N3 U* r- {; B4 ?# Z
}
# `! F! V1 ^0 v: E- `
2 a. D" q0 P# p- t! q; s
// 返回帧信息处理
0 u9 F( z, p- I$ M! S* i
function frame($s) $ K; S! ~" R& \
{# A/ J* c; r: S* }" S: i1 H
$a = str_split($s, 125); C1 D, L5 y( M8 m
if (count($a) == 1) {, x# S! J- \1 W& Z4 N* V7 j
return "\x81" . chr(strlen($a[0])) . $a[0];' S0 r E( C0 S1 M, X
}* \5 p, X, @# ~4 S
$ns = "";
, a, z0 n2 @3 r" N% M; v+ G$ v
foreach ($a as $o) {1 u8 {5 c( n! q
$ns .= "\x81" . chr(strlen($o)) . $o;) k' ]2 m: d+ c0 b& m
}9 a9 C) ?) b3 L% n
return $ns;
6 R( ?, b& R& D, L
}/ K. T, u3 \0 j! o! a+ P- y+ w
0 ~5 ?& D8 o$ _ o+ j
// 返回数据5 J5 h1 K+ f1 y9 f- @
function send($client, $msg), n& Y' ] J4 U6 h
{
1 O! k3 \1 F/ A% {4 J% N a+ P
$msg = $this->frame($msg);/ e. M" B9 X0 l9 u0 T7 H
socket_write($client, $msg, strlen($msg));: p! j: h* V: k/ S5 E
}) Q$ @! J3 Z |# A+ X
}
* I% h- O$ X3 p5 a, f8 o
; g0 D& e& ~& z1 Y
测试 $ws = new WS("127.0.0.1",2000);
( ~, o3 O, h4 r7 D* o
8 {& Z k. i }+ i% v$ I

复制代码

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

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