PHP 简单实现webSocket

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

1）客户端实现

<html># J! O: `0 c) N2 Q+ P
<head>/ D$ O, i! |+ {- B, ~( Y4 P$ @
<meta charset="UTF-8">6 e) r/ p' ?3 v+ g3 e) o7 O0 J
<title>Web sockets test</title>( I4 \1 F/ U. I
<script src="jquery-min.js" type="text/javascript"></script>
% f- p4 K; y$ e/ B
<script type="text/javascript">/ W$ a# K( N2 l Y) x4 r. @: y1 S
var ws;, W0 W, S1 E6 [/ a# F4 M' k6 O
function ToggleConnectionClicked() { ( Z2 `! C3 }$ X: r/ L- Y& f6 G
try {8 a o6 [, J% |9 q* Y1 v
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器 ! Z; A6 K8 V5 h$ o7 M' X
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};9 ^* d4 t3 W/ \' U4 e
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);};9 u0 n4 L2 s6 m0 N
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);};
5 L: l! f5 G N
ws.onerror = function(event){alert("WebSocket异常！");};0 h. k6 z5 g. [1 v
} catch (ex) {
9 l% I7 e# d$ z' x5 [
alert(ex.message);
$ y* r- }) e7 I8 r, e
}1 [9 z3 x* d: k" Z2 c8 n/ `! ~7 J
};' X: c3 I. @+ B& p+ ]6 |
" g$ T9 V8 {: U7 H( D
function SendData() {
' {5 e+ A' { f+ `) c3 P
try{
1 m0 n7 `) F' T$ q# j2 J
var content = document.getElementById("content").value;
3 L' B o* Y" N$ ]0 g6 N9 H
if(content){
$ w! W0 Y: V7 S' e3 K1 s
ws.send(content);6 A( p4 Q( c7 } |: l8 q( G: ]( c
}7 t+ @$ r# J: N% ]/ Z" ^6 c: D
t$ y& u/ B- c9 p9 J R
}catch(ex){$ t2 `2 ^/ t T5 q$ l! n
alert(ex.message);' y. [$ Q, W0 Z; r2 _
}: l$ w i3 r% H* U+ I% r/ |2 o
};
$ w1 _. \ m# n" w* [3 C
! t$ L) b. A& ?
function seestate(){# ?2 D; _$ v* k
alert(ws.readyState);
u5 | w/ \4 P2 e
}4 F2 A5 o' h1 o$ @( i; \; p
! C/ w" X, @- a2 ~9 v* z$ h
</script>$ \, ] A) |" D. K6 L
</head> B: i' a/ ~& }. x6 M
<body>
7 F# i; s/ r6 T! b# B* P
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br />0 b/ [# ^2 [5 b6 ^0 Y
<textarea id="content" ></textarea>
5 n3 j/ H9 }9 D
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br />
# a, q! g$ T: v' a, s
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br /># @ J! s- {/ n* |3 i
* X: E. F- i2 _2 @
</body># I5 c3 T% e7 Y2 ^2 d/ ~2 O
</html>
6 b& |/ S% O) j

复制代码

2）服务器端实现

class WS {
1 x2 @+ ]. t$ y; l. i
var $master; // 连接 server 的 client+ Z8 h% P$ y" d2 v
var $sockets = array(); // 不同状态的 socket 管理
! X5 C& R6 P% F, v5 ~% h
var $handshake = false; // 判断是否握手
' W$ t( y* b: a w+ y7 Z5 f |
: u/ B# S1 E$ C- A0 R( N
function __construct($address, $port){
9 x: A2 s8 }" c- @4 K% R4 P
// 建立一个 socket 套接字* _, Q5 m L% J( E% G) ~# _3 F
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP) ( n$ Q+ c# E5 x: }# B e, j
or die("socket_create() failed");: l* W" E( P% b' c0 N1 T$ _6 t0 M
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1)
# w/ ]/ |$ [% q2 m& a6 M9 V% Y+ c
or die("socket_option() failed");
7 l) \- E( T) `
socket_bind($this->master, $address, $port)
4 m% ]; h1 O3 A7 Q
or die("socket_bind() failed");* `/ z- I) e- j* l c
socket_listen($this->master, 2)
! [' x3 e. O: W8 {
or die("socket_listen() failed");
0 _! H1 Q; |, [3 G. U$ C. Z- Q3 L
$ W" \# J7 g9 \" E% Z I) X
$this->sockets[] = $this->master;" }# D8 O( s9 z% S- B; l3 `) E
, e q7 Y0 k9 {9 C7 l
// debug5 H4 A" S* P2 g8 j1 s9 F
echo("Master socket : ".$this->master."\n");6 }" H! `7 I2 N( m, {' y
- q0 O! R" _/ x# G' U' `/ Z2 k2 Z
while(true) {
$ V$ D, p# l4 N* o
//自动选择来消息的 socket 如果是握手自动选择主机
7 {1 u3 F0 A2 v# s: y2 U
$write = NULL;
1 M; {. B/ {* E' x% E3 X# T
$except = NULL;8 v: d/ n o# }# c7 `# ~
socket_select($this->sockets, $write, $except, NULL);
* b% u$ [5 I& x; ]$ H. \9 n
* f; X* @# P1 k4 \& F: G8 ]: w [7 k# b
foreach ($this->sockets as $socket) {- `' d# M/ G% u5 C) h Z
//连接主机的 client 2 w" s# ~3 E! y- P6 m
if ($socket == $this->master){
7 L2 a1 o& n- k
$client = socket_accept($this->master);
; j/ ?' H, h% @: K+ c
if ($client < 0) {3 X$ f6 \1 E! h% |. i y( u' w
// debug) m5 }5 m1 i" K$ X
echo "socket_accept() failed";
& ?/ k3 ^* H8 k1 r7 T
continue;
/ W$ F. i( a* y: L* j& N
} else {
# G. H Q- {* F4 N
//connect($client); a1 I- @2 |+ [6 m1 g3 m! e
array_push($this->sockets, $client);
: _: I4 c$ t8 D1 ^7 P
echo "connect client\n";7 U9 a+ R' W1 y/ v) F5 { o
}, S `6 p8 C) ~) m+ W5 }
} else {* G. e$ \9 [: c' h6 _
$bytes = @socket_recv($socket,$buffer,2048,0);: {* O6 [3 ?$ l
print_r($buffer);% W- Y6 y# n6 D% Z
if($bytes == 0) return;0 [/ W3 d4 x5 \1 D
if (!$this->handshake) {4 N. N w3 U( X( q
// 如果没有握手，先握手回应3 Q$ H# _# D. r( j* t# ?0 C+ T+ e
$this->doHandShake($socket, $buffer);
/ u' h" }5 t' }, e' J; @0 f; L
echo "shakeHands\n";
, \( ^- E3 Q' H# v" ?+ ^/ W
} else {
+ j. Q0 e* h7 ~6 l5 s
& A/ j/ W- q! x' l$ ?% Z4 U
// 如果已经握手，直接接受数据，并处理 d7 ]& }. P4 `2 b
$buffer = $this->decode($buffer);
M. w2 a% t: I7 y, y/ K
//process($socket, $buffer);
5 c9 d% z5 d. w5 ~) t4 T9 e, N/ O
echo "send file\n";6 u# ?7 [+ R3 R$ `' a `* K
}
$ y" A) n2 y4 \" l" v0 J. R" J; p
}! C! K6 [, O1 d$ L. r! L; G$ y
}
/ [' Y- w, A. u, V2 O4 R
}
1 q6 F$ D2 f9 n2 d8 T
}% L0 z) s3 I8 r6 {; h5 C8 c
' R# n3 T, d8 v3 y0 l6 T' m
function dohandshake($socket, $req)) Z3 l! V5 a' Z0 Y# @9 P
{
' R! ~0 P4 o/ Y j9 X9 V i1 ~
// 获取加密key* k7 i ~! E& O: q! p- O' S( m
$acceptKey = $this->encry($req);
+ y2 W) @9 ~: _5 K: s
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" .
/ d$ B6 v a r2 P" q, `. T( J
"Upgrade: websocket\r\n" .
# U5 j4 K- e2 O. Q
"Connection: Upgrade\r\n" .
9 {% M9 Y' h, v8 [8 b8 [8 n# \
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" .
8 W- ]& j) u3 F
"\r\n";
* {0 k# m/ p* |! p
# G6 L5 O! h+ T5 G; w) r
echo "dohandshake ".$upgrade.chr(0); $ X5 t* j; J3 f3 `
// 写入socket) b5 L, S- j4 N4 i; y8 D
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0)));, [/ Y! k( [: r- \& L) E8 |( C# a
// 标记握手已经成功，下次接受数据采用数据帧格式. `. {) Z/ L: S& `
$this->handshake = true;
* [5 F) I$ g6 g0 I
}* j: b- f: o. @+ ~' o% L
& ]% V, Y! C9 e, C$ Z# l, N: o! x
6 k9 |" L, q% A# J
function encry($req)
& d3 Q9 `/ i1 ]$ E+ h
{2 p5 _' l! ~8 y
$key = $this->getKey($req);
* M0 Y+ M( C- }) D, n2 [8 m: r
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";; u1 h4 \; c/ c4 ~$ D) O
5 V9 v+ \2 L8 Z( a N) n% ^# d
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));
4 H, ^1 z. D# B2 G/ S
}
5 J! d7 B1 T L- h0 G
" I0 T" T$ A, W/ X" ]/ f3 g
function getKey($req) ( m: m8 w7 T+ G
{. H5 j, ~$ G( ^4 x
$key = null;# L; s2 W" G1 c6 k/ [" k( J8 |7 f# D/ o
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) {
% V [9 N9 }3 h0 p* B
$key = $match[1];
! k$ ~5 ^3 T: t# A
}$ I/ I' z- d- F$ ?, R
return $key;
3 V* g! Q3 J5 S$ \; {/ e
} {/ U" {/ M$ r+ Z6 m4 O
+ M% J9 y3 E3 v. N! W
// 解析数据帧
" K2 ?$ m! ^) B# @+ j$ M2 x9 o( f
function decode($buffer)
( v" h" I3 H$ w' T5 s
{6 z' D+ H: F! `3 J/ @+ J/ V
$len = $masks = $data = $decoded = null;$ V9 d5 h& o2 K# c- Q6 q
$len = ord($buffer[1]) & 127;# k& m- U, `7 { i! h- N
9 ~8 i7 n5 b) n- X
if ($len === 126) {' f- r2 Z( S7 q) a# s
$masks = substr($buffer, 4, 4);
) l9 Y3 {- n& j% S
$data = substr($buffer, 8);
8 o2 Q9 [4 f; `' J; ^
} else if ($len === 127) {
* _% _/ ~* x/ X7 k2 p$ V& v
$masks = substr($buffer, 10, 4);) U0 R8 [) f* u8 v' t
$data = substr($buffer, 14);
9 P; t3 o# r+ b
} else {! Y; g9 p; { E6 a7 M* |
$masks = substr($buffer, 2, 4);
J8 X6 z* ~ E* _3 s4 W( K
$data = substr($buffer, 6);* \7 j$ i7 `4 r6 U) h
}( ?7 H$ A: c% W% _+ ~1 d+ U0 X
for ($index = 0; $index < strlen($data); $index++) {
5 g; u! D6 v8 o/ z
$decoded .= $data[$index] ^ $masks[$index % 4];+ g) F( F0 P' [ X0 ?
}
( b( Q2 H5 M9 q: j9 B; z
return $decoded;. m: A( y z9 k1 Y, r2 j' _
}% I* q, x2 c" q# i7 n
. e) Q! X `- o0 N( j$ E V
// 返回帧信息处理, V! L3 h0 ]3 M8 h- l$ a
function frame($s) 8 U# G' m; C& l. @! m8 V6 B
{
. z: n2 _. r8 R# p2 M( d: i
$a = str_split($s, 125);
7 N3 ~: r$ {3 V1 R7 a, m+ x& B
if (count($a) == 1) {7 _4 u2 t3 o) O* x1 m
return "\x81" . chr(strlen($a[0])) . $a[0];; ~+ v7 @$ f. `- E L* Z
}2 R6 O9 x8 K9 h# h' d! c2 x
$ns = "";
f1 i: l+ e# f6 p
foreach ($a as $o) {" ~7 A, W7 C1 F# Q0 h# s! P
$ns .= "\x81" . chr(strlen($o)) . $o;
+ X2 M/ x: m6 W2 y
}
. ~* Q: J2 x( U) C0 B& T. n
return $ns;% d1 K1 @* R" [
}9 a: Z2 E: D' R+ A$ \
6 D& H& W" n, N* R0 c% w" u) y
// 返回数据
( y1 G+ b4 D7 O# S7 F7 O+ G' h1 o- j4 Z
function send($client, $msg)
) f3 N! \6 c* w$ \! F* l5 ^5 D
{3 {. K4 T+ {6 R3 E8 u
$msg = $this->frame($msg);6 a% I) ?7 {2 h6 S# ^$ j2 D( d) I
socket_write($client, $msg, strlen($msg));5 Q7 y {- v. j }# R0 u
}4 y4 w% H9 g4 N' O* d
}
+ v3 I# o& a0 T+ L! Y
3 a# o$ G" s2 e5 E
测试 $ws = new WS("127.0.0.1",2000);# r4 O& J3 f1 ~" \7 g. W, r
/ X8 ^5 R! a2 ^; k5 k* R

复制代码

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

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