PHP 简单实现webSocket

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

1）客户端实现

<html>
7 M! Q. p a" q6 s
<head>
/ \- ?3 W& C3 D9 ?
<meta charset="UTF-8">/ `6 Z8 L& j( [
<title>Web sockets test</title>1 E. }% E |' z& C
<script src="jquery-min.js" type="text/javascript"></script>
. U/ b+ E# n5 R L: j: L
<script type="text/javascript">
* u* X W* |" V% F$ x8 R
var ws;$ F, d7 p2 F8 V8 q: ?
function ToggleConnectionClicked() {
- n1 J! w3 g% x0 B3 E
try {
) o5 q7 a$ N! b; h
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器
* t% } i! s3 G3 S3 q
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};
8 I! P: T2 z( c9 z6 E
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);};
s8 n5 M' q$ K+ B( I# D* ?
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);};* x4 z% P& j% n. S6 n
ws.onerror = function(event){alert("WebSocket异常！");};
& H9 ~) {2 h# U- g$ J7 J; X
} catch (ex) {0 F: Z9 U6 f# A7 J1 |. o
alert(ex.message); 5 w0 Y$ d/ e, R9 p
}
. z+ `5 A" c) i# f; g
};4 W4 B0 E; z A4 M1 O
0 E' q8 D4 V( c) z+ G( _8 l+ I8 ]9 c
function SendData() {/ U* v. b% t- F9 Z4 s) O4 A" o! j: H
try{4 e* v: R' y; h, b2 t
var content = document.getElementById("content").value;" ~ n4 Z7 N. t
if(content){
* u- D3 ~2 g ~
ws.send(content);/ n# f/ D7 b5 W1 |) y& A L/ v
}
/ @' w( r1 \. u1 W/ ?8 d Q/ A
% @: m5 [7 ~$ Z9 b, y+ V( J
}catch(ex){( Q# m% w0 L2 o8 ?2 ]. c( t
alert(ex.message);% E5 d; ~! ?9 o+ \3 l# N+ c
}
, s: y3 P: ]* z! p3 k P
};0 |9 E- ]9 a. h+ k' U! ^+ b
+ w$ n7 W( Q& {! p% N" O
function seestate(){
- v# {0 ^7 P! q; s7 g: L" D) B
alert(ws.readyState);
( f) k& X. [5 F# s9 Z# m3 W9 T4 b
}) t7 I) ^" V2 x$ e5 E
4 X% t# H4 S. L- F* R- W
</script>
0 ]7 c0 z/ y. E+ Z1 D* i
</head>! l9 H3 }( v: Q/ @9 L
<body>
3 Q% r7 c) j! X+ z8 ?3 J
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br />, {% M$ @3 s: I
<textarea id="content" ></textarea>
A3 L8 ~- h6 Y. T; U4 C; g
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br />
% ] ?, z! x" N+ H
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br />
- B6 Q& [2 n- {" P6 ~
& _7 f3 Y2 j9 z: o' q
</body>
3 z% W' E0 R3 e& E' e5 z. ^( Q
</html>
* y) V, U8 F* o0 t" @0 z. U& n

复制代码

2）服务器端实现

class WS { `3 @8 |# {" t K7 V7 W1 O
var $master; // 连接 server 的 client
6 L7 L5 x* c3 }; Y
var $sockets = array(); // 不同状态的 socket 管理
$ N G3 j9 Q# l# H# Z' V* E# V3 `
var $handshake = false; // 判断是否握手, w# q4 D; n" L
! E# A& _/ Q; E3 G2 e
function __construct($address, $port){6 _4 N9 G) v' C7 R5 V; P
// 建立一个 socket 套接字
8 T7 k3 A4 n6 t8 F% ^
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP)
6 D0 y8 t% a- R% D; _' h& s
or die("socket_create() failed");+ {. O6 t" m( m% O
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1)
# b8 c- P! B# [8 \" M, W
or die("socket_option() failed");/ h8 a+ k9 E! V1 i% p! H4 g, B& k
socket_bind($this->master, $address, $port)
/ o! g( Q% M) c2 Y5 ?/ X& I. q
or die("socket_bind() failed");- F- i @+ z, H3 V Y: N: ?
socket_listen($this->master, 2)
! z) S: S; `1 ]5 F ^
or die("socket_listen() failed");- o5 u1 q% N3 G# w
- J$ G. c* c' ]+ n% ~- n
$this->sockets[] = $this->master;, h5 p( ^4 O/ J# F
# N3 m$ m$ Q# }( V* @5 B; V+ _
// debug
" r2 ^' b& \ L- t& o
echo("Master socket : ".$this->master."\n");
% C% h0 S4 Y5 B" R8 C! [
. G* j, W. y; H" T7 ]" C
while(true) {+ Q7 j( T; Q6 K9 f: w4 F4 k0 F
//自动选择来消息的 socket 如果是握手自动选择主机3 Y- e; M% G$ u0 w
$write = NULL;
9 A; x: K! K: x9 T% @: t
$except = NULL; x, m" r& v2 ^. v* g/ z1 l3 D6 }5 v5 M
socket_select($this->sockets, $write, $except, NULL);
4 |, H) G$ R1 Q1 E# |/ C1 i+ I6 z
5 j5 i3 y' ?( G6 U$ i. _2 H5 L
foreach ($this->sockets as $socket) {( k1 v% e4 m: i5 S" }: R
//连接主机的 client . @$ l) t$ _ P& C2 v0 `1 E
if ($socket == $this->master){2 h7 Z, @' O' h9 n# Q
$client = socket_accept($this->master);
9 _6 {/ g/ H9 n( {5 n) y' D* M
if ($client < 0) {
( J* h% k" e6 p" N% p
// debug" Q4 {& l' A) N( D% r8 f+ w1 P( W2 ~
echo "socket_accept() failed";1 F0 ^* |7 N) P3 D6 Q
continue;) L" u7 J: z( J
} else {
# k4 f8 G; J: l; V+ G) M. c3 c/ A
//connect($client);2 [+ Z" V k! K$ C
array_push($this->sockets, $client);
" A6 e- v- x$ o% j
echo "connect client\n";9 ^- a$ l4 d5 z: D: s
}( b( ^/ F1 G) F# D
} else {6 Q" e1 s, B9 E( i; N
$bytes = @socket_recv($socket,$buffer,2048,0);
9 w3 _$ F m2 |9 C/ Z" X4 y
print_r($buffer);
' a# I' W. p9 k0 B7 t
if($bytes == 0) return;% }6 F+ ^+ s. [6 h
if (!$this->handshake) {
% n, w- Q' h' t. `" F
// 如果没有握手，先握手回应
3 I3 U5 h% W ]% w
$this->doHandShake($socket, $buffer);
9 t! z L: I* u( I4 R# X
echo "shakeHands\n";+ O/ X# M. }1 n0 R; G- [
} else {
& N0 J* F- ^- Q4 @4 `, S2 @
' i# E" z* S' g" S! O4 ]
// 如果已经握手，直接接受数据，并处理
* V2 v4 x4 P8 j* b! w
$buffer = $this->decode($buffer);# |+ t/ r, d6 j3 }) Z
//process($socket, $buffer); 0 a& R. ?# _7 m2 |
echo "send file\n";
3 S: [( s9 H6 b& \ }! z, Q
}
9 _* M x$ x7 [3 D0 _+ `
}+ o( B2 o9 t9 i- U' \
}
" Q- ~0 O& l+ C2 h; Z* E6 J$ I
} P1 ^ X1 h, ]% v E5 u. ~
}! Q' x+ ?9 s9 L
$ Q' g8 X3 h2 M2 o0 s ]0 ]
function dohandshake($socket, $req)- h% q0 k! ?, ?( N% ?% ]
{
+ f8 S; e' s8 U- v: j) x
// 获取加密key0 f5 N& W, Y& I& F& d2 L% u
$acceptKey = $this->encry($req);
/ T g4 }0 ~( n/ R- D, d
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" ." S2 d' s0 R0 l/ T
"Upgrade: websocket\r\n" .9 W q" P4 L! I* c$ P/ R c
"Connection: Upgrade\r\n" ." u7 x2 m$ m+ D& r/ ]4 [
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" .& k! f& j" o0 p8 a' L( M: x
"\r\n";( a# }( I' s6 q4 T9 n( D8 r
4 O3 F) K* g! F- U& a9 p3 u
echo "dohandshake ".$upgrade.chr(0);
3 Y% A1 R" e" V z$ p; B$ S! u
// 写入socket
2 n0 V4 P- e) P7 j/ }3 T! l
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0)));
8 m+ j. n9 q: X- r6 ]/ o. e) ]2 }
// 标记握手已经成功，下次接受数据采用数据帧格式' Z$ p* h6 E3 A9 F$ `
$this->handshake = true;
+ _& K9 o7 r) {3 W0 ^4 T6 S
}2 v9 s: S7 C$ v P5 B4 `4 P
- K: h+ j F. e9 G7 @
/ \' c) {! g. Q4 @6 Q: o
function encry($req)8 ?+ s w# F* q6 Y5 o; w: v
{* {- Y3 U+ y: U: w0 C( f' B/ @
$key = $this->getKey($req);
9 y1 g$ B, L, ]+ `% L H0 p
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";! G+ T% j- p' q, o( v
: F3 ]& F9 K( r- M6 j9 n+ F6 t
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));
! c( |# p8 c: f
}; y) ]. v" {9 `6 `+ B$ X8 Q- h
) I$ B7 k5 s% q
function getKey($req)
& z3 `, L: F7 J7 x7 s
{
d! x y3 W, O1 K( H/ X/ k
$key = null;. E/ m1 ]8 Y/ Q9 e' J" t% E" U
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) {
) T8 k+ q1 \2 I( [) Z R' j
$key = $match[1]; 0 K0 r7 Z& C( R9 V8 W; J
}8 [7 S8 Y% @3 u( [5 g
return $key;7 _& Q: S2 S4 H# u r5 N
}7 d+ l" v. D; Y/ Q! y* J
$ g! `. s; Y5 F) O& Y/ r/ g( J+ t
// 解析数据帧
$ k( W% Z/ q! [% F
function decode($buffer)
8 d4 ^& b( f/ Q) Z
{1 J! f9 [+ j4 K2 M4 H
$len = $masks = $data = $decoded = null;
D1 i' @8 J+ Y2 k
$len = ord($buffer[1]) & 127;
) \5 Y: T5 u; Y: N, I
2 E: X, {2 h. e& H' m3 T
if ($len === 126) {# M: K: C# O ]$ B
$masks = substr($buffer, 4, 4);) s( v- e# `% R% d: z2 p( g8 z
$data = substr($buffer, 8);/ X4 t* I. T E9 @) y
} else if ($len === 127) {3 \$ j& N1 U. f+ s) }& D! ]: V+ f
$masks = substr($buffer, 10, 4);
' f( M) y8 W* E% n$ O- m
$data = substr($buffer, 14);
% A3 N0 b# u4 s g7 F4 Q! q
} else {
9 D' I$ Q8 q! W7 p! [! z
$masks = substr($buffer, 2, 4);
+ w5 c: [4 K. @9 M
$data = substr($buffer, 6);
+ S( N& X. F) Z- P
}2 G0 c% v0 _) J# _9 A7 N
for ($index = 0; $index < strlen($data); $index++) {3 a5 @1 H8 v3 b% }: ~" e
$decoded .= $data[$index] ^ $masks[$index % 4];6 ? \7 D# @/ _# h, m! E/ D
}
2 k" I0 v8 t1 V+ A( F5 b
return $decoded;
- F# o5 X% `) o( y' w8 f L
}
4 s7 h. f5 [- L4 W/ P/ y
9 W5 n- ~4 c1 f; k7 m1 S
// 返回帧信息处理
0 q9 d0 U/ g2 o$ d! Q
function frame($s)
{6 b1 p3 a$ j2 S- W2 x
$a = str_split($s, 125);
2 n: C8 m( M- N: Q9 P) [
if (count($a) == 1) {3 ^/ b% |9 Y7 s+ `
return "\x81" . chr(strlen($a[0])) . $a[0]; a9 v# j4 \. L+ \! R
}
# ^- B! F% ~" P+ z
$ns = "";
+ R e+ P T' U. h. [
foreach ($a as $o) {
6 H/ U8 v4 q# S$ Z7 z
$ns .= "\x81" . chr(strlen($o)) . $o;, J: ]! I( ?! a5 G6 v
}( n7 D0 M- q$ q3 R7 J, l. a
return $ns;
9 M6 D3 [' C( N& u
}
: o- M1 N/ |9 i r- N3 m$ T
1 K. d+ D$ V- l6 p% y( |
// 返回数据
, ^+ b+ J% O" i: e2 k" w
function send($client, $msg)
; S: J! J2 c! [9 X
{
) @ R$ B8 u6 S8 G o9 r. u- q; b
$msg = $this->frame($msg);
" Q( I0 O ?: m& } H4 }, a, n
socket_write($client, $msg, strlen($msg));
4 g5 w; f4 V: }
}
v! s$ F; _( ]: u, L) s! M
}* U* a' o# I' k! j: L! V0 A- D
8 P% E8 f. N% }1 D% X' m
测试 $ws = new WS("127.0.0.1",2000);
( \0 d' B2 R i% j" T
. i4 [4 ^& H. _! q$ Y! i$ X

复制代码

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

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