PHP 简单实现webSocket

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

1）客户端实现

<html>
. h7 E9 O# s: }0 a L- d
<head>$ U5 C+ r6 c; z+ t8 y8 c$ n. q7 K
<meta charset="UTF-8">
- a! X' j7 C7 Z$ I; m5 ]% |
<title>Web sockets test</title>( i. u5 T2 b" s# [9 [6 B8 ?
<script src="jquery-min.js" type="text/javascript"></script>' T# X3 Q/ K5 ?1 A& C0 e% A6 S
<script type="text/javascript">0 t, S. K* H' y6 i3 D3 ]& [
var ws;
+ F" _. U) D w% M. }, p) |' C" U
function ToggleConnectionClicked() { 5 T7 k$ z/ k& W2 C# X( s& r* |* z
try {* i! ^6 {! r: Z3 U- k3 h1 p
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器 ; F+ O( I4 I1 T# ~ P5 g
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};
& A0 s" L" q( z* F- P6 C& L
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);};2 G0 _$ N, X3 [5 k
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);};
: ?0 E8 L [, ?1 ]* v0 k
ws.onerror = function(event){alert("WebSocket异常！");};
5 B ]9 _) Z$ C, R' G' p0 a
} catch (ex) {
8 _* H$ X* \. _! ~6 g' K9 ^
alert(ex.message);
- m2 d6 ~/ K+ ~: _
}2 `& n8 m% ^3 k- q% M2 Q. p% v ~' k
};0 d& p/ t8 c: L( h- g7 z
+ m3 u( ^. m2 _. u% u' Q
function SendData() {
# n+ `8 M* N1 y7 K0 t0 r( d" n1 v, ^
try{& I/ ^3 z7 g- b4 P/ ] z
var content = document.getElementById("content").value;4 _9 M8 s5 e8 h5 O% _. J( ~
if(content){" A* O. l: p% }, _; i0 B
ws.send(content);6 t% v0 w0 i* u$ ]9 @
}
( i/ S' t: h! N# B' V7 P/ o
6 o1 a, L- o z2 Z$ |
}catch(ex){
$ x" `4 O& C& D* C1 {
alert(ex.message);
3 V z* g# {/ {0 D# N8 @
}0 ~# R' [5 w- c, e8 l5 z
};
4 @, S) `: |( O. _
; i$ h5 g& v5 X1 R5 \
function seestate(){
+ W. D( N9 Z E
alert(ws.readyState);2 L$ {2 P$ \3 b; @0 H( t
}
- G4 H) h6 N& [+ q/ N6 i
9 g% m g; W C
</script>
" k$ \# v4 O" ]
</head>* h" O% u5 y4 ` V# x% I+ X
<body>+ e0 u! r' Y; N
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br />
8 a0 S9 B) U, s4 v# T0 y
<textarea id="content" ></textarea>
- T( L P1 o: z1 C7 D9 v' V
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br />
* V: l6 b+ _4 Z
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br />
M! o1 z% u& x
, v5 D/ M! N; ^2 L' Y- r, P
</body>
' W0 I# \3 J( M6 g5 E. L# \( [9 l
</html>( m8 D% }; T0 h

复制代码

2）服务器端实现

class WS {
. z F: a! C- y+ [5 G! Q3 b
var $master; // 连接 server 的 client
$ h( @3 v9 F$ ?# u. v4 ]
var $sockets = array(); // 不同状态的 socket 管理" T1 @ O$ C( U5 `
var $handshake = false; // 判断是否握手8 @5 L2 q- D' ~$ Z3 E# Z
' H4 e" W7 j+ m% `+ l) `
function __construct($address, $port){4 u- a7 Y$ F! Z7 e) ^+ q) y2 ~8 Y
// 建立一个 socket 套接字
! c9 e6 L |' v( e9 `
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP) 5 k4 ]7 y8 V& L3 i
or die("socket_create() failed");
4 f6 g: E# g2 V! i
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1)
- [4 ^ f. t, J: `* X% q# q" `
or die("socket_option() failed");5 u8 T% ?2 m2 ^
socket_bind($this->master, $address, $port) " e6 a1 c6 K" w K
or die("socket_bind() failed");2 ~* H$ k) F4 t% n/ g( e0 @
socket_listen($this->master, 2)
& B Q2 q5 t2 [0 q. ?& d0 t
or die("socket_listen() failed");4 x& U6 B; \0 F0 o$ ^# h
6 b+ g: i# u l1 j! k- A" X: O/ \4 E
$this->sockets[] = $this->master;, g0 P2 y6 c; ?! P- O) @; P( x
/ V. Z c; a& [/ L/ \+ H
// debug
8 s& [0 f( V5 E
echo("Master socket : ".$this->master."\n");
$ K6 ?4 q! R6 s1 g( g; F) L. ^1 k
( A- l' y7 N, |3 Z4 t$ R
while(true) {' ]$ ^8 v" } w: e4 x+ y
//自动选择来消息的 socket 如果是握手自动选择主机9 s3 U5 J) A& U
$write = NULL;
, j, [5 l- n) Q8 @( K- u. i
$except = NULL;8 L4 _, X8 o# I
socket_select($this->sockets, $write, $except, NULL);
3 C8 R! F' D6 Y3 Y1 N0 [/ Y
* `9 R1 s; T4 m6 t
foreach ($this->sockets as $socket) {& w4 ^! p2 t; V9 K( z& q
//连接主机的 client
/ l/ R# O& {8 r3 p) G5 Z! o
if ($socket == $this->master){: F8 `+ b# Q# u- Q" l5 l1 f( J
$client = socket_accept($this->master);+ X5 u. D c: n0 [
if ($client < 0) {
, B; ]: _* D% E8 e
// debug0 }! e6 X& ~; @: w& |
echo "socket_accept() failed";' ^) K! h8 c9 ?4 s4 R4 S+ {4 f0 ?( S
continue;
4 @, G; q5 i1 ]: C+ o- ]* \) p& {
} else {
1 F. c" U' G K5 @$ |0 c
//connect($client);
; S, _" o% I3 ?3 t' W$ b6 T
array_push($this->sockets, $client);' o5 y8 W# y' S. I7 `; |/ W5 X4 o
echo "connect client\n";
. {* ~2 z( {" ]2 d# f% C
}* j: M# n- K1 O$ B) Q8 }% a0 G
} else {
+ U2 P) B& m8 d$ f8 M# m
$bytes = @socket_recv($socket,$buffer,2048,0);
. _/ O4 g, n6 ]' I' J, a
print_r($buffer);, y" c" x0 U& W9 C
if($bytes == 0) return;1 E% i# R2 A/ C5 U |4 K/ ]8 f9 E
if (!$this->handshake) {# c* O1 m4 A- [: N6 E$ `
// 如果没有握手，先握手回应
6 @) d) \" z9 J
$this->doHandShake($socket, $buffer);1 K/ M" I) n5 R& R6 W4 Y3 ]# d3 \
echo "shakeHands\n";
& x3 `" g {: w/ I; D
} else {
$ }" g8 T8 G# j0 o& ^2 [1 u# z
& G6 _ A$ `6 i0 L+ V2 R' d6 ]3 Q
// 如果已经握手，直接接受数据，并处理6 O; @# i ]4 x1 S+ [. J! ?" S$ a
$buffer = $this->decode($buffer);+ v; @# d3 m% ~0 v+ `3 @
//process($socket, $buffer); 8 H7 o; e- h& K* a/ |" I
echo "send file\n";! X, r1 P% [ d n, W# Q
}5 U# ]1 {& m$ f3 Q4 B) B
}
) C) z9 H- v: O4 J7 a E, z
}$ l* `9 K B% a1 H
}$ M; C& G. A, u1 J4 t
}
& W' v' T q$ `0 u/ @8 l
1 A8 M; e( A$ m: c& D( I
function dohandshake($socket, $req)
+ `# o. H& A E6 k9 Z' {, a
{9 k" e' Y: E& D
// 获取加密key
! W; I' E; B$ D/ @0 K4 s# D3 T3 y
$acceptKey = $this->encry($req);1 M4 q2 `2 { W3 m+ Y
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" .9 Z7 h( O' ~4 P9 ?
"Upgrade: websocket\r\n" .+ j) z; r0 Y# l' o0 L4 B+ M. \" }: s/ f
"Connection: Upgrade\r\n" ./ D' [! x, i* j7 d- s! F* q
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" .
( G1 x1 u4 C- I7 j# a
"\r\n";
# n& T. U$ F$ y
/ B5 J/ H6 ~1 N4 i; d' c
echo "dohandshake ".$upgrade.chr(0);
8 M: C2 U1 R5 P( j
// 写入socket/ n# G5 [/ s: i( }5 P+ O
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0)));( S: t7 w3 Y0 a, w4 W
// 标记握手已经成功，下次接受数据采用数据帧格式# \+ \/ m1 s( L' o3 e4 m6 w
$this->handshake = true;* v3 E* Y8 T3 x' |7 p4 T2 {
}* H+ {7 C2 H8 A* z1 }
% L5 Y( u# y' D
% S; V* K9 E. a/ x0 P, t
function encry($req)5 {% _: ]" `4 V( [2 `
{ Z% K+ P. J; e! y
$key = $this->getKey($req);
% l% E( k ?! K
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";; o/ r7 ]& X. I
( E" t+ d3 I/ D G9 q
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));
+ j: [: z2 ^- t3 x8 ~/ v/ C- O8 X* u) T
}8 q: h* |) H* m3 f" X2 p. Q0 t6 b
7 m8 d" o* {- H3 d
function getKey($req)
3 d& k7 a9 `3 L
{$ }0 G0 f, K. G1 c1 J: F
$key = null;
2 |+ A: o. ~! T" L& D8 l: o
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) { + _- {% S' e/ l6 S/ i% h/ L
$key = $match[1]; " W# y* a$ j z
}
- f4 ]$ t4 X5 m* ^" N% x% B
return $key;
; s% M" q5 P1 X% f: a9 u: V) g
}& O6 T( | D( }: X+ Q) F
+ C- [2 T/ N `! [! R0 i( U7 l* B
// 解析数据帧# b5 l ~+ p( `. V9 k
function decode($buffer) / ^) ?2 ?$ S) e8 X- d( E' p
{
* R) T4 l- u8 j' H
$len = $masks = $data = $decoded = null;
7 M2 j! Z0 n% ?6 b9 Y F" y5 [9 z
$len = ord($buffer[1]) & 127;+ ]; G6 P5 ~9 U
! [* ~/ k& f, o a' O) {! q
if ($len === 126) {. A+ G* p3 x) K
$masks = substr($buffer, 4, 4);
( K6 i. H1 ?: |- f
$data = substr($buffer, 8);
/ G8 e' s1 }6 w/ x6 v% w+ }& O. `
} else if ($len === 127) {
% T3 T9 X- ]4 u$ M
$masks = substr($buffer, 10, 4);
* B) q) L m2 P
$data = substr($buffer, 14);9 }/ B, N1 A0 z/ d+ K& V
} else {1 Q" u' n# H# k
$masks = substr($buffer, 2, 4);
0 x; W; n) Q: f
$data = substr($buffer, 6);/ O7 o n) H% {' w& m
}
' a- U$ v5 t" I# w& y( I
for ($index = 0; $index < strlen($data); $index++) {+ V9 t1 d1 |% r# |
$decoded .= $data[$index] ^ $masks[$index % 4];
% D: M+ [! `0 N1 \ M4 f
}
$ C0 }* C0 {" o: W
return $decoded;0 R) W( g- k' [$ \6 }/ a6 F
}
' b/ b, O/ U1 k7 ]$ x3 ~9 a# C
) n3 s- a' J* j# t; ~
// 返回帧信息处理
" @* @; `5 x" a0 J3 d( A9 A
function frame($s) 1 z8 h# M W2 o' Q$ h
{# d0 N' W1 _' Y2 V
$a = str_split($s, 125);8 X1 p1 n1 Z$ F+ _$ r, i! ?5 c H
if (count($a) == 1) {$ F+ W% e: z% H8 w- l* Z3 F
return "\x81" . chr(strlen($a[0])) . $a[0];
}; u- H5 W5 Y4 u0 [+ C u* W' l& q- d
$ns = "";
& g8 M8 |3 `" U$ D
foreach ($a as $o) {9 {6 c e( C, \& D5 T8 o8 D
$ns .= "\x81" . chr(strlen($o)) . $o;
* m# D' [& | b4 h
}
" Q+ I3 F# s- ]
return $ns; L8 v- J" |$ ]2 s& d
}/ B8 @3 n6 D( E
$ ^ T. G: |0 o8 j% J
// 返回数据
6 H/ ?0 K0 G Z4 I, \
function send($client, $msg)
3 I5 e1 ?/ r/ R& R" z6 k
{
1 ]3 k, Q2 V/ _ ] }, d1 n- D2 t
$msg = $this->frame($msg);
4 _, |5 ?4 f# y7 [ Z% z2 j
socket_write($client, $msg, strlen($msg));
. `. {! R9 Y7 e) W p" A
}9 J9 k4 }8 Y0 H4 g8 A; C7 ]- ~
}& i# U* h8 w7 ]
& ~' M( [3 j! d7 m$ ~
测试 $ws = new WS("127.0.0.1",2000);
; f6 m/ j: n2 j8 y5 T
9 ?2 _3 A. B4 ^5 T, R) c

复制代码

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

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