PHP 简单实现webSocket

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

1）客户端实现

<html>: ^% S" G% z# D; n8 l
<head>' a9 G x( R" M; C
<meta charset="UTF-8">. {6 M2 V) ]& N. l. f2 L- q
<title>Web sockets test</title>; A8 ]* K+ ~; F Z* S$ M) ?
<script src="jquery-min.js" type="text/javascript"></script>
! N. b9 O7 {( `- H8 a
<script type="text/javascript"># f6 l1 ]0 R" O* n
var ws;" u1 b2 Z- F' d% y- S2 J. r6 r
function ToggleConnectionClicked() {
/ o3 j& i- l Y! t& ~
try {3 ^6 z9 _6 D8 w0 g" ^3 x: Z7 _3 w
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器 6 G) f: Y- g+ u9 T# S9 b( N9 M
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};
2 ?: Q4 z' B n! t2 g
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);};5 c$ }; z% ]3 u& \* l
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);};
1 E( e/ p( }+ a4 l! }/ ~
ws.onerror = function(event){alert("WebSocket异常！");};$ V* b0 ~) C) U+ u6 E. q# ^
} catch (ex) {
3 ~' T C2 A6 j( H. h
alert(ex.message);
. G% L8 O0 y! w
}
& P5 J3 K7 c( d
};
9 J. Q1 F' Y# X. t" ~
" Z, p H% a5 V; F* w6 J
function SendData() {, _$ D$ \$ [- n4 P. v
try{
* w% R8 \3 ?2 s( z" l" m* Z$ l
var content = document.getElementById("content").value;
1 x6 a' n" s: ?) `0 E; y2 U
if(content){0 e9 u# M8 B% J- t! U) u. b/ Z: b
ws.send(content);( x, B) a5 L% g! F/ g
}
3 N9 |/ R4 f; h! ~
" B, J4 l( G2 k" _( k# M
}catch(ex){- b4 p* r0 k X9 P. A* y, j% f
alert(ex.message);
0 f3 l9 x$ U9 A& V. G r
}" c8 X+ B% }# w" g' N; u$ S. x4 ?. E/ o8 [
};( a# w0 b. _( P1 m. N
8 I, u2 H" s. z
function seestate(){
" w4 v# c3 x$ ?1 {# V
alert(ws.readyState);% P! {7 I' J, p. l- f' w/ A! N
}' H, w" U; N7 A& s& L4 H6 E8 w
4 R$ |0 {8 T9 O, a/ p1 T6 R1 e
</script>3 Z7 m' F6 I7 ]2 B* I- R
</head>3 e# s5 l* v- h% i* i2 M
<body>
2 ]1 D: u. O" [
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br />
' R9 I0 |1 F: k5 p3 U
<textarea id="content" ></textarea>, ?. x- m9 L, x$ b, ?" A% o: s
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br />& W/ A) m2 C+ X# A
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br />
i$ r: a4 P: B; }+ {3 r
5 j/ N7 C0 X% Q- N6 p, d
</body>
7 y# H! O& k6 {
</html>6 u9 p6 P$ F5 W4 V9 x

复制代码

2）服务器端实现

class WS {
# D+ `2 B4 R6 U" ~8 l9 C: M
var $master; // 连接 server 的 client% ~9 ?: `: ?3 |! K. K' W
var $sockets = array(); // 不同状态的 socket 管理8 {5 ^6 p8 S6 K1 A6 r
var $handshake = false; // 判断是否握手; g3 G' g( D6 C
* t9 z8 B k6 y6 G R, a. a" {
function __construct($address, $port){' b; |2 K" e7 \7 C# F, a
// 建立一个 socket 套接字/ G8 `* h; |; X( r2 k( R7 l5 k+ Y
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP) ( [5 [1 J6 N0 T' y5 x
or die("socket_create() failed");
9 ]0 J7 z. E( t3 [' C
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1)
* y1 k G7 q* m' \4 B4 H
or die("socket_option() failed");, J2 Z) H" f- |4 l# T
socket_bind($this->master, $address, $port) $ s S8 h Y2 k: j0 e0 |; R
or die("socket_bind() failed");2 g5 a0 i& v3 [! G2 O. I) [
socket_listen($this->master, 2)
. [3 O" o6 v; [# b5 I- P6 r
or die("socket_listen() failed");
. T1 E# Y% K1 p- W$ D& Y) B
+ K: l: t$ g5 B7 R7 `0 e" o
$this->sockets[] = $this->master;- }% X6 _2 j8 W8 ]5 x
# b% F( Y! r2 \* B
// debug0 J6 L7 B# @( v+ ]9 N
echo("Master socket : ".$this->master."\n");
4 [! [$ @" x! `( g7 O$ ?. i
+ I: \) Q7 t5 Q& v
while(true) {3 B; n& k& q: `' p/ b
//自动选择来消息的 socket 如果是握手自动选择主机
3 S5 H7 _/ Z/ K7 w: J( k' Z
$write = NULL;
4 h$ B* K( L$ l N8 r) |+ Z
$except = NULL;
5 r" i" {2 c2 {0 h
socket_select($this->sockets, $write, $except, NULL);
% r1 ^! e! m# ]5 l3 O
[( v0 M3 o$ g" \0 d
foreach ($this->sockets as $socket) {. G) |: t! W" C4 T( T
//连接主机的 client
n1 D9 Y5 `. D' P
if ($socket == $this->master){: o: p% N r5 k% ?. \
$client = socket_accept($this->master);
( c! z: F/ i0 J7 R- K
if ($client < 0) {3 J. v/ c6 S0 v
// debug
( j2 F; P/ l% z9 R1 D. ]3 {" ^
echo "socket_accept() failed";
0 h/ i/ I" n7 x7 ?1 P$ T
continue;5 V5 ^0 v$ r8 [$ R# x4 }" j
} else {: h$ N( `: O% x# U! o
//connect($client);" }8 f/ q6 m5 S4 b
array_push($this->sockets, $client);
. k: O, H- P( i
echo "connect client\n";/ U6 l( |! W3 S; p# y* |! w5 A
}) y" p7 m4 Q3 B0 w7 } ^
} else {
9 R& k* v& r: |% I, ]5 J
$bytes = @socket_recv($socket,$buffer,2048,0);
% W/ \' U; w& R( {
print_r($buffer);
' t5 J( b @% |8 X8 i9 L5 y/ [
if($bytes == 0) return;
5 @3 f! O8 o6 ^' [/ [* U9 k
if (!$this->handshake) {8 }" y, V% q$ s* q7 i ]# g( n N
// 如果没有握手，先握手回应
, z/ t8 `9 o4 e
$this->doHandShake($socket, $buffer);; B) @3 J& m& f2 N5 [+ `* T! m
echo "shakeHands\n";
8 V0 I. S. O0 h& ~3 Z# Y
} else {
( e: E1 F# N( B) k5 F
& z0 E: Z( h9 `& u* Q# b# C
// 如果已经握手，直接接受数据，并处理
1 ]# r/ a. E2 U
$buffer = $this->decode($buffer);+ ~6 \. {( n5 z- J% M* c8 \3 v& e
//process($socket, $buffer); - W% x" E: \* [8 Y
echo "send file\n";& N' J2 M, q% S# ?& t/ H5 l
}* Q6 m; A' V3 J6 M
}
& R& l( u w/ w1 M
}$ s# L7 D) j$ s* {0 A* X
}! F& Y/ H* M% _# Z9 w5 w5 f
}
& e) s+ L: t' A$ L+ ~% b
$ ^* P( ]* x1 j! P) S0 o" n
function dohandshake($socket, $req)
: C8 [0 R, W* D7 l
{
& @. I& g, t: c( J# N9 O T& b
// 获取加密key
" v) n$ i" @# F3 U
$acceptKey = $this->encry($req);
5 x) H6 O. r% m
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" .
, e P2 }0 N. d
"Upgrade: websocket\r\n" .# @" o% B% o7 Q D6 y. r* f3 N: w
"Connection: Upgrade\r\n" .* K4 _6 F3 p! e4 k
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" .
5 h1 I, X; F" r! d- Z
"\r\n";
1 T5 h; r8 y1 a' L5 \5 ~
( z/ j2 p+ W* V- y @: m& J, A& ~
echo "dohandshake ".$upgrade.chr(0); 4 f& r q& A7 q
// 写入socket
* _) D8 D6 _+ }
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0)));& J( G# Q0 V/ n) o: d
// 标记握手已经成功，下次接受数据采用数据帧格式% N0 s- K7 i! c" ^) f; v
$this->handshake = true;
% }3 {: W# r" A/ T, B, [, c
}
/ i( w5 W+ T* f! G
; I; k) ?. c9 g: [2 X- ?' p2 g
& i% u6 A w* `# f! E
function encry($req)4 S6 u% R& ? I# E, o' b) N& \
{
7 \% y7 \* X7 @; U0 i$ |$ m
$key = $this->getKey($req);
0 o7 M# E! r+ b" O% F' Q" j
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";
( { b- z7 H0 g# G) K
2 q, N7 c, h# b/ R6 z) S5 r
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));
8 q* X/ O& P( s7 z2 t1 L0 {
}) Z: h5 v4 W; Z* Q$ h
9 d8 Z l+ o( I U
function getKey($req) " F4 u' h* e' [$ C- y9 V, K1 p# q
{. n- N3 [# e* X) E u, |9 p
$key = null;2 r* p- j% v! v9 J6 j
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) { 9 o2 L2 b( P# t) A
$key = $match[1]; , w2 F+ S* C! \' w; I
}+ v' o. s3 j0 p% ]
return $key;
2 e2 s3 S1 `. i, l; R
}7 O% i3 }3 H/ R3 E2 n7 W+ K# L
# F" q8 j+ U& y/ r9 D
// 解析数据帧
5 r: ]/ l P* r% |5 i
function decode($buffer)
! d- f- ]( U7 b8 ^7 o, o* v6 T
{6 |( c- C( O4 I5 v8 k
$len = $masks = $data = $decoded = null;# J6 U) r6 q; Q2 Q
$len = ord($buffer[1]) & 127;
* U+ t5 Z2 ?6 G# I8 x9 _. p
/ z; S( T L% v9 }
if ($len === 126) {
. H! H/ e) u4 o+ }6 ?( X6 q
$masks = substr($buffer, 4, 4);
8 W1 q, I6 P+ W! ~* t
$data = substr($buffer, 8);5 d8 m- V# I1 ^3 ^ Z, v& S6 W
} else if ($len === 127) {- U8 Q: M, w5 A' p1 ^
$masks = substr($buffer, 10, 4);
7 @. Q* V! u8 y: Z; `: ~. V, p
$data = substr($buffer, 14);
1 E$ W* q$ G: C6 d& H
} else {2 ~9 N& Z3 E+ `! o' g, }
$masks = substr($buffer, 2, 4);
7 X8 [4 w1 d. u5 E) [3 K5 q
$data = substr($buffer, 6);* u2 @' Z# s4 {6 S, L
}
9 t+ U2 w4 [ ~7 e
for ($index = 0; $index < strlen($data); $index++) {
% @4 T) N" F+ j5 V
$decoded .= $data[$index] ^ $masks[$index % 4];, ]3 b+ t9 B) b7 x `1 B' \
}
6 Z1 Y( Y6 _1 @1 m7 h9 ]; S
return $decoded;* n- C0 q0 n! ?0 k( r- j$ A O! J
}
4 d* I* w( a7 ]4 F. a1 v
# ]" G0 n( s4 D2 J8 q: {; N
// 返回帧信息处理
" W8 i, H/ E* y2 ~4 J6 y
function frame($s) 2 [7 a" x' M& Q5 x2 f
{- s( i+ ~( r( [5 n2 B! Z$ e
$a = str_split($s, 125);
5 V# B/ L6 g( I4 m
if (count($a) == 1) { I: Y4 o$ b# n3 u; t
return "\x81" . chr(strlen($a[0])) . $a[0];" j6 H5 n2 k6 S" p# V
}
2 I' N2 W ]: e' i& d
$ns = "";
8 `+ ]( R. [) J2 h8 }) ]* B$ m! }
foreach ($a as $o) {; u) ^* X4 p( s& `# M; Y
$ns .= "\x81" . chr(strlen($o)) . $o;) Z9 c5 T0 f- `
}
! _( r _5 }. `4 Q9 |
return $ns;* p) w% u0 f8 R6 K! i+ A4 ?0 z Y
}; Z2 i* B( k' ]6 G& H8 k, x$ B
# G" D7 K% }* c' ?0 R; q: A
// 返回数据
* t1 N) E& B8 z+ w: U6 |0 s
function send($client, $msg)
4 D5 W. Y' C. I$ q
{2 W6 ~$ D2 h& q( Z+ ^
$msg = $this->frame($msg);- y: V6 o1 p+ H
socket_write($client, $msg, strlen($msg));5 N; l. k+ \6 ~
}
( c8 n2 A$ k5 v& s: f2 U' g
}
. U1 ~( [& Z" [2 |% v
# S& f6 D- c% j9 c% e
测试 $ws = new WS("127.0.0.1",2000);
- A2 B: J8 _6 v r% [& g" N/ i
( k$ z6 F5 p, [; Q. j2 Z

复制代码

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

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

浏览过的版块