PHP 简单实现webSocket

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

1）客户端实现

<html>
9 |8 r* y0 ]- T6 p2 O6 v9 Q
<head>
% M& K o4 m R& w
<meta charset="UTF-8">
) V9 D0 F1 p- ]) }
<title>Web sockets test</title>4 C. u r* g$ p% Q l! D
<script src="jquery-min.js" type="text/javascript"></script>* ^/ s# Z# G) S' }/ r! e# C
<script type="text/javascript">
' w' E) y7 b) b$ ]4 c
var ws;" p" M: F# I. f1 @/ u
function ToggleConnectionClicked() {
7 k5 M6 h, |5 c! {
try {9 Z9 d* q7 o' j4 c8 e
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器
# b/ W* S& u3 [) k( S' ?
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};# T4 q! _. H, a5 e9 q
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);};& o( x' e6 W0 @7 P, A) h
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);};, }5 D: J0 D! B( f8 m1 ]+ q) E
ws.onerror = function(event){alert("WebSocket异常！");};
8 v0 w. u7 e! m5 G T `
} catch (ex) {: R( {% Q# ~7 ]+ S. A) _0 Q
alert(ex.message);
* \+ l( g) b% h3 W
}
, }+ h, v: T! ~( Z+ E
};2 r# n4 W) R( H+ a5 g
% J: U' L% G: I, L1 ?7 e
function SendData() {
# I8 }3 Z3 d) F3 ] O4 M9 R% G" Y
try{
( P' U6 n2 ~# R1 c/ q; b8 n# T6 c0 P
var content = document.getElementById("content").value;% j: V4 W3 }3 ?) ~) R3 G
if(content){$ M" y" Y3 l N) a. e" w# q
ws.send(content);$ f5 e5 r# P( H- T
}% J( i. K5 S2 V+ n1 a
3 V" S9 N; P1 p( n# o' h4 r N
}catch(ex){
& f# c2 }- v9 F; W0 }+ Z# s
alert(ex.message);
4 O- |# c4 _, ?* C2 c
}4 K6 x) `, u$ F# F1 ^# i& l: p1 g" c
};
3 Q. M9 c0 m; g' _
. D- y5 F! x3 X
function seestate(){
* a% @& u9 e7 @2 Z8 T' |& a' ?
alert(ws.readyState);, r+ |; s! Y& X3 o# G8 G
}
) S1 y2 k4 E7 _4 a/ {, P8 H
/ ?! w8 d9 c! h
</script>
9 v% \0 g4 L% ~1 p
</head>
1 m: O) f! z* i* Q$ F
<body>1 b5 I; J* X* I
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br />
( u9 F" c6 Z3 ?; S7 i( b8 m& f' K6 E
<textarea id="content" ></textarea>
9 s" i. o& e5 e9 \
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br />
+ O0 }" X2 t8 d% \: N- z
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br />
- Q/ V0 S* X+ q' I0 A* o
# L3 K) t% y8 M; Z$ B2 t) @
</body>' ~ h' {5 M- H# G, y; `. e
</html>. U" r J* \$ J/ X: u- ^

复制代码

2）服务器端实现

class WS {
7 o+ [7 g1 S o* F8 }) C
var $master; // 连接 server 的 client" ?/ i+ m5 q/ d8 C
var $sockets = array(); // 不同状态的 socket 管理# T. D+ i! I- R! \
var $handshake = false; // 判断是否握手& J0 f/ t) |# H- k% @
8 Z/ S% P4 R- {( K& x
function __construct($address, $port){
5 ?0 a& R1 h& h* G6 w
// 建立一个 socket 套接字
9 o3 w* n+ h. L* g# ^4 u# g% }
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP) " s( V d3 e' H `5 t2 i% I
or die("socket_create() failed");/ K# v) |4 a3 u4 }; B- ^/ M
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1)
) I! `, \" t J2 _
or die("socket_option() failed");% T3 ^, S/ f& d/ ]) e6 ]
socket_bind($this->master, $address, $port)
4 e5 O, T' o" y1 X
or die("socket_bind() failed");
# g$ E4 V+ o' v/ A+ T
socket_listen($this->master, 2)
# S4 d( Q4 R$ a
or die("socket_listen() failed");8 `+ A: o+ w8 B" t) s% X
3 | W" S- C' f" ?% @
$this->sockets[] = $this->master;
- ~7 b |1 P2 J+ \0 B* y4 n- Z
& _% T0 Y: S% s
// debug; Q8 z* a7 q1 I1 d
echo("Master socket : ".$this->master."\n");
2 R: E/ p5 d- M( V) S
$ V$ ~* e7 e r0 k" S; `. ^: g9 o, g
while(true) {2 T& {; ]9 y* U j0 Y, {, k
//自动选择来消息的 socket 如果是握手自动选择主机
( f4 p2 f3 b* [" l! Y- ~- R; K
$write = NULL;2 I2 q) L6 @9 D+ n- o
$except = NULL;, C/ x/ p: [* E( q) M J4 Z
socket_select($this->sockets, $write, $except, NULL);
* Y. W; H2 |0 l3 B1 R' r! j+ N
4 C/ s$ O3 ^# m( Y2 U& p
foreach ($this->sockets as $socket) {
* [. o; s+ T8 p9 H
//连接主机的 client
1 q+ N% {/ C- j0 m+ b. E) x2 ~; X, s
if ($socket == $this->master){
9 I! J _- C1 w, L1 C
$client = socket_accept($this->master);
2 U9 M0 U" ^3 s" w, C
if ($client < 0) {
* Y+ a: q# G3 r7 m
// debug
8 d4 N3 o" t6 N5 I
echo "socket_accept() failed";
7 _: f/ k/ T& m4 E+ k& J
continue;: {% W9 M' z/ m
} else {
3 S( B+ ]4 n$ q* F$ P
//connect($client);8 J; w- f- j3 @1 q' a1 G
array_push($this->sockets, $client);" C- L- r% y7 D7 B) K5 f8 r* o
echo "connect client\n";
# l4 @; `7 [+ Z. K, g4 |
}
: h; |1 y7 B% }
} else {
s0 ?' F1 ?1 _3 d; u \
$bytes = @socket_recv($socket,$buffer,2048,0); i) v1 S# D' i5 @' K; h- k8 b ^
print_r($buffer);
4 J2 v# d& r! R1 a* W
if($bytes == 0) return;# V) w6 ^) w7 ^" Z5 X3 a$ X6 E
if (!$this->handshake) {
z$ j# F' u8 ]4 _
// 如果没有握手，先握手回应
8 K: O. E" ~" f7 e! {
$this->doHandShake($socket, $buffer);: Y0 B+ o7 V( s' \
echo "shakeHands\n";3 k h% ]2 E4 J/ l
} else {5 q; w( P9 v5 @7 @, H# D
1 o7 A W9 ?3 W' Q5 S2 F2 ?
// 如果已经握手，直接接受数据，并处理1 M1 e1 _: }9 @9 |+ l( O# U+ z
$buffer = $this->decode($buffer);# X* s1 P) `$ @7 p7 a5 m
//process($socket, $buffer); 2 P* U; I& F# |
echo "send file\n";! [3 w% E9 L0 j( n K" Q
}
% d# ~9 P* h0 j
}& o3 j! U0 x6 N! P& V9 ~
}3 c. Y$ x# R; t6 e
}
- V4 ?& S2 F5 f: f6 p
}/ ` d ^8 M& n/ O+ m
. U1 }6 G3 ~9 D. [0 M; G, v
function dohandshake($socket, $req)5 R" C/ H9 l R+ Y" K- }& R, I
{+ a r B2 j0 {" @0 ]7 ^
// 获取加密key6 E& k' C# B8 ]) ]: o D, W
$acceptKey = $this->encry($req);( t/ u9 ?: C8 l- V
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" .
& M H0 ^8 Y' d, c2 w& V
"Upgrade: websocket\r\n" .
" ~ v3 Y& a/ T, g4 R3 H
"Connection: Upgrade\r\n" .
( W: K8 d J8 j! {1 b7 A! ^
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" .4 I3 D( y5 l2 x1 \
"\r\n";
1 Z# l( i4 _) e2 d
0 I K' E( A# [1 r, u
echo "dohandshake ".$upgrade.chr(0);
, Z( |+ q0 n/ c+ v: V
// 写入socket
5 c/ v+ F8 [4 o! g" l
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0)));
( N4 \) A* s% H6 l/ j' A: [
// 标记握手已经成功，下次接受数据采用数据帧格式3 s1 Z$ K8 g' [, V! j6 H( J
$this->handshake = true;
% k- {* m( H, ]
}
9 C2 A4 z: d9 L
6 O- Q4 K. l( c
6 y! B/ l% s+ B* D& E* G' \0 n
function encry($req)
( f# m3 y- v8 I8 I. s0 N
{
) R( @, t" v- }$ } a' g- z
$key = $this->getKey($req);8 O" {" T# U0 S+ |
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";: @( Q; s$ T% i, M
5 [& K }- r2 w
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));) b) O ]) e( B
}0 `7 @+ c! K2 J' f8 F/ l3 D
/ w" j; C, ]* A" h7 L
function getKey($req) ) V1 B: Z# G) K& |
{4 i, N) c% I" L' e% T. y
$key = null;
0 p7 z* U- K: {+ }" B
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) { 2 G; g3 \0 R$ i# x0 }/ b
$key = $match[1]; 7 T* s ]- t% |) \! P
}
% l% c, Z; d6 Q' I
return $key;
}9 g6 |: N! T4 o$ a
; w" J# x8 p. T/ X( Z/ |0 c
// 解析数据帧- F2 s: ?8 u' c3 C+ }" s! V" J
function decode($buffer)
% g( J1 f/ w& |. D8 P
{
/ z6 E% S! I h7 q* \9 _
$len = $masks = $data = $decoded = null;
! l3 H% \- [; A, P' n
$len = ord($buffer[1]) & 127;
- P$ o; I7 l) y1 l/ t+ g
, P/ \: }8 \3 g5 t2 i% F, t
if ($len === 126) {
2 N2 w m9 W- b9 R
$masks = substr($buffer, 4, 4);2 U0 H' X+ Q0 v
$data = substr($buffer, 8);7 V6 O% K8 `2 b- g& G% ^
} else if ($len === 127) {
, ]* _; c* U- i6 V
$masks = substr($buffer, 10, 4);0 D9 n1 t; \' P% A% }
$data = substr($buffer, 14);( s+ o+ V( W5 p" T+ e
} else {
7 o1 l; d8 m6 f& f! f. s0 ^
$masks = substr($buffer, 2, 4);& s5 K: {! h! N5 w
$data = substr($buffer, 6);
/ z/ Z# G% j) j( p; }# \
}
+ o* z3 a' m( v; r6 |* }0 C
for ($index = 0; $index < strlen($data); $index++) {( l# Z/ G7 M) M) B/ [7 e
$decoded .= $data[$index] ^ $masks[$index % 4];+ q2 F- t, q, a5 K8 }1 h) {
}5 a& t) M; [2 \1 I% T0 \
return $decoded;
* Z% ] z6 L* m
}! I y# f4 Q) y) }# j
$ H( p0 L& z+ T9 o- y/ \1 d
// 返回帧信息处理
4 b: H0 f2 S5 D/ S
function frame($s) 6 K# C) ?$ C. w% Z
{
9 E4 B6 K2 R) C d
$a = str_split($s, 125);
9 s7 {, N- Z `$ |/ g' Z
if (count($a) == 1) {) k) C* p& X* d# g+ D( p) n
return "\x81" . chr(strlen($a[0])) . $a[0];
+ M/ c6 _6 U8 M
}7 O% t" z- n! Z
$ns = "";
- r/ ]# L3 N4 q3 l
foreach ($a as $o) {6 @! s5 U6 [: h5 y& H# ^( f
$ns .= "\x81" . chr(strlen($o)) . $o;, e6 }2 k: H- `' p4 S6 y$ z$ ~7 j
}! `1 Y8 ^; U' `: U6 ]5 { \* S
return $ns;
# \1 P5 V$ Y, y" J* U
} e1 L- g$ W) m$ o C5 c
: T4 O0 m$ V2 w8 c
// 返回数据2 C% X" \& J: U( ?, J6 l
function send($client, $msg)
; A1 X1 s+ \% a
{
. M6 m W" {5 {' X
$msg = $this->frame($msg);
& h; L# Y( ?: v ?" n; F U! D
socket_write($client, $msg, strlen($msg));+ ~- ~. \ ]# Y) O
}
( Q+ A, C: e1 f) h2 g
}
" C* l" E4 g+ c9 `( ?2 Q- F$ H% F( P
$ A, t) n, t! b" e4 ~
测试 $ws = new WS("127.0.0.1",2000); f/ o- l7 G4 H4 i8 z% h
$ v( e: j2 w& O& \8 `0 L

复制代码

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

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