PHP 简单实现webSocket

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

1）客户端实现

<html>
" |( @6 W6 F0 z
<head> t" c' O! A4 n1 ~! j& t2 M
<meta charset="UTF-8">
& C5 D/ \( K$ w7 o4 f4 }3 P( v% m
<title>Web sockets test</title>
7 ^# ?; J. S+ G9 [7 F- t3 n
<script src="jquery-min.js" type="text/javascript"></script>
5 v8 C! P9 e3 S" j5 X% [( T
<script type="text/javascript">" Q# I i, v' s ~1 u b+ v
var ws;# [' e4 m# L. \
function ToggleConnectionClicked() { : a) o/ _5 l1 ~: `1 D; q
try {0 q6 ?) E9 x' U& \ q& u
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器
3 h- c" m& @* c( ~
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};9 `5 u' Y2 d) Z$ p5 w0 j+ p
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);};' I% C% _) H/ ~; S, \5 r: A
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);};+ {5 |2 H/ d$ d
ws.onerror = function(event){alert("WebSocket异常！");};
' W& q/ n5 O8 s' J4 Q8 R
} catch (ex) {% u: _2 c3 L5 I) H/ k
alert(ex.message);
$ N% P1 T# U. p4 J0 c
}
- k- ^1 m. }6 p f% y) ` f
};8 g9 L8 R9 R6 Y' C+ M$ R1 c$ T7 s
0 R' h# J G% G& @* W+ `
function SendData() {
5 j% y9 ]8 t& U- e3 x/ \4 B
try{
. l( P0 |0 D; X- X4 y# \
var content = document.getElementById("content").value;
$ {! o( i$ Q! o# e5 c
if(content){
& ?4 z3 y( W7 R% e4 b
ws.send(content);
* o3 J7 u0 V; `5 ~4 \
}
% r* o3 g/ u; P$ ]4 p
) y7 {6 e( e H( j5 a9 Y
}catch(ex){; A: s& w. \7 J r+ {2 {( C
alert(ex.message);/ ^ D T# f% @: n1 H1 h
}
* I8 f) P% R& Q$ ^% Y; J
};
( k0 M/ m3 }' z3 e! n3 Z6 c) B
: C5 U7 @1 c. m& I. g& O7 l
function seestate(){2 T1 g5 k2 r$ _; ^, V) i9 v! @1 }+ G
alert(ws.readyState);
" h! S+ ], c. F! t- n
}8 _+ O, h u* [( I# l% T
! M% l( K4 \1 L2 ~' @ S- S" n
</script>
* X/ ~* v9 W) L
</head>% V# R; w" o+ T2 ^- p
<body>
, m: p8 r: V+ O& l6 ~
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br />
, g* k% s5 Q t( H; g. l2 @
<textarea id="content" ></textarea>) c/ ~ M! k1 ]. F/ n& ~$ m# w
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br />" z( C2 D( A/ @/ [ t: E
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br /> J9 w e& S: S1 u" Q
4 Y/ s# p M+ L6 y1 b
</body>
( l- Z8 I3 O$ l$ a; A* m% o! u
</html>
- u# X3 b) L2 a. k7 s+ V! l/ h

复制代码

2）服务器端实现

class WS {
9 Q4 ?; a( H2 ]$ H8 S
var $master; // 连接 server 的 client6 k: ]. m' A6 O
var $sockets = array(); // 不同状态的 socket 管理- u6 F* V0 M6 q o' S$ Q
var $handshake = false; // 判断是否握手2 S' r5 s0 b, V& n0 Z5 ^
2 Q" L; c. q4 y; {% P7 V, O
function __construct($address, $port){
' q* Y6 j+ E4 V, q" g ?
// 建立一个 socket 套接字. N) s$ ^# _1 \5 B& X& f
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP)
8 ^- K: L' P$ e& C
or die("socket_create() failed");
# @- [( Z6 Q' x2 c+ x- c- }
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1)
) g! K1 W1 N& u9 l: A
or die("socket_option() failed");
6 k, H2 j0 p: d2 w+ k S
socket_bind($this->master, $address, $port) 0 V, o5 p: }3 @, V
or die("socket_bind() failed");6 l8 ]8 Z s- a4 D
socket_listen($this->master, 2)
. ^8 d& _+ [; N; p. [' n0 L0 d: N
or die("socket_listen() failed");! r4 w( Z9 s0 u/ Z9 r
) l& F7 I" F" W; }9 O$ b
$this->sockets[] = $this->master;
1 F4 J; R5 d4 Z7 B
% Z) F3 d# G) U
// debug
' }' m' C9 ?: r& k; p/ J- O- [
echo("Master socket : ".$this->master."\n");0 h9 Y1 L7 R# w
# m6 p2 W5 n2 ~% B% z8 U1 B
while(true) {& R) U7 A8 z3 L0 u* Q
//自动选择来消息的 socket 如果是握手自动选择主机. h1 o+ K; u6 N9 d* K! S1 _, u
$write = NULL;
6 G, v' O$ a8 y7 u% N6 T
$except = NULL;
; f. V+ P1 {9 \2 ?& u c/ W! |8 d
socket_select($this->sockets, $write, $except, NULL);
7 G/ Z; @1 Q3 J6 M
+ E( G4 j7 L, N+ g! A Y3 v, [, M
foreach ($this->sockets as $socket) {
) N& w' Q& E! z5 t& P* m
//连接主机的 client # M# a$ T& m9 ]4 ` `
if ($socket == $this->master){) Q7 F8 a2 M4 ?# K
$client = socket_accept($this->master);
. H8 @1 r- k. P5 {& h) t/ i/ l
if ($client < 0) {, s3 {3 a0 a- C
// debug2 V7 s- a: D6 N& J! Z& T
echo "socket_accept() failed";
* `4 ~# u- q, g
continue;3 J* h. P0 Q' P/ n% m5 K; J
} else {
6 e" _5 g1 n# ]( H$ l. q8 L/ W
//connect($client);
' D0 W3 {9 B! C
array_push($this->sockets, $client);
! W6 f' \1 O/ x- ?& ~. ~3 L; d
echo "connect client\n";6 J1 u- C* N0 q0 F. v( g
}& P: ?0 W& w1 Z
} else {
2 `. f1 m% W- r/ |( S+ e) |9 x
$bytes = @socket_recv($socket,$buffer,2048,0);+ U0 O# g/ ?. E
print_r($buffer);
c& B% t' n- Q( a
if($bytes == 0) return;
# o# P% P( F: W( l2 z6 Q
if (!$this->handshake) {
& ?% t2 h, P8 _: X/ X1 B) T
// 如果没有握手，先握手回应* \* r9 v% B% A0 ?
$this->doHandShake($socket, $buffer);/ m/ B1 E7 r. D" ]3 |. `
echo "shakeHands\n";3 U/ ?2 f6 V( U0 _( U
} else {* h# |: \& Y4 p9 F5 [! q$ n
, Y( z, ]7 s* s
// 如果已经握手，直接接受数据，并处理- g* W) C( ]2 N( P# c7 O
$buffer = $this->decode($buffer);
& {( M, X5 }' r3 q6 W5 C
//process($socket, $buffer);
+ t. U" z0 S% c9 R
echo "send file\n";
6 V# G1 |. Z! X5 g4 O
}
) I, Q, y8 d+ ~0 P: ?) [+ Y
}
' s) N' i3 N' T% e
}
# [9 ]) B0 n+ ]+ F
}
, ~6 u' I1 D; q1 ]/ U' M9 \$ q4 N
}
2 Q& P9 D2 R5 c4 F' N
K8 y' g% | i4 w# C
function dohandshake($socket, $req)* K( H$ R |7 N" @
{
2 ?0 e6 K/ W& ~$ n8 n) [
// 获取加密key6 L4 F C. N `, g' {) i9 |+ M
$acceptKey = $this->encry($req);$ O9 n% q& w. l
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" .
! E! g: O# |- O* o- b5 |% X
"Upgrade: websocket\r\n" .
* ^9 p. \6 o/ M% W
"Connection: Upgrade\r\n" .% h$ Q+ ^) \$ m; m1 r0 h( {" a
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" .5 q4 V- r o% [4 U
"\r\n";
5 w7 U3 [. @! s2 o6 k+ @
$ Z0 @3 C" @7 Z
echo "dohandshake ".$upgrade.chr(0); " D% U5 j" ^+ Q9 v1 F" r
// 写入socket# k; d4 e/ F( x
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0)));- x, K# |! }5 \+ t, p" Z# O
// 标记握手已经成功，下次接受数据采用数据帧格式
; d/ m# O& j1 {6 s* a% a% [5 I
$this->handshake = true;/ o& _' q7 a2 Q* V& K! ^
}; d& a0 S* ]# }( w0 e
: y! Y! Z- H1 o
- X$ @# r/ h* ~& e4 \; k
function encry($req)
" M! l4 [ I8 S! E
{( o* Z) R! A! ^- \& e
$key = $this->getKey($req); G5 | K/ g( P$ a
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";
7 {1 t6 l5 W9 s1 ]8 J/ Y5 p S3 T- \
" {# P! A+ c* d+ b# D) [
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));8 e% }( y$ A& Q4 y/ I
}# E! }! o5 u5 j8 F c- {
+ R" H! l. E4 r- T1 S* }$ W4 W
function getKey($req) 9 ^0 e6 ] [5 K) W
{
; z" t* N, I- y8 R# u& d
$key = null;& ]4 Y; n1 |' ]- W2 H8 Q& I+ X
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) { 0 y' N, J2 h/ l T1 J
$key = $match[1];
/ T& _9 U9 p/ n i
}
# E! `5 e- s( W4 D( j4 y
return $key;
/ h2 x0 B- {" Q( {5 Y
}
) D- V6 E- Y# H+ T; H
# N1 ]/ ?2 h- i4 Y6 M# J
// 解析数据帧
( Q6 S. w q0 r. f& v. @7 G- j
function decode($buffer)
0 E& y% K* ~* l, `4 s' v* X
{/ S2 h( r, |2 C) C( @
$len = $masks = $data = $decoded = null;
/ _8 X7 v! Y( r8 C9 Y
$len = ord($buffer[1]) & 127;! x9 R h w2 @: p& H8 q! C( m2 @
e N. U! h: i. t$ ~3 b3 l( h
if ($len === 126) {; R: `* X1 N* z: ]; p) t9 B
$masks = substr($buffer, 4, 4);
7 E4 R0 c4 I7 {9 c, X3 {) l
$data = substr($buffer, 8);
) [" [, {; g" q+ I# b
} else if ($len === 127) {
9 K, X& k4 Q% e; d5 n
$masks = substr($buffer, 10, 4);
& K _6 e# j6 B7 h& L U w5 \/ |
$data = substr($buffer, 14);
- ^. N- G$ X! _ K& D
} else {
3 H- z+ g% V- x( h Q% \
$masks = substr($buffer, 2, 4);
6 ~- Z6 c9 w% D) ], {
$data = substr($buffer, 6);! S! N3 v' v6 S( r; u7 A. m
}
! ?# ^( j9 L1 k8 o! U& a/ Q- T
for ($index = 0; $index < strlen($data); $index++) {
/ x9 p/ H7 d; H' f
$decoded .= $data[$index] ^ $masks[$index % 4];) o) V5 f: L* E/ k3 }/ M2 y
}
" c5 s; K( }+ E+ a+ w7 V
return $decoded;
2 ?; ^" [) f# R6 o8 q( S7 n
}6 a( d9 f! Z: o/ g
; U0 D R4 ^ l
// 返回帧信息处理
1 F0 r( ^! q# F9 H# D6 _
function frame($s) * [$ Q( c- k/ p
{8 R, k8 v& X0 B. @
$a = str_split($s, 125);
2 m) d9 \7 L3 ?. e( }6 D Q
if (count($a) == 1) {
; p; G8 _* T! H$ u1 k! M
return "\x81" . chr(strlen($a[0])) . $a[0];
) ?7 Z1 c, v8 z" b$ [/ Q
}
; y* C9 d4 G+ ]1 b, a1 }, F
$ns = "";8 m4 G( u/ P0 c* \) q: I' }* C+ I
foreach ($a as $o) {
; Z* [- U/ k- l$ G I
$ns .= "\x81" . chr(strlen($o)) . $o;
/ _0 Z& z6 A1 ]3 B) p, u1 r
}
( F% F: Q% e' }
return $ns;
- K: t! ^7 }3 U5 C" B( B8 O
}
( ~5 ?- d- k' u V0 v3 q+ W+ ]
5 b9 B, u8 ]; m( Z
// 返回数据
~3 P v7 a: D: X+ B
function send($client, $msg)
3 y3 z8 j6 T. o, E( r& e- K
{% M& V# y$ i, \" A' u9 K
$msg = $this->frame($msg);
5 b7 h: k. R' H9 l8 f$ v0 r/ Y4 w" _
socket_write($client, $msg, strlen($msg));8 b/ o. ~! _, m- Y- H
}8 {: |" `4 J* H
}; M1 j: n7 c6 R+ @8 V$ t0 R/ V3 I% ~# X
# H3 R; s+ z1 G" \; I
测试 $ws = new WS("127.0.0.1",2000);
: u; @& B3 z* \- o) J' b* [$ ]
4 m" _$ {% W; b6 c

复制代码

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

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