PHP 简单实现webSocket

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

1）客户端实现

<html>! ]; Z5 R. r! g+ W5 E8 p1 M s9 r5 R& U
<head>% R- u+ u! q Q7 h* l- Z
<meta charset="UTF-8">4 [; _ N4 O- N# G4 {! q1 U
<title>Web sockets test</title>& a1 E. P! z$ u
<script src="jquery-min.js" type="text/javascript"></script>
& C* e# D0 U e, G# D
<script type="text/javascript">& }9 k4 V% q& { k. v
var ws;
8 x; A! W+ F/ x- Y- b+ O
function ToggleConnectionClicked() {
) o/ [' D/ d( ]/ u( ^$ a
try {! i! \- y6 g8 a- r5 I0 h9 I, D9 x* l' R# H
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器
2 K6 }! x" U3 D, h3 ~7 f
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};# Y! ]5 u4 u {
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);};% o3 P y9 X% m( C
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);};
6 O$ _$ q9 o. t
ws.onerror = function(event){alert("WebSocket异常！");};6 Y4 y2 ]. `& a6 B, U
} catch (ex) {2 S1 h* a! @7 s$ `7 H3 _1 E
alert(ex.message);
7 I l2 m$ U0 r9 q8 k T% ^( u& l
}
+ S7 `5 `; h) u+ l' D8 T
};
" |5 ^" N* E# I
8 r! K) Z* g! l
function SendData() {+ k# |; u$ q* B; h
try{
3 `5 Z2 Q8 G( C. u6 z( W8 c. L
var content = document.getElementById("content").value;; V0 V3 T! }7 k0 y# b: G$ j8 [0 J- r
if(content){
% A; Y$ { N& {: l
ws.send(content);, m" C. ?' J( H4 t
}
: g* c6 z8 I2 B
( h& P Z: K. H" O4 V: w
}catch(ex){
4 E6 \# H/ M: u$ k
alert(ex.message);5 d; j; t7 J7 I
}
6 J' B( i$ J2 f/ A& B- L6 Q) u
};8 j! O' p7 U; f& q- q7 I
9 Q4 `0 n5 D( c, f/ Y
function seestate(){
4 A' G. J O; {6 W2 h$ ]
alert(ws.readyState);
- V& B$ L. s+ p% u2 [
}
! S& x, H7 ^; }
/ ^+ F+ p! G g5 u5 W
</script>4 H4 a7 G4 ]9 n. }/ L8 F( Y$ T% Z
</head>
% J" s5 k. M* M B
<body>$ {2 l& p" |4 @2 o7 p! [7 b
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br />
# i8 ~3 e% L( y1 ^
<textarea id="content" ></textarea>
; W, K9 U% J d$ b. j, m$ m3 W
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br />
' z7 ~5 F6 {1 U8 z' ^; u
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br />2 l* C1 l8 n& {3 Y
& [ E; T7 |6 Y$ J! ~/ F
</body>4 K U6 Q" R7 h; |3 W
</html>
" j1 E+ \6 W! E/ r. z) @( @" l

复制代码

2）服务器端实现

class WS {
0 t ?8 `* H$ z% H
var $master; // 连接 server 的 client7 X/ U3 E; Z5 X/ {1 @
var $sockets = array(); // 不同状态的 socket 管理
1 }0 A# y" g! Q" O$ D! g
var $handshake = false; // 判断是否握手- I/ p: k& t$ E& f* y+ e" z- J
. e" D/ k0 W* x3 `+ t7 x. v) ]
function __construct($address, $port){0 N/ x5 F' N$ v+ L, k" A) {9 Q+ |0 S
// 建立一个 socket 套接字% s' s: }9 T! C# E. G
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP) / d7 P* J. e$ v# Q2 c: v& \
or die("socket_create() failed");
2 |6 }' p- i8 X# u, p
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1) Q6 |2 r! J: l2 e! w# j& @
or die("socket_option() failed");2 o5 [; P2 l* N
socket_bind($this->master, $address, $port) 5 M i+ M* n9 l$ P# S- @
or die("socket_bind() failed");
& l" [' O) w4 @5 v
socket_listen($this->master, 2)
8 t" ]4 T2 I2 e8 X5 X( J
or die("socket_listen() failed");
. O/ K5 T. v* ^* ?
; S* t( |. X$ \8 j
$this->sockets[] = $this->master;# v, h9 x+ G# o
! K' f# E. {, [% K. ]
// debug
' Y; V$ m" j. T
echo("Master socket : ".$this->master."\n");
( `/ l6 a" e5 K7 D6 [/ J! n
A3 Z3 J+ c! l0 O4 n9 h
while(true) {
' h) N# N, z6 [2 V* y2 i
//自动选择来消息的 socket 如果是握手自动选择主机
f4 X# A4 m, p' l8 L
$write = NULL;0 s9 l u& F) ^
$except = NULL;5 U2 m, b! e% a- r u% ^
socket_select($this->sockets, $write, $except, NULL);
4 X/ h7 c" p' t% j; V: r) l7 h
/ u6 S9 D4 w% _8 Q7 V
foreach ($this->sockets as $socket) {- W% `2 d; A3 k$ d* c5 k0 n
//连接主机的 client 4 }, ^3 M: E: Z3 u' v
if ($socket == $this->master){
! g: l& p% _0 O- H/ {
$client = socket_accept($this->master);* k' p$ r( N! U c. f
if ($client < 0) {
; S# ]: u. Z" G% [; h4 _% t/ [
// debug
& O( o5 n6 c! X9 r0 }8 m C" y
echo "socket_accept() failed";2 M7 C# @& e( |! d$ d' ^" `$ J9 ?
continue;# h$ }- c& d6 s, c- g
} else {% Z) Q5 T2 d) h) o" e; ]
//connect($client);3 K3 ~% l7 ]0 E5 q- Q2 U! R
array_push($this->sockets, $client);
# p' k- S) I$ X
echo "connect client\n";7 f6 p5 S4 Z$ ?0 m9 J. ` I
}
: g4 x# X3 T$ G
} else {4 i" p! g% R. s) C) M+ ^$ p2 Z
$bytes = @socket_recv($socket,$buffer,2048,0);
4 X) u! @$ p! f2 _' M$ |3 E, E
print_r($buffer);- f9 i9 ~& D! p1 h1 v7 M, w" e
if($bytes == 0) return;0 \4 o0 z. h( W8 ?! |* u
if (!$this->handshake) {' J* p5 d" B) U
// 如果没有握手，先握手回应
6 N2 I: t0 v# @6 ]/ p% N3 ~' ^
$this->doHandShake($socket, $buffer);' o& m, K" X5 e, J& p) y' ~
echo "shakeHands\n";
) R; ]% M& m2 i3 g9 R. q# ]; s
} else {/ d; y2 \1 v: N1 V; y. W) ]. C
/ y9 B% S1 B8 t
// 如果已经握手，直接接受数据，并处理% p& D, h3 ?! _) S
$buffer = $this->decode($buffer);/ u# u8 v1 z5 m: S$ W/ S t% Y
//process($socket, $buffer); 9 D. L- n" A E' T$ f* U
echo "send file\n";1 k# o& |$ [( l% c
}- y1 p7 H/ b( }. u
}
* Y6 ]/ k/ R Y- D" r% Q
} {! h, ^1 p+ i( k* j9 w8 x5 `+ t
}% |. n( ~& R" D$ L
}
4 t. @& j i' A& @/ \0 S1 r
- U& b% \3 @. q* @" @1 F* l
function dohandshake($socket, $req)& o# s* l9 e4 }. A2 j
{
. e6 Y; A- j% m" u2 Z/ D6 y
// 获取加密key
9 p8 j: k2 f* w
$acceptKey = $this->encry($req);9 _% z0 y- J! |
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" .
; { Y+ G5 ]' d6 Q4 V
"Upgrade: websocket\r\n" .
2 V+ F6 I6 Y/ }. ^$ O( ~- J
"Connection: Upgrade\r\n" ." M; K# @3 p, |0 K. t) }2 P/ x
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" .
) Q9 p4 ]' G* z% S! z7 ?0 C
"\r\n";
. U; }6 H+ D% \* W2 R
0 A2 e2 V' T; Q+ h$ z7 Q/ C
echo "dohandshake ".$upgrade.chr(0); + C4 Q0 w9 g, Y( b) T: Z
// 写入socket
3 W1 N% r: s" o
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0))); V: Z3 T: w; \1 r. v8 k* ]
// 标记握手已经成功，下次接受数据采用数据帧格式
: P- H; N& Q$ g/ E( y
$this->handshake = true;
2 ]" f% R+ B% y$ f- C
}8 p: N4 g+ ~+ |! p8 k B
2 O2 \5 `' c3 J$ Z# I+ a' B+ U
6 P' [4 m% O* I3 g' I0 L, B
function encry($req)
2 D% t- L* i: \ p
{
$ }& n+ @3 W& x9 v1 y3 Z$ ?
$key = $this->getKey($req);6 d! ~4 S0 j8 e* {! C
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";2 A- x, J* Q7 I8 y; X
2 S. }7 O/ r: D( G: D/ {
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));
' q6 Z" ?9 F2 ~: F/ V6 f6 ]5 p
} D* K: v9 S* w; O) I
) C2 u9 f1 {* ?! N
function getKey($req) 1 e: P+ A0 l) G& {, o- `
{( J3 P4 g% ^& ~+ a$ t
$key = null;
. w0 l) T0 b$ u, `' z+ M& f
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) { ' _ ?0 T1 v! A$ T U' R
$key = $match[1];
# P& `: E; C: h
}& p6 [' \9 d; G# j
return $key;
0 q0 @! D' B2 t- S
}
8 E! n5 L% p, j
4 x( @# T; J9 {$ U
// 解析数据帧
: W5 P5 W) @# |3 T5 v& X
function decode($buffer)
& X! M9 V+ K/ P6 d
{1 q5 f% D0 F+ H% g* ]6 I
$len = $masks = $data = $decoded = null;- J' Y: ?' J. m5 G) a: U- v8 {( y
$len = ord($buffer[1]) & 127;* a/ z k" `& I
' [, l7 d1 G& I$ V0 H! r# R
if ($len === 126) {8 n( }% a1 Y- n8 |! |7 I
$masks = substr($buffer, 4, 4);
, g: `6 G. k$ W3 L5 |4 A
$data = substr($buffer, 8);
% z8 U u# J5 E' R, S% x
} else if ($len === 127) {' g) O1 v# T) s1 _3 j* W
$masks = substr($buffer, 10, 4);
6 a: Z2 U0 W3 M
$data = substr($buffer, 14);. R; E. ^' V K9 D& F; U4 r
} else {
d+ [5 o/ c7 i- q" ^$ ^) G; _6 m& ?
$masks = substr($buffer, 2, 4);+ m9 }& Y# J4 ~6 G; G* o
$data = substr($buffer, 6);
3 P& i% v x- E, q) o9 h
}# O& E7 y5 k2 ?6 D
for ($index = 0; $index < strlen($data); $index++) {) O8 _3 j# _/ A9 G& x
$decoded .= $data[$index] ^ $masks[$index % 4];" f4 N) j$ `1 h
}% N7 U5 @* B) B# W
return $decoded;5 P5 i1 p2 o u ?
}
" s v& n7 ]* c: V
/ k& W" G8 ]- Q% [
// 返回帧信息处理. i3 p) B$ k2 T. W/ `
function frame($s)
+ B) W' G$ b( D# J! T6 ]! I
{
! V! T# }9 n; O/ v
$a = str_split($s, 125);- L0 @7 h/ Y: d2 ^6 n9 W1 E
if (count($a) == 1) {
: M6 r. q; ]5 v" o9 Q6 r
return "\x81" . chr(strlen($a[0])) . $a[0];- U! S) L" {. L) p* W' f/ w
}+ b! O/ r' h9 X: c
$ns = "";
3 m( x4 d$ F- w3 n9 Z2 H' \* f
foreach ($a as $o) {+ J( _, y4 O; w! V
$ns .= "\x81" . chr(strlen($o)) . $o;" A& n0 P' c' d" ?) H6 f
}, b/ T: ?2 U- G
return $ns;
. Y0 R" `- ^# l6 ?9 L0 w# b
}; I+ S2 m5 ~# A! P5 J; u/ `
& M# O/ ?6 O* y. C
// 返回数据
- ]0 V, C J( p! c
function send($client, $msg); ]3 V$ y6 m% J) O& U. N1 v& V! _
{6 [9 N! V9 o) J" ?; k
$msg = $this->frame($msg);
( R" ?2 o/ c) v* O+ i
socket_write($client, $msg, strlen($msg));' Q6 G7 e' x; P1 Q
}- H9 P. F/ g# k- ]
}
E( A) }( Q; ]* ?- K
6 {! y' U+ E- f( x" }6 o8 [0 c' m3 R! O
测试 $ws = new WS("127.0.0.1",2000);" l* N) H4 n3 N% M+ [ A" y" V( r+ f# H
8 ?: c y( B1 z; c9 N8 x

复制代码

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

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