PHP 简单实现webSocket

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

1）客户端实现

<html>5 ~/ u0 v# u# v# g& C7 x {
<head>
* S$ b9 @6 w; ^
<meta charset="UTF-8">' q h: r' O8 l4 I2 t
<title>Web sockets test</title>
3 U) X& I+ q, C0 J6 |3 p
<script src="jquery-min.js" type="text/javascript"></script>0 p% y+ {: {. h/ ?% h
<script type="text/javascript">4 I# g! \6 y4 L
var ws;1 S" X8 s# l/ \% _* A# z* `+ a I
function ToggleConnectionClicked() { 6 c) _7 P3 x5 U! @; ?7 U) Y, Y
try {
5 b D7 K. `' R' m! J! L" G
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器
+ r( o$ l2 k5 b* x
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};1 l+ c# _5 m) T& Q3 N3 u, V
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);};9 _- O! p& n! Q! h/ P
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);};
% f. `, {6 V( D# D1 E
ws.onerror = function(event){alert("WebSocket异常！");};
$ G! C1 U ]7 L E, x
} catch (ex) {
4 q2 p# H* [( F0 ]: _4 d8 ~0 X
alert(ex.message);
( l( |1 H" A U6 _
}
- R/ i5 h2 \0 b8 }1 F/ y4 H
};
+ |: u, \5 i' b& v3 |
0 x. w7 t+ ^" ~! c/ y8 x" J
function SendData() {9 M9 _4 D% G6 y; w/ k. @9 a
try{
% e1 c1 m) I: _+ e! B* r
var content = document.getElementById("content").value;
* g. I7 O4 w, D$ b+ p
if(content){
2 |- f/ I" L! V0 ^* e% W1 f) \
ws.send(content);' T5 G9 P0 C0 c. M
}4 d8 e l9 n/ a0 ~) B
* B2 [7 ^- R' [( t5 T
}catch(ex){8 z# `0 k0 @+ Y, {$ q1 I; q- W
alert(ex.message);
( i O+ o6 h p5 \. {4 M u* |& A
}
. v$ Q8 T% x5 B1 X0 E' J% E7 c! H
};0 P4 q# o, [. d+ i2 B; s
, i2 _) B3 o( J2 R
function seestate(){
+ |( \% y6 q: e- i7 H
alert(ws.readyState);
! x; W' T `" P) C. N h. L* w" F
}& P* e. I1 |. a
, B4 I( Y, G5 j. A8 ]+ J t. q* m1 c/ v# @
</script>; O) F0 |! h! T _% ~; C/ a% ?
</head>, \8 @- _# ?3 \6 a
<body>0 r( `/ I8 {% M U. N: C- N+ a7 }7 R
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br />: D7 T+ {. R# Q0 J) C4 X8 ~
<textarea id="content" ></textarea>9 c9 |& X+ I V" l+ t1 J7 h4 X
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br />' ~7 e3 e5 f( P
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br />
7 t- W+ Q) s( h
/ ^$ S! ^7 E0 w* @% }; n
</body>
1 e8 G: r9 Z/ V! |
</html>
+ ]3 q5 Y M7 V; o' E

复制代码

2）服务器端实现

class WS {9 S T- ^/ c% c' a9 T" [
var $master; // 连接 server 的 client
4 c0 S1 \2 ^0 f& ~" T
var $sockets = array(); // 不同状态的 socket 管理
( e' v9 n5 k; K6 d
var $handshake = false; // 判断是否握手
! P$ b5 N# D' c& A
# ]0 A: A, m: R; @/ S! w
function __construct($address, $port){
& ?" t& x; i% [7 m6 ]/ n9 P
// 建立一个 socket 套接字
4 Y. [, ?# [8 A8 z9 `2 d7 ?0 D$ Y
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP)
3 o( e4 Z7 ]- y) ?! p
or die("socket_create() failed");$ x0 \# Q- O+ L7 U
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1) # F- [. A; b/ Q& n8 f, Q! V) F. s
or die("socket_option() failed");
& I; O; t5 A6 Q
socket_bind($this->master, $address, $port)
& ~( V8 h$ h% F% P3 ^* N7 k
or die("socket_bind() failed");9 U; W( g8 H7 p8 I% N
socket_listen($this->master, 2)
' B. u. R4 h" [/ H/ |8 {" A/ ]
or die("socket_listen() failed");% [( C, D( d; g+ v
3 M9 \* R* R, }+ O5 J; }
$this->sockets[] = $this->master;
7 {' B( t' W7 @) F0 y7 z
" Y5 ]' d2 E# O# O7 [0 _
// debug
- G \, T7 e' I
echo("Master socket : ".$this->master."\n");. [: t9 O+ W4 @( x; H4 N. S( Y
4 `) ~2 \0 R( @- ~7 n( X' B
while(true) {5 r, @7 ?! M: {' h. J
//自动选择来消息的 socket 如果是握手自动选择主机
8 ^1 N0 x7 F' g( u/ t4 v/ t
$write = NULL;
: D; R H9 L4 P3 S3 q
$except = NULL;
- c) Y' B% ?9 D, }: z
socket_select($this->sockets, $write, $except, NULL);
, u* ]! n2 z: v9 U6 }0 {% `/ r
( v7 n( ]7 @% h1 q- j4 }
foreach ($this->sockets as $socket) {
& e4 h6 g7 A! s0 e2 [4 c, J
//连接主机的 client
0 T" x% T1 `* d* m
if ($socket == $this->master){! o3 p* ~; M+ m) F3 a$ L5 j
$client = socket_accept($this->master);# {6 ^0 i3 A$ J
if ($client < 0) {2 Q: w, m, Y. A" H: _2 T
// debug
0 t; }2 q5 Z5 [5 ^7 X; V
echo "socket_accept() failed";
% y7 f u5 @4 L' j x+ F
continue;
! Q' \+ |0 ~& t) j
} else {
: n( L& D4 t5 E9 X- [: {, K4 \
//connect($client);3 |9 L* M2 i3 G) b; s
array_push($this->sockets, $client);
' w( R% }5 ]4 a: T& H) g
echo "connect client\n";: b/ {9 G/ s# c& p$ M& r1 g
}
; a% d6 j5 p" _0 I+ J6 W) K
} else {
4 J4 A( p& y: _+ M% @* M: e3 G* u
$bytes = @socket_recv($socket,$buffer,2048,0);& D4 a! D- f" ^& g- b
print_r($buffer);) F1 ~2 \1 |' O+ C
if($bytes == 0) return;
$ P0 m1 M: F, F& i
if (!$this->handshake) {
$ H. H- s6 c# V
// 如果没有握手，先握手回应2 T! Z$ z4 V* \- ~
$this->doHandShake($socket, $buffer);
/ y8 Q/ |. ]( O7 [% q
echo "shakeHands\n";# A+ `# f* i- N" E/ a$ \
} else {! I; u$ e# U6 C% r4 ^
/ O6 m2 `0 v, E. q* ?4 R
// 如果已经握手，直接接受数据，并处理7 u( [3 }! i" G
$buffer = $this->decode($buffer);# R1 |! f) n' O7 z1 W/ `; F
//process($socket, $buffer);
$ [- Y& @$ {4 u' p) ~( B- [ k
echo "send file\n";% g, V, H% V# l- r7 D1 s$ Y
}
( y0 A, Z: h/ `& T1 Q, m
}+ Z* N# q5 K0 c( s2 g! @8 g, c
}
/ |1 J* o/ @1 o) q9 T* D/ [
}
' h6 H3 m+ j8 ~( j: J- S; Q; n! k
}
/ [ Q; I/ c3 M6 L7 Z4 F" Q
& @8 u$ ]; F7 X% s [ V
function dohandshake($socket, $req)2 Y3 y s. v+ k
{
& c% S3 x( d' ]2 j- P
// 获取加密key. M4 ?& p' ~# u; ~
$acceptKey = $this->encry($req);
. A0 g: D# k, a9 y5 g. I
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" .) V4 F) t3 k! D. R5 _1 v I4 d
"Upgrade: websocket\r\n" .
& }5 j% C/ @2 q/ T9 m8 {) @% w
"Connection: Upgrade\r\n" .% g5 |+ u# S! l0 w$ C
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" .
' Y6 Z9 d9 `# @; [& J6 |* V/ P4 M
"\r\n";* m% D% M" q6 R$ L( G
9 t& P3 A9 |- A* [' T
echo "dohandshake ".$upgrade.chr(0);
6 L0 E3 k: l* M3 \% U J7 z, r2 Z
// 写入socket& w: G$ i' n. p) }( F- @
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0)));4 a: I" ^1 Y2 R3 V# S
// 标记握手已经成功，下次接受数据采用数据帧格式4 P2 }1 c- Q* l z' w/ S8 w! F6 K. c
$this->handshake = true;
" g; u( B2 c. q3 X
}
" P# f! f' A3 |; i$ I9 W, X3 C4 s
3 f! n) ?* o; I$ v1 h
- X) Z! V# x( E7 [- m, |0 ?& E
function encry($req)
- v1 U* ?' P, i3 \3 y' s
{; D- u9 {# U7 Y# b) V9 j' b9 v- F$ q
$key = $this->getKey($req);0 k- T9 M9 q; E! u; `: r
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";
. e7 z) u' U+ ^" m2 r* A
2 T6 I+ [5 Z# e$ ?
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));
5 B, m/ L1 d# X) J. B( L( ?) K7 v
}$ m9 a. u2 G+ P% B' O- o
! n& P \& q: @/ W! _* k
function getKey($req)
: a4 I: [; D* D! X' |
{( C8 w2 j1 L" r8 u- L! L# J/ q: U
$key = null;
2 Z7 Y: G% o2 n5 s# t( P* N/ G o
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) { W6 k& ^2 v4 v1 A1 L! x
$key = $match[1];
" Y9 C1 Z9 d- c% s _7 z, \
}
7 g( [8 a' X) C6 N4 S( `
return $key;: z" ~+ \) A3 A6 r( w/ Y
}
1 n6 q6 q9 {# [/ o J6 S
, C8 O6 C) ?9 ?7 V
// 解析数据帧0 j% P, ?6 X+ V) H/ R3 j1 d, Z
function decode($buffer) - W7 ?2 W" }- F; V
{" s6 T0 B+ V& N5 w# @. g9 }
$len = $masks = $data = $decoded = null;
- ]. X, w- Y4 S' E2 p2 X, {- U
$len = ord($buffer[1]) & 127;8 H) i2 h# I$ d: D1 e7 k, t" c7 q1 R7 p
! A& F8 Q& I- X+ w/ o* F9 P
if ($len === 126) {# o$ ]+ b5 v% z" [! I1 k4 w p
$masks = substr($buffer, 4, 4);$ F. _$ }$ ?( q- G. p4 Y$ D" V
$data = substr($buffer, 8);
8 B4 j! Q) d/ x9 P, `! G" @
} else if ($len === 127) {
- q0 B& {6 L# I6 u" _! M! w% v
$masks = substr($buffer, 10, 4);
8 C$ N8 W5 j1 Y) ^
$data = substr($buffer, 14);) e0 b: I6 F5 y0 f: z; L
} else {) U- B4 }8 M% \# W J
$masks = substr($buffer, 2, 4);
0 ^( j' E" H ]/ g+ M; B3 Y* o
$data = substr($buffer, 6); C6 z# N4 P0 ]/ b
}8 j! m5 t3 l9 F* V) M+ e6 @
for ($index = 0; $index < strlen($data); $index++) {
7 k( B, j' O8 t; V4 E- X1 s
$decoded .= $data[$index] ^ $masks[$index % 4];
3 S7 K$ u2 x3 ^/ l
}
1 j `, Z& w2 q9 \7 E7 @
return $decoded;
7 {3 Z9 |$ |1 j
}
2 @4 q5 s1 S& q* `' U5 p) k
1 f* K. h% Y0 w5 D
// 返回帧信息处理8 u6 ^: r1 J# Z
function frame($s) q6 k \2 l/ }1 K# z0 g
{8 y8 v( V4 Z0 c! g
$a = str_split($s, 125);, |0 O5 R% U, _: X3 y. X
if (count($a) == 1) {! J# I5 r" C/ K
return "\x81" . chr(strlen($a[0])) . $a[0];
i& Y, S' }: s4 ]
}
$ns = "";+ _' h1 c. t1 t/ X
foreach ($a as $o) {
& T6 ~0 x" a4 T
$ns .= "\x81" . chr(strlen($o)) . $o;& U, v4 \5 ^2 T, O3 V
}
% g% R7 L" q: b% h
return $ns;# c+ z* O2 n* ~* D/ ^8 X+ j
}
0 ^3 H) C6 P& \! s6 l
* a* \+ ^1 y2 u$ Y: r! [' ?+ Q
// 返回数据- h! t0 \" @0 ~4 o2 Q0 N
function send($client, $msg)
8 w* ]6 l4 F: E
{: A; d5 B2 g. o( p
$msg = $this->frame($msg);) |7 n5 L2 L+ t9 a7 N
socket_write($client, $msg, strlen($msg));
7 X) z7 h, o$ F" K9 W* }. V: I
}+ A5 h& f+ `+ U7 P8 T
}
9 ]( d6 k( m- v1 r/ S
* n! E2 j3 S! o0 ~
测试 $ws = new WS("127.0.0.1",2000);
3 @' i" D9 {: p
% w4 l% E7 W( \, a

复制代码

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

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