PHP 简单实现webSocket

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

1）客户端实现

<html>
# T2 Y/ v! m& L0 R x4 D4 j
<head>
/ O0 w2 a9 q. \- K2 K+ K
<meta charset="UTF-8">4 s* H$ f4 O! u* [) j2 x3 ]- h0 V
<title>Web sockets test</title>
n b5 x" ~5 d4 `
<script src="jquery-min.js" type="text/javascript"></script>3 p1 `$ k7 F! y8 A, n' W
<script type="text/javascript">9 @8 Y8 y' a) D# V4 U, }
var ws;
5 g% u0 ^* K% y# x- z$ Y
function ToggleConnectionClicked() {
+ {4 t8 z4 t e4 v `
try {- A1 z7 |% R" [. J. b
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器
1 M2 A$ g# `* b9 n$ `2 _8 _+ G3 G4 }
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};6 V' }+ R: \8 E% r
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);};
- V! r7 v- S# y! J( q
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);};# ~1 D- a$ `+ n2 q( }/ X8 }9 l$ W& ?3 W
ws.onerror = function(event){alert("WebSocket异常！");};
1 X; |6 M# L+ d& G2 d5 g' h
} catch (ex) {4 O% B* K, D9 K: u* j# L7 I
alert(ex.message);
0 E" O" F( t% m$ f" X6 l1 v5 [
}
}- q4 P8 P+ d+ l* [( W( Y
};
" D6 Y5 O4 f* O) ?# k) g& o* y0 c
' y' T* f3 N3 X6 h5 E1 g! t
function SendData() {, Y$ K- v$ B2 H8 O4 V" J S
try{1 b1 {' h& S2 g
var content = document.getElementById("content").value;( f- K, N! G& J% m, S" T
if(content){
) h. B r8 O8 [8 i6 h4 K
ws.send(content);
z9 U T* P) L2 q6 _
}
6 d+ X6 j# k8 M- P* U, c. R
- h+ C, Y7 }- v# E2 j3 O- A% ~7 b. b# m
}catch(ex){
" b( [- O" k0 k6 J G
alert(ex.message);% u7 @# o3 e, @: K3 x& }; H9 M
}
1 m: F4 Y* s. S# I& @, n4 K
};
2 Z! G4 i# l3 t4 B/ P0 |0 s7 Z' n
# _9 E/ C) w4 c8 ^8 Y! t- R1 t9 {
function seestate(){5 L9 c& Y. \7 U+ S: J! c6 w, D5 G
alert(ws.readyState);2 `' q; K' ~* `8 o8 M1 H2 B
}
5 O! H! U$ @( P8 M6 o
, L, k4 N0 e# f0 O! V# ^& ^/ j
</script>
% c4 b* A7 r) c, [# u
</head>; H0 B( C& e* W6 n
<body>
- `4 f0 ]/ c# F5 b$ A
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br />$ s+ C3 J w) ]+ j; ~ N) f) i
<textarea id="content" ></textarea>, v& c6 Q+ b7 m( ~( V) I
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br />
& x* o8 [; a. S0 Y
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br />
" u- I. W0 N6 @, {" t. z
& W" Q4 k8 R; ~7 Y$ Z( h! N3 f
</body>
! J6 o/ K- L& ^- b
</html>& S! V. y* }( e0 C# Q

复制代码

2）服务器端实现

class WS {
$ N& k" U4 m0 O
var $master; // 连接 server 的 client+ o" e- H, O0 g6 l
var $sockets = array(); // 不同状态的 socket 管理
; p" V2 Y5 b$ z# E
var $handshake = false; // 判断是否握手( C j7 a) c9 i4 a
, ]. m; C4 H+ |' A% e9 Q
function __construct($address, $port){2 @ p# L. |; l( ]7 g2 [) A" A6 H5 l
// 建立一个 socket 套接字! V3 _8 J( t8 |( F
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP) " K0 m, B5 z" e4 _+ z2 b, b: M
or die("socket_create() failed");- h q! M- @% t
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1)
1 Z4 R/ |; s" M+ k$ ^
or die("socket_option() failed");
6 m9 W" g; J3 _5 ?* B, P
socket_bind($this->master, $address, $port)
; Q" h; u! O1 t# u2 O! b3 k
or die("socket_bind() failed");, Q: d# a, L1 h
socket_listen($this->master, 2) . e/ ]) A: q2 H- B0 C) l& `0 |( u
or die("socket_listen() failed");) U M* _& K. }% C l: U* ~3 c
: T. d4 [: }+ P: ^3 c2 b1 P/ V
$this->sockets[] = $this->master;
; S6 a. C, v6 f4 w: }* _9 V
: w; M6 L% {. n3 B1 n! R! U! P
// debug
0 M r* V+ M. q* _) }
echo("Master socket : ".$this->master."\n");
2 l7 I+ i2 O: p& s
+ E; ^! J! B% s2 a* {' A
while(true) {
- F: A. v1 C( R' |; R5 U
//自动选择来消息的 socket 如果是握手自动选择主机* t, i8 `$ }! M! g6 ^9 j" p
$write = NULL;$ A; |7 y% @( o0 p
$except = NULL;
socket_select($this->sockets, $write, $except, NULL); R% m& t* x \% K& o7 u
$ _8 w( y$ L, x K+ i7 O
foreach ($this->sockets as $socket) {
/ @2 L$ i9 B' [% ?
//连接主机的 client
4 J( h/ N7 H6 X0 a& y& C2 |4 {
if ($socket == $this->master){
- i+ X% R! `! }1 ~: v' [1 k: v! D
$client = socket_accept($this->master);
8 X% b" _! ^3 r: P) ]6 m
if ($client < 0) {
5 f7 x! }: t, u. n4 O, S
// debug
2 C, N- }0 q6 V# z
echo "socket_accept() failed";
7 q+ m! h: \* F3 K2 j* H Q; L
continue;. ^' N- J% v; m3 ?$ e
} else {0 O0 \! v' E5 f( b$ l6 u
//connect($client);
/ b! O: t% g- H# e6 D# u; l
array_push($this->sockets, $client);9 h( |. A3 H+ K& `7 Y
echo "connect client\n";
) B( d2 ^5 E" a$ S# C4 ~7 h+ v
}
5 M" _ `' _9 y- @# w5 k4 [
} else {& ?5 @8 O7 n4 O5 W0 S1 T
$bytes = @socket_recv($socket,$buffer,2048,0);
5 o) y/ k& |. g |" K
print_r($buffer);
7 [% l3 v8 [1 @& o& n% P2 D8 \
if($bytes == 0) return;
$ L! C8 w: G K0 ~2 a
if (!$this->handshake) {
* _4 b6 W3 S/ T% j. Y
// 如果没有握手，先握手回应
5 ^0 B% X: n2 C, ^' T6 n
$this->doHandShake($socket, $buffer);" w1 B9 {* S5 Y `9 h) L
echo "shakeHands\n";) Q3 v. v* B0 g1 q
} else {
0 T( p6 L0 \7 O8 _9 |1 u: Q l
( I1 W: [+ I; ^6 J+ B: t: r
// 如果已经握手，直接接受数据，并处理
8 [5 [. Y3 r+ o7 d, V; U
$buffer = $this->decode($buffer);3 @1 J2 ~5 k8 P& U5 ~6 J) a9 E
//process($socket, $buffer); x$ w. R4 w/ ^. z! [
echo "send file\n";
1 Z3 `8 ]9 Q1 A. B
}" v. P8 Y; i& ~1 g, F
}
) S* B5 }. x" P+ j: @
}
: J* U( G5 l1 x8 t3 H: k, g( H
}
- h/ Q3 M# z: V4 B. C( E
}
1 `9 \6 |& y8 x6 ]
" S" k4 }5 n5 _" E, l
function dohandshake($socket, $req)
: l, q, ^& v `7 @
{9 `; p$ ]) i5 N
// 获取加密key
$ w, u N8 x0 q! D, J- ^
$acceptKey = $this->encry($req);
2 N% p2 q% ~3 ~5 S2 K
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" .0 a |0 B* W( e( `
"Upgrade: websocket\r\n" .
Z7 |0 b: {2 `, Y# M) Q
"Connection: Upgrade\r\n" .
/ Y$ | E2 ~, q! b6 }1 }4 W
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" .
P# `* V2 e* F" G0 U- {) R5 B! [
"\r\n";
! `* p$ b3 }2 W( U) q
" P0 j) D& G$ F3 K8 ]8 B/ |
echo "dohandshake ".$upgrade.chr(0);
5 f0 n- E/ w* X+ v
// 写入socket
2 q; j8 |$ n% Q I/ @3 s
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0)));
) U0 a3 ^& v" _8 U3 f; o
// 标记握手已经成功，下次接受数据采用数据帧格式
2 S! D, b% q* }) }
$this->handshake = true;2 l7 ?# i% W# @3 u$ H8 y
}
8 m1 n" g# K- |( i8 x# X1 H
4 e# J. E8 v1 C
' p. k; o* H8 X; _4 m. n7 L
function encry($req)
% v' v! I& |; l+ N0 r
{& D5 S2 i" s& I5 ?( T, o% m
$key = $this->getKey($req);
6 H0 ~: b( L8 G: \$ x- A4 S& Y
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";0 O) R- Y# z; N% \8 A. o4 _% p! p
/ @. t8 f7 B" B/ M3 Y, Z
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));! e. v; M# x% g
}- `' z; o6 t; q# g0 k# Z
* J* H& r- ^1 ~. E2 l2 d* j
function getKey($req)
$ v+ e4 n9 e- r, y0 L( f
{
( _, t* | Z; ?3 C
$key = null;
& M6 c. N) G, p- p' d" e- {
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) { . ~. p1 z1 |8 _1 ?& o3 c+ B! D% e) b
$key = $match[1];
! @: C5 W5 ?' G- J' Y8 r
}
' v) c O- x; ^2 B
return $key;
2 q4 V* r$ o7 `/ @, M7 Y. t3 C
}1 r5 g3 [! J9 n# p. B% n. z4 I2 I& ?
7 ]) }* U0 E% R* q0 @) p
// 解析数据帧
8 }& u7 A& J# o9 m& u$ D; m$ l
function decode($buffer) . [! D) Z* B, n" r
{" F; e0 J/ @" C; @( _& p
$len = $masks = $data = $decoded = null; R: Q* m, i; [* p- I; X# }
$len = ord($buffer[1]) & 127;9 V- M- a" G5 o" O2 {
6 w& ^" c: F, V p- e3 v
if ($len === 126) {
# j% C, G% \; [( d- s k, ]* U
$masks = substr($buffer, 4, 4);: F8 c/ F. s; |5 h6 l2 n9 i
$data = substr($buffer, 8);2 R3 w, R, R9 Q) @" @% b# ^# _6 F2 ]
} else if ($len === 127) {
" W; Z b: M8 G! s% F+ F4 _
$masks = substr($buffer, 10, 4);
; t8 ~. H- W6 j0 n8 ? r( }
$data = substr($buffer, 14);" n! ]7 _. Y# {
} else {2 W1 V2 J9 x3 F' J B3 P2 w
$masks = substr($buffer, 2, 4);5 `- H2 F) e0 D* d1 ^9 O7 C; P
$data = substr($buffer, 6);& D, x3 S% p) E) q7 d# _
}. ?, A4 U; \$ v1 f% W7 O; I
for ($index = 0; $index < strlen($data); $index++) {
2 v8 c+ d N$ A
$decoded .= $data[$index] ^ $masks[$index % 4];
7 Z( c9 |! t5 L* `$ r( j1 [
}. y8 F1 O' {8 l! P6 {/ Z; C7 s
return $decoded;
6 m' y- Y# u* Y
}# X4 }% E( R1 n# X. t
+ S$ x# ]# |: F8 C" b3 `
// 返回帧信息处理
$ a' y$ a% `0 {( f! j) ^
function frame($s) & E5 S& g) D7 r( B. I6 X; W" H; R7 J
{$ j( L2 c3 h& {$ j6 e
$a = str_split($s, 125);
7 b0 ?5 o) p; R8 b# P6 _2 p
if (count($a) == 1) {2 \ m2 K7 G+ R5 X( S
return "\x81" . chr(strlen($a[0])) . $a[0];8 m* r% Y3 Z( ~! M7 k0 R
}
9 j0 [% O& T+ Q$ l1 X- S
$ns = ""; X! g S- r2 C9 K5 u
foreach ($a as $o) {& L1 N' f# s0 D! F$ ?/ f: C
$ns .= "\x81" . chr(strlen($o)) . $o;; z+ ^+ U) C" R8 d6 e
}
: c0 A; D* E" M8 v7 e+ _- e
return $ns;7 A6 p6 H1 g; Q/ V# f6 @
}
" F$ |: H4 R% n9 v! l: G
. d7 o: q8 p7 b1 a
// 返回数据# ^' t7 j4 w& H' m
function send($client, $msg)- x0 W. e! z+ Y) e T% t
{/ t- {/ o: f$ Z: |# z
$msg = $this->frame($msg);! V' {4 m' Q' S7 }# M0 V$ y
socket_write($client, $msg, strlen($msg));. |6 L, Z' V( G5 r1 q8 v9 r9 S
}
/ m& z1 q! ]0 Q: T: o$ d
}
3 s# @7 Q2 j7 Z: \+ e1 E
/ [- ~& @% W, Q; Z% C8 s
测试 $ws = new WS("127.0.0.1",2000);
3 F( w0 b7 t' h% }% O* H' {4 W
0 \# m/ O2 f/ j5 R

复制代码

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

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