PHP 简单实现webSocket

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

1）客户端实现

<html>
1 ?. \' P3 W- o" l! Z# N
<head>; ~2 A( }( N; ]& f- L) _. v- {( V/ t
<meta charset="UTF-8">
; l7 Z' U0 E: |) ]$ c; |8 k
<title>Web sockets test</title>
' V- ?/ {. F" t b0 s4 M8 m; O
<script src="jquery-min.js" type="text/javascript"></script>
3 G7 F( e4 H1 Y3 k$ n T# z
<script type="text/javascript">
0 H6 V* _& O5 y9 d2 ?) s. K0 k( k
var ws;1 f* B! O. J" M
function ToggleConnectionClicked() { - q; {2 q _/ o4 _
try {* I% U" i# c& t9 |6 C9 b
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器
( f* |4 E- ^" M5 @% q
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};
9 w, c- b: k7 r/ f0 Y$ C! h' w
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);};
4 Z7 T1 J' s, P- h; C" Q# e6 I
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);};
) W1 X5 t0 E g) {$ s" o
ws.onerror = function(event){alert("WebSocket异常！");};- G% S6 U4 e1 s! v$ i7 [
} catch (ex) {+ R$ ` N7 p1 e) |, N7 W% \/ f
alert(ex.message); 9 b$ X5 o' f& G; i( z$ U
}
4 y1 s$ l9 P& m; \( V3 ^9 J
};
4 a- q- M( C \4 u/ G, o2 b
. [% N0 A- ?) A5 A3 [3 B2 A0 E [
function SendData() {
) r" _7 m" R! U) W; H+ W
try{
# H, s7 {0 ^3 G6 S! ^
var content = document.getElementById("content").value;9 A+ B y9 {5 l( ]$ v7 `
if(content){
. Q+ Y& _9 Y x" g& M% t) Q3 f" ]
ws.send(content);$ t# i! V1 o$ K* P0 q
}, |( U& q2 U, k% G
: [; F9 j+ F `! S) X4 X( a. X
}catch(ex){
! N* @' E% i7 Q* T8 L
alert(ex.message);8 H8 C4 p8 ?, \- b) V* ^
}
5 _2 N* I0 T1 j0 I& A) O
};- Z7 @ R9 x4 i t
9 A3 T( W/ J3 u4 k1 |, e
function seestate(){
' m0 Y+ l" \5 k. f' q% }
alert(ws.readyState);; `3 X5 } C% y
}% b( Y- W8 O2 S5 h0 ~' g0 A( m$ P
$ M+ X9 z% z8 [
</script>- G' }9 J4 x' _8 u9 m# r3 O% H! y% J
</head>6 `" d c' B7 C) @3 P g9 v+ `5 f. [
<body>
- S/ }1 L# s; }% h9 J( N
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br />
8 V1 r* p" @0 z+ k6 U
<textarea id="content" ></textarea>
& o* ~ O! ?, |) j5 T9 K3 U3 U
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br />
% \6 f$ C7 s3 B% g! I
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br />" O* e# J$ v! k, f+ A6 y9 B# U
2 C) i u5 Y3 Z% u; J6 D4 V
</body>. t( \$ y, y7 m) Z+ v' e
</html>
$ [0 I4 P' x2 r" s4 }2 n$ X

复制代码

2）服务器端实现

class WS {2 L( n5 l. f# e9 s# p/ d3 y& g) t
var $master; // 连接 server 的 client
/ p d, j0 Q) `- ~" U: k8 l* A. R
var $sockets = array(); // 不同状态的 socket 管理
9 z$ D; p' p5 C
var $handshake = false; // 判断是否握手. n, l; a X5 F# Z2 t9 x* k
% Y X/ ]9 S) l( @2 B1 J
function __construct($address, $port){; o$ {6 ~4 q @ I- i- n
// 建立一个 socket 套接字2 d+ w$ x. _- G! C
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP)
' @- h$ v" r9 Z1 [
or die("socket_create() failed");6 w1 U. ]* H9 a9 `
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1)
1 ]0 j, [3 t n
or die("socket_option() failed");
1 x; {5 Q$ Q/ P* i! n' Z
socket_bind($this->master, $address, $port) * k* S$ [. \ U5 `, I- X6 m
or die("socket_bind() failed");& B7 P2 f$ t" Z: E0 _
socket_listen($this->master, 2)
9 ?% F+ L$ Y8 r6 O
or die("socket_listen() failed");
7 m1 T ?- `5 V8 w# b$ }
@7 i- g' b1 D
$this->sockets[] = $this->master;
, [( G; P- s P1 M/ x% h! O
2 K$ t( P! O7 S6 Z5 k( e
// debug! f: ~8 }0 R: w6 A& k: m2 P# x. R
echo("Master socket : ".$this->master."\n");
2 a3 C; G- J, z& [1 a b0 j* _
% Y3 |4 o5 p: p7 p0 N
while(true) {: D: O! b7 M! l$ \4 w- i8 C: @5 k# C
//自动选择来消息的 socket 如果是握手自动选择主机. k- t/ M; j& P5 J% M
$write = NULL;
. k1 I; L( D1 b8 G1 `
$except = NULL;
) y* D( x' `2 g/ u
socket_select($this->sockets, $write, $except, NULL);
, @( |- t. i; u. t9 L( t2 O' e- ?
. i1 o* [. J4 n
foreach ($this->sockets as $socket) {1 Y+ h" D9 E1 Z- v
//连接主机的 client ( [5 {6 k, a0 p$ L4 ~# W3 G; p
if ($socket == $this->master){
9 s3 [$ B& t# t4 ]/ |: w% X7 ^
$client = socket_accept($this->master);
* c1 S' |4 s# l
if ($client < 0) {# E! Y k8 \. ], [- F
// debug
! S4 y' l2 b3 }, d7 x" S
echo "socket_accept() failed";, H( Q m1 }/ @" w8 ?
continue;
A; H9 V& r* T \; n* u; \
} else {2 F2 o0 O8 G' ~' ~3 [: m P! E" B
//connect($client);
9 i/ {$ h$ z G, C0 _+ R
array_push($this->sockets, $client);
' A0 R% k6 R9 s$ _ h$ F1 x
echo "connect client\n";
" S* R) f3 \# s _ R& s: B6 i
}
; g7 E) @/ E" p! L* @+ D
} else {/ Y3 T j$ m: X& G6 K
$bytes = @socket_recv($socket,$buffer,2048,0);
# h; E, \2 M& Q: t! u M
print_r($buffer);
5 b3 x" v; ^! K* K# N" x1 a
if($bytes == 0) return;6 X) O% y7 `$ M7 U L# n8 `
if (!$this->handshake) {. q/ M0 k* _8 b9 `( I; T) x& p4 p
// 如果没有握手，先握手回应( N9 ~& b' K4 \* Q6 w, O
$this->doHandShake($socket, $buffer);
& x6 l7 v# ~, L) L
echo "shakeHands\n";9 M% {5 W* u1 ~5 d% j& c# o
} else {+ d0 l6 y2 g9 t8 p; Z
8 h; }0 o. h, n- z5 S
// 如果已经握手，直接接受数据，并处理' y# m0 V, A. ^; n3 S- E
$buffer = $this->decode($buffer);) G& T) ^: {, W* O* U7 n
//process($socket, $buffer);
4 i8 w. s+ z2 q+ C$ K" g# p
echo "send file\n";3 T) \; H4 z2 U" _
}# H; a/ }( N: p4 \* w4 S M' \
}
+ u1 \8 S/ X* D/ ]
}% D, R4 }0 p; G# }1 o
}3 ^- u. s( v; e. t; g
}) B4 v7 Z/ G7 ? h$ _" G
" c9 }0 s( J, s6 t1 \" r) m
function dohandshake($socket, $req)
0 U0 _$ X* ]- A, [8 ~
{
3 ]- m1 Y& i; t2 q3 v0 d
// 获取加密key% P `& A6 E, ^0 W
$acceptKey = $this->encry($req);* p9 _7 H" l6 S3 o9 ]
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" .
: Z9 l3 y E1 z8 @. `
"Upgrade: websocket\r\n" .
: n6 ~2 y9 } O
"Connection: Upgrade\r\n" .
( W, y4 S+ R( S
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" .
% C! T# N% r. Y9 b3 y; [
"\r\n";; [- h8 }( n9 ^' s7 K5 M
) I+ }5 p& f$ o
echo "dohandshake ".$upgrade.chr(0);
# N$ H0 L2 E" i( M1 y7 U
// 写入socket& C6 L5 k" H, D6 U b
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0)));
( L& p; |- k3 r1 M( K- e) r3 d" K0 P
// 标记握手已经成功，下次接受数据采用数据帧格式" ?/ ?" J" y' A) a, {
$this->handshake = true;
+ |% t3 {6 [1 Z
}* J2 _# A( ^: K! H2 M
- a* @ f: g0 k) X* B
' l1 l9 H& ?6 g# O! p4 k
function encry($req)
# x0 T; a9 | z+ }$ \, W
{+ w; Z' } h; \& k( T2 v7 d
$key = $this->getKey($req);
9 q# |4 f8 q( @9 i5 d# D
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";
/ o- A5 N) i) ?9 ?
: H# w) L% C6 J: u
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));% x% u6 a3 z/ Q8 f1 z
}6 d7 ^/ K" ~# s; p: I: a
3 A" X k: ]7 s- b* N0 E. u
function getKey($req)
1 o$ C9 t: M$ p% ~ D7 v, j' T
{
- N! F, }8 }) K. k2 {7 Y9 Y. Z2 e
$key = null;+ ^2 G2 W0 t0 F2 |
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) {
. y9 f7 S( V t& f/ S4 j: C
$key = $match[1]; 8 [8 X" ]6 _/ _; u
}
8 b! Y# ?7 L$ V, x
return $key;& L0 @" L! p8 c' O
}
- b! P$ j% z! H% w; W' d3 u4 c
# ~0 H2 e( I. D! _& {8 X
// 解析数据帧, x _0 b& |0 o: w* d% ^
function decode($buffer) ( I8 r. K( o3 {6 C& K, `7 A
{
. P( Z. Y& p% U8 z/ y: T) X7 c
$len = $masks = $data = $decoded = null;4 L% n5 U- x6 n9 w4 _& K- E; o7 |
$len = ord($buffer[1]) & 127;3 Y: }7 |; w/ h9 k+ p
) D3 f6 e8 x5 h' ^& Z O% ^4 a
if ($len === 126) {
& b L& ] [/ p# a1 r
$masks = substr($buffer, 4, 4);" @2 u0 k9 j3 k z( a- O; T+ h* H
$data = substr($buffer, 8);
; t9 H0 u4 P2 j: G! d
} else if ($len === 127) {( o0 J+ E3 ~4 I
$masks = substr($buffer, 10, 4);! [! F: [/ k5 \% G! h
$data = substr($buffer, 14);: b8 x. I2 d* w6 K8 z
} else {
2 k+ U4 s3 ^" A' m! w
$masks = substr($buffer, 2, 4);
& q6 ^. f; x- Z* a$ M( \
$data = substr($buffer, 6);/ d1 W% D5 ~/ Q2 G/ |
}( i8 ^( ^! Q5 E; l. t- U
for ($index = 0; $index < strlen($data); $index++) {' q8 m! [2 Y* n7 d! N3 Z
$decoded .= $data[$index] ^ $masks[$index % 4];) T1 N' N$ l) Y. p
}' L1 U; z; h5 ^8 s0 g. ] `* S# H
return $decoded;
7 G9 L* b: |* n7 p. e* O9 }" l# u
}# ^1 Z. ~6 R a
+ S( i6 Y( v1 i: Y3 l
// 返回帧信息处理5 ^0 `- ~; {" O3 R: N8 \ L7 d
function frame($s) " N/ v) B9 M9 i9 l1 G5 k
{; Z% j, I( X3 f
$a = str_split($s, 125);
& y' `- g+ O5 E4 w/ B
if (count($a) == 1) {9 ^4 {+ ~- X7 M: v1 X
return "\x81" . chr(strlen($a[0])) . $a[0];1 G; O' g3 C5 z# l
}
. f( z2 d! K( n" P* D3 @
$ns = "";
& o6 V* B+ a8 }3 T' |% @
foreach ($a as $o) { f+ {8 `# k" ?6 ~2 g: }! w- U
$ns .= "\x81" . chr(strlen($o)) . $o;! X% F" N" O! P6 D% Q3 j" T$ w
}1 s; k8 ~* ]2 q( V
return $ns;$ ^" @/ Y9 F; k6 ^+ z: G. W" q/ r
}
; _- P, {. v. }" k" T3 Z4 j, z* f
; ]( j$ ] Q7 B* O
// 返回数据
8 E4 d% p" H8 c7 I7 {% X# {. T8 h
function send($client, $msg)+ [* T/ X+ k5 U& c/ E! o
{
0 J& i' w9 H" c
$msg = $this->frame($msg);
5 Z+ G% o* g. i, G, T* w1 K
socket_write($client, $msg, strlen($msg));
# C" c% T7 a" l4 |) r
}
& L5 ?$ ~6 X' A% }: i7 r. S
}3 X& h3 ]- y' f( A$ N3 P% Q$ I
0 x7 d. y' g5 J3 j( d
测试 $ws = new WS("127.0.0.1",2000);
; F K: q+ B: T
( Q8 T% z3 s; H' C' h& |

复制代码

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

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

浏览过的版块