PHP 简单实现webSocket

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

1）客户端实现

<html>
1 K0 s- u3 g6 l+ }
<head>5 Y* l9 j( b) b! O
<meta charset="UTF-8">
, `) f' r& d9 d5 j
<title>Web sockets test</title>5 G! v7 I4 v% W, n* X2 r; [! s0 z2 [
<script src="jquery-min.js" type="text/javascript"></script>
- M0 @+ { G" E3 k1 l) u
<script type="text/javascript">
' [/ @. y& a) o
var ws;7 c- K! [4 T2 x+ N2 g
function ToggleConnectionClicked() {
2 [ q [. g2 Y2 w6 A' w/ P
try {. c( |* c; Z K& t
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器
; ^+ e# @* y4 Q% A
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};
7 H1 w' x5 m0 C. t
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);};
# k9 O, e! @- R8 r
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);};, V" T& |4 r; _0 `! u: T9 U
ws.onerror = function(event){alert("WebSocket异常！");};
" Y& ~1 R7 F# A' s0 {; |* S# R( ?
} catch (ex) {
/ q% [- y. }- G6 U9 w
alert(ex.message); 1 v- a i8 O$ a2 i6 e" x7 L
}
+ k7 ]* N1 }3 Q1 S7 L
};! S/ ?! U3 u6 ?1 J! J% v& g
4 s( w! d9 Y9 s1 h3 r5 M
function SendData() {
5 Q5 K5 p8 X k) o4 i, P+ g
try{
$ p8 k2 X" [) l7 z& T( W- c& J
var content = document.getElementById("content").value;# r/ z6 i$ s! y1 ~. [# t% e0 o
if(content){
: L$ f# U% ~) M
ws.send(content);0 J: i9 |( u3 @2 U* o f& q
}
1 _6 k6 E& L% E. p3 n; y1 S
- q7 I4 ?+ D4 K$ z% X
}catch(ex){; ?8 m. I3 r: R# x$ T x$ Q
alert(ex.message);* m/ D4 |3 y# C2 r; b
}/ y: k: v( u( m0 P1 {% O- ^/ K. C, f
}; c1 E+ u0 l' I
1 v, K, T1 \: |5 H& Y9 e5 l6 j% P8 J
function seestate(){. R) R8 a/ u3 F; e' l
alert(ws.readyState);
/ o1 J/ s0 T1 k" N, q9 t
}
8 s8 c' E) }+ I5 R+ G
0 d" p) U7 Y% c T" e/ b9 V: [
</script>
# X* W% g3 _7 N2 Z( S+ {
</head>) W; P9 F1 t. I, s, E+ U
<body>
# ~& P$ N5 P t- w" R/ u
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br />
# Y1 m5 c+ F9 z3 k. v8 Y
<textarea id="content" ></textarea>
, l+ @/ U" V( l1 @2 Y z1 J0 I( r
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br />. n* \: S# z- R. l: R! y
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br />
* x( a- W# L+ r/ Z- Y% |5 f
* j5 v) f) L. {! c2 a% ]( d
</body>
) B- @6 a9 H. z! d8 x& Y- H: ^! n
</html>+ Z- n3 O2 w5 Z9 C' e* o6 Z3 ^

复制代码

2）服务器端实现

class WS {, T8 X( E; F( `# p- n- a! z
var $master; // 连接 server 的 client; a8 U; }6 K* E& o$ R: p' D& l
var $sockets = array(); // 不同状态的 socket 管理6 z9 M: Z" @1 |: ~) n2 ~( s/ B4 y. {
var $handshake = false; // 判断是否握手
$ h, a' l( X% i
9 f1 b4 @6 i; o
function __construct($address, $port){7 [: c: u2 b# b% d
// 建立一个 socket 套接字
4 _3 z: N5 j7 @6 [
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP) ]: d% j8 s9 A2 d' O
or die("socket_create() failed");
" M& c u) N6 d" o7 ^
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1)
# q# @; I; {, r
or die("socket_option() failed");: r7 U7 A% l" ~: I( W9 O1 R
socket_bind($this->master, $address, $port)
7 N3 \5 {0 f5 H+ a4 f+ D1 J+ s
or die("socket_bind() failed");# ^8 E& f% o( S- q! ]
socket_listen($this->master, 2) $ @" a, L$ J# p. b4 d r* T
or die("socket_listen() failed");9 C. }8 b3 {7 W( F2 J
. X# x3 I3 w8 }
$this->sockets[] = $this->master;9 p) k4 I3 F3 P) T q
5 x0 c' i/ x3 p! b
// debug
6 l# p: U$ V" K( V
echo("Master socket : ".$this->master."\n"); r! `0 U+ @1 X7 ^# F
& z" F! a% V/ k$ y! L' L# T. Y: N
while(true) {
5 |. ]& K- R/ h) A- x" ?& L
//自动选择来消息的 socket 如果是握手自动选择主机 L" d6 Z: D) o6 ~2 W7 u
$write = NULL;9 C$ m! G5 M% i8 k& }: V4 e9 }
$except = NULL;$ |9 {# G* S: \4 R% R0 E+ c
socket_select($this->sockets, $write, $except, NULL);# p' o* o: N$ i3 p( p/ F* {
) D. h- z' A% J0 t4 l
foreach ($this->sockets as $socket) { E& ?4 C4 H% f1 v) q) X. l
//连接主机的 client
! F0 A5 ^; Z# s- B- @$ @/ Z
if ($socket == $this->master){( l: A& N2 W/ S2 V1 K* s9 q
$client = socket_accept($this->master);
, H4 o! L$ Q; M! Z% [' ?' }6 M
if ($client < 0) {! W( O3 M+ Y& ]/ G# t: C- \
// debug
) ^+ W3 m( \, m- x2 t
echo "socket_accept() failed";
: Q+ P6 S( O8 W, ^
continue;
' c: |( \" K P* W9 @
} else {! K5 j: |) j8 u! J5 t4 c- N9 U" u
//connect($client);
" l5 C7 W0 G0 ^, W! F
array_push($this->sockets, $client);6 j5 b8 y: m+ z" W+ k4 l, i
echo "connect client\n";
1 d- O% ~: h% e1 k3 g& x8 [
}3 V" b2 W! @+ ?7 }, |7 r
} else {9 ~ [0 }4 Q" h: c
$bytes = @socket_recv($socket,$buffer,2048,0);
1 u u! L1 H# V" V/ S" _
print_r($buffer);
9 k: W6 R2 S$ L5 C* e$ b
if($bytes == 0) return;6 W7 v7 V7 M5 v$ q+ a
if (!$this->handshake) {
0 h. `8 ` ]* x3 Q) S) s
// 如果没有握手，先握手回应2 ]% J* o9 [# D6 v2 J9 t0 K7 x
$this->doHandShake($socket, $buffer);
" m' R/ k; K9 f& p6 r- I- z
echo "shakeHands\n";' H/ n: J9 X3 s0 j
} else {
" N" ~* D4 D4 Z% d
* F% Z1 A" @( C/ @2 f, x* B( C9 V
// 如果已经握手，直接接受数据，并处理
+ l( ^% }5 Z# `) K9 J9 q
$buffer = $this->decode($buffer);2 {3 A Y( m+ ?: k1 @1 U
//process($socket, $buffer); 2 w# f+ Y! s: c( ?9 p& N$ A- D6 E
echo "send file\n";
5 ^' A5 l j: E
}0 ]4 s! n$ ^* A4 |* }: N |
}* Y9 o- v; ]7 Z5 N
}$ ~ Q( i) o. x$ x3 l
}
. Y0 X( r% u: ^+ J2 o4 i7 `
}/ L C/ J+ L X5 e
6 F o' ^4 ?* J$ o J+ V( D
function dohandshake($socket, $req)
3 l! k8 i; O* t; |, `5 t
{/ v3 O+ ]* B8 j8 G8 }
// 获取加密key
; f ~! L; b# V, ?! N! u
$acceptKey = $this->encry($req);* J# p. l* s5 k) X' t* t. u
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" ., k* s0 R& |' [
"Upgrade: websocket\r\n" .
, n* `0 D2 @; y% U# F0 a
"Connection: Upgrade\r\n" .( V0 h( a% [ Y+ P! `+ h
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" .: p) L7 J1 Y3 T( l, H
"\r\n";
% D; f, b$ ?1 r3 D2 z
' m/ V, H( L6 M9 P- p
echo "dohandshake ".$upgrade.chr(0);
3 C- z R6 r% l5 l. O
// 写入socket2 @5 U( O) |, u" \* Y& o
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0)));# h& A! W. ^4 z: _( N: f5 G* i
// 标记握手已经成功，下次接受数据采用数据帧格式 ?1 @. o% |$ {( M# O! i& x
$this->handshake = true;
e4 W0 s, I1 f2 a( ~' z- k9 }
}
% {! ^* x- D1 F2 o0 }% h
2 v. ^# u7 o+ s z; `
! ^$ v; h1 u# U! B! _! X3 d
function encry($req)
1 B2 T9 j* B( [- X
{1 q1 ?# }9 p3 r G
$key = $this->getKey($req);
+ f1 M) W* G! X/ b. s2 m: M
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";
' \& i v1 _6 n# Z7 b/ c
, o! Z3 g! ?& D! W2 t
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));1 }7 J6 A+ j$ B, u: g
}" m# K7 O& W! r# X: h
( k' M7 m. d6 Q" g* f
function getKey($req)
$ F# F6 a" a9 A. S
{
( X( P0 N, y) ?, S3 e1 Q3 w
$key = null;
, J, X8 P! c1 f. |# K. ]( G3 F- O
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) {
, [# {9 N& W# d m. W
$key = $match[1]; ; H, H5 h- V( O, g6 m' E+ a
}. a* J" p" f; f
return $key;
2 Z+ O) e. m% T
}
% G0 i0 r9 S8 ]# j+ s A3 m9 p
4 w2 A: A- }/ ^4 @$ P; I, P8 T6 a
// 解析数据帧* s: c, E @# Y) K+ ?
function decode($buffer) W$ t( D: w+ F8 s3 ^
{
% _) V# @' l# ~
$len = $masks = $data = $decoded = null;, \( V% l8 C" p6 H2 ?
$len = ord($buffer[1]) & 127;, Y z3 W" s7 z9 @* Z4 g4 _
. \! M$ b- R4 N' m
if ($len === 126) {
4 l7 K0 U9 ]5 ^$ b9 l( i
$masks = substr($buffer, 4, 4);
/ ~2 S, W+ U7 \0 D
$data = substr($buffer, 8);: ~8 |' r1 w# I
} else if ($len === 127) {- e2 C* i2 [- w/ d s1 z1 e
$masks = substr($buffer, 10, 4);: p& T5 m* n$ M% U1 _
$data = substr($buffer, 14);; e; c* f$ m5 S
} else {, I, U) K0 o/ a) ~' n* E
$masks = substr($buffer, 2, 4);
. N, x' Z* A0 d# s
$data = substr($buffer, 6);
) C9 k- X6 F0 E8 E. I* T+ c5 c
}
4 l0 d( E# j( H9 t2 h" w
for ($index = 0; $index < strlen($data); $index++) {/ l: V4 ^; ^8 G
$decoded .= $data[$index] ^ $masks[$index % 4];6 q8 }/ T! U0 d, D
}
( f/ h0 o5 k0 ?: P4 e4 h- ?
return $decoded;( C+ ?) D! g( M/ r J
}
, i5 g& a4 n2 C6 \
: S8 d1 n" k: e$ k
// 返回帧信息处理
; Y7 G/ O) {9 t/ {1 t" \4 b
function frame($s)
1 u8 r! j9 O; Z6 u
{% `5 m( R. P& q
$a = str_split($s, 125);5 s, Y8 O( C8 ^$ V+ W5 b. T
if (count($a) == 1) {
: A; A9 |7 J& Z w2 P L( k
return "\x81" . chr(strlen($a[0])) . $a[0];
4 L$ b' }+ ^- a6 F: U; e
}! u: t% E# A; e3 Q2 X
$ns = "";
# H# G2 W) G# B1 V d0 y& H) p
foreach ($a as $o) {. _4 V1 a) i" G$ P3 {8 m, O8 W6 T% D
$ns .= "\x81" . chr(strlen($o)) . $o;
; P3 U/ r$ V1 ] e
}' i- _$ F" F, x2 o/ w
return $ns;
% m: P. V" i: h$ q" B. o
}6 h5 \2 ~4 L) ?
. s9 x9 n% w' I& B* J: T
// 返回数据! ^7 M% M: Q& X) t
function send($client, $msg)
: C7 O+ E. p5 L, q5 d
{
+ s" G* Z5 q. `2 z
$msg = $this->frame($msg);* X) u$ a' {$ B% U4 W1 @
socket_write($client, $msg, strlen($msg));
7 n" c" o+ ^+ N- ?6 G4 K" g
}
- Q1 {9 ?5 C: C, ?/ Q
}3 V- x( j; l, p8 P ?# S% z/ a3 n
2 A/ D/ n# i ]# X$ D5 @1 O. L# c
测试 $ws = new WS("127.0.0.1",2000);
+ w0 g3 u# `/ ~0 ]8 ^3 k
7 ^6 h: m: j+ T9 A) N3 _) M

复制代码

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

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