PHP 简单实现webSocket

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

1）客户端实现

<html>, F9 i8 j+ {9 b$ S1 d" ^/ P0 }
<head>/ Q3 ?8 J' M% E+ j+ ?. `
<meta charset="UTF-8">3 N- o, u- R) S: }: m( y
<title>Web sockets test</title>
7 ~7 f; c. }; e6 S; B' Q) B) j
<script src="jquery-min.js" type="text/javascript"></script>7 V$ ?0 i" _" r7 u/ a0 D1 y
<script type="text/javascript">& h1 L/ G1 Z7 I" V U- V
var ws;, A: X* A5 [# x
function ToggleConnectionClicked() {
t6 p' Z, I. t: ]
try {
0 M: W w* I4 m6 y
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器 ; @0 @9 Z, k9 N% R
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};) N: ~5 e& S/ \6 v- j
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);};
5 Q% Q1 I7 ?' X" G( W
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);};
5 o8 G0 z! l2 G' p9 c( y' |# J7 ^) j
ws.onerror = function(event){alert("WebSocket异常！");};% D+ v- B: k7 B8 c6 @# f3 }
} catch (ex) {
4 u3 I8 A0 _' w% r- C9 I+ M% d
alert(ex.message);
( I8 n( d+ U' _2 n3 l" P8 v
}. K4 }) @3 ]" d I& K' G
};8 [! q9 h! }, } k% q
8 ^; D2 s/ O, J8 y. y* c5 b6 F1 A
function SendData() {3 \ p6 c M) D- p6 \. a b3 C' [, u
try{
% q, d a3 ^+ N3 K; L
var content = document.getElementById("content").value;
! x$ A9 V3 W Y- H; M( D. O
if(content){: q% ~ D6 X2 [/ v
ws.send(content);" }! ^5 e5 d/ `
}9 w* t# G4 z5 |1 A# o" K
% U9 I" X8 K& y
}catch(ex){) n- Z L2 [# C# p7 t: |$ ]
alert(ex.message);, E) y! }6 m" p- \* O5 t/ y
}+ u* O, @$ Y: u) J% r6 J7 d Y
};
4 X$ r: t E% u5 |6 D3 |
9 J4 O$ v8 z" }# T4 b
function seestate(){+ j$ z3 o' \* Q& X) v e* n8 R
alert(ws.readyState);
, e# |1 p% I; k
}. o: J. \: N6 P( ~# R. q
) s' q4 M9 }( E9 i, ~
</script>2 [% D( L9 w% O4 Q4 x9 }
</head>1 f1 k- m8 ~5 ]% P
<body>
3 _' i5 O% N! _; ~: D' I
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br />& i/ j# @8 K% F3 r$ A) ^+ A1 S5 |! O
<textarea id="content" ></textarea>
9 R) F; } b# z1 g7 S
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br />
- b6 p( {, E, s! I% c$ G9 j6 |
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br />
' b$ H* O* A! s: d' I
* ^; J8 E3 U; A1 w+ w+ i
</body>3 z) u5 M; _) |5 u6 `
</html>+ c& l5 z+ s4 A7 L" B; t' |, F

复制代码

2）服务器端实现

class WS {7 d; w# a) E. E1 d, U
var $master; // 连接 server 的 client
% U8 V: x3 u$ S6 M6 W
var $sockets = array(); // 不同状态的 socket 管理. B: a0 R& |0 ]* N4 ?
var $handshake = false; // 判断是否握手+ N u- ]$ Q) N! C; D
+ E9 O! p2 C' s5 B3 i
function __construct($address, $port){
6 M* U/ W% [' h+ r2 `4 K
// 建立一个 socket 套接字) G& g! B6 U+ {- A0 t6 q
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP)
3 ~1 G- D, Y3 v. [: K
or die("socket_create() failed");
3 g* G# z7 D( |$ w
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1)
5 P! f4 Q8 Y+ u$ Z6 ?
or die("socket_option() failed");5 Z' H6 H- A3 o, W
socket_bind($this->master, $address, $port) 1 E. o7 i- i4 s* ~$ F; Z% ]! P
or die("socket_bind() failed");
8 y+ B# a: G9 n: y
socket_listen($this->master, 2)
0 r* E6 z* b% m
or die("socket_listen() failed");9 x# C# x0 o) c, l' `4 m2 k
+ G# t- I( S: C+ e3 `7 [
$this->sockets[] = $this->master;
3 N& b5 I$ _" j3 G9 X2 X6 J
O! x, h' g% |7 ^* C0 I
// debug7 l% \- o- {' z4 C$ ^' H
echo("Master socket : ".$this->master."\n");' s8 {" l( o/ M; ~% {: i: c
: h; a. N& S: i3 Z2 U r1 C% A
while(true) {
: I% w0 Q1 q5 @5 K5 O% {0 i- s% X$ r" ]
//自动选择来消息的 socket 如果是握手自动选择主机
5 U0 l( D: [' z. b0 G* N; ]( T: g2 {
$write = NULL;
2 ^# b6 m4 g$ c8 g
$except = NULL;
8 o1 G+ F7 S5 R
socket_select($this->sockets, $write, $except, NULL);
- [0 R3 ?# s! |) Z
- c; r6 f, ^4 w. Z
foreach ($this->sockets as $socket) {
) H7 m# I1 k: E) o' K# p' v
//连接主机的 client ; g9 u3 {: L( S' M
if ($socket == $this->master){/ e- P" c" S8 o! a. W3 T
$client = socket_accept($this->master);
; d' m% k4 @' a+ H
if ($client < 0) {
) w) q2 M' `. T" C; P
// debug% l: _7 V8 N' w" R( I7 A
echo "socket_accept() failed";
& z: M8 S6 \& w: l7 | I$ D
continue;
, ]8 v1 C. d) M0 D3 `) T
} else {+ a3 P+ ^0 T5 S
//connect($client);
* @$ K5 k3 ?+ K& G0 ]
array_push($this->sockets, $client);
, ~* }* Y# }# A- _' R6 ~! a) M
echo "connect client\n";
8 F' U$ Q, b0 L2 h% K
}
& C9 @8 `9 n8 T, |! b
} else {
; `! N* g& `) a8 T: f- D
$bytes = @socket_recv($socket,$buffer,2048,0);
1 z" n6 J* ?7 ?2 Q/ E" T
print_r($buffer);2 f8 L1 G2 e) p0 y6 z
if($bytes == 0) return;' s5 V6 ?: h' S! d- g" n& T$ v
if (!$this->handshake) {% _ D# b" F% k
// 如果没有握手，先握手回应
- @0 P' r! V6 }2 w
$this->doHandShake($socket, $buffer);
5 S1 }% b* ^" t) [
echo "shakeHands\n";
v( ~* f; O3 }0 c
} else {! h L! d( h6 E$ C- O
% w( _: f4 z8 ?% R
// 如果已经握手，直接接受数据，并处理8 N& n' \: ?+ F7 O y
$buffer = $this->decode($buffer);3 w" m3 d8 M, z F+ | w
//process($socket, $buffer);
# t" N; S; L X; V. `9 {, x% D) L! S
echo "send file\n";# @* g5 {( q6 G3 D3 g. x
}- q, P% V2 b( [$ Y' L( Z
}( L( l* l4 C M8 f# J. U
}( E8 j, P! t. [) W g" B
}% \" ~2 m3 i! u$ e; g# w
}9 t& @, Z% u1 f
3 u3 B& ^% |2 X8 p
function dohandshake($socket, $req)
9 Q' _4 S7 c- Y" m/ B( C
{. n( D' U' x w* |* {" O
// 获取加密key
; H* n4 y' v' I2 B2 u- X
$acceptKey = $this->encry($req);
8 [7 Q, ^, I0 M- _, _
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" .
9 J" e0 [7 T$ H3 w& j5 M
"Upgrade: websocket\r\n" .
( t& r- P: I# d9 z7 l2 D$ m
"Connection: Upgrade\r\n" .
5 M) O" ~' ?* o7 z5 I
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" .
N9 T8 h/ M/ S1 Z) V( h
"\r\n";( O" Z5 {; {6 J7 F, K% i. a# W
4 K0 m3 a, u* K1 k0 p8 _. ^; P. p
echo "dohandshake ".$upgrade.chr(0); n1 { t7 |- L5 |* s' g& J
// 写入socket; L5 t+ c9 Y! |3 L2 w, D
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0)));
0 }) u& V0 S. \& [& V8 R
// 标记握手已经成功，下次接受数据采用数据帧格式
7 y) i P# e/ I# M9 l1 c
$this->handshake = true;
/ v% t2 @8 t8 s6 L/ D, m9 a! S
}
% W9 j/ {6 {7 v
" k# h# ]9 G: J, W9 b$ M0 G
% `0 R2 F7 [1 L [$ F% X
function encry($req)
+ w5 I' w( B7 C8 G+ o
{ O( J( y* J; r# c6 B
$key = $this->getKey($req);2 k8 U# d' `6 @ J" `3 x/ R
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";
/ p2 _2 w. \' _/ m
: Z6 {. ]( R" I G( k+ o
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));
5 d+ K; L3 l3 P K2 N
}0 l0 p) I1 d4 C q# Y1 O* V: M
5 z, V( Z, |# A: d9 v
function getKey($req) 8 f b3 M/ F! u( x/ I4 ^+ v
{. l) p8 R9 t' y5 A
$key = null;
( A6 X4 l( A5 k' N
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) { 1 R2 q* a( @& k2 n8 | K$ D
$key = $match[1]; 8 t. H8 z3 X1 u- ^9 L
}7 U" D* \ t! F& X
return $key;
8 }8 `1 r: y" m: W
}0 ]8 d9 i) Y; W( I
& u# V. ^, Q4 f: O# Q0 t7 `
// 解析数据帧
; c' ?; f0 E. i, p0 Q
function decode($buffer) 1 F" o4 \9 Q; G! H+ |9 T
{
8 @0 z5 T/ } `- [* J) Q
$len = $masks = $data = $decoded = null; a1 p& `, q+ }. ?
$len = ord($buffer[1]) & 127;
- ?7 R3 [. Q: p4 Y/ z, l
$ G6 y0 U% L& k
if ($len === 126) {
: E2 _" Y) ~6 [3 D
$masks = substr($buffer, 4, 4);
9 X6 b: R& K% Y9 y& B
$data = substr($buffer, 8);" Q6 K+ C( @0 h$ V* `8 V, n3 e
} else if ($len === 127) {
8 k( l4 s7 C4 }0 }
$masks = substr($buffer, 10, 4);6 _. c; h# T& O' R
$data = substr($buffer, 14);* A3 \- m' ?" L4 ]6 K; R$ F& U
} else {
$masks = substr($buffer, 2, 4);
4 a3 s) v( H9 M3 P7 }2 K
$data = substr($buffer, 6);
5 a, N8 u( x$ Y2 T
}
; i, R s m- J- ^
for ($index = 0; $index < strlen($data); $index++) {
' \4 d5 O3 q p5 |( ?2 }) x* s
$decoded .= $data[$index] ^ $masks[$index % 4];
* E) s7 B0 p$ y" J7 s
}
1 x5 ^! ?9 q! ^4 }. c9 B7 p" d* \
return $decoded;
- R0 Z: r* }' b5 H
}! `% W. [; r- Y+ h) }1 j" E3 E
$ o; @ Z" _4 t2 y6 z
// 返回帧信息处理* a) u8 I# b( P3 M/ E
function frame($s) - V7 ]5 L+ m& ^: J- z. }: [2 d
{
1 i$ }3 ^5 P+ C1 B% @1 S3 x u
$a = str_split($s, 125);: a6 Y: a" k1 J% D6 V
if (count($a) == 1) {6 H2 C( ^/ y- v. r
return "\x81" . chr(strlen($a[0])) . $a[0];( s7 F) V1 P+ O$ |1 O% [# |
}) r( q# v3 X# f E0 @0 W4 k
$ns = "";
0 _' e6 g" Z* N ?8 w
foreach ($a as $o) {
: P/ z! E* X; Y- J
$ns .= "\x81" . chr(strlen($o)) . $o;1 b3 n. s$ b0 q8 q2 v S9 \
}7 b" ~" c) A- b1 i3 ?" o
return $ns;, D5 [0 {$ |2 j
}
$ D0 |1 q- G4 |7 K
( X5 I8 C% U( a
// 返回数据( H* h2 x" Q, I% B
function send($client, $msg)' j8 w0 S- N9 I, v
{6 F1 b% K" y# n
$msg = $this->frame($msg);
& Q& b4 X0 W' q7 l' j9 h
socket_write($client, $msg, strlen($msg));
# n. S6 L h* f' ?2 Q1 f8 b: W
}
! o5 r6 ~2 d7 _- u+ {& l
}
# d4 U; t! U# C! }' f4 z
, }1 O+ [) R0 E8 k/ B
测试 $ws = new WS("127.0.0.1",2000);
* a5 W F K' r0 Q! d; y# S! J
! _2 l% L9 ]$ _& N) v1 W7 B

复制代码

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

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