PHP 简单实现webSocket

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

1）客户端实现

<html>1 w' L* @1 m. H8 i% Z
<head>! L- Y* Q. x2 h# O. c( o, `
<meta charset="UTF-8">- ]; d; O) k4 T+ t" L+ c. g
<title>Web sockets test</title>
, |0 [5 ?7 X8 z
<script src="jquery-min.js" type="text/javascript"></script>
( A j; m6 J* Q }# \
<script type="text/javascript">6 E( o- ?! P# q" ]5 _3 h6 T3 _
var ws;
$ ~4 f' I( `; b( J3 ]* B+ p
function ToggleConnectionClicked() { : _& J4 w' `5 N1 @' M. i- \
try {( W% Y4 e/ g7 n2 r5 `! J5 M. p
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器
: ~5 q+ O# S' A" y" d# m
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};
7 F; E5 }2 @/ l9 V" v
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);};- l ~9 `8 w6 n. s- F! R( e
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);};
- @& R; i) Q" t' J0 E
ws.onerror = function(event){alert("WebSocket异常！");};4 H+ G8 z9 Z7 `. o! C3 A
} catch (ex) {( A+ c4 m$ t0 P. {9 {: J
alert(ex.message);
+ J( O! D$ t% M! H+ w
}
3 ]5 V$ K5 K/ j2 e
};
4 ^) l0 q4 s c a
& { l7 o% x" ?2 \ _
function SendData() {
9 D# E$ x# p5 [) v; ^* a. L
try{% Q% L( d9 k9 C+ d5 }$ ]
var content = document.getElementById("content").value;) C0 s6 ~* D' F+ Z# F7 l
if(content){
0 `2 F+ d" P m$ N8 n
ws.send(content);$ N3 l4 _' M4 \/ y; U. Z
}
- M% D8 m2 q f' y1 V
$ _! ]. @' ^1 L4 ^' ]1 V4 M
}catch(ex){
% s! M" ^5 p6 N8 y
alert(ex.message);2 K$ \4 z. X/ M, P1 a, a7 \. v
}
$ a6 I$ d# r4 P+ h0 L$ i) J8 I
};
* K; q$ L# ?, s% L: g
0 F" C( }* I" q- a$ T
function seestate(){
. H5 a- }& x1 K5 H. Y: ?( O
alert(ws.readyState);" B9 j+ f2 Y/ a
}9 Z; K4 z2 z5 U2 e% Q+ b
: r1 ~' s3 ~7 ]( n' [% |
</script>
0 F! ~3 @: P. R
</head>5 e5 ^6 l2 I6 P. m- O3 g; I
<body>
9 f( `- s4 k1 ~6 K3 Y5 ?
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br />3 p( r+ G! W5 |2 x8 o! H
<textarea id="content" ></textarea>) o# B9 x' N' f8 g2 }+ D, l. w
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br />( }% J3 ^* j# G" N% e
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br />
! G. ?9 Q9 j& {1 N O) |' X
2 F4 ?5 p5 E$ H. D$ C( V# M: v5 L
</body>
* S' |" _! n, U) W; H( p
</html>* b: B& r, {1 h! `! g0 a

复制代码

2）服务器端实现

class WS {. t" B, j, A' Q
var $master; // 连接 server 的 client# K3 K1 h( d O8 X
var $sockets = array(); // 不同状态的 socket 管理
O' J3 b, Y1 g
var $handshake = false; // 判断是否握手; Y/ U3 j6 U, U, s6 G0 M; C( T; o
8 R, r5 Z. m9 q3 D9 U7 K G( {2 ]
function __construct($address, $port){
( s: V+ K4 g& y% a( b
// 建立一个 socket 套接字0 x! T& i' ?& o
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP) % O7 ~, a7 D' f+ u0 F* V9 b
or die("socket_create() failed"); K0 A& a: q8 [4 Q1 o) w: n/ n
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1) H' D& B6 H) S* @& D( Q2 O. R
or die("socket_option() failed");9 l. K1 _. ^6 E! m. X) z( B
socket_bind($this->master, $address, $port) % ~3 Z @ _4 }9 T$ ^: U
or die("socket_bind() failed");
( a, P/ R2 r" S/ p. h: A& t
socket_listen($this->master, 2)
9 f. Z, A$ Y: E9 }
or die("socket_listen() failed");
5 ?# L- w @2 I: i$ Q$ Q6 o8 j" w
4 Q+ s" {* N, P" G! B6 e
$this->sockets[] = $this->master;
9 n6 a& ` C6 g5 z. `7 y) {5 o
7 F+ G* u& c( n; B; }% x! ]
// debug) F' A& N& B D5 {; b
echo("Master socket : ".$this->master."\n");1 o' s1 i- K# S& N: R. R& i
/ w# R! g/ y' D8 |- E. V+ G
while(true) {
0 l7 y: I2 I3 Q* Q; s, J( n; w+ m
//自动选择来消息的 socket 如果是握手自动选择主机
' b' h$ f6 f( a: Q: ~5 c
$write = NULL;) x6 w3 I; i j8 J, o" G
$except = NULL;5 w3 m' z; T6 G
socket_select($this->sockets, $write, $except, NULL);
5 j* T2 o2 R. K8 f4 |4 H
' G6 y' W) d& T: L: o' M
foreach ($this->sockets as $socket) {; R5 V9 r5 d3 Y5 [% m, `
//连接主机的 client
, E- o! S7 E9 r
if ($socket == $this->master){
P' Q+ f3 ~) a* J# S6 S
$client = socket_accept($this->master);
, e- @% N2 K) R5 q4 h) d
if ($client < 0) {
7 t5 R0 D& r7 X' K' O
// debug; Y" z+ a: R% X- p, o6 d/ b6 E* J+ N
echo "socket_accept() failed";
5 c% C# W7 k6 R. t9 V
continue;
/ S m- t- [, {4 E! ?; D
} else {/ y2 }, E: p1 S; g
//connect($client);
" R8 T$ ~5 Z5 u; }( ?
array_push($this->sockets, $client);
4 o" o0 A2 t: ?- q% q3 I- f
echo "connect client\n";# H' j$ g0 z; J E
}
, S( ~0 I3 b; E" l. e2 ~
} else {
% {% U0 r9 D( h/ t! x
$bytes = @socket_recv($socket,$buffer,2048,0);
1 b$ Q$ W3 I; @ ^
print_r($buffer);
* H1 G6 C& N' |" L% B4 |! a: m. r
if($bytes == 0) return;
, [* r' x5 Q, `$ c! d: f5 x2 ^: g
if (!$this->handshake) {, V: O2 d o% f0 s; z: U; O
// 如果没有握手，先握手回应
, {( g9 r9 U7 L) M1 y Z$ P
$this->doHandShake($socket, $buffer);
3 m! I& C# h v5 s r
echo "shakeHands\n";$ i& K# G p# W1 q6 p0 @0 b! g4 ?/ q
} else {9 t6 x+ ~! X, l2 [7 |
& [+ {! O! h% c8 l, f) ]) a; R
// 如果已经握手，直接接受数据，并处理) S) O( x I+ D+ p# w8 Z! u
$buffer = $this->decode($buffer);$ W0 F5 T; ^, ]( C p
//process($socket, $buffer);
+ `, f7 G& x; B& {4 A. q
echo "send file\n";6 Z3 e# T( q( Y$ ]4 K
}
K' L4 x# W F' K# Y) l
}
! S) Q2 x; Y5 r
}
% s" O, [. X/ u) R5 N
}0 h# s9 }" h0 \( W
}
; s6 g7 l' X. J
0 ?7 j. G. X+ E
function dohandshake($socket, $req)/ v0 H8 i( H0 F: ]" M* [
{
" e' Y# x m% Z" y+ y4 C
// 获取加密key
- g8 Z l, F/ Z4 ?
$acceptKey = $this->encry($req);# f4 @0 @* M3 \1 u. x# w
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" .: B* p w0 V+ ^- X# i5 V* |
"Upgrade: websocket\r\n" .1 ^) o8 n- E9 w$ |/ o6 e
"Connection: Upgrade\r\n" .4 j8 w3 \" k- q! |; s$ u! N
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" .
9 J8 D7 s" C2 q$ i- K6 K z1 M( D
"\r\n";6 O( S* E( q8 i% b% H5 u
! P6 {" H2 _# v, m; M( D4 Y' L) {: z
echo "dohandshake ".$upgrade.chr(0); ( D5 n! Z A% k0 S: F3 s; W! J
// 写入socket. M. y" \+ P# H$ {0 f# L6 ^
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0)));& A3 P2 i* t2 e* I
// 标记握手已经成功，下次接受数据采用数据帧格式: Q% }1 h0 p; R/ w
$this->handshake = true;+ I7 L$ {6 m; R
}% N# H) v/ P% S( s' ?
8 e$ A( `* r$ t# @
0 y7 s/ }9 p' ~/ _$ g5 `# p
function encry($req)
% E2 X7 X, k% j6 X4 D
{1 d2 s& O! Z5 s/ q. J
$key = $this->getKey($req);
2 R/ E- L; R! J
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";: @0 G, v& p9 T5 N6 Q/ V
8 W2 y3 {9 {. G. a7 }
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true)); e- U7 B9 C5 n8 N! w
}$ d( g# c Q: y# i& i/ Q5 a
# Q3 U V1 e! \, [# V: f3 \& C
function getKey($req)
+ N+ m$ C. w7 Y( u( x% I2 h+ i
{" [0 {; F- y# I% |# U6 ^% m7 Q
$key = null;4 I! R7 s/ q) x$ z
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) {
5 j& |! O5 S; n1 c' ^% X
$key = $match[1]; / H5 {& w; N* {8 y; ?4 M
}4 }. {' S% K2 L8 ?! g" t% {, {& X
return $key;$ \/ K$ N$ o. d% M4 H% F% O
}
% b3 H' K6 S2 W6 }2 H: G9 t
" J3 R# r' y- |. V1 a' [7 o
// 解析数据帧
5 v* z% A8 \# Y
function decode($buffer)
4 H; P+ B5 k* _9 ], B% k
{, a, _ x" i* R+ a, B! `0 j
$len = $masks = $data = $decoded = null;
2 C. h( Q5 N1 ]8 w3 p
$len = ord($buffer[1]) & 127;
5 @7 Y/ O2 g1 u" v3 }, }3 q* F
5 n1 w, V: ?( {7 v: L
if ($len === 126) {
: I! w5 h( ^* n a; o
$masks = substr($buffer, 4, 4);
$data = substr($buffer, 8);
9 P2 s8 G! M, F
} else if ($len === 127) {( M5 s8 t& ~, l; J9 X9 i
$masks = substr($buffer, 10, 4);% l" s) i" Q% R2 ~8 r/ _
$data = substr($buffer, 14);: Q+ r2 R/ }6 y0 R3 M& ?+ f
} else {
+ ~( r: `: o$ }0 H5 G
$masks = substr($buffer, 2, 4);
* E$ I2 B W3 |" k6 Q
$data = substr($buffer, 6);2 [ B* W5 X9 M9 C6 b
}
! t; J' l( ~; l7 z. {* `8 A
for ($index = 0; $index < strlen($data); $index++) {, P, A) y+ g- a0 d/ L
$decoded .= $data[$index] ^ $masks[$index % 4];
2 T2 o0 f1 x x4 S
}" p) V3 P7 I/ k% w9 ~- u
return $decoded;$ q4 y) @; X/ v4 f: F. V, F; A
}
; l D! Z- t4 O+ A( q8 I. H
/ }# F- [: Z9 k' Q" D
// 返回帧信息处理4 n5 W0 P, U7 F8 e. b6 |2 T
function frame($s)
4 s5 U2 }6 X% o+ |: S- N$ s' J
{0 }# }; z5 c* S! [7 J
$a = str_split($s, 125);2 z) H: p& ^$ u3 n5 N8 M* P
if (count($a) == 1) { ^; I) D) E2 F g7 H6 m, R
return "\x81" . chr(strlen($a[0])) . $a[0];
$ I Z6 s" N8 e
}& ^" t) U) J) |' N6 F0 H6 q0 A
$ns = "";
2 W) b: R0 d3 @1 a' W
foreach ($a as $o) {
# [! t; T% L: I
$ns .= "\x81" . chr(strlen($o)) . $o;5 j# |5 \; C6 m' h$ [% K
}4 i3 E+ U& u _
return $ns;: x) M3 a: ^' V/ X6 J" z
}
) H c! H3 t9 E) |& d0 { I5 Y
. R/ _, H1 e+ J! s6 {% M1 _) p1 R4 w
// 返回数据8 e x" P$ v! F9 j3 g0 v1 l0 S6 F
function send($client, $msg)
; J0 r8 j/ I X
{8 s4 e8 U! y; ?- s8 H5 t- u
$msg = $this->frame($msg);! ?9 x1 u# ]2 l7 ~+ l/ u3 j! g
socket_write($client, $msg, strlen($msg));& W- l' r9 E, ~" I
}
& R' J6 _0 r1 Y5 ]$ j# `
}/ _; n6 f9 t5 T* ]: e9 H
! k! G! x7 S0 f
测试 $ws = new WS("127.0.0.1",2000);0 U( F, r9 z4 h7 O) [( z' w. z8 G
( R/ ?* E. g) W* Q# X0 _

复制代码

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

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