PHP 简单实现webSocket

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

1）客户端实现

<html>
0 a( O' ~0 X, [( N: @! B4 j) S4 Z h
<head>
& v7 u& y ]1 `% |( h( {
<meta charset="UTF-8">
. t5 V, w1 H6 T+ `# y
<title>Web sockets test</title>
P2 t9 r# x4 z4 c2 j) W7 j
<script src="jquery-min.js" type="text/javascript"></script>
; w+ t0 h A) l. Q5 ^
<script type="text/javascript">
" f8 w2 Z3 I! D& ?+ M; R* Z
var ws;
) ?( N7 t# O: a j9 I0 T( x# f
function ToggleConnectionClicked() {
% Z" j6 g9 ?0 H8 q5 s+ v% ?
try {3 x! |* ]0 ?# Y* \0 {6 L1 }$ v6 p
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器
+ x. i& c4 y$ p& W5 |/ H1 I4 B8 ~6 V
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};. o3 _; [* }* ?' ?
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);};
* ?; j" D* N) _. {( A3 W8 h
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);};, S) b* N. `/ |" W- A
ws.onerror = function(event){alert("WebSocket异常！");};& ~$ b$ ]8 k2 m) @8 g J/ T; C
} catch (ex) {
. k; N( [) e- F! K
alert(ex.message); 8 r+ G! z w1 V% [
}
0 B2 m( ?9 S9 J% F: e4 M+ u
};
! x6 G( F6 [% n. a2 x1 r
" r$ ~- p8 J v" D6 f$ D5 G
function SendData() {! A9 c5 z( ]1 G7 ~6 l% F
try{9 n9 ^: F: r3 d$ @* |; {
var content = document.getElementById("content").value;
0 T# w9 s) e1 i( B
if(content){
5 q2 B" ~! X+ _( B |
ws.send(content);) Q& E; g4 k" P# ?8 O; w
}% @0 y, q* O2 y8 Q6 S. Z; X; c9 n
: d: n" G! p, p
}catch(ex){0 ?0 G" o6 Y0 p ]2 h+ r W* Y( V
alert(ex.message);
# W: }8 Z6 w+ O9 F
}) `) w% W6 m/ v& N/ p
};/ O# e' ^( ], Y* \5 m$ K8 N
) j) P$ X d4 V g Y: N
function seestate(){
0 B5 X% ~2 p6 H2 I
alert(ws.readyState);
# r4 ~' t M8 ?4 |9 N0 A& \2 W
}1 e( u9 G" ?& s
6 W( M/ e$ l, T* e
</script>$ _: o" D* i" H
</head>3 b, r$ M) v% s* G
<body>( n. O# ]& {7 c) H4 I9 f( s
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br />
. {: c; e- Z( P1 i4 f+ l, l' N) D
<textarea id="content" ></textarea>
5 k7 O- n/ n" i: J1 z8 n+ ^( c# @
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br />! {$ j; V( p$ Z( U. j$ P
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br />
$ |& G: j$ [+ W" T; C' Y$ S: C
9 B7 K8 T; a8 E! j
</body>1 s! `9 y$ Q- u7 h2 |' S6 m
</html>9 c% U: M8 h) y6 F% t

复制代码

2）服务器端实现

class WS {
- S; N6 w2 q2 o* s# G. Y# y
var $master; // 连接 server 的 client- A* u \: d( U; I
var $sockets = array(); // 不同状态的 socket 管理3 Q3 L! c# y4 J1 Z, W6 Z9 h& P, j2 [
var $handshake = false; // 判断是否握手) K$ O$ d* ~, A6 z, m
9 U4 s3 M" p( X+ v& S) G/ E* Q
function __construct($address, $port){
: Z. }6 `9 d/ q: J F) y
// 建立一个 socket 套接字. I) O* u$ D' ~% K& y. |
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP)
$ K! n' S/ j$ M* Y
or die("socket_create() failed");- Z4 N+ X9 a) r/ M! d' v: W1 f& H
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1)
. N* r$ y' B4 `- @
or die("socket_option() failed");
0 [5 N' n- X# l$ T. o* Z
socket_bind($this->master, $address, $port)
( L4 C- K& C* X" f `% Z
or die("socket_bind() failed");; G9 c( S3 i( }, P, w; E% j
socket_listen($this->master, 2)
$ s# A4 ]3 N6 y9 U B
or die("socket_listen() failed");
1 z5 ]. M3 s( R9 T5 r4 d
: P/ v, G- q) z8 [. O
$this->sockets[] = $this->master; L3 S6 N2 K5 X. |' H$ D4 ^; P& Q3 @0 e
( j) J7 O x& {. i4 K9 K' }' T
// debug
( H& l- O! Q) d. c' P0 e. c3 w
echo("Master socket : ".$this->master."\n");
# w& u6 z& l \' s0 k0 R" |! M
0 k# A9 r1 u% t
while(true) {
3 G5 @$ {! k/ H8 E; A% E; T2 e
//自动选择来消息的 socket 如果是握手自动选择主机3 f+ Q/ d/ r& ^, q! z
$write = NULL;
* @( N/ F) F% F% J2 h
$except = NULL;. a* r# S; O8 j% \& [# i
socket_select($this->sockets, $write, $except, NULL);1 ~; v1 B) x8 i& z' Z' t
7 A* u1 A5 P n( X
foreach ($this->sockets as $socket) {: ~ a( a/ O% g- T
//连接主机的 client ; o+ Y6 m T9 c+ i" u
if ($socket == $this->master){
. E0 @! a4 u3 g# S/ w
$client = socket_accept($this->master);( j- S0 }' W, I6 t
if ($client < 0) {/ G: w/ ~. X% g, z
// debug+ P# }' S. o& r3 f. y
echo "socket_accept() failed";+ L+ p' w! J3 P0 I8 M& J
continue;; T) Q& A2 O" b
} else {9 q1 q, J$ q: L5 \
//connect($client);
& u; l& D: }6 M
array_push($this->sockets, $client);# P! H( d+ P2 x( i& r4 f
echo "connect client\n";! H0 l& I( f0 I, D3 t
}
5 |) O0 } x0 y& Y4 w2 D
} else {
( V7 P& X8 [9 X$ C* O
$bytes = @socket_recv($socket,$buffer,2048,0);; _+ f" D' _3 I/ x9 T; N7 m
print_r($buffer); r% ]) o. i# K6 N
if($bytes == 0) return;( L; t' x% I. L. m0 d2 R1 j4 `( g
if (!$this->handshake) {
$ b4 I; [$ W) @8 x/ J8 U
// 如果没有握手，先握手回应' m1 D! E* z, {7 a' B+ T7 k
$this->doHandShake($socket, $buffer);
1 g' Z3 X' r& w2 ]8 W) F# ~5 h
echo "shakeHands\n";3 Q% g5 l3 r) C" M3 n( Y7 W8 Q
} else {
+ a+ C: e6 u$ F1 `' r
3 Q( U( u) t5 |# T5 C
// 如果已经握手，直接接受数据，并处理
1 Z, F g6 r! ~! z+ F8 o( l
$buffer = $this->decode($buffer);$ v! t2 j* W9 _# Z5 y" l y
//process($socket, $buffer); / f6 g' r# m4 T1 Z1 j% k5 o
echo "send file\n";( _# i" z- H5 a" k7 E Z+ H6 [
}+ G# C5 n' p1 M; T3 }
}
2 D+ ~1 H9 |) W8 d
}
% P. o& d, j3 m' _2 o
}8 D: T8 E- e# ~( n/ m
}# K" M4 I% Z# Z3 S( n4 _
' C3 N0 C6 i. D' k4 a
function dohandshake($socket, $req)
3 Z' O3 c3 P# x$ Q6 h0 z& a
{
+ t& W6 G5 l9 ~- I$ j0 L7 A9 r
// 获取加密key5 ?1 q5 X; _1 \5 I( d4 Y* p/ m* q
$acceptKey = $this->encry($req);0 b7 p( B A' }
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" .
3 g* {. A/ `! ~( Q
"Upgrade: websocket\r\n" .
! ^/ o0 f9 k% S( ?
"Connection: Upgrade\r\n" ., Y3 O T$ v$ _ x" J1 s( @- v
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" .
) H/ @4 r- M$ f2 n i
"\r\n";
! z- V: n& g- d: | S: [
8 w+ t! W* v; n' [
echo "dohandshake ".$upgrade.chr(0);
9 E. A* d9 f9 N' |& E3 |7 m+ A
// 写入socket% M2 }4 ~" l6 W, R+ k
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0)));8 y& F6 \5 q; C7 p2 B
// 标记握手已经成功，下次接受数据采用数据帧格式- r; ?0 x) F% `9 T
$this->handshake = true;! h& @' ?+ H1 D( u2 ]. a
}
1 V. e! m) M4 F) g6 u
1 ?* S; q* t+ ]: |1 S3 b
( t+ `) [/ a+ z4 z& J9 }# `
function encry($req)
f6 X; I$ \ `7 E
{& v, O8 Y! r( l3 b% V* K1 h
$key = $this->getKey($req);; o' ?+ s! z3 _4 P6 i
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";
+ U* b* ?" L0 v; B: A
0 R: _2 j- N. o8 R2 y
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));
) J6 H' J1 }/ R
}
, f+ z9 B% d, L1 y
E5 x( e9 K4 `# d- {6 n( ^. E
function getKey($req) 4 Q7 ]7 A: U. k: ]
{1 s' M# ?' n0 F$ t$ @7 D; \
$key = null;+ K% _0 P; P g: ?
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) { 9 V2 q, @4 J ^- C' t D& F
$key = $match[1]; % o" E% X5 F: f* m" u
}
, K0 X# x6 P8 c$ J* S `5 e* @
return $key;
% Z1 s5 l% p/ E
}
. Z$ n5 D: R9 Y+ W
0 b5 h; Z4 p# G2 p& @- J4 U
// 解析数据帧
$ L* U- Q0 M; o; O9 m# o
function decode($buffer) ( z# F' ^+ p; t! S5 |9 g
{
# q5 l0 f1 R0 S% M( N* Q
$len = $masks = $data = $decoded = null;! E7 O4 Z2 I0 b& v U! a( \ p- b
$len = ord($buffer[1]) & 127;
& \8 u" S. G. F
0 A5 \/ y. U0 @% O' M- k
if ($len === 126) {
( [8 {! s: p# _3 `
$masks = substr($buffer, 4, 4);/ `& k: h8 D! [2 R/ w
$data = substr($buffer, 8);# L# g: d1 n# ~! u* j" n
} else if ($len === 127) {. j, z0 C Y G- I9 V
$masks = substr($buffer, 10, 4);
& d& y, S+ a# J; i0 |$ C6 H+ U
$data = substr($buffer, 14);& o7 y! i/ ~% p* N7 M. o2 J Z l
} else {
& F* E. x( h: @, r7 k6 o% E+ w
$masks = substr($buffer, 2, 4);! h4 T$ Y' | p/ k' f9 N: h/ ?- b1 k
$data = substr($buffer, 6);
9 @) S& `3 s: h/ v
}" y, F! X1 D. d- C" o( K; P
for ($index = 0; $index < strlen($data); $index++) { s6 e- z( a4 @
$decoded .= $data[$index] ^ $masks[$index % 4];
2 q- O6 x, I3 M& P4 M
}. a# B7 q8 f |3 x' S& U
return $decoded;
0 L* B C* ]2 O7 a# i; z% A) w1 S3 `
}
8 |+ V0 |7 C4 {+ c6 W; E& f6 ^
( b. h+ \4 i+ |7 c: j
// 返回帧信息处理 @# F3 j4 _3 q
function frame($s)
{
4 Z2 ?' ^; d. z# [
$a = str_split($s, 125);
8 F3 o- u' t! c! b
if (count($a) == 1) {
7 E, w3 T9 W, V3 [
return "\x81" . chr(strlen($a[0])) . $a[0];3 q& N9 N- E0 K
}
& }8 b4 \% K- Z# x, H+ z/ y
$ns = "";, T8 y+ ? d( H Y3 p+ X5 M# M! j
foreach ($a as $o) {: P) R! w0 X! Z3 V; K9 S3 D
$ns .= "\x81" . chr(strlen($o)) . $o;
- m5 ~7 K: M r& b2 L+ Z
}, {& l. d6 }& N/ A* z
return $ns;
, X- n( T3 Y/ J
}& d5 G1 m- U- O7 w
. r# L5 R/ a* J6 c0 U7 R0 Z
// 返回数据
$ b4 g9 Z, n% v
function send($client, $msg)
7 e3 O2 G+ X5 T% h9 C
{$ B. z; i6 {' P! {
$msg = $this->frame($msg);4 A, ~; J% Y, O
socket_write($client, $msg, strlen($msg)); H3 F1 S6 B$ T% g0 j
}
1 V& ~7 G5 r* N8 U! l+ k
}
) g- s; o; b1 L' n) }+ T
2 v! J$ v5 N* H5 w# y+ ]
测试 $ws = new WS("127.0.0.1",2000);- f1 |: z# C) }9 A$ N
/ l+ y7 n6 f& U2 F& M

复制代码

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

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