PHP 简单实现webSocket

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

1）客户端实现

<html>. k, g3 A/ z* R6 U8 m5 m( Y
<head>
5 j1 ]: S4 H- X5 H U
<meta charset="UTF-8">
0 p. Q& Y. C" q7 i$ ]* u& F; X* u
<title>Web sockets test</title>
9 h4 z; ~4 n0 l
<script src="jquery-min.js" type="text/javascript"></script>
! \: y7 v: E0 U
<script type="text/javascript">
+ Q4 U8 P9 R2 ~3 o1 B
var ws;( A, G3 m# ^1 U' e" I
function ToggleConnectionClicked() {
X8 J+ v( C, E$ g4 b
try {9 S$ h, F+ a: Q0 j8 V* w
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器 / t8 ^8 ^9 O& p
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};) k2 v! |5 A3 }( [0 e
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);};: g/ E- Q9 \! W8 E+ c
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);};
9 F7 b, M) E7 @7 N) [
ws.onerror = function(event){alert("WebSocket异常！");};
I. [4 W1 X. X. N1 @ K
} catch (ex) {0 \2 _( J) c& N
alert(ex.message);
' n6 n3 q$ q+ P8 F, S$ k% F
}8 L$ `4 U; t9 u# W
};
8 U6 Z( \4 X7 y. @) M' A2 d% c
2 n$ J6 g4 B2 ]
function SendData() {
1 m7 ~3 a' C2 H* i2 L, g0 K2 C
try{
s ]& w2 Y( G- Q# f* u* H( L
var content = document.getElementById("content").value;* g3 `3 T' f7 H7 d: A! e: ^0 j
if(content){* I2 x/ H c' p/ p. o6 p
ws.send(content);
5 l7 }1 s: F3 G; Q: V
}$ G/ E @" V6 F A) ~ |
, v% \+ c# O; B# w7 ?+ n9 c4 {
}catch(ex){
- n4 K2 ^# \" ^) ~4 k
alert(ex.message);$ v- v( c# k: ~0 |; b& u3 D
}0 t. ?+ f) E4 h4 H1 l
};" F8 F: o8 n+ f2 o
$ x7 [. Y( G' g6 n5 c9 L
function seestate(){
* _; a0 Q( P# f7 ]+ [ Y5 ~" O
alert(ws.readyState);
' L+ g6 o% S( S- I
}
2 H& m; o. p2 P+ _" V% [5 w a
5 O. M; M1 S; T1 t
</script>* x. U- c* h; N# E. ~/ }* o. C( l' A Z
</head>
2 @* c) C; n" W( S4 b
<body>
. ]0 @, J9 l2 w1 i4 g
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br />: Q) D' \( h+ |3 W+ J) @" {" G. V
<textarea id="content" ></textarea>7 u" `* |9 B" S$ M* y. {: ]( i
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br />
0 [, T3 x) u# M$ @
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br />
% J0 G( t) a( y, [) d
0 q, t" n( X& H6 [, k% a& w9 ?
</body>1 d! Q0 r# Z' G7 P6 ~+ x) K1 _5 V
</html>( p# R. @' w- a( ~; U4 I

复制代码

2）服务器端实现

class WS {
8 s& V, G: }- A8 O. |, i# `
var $master; // 连接 server 的 client# _! c6 E% e; @1 H/ L
var $sockets = array(); // 不同状态的 socket 管理
0 }+ Q0 I; ]4 m6 N$ F9 \; e
var $handshake = false; // 判断是否握手, s6 M) Y7 `. O, [
6 H" Q K" X" \- ]: B; q9 k5 G
function __construct($address, $port){
& C( ^: u- }0 y+ M3 J$ K
// 建立一个 socket 套接字
( }4 U: j' P. M- v. F7 a" b
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP)
( x: }" D$ a& O& c# F# ?$ Y; J% {2 ~
or die("socket_create() failed");
1 ~; ]" K& M/ ~# a% F( D+ K. w
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1) . f" l4 {! S8 t" u! z6 ]' [! y' _
or die("socket_option() failed");0 S- X `, d- b
socket_bind($this->master, $address, $port) 6 A5 b( D/ ~ {7 e6 w8 ?) Z
or die("socket_bind() failed");& Z# t3 S; q; S/ i' M; y
socket_listen($this->master, 2)
, ^# r6 o- r0 G
or die("socket_listen() failed");0 P, S6 o7 U$ j
( Y5 \$ Q. }5 p! y3 S
$this->sockets[] = $this->master;! E3 P3 ?( Y( |8 t! ?
1 w. r( Z- s. K/ y4 J, C1 j2 g
// debug2 J& B1 L( |4 h/ S) \2 _, P4 {' i+ B
echo("Master socket : ".$this->master."\n");
7 L: T0 q7 X8 g9 y
( O: c8 o9 [" U) y
while(true) {& x. @. d! i# I9 S) N, y) }
//自动选择来消息的 socket 如果是握手自动选择主机
* U2 k, U# z' [. h
$write = NULL;
- z; f( D2 }! a- ?0 z' c: n
$except = NULL;
( t) ?: \7 d5 W h4 u
socket_select($this->sockets, $write, $except, NULL);
2 ~+ N4 K6 p' i: x$ i$ _0 r
$ W% D5 m/ f% } g3 N3 d0 d- a
foreach ($this->sockets as $socket) {
$ `& l K# i3 L9 A* _' z1 q
//连接主机的 client 7 O6 e. u4 {' h; P$ [- s4 p
if ($socket == $this->master){- d" c% v6 z* e& u- |9 ~
$client = socket_accept($this->master);' ?; \5 p3 v% V& Q
if ($client < 0) {1 ^ J7 a1 p; e8 w/ w
// debug/ C S9 \( m% e1 d$ w
echo "socket_accept() failed";9 L5 M# }) F/ @! v
continue;; `% v: G9 ]3 W& H7 y8 l* ?
} else {0 \& s, y& X! u0 ~; }
//connect($client);
' N7 [7 O7 h0 A: l
array_push($this->sockets, $client);& f2 \& n7 k2 b$ q9 Y
echo "connect client\n";
5 w$ X) o0 f; j& Z: E
}
. I9 x. L; Z* u
} else {
- h, f D. Y7 h" \9 }% |9 u
$bytes = @socket_recv($socket,$buffer,2048,0);# r) v- ~- t+ Z
print_r($buffer);, s$ G9 s* X! |3 W- v9 J9 T7 x
if($bytes == 0) return;
% I) v& M$ Z/ ]7 e8 V( G4 [! d' ?$ R
if (!$this->handshake) {$ L. _9 Q$ x) M" x/ p/ @8 g
// 如果没有握手，先握手回应
9 o7 e) J- a5 |9 B5 x. o D" Q
$this->doHandShake($socket, $buffer);7 u( z% x' R+ m/ |! X
echo "shakeHands\n";
} else {: _$ z* e1 G! Z+ e( U
- ` V8 y7 l& y
// 如果已经握手，直接接受数据，并处理5 O/ f! j h$ M
$buffer = $this->decode($buffer);" K5 j) n- Y7 J( V
//process($socket, $buffer);
1 \+ b( e! @6 X: }, U
echo "send file\n";6 P4 f. H! C' a9 } c; k7 N
}& R) Y5 m+ Q8 v3 c
}
! U% G. k/ L4 g& ^2 ~8 [
}7 \0 {8 q1 h# }1 i9 Q# G( j
}
6 B6 {" g6 k& `- }; p, s
}
c3 x9 q' v9 y
' Y( {' {% e- d$ R3 o
function dohandshake($socket, $req)9 I/ ]& {& l- {
{8 ]% w. a: k2 a' A6 ~3 v; |
// 获取加密key
6 z" Q# l9 x0 {- Y! F6 N$ i# g8 ]
$acceptKey = $this->encry($req);
% L$ @$ e9 b( S7 Z# D# p3 [$ L* R7 B
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" .
8 S+ k* f3 R4 ~8 r/ s5 E+ m) f/ K
"Upgrade: websocket\r\n" .
; O: R0 b3 `6 \2 ^- M% O5 S8 h( T
"Connection: Upgrade\r\n" ./ z! J, {% F& _" k' p0 @3 i
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" .; g* d5 P* Y2 t( d% }% ]$ f
"\r\n";
, x# H) F2 J! T2 f- S1 M0 z
4 A2 |" w" O8 x6 c
echo "dohandshake ".$upgrade.chr(0);
! M' [& H% E0 Q' _/ q5 Z: X( G# a
// 写入socket
. R! N ]+ ]& A1 b x/ k3 L' J
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0)));
/ ^: T1 O- a4 N
// 标记握手已经成功，下次接受数据采用数据帧格式, n, _7 W. F4 i( W: M
$this->handshake = true;
+ D* h8 s( h \3 g( B: ^6 C
}
4 W; k* b/ j0 E8 j& O2 A
3 S( `+ g7 J$ N4 {. V, N
. k; P% z* F: ?& G$ Q8 s
function encry($req): v/ @, K( T5 w
{5 x) A& r D. \0 f, G7 O" F
$key = $this->getKey($req);: k4 Y1 o: l0 g! U2 C) n
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";) u7 `6 @. K& e: n0 ~0 L S& c& Y
w* S4 g; V {5 T7 d! _9 k* g
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));
0 T1 k/ Y/ E- X5 i, _1 ]7 E
}
5 i" a, `% {/ E3 b8 d& W1 T {( ?
) S+ p6 u+ ?! V* b6 @9 R2 o
function getKey($req) 3 J3 i3 c4 O/ ]( \8 A5 H
{
/ Y! u2 Y4 T( S* H5 V
$key = null;) N/ I. }/ l) J. L) I
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) { 7 Z5 o8 u }7 M$ K- G
$key = $match[1];
- I. ]: v8 S: l3 T9 p/ z: r) G
}
. k! H" F1 C6 b
return $key;1 s* D7 ~% P V9 t( q) C" V8 ^
}
3 b; p. h& E$ n# S2 }2 o! M) c# @. T
; G- m9 H) F0 Y1 t) H2 l+ v1 c
// 解析数据帧# z A+ T6 w8 }& Q( w* H/ ]
function decode($buffer)
) |( G4 Y. n2 K3 U1 Y/ y* G
{1 V4 A/ S* g! b. U0 _9 x
$len = $masks = $data = $decoded = null;
( k! G* I& ]" Z9 ?
$len = ord($buffer[1]) & 127;: g0 E0 j. J& b
+ U0 y; z# y+ n2 {$ y4 X. ], ]) M) w
if ($len === 126) {
; ?3 c8 }6 s5 ]2 ^
$masks = substr($buffer, 4, 4); z& Y, K7 x, n& p( R
$data = substr($buffer, 8);
1 d6 W4 O6 d3 \. s
} else if ($len === 127) {
* F6 N- U' R% s8 y9 D
$masks = substr($buffer, 10, 4);
5 F' a+ s( [6 m2 L$ [
$data = substr($buffer, 14);: P; j) h3 W* Y7 N
} else {; k. i# Q" U( g, z1 }% n5 w' p
$masks = substr($buffer, 2, 4);9 `8 q( w- V# X7 n) j, T
$data = substr($buffer, 6);5 T( M' T/ f5 I0 C' X3 w* M
}
x! ^( a9 `1 K! e! A& p5 Z
for ($index = 0; $index < strlen($data); $index++) {: C0 P4 q0 X3 w; @/ T2 w% n
$decoded .= $data[$index] ^ $masks[$index % 4];, \$ k7 H- V7 D5 B
}
7 M2 s5 h( B8 N x; {! P l; @7 |
return $decoded;% Q: p5 E, y& u+ r% o T6 m
}
- q. f' _0 n: H! m* E# T5 d
& H+ n' r- b; }- G! _$ l |
// 返回帧信息处理
9 P9 p6 A5 s( @) a
function frame($s)
& o2 G( L% D- V) G# c0 i
{
5 Z$ i' ^. W/ O: l# ~8 j6 M
$a = str_split($s, 125);
4 \& Z7 I" A9 ^, O8 h8 V
if (count($a) == 1) {
- h# Q0 J9 M7 o. x" W( r6 N5 n
return "\x81" . chr(strlen($a[0])) . $a[0];
1 e' w" l4 |' D4 y V' W
}
, y5 b+ L0 p6 C- a% y4 E1 k
$ns = "";
8 b3 K7 m3 _% M/ ]+ ~# o9 E
foreach ($a as $o) {# v4 m2 ^2 l! D+ F) q! {
$ns .= "\x81" . chr(strlen($o)) . $o;4 f8 n! s8 A6 `
}
; s: R2 I- m6 \# G; d% Y2 {% w
return $ns;( }) }6 J7 b2 [# V( l8 e
}0 D1 g3 ~% B8 O
/ g( I- J( \( R; u$ L1 b
// 返回数据
J- \4 e* T9 v: w/ O- V3 |
function send($client, $msg)
0 G3 w4 \" N. w' g8 E
{9 d' {( [: \$ Y% A
$msg = $this->frame($msg);
+ B: x S4 I6 w
socket_write($client, $msg, strlen($msg));
" q ]. w* x1 W* Q
}, s' l: q0 Z$ ]' u' X9 M
}! O5 h) U, K( j& B; r; N! Z
5 S8 C6 f5 [; J9 d% }: W- X
测试 $ws = new WS("127.0.0.1",2000);
' @0 `" C% G$ q" {! [( D2 A4 t. B
' G1 I, t' a5 T) m/ Q

复制代码

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

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