PHP 简单实现webSocket

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

1）客户端实现

<html>
2 v2 S- W6 Z c/ S: S7 R
<head> ~5 |6 a+ k) `9 \7 i
<meta charset="UTF-8">5 v( z6 H" i! d" |' E
<title>Web sockets test</title>. m* `: i, U7 g- u: E
<script src="jquery-min.js" type="text/javascript"></script>3 {1 d. I L% k9 X* s- |9 I
<script type="text/javascript">0 Q, R+ Y' b* j' {
var ws;
% V: q; k W1 J9 b+ [7 m9 ]% h
function ToggleConnectionClicked() {
9 D8 G& t2 ?9 u
try {
1 R% B `( U7 k& K5 P0 \3 Y
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器
% w7 k. T4 A( T0 D
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};
- A0 ]$ E( m$ X# q7 c. w
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);};
- f0 w! n; c: M: x( `2 O
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);};( L% E1 B5 M% H# Q6 d
ws.onerror = function(event){alert("WebSocket异常！");};
' q) d& I0 h& p7 k4 L0 i
} catch (ex) {7 E: q4 p/ H# T6 y8 M
alert(ex.message); $ g0 T2 }% B3 s' _# S. |
}6 l @5 M; u) M( s5 {+ R
};- q) P9 h' d- a' E( {# y2 m0 a
( Q' [" ]% o' j! S1 H# E; D6 h
function SendData() {
3 o$ e5 {" h; p% P& f. m' [( f
try{2 c9 G7 g( [- Z2 p2 Y
var content = document.getElementById("content").value; W5 D/ k$ E9 T2 q1 s
if(content){: }* F. d; d$ O- ?# W
ws.send(content);
}
, A, O+ ^5 s$ m- x4 `$ ] t
5 h5 b" s& s6 D7 |2 h* U0 @2 ]! [; N
}catch(ex){: p& L' _! |# _9 {+ f
alert(ex.message);
8 \: |# X4 y f5 t
}% b) [8 G G1 P# D0 k& B
};
# v8 Q$ R7 D, R. t' {# _
: G; d x$ D8 d
function seestate(){7 c$ O. i) v) S9 H
alert(ws.readyState);
7 Q+ k( Z! Q ?+ I# X0 d! D
}' c! k& e: K$ B, F- P7 J
# S' ?) ?6 y# q/ M( @- H( m
</script>
E4 ~- k+ p8 j4 ?! u
</head> a* V4 p# w: H! D
<body>, z9 p% K: f) ?" i2 X
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br />+ X0 P4 `% G. W6 E
<textarea id="content" ></textarea>- R1 {# S/ `1 T' I4 n* E0 Z9 \
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br />, a8 [$ b& `0 L+ `. N5 Q4 g& ?
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br />
$ R" n& H$ h' ~5 Q- i% @) b
" {" I3 V8 R" K4 F6 A$ ` d% O
</body>
' ^% m: d) X2 r! d4 C# [; g
</html>
8 C. O N; }! y# P& u

复制代码

2）服务器端实现

class WS {
4 W7 m. X: \2 T* C! O9 y
var $master; // 连接 server 的 client" r- D( t5 u/ |* H* y6 o/ W
var $sockets = array(); // 不同状态的 socket 管理3 \3 j/ R# \6 t5 Y- k3 a j
var $handshake = false; // 判断是否握手
]! n$ _/ J9 _% t% z# ^ W
% i' m& z& {& ]) O4 O1 ~* z0 B+ O
function __construct($address, $port){3 K3 Y/ D6 L: x; l% Q! m
// 建立一个 socket 套接字0 B4 A3 p) f( J7 M3 @
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP) + f$ c3 t" V) b+ A% e
or die("socket_create() failed");
/ ?! R( {: l. U8 W$ Z
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1) ! x& H5 P8 M* _, L/ {8 C" ^8 k/ i) t
or die("socket_option() failed");
; E& C Y) x) Z$ X- S$ o: V6 M$ z6 I
socket_bind($this->master, $address, $port)
" R' b/ P/ s* r* u/ @1 D- ]0 l
or die("socket_bind() failed");- c$ {/ o6 Z, I! z) S- }2 K8 c K
socket_listen($this->master, 2) 9 A, I' @; _ y$ @7 V
or die("socket_listen() failed");2 [2 q3 T# K5 d# m/ H' ?
4 m9 K: N2 ], s1 A4 U8 }" m
$this->sockets[] = $this->master;
6 i7 V- n) o: A3 q4 l* L
7 ?/ f; R5 x: h
// debug1 t" h5 d$ b0 z. x, U7 i
echo("Master socket : ".$this->master."\n");9 l( t/ h) q7 @: c- ~& |: K
t9 E) G4 D8 n! r4 D
while(true) {" v1 Q9 G$ m, S/ e/ F
//自动选择来消息的 socket 如果是握手自动选择主机
: Q8 H3 e% i- Q8 p
$write = NULL;
/ ^5 t; M$ D7 |9 z
$except = NULL;' j8 E* M) _/ S$ L
socket_select($this->sockets, $write, $except, NULL);) ?! C/ H0 B1 }$ B% ]5 G& i, o9 \
. F" B) x7 r* x B
foreach ($this->sockets as $socket) {$ X! J5 t6 O' H
//连接主机的 client 4 d" j' O9 Z, _$ a
if ($socket == $this->master){
4 |* h0 u6 l) B% U. }& S" V/ P
$client = socket_accept($this->master);4 p- k+ N2 |% l# C
if ($client < 0) {
7 Q/ d% _# B; m: @2 b
// debug
a# o5 y5 s1 _/ i# g
echo "socket_accept() failed";
8 J6 c9 o2 T! h7 e
continue;9 t8 S& f, O) p' p. ~1 A
} else {
' |* S, c: f: A
//connect($client);, p2 l$ b9 k' _3 _2 Y& z
array_push($this->sockets, $client);6 |* x7 T- u5 r6 [$ q
echo "connect client\n";. T; q: k* p+ Y' a
}; |& Q3 o+ Q% y6 n6 Q! b* U, m
} else {% S7 | f9 C+ r. m* i' U4 K: J
$bytes = @socket_recv($socket,$buffer,2048,0);0 k& O" l+ ~* [) V+ @" R
print_r($buffer);
- l, }# }0 [& I" B2 b
if($bytes == 0) return;# t( O2 C u/ I Y3 L# T
if (!$this->handshake) {
! Z- P% q4 S$ Q
// 如果没有握手，先握手回应! o+ K: d4 v) q. @% r' Q
$this->doHandShake($socket, $buffer);
1 ]# m8 |. @- o; l
echo "shakeHands\n";4 h9 b: Q& C$ B
} else {
0 s5 @0 |% K4 M0 @" o1 x( F* q
" g. b1 u L! g
// 如果已经握手，直接接受数据，并处理. P% i! ?! Y; b
$buffer = $this->decode($buffer);$ @" o) b" a; t# y
//process($socket, $buffer); $ J8 Z8 T9 Q8 O: Q" {9 Z+ n% S
echo "send file\n";
. {% i* [& X( t0 }0 b7 e* x% Q9 S
}* O5 o) g3 ?$ y8 K% r
}
0 Z: B# [) k. a5 x Q
}
2 T; Q$ q$ X6 G6 L( U* q
}
+ F$ H( F' d* A) }) s( v
}
5 q* L7 T1 @" M2 _ ^
6 I/ R4 b _$ L) ]1 H) F7 O
function dohandshake($socket, $req)$ I$ e- K6 G! e) ~% q8 O K
{
9 m/ a+ J$ `8 I" w/ ]
// 获取加密key
+ Z! S+ Z1 E0 N3 {- n3 I
$acceptKey = $this->encry($req);. I8 y+ Y) y9 \5 x ]
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" .% Y. M2 E5 y' F- @
"Upgrade: websocket\r\n" .; H) E' m9 Y, S: @7 s; [# X. K
"Connection: Upgrade\r\n" .9 F- C3 x7 \4 u7 z5 B9 P) [
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" .
# A& ^% j0 z' n3 \
"\r\n";( J) G# a( v, H% a0 z2 k9 K' y
( E, @# H' L, w/ ^
echo "dohandshake ".$upgrade.chr(0);
4 D8 b5 X# \+ |. k
// 写入socket8 j" T9 B; }3 w4 V; N8 g! ^2 I
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0)));
3 u1 D5 e" N# ]) k. C( X7 F
// 标记握手已经成功，下次接受数据采用数据帧格式
: |! _* L2 z% h E) D
$this->handshake = true;
/ n9 ?: A2 V- r D6 Q
}5 `( U" ]" B1 H5 c7 R: j5 e
2 g8 t3 m5 k/ O/ @' U, c- J* z+ Z
8 ^1 k" M5 }1 N9 _& b
function encry($req)
5 G1 B3 x0 P: e* f* o
{3 G) h8 O5 e; U1 ?
$key = $this->getKey($req);6 t1 L/ y j4 K: E7 s
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";0 ^: j# f) y! t- d* F) y" y
9 Q) l7 s" T8 F. _
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));
: }6 M: U* ?0 G
}
, [4 H& n9 V0 ?# q8 _; y
2 c) @/ J* R( s8 n
function getKey($req)
$ D- l- v1 `: l5 w9 Y
{* d, @9 V( o' Z8 C: j7 L1 E
$key = null;
* S7 ?. k2 ?* l( H. o) h
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) { + |. u K6 h4 z: Y& q; l/ }
$key = $match[1];
. V3 U. Q2 | L/ a! Y* g6 k
}
: @- K/ B1 c( O: g0 i, |7 l
return $key;
6 W. B; }$ g6 K+ X5 x) v
}, |. |6 j6 N6 S/ I7 T" T6 |
. t) ~: M* w2 [/ p/ F- n
// 解析数据帧, R" H, M1 _: o3 B% T
function decode($buffer) 3 M7 j5 H9 ~" C3 Q5 ^
{
$ R! K0 U& T* r) d) Y. h2 i1 n
$len = $masks = $data = $decoded = null;+ L' U1 f4 b$ Y4 a" M! G$ t
$len = ord($buffer[1]) & 127;
5 j5 X# \' Q D# R
( ?0 h+ }5 u, @$ N) y2 t+ h
if ($len === 126) {% |6 n/ s" _( R5 q7 A6 U- o$ _. Z6 G. G
$masks = substr($buffer, 4, 4);9 E+ I: w+ {) E0 y" A" g
$data = substr($buffer, 8);
: J" b% Q1 e8 S: k
} else if ($len === 127) {; l8 m. V: I8 F+ d' Y, e4 A
$masks = substr($buffer, 10, 4);
: j+ @% x3 K# k/ N8 [
$data = substr($buffer, 14);
* ~; y! [- z( Q( m7 B/ f/ \
} else {6 h; G6 L: e- [, K$ c, r
$masks = substr($buffer, 2, 4); G7 i8 w0 c( `& M% p" b7 a( D
$data = substr($buffer, 6);8 x$ M# G$ Q! I. j
}
( Q( _; B" g3 O" `) O$ V9 R
for ($index = 0; $index < strlen($data); $index++) {5 `& G C$ X% `1 m, W
$decoded .= $data[$index] ^ $masks[$index % 4];8 O4 N4 H# h2 M. R
}
6 ?) F. P8 E& t
return $decoded;
& h5 |; a4 W5 _9 Q% I. S& Z
}& \ }" D0 C& D7 s+ V v$ Y$ n
5 v8 F; M9 \) T9 }4 b! s1 w' x
// 返回帧信息处理6 w/ _# t6 N0 X# |6 P6 Y r" V0 K
function frame($s) % O* ?6 `: H) K4 _! Y
{
; {0 p: P% _- @! e# _0 F+ F+ D
$a = str_split($s, 125);1 E1 F1 m/ [7 ~4 d$ L2 l
if (count($a) == 1) {
$ K* R; L; B: W3 V
return "\x81" . chr(strlen($a[0])) . $a[0];
. b3 f" C0 p( V0 v' t
}3 Y/ R# _1 D% U- Y; s3 `5 A
$ns = "";
/ a, n6 M3 a0 ?
foreach ($a as $o) {
* N$ h8 @. l% h3 u8 ?
$ns .= "\x81" . chr(strlen($o)) . $o;
z$ N2 i1 U8 |
}
7 |* H: {) W1 d8 Z. P- `* b
return $ns;
( u& B/ N) b, K/ Z
}3 l( g D, y% @' ^# u
; A& z* e% b5 g& I4 \* y$ A
// 返回数据- K3 q+ g: H ?8 g; s+ Z# g& z
function send($client, $msg)
$ U N0 _# p; t
{% e8 w2 e* H: Y
$msg = $this->frame($msg);
3 z6 N, Q* I; R, H2 e
socket_write($client, $msg, strlen($msg));5 ]7 ]+ Q- b! u) `0 G/ L! r
}7 N4 j% g q6 L/ \3 Y* c" p
}8 |8 t) @3 L8 C) U
|; j. f& I* |! y0 o
测试 $ws = new WS("127.0.0.1",2000);
8 }4 l# W8 h `2 I
, ^' {0 y7 h* o6 G k( G

复制代码

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

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