PHP 简单实现webSocket

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

1）客户端实现

<html>7 I! K; S2 {: F9 Z2 V: a% u w7 O" s
<head>
# U4 `+ J# `, m/ _
<meta charset="UTF-8">
; _4 s6 C, p: w9 ^0 @; i8 k
<title>Web sockets test</title>; }* @, s6 K# X. [- f9 T/ H
<script src="jquery-min.js" type="text/javascript"></script>
( B5 D4 c. v4 T( G
<script type="text/javascript">
6 _4 f9 A. t- x, Z, Y. V
var ws;* M0 `" q' z/ W% N! J+ I
function ToggleConnectionClicked() {
2 i, r+ U7 g) |8 r8 R1 q
try {
3 b% w! [: U/ ]3 H# s# j
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器 " P* |5 Y- @$ F& F
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};
! Y9 D/ y. s. N0 m
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);};! c7 y7 |, M, a7 E
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);};
! {! Z& j. P* ?& C+ u. `5 J/ |5 ~
ws.onerror = function(event){alert("WebSocket异常！");};
' ^! T/ K9 } k, C4 r
} catch (ex) {$ X m a: x0 w7 _! T2 r) K
alert(ex.message); . F9 u6 `2 @( ^: M; B H, D- \4 s
}
+ b/ D$ r6 G; h0 W" V$ d; j" z
};- Z- I# x \( |6 H) S3 H" A9 O
3 S4 U' P6 p9 x0 A1 U. e8 k( ~
function SendData() {2 P% K2 ~' f4 m! U& l L2 x
try{% ~) Q5 {: z# U1 x# V6 x
var content = document.getElementById("content").value;
' k, w2 V6 M3 R& p7 [ i
if(content){
- N8 A9 E9 ` R; v5 s. F
ws.send(content);9 d' W/ R2 h" x
}" v$ Y/ A6 d% H- z7 _
. r1 o" \- h; t4 w: e, E5 |' v
}catch(ex){
: X7 s+ {8 w- R: ~/ C. \. Z
alert(ex.message);9 x" M2 i% Y$ _5 _8 M/ u
}* K% c7 i$ j$ p7 g8 x( u1 h: ?
};
3 ~( ]+ t8 |/ K1 |) I4 t
. W ^, |4 l9 P* ^2 o
function seestate(){/ ]+ A& f0 e: v2 C/ f( s" {# e9 u
alert(ws.readyState);
) x3 p8 J9 U5 f8 H( e+ @
}! F- D' g$ v2 {+ ]
* |5 M1 f) k4 P' W& v" Z
</script>8 W) H2 G9 @* ?& u& e& Z9 V
</head>
6 X8 j: ?9 V; b( W- H/ N7 Y
<body>
( L( ?' [' u, l$ l# L" `" ^
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br />
/ I' ^7 g5 v1 ^% ^/ k
<textarea id="content" ></textarea>
" Y# {# |0 S# b5 |7 z
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br />
: p3 Z. a2 ~ N
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br />( O2 `) ~, ^# u5 @* ~& E
& E: c) Z+ N" X9 t7 u5 j: v
</body>6 n5 `) }. @( o, W, I' T, r
</html>4 U8 f6 _5 z' i) o& l. B7 L6 c

复制代码

2）服务器端实现

class WS {
0 s5 W7 M3 y; a2 j" c+ ^' t. C& t
var $master; // 连接 server 的 client
# I: e. H n2 w$ V0 G5 S" r$ a
var $sockets = array(); // 不同状态的 socket 管理
) e0 T/ F$ t9 O$ ~0 m" E0 Z2 H
var $handshake = false; // 判断是否握手1 Z1 s2 f8 J# Z
* ~2 |: \7 q2 e
function __construct($address, $port){
+ }1 K8 e9 d7 ]3 N2 D Q
// 建立一个 socket 套接字
8 [! \- o, R2 l/ g- D
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP)
2 u1 }! X* ], y- o
or die("socket_create() failed");7 t+ @. W( N; ^- h
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1) ! y& |3 M3 r' o* F( O
or die("socket_option() failed");
( \, z+ Y+ I0 x4 ?0 ?4 K" q
socket_bind($this->master, $address, $port) / d; r8 y/ I, y2 f
or die("socket_bind() failed");+ }% {' t! T8 z: j' {. P
socket_listen($this->master, 2)
1 v: v* S' R- \/ P) R
or die("socket_listen() failed");2 ~% Z) X6 t% H+ n7 e2 l- G. L
2 N. P) M/ Y' B% P
$this->sockets[] = $this->master;( y6 q" s# i, n0 f
( m4 w, i, i7 D" ]; p
// debug
& g+ S! E- m. L) \. Z+ d8 H' \
echo("Master socket : ".$this->master."\n");* P7 R7 ~5 s ?; V Y
8 ?$ ^4 M$ I" F: j
while(true) {
% u3 V0 [+ K- M* f% j
//自动选择来消息的 socket 如果是握手自动选择主机6 {/ F8 _! V z3 ^; M2 _" b
$write = NULL;& t0 D+ V. ^& L* f
$except = NULL;1 X: U9 A7 ^. |! ^" E
socket_select($this->sockets, $write, $except, NULL);
/ y/ k+ `7 n2 b
' c% O7 g6 h# a% L7 A
foreach ($this->sockets as $socket) {
- T/ m, T A% b' y2 W! U
//连接主机的 client ; e4 s) v" k( @) j) W" Q
if ($socket == $this->master){/ }) _0 t2 W- B- c3 O
$client = socket_accept($this->master);/ ^, R; z+ d" q& a
if ($client < 0) {
2 C. [; H* y W
// debug1 z G6 |* E% c# ~0 Z+ \8 }4 S6 m/ I! l
echo "socket_accept() failed";
; B# U% p! t+ |9 B( I7 Y+ V: L
continue;
8 h* w5 H; O9 c2 @
} else {
7 K. X4 Y: R( h9 M- R
//connect($client);
/ P- H7 s9 z" _
array_push($this->sockets, $client);$ \' M" V$ B' r9 @' }2 ~$ c
echo "connect client\n";* f4 X2 a1 T, t* G0 Y
}; z: B8 _: T8 [9 x" f4 M: y" b
} else {
+ E7 w( ] O7 K! \1 N3 E! b
$bytes = @socket_recv($socket,$buffer,2048,0);' W$ y/ o' V( k, X: c
print_r($buffer);
0 g! R! m' m4 R; O8 n& I
if($bytes == 0) return;7 V2 f: N* E' i, w& B- Y
if (!$this->handshake) {
" W U2 R1 \! V' j, n
// 如果没有握手，先握手回应
( X' ` G. X m: o2 Q& \ \
$this->doHandShake($socket, $buffer);9 Z% V3 |9 D6 Q( _, i, O
echo "shakeHands\n";4 y, W$ S& J; `1 H; o
} else {
( [9 s0 n0 W$ C& E* X
3 V4 d4 p' P5 k) X2 `6 p
// 如果已经握手，直接接受数据，并处理4 o5 v7 M& R. ]5 D2 }
$buffer = $this->decode($buffer);
' S7 q0 b. [; l6 t% T: H2 x
//process($socket, $buffer);
3 m5 r* `! o- p9 q! Y! |5 G; z; J/ T
echo "send file\n"; ]9 d* X+ A, l# t1 q
}% D& a) A& g9 B. a6 z+ E
}3 J1 B" ]: q9 }& ~
}
8 r: o: i3 H8 i
}: J, a# C' ~' `
}
! m2 D5 f3 y) s% m- d! ]& e
- E7 ?8 P U. ]5 D# c8 u
function dohandshake($socket, $req)
( ]! _1 r4 n C. Q7 x) k% x
{
9 C+ K! u* J# N+ U) q, ]# o6 d
// 获取加密key. ]. g, C. F+ d6 H8 H% p
$acceptKey = $this->encry($req);
; l' ?. b+ j B: k. l7 W4 ?3 X x
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" .+ G& N8 [& m W {5 M) a% y3 f
"Upgrade: websocket\r\n" .
! e1 T) J- w- D( D: ^ D
"Connection: Upgrade\r\n" .
; ]6 O/ v8 ]' h: z
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" .
0 e% o! d Y3 E7 @
"\r\n";
! Q k5 b- {9 A& }$ D
" ]9 {0 c/ Y* n* c2 q, m( O7 x
echo "dohandshake ".$upgrade.chr(0);
! u6 A3 ^6 v& e1 i! o" c
// 写入socket
) U3 N" L2 t3 K8 t V2 O
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0))); ^- U% ]9 I" n# m- {7 u
// 标记握手已经成功，下次接受数据采用数据帧格式& S. [7 i7 f, O
$this->handshake = true;. V( q) m+ ]" r
}
+ m0 @6 N/ E9 ~" z: ?) D
% F2 N5 r' [" [/ q' u( V0 U
1 [# }6 R$ }* M
function encry($req)
* g6 V8 G7 ~5 V4 w" i) I
{
( G( K& b7 a: |
$key = $this->getKey($req);8 {0 b8 D$ i6 `
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";
! C$ q6 V" g r6 f2 V1 d
: l6 d9 h+ W& |9 z9 P
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));, y8 ^8 C0 y8 l
}
" e7 y; F: R" ^* ?) m3 [* e
6 i# R. E0 }' o% m' B' l
function getKey($req) & i3 b$ ^' E ^
{! M/ {- q; Q: U, J3 X. G
$key = null;
. |, P, x2 @6 q L
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) { - s5 ^- D, r& R6 U) \$ H, x& j6 ]
$key = $match[1]; 2 L. v/ e9 d s7 S" H
}( t" h; ^# \7 f6 T: F( T
return $key;
$ |, _" c8 F$ A1 B9 T4 |6 o
}
7 j P6 N ^! h5 X* l
// 解析数据帧/ w+ I* s6 r3 ^0 x# Y
function decode($buffer)
& o) m- G) R2 q- d
{$ n% d* L0 P# X5 g* t; Q
$len = $masks = $data = $decoded = null;
. r, O) Y! @& N, ]& x% n
$len = ord($buffer[1]) & 127;# ~5 L2 {8 h6 u9 C5 W; Z. ]- h) A
, y4 S. h1 _" o o# Y
if ($len === 126) {! W% w, `+ K4 x! X
$masks = substr($buffer, 4, 4);) [; C, ?6 l( q2 g0 |
$data = substr($buffer, 8);+ K/ c$ e8 q+ A, V+ A( e
} else if ($len === 127) {
. Z4 z% D2 e$ m2 l& d
$masks = substr($buffer, 10, 4);
7 K+ Y6 f4 e* e5 x7 t
$data = substr($buffer, 14);
5 O% `/ h% \5 B
} else {
2 Q+ B( I3 q5 i7 d3 o. L# L
$masks = substr($buffer, 2, 4);6 s4 G% e$ A: c" e$ r' \; |
$data = substr($buffer, 6);
# |* D, R* O7 r1 C
}/ q" z. Y2 \6 l# B
for ($index = 0; $index < strlen($data); $index++) {
9 Z3 l1 q- R0 o. f! U ]
$decoded .= $data[$index] ^ $masks[$index % 4];+ p* L, o% F& H; _9 I
}$ Y/ l4 k- c5 C4 p2 G- D7 `
return $decoded;9 c' ]& U% Y! ^$ R" ]' o" k
}
8 v4 r5 W* d' b* ?
, ]1 [+ Y+ X+ M. W
// 返回帧信息处理
0 U, E* y! ]5 f; _
function frame($s)
. n7 H( h( l$ |/ z L
{
, E' s8 Y/ r. Z ], @( ?2 H
$a = str_split($s, 125);7 z+ P/ \: a0 x& {& d" @5 u' Y
if (count($a) == 1) {2 P1 y8 `2 [+ G" O7 { A
return "\x81" . chr(strlen($a[0])) . $a[0];, |9 k* S. _9 A) @0 C" V
}
l' j: n* e" a' }2 x% k
$ns = "";# O4 y$ ?* b9 |* I
foreach ($a as $o) {
: Y5 k; i' J! N1 p5 ?
$ns .= "\x81" . chr(strlen($o)) . $o;8 S# g% S5 ~# i7 J7 k; [1 B
}
# t1 a k' v7 K9 T* [
return $ns;) ]! m- l: G% F. e2 P
}
8 u4 o5 P) ~7 [( H% l
/ W- ?# t) l: L3 S m6 z6 e8 `& p3 D
// 返回数据
7 L- [9 G) Q! L0 y( o8 u8 H
function send($client, $msg)1 j& A {5 t( a: j5 K
{$ r- [* T2 A, [9 a
$msg = $this->frame($msg);4 Q* ^4 I& O+ R2 F4 m
socket_write($client, $msg, strlen($msg));: y: u) J6 l+ W. V$ y% M5 l5 y
}- P3 y) \+ W3 q$ p8 N4 l7 Q6 k( S
}
* x! e' H# Y( I; H9 V8 \" ~
2 y% I: O$ ?0 h3 e
测试 $ws = new WS("127.0.0.1",2000);2 P/ I Q M; H y1 s! n
3 [3 [/ [; C% l o6 N

复制代码

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

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