PHP 简单实现webSocket

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

1）客户端实现

<html>
* g* v# S* H9 Z ^" ~
<head>
0 v5 C& T0 J, b) z/ q
<meta charset="UTF-8">
+ q8 o( C$ x- L; z3 k* ]3 z
<title>Web sockets test</title>
. k& ?) G7 v" h1 D8 A) H7 u: g, w
<script src="jquery-min.js" type="text/javascript"></script>
' h/ M- o& }, w% M$ w& W
<script type="text/javascript">
9 \* H" n) J2 ]/ ~ g
var ws;, H9 P( S1 n$ g. }8 c, K8 O8 c
function ToggleConnectionClicked() { 8 f! L0 [- W s6 S6 E
try {
% C& ?+ l1 M+ N+ q4 a
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器
* m$ X+ F- M' l4 F7 ~
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};* u% e+ A: T, d' C: ]4 W8 t& W
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);};9 @* c$ H0 D) c/ Q
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);};
p% i% X% o/ t1 {' D' a+ P( y
ws.onerror = function(event){alert("WebSocket异常！");};! @' S* x% @: y
} catch (ex) {
alert(ex.message); : N) g$ t D0 \/ Y' g
}
5 J! f* \8 A5 B% K+ C& p2 ~# X
};" k4 _% l7 m' F: c+ M
5 E' n. c d1 [6 F X
function SendData() {1 h d" y5 x% q
try{
: v0 T0 }6 A: A' Q' Y
var content = document.getElementById("content").value;
; Z% l+ x0 x6 I$ ~, S' o3 I( n
if(content){
9 n* h* W6 c2 N0 q
ws.send(content); Z/ `/ N& C0 |4 B1 _5 |9 i
}
, I8 z; z3 J. W6 G4 L+ d
0 x* H- m) `! J' E- n. E
}catch(ex){
8 m! M' B0 l" J4 L. U2 v+ R
alert(ex.message);
, I2 w: Y7 Z6 @, q) Y# b/ F- q- m
}: \, R8 d! @! |! ^. q7 F2 a, C
};
! R& E. l* F: C% @6 c+ K
: A/ g2 N% _5 m Q- [
function seestate(){) t- c8 o. u7 t
alert(ws.readyState);- o, r+ i& z% n, Z; j. s; q
}
9 S* l( R1 D7 s7 n6 A- B5 z3 p% ~
9 i0 U3 `1 H/ F: e2 [0 n" V
</script>( Z8 N, x# ~3 n/ x) b8 w6 L
</head>* c, @8 C) c, K& m# L* b
<body>8 t/ ?. H4 i- L" N3 a
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br />
: B. W# `% {/ Y9 ?! N
<textarea id="content" ></textarea>0 @7 d, F0 g/ D6 i+ A
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br />2 X2 k. C; U/ r ]/ j0 f
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br />$ ?9 m2 E, D# S/ z
: W* t% R9 z/ \9 V u
</body>1 l6 ~6 j4 o% c( S8 D& W
</html>; x8 w9 S* Q9 m5 E) u

复制代码

2）服务器端实现

class WS {; Z8 W; n. F4 w) y1 x
var $master; // 连接 server 的 client
9 n! s. _; Z, n
var $sockets = array(); // 不同状态的 socket 管理
8 k$ ?+ O- K# e5 X! K; c
var $handshake = false; // 判断是否握手
3 _8 }! X$ d% i& c% I0 i4 | T9 I, P
V; h0 a& H: c
function __construct($address, $port){/ P# f1 V3 ^8 V
// 建立一个 socket 套接字
1 I4 q! i J0 X9 |8 J
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP)
3 e2 F5 g; u: ^& e
or die("socket_create() failed");
5 r7 W' @6 U1 U$ X* F3 l* Q& R
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1) 2 g! |$ ]7 y! y* Q
or die("socket_option() failed");
+ h5 r, s& Q: k; f) o5 s; I* M
socket_bind($this->master, $address, $port)
% }& B9 z) Y( H2 q$ C% F0 j
or die("socket_bind() failed");
4 o3 C, J4 m5 f s
socket_listen($this->master, 2) " W' C* n# v7 z& h0 V* ^" D( h
or die("socket_listen() failed");7 ]1 q" |7 ]* j6 [+ Q5 }
) A- q" I8 Z2 Q9 ]- s# E
$this->sockets[] = $this->master;/ f7 o5 L5 }8 T8 r7 ^
) L# G% L( ]7 v6 h2 O, S3 G
// debug; P$ a, t2 O2 X, ?" d0 M
echo("Master socket : ".$this->master."\n");# [3 ^$ b6 A; M7 T- ?' `8 e" S
: q8 o3 J I5 u- |3 M0 Z
while(true) {( N( g+ R( ]+ J" }" H d
//自动选择来消息的 socket 如果是握手自动选择主机
$ s" c/ w V0 Y+ z
$write = NULL;
+ [4 x. p H' F: c8 t
$except = NULL;0 g2 ?7 r! n$ x& V
socket_select($this->sockets, $write, $except, NULL);$ N* G; u K- j5 X* M
& I# u3 t7 }4 k# r% H
foreach ($this->sockets as $socket) {0 |" H: D0 E( T, `% a' D
//连接主机的 client 1 Z* c! h l( t* \
if ($socket == $this->master){! N. T. u& y' E
$client = socket_accept($this->master);
$ ^( I/ |% X8 m0 w9 P# \
if ($client < 0) {" v$ y# A% t3 j) a$ H* Z
// debug0 W3 _ f) G$ ?; Z1 a$ }6 t
echo "socket_accept() failed";6 n. h# |+ j# o' ?7 w" L9 _
continue;7 P' }4 {% g' o( i. k; ?+ i
} else {
. G% Q8 l7 v" B7 N) f8 v
//connect($client);* X7 m9 C0 ~3 U1 E- [
array_push($this->sockets, $client);
! L( D9 [+ A+ N" D
echo "connect client\n";
' P* a" b4 P# |6 R) b
}3 v5 ?$ F' o* n1 \- `
} else {
& B3 z# t; R* E. _* {4 \7 J, Y
$bytes = @socket_recv($socket,$buffer,2048,0);2 S R# l! F* _
print_r($buffer);
3 r l4 X8 |+ b) h# v( ^
if($bytes == 0) return;
: ^3 \) Y. e8 p1 l( D# A
if (!$this->handshake) {
U; A5 t, d& ]% O
// 如果没有握手，先握手回应
$ j' \+ W2 y2 `! ~7 f
$this->doHandShake($socket, $buffer);
( h, I9 i/ e2 m- t) K
echo "shakeHands\n";' E$ K$ l4 M- `6 T
} else {0 w7 b" G/ u2 {( a6 R
1 i$ p) [" t, o8 U7 } c) Q
// 如果已经握手，直接接受数据，并处理
3 s. B' v$ y/ d- l$ A6 c
$buffer = $this->decode($buffer);$ K. S" t7 }* I- m7 Y n" A, p
//process($socket, $buffer); & \& J& e+ g& A& n) Q" F5 J) Y* w, F
echo "send file\n";
3 X. D* W4 y7 v t) Q
}
1 } C4 c- O( E6 F7 u" w
}- [0 w$ Q6 S' H O
}
1 L- V: y# N. k1 G1 {! V5 a
}2 b* l- K+ I0 @5 \# q
}
5 b, E7 O) M& ^/ Z2 m
0 c1 U0 s3 ? A7 ]2 T
function dohandshake($socket, $req)! c% P, j; N5 \$ [ K; V
{6 g9 d$ d$ W! X/ @- _
// 获取加密key
3 v2 T2 v/ r. n6 c5 H3 [
$acceptKey = $this->encry($req);+ @3 B" H$ l6 k. R
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" . n( E( J4 S& N2 l
"Upgrade: websocket\r\n" .. \1 n! `& K# K) B# ?
"Connection: Upgrade\r\n" .
! O4 C1 O [% W! P0 y
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" .0 h. p9 `3 D+ O, x
"\r\n";
# m) x6 U8 P1 |1 x4 X; _
) O" ^! h. `# X% ]0 v7 Z9 v
echo "dohandshake ".$upgrade.chr(0);
* \: h! w+ v! ~! V( I# u
// 写入socket. W: P" z7 Q5 N5 I, _$ T
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0)));
o2 J7 z" ^7 S5 n4 K8 \8 {* I0 Y
// 标记握手已经成功，下次接受数据采用数据帧格式
1 ^. Z% M2 A: q" f3 V; n7 ]
$this->handshake = true;
* I0 e: v. f" Q
}# \ S0 R, _8 g9 Y1 }) e7 P1 _
: L& S- @) d6 P
1 J9 P4 E) q0 X' y6 O3 W
function encry($req)& v$ `2 [# ~% s4 n3 F' o6 G" l
{! t* I% B; j& e( |! I2 I
$key = $this->getKey($req);
9 j! W' O$ O! o# [. ~+ j2 s! Q
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";6 |* N# E1 ^; o( C4 `: f' [8 s' ~* D. f
6 c4 c& K: @' s7 G- U, X u
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));1 U: y! y" n& h* _" F
}8 E! c2 b2 w! @* D" D
. s; j0 q! j) M# k5 d
function getKey($req)
) { g/ V! L( W6 ^ J
{: _. Z( u; ]" E8 t4 E) x& R
$key = null;4 _7 ^) x9 l s
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) {
) M2 r& f9 U C i* f6 W, Q
$key = $match[1];
) L6 A3 J( _7 U, h! m; E
}
( I& [) i$ O( g0 K
return $key;) f( o( Z/ m$ O; p+ J6 {9 s
}& q* }* @3 x, F1 o) M
^& B8 a" }* Y& J( E/ u2 C& x6 v
// 解析数据帧
8 N2 H- p6 d# ?# f" ^
function decode($buffer)
( H" B4 G6 `: P' ^; K& O) a+ p
{
( e9 ^1 u3 V; |4 ?4 O$ M' e
$len = $masks = $data = $decoded = null;
" v5 I/ ^" {0 n- v
$len = ord($buffer[1]) & 127;
: @9 I, D2 [) D# U1 b
0 Z* ^7 ~; A J2 e+ _: q
if ($len === 126) {2 ~: ?0 [! M% z8 m2 H/ {: Q/ A
$masks = substr($buffer, 4, 4);" K* b- K @/ v: F9 X( C! c
$data = substr($buffer, 8);
; x# r$ J9 W; q8 K
} else if ($len === 127) {
4 {0 v" S7 K5 D4 q
$masks = substr($buffer, 10, 4);
7 F* p1 L) a* e0 }: n- ]( L
$data = substr($buffer, 14);
( J0 ?0 I8 S) Z1 Y9 r! x; \
} else {* t8 {3 j4 [9 G
$masks = substr($buffer, 2, 4);3 ]- [5 y' K& y8 N: l
$data = substr($buffer, 6);
}
; {% D% I' M$ m: H
for ($index = 0; $index < strlen($data); $index++) {
. A9 h" }& I" H! \3 A3 b
$decoded .= $data[$index] ^ $masks[$index % 4];# T! V+ w% K" z+ a! z7 `8 o. N3 a& H
}
6 k9 G# p0 p( H# V( B/ N& N' U% U! G
return $decoded;9 z) t( X. X# Q
}- [, j( {4 ?2 r- i% B4 t
( ? v h# T6 h5 S' S' d
// 返回帧信息处理
+ P% O3 h9 [( Q
function frame($s) . z! g4 }) L1 s# Y# ~2 }
{
+ X: h ]" M3 E0 { U
$a = str_split($s, 125);9 e# e9 t: L& Y. [
if (count($a) == 1) {
9 N9 y- f- J; [0 a
return "\x81" . chr(strlen($a[0])) . $a[0];
+ b# ] R, {! O- P
}5 n W; L. Y! ^: O0 z
$ns = ""; u* u, `3 W0 x( x, s6 |; H* O/ \
foreach ($a as $o) {6 \0 W6 j3 t9 g( i" V( ]4 @ w
$ns .= "\x81" . chr(strlen($o)) . $o;" W. ~) s2 G. _3 U& S
}
" ? h1 p1 @ ]1 n8 G/ Z
return $ns;9 j+ D0 [: ~8 A* ]3 @6 ]
}
' J" j9 x, L- R" G0 s" y
0 g2 W$ c/ m+ k" w8 P3 E+ t
// 返回数据1 ~4 B. a: L1 J9 l) G
function send($client, $msg)
, d+ N! l$ N7 q& d, I# R1 b
{# R* [+ T; k! @
$msg = $this->frame($msg);
" [+ i: M$ a, Q# {' A
socket_write($client, $msg, strlen($msg));
* E* s7 o# Y* y
}
8 Q$ c6 O9 F* {$ q) n& }. I
}+ n& o7 @9 o) g
! v) x/ b: {& p, }( h7 G
测试 $ws = new WS("127.0.0.1",2000);& M( v1 ^& j& E: C; G; ?
* B7 I- }; W+ G

复制代码

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

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

浏览过的版块