PHP 简单实现webSocket

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

1）客户端实现

<html>8 ^) h0 j, Q- e7 R
<head>
, l% S0 ?' p* m; k4 T
<meta charset="UTF-8">
7 z$ C6 J4 X. p' ]5 i- H0 Q6 T
<title>Web sockets test</title>; t$ m( v/ @1 n$ W! e
<script src="jquery-min.js" type="text/javascript"></script>
c1 N( L0 m0 R7 b/ i! H1 j
<script type="text/javascript">4 b4 g2 i7 Z* f4 ^" Y. U/ F, [+ u
var ws;
- A/ R1 a# k7 l% K
function ToggleConnectionClicked() { / N" D0 b( \1 \+ R& g V. S& g5 L
try {2 q4 ~1 d- ]3 ]2 n4 [
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器 , ]# u6 p( @- G# Z2 P. D. u! J6 g
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};% ]# r3 t+ u6 c4 y
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);};
3 g' L/ s4 n* r/ I
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);};
) j% Y7 e" P' c2 D
ws.onerror = function(event){alert("WebSocket异常！");};9 F- U6 e3 \' w, g1 Y: O$ @& J
} catch (ex) {
# Q; n4 E* \& c' R, c
alert(ex.message);
2 U0 |1 @: o1 I: R& @: c
}: P( H$ c+ T. S5 }) E# g
};
% A+ z2 O/ I4 C# c
5 M# y! V9 [% D" R: ^
function SendData() {
& @2 H! O5 ^! W2 A6 g6 \
try{( I* z. o5 z4 n2 x; O
var content = document.getElementById("content").value;
' A; O7 {2 o. T2 O% o$ V! ]" t: \3 N1 X
if(content){
3 Z9 j. y; ^! k8 ~
ws.send(content);
/ t. i: N) S7 {0 \( G+ P% S
}
0 e8 [& u: t7 i' S( y- @ J
) W R2 W! |/ ~+ |3 X/ B
}catch(ex){! |/ V9 s& x9 t* t# H
alert(ex.message);$ z/ x. S4 Z3 [" Q$ [
}
; v" Q9 p7 s1 G" u- Q
};, V Z& V7 @% a' X. r
8 Q* `* K9 E# F- x
function seestate(){7 s+ A$ O% g$ \8 O
alert(ws.readyState);/ n) ~" N. r) j5 H# r; z
}" U8 c; Z' ^$ J
' O' y: M1 g0 u" {+ K
</script>
8 i' B A& H, |& n! f. n
</head>
4 f& X. l/ m8 E( p& S
<body>5 w7 z: C) S: e" R% {) U
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br />
i- F/ T; @1 Z: `/ R( F& A
<textarea id="content" ></textarea>" i: B7 B0 V1 b
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br />% f' f# e4 h3 _9 Y/ h
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br />3 I8 \% u! M) p) [- H3 Y6 m7 w
1 j; z$ P _) d0 A+ w" G
</body>3 j m& f# m7 A
</html>& Y& l/ d2 Z$ l: [' _: \8 K

复制代码

2）服务器端实现

class WS {1 J3 F' B; q' S( o$ g' o
var $master; // 连接 server 的 client
" L% v( f2 }7 x% N
var $sockets = array(); // 不同状态的 socket 管理
" n1 V- N# @, I. A; N
var $handshake = false; // 判断是否握手
# I( [& O; z- j$ R( ~) R
; l( B O3 F3 k$ U
function __construct($address, $port){
( |1 w2 s2 ^2 {) [9 W- U% y" t2 o
// 建立一个 socket 套接字
. n+ V5 e* ^! O* \1 x% S# A+ ^
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP)
$ \4 g# i4 u, i5 C$ ]8 h% y/ E) @6 S
or die("socket_create() failed");3 ^3 k0 l: S3 s( y" L
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1)
0 ?* _* I' x* ^* I- {4 W3 p
or die("socket_option() failed");
) \4 c/ {+ f2 ]" o9 w/ q
socket_bind($this->master, $address, $port) 1 b8 q6 r% `/ u4 d+ H
or die("socket_bind() failed");
" i; u5 N/ A( ]* X
socket_listen($this->master, 2) Q3 U+ k6 k5 D! r1 [) p' \
or die("socket_listen() failed");
: B- {+ C* i {
+ l2 j+ D0 f. c6 y
$this->sockets[] = $this->master;" o \- F0 x% n+ A, y5 i& o, s* G2 {
3 V- f3 D _, G; M* h8 o4 e) H/ N
// debug! W T- j6 y; B, r
echo("Master socket : ".$this->master."\n");8 W+ G7 i/ _( }% i! n7 w
' P: C$ @. ^2 |4 [4 ~
while(true) {
' n3 K+ V9 X/ }- R5 s* C/ @1 E
//自动选择来消息的 socket 如果是握手自动选择主机5 ^. ]- F0 b, ?% X
$write = NULL;9 o+ F, o1 O0 w# N! q1 G' m$ b$ D
$except = NULL;2 {! k- h1 \8 [0 U' M
socket_select($this->sockets, $write, $except, NULL);/ O/ b) W* ]# e" ~6 i' i
: M. ]! W# S6 h- a q
foreach ($this->sockets as $socket) {
U# B1 J. S( p4 T
//连接主机的 client
* {9 x C9 f( {8 p2 K
if ($socket == $this->master){4 _+ g7 Q, E, F* M; o6 x) \# K
$client = socket_accept($this->master);7 o: q2 ? X2 |3 U7 d
if ($client < 0) {5 X5 Q1 g# }( m. H3 U
// debug
9 I& o0 M) Q0 `2 R! N( q
echo "socket_accept() failed";* G* l% d1 ^7 z$ z( i: v
continue;9 w* h3 T; {" V7 n( q: m
} else {
. z- R4 J" V7 W, s- ?
//connect($client);" \5 ~: `& U4 w8 ?
array_push($this->sockets, $client);, ~% x+ M% C1 O/ A- u4 B- e
echo "connect client\n";2 [! k8 q' U& H/ l( t, ^$ k3 j
}
2 _: @& `: u) m$ m) J$ m, Z
} else {
; k4 p e, }3 v3 M$ V; k `' J
$bytes = @socket_recv($socket,$buffer,2048,0);0 t8 V8 k* L- _) Y7 J
print_r($buffer);
( M, n: N' _4 c& @( y4 n9 g# B. U
if($bytes == 0) return;2 J! r3 T( ^' ~, p7 _- [9 Q
if (!$this->handshake) {* u/ w2 U- P K8 R
// 如果没有握手，先握手回应
% d5 x; U9 a; E( u7 Y) s
$this->doHandShake($socket, $buffer);8 H* z7 S. L5 ~+ {4 t0 l: a! N
echo "shakeHands\n";
8 x7 B, |2 S7 @5 s
} else {
" R5 _, Z- n" U7 E% X
( a0 O- ?, O+ B* y
// 如果已经握手，直接接受数据，并处理
% q2 g r: r2 T3 G: a
$buffer = $this->decode($buffer);0 m/ H+ ]; {4 f f! Z
//process($socket, $buffer);
4 _4 l; ~1 o$ S4 J; Z' {
echo "send file\n";
' H, {' K3 o9 y
}, s! r+ f2 f1 C2 a$ [. R& Y- x
}* K% a0 p+ K6 B% c+ S, m
}
" I; M, F1 M/ n; n
}
+ \6 q$ \5 E4 I4 q- a! {& s0 U
}, m$ }/ H2 \6 q
& v/ j6 K+ k5 B- `! W: ^
function dohandshake($socket, $req)0 x. S; s8 o W* q$ _. _- r
{
9 O0 {" c- `2 s# ?
// 获取加密key$ ?* v" m+ R# Z0 Z' k
$acceptKey = $this->encry($req);
- g! W& r, G: O8 ]
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" .
, P& _; H1 ^! |/ w N3 P6 L- a
"Upgrade: websocket\r\n" .
6 v l; N# R+ J/ f
"Connection: Upgrade\r\n" .4 c$ t" @9 f1 v) L* b; x7 h# P* }2 u
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" .% A9 S6 e! x( A% m3 |' W+ q: @; Z D3 \ |
"\r\n";
' N1 o, g; {- j$ c- q
" O" B* I& i. K
echo "dohandshake ".$upgrade.chr(0);
8 O* k: M) Q6 |; G% y$ Y0 X
// 写入socket+ F- D, G9 N: u; ~- I
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0)));* |$ a6 |$ E- \ g3 C/ d1 e/ c- {& i+ b
// 标记握手已经成功，下次接受数据采用数据帧格式% I3 }7 S1 [8 l7 Z9 D
$this->handshake = true;
; u2 R+ B2 T2 B3 m8 }% T
}
2 l/ ?# U3 U: x$ B7 ?1 O( b$ ?
& e# k- G c& g$ _
/ _ M9 d5 K; b( Y' o
function encry($req)
) u; h, b6 c- ?% ?
{
E: d6 N# b- ]2 u2 b# E0 X2 D. m( e
$key = $this->getKey($req);- `* b, G4 L* I/ z/ `0 i2 i3 e" N% l
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";
! S0 D. \' b) Q- I
# y8 D5 b5 ?8 Y8 m7 s
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));
( G7 w; Z' T) ?9 \1 s
}
( c1 p- w2 R: C( G# x0 {9 p
( z; |" M1 b1 C! x. }2 P" i( j
function getKey($req) ) R7 I5 p6 o/ T1 Q1 R
{$ i) x2 p; D' N' F
$key = null;
/ Q) W4 q0 g% B5 o; L6 N( ~/ S- h; [+ P
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) { 5 _8 p. U* D& i) o, U4 r B. A
$key = $match[1];
7 Z6 b5 h7 ~; p- j3 \6 @
}
$ F0 t' u' }6 m+ J* H% V
return $key;
# U( ]7 o" T) u: j8 T
}
; H: g8 q N1 w+ w& k
. T" z1 ~$ M) e: [
// 解析数据帧
4 }/ ?& _# o: Y
function decode($buffer)
0 a4 R) K ]9 ]3 J8 ?. o
{
+ R% z7 D2 o6 \) ]) p3 Q8 t$ R. ?
$len = $masks = $data = $decoded = null;: X8 h7 `) A' u2 o6 {
$len = ord($buffer[1]) & 127;8 p; q" W5 P) G- L* ?
! C" t6 R; w- D& ~" Q' K
if ($len === 126) {* y7 ^# a4 o4 F5 _+ g6 z
$masks = substr($buffer, 4, 4);* h, f ]. V! r* v& G3 Y
$data = substr($buffer, 8);! R. n8 K! n; j
} else if ($len === 127) {
* R1 o( f: ~7 A) _3 J* F2 v
$masks = substr($buffer, 10, 4);
& w8 p3 \' B7 _2 J+ q
$data = substr($buffer, 14);2 l1 t S4 C5 l2 C) {) A
} else {! A4 C, g) Q( Y. P
$masks = substr($buffer, 2, 4);
! g, t& Z. o- t- N4 ?
$data = substr($buffer, 6);: f0 a. g& l% ?1 f
}. G' K6 M( v7 c( E( t8 c
for ($index = 0; $index < strlen($data); $index++) {
) l# _' S6 D8 L3 O. T) V. K$ e6 A
$decoded .= $data[$index] ^ $masks[$index % 4];& D, b6 M2 S% \8 `* e
}
* I: t# J/ Z9 Q* h5 W: H3 z
return $decoded;
) Y; b" t- x9 L2 i: e
}7 K4 i. X: p1 L3 Z' D. ~
& f. M$ N8 \7 e
// 返回帧信息处理
8 q' e- n- O0 j0 w
function frame($s)
5 c5 I' S3 z4 p! o, [. ]' f1 C* H0 y
{* }- X5 i2 N3 _/ ^! W* N3 {4 c) S
$a = str_split($s, 125);
' Q. X9 Y+ _( p( h
if (count($a) == 1) {
$ I8 ]. a5 W+ x9 S
return "\x81" . chr(strlen($a[0])) . $a[0];
( {! [+ u0 f- t4 L5 j1 E8 B
}* S. r1 s5 d3 |. s; I# h! u! m5 O
$ns = "";
; c& F) h1 J6 w0 W3 B
foreach ($a as $o) {3 P% _; V5 `. w: J7 L4 w1 u" W( o
$ns .= "\x81" . chr(strlen($o)) . $o;% O+ g, }: S, R; W/ E
}
0 K/ B) [6 L) d5 @4 p
return $ns;; \) O, V6 T( o( d
}- ?' i6 J+ D i" D
* B& v" r0 }5 S+ J3 ~8 k* w0 S
// 返回数据
; L* ]0 y1 T; r
function send($client, $msg)1 C# c) o4 W% a+ D. F
{% B& N) t9 x0 Q* \' t( V
$msg = $this->frame($msg);, ]4 l2 p9 G, }. m4 o! u% }" x0 t
socket_write($client, $msg, strlen($msg));: z: o3 ^- c7 k0 Z" q, T
}
* I! v. p9 w8 E2 F
}
9 K( D& z' A& s" d
" L+ D( G$ r% Y$ ^% n# |
测试 $ws = new WS("127.0.0.1",2000);
" s4 ^7 j' W3 l
& d! U T4 C3 J% R" Y. k* s

复制代码

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

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