PHP 简单实现webSocket

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

1）客户端实现

<html> ? S1 ?- `! V9 C
<head>
) p% H# T T* I& g
<meta charset="UTF-8">
: \8 g3 i2 y2 }. F; V1 |
<title>Web sockets test</title>+ V# c8 [: \; ]6 B
<script src="jquery-min.js" type="text/javascript"></script>/ m! z$ p) l6 n
<script type="text/javascript">
, X0 R9 D' Z4 q- c, f' ~
var ws;
* Z$ ^& _% G/ h. {# s# B
function ToggleConnectionClicked() { ' o% j3 f( c2 a* p$ ]7 E
try {2 ^4 z7 }) @$ r. Q7 A
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器 ' \; x l) c1 G& c. ^' g
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};
. ?' |" o1 }5 u. j: v" \% ~
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);};& G* ?! _# Q. y
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);};+ P: v" w3 Z- s
ws.onerror = function(event){alert("WebSocket异常！");};
0 I' A" v6 g. O$ [3 E/ N# C$ A
} catch (ex) {( M. B$ G. w: @$ V# K
alert(ex.message);
. T) }# Y2 w \% t
}5 q% `4 Y1 O- [+ C
};3 i! q. `2 K5 i
4 d( q _( c/ a4 |$ W( j
function SendData() {- F: p5 W- h! ?4 l
try{3 F" {/ U# o8 K, @' b
var content = document.getElementById("content").value;
3 Q# Q1 ^) j+ R
if(content){: g/ l+ Q) {7 ^& X
ws.send(content);
1 k$ i* `) j' }. I2 Q! {( E% a% }" k% D
}
# E: X, I u& X6 I7 c. F& }9 E
3 n9 ?3 ~8 g5 x4 F8 c; ~& }
}catch(ex){6 l, }4 A" O' m5 Q4 K- m
alert(ex.message);8 {* }% G( a8 `( M8 T+ w9 n/ t5 N
}
/ m i2 z* U7 i( B2 t/ j) k
};" V2 V9 ?( S5 O: q6 n
) ]/ E5 ^& k9 }3 W! I* g2 P! D
function seestate(){- [" w4 z( l1 h7 K6 s& G# D& D
alert(ws.readyState);0 b) S* ]! \: F
}
2 ^% ^' H4 h! V
) I$ D! N% g& \2 t8 g
</script>
" z# {$ W0 D0 ^
</head>! P( O3 ^4 l! D
<body>. ]$ o2 M1 V6 \8 F( s
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br />
( [6 S9 y Y2 k5 U9 U ~1 c- z
<textarea id="content" ></textarea>: h* H/ `! |* J
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br />
! e; e4 q$ T1 u7 i& |: S0 e
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br />+ Y9 r. {9 V8 {$ o6 c- L B! R, E
2 i, P$ C, F- @& i& t
</body>
& w1 x3 f5 J1 _' _8 \ g* R+ ]5 ]
</html>0 z2 ~* d6 l3 J( h& P, j

复制代码

2）服务器端实现

class WS {
5 y! d: @' V3 T! y# w
var $master; // 连接 server 的 client( h4 z; N4 _: c( Q, O& q
var $sockets = array(); // 不同状态的 socket 管理$ Q# `% I6 k! X8 b7 _/ ^8 j
var $handshake = false; // 判断是否握手
: F9 u+ E$ i3 N+ u; a! w
- e9 t& Q1 B- x; Y% \
function __construct($address, $port){# m3 ^/ {. l) x1 d1 c" L: ?2 q
// 建立一个 socket 套接字
5 v+ v" K; ^6 n, _5 ]- L
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP)
" _' h8 J( f* v' |4 _: z1 f
or die("socket_create() failed");
% E. I& U; m O8 t
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1)
& @4 ~/ K8 s" S
or die("socket_option() failed");8 ^, n& S5 I$ Z& V
socket_bind($this->master, $address, $port) $ X# o& X& T# f$ K0 {. F/ J
or die("socket_bind() failed");/ u( T. t! g- ]. k E i
socket_listen($this->master, 2)
- i; F( v* x- K% y
or die("socket_listen() failed");
" P3 I/ p" V! `; I
6 E y* z; f* L! B
$this->sockets[] = $this->master;! C) z5 b, c& _7 {& h9 P
& U; m& C& g. ]* D5 s4 v4 G2 r6 {
// debug% {: O7 Q0 }- ^( Z( M; R+ S! d
echo("Master socket : ".$this->master."\n");' v' J: ?) L1 j E, u) ?. v
+ S* j+ K4 X% X4 {% @& g3 |! n4 u
while(true) {
, h: l" w& k3 S% B6 `# g% Y5 S9 H
//自动选择来消息的 socket 如果是握手自动选择主机0 ?* o" u0 b, k( i6 E3 q$ C% f
$write = NULL;
" T" S& q) q( }2 R, \ Y
$except = NULL;
1 L( Y" y& }9 i6 N. @- I
socket_select($this->sockets, $write, $except, NULL); ?8 W! V% ]- E5 ?; }) n7 f. u1 H& {
$ M- P( T5 k$ a
foreach ($this->sockets as $socket) {
6 W- e# s" Z/ ]) O/ Q1 {: n( L& D8 }
//连接主机的 client ! c# o3 V; T+ q( c7 E
if ($socket == $this->master){, P. O6 g" c' [: Z
$client = socket_accept($this->master);* f) R0 y! a& v7 o/ f& l2 B
if ($client < 0) {
0 j# D8 e3 a$ f& P
// debug
% r# c! i8 c5 A% ]" b `! o
echo "socket_accept() failed";8 ~( z, v; O1 D
continue; F1 o: G* ~% Q: i# ?4 ~" q- }
} else {( A: A; t" M. v+ t+ N
//connect($client);
8 `& f* u2 r2 b8 I* p& j/ Y9 p
array_push($this->sockets, $client);' ]$ X' _/ B7 L& r6 U3 _8 w! i# j
echo "connect client\n";
* c- T# N6 q1 }1 z V9 W' z: Y% D
}4 O, b# [4 r8 h v/ x4 l/ _
} else {/ S* M# E9 w3 E& j) X+ ?
$bytes = @socket_recv($socket,$buffer,2048,0);
% M; g( ]3 O" \
print_r($buffer);
9 h5 b1 A; ~( j$ s2 s, _& [
if($bytes == 0) return;
8 q' z0 x& u5 A. [
if (!$this->handshake) {1 C" Z& f7 }- B6 i( j/ n" q5 u/ ?
// 如果没有握手，先握手回应0 o0 o3 t" z; a) A q3 u
$this->doHandShake($socket, $buffer);$ |4 a- B8 W8 d: v+ \# d
echo "shakeHands\n";
; Z, ^, o+ x! y7 Q* P, F* Z0 W
} else {
h" I: Z; Z1 g: P& f# U
1 u2 f9 U& T) n8 }. G: B
// 如果已经握手，直接接受数据，并处理8 W3 s4 L. h& Q( H% o
$buffer = $this->decode($buffer);! S1 I8 q4 p% n4 p( a: H; ^7 ^
//process($socket, $buffer); ! |( q+ y; f" Q0 f
echo "send file\n";
! ?& j2 J0 J9 A
}
+ J' f! {. P8 x( |( ?3 Z
}
7 x6 o# Z$ J0 P0 W: `, m1 O' Z
}" p; c; y! j" s8 j& ?7 @ D! _" @" ?
}8 {- I9 n. [) F; c
}
8 ^! _8 f9 v- f
% i: k) w4 L4 `" p
function dohandshake($socket, $req)$ E( |. b0 O" {6 u% j k( n! A
{+ W4 P* X9 D/ k: v1 J
// 获取加密key/ e) r7 L( s( n9 E5 r4 S* X
$acceptKey = $this->encry($req);2 b! U3 |. D, t' Q$ e- O) L5 p
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" .& R( I7 R4 x( |* Q" i" M- y
"Upgrade: websocket\r\n" .
" ~) L! a# W6 U; n4 Y
"Connection: Upgrade\r\n" .5 ^8 A& ?. C/ _- @. J: C
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" .
* \2 S+ ~" {- V# h. {" r& ^- q; _% a" l
"\r\n";& g+ P' e! B% Q- N
& v: F% g7 }. m( j& F
echo "dohandshake ".$upgrade.chr(0);
7 W. y! I& z/ C, o8 {; t. ^. Y
// 写入socket
: V9 t; ^+ _0 g5 l3 g% W
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0)));) J2 d7 O; {! Z$ Y
// 标记握手已经成功，下次接受数据采用数据帧格式
0 U, r4 [% y' U0 G8 u# R
$this->handshake = true;
) s& P9 G4 m# Y, b# ^
}
2 T" G2 e+ M' ?( b/ X7 a" X
; @( E. A8 T* z1 T
% H- X5 B2 G% o
function encry($req)8 B/ |) E; j9 I l6 Q
{
8 ]5 |0 ^0 _0 f! W: p* y7 V
$key = $this->getKey($req);
% g a/ Z, z9 T3 \, d
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";
+ h. Q7 R- E8 I- J% Z
- s5 L% {! Z g" m3 d
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));
6 g% }6 ?* o, K
}. `0 b9 x3 ~5 w0 G& j1 m
6 j- V5 G. P. p; I' g3 k8 [3 n4 e0 d
function getKey($req) 4 I, n3 s5 b( s" ~
{4 G) l: M2 ?; n1 z2 `8 p* K
$key = null;
+ R d; v9 c; C: D5 c& {2 W
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) { 4 A5 A0 u5 }0 P% C6 i' u+ l+ _, |
$key = $match[1]; ; ~* g. l+ }( \* N2 u* q
}
^0 t) q: l4 o: ]7 M# E" S
return $key;
5 w2 m7 d" W2 a$ @/ N) r( `
}
, ?" u1 g$ ~, D5 e/ R, Q
# S0 Z; e# k8 w! Q% q/ ~1 A
// 解析数据帧
0 k" t) n& ^ Z! I1 @% H
function decode($buffer) / e/ l5 S7 O) X w" W* u8 n% s; e
{
! S, j, n( T) W# W1 `
$len = $masks = $data = $decoded = null;* j2 u+ W& F3 \2 ?$ x
$len = ord($buffer[1]) & 127;1 w- [* [7 X4 c" n9 Z3 n
7 z* y7 ]: B- e- ?& V6 N
if ($len === 126) {
+ E1 m% ~: o( r( ?9 X9 l9 c
$masks = substr($buffer, 4, 4);
8 D, n$ P" w* l* ^1 e
$data = substr($buffer, 8);4 l3 J: Y" a% r6 Y
} else if ($len === 127) {
" k" q* {& @: M9 o6 c+ d, ]& \( K! f
$masks = substr($buffer, 10, 4);
% c3 J l# `! e6 h2 e# T
$data = substr($buffer, 14);! i: l1 d9 a6 A2 U
} else {
, P# l2 M; J( d# p
$masks = substr($buffer, 2, 4);2 K$ Z. C: j/ F, U
$data = substr($buffer, 6);6 ~" H+ t/ F( F: U# I- e
}
& N* C' U& |7 Q" q: R8 i2 E" Y
for ($index = 0; $index < strlen($data); $index++) {
( y/ _* f" u2 U
$decoded .= $data[$index] ^ $masks[$index % 4];/ n/ o: Y; n: s# ^* }
}4 \( v( q2 y* b7 M
return $decoded; L0 C" I0 \6 P' A* {
}3 k% C$ ~1 ^5 U
" q$ `2 r; i$ ~7 ]. r7 O! g% [
// 返回帧信息处理) E' ]( B2 i# S, h9 N
function frame($s)
* y$ m: o: f' @8 k( y9 R# e" f5 O" x
{
5 R- X* g- o8 Y$ i' m# _+ R
$a = str_split($s, 125);6 \' ~) D2 X7 P& [; o3 Z4 @
if (count($a) == 1) {
- [- [0 [# N) Q. ?. ]$ b
return "\x81" . chr(strlen($a[0])) . $a[0];* L$ U8 l! Q' y A+ G! m# e
}
4 F, r& L5 Z" y& \$ T( n0 F8 }
$ns = "";. ~+ {, v$ q1 Z0 x8 u
foreach ($a as $o) {1 Z6 k$ B8 q# D! f2 Q. n7 n
$ns .= "\x81" . chr(strlen($o)) . $o;9 D' V8 f! B, U: K. E
}
0 ]/ ?. b2 P! Q$ l( v
return $ns;
7 D3 t- {: q9 K
}
. ^4 B3 @9 h# i3 J# A# x1 n
8 P; D. ^/ ?0 [$ ` M3 I
// 返回数据9 y+ e! k: ?$ S; o$ C; n/ M
function send($client, $msg)
; |( n! M$ Q/ [+ j# G' K
{/ O( k$ h( s0 k$ E
$msg = $this->frame($msg);, c% a3 d; d7 i5 _$ B/ H) F& w- z
socket_write($client, $msg, strlen($msg));
5 b/ ~ L* q1 l
}- D z0 V B& ~: u
}
6 [$ l( C6 W7 b4 @: N2 a( ?: l
- z3 M+ S0 l( T: p# |/ e
测试 $ws = new WS("127.0.0.1",2000);
* b+ r6 d) b5 J2 W- N. y
" q, G+ g' W) D B: M& _+ |5 F: \' X

复制代码

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

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

浏览过的版块