PHP 简单实现webSocket

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

1）客户端实现

<html>9 m- f% ?+ m& {
<head>3 {/ o( A; k0 D" x: c! L6 p: P \2 J
<meta charset="UTF-8">
( w1 L% Z/ ^8 }! f# d. ~6 q
<title>Web sockets test</title>
# V) x; A6 v; ~' j
<script src="jquery-min.js" type="text/javascript"></script>4 U% p& }* T$ [
<script type="text/javascript">
2 _, f0 `' [+ O) i3 `
var ws;5 C6 {- B$ A ~/ h+ S g4 f. u
function ToggleConnectionClicked() { & a8 ]* ?: k& T
try {5 _3 V( j1 I0 G; k4 v# Y8 m
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器
+ P( j! J1 q2 A7 [4 }* e+ W9 h
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};
1 s0 x% F8 _" |) ~
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);};
8 F% k; G( s% N) b
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);};9 b, [ t t7 o# }% `
ws.onerror = function(event){alert("WebSocket异常！");};( h# s7 @, |) G9 k% R
} catch (ex) {
5 b! B! \; d& G3 r# @9 I t
alert(ex.message);
; Q$ _; Y! G2 A3 K) j/ N( z) q, e [
}/ \# c0 M" z6 @) s1 o
};
# Y, U- c: w) |+ h
" Y% A7 A* a$ \5 J$ R0 P2 q
function SendData() {
4 N- J7 h# N( l; d4 \$ ]+ D/ i, l# P
try{
0 K' R0 c1 m- i5 F% A9 n) c
var content = document.getElementById("content").value;* b* d+ l& w$ H. W0 ^" }, F( j b0 a
if(content){
' x4 [' k7 J7 O) m
ws.send(content);( O$ x* F4 T/ c p+ o
}! I4 n7 e) A) e, j. N, b7 M5 G+ `
. M/ R1 b: Q( x) T: z x% _4 \0 g; L# v
}catch(ex){
Y& ?% m% P# l
alert(ex.message);
4 G# m& ^9 h( z
}
$ u- w, {* a4 _- G0 Z
};
, y- j {" T0 ?( `, l ^
7 u4 ^# Q0 v e# z- e# U: ~* n5 O
function seestate(){, J- w9 W. E9 d; S
alert(ws.readyState);
: y% {: I: w. U! |) G0 h, J7 n
}
* b9 s2 F2 W' p z; G e9 a
5 @: i' h3 ^& x
</script>
; C8 \% G/ U# o o+ s. e& F
</head>
$ c6 K/ b/ N* l8 ^/ N5 H
<body>1 X7 C( e! a9 ~- g) o( e
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br />% l2 i3 c: \' V4 ^
<textarea id="content" ></textarea>
$ @1 b) J9 S) P! n# {, I# T$ |9 K6 N
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br />
% U( J% y4 B* e1 m0 [
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br />, O1 Z9 ~0 P V" D, f' ~7 O5 w
% S4 g# f$ Q C3 I
</body>9 p2 ]6 G+ p B! O9 b2 A0 J6 N8 s
</html>
- w! [8 b: [+ |5 u3 L

复制代码

2）服务器端实现

class WS {
) d. M- o8 @1 z* x- w
var $master; // 连接 server 的 client
+ K4 m H' P0 R
var $sockets = array(); // 不同状态的 socket 管理
8 v8 B8 e: [2 J
var $handshake = false; // 判断是否握手 N: h& V3 O# o( j
: W3 _* F1 g0 P6 U% @* V
function __construct($address, $port){
% g- f7 A# M$ o3 f% k' f- }
// 建立一个 socket 套接字
" C6 P( E& F1 S7 ~; J4 p
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP)
# l3 H! O7 r5 ~- R
or die("socket_create() failed");, m& s+ f) _; e) ?1 ]! x
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1) ' w) N2 }! K+ x# l: N
or die("socket_option() failed");
2 ^8 z( o0 ^& x! A1 b% v
socket_bind($this->master, $address, $port)
8 s6 S- b: N! \) {
or die("socket_bind() failed");. d5 `; n# x: b4 V: A- {
socket_listen($this->master, 2) 9 B% a1 W/ u& ]$ A" |, R
or die("socket_listen() failed");; J0 ]% p, y, Q) r9 ]) {
4 m5 D+ a# U" x- S% X8 }
$this->sockets[] = $this->master;
0 |3 k- Z' L K+ x0 l: y3 g) C' i
# w6 W; B6 |4 r7 C/ b
// debug6 b. y. z$ w' c& M+ x6 |
echo("Master socket : ".$this->master."\n");9 J2 Z) o% ^$ b: I2 ~+ R* G: |
( K" X3 P! n* T( `8 r7 A
while(true) {
( `/ v1 G% Q6 N- s: }
//自动选择来消息的 socket 如果是握手自动选择主机
9 w( r6 ^' _5 @4 o
$write = NULL;" n, X! C7 V6 g# k
$except = NULL;4 |7 g9 v: {' h, Y) w
socket_select($this->sockets, $write, $except, NULL);3 E* [6 d: k, S3 c* X& ~6 _
4 ? X5 Z- u3 h8 R) ^: R. U& n
foreach ($this->sockets as $socket) {! ~9 a2 X2 u/ ?9 u8 s
//连接主机的 client
3 ?3 S5 }: x/ B/ B; b
if ($socket == $this->master){- {' N0 m( y& c
$client = socket_accept($this->master);9 \* f/ B+ k' J* F8 v
if ($client < 0) {
* n, V$ t% g5 L2 D
// debug4 L3 Q3 J# y- ]& B+ J. y! s
echo "socket_accept() failed";
7 B( O/ k3 e, Q8 x
continue;
9 R. r1 S4 v9 n
} else {
! U& t0 a. {+ _! @* F9 [2 W
//connect($client);
- Y+ N: L0 F% ]3 ~8 b; E
array_push($this->sockets, $client); [4 o9 L" U2 I3 J x4 V( X
echo "connect client\n";9 P; `$ g9 `, _7 k- a/ y
}
0 g5 m& X: ~9 Z; Y. P' o
} else {
* S4 e! D3 q0 X! G# m* B
$bytes = @socket_recv($socket,$buffer,2048,0);0 R" a4 o) z8 _( U
print_r($buffer);+ ?% w6 t6 A5 {) C
if($bytes == 0) return;
( i0 i; z6 A$ _: c0 f! ?
if (!$this->handshake) {
1 w1 r5 ~/ y% T' l: N) n" e5 k
// 如果没有握手，先握手回应
( R* A) K! @& {4 g9 R3 T
$this->doHandShake($socket, $buffer);
# @( l8 S- ]" c. x4 v1 L& i
echo "shakeHands\n";
/ r2 q ?: \; f( c
} else {
5 U3 X" g7 y! W
5 a- G* ?" ^$ ~% q' ?0 z: V
// 如果已经握手，直接接受数据，并处理
! i. ~5 @' q9 C3 a0 I5 m, C( T
$buffer = $this->decode($buffer);
+ F- S* J9 Y" n! M1 ] n$ U& p
//process($socket, $buffer);
* V8 a! |7 g( G# d+ h
echo "send file\n";3 }8 _! t' U/ _: [
}
% q# a! a+ N3 Y4 ^( O) q
}( w! X) w6 ^7 V/ P7 i9 X( L
}6 M4 z& J( B. {7 W; [
}
& b6 x5 |1 h5 m6 ?
}
+ c" {: `3 b& `# i
- m' P8 C' E4 C# g
function dohandshake($socket, $req)
, v1 w9 [, h3 g1 P" h: C- ^
{! P+ x7 o% L$ e) h
// 获取加密key( a& Z1 c/ L1 U' {
$acceptKey = $this->encry($req);" p8 N; ~8 _# e9 E0 ~' h
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" .
, P5 g7 ~+ Z e" f' z0 s! a
"Upgrade: websocket\r\n" .
) C! C, M# v* i" f
"Connection: Upgrade\r\n" .
9 c6 W1 { ?; ^6 w P
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" .8 J4 M2 G- u1 x5 J5 R7 Z
"\r\n";; b* x% I/ P% `: H! X
( n. m' R% s$ f( V `
echo "dohandshake ".$upgrade.chr(0);
5 N# F, W( o: q) R7 V
// 写入socket+ l. s) U6 D! }# @- K
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0)));
+ O: X8 d- B d7 G; r2 D
// 标记握手已经成功，下次接受数据采用数据帧格式
; n+ e8 t1 Z6 H/ L) R
$this->handshake = true;- J, Z" r1 i) `0 y, ?) O$ H! H& a2 m
}( j" ~5 s d) B7 M/ `5 p
2 [$ h- g" v& ~
, G/ R, B# Z) f
function encry($req)4 ]7 v* ]% _% \
{
8 e1 P. V; B2 b2 h2 L ~: T* V
$key = $this->getKey($req);
, F( K* R4 e6 r. f5 D2 {
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";' s, H7 i+ e# p3 H% f( q
# |! Q8 I. ?+ _
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));
9 m! T- }( r# c( f' a1 @
}
# B8 y& H+ Q' ]* O
# u& J( S) M0 \2 l% K, v
function getKey($req)
5 L: [! s& _/ w: B
{
4 D7 `' H$ }/ J
$key = null;) e# [/ o T+ C; t/ \' T
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) { 9 [% u6 S( C: d+ C# Z. d% e
$key = $match[1];
. x5 E1 Z6 z/ A+ B5 w4 s
}4 ^( X3 u. k* Q5 g# R Y" G
return $key;- d7 U0 \$ b9 C
}
+ ]: j/ o0 E. `% l
4 y9 \2 m5 J L2 i5 p: b
// 解析数据帧
3 q- G4 ?) j2 \; @; X
function decode($buffer) + p$ g% p# ?* S. w- O
{
- W5 W. i2 r' j% C1 M8 a
$len = $masks = $data = $decoded = null;
. e9 k2 z( {4 O0 {) @
$len = ord($buffer[1]) & 127;- K& B/ F7 H, s& f. f
% k4 F0 I% c$ C& K$ |( l
if ($len === 126) {
, T- O s7 T# o7 `" }
$masks = substr($buffer, 4, 4);
2 P% z/ K$ ]% B5 R1 I" k j+ ]
$data = substr($buffer, 8);
3 e; w% f6 ]) a/ o4 Q1 r4 _$ q+ Q
} else if ($len === 127) {. z6 U; `0 y4 `8 P# S& u! W' l, F
$masks = substr($buffer, 10, 4);6 b" t; X# n# R. C; o( R
$data = substr($buffer, 14);( ^# a) S* t1 H) @
} else {5 l$ X9 I5 s3 x8 d$ m" [9 Y4 P
$masks = substr($buffer, 2, 4);
8 G2 K# v! R7 r4 s- ~4 X7 ]* Z G
$data = substr($buffer, 6);
/ w- v' C+ l4 V6 O
}; @/ ~6 w1 J [) i
for ($index = 0; $index < strlen($data); $index++) {
# C; ?& Y; L f A" x
$decoded .= $data[$index] ^ $masks[$index % 4];) l; ?, W/ a% ^# O! ~
}; L+ z0 t7 ^' g b4 Y
return $decoded;
& {1 |, U- b) d5 ?) `- G ]+ U
}) {" K4 S' q0 D. W
_/ B+ G8 z l5 K& g4 j9 c
// 返回帧信息处理- B. n. B* g) T
function frame($s) : M8 d+ g( ^ I3 L
{
/ c+ d4 a, O$ B$ u# O+ V
$a = str_split($s, 125);. H: U2 Q7 f3 j, m; U
if (count($a) == 1) {5 A) _! b& H; j) E# E) W2 g+ d8 r
return "\x81" . chr(strlen($a[0])) . $a[0];# G7 c' Z) _) W( B: n
}7 S0 Q# O; d: H) E
$ns = "";
4 Z' Z5 e* E8 W) e
foreach ($a as $o) {% }4 `7 W; v% h, v, m' P: u
$ns .= "\x81" . chr(strlen($o)) . $o;" W5 ? c( M$ W( v, O0 M% b
}% p4 Y) I0 a5 t5 m& p& a2 q
return $ns;. f$ L1 T* Z- r% s' f
}
5 F0 S! u+ o m, Y3 R' l/ C8 t0 b
6 G$ ^9 }3 z- H3 V1 M+ C* X" z
// 返回数据 \/ @) q N k
function send($client, $msg)
9 C1 ^" e+ ~' l6 R) _- p: y
{- Y3 S, D8 a4 N+ p
$msg = $this->frame($msg);" t( }. T* R7 r, p2 P
socket_write($client, $msg, strlen($msg));& C" @) y3 R2 d6 T& J
}
5 F. D' ^3 d: M' n- C/ l3 [
}
$ `* S% m- h/ V) d& W/ m
( _5 L, i& e! K* i5 i
测试 $ws = new WS("127.0.0.1",2000);6 P) N' N: a3 `; z5 n; D" m& t9 Y
5 T! W( x2 s6 ?; [" A7 Q5 e5 r

复制代码

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

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