PHP 简单实现webSocket

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

1）客户端实现

<html>' E9 ]- k/ f- Z
<head>
( T* j6 g" f7 D) k0 Q% W
<meta charset="UTF-8">! c$ ~3 Q. \4 g3 U W6 J
<title>Web sockets test</title>7 l8 Z! Q- h' W, S
<script src="jquery-min.js" type="text/javascript"></script>
& o. ?% h% t( w& v' P! n1 {
<script type="text/javascript">
' W. L5 ] \8 Q6 G6 X3 k
var ws;
% ~ \ r/ q) ~5 \) e
function ToggleConnectionClicked() {
$ e5 h* }7 a& K4 g8 f. `
try {0 U4 N; _3 N5 ^' O0 }: U/ E2 x: [! F4 [
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器
9 b/ {3 K7 |& m
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};# |# o8 v: g/ h$ X0 _% }: v% e
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);};
$ L) P! _+ s. r/ k" c
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);};
) j0 {6 j& s$ P! ]9 _$ B% g; }4 |
ws.onerror = function(event){alert("WebSocket异常！");};* b6 J/ j8 a c f) ]; @
} catch (ex) {9 a3 [. h t/ ~, Q. W, N$ ?* {
alert(ex.message);
6 b& B: p3 R2 `6 F' W V
}
& @1 ~& h% F$ ^. B; O& }0 s8 r- j
};3 |: e: H* r7 `$ Y3 s- r
1 v g, \9 r; H% E; J; @! Y7 W7 Q
function SendData() {
1 W& w$ H) u5 |3 o
try{9 l+ ?- W7 _6 j8 b* k: I
var content = document.getElementById("content").value;
8 l* S9 c' z0 Z" o2 _) Q7 R% S- J6 i
if(content){! V3 M+ c" c; `5 ?
ws.send(content);
n% W, L- z# ]4 h) k# G
}9 M3 k$ f4 S5 \, T, K, e/ t4 S
) E" r2 R0 u3 L- w1 i7 n5 G) I
}catch(ex){
1 M8 {/ A* |5 Q' E( R9 I8 R. v7 p
alert(ex.message);6 k/ ~( v! f$ O4 ~" T, P* t
}
! {4 K2 I4 s' E* _+ q8 K. p; h
};% ~% P3 h( M% w* w1 K/ j. y @
4 f* B5 `. }) S d+ j
function seestate(){2 Z& o% k t; O; k
alert(ws.readyState);
8 C$ i e8 k# C' K9 r7 ]
}4 t& k: x! p% O2 B! [$ }
</script>" ^ U- G# U/ ~
</head>
$ d ^ M7 x- ]( E1 s
<body>
- k* L1 F( f! j. i8 c9 y {+ J0 h
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br />( ]' M2 I- v4 B: a: s
<textarea id="content" ></textarea>' a9 G/ `$ t; {7 p5 m0 W
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br />
+ `' Q7 V2 E0 T0 C$ l' x
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br />
0 ~' K1 c( e/ N6 f
9 o, V' t; \% U( T/ l. _) D
</body>
% a1 D8 G. _0 D& N
</html>
+ c$ j% \, y6 y/ M, V3 g

复制代码

2）服务器端实现

class WS {9 x! Y$ {1 P! e
var $master; // 连接 server 的 client
3 x. V- S7 o' S
var $sockets = array(); // 不同状态的 socket 管理8 m! m* a6 _" A; Z! _* P
var $handshake = false; // 判断是否握手% Y. q4 L1 B5 X! @
7 A" E5 E5 H0 d/ Z7 j4 U" q
function __construct($address, $port){
9 e: B; _/ n9 L, A5 C
// 建立一个 socket 套接字
! A% {7 M, V% u& G4 u( T
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP)
8 e* |) n7 b. N: d, @8 o& x& K
or die("socket_create() failed");! k5 n2 E3 P$ m
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1) 2 x% k4 B7 N r4 N
or die("socket_option() failed");5 O! S) G0 ]- b4 c% A+ M
socket_bind($this->master, $address, $port)
: o+ k* R9 l# P" ?6 S# G
or die("socket_bind() failed");
8 l& [ J+ c2 o! X: j5 l
socket_listen($this->master, 2)
5 ]: w: t& G7 i. w0 y
or die("socket_listen() failed");
p: R7 }% T, k; [! Z9 l) X
% n3 K+ _5 ~& d9 k8 w
$this->sockets[] = $this->master;
$ [. o. @) w" t6 A6 O/ ]# k" K
0 E$ c5 q# w/ Z
// debug* e* P0 {9 X g
echo("Master socket : ".$this->master."\n");
4 g1 E* I% n/ L6 f
# Z1 _: C7 U+ W: ?
while(true) {
+ W( P! F3 n- k
//自动选择来消息的 socket 如果是握手自动选择主机
/ F6 H7 z' m1 I. u" i$ x6 S( c
$write = NULL;2 R- _1 j6 ?- Z5 Q3 o- J) S# R
$except = NULL; {! ~$ G! T' h$ ?
socket_select($this->sockets, $write, $except, NULL);1 Q& u; Y% f; f$ p/ }
2 Y8 S4 j* y2 [9 X
foreach ($this->sockets as $socket) {; b, y3 i9 Z% a u
//连接主机的 client
" d: {9 [! A& d. F4 S( R7 M
if ($socket == $this->master){
6 f# `: {# P' ?2 y
$client = socket_accept($this->master);9 T6 u1 w" \2 {! L0 Z3 q
if ($client < 0) {! B3 _ Z$ m i* Y
// debug
3 o9 V+ E* P0 m( F
echo "socket_accept() failed";9 ?$ y+ `* `/ l3 p7 o
continue;3 w1 m( K ~6 F) ~
} else {
6 F4 u# p2 [1 K5 Q6 G, c& U
//connect($client);6 D3 G( z3 b4 W! E# b: O) \
array_push($this->sockets, $client);9 n9 f4 W& S1 p8 g# k
echo "connect client\n";
+ t1 F/ ]* W& l1 ~5 w( Q. {
}
8 J* _$ r8 ]+ E, W' b
} else {9 @3 V( P8 K9 W, `, ]4 A
$bytes = @socket_recv($socket,$buffer,2048,0);
. b# ~7 D9 \. \$ {
print_r($buffer);# r" G# W8 m5 d9 U% W% ]( H
if($bytes == 0) return;6 ~* i" Y, V" a( B$ F& R4 P8 e& [
if (!$this->handshake) {! F0 v# l' b6 M- n K; _
// 如果没有握手，先握手回应
- A4 t3 E# d" ~: }# K
$this->doHandShake($socket, $buffer);
1 a8 T% S$ s$ e4 d
echo "shakeHands\n";$ H: e/ b) L" U" W) ~
} else {5 i' c" \5 @) q% \9 `
# i& {6 q9 o1 J$ k) w* w; J
// 如果已经握手，直接接受数据，并处理
) I6 _/ s1 [. q" b E# h7 F
$buffer = $this->decode($buffer);# o0 A: q! u! j8 {: q
//process($socket, $buffer); - v- t" U) H& l/ M3 ~ F/ P
echo "send file\n";: B9 m. M. P2 u9 s" e& J7 O
}5 x6 \6 ?4 O( n* I6 M. u8 `5 M3 @
}
9 `' f* G3 E8 x
}
' r+ w0 H. B, r0 D5 e Z" F
}+ R5 U- S- C% i- ?/ i2 E
}6 ~! V0 P" w5 U
3 z; y3 \' x$ U# p) j9 d
function dohandshake($socket, $req)
" }& ^- u7 p R) s7 r2 p
{( o. ?" K1 {6 }# Q
// 获取加密key) u5 U+ C9 ^7 H# f4 x) w0 @ A1 T
$acceptKey = $this->encry($req);
& ^5 Z3 [6 `* W7 ?
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" .5 U' @" y- W6 @* \8 B
"Upgrade: websocket\r\n" .9 l; v- B2 d4 k% [3 R
"Connection: Upgrade\r\n" .3 d1 Q$ U/ d# c: S" D# C! }
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" .
& N. O; l0 f: @& a
"\r\n";/ x' B4 `6 S1 R1 S* {$ V
# d$ M- [! }8 N+ Y p
echo "dohandshake ".$upgrade.chr(0); * N0 Y1 \! H; L3 u0 R/ {
// 写入socket$ q$ c# ` G$ |9 Z
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0)));
2 Y. Y& l* D/ w. u
// 标记握手已经成功，下次接受数据采用数据帧格式
" X% v" L* [9 Q* O4 S2 Q
$this->handshake = true;
/ O$ s& o9 R. c8 v3 t8 \
}# r. M8 M% A( Z$ S
& D8 ]' d+ H- Y4 z
5 H6 b5 B1 J7 D% h+ ~
function encry($req)
q% G5 a8 g& C5 E q
{
! w% r) Q( u y6 K# Q
$key = $this->getKey($req); \! P5 o1 q4 l( i2 _& Z
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";" p z2 p! T; O3 s# {0 Z% O
7 y% N8 Q4 g% D" B) |
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));
" S$ n8 l) C7 _7 p% V
}5 u" b: K" V1 k3 k9 i
* `2 h& f# C7 O4 v/ H
function getKey($req) 5 m5 s; ~5 h/ @2 ~+ d" S G
{
$key = null;
. w+ g" E5 A# \4 F R7 Y% ]+ M5 {
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) { / R1 i$ q7 e% E4 p |
$key = $match[1];
6 p5 D; s4 m. f7 d. p C) D
}! @- { p" O4 G5 B }& E
return $key;
( s5 K/ [) D8 @, n
}2 P4 |+ g2 `; o8 x9 {, i
+ |; ]5 B, v2 N* H
// 解析数据帧: p$ T& v4 K+ R
function decode($buffer) , r3 C+ H( u8 l) W. x
{5 o; v" ?1 |/ r$ U, i! }, }3 v
$len = $masks = $data = $decoded = null;, D) b9 i1 t9 F$ s
$len = ord($buffer[1]) & 127;
' S% P8 `( z- v5 S' W
* }# J2 \& M$ ~5 x
if ($len === 126) {- V, K* {7 M! w% U1 V
$masks = substr($buffer, 4, 4);- E$ b/ H0 J( e- l9 n
$data = substr($buffer, 8);
' b" Y) j$ a" B/ F9 z( `
} else if ($len === 127) {& C- ]0 a4 I% i+ U
$masks = substr($buffer, 10, 4);# \# L& w) D% r/ @ q
$data = substr($buffer, 14);
1 H9 x0 U/ c% v( T2 \1 [
} else {1 g5 n" M9 N6 f# W* J2 p, N' O. }
$masks = substr($buffer, 2, 4);* P; D! `, F7 @$ Z
$data = substr($buffer, 6);# v& O H6 _. d3 Z
}
/ V3 ?, S% D- r1 K1 a7 t
for ($index = 0; $index < strlen($data); $index++) {
; ^- D: P5 O. {' Y R# L
$decoded .= $data[$index] ^ $masks[$index % 4];
0 l' j# s' i# Z. I! Z
}2 S6 @9 C$ R7 a" q( [7 I' t
return $decoded;
: I# B* |& ]+ _+ l. o+ p
}
6 `/ k2 S( x$ y, Z: M: _
) W$ o r: r- o( U1 E Z4 P% v
// 返回帧信息处理
4 A. B; c9 j8 P
function frame($s) # B; J# D" v% q. I# C( @
{0 F( ?3 ]* P. l* Q: c5 Y( I
$a = str_split($s, 125);$ }$ Q. j+ a# L: K3 C
if (count($a) == 1) {2 e8 S: B, ^+ r( J: o
return "\x81" . chr(strlen($a[0])) . $a[0];
6 }& f! P& N9 F' x9 M7 X
}: _& o% H8 X( O* b& }( a
$ns = "";
1 O }- f2 ]% @/ o! l
foreach ($a as $o) {
. f W. z3 S3 N- }+ J# H: ?
$ns .= "\x81" . chr(strlen($o)) . $o;
0 J7 j) h* ^+ H; X
}
2 ~0 |/ i. r+ l( t7 k
return $ns;
3 l. s* z3 X2 R$ S& A* e2 r; e! ]2 R
}
B3 G0 |4 b+ _7 o1 g
$ ]0 g/ }$ r4 u
// 返回数据7 h+ w! l' ~3 z% V2 C- h2 P
function send($client, $msg)7 r% k1 I4 D; j
{! a* n" ^% o! \3 K) S
$msg = $this->frame($msg);
% U5 u" q- [& `
socket_write($client, $msg, strlen($msg));& d, S: Q/ ^) p. ?7 |* y
}
# n0 e8 w! x5 V( Q
}. q+ p3 @& Q3 ]
( l5 z! S9 ~! p6 v5 s% t# u# U
测试 $ws = new WS("127.0.0.1",2000);
0 Q# T0 f" m$ Z5 x
; Q1 K1 F. Z9 D0 W8 K; ^

复制代码

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

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