PHP 简单实现webSocket

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

1）客户端实现

<html>
% W, i$ g7 e" z' h( M; g8 i
<head>4 X' |$ p; A4 o4 C+ d6 Y* f2 V" P- ~
<meta charset="UTF-8">
& Y, n8 v" x, V9 I0 T: I2 r% W6 v
<title>Web sockets test</title>
/ J; [4 P Z! a- v
<script src="jquery-min.js" type="text/javascript"></script>3 Z3 }1 H* M9 I G6 m' L
<script type="text/javascript">
2 }! } ~, B9 H7 h. Y8 n
var ws;/ l) m$ F2 Y }' ?& r" `6 _7 T
function ToggleConnectionClicked() {
) [' ]8 s: x4 X) x9 M# D! J( E5 x
try {
; j- `7 }* `0 S2 R
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器
- [7 v+ }2 h8 c
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};
! {' T1 ]. J2 q6 J& X2 p
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);};
* e. V( |- }/ t9 D
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);};
0 }: R" K/ H& M& F$ S0 H
ws.onerror = function(event){alert("WebSocket异常！");};3 `& i7 |. T& ~/ E8 I' t; ^! q
} catch (ex) {
$ y4 o& N6 q1 Q$ }# H( o( y% A! S) T; B
alert(ex.message);
6 H" u7 j) N/ {( z) R* k! Q5 }8 I( h
}
I H4 b9 P; z( l3 w; Y6 w3 r
};
6 w" Z2 n3 {- D) E* _* j9 T! Z* T
' P- V" A; F6 s# b' K
function SendData() {0 k% q ?6 |5 Y. b
try{
8 G& x, J) }8 N! ?. g+ Q5 v
var content = document.getElementById("content").value;
4 U1 b1 R/ f3 P
if(content){+ C4 o0 `9 u3 t, ^
ws.send(content);
& P- h9 @' _" D2 O8 m5 Y- D
}
' V$ U' O9 k" k2 S: q/ c/ A
, g. T% N2 {4 u- x) g5 e2 g$ P- ?4 ?
}catch(ex){/ s+ R" ]; r; |& m8 z8 t
alert(ex.message);# p9 n: J/ u9 o1 a
}) J- } N% \- |5 t- r
};
: | O) w2 c2 J' t$ \; {+ E8 E2 l
9 r* M+ L- T; Q7 g. S
function seestate(){
+ U& R! D7 f3 ~ i g% m
alert(ws.readyState);
# v, \1 Z# B: n
}
3 P+ ^3 J- f2 a- c d# @( K
6 K6 J z9 s0 H( q
</script>" ?/ ~$ E; w) q7 i0 S M4 [4 ~2 }
</head>
) A( ~* i& Z& Z" Q) Y6 k
<body>
8 {/ R! w5 W# K% s7 A
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br />8 K! z, r2 G. C# ?3 y$ r' K
<textarea id="content" ></textarea> t0 n, n1 b+ k; \# w
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br />
( k) D# I# A- f9 e! u" q( r
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br />
# T% w/ Y. @0 w6 t8 L1 v
$ x! W1 W5 {+ ~1 n
</body>' o3 d: E% C( _3 J1 w3 J' h
</html>
/ P; Z5 ?& @6 _; |) v6 Y( \

复制代码

2）服务器端实现

class WS {7 ~# M3 G7 t) d) V
var $master; // 连接 server 的 client3 S: L! `" _1 p: t
var $sockets = array(); // 不同状态的 socket 管理& M! Z& r/ [ W- z
var $handshake = false; // 判断是否握手
& _# q! ]: H$ {$ Z+ \
' s6 I2 n0 b' P$ |& _# j1 f
function __construct($address, $port){
# {' M" R' s& X( M$ b" p; t
// 建立一个 socket 套接字6 Y! o1 u+ ]( l3 G* L6 m* ?9 l" y
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP) 4 `$ Z1 Z3 j3 R [5 a* j9 G, W/ n% I
or die("socket_create() failed");( e: n9 ]% i8 O% B9 Y
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1)
* [( i; q4 M1 G; E8 o
or die("socket_option() failed");
( V" S. d0 _1 E# p
socket_bind($this->master, $address, $port)
& L/ J8 m; k8 C9 W' Q9 N- U( Y; p
or die("socket_bind() failed");3 s- w: M- B5 y3 Y# H
socket_listen($this->master, 2)
7 Q6 q+ h& r- F3 C. g+ W& k
or die("socket_listen() failed");: k2 {+ g% f; W0 s9 H! p
3 }8 x) z6 G# B* c3 Y6 a9 |: [& ~
$this->sockets[] = $this->master;! P0 T! @( h( B6 z" H7 w* F% u
) `2 u5 t2 ~% i, ^" e6 _
// debug
8 @$ m) n& E# h
echo("Master socket : ".$this->master."\n");7 ^+ Q# m( f( ~8 u9 {
6 A$ O% B9 ~, L @* K, J
while(true) {" c, |# s6 w0 d9 h0 _
//自动选择来消息的 socket 如果是握手自动选择主机" F8 l P) r" b, H
$write = NULL;3 u; a3 q; ~" `; g
$except = NULL;3 B9 M( _6 E; x" ^
socket_select($this->sockets, $write, $except, NULL);! o a) X0 @, X
$ k5 k0 c, e+ [( h: p
foreach ($this->sockets as $socket) {3 j! P: V3 d% U/ I6 V
//连接主机的 client
- S$ m4 L+ k/ ?* n% I2 \$ ?* [
if ($socket == $this->master){ w+ l6 h! s6 \: v
$client = socket_accept($this->master);9 t3 U, M# ]: g# l% Q( m9 T/ ^
if ($client < 0) {
* U( M U7 U' ?) J
// debug+ I8 @& L# o1 |" f& F/ P9 h
echo "socket_accept() failed";* c; d" u# e* k1 t
continue;7 K" |' n: z5 T3 _& e- H3 L
} else {
, b& V1 c2 N$ x. t4 P2 N0 P6 n
//connect($client);
o3 o# N z; |* e
array_push($this->sockets, $client);
# W- V6 @: W( J' J( T. a
echo "connect client\n";3 f1 w' E2 o+ C6 |7 e5 y
}
2 e, H+ [- h% l6 Z. ]( \
} else {" a0 I" f4 G/ z; u: G
$bytes = @socket_recv($socket,$buffer,2048,0);/ _, w; ]$ h2 k$ @5 i4 i
print_r($buffer);* m- q, f% V! ?5 R- I6 F
if($bytes == 0) return;5 m3 ~4 c n# \8 _
if (!$this->handshake) {, j( ?* \; O, |: e; K
// 如果没有握手，先握手回应- y1 i1 t0 {# I4 a3 E4 Y' [
$this->doHandShake($socket, $buffer);8 {% |+ E$ {1 i) x
echo "shakeHands\n";9 G1 a3 n8 h/ a4 |* I6 `
} else {' H; [! m3 B& O, r+ n- u1 Z% U
* b; B% q2 F M) M* G- |! r/ Z9 z
// 如果已经握手，直接接受数据，并处理
5 {9 O \8 ]" |) ]( V; y
$buffer = $this->decode($buffer);$ Y5 s& F4 t3 f" T1 y
//process($socket, $buffer);
" v7 D1 w+ C; }
echo "send file\n";
* B9 k! {$ ?/ [( a" j. S
}
( S! j/ V0 v( }8 V. j2 y1 N
}" ^5 d. @$ A L( g
}
& `0 q: M. {2 j* z
}
3 u1 P5 C% ]$ \5 J( G
}5 a& v4 e4 _8 H
2 d( C _% z7 I& c1 B: o
function dohandshake($socket, $req)2 O( e9 Q$ O* G
{
# C# j# k& g; Y/ T; @0 J
// 获取加密key
. L( i4 E& P4 ^1 \! ?: x9 W* t
$acceptKey = $this->encry($req);
' y* n, N3 C) {+ o
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" .# a7 y* |5 k% k" f3 p
"Upgrade: websocket\r\n" .
. L) Z& J# ?8 N! R
"Connection: Upgrade\r\n" .
& O5 V7 Y; k O N8 z/ b
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" .. E' q- h& R% ?) M8 t9 e0 J
"\r\n";
% {# Z) n2 h! `* f
b. m: `! a* U3 C1 u8 Q
echo "dohandshake ".$upgrade.chr(0);
- ?* ?% L* t' D& V$ o2 x% |& |8 {1 U: l
// 写入socket
6 Z! w, O& I0 t; k6 _6 c+ E
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0)));# y9 u |5 N9 M, i
// 标记握手已经成功，下次接受数据采用数据帧格式
8 N0 J) R3 L! J" [
$this->handshake = true; c9 z. U, Z8 u) C
}1 m* J9 `1 E. K5 z B( |( r
7 a/ c! ^$ K5 D- J, @
) ?* [% x7 N8 K2 T6 U
function encry($req)
% Y) U6 s) }' D( `3 w' k' e
{
& i) v. m% a' S2 {9 T$ Z
$key = $this->getKey($req);: E% o4 g+ |; X# Z6 Z7 j
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";, y5 ~: k3 v# `7 k" }$ [4 o
* P2 T1 Q7 p* }7 P" j
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));0 k x* G+ [! F3 P
}7 j: G: F) i* w3 b* ]" j
( I* }- y7 ?4 @4 `) a
function getKey($req) ) `$ d: m; M, u3 V
{
( a5 X+ [9 X0 F. W5 V7 m
$key = null;
* n$ M/ g1 u0 E' q' v' y- P
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) { ! n. T. P& I% T: e
$key = $match[1]; 1 p! j$ v) \! I# B0 q) X: H
}. f6 Z7 s7 o2 a7 l8 f) i% J# k
return $key;
% e* L ]4 W q" F8 |
}
% l B+ e6 F- \1 i
2 l9 v( F" s. ^1 U$ m% I
// 解析数据帧
& m5 _% \* _( P
function decode($buffer) 2 H2 h# ], w- J$ F1 G( V- q
{
8 B! k3 [' D0 v3 u
$len = $masks = $data = $decoded = null;
, W& h1 X# d& I* u" Y5 \1 B0 u. _
$len = ord($buffer[1]) & 127;0 E& Q6 E# C1 U- g- Q% V# w
* o L v3 _; s0 a# R% _
if ($len === 126) {
" C% k4 r# q- P5 n
$masks = substr($buffer, 4, 4);: e5 J# e7 Z& w& g" s3 R) J5 }
$data = substr($buffer, 8);2 @+ c" B% f) H4 r
} else if ($len === 127) {6 ]/ V( [# S& h5 I5 r
$masks = substr($buffer, 10, 4);
* k6 s3 Y2 ~4 b
$data = substr($buffer, 14);. Q5 z3 r$ h4 U9 Z" D
} else {
6 O) p" P9 ]0 C$ V
$masks = substr($buffer, 2, 4);
; N% s# c1 r) B Q2 U5 |" L9 {3 B* M
$data = substr($buffer, 6);
9 M3 `& Y$ l5 D7 k6 A- H) P
}5 B. ]7 a0 r4 f; Y5 [' ]6 w6 [
for ($index = 0; $index < strlen($data); $index++) {$ \( O5 {. v+ a7 o" y1 e
$decoded .= $data[$index] ^ $masks[$index % 4];
) {, r/ W+ @! B6 B
}
8 ~) J) Y9 J8 G; w( q
return $decoded;
! X& w. B4 }, U. ?( H
}
0 G* y9 p. d/ J( F/ ?& {3 f
# p( L5 o9 O: W3 y! g, y1 p, M! g
// 返回帧信息处理
7 q& S+ ]/ F5 H# c! f% R, j) K
function frame($s)
: d' v* `7 }9 ~
{
: w6 Z* q& ~$ q) ~% e
$a = str_split($s, 125);
N: V$ ^( m- x% n
if (count($a) == 1) {
% H$ x, E! \8 ^ T- I9 }
return "\x81" . chr(strlen($a[0])) . $a[0];7 W8 l0 P& P; K4 p
}
& ]1 p' u: D' y/ S5 a
$ns = "";
b; O0 A; I$ H+ t. W0 {" M% L
foreach ($a as $o) {
$ j |+ Q3 Y' v3 `6 g
$ns .= "\x81" . chr(strlen($o)) . $o;
( R0 X7 k0 K4 \; H; {6 X/ j) d
}
" J8 Q& F) T s. c
return $ns;
( j6 U5 I, g3 }3 M4 Y) F: D+ {
}
( y0 j7 E/ D. S* [9 F3 G
" w b- ^: R. _: {0 R
// 返回数据
- q( W, T. [, w
function send($client, $msg)3 ]( H, L4 `% f; W6 n
{
$ V4 W5 f0 K* \ m% Q3 J0 p2 P4 R
$msg = $this->frame($msg);
' r% N* y/ _, a
socket_write($client, $msg, strlen($msg));7 @0 g3 j' z/ J! _% T+ ~1 k5 S' L
}
) @4 W1 {7 N4 ]& z& p
}, Q1 X* o4 ]6 y
: V# Y, o* J0 m; q
测试 $ws = new WS("127.0.0.1",2000);) x- X9 q/ T& x
3 N! }! W! {( a) Z$ T3 |! A! X! D+ a

复制代码

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

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