PHP 简单实现webSocket

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

1）客户端实现

<html>, N, ]2 ~# _) {2 ~; i C3 `
<head>6 [- A7 z# j6 ^' C |: m7 T9 |2 A; ?
<meta charset="UTF-8">
3 r( @. R* J- y! s7 o) C
<title>Web sockets test</title>
/ E' s6 C) u- I* F3 L' N9 a/ l
<script src="jquery-min.js" type="text/javascript"></script>
8 T4 j& h# n" A* s
<script type="text/javascript">6 L+ P. ]/ ], l8 K3 |; E1 ]! F2 S
var ws;
! }0 a% E F- n' f
function ToggleConnectionClicked() { : c: k$ t% q1 h Z7 Y- Y1 {
try {
K! ~& o' q- o- N2 v4 X% p8 M( ~
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器
. m/ a0 k, A2 g
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};
# Q: h! e Y. |! e6 a9 ~
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);};6 V* w. Z' |- \% h% R+ {
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);};
4 G( P3 u+ i8 v- j$ r
ws.onerror = function(event){alert("WebSocket异常！");};6 T7 [3 o( d: x0 J' I) ~* O' z+ ?
} catch (ex) {+ U; @* B: O' p$ |
alert(ex.message);
- e! D, v T' \4 z: r7 i
}" h) [# t$ ]& v7 M. j
};2 C) {5 ^5 s3 P( j
$ @% t" ~, S+ h- B2 G
function SendData() {4 h7 h" G% ~! y8 Z4 n3 I
try{
+ J- X1 m% g- @# r( H& \- `
var content = document.getElementById("content").value;8 H/ ]. p# R4 H. ^: t j
if(content){* N; |$ I! U- P4 ~' O s" i. W
ws.send(content);) Q ]4 V7 t8 j0 {4 V; x/ H. B
}- }' ^) T @2 b
v% a( e: s1 V
}catch(ex){% s! G, }' A- |
alert(ex.message);0 |7 \# Q7 H3 _* g8 w
}
, Z1 j9 [5 }) y& s4 V0 S" f1 ?
};
! `6 P; a, z: m+ s% P
8 {/ K5 O" V4 j8 ?% s
function seestate(){0 V; K/ y' Q9 q/ X
alert(ws.readyState);
. I9 T3 x' h, `7 U
}
3 O" }8 f, N/ K' h' E, i
2 [! j& n& }5 k
</script>; S9 z1 z o& Y# \
</head>+ [0 e: V8 B* |/ s1 h* p0 |) c D' J
<body>% q; g' P! |: j& z- ?& h U6 f
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br />+ }, V5 A2 g& `4 i0 B8 d3 C$ O; E5 t
<textarea id="content" ></textarea>
1 X5 v! g/ u& n$ x: ?9 l) j9 g
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br />
3 ?- |3 c, A4 e; E% a6 v# d5 t! y: c
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br />" r& s' u0 m- z' ? h0 Q: N% n
- e* T% _4 i D( E% a; I K
</body>4 A' W0 _* H. z% J
</html>
- }. E7 k( n( O& }$ c. N! Q) d8 F

复制代码

2）服务器端实现

class WS {: o, ?, B" }3 J
var $master; // 连接 server 的 client
8 |! D) s2 c3 P0 d- m+ G9 e2 d
var $sockets = array(); // 不同状态的 socket 管理
5 V. m# `0 v6 U5 { I1 h5 |9 T' X
var $handshake = false; // 判断是否握手
% I/ z; R! x9 U7 g$ h, Q# N- G5 F
3 [2 g# X6 S5 q/ x- \
function __construct($address, $port){: D4 [8 L* C2 W
// 建立一个 socket 套接字( O. ?9 ?$ c+ s
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP) " p0 j9 F* m9 y) F
or die("socket_create() failed");
- [1 Q7 H- N6 [; T5 X3 [0 D
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1)
( \& H) L8 ^5 u
or die("socket_option() failed");
. l0 o, H- }; n8 d( O) o
socket_bind($this->master, $address, $port)
- X; r8 k6 S. D" f
or die("socket_bind() failed");. T& t) T i% Y0 t
socket_listen($this->master, 2)
% n0 o8 g7 Z u0 M' ]8 v7 ^
or die("socket_listen() failed");
* `) H) f# A% y) s) z
: P3 @) ^ G4 S$ `9 l
$this->sockets[] = $this->master;% B- G/ |4 p' a
/ ^( B; o! _8 `
// debug3 g) E0 R* |& h Q. U
echo("Master socket : ".$this->master."\n");# j7 A8 t0 p1 v q6 n& k
9 _. s- Q* H# P) P$ @/ ?* q: P
while(true) {/ o5 t+ w0 T2 v8 o9 U
//自动选择来消息的 socket 如果是握手自动选择主机( U" J# J, J" J/ R, Y
$write = NULL;/ _9 G4 n' y% ?' O
$except = NULL;: i. O% K# F7 ]) @& v
socket_select($this->sockets, $write, $except, NULL);6 n- }* s3 L% q/ r: q+ |; K/ Q+ c$ ^4 ]
0 ]$ H( u$ F" H8 a4 }
foreach ($this->sockets as $socket) {& L* m& }/ t0 z8 w% p
//连接主机的 client
7 H) y2 }2 E/ Z+ c; o, _
if ($socket == $this->master){
i& c( x9 g. H/ X6 H* ^
$client = socket_accept($this->master);4 ^. t& |5 _7 B4 B S& {
if ($client < 0) {, v5 M7 ]" G2 j1 p
// debug
" J9 r* w7 ?2 d
echo "socket_accept() failed";5 |, N* c- L+ U) K4 C2 h: k8 {
continue; P9 P' G3 J; r# U7 b. l
} else {$ I! n: X0 B9 D6 }, L
//connect($client);
: @( l+ ^8 \: K. q1 O( Z
array_push($this->sockets, $client);
- L- _% |, w; u ?4 b' J
echo "connect client\n";
2 a4 y- A7 l5 A1 l$ G
}" T; P6 h$ F- R) P6 P8 Y9 _
} else {" }- I3 U7 G5 @7 ?' J
$bytes = @socket_recv($socket,$buffer,2048,0);
: c; s2 a$ @' Q
print_r($buffer);
2 i; D+ f6 g0 y( Y
if($bytes == 0) return;2 B! J9 [7 L3 l5 s* r
if (!$this->handshake) {8 z; z" Z4 o8 \( ]# U- Y4 o: P
// 如果没有握手，先握手回应2 Q; N: u0 S1 ]! R* ]# d
$this->doHandShake($socket, $buffer);
! N& h) D, ]9 a% x$ e3 d1 V
echo "shakeHands\n";
# _" p9 e! ^% h, k/ u8 }
} else {$ C0 Y. r7 ] a; u6 P2 o/ X# W7 o$ s
0 S% s: o& T1 {3 {; c0 H
// 如果已经握手，直接接受数据，并处理
Q J. i# J1 S+ Z u
$buffer = $this->decode($buffer);
" I3 L0 R1 l0 M, S; H
//process($socket, $buffer); 6 a9 n. _0 Z% m$ W$ y! G5 E
echo "send file\n";
/ r- ?- q9 P4 q# K2 ~' I; r* M! `
}
1 B9 R, q) B0 R0 h: k
}4 b+ H- B$ d, a+ \. Q, k2 u
}
0 k- c/ B2 \; z0 u( \& O' C
}
: ^& S7 z& o6 i
}+ t6 w4 A$ w, N0 S
: @# L: ~ c/ x8 s* V2 H
function dohandshake($socket, $req), F/ t& l) `+ m4 r2 V3 l
{
- Q s. }% R2 {+ C9 x: U
// 获取加密key
. U0 @5 e0 e2 k7 y
$acceptKey = $this->encry($req);
# Q4 b8 p$ n* A7 |$ Y, ~' @: w
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" .; j% S# X3 P5 M
"Upgrade: websocket\r\n" .
4 X3 b8 }' R1 s1 ], a) l0 B1 \
"Connection: Upgrade\r\n" .
; t5 o- o" M+ U# q" m- Z/ ~
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" .
" x7 Z0 ~. |. ]& p
"\r\n";+ M1 U; u7 K6 W( j- c' l
: M- \/ |: l0 e$ [8 I+ z2 r5 _
echo "dohandshake ".$upgrade.chr(0);
% ] z; Q$ a) X; w2 ^
// 写入socket
; j J4 m- W: Z/ N( O! l- u
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0)));
6 L* V: P: V4 _, U- D i
// 标记握手已经成功，下次接受数据采用数据帧格式3 W, F b( W6 F& _. W4 k& ^, Y
$this->handshake = true;
} ?7 c( C0 `, h9 y; ^0 ]/ Z0 Y
0 F4 t9 }# k x2 Q! _. J: y/ [/ M
: N8 w* j8 n. f) d! E
function encry($req)- j( N9 k% a( E1 Y
{
/ H' @8 ?9 a) h/ o
$key = $this->getKey($req);' C2 V1 E, U$ o! F
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";
; \" d- P1 z5 l6 u; y; M
$ Z- g6 v; e/ ?+ ^
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));
; Q. U7 R0 C% ~/ f
}# I( J+ N* S) D0 \2 C1 S% N
9 I/ | c# Z0 b7 y/ X
function getKey($req)
: q, _6 b* x2 V! m4 G+ l6 A
{
! n' r1 L; q/ z
$key = null;
* q2 T5 h% ~# Y& t9 B
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) { 0 \8 M% d* t1 V+ x0 \# c
$key = $match[1]; . C( c" i$ j6 r7 c# W- o
}# d) |! q$ q9 g, r' @4 J
return $key;9 \$ h8 |' C# [" Y0 x/ b
}
; J8 F" m6 l3 V& d4 R9 P
( @4 [) v6 X6 g' B4 n& |* h
// 解析数据帧
0 r) O7 t% M( ?# K, a
function decode($buffer)
8 I$ K+ U+ j( h0 D) \
{* u, L8 N' k) v; w
$len = $masks = $data = $decoded = null;
- D- B q* H3 r! q; L9 ^" B" G. j
$len = ord($buffer[1]) & 127;2 i* ]% A# E$ z5 ?6 N1 N
( F0 J5 A$ O( \/ l
if ($len === 126) {
% F. u9 G& y2 J& X' D% ^% L7 i% Q8 {
$masks = substr($buffer, 4, 4);; M3 K7 V- Q: L% S" P' Q" E F
$data = substr($buffer, 8);3 K+ H# G3 S2 c; r- _
} else if ($len === 127) {. a- u( F4 m0 L
$masks = substr($buffer, 10, 4);
% O' L ~- I, y7 I! z$ C |( q
$data = substr($buffer, 14);
- w- W- C* T: R F* a- y$ U
} else {1 Y, L" X7 C5 ?( \# g; r) I3 w/ s# l
$masks = substr($buffer, 2, 4);! k" P1 y6 A. U' T: {+ T4 l
$data = substr($buffer, 6); c) C+ ^/ W- O
}& b4 m g9 ] H4 b* J
for ($index = 0; $index < strlen($data); $index++) {% U0 m8 b; F) j+ K4 y( x
$decoded .= $data[$index] ^ $masks[$index % 4];5 K q# \9 ~. k, \
}: u6 J, K: G" x9 M. B/ N
return $decoded;, ?6 k% X6 x* a; H' u, C% D i
}! w# m0 d9 }5 g5 E- ]; i
) M, I6 Y, z# \$ z$ n7 A
// 返回帧信息处理
7 s. {: z% P- K2 q5 |. L, P
function frame($s) 7 v; C+ S1 ~( J, s. r/ ^- }
{
( m- F- J4 o4 c; r, h
$a = str_split($s, 125);0 E0 H. u K& ~% M4 q
if (count($a) == 1) {' w8 I' Q! C/ d" f4 E, s# X
return "\x81" . chr(strlen($a[0])) . $a[0];
& e+ v* C7 q, I4 P; R* T
}- D4 q: L$ f! i% n t/ e
$ns = "";
& t6 C: `5 `8 n' v- P' B
foreach ($a as $o) {0 A; }6 i3 o5 p
$ns .= "\x81" . chr(strlen($o)) . $o;2 Z" h9 ~# v! l( h O* ^
}! \. |2 V; z# K9 f& K) }" E8 y
return $ns;
" D' o9 Q6 i) W9 O1 w
}4 P" P/ b" d& B9 M+ B
# a, R S$ _4 D( h1 {* u( K1 i
// 返回数据# }# {( q' ?" X. j; b
function send($client, $msg)
/ T, @9 S% e) p
{' O# H% Z; q/ S1 n1 _* L0 y! d5 a
$msg = $this->frame($msg);( ?8 r8 b+ @, C% R" d t5 O; s
socket_write($client, $msg, strlen($msg));: T8 I1 v* d! ~
}
! o g) N- g0 J& n4 l5 p: l1 H0 F$ M
}% z3 G7 }) Q) L2 d
: K) M2 I% F% Q# f7 G( }0 O5 G& h' K
测试 $ws = new WS("127.0.0.1",2000);
+ k9 L+ ?! D& K
/ o" _) ]3 }2 }' }# k) G5 b& W

复制代码

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

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