PHP 简单实现webSocket

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

1）客户端实现

<html>: E& e& R8 G! f( g# x9 c o
<head>
) ~ i- @% U/ n6 k
<meta charset="UTF-8">
$ Q. d+ G" [" \# r$ z
<title>Web sockets test</title>' l5 F8 D6 t. X9 I( h# n+ J2 X
<script src="jquery-min.js" type="text/javascript"></script>
* a/ U2 P5 F3 r
<script type="text/javascript">
+ X, f5 n0 s* G' ?8 k; _- M4 ?
var ws;
6 p3 C* U G) d" Z+ U
function ToggleConnectionClicked() {
0 r2 p1 X: U# _2 t: [% T
try {
+ j7 K! ]: |9 I
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器
6 |6 p) |4 J. L! x& c+ H
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};
: S( I6 k/ } W" Q
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);};
9 m% d- O# I, q1 p! q
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);};
% i: t* Z2 w+ u5 N s0 g
ws.onerror = function(event){alert("WebSocket异常！");};9 n" t" Q3 U0 b* X" L
} catch (ex) {
: I/ J6 c% W8 i
alert(ex.message);
# @+ P! W* C! `; G$ I
}8 I. c1 a! W5 ~3 w. |/ p! a
};
$ {5 A' [+ ^$ G4 n
9 q- b3 D" U' n( \$ q+ A5 U9 Q4 I
function SendData() {, E8 T/ V/ l) L! g9 Q( i
try{
. {4 a6 _1 f( b; c" W) E
var content = document.getElementById("content").value;0 G4 z9 N" ~! ~ H+ A3 t( Z
if(content){9 _) O1 Z1 l# S; N& T9 u/ Q
ws.send(content);
. Q0 F6 X6 P- v" M! P7 ?0 {! H0 Y; v. ^
}
% b& q; J6 H+ Z0 I
6 h/ R9 w3 b2 V, s/ x' X5 j* [) z: N
}catch(ex){
, j8 x% L4 Z, j7 I
alert(ex.message); e# _7 x- X/ ~9 K/ I9 E
}
" b+ l- e1 L0 l( z7 G+ r1 e
};: e0 L8 D$ B: X' Q0 k; U
, G: X: H% @& J/ O2 [0 k) e( }
function seestate(){$ Z V% `5 C% ^! S$ \% [6 ]4 S1 h
alert(ws.readyState);
. T# Q: K3 d4 ?5 h8 I9 h
}0 t+ |4 f' S' ]5 D8 X" I% m2 R( M
( @; X. Z; c9 c1 E
</script>
- S! D7 Q7 T5 l7 ^5 ^7 g4 ~
</head>* p: C+ i- G4 j% B5 Z( e
<body>1 c( C0 @- M% D1 Q
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br />
2 C3 c) M: [- v" y) _
<textarea id="content" ></textarea>
O( n9 z, f" T: q E s) Y! x
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br />
( \/ M8 R* N5 n- X- c& J
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br />) o( U: y2 O7 z4 M9 C& Z6 j
: T9 v" Y) Q2 k. N: A+ l( B% t
</body>
) b' [8 D3 h- g6 r \
</html>
1 _) T- \2 D+ {- q7 V% l3 n

复制代码

2）服务器端实现

class WS {( [" E& A; I) E
var $master; // 连接 server 的 client2 ~% W+ y% ?6 `& O$ {; ]2 S9 H6 C1 [8 X
var $sockets = array(); // 不同状态的 socket 管理% A% [% u; E7 Q5 a% }
var $handshake = false; // 判断是否握手, w9 R$ `" f1 B& }) k1 V3 e
* ?7 u" H' I! U* j! m. H( Z
function __construct($address, $port){
K- G; o! b* J( e# `
// 建立一个 socket 套接字6 j4 L7 m4 N! @" P. Y
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP)
% h) ]! ^" ?2 [& O' g1 ?
or die("socket_create() failed");
C- b# H c+ k4 o
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1) 5 O+ l7 J* t" G
or die("socket_option() failed");
6 S8 L5 L. O' O4 |. R$ t
socket_bind($this->master, $address, $port) , k& d3 e" P1 K
or die("socket_bind() failed");
$ B: s7 }) \: w4 P1 ~* K5 G
socket_listen($this->master, 2)
! o2 j. t; Y M: Q* A, v d
or die("socket_listen() failed");
0 r3 \3 `9 k3 Y9 }0 C
8 p g; i ^& l1 l7 N8 f% ?
$this->sockets[] = $this->master;
9 g# ~3 r2 \6 k
% ~# ~) s4 o" X- ?1 |2 Z1 L0 N& R
// debug
: s/ j. q3 S9 C! S: w
echo("Master socket : ".$this->master."\n");. }+ V0 d! A4 t+ }3 A+ v
" g1 Z& M0 a" V# t* a. _
while(true) {
1 g" Y2 ? n# [' t) S. i0 d/ @
//自动选择来消息的 socket 如果是握手自动选择主机) W. Q9 c0 r+ O2 n+ G
$write = NULL;
. C* g2 ]6 X! f. l1 c; |8 m
$except = NULL;
3 d9 |. p7 M3 c& F2 h5 ` v
socket_select($this->sockets, $write, $except, NULL);
+ }8 V+ x. e: T- q) T
3 h; [: P5 C0 _4 _5 Z4 E' ~1 [
foreach ($this->sockets as $socket) {! N! Z$ _9 D9 q F! n" A7 g, j& J
//连接主机的 client ( z, G/ c" Q" `
if ($socket == $this->master){* p P3 p' P; b6 [3 Z% U; s* @
$client = socket_accept($this->master);
! f# t: L8 r. `) K6 b
if ($client < 0) {' A8 b' z J# T6 ^
// debug3 j( k7 \% R+ l4 C: i5 l* @
echo "socket_accept() failed";$ w0 R1 X( C9 L9 _; z
continue;
# o8 i/ i' n6 C! F2 j/ o/ t) j# {' P/ j
} else {( F; a# Q6 J6 p3 b: H/ E" o5 w
//connect($client); I1 H: D* K: N
array_push($this->sockets, $client);$ _ q6 U. r% F, N! e
echo "connect client\n";
2 I. J8 a. L5 q7 v" `7 V! a
}- B9 T, ~, [3 c: B: F6 p
} else {
5 H! g+ d, @" \$ k8 Z
$bytes = @socket_recv($socket,$buffer,2048,0);
% k$ I5 Y( F% w- @7 C
print_r($buffer);. I h9 g* k) r4 R
if($bytes == 0) return;6 P9 K8 S1 G. I
if (!$this->handshake) {5 A: j; Y! ]+ |$ O: ?
// 如果没有握手，先握手回应9 |9 Q4 b- |/ u% U0 ~+ k' K
$this->doHandShake($socket, $buffer);7 s7 l, z$ y( _
echo "shakeHands\n";
( c( I0 W& \7 Y/ x
} else {
1 \& K! }9 o! M$ m: C6 S
4 b' E7 v* o( E$ M5 m5 a& {
// 如果已经握手，直接接受数据，并处理7 ]- u" \& e8 f8 R% j+ C
$buffer = $this->decode($buffer);
- X7 F: g5 D& }7 A$ _
//process($socket, $buffer);
3 z- @* u I, t+ ]' e# s8 V* h
echo "send file\n";
0 R! M6 A& b6 u6 Q! A2 W8 c
}
# @6 C! ?0 q% _7 {
}- p# |$ \/ Q2 U! H" @
}! j6 |6 [, o# H, {9 I
}2 h6 x6 \, o. ^/ h
}+ S9 Z6 Y' K9 @: a
a0 j8 k7 ^, m* ?
function dohandshake($socket, $req)! ?. e" L( t! ~$ r+ ?% E1 y
{0 o2 {+ W) L0 m* G, D# I4 K+ _( D% R
// 获取加密key8 _" P+ K) z% c0 U9 f
$acceptKey = $this->encry($req);
) W( \5 u- |8 r- u; M5 g- f% l
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" .8 n7 t3 \* N7 j4 v; a) X! ^
"Upgrade: websocket\r\n" .
, b; K2 Q. W0 f; v/ d; F M3 L, U
"Connection: Upgrade\r\n" .
Q$ d: r2 ^- n) G2 m
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" .: |; L+ t; I/ s& v
"\r\n";
9 ]5 x, t* _ R$ [% R
5 k! }, i) A& q
echo "dohandshake ".$upgrade.chr(0);
: L W3 D! ?: W& Y! Y8 Q
// 写入socket
5 Z9 ^2 V# ^4 v" Y `0 x; g
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0)));
/ ^+ O9 l0 _; d7 p- Q4 y
// 标记握手已经成功，下次接受数据采用数据帧格式4 Y- t r- e) f! Y
$this->handshake = true;
. t+ M+ b) J9 }% u8 Z0 L
}; ?2 I: c; @4 m" C0 h9 s& {8 A- A
: L; {8 [% X5 Z% ^$ @* ?6 z
[" m1 S& v7 Z
function encry($req)
3 ?0 V+ o3 g! p9 c( y
{) e8 B7 f) S" b
$key = $this->getKey($req);
! E& W7 n3 q' k' ~+ q
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";
# w! V9 h+ e) G* c
4 D% L) r$ `+ J* h' X
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));
0 Q/ m9 B/ F! D
}. B# Q% S$ Q/ k0 _) ]
0 D+ e# o- c5 A2 n1 L
function getKey($req) 9 h- l. _% h8 Z
{
7 ~1 D c% y0 K x$ y: k p: F
$key = null;
7 m( G* w9 k5 ?7 x. q) O# w, g
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) { ! S0 z/ G( o, o# A6 w
$key = $match[1];
! I' t/ s) I! ]+ F( t5 G
}& B/ @4 i; J" h- e- J! K
return $key;+ o# D5 f4 y6 e2 N
}- [( s2 X* s! `% ]) f; p z" {
! X) K# R0 U/ e2 u% J
// 解析数据帧& m2 l! u+ ]! x* \
function decode($buffer)
* h. y& ]7 F2 X! Z' J5 P- k
{4 t4 N) z0 K# r& w$ T! N
$len = $masks = $data = $decoded = null;- X5 j- J* h3 j$ Z& D
$len = ord($buffer[1]) & 127;9 |7 {2 F( n7 X! D
- s3 F" `' H1 X7 } D/ y
if ($len === 126) {" Q5 ^1 [" `) M y% T* `8 [6 Q: O
$masks = substr($buffer, 4, 4);; Q/ L& r! [$ w( ^0 B2 v
$data = substr($buffer, 8);4 e: d3 Z5 z0 q/ w
} else if ($len === 127) {
9 p% B( T5 c+ \9 u1 \6 E
$masks = substr($buffer, 10, 4);4 V" _7 K5 O7 \: F u; p6 X
$data = substr($buffer, 14);$ o; N8 \0 f7 \+ \( ~: W7 q
} else {% [; E7 ?$ j, x! t3 o6 S: S
$masks = substr($buffer, 2, 4);
$ l. H# `2 b- e
$data = substr($buffer, 6);; H8 V$ W0 g S/ m% G1 D, z
}
# H( X4 G. i* ^0 `" \
for ($index = 0; $index < strlen($data); $index++) {4 u: N6 X \: }9 E) ]
$decoded .= $data[$index] ^ $masks[$index % 4];( t$ r9 x4 S+ H8 v$ }
}2 X5 f. X! a* c& Q3 C1 U
return $decoded;$ m# O: g6 [( r0 t
}1 o/ b; w* Q; R5 k' i5 ^
- {+ W6 x( `1 T# q
// 返回帧信息处理4 k& G4 S& P& A6 J+ C, n
function frame($s) ) \1 a# ?' P9 q! `, X6 M( X
{/ [3 G; ?& P4 {
$a = str_split($s, 125);0 m4 A. W7 ~" W' `
if (count($a) == 1) {
! v$ M Y6 H7 r1 X) a* B# z; ]
return "\x81" . chr(strlen($a[0])) . $a[0];! E/ i% |+ o* q2 t/ G
}, U+ r" O0 K* D/ P" T7 y
$ns = "";
9 V$ a! H+ U. ~7 P' T7 `: d
foreach ($a as $o) {
+ J6 Q0 b( E v0 e* K6 q8 A+ w/ d) i
$ns .= "\x81" . chr(strlen($o)) . $o;1 u8 N$ C! b; R5 Y9 W# ~' R. J8 X" s T
}+ N& H1 [; J, W2 O9 d! b
return $ns;
: X& {8 _; {; p J9 g
}
- A: k5 N) z0 ? ~4 @2 S: w
: n5 ^7 }* W+ `3 D
// 返回数据
; u7 L9 a3 l8 D
function send($client, $msg) @5 p% m+ \; L- O: e6 j
{. z r; ]. m9 m; G6 l3 P0 c
$msg = $this->frame($msg);2 u8 _# v2 M! _6 ^4 q% }, j& M
socket_write($client, $msg, strlen($msg));
% ?' G5 `& a& z7 R+ K
}
& p" o4 w4 y, K
}
, ]6 z; F0 Q- X3 |
4 Y. x" l+ v" [2 R. ^3 o% o: @- y
测试 $ws = new WS("127.0.0.1",2000);
& L; @# S& w* D9 [
# U# y4 f; b7 A/ \, E7 F* w

复制代码

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

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