PHP 简单实现webSocket

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

1）客户端实现

<html>; G& u% u4 d; n$ g/ N
<head>
* z0 W; D; C! E2 L1 j6 h
<meta charset="UTF-8">
, R0 Q% }) P8 D9 W! z% u9 }9 z8 {
<title>Web sockets test</title>
p& k i& t2 C) j# J4 D% k; V( p
<script src="jquery-min.js" type="text/javascript"></script>
& c* ]2 A) e* s; N
<script type="text/javascript">8 I; e; {0 p9 M/ T- f3 X
var ws;
# H" v4 m5 B& G7 X% G
function ToggleConnectionClicked() { 2 b" m1 t3 v; G
try {
) N4 \2 S0 |! R4 h7 T
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器 C1 b; n# f7 M+ V* w
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};
5 Z$ K( l% ?! h* c% r' @8 y; }
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);};- @! y, G8 Q) q$ C
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);};0 q0 o* D8 o. L, e* a: P! P
ws.onerror = function(event){alert("WebSocket异常！");};
0 @" A1 f9 Y/ ~7 u" m6 H" M& `
} catch (ex) {
' s$ Y) m" G; K' U2 l& b" E" N
alert(ex.message);
5 s* R$ |" |- o! f% e6 {; v
} c1 t% J. j0 H# X+ `/ c' U
};* {& m& \5 r5 _
Y; u# A' K( Y( P4 I% @" g6 r" m/ J
function SendData() {) J, \7 u) e" ^# b, f$ b% M
try{
$ e9 V6 M$ x1 z3 z% X# o
var content = document.getElementById("content").value;! t/ O' u8 E9 |3 B* y/ t
if(content){) r [5 K1 V* g: s n; N
ws.send(content);
4 p) K% l% |7 J* p* z6 E
}
' S) V1 W: ?7 e9 k) X! `
$ A; { h, m! Z
}catch(ex){% G8 v' l$ X. W, o* |
alert(ex.message);6 ~ E$ P( t. b/ c
}
4 ~7 R9 U+ l3 j
};* ^% S5 {( s: E9 T, E
5 [# o9 P0 I9 f: p4 U J
function seestate(){* f4 L- _, w2 ~7 Y! o3 y
alert(ws.readyState);
8 m' y4 Q8 s5 Q2 q4 U
}( s' |, ? q; q+ @1 w8 j
6 D! S0 h1 Z! o. J8 j9 Y% l
</script>
+ f" e+ C) @) l/ _3 k, }
</head>
5 w, V3 N; B- ]" W. R
<body>
# R. I+ v' {3 a/ B' J
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br />
: _, L0 N1 w8 l( F3 _' A
<textarea id="content" ></textarea>
2 C5 L7 _+ d) i( V e o! L
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br />' |1 H( w* P- I) h
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br />
4 {" U7 b8 G+ R2 U; G ?
, r+ ]+ T3 y2 H% E W
</body>
v( }: `, X' U9 V; q% M5 l
</html>; I0 u7 m' s9 f. k

复制代码

2）服务器端实现

class WS {; a$ B \, Z' e* i/ C" v
var $master; // 连接 server 的 client2 C3 ~) W" n- y; z/ C+ p
var $sockets = array(); // 不同状态的 socket 管理9 k# w) s6 l; v/ b. [
var $handshake = false; // 判断是否握手4 ]: d7 p6 ?* v$ D& j I
4 y3 ~5 M4 A0 c" E
function __construct($address, $port){
1 P/ T4 R7 z2 B4 t: X/ S
// 建立一个 socket 套接字9 |, @. ]0 G# W( u/ C4 G( Q& N
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP)
, E8 C2 S$ S Y1 e) T- E, Y
or die("socket_create() failed");
+ A/ U2 [4 |( ]1 H% s& V
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1) " T& L" L& `! J' n8 ~2 f
or die("socket_option() failed");2 o5 F1 W+ [! Y! G8 b& n
socket_bind($this->master, $address, $port)
$ S, t7 w+ T7 n A* [0 Z. _' C
or die("socket_bind() failed");. D4 {) A4 t9 S! X; T
socket_listen($this->master, 2) ) J. _9 t, W" ^2 C
or die("socket_listen() failed");' o$ \4 e% G6 n- p, V
3 y2 E5 [1 ^2 ?# b- J- l' @4 @
$this->sockets[] = $this->master;
' \2 `( h$ v6 q* _; o9 {
# z }( q2 h( w% M
// debug( z# Z& Y( V' Z! W! M+ U
echo("Master socket : ".$this->master."\n");
6 G0 w* p# \0 a1 D) A2 G2 ^8 @
6 S+ k$ J* U4 ^
while(true) {
. ]0 d( b$ r. V
//自动选择来消息的 socket 如果是握手自动选择主机
7 \& C# [! h2 n8 o( ^7 v8 d) X4 D
$write = NULL;
1 P" l4 m( D) J( E7 N6 \- I
$except = NULL;; n2 Y" w: i1 e3 k1 S6 m5 \- l, C+ a
socket_select($this->sockets, $write, $except, NULL);" z( q9 v! }8 Q" W
& |% x. o3 D, a( D/ s+ C
foreach ($this->sockets as $socket) {. l' }6 U# T) s4 l- G; q; A' i
//连接主机的 client
7 B# }1 I# u) S4 ^
if ($socket == $this->master){
' o1 V( K+ F8 x7 }6 ?. ]+ w
$client = socket_accept($this->master);! E+ ?7 P/ k2 b* A2 L6 h: Q# z
if ($client < 0) {; A* |9 @" q6 w1 P, T a
// debug
! ^% W( L) h1 G- G1 C. v
echo "socket_accept() failed";
; K' G& z N+ m! J' j* U
continue;0 z' r8 ^& g9 Z: ?) c8 ~; y
} else {
- l% u- y7 `. {/ G* U" P
//connect($client);8 P" h4 E* D e9 _% l! v
array_push($this->sockets, $client);/ w2 y) E, }) ]- v: n0 `
echo "connect client\n";
( `) S9 q& X5 {# a
}2 u3 g* E2 k8 B1 g. l3 y4 C. `0 l
} else {, x6 r5 \. d$ }9 e7 k$ C
$bytes = @socket_recv($socket,$buffer,2048,0);
, Z! M4 r ?6 z5 l
print_r($buffer);
( d. w1 h8 S* z @$ V# u, o
if($bytes == 0) return;; K- m: h; l5 U9 `
if (!$this->handshake) {
* J9 S. s, z2 k% H9 z
// 如果没有握手，先握手回应0 o9 [% V) E/ s, n# v
$this->doHandShake($socket, $buffer);& W& r( Q$ m; y( d3 s. C8 ?$ f
echo "shakeHands\n";
' l( s& }( _. d: W
} else {
% h, l2 q4 J" l4 F
. z) q7 R* N8 E5 E1 s
// 如果已经握手，直接接受数据，并处理5 r, I- B$ ?4 C. S
$buffer = $this->decode($buffer);7 i, X( e& e" ~" ?
//process($socket, $buffer); * P' }4 E* I& f9 M- @
echo "send file\n";
+ c9 Y! m9 E, `- J
}
# C9 F) L/ e8 g1 ~& e, v3 p+ @( c
}
" n( P$ }; p$ V ~4 O+ D
}7 X6 f, V( N& g
}
" D3 c( \ P0 Y" \" D9 ?: x
}6 s n( }/ G% H6 V$ E
& l H C2 q# b1 B$ @, M& \
function dohandshake($socket, $req)7 b5 G2 S$ J6 r5 D& T% ?8 o* R
{0 L/ w( C# D6 v! q" Q3 t9 T' E" X
// 获取加密key* B6 H) k1 ]7 E" ~
$acceptKey = $this->encry($req);
% @" ]0 r2 j* o( L) o* M9 L- i
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" .* J2 |1 U! l, h3 v9 K$ U9 N
"Upgrade: websocket\r\n" .
1 x- n* k b( z) ?1 T: |, L* K# H' B
"Connection: Upgrade\r\n" .% [% z7 T+ b, Y
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" ./ L& G$ W8 P7 W' o, K* N' O5 s
"\r\n";2 t1 o; O5 ~7 S7 O2 M& N
" V5 T9 \! z' v( G. h" c
echo "dohandshake ".$upgrade.chr(0); ' N' w1 g! F# w
// 写入socket8 C$ u" q7 V6 ] J" R" c
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0)));9 Q' e' T+ A7 R$ H( J1 E
// 标记握手已经成功，下次接受数据采用数据帧格式
( n) G$ O3 t+ h0 l4 X/ Z( Y% |- c
$this->handshake = true;/ Y" V+ h. ]2 l5 z6 m6 v
}0 ]7 f S3 T/ Z8 Q; f
5 T& [2 _$ U8 I7 h. F7 S
2 m* A& L& \% u' u" m" ~7 r
function encry($req)
. o5 l; S4 T( i/ c3 C. r& N7 n# i: Y
{
; s; q) e/ }% k5 q E. b
$key = $this->getKey($req);
0 e4 T U1 v% [8 q* S
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";
" \: w! H0 ]. p9 K: z
% H" M; W7 D, \4 _
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));
0 j- I6 H4 r( A4 \. A$ h0 C
}
1 I$ z: N* ^0 ?/ h, ^
, `, q' b9 b8 g5 c8 I, Q
function getKey($req)
6 c$ u8 e5 _* x1 F$ W
{1 C3 d/ D" z( B, l5 f
$key = null;
: q& |- ` p3 L* ?+ g9 g
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) { 7 \' ], N5 Q# J2 A
$key = $match[1];
5 G. D5 s. {$ n1 q( f0 `* n3 G# M0 a
}$ ?, K; o, ^) H* f8 v n
return $key;
% F4 q& z+ a9 z- [
}
2 J; X/ N& z$ u% d; g- F
+ ~* b0 J; k. A: d2 Y4 x
// 解析数据帧) i' A6 I$ T' Q; u4 [; Y
function decode($buffer) # c$ j6 L8 ~) y9 |& r* y
{4 E8 ^2 F6 S6 P, o! f2 ]
$len = $masks = $data = $decoded = null;
h3 U) i8 I d% z& z# M
$len = ord($buffer[1]) & 127;: u+ X$ |# b; g3 H8 N2 N
" `5 w( R/ w& p! `. l9 \
if ($len === 126) {$ R3 p& K0 [/ R; M2 N6 \' C
$masks = substr($buffer, 4, 4);
7 J+ d/ { k! S; C5 J8 r6 U
$data = substr($buffer, 8);
5 q; r: Q% g2 R- ]" J; [
} else if ($len === 127) {
" o: v) M( u* S+ o$ L
$masks = substr($buffer, 10, 4);
( L' e6 E4 m8 G" y$ w& p
$data = substr($buffer, 14); M0 h- X2 I; L- J# U+ ~
} else {( K" A B9 D5 r7 ? x/ x
$masks = substr($buffer, 2, 4);
- f" u- ^7 @' r4 i+ H$ D* f @" g" t
$data = substr($buffer, 6);
! [5 m% Z2 N/ O+ p, O0 Z+ l4 F
}; s. K: e0 q! w/ ?. o3 P) N/ m
for ($index = 0; $index < strlen($data); $index++) {
2 V9 C0 M3 Z+ Z8 \* i7 i& ~0 _
$decoded .= $data[$index] ^ $masks[$index % 4];8 I0 Q+ V; _: M) F6 W
}. H: N; h3 s' d- y
return $decoded;/ y8 O' T0 _$ s
}
/ f7 h3 f( o- Y" E
* D+ g4 j7 [$ E3 ~( G' c3 i
// 返回帧信息处理3 A6 \. b9 Q- O8 ?6 y: t+ k: I8 Z
function frame($s)
$ ]8 z' O* ~2 n. s y" }
{
; I3 a, b, v' Z3 M6 @, n
$a = str_split($s, 125);) j+ d0 u& e4 z2 v
if (count($a) == 1) {
0 h0 }6 _: _$ I0 I$ M
return "\x81" . chr(strlen($a[0])) . $a[0];( d# x8 J+ q( O: a, v8 s
}
/ |9 ~; M$ L: y5 f! f
$ns = "";( J7 Q1 B- N& g
foreach ($a as $o) {/ D9 s. i0 H/ Q' h4 ^/ I8 O
$ns .= "\x81" . chr(strlen($o)) . $o;
% G `4 c% C) ^2 I x
}* N$ G/ |+ G) L0 t& a* w* { a
return $ns;
% Z9 @8 F0 R: J6 m. x
}6 A7 E. w8 t. F+ X9 v, T- O- b3 l
6 X$ M b! o9 w7 \2 K' \
// 返回数据
: s+ c' B/ `9 [5 d' Z
function send($client, $msg)0 c( x3 d& p* C7 C/ N3 A
{# v6 U+ F. C2 J7 k: J
$msg = $this->frame($msg);
- v5 P& K0 A1 w; @$ f( v
socket_write($client, $msg, strlen($msg));% s/ ?# |3 D# _: c" @, ]7 }
}
6 X3 M! i/ j6 N$ x! v& i" ~# l* U
}* `0 J. {0 k! a/ h
' I. x) G% d& s. s4 I
测试 $ws = new WS("127.0.0.1",2000);
7 X, A8 S! Q8 {8 h: A- w3 ~6 a
& p3 X- x5 X/ u; X

复制代码

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

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