PHP 简单实现webSocket

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

1）客户端实现

<html>
]% T5 y- B q" W$ g; k
<head>: x! A4 ^, o: s& E& Y) ]1 A% b
<meta charset="UTF-8">) e7 a. D. \! K/ o1 G& X) T
<title>Web sockets test</title>
& w2 T# t. B( ]6 g3 ^
<script src="jquery-min.js" type="text/javascript"></script>/ `5 v2 h9 z( x6 D9 Q
<script type="text/javascript">9 Y* u; I/ f8 t+ U: U0 K0 Q$ N6 ?7 v& L
var ws;5 u% x/ b) Y& |2 o* o
function ToggleConnectionClicked() { 4 U6 d% x8 `; I3 y; c
try {0 D3 B4 c2 U, S( i3 E+ B
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器 5 x/ @1 e" L& C/ k+ d) }
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};
0 r1 e& e1 {( ^3 u/ E1 l
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);};1 c3 X. A4 U# ]2 t! @, l) a
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);};+ k* }9 K4 d7 J
ws.onerror = function(event){alert("WebSocket异常！");}; Q$ D- s. P2 n; X" S
} catch (ex) {& `/ g- |: g, r6 O: f. I5 m
alert(ex.message); / ]* f m1 A5 O i( t* Y% [
}
1 v- H+ u5 o2 y1 m
};& z; j) K1 O' A+ d
& t6 Z% e8 ~1 j! P0 N) O
function SendData() {7 V' D: c4 v& W' i. p
try{; L$ @4 L+ {& [+ e/ s7 S3 z
var content = document.getElementById("content").value;0 g" S4 v6 Q9 J' P% W
if(content){+ i- `* N( A1 c, h+ K
ws.send(content);
% ^$ ]! k+ n! c+ k# Q/ L( I
}
% ^. U/ O) x* v+ f1 ]
8 V6 S7 z+ M: t; U- Y
}catch(ex){- u9 @# u- e; w% ?* J$ Y
alert(ex.message);: |+ s. d0 ~/ W! v& t
}
' r+ [% c8 B, I+ H8 C6 Y4 `# W! j
};
* Y$ h2 Y' w1 H5 O/ o' U- c3 t
4 D0 f2 w! g2 Y! X
function seestate(){* c* t# b# s* b2 |2 ^
alert(ws.readyState);2 |: L- J9 O* l3 n6 P- `
}
+ Y0 R: l4 L' u: P& h N
! ]6 M0 D3 |, n5 W: a8 G
</script>/ u7 k6 M0 v0 {: L$ X& M$ f3 D1 x+ f# R3 Q
</head>- k8 E6 R4 v" q' l5 A ~2 |
<body>( Z7 C; a1 i7 D
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br />7 {" c# J* V' _8 p/ ~4 }
<textarea id="content" ></textarea>
i4 y3 d7 G2 {# o
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br />
: }) H7 l5 U0 s$ \: U5 H
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br />. ~ b' f& t% p0 B0 b
! ]1 d' S& b! j4 F
</body>: C6 b! p' ~8 s' P2 W
</html>& U# e7 ~; a4 Q6 o3 A

复制代码

2）服务器端实现

class WS {7 ~4 T' x6 c$ Z1 |! I7 |7 ]5 z
var $master; // 连接 server 的 client! X1 |0 z. K! \ r% a3 q1 p) h
var $sockets = array(); // 不同状态的 socket 管理& b. e8 Z# h7 r! w9 `
var $handshake = false; // 判断是否握手
. B" V: o( p& h) N' e% @5 o
# C8 n$ K# Q0 n9 `
function __construct($address, $port){
7 }& K! q. o( S7 c/ ?5 F
// 建立一个 socket 套接字
! [2 A6 G7 ?! B* e: l; t2 M+ z+ J; \
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP) , F! n3 ^. O) E7 K- ]
or die("socket_create() failed");( q% r$ m1 S+ K8 r/ O4 }
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1) 5 |/ L1 j8 Q/ v4 u" x# t
or die("socket_option() failed");' B$ p' N# T+ _, M: p' c
socket_bind($this->master, $address, $port)
4 {, |4 }: p' N/ |( E4 T$ @
or die("socket_bind() failed");
/ r" C: |9 w' T. N( R
socket_listen($this->master, 2)
+ s; q3 o6 F3 J; S
or die("socket_listen() failed");
* B( f8 p9 x5 E& D# B% A
I5 h3 P* \& x" m$ j
$this->sockets[] = $this->master;; k' S; j/ j. N5 Z2 p# h
/ }; a. E! k X, t* L
// debug
! ]2 B5 T# U" N6 X
echo("Master socket : ".$this->master."\n");6 A' D& c' g+ |
/ F9 N3 `, |, s* m3 F& V
while(true) {
. T. x. ?0 j/ y" |6 c
//自动选择来消息的 socket 如果是握手自动选择主机
2 K2 @, P9 q2 q6 a; E" R
$write = NULL;* M! Z/ j1 B* k2 x* A
$except = NULL;( }1 z( O$ a3 w" h! L2 Z
socket_select($this->sockets, $write, $except, NULL);
3 v/ m- K- g+ c' n
/ b+ k0 P( I. f, x0 `
foreach ($this->sockets as $socket) {
$ q) e1 ^) q$ {, b) ~ r
//连接主机的 client
5 I# K) C6 ]. m8 w1 Z
if ($socket == $this->master){
* |* F5 z0 P$ @. z" P2 `
$client = socket_accept($this->master);
+ f1 k1 p7 S. u) O! B( X3 r
if ($client < 0) {1 \& a( ^2 `* ~0 d0 j/ O
// debug
- \% |3 U+ Y( ]7 p/ w' h
echo "socket_accept() failed";% ]( u& J$ s/ v1 g& H- Y" ? a+ X
continue;
% f) g, v: C$ D" s0 q
} else {# T1 B# U" V3 r. }0 d
//connect($client);
( q9 S# m% P. z% f, T/ @' p5 Y- h1 l
array_push($this->sockets, $client);& H5 N$ N1 E9 F9 d; @8 P
echo "connect client\n";
' X8 I3 `; u D$ [+ U$ S' `
}* ~. X2 | K2 x9 i* W
} else {0 l6 ?2 n' l% W) o, w) U' A
$bytes = @socket_recv($socket,$buffer,2048,0);# [# Q7 P" x. k3 K# `, n( J% _0 w
print_r($buffer);& M' S" D3 B0 l& `
if($bytes == 0) return;
9 C, H; X% \" s, I, v/ t% a1 @
if (!$this->handshake) {* E; }! v n* {! a3 v' O2 ]" c) x
// 如果没有握手，先握手回应
6 g- l' |: g* T$ C% m5 E- k T( |+ J
$this->doHandShake($socket, $buffer);
4 q, U9 H1 N4 z3 S! J( k. j6 L; M
echo "shakeHands\n";. u- u6 H" n* I5 D- T8 q
} else {
9 q' G* {3 B# H
) j6 J+ U; L9 h( {! M
// 如果已经握手，直接接受数据，并处理# b8 ?, ^5 s) B) B7 p* J
$buffer = $this->decode($buffer);
+ ^+ |. m, p7 Y6 H
//process($socket, $buffer); + C4 X2 E4 `7 {9 C: m
echo "send file\n";9 j! f2 b! T* R; S3 `. n
}
+ w* u" s) S# J* A# E! w3 O
} H5 H# ~% ^1 I. \, Z' K# [
}
/ G3 t1 P0 B9 Y, M
}& X( L& T. u `
}7 ~% }( D$ S! j1 |" e
$ F8 \; ~/ u3 F4 J, q
function dohandshake($socket, $req)
: d+ c* p+ R) f0 m
{- i1 D& [% y/ a. \
// 获取加密key
8 I+ v3 [' E9 b7 G$ m4 n" X
$acceptKey = $this->encry($req);1 V" H, M5 C) P1 x9 s
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" .9 P2 S9 U7 Z7 I; O4 x, f# o
"Upgrade: websocket\r\n" .
& f& y* M2 t2 S: v5 {3 ?8 {" P
"Connection: Upgrade\r\n" .- a C+ i2 S$ ~) o8 |3 G: O9 U
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" .
6 ?+ Q- } i D2 }. e" ]
"\r\n"; y! y* m3 ]" ]+ U A, \ x5 L+ v
: c; k5 l d, x$ J9 x0 F- j
echo "dohandshake ".$upgrade.chr(0);
, \. h( K" p- W* u
// 写入socket4 b; x5 Q$ R- ?5 c3 W
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0)));4 e* D6 F0 ]. w! P
// 标记握手已经成功，下次接受数据采用数据帧格式4 w+ ^2 p7 @7 A4 J# ]% E, [
$this->handshake = true;. K S7 A# [& G
}4 ~3 W! V7 ?, {3 m, [9 l0 A
: G9 Q3 o+ G% ?2 v
0 O% P5 R& Z' J
function encry($req), v9 U9 U3 x& y, c
{0 h, x ?+ R. d# P7 [
$key = $this->getKey($req);) ?$ a8 ]- a' o+ V6 R9 @, q
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";( _1 d t5 ~5 R, t" S
& M! p" P/ o) w1 T1 H. M+ a
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));
, Z' q% M% Z9 h( I5 o; j7 ^3 Y
}# s( g% Y$ ?, V% W0 G: R+ G$ b& r
6 X; C% L4 H4 t5 l3 I8 T5 y7 a! B
function getKey($req) - Z: m- e3 q5 f. e1 {# ~
{! F2 S6 t1 P, W
$key = null;
3 k/ K z3 ^4 r: j: | I
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) { 0 a0 F% y6 \1 |& g: b N. W
$key = $match[1];
/ M: p9 O1 M& n: ~5 G9 k' g
}% A( s0 S' g# Y- Z* }
return $key;! H) ?1 k/ j+ [& j
}
; Q2 Z3 ]0 k! M" T) L9 }* p
1 V& S9 L# f+ E- ~
// 解析数据帧
% i6 |" g4 |" {* G
function decode($buffer)
5 K. t7 E8 k; k5 C3 O; t
{
& h+ i$ W( F* _. T8 S6 n1 t+ d
$len = $masks = $data = $decoded = null;1 d7 S* H8 ^) P+ x" p3 T; x: w
$len = ord($buffer[1]) & 127;0 x3 A4 Y* Z- i6 f/ ~5 v
$ M/ j# O& m5 n$ p) k+ H, h. m
if ($len === 126) {
. L7 z* r' \1 [8 @
$masks = substr($buffer, 4, 4);
\* W. \8 j* _" F! F
$data = substr($buffer, 8);
$ F3 f! d, D9 C5 C
} else if ($len === 127) {
3 C0 I4 ^* W; t6 O
$masks = substr($buffer, 10, 4);6 J. D% Q& V" a6 J% `: s
$data = substr($buffer, 14);( m/ O; D, N/ G! @( X
} else {
( j2 |% L: w. ?4 q4 Q! {* f% h* K3 J
$masks = substr($buffer, 2, 4);
9 i8 y, E& N% A4 H
$data = substr($buffer, 6);; l9 w4 R5 G) ?* k; C& Q3 \& H
}* k2 J) w7 S; I! ]
for ($index = 0; $index < strlen($data); $index++) {3 r. v: X1 M. m: F/ ` ^- P! D* D: p
$decoded .= $data[$index] ^ $masks[$index % 4];
* m1 w A0 Q# K: n7 G9 R
}
. f. X# l" T/ b6 T* q7 S
return $decoded;: \8 U7 F; F0 s8 C* G
}! W8 p- P& p. p/ p
1 p8 H' l# K4 h
// 返回帧信息处理8 g: e; z: h; h
function frame($s) 8 ]. [1 G# K: X+ H
{6 P& l4 O3 X" n$ L" `, C- u. S- z
$a = str_split($s, 125);
+ C0 x D- ^0 s, n( ]2 v3 Q
if (count($a) == 1) {% n. ~% c) J! |8 m- }; r6 C j5 a
return "\x81" . chr(strlen($a[0])) . $a[0];6 }) t1 j3 V4 Y) {; l7 z9 @
}* t) Y2 X6 c. @( a4 j0 J
$ns = "";
3 m& y6 L) C3 O+ {0 ? @$ c2 A
foreach ($a as $o) {
! P* _2 f/ G+ z, D/ T# B
$ns .= "\x81" . chr(strlen($o)) . $o;0 K# F9 `0 D* h- G T
}. K; ?' ~# j( s3 T& X$ N
return $ns;
) B( a2 k. l& @/ | k1 b
}
N5 I4 B4 u8 {1 C
* @9 n' C8 E1 H
// 返回数据
9 N/ E1 c n' |! W! w
function send($client, $msg)$ n" z/ z) \; m' L& R0 S" }- Y7 Y& D
{: `0 R/ C" y; Z2 L
$msg = $this->frame($msg);1 D) l. ^* d) e% R; Q
socket_write($client, $msg, strlen($msg));. O) K' m8 `& Z' |7 z
}
1 F$ M3 W# z3 K& \* X# `& M( G3 U1 j
}
0 F) d9 q6 F n1 H: A% _3 J8 I0 H
4 C3 L- ^+ H ?7 f, K; C' I
测试 $ws = new WS("127.0.0.1",2000);
" \- q" _8 J% Y$ B" |' R9 _
1 z$ S5 g) s# p: R* p1 O

复制代码

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

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