PHP 简单实现webSocket

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

1）客户端实现

<html>
) g/ G/ }' o5 B, v* n
<head>6 J. c7 I+ {0 i& L4 O, x1 R
<meta charset="UTF-8">
q7 V2 F4 h4 Z9 T) `5 A& S
<title>Web sockets test</title>4 l0 _* l6 s! F
<script src="jquery-min.js" type="text/javascript"></script>% E7 l6 U. _5 \* P
<script type="text/javascript">: Z/ j/ ]9 g$ e1 D0 t: i
var ws;
! c" L5 X% Z$ T$ I) {8 V: H
function ToggleConnectionClicked() {
! w5 P _! q i/ w- b
try {/ d" h9 y* b9 ~6 m/ S! X4 b
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器
' Z; {! e4 L! t* q; m
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};" p/ V+ ^8 q1 b$ E2 _8 G' a q9 [
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);};4 |' h- q1 v8 G2 ?9 z k2 f: q2 E
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);};
" G& L7 M! A1 l _" _" j
ws.onerror = function(event){alert("WebSocket异常！");};; W, x7 M4 b8 g3 E: U' P7 R
} catch (ex) {
z- u& U7 ^) R* _% b( p z$ m
alert(ex.message); 2 I: e* }7 v! L) G& ?
}
8 B7 c4 O) B: m
};$ Y$ l h* I- \% k: b! C2 c& R
- W; V& t- a. a- y( C
function SendData() {
; g# x3 p# L9 \8 U- k! i
try{
7 P$ n+ K( T+ W1 J
var content = document.getElementById("content").value;
: J: M5 i5 V+ f* [" J( F9 t
if(content){% P2 J/ Z4 ^: X0 B
ws.send(content);2 C+ x$ B9 P4 k/ l( w# }
} _. K4 Q, X+ c$ @, N& L
4 D7 y6 T6 p7 n" [
}catch(ex){: R. U( y1 m1 E: X
alert(ex.message);
' X$ D6 H i' G/ K* o2 b
}
9 B _, O. i0 W3 Q0 p Q
};4 ~8 O& N/ g2 v" w7 r s* T
' g8 q+ c8 n$ n( g" R( P
function seestate(){( _0 d: W" u8 Q6 p0 ?$ t
alert(ws.readyState);: ]- Z$ o _$ w- k
}
7 n$ z0 ~0 N6 ~! }
& J; W0 Y5 F1 x/ P
</script>( t1 G+ G$ V( `3 g$ u7 D
</head>8 l' {; R0 `4 G, S, Z
<body>
y' `- ]1 \ w2 T' n& d
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br />- S5 k' V9 G: x; _8 x
<textarea id="content" ></textarea>$ K4 i4 K# N7 x6 M: a: c) i) j* Z& o
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br />
" U9 Q* w9 ~$ P( T$ ^
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br />) r; I% ]+ g, ?$ a
: P# P9 r6 `8 j8 q6 L5 J9 J
</body>
( Q: K* f8 B, E# O' B6 [
</html>
1 C9 H1 C# x9 V: c, a

复制代码

2）服务器端实现

class WS {
9 q0 q# f0 j& q( u! M: h4 h
var $master; // 连接 server 的 client
1 K/ H/ U( n6 T3 W8 t) ?# }6 ?: T: N" A
var $sockets = array(); // 不同状态的 socket 管理* ~; i# H3 |4 O' c1 E
var $handshake = false; // 判断是否握手1 Y* c0 c6 j7 i, ~- D: y
1 p. z, ^7 G. P+ y+ E
function __construct($address, $port){
2 Y+ L& Z3 M. d/ U+ D& V% @5 j6 q$ ~
// 建立一个 socket 套接字( K9 i ?. i# m
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP) 0 @& Q7 s8 O0 `0 g* R" O$ z& X
or die("socket_create() failed");7 r' X( q5 y+ `7 k+ c, @9 D' I/ ]( H
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1)
, O# T- m, z3 v3 e' n* ^
or die("socket_option() failed");. n u8 _( Y% v8 U `
socket_bind($this->master, $address, $port) J) Y9 {6 {( {+ M2 u+ C1 @1 s4 x
or die("socket_bind() failed");' T7 d7 Z3 w8 K( Y7 D
socket_listen($this->master, 2)
) `+ ~7 x; ^0 Q, ?" }' I$ r
or die("socket_listen() failed");7 }+ ]" B3 a5 @- e/ S7 B1 w# G1 J
: q3 b* K7 {" J1 t9 K8 S8 m; \8 Y
$this->sockets[] = $this->master;. Y" m$ j t B. C; K( g
6 e1 |4 U8 `: f4 c
// debug7 u& q3 O- J) u
echo("Master socket : ".$this->master."\n");0 [# W8 Y& o1 Y+ @" T# X# e% [
/ q% V4 a) h1 V& R2 ?
while(true) {" m5 Z6 f# V3 q: B, z+ L- i5 d
//自动选择来消息的 socket 如果是握手自动选择主机
G! E- g3 l: U- V* K5 Y1 d% x
$write = NULL;
; _% @7 U' T: i0 ?( \2 X& p$ h
$except = NULL;
( k5 t7 ~. v2 D, @
socket_select($this->sockets, $write, $except, NULL);
/ A0 I1 i; e7 b8 {7 @4 M) ?
# P; m0 e7 a, a
foreach ($this->sockets as $socket) {
9 T: M7 h) v2 Q, @, h! b
//连接主机的 client + l" j% Y D% ]2 A' V
if ($socket == $this->master){& p9 q. a+ G' Q& \5 k, `
$client = socket_accept($this->master);& ]' ^ U3 D* W8 d0 e1 _" [! q
if ($client < 0) {% n1 O2 u: q9 \- k; I" A
// debug
& @8 @8 K& `" f5 F* P' P
echo "socket_accept() failed";
" y$ F0 t) X+ b" _0 m
continue;3 ?% U! e q3 n% \# G
} else {
d; R8 d3 N8 Y( k; u" |
//connect($client);
. c6 h' Q+ e, _4 e* j
array_push($this->sockets, $client);
: E& t7 F0 l/ f( w# |
echo "connect client\n";4 W# s0 W' T$ d* B
}
7 a0 E# v: s) L! }; b% T
} else {
: z3 J e* [* j6 h' N
$bytes = @socket_recv($socket,$buffer,2048,0);
; A1 a6 s& _7 J2 J% t( E: t
print_r($buffer);9 l0 ?' O+ ]3 U( M& T3 n- O
if($bytes == 0) return;
7 m; H4 v; h3 M! h$ _9 p3 ~3 m
if (!$this->handshake) {4 d: h1 X0 i/ j( I. R7 i8 S6 T
// 如果没有握手，先握手回应7 J- M8 }6 b- w& k3 j+ L2 B1 @3 B
$this->doHandShake($socket, $buffer);) K+ K7 I7 j9 T. c
echo "shakeHands\n";* h4 f# f4 g- w+ ~; q* u& X
} else {
% J+ Z# \. Y( ^+ H" x
6 ~/ I( t/ A r3 x* ~
// 如果已经握手，直接接受数据，并处理& [$ G5 D4 M. y {; w x% o
$buffer = $this->decode($buffer);
1 Q: F$ c, Z/ X; j0 q, @
//process($socket, $buffer);
3 C8 V% y6 P1 t; Z4 @2 d1 c7 [( @/ y
echo "send file\n";% F9 ]3 c7 l2 Y* w4 m6 E. k
}3 f8 ?0 P6 O: g$ \' H( G& e7 z; O
}5 |! ^" j: m" g7 p5 s
}4 I7 H/ e! z- g8 b6 J% T
}/ F' L+ v% u* Y
}
5 P$ |+ Z9 i7 k+ ]
2 M# o; O$ d% P @% t4 L/ H5 Q3 K+ ~
function dohandshake($socket, $req)! H* f$ K& H% h4 F, d6 L' f
{
- J# t7 R4 u1 J$ f" j" p0 M: D
// 获取加密key% ]9 M& t% I; A6 y: {- f' x
$acceptKey = $this->encry($req);
4 \6 A1 T" l c4 E
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" .
M9 o ]+ c! o; S- _0 N
"Upgrade: websocket\r\n" .
2 n* I6 l: S2 G, i3 s( D, U
"Connection: Upgrade\r\n" .4 g: ~6 L# F4 Z% M" ^* [
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" .* P8 D& n; `$ l. i6 N
"\r\n";
M1 R8 j+ L, f5 q7 A
1 |; ?3 j9 X* B+ ^/ ~: g3 I
echo "dohandshake ".$upgrade.chr(0); # P5 z; r5 e; k* Y
// 写入socket
: J' R& c2 T$ E: Y, R' u) ~4 L& \8 f
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0)));
, ?3 \, J- J2 Q/ [' e! W& A3 v
// 标记握手已经成功，下次接受数据采用数据帧格式* Q; ?9 d/ X, c$ P: \& J; A O
$this->handshake = true;
* y+ @( H$ T( P7 \
}* u2 K4 _' |5 L8 Y- y7 `$ {( F
+ c |7 ^+ ? I' h% {) L& V7 V
* q+ G: q7 P1 k) i* x
function encry($req)
1 R6 B* l! q) l# k5 F+ y, U; x) ]
{7 Q' F, L4 C( y; p2 N
$key = $this->getKey($req);) y) A9 F/ G7 }1 F/ h: T
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";
- t& Y6 ^7 ~; l3 ]. g
3 e9 z2 I( r+ S- [% E) R0 W3 Z2 v, t
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));
' @+ C3 Z' s3 `# x7 z8 e3 i5 j9 h
}
- A6 d% o1 e% W/ c d: ?9 b
0 `, I2 D" ]2 A% ?) Q2 j0 I
function getKey($req) 5 A5 N7 K- M+ ^& x3 i. a
{# z5 c# D% a' ~0 V
$key = null; v4 V6 h$ L1 t* i8 e- e
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) { $ C0 [& y; |5 M' p( w
$key = $match[1]; 4 e) y: h9 G" N' t3 n$ [, k0 v; P
}0 I* O0 [) `' z' w9 i
return $key;
c- w1 ^5 S6 U- s1 \& f7 o
}& a. ~" \7 L; W
/ } t- M0 _+ D3 ` S' I3 M! ?
// 解析数据帧0 ~* } o( Y( V4 @9 o* N5 ]
function decode($buffer)
6 Z8 k/ `* b% N0 ?7 {2 M
{
7 t0 v7 |' V/ M4 ~6 Z9 M+ v
$len = $masks = $data = $decoded = null; L) `$ G1 R6 r5 }9 }
$len = ord($buffer[1]) & 127;
- ]7 x8 q5 V9 P4 G( c
* p! N/ B% S2 O% Y
if ($len === 126) {
+ N4 }- H* k N, I* \
$masks = substr($buffer, 4, 4);' n1 a# N6 \8 v' F; l
$data = substr($buffer, 8);( R# Z* m( ]! i4 a6 V6 R( u1 {6 X
} else if ($len === 127) {
2 d: W+ d3 R4 q. ^
$masks = substr($buffer, 10, 4);: R: D" d' b: v$ J: s B1 D* [) i
$data = substr($buffer, 14);
( E8 Q4 K1 x2 @, ~0 u6 E
} else {
$ e. F. Y- @" a+ T2 q4 J
$masks = substr($buffer, 2, 4);
! a( h: b; ~" w" | c
$data = substr($buffer, 6);
5 w! O2 s: m }, s
}
5 R/ g( v* o* {. R' {7 O
for ($index = 0; $index < strlen($data); $index++) {" d! E/ B( C7 Y- F
$decoded .= $data[$index] ^ $masks[$index % 4];
2 X, I) M6 Z* T# `9 t1 i# K
}! o" ~+ ? _6 w- n; L
return $decoded;
7 I! I# v1 _& q( t7 y
}
. e$ _/ w, H, O2 Y# t
4 x% U& h( n( o: j5 ]4 `& H
// 返回帧信息处理% c/ o$ a7 q' [) a6 f7 y2 E
function frame($s)
- L) y/ D% ^4 g9 O
{" s0 l& B6 o- |) s5 \
$a = str_split($s, 125);
7 N, P1 @( P! ?
if (count($a) == 1) {/ q* G k- |8 ?$ [: w1 Y, b
return "\x81" . chr(strlen($a[0])) . $a[0];
8 O- ~# R! L; K8 |" v
}
6 c$ c+ s/ O( f
$ns = "";
9 W: d: m- r1 f- f
foreach ($a as $o) {& A2 q8 ?7 O/ W: f( k3 P) {
$ns .= "\x81" . chr(strlen($o)) . $o;6 j+ j# i5 k1 E8 W$ O( T/ P _; f' ^
}
0 T, u3 W0 ]3 H+ f) V
return $ns;" z0 _1 n+ \1 S9 C$ C
}
5 g' \* p9 ^, B0 ]3 [8 Y
M- o' ~. [; T
// 返回数据
* v6 e; y5 g, @$ u& d
function send($client, $msg)
4 X, w, ~7 s- q5 k- i& r2 J8 Z$ n
{
) U4 u: A, C2 J) p1 s3 @8 i
$msg = $this->frame($msg);
5 r" H9 w4 Y" q, f; D3 S( L- G
socket_write($client, $msg, strlen($msg));. o& A7 [ W) T) ^2 K7 D! Z- ]
}1 E, H- k7 E/ C5 e* V2 f
}- P- H8 \! L/ c9 M
0 d# ~$ Q2 [( N* \; A i' @! C0 F
测试 $ws = new WS("127.0.0.1",2000);
6 c9 D' X) z i% C- A3 J
8 s9 B; L( n) p" E A

复制代码

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

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