PHP 简单实现webSocket

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

1）客户端实现

<html>
; {" o( v1 C o2 A4 I$ l
<head>
5 u% d% z, \7 x' I
<meta charset="UTF-8">
# ?7 }% G% v6 H ^& ?
<title>Web sockets test</title>
* @/ D3 x, L1 d7 ^( E' @6 A; z" f& \7 |
<script src="jquery-min.js" type="text/javascript"></script>7 p) W% L! p. i1 I0 h9 ?
<script type="text/javascript">
. F3 [" J( M O+ H' h$ r
var ws;
, h/ k6 Q$ V& B1 u3 J9 d3 j* Y
function ToggleConnectionClicked() {
) M0 O- I! o! q7 m
try {/ M% @/ b* g+ y
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器 . a! F' i" f- z0 E
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};6 P2 T7 H8 h e: a* A& H$ }8 V
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);};2 G" \' u8 S' r0 y7 S; K9 q \* J
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);};
# w# v n0 B/ v9 ~
ws.onerror = function(event){alert("WebSocket异常！");};& E- l6 {# p# o ^
} catch (ex) {6 C$ J8 x+ _+ c% Z6 J
alert(ex.message);
1 x$ |: i; h/ n; [3 r$ L
} V) V- p; M0 M' R3 a7 |, H
};
6 m$ T6 x* B6 D! _
- z' u! k( L1 H0 ?! w& ]7 i5 X% Z
function SendData() {5 ~" p7 w/ q z T6 f6 t7 f& H
try{
6 m" V3 }/ w3 e: m3 u& e
var content = document.getElementById("content").value;
/ U7 ^ [: ]1 H
if(content){: |4 F8 k( p6 c2 E( G- _
ws.send(content);# T) e* }, ?/ Y
}8 X4 d' t o' S1 e" y) }
]* S! |0 m0 a( I4 h
}catch(ex){
9 B! O2 y/ X. Q3 J. ?! Y# ~
alert(ex.message);
4 t6 B h6 V8 f: X& v) L1 q2 d3 s
}
! ?: v7 x: e$ I' V/ ~
};3 [, [& M6 Z J y W
& x& L0 Y) U5 o% e0 u' a5 k# n
function seestate(){3 r- l- g/ Q% e
alert(ws.readyState);+ d1 K' I4 P' `0 j$ f
}" N6 s$ n, S0 {
$ Z# B' w9 G# H1 W$ h# q5 f7 v
</script>
' B8 }/ [0 f* ?( u9 r) N
</head>
' L! c2 b& w9 I7 ?, ^( x/ d
<body>/ F1 _& t$ z7 K$ l0 ~
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br />
. k$ y0 s0 P' |
<textarea id="content" ></textarea>
& B8 F' v' U d
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br />/ \* E4 p j8 ], u9 [1 K& `
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br />
' n9 i( B7 }. t, `
# C+ e- j2 n, |2 f: E
</body>
2 t% U9 h$ W3 y" p; p$ S
</html>
" j$ N0 J4 x; A+ e8 b. P

复制代码

2）服务器端实现

class WS {& q7 Q$ x& \( d/ p3 A
var $master; // 连接 server 的 client8 a" J. }8 @. k8 e8 n
var $sockets = array(); // 不同状态的 socket 管理
2 O$ z U* p3 S; L( r/ `7 R! u/ J
var $handshake = false; // 判断是否握手7 H0 i. e, t) [8 o( [
6 C. B1 Z( h$ }. w
function __construct($address, $port){& V* C0 s" J; f& K
// 建立一个 socket 套接字% i- J5 B( Q: o' M6 ^9 ^
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP)
' i$ b3 _3 G. `3 O% R+ i( Y; B8 |% U
or die("socket_create() failed");
- S0 ]: c2 t& s. [, g0 S
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1) 8 B. k' h; G& _6 O+ u* {9 N6 I
or die("socket_option() failed");( `7 H+ C u3 g2 c7 u
socket_bind($this->master, $address, $port) . z w) s! E( Y, o1 H3 j/ S
or die("socket_bind() failed");
T8 O* [( u* F1 b5 p% @
socket_listen($this->master, 2) 5 q6 g _7 _8 Y/ z% U: p
or die("socket_listen() failed"); k& g# p& d% Q M) R) L5 u8 ^
/ \ \8 v9 L$ n1 T6 R+ k
$this->sockets[] = $this->master;- G/ `( x! D2 y% {$ o0 T
! u6 I' ^- L# `. P3 A r
// debug7 G* x+ H* T8 i- O0 Z, O5 U
echo("Master socket : ".$this->master."\n");
/ f& k0 ]0 W* R. `
0 V5 A" F; ~8 \7 f/ M
while(true) {
& ^( o) M) d2 }) o- z3 ~
//自动选择来消息的 socket 如果是握手自动选择主机1 X7 T. I9 {( o
$write = NULL;
' q! u6 z- C+ u4 K0 b8 T
$except = NULL;& q: r, l4 {' G/ k" r1 L. _
socket_select($this->sockets, $write, $except, NULL);. ?! ~2 H7 P) ?1 B( s
1 R R( P# x% n3 a7 G
foreach ($this->sockets as $socket) {% o3 N1 `- p/ k6 p
//连接主机的 client # k+ a: c. m' n2 _9 L, H: C8 ~/ d
if ($socket == $this->master){$ }: R. x; H% t* |9 v
$client = socket_accept($this->master);
3 c, t9 w5 ^ _1 G! J- r0 C
if ($client < 0) {
" J$ Q% m ~, }" b8 o
// debug
* t% e5 w- }& I8 u
echo "socket_accept() failed";6 b$ z8 f0 K+ |" R9 t4 e1 K4 ?" W
continue;
' | r0 U" ~1 U& y& i
} else {' P. E* H6 H" _( L2 e9 R
//connect($client);
- w+ y3 @) X8 [) ?& A& J$ u/ C
array_push($this->sockets, $client);
( J# I A9 U3 W
echo "connect client\n";% V) i7 o/ z9 c- [* o9 F
}+ j1 L' \3 r \' b; _5 d
} else {
6 c2 n% G4 x5 o- c6 J f# o
$bytes = @socket_recv($socket,$buffer,2048,0);: A; {2 B4 q) p
print_r($buffer);" \7 G1 f8 }) i( f2 Q- N
if($bytes == 0) return;5 i( v1 t3 }, \6 G) R. R% ~; r' P
if (!$this->handshake) {
% B1 `* Q' P( C) D2 q5 v |) m4 F' J
// 如果没有握手，先握手回应
. \7 r: l) O3 _7 T
$this->doHandShake($socket, $buffer);
# c: i) x! y: t* ~
echo "shakeHands\n";& O ?5 ^1 x" G T5 F
} else {4 d7 Z# C+ F8 v. q2 a9 Z
, c7 W2 e$ b5 b; D' `" i
// 如果已经握手，直接接受数据，并处理 |9 }, \' b7 B6 h. [
$buffer = $this->decode($buffer);' O3 E V0 ]2 [5 O7 c
//process($socket, $buffer); $ m5 R- [$ }6 U& f, t" K
echo "send file\n";
4 _" X1 k5 Z% _1 J- k0 C2 w
}6 O) |2 m9 [4 J& W
}% Z9 ?/ t4 Q+ P
}2 C% S; Z- T: e) {
}
' v' |- j, L+ n0 U
}
9 Q ~; s0 G$ _8 i
: b: g* y$ Z# _1 F
function dohandshake($socket, $req)
9 L. B" c8 I3 L
{2 y# u( J: ?7 B0 \% b8 W
// 获取加密key
+ [$ K& q9 a; T w
$acceptKey = $this->encry($req);& u$ l3 R2 R) {- ?
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" .
" y- n+ u* H k E9 V
"Upgrade: websocket\r\n" .% d, O, ?+ [5 k6 y5 b
"Connection: Upgrade\r\n" .9 D+ G- k$ Q6 m1 R
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" .
7 N5 ]; z/ U1 T3 J1 L( M
"\r\n";
7 ` p2 D8 H( k* D2 y9 H
3 ^% X6 [3 x% a- F6 ^5 \
echo "dohandshake ".$upgrade.chr(0);
/ S, f+ }. i5 `# A0 m$ ]
// 写入socket
9 L6 r6 \! |9 A7 X, s
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0)));* w1 y* E, w9 S; k
// 标记握手已经成功，下次接受数据采用数据帧格式/ {) `) H& G" w5 y* t! w+ b6 q, Z
$this->handshake = true;
' ~! ?8 V2 v7 V, N& r4 f& S
}
) \0 ~/ _5 c& y% ]$ H2 s/ C
1 }& d" F( c" P9 l% V1 y5 d6 o5 t
' ?% P5 @3 D# r+ P
function encry($req)- S7 ~) N( a- E* S& K! M
{6 }; l5 h/ f0 e4 @! i& I9 R
$key = $this->getKey($req);; F" i5 H6 d% \" t8 s; Y5 X2 H5 b
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";
' i3 t* L/ ~- p9 n
( c9 t' n3 ?6 S7 U
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));
: _8 E8 |0 z4 U3 y
}
* n" c y8 _+ j2 U. v8 k' M
' n* E$ a$ j) ~2 }* \/ f! q! l
function getKey($req) $ ^" v6 ^$ k6 l6 {2 X6 [" T
{
6 I& D7 Z' K2 p& I$ u) H1 S
$key = null;7 A% D1 r) w. ^: B: Z7 L% d: k
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) { ( e8 ?5 g, e% s6 v
$key = $match[1];
6 k% _3 p$ w6 h. S
}# _/ W1 \6 h' Q* P* b# G c; b: B
return $key;7 J% V% y, ~0 c& d0 Z! \4 p
}8 ]7 O4 {% u: Y
! R# q3 c/ ^8 @$ y
// 解析数据帧
0 B# V& A Y1 r
function decode($buffer) ) J$ c. z9 U7 `: P& w/ J- ]8 G
{
9 t, p% N6 d" F! X9 m# p+ E
$len = $masks = $data = $decoded = null;
& p, ], o4 q" f% D# \( ~% o3 b4 w
$len = ord($buffer[1]) & 127;
7 r7 m G4 y/ }0 K" P' q
o0 n/ F; f! U8 {- Q3 W
if ($len === 126) {
) a: A1 e9 ~$ {0 F/ j4 n4 T+ n
$masks = substr($buffer, 4, 4);$ M6 y1 p% g1 P: p3 k- w9 {8 `! u
$data = substr($buffer, 8);% P: v. q* S" n9 b
} else if ($len === 127) {) L- t+ `6 \0 n- g
$masks = substr($buffer, 10, 4);% W9 T. M: P, N* s$ `, I9 Q
$data = substr($buffer, 14);
" p2 j9 n% v6 e) ~- n
} else {/ w3 c1 C& P/ z0 u) Z" A
$masks = substr($buffer, 2, 4);
( D- _: Y8 U& l [5 \& c) x1 R' _
$data = substr($buffer, 6);; ]% e; F9 z* Z! T
}
% m9 {* r+ H |* G$ m
for ($index = 0; $index < strlen($data); $index++) {
0 M K! Y0 X$ a; g+ r. H
$decoded .= $data[$index] ^ $masks[$index % 4];
5 A* E2 d, y$ q( s8 A4 a$ J6 [
}) A/ i3 E6 Y8 P& L
return $decoded;
% O. f- M9 u t% V
}
; j E6 ~# S( C( G- H
+ {& A- k- }8 O5 ^6 r( g
// 返回帧信息处理
; s8 B; c. I! g0 }
function frame($s)
& r5 V# B6 h1 h+ l
{4 e9 _1 j7 [2 Q
$a = str_split($s, 125);
# _- n3 ~* M- t8 Z5 v) d
if (count($a) == 1) {
6 I* T# @# P. ] ~ O, m# v
return "\x81" . chr(strlen($a[0])) . $a[0];! F) C* |+ A+ Y$ f* n2 m, M% f
}6 U) P5 u* X2 [4 V: p8 G; d
$ns = "";
* O$ c5 z' e/ s6 T+ d# D: r
foreach ($a as $o) {& ~) M' c* t7 M, r
$ns .= "\x81" . chr(strlen($o)) . $o;- w+ N2 A3 w7 `8 i) u" c
}7 ~* M* ^# n2 g) m3 Y# j& z4 K; w
return $ns;% o; T7 U0 i0 D. r0 H
}
4 n6 w! W6 u9 T
( \& {& a7 p5 h. }3 b& d
// 返回数据5 |3 E6 H7 q7 N; J( E
function send($client, $msg)3 ^4 p' l. w. U; G: ]
{
9 K0 A* c* ]) [" s, P- i# T
$msg = $this->frame($msg);
, k1 K' r1 V/ o
socket_write($client, $msg, strlen($msg));
1 @4 J1 }5 |! D' s' v
}
( R3 D9 O" p4 d1 \, T1 y
}) _. |1 H% s' y7 k ~! [& y
- l8 I, m* L- b
测试 $ws = new WS("127.0.0.1",2000);! N5 L( Y0 q/ {; R. V# g+ u
; D2 P7 t+ a a2 y

复制代码

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

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

浏览过的版块