PHP 简单实现webSocket

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

1）客户端实现

<html>2 |/ W$ N* A1 L; m
<head>7 i" Z+ x d2 X* F4 x1 Z
<meta charset="UTF-8">6 n% d8 o. j3 b1 `* T. q
<title>Web sockets test</title>
! I& \) D6 `% T3 _& ~6 j$ }
<script src="jquery-min.js" type="text/javascript"></script>
* I: d1 I N! @6 K( C
<script type="text/javascript">/ ?' C, D5 m, e$ T
var ws;8 l" R0 D& H2 t2 }
function ToggleConnectionClicked() { % B; Q0 ~/ F& w: I' D3 C) M1 m: s9 i
try {/ b/ G9 b, [+ S- X) [( _
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器 : o7 L5 ?3 B4 l' w) K: p
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};
8 b! @. E! | g& n& S1 R; u. W
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);};+ v! Y/ f' _2 }; M$ V: w" z+ W
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);};' O# u, Y/ ]8 F/ ]7 F8 Y! I
ws.onerror = function(event){alert("WebSocket异常！");};1 q3 s0 t5 s; l# i
} catch (ex) {- ~- R. C$ [; J6 g
alert(ex.message);
3 C/ m: l. D% c5 V# a9 v' F
} |9 h/ @6 P! f! m6 t* z: t% S
};, M& Z- r1 v* G" p7 H: I+ {
" N; s' o" a+ P& `4 W7 i% C
function SendData() {$ t4 M+ V& N$ f5 E \3 Z4 Z
try{* T: d; s( ~% o+ s
var content = document.getElementById("content").value;" G; g; [" s1 U, d
if(content){
/ H" h! O0 M- a! U+ R- @) N! Y
ws.send(content);1 g" M; R2 S" p) G* n" A; t* z
}
) ]$ U0 d6 @0 R
- ^7 X& x. [! O. J$ b, S: e
}catch(ex){
* _1 d: V! C4 g) y- J" z5 a
alert(ex.message);
' o. [6 E# \4 o& G- T0 F) a2 j
}% N3 a$ _6 U" v" |, D
};
8 W. t% D; W6 g5 Z$ P
3 c; ~, n) C) N) H0 C
function seestate(){
4 W# m, d. q8 ~; R- @+ c
alert(ws.readyState);
1 i/ [- P+ ~) T. w7 @% L
}2 K) `) a9 t- x8 b
" v9 B$ \1 t1 t' Q0 A
</script>. N6 \2 z9 Y* x( ~9 t" }
</head>
+ S. o2 x5 F/ H
<body>: h+ _) ?0 I, K/ y+ b- v$ C9 Q4 P2 m
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br />
% ^; F" A4 U1 ^& W: J# A
<textarea id="content" ></textarea>
' k0 `2 e2 o, z8 T
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br />
% d; {, g- t# b6 i' m& Q
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br />
$ }$ I2 V6 r5 J1 r' @
/ ?; C; f" W' E/ Y, I4 Y& M( e
</body>/ M9 v' l ]% H8 q
</html>4 [' \5 c4 Z( P, e. q/ O/ W

复制代码

2）服务器端实现

class WS {
( ^: ]5 H# U5 \ E% k" X4 X
var $master; // 连接 server 的 client
- @% F0 J6 ]4 B# w( ~7 y
var $sockets = array(); // 不同状态的 socket 管理1 G$ o8 q, e2 N; I" D* j) q5 J2 l
var $handshake = false; // 判断是否握手
+ R$ `' j2 d2 A6 L6 B5 E
8 `( G# O2 g8 J+ _) L8 `1 T
function __construct($address, $port){
1 E" s5 D5 {- x& P4 w- B$ r
// 建立一个 socket 套接字
" U/ U( Z* a4 C1 G9 E3 C6 o- l
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP)
+ b3 u( H& `* D' q# {
or die("socket_create() failed");
! z5 D2 d4 F1 Y1 z" f. ?7 m
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1)
8 x! r% E- O! F0 p+ L% I# ?8 C! z
or die("socket_option() failed");( V- P) Z; f5 X, p* U4 }9 T2 E
socket_bind($this->master, $address, $port)
4 k. ]1 E. I/ i# s) \& D2 F
or die("socket_bind() failed");9 c1 T3 v! v6 `
socket_listen($this->master, 2)
& Y/ C/ ]# L9 @& W" o3 P
or die("socket_listen() failed");
1 d7 x' n& N* u3 x( s
( h+ u g$ v4 T* p1 ], e+ y
$this->sockets[] = $this->master;
6 B' b* ^+ }, m# f( d! D4 O% c
, g3 Q% n8 m6 q4 [- ]1 ]4 K9 z+ F
// debug
3 I A8 h, g; @8 J6 q! y9 }8 C
echo("Master socket : ".$this->master."\n");/ ~- p' c0 Y2 N) B, i7 M+ b
( I1 W2 l! d" ^ f @6 Z. u" ^
while(true) {
2 A$ D0 X% G* ]
//自动选择来消息的 socket 如果是握手自动选择主机8 ]* b+ V4 g3 o7 m7 f% L
$write = NULL;" c; p$ V. E" m9 D! V/ }
$except = NULL;& k; Y+ S. c6 `4 R
socket_select($this->sockets, $write, $except, NULL);
: K" O0 R+ n; E7 `4 s
% M2 m( u8 [$ F: i' z Z+ R
foreach ($this->sockets as $socket) {$ g1 `% n. d/ G' s* q
//连接主机的 client . U/ T6 z/ f, ~& \8 z
if ($socket == $this->master){2 F, l2 q/ d9 H2 o0 D, t: ?9 j1 k
$client = socket_accept($this->master);6 i$ o! p8 G+ [- L$ G
if ($client < 0) {
. t+ ~3 b Y, F% D% M
// debug
% I9 k6 l) m$ s4 m4 O& C! ^
echo "socket_accept() failed";7 N" X. F3 ?: h# ~, d
continue;
$ e7 T- E! z& F" A4 Q+ V( k- S
} else {) L- u7 f: _* d5 y; F
//connect($client);
$ Z& q0 q; `1 e& [4 j- a5 `
array_push($this->sockets, $client);
Y+ X% V; N6 u* L
echo "connect client\n";
3 |( |: C- b4 J0 @
}/ B. u f7 O! j' D1 n
} else {' O, F; N: B7 M& O$ R8 N% W
$bytes = @socket_recv($socket,$buffer,2048,0);8 M8 h& p6 P1 q) V. Y5 U
print_r($buffer);; X. G) q* ^! U) R
if($bytes == 0) return;/ ]! s& n1 D2 s
if (!$this->handshake) {* X+ E# s) [5 a7 M% i! s# E" a( F; {3 ~
// 如果没有握手，先握手回应# e0 I! `* e. B# `0 V; }
$this->doHandShake($socket, $buffer);
3 v2 _ A+ u) ?0 \ D: w
echo "shakeHands\n"; Z& j. D: ]" L
} else {
. T# o5 T, @4 p9 k
6 X6 J% J8 y' }/ t2 m
// 如果已经握手，直接接受数据，并处理
4 \ c( [7 ~$ Y! E1 q6 W% x9 h6 M/ s
$buffer = $this->decode($buffer);
: M: X3 ^% {0 R9 H
//process($socket, $buffer); ' {' d& q+ _2 X7 Z1 a8 Y5 r
echo "send file\n";0 `" _* n' p( J" B: j
}
5 ?" V- E( m( \ v
}4 Y% ?! T% C$ a# R9 \
}
/ W1 l2 G2 L2 m. a# l: v
}
( {7 k$ z8 ^# `1 C0 z
}
* k! t, |6 h* b: b0 t, @
5 t7 s! \ z8 d; w, g1 b$ H0 F
function dohandshake($socket, $req)
* t) ~: I& m0 \; i& N0 G+ ], K
{; ]9 F. {. {$ Y5 Q- V) L: ?
// 获取加密key8 l2 \3 I/ d8 H7 ~9 K$ j
$acceptKey = $this->encry($req);! o. I0 J! Q) r9 L" w5 U4 l
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" .
9 G1 T8 }: `" G( z# @" I
"Upgrade: websocket\r\n" ., u. x4 p$ X! X2 K
"Connection: Upgrade\r\n" .6 `) |/ P, b+ u) Z2 |
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" .
- V' ]$ b$ {: c' ?' Z4 K
"\r\n";
3 X% A# r: Y- w3 _5 K) G5 T
4 u" K0 F* Z( ]% H/ A1 d) Y
echo "dohandshake ".$upgrade.chr(0); 8 O5 D: V* N; D! f& O# J
// 写入socket
8 n; @- }$ ]7 t& y% R
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0)));+ s! @0 D5 o! }# x
// 标记握手已经成功，下次接受数据采用数据帧格式
* f) h: P8 G6 s* \3 {+ I
$this->handshake = true;( t- N& @- C3 b8 z. H. I& f1 G
}
3 a( R3 v, ^2 `3 r9 L5 a( g
6 @; f4 b( t9 B' u" E. K7 O1 l
5 x: t/ e* K3 m6 x- K
function encry($req)' |' K. ^& k: b! @
{
! i$ {6 E$ {" I6 m8 U
$key = $this->getKey($req);
" L. Y5 L$ z( B! `7 i$ h: B
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";& ^) b3 i6 S ?5 c
$ i% P4 \* G5 i. F. D1 s
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));2 b2 @$ W# p3 c7 H) p {
}
, H- E! x7 j( s0 a' Z
! e4 D4 n/ Y% j5 U- X. Z
function getKey($req)
- `; n9 f0 f' h
{
1 j8 y5 G9 o7 c6 W. `4 ~
$key = null;
* F) K* @/ U2 i& \; C
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) { ' D1 y) H( f6 G! U3 U
$key = $match[1];
7 T# T- R9 k0 t
}; b) S2 r# ^- s6 I
return $key;1 u- K" h5 {9 M, j
}
6 }6 X) M& Y$ V8 _: W
2 {, \6 ~+ u, E
// 解析数据帧
+ u9 h& x/ c% ?: i2 W+ T/ _/ z
function decode($buffer)
- r. u& W4 E4 P6 x9 L* g6 O
{7 U$ k8 w5 M1 V! v/ S' T4 N
$len = $masks = $data = $decoded = null;
2 b2 c9 P) V$ f; ]1 L
$len = ord($buffer[1]) & 127;
. H+ a, l$ ]. k" y9 T
( [ \. ]- P4 @/ w" s8 n
if ($len === 126) {
2 q z0 _. M5 x r0 X: T
$masks = substr($buffer, 4, 4);
. R @0 t4 O& I/ I/ x+ w7 V7 N
$data = substr($buffer, 8);* X& n2 B, T& E7 `. h
} else if ($len === 127) {/ d- f0 h, E! T' r' {
$masks = substr($buffer, 10, 4);
8 x, X, h: Z, T$ a4 y/ A, c3 i
$data = substr($buffer, 14); J2 b2 ~2 `) {% B: H
} else {- V! `, T [5 |# `0 {/ l
$masks = substr($buffer, 2, 4);
. f. O$ Z. H9 T4 W
$data = substr($buffer, 6);
8 L) I$ t8 d7 e' R8 n9 H) g
}$ r0 U, X" \6 \1 E4 r6 K
for ($index = 0; $index < strlen($data); $index++) { w, T: V' n# M4 G! r; S6 E
$decoded .= $data[$index] ^ $masks[$index % 4];
6 X" H" I8 O- o" B0 n0 b! p9 v
}
* d. `3 \% B; O5 Q3 V% Y
return $decoded;* m7 J' [; B/ g C7 ]
}$ K2 }6 [1 d S8 A' q# p) Q! F9 o
% C# k2 ?9 @. B$ H: N5 [2 u5 n
// 返回帧信息处理
. O: A' @1 c# G) r5 N3 ^9 B
function frame($s) / Z, ]8 N. C% u+ U" E
{/ `. M9 C0 p/ L6 ?, G/ ^ F
$a = str_split($s, 125);
9 _9 `3 @# i: g8 F% G6 W
if (count($a) == 1) {2 Y* t# g2 Q1 ^3 w& W0 Q
return "\x81" . chr(strlen($a[0])) . $a[0];
* B" S# o$ f; L' D7 W0 T! R# J6 u0 ^
}- F' E, m1 D, y" k8 _. p
$ns = "";
& _% [5 T ~4 m2 g6 a6 B. R
foreach ($a as $o) {0 x( _2 \1 g! J% \& i* U
$ns .= "\x81" . chr(strlen($o)) . $o;
6 {9 r' Y) S3 R0 U
}1 o' ]; R$ R( n9 D0 {+ z3 Q9 y
return $ns;
1 l$ \! t7 O1 J2 w7 F% U
}5 Q8 w6 P0 z r+ D: E
8 z* _) X* x1 Q2 x; ~6 g: x
// 返回数据8 L/ U: L# @& B& d) v2 P$ u8 `
function send($client, $msg)
2 j3 \# q! o2 H* R
{! Q% a7 }! k9 Q0 Y9 g
$msg = $this->frame($msg);/ ]+ ~3 W" U. p% t* y
socket_write($client, $msg, strlen($msg));
/ j1 G4 k7 F2 P* h* N
}3 m+ R# z* t2 H; m5 Y* F. r
}
$ K' u) e' I$ J0 n8 ~& v1 z8 e
" H3 J: X) c7 e
测试 $ws = new WS("127.0.0.1",2000);
( f- N8 ]0 Z6 ?" z
/ V w* T4 C' l2 X' ]" F

复制代码

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

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