PHP 简单实现webSocket

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

1）客户端实现

<html>
* l& {$ j5 g+ c, w
<head>' U& D ^+ U4 U0 G
<meta charset="UTF-8">
0 {' b. @3 ?; A6 ]% C, i2 G) A
<title>Web sockets test</title>
: r7 S) g: Y- r& S) G) B' M
<script src="jquery-min.js" type="text/javascript"></script>8 ~9 i7 U6 j% J2 W7 ]5 p- U
<script type="text/javascript">
$ ^+ I, G# X; `7 x6 `( u
var ws;3 ~- |) H* [* J* _ W
function ToggleConnectionClicked() {
) f. C& y9 C/ D) a: Z
try {% L% q. Q% r( @. }( T1 P: Y
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器 % i% H! Z* x o2 X4 n
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};
4 d6 P: `: z+ P& U) J1 ^3 o
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);};, E. b5 @) I" J, i3 U
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);};7 K8 e- k. r3 e# |
ws.onerror = function(event){alert("WebSocket异常！");};+ }- U2 z( r1 o- w4 p
} catch (ex) {1 p( \; x+ b1 A) D, s$ Z
alert(ex.message); % q1 r! I k2 m- o
}+ V0 m: b/ D- T q
};. I% w, g+ O* I& v6 s
& a- r0 J/ C0 u% M: u# h3 z
function SendData() {* y/ S* {6 P9 U& g
try{) {' Q8 M; s7 V5 d) A1 @$ O" R7 D
var content = document.getElementById("content").value;; A' h4 z' @. Q7 b
if(content){5 T& s0 f6 N+ a# B* J, \9 a
ws.send(content);* g$ Q1 U. i4 |8 p9 w7 V, y4 w: O
}
: d1 |( m6 u, m |
, U4 P6 t/ y* {( O4 W0 w1 ^/ H
}catch(ex){8 C; {: {0 Y- D) q$ m, T, t
alert(ex.message);- Q- c& t" J0 \) i4 S! k* i7 j
}
8 E# i. _3 ~6 I( X5 b
};
j5 m* w9 R; N8 A2 ^ T
- O: {( V# j- y- V
function seestate(){
7 L$ g. x# p1 E5 D B* D
alert(ws.readyState);
* J( G, S# Y7 n% `
}7 p- O7 r( R2 P9 o
% e7 z- Z4 {0 L! e
</script>5 j2 x1 I! K/ R
</head>4 j$ _( Z. V# V- J1 i, g
<body>. l. l5 X/ e& M G
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br />
' ^" r! F8 o" w2 U I* B
<textarea id="content" ></textarea>- S" O I; V* B
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br />8 r+ I7 t$ _& z' r7 C
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br />
% K' ~5 G& ^$ U+ [( \. ]* O' s
4 T( W n f; {, u9 D0 R
</body>9 F7 ~ t, C7 I. S; `
</html>
( C: p- j5 O! O9 V. _' p

复制代码

2）服务器端实现

class WS {
6 b! S7 d \2 V) J
var $master; // 连接 server 的 client3 o0 k! t5 {6 u6 h1 k
var $sockets = array(); // 不同状态的 socket 管理1 _' b# k9 D$ h% |
var $handshake = false; // 判断是否握手
# D4 h+ r/ \. J
9 t: i8 p) B3 c# X" t+ q
function __construct($address, $port){. K8 T" u* L2 _: a! W/ s3 k! X* O
// 建立一个 socket 套接字
6 h8 a4 {5 S8 v7 q
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP) ! h# B; ? R \& e. x- V( p, v8 q
or die("socket_create() failed");
* g" R# P8 D8 u; C7 T
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1)
, a/ V4 y7 t3 h5 M# k; r
or die("socket_option() failed");
7 y$ T5 o2 x$ s/ O4 ?2 @
socket_bind($this->master, $address, $port) - a" U0 }) D3 S3 i$ K3 ^
or die("socket_bind() failed");% s8 Y: k! W( f4 {" d; D3 J2 B/ V/ z
socket_listen($this->master, 2) / P0 z( S/ F2 O C; e g
or die("socket_listen() failed");
; Q( m( R6 `, F' p* ?9 I$ P
7 M; A- p8 r- i/ N
$this->sockets[] = $this->master;2 R, o, e) {3 N9 D' }* l- |
" ~4 W ?7 Y% i
// debug1 y* V) `# j3 X( e) m3 e; X
echo("Master socket : ".$this->master."\n");
; y( F+ P# h; _1 E% Q
+ i7 g9 `6 K8 |, G$ P* m; a
while(true) {
' w3 W( c) _% n
//自动选择来消息的 socket 如果是握手自动选择主机
+ v5 t, H4 q) q) g* R0 j
$write = NULL;
# T& P+ E. U4 y
$except = NULL;
7 X7 d( q4 r- h; x/ n. R
socket_select($this->sockets, $write, $except, NULL);5 u+ x' c. ^& j: I# y" g
6 J& ~" u& R0 z" T1 p
foreach ($this->sockets as $socket) {
5 E5 w3 V. j* R2 `; f
//连接主机的 client ' B$ W% P% C0 k' L) Y' O6 j
if ($socket == $this->master){
; L- q" u2 @ g
$client = socket_accept($this->master);& m& T; {( p- | @
if ($client < 0) {
7 U! t/ ]3 {$ F0 k4 n
// debug) U; Z4 }& ~* F( n) `
echo "socket_accept() failed";8 \4 {9 G3 w8 |5 q- g6 e
continue;
1 b" _) H4 n+ J# U! p1 G$ M
} else {9 L( Q- I& E) s* P$ _& k, }. r3 T% l
//connect($client);4 n# L0 ?3 N( i, _3 _
array_push($this->sockets, $client);/ X9 J6 i: D) D% l2 W
echo "connect client\n";
2 ?9 e: \. {2 y6 y7 e, E! ^% h
}
4 @4 Q" s2 [% @
} else {
6 x+ G% U. m& k& G) Y, Y( d6 X: p
$bytes = @socket_recv($socket,$buffer,2048,0);9 e9 U* v+ _. Q
print_r($buffer);
" _) I j2 \* H( U( b7 X9 L$ m
if($bytes == 0) return;
9 H8 D2 p' [. @
if (!$this->handshake) {
- v. _, n+ h% _5 x: g, I6 {+ \
// 如果没有握手，先握手回应
! [9 d5 {, Q" T: F# _/ A
$this->doHandShake($socket, $buffer);
9 J' c2 q: A7 o+ B1 K' n. f
echo "shakeHands\n";
" R! X6 z! D, k. L
} else {9 |, R3 y4 ^5 K/ F1 _
; Y* d+ [1 F- ?0 C) x
// 如果已经握手，直接接受数据，并处理9 Q! C' e. \/ d) r+ E4 q
$buffer = $this->decode($buffer);
7 A* N/ }, {# i# L6 x5 U3 G
//process($socket, $buffer); ) M+ p, P0 X( X
echo "send file\n";2 N9 d) {, W( d8 k
}) S; H$ c- I; m) ]
}- a6 c% y/ m! }
}
! |' O8 ~& k& m$ o9 S8 J
}$ }9 G6 [7 j* Y& [% ~
}+ ` i: @6 W/ D5 S$ ]) i) [: u
# J) q9 I2 v$ H& P) @# ?
function dohandshake($socket, $req)
- s; E$ M3 W) r# S6 @' O- z3 ]% x F
{7 ]! y9 P% [! d
// 获取加密key2 W1 c: k- Y) q6 B& j2 n; @$ Q
$acceptKey = $this->encry($req);
- ~$ r4 ]8 b: x. g
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" .- e( ^) P/ m0 l! O( { _% s
"Upgrade: websocket\r\n" ./ U9 Z4 A t2 k
"Connection: Upgrade\r\n" .
8 C: F- |: f3 X7 I
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" ., D F7 ~$ d+ T/ T. l$ l# D
"\r\n";, S0 U2 n8 a0 j, S P
7 b, L. t- m* X5 e! m/ B% W
echo "dohandshake ".$upgrade.chr(0); % ~5 n9 j3 r2 x1 n/ n$ r1 ]
// 写入socket5 ^3 D" N0 I" n" x8 G }4 J
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0)));
& p9 O& j! A) P0 t9 ~. m
// 标记握手已经成功，下次接受数据采用数据帧格式7 M: t+ ^' Q- z) @0 |3 f% k; _
$this->handshake = true;
8 y2 S5 j* P3 t( l3 U
}
9 W* C0 B* r9 i9 t* Q7 S O- K' o4 H
3 t& ^7 R( ~* g* z1 M
0 h m4 f: c6 Q5 ?( c
function encry($req)
3 G* s! c4 Y( B0 W1 `$ {- ?# u
{! t# u! V5 L! I! q, e1 q3 D
$key = $this->getKey($req);
7 {5 T, W) ]- V0 X
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";
2 e/ M5 P1 @: N
$ X5 O) d' K7 ?4 Q' n
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));0 p+ J5 B5 m; T1 e& _) i' J; p/ ~
}: `2 ]0 m2 Q% L7 z4 \
+ K- l" _+ u) M8 y
function getKey($req) 8 ~& w" ?3 z9 {. h6 A
{$ z; G4 k* H" I
$key = null;
& Q: D; z# E" s6 O C
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) { 5 R3 j) j5 t" N L) e
$key = $match[1]; ' x ?/ T R6 U: G X
}; V6 r9 Y0 C# ? B' r) W3 n
return $key;
% o1 @7 {; n7 R* k$ _
}
! h& `3 x- D$ G4 k6 I" D
, l. C3 d V* |8 a; b% x; `
// 解析数据帧: p: l; }( G1 L% y/ c
function decode($buffer)
$ [4 z; P, [# S$ r8 l3 ^& j- d) Y7 ]
{
4 Q9 q2 G: \0 p7 X3 D! ~
$len = $masks = $data = $decoded = null;
8 w9 i% O3 j1 G1 F/ E
$len = ord($buffer[1]) & 127;
$ ~/ f, X3 C" E/ v: O) H
0 M% A% b2 h% C, @' a9 j# e! R
if ($len === 126) {% X2 y, b3 C9 _2 W7 d' Y% n
$masks = substr($buffer, 4, 4);
( @6 n8 H6 [5 m+ X) F+ z
$data = substr($buffer, 8);
2 _- `4 C' k3 F- L* g
} else if ($len === 127) {
3 o, G( [, A7 X" f! T5 z
$masks = substr($buffer, 10, 4);; W& y3 v+ p0 v3 k2 \
$data = substr($buffer, 14);" ~) O' U; E5 J( }1 o; z! _* z
} else {
7 M0 p. Q+ u/ S0 V& F: ?
$masks = substr($buffer, 2, 4);
5 H) s0 c/ e: ?% E
$data = substr($buffer, 6);
3 c5 n' |& W8 {7 R3 H: F% z
}
! \- G9 z9 r" U1 O
for ($index = 0; $index < strlen($data); $index++) {. `4 e" ]- C+ w6 n# N4 s" W3 ~ }) |
$decoded .= $data[$index] ^ $masks[$index % 4];
. j9 S; E& ?: t+ M% G
}
; O: f' ^5 ~, q& p4 S
return $decoded;
: u1 M9 g3 \! D- U% U0 V
}4 D" z# S; p% z1 p
s1 u# r# }+ t" ^
// 返回帧信息处理
" x. b5 v, Y& V
function frame($s) ; f9 D7 Y6 a6 O. s @! _; z% _
{+ S2 G! ?- P, p
$a = str_split($s, 125);4 H/ E/ b9 ^8 r7 l$ p
if (count($a) == 1) {9 |$ ^) C* q2 l/ `* ^
return "\x81" . chr(strlen($a[0])) . $a[0];
. S* C: p: C8 I# l
}' W2 f/ R& I' ?0 m4 F+ i
$ns = "";1 u- Y5 g+ X' [0 J7 ^' G& u
foreach ($a as $o) {& }% k; D: Z) ]9 t4 [$ ?1 P* [' G# x
$ns .= "\x81" . chr(strlen($o)) . $o;! Z8 ?& j4 z: U$ K; f5 k
}
! }/ H' k& g, G H. }
return $ns;" L( {' K8 i/ P
}* }2 u# {! s# ]; `1 O m3 T
3 P3 r, g2 u# t) _) r. l1 n
// 返回数据
1 P( k7 N2 t7 l/ [# W
function send($client, $msg)1 w' A7 y- r) v
{( J* F. ~: u' L1 a3 w+ M, ~
$msg = $this->frame($msg);
( `' |) k6 |" P6 U, p; ^
socket_write($client, $msg, strlen($msg));3 R2 ]! F% m7 [& |
}9 U; m4 _# D: J: k. O; l
}
; u1 p+ S0 @$ U( y% Q- n- x
& r, G& L. N/ p
测试 $ws = new WS("127.0.0.1",2000);) r0 f6 k* l8 b5 f
, w+ j" Z' }% t" J' n5 x3 P0 [* ]

复制代码

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

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

浏览过的版块