PHP 简单实现webSocket

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

1）客户端实现

<html>
4 b/ T- Z/ ]1 w7 l, P1 y* z( ~
<head>4 g( v# x8 f/ e: U7 k) X
<meta charset="UTF-8">7 K: e& D5 j6 A' q3 [, h0 f: p
<title>Web sockets test</title>$ g0 S- A: t* E2 A
<script src="jquery-min.js" type="text/javascript"></script>
9 z. d8 b# v7 x& {. ]$ W) ^# |
<script type="text/javascript">
8 p1 u. }% Z$ c3 o" W5 Z
var ws;
$ N& x, E7 _" ^) p( ?9 |
function ToggleConnectionClicked() { - |/ D5 u' Y; C; H5 v9 ~5 M+ Z
try {
1 h2 a' e7 _8 }3 N3 a. E
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器 ; e% x* F, X* Y; T7 ~$ x& |
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};: T8 h$ n8 a( K
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);};" Q: q# E( v4 D$ d! S
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);};
4 Y* \; ^; ]' L; ^4 D% e
ws.onerror = function(event){alert("WebSocket异常！");};% n+ x( e3 f6 M% W
} catch (ex) {
# d7 n! ^5 D& ~. j; O
alert(ex.message);
9 |( U% G" _. W1 d
}7 T- ?0 Q1 o, i# W5 Y6 {
};
- D+ A7 D4 T- X
& [! J5 L2 b$ V2 j- {6 X0 a
function SendData() {5 `# c8 T2 _; }# A# [1 M
try{
/ u* u3 P& w/ o t
var content = document.getElementById("content").value;- \. o V4 v8 ]+ w4 D1 S
if(content){4 T9 t% J: Z. K$ [; c
ws.send(content);
! r6 z! j$ O5 G; v' _1 i
}' A' H) u6 X: [- w0 C5 {
4 b: \0 c1 o, d3 d2 x
}catch(ex){2 Q) U; ?) D/ ^5 w
alert(ex.message);
9 i. a/ x1 n/ d6 g9 {1 [
}+ W+ k0 F8 S U% V8 [# R. E
};" V' g+ [; C9 V0 g: {
# x7 k' P ]3 n7 m
function seestate(){
5 z7 [2 @7 ?+ A# G" B
alert(ws.readyState);6 [( u; ^% t# g2 T, @, _2 [
}4 q, q# \6 ?6 G6 ] P5 ?
9 M0 |0 M9 W7 h; Q! \8 u
</script>
! y5 U) Q( p* y/ Y0 j# Z
</head>/ c7 m. e" u M0 y0 b5 ~
<body>5 L5 A+ r3 i& V7 e" B2 Y4 g0 G
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br />; z( Z6 H( C3 N6 s+ J$ k
<textarea id="content" ></textarea>9 ]1 |; c3 m+ u- I
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br />$ l4 h9 v/ w$ O+ @. q6 j( u& b
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br />
' A' ?9 \$ _6 e, E% B4 F1 Z( m
0 |1 g. m/ y, [5 m# w5 ~
</body>( i4 Q1 S6 Y9 I# V/ L) g1 `. u
</html>
4 \, V" i- U8 O# f, ]

复制代码

2）服务器端实现

class WS { ^) M* [( k2 [, P$ J9 d$ P
var $master; // 连接 server 的 client' E5 d; X9 j# X
var $sockets = array(); // 不同状态的 socket 管理3 M. I# _0 {8 L- _9 a6 y. r
var $handshake = false; // 判断是否握手
: s8 } K0 A6 W6 z) g
# V2 V3 t3 @% X2 q& P
function __construct($address, $port){ c% i `* y2 v0 c" T2 v1 ?7 L
// 建立一个 socket 套接字7 g( W/ P) `& E1 R- k2 R
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP) * X8 |; f- s% Y9 L$ \9 K
or die("socket_create() failed");
8 B3 G$ j9 d; @) u
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1) 4 E) k) n8 A9 l2 W' ?
or die("socket_option() failed");( H; [5 m6 j9 @/ Z# V
socket_bind($this->master, $address, $port) ' h7 s+ x3 s% c- v6 D0 @; T
or die("socket_bind() failed");& X. e2 l( W c) {
socket_listen($this->master, 2)
* h6 W6 R; b m4 B
or die("socket_listen() failed");* d6 |% S' n6 P+ L0 G, n0 U
5 E! ?9 R+ e8 F- M! W1 s; f0 ^+ h
$this->sockets[] = $this->master;
! c6 Z/ p. T7 U* [5 l, b- F
5 g& Z7 G: t' n) _
// debug
# K5 S! m. D7 n8 h2 l: f/ N
echo("Master socket : ".$this->master."\n");
( p. @; h% K4 l# B+ I& q
1 v9 L9 Q' s9 v/ X
while(true) {% K# `( g, g! F8 ~: @) I- D
//自动选择来消息的 socket 如果是握手自动选择主机1 m: y* A) O5 F; I! p
$write = NULL;+ M' U9 |* q; T
$except = NULL;. } S: I7 N. x) Z
socket_select($this->sockets, $write, $except, NULL);
/ q* x. T! Z- b- V: r# O* W
foreach ($this->sockets as $socket) {
* ?( G9 g Q- ]
//连接主机的 client # T% e: h% ]5 U- i: C, q) z+ y
if ($socket == $this->master){# S9 x( i/ R1 A. \3 U
$client = socket_accept($this->master);
% K) _2 C6 O: f" u7 v6 b
if ($client < 0) {
2 M1 f' W$ B1 V& U$ M% e
// debug
7 ^( ?/ [# R1 Y/ u, E6 `
echo "socket_accept() failed";7 T+ u. Y# f. l1 q
continue;* y5 J- t- d) q
} else {) i$ e+ V+ J, h2 A
//connect($client);3 b3 J7 k7 A0 j2 q
array_push($this->sockets, $client);: e8 b" h# ]2 B1 A" u9 w. C9 w) G1 w/ E
echo "connect client\n";
$ U) ]$ H" h I! i3 p' O
}
# n# T/ D& a# ^) u, [' Q `/ \
} else {/ L% a" U! B+ M5 k) f4 [
$bytes = @socket_recv($socket,$buffer,2048,0);4 E% O' ~% H! D6 m9 m2 i3 H4 t
print_r($buffer);% G+ d# j' ~+ K. T+ [3 Y' ?
if($bytes == 0) return;
5 W$ `' B0 Y, o% e
if (!$this->handshake) {& e* G- @* J% v% j0 \
// 如果没有握手，先握手回应3 m2 L9 ]0 z! E, a% u
$this->doHandShake($socket, $buffer);) G) Y/ `$ i- b2 F( B
echo "shakeHands\n";# c: E: a3 _8 H8 Z; A% Y
} else {
2 c2 J9 C. f/ P; g: y6 Q
/ x6 j6 o, h. {
// 如果已经握手，直接接受数据，并处理, J( F9 R8 ?% k: }: R' d
$buffer = $this->decode($buffer);( j: i) m3 Q6 ?2 N
//process($socket, $buffer); - @3 Q' o' J, c% j2 W( V
echo "send file\n"; I1 O' P$ L' _- m3 X$ Z% }
}
7 s7 m1 }1 f. n) c0 D
}: z J( c+ u- d/ S$ A A
}! i6 s9 p$ t: q. W) l
}
}# R2 G3 R9 [ X
9 r5 D" P4 Q& b& w+ ?0 h) n
function dohandshake($socket, $req)) x/ l+ Q5 I1 V8 l
{; s" j3 x$ T1 `/ s; [' n; z2 h
// 获取加密key
( O" z; F% u: a2 Z2 P
$acceptKey = $this->encry($req);6 B# N/ F: g& y9 l0 V* `2 ^
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" . V- H6 c8 D; J, ]3 {9 D) M
"Upgrade: websocket\r\n" .
/ f& z1 D4 K5 t" g2 L! @
"Connection: Upgrade\r\n" .: ^& i/ O; O4 t8 N% J2 L4 H: W# T
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" .
% I" F" a9 ]" E/ S# Y, T2 R
"\r\n";' h4 V* L4 `: O0 T0 E
# i7 i' A6 Z) X6 n( w! N9 w* z! k
echo "dohandshake ".$upgrade.chr(0);
# @) F( T0 P1 C; \
// 写入socket( |! @) d: ^$ n# ]" E4 } c3 I
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0)));7 T9 ]% v0 d+ v5 q
// 标记握手已经成功，下次接受数据采用数据帧格式
! s! J5 `+ w2 t
$this->handshake = true;$ O+ | E8 X8 a( r' [' K7 v
}+ ^% q# b# b8 ]& C; A; K, e
3 b6 E# R* D6 S6 K
9 |+ b. D2 M& h5 Y+ j/ F2 ~
function encry($req)
* f; K% n* f- H; X$ V/ T
{
! V/ X: l6 a! \/ o$ W1 K# w
$key = $this->getKey($req);
: i7 K; G7 \& z6 y+ F7 [
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";
" c2 v9 a( I# T# f# \* ?+ k
6 E% \5 z: w7 c! v5 T$ d: _1 X
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));/ X+ l1 f; f; U0 W$ U
}. }' t0 z% g* j: }6 h2 C& [$ M
. k& V& t: ?5 L e2 v
function getKey($req)
4 ?& [9 ~9 v: z% e9 l
{
: K6 p' g: Q, W g) K
$key = null;( E5 M3 i: |0 B! k0 X4 K
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) { & l& l3 v/ [ \# g
$key = $match[1]; 5 d e1 L+ F c3 O
}" ~( S/ @& Z! K: g
return $key;
0 x' E. O' A( [: n
}5 }4 A4 V( w2 s: z9 Q8 _/ Z
4 e# d" J! _' ~( Q1 r; p5 e
// 解析数据帧
' i' G; H) _/ O2 z/ j' j, T
function decode($buffer)
* `: R2 }* T1 u
{' n' u+ R2 c* J, @& ~
$len = $masks = $data = $decoded = null;% \ k. O! y/ s3 O
$len = ord($buffer[1]) & 127;' X; N* e( X( _1 V
. O+ N- S8 w1 h$ y4 w
if ($len === 126) {# p" c x2 I7 \; |) M4 _, U# P: ~
$masks = substr($buffer, 4, 4);7 Y; `/ [. b' o( t, V5 _* D7 a
$data = substr($buffer, 8);/ s$ ]8 f/ V& j! T
} else if ($len === 127) {. c4 j6 }2 M8 `: |- P
$masks = substr($buffer, 10, 4);4 ^7 `, r7 l9 h% t1 X: W
$data = substr($buffer, 14);6 E( a$ u) z7 t$ H1 A& N
} else {
& `9 V( C+ k l2 ~- H5 G/ V
$masks = substr($buffer, 2, 4);
% |: Q' c7 Y* m- | c; j
$data = substr($buffer, 6);2 Z( Q/ s" ?7 Q
}
0 D" q) X; T* x* D4 X1 L ~
for ($index = 0; $index < strlen($data); $index++) {/ w* t* \; H$ x, I% m
$decoded .= $data[$index] ^ $masks[$index % 4];- s# _0 \6 [8 ?2 `% v1 `6 k% }# d5 r
}
" N8 t A0 A; c {+ b; W* U) o0 ^
return $decoded;2 L0 \8 [9 s/ U4 S
}( \6 a2 m- C$ M; |
7 ^) w6 I9 L3 [- ~
// 返回帧信息处理
2 A' I5 x) j _6 N6 ~; d: L
function frame($s) 2 b1 B' R/ {% G9 i5 _* \3 @
{
5 {- [4 f) {; t: r3 n! g
$a = str_split($s, 125);% z6 E) c1 r- o5 ]% o. s& O: K
if (count($a) == 1) {7 ^1 B, S. C# l T7 Q0 G" ]
return "\x81" . chr(strlen($a[0])) . $a[0];8 S+ v) J4 U4 W3 [: @
}
6 [3 X/ A$ J9 I z) J
$ns = "";
, f6 k# @) W. ?- M# C: m
foreach ($a as $o) {
% F3 f4 c7 H" O- S8 n) Y
$ns .= "\x81" . chr(strlen($o)) . $o;+ F8 s/ }: ^$ d9 e% i
}: t0 a# ]& p. X" X) e
return $ns;& x1 E# Z$ s% y) p9 j, _1 z* F( ^
}
& m: k4 x3 G. k; h, A1 _
* C B9 G3 e6 a+ A' m# p" Y! h; L
// 返回数据
- `: O' r6 c7 M. g4 \- s% L, ^
function send($client, $msg)
3 {* Q1 D. g- I9 @; B
{
. p0 ?3 V" |9 J4 l7 N' k1 ^
$msg = $this->frame($msg);
4 a& t a8 m4 u1 @# r! n
socket_write($client, $msg, strlen($msg));5 w. j, g0 U" D
}
8 n6 m" E7 l2 k7 C$ c/ `2 x+ k
}4 U0 q/ g7 s: r* s
6 F2 P2 _! U: ^( l: U. W' U3 L
测试 $ws = new WS("127.0.0.1",2000);
" U% N* O% `% i/ C
4 X1 p8 y) o- T1 O

复制代码

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

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