PHP 简单实现webSocket

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

1）客户端实现

<html># y, r) Q# W$ f0 D: X9 \
<head>! U5 y5 T" t3 I' J" ^6 x
<meta charset="UTF-8">
& B) p& D# e- x/ V% a! S
<title>Web sockets test</title>
! F/ S: Z4 i# _) a! a9 O' i. J" @
<script src="jquery-min.js" type="text/javascript"></script>3 U9 o" |) F5 w9 P* ^5 R% ^/ r! U9 _9 r
<script type="text/javascript">5 D6 F1 _, z [- `0 j) ]! j. e+ z
var ws;
0 C+ d$ p6 E5 o# @
function ToggleConnectionClicked() { * W8 ]( W' n- P$ R8 m
try {5 `) L2 G A2 ?" p/ U" h8 J/ f t
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器 ' \' T. J3 p5 M: e3 R
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};7 N2 l5 L# S8 a6 K. y
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);};
& X! L' E# E& Z! O F' I% t
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);};+ m+ Q2 m9 U+ `1 b' x7 h8 Q% |' q
ws.onerror = function(event){alert("WebSocket异常！");};
5 [( A' P, N. V0 E$ ]
} catch (ex) {. o2 S# x+ g- r" c% I
alert(ex.message); 5 V+ l! G5 ?1 v5 W* f) @. ?
}
: x8 @: n/ I) n: z8 b2 }
};
9 w8 C, l4 B* G+ ?: r
+ v/ b" W% Y p
function SendData() {
8 S0 \1 z& H; C1 W5 E3 E
try{4 _! F" u# O2 Q) }. J2 l
var content = document.getElementById("content").value;
. x& d! o0 n y8 C+ [
if(content){
% L; M. m. i4 D, q2 w* |
ws.send(content);& }$ t4 s+ J0 K8 ?
}( t! U% ^: F& B4 V. n. H7 `8 C- j A
; e( q3 f# |% M. o1 e- {
}catch(ex){ K+ G" @) K1 N% Q+ x+ p
alert(ex.message);2 Q3 ^! i$ a. B- ^6 W* G: a
}
) A# y' f7 r' l
}; H1 x$ r6 Q n4 v
- q' ]: k& c1 |- `; C
function seestate(){* Q1 x, B3 d& B0 F
alert(ws.readyState);
8 G& g! Y1 S# V
}9 I% w: N* _# R
" {, M/ P7 @ [1 t% G4 G8 a
</script>2 r. v* }) o; g. }
</head>% F0 L! j9 n& |) K. s" w7 l
<body>3 Q+ }" T$ F% Z4 S9 w* c. L: D6 T, A
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br />
1 v- p4 k2 d+ [
<textarea id="content" ></textarea> ]& L. B+ R) Z
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br />8 J3 |4 _! c* @! x" ^, A
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br />
5 O( K$ j. N" s: q! W
2 R% N* f& R/ w5 J$ }, r
</body>
" _8 ^) Y% m" G: ^; P6 \
</html>) z5 Y! C5 f1 ~! c, v

复制代码

2）服务器端实现

class WS {
5 \, T8 }/ l6 ?) ]% R
var $master; // 连接 server 的 client
9 S2 }3 {; D& M
var $sockets = array(); // 不同状态的 socket 管理
; T8 q+ K8 N( u* i1 T3 ^
var $handshake = false; // 判断是否握手) }! \) {+ U& @; I2 k7 h8 T2 [2 S
9 \, s8 P. d2 d" V+ [- t
function __construct($address, $port){
" k% _8 Z( o& ^
// 建立一个 socket 套接字: i, K, G' W" o5 {* L( a
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP)
2 }* D e& n9 A& B! |. e; V$ b
or die("socket_create() failed");$ p" N. ^4 I* m2 `- W& Q8 g
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1) ) g# T2 e- Z) F7 O
or die("socket_option() failed");
) T7 u5 |( p+ s% h
socket_bind($this->master, $address, $port) / v0 i6 I) m ?
or die("socket_bind() failed");% {3 a% Y0 V, D5 F
socket_listen($this->master, 2) / G+ x$ r N5 w3 Y: x; S
or die("socket_listen() failed");$ ~8 X/ [3 C' d- m1 d
! B% t# c8 ?1 `. `" @5 a2 I
$this->sockets[] = $this->master;
1 L- j# h' Z8 Y) j+ |3 `
// debug
' R) I) `+ h, y t3 l9 q' k
echo("Master socket : ".$this->master."\n");+ H) e5 s' _* ^6 y& ^& @3 z0 h
7 ]5 W% E7 u/ s
while(true) {. w' ]9 \5 @7 z9 F2 A
//自动选择来消息的 socket 如果是握手自动选择主机: F9 ?6 w$ _; ?
$write = NULL;
/ F$ q+ L( j `; Q) Q) C
$except = NULL;
" W# H- I2 K! E- {
socket_select($this->sockets, $write, $except, NULL);
. w9 e; n: v; x' T1 p8 }
6 h8 w- x8 A4 s J( o. \. g/ w
foreach ($this->sockets as $socket) {
. j0 J' ~ {% ~- M
//连接主机的 client + a1 A0 H3 f% ^. B% K, T3 t
if ($socket == $this->master){, P! c8 Y; F1 N4 ^& x
$client = socket_accept($this->master);) X! s% B, T: Q$ o
if ($client < 0) {
+ N+ _' V. Y% ~9 B
// debug
+ \/ ~0 ~+ r' _- z' b
echo "socket_accept() failed";" Z6 c: i, G% G. g* P9 U
continue;
l) [5 }* {4 c) B9 i$ {- n4 k
} else {
, J3 z0 U2 I) _% H$ m- I
//connect($client);
0 e' W: |5 C. ^) A/ Y
array_push($this->sockets, $client);! j, c; t; M7 h. g% l4 M
echo "connect client\n";: U. c8 j* X5 Q$ q; S8 N6 s
}; Q) P* O. \, B; O, Z
} else {
# I9 k1 H* m* i; m8 r, S3 E
$bytes = @socket_recv($socket,$buffer,2048,0);9 w+ K# [" k( J7 G
print_r($buffer);, _$ j9 A8 m$ \% C
if($bytes == 0) return;
! m! B9 u& I# a
if (!$this->handshake) {/ f! U. w9 S8 Q* C6 V; w# |0 X
// 如果没有握手，先握手回应3 ?; _: W! k6 t) e& u2 o, H4 l
$this->doHandShake($socket, $buffer);
. }) s1 [; G) n5 u
echo "shakeHands\n";4 F5 ?* J, d- _% Y ?2 }. h8 y
} else {$ J* j g6 y$ R& D5 G
7 P& R. @/ f+ w/ y$ E: G$ O# m
// 如果已经握手，直接接受数据，并处理/ N2 ]0 e) M# [* }
$buffer = $this->decode($buffer);
! L i* G- |: _6 e) S: f
//process($socket, $buffer);
! X$ B+ U; J8 U8 M
echo "send file\n";1 m- x+ X; ?0 i7 z* d! }% L# K
}
8 {4 H( [8 x5 K. B% \/ g
}
9 @, L/ u3 @, P, `- M* B
}0 w, m$ R% W3 `' V" K
}2 X! @8 X- H8 i: ^/ \- g' W
}2 a9 \9 {1 c9 H+ R1 k$ M7 f
/ Y$ f% n. S, g& T
function dohandshake($socket, $req)
# e0 y, R6 [9 S
{
8 c. m) Q! @* _4 |5 |& O
// 获取加密key
# x9 q/ [; t( ~1 }4 h: {! I
$acceptKey = $this->encry($req);
/ n) O4 a5 ?1 ^! d' V
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" .
. h. D- l I" ~8 d, Y+ f
"Upgrade: websocket\r\n" .
0 D4 Z9 r8 S. J& `
"Connection: Upgrade\r\n" .
8 f7 A, V* k/ }: w) _2 F9 b( Y
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" .7 @* t* O; f& }9 }* F
"\r\n";
4 s+ O5 V8 U* ~0 z9 N
W! P( i' w" m4 T
echo "dohandshake ".$upgrade.chr(0); # o2 I( e8 |7 r6 j* S, a4 w g6 y
// 写入socket
* }' ?7 s2 D! M6 h! K u
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0)));% N8 ~& ^ O1 `
// 标记握手已经成功，下次接受数据采用数据帧格式
( X8 c6 c' o. V; E
$this->handshake = true;$ N9 B, U) v1 B% E s' _
}' l' w" `: y/ s& w2 `! o
/ Q6 s3 J7 a; d. T0 V
1 `$ b1 d( |' e0 I# i
function encry($req)1 x/ ^1 |8 w g
{. h- B' m# u+ e$ b+ B- K8 G
$key = $this->getKey($req);1 Y4 L) y' W, c/ [
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";, @; O' K/ G& H- O d4 m( \
. v3 u% G$ l+ r: a
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));
; i0 M6 R3 H) {
}: z0 m" g' f, c1 V/ Z" I
, ~8 v& r9 ]1 N& W
function getKey($req) 9 h& M6 r, |4 w6 g% r
{4 o0 n6 O2 W6 H- y
$key = null;8 h D; \, g! U" _; I! I% O! {5 Y- _
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) {
: c' K, k( e) [; c# j+ F" J
$key = $match[1];
0 d- X/ ?# m' P6 ]* e, j
}% Y; Q/ G* i8 a, e* s, u
return $key;! ^2 `* _& P! q. @6 Y3 p8 H' ?
}
) g/ A" W3 `. V# B; q+ ?8 w8 ]9 D
$ N% q/ i9 I( d8 x9 Z5 }, {2 W: v/ I
// 解析数据帧" k: N* i/ ]% q* L' J
function decode($buffer) # Y) B0 t" }2 E" c
{
5 {. J! x: K2 |6 G4 K
$len = $masks = $data = $decoded = null;% M- d# C! G0 U
$len = ord($buffer[1]) & 127;1 I& ?0 x/ N- I1 N5 T$ K
$ j i% Z" I7 {2 c" ?, V1 X
if ($len === 126) {! D6 U: j# n6 E) B
$masks = substr($buffer, 4, 4);
8 K! H- @ J' E$ M4 C' s
$data = substr($buffer, 8);
) y) M+ J. j7 J+ Q% ^' X
} else if ($len === 127) {8 g8 G1 h8 t( L# b& o5 t
$masks = substr($buffer, 10, 4);0 z; O, I' L+ a/ I1 J
$data = substr($buffer, 14);+ Q% `# ^9 @9 |7 L
} else {$ c% H* u; m4 C! q! ~" Y
$masks = substr($buffer, 2, 4);
F6 V: Y9 R F; b$ c- v2 A% l9 p
$data = substr($buffer, 6);- s) r8 K+ ]' \: o$ X* X" q: U, a- m
}
+ v$ R/ m2 I; s9 m
for ($index = 0; $index < strlen($data); $index++) {
( T0 W( u& p' v* ^! w9 Z6 Q
$decoded .= $data[$index] ^ $masks[$index % 4];
6 P5 [; A$ S7 A# t
}
. t# S0 Q5 S1 D1 u+ I
return $decoded;) Q) E& I2 G8 B
}
4 q$ }. W7 U. j; F$ P
# g5 q; V$ r" N7 d$ S% C1 d8 L
// 返回帧信息处理% j# D; A. i0 M1 a8 U; L
function frame($s)
" a1 h4 ~! w; j/ |4 |+ R( h+ h
{
5 f9 T# P9 H6 m. q) G( }
$a = str_split($s, 125);
( k9 v6 ?; u8 ^" u+ D0 |# ?
if (count($a) == 1) {* t) ~' ` M* v A# _' j
return "\x81" . chr(strlen($a[0])) . $a[0];
5 ^ G( b3 t' O3 h. W7 _
}4 C4 a/ y( f3 v) x. g: f$ `
$ns = "";
! o# Q' V- r1 I) n
foreach ($a as $o) {
% Q% H4 J9 ? i5 `/ j# R0 Q; k
$ns .= "\x81" . chr(strlen($o)) . $o;( ] X& } q3 I/ L
}! m% G" a* |/ x# v" y U
return $ns;" u# D% J# H; H, V, M% X3 Q
}! [2 I* B, L) F3 V9 @- C* ~
/ s" K. ^6 W a6 V
// 返回数据
7 [. C# A* i, u/ B+ F8 i' ?. z a
function send($client, $msg)8 x; _5 G7 l5 }+ R- J* V
{+ i1 r$ Y- M1 R! y6 v) {
$msg = $this->frame($msg);
* M$ I5 E$ ^ V/ f
socket_write($client, $msg, strlen($msg));; s# H& }3 O. X" b
}- y( B+ k5 ~; B0 h
}# ~: v2 Z! L8 k# a* ~7 A
7 i% v/ c& K3 E" ~+ w" }
测试 $ws = new WS("127.0.0.1",2000);
+ [, W8 r8 T- N& K8 J6 S6 T! A3 d6 x
( T. [- P/ @( [" i: B! C6 k, d+ ?

复制代码

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

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