PHP 简单实现webSocket

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

1）客户端实现

<html>
8 E& Z; X0 L" N& }1 y! v# N2 Z6 K! }
<head>4 d1 i8 v* k+ Y% }- A3 R0 J8 b
<meta charset="UTF-8">
: m! @6 j ^+ V2 C" _, D
<title>Web sockets test</title>
7 a; }- d$ ?! U# I& s: Q. _* _# w' W
<script src="jquery-min.js" type="text/javascript"></script>: @3 \2 U: H# ^$ j
<script type="text/javascript">
4 ]4 u# x [, r- A" A
var ws;
. I. J9 B- D% @$ m. J2 p. q$ F
function ToggleConnectionClicked() { & \1 }, u" }8 {" Z; m2 I
try {
8 p% E+ x, n0 F9 O) f" y+ P+ |
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器 ' S$ h+ q$ V. `% P; W
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};; @* _: u' P# X1 j% p% \1 a7 [
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);};. ?8 s* s, o$ `( H" U
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);};
& G" f1 m& N: Q& Y5 E$ b: v1 H
ws.onerror = function(event){alert("WebSocket异常！");};' U7 L% }* J& q4 B
} catch (ex) {
& W9 D0 k* ]3 r9 m; t4 ?
alert(ex.message);
) F- O7 B2 b$ |: m2 t
}
+ l! j* T7 z% E& \
};
, W+ K! T6 o. j2 h1 Q
- F9 B3 j, y! v5 c+ X( J
function SendData() {7 ]7 C( R' x7 u" @4 V
try{ Z1 }; L( A! |, i
var content = document.getElementById("content").value;/ M- q) P% G- C" p6 \# ~
if(content){
8 L; Y6 ^4 ?" H }
ws.send(content);
) |1 @1 t8 ?& H- ^, r q
}
# \9 z6 x" k5 h$ W( ~5 A# `2 h
2 A2 r3 m* K" e( a. V/ c# w! p
}catch(ex){$ {2 X4 b& K# c$ V& e7 L2 [
alert(ex.message);
4 h0 f. g$ |) k3 T' C6 U
}
+ I5 q9 |, ^" E2 o: f6 H) @
};
% L( l% _. H: } N1 B0 x5 t
4 z& o) [& a9 O4 N/ q4 X1 b
function seestate(){. `( r8 Z: ?/ S# M
alert(ws.readyState);
; r7 c0 }3 ] ^0 ?( Z0 s, z; e
}' a5 A8 x' L3 q) e
" U; }, G5 K( M. B4 j
</script>
8 G5 ?% Y0 q' @9 y; D; a
</head>
+ U% V/ [, s. ]: w5 d* Q
<body>4 ]+ z4 w/ p) x G% {; D2 G* {. ~
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br />+ ^' o: b4 j5 H% e+ J8 b
<textarea id="content" ></textarea>* b' Z `) Q+ S7 }1 f
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br />
- G' ]/ q1 X8 }; u" \. g9 U9 T
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br />+ T' H$ J* o% c3 x2 `
7 q9 T0 e9 M7 j, N1 V+ _6 O
</body>' i& s& h3 s* n- J& O8 C/ `
</html>
) R3 f4 t* _# Z' D/ P7 Z9 r

复制代码

2）服务器端实现

class WS {
% T: s" @6 U% [$ R
var $master; // 连接 server 的 client
: z+ N2 p& z y& V5 X p% [
var $sockets = array(); // 不同状态的 socket 管理" t3 j0 c w* g2 T$ E
var $handshake = false; // 判断是否握手
$ y# u$ s9 @( R( n: u+ @1 A# r2 T6 L
. f3 f/ }; P* [
function __construct($address, $port){ }: l& _% `4 L! U
// 建立一个 socket 套接字+ e2 e" k* M8 d) _9 R B
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP) ( h5 k2 D* ]9 q3 M
or die("socket_create() failed");; Z* C5 A" X% u; n$ b5 {, p# H
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1) * k) k; } e: B& F. m( n5 E
or die("socket_option() failed");/ J" R' g1 ?9 L% d1 Z; Q0 b
socket_bind($this->master, $address, $port) / T1 Y* `% D/ @$ [( o/ ]. A
or die("socket_bind() failed");
socket_listen($this->master, 2) ( [' C; T4 ^, T% p9 s
or die("socket_listen() failed");
0 @/ }) q* u7 e9 u5 A$ o; J
/ B" a. {( }7 d5 [% o; @; T
$this->sockets[] = $this->master;
1 K. Z" T0 K/ H$ f& g! ~
& \: E! q. @. D: _" G
// debug, z7 `$ C3 ^3 I' B% t* R
echo("Master socket : ".$this->master."\n");
. B* |! t/ j3 r6 C! e0 G. g
6 ~) s5 M9 N9 [/ p& [6 X0 q
while(true) {7 `/ S. L, p# Q& N
//自动选择来消息的 socket 如果是握手自动选择主机
9 O5 V% G( R2 e5 [! [1 P2 W7 N4 V
$write = NULL;7 j8 L1 ]5 p0 N+ a+ z; r& t/ O' N; X
$except = NULL;) h! Y; l" K2 B3 r
socket_select($this->sockets, $write, $except, NULL);
, H0 b `7 H, O" g! `/ g
7 _4 u4 @* @. X# e4 r! h5 I
foreach ($this->sockets as $socket) {
5 S6 n+ @) y+ c, ^8 R
//连接主机的 client
' X; E7 K0 f' r% M- q# D
if ($socket == $this->master){
y1 y. P+ s% q/ D& _
$client = socket_accept($this->master);
, M" c" P7 @& M
if ($client < 0) {
; N) k( ?$ a$ P" m/ w/ B, G
// debug0 L, G+ _# c, S. D" S+ E0 H
echo "socket_accept() failed";
# r9 D* n; C% e- L, Q* E- R
continue;7 ]( B4 B! x e5 S2 I
} else {1 T7 M$ N: o% c% n: }. A5 l
//connect($client);
/ S9 [! i! E5 d1 @/ v1 O$ k7 S
array_push($this->sockets, $client);' e. o: L" W8 B4 k% C4 [/ y/ ?
echo "connect client\n";6 W6 o+ z( p' s5 J n
}4 ^) H8 |" x# W( P! O
} else {5 e* W6 k5 R' I6 ]- H
$bytes = @socket_recv($socket,$buffer,2048,0);
5 C, }4 L/ x, E9 H( Z
print_r($buffer);! E/ W) u6 N& J
if($bytes == 0) return;6 S& C9 i D3 G: g. K/ E
if (!$this->handshake) {! B! k8 E5 X n, g: `6 _! E
// 如果没有握手，先握手回应3 T1 E* M0 r( o
$this->doHandShake($socket, $buffer);% |8 R/ m2 _+ J/ `7 \$ I* l+ M
echo "shakeHands\n";( ]: J# V; l Q5 q/ c4 @
} else {
) U) g* l( Y& @% B2 U6 J
% T) S3 x- Y5 h6 x
// 如果已经握手，直接接受数据，并处理
. j* f1 p6 y6 U$ Q: B Z i
$buffer = $this->decode($buffer);
& E+ X- ^( t' t
//process($socket, $buffer);
8 D- ^3 H( A) l! ^, }6 S |/ y
echo "send file\n";
- g& C8 l u# m7 ~/ K! |
}
, q6 j2 ^& [/ W( c# l9 F* _0 Q
}
( H& f0 H( N& V$ a9 I# O3 u
}
/ o& i% H7 c7 i
}
: K, R+ j, K3 D! @6 [6 l
}; P. E# `; U1 Q: Y0 D4 A$ p J
7 ]. ], Q }' f2 ^% P- r; |0 K
function dohandshake($socket, $req)5 H- V! @5 e8 N) }! z! @
{
3 n/ d4 [ k( h) u9 S7 M6 g
// 获取加密key- A5 F! Y; B1 | e0 j6 L- H, y/ b
$acceptKey = $this->encry($req);& K) d) i5 m# [1 K0 O: _, }5 e
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" ., k f6 V* W6 r- I7 T+ d
"Upgrade: websocket\r\n" .: d" M. k5 w6 ~5 t/ [/ U8 N9 ~
"Connection: Upgrade\r\n" .
8 u1 l, z" d5 M7 ]
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" .0 n0 _0 h( O$ `- Y, \; I
"\r\n";
( g3 S9 y( A/ \) }8 e+ n& v
; B2 h2 }2 h. A* G
echo "dohandshake ".$upgrade.chr(0); + k9 t7 C5 a3 O
// 写入socket+ x8 A1 K/ A: }4 M
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0)));4 `) V: _# L, ~# ^, ]
// 标记握手已经成功，下次接受数据采用数据帧格式
- ^# g j$ M% M3 j. N
$this->handshake = true;
+ s8 A' v( R5 d' V7 Z+ M
}
: F6 U4 D2 a g# ]* j
^ }* H& _% D; D `4 p/ M
: M% ~' ^* _7 G2 q: @
function encry($req)" W4 C+ ]$ `5 Z; x
{
2 ]' x: l! D4 j2 ]& {
$key = $this->getKey($req);4 m- F0 S7 j9 N- P
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";" ^. {0 i! {2 a" k
% _! R% D/ j" c$ ^# q5 L* d, J7 q
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));
+ L( V/ g, q8 _ o% j% [3 E
}
/ N* ]5 [' ~. e/ B. f- @0 q) @
) p6 l7 Z" z0 t$ X! }. r
function getKey($req) . O! N" @# v/ O. o+ ]3 H
{; p/ b7 T) _4 V" w4 V: Y3 ?+ {
$key = null;
& R8 n+ B/ ?. R7 c7 v
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) { 6 O0 v/ J6 z- h% T. V6 u( v
$key = $match[1];
& T" [, T2 i$ g; o- J0 J, Y
}
# x6 I9 f& y- v( @# D
return $key;, A R5 \$ Z( N+ o0 o
}
7 k* ]8 X2 O, r/ W! e
! t. t, c5 I/ h# R8 R3 j8 U
// 解析数据帧4 V& w: y2 t7 [+ K+ ]% D; F
function decode($buffer)
- c5 C: M- C) R8 C( G8 [* m
{
+ V- f* u: J% W
$len = $masks = $data = $decoded = null;7 U( }+ u0 N$ G4 X2 z2 l2 R, k
$len = ord($buffer[1]) & 127;. L7 }0 b' A! |2 n( H* Z. E. m! Y7 M
/ W" @# H3 j' s7 `% ^) S) B' E
if ($len === 126) {0 O. m4 G9 V: T! J& a1 u: G- Z
$masks = substr($buffer, 4, 4);
: n9 U8 K n% \5 G# b
$data = substr($buffer, 8);
' [6 D) x$ D8 P$ Z* n3 ]) m# T
} else if ($len === 127) {) w! P' i3 q1 t, W
$masks = substr($buffer, 10, 4);" m% }1 |: B! Q; Q' r! Y
$data = substr($buffer, 14);
) c6 W% `7 L3 T; z
} else {
+ t3 f& S' T6 I8 W( P; |
$masks = substr($buffer, 2, 4);$ z& ?2 |. [3 z9 z2 L
$data = substr($buffer, 6);1 f }; G1 q4 H* r2 [7 g- D; {' I4 A
}
) p& T0 s" R; D# f' W& U
for ($index = 0; $index < strlen($data); $index++) {0 Y( A; f8 K0 ?4 I( |+ r5 g" M
$decoded .= $data[$index] ^ $masks[$index % 4];
3 l+ e+ N3 c3 C. `( w
}2 O- G9 G5 q2 b0 t6 m" J3 n B- O
return $decoded;0 t) J) x7 c; X: `. E
}; o4 o6 w% j5 \2 \1 b# S/ {
. L1 `5 w1 f- ?7 Z; u
// 返回帧信息处理6 `+ k# v8 M0 b4 p; d G% Z$ t
function frame($s)
( c, ~$ q! B" C8 F
{4 S' c6 M) \2 `7 D- F
$a = str_split($s, 125);! o5 F0 n9 C9 [- M& L1 t
if (count($a) == 1) {( {0 C$ O! ]4 [. Z' L% V( H
return "\x81" . chr(strlen($a[0])) . $a[0];
2 J. T1 i; s7 P: Y3 V$ g u, }
}
* q3 j5 Y) N0 b6 L* f* ^9 P6 M
$ns = "";6 S' E( Q9 W7 @ _' w& o0 L" P6 Y
foreach ($a as $o) {) h( ]7 Z# [1 O/ p1 H
$ns .= "\x81" . chr(strlen($o)) . $o;
- X3 O. O7 R- `3 O9 E; o$ k. ~
}+ |7 s2 B, \% ~5 }. A
return $ns;
+ D7 w( C. A V& s
}( U; x) ]+ w3 m
T# j) i( `) J* W" u! ~2 r
// 返回数据- W- [" H. S3 l+ [% C' [1 M
function send($client, $msg)/ V! p- J0 K. U" B. K! { z
{5 Q; ?+ j' c2 I$ E/ c
$msg = $this->frame($msg);
* k8 E$ Q; J$ S* }8 a" y; E
socket_write($client, $msg, strlen($msg));1 i" v- @" i- R* z: Q6 k0 m
}
; n' x- t, L: i
}
0 I4 `9 D; v, l3 c
# m; p/ U" V/ i! x6 X- n
测试 $ws = new WS("127.0.0.1",2000); P4 D* m& o: i* N2 f
8 j" x% f0 K5 }& N& O

复制代码

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

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