PHP 简单实现webSocket

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

1）客户端实现

<html>
6 J; p |* z: M0 z
<head>
X+ d) i, u" _ v. s9 _
<meta charset="UTF-8">" x5 A9 B7 X) [9 g5 a
<title>Web sockets test</title>
$ i: N5 Y) i: O% `
<script src="jquery-min.js" type="text/javascript"></script>
- K! `5 d/ E v9 Y8 `
<script type="text/javascript">2 M' X$ m+ e% X+ F6 j |: W$ p
var ws;
( |. a3 }0 H3 T3 p6 |
function ToggleConnectionClicked() {
" x4 ~1 p' Y: K9 Z8 y7 q4 w
try {& ^1 I( n- o* L5 M* l
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器 # c- |( K6 N* e! G. |4 B
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};. P6 U7 g; Y% q& ?( ^
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);};
( B% V! r5 H2 F6 b5 o
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);};7 j/ H# J; u: }& `- a% S- R
ws.onerror = function(event){alert("WebSocket异常！");};5 s6 H- U# u: m, J- E
} catch (ex) {
. |* v y# H$ }( f) l+ E7 k+ k
alert(ex.message); 0 i; I5 p+ g; V4 d/ T3 a) w) `
}, f1 C1 [9 h6 R/ s
};" c8 R: ~9 W# T% {
9 Y' o7 @5 ~; B0 q
function SendData() {
j8 i, T' U4 j2 |7 C- a" k
try{9 O/ |" g/ ]% Q/ f# ]8 O
var content = document.getElementById("content").value;5 V& x3 c$ N* H# o3 K* Q7 Y
if(content){7 a2 k0 i/ ]4 b& M" ^! p1 Y
ws.send(content);
( s- Z6 B; W8 u' x# R8 ^' }
}& Q! _7 d5 X0 z3 \
% _ ?" Z/ i8 D$ L3 p9 b: `# ^( ], i
}catch(ex){. x* V4 |" |$ B6 z- h! E
alert(ex.message);
( h/ W J" s0 M
}
! b9 K8 H! s1 Z0 f/ p2 E
};8 d% c5 b, e9 D* _
4 O2 z; j0 N" ?8 b5 i$ S0 e! u
function seestate(){
& {( y" i0 r3 r8 n" j# |- S
alert(ws.readyState);$ z6 A' o9 V+ L# _* Q) t
}$ p0 v3 o6 c Q/ s5 E$ Q3 C
% D4 ?- V% P- ]0 C9 ^ ^
</script>2 s% _: y8 x1 G% S% K5 A
</head>7 S: }. D4 {5 \2 X2 E
<body>
Z2 d4 x2 x9 K u. C. H! W
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br />
" u* d" M0 s6 w6 L
<textarea id="content" ></textarea>
2 L( M4 x# [- `
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br />
$ ~5 g' b3 \1 x: `% f0 O
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br />
- }! e3 b2 B* f! x- Q
4 i$ K; T. m; k# O* t
</body>$ Z* T+ ~/ ?' b) Y, U9 x
</html>
5 w7 D8 T y( J* h) K

复制代码

2）服务器端实现

class WS {
4 c2 Y }* [, F `+ V
var $master; // 连接 server 的 client
Y2 }- G" H1 _
var $sockets = array(); // 不同状态的 socket 管理, E" C7 V0 J( t6 G
var $handshake = false; // 判断是否握手
" N2 O. }3 S5 }" f% o
: H) e% r8 X1 i2 r
function __construct($address, $port){: L7 e7 s- W( o4 F" J
// 建立一个 socket 套接字
5 {* V5 |1 U2 }0 Z. y( u4 |
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP)
2 M) Y3 B: ]3 g7 p5 c
or die("socket_create() failed");: l) a I: j: Y! ]7 {( q" e$ K. o
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1)
& Z( h. h1 ^* y5 P6 u# W
or die("socket_option() failed");
1 s0 E% T: u! u, h* M
socket_bind($this->master, $address, $port)
2 _: {* Z( b/ D; X$ [3 T
or die("socket_bind() failed");
9 P" Y' C; }. Y8 \: |
socket_listen($this->master, 2)
5 \0 l" y- S2 G" b* K+ Q! R
or die("socket_listen() failed");
+ Y% g6 L2 d$ x+ U
; o) |( u; ]3 \, m
$this->sockets[] = $this->master;) Z, p" V6 S* f8 E4 Q$ f
9 `. F* M: \* M# ]9 W( e0 Z) y' K2 n
// debug) [2 W; c; D& l' X X y' b1 o
echo("Master socket : ".$this->master."\n");
1 F7 }9 e( [% }' q0 A, X
! g8 z/ G& l, z t( K* }
while(true) {
9 W' a; ?: J {2 @$ t
//自动选择来消息的 socket 如果是握手自动选择主机
( m0 e6 P; \8 o" Z, M) ]! ?" ?: @
$write = NULL;- l: e* M, ]0 ?2 x7 B, e# M* B a
$except = NULL;
. p6 m$ E/ f5 `5 {9 _9 J
socket_select($this->sockets, $write, $except, NULL);
5 v. [* ]' a$ x* z$ t) g) ]) L
& W9 F. l8 k3 e3 M
foreach ($this->sockets as $socket) {
; K8 J" Y# P; [% d# M7 H% Z3 n
//连接主机的 client
' n( [4 K% V0 D" p5 E& R1 x. P! ~
if ($socket == $this->master){
: E( B" n9 I2 W9 S7 k* f4 Q
$client = socket_accept($this->master);1 k9 D, |( w/ f, b
if ($client < 0) {# r# N% j z: Y) M; c2 L+ b% f
// debug
: w; f& Y+ d) T6 V1 {: ]/ r
echo "socket_accept() failed";3 @5 a" h+ n) Y
continue;# D5 t! g% @: Q6 \
} else { ?7 Q, Z7 _; s4 Q- x- G
//connect($client); [+ \6 @# U' r: D3 H; ^
array_push($this->sockets, $client);
3 S0 ~% J% W, _$ D3 A6 z, m H
echo "connect client\n";' ~' |3 v7 y& w) q7 @% v2 S* A) `$ s
}
/ z W8 d7 H: ]
} else {
2 a5 `9 e" x: m ~! ~
$bytes = @socket_recv($socket,$buffer,2048,0);# J1 }; G, d' x r
print_r($buffer);
. B+ r& q1 n' e2 I! F6 `* p
if($bytes == 0) return;9 l7 I0 w& M, R* L' g0 H& i
if (!$this->handshake) {
% F4 a G2 F0 x! k
// 如果没有握手，先握手回应
{! @+ {5 ~. I$ X p1 A- G
$this->doHandShake($socket, $buffer);' M7 }' V2 p# a# v/ W! r# y
echo "shakeHands\n"; T6 B0 U; b' w4 T7 D
} else {
6 l0 l3 u6 r3 G7 e: ^' r# X3 u% ]
& f+ M/ l+ S/ ~
// 如果已经握手，直接接受数据，并处理
; P$ S1 v3 ~9 i- n! j% f
$buffer = $this->decode($buffer);$ L' I, K3 K0 B( p% z5 u' g$ t/ n8 b
//process($socket, $buffer);
7 e8 `( J+ ^9 f8 @
echo "send file\n";6 T* q6 B$ s3 c' u* D( l
}
# S0 I6 D. s8 q( _6 V" X
}% k1 _, z0 {6 D2 @4 ?5 ?0 w8 _
}
2 |4 K* G+ G* ^+ p. o6 w7 ]/ @/ P
}9 b( S4 Y* w4 e' Z
}2 L( s/ S! Q' w* K j" X# T2 P* Y r
( T7 ~ s, ?/ r
function dohandshake($socket, $req)
, N9 U) r& a/ ~* Y& {% i3 k
{
7 ~# g8 p V. G
// 获取加密key
! j2 e: l( b/ k' d# v- H2 L8 _
$acceptKey = $this->encry($req);
6 U/ R. R/ S7 H- u
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" .
& d4 m1 u3 d% I1 x! L7 h) c! H
"Upgrade: websocket\r\n" .
8 {1 ^) V- [3 f( n {* g) g
"Connection: Upgrade\r\n" .. ]9 {5 {( y$ C$ g E
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" .0 M* t I A G/ l5 w
"\r\n";0 u8 g) \0 I$ _. ^/ F, |: K
2 I) k/ c% T" B9 o0 G: M
echo "dohandshake ".$upgrade.chr(0); 9 o7 _/ C# O$ D5 `6 P/ }
// 写入socket
^ k, v. f8 v. [9 S+ y) w
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0)));
5 A5 o. H, O2 a3 Y8 Z9 W
// 标记握手已经成功，下次接受数据采用数据帧格式
" M' ~0 _" g7 s C9 J2 F- L
$this->handshake = true;
9 R" c5 ?0 C0 M3 d" x3 R
}; N; p5 x8 U/ E* Z; }# I% p2 G" R
. u3 j8 m3 ?6 l
: ], q" s: |: E7 H) i
function encry($req)
1 e) z4 D4 l& S/ T! H
{9 J& e) f; O( V4 e( m9 j% p" K
$key = $this->getKey($req);5 q+ c6 c0 H g% R( i
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";! u, U4 `9 m9 p
' y5 n6 k0 i8 Z; I& s) R& t
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));
4 t) @, o6 a( K( d& J9 V
}
8 Y8 |8 Y# x9 l, x% G
4 X; N7 X* L4 N
function getKey($req) 4 Q( U/ e1 Y% o$ u W
{; W6 J; i. o! x. x0 P5 m
$key = null;
5 C" t2 S: |" u$ k
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) { T$ J5 t) [) ?! ^/ j/ m
$key = $match[1];
) C8 G+ i# e. ]! r1 {+ G: ^: t
}
1 v+ y9 G5 `* ]0 J" Y+ @. a
return $key;$ ]7 h7 b+ n6 S0 e5 N
}
' x( p# P6 \2 Q, S$ O! u; T5 W
, c$ K' e$ }! I. W4 ^* E O. G- x
// 解析数据帧
9 f) t6 ?/ q9 c' A
function decode($buffer)
' K- f$ K- ?. n
{7 h: J( a, G* B& L8 Q5 |
$len = $masks = $data = $decoded = null;- r+ w! b5 p) w
$len = ord($buffer[1]) & 127;
" N( }, c2 t8 t/ s/ }7 m, K2 F
, Z( e% y# Q1 C% X8 B. a: C
if ($len === 126) {
( j9 Q2 |: h4 k- U% L
$masks = substr($buffer, 4, 4);1 i# Z( H# V* m$ V( T }# Z; B- _
$data = substr($buffer, 8);
, [4 ~8 F7 Q7 s" C) d8 b
} else if ($len === 127) {
: R4 j, ~9 D3 Q- N3 [7 h" U
$masks = substr($buffer, 10, 4);
* b# V1 Z/ u4 n6 |. z# ?3 ~
$data = substr($buffer, 14);/ T/ `9 E8 u+ e/ ~$ P
} else {% v! e6 u9 e% N Y
$masks = substr($buffer, 2, 4);4 S* R: z7 \( N' D3 {
$data = substr($buffer, 6);
3 ]' T8 _3 q9 J
}0 t3 W V2 y+ h" r, K
for ($index = 0; $index < strlen($data); $index++) {
. z- U" d7 S7 ?4 Y# W' b7 U
$decoded .= $data[$index] ^ $masks[$index % 4];% L& s# N( P g& c
}
9 n7 w* ~4 _6 h9 U
return $decoded;' I4 z! O( L* e8 V- @- f2 L
}, x+ i' L' F# P
7 W/ ?4 a: }" A4 W6 S4 ^. X' s2 e
// 返回帧信息处理0 B E) y2 q2 j1 E: R+ ^
function frame($s)
4 A+ p8 [% L2 R, ]
{
+ P" R y0 J, U" s4 B4 q" L
$a = str_split($s, 125);
- d- N0 c# V; H
if (count($a) == 1) {/ m: I. V4 x- d! X: c
return "\x81" . chr(strlen($a[0])) . $a[0];
: E% o% O* \0 s8 G5 F0 Q; s
}$ X* ?8 ^# @8 D3 y$ u: K$ C
$ns = "";( f$ e$ n# n' s- D
foreach ($a as $o) {2 E( z3 h3 |+ Y1 w" ~9 Y+ [
$ns .= "\x81" . chr(strlen($o)) . $o;! ]+ i+ S9 o& t( r3 n; J5 ^3 U
}( b7 C5 p% L3 U3 X5 u# }
return $ns;) d8 }; a9 H: d* R0 J
}' {2 v& ~/ k3 n0 R& G
9 E; f7 R+ \# k
// 返回数据
" B, y5 [- f# h' `/ x1 h) z
function send($client, $msg)
6 w5 J R( B; k5 S o
{6 v3 g& \0 w9 D" E, H
$msg = $this->frame($msg);$ C9 ^# N0 [9 |2 u; V7 L
socket_write($client, $msg, strlen($msg));! @2 X0 _ ^7 m: C; J
}5 T8 e, I4 ~5 W3 w: ~% H i
} s. g) ~. [1 a7 c( _# d+ a, y
3 `/ V" i8 K8 h
测试 $ws = new WS("127.0.0.1",2000);
. Y$ t ^" q: O
2 n) R: B) @0 o+ n: U/ r

复制代码

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

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