PHP 简单实现webSocket

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

1）客户端实现

<html>
, B) j9 Y0 f, j6 {6 [
<head>+ T8 r3 Y, _" p
<meta charset="UTF-8">/ E( t3 t$ f- t+ L F1 P! k
<title>Web sockets test</title>
) o$ C$ M: |+ f$ p$ w% @$ ?" u
<script src="jquery-min.js" type="text/javascript"></script>
# a( x) {/ A: \# o7 B( V
<script type="text/javascript">5 H/ B& i! |! E# w( s% U: P6 g
var ws;
: U3 v: _, n. w0 M g9 H
function ToggleConnectionClicked() {
2 x/ j# d4 ~/ P+ U; s$ b
try {
! E1 p3 s! ?6 L- c8 w( r& F
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器
; j( o. [5 X: h( X2 @" i" L4 C3 v
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};
$ `& C$ c& v/ x6 D9 L% _
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);};! j% ?8 a, Z2 n7 U
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);};
' s* ]1 [8 _0 Y- U9 ]) B s
ws.onerror = function(event){alert("WebSocket异常！");};# @0 k% @/ B x/ O" Z$ ^
} catch (ex) {
; h% m2 q) J- B' `; i* r
alert(ex.message); 3 v. Z& l# Y& x8 Y: g2 t& |; X
}
' f* B* Y) Y5 S% P. j7 A1 {% ?
};
; N0 D" ]2 i# ]8 [2 ?) K2 ]9 _8 B
% K8 W: ?+ L6 c7 W3 F
function SendData() {1 E, W% }: H0 ?
try{2 a! C9 b J& u; {4 U
var content = document.getElementById("content").value;0 P0 e- i1 ^. T
if(content){6 O7 d$ h2 i0 A! p
ws.send(content);
1 l/ [. u' F! y1 b' C# ?* m
}
0 w; S' Z: K, b* W
' F) t' \- Z1 M& m& d! P, M
}catch(ex){
, N5 n' l; ~# p# u5 Q
alert(ex.message);
* l5 S a) x2 W- Z: Z
}$ p; G+ N" }/ ^3 j: q
};$ r1 d7 ~& A$ s D6 r# }
9 r; b& [. G4 v
function seestate(){
- B# I- O/ J% [( t& o# E+ j
alert(ws.readyState);
7 n) |7 R- r, t4 n, e8 N
}' g' o. \, e$ q' ?9 |8 u* X
( a* ?! I- l! R) |4 |4 q$ i
</script>4 l& ?) E% Q" G9 N6 v1 e) o1 v
</head>
" i: B" D4 E1 V+ O( @) Q
<body>
9 {& t+ T5 E$ h7 t
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br />/ ^7 w/ X% ?3 D# D I$ j4 f# n1 x5 l- z
<textarea id="content" ></textarea>( T8 n Z3 } m% U# v* Z4 @1 p
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br />
1 Z3 n& W ]5 c4 Q V* X$ C
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br />1 O, m; @) f3 |7 v
7 c: Q3 b* c+ V0 T: g* M6 B2 g" \
</body>9 |9 ]; _' n1 l6 l1 ~0 p7 r9 H' \
</html>
$ z1 l$ z* M# S) S

复制代码

2）服务器端实现

class WS {
$ ?. }' k5 t. w8 U8 t. ~
var $master; // 连接 server 的 client; t' N$ m' ~9 \( f) t* D# r
var $sockets = array(); // 不同状态的 socket 管理/ ~, p5 e6 \ @% R8 K
var $handshake = false; // 判断是否握手
2 ^# W4 `6 D/ c" K3 r+ U% Q
3 S+ w% z- m! F. Y' z& C& m& E
function __construct($address, $port){
& H6 u4 I5 B% F7 |" n
// 建立一个 socket 套接字9 h, s, x) i% K" f
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP)
$ N. M( V7 M* U$ ?
or die("socket_create() failed");& O9 V4 z* s0 ?+ k5 x |
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1)
" A O9 L2 `, r6 W* f# U
or die("socket_option() failed");
socket_bind($this->master, $address, $port) + N& Q" V0 X9 ^7 u: t
or die("socket_bind() failed");
; t% \2 t3 E, B+ R
socket_listen($this->master, 2)
' P6 r) T3 W8 _# D
or die("socket_listen() failed");/ Y# ]7 I( [4 Z* ^2 v
; d* H3 Y% T8 V. ?3 K0 C
$this->sockets[] = $this->master;% T) ]) S5 q7 W4 M6 s) U a2 y
. ^' t! ]. R+ v
// debug
: g7 Q9 P. Y9 U- Q9 X' U
echo("Master socket : ".$this->master."\n");
+ ?+ O6 {6 z/ w* L4 p3 }
) T9 \1 h! i) L0 E
while(true) {2 m9 J. `9 Z* Q' F5 K
//自动选择来消息的 socket 如果是握手自动选择主机5 T7 b' I4 N! k7 \3 g# M. ?
$write = NULL;
% B: X$ q2 Z3 l" ~; a# D3 R
$except = NULL;* d. A$ K7 Y( i' B
socket_select($this->sockets, $write, $except, NULL);1 K0 j1 [- f) m g3 V5 h2 N1 `
/ l3 z9 y6 h N
foreach ($this->sockets as $socket) {. l! \; X2 B; j
//连接主机的 client ! b! ^* d/ c& w; g* I
if ($socket == $this->master){
8 \2 Y5 c& S! @
$client = socket_accept($this->master);/ o2 \8 E* H" @% N! v
if ($client < 0) {
; k+ }$ `- ?8 E, T, N
// debug- F/ j' L2 R* E7 t; E5 @
echo "socket_accept() failed";
6 E8 L( r- G1 J3 g b8 B
continue; J+ d$ t4 v" q9 n
} else {3 [9 G3 K' ~3 S; B
//connect($client);8 z9 j, v0 }' k& s& w
array_push($this->sockets, $client);. @+ k! F3 g% U
echo "connect client\n";
( _% D6 `; z3 j. B* I
}
! T9 `- Q4 C" S, Q
} else {6 T' A8 }, N/ N0 h: N4 i5 @
$bytes = @socket_recv($socket,$buffer,2048,0);9 R/ P5 t# ~% q. ^& c
print_r($buffer);
- t; ~& r( k8 c- E
if($bytes == 0) return;; k- |0 j4 a% |9 G* l: h O
if (!$this->handshake) {0 S C7 J$ r6 h" `- K- c
// 如果没有握手，先握手回应7 t% h4 e; U1 A/ ~
$this->doHandShake($socket, $buffer);
~% ]$ F$ t1 Z$ q. H- h; D$ J+ U% W
echo "shakeHands\n";
$ x$ G/ w) ^; Z3 ?$ k
} else {
# F5 _: f( \" {. i. l5 O! L
G- S }- g$ z, I. h1 d
// 如果已经握手，直接接受数据，并处理
$ h2 Q! |5 o1 d+ X3 @
$buffer = $this->decode($buffer);
: k" Z* Q4 I/ h
//process($socket, $buffer); 1 O T' G/ d- S, w: C1 ^5 D9 C0 t4 [
echo "send file\n";
' a2 o- h# g2 G* I
}
+ h. N# _9 {4 O$ r' H
}
& B$ X0 [3 f, {8 r! v* d) Q
}
$ t1 z4 c. n1 T6 X7 d
}
7 n2 o5 z: D- {4 W) u& E
}
1 Z) |' m* ?$ W' t& a
. _2 h* X) N7 f1 e# s( C! F. A/ r
function dohandshake($socket, $req)2 m1 \/ V8 ]$ z
{* K! m$ E( V8 ]. u
// 获取加密key7 q5 L5 z/ C0 ?, `
$acceptKey = $this->encry($req);
d( b2 s; W$ [
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" .
- w* f `5 t( O3 N) { d, {$ G' Y
"Upgrade: websocket\r\n" .
1 G& h+ e: D% C
"Connection: Upgrade\r\n" .
" c/ }2 T' s0 I- N$ o! t1 C$ ^
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" .# o9 e1 y- ]' Z( o6 l6 w3 Q
"\r\n";5 I- S3 L( {* V' W1 q+ F) i" {
. K9 M2 i; i' C3 {
echo "dohandshake ".$upgrade.chr(0);
6 m7 W5 Q( _( b7 Q8 }
// 写入socket
" b6 d* t; A8 L7 B6 A% R; O
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0)));. I0 \0 n# b5 L6 Z" F
// 标记握手已经成功，下次接受数据采用数据帧格式
; u* N0 d6 O. S' s: u4 {; M" V# I
$this->handshake = true;
6 n u: d# j$ B& n1 b
} G! c, B* s+ \% m4 U1 q8 ?! R/ N
" I( ~4 w& U9 H: G; P1 \7 `% x
3 M, Q% }2 o! y
function encry($req)
3 d' K# f9 F* e* h Z
{5 b E/ R1 X5 q! W
$key = $this->getKey($req);, \/ j, O8 q" }% C7 ?
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";
/ ~1 b8 E- Z$ u& m' t
) U# Z: K0 ~1 X* o6 j7 B
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));$ V7 ]- e$ P1 e
}0 ~0 W6 y7 i( X. c! z
2 P4 H/ A% f5 n; W b
function getKey($req)
/ g5 {3 |3 m* i# T
{8 r9 ?) d. }2 }# G
$key = null;
9 d! ~# \9 L+ P+ H+ {
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) { ; U; t% l. {8 N, n- N# z
$key = $match[1];
~$ Q" E7 S; [- E
}
9 X) ~ d# }3 R y
return $key;$ l4 h0 y' d# h7 o$ A: @
}$ f, F: s5 ~2 W3 w- y
: ^) }: }& b/ K: a, u0 T. ~
// 解析数据帧- G" G* |0 z5 I7 S$ r* f/ ?' p
function decode($buffer) " U; i( s5 r; i+ U
{) y4 Z- s: o, N3 H7 z. y
$len = $masks = $data = $decoded = null;7 {# J5 a4 o U* i% i3 j
$len = ord($buffer[1]) & 127;( L+ T; D/ m" x- z+ y- |4 Z
2 F$ o" i/ ^& U. V2 E
if ($len === 126) {9 _- p& m4 E8 H8 c/ |
$masks = substr($buffer, 4, 4);
s( T0 Y" _# l% l6 Z0 X) N" [
$data = substr($buffer, 8);
- W( y0 B# O' T2 L
} else if ($len === 127) {: h0 D( T% `0 K7 G2 c: ^$ G ]) m+ }
$masks = substr($buffer, 10, 4);
; p) Z- }2 `$ b7 `
$data = substr($buffer, 14);
/ ?/ `3 t6 s- K7 k. Z- b
} else { A4 Z) ? g' G& }' L
$masks = substr($buffer, 2, 4);1 I, ?# ~# V$ [% h9 U J( u8 i
$data = substr($buffer, 6);9 L" h2 w# h" F; g/ ]1 p W' T. ~
}
3 ]7 D, s( ^: Y6 |% k0 [" f) j
for ($index = 0; $index < strlen($data); $index++) {# o1 H& p8 T4 P9 [
$decoded .= $data[$index] ^ $masks[$index % 4];+ k: P0 c/ O0 T9 ?
}7 g$ K$ W5 w. `6 [/ j0 I% }: t, |' \4 d7 o
return $decoded; p, h* G+ k; r* q* c
}
0 a- R% J# X4 H- k# [; v9 J, F
; }" F/ Q% `- [! T
// 返回帧信息处理
5 ?1 s. I. X: N1 ]
function frame($s) % G, x& L9 r; u( j" m( }+ T9 }4 s
{ y4 }& K* `1 U: N0 r; E
$a = str_split($s, 125);
6 J7 u" g7 Y- K+ u. Y
if (count($a) == 1) {
! w8 ?8 a: D( O2 F
return "\x81" . chr(strlen($a[0])) . $a[0];
: M2 c' x8 |1 v# h, S1 U
}/ S' O G* H, B* b* c/ L
$ns = "";) ^' z* z Z$ O! u$ i5 r, R
foreach ($a as $o) {( g8 [) A' c" w+ P4 y
$ns .= "\x81" . chr(strlen($o)) . $o;# O c! U7 l% E6 @3 e0 u
}
/ u9 `- p! v2 B+ x E. |
return $ns;
. p. o/ ?& A0 l8 o
}
# m) i/ X/ A9 C6 j
. ~# W: b$ q0 @- g* x/ {( q
// 返回数据2 Q) N% C+ Y( }) Z+ c4 p. }
function send($client, $msg)
/ J+ T3 i- H9 T: G4 V3 y, `
{1 y; z ?. |6 E
$msg = $this->frame($msg);. z5 R8 |# R! p& G
socket_write($client, $msg, strlen($msg));% O" Y8 j/ U! z9 \( J' P$ o* s
} l0 i9 S3 J, b5 H' o C, b
}
u% a |9 g! f# G1 w0 _& P
: |$ x" l8 e" Y) Y8 W: v
测试 $ws = new WS("127.0.0.1",2000);- W* W/ b: L5 z7 A" Y) m
: n9 `1 N$ {. _0 v

复制代码

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

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