PHP 简单实现webSocket

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

1）客户端实现

<html>; s5 L) X+ z" n
<head>5 g0 F3 Z4 {& ]1 z' J- ~- F
<meta charset="UTF-8">+ x5 u) T: s$ n& A3 E9 v. \
<title>Web sockets test</title>
9 t0 K9 g+ d, R
<script src="jquery-min.js" type="text/javascript"></script>" v! Z" A: ?0 N
<script type="text/javascript">; r* J, W9 P" i7 M4 w) Y
var ws;, c: p: n3 U1 {" D/ y/ a) Y
function ToggleConnectionClicked() {
0 N4 w8 | k# G" ~; f) o5 |0 f9 v: q
try {
6 p( r& Z/ C2 k
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器
" q: A2 \& R- X5 D
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};
" f; q9 J/ G5 |! D% Y5 ^
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);};
( f8 ~1 q6 m% [' ~# ]
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);};
+ X2 j8 _; T% _% E4 Q
ws.onerror = function(event){alert("WebSocket异常！");};! a" i* q" E5 a8 _/ A4 j
} catch (ex) {" w& N# h7 E0 B( |
alert(ex.message);
1 {; S0 _, o2 b+ ]- a+ Z
}. P% r, |" Z' {+ y
};0 E* ^4 v) Y) K9 Y' f0 i% N, V3 m
/ Q, Q* `' V" q4 W' G2 a
function SendData() {
; Z) Y# k/ k% f/ M/ B
try{3 B7 S' w$ G$ g; m! u' Y' a8 u6 l6 i
var content = document.getElementById("content").value;
$ |) E |) l: w; e
if(content){
# ~7 c( }) k, ~
ws.send(content);
$ m+ P7 {) k/ S' g6 f
}3 _1 `. o# ?1 |6 f# r
" s4 f2 v9 m( U
}catch(ex){- y6 M8 I# ^: K. P
alert(ex.message);2 z9 _' ]7 i, y8 V- d
}4 U; ^3 d, Z o) {+ n3 {
};
& o7 Y; ?9 t; B; B1 u
+ ~" o0 e; m! Y6 Z% G& h Y7 l) G
function seestate(){
& e4 K3 s4 V0 f W& v
alert(ws.readyState);
+ `1 m7 k3 {3 W1 n! M
}; N& a: e" f; `# H3 B9 ^( D/ V
1 G' N# ^& ?( P5 q& R
</script>' i/ k, X; \" {, r2 ~! q* l
</head>& [: F& d6 {' @0 G+ K4 z
<body>: a1 r2 T4 D# p" p( i
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br />
1 h/ ]7 K9 p' u" L) p
<textarea id="content" ></textarea>
& O- ~( o: c" C0 T. S, l6 J& O
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br />3 ?! ^6 { q$ H
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br />5 l9 o6 Y3 ]( ?1 P7 _5 ^
, r( Y% D. T8 u |+ k8 ]
</body> R5 ] J1 E( U* {' O- J. }. I6 ?$ k, p
</html>
' Q9 }/ W1 Y; V9 C }% t& U

复制代码

2）服务器端实现

class WS {. }( P# R2 v5 m$ J
var $master; // 连接 server 的 client/ @' n8 k( a! {' i3 W4 q
var $sockets = array(); // 不同状态的 socket 管理- L+ Z3 B. n( U, e# e
var $handshake = false; // 判断是否握手4 d4 K* I, G' _% v: k8 m8 C) W
5 U8 E! p8 I( @0 L: k
function __construct($address, $port){1 g! k) s3 @) W
// 建立一个 socket 套接字
- l5 d/ b( e* U7 I* D0 V
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP)
6 A: l& J5 k9 Y; P/ {% L
or die("socket_create() failed");
0 g% f2 c+ p4 z/ ?
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1)
/ ?# C1 P& {+ Y$ Y9 W7 p; }
or die("socket_option() failed");& e: n6 B8 I/ U# c) r: r: Y+ U
socket_bind($this->master, $address, $port) 5 M' s3 C* z$ R+ X
or die("socket_bind() failed");! N* \; N: _7 `$ n: X
socket_listen($this->master, 2)
6 M$ [; P/ I$ r' U
or die("socket_listen() failed");! S5 I' L( ^( m$ y! M/ [5 `1 h d' e
- I" L* k0 _' }' e' @, J9 a2 d
$this->sockets[] = $this->master;! e! |7 A' e6 w+ Z3 J& E
+ X1 M( ^* A/ [1 z: D; ~
// debug6 [6 x7 P" B& O1 q- ?" W
echo("Master socket : ".$this->master."\n");. Y O, V+ s8 _0 ?/ R
: D! }7 t2 P' F6 z! B
while(true) {0 g/ D+ I8 J6 t% ]& |2 f
//自动选择来消息的 socket 如果是握手自动选择主机
6 A: q* T9 I$ G7 P
$write = NULL;
- i/ C" U$ D/ ]' W- P: Q
$except = NULL;4 h1 r5 O5 u+ E1 n
socket_select($this->sockets, $write, $except, NULL);
: K( G! ]0 M4 V7 m1 t, B# ~
3 k8 ?9 _# ]4 x/ s; u) z7 [5 S
foreach ($this->sockets as $socket) { H c! R f! ?
//连接主机的 client
) F* [% H, S" d0 j9 U9 u
if ($socket == $this->master){
8 ?1 F p0 t, s) u/ i- Z
$client = socket_accept($this->master);9 y3 h: }: P2 b$ X( B6 _ `# o
if ($client < 0) {+ z2 _( f8 Q3 e- L. k: T: x
// debug* `# ?$ _ |2 m. Q1 @
echo "socket_accept() failed";
' W( \0 x( Q. }2 t1 u1 V( C
continue;
! N" b4 ^! a0 @1 S0 E6 M, f. Q
} else {
' A6 H. k8 W3 ?9 J [5 r
//connect($client);
& D0 m( [4 ?" h' z. f1 B2 X3 N
array_push($this->sockets, $client);' O' C7 i/ h9 u% M
echo "connect client\n";
! l% K' j* |" ^* L
}
, ~) D* Q: f) e$ l2 N8 B& E3 W
} else {! Q/ f2 d) ]' {+ B
$bytes = @socket_recv($socket,$buffer,2048,0);$ e/ Q- x) B1 [ j+ I1 }4 Y
print_r($buffer);/ f% {9 N& w( u8 J. G5 a
if($bytes == 0) return;- q; ~2 T( U7 }) a4 _
if (!$this->handshake) { }1 z- N5 p& Q9 s4 u
// 如果没有握手，先握手回应& u2 y# c$ H- [3 h- v& Q
$this->doHandShake($socket, $buffer);
. ?* o0 |/ K3 f, l* k9 p; U5 O' j
echo "shakeHands\n";
# v, d% w% R: u# J
} else {" P, a- O, p* `
* j4 X0 e& {+ m
// 如果已经握手，直接接受数据，并处理
! C- A- x4 F; ^- V9 L' U
$buffer = $this->decode($buffer);9 c' u4 E$ K0 n: |7 A
//process($socket, $buffer);
8 c7 Q7 D0 L1 E# y5 I4 u+ B* W
echo "send file\n";
% ^$ H9 C' D; L- u; h
}, C$ c2 k- C" U- ^
}
& m4 z; ]+ k; K- V n; w# [5 q) J. d
}6 [7 p- Q: K" O7 C* \; B1 a6 x
}
) U" ^0 R# s& g9 V
}
& Z$ w9 c, r0 s! k1 Y7 z
, e3 W5 E$ h, D0 l: O1 S9 M. k
function dohandshake($socket, $req)8 C3 N/ A+ M0 A. F
{
+ ~$ e6 H- @9 v1 a
// 获取加密key" d+ G1 m3 e2 y
$acceptKey = $this->encry($req);! H+ b* z" `. p8 v$ _3 X: H0 [' y4 Q
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" .7 [) I0 A' m: I1 d4 ?7 F0 h$ X/ e
"Upgrade: websocket\r\n" .# O- U1 S/ N3 T: N* e
"Connection: Upgrade\r\n" ." v" V# Q; }6 I8 J% \2 @ j8 B
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" .5 Y. {2 {9 Y& E6 s- g( t
"\r\n";
5 B6 r1 D. B/ {
3 h D3 Y; D1 ]& o0 d& j; Y) I
echo "dohandshake ".$upgrade.chr(0); * n9 a, R8 S1 o# h% d0 D' O8 q$ K
// 写入socket& D. j. C. Y# U) X$ m7 J
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0)));! I$ g7 a7 d5 a$ F- X0 ~
// 标记握手已经成功，下次接受数据采用数据帧格式* t. \% l& A/ k( z0 F1 ^" i+ n
$this->handshake = true;
* J6 i/ p9 }$ c
}
- t; g$ r5 h9 z) l- Z
7 k" t, d8 I9 E% ~: ]; Y" p
& q6 y! e( b4 u9 X, c2 ^
function encry($req)
2 c& K2 ]" p) K3 t
{; d) F4 y( h- H$ m l
$key = $this->getKey($req);
- P U$ K3 S ]2 D4 U1 v
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";
w1 F6 c& g9 {( b: J
3 W, h. a- y, c- ~
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));0 [) r" Z: m2 F4 O4 e$ @" E. T
}% X8 l& l* B. G) ~) W( q3 f
" |- R" N, J2 N
function getKey($req)
. X O+ h' p/ Q" Y
{$ _6 T' s0 h2 r: R
$key = null;# L/ c) M: r; F% n7 B7 ~
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) { + r- R4 s# W m8 C! O$ \
$key = $match[1]; 5 a0 o% r7 w( t& k, l
}. Z: h; @6 c: x6 w' D* j' z
return $key;
9 T) N% s1 X( n4 l5 D9 T" h3 k6 I5 J
}9 y, F- y8 {! ^9 \. ~) ], U, G
8 `: z* v" u- e' K8 i0 Q8 Q5 ^7 f
// 解析数据帧
* ^" b( n- m' j$ V( p
function decode($buffer)
6 s0 R, q" I- J
{
" p- ~) F% ?: H, }% I7 u' P
$len = $masks = $data = $decoded = null;; ^! r2 m4 Q: Q0 ^7 h9 H
$len = ord($buffer[1]) & 127; n( A- y9 T+ f; B3 I. j! z% K
) E; R2 \, o. D/ h, x
if ($len === 126) {3 z v! m' q; U* b0 ]
$masks = substr($buffer, 4, 4);! ]: l' g! n# K* s' I* f1 E
$data = substr($buffer, 8); K3 C( T9 B) U: X
} else if ($len === 127) {5 P5 ` X+ f4 K0 S4 ^: }3 I
$masks = substr($buffer, 10, 4);" S! P7 y9 s# K
$data = substr($buffer, 14);" A" x3 Z2 x( b b u* R* W6 o% S
} else {* x- x1 U6 C1 x, M
$masks = substr($buffer, 2, 4);
* m2 q4 L. F- I. I
$data = substr($buffer, 6);
7 s0 V6 k1 k( U, L ]
}! \( I8 y5 r* V
for ($index = 0; $index < strlen($data); $index++) {
6 b. \6 n6 p/ c+ S1 c
$decoded .= $data[$index] ^ $masks[$index % 4];
# d; Z" P8 q$ S. g; n9 J
}, _( `- L8 f, K/ Z
return $decoded;& J6 n$ A( x" \* W+ @6 w
}
2 N3 M. Z# Q: L
% K: v8 K+ e5 j) V/ k+ S+ i- Y( _
// 返回帧信息处理0 ]6 w: i. e( n9 M& W% m. o$ R% X7 Z6 l
function frame($s)
0 s8 E6 T5 j1 ~* B% K8 U! r/ G
{
. Y3 B* M+ C1 g7 l% k
$a = str_split($s, 125);
: A2 Q+ @# ]( [- z3 Y
if (count($a) == 1) { O$ f+ R9 D9 c8 n4 O; _0 ^
return "\x81" . chr(strlen($a[0])) . $a[0];. H2 u5 i+ _- X3 n. S. L4 K+ o4 d- d" ]
}
2 B9 t- Q+ T% x, H
$ns = "";
% j0 ~! Q& }# Z
foreach ($a as $o) {
+ S5 u. [! I4 B3 D$ ~/ r B
$ns .= "\x81" . chr(strlen($o)) . $o;
2 o8 {, U( q4 ?9 C. R
}
3 H! s# j0 d) _. v, M. q. t
return $ns;
% I4 T7 M) C ^! z/ k3 H; C3 K7 g
}
) y$ x- ^* D5 ^" V, c# G* P( m& d
' o1 N- ]& @+ R4 P) c/ p" ^# t
// 返回数据; w. G5 r4 w% l: n) \$ X I: [5 s
function send($client, $msg)
# ?6 s/ ^' B+ r7 T$ p
{" f6 t2 ^1 l# Y% S
$msg = $this->frame($msg);
& d2 M8 L# g# M# _; L7 J
socket_write($client, $msg, strlen($msg));
/ V: K+ [6 L; m& E; n- U5 `. R
}4 b g! ~3 X0 c# R7 U
}- m+ W1 ]* C& E
' D; x( t' r: d$ z& {" l$ }
测试 $ws = new WS("127.0.0.1",2000);
0 S8 e' \) e- F
. h0 ^0 W) _ D$ q) |

复制代码

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

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