PHP 简单实现webSocket

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

1）客户端实现

<html>/ e8 M( s9 p1 N: o8 d
<head>4 ~+ l2 s) L2 N3 m+ \6 f
<meta charset="UTF-8">
$ i! v$ G1 r$ s* X7 s4 b
<title>Web sockets test</title>
2 j; ^9 a! R1 t2 g# X
<script src="jquery-min.js" type="text/javascript"></script>4 O, \1 ]' v' G7 D4 h& Y# U/ a; D
<script type="text/javascript">( Z6 i( |) {% x* E) {, s4 |) J" n
var ws;
1 r+ a+ O7 t4 x. U
function ToggleConnectionClicked() {
( o9 K8 r; B% f; B0 k. P0 c
try {
; L4 K, {& y3 r5 R8 C* f( V" L$ l0 c
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器
$ b9 O& i7 |. A5 I9 i
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};5 g+ w4 Z# K3 P. u& J
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);};
g- h+ S( \) V0 a
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);};! `+ J0 J# l4 W" p. ^0 ?5 [+ }
ws.onerror = function(event){alert("WebSocket异常！");};
4 F, z/ [! }! i; F/ X j
} catch (ex) {
: ]% P0 C1 b, A: O& {
alert(ex.message); ( b4 v2 {& \& c9 P) R# U. `" @2 H" l
}
7 \0 q9 X/ p* \; S" @
};- O2 w+ C) Z& J1 n0 _) e2 I- t
$ s! T7 R$ P: p& s+ @: g
function SendData() {5 d9 w5 X+ `4 {8 q; ?
try{
9 s2 Q% ~. \7 U* \& Y
var content = document.getElementById("content").value;7 n: q2 G7 c: Y2 K" h# }% ~0 _
if(content){6 w4 L6 C" N6 x; M5 L6 H2 u
ws.send(content);& @4 q. B. w2 x" l4 w
}
8 h4 @ {) ^6 X( k1 Z! H [8 R
6 q; u: q& o3 P x( \- Y, l
}catch(ex){5 Z L* z1 O. H+ R& ^& f0 k/ a
alert(ex.message);% b# b/ P8 }2 T3 V4 W$ m, O6 V
}
0 t& D$ s! Y! x0 f
};
3 X$ t2 I, H" K# P' L* m
9 A) Z* k4 n) r* [
function seestate(){# ]+ C2 l) \* i7 S6 Z$ i6 @
alert(ws.readyState);
$ v9 |( n4 p g- s
}
$ ~' u& X& D) h& u% S
- U/ W/ X9 I- V/ E
</script>. k! y! o4 a3 h/ \3 g! k Q9 p( l: l: O2 v
</head>$ ^2 _; L+ r9 l
<body>& S# X9 H2 [# c. f; T; l
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br />/ G" W, i2 k& d) j( j7 s+ Q
<textarea id="content" ></textarea>
' s0 l, M: {0 i; t9 i9 \% z5 }
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br />
* T6 L! J; A! e: q$ |2 O/ v/ N% z
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br />; x$ x& t) f7 w- z( A3 ^# b
. K. t2 w% V9 o. i9 ~6 p' [' j( V
</body>
8 g! [. C- J( c5 v. r* {
</html>
1 n) @" \. u# A! g

复制代码

2）服务器端实现

class WS {$ g- v1 q N l+ {* p6 B& c* g
var $master; // 连接 server 的 client
6 V! m( l, c; g6 G$ R, ] f+ x
var $sockets = array(); // 不同状态的 socket 管理+ ?' |8 V8 G# K) O1 P" C/ ^
var $handshake = false; // 判断是否握手/ i2 c8 L, z1 x7 L" ^
& ^4 T4 }$ ?, \' Y- e0 [2 M
function __construct($address, $port){8 x: t8 P! \& l, v4 [, s5 Y- w
// 建立一个 socket 套接字6 [, G0 c" p0 @, q: {
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP)
' G4 p% d' K( @) j
or die("socket_create() failed");
1 Z) M) |6 V; g4 a/ l# D' p
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1)
+ ^; q/ B: K/ q4 @2 X3 s0 Z
or die("socket_option() failed");
4 W+ {' F6 Y0 @/ a( d; J& v# ~
socket_bind($this->master, $address, $port)
# S5 P; ^% @1 s. ~- [
or die("socket_bind() failed");
7 o. k9 ]& w# P, W B9 a
socket_listen($this->master, 2) ; C0 U; A1 w) D6 B2 T* d6 P- `& O
or die("socket_listen() failed");) x) t) |; m- B+ N: y' |
6 n9 q t3 Q! t7 s; G7 W7 d
$this->sockets[] = $this->master;! A7 a: ]; N7 r! ?
4 f0 m! J8 \7 Y( ^. V4 e- Q
// debug: a0 q8 v' s* Z# F
echo("Master socket : ".$this->master."\n");2 n- A l; C0 m9 H, d$ _) B
) j0 y! I6 U* A& H! J% B" w
while(true) {
5 H( `7 ?: T9 h- v8 [
//自动选择来消息的 socket 如果是握手自动选择主机
0 L9 t% Q; J, B1 Z0 ?9 L% B" S
$write = NULL;
! U; |* W; x/ ?
$except = NULL;
; h Z/ x6 U& c$ d! P. Q: G
socket_select($this->sockets, $write, $except, NULL);% i x8 H+ b$ ^5 ?% z
+ `/ t; v/ e$ L& }* U! `: n
foreach ($this->sockets as $socket) {8 |4 n% h5 @0 O; m* B
//连接主机的 client
) u# m( h0 U5 E& s% G
if ($socket == $this->master){
* P2 J8 c2 I8 d- _. |; u" k
$client = socket_accept($this->master);
! q% [+ O2 ]$ V' ?% ^; e) x
if ($client < 0) {
; T9 U3 @) A8 G
// debug& j. r( b5 f) s( k& x
echo "socket_accept() failed";
; O0 y# ~5 d% ^0 H/ c; f% E$ n
continue;
% q6 r+ x C$ m4 |
} else {
! e! h6 t# J% y2 [# \+ g: m3 Y
//connect($client);
$ w/ m: a; o. ?+ R
array_push($this->sockets, $client);! N0 ^2 i; ^' [" m- A# @; n, A0 B
echo "connect client\n";
: \8 I& u- X% P* z' q5 e3 p# G7 t# }
}, `8 X, B' z; X% d% L5 D( K" A
} else {
# R0 @( f, I+ F) U. l" o/ ~
$bytes = @socket_recv($socket,$buffer,2048,0);
1 z2 P- s# Y! {
print_r($buffer);: o( \% b% V/ x( g: f9 }
if($bytes == 0) return;- x! r9 L# H' j- r3 f1 {
if (!$this->handshake) {- Y' |0 M* x- U4 U
// 如果没有握手，先握手回应
E) o% A2 E9 J/ ]9 i
$this->doHandShake($socket, $buffer);
D p6 ^2 Q6 K* ]$ [( K' G0 {4 t
echo "shakeHands\n";! q4 V% `$ X: N: P7 F- y: h5 z( ~
} else {
4 }- C, z8 s% Q" t6 t( b
- m% S# n3 G9 {1 c, l' @
// 如果已经握手，直接接受数据，并处理
8 n/ F" l+ d3 n% y8 z7 z
$buffer = $this->decode($buffer);
- b) s7 e: O5 J
//process($socket, $buffer);
7 z2 \! |/ l. G# K$ G
echo "send file\n";
3 D/ b" X; F- v! |' y
}
# W* N. T& U" j+ D
}
0 t( v0 s2 \5 B* }3 ~6 D
}
6 c7 B2 E |0 F; P
}
, q" O" U" C! A2 l- u/ Z, [; I
}& C' ~ g# m. H: C: x* {6 h6 ], O
: Q8 @$ r- m7 ?, j/ P
function dohandshake($socket, $req), `; C" B/ k( H% l5 \
{
3 c% v/ _/ z0 F: k
// 获取加密key- \& G+ k# b* d b: n) u: G# f
$acceptKey = $this->encry($req);- n* l, T1 _% J- M% ?
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" .6 _1 y. ~* s1 H. Z; v+ i2 D- `6 V' V
"Upgrade: websocket\r\n" .
; \. `0 `) A% m* K6 L
"Connection: Upgrade\r\n" ." y: p- U/ _, Z$ |5 F7 ]7 ^+ f5 t
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" .4 J% o9 p7 ?% U9 {7 ?( v
"\r\n";1 @. M5 Z+ c$ H
: Z: J, `# t: `
echo "dohandshake ".$upgrade.chr(0); * U$ Q. ~# b$ o2 J7 W9 }4 H
// 写入socket1 ~3 r9 w7 G+ a) e$ d
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0)));
: @$ W& u) e/ \& Q
// 标记握手已经成功，下次接受数据采用数据帧格式
/ J3 @ T& F Y7 `! ?) `
$this->handshake = true;
7 C, o( @# d/ `# E: J( q. V7 `
}
) q: V! U- ~! l1 G3 W
2 {2 h: `1 ?/ v3 j. H0 K& i Q: A
! M: y; H/ e* G- `
function encry($req)5 q3 @# m3 I" G+ k- z2 n
{
' ?" D; Y5 e! l% n2 W% z
$key = $this->getKey($req);
* M1 d2 a2 H2 c$ S3 E/ u% n
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";$ b% K' Q& n5 ^* s( k
! y- D: u& C2 W9 d* m0 D
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));
! `" ?- d8 V: a' b2 K9 j
}/ O3 K7 t+ B1 N1 G7 M/ I7 F7 h* C
& z5 n( }) h1 Y6 h: ^# a8 I
function getKey($req) 7 u6 c) q. A* g0 B
{0 o; v4 }, g# H
$key = null;
9 q H3 Q* C: W# b! \% e) Q
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) { ) \. Q! Z. Q7 ^; U
$key = $match[1];
! [7 _9 T3 i: h2 \. G1 x) n4 W( i
}
4 f. r% Q0 S+ B) P, w" f
return $key;( `; b1 r( D( a
}
! p/ f3 {2 F; i5 X, P7 [
1 N/ `9 G; V6 r6 s5 W( G
// 解析数据帧- D/ G- v: K% l6 |/ f3 h, S7 p: G
function decode($buffer) ! o; N* \4 T/ b% N7 y8 n
{) ~/ T+ {0 G# S5 X( v) o9 m
$len = $masks = $data = $decoded = null;
9 g6 a, q1 ]& l
$len = ord($buffer[1]) & 127;6 i) [" w6 c5 ]; J1 i9 |
. S$ \' C6 Y9 V% d
if ($len === 126) {1 ?' h" Q4 J& ~) B
$masks = substr($buffer, 4, 4);+ Z7 |6 `; R7 `5 A
$data = substr($buffer, 8);
# c, p3 e8 A$ T4 b0 o& ~% {% g
} else if ($len === 127) {: n& n; v1 f& l7 r& H+ N
$masks = substr($buffer, 10, 4);3 {1 S) Y4 E. n& K! u
$data = substr($buffer, 14);
, ]: u; B6 U% U. j! L+ u9 s
} else {
( z$ t0 G6 `4 [, U- L# X- w; _, O1 m
$masks = substr($buffer, 2, 4);
* ]- t7 T8 h6 b% D1 M3 z: c/ G
$data = substr($buffer, 6);- Y% o; {, y7 x. Y7 W3 X3 F
}
+ e# H! d+ } Z0 I9 D9 }
for ($index = 0; $index < strlen($data); $index++) {
% y, ]( o4 S8 z& v
$decoded .= $data[$index] ^ $masks[$index % 4];
3 v/ H5 E3 @# S8 Q. T9 U. c/ I
}2 c2 R1 A: H( ^) S, _3 i- s
return $decoded;
' K: d# G( E+ _. ^7 N# N4 y
}
3 w8 c% c" g5 E0 A
; d u/ Q+ `* X4 T# w! a- J8 U }
// 返回帧信息处理8 m) d6 o% _/ G6 O0 ^7 @; L
function frame($s) - R0 B4 Z; x; b7 l# e9 U3 }
{
7 Y, n; H1 U8 Q: b
$a = str_split($s, 125);, U; l0 @3 K& [/ Y" C
if (count($a) == 1) {& S# F4 D+ A2 A: S# b% f
return "\x81" . chr(strlen($a[0])) . $a[0];/ o* i9 I# v% }% p
}
" I) H0 _6 `, H
$ns = "";' M3 |8 l3 w. j1 n+ J1 J
foreach ($a as $o) {
! i, L$ }$ J+ o! [; y3 L8 b1 Y, r
$ns .= "\x81" . chr(strlen($o)) . $o;
5 y9 D% D+ U8 r4 S/ S$ b1 n
}2 M+ I3 n. V3 F3 S4 A# H% Q B
return $ns;: ~# B' u8 t n+ P
}+ `( `% ~) j0 Q: I7 T
- G9 _2 w/ C! w0 d3 Q/ I
// 返回数据1 r# ~/ y& \* s# t. w
function send($client, $msg)
" k3 b( p% j6 I" L$ v F
{5 b6 `5 B$ y* d7 B. y2 K
$msg = $this->frame($msg);( _( l* e4 W" k3 {7 E8 ^! t1 S3 P
socket_write($client, $msg, strlen($msg));
. L$ k5 r+ l: H- @, R$ k0 r
}
. H( e+ L5 m+ _8 ]
}
) F" v6 F/ f$ `! b$ N
3 z% G4 k N1 e( B( ~! v
测试 $ws = new WS("127.0.0.1",2000);" g" E$ ]) j" q4 y+ p$ n
# @( b' N$ @0 p+ F1 t5 N; z9 w1 `

复制代码

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

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