PHP 简单实现webSocket

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

1）客户端实现

<html>
0 n; \) G+ \0 _8 o# ^/ [$ ^
<head>
0 e3 a- E% H& S' X, P$ z
<meta charset="UTF-8">- X" ^4 Y1 V0 V; v
<title>Web sockets test</title>8 |8 N; Y' t) Q( h
<script src="jquery-min.js" type="text/javascript"></script>5 c/ b d0 }+ z# K8 ]& l4 \: t6 C
<script type="text/javascript">
/ U& p7 U0 K4 h$ G# O! \/ W/ _
var ws;
! B7 k) h- D) l* R, V5 H
function ToggleConnectionClicked() {
" @; G5 }+ U. M; b# i: P. I) U
try {$ i' X( E7 ~9 D4 B# @
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器
; Q& d7 _+ h* Z
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};
, S2 Y) b- x& p4 \
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);};9 a: ~% D6 j M
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);};
2 T( {9 v g, }/ P' l7 a. _
ws.onerror = function(event){alert("WebSocket异常！");};0 s3 l3 W; ]1 Z1 n! T* z
} catch (ex) {# w* _/ h, K3 c6 U
alert(ex.message);
6 u K- N* @/ I4 x7 p( T3 r
}
* b# A5 r, E, d3 J( M' [2 p
};
" f3 i8 v. z, B3 x6 U T
: q# g; B4 Q$ I, b+ E
function SendData() {! x& P: J! m9 J3 ^' G
try{# Y2 M! w# Z# w% [' z
var content = document.getElementById("content").value;
6 y5 W1 q0 J. R- h
if(content){
: |+ V' l( P [) s+ \! M& v
ws.send(content);
. W) Z e; H' Z- M
}
: Q' n0 p% z4 @/ ^) d; Y
, }' `5 d/ u' B/ G! E3 M
}catch(ex){8 C( N" v1 {' X. m/ m9 z( O# ~( H( ?
alert(ex.message);/ s7 d3 ^7 h- w# Y8 X+ d1 s
}+ B: |) v4 B8 t w; `3 O+ Z
};/ K5 U& S% J" p% k# W. r
9 y. M+ U9 [5 G( O
function seestate(){
' R. I! l# J5 O, t- W; L+ @
alert(ws.readyState);
) P$ Y# A* E+ K+ }- N& D% c
}) a$ [5 O; v9 H5 X
+ Q& O: x9 \3 k# E5 x) g. U! _
</script>
1 k; c$ ]+ q K+ ?3 b
</head>
& T+ D+ O4 f9 T9 u/ h/ B
<body>
" S. e3 h4 _ z" @) `2 i
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br /> T4 g/ g. P$ q
<textarea id="content" ></textarea># m6 E2 h1 R0 u$ |
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br />( \9 L' M1 r, f1 p
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br />8 ?* Y. W+ s" x! T
3 I9 S+ P: e/ f6 z
</body>% j5 `/ w7 r6 {" {6 W1 U
</html>6 f% X2 H- h& d

复制代码

2）服务器端实现

class WS {$ f2 E; K1 h# o- x9 v2 A
var $master; // 连接 server 的 client
6 ]- O8 ]3 w7 Q( X9 `
var $sockets = array(); // 不同状态的 socket 管理
4 W- n+ {; [2 m5 X: Y9 M+ c
var $handshake = false; // 判断是否握手
% _, }- m0 S- Y3 D! H
& |& d" \7 H& p$ W4 q2 U
function __construct($address, $port){, l8 h5 }4 V( c: T$ a r- T% |* \
// 建立一个 socket 套接字
' U" g" u1 i: p0 p' |% {: {
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP) ) T8 z; W( x8 ^
or die("socket_create() failed"); x% r% n, W- @; [* i; v9 f
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1)
+ D+ N: U" Z# G& J: _
or die("socket_option() failed");, N- R" w; _ k2 M) V2 S4 r" I
socket_bind($this->master, $address, $port) ! D) D4 b$ J$ e2 I( Q
or die("socket_bind() failed");
2 W* [0 `2 P, Q+ V* {
socket_listen($this->master, 2)
& Z/ T5 T# A* y6 {! }1 A
or die("socket_listen() failed");
" _9 V* v2 s1 v0 N% ~1 i
; ?, Y9 t9 x& z1 S v/ U
$this->sockets[] = $this->master;0 O6 S" K; t/ E
3 E5 ?( `7 w/ G+ e, T, H: D) \
// debug
) D) m* l7 J% q$ g4 {$ {% E
echo("Master socket : ".$this->master."\n");
1 R' n8 H( Q, |( Q6 q& K
8 c; c0 G3 T( i
while(true) {
1 u: L2 R! R' Y7 v% i& m; e
//自动选择来消息的 socket 如果是握手自动选择主机6 u* U! _0 h; C7 g! x
$write = NULL;4 D: i. e7 o5 W: ^ G
$except = NULL;
' H2 C, x0 @% I
socket_select($this->sockets, $write, $except, NULL);, H, M) k1 n r) L7 e5 C7 t% r
4 ~% x2 u7 E; V9 T( Z
foreach ($this->sockets as $socket) {; Z; u3 W3 {! @! K u" x
//连接主机的 client
; [% f# F! S x% |* t( Y. Q3 ]# B
if ($socket == $this->master){
0 ^! y e4 o$ {
$client = socket_accept($this->master);, j2 a* d* [5 z. z5 R" v2 W- A6 I
if ($client < 0) {
8 v7 x& Y( y/ @0 w; c
// debug$ B0 a& ]6 u0 Q
echo "socket_accept() failed";- M- Z6 R% ]* [; u+ X
continue;4 w4 s9 W1 \2 U) O- k) J
} else {
4 B H1 O$ q6 _0 s3 E+ |7 \9 B: y
//connect($client);
9 k0 J% R4 [9 o8 B% ?1 S
array_push($this->sockets, $client);
' h" @+ B( K" y3 C8 F" w4 k! \
echo "connect client\n";
* g1 y2 q0 F% ^" L
}/ O- y% d$ P7 |& v! o X$ J' u# D
} else {, V* n4 U x5 Z! I3 }2 h5 u, K( P
$bytes = @socket_recv($socket,$buffer,2048,0);
+ L1 N/ H0 P; V* q( Z; M: r+ K
print_r($buffer);8 P) k2 K0 a5 B( a4 Q
if($bytes == 0) return;0 }+ u' N! ?# j' p
if (!$this->handshake) {( Y. c' o( ^, c) o+ u* G; f
// 如果没有握手，先握手回应. S4 f s& P* D* R, m2 r
$this->doHandShake($socket, $buffer);
. t# j; j& M* M) _5 _$ o% y
echo "shakeHands\n";
+ [9 y" F$ n, w6 i! [) m
} else {( ?2 R# W9 e; @8 Z
/ @+ }2 [1 ~9 |/ j; k4 M+ u& Z/ a
// 如果已经握手，直接接受数据，并处理
, D, N/ j( l d* C) @
$buffer = $this->decode($buffer);
2 w4 {2 s5 Q' ~; x
//process($socket, $buffer);
- v; d4 q6 J8 x( \' c) t6 P4 t
echo "send file\n";4 k) j5 ?. D4 P3 |* V
}3 E% s/ h6 k9 Q' D
}+ e. L X- [/ E0 ~
}
% t9 ]; d( O' M1 P1 x% l/ t
}' N" n0 d" K! I) n
}3 B; d3 j- a5 i3 U4 L1 a
& I8 r1 P8 f9 h* s% B' j b3 O
function dohandshake($socket, $req)5 y" h9 w/ x l( E5 ~% P( v
{
+ @ z/ U. X# X" k2 O0 h
// 获取加密key+ A' F; \/ P' Z7 x: S8 [
$acceptKey = $this->encry($req);
9 J* @7 V& f/ F0 ^6 o
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" .' I+ I8 f# h# Q5 ~' F; J. N
"Upgrade: websocket\r\n" .
' k$ j. u; U* I7 I/ a( y1 U
"Connection: Upgrade\r\n" .
: @( }/ h" M u" x/ l
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" .$ ~2 b! N. _1 ~; ^% s- P, f b7 ^
"\r\n";
9 _! i" v0 K% d$ c$ o
& R5 Q' K- {4 x4 J9 S
echo "dohandshake ".$upgrade.chr(0); 2 w; {( \1 {( I
// 写入socket+ F$ Z3 G7 W* l
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0)));2 Q _/ F K6 j2 l1 u
// 标记握手已经成功，下次接受数据采用数据帧格式; m' ]3 |9 m2 V% w" _8 b
$this->handshake = true;
) ~. b. q% ` t0 m
}* l. j+ e I: a* P$ b( Z- P7 B
: G: Q2 @# |' L& p
5 b$ @: P( j( V* l) ?
function encry($req)" l0 o3 q2 I6 P0 ]0 {! L7 y, @
{# [9 h7 N6 O: t( W& G
$key = $this->getKey($req);
1 E0 n% [( M% b) ^0 m9 V, j0 X
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";5 x6 Y- \0 N0 s
* T& S/ w3 ?( j* }7 e! f) q
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));/ N# S8 k5 k5 \' a$ m! b
}
6 v, z: K9 {5 w q
! v) V' q/ f8 i" w8 ~: v
function getKey($req) 0 ]. i, T* m3 V3 x+ l
{. `. K. f( k1 q- O& ^8 P/ ~
$key = null;
$ @* g- }9 D M3 Y1 f
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) {
4 N6 U* O- m8 d4 s# f/ W* m' V
$key = $match[1];
7 X# O$ Y! D: n& s
}
# \. Q' I6 c* i) q- T. d8 Q" Q
return $key;) h) E4 ^: R) n* i4 U
}# k5 a6 e- t+ I$ p q9 f1 l
/ ]( h& X$ o8 V- z
// 解析数据帧% E7 G* J8 E5 X
function decode($buffer)
4 b% _ O. D4 Q7 ^* H2 P+ }$ q
{
% X$ w, K8 u2 v, B: T+ {
$len = $masks = $data = $decoded = null;2 |1 N9 W( @0 t6 A" O3 ^9 @: s$ ^
$len = ord($buffer[1]) & 127;
9 A6 Y0 B; f- [7 q% _1 i
/ P0 P& p% ?1 }0 H- X8 p% m3 t
if ($len === 126) {
) s0 Q) y+ |" U
$masks = substr($buffer, 4, 4);! H2 a) F7 @: T! {7 `- F) u! q3 H
$data = substr($buffer, 8);
4 {8 f' z% f6 ]6 D( |
} else if ($len === 127) {; A9 H: x3 ?! u3 d3 f4 w
$masks = substr($buffer, 10, 4);2 s. p4 } I6 F( l6 c9 S2 o% I
$data = substr($buffer, 14);0 O- X/ ]) j0 ]1 O! y$ T1 J' _
} else {
( }2 {' b' g& a# k) ?3 z& W& T
$masks = substr($buffer, 2, 4);
& w8 w" Z2 ^5 t
$data = substr($buffer, 6);4 ~% ` m/ x i
}
7 r, d/ l5 }$ R3 F6 G
for ($index = 0; $index < strlen($data); $index++) {
" w# m4 g& t5 T1 L0 e
$decoded .= $data[$index] ^ $masks[$index % 4];" u; s, {8 w) J3 l; b$ Q/ f
}; s9 v) L5 E; f8 h
return $decoded;
* C; c% Z3 d& s( r) L3 b1 D7 Q9 W: u: S
}+ C9 |$ `; a7 ]9 X3 }7 j
9 x1 x& M! \; F- {
// 返回帧信息处理
: \; s9 o9 E: p$ S- i
function frame($s) 3 s2 ~2 |# x& _
{" n/ ~: k+ @1 M8 C4 ]( r6 ^. w
$a = str_split($s, 125);8 w2 A# `3 v) @9 h2 @
if (count($a) == 1) {
7 ?, j8 u( E5 m8 Q# K
return "\x81" . chr(strlen($a[0])) . $a[0];1 U7 u! R! O5 y+ f1 l) O
}& ~5 p2 x: k e2 _0 q
$ns = "";
! T% s' |3 r- x0 `3 E( V" B6 H' B
foreach ($a as $o) {( [/ k! |/ |( b! V% y, |9 H2 L
$ns .= "\x81" . chr(strlen($o)) . $o;. `% B% B2 ]! S" ]& i+ c1 Q' ]
}2 n1 {3 U* ?* i4 p% d$ u) H
return $ns;) p5 V8 e# I# q* Z
}1 j) B; B6 |8 Q6 u+ L7 `" o
6 k) T) h. E Y
// 返回数据3 j% x% Q, ?+ f& g: ?$ f
function send($client, $msg)
$ y5 ?9 B* p+ j
{
8 Z6 @8 `) c0 [
$msg = $this->frame($msg);4 V( n. o2 a6 Y* z4 V1 t
socket_write($client, $msg, strlen($msg));
6 V4 A: i& o3 u4 q/ `4 G
}
# h7 W9 b- l1 `6 @
}
- `$ i5 [0 E$ @$ L
* p- r+ [+ P( }* W$ z, K
测试 $ws = new WS("127.0.0.1",2000);
! L! ^0 L: x Z4 `2 B
v% r- v! R5 I8 }- C( e- V

复制代码

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

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

浏览过的版块