PHP 简单实现webSocket

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

1）客户端实现

<html>2 Q3 T$ J$ B8 t, X7 @
<head>5 K+ ~9 n) m- {% c+ y
<meta charset="UTF-8">
* {5 m$ u- x- S9 c) `! i# o
<title>Web sockets test</title>, w4 c) ]3 }/ H# G% j2 S# L: L" w
<script src="jquery-min.js" type="text/javascript"></script>+ d5 B3 U4 I- N" t
<script type="text/javascript">
8 n; A! f6 S+ e/ [ j8 d" Y
var ws;
`, T% W& A; O( c8 ?2 E; n; e
function ToggleConnectionClicked() { & n% i0 \: m! x1 [+ i3 ?: {) s
try {/ B& @" D1 k* Z6 V% o
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器
) k- u/ C& o! q5 I
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};3 B% u! c F0 X& {0 N' U' o# C
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);};; |. w1 I6 V# _# {3 }# b
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);};
/ a1 Y" V3 W+ D
ws.onerror = function(event){alert("WebSocket异常！");};
" G% P' L. U6 F6 c8 B
} catch (ex) {/ o* H) ^) Q* X( b i$ y
alert(ex.message); . o' P+ C) w5 T7 h' L7 ]0 c
}
8 A" j7 ? ~5 h5 J
};
) ?0 U6 k+ e k" r5 t/ y- |
$ B5 Q( `, F3 N3 Y7 J' m" Q
function SendData() {
! q& N2 ~+ J8 M9 Q- `8 U3 e
try{1 n+ Z9 e7 G- a1 W& G; g) s' o, x
var content = document.getElementById("content").value;( L5 Q+ R9 n$ {' m" j
if(content){
2 r: j8 X3 j$ S
ws.send(content);
t$ @4 X6 ~1 ]4 e/ R# M* R
}9 u+ H$ G) U# Q: q; U: ^
1 J. h0 V: P+ V- t1 f2 \2 z* q
}catch(ex){; C; @. V! Q$ d' K6 ^( W
alert(ex.message);
. ~9 d- _2 I6 h6 r
}
. z% T% z: u1 p0 n
};
1 o- F( ^+ h7 i& t' f: g9 l5 K0 V
7 J7 F6 K, X* k! U1 P: n3 x
function seestate(){6 y- ~' ?2 ^; v4 H
alert(ws.readyState);
; f* J4 E, a! `9 `* N+ W
}
; L4 v6 f5 q4 {) F1 v
9 ]3 z" }8 z/ G% d2 g5 B5 r
</script>! x# C8 W8 e" U0 u
</head>
9 W0 f' Z* N9 C
<body>- K: j' k; P+ q8 G4 `6 a) p" W' w
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br /> c" s. T7 p! ^, `
<textarea id="content" ></textarea>' T- i; W. G8 U" D" ]. k( A' u5 {" Q
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br />1 D4 j' W/ u7 H1 n9 Q2 w
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br />
. e/ Q8 m' ?- F' D) R* z, i
: b2 B0 \( f" ^6 h+ {- N- j6 A, V
</body>+ R; q/ ^( }( m! t- q4 ^
</html>' w8 Y, f% W8 e' Y) ^

复制代码

2）服务器端实现

class WS {
6 l0 }$ x _+ C2 W. G5 b5 q4 l, G; f
var $master; // 连接 server 的 client9 G1 L+ p2 N8 Q2 p" M* k; H% G' K6 H
var $sockets = array(); // 不同状态的 socket 管理
0 V2 @0 }! c+ g- v/ l% g" T
var $handshake = false; // 判断是否握手! k, f. K: Y4 {4 ]/ a
. M7 u( ]7 v- d9 b7 @6 o' P& B
function __construct($address, $port){, _' W$ E3 Q1 H" ?& ?/ V$ k
// 建立一个 socket 套接字& S' Q9 _* {( Q
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP) ) ^* E6 q( j6 w/ u3 \5 n# n
or die("socket_create() failed");
, s0 @& t! O& t( z) A
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1)
4 Z) I/ g3 Y' w. M
or die("socket_option() failed");8 W/ a- h0 ?2 D$ |* H
socket_bind($this->master, $address, $port)
+ l+ ` }/ s( y$ a
or die("socket_bind() failed");9 q3 T' g" D# W! ^
socket_listen($this->master, 2) # Q/ i7 T6 |0 t1 v$ u8 u7 X; E4 B
or die("socket_listen() failed");) J- B0 N; B4 S
$ t* T! h. |1 D, `; V B
$this->sockets[] = $this->master;( A1 h) z, n- J Z
3 C$ \# u8 Z) s+ q! N: F; g3 l6 B3 l
// debug1 g+ ?: h6 n' F
echo("Master socket : ".$this->master."\n");6 {* [! O6 B) b) m/ A
" t6 L: A" C8 H, L
while(true) {
9 f8 X! H. t/ V# ?7 \- n& n
//自动选择来消息的 socket 如果是握手自动选择主机8 ]) K5 R- ?2 W4 B; V
$write = NULL;
' Q* [+ z+ T. p* U
$except = NULL;
: w* j5 u/ D8 W* ?9 ~
socket_select($this->sockets, $write, $except, NULL);
# M7 w+ q2 a6 z( s. \7 b3 ^( I
7 ]! m. ~* i- E/ g
foreach ($this->sockets as $socket) {
. W" y; H; c+ X3 J+ X
//连接主机的 client
6 X- p1 U+ y0 x1 h# V8 Q6 `1 ?
if ($socket == $this->master){
9 N' N, C* M9 `4 T5 N3 n
$client = socket_accept($this->master);: t; a/ V/ W: J
if ($client < 0) {4 L8 s `+ {% d5 u! ]* X
// debug
+ q/ F: w* n' k" Y8 Y6 Y
echo "socket_accept() failed";
- A. ^7 V5 s) J% \6 `/ N; g
continue;
( r1 W L* A+ L% M7 n: X
} else {# w2 ^: D6 o) {+ T' }
//connect($client);0 I: ~7 X) o) B+ n
array_push($this->sockets, $client);
) L! [9 V. Y7 Z& C H; @
echo "connect client\n";
- }) v* Z+ q0 ^5 X0 d: Q: S# o: x
}" t" T& k9 T) T$ u3 m: f! e
} else {; t }8 G5 I+ l( W$ k' p
$bytes = @socket_recv($socket,$buffer,2048,0);1 |" p7 @) n5 f7 v" x/ W# A
print_r($buffer);
$ g- ~( C/ H9 ~! N: w
if($bytes == 0) return;& k# B2 o9 N$ B6 H8 B4 Y- D) W/ j8 b
if (!$this->handshake) {
0 j& P- l/ k, T- U- ]
// 如果没有握手，先握手回应, r7 I, S. J/ G8 C( o& R7 P7 {
$this->doHandShake($socket, $buffer);+ X' ]2 z7 v1 X6 o. C) Z
echo "shakeHands\n";! ~( n. \1 N Q; @+ t" n
} else {
' V( s: i- p( s- l1 x' l: ^& k
n" ~! }" p: a) Z
// 如果已经握手，直接接受数据，并处理5 w, ^, v5 Q- u6 R6 e0 O
$buffer = $this->decode($buffer);
2 h- j9 e; W0 w
//process($socket, $buffer); ; x0 [# h4 A' O' s( I
echo "send file\n";
/ T* n8 ?$ {. R* U% i; |; @
}
5 u% }: I* e$ R% G8 ?
}4 a+ N5 f' J* g8 j
}2 d( Y7 N/ Z$ k
}' R; P# @# J9 Y8 A! \$ v9 ], R
}/ A/ r3 A$ w! ?9 A, V m! {* b( S
* C0 `1 g! w- a6 e# [% H
function dohandshake($socket, $req)
4 q' E7 `: F! ?1 b2 m3 d
{9 `3 H3 Q4 E% Z% u" s
// 获取加密key8 c0 G5 q$ [6 ?+ R* A' }% |4 c
$acceptKey = $this->encry($req);3 L6 V8 C [6 y9 q9 Z: ^! R- Y
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" .
, q0 L( w4 O- ^9 c5 N3 D
"Upgrade: websocket\r\n" .
}" N# W! m0 u x' F
"Connection: Upgrade\r\n" .
/ V9 E6 L- c! k2 o
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" .
' N/ n v! O( ]! g* T6 W6 |
"\r\n";
2 N; y! [& G$ R/ B8 _. d$ l
6 ~" f- v ^/ A* Y# z0 j
echo "dohandshake ".$upgrade.chr(0);
/ T& ^4 u. U6 b* b6 ~3 f* a S
// 写入socket
3 @- R/ K$ j- P# |0 @! C: [
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0)));) u9 l. B+ P, G; _2 q7 s
// 标记握手已经成功，下次接受数据采用数据帧格式
m: \# P, i0 O/ v2 H. s4 Q
$this->handshake = true;
; b9 V6 X4 m$ P* u# S5 ?+ _
}% P& q6 R3 s8 X4 ^1 j
: v( i; e! i+ I. E* L/ O2 ~
4 N) S% z. \- H- H( u- z& y
function encry($req)
9 k1 [: V7 }0 j# D9 e/ M, W
{
8 U8 o3 \# ?6 |1 m/ g; R- u
$key = $this->getKey($req);
* {0 E1 C4 p( W$ @
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";6 l! q' V3 R5 y$ x: O
$ x6 \# t$ L5 y4 O7 C7 A
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));, i6 ~ j K0 ?1 B* E( E* u' p
}
9 O$ V4 W5 s" A- E0 `2 y
) u# Z) @! ?# q$ s" \2 @9 U
function getKey($req) " s2 G) @! Z0 H. p- x
{
% A' G- M* ]' T
$key = null;% j% R3 S* I8 s
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) { ' u2 H B1 P L
$key = $match[1]; ( w! @" L% Z3 K- ?+ v
}
4 v( Y h7 t& g, z0 D
return $key;
" t0 y' q; u% w7 S3 G
}
9 S; W, ? n, o8 g# P
: j: D" w4 ^* R
// 解析数据帧$ O4 Q6 e; V+ q# N! b7 R# [
function decode($buffer) . _* a `) p, K9 _* [
{; z) u% }& C c& g ^% e
$len = $masks = $data = $decoded = null;
' _. f. q+ U! j9 t1 A# j7 M" D0 ?
$len = ord($buffer[1]) & 127;2 n+ K, H( y" `
: Q7 U. w( ]9 L; j
if ($len === 126) {
' B# f% J' q8 H; K
$masks = substr($buffer, 4, 4);
% h) h* Y; Z2 ?0 w! m7 _
$data = substr($buffer, 8);
) W9 m2 y' T8 h
} else if ($len === 127) {
2 m- K# q9 R* U0 a* W- F: G5 m( D
$masks = substr($buffer, 10, 4);
3 X1 g# ~" g* A9 ~' w+ T% z
$data = substr($buffer, 14);0 J! v- _7 T {# o
} else {
, S+ k) o; L. r, R% K
$masks = substr($buffer, 2, 4);
5 R' m/ O4 L& n% `2 P
$data = substr($buffer, 6);
- p3 k1 L6 b' R
}* y, _. l* V- Y% H6 l/ W2 m( `; v
for ($index = 0; $index < strlen($data); $index++) {6 C+ n: _1 U% R" k5 G: j( `" i6 M
$decoded .= $data[$index] ^ $masks[$index % 4];9 ?2 s* O6 n; Y/ v' e! {
}0 g) u1 Z l4 f7 Y. a+ }' t
return $decoded;
0 ~( M4 @2 ]0 \+ r7 Q
}
# Q8 N& ` z4 N: e9 v
9 ^1 E: X6 x( O6 _8 k3 o8 V" ^* B! Z
// 返回帧信息处理+ k, D: ~4 |0 Q# Q. H
function frame($s)
) o1 q6 h6 ^5 }2 [7 p$ V# J: ?
{
2 b. S, j) S5 L
$a = str_split($s, 125);- i. J& c' H& f
if (count($a) == 1) {
. Z t+ N: x; r5 C7 ~4 x
return "\x81" . chr(strlen($a[0])) . $a[0];
+ K$ k5 T* _/ w+ b0 X2 a8 b! f
}
( E! D' ?- h) c7 [$ q5 S1 r' z
$ns = "";
7 q# N7 }4 p) H7 q2 T
foreach ($a as $o) {4 v" r% p: B7 N2 ~2 B
$ns .= "\x81" . chr(strlen($o)) . $o;* l3 ^1 P6 i" b
}; p1 u7 _/ \9 A7 ?0 _; P9 A9 B
return $ns;
' N2 f) F0 w" a' R, s) F; I G
}
% `$ ~$ _7 e8 \
2 }& I/ ]. O$ ]( `) U% _
// 返回数据! v, v1 O* C. ?2 ~- l" h
function send($client, $msg)
. y! N0 E) y! z- @ d
{
/ |9 K7 D9 j( r5 N
$msg = $this->frame($msg);4 W. G: Z6 @0 J& d) L0 C* @
socket_write($client, $msg, strlen($msg));7 O5 {/ Z; z, j y* W3 C
}
2 F$ ?" v1 N8 ]$ F6 \
}$ d7 ]# B: h q% f$ k. c
9 G5 {: l$ p: n+ D$ O) O
测试 $ws = new WS("127.0.0.1",2000);
n- ?) C4 O1 [9 C% w4 X+ }
4 d* l8 X! O( R& |) b$ a/ g: O2 ~

复制代码

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

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