PHP 简单实现webSocket

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

1）客户端实现

<html>2 i7 d E F4 }8 O# c0 ]7 Z! K
<head>% u! J7 i/ |7 a1 T1 W" `7 O D
<meta charset="UTF-8">
, g5 J8 G$ N& L% L8 ~! A! |
<title>Web sockets test</title>$ ? @4 J) B* i& ?7 N
<script src="jquery-min.js" type="text/javascript"></script>0 I4 V3 J0 v) [& P( z" r( P \
<script type="text/javascript">
8 h- M& d. E( m# U/ F
var ws;) ?+ T+ x, u& h2 U* W# _
function ToggleConnectionClicked() {
" |, [8 o7 T; v- x7 Z& \. g
try {
* x. y) Q9 w' y1 i" F" G! I
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器 ) l* m, e0 d. J* e" r& U
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};
" a1 M4 v- M+ i7 ^
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);};0 N g7 F! p/ N; ?5 `; G3 x
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);};
7 C3 Z5 F0 [0 c3 X# x, V; M0 C& u
ws.onerror = function(event){alert("WebSocket异常！");};$ A3 L, A$ Y* p% Q) x8 M
} catch (ex) {2 g, O, h6 x. ~/ Z" G
alert(ex.message); & ~2 `* A- ^; p1 g5 f5 A
}
8 p; K0 N9 d! C. G! i8 K
};
6 |" B5 `- O# \6 w4 A, b }" w
( j$ j' h9 b3 }" y
function SendData() {; a' W, i* @! |; z4 T# w$ p
try{
$ r! a% C7 p) H& p9 w9 d8 j: [# D6 y0 @
var content = document.getElementById("content").value;2 O* ]# e! j+ ]9 g
if(content){* c0 N. H) f( A7 M) z- J" G' l0 R
ws.send(content);; b7 W) `7 O0 `; M: n# N j
}
8 a9 g4 r' e& h! a, U/ ]5 D
; c: F$ u; A* z4 i9 Z) F- r
}catch(ex){
2 a' J0 M: t+ d5 E. L7 i( C% \
alert(ex.message);2 N* A" G( h! \; b9 }* `7 K+ p& X
}6 ^, D" p* }/ a$ Y, w! Y9 ^# k5 U
};
& G* r4 b. {. m R, }
% r0 O9 {7 R) |4 l2 i, L9 m
function seestate(){' f C) }7 e* S
alert(ws.readyState);
/ w8 A1 a, }; [; S) V3 d$ P! }; J
}/ \( @5 O3 O7 f1 t4 u1 R3 u
7 h" I/ W9 M/ U7 e6 H
</script>
1 `7 M+ ?- M0 U0 d! { _
</head>3 V6 `) j: m# n$ [, R: Y
<body>) j3 R- q& q- r/ u1 D
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br />
+ e# j: E2 S" K1 g% M3 J5 o4 c% }
<textarea id="content" ></textarea>) Y6 C% [3 V7 w( k ^- v$ |* \! B6 A
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br />: L7 K% X, D, z+ v
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br />) Z0 V2 V4 t/ [/ A. W
* U0 q0 Z6 Z; X( m# f t
</body>) v2 O* I. g, N0 u
</html>" D+ B! \7 k) g" @7 e" ~! ^

复制代码

2）服务器端实现

class WS {
* Q! B5 F e7 p! ^
var $master; // 连接 server 的 client
. H4 j- j7 }$ e+ W0 l3 n
var $sockets = array(); // 不同状态的 socket 管理
' [ ]7 H* _8 z0 }. @
var $handshake = false; // 判断是否握手- W$ v3 x' C0 |( |
( u: f, p2 x* y* }. w6 L
function __construct($address, $port){
# |+ t* B3 Z( w
// 建立一个 socket 套接字
+ [0 p v7 @- _2 u1 N
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP) ( r/ ?8 C/ ~+ z. Y: K) b6 {
or die("socket_create() failed");. c W; t* M$ r4 J$ n; d7 f
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1)
) s& u' V9 e7 {3 I$ X$ U6 p
or die("socket_option() failed");
# y8 w4 v. `) s0 p2 q" [
socket_bind($this->master, $address, $port) ; k- D+ Z K' ?+ n% G6 Q
or die("socket_bind() failed");- n" v. z! G, O& C1 f
socket_listen($this->master, 2)
; P$ _; Q* e! J3 F
or die("socket_listen() failed");
: V: u6 j- r% }3 p0 s5 }. }
$this->sockets[] = $this->master;% D: l5 _3 _8 I) C# ^9 E6 R
: `& u: s& ^( }& ^+ |0 ?* U) t+ n2 c
// debug# g/ h$ m+ u" R) i4 \6 V1 G# P2 @
echo("Master socket : ".$this->master."\n");
; `8 r4 e8 H$ g9 H. Y! d
! S6 ^+ Z# _+ V8 h
while(true) {
( e% J+ @8 U I( N3 \% K. f& Z2 f
//自动选择来消息的 socket 如果是握手自动选择主机
! m& e% x+ D; \# ]* S
$write = NULL;/ v) r$ a* k+ O3 D* X% _; E
$except = NULL;
3 }9 B/ y! s/ e& I9 B
socket_select($this->sockets, $write, $except, NULL);
3 A/ L; d9 H) P7 g" ~. m+ X: _
7 L0 X, v4 F5 W# Z! ~ ?2 K" f
foreach ($this->sockets as $socket) {
& f# p1 e0 x9 w+ b. ?
//连接主机的 client $ o" d0 E1 ?" K! H4 y' F' N
if ($socket == $this->master){
+ }% R' L9 \: v5 \
$client = socket_accept($this->master);
$ U2 Q+ l, d& h8 c0 Y& m+ C
if ($client < 0) {
0 V; y+ q0 \: H( d1 ^4 {
// debug
# _; H c" s7 A" ]3 I, T+ b! A+ z! F
echo "socket_accept() failed";$ Y' `# d5 J o6 Y
continue;6 x6 ^( n& V8 D6 y
} else {/ k3 H% W$ \. d$ R2 f& o, Q
//connect($client);$ I0 e1 G+ y$ s8 m) n i
array_push($this->sockets, $client);
/ Y' A3 L/ P7 |, m( h
echo "connect client\n";
; ^) E2 J8 }: ?+ m. N0 G
}6 A# G5 x2 U) C `/ o
} else {" ?4 I6 b J/ X/ W M1 V, ?
$bytes = @socket_recv($socket,$buffer,2048,0);! r5 t% _ y8 B2 q# C
print_r($buffer);
% S* K3 L }, l8 E1 ~: m: [
if($bytes == 0) return;
5 F/ B' S7 h1 c3 e; j
if (!$this->handshake) {
" r5 q: @3 q+ K2 E- A1 f" d
// 如果没有握手，先握手回应
8 A1 q( S0 _) c) G: K
$this->doHandShake($socket, $buffer);& I3 M) i% e8 s Y
echo "shakeHands\n";5 \" e# z3 K! Q8 ^& g7 m2 Y$ U2 [& o
} else {2 A3 _9 O( a1 F6 o7 Q% \
5 F6 @; t8 e0 V6 [: g6 L
// 如果已经握手，直接接受数据，并处理 L- r7 L& c; \' V
$buffer = $this->decode($buffer);$ t8 U/ h- j3 L% W6 X
//process($socket, $buffer); & h) u2 }0 |, j" v0 z7 e3 s
echo "send file\n";% V8 w, `, {7 \' G$ [' t
}
" K' R/ s3 C& I$ e( L2 J; n( P% N
}2 I2 K/ c: x4 }9 |2 w
}
, {$ ~( u+ e: L' y4 k0 W2 r
}
# _; N5 m3 h( v# I L. O9 v
}5 R" [ D1 c0 O1 s
% n0 S5 Q' E3 j7 I8 R
function dohandshake($socket, $req)
3 D) b- E& l9 E) R8 Z
{
# c& Z$ E2 z1 o$ O
// 获取加密key
1 p+ @1 a- T7 [9 W0 I
$acceptKey = $this->encry($req);
; h/ s5 V9 l' Q) k% Z3 S; {0 `+ l5 _
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" .6 e5 w9 a) Z& x
"Upgrade: websocket\r\n" .
" ~ N+ d9 k! h6 n
"Connection: Upgrade\r\n" .
' h4 M2 s: G2 u6 j7 W. P
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" .: U Z* l+ a$ ~
"\r\n";
8 I( N7 I- u4 A g
: \) F4 M+ a K+ }+ N4 ?4 c. x6 _
echo "dohandshake ".$upgrade.chr(0);
/ P4 c( A: g1 s6 ]) r% `
// 写入socket; F ?8 D" R6 {2 S2 ?: @ f# L
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0)));6 H& E& O+ @; `) r
// 标记握手已经成功，下次接受数据采用数据帧格式2 }5 s: A, t% n# [
$this->handshake = true;
/ P D' D( |! a% D
}
/ ]$ C: s0 T( f& k! d
' {0 }+ V: L& h9 }+ T( u
2 q8 I) U" f9 ^" h1 Q4 C/ L
function encry($req)$ A$ W: V( H. \
{. G* [* M, I4 p7 l
$key = $this->getKey($req);
' h# w c9 T g' T6 B
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";6 c4 y6 Z8 Q) s) N; b5 n
' ?' M& w9 y7 x: E# d
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true)); ~0 ]" O+ J, b( X1 O
}: R" s# t2 v$ a
D/ c% Q8 l% I- l, s2 ?
function getKey($req) $ V8 ?) N1 l7 |" `
{
5 {% q* a# f1 {
$key = null;% g( j6 w+ ]! Y: c. K9 t
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) {
2 t6 q. U/ L0 W- `# Z: D
$key = $match[1]; 4 p/ E# [4 x! J* g5 t9 H( w E
}
% V5 s1 L/ H* r6 t. C" j
return $key;
B$ b; @, I; E, [# ^2 u3 x+ K
}: g& _( [ A- _" I
6 B, i* z# Y* p f
// 解析数据帧/ f8 g& \ h6 j# X
function decode($buffer)
* b& j: G8 z& h Q( m
{
1 y. g" |4 B' T4 c
$len = $masks = $data = $decoded = null;8 c! F0 e a' }6 q3 a- L4 c
$len = ord($buffer[1]) & 127;, M" P; O: o1 [2 a; V' L" ^) T
5 |2 S. s" s) e$ a. I$ n) j
if ($len === 126) {
6 Y; [8 }& u y( ?/ N* N2 _' w V
$masks = substr($buffer, 4, 4);! c8 m* U8 p F. _2 Y& e' B5 G
$data = substr($buffer, 8);2 S1 h+ e0 C! G6 }9 m9 F4 o
} else if ($len === 127) {% j8 h) W3 V2 \0 n
$masks = substr($buffer, 10, 4);# l: ?) P6 q4 F3 Y( _
$data = substr($buffer, 14);2 v$ G @& Q4 T( J1 M4 M/ Z
} else {) e7 Y% i* W8 Y
$masks = substr($buffer, 2, 4);# p* L6 `6 n4 o9 F
$data = substr($buffer, 6);3 A0 N$ C- f% Z
}2 j! q6 w1 U1 T% n! c+ C4 [# h
for ($index = 0; $index < strlen($data); $index++) {
! Z+ z. n0 ~2 k3 M( c2 t# c; X
$decoded .= $data[$index] ^ $masks[$index % 4];# v7 s1 e8 U( ?" A
}. t7 A+ V/ ]! c
return $decoded;7 _2 t% u, E# e% Q+ s
}
; U6 t+ A5 Z5 u9 I7 E% M& z
# x5 A1 N2 F$ g" D2 v
// 返回帧信息处理 X: b* S6 t n- h! c; t9 y: C: L
function frame($s) . H1 _2 n: o) ~. o
{
+ s1 P1 O1 b0 v8 k0 ^
$a = str_split($s, 125);
; Y% n2 j& P$ O1 `3 r6 b: ^' I/ A
if (count($a) == 1) {9 a$ T1 y4 u6 {, Q0 L
return "\x81" . chr(strlen($a[0])) . $a[0]; \ i+ t$ V5 ?5 l# U! ?; Z
}
. l. q1 L: x+ h M) V; K
$ns = "";
- k9 f6 X0 P1 ?) r
foreach ($a as $o) {
6 _: r8 M: `$ f: f
$ns .= "\x81" . chr(strlen($o)) . $o;
V8 t1 C$ Z# V( j# T' ?2 p
}' u. x, J( p6 w/ q/ i# R
return $ns;
& h) t; U2 e& v" S: H
}, Y( y' @7 x: ^7 \* z. L; S
) K. h5 q9 Q7 n3 Q. V
// 返回数据- I0 H6 ?+ `3 n" |' V& w9 t
function send($client, $msg)5 t2 L C3 F3 _$ R; w$ D
{1 \ `* u1 d* j& q1 }
$msg = $this->frame($msg);& q% k: s' x! W% i# B. M% q' J5 z+ g
socket_write($client, $msg, strlen($msg));+ Q7 f" t @' C: I; K x
}
" c" A9 s+ C! }3 M7 h
}
) Z5 l+ T- z) b& U
: M2 y( c6 B. D% t Z5 O0 u
测试 $ws = new WS("127.0.0.1",2000);" R0 E& M7 y/ S" b$ f6 k
" Y# X. D( v6 m2 x) s- \

复制代码

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

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