PHP 简单实现webSocket

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

1）客户端实现

<html>
# @; p) ~3 j* O8 ?& m: k
<head>
( @& a6 @1 p: E$ ?, j3 b
<meta charset="UTF-8">7 _9 F2 Q9 y6 t B+ Z
<title>Web sockets test</title>0 c+ ]1 Z2 S* a& |5 C
<script src="jquery-min.js" type="text/javascript"></script>) u' A/ r8 k- k/ J. [* `
<script type="text/javascript">
6 K6 N4 N( F. x6 O) T( g
var ws;* b9 V3 r. f* Y' ?0 m
function ToggleConnectionClicked() { 2 d0 V) Y' {0 S. U5 X
try {8 L9 f. E, U7 w
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器 " |/ M4 Y0 C" Y. w/ Y
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};" z* ^( b7 i# t! p4 b! L
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);};
8 j* J3 m, B! L% f2 {% I7 }
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);};
5 j) L- n- {6 |: c. L7 `
ws.onerror = function(event){alert("WebSocket异常！");};
) H% p' s4 P, m2 x
} catch (ex) {
4 F' E6 t6 g8 @% E9 L
alert(ex.message); 6 ?% K' |' p9 C& |: A& f; ~
}% A: t# N5 H; ^
};
- j; v# O3 T1 P+ R" d+ Q
# n, L! p2 Y& T( _+ {9 }) H6 t9 p9 t) f
function SendData() {2 N+ ]7 p/ N- p( g
try{
E# V/ Y9 V# h0 c, u z% A
var content = document.getElementById("content").value;
, y5 U) b4 e8 W( A* D
if(content){9 {" M- r1 t8 |8 P! e+ _/ O
ws.send(content);
( p, o& v# c: Y3 B, @8 S* I
}) W$ z9 |( u4 G: H* _. D$ X! Q2 c
2 }3 X0 d8 [4 \! P% O
}catch(ex){8 G/ b2 J5 N/ e- K
alert(ex.message);
" m: n, D8 @" o1 q/ ~* H0 w
}& L ]) [! Y. q3 f5 Z; G( N9 z
};7 [3 b- A( c: F; ~/ Y
1 f0 y# O$ \; @) e% g1 p: D
function seestate(){% F& o4 [' O' R) A0 A- j& Q) J& o3 D
alert(ws.readyState);5 e1 g' |0 b1 M0 Z4 q1 M! o
}
8 A" d9 B, P$ y# ^3 T& F
4 H& f6 V9 j! z4 ^2 s
</script>
/ O, `4 D) C8 D2 c9 L. z
</head>
+ F/ ], z n+ L/ A$ [2 v5 W* S' U
<body>8 o% t7 y( k: u1 @& J3 Q7 a8 |
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br />3 ~* s% P8 T; N/ ?: @* H
<textarea id="content" ></textarea>+ e$ R( \5 V1 H, ^# `, ~
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br />
/ K8 T. e; G( K8 e4 Y- ^
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br />
- H2 z. N7 ^, l) N3 a/ v5 V
8 I3 S3 N* H3 x! \ u1 K1 P
</body>
2 u6 [2 A2 M" J l/ f. q) k& x& ~
</html>. L- j" q+ T5 L' J# ?4 H6 N

复制代码

2）服务器端实现

class WS {
# w* s) [, a/ l& ~
var $master; // 连接 server 的 client
' E5 F$ k- `& z& Z% m' i# H
var $sockets = array(); // 不同状态的 socket 管理
# k0 D% z. D1 ~# }- q( d, L
var $handshake = false; // 判断是否握手0 P& L8 a4 D& W) d. h
8 |5 X) G1 B3 C- n, }' ]
function __construct($address, $port){) N) |2 h3 B" Y: q# A T
// 建立一个 socket 套接字
* s5 A! R0 Y8 q/ o$ B
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP)
1 ]4 @( F# ^& V( J
or die("socket_create() failed");
6 A6 Y. f% T) d. |- O! ?8 U
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1) 6 H+ @5 d; B- ]" W2 R8 }" \
or die("socket_option() failed");1 B9 V1 I1 s8 `) T
socket_bind($this->master, $address, $port) C9 f1 g9 E! h, i9 D5 ^: q
or die("socket_bind() failed");* X( Y5 g2 i$ k6 s3 r+ x Q7 e
socket_listen($this->master, 2)
7 `$ ^# E* h( M0 K" _$ H f
or die("socket_listen() failed");6 _& f- a6 j) p" a
B) p- L2 b; M3 i6 h, f% z
$this->sockets[] = $this->master;
$ C. J' q b) ?, N
$ T, X5 o) `+ f+ \
// debug% d p# U; t! E3 x
echo("Master socket : ".$this->master."\n");
: u1 s3 M: s% i+ R+ i* s
" C+ I4 Y+ T5 X5 ~9 J( R+ z
while(true) {$ n I. k: m/ H. Q% _
//自动选择来消息的 socket 如果是握手自动选择主机
1 N9 \: u5 W, P9 O+ Y0 H r
$write = NULL;
- b/ w0 D- w; d# \
$except = NULL;& g9 H" |8 q2 x! y/ I! E
socket_select($this->sockets, $write, $except, NULL);+ K8 X1 l3 v% G v0 S
% j3 T% Y5 q' q, C( P
foreach ($this->sockets as $socket) {0 [3 Z5 t+ T! r; e, k* K' _
//连接主机的 client 7 \, C9 y+ L- K* g$ p/ E# l5 l' E
if ($socket == $this->master){7 J2 x% ~( o3 H% K G1 ]
$client = socket_accept($this->master);8 Q* a. x! e- ?# W! N& b
if ($client < 0) {
, i& Q1 g3 h' B: T6 J& A) Y1 {
// debug
* S! U; }* e) e M' R6 Q
echo "socket_accept() failed";/ S0 ~- C3 o. |1 ^. ~
continue;& E) \) m: _- r6 o r+ R
} else {: @6 A$ ~% A' a1 f8 V6 F6 j Z$ }
//connect($client);0 H) o3 [# \* p; z. q `1 \$ a
array_push($this->sockets, $client);
& e+ p. W1 b) F$ c# t
echo "connect client\n";! v- j+ ?1 \6 I8 n
}
& z" Z& u) s. D( {
} else {
% g/ _0 H. F" I. O
$bytes = @socket_recv($socket,$buffer,2048,0);
' V G! e/ e" P5 _1 X3 a Y' F
print_r($buffer);
* k$ [, U* Y4 W4 V! q# s4 y
if($bytes == 0) return;
+ a* z' S; w r# T1 \8 O q
if (!$this->handshake) { e. A; J% @) B; J, y2 @# i a
// 如果没有握手，先握手回应. b3 ^4 w0 l/ r5 o7 e: ^* k
$this->doHandShake($socket, $buffer);
3 J/ H, ?9 Q; u% k1 |, [( M
echo "shakeHands\n";
. c7 e- I% B1 j7 K2 t( C
} else {
' z1 N/ v% \1 o; R+ a; b
2 s* _9 W0 `& P, A3 M
// 如果已经握手，直接接受数据，并处理% L. t, g: R7 Y A1 m6 A; t9 i
$buffer = $this->decode($buffer);5 x. b: q% G/ L" g, x
//process($socket, $buffer); , F, k5 ]. G$ x# ?& }$ y% w
echo "send file\n";) {: E# L" b* r( P% B
}0 g1 K! x" ~* ]' _, a
}
0 P9 r- r" Y1 c! s& W
}" l j# j+ D; M2 C4 K
}
( r3 u8 A$ A1 n
}
5 \% J8 y3 a( C+ s6 g3 n. t& J
2 @# V; h3 l9 O4 r {: x
function dohandshake($socket, $req)
' Q3 u( |, ^( b2 ~
{' q; X9 w% E$ o$ {0 L; N
// 获取加密key
6 D. a- i2 N0 l, x6 j
$acceptKey = $this->encry($req);: N' _ P) k; S
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" .# s7 s$ R& w3 A5 ~% V" G
"Upgrade: websocket\r\n" .! h3 S4 y) |. D4 |: n4 X! [
"Connection: Upgrade\r\n" .* L5 j# u% {* T$ ~5 @
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" ." R" X% j5 Y. y7 a, ?5 v5 A
"\r\n";( N# o' d6 p. p
+ U! a5 ^; K+ C4 [+ C- Q
echo "dohandshake ".$upgrade.chr(0); 0 w6 M s' Y1 o. }* R
// 写入socket
% `; K+ `7 t4 m& r* K6 h
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0)));9 N5 y8 E8 B. ~# U+ S
// 标记握手已经成功，下次接受数据采用数据帧格式
/ G& o1 b) i2 _3 j* W& w( v
$this->handshake = true;$ [6 f8 R, @/ R+ e/ h
}4 ]" Q6 f; P$ Q, L% e* Q: w
+ ^& p8 D# @ e$ m2 V$ ]: G
0 B- o4 g/ J9 S
function encry($req)
* X# I* j; E6 M6 l
{- C! a b- N* h. ?" n" S7 i
$key = $this->getKey($req);. C5 F. i* H% ~* n/ l9 Z0 X
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";# k. [# k6 t/ e1 i# A
+ |6 G% i, i9 r% P$ m
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));
: r5 f, W5 j" }8 J
}
- W! h5 X6 ]- E
; j' S9 E0 |+ { @& t
function getKey($req)
+ t( Y2 Z0 T, P' R+ v
{/ y. @2 x& S' g |" A
$key = null;& O- ]# ^+ I y; A( b$ A
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) { 0 `5 o- V; J) e4 r h
$key = $match[1];
) C- y3 B, l. k9 t; s
}
$ ^7 {: ?$ r* S5 G5 _
return $key;( n: a# J$ s5 y2 ]8 c( U
}
' z6 ]7 b& z' q+ `$ ?
! [. ~- X2 _8 k( A, Z
// 解析数据帧
7 G7 p* f& ?: Z! B
function decode($buffer) - x0 {: |9 W" l1 x! ]9 W
{8 p4 g: r- Q" W7 D
$len = $masks = $data = $decoded = null;# ?) F# t% ?' r# A* ^$ c
$len = ord($buffer[1]) & 127;9 h$ C! r3 ]+ t) i' l) u
. u1 C% G8 i1 E- {- T
if ($len === 126) {
$ z! ~+ [) ?9 [+ U2 G h+ M2 f
$masks = substr($buffer, 4, 4);
/ B) I2 [' `6 r, T# a$ |4 `
$data = substr($buffer, 8);
/ V/ {' n( J L7 u9 J6 v
} else if ($len === 127) {8 Z& d. l+ m6 y* [ T
$masks = substr($buffer, 10, 4);
" B. R# w3 ]; x$ U( n4 c$ L
$data = substr($buffer, 14);
3 O% t* Q L9 {+ ^2 { J
} else {1 x( ?* `8 o" v1 D, H6 i
$masks = substr($buffer, 2, 4);
( E& `! z# g3 `0 k& X" f, S
$data = substr($buffer, 6);
/ I, F; K2 D# z9 a( ^) ?! Y
}. Z7 X1 u; i' k6 b4 n4 _- f
for ($index = 0; $index < strlen($data); $index++) {) C! s8 R: w( R! u- I7 ~! X
$decoded .= $data[$index] ^ $masks[$index % 4];
: M5 j' U* J' R+ C
}
7 _9 J0 V c& k3 @
return $decoded;
: w8 H$ P0 M) x8 A
}9 r3 K% g$ n4 k& s
c1 S5 O/ |2 {9 K4 A& F5 E/ u
// 返回帧信息处理, N1 l! m* i$ \" Y; R
function frame($s) 7 }' v" t* j. h
{
0 L; D1 ^# ~8 |" A; W
$a = str_split($s, 125);/ l% j2 y1 g1 D1 k1 X7 f
if (count($a) == 1) {# L2 X% n% p. a$ N c+ @
return "\x81" . chr(strlen($a[0])) . $a[0];
2 O; Q7 ^& T4 R& @+ y
}
- _2 v8 v% J5 B6 g
$ns = "";* R! X2 c* h' a$ B
foreach ($a as $o) {! R; W! I$ A; \- \" Y
$ns .= "\x81" . chr(strlen($o)) . $o;7 [- T; j- v z+ J8 J
}
8 E& _6 O) ?' v0 b! \
return $ns;
0 u# J! V6 }3 k7 {; v% w
}
3 `$ ^: W' F& t7 X+ v" y" ]3 @
6 x' G2 e$ I a8 P, n+ t! o9 [+ x
// 返回数据. D q0 `3 [5 H( k/ J6 ~3 W* I
function send($client, $msg)
g. q8 X2 g- R% W4 Q; m
{2 `$ d4 q( c f- J4 K; T8 D9 T
$msg = $this->frame($msg);
# T8 p8 K' t" K) n/ \
socket_write($client, $msg, strlen($msg));% J( j6 s; B/ c- z- H
}; v3 {! G, Y7 I3 b* ]/ n; e! s5 Z8 V
}
7 R+ i5 e- a4 \" G4 E4 m) z
1 E) L3 W. C* X# j' E1 D
测试 $ws = new WS("127.0.0.1",2000);% Y7 U9 J! \2 \" y$ H
/ Y1 q8 S5 r# {! j* k7 m. a

复制代码

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

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

浏览过的版块