PHP 简单实现webSocket

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

1）客户端实现

<html>- ]: G. ~: V. U
<head>
7 b2 W5 ^0 N/ f
<meta charset="UTF-8">. u9 Y' E/ s+ a! I
<title>Web sockets test</title>
" B' }/ n8 U4 P9 H
<script src="jquery-min.js" type="text/javascript"></script>" A- a" |, }! ]- Y
<script type="text/javascript">
7 z/ L* T% d! f- n. U' Q
var ws;+ |( b! @3 M& }$ V
function ToggleConnectionClicked() { . C" z5 D9 ~ s7 g* q
try {
# C, x3 ?% E, K1 b7 D. R
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器
$ l& h3 Y0 @6 e5 A1 r; X
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};+ i7 ]/ a# x* o; l% ^! ^. F- [, ` m
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);};+ `' u+ s" N% U: C
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);};7 [' C; s0 M1 u* w4 f. G, |) q/ X
ws.onerror = function(event){alert("WebSocket异常！");};( ]' `# F# o6 }7 _. P) H3 d" c
} catch (ex) {
* N0 U5 l L: m5 ] u
alert(ex.message); % u( x% Q1 V3 Q2 q* @. V
}
! ~7 Z ~+ M1 z1 M G: v
};: l% t4 t2 H1 |# c* l+ q7 j% C
( i; b; g% U1 Q! |; u' `% o9 |1 B
function SendData() {- `, j$ A/ i1 P7 b# f2 K
try{; n& P- Y( a9 X! x! ?3 D
var content = document.getElementById("content").value;5 _9 F9 O# w8 y' A: N I
if(content){
! a' [- x) s% j9 H, u0 l+ Z+ i
ws.send(content); V& U2 l1 U# e7 p" l! @
}9 @$ {- n5 O" y0 Y7 C3 e
; I/ l4 t) ~8 M4 s6 l
}catch(ex){. g% w/ J: a0 L4 }. j# s
alert(ex.message);
" |3 ^' K4 n5 c8 C+ m9 C i/ M
}1 S' n; w/ N* X% w' c' Y
};! R) y6 I( U% p/ s- \
2 T* I# x% ~; `
function seestate(){
6 q7 X9 j; |) O/ b* F+ J
alert(ws.readyState);
5 o/ ~$ r0 f3 o/ c5 _* n# ^
}2 {6 C( Q9 a4 W$ V! y6 V+ j3 A
# s. K; B- o. ]7 P+ Z7 j
</script>* u2 ^: J; c' U8 M
</head>
, [: }: ^% \, y$ o! m$ U4 x
<body> k$ W9 j5 X5 F. O1 o; s7 k) D) T
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br />* G! U0 Q' H7 [! a2 ]% l9 l
<textarea id="content" ></textarea>& V2 N* S% X9 Z$ C( p( J
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br />; @& ~, w4 e- u1 B! g t
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br />8 E T! G7 I6 U- R
0 E7 r4 Z$ P, Z3 U4 ~
</body>
* i; G6 `5 H# [ g2 V% n
</html>9 _1 k/ \: v9 F. [. x# b @

复制代码

2）服务器端实现

class WS {
6 w" u; n+ g" }( J9 S; O$ j
var $master; // 连接 server 的 client4 |6 w" S$ t6 G* L$ a5 K* B4 o
var $sockets = array(); // 不同状态的 socket 管理" K% u3 u+ T0 o1 E
var $handshake = false; // 判断是否握手
$ M" h# A' ~" M) h, i: [
2 Z5 K2 n0 @. m2 k! [' b& L
function __construct($address, $port){, t- _7 h" Z5 g* e0 W; |
// 建立一个 socket 套接字
- q+ e, F- p& ]
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP) # B( S7 n% _5 h q; I% P1 i1 a
or die("socket_create() failed");
) y. L( t2 p1 |
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1)
9 _8 e+ f$ s4 b* f) W( y
or die("socket_option() failed");
" U) V+ y9 U; n2 K9 M' i
socket_bind($this->master, $address, $port) $ Q2 F0 e7 I0 ^! N8 D+ ]
or die("socket_bind() failed");
: T3 }9 e: b( y! y- u; B8 R5 t
socket_listen($this->master, 2)
5 S ]! q0 B( e% D
or die("socket_listen() failed");
! v% ]4 F+ p5 c. p ?1 J+ f4 `
6 m& Z0 C5 B5 t" `
$this->sockets[] = $this->master;0 _/ d- c- {3 { Z. j
0 k/ U( z' W0 c P0 \8 @/ o
// debug, Y" r$ W6 {+ w* |- U( P, z
echo("Master socket : ".$this->master."\n");
0 D( [ T) U. H. y
+ D% ^1 T5 |) V+ _5 @) c- `
while(true) {; p+ M3 @2 f" D8 ~0 N+ r# T
//自动选择来消息的 socket 如果是握手自动选择主机
! g5 e* }7 a. W o+ M |
$write = NULL;
5 M: R# V- w- I7 u; y! f$ ~
$except = NULL;: w7 e# z% y2 g) L- d% E# G8 j' V2 ?
socket_select($this->sockets, $write, $except, NULL);
; ~8 Z0 J4 P' A4 g* ~% Z
& w- t5 v" H' ^- G) {! i, f
foreach ($this->sockets as $socket) {3 E+ Z+ Q4 k( {% A8 t
//连接主机的 client
6 h& N0 W Z, e4 i& X
if ($socket == $this->master){
) v4 }) S! X' Y* f
$client = socket_accept($this->master);
4 w" @) ^* q8 j3 E: K" i" @
if ($client < 0) {
6 \8 y( b( ?% e, r1 W" c# h
// debug
3 N f1 D' `" P) i, A, q! R0 b0 g
echo "socket_accept() failed";1 a& ?& k5 P. w5 ^
continue;
2 y6 `5 ]1 `; x3 f; L
} else {
6 l+ U K. _2 ]
//connect($client);- Z7 @ h. D4 I% [
array_push($this->sockets, $client);
, D# G, ` _5 z2 u8 A. H2 p
echo "connect client\n";
, H% z* W2 \$ U l% t# ?
}2 Z1 `$ x6 V3 `5 p2 d2 G+ z5 K! o: {- F
} else {0 V" E4 h/ ^+ r. X* m; d7 z. [9 a
$bytes = @socket_recv($socket,$buffer,2048,0);1 d/ C" ~% s7 B$ W2 W& ?( w
print_r($buffer);5 H' k6 D7 A# y5 v5 h( Z
if($bytes == 0) return;
9 h4 Z2 y/ J# F6 J! `" ?
if (!$this->handshake) {
! E- n; H s C: C8 y) N
// 如果没有握手，先握手回应
: O" G% T3 `7 F/ h% x
$this->doHandShake($socket, $buffer);
* L8 y5 z: Q: q4 _4 Z
echo "shakeHands\n";
$ {, a/ H, r) b% Z: q' S3 ?
} else {
- n- K; p- d6 |! ^+ U2 d8 N
5 ]) `6 p9 J2 @6 L& `
// 如果已经握手，直接接受数据，并处理; T9 Q \/ i* o$ n
$buffer = $this->decode($buffer);
5 J& n7 ^7 ~4 J
//process($socket, $buffer);
echo "send file\n";2 T5 c: V4 J' G y* Q2 q
}, G. }0 S& K$ v: T& f
}
% p& d3 ]: n) P0 y2 p$ H! Q7 K7 w1 }
}
2 z, D4 D# ~- ~' F3 e
}" u, Y$ B! h6 a& j9 s( j
}) o' S' N- l6 b2 v! t) i
2 D3 g' F- @$ p6 r$ k u9 a
function dohandshake($socket, $req)) d, X+ V* v( k6 [
{
) [# N+ x' o, B O4 z
// 获取加密key
6 f4 t4 W$ V9 A6 h( ?/ `5 c* H
$acceptKey = $this->encry($req);9 {& x! N/ d$ l8 }8 J# n
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" .
h1 e0 M2 R/ `+ t
"Upgrade: websocket\r\n" .
& c/ P' a% P! d5 @" T, R
"Connection: Upgrade\r\n" . U; n6 D e, ~7 t! b
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" .
, u: w2 J1 Z' ~$ t. f" X/ K) U
"\r\n";
) c' M% I; m0 D, D4 l( P
9 v' G" |) j/ w5 R
echo "dohandshake ".$upgrade.chr(0); 3 @* E2 y" t9 G1 _
// 写入socket
1 e, F& [0 z$ o! g
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0)));$ @; a3 d/ t9 R* F1 ~- e
// 标记握手已经成功，下次接受数据采用数据帧格式
3 S' E+ R( z- S
$this->handshake = true;* A5 ^9 [% F2 ^7 A7 A* m& T. x
}- S( @7 n5 G- I2 f$ u9 Q+ Z
/ Q. D' k5 L7 |+ W. t0 H" F
7 U. |' O0 S; u" {3 r8 ~
function encry($req)
3 H# y4 |2 \3 H% K+ ?' j) R
{
# d4 l* `3 L& k! N& z
$key = $this->getKey($req);6 Q' J7 R/ }9 n; R9 o# S
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";
0 G% X O ]) c- ~! D! E6 I
0 K4 w" a6 _+ \! J
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));" e I: _+ n" i# b7 t
}
* J% n: g: ~1 q1 b0 S
% ~ }* q3 o- c. ~$ G( g" y
function getKey($req)
' [$ S0 t( D, d+ y2 Q9 s
{) q- Q, D) F- H, v, A( w
$key = null;! U* Q+ }3 H! h: g3 ~! t
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) {
# M L: L. j8 o0 d3 v
$key = $match[1];
! C) @9 B, r1 i: S) x5 e
}5 P( O& f- p4 ~
return $key;
4 R7 Q$ g* |: p/ W2 K$ e
}
& u4 K, z3 A, D) d6 W
% a |) A9 V# h$ c- {. {
// 解析数据帧
8 e& l3 t. H0 X1 Y
function decode($buffer) 8 M# k: o, k& {2 J
{, Y8 M6 m6 L( f7 m8 [
$len = $masks = $data = $decoded = null;
' k* F; |8 S$ ~. A2 p" m, Q
$len = ord($buffer[1]) & 127;% j& q* R8 s1 k% H5 z+ W
& ~) @$ L1 {( {3 u8 `' j
if ($len === 126) {
4 H1 [+ ~* w0 o2 X4 p- Q) W
$masks = substr($buffer, 4, 4);
' E% l5 @2 G3 x# {0 Z$ J
$data = substr($buffer, 8);4 p: h2 W3 l/ i0 e9 ?2 @
} else if ($len === 127) {
: T4 a6 O( R+ D! f. L9 D& L
$masks = substr($buffer, 10, 4);
: ?4 O8 L& p4 R& R
$data = substr($buffer, 14);* m7 F/ }. y1 w' {! |
} else {- _. e& g0 x' {0 ^9 Z9 X d3 q4 l* B
$masks = substr($buffer, 2, 4);: P9 w2 i. l; S! s# L3 X$ W* T7 O! }
$data = substr($buffer, 6);7 w: A* }. h% G8 `8 R
}3 c4 k9 r# V, y/ o, g
for ($index = 0; $index < strlen($data); $index++) {
! k9 K, p/ Q: |# |7 H) h
$decoded .= $data[$index] ^ $masks[$index % 4];( @0 Y3 r4 v% B9 d# {- h0 }* K
}
) d: y+ S4 s$ q( j4 N3 F
return $decoded;
}' B$ a3 \* ~ K- d" a# `7 ~, u& j
2 L% g. Q$ B1 {2 @7 R
// 返回帧信息处理
' b1 I: J7 q0 h1 m
function frame($s)
( ^4 u! |3 p! i* Z, u" ~; {- c
{
( f4 i: O7 [; D8 }
$a = str_split($s, 125);
" Z: x6 v7 S+ h, I
if (count($a) == 1) {3 N4 _, Z4 y$ C+ C: W5 H* h/ O* y P
return "\x81" . chr(strlen($a[0])) . $a[0];; K |# C' `4 e+ M
}) `, C# o" D+ B+ x8 w5 R _
$ns = "";
0 V( ~$ _5 b* k$ u0 `
foreach ($a as $o) {$ T. G U$ A' l! F6 Y0 T
$ns .= "\x81" . chr(strlen($o)) . $o;* i: B# a( J4 h7 W$ V
}
4 _; C" H! P8 h9 f
return $ns;
' \3 q; V# q ]3 n6 l4 k) P
}# @* c" z/ a$ I( @6 W
/ t8 e& W' F- x# X( c, U
// 返回数据
1 ]( _ [8 X; c7 J4 E( X
function send($client, $msg)
" }4 h. {. D9 d7 L
{8 f$ c- r1 x8 Y$ R' H
$msg = $this->frame($msg);' V7 {$ {) T! x3 c
socket_write($client, $msg, strlen($msg));
4 G7 |: X- M8 K! |
}& `) B0 C a' `, @; n# @' \
} z0 d; S* {* ~6 r) Z" N
# j! X% n. v1 a0 I0 ~# V( y
测试 $ws = new WS("127.0.0.1",2000);( i4 L7 F& O: E( {" U7 I
2 x6 P" E. R4 T& T, n+ w8 ?

复制代码

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

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