PHP 简单实现webSocket

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

1）客户端实现

<html>. L. F4 D, q- g9 g- C
<head>
$ l& x Q8 I0 t. H4 ?
<meta charset="UTF-8">
3 b, C9 Y6 Z, O
<title>Web sockets test</title>' P2 Q' ], Y L4 k
<script src="jquery-min.js" type="text/javascript"></script>+ b2 S, F0 G1 A" J
<script type="text/javascript">$ W, p8 O7 u. T" f4 E; Z
var ws;% ]8 Q( Q8 T9 K7 G& ^. d% k
function ToggleConnectionClicked() {
4 p! d! B- u% w# m
try {
. G {) F @* M- i
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器 9 {3 Q4 O9 Y H2 Z
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};6 H, Q: i: e' |) M6 @, A
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);};
5 y5 X. C) `6 J
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);};
: B7 J4 X7 x1 X S1 b
ws.onerror = function(event){alert("WebSocket异常！");};
+ k) j( c/ p& e
} catch (ex) {
. I4 ]8 `( d+ {! g) {
alert(ex.message); a E \5 }3 e4 L! _
}5 K+ ]9 R( s& y
};
/ r7 s8 g/ |& b9 H
' ?$ V- u9 I& r4 K) Z) X
function SendData() {
5 j- N$ p. ^9 a0 t, q- o/ j
try{' \3 e0 M$ p$ B8 b* Z- d) T
var content = document.getElementById("content").value;
4 C! r( \" E% u) J
if(content){" T+ y5 m6 p) {: V3 ^8 L, R4 _
ws.send(content);# _5 n' G$ d* ^# e5 a- ?" a3 ]: x
}
% H0 ^# ^/ X/ h* [) q+ e
8 n1 X' {; n9 p3 b% @; v" x/ e
}catch(ex){
: p v/ k7 X& ?
alert(ex.message);9 v: |) q: P8 q6 \" n+ B
}1 v2 e. U& D4 `) V1 P# D$ r! \
};
8 w0 E; g5 w2 N U2 r- t3 |0 |% I
( O) I" g v8 d- V
function seestate(){
) o$ `: C5 Q j
alert(ws.readyState);
" j$ h2 y: P D6 p
}. b V, u' z' Q. _5 h+ ?, F; \
& c, A3 N* ~: F. T& b& X. v4 W9 U
</script>7 [9 [. y6 i- i2 t
</head>
' ?3 H& z0 f4 ]- \2 K& h" g$ a7 J8 }; F
<body>
/ c) y, ^8 k7 u/ K2 u# t2 s1 M
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br />
% q/ P+ W" V5 F
<textarea id="content" ></textarea>
" w% [$ b; {0 A) y5 z- _/ a
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br />
- Y' a! v' g6 Q0 c4 X
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br /># L, A; a% t+ Q! }8 F, a
4 B) k# H8 a8 a& _$ O
</body>. w6 A) \6 t1 G& ]. R: u
</html>
M1 Z" t, v! B# g( s7 w3 z* ?

复制代码

2）服务器端实现

class WS {/ g2 @1 A1 S. N* w2 Z8 H/ Q
var $master; // 连接 server 的 client
1 G E: Y( A4 N& o0 {* _, Q
var $sockets = array(); // 不同状态的 socket 管理
( Z0 k" \) D, b6 r" ]+ a
var $handshake = false; // 判断是否握手
o. A, G6 \! N1 S( r
" B+ M X" v4 w3 c/ O8 F. B
function __construct($address, $port){
/ Y) i5 e& {; R0 B5 d
// 建立一个 socket 套接字4 K, ]1 K: }8 u+ _, a4 F5 H2 v
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP) + p6 q: @ [$ j4 k4 y9 p
or die("socket_create() failed");
, b K2 L- o; N" ^) X0 P
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1)
3 {. D( |6 m2 f1 l0 a
or die("socket_option() failed");! ~4 U( Q. Q- _5 e" J
socket_bind($this->master, $address, $port)
( q* } [, O9 A" n! B: }
or die("socket_bind() failed");' }9 i, E3 r5 q: A
socket_listen($this->master, 2) & Y% h- j' J& W+ b* ]+ _
or die("socket_listen() failed");4 x, j( y- g9 ?9 r- G6 p
) V$ l& s! K; D
$this->sockets[] = $this->master;! K a! ~1 T \, H: L# ?
- M8 m/ X5 m! d! ]* S( U9 z
// debug
3 K& x8 D% ]: X: o$ Z
echo("Master socket : ".$this->master."\n");" y4 J- e: J9 D; T2 Z' f
) }' p. V# \' d/ p: D2 j3 {4 P! F
while(true) {" w5 \' g. i3 w2 ?9 L' }& C
//自动选择来消息的 socket 如果是握手自动选择主机
" ?: s8 {6 f( ?# N2 f2 r
$write = NULL;( Q4 }( V) y1 W9 I( `& ~0 G. d
$except = NULL;' B2 y: ^+ D/ L" T7 r
socket_select($this->sockets, $write, $except, NULL);1 D8 i5 h8 z1 O
, d- k" g5 a& S4 X! U3 B5 W
foreach ($this->sockets as $socket) {* l' J. R; R& d" }# u- |
//连接主机的 client - W: Y( O& w" b* y' K- m+ `* ?+ D! |
if ($socket == $this->master){7 r( X+ H8 r% W3 h9 g9 A6 ~& G1 n9 r# [
$client = socket_accept($this->master);4 M$ ^1 H6 {/ \% S& x- i+ D* r3 N
if ($client < 0) {; h$ t6 A O' b0 K' g
// debug8 [4 `2 T& |! k# j6 V/ W* d
echo "socket_accept() failed";: t, m0 e: [: P+ w
continue;5 q1 E5 I# Z( q7 M! _# j `7 u! Y4 I
} else {5 j; Z2 S- Y7 C* t- P
//connect($client);
( |) N; f# ^7 ~( e* A" t& m
array_push($this->sockets, $client);
$ U+ H% _4 P2 E+ b
echo "connect client\n";1 u) y7 n# A: w$ ?8 t0 p6 E, X
}3 ]3 h ]; C" u6 y/ o$ D
} else {
5 Z# w0 O! C% p6 B1 r- _
$bytes = @socket_recv($socket,$buffer,2048,0);! B" Y+ C7 A0 d" ?( Q/ c% g0 A2 L: p/ N
print_r($buffer);! F/ E& D _1 ?5 U
if($bytes == 0) return;
5 E: @# {, V' D% H3 w* v% \
if (!$this->handshake) {
7 M& `6 I5 \* B3 j
// 如果没有握手，先握手回应
: Z/ S) E K2 `- m3 W5 m
$this->doHandShake($socket, $buffer);
) R$ B# H2 ]: h! l3 Z0 d- b
echo "shakeHands\n";
1 T0 c) W- A( R
} else {
. l! T$ B1 b8 n3 U! O
$ V2 r- F. b) g4 c @! j
// 如果已经握手，直接接受数据，并处理
, Q( s+ A8 G6 C" A! s
$buffer = $this->decode($buffer);% u' ]: y( p0 F2 C- \
//process($socket, $buffer);
1 [# d4 k2 ^2 z- k+ Z7 w3 G7 S
echo "send file\n";' i1 [+ [) }8 p' [
}
2 c9 u. m: V8 z7 L8 K/ c
}
9 I# K5 w. e t, }1 x# m" h" u
}
: c) k- A- J( A
}
$ `/ l; L: Q& ^8 K
}) L) u5 X2 W9 s9 W
1 q& W% n C7 K4 R% d
function dohandshake($socket, $req)/ H ^; {2 h! v1 E
{( K0 x" T1 U1 ?* I
// 获取加密key0 o5 l" i0 O5 B; L. d
$acceptKey = $this->encry($req);( K6 z' o5 v B* P/ [
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" .9 k! g, V: z! `% V
"Upgrade: websocket\r\n" .
% d! K. v4 d; g2 s
"Connection: Upgrade\r\n" .9 J* m7 u. U3 e
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" .
8 u4 q! o% C9 |% ~$ }/ k! S% M
"\r\n";
' Z* j- I5 }$ ~! s) M
0 y2 @6 p' x6 E [2 b. K
echo "dohandshake ".$upgrade.chr(0); / W/ I& t; c, r0 S; M+ c9 }
// 写入socket
0 {8 i2 ~' _7 d- _& R9 n/ G/ S
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0)));
: m2 u% u7 t C' L( w0 A. ~
// 标记握手已经成功，下次接受数据采用数据帧格式! V4 w1 l6 L0 [+ k+ r
$this->handshake = true;+ f( ~, S, J& H& `8 C' c& t
}5 v1 s' Y2 Q- q
& r* S9 B7 y. y
) Q: M' j+ E) Z5 x
function encry($req)( Q$ O- t+ w+ u* h6 N
{
* e7 w% k" x2 y1 w. z3 O' F6 | n
$key = $this->getKey($req);/ b( l6 K) V5 V& X; A0 v, O, l6 M% n
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";, U ^( I) i! Y& m1 G0 P
' e/ o" W' J9 I- x
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));
$ X5 m. c2 E% W; b5 i
}
7 z, T4 g9 F, }
2 _5 k [- j6 P/ o
function getKey($req)
: W* i( t+ c" y2 x H+ k1 H9 A
{. g" n6 R( E, c$ E
$key = null;
$ A! ?, T& N0 P! \4 w
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) { & a" E, N M* A+ S. S
$key = $match[1];
G" J& h2 _9 D: F/ c
}3 V) b* I# c' U/ p
return $key;
O( i) V. ]& f/ `' Y1 I* ^# p
}$ Z* H2 z& }6 d9 \: w" _: D
) n0 S( [8 a. L! ^0 {4 K1 p
// 解析数据帧
4 O0 f3 s M% i- b( C1 b+ ]
function decode($buffer)
# P& [" j' W9 ^, o7 y! h
{
' P! J7 _* p5 q( |3 r# S
$len = $masks = $data = $decoded = null;
# s& M% d. x0 z4 j! q
$len = ord($buffer[1]) & 127;$ B& H: \$ }7 a* d: |
' W4 X8 Z8 x6 y ^( F3 d5 B
if ($len === 126) {3 F* \ t3 h3 h' L* ?0 X8 U
$masks = substr($buffer, 4, 4);9 L2 z3 P( K8 ]* o0 |* ] A
$data = substr($buffer, 8);
3 s4 Z( i+ k2 Y' t1 O+ L
} else if ($len === 127) {
, D" S( v5 T3 z' L5 g/ c, k
$masks = substr($buffer, 10, 4);/ x* P" z# _& l
$data = substr($buffer, 14);
& Z2 X& d6 U3 w. i7 z2 d" ]4 T/ e
} else {2 S3 z+ } Q' a0 B$ e4 u3 y
$masks = substr($buffer, 2, 4);
0 d: z: \* t+ x
$data = substr($buffer, 6);) Y9 Z& }! I: N6 _
}& F. `) A4 f1 J7 F
for ($index = 0; $index < strlen($data); $index++) {
3 Q3 ]" }9 l; |3 N+ |
$decoded .= $data[$index] ^ $masks[$index % 4];9 A7 l# t+ l9 j, `
}. V+ d- \( x& Y5 Z- b
return $decoded;% b- q A" i- h& g) n: r% H
}
* K6 M1 X4 l5 Y
/ Y2 I1 e5 ~, N6 {: n8 M- \
// 返回帧信息处理
. z, m2 m" h& {% {6 h6 J
function frame($s) 4 J, @8 G, J* P2 t* `& V1 D
{
6 q8 r$ e6 B, ~8 B/ `
$a = str_split($s, 125); T2 [! Q) J& Y1 v+ Y$ ^3 S
if (count($a) == 1) {7 F7 E7 H' z0 @8 R' i8 l
return "\x81" . chr(strlen($a[0])) . $a[0];
/ G }, M" z& g* y/ S
}" m) u S( ~/ Q& L
$ns = "";
( ^2 s1 f) B' [9 h2 I$ r( R
foreach ($a as $o) {
/ ]& e @2 L) v" b; Q9 G5 B+ }3 _; X
$ns .= "\x81" . chr(strlen($o)) . $o;
3 m6 i) P6 I+ Q. ^
}, ]* l( D4 e( {6 b
return $ns;2 ~' }! ^& ^3 s" h1 o
}
! B1 n6 K. @8 y. N* @! L
~( ]% j( A5 k) Y( J7 A" \
// 返回数据' s8 a, m! L* U' M+ _8 T' \
function send($client, $msg)! V8 k$ N' A( K# I7 P
{; K" @- ?. B4 s. Y
$msg = $this->frame($msg);
2 L5 ~ m! a: @2 P/ L7 N
socket_write($client, $msg, strlen($msg));; G* F2 N+ a3 o/ z
}
/ ^; @9 z, h4 u9 E# I
}
9 `/ x2 i' [6 m9 W* ?* F1 M
/ T# N! h7 Y( G0 v6 K
测试 $ws = new WS("127.0.0.1",2000);# v# o* T+ Y/ |" b
# I; z3 v8 C* u6 h. L

复制代码

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

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