PHP 简单实现webSocket

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

1）客户端实现

<html>/ |0 j8 N, W) d |5 u
<head>
- |' K( Z0 \$ a/ Q
<meta charset="UTF-8">7 L0 ~5 K0 |0 T2 O* i* H7 W
<title>Web sockets test</title>5 e6 f n( V/ c- o% x
<script src="jquery-min.js" type="text/javascript"></script>; N; V" [* n3 g, D0 r5 }( z ^2 u, o% F
<script type="text/javascript">/ S% z% f+ D4 q
var ws;
& x m, C V2 Q5 A3 i+ p
function ToggleConnectionClicked() {
" P7 Y$ \0 \( j4 o
try {
, `, P1 K4 G3 s3 T5 i E8 k
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器 7 D+ H- t4 @" s: p
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};
! \; q0 B( T! Q+ X& j; O3 Z
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);};
! Y* x+ c5 y3 ^3 A' u; v# c' k
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);};! `# B! `. d, G1 h0 c( }
ws.onerror = function(event){alert("WebSocket异常！");};
3 }) L6 C, Q5 |- u1 `
} catch (ex) {% b2 X. X# j T K* y }
alert(ex.message);
' g6 b; K1 _$ x; G3 z
}
8 [) E g+ T W8 O6 _4 w) t7 P0 {
}; H& d7 G) b a) q' O
" Z6 `* S b) J& B4 w
function SendData() {
' e& j9 ?. K9 S2 t7 C3 [$ x
try{
# W" z, q4 ^+ _( h
var content = document.getElementById("content").value;
6 [) Z: D; q' A/ E& V) q' ?( k! F
if(content){( I3 c* Q% G/ q4 J/ L7 d
ws.send(content);
$ r3 K$ X, O' `; v! X# B* N' \
}
4 c8 ?4 T2 E* t6 B3 j1 L
' ~7 r5 @' s$ J" c A" }
}catch(ex){
3 ?# ^; W8 z! m0 C( O0 f7 @- r
alert(ex.message);
) c7 ?6 T9 i9 d6 N' O
}
~+ [8 ?, C) O" t5 k1 ~+ p3 t
};0 K F$ N* m6 J, S' U
4 w. b; g. o4 ^6 |
function seestate(){
4 M' C# y2 e/ E4 v0 C5 O, w/ t
alert(ws.readyState);
7 \; A R3 d1 q% [9 N8 j
}
8 b% j, e6 D0 C/ B o6 ^% ]' ^
/ |; C7 T# @ ?4 h6 G7 [( X J
</script>
3 a5 y" d3 P7 P
</head>
; \: }8 H; D( C+ E
<body>1 `7 @5 ?6 O* O( t
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br /># c6 L- H+ ? g. f
<textarea id="content" ></textarea>. _, [2 I/ g* o O8 S; m, [) G1 z
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br />
+ {! V* r8 T( }9 B! o
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br />
0 L4 C7 {" `" X* [
8 f7 q2 `$ c1 Z4 t% I8 z. f
</body>
- s3 X, Y, H, |3 C I/ N* f5 q
</html>! A/ r/ Q& G. w9 u# e

复制代码

2）服务器端实现

class WS {
) H2 ]0 Y# L+ I9 I* _5 M; z) Y
var $master; // 连接 server 的 client8 g2 k- U) O5 W7 V2 ?& `- b6 d
var $sockets = array(); // 不同状态的 socket 管理
+ c c m' ~7 ?& \$ ^% m
var $handshake = false; // 判断是否握手
( y1 s5 q$ O3 C, _& |
3 t7 i& N! I: Q* X% s$ a& Q9 m
function __construct($address, $port){5 P/ ?9 h( y9 \: O/ n. [7 H/ `
// 建立一个 socket 套接字6 E2 e' @) K% S; y+ l$ Y0 s- f+ J( }
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP)
/ C) q5 }& U |: G
or die("socket_create() failed");; E0 e% w# k0 G, ?- f) a
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1) , z7 T, ]6 ]. B0 a% W4 q
or die("socket_option() failed");6 R4 ]& J3 {: R
socket_bind($this->master, $address, $port) 5 q3 H m! `' K" h/ x$ j5 d" X5 {* j
or die("socket_bind() failed");
: L8 L0 s T' F+ X2 n
socket_listen($this->master, 2)
; I0 H) _& K3 q6 S9 y2 i
or die("socket_listen() failed");9 L9 h6 H. D) b' I3 u. E5 J
: P+ A: a2 y4 S& a* K* E
$this->sockets[] = $this->master;
9 t4 T9 W7 A$ ~% [
2 Q! M& p/ `* A
// debug6 x3 h/ D" t3 u/ T
echo("Master socket : ".$this->master."\n");
7 w8 B" l4 U6 M) d- v; y
. V/ L5 I. s7 ^
while(true) {1 u2 B8 F5 n( m3 U
//自动选择来消息的 socket 如果是握手自动选择主机
& K+ d2 u0 x' A/ }: r
$write = NULL;
) e6 i' O3 _' k4 @0 l
$except = NULL;, t9 ~( F. M- r, k7 J' l( B5 a2 e' a
socket_select($this->sockets, $write, $except, NULL);
, l7 m3 U5 ~% _8 X, ~
- ^9 X0 x. A% z4 Q! A* D
foreach ($this->sockets as $socket) {! |5 r+ ~. S5 R. W8 V6 v
//连接主机的 client : z0 Y, S8 p4 P# m
if ($socket == $this->master){1 V0 ?6 b' Q/ V8 Z
$client = socket_accept($this->master);
! h7 u* k* r* L# f
if ($client < 0) {
// debug
- |$ V6 {1 A* i; r/ r
echo "socket_accept() failed";
, t/ |1 Z/ _, H |
continue;; m( C# ]8 {6 H K4 C( x
} else {
5 D7 h2 _* r) H; l F/ z
//connect($client);* q6 G/ F: p+ H8 F, G
array_push($this->sockets, $client);
. ? c2 f2 }5 a5 [0 Y" F9 M+ `
echo "connect client\n";, _# O, d) s- K4 B! ]7 s8 c2 m
}) J. Y. i$ Z* E: W* O4 x& C
} else {5 T) W3 z/ [6 Y0 C5 Q
$bytes = @socket_recv($socket,$buffer,2048,0);# D+ A2 s2 A4 s8 s6 E5 T- `
print_r($buffer);
, F5 U1 h' U. ^) U n3 K9 I
if($bytes == 0) return;$ s0 ~/ Y& m: X( ]2 k
if (!$this->handshake) {# B5 C2 w& B1 }* Y
// 如果没有握手，先握手回应
" i9 _6 e& l7 P( }, M
$this->doHandShake($socket, $buffer);
% R2 d9 z% N! X
echo "shakeHands\n";8 }! a3 f5 T. w
} else {
3 Y! E. t9 j6 d( Q9 {
; H7 O: x! F* |
// 如果已经握手，直接接受数据，并处理
9 B0 | e4 E# j
$buffer = $this->decode($buffer);" ^3 a5 b: P0 f) N+ i
//process($socket, $buffer);
E- @* S1 E4 _$ s! I. H! a, `; X
echo "send file\n";
" o0 X/ ]# m, i5 D5 Y
}
4 R, C# ^6 K( J: X+ C, z) p
}- P3 S- ^2 G( B. R- |! v: ^
}! s2 b9 b9 Z7 e9 i, D* w1 S
}
' M. {8 }2 X& Y5 {2 u# A U# G
}
6 _3 f) Q! d/ H. K) U+ o
1 X+ v' t5 t/ Z
function dohandshake($socket, $req)
+ |$ W4 ^3 B. B# }
{0 m8 x0 O2 Z+ @1 \
// 获取加密key+ m7 Z. \; Q; H3 Z$ c4 n3 z
$acceptKey = $this->encry($req);
' l$ K3 q( o+ \! |- P! c% o- l6 `
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" .
6 k0 t- } U) E2 o3 l: \" |
"Upgrade: websocket\r\n" ./ \* ~" H$ {4 Q/ l( C0 K. M' k
"Connection: Upgrade\r\n" .
! [! @ p$ O- R
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" .
$ U; o) K$ I6 i! |# d7 J
"\r\n";
: G$ h* p( c7 ?2 O0 C* B3 q8 l
, W. J, {8 l9 U3 \) t! l; z2 ]3 T
echo "dohandshake ".$upgrade.chr(0); / [5 y) l& M+ \) c8 E6 e
// 写入socket
6 R2 e# B2 `! M* w5 O; _
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0)));
$ ~& I0 s `8 B" _+ ^+ d' A# W. ~5 u
// 标记握手已经成功，下次接受数据采用数据帧格式% u. _/ C+ X6 a8 l/ m2 P9 g- G1 @
$this->handshake = true;
& Z5 Z0 _* R' T! A" C3 ~
}/ @1 ]/ k( b( m J
0 |8 r1 ~2 P. h7 h. L
* z. k8 V- b% t: L
function encry($req)) o" H6 o5 K; }( l
{& g$ h0 p0 F) _6 y
$key = $this->getKey($req);# J9 b4 P6 X, Z6 r. z
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";1 F- d9 Z2 `* n: g: n4 X1 f% ]
' r1 @4 f% c8 T5 ^( `: ~4 S
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));
) v- p/ d M/ `0 G+ C
}* n5 W$ ]) S; @, [8 |( E
; }& N3 y9 e& B3 [+ Z* ]
function getKey($req)
1 w" W9 r- o8 D5 J3 V
{# B: D; q' N8 T3 d
$key = null;( Y m% C- e& h. ^6 E6 n- ]1 S
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) { ! F$ W2 Z$ Z. H! p( Q+ I
$key = $match[1];
4 [" B/ Q$ J/ A9 k7 Q. J/ [
}
" |9 N; y$ ?: Z
return $key;6 s' [9 B/ C" c/ t) D4 }8 K; M
}( d; L+ `4 T5 Q* [9 `- [( K
" H& V3 W( Q9 ]- E/ z2 ]6 u
// 解析数据帧1 j# H$ a6 C, Y* k1 b9 P- J& @
function decode($buffer) 0 H# d, Q0 ?& R! _ z/ t/ H
{
% M) w6 [3 M: f8 g0 g" C
$len = $masks = $data = $decoded = null;+ e, l7 B* A+ D1 F
$len = ord($buffer[1]) & 127;$ \4 \& I# M r4 Z* N
2 {' Y. K+ k. P3 Q& G
if ($len === 126) {" z9 B7 K# b1 E! J
$masks = substr($buffer, 4, 4);" `( f5 y) ~8 l3 y' D8 I# s( \
$data = substr($buffer, 8);3 l7 G7 l* \; r7 I# j8 Q
} else if ($len === 127) {3 H, o% [3 ^& Y! q5 F) i
$masks = substr($buffer, 10, 4);6 W$ _% `% \2 c" @
$data = substr($buffer, 14);
& i2 ^) t7 ?& n: {
} else {8 W/ f+ V% f* f% A
$masks = substr($buffer, 2, 4);
# j7 Q3 z: T5 Y5 ^ D' v
$data = substr($buffer, 6);
2 S$ B6 [/ i( G% g2 b
}2 s! E X) G" y' ?' x) D
for ($index = 0; $index < strlen($data); $index++) {3 M8 |6 g* H% z2 t9 h! x
$decoded .= $data[$index] ^ $masks[$index % 4];
( I8 Q* D) l' A5 ^8 K, D
}& K9 F9 g- R( e. R- h0 M
return $decoded;. X. b$ Z2 f6 n6 L+ k7 t
}
: T2 g+ U" j& [6 o7 y
, Z+ c. S7 U; m. ^+ ~; P
// 返回帧信息处理
& v; \2 Z; x2 r4 V$ l; g. z
function frame($s)
9 k9 V$ J8 u) _9 [* H
{6 [2 n2 E: s* v, q- e8 k# j
$a = str_split($s, 125);
( B3 G3 O0 x* h
if (count($a) == 1) {$ P, u- p" S0 ~% l& o
return "\x81" . chr(strlen($a[0])) . $a[0];
7 M3 d+ \+ t# f; T5 V# m6 g& l9 |
}
# z6 f" \) T, a8 X% @
$ns = "";; x$ A! B4 |! }* D# h8 M. s
foreach ($a as $o) {
l% l3 r$ J$ S6 D0 c. F
$ns .= "\x81" . chr(strlen($o)) . $o;
& m8 G& z! M: _/ `+ n9 l0 Q
}' X$ }# ]) C- O u; s% L
return $ns;
& m' o% d" _" N
}
& ~0 K( E# z* Q" w
. o4 F+ j4 W( Y9 t5 M8 @% D
// 返回数据+ a# s% \* N, L W
function send($client, $msg)6 b- t- C5 m3 e, g, L$ g, [6 x- B
{2 s/ z) E6 }3 _ V8 N" z
$msg = $this->frame($msg);
8 n! E2 @/ Z% E
socket_write($client, $msg, strlen($msg));# J6 U# T- d8 j/ f: r
}
4 h3 h. I' i* K" u. X& e! ^, s
}7 U, i/ L [0 v- x: B
3 {2 W, T* f6 S
测试 $ws = new WS("127.0.0.1",2000);
, k; u% X/ {# m& L
3 ?9 K v' _: ?6 N: w

复制代码

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

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