PHP 简单实现webSocket

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

1）客户端实现

<html>
, c7 B* R- N2 |2 |: G2 Q; [
<head>
/ O" R+ E& S' E& V4 k+ V
<meta charset="UTF-8">* O3 q! X6 t! j
<title>Web sockets test</title> V+ t! L6 C2 F
<script src="jquery-min.js" type="text/javascript"></script>
0 O+ a& A; B' o5 S
<script type="text/javascript">5 k$ r1 R) i t- @* f9 l1 Q& ?3 g9 l
var ws;
: F Q1 U! H8 P) Q
function ToggleConnectionClicked() {
" r1 v; M2 e" ]( X- s
try {
a1 }! D; X" e0 [+ _- {
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器 + o, c% r3 k4 G8 Q
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};
: h+ c8 ?4 j$ M5 M
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);};
5 S8 e( @- D, |+ U
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);};9 t* m+ o b" t! ~
ws.onerror = function(event){alert("WebSocket异常！");};8 Q1 r3 F( O: L8 p
} catch (ex) {
5 T" ]7 ^ w# K7 H
alert(ex.message); ! G( u5 L0 }8 K/ p. |8 y- m
}5 R, W; W( V4 T9 L& h6 b1 ~
};
) V5 X, O: g3 Z1 _
" U' N& x7 Q( M" ^; _6 \
function SendData() {
* t1 e& P2 b7 I' s# s
try{
; i+ C7 G! e: w. ?7 D
var content = document.getElementById("content").value;; n# ]* r0 C( s0 V. F1 j* T
if(content){8 V8 C! ]/ E0 P! J- Q: q8 E3 k
ws.send(content);" U! U: ^" j2 I6 n) i8 ]
}
9 `( e( r& O# o% N6 W: f# f$ ]- I
( o# X) m/ y& l8 \
}catch(ex){
8 o; l6 o+ k; g; ~+ \1 F7 R
alert(ex.message);
& Q# q9 ]9 K6 I2 M$ J
}& M* Z. p- ^7 J. v L
};
6 J; F! R1 M5 ^) T) l1 n6 u- h5 H
& D' w B3 v" d9 a1 ]0 E! d+ j
function seestate(){1 o* O; D4 `8 s5 A. X2 U9 E/ u
alert(ws.readyState);! K" [" s3 \, ]/ ^5 C$ I
}( d- S1 o5 y! E7 t- u( B- T3 o
% D! i7 w4 ?" t
</script> |" c3 l) f9 H V" l
</head>
, ~( V& s! \! M" x, t l
<body>
# O% b, p' c* J) H- A# I8 Z& s0 A- W
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br />" N5 C: w9 q1 C0 D, f& A0 V' h; l; Z
<textarea id="content" ></textarea>; t9 L) _, `$ X2 Y1 M
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br />
+ X# g& C2 l8 O% p4 {5 w# I$ |
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br />
/ v$ P! T9 s! T0 T$ x
6 x2 {2 [+ J8 z" S9 }8 J' `' K
</body>
& p+ H; W7 L3 S
</html>! }) g9 `" p7 C3 d

复制代码

2）服务器端实现

class WS {; w0 b! Q# f w& F* v8 {. u( ^
var $master; // 连接 server 的 client% t _$ L& E; b5 I
var $sockets = array(); // 不同状态的 socket 管理
0 k: o6 L% h% F$ x+ v- E
var $handshake = false; // 判断是否握手' [9 B* D a; ]! C8 C
. ~9 [! i, s8 L% B
function __construct($address, $port){
, ]5 |7 A- I( O( R
// 建立一个 socket 套接字
; R* k4 i) S( t4 g
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP) - N2 o! i0 O! x0 a9 l2 Y: q9 j" D
or die("socket_create() failed");; k1 H; l8 k% P) b+ _4 h) ~
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1) & w6 h/ ?4 O3 n) ]
or die("socket_option() failed");7 _9 T; ^+ [, W9 N9 Y+ U
socket_bind($this->master, $address, $port) 8 n) Y; f9 L8 o2 A" @ F5 R2 {
or die("socket_bind() failed");
2 M3 D7 G/ N h* `
socket_listen($this->master, 2) + i: l7 O/ O4 A' x& D
or die("socket_listen() failed");
7 P/ K9 _& n9 ]) p
2 J3 M1 G7 z# \- C- S& i
$this->sockets[] = $this->master;& ~* E; E; ^4 S+ }% i3 t
i1 e7 j$ D3 j0 g @; j( |
// debug
6 t+ m0 g+ O% r
echo("Master socket : ".$this->master."\n");. c2 e4 H5 K: F
1 K8 e+ |- \; b; F ~' v& S9 H
while(true) {" p: y( _! E7 L6 g' H) j K+ ^
//自动选择来消息的 socket 如果是握手自动选择主机* A. @5 u1 B. g4 i C: {/ |
$write = NULL;% r# Z- R' o# r( \
$except = NULL;; Z C0 k; E# c1 q/ C% D3 T7 U) R" x
socket_select($this->sockets, $write, $except, NULL);
) o; f& ?. Q' ^4 |" t( l
: Z: J) n0 H1 y( w/ C0 a
foreach ($this->sockets as $socket) {1 h' w3 K- d+ ]) M# ]5 L
//连接主机的 client
; ]/ U% v! J% F9 i
if ($socket == $this->master){1 z: g# u# F/ U6 Q, s
$client = socket_accept($this->master);0 P% U7 |* q7 M0 [
if ($client < 0) {
H: L+ U* n" `
// debug
3 d/ u8 P( q+ ^" q& h, T! F5 I( T% M
echo "socket_accept() failed";
+ G( E1 y) q5 |+ w+ @/ N+ \: E. J
continue;
$ R. h- p7 I; r: p/ ~3 P4 L) G
} else {
' N4 t6 H1 C/ S
//connect($client);( T8 a5 ?9 }' v
array_push($this->sockets, $client);: E: s4 i- [% i$ @
echo "connect client\n";
. W2 L) g" v8 V
}
$ M- E4 ?3 m! }/ o1 M
} else {
1 M- p% n4 @/ z2 T8 Q* ?, l
$bytes = @socket_recv($socket,$buffer,2048,0);
) h9 X" u* {/ z' D
print_r($buffer);
! M9 g% p `9 t4 {* e: ~
if($bytes == 0) return;; e8 V! w2 }0 I1 p
if (!$this->handshake) {7 w" P) G* s {: o/ K8 C
// 如果没有握手，先握手回应
9 p( E8 E( S8 L/ b5 A
$this->doHandShake($socket, $buffer);4 x" C; T1 m6 C# c8 h) j. k& j( [
echo "shakeHands\n";
2 b1 b3 `4 y, a
} else {3 O- l9 Z9 B5 [* W+ j. X* s/ V
7 P5 j* C- E& V5 W/ g
// 如果已经握手，直接接受数据，并处理& c/ u) `( v& _
$buffer = $this->decode($buffer);# j1 V. ^( P3 k4 _. l. V3 P+ x( e
//process($socket, $buffer); , ?9 E$ E6 Z" D3 A
echo "send file\n";1 P6 n q4 P) [
}
, N; `8 D, |& h& |
}
, N# @% F# }* Q$ P, K7 a
}
6 H3 X3 S; W; ^3 _4 p* O% Z$ B
}' j! q/ k+ T: i' G! L; H- E6 s' I
}
) k& l( |0 l# {3 b* [" G7 x' j1 w
% C9 [: Q8 T; B. N& U
function dohandshake($socket, $req)) |6 n* n W3 ]" ?% o
{3 X" g! H7 \. u
// 获取加密key
1 p6 o; a" b9 r& p5 ]
$acceptKey = $this->encry($req);9 }# s" L3 X6 Q: Y
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" .
+ I/ W6 Q" d4 ^$ h
"Upgrade: websocket\r\n" .* s8 u# f& w: s
"Connection: Upgrade\r\n" ./ e, k8 `& ?, j+ B7 R; ?+ ]/ e7 z
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" .
3 r) }, @9 [) t) O Q4 `
"\r\n";! E; ~3 g5 ], S0 k6 s; g; Y
9 q" e% ~6 K( x
echo "dohandshake ".$upgrade.chr(0); 3 J6 R1 ?$ j2 F- [8 ~+ B* x" [/ f
// 写入socket! H% Q* r# c M$ ?9 P
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0)));
0 i4 { Z* }# r5 w, e# G$ Z
// 标记握手已经成功，下次接受数据采用数据帧格式( A2 |( L% j& O* }
$this->handshake = true;$ o B& `! W% K% E
}; i- ]( F' W; T* B/ m9 i5 b
( S8 ?% _, S) [7 {: }
6 |7 l5 W! `+ q( ~
function encry($req); I g2 V4 ?' a* r! W. f; B
{' t! \6 d% b* r- p3 ?. v# l
$key = $this->getKey($req);
( m+ ^* ?! X) I l
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";8 X6 \& u/ g' S9 [4 X
, o9 h' P# Z9 J9 X* }
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));4 L1 ^4 h0 Z2 e$ m
}
/ o9 U1 b2 [* ]0 L' I
: F; Q1 b0 q% @
function getKey($req) # e4 @% a8 D# f2 v7 Y- ^; o
{+ t" H; p& }0 ]" g9 r& | W+ C
$key = null;
$ |& {& ~- j4 W: [; l3 l
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) {
x3 p+ T0 a- @0 X5 b/ E/ d
$key = $match[1]; / K& C% `; e5 m5 L
}
6 I6 K# p% W3 I) \1 f: X/ G
return $key;
/ {6 K. E+ I, u0 x3 h0 w
}* Y: }( { @& ]% z% `0 W/ i& p
9 J; s4 O9 n( v- O, i
// 解析数据帧, e4 n, ?, y- Y- G4 I
function decode($buffer) + q' i1 I. z- K: Z8 u n8 z
{8 V6 X8 `* N g% e
$len = $masks = $data = $decoded = null;" x I" F+ K2 W
$len = ord($buffer[1]) & 127;
& @& B n0 l) W6 C
0 ]3 r1 e' k' s7 I
if ($len === 126) {
3 N5 B- T/ k% u6 g. @- s
$masks = substr($buffer, 4, 4);
, ]/ n8 l& z' o4 c5 K6 o( h
$data = substr($buffer, 8);
1 p. ?+ O# b0 S- S4 o* K8 `
} else if ($len === 127) {
6 z# e6 V7 W* ]/ M8 N- l$ r$ v
$masks = substr($buffer, 10, 4);; U/ K- c+ E: i% M* R+ l
$data = substr($buffer, 14);
, h4 Y) {. C6 X( [7 t) a
} else {
& C( {+ j( `( z0 s7 G O9 e
$masks = substr($buffer, 2, 4);
+ g3 m. D3 x# z6 Q3 @
$data = substr($buffer, 6);
; \8 o, x7 g+ @( R2 O
}
! Q b' ~' d- P
for ($index = 0; $index < strlen($data); $index++) {
# Z! v' A! C, ?+ y" K9 A$ M
$decoded .= $data[$index] ^ $masks[$index % 4];/ N9 x) @9 O" [! v" }' }" w
}
; L2 c' v' `; u7 d
return $decoded;* {+ {9 @ ]/ D+ i
}' Y0 N" p. B7 t2 I e
' P( [ L; u j8 K& x, o
// 返回帧信息处理) ]/ M. S8 ?; k4 Y
function frame($s)
. d1 [8 L& g5 C8 ~
{
3 v& L9 G: U* T: e+ P l! y4 G
$a = str_split($s, 125);; C8 G! k3 h/ z
if (count($a) == 1) {6 s0 P* i' ?4 B
return "\x81" . chr(strlen($a[0])) . $a[0];. R# l% o/ O7 `3 r; _5 v
}" _8 i3 m- C4 @0 H/ S
$ns = "";4 @4 J' {' V' r/ T
foreach ($a as $o) {
7 V; {4 B! S K( F7 b! K! z8 H
$ns .= "\x81" . chr(strlen($o)) . $o;3 @- ]6 t) I9 | b [4 z
}
0 J& V' h O) P9 [5 P; e
return $ns;
- y+ @+ C4 B# h6 q. [0 A* H3 H
}
, h$ B3 P3 F: c* ~" `( m
- f9 P% q( _# O) y, i
// 返回数据6 [6 G2 v8 [3 s' E c# ]
function send($client, $msg)
* A2 O+ W# i. f
{/ O+ {; h8 }3 s0 h' |8 ]
$msg = $this->frame($msg);7 j. ?' i* h) {& |
socket_write($client, $msg, strlen($msg));8 ]% V; ~! \9 C3 O; m5 ?* q6 i
}
7 |* l6 c5 [9 }* P0 C" t
}; b$ E! h4 U/ U
# V% v8 p8 l8 k) I7 p! I
测试 $ws = new WS("127.0.0.1",2000);
3 P" D& L& _4 S' D; G
" i+ D3 q8 ~ e! S' m

复制代码

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

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

浏览过的版块