PHP 简单实现webSocket

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

1）客户端实现

<html>
8 S1 I! e. m5 T8 x9 m+ d6 R# l- o
<head>
" }' H9 j/ ]; X+ f3 x
<meta charset="UTF-8">
4 E3 x% x: C- P. o/ }* a" J# i
<title>Web sockets test</title>/ e j% Z' k$ L# `) `. e+ a' W
<script src="jquery-min.js" type="text/javascript"></script>6 f0 d5 F5 ^1 ^+ s
<script type="text/javascript">* v: K. X7 W/ X9 m. g
var ws;
# n! A: P$ H& T: i6 Y
function ToggleConnectionClicked() { % u4 V% h7 C" o* ]% ^
try {
# M/ s* y! Z- @1 e
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器
1 ]2 J* x; v4 B8 i0 Y8 h! i2 K
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};! N: s' x+ s0 c5 |8 ^, i7 g( _
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);}; Z6 Y8 e* _0 U
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);};
* w3 B& f8 [$ G, ^# @2 l3 G, ?
ws.onerror = function(event){alert("WebSocket异常！");};
2 f" h, m7 v; U* \
} catch (ex) {$ x9 D( }! J- c; l+ [+ P7 S. b( k
alert(ex.message); " t1 u: x- u& l- K) a( L& V) V
}
5 q: b& |8 [% _+ S0 _3 U
};- r9 |8 `3 a( C+ r8 d+ @" d4 L; e
7 o' b. v5 F" y/ v, d
function SendData() {
7 d, o! Y: ^5 N, ?3 r' ]
try{+ @8 Y9 F9 @, l" R9 D
var content = document.getElementById("content").value;
- e) m# k& l4 r |) _; x5 F& m; \: j
if(content){1 l" n8 ^4 g. N/ d# l0 ~! Q8 f
ws.send(content);! N- [$ g G; \7 O$ ?$ D
}% T- K$ t& e: B7 b4 I' L/ o. O- ~
3 d! q( l" w) Z" b7 L" T& g" J
}catch(ex){
* a$ d! l4 j9 e( f/ Q
alert(ex.message);* |: I" ?1 B" X
}
: K! ^/ j8 O5 M8 z$ C$ m+ o7 |0 e
};
# ~; Q) t+ a, R" |0 i5 A$ ^; O
" ~6 ]0 D; O: d3 e
function seestate(){( Q( S" K* W$ s" H1 c
alert(ws.readyState);3 I; N7 ^+ ?1 A9 D" u* R: R
}5 A) M, |0 [( O
1 r; j: N, r' o0 C5 ]5 W
</script>
; l# y! a0 }# U! R) E* S) p
</head>* }1 C ]; T9 L6 g, z
<body>
; f4 {( e. a* D: z
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br />4 p y& b% E! n0 C1 o- M
<textarea id="content" ></textarea>9 r$ f# M% p- B" V9 B- n: Z5 W
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br />
, V" q5 w" C& E
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br />* I" c5 r0 z9 S {( p) [5 G
( q/ l5 M7 S" |" F; b% a" D! O
</body>
, u; V" P$ x$ A3 u! s" U
</html>" N: |/ _! M8 ?( H7 V4 N

复制代码

2）服务器端实现

class WS {6 { v: o/ r) V4 Y( ]2 j( X" X4 s5 o
var $master; // 连接 server 的 client; l0 Y, G) e/ ]; W& v$ x
var $sockets = array(); // 不同状态的 socket 管理
: L/ p: z9 ?2 {
var $handshake = false; // 判断是否握手
( f" I' ]! X( J. ?" o* m5 e
Q2 w( T8 v( D* j0 T/ C3 t0 N
function __construct($address, $port){: G% C; v8 O3 o2 E( g
// 建立一个 socket 套接字7 f4 o* u) l% b. S6 R
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP)
5 I% Y7 J* t$ p9 Y
or die("socket_create() failed");
, w, U' l6 h& M7 z6 A, P' u
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1)
6 |5 v/ q! L( t* N0 a
or die("socket_option() failed");
' Z' V6 H; j* g$ t/ l* r
socket_bind($this->master, $address, $port) 7 w0 z2 h9 a- m$ ^- H2 O
or die("socket_bind() failed");
$ @8 G! f1 S7 E4 o" P$ f( k
socket_listen($this->master, 2) ) ^* M+ {8 b. V6 {" {1 C* ^+ u
or die("socket_listen() failed");
5 U* x- s& Q. A; O; S
3 H/ Y9 s0 Q4 k, G
$this->sockets[] = $this->master;
* q7 w: S- M6 x1 U p' t( z, O
5 K- Z2 R: k+ g2 J7 E- W
// debug# {' v3 h# A9 O# d! M
echo("Master socket : ".$this->master."\n");
9 P6 k6 k3 P5 C4 V& f
6 N E! d: |+ t, D. s# ~7 F$ N& V& I
while(true) {, Y0 h( v$ r. l8 h8 r, O6 V+ w& `
//自动选择来消息的 socket 如果是握手自动选择主机
. m% x( T9 R0 E/ Q# V$ h2 P$ `
$write = NULL;" U, U0 T8 K7 j4 N3 s
$except = NULL;+ v6 P) t- N8 M. A
socket_select($this->sockets, $write, $except, NULL);# F! J* G+ ^3 R
7 t9 ?4 g- Q4 d, I% r
foreach ($this->sockets as $socket) {! q& ~# L$ b# Y9 |% Z# x. j+ I
//连接主机的 client
2 j) u! t2 C( K. y" O3 ?
if ($socket == $this->master){" H2 Y1 o8 u2 B# t
$client = socket_accept($this->master);
1 d( l. s3 K4 W* g7 B; w$ J/ u
if ($client < 0) { i) Z* B. u3 u3 S
// debug
# w! u( `& N9 L3 I/ }8 m; ^
echo "socket_accept() failed";
t# B4 y# R3 ^" x9 w# I, K; q; h
continue;; T! O8 k8 F4 a6 k. ~: s0 I
} else {
% M/ S/ `- C+ c4 t: ^
//connect($client);1 |/ f7 H( g7 K) R+ E- Z- f ^6 n$ H
array_push($this->sockets, $client);
3 \* ]8 p. v3 g9 w3 \' Q* y
echo "connect client\n";
% s- T8 i" s% _( V3 }! }
}6 G: W7 e3 {6 A" T& E0 t( o& ?+ ]
} else {
$ I( f2 g& }) p' m
$bytes = @socket_recv($socket,$buffer,2048,0);
& M# U% m$ C" {: }8 W" T$ C4 J
print_r($buffer);, B* q1 A2 H0 y& E1 x; _" {+ V
if($bytes == 0) return;
# {3 K8 [" B$ o
if (!$this->handshake) {
: E! Z: C: H; A: ?& o
// 如果没有握手，先握手回应
9 t3 p2 @8 \; b
$this->doHandShake($socket, $buffer);
# W( v3 ?) `# U8 F& P- }6 z1 n
echo "shakeHands\n";& x0 Y8 K4 {6 H5 F' s/ p+ [
} else {4 M7 j! }6 S) V2 K
; e) O8 x* u# d
// 如果已经握手，直接接受数据，并处理" n) {$ f% w8 G/ l
$buffer = $this->decode($buffer);& l' ^7 I( D- Y6 `
//process($socket, $buffer); 5 J3 C& P% D. t* n6 U! _
echo "send file\n";4 ~9 c0 k6 V' {' m! V5 Q$ K: g
}
- o; _$ e5 R3 |. T D5 w% F
}/ N2 d7 \. k/ J# d8 H3 Z7 Q
}
: c' o/ L- j1 |2 j
}
, x, y2 U/ M6 b6 e8 D; W: D
}5 [/ ^: C( [8 r( Z, b
4 _2 q( C3 {% P: m( k/ p1 M
function dohandshake($socket, $req)
3 H* h! j7 f0 x' R
{: l) o( b k5 R# }$ ^& D
// 获取加密key
x# ~# R4 `3 r6 |, J' ]
$acceptKey = $this->encry($req);
: M: a* ]/ Q; m J9 \, T
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" .
: }1 R- Q( U, v/ H2 [
"Upgrade: websocket\r\n" .
1 [% w8 v6 i1 r9 p$ l
"Connection: Upgrade\r\n" .
% q: w, `" r! r( H; T
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" .- ?, j1 S3 B* R. t! O2 Y
"\r\n";" ?3 ]2 n1 \5 I) u4 z! N, [
% F) e4 J9 `4 F6 _' G! X( X1 w9 M
echo "dohandshake ".$upgrade.chr(0);
/ C* z+ q' k: ^4 w4 s
// 写入socket# }; d: l' j3 {* {2 r9 y- t3 c
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0)));7 u( b1 |3 ]% v
// 标记握手已经成功，下次接受数据采用数据帧格式* J4 g: \& O; F; W. `
$this->handshake = true;
' V I( {: P& ?- M
}
) z/ t" O3 @5 o1 p7 z
$ R1 ]4 }* m! r0 I' Q7 d
3 G5 M3 {. N6 z6 }4 d
function encry($req)- S$ u5 S' q' f8 M6 D. R9 U; j
{
- a! [( P( g+ a7 G
$key = $this->getKey($req);
& o/ i: d$ N- u& i5 u
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";. M5 i* Q: n7 U6 @
' u2 n- o0 r0 w$ Y
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));
0 b" `5 d4 h9 J3 B4 a1 V2 l
}
) I$ [5 f8 M" `; ^
2 u6 U/ s! m& j9 R) f/ g. R9 M
function getKey($req) 5 A5 Z* Q) l( C$ |% G% `3 ~/ `
{6 F; V1 t5 F0 k' ]; ^. w& X* C5 |
$key = null;
% g/ ~' ]' @" |( K8 Y3 Q
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) { + D; r1 R+ N* B5 h: L" F
$key = $match[1]; 6 ]3 S0 l$ R. ~! P
}
# _* \! S$ s2 `
return $key;
0 R& B; f5 i s9 B! X
}4 \/ ]/ E, P4 J* [, W9 m
2 {" \$ C4 a( ?9 a1 S J+ M; f
// 解析数据帧
9 @ w; \3 a% s6 O5 S: M
function decode($buffer)
* l4 k6 b& Y" y# _0 F1 H8 Q
{% u) K, a1 u9 D2 X! ^% i- c' e
$len = $masks = $data = $decoded = null;& b2 L0 w, I8 B* a; H! ^! H4 M
$len = ord($buffer[1]) & 127;/ C9 \ A) s. N4 G/ U; i4 D5 J0 _6 h, _ l
- Z2 ~" i+ P% @! W
if ($len === 126) {& V& O* K" m. M' ?: h) t4 b W* W
$masks = substr($buffer, 4, 4);
1 F7 h- w1 {( b3 I0 e* P2 W
$data = substr($buffer, 8);9 }! ~1 M) ^) l6 B' y" d1 W: J
} else if ($len === 127) {/ N' j# l% C1 U- w: W
$masks = substr($buffer, 10, 4);
" x% j7 v% z9 _- }+ p
$data = substr($buffer, 14);
* c8 @. H. i. o: h. O
} else {
+ ?. \# @& p$ U
$masks = substr($buffer, 2, 4);
8 \$ X0 y/ A/ y6 a) O
$data = substr($buffer, 6);- f+ o/ i* o+ i$ g2 v. }0 W
}
5 N8 s0 q7 g3 z1 V# h
for ($index = 0; $index < strlen($data); $index++) {( M0 b% A1 j: _1 b7 f
$decoded .= $data[$index] ^ $masks[$index % 4];
2 t+ w5 I9 Y& P
}
6 ]" ]& j0 [+ F/ f
return $decoded;: `# b8 n) ?7 `" m+ q8 E: Z
}. g! P$ G+ V3 x; I
- I5 e% y9 @* c. I. i- b
// 返回帧信息处理
7 Z/ P8 d% H7 n) u" Y
function frame($s) : |# t. v" g& n, n3 f
{
0 T8 J1 u/ i/ { ~1 S
$a = str_split($s, 125);
* B, T4 P; K: u7 c9 ?/ y7 F/ M
if (count($a) == 1) {
# k' _( H9 G! ?4 x% J0 `4 L! A
return "\x81" . chr(strlen($a[0])) . $a[0];
e" [9 D. o7 w& U% i L: A
}' q* b$ R, w% \4 T
$ns = "";2 b. _. |' W; i5 x2 G
foreach ($a as $o) {: F% Z W: u" z) K: _0 b6 I# b
$ns .= "\x81" . chr(strlen($o)) . $o;
) i- L0 Y. S# _ \! N' _3 s
}
V4 X* Q/ F' [7 `. U2 Q) y5 y8 d% @
return $ns;: H" z7 X. `# N5 ]
}" M; z* Z: ]0 J! k2 j
+ @( M% V: W) A" t% T b
// 返回数据. T- D/ P% U! g5 ~% k
function send($client, $msg)
7 n2 K3 I2 W5 E1 j- H
{
5 ~; N$ Q& z# l8 w4 K
$msg = $this->frame($msg);
1 h" y4 V) _% L/ a( o/ i, z9 _
socket_write($client, $msg, strlen($msg));
% Z" f5 L! H, L- [) V. U2 x8 \: h2 @$ @
}! V! ~% O M _: e
}
! h* D- D" `" ~; v2 P( B2 l
; s$ W5 c1 w/ w, J. c- s( L8 p
测试 $ws = new WS("127.0.0.1",2000);& [" g; @: [5 Z+ G7 C0 V0 Z( ~4 i
! f1 P3 |% f& ]. O, g2 r4 R

复制代码

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

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