PHP 简单实现webSocket

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

1）客户端实现

<html>
) X1 |/ \5 r2 z1 s6 A
<head>
6 n+ A5 g) X; Q- I7 \" P6 y- T
<meta charset="UTF-8">
/ M$ @& H$ q. Y, s
<title>Web sockets test</title>" r6 Q" L8 r1 O/ a$ V$ G
<script src="jquery-min.js" type="text/javascript"></script>( R. ]# R5 G, o! V$ }* G0 J
<script type="text/javascript">
3 Q* B' [+ H# E9 h) z
var ws;
9 o( ~! |5 Y' x" S
function ToggleConnectionClicked() { ! Z/ J- ]6 a) h, h* }9 R+ N1 w- z! ~: e
try {. i: B+ x1 r1 g* Y+ i+ B
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器
$ ?2 {+ }/ r* o% M4 _+ W
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};
3 ^6 O$ I& c- D- X, A; V* B
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);};
% U& a3 x3 q( N9 w: z8 A, q
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);};
" Y2 w- k. G/ b# x9 j
ws.onerror = function(event){alert("WebSocket异常！");};5 y- Z) G F& |/ m% S: b- a2 y
} catch (ex) {/ h: u8 G' z6 h- S' C) X
alert(ex.message);
. v9 K( x, j; y5 s/ N, ^- n5 i
}9 l! |" h$ t2 U$ {* w1 W% D/ }
};8 ?' V* v7 Q+ i
- K) d- l2 c- I; a, ]/ T/ n7 {
function SendData() {
3 j' H6 L2 r" `
try{" Z c, \1 l$ x" D! N# V* s
var content = document.getElementById("content").value;8 v y5 |- d! b Y4 V
if(content){
2 ~' S9 C* b: D7 c
ws.send(content);
. `1 S; E9 [2 |* d
}3 ^6 X' ?4 G. K, L3 B8 R' Z
5 k- j5 g6 Y$ W* I n
}catch(ex){* ~1 n3 `9 Z D6 J3 z/ T
alert(ex.message);! d0 C3 w$ C" C% i0 ?( e
}. k5 \) p, ]/ n" e3 W K
};
5 v, n5 W; G4 y& R
0 R3 h' R0 v& H1 d3 `
function seestate(){
. f# G4 J+ I6 e/ b, _/ l- u# M
alert(ws.readyState);
8 K. D+ j; r1 w
}( Q# D0 e8 `' J \2 T
" N* \8 C( U- ^. k# p
</script>' Z, B6 L+ s4 x" \
</head>
; k! f- h5 A* S4 }
<body>3 x1 L- Z) P8 O6 s+ m3 T
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br />
0 q2 V$ i( P# ]6 x0 {/ D. f
<textarea id="content" ></textarea>
3 h. ?+ ]* u& V2 J+ ~" E* \
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br />
; u; B7 c% v7 P" P
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br />3 a" k2 L# |2 \7 q
+ [9 U& C: a$ t: j2 o
</body>( A) s: i: E e# p
</html>) Z; v2 E- M7 Z/ l6 e0 ~- A

复制代码

2）服务器端实现

class WS {
, a9 `) d# n: B3 W% G" X
var $master; // 连接 server 的 client
8 ?% j8 W/ ~# r, r+ X$ L( x. }( n
var $sockets = array(); // 不同状态的 socket 管理
: e; ?* `5 C5 `
var $handshake = false; // 判断是否握手& U# c) ~: R8 @/ T1 H
- U! s% Y, |0 N O% E" z+ u5 s
function __construct($address, $port){- w \0 L! e5 e* s, H3 E
// 建立一个 socket 套接字1 f- n0 Z( P# l( `) o
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP) 2 l( r9 C# V h$ z1 I! {$ W) l
or die("socket_create() failed");
5 C) K2 {; H& }0 l% S2 U+ O
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1)
& e* K9 m! y! x' s, G" z% B2 P
or die("socket_option() failed");4 V1 b; x6 e- w; W: _
socket_bind($this->master, $address, $port) 0 t3 w& B b' x0 Q, `
or die("socket_bind() failed");! U7 u* |! f I- b2 e6 Q
socket_listen($this->master, 2) , d1 X$ J5 }, I# u
or die("socket_listen() failed");
* o; F: B& k: [% q8 z
" c7 ~ s U+ W. h$ h% s' ~6 `
$this->sockets[] = $this->master;
9 A7 {/ {! W6 Y0 \7 h K% e0 G
8 ` U' ~$ {) R
// debug
* i; B( b$ N# S5 W! Z/ n( w
echo("Master socket : ".$this->master."\n");
: s+ P4 K! v. ^# M. V3 K& A
, T# e9 f/ K+ T. ~) ]
while(true) {6 D# ~; o" W$ G: u. S! q+ `
//自动选择来消息的 socket 如果是握手自动选择主机
4 V$ ~8 M9 j, \- r3 F, y
$write = NULL;) I, R @: f7 X; i* t
$except = NULL;9 x6 r, x4 o2 ^( f) g
socket_select($this->sockets, $write, $except, NULL);5 e7 |; T' N: t+ h: [
' L" L, j, d0 ?: y: G8 [; `3 d
foreach ($this->sockets as $socket) {
( w2 p! e9 P' Z! X9 w4 Z& I7 @' ~# x$ n
//连接主机的 client % J2 \+ v3 t# x% m
if ($socket == $this->master){
7 j; w5 c3 z. ]6 `* ?
$client = socket_accept($this->master);- J% }* ^; ~, @
if ($client < 0) {
L8 W2 I' l5 h6 U% f) R4 d$ L( q% `
// debug8 M/ A" _2 G" G5 M1 T
echo "socket_accept() failed";
: @$ N) Z& N- T0 Z7 Y( x) r H
continue; u3 q: }' }+ W
} else { u# z/ V, N, u# Q' v/ Q4 {
//connect($client);5 m' X- ?4 G& Q' y
array_push($this->sockets, $client);
7 J4 ~& t' W1 w& M1 G' O. ]
echo "connect client\n";
8 S% u3 V$ j1 ]$ W3 m
}
0 d: V7 y! g" z- k/ a! q
} else {4 e. }6 V9 M5 ~. G" \; o0 L
$bytes = @socket_recv($socket,$buffer,2048,0);
G: D- V8 G# e, |
print_r($buffer);4 [' o c2 W& y$ _! N t/ |
if($bytes == 0) return;3 f, _! G# t- d, E9 V
if (!$this->handshake) {
( R, P/ V. ?- d. z
// 如果没有握手，先握手回应 v9 K& j* _7 n
$this->doHandShake($socket, $buffer);9 @# s& U5 K g. `. Y4 ^
echo "shakeHands\n";
" q' N! ?9 C' }2 m6 v
} else {+ L- e" l6 j! [/ e6 A8 X
3 e B# F# W' K% H, F# \9 F, y
// 如果已经握手，直接接受数据，并处理
* f$ l+ A' y/ [7 F6 b2 t$ z3 q
$buffer = $this->decode($buffer);
! f3 d; G. l9 F* b
//process($socket, $buffer); ( Y7 o1 I5 s5 B+ L1 |( X1 |! ~; ^
echo "send file\n";5 t: ~0 k6 e& ~! O7 l
}1 `: C4 z0 T" {+ Z3 {" f/ M
}
$ n1 ]5 P q @2 y$ H! w
}( S' y6 r3 p# ]+ C4 C- B6 o
}
: W0 u; z. a: y/ w3 Z) L' J
}
! E$ u+ w9 A# r3 A1 y5 B& A' w/ c
% c- l/ B8 z# Y+ |6 Q T
function dohandshake($socket, $req). \2 B7 L2 ^: }: r7 R: s- }
{1 w" R9 Z+ b5 Q+ q& t4 O% I5 m$ p
// 获取加密key
# D+ {5 M) O+ i
$acceptKey = $this->encry($req);1 X% C+ j' k2 Z: u
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" .
"Upgrade: websocket\r\n" .
5 R$ t* ^5 J9 s# s8 G3 L8 M g
"Connection: Upgrade\r\n" .5 O3 l& s- t" e7 S% w
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" .
! g# H% l2 H* o9 ?3 O$ l
"\r\n";
% f; x* D$ H& {0 r
) V3 h& X7 V4 M
echo "dohandshake ".$upgrade.chr(0);
2 u9 v* Q/ o7 I/ I
// 写入socket
! @4 s( O& p; V* J* \
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0)));# ?7 E3 R# w1 {) y& R
// 标记握手已经成功，下次接受数据采用数据帧格式
3 t" D: ?* ?& d5 S8 f! \. M$ v) L
$this->handshake = true;! g" o% d+ c9 M" v& S1 B0 D+ ~ l
}
8 i3 v) a# |$ f2 @- w4 _5 W
" X/ M8 @# o/ Q) O% f4 h f
: u: ]' Y2 R' f0 {# {) W
function encry($req)2 G+ u' x( A9 b% f- g+ u3 }
{
3 {1 P$ J+ Z% Y$ {, `, d6 K; M! ]
$key = $this->getKey($req); _# n% }: T2 z
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";+ L% j8 L% h8 l' l4 ^1 m" N
$ v/ L4 C% F4 a" R1 v/ k1 H
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));
" y2 H- A$ i1 K# M
}, S# K. k7 ^. W( k* a
7 E7 z: N6 m3 j
function getKey($req)
- @: K9 o" ]5 H4 \5 T# f \. l% @
{
7 G5 m% T- G8 l& L; b
$key = null;
/ @, {, n5 }4 T( L& t% M
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) { 6 j$ k) M- K) B$ ~* N6 F/ q
$key = $match[1]; % W1 h# i2 c' X3 E1 n
}) K0 J: S x, b
return $key;" b& A2 q/ M i7 {1 r
}
/ x. Q | |: e1 ~ I% M8 w
/ z- V1 o ^1 l7 N
// 解析数据帧
- G( G4 T( L6 ]% l3 {
function decode($buffer) 8 D( j* W( t% c2 G
{3 a2 F5 ?8 \( {0 t5 G
$len = $masks = $data = $decoded = null;# u/ m* e! k \3 x% e
$len = ord($buffer[1]) & 127;
$ \& f$ O$ @5 t+ [ @: R0 \
" |+ ]$ z: J; }4 F" n9 o
if ($len === 126) {/ M% N$ q1 A( S/ t
$masks = substr($buffer, 4, 4);' N0 s$ d5 V; S4 Z7 ?$ ^+ D
$data = substr($buffer, 8);
1 N0 l0 f1 K- N4 W1 s
} else if ($len === 127) {
; c/ `( q3 [- \; ~' Z9 z" j
$masks = substr($buffer, 10, 4);
/ g0 A9 _( X: n- U& {) @/ I
$data = substr($buffer, 14);* w# _& L; b0 O% V; s1 j0 C" s
} else {2 b7 A: J/ f- l$ J4 c# H$ h4 w
$masks = substr($buffer, 2, 4);1 `6 L; ^) N: X# U* o% W, ^
$data = substr($buffer, 6);
# d3 g. |. F1 ~$ T
}
2 t+ W4 w$ ~3 R6 X* R* F
for ($index = 0; $index < strlen($data); $index++) {
, \3 X, E4 l: z& Q" U9 l
$decoded .= $data[$index] ^ $masks[$index % 4];
6 x9 U( B4 D w6 U% u/ Z
}
: L0 e( _; f( n! o
return $decoded;
2 H. f: R( Z1 f+ [/ m; r( k
}& Y( M! i) ~4 n0 u5 I1 m% M
( M# O3 x5 F2 ]1 i
// 返回帧信息处理- \( L6 i* f. J$ B+ C3 O. b% G
function frame($s) 4 U1 x6 u3 C1 h9 c8 n" t3 H
{1 i2 f# I, y; T
$a = str_split($s, 125);
@% w( F$ G, n- Y* b9 C
if (count($a) == 1) {* @& d: ~3 L! |" C; s- u& K
return "\x81" . chr(strlen($a[0])) . $a[0];
& L) M$ m& b0 ]' C) r: |" W/ a
}
% d: G: E* r4 D2 ^
$ns = "";
; L* `- v3 Q$ `/ ~* L$ p% I% I
foreach ($a as $o) {
5 `0 k) l$ c( g. k
$ns .= "\x81" . chr(strlen($o)) . $o;
! o. m% C V# V4 O H+ Z
}2 C! T5 a' s! c2 A/ J
return $ns;$ i2 \8 r% ~# L4 d0 m* C' z- F0 @
} k( S: O1 y# l2 F" q) g- J
* q# O0 ]/ c( r
// 返回数据5 i. a$ r3 o9 n
function send($client, $msg)
: q& S& `: H% m3 ?
{
& W! Z% @" \+ j V. ^
$msg = $this->frame($msg);
; [9 H* C/ I, {# \8 _, D* I# f
socket_write($client, $msg, strlen($msg));
4 z. r: D$ T* m! l0 c
}
) A/ K7 S1 U A1 V* a
}
" V5 T! F" |' z+ L! O7 I
. M' l. c$ D, L/ L
测试 $ws = new WS("127.0.0.1",2000);
! r& ^# p9 a m, o, ~8 P
K, Z% o) R/ D+ H

复制代码

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

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

浏览过的版块