PHP 简单实现webSocket

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

1）客户端实现

<html>
" \3 K. a) \$ ]6 Q! X
<head>9 [ c7 V4 j% z- T! J- ~
<meta charset="UTF-8">' b% V* R' ?# H3 e9 p1 |
<title>Web sockets test</title>
: z, Z4 u F5 g
<script src="jquery-min.js" type="text/javascript"></script>1 e8 N+ F* W+ F( o3 K' c! g, j
<script type="text/javascript">9 [! {3 `# ?' s, x
var ws;4 j9 u7 A% i }* ^" |6 x: ?
function ToggleConnectionClicked() { 5 i% H; \7 S; ]
try {
5 V/ V# V# v$ r+ j' h
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器
/ d- l3 C# S" e5 o9 G; q2 @* I
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};
- Z# C7 J8 _( D4 M+ b- W: a
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);};
5 k$ P. B; H3 P& ?
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);};
% w5 n6 a# h+ `% C; x3 k" S- h
ws.onerror = function(event){alert("WebSocket异常！");};4 a! `. H3 j0 \4 i$ R# b3 B
} catch (ex) {* s. S- O% s, l( }
alert(ex.message); & s0 D2 K/ A7 i
}0 h+ i' E, f) z3 V* O
};$ Y( d# Y3 L% `6 i
`* P7 D2 E4 i
function SendData() {5 V3 a, F9 C( Q5 G
try{# W# o! A) V5 M8 p% L' ~/ `
var content = document.getElementById("content").value;: h4 _$ [% [+ @& t' ^& v; c
if(content){
" L2 v+ f) D' S
ws.send(content);. [, c0 \6 C6 a; P
}3 H8 l+ h; B0 r) T' k
/ ]# M- Z0 q: K' M0 L4 p6 u8 N* c
}catch(ex){" f/ e/ P- M, g* F( g! ~1 y5 `
alert(ex.message);
+ C3 ?7 e* n# j/ s1 z
}) ?" a! h9 I& P7 Y" J* o
};; O& D0 K" T5 R5 l) `
0 q) l @" i4 b8 P" U$ N1 Z
function seestate(){
' Z6 H" c/ O' K) u
alert(ws.readyState);
* x6 N4 a% T/ q' _$ [+ z5 {. _
}9 J, J; Z: b2 U, r' b! M
: w5 S8 z6 n) [. B! f5 q7 I
</script>
, i6 R; t9 l6 ? u8 @3 J
</head>
1 V9 L. B& N# r1 z4 z' P, h
<body>: D7 A% Y6 l, \+ ]
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br />" V. \; P7 r, J6 w# F6 P! i: s I
<textarea id="content" ></textarea>
J% {0 A' e7 Q8 L4 v
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br />
) \" W* d* V- E
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br />/ m: A7 n4 D! y% c
6 U. R! r: a# V' f4 t4 P
</body>
* V9 I5 V3 E0 y' j( Z$ z: f' X( F3 |
</html>9 l+ N4 l0 r& _& t

复制代码

2）服务器端实现

class WS {9 p+ i' \/ U- u( ?! c) N
var $master; // 连接 server 的 client
/ @9 D m2 i# p) ^
var $sockets = array(); // 不同状态的 socket 管理
M6 S% L) x" G0 `' a
var $handshake = false; // 判断是否握手/ i ]( s. V. N; R% L9 z, S1 `
! V, s; X1 t% |; @! T9 B
function __construct($address, $port){- q* u, E9 p7 W- _1 j, ]
// 建立一个 socket 套接字9 g! t8 W& A% K# r
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP) e' A; G1 l5 K* u9 F
or die("socket_create() failed");% u8 B" a: E; M$ Y
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1)
& v4 E1 E" X% c4 L
or die("socket_option() failed");& Y5 e2 ^' J+ L4 E: {( l
socket_bind($this->master, $address, $port) , G, j. N( y% `* N: J7 y) Y7 @
or die("socket_bind() failed");
+ w( n" ~0 x b: l/ G
socket_listen($this->master, 2) {/ v" I' {( M2 g2 S- ]' v
or die("socket_listen() failed");# f% F6 U* z( ^; y4 [$ H
% O* C8 k" a8 \
$this->sockets[] = $this->master;
5 s% I+ j7 R& m: R
# a( S1 l3 T* @" E, z& P
// debug2 x$ W, {5 {2 ^! u+ G+ B* \
echo("Master socket : ".$this->master."\n");/ W& Q# q4 n7 v& o7 Z5 B T
* t _- ]. f& b' F6 Y6 ]
while(true) {
$ ~) o- T$ I0 x2 H+ G
//自动选择来消息的 socket 如果是握手自动选择主机, ~6 ?7 [+ I$ ^
$write = NULL;
7 L: T& |3 D* q9 j
$except = NULL; B" U: H3 V# H
socket_select($this->sockets, $write, $except, NULL);7 o/ g! k' }. q4 n- }
7 [) \1 A# Z* p3 q. O3 K
foreach ($this->sockets as $socket) {
% z1 O0 N: d& k5 B
//连接主机的 client
2 L6 q) B9 G% g& x
if ($socket == $this->master){
@+ Q/ k f+ M% H9 @
$client = socket_accept($this->master);7 m+ ~ u+ l1 r. X4 h( o1 A
if ($client < 0) {
! @ x7 e5 f- ~ G, V/ L: H* g: c
// debug1 R4 v9 ^% j- y7 s5 q: Y/ {
echo "socket_accept() failed";
9 N$ ^2 K: Q3 j! t9 [# n% f9 U$ c
continue;
2 g$ z; [9 W1 f U6 P
} else {! u$ D6 }4 W1 k) W' I
//connect($client);1 [8 E* ] e9 ^4 ]8 m2 ]
array_push($this->sockets, $client);8 F9 K* I/ T" v
echo "connect client\n";
; ^+ ^' G8 C, j L( E8 `8 E
}. G$ c3 C. `/ P
} else {
$ z) P* e0 N. T: D# X/ v
$bytes = @socket_recv($socket,$buffer,2048,0);# d! d" ~% i$ H. S
print_r($buffer);
: ?9 F! g3 w6 C' z" ^
if($bytes == 0) return;
, n) |4 C0 p+ ]7 G, u' L+ k
if (!$this->handshake) {" c+ K% f2 m7 x& `
// 如果没有握手，先握手回应8 G% x' R) d1 M. A7 J; \% i+ i
$this->doHandShake($socket, $buffer);
# I- ?. F2 q# V6 h4 F4 F
echo "shakeHands\n"; w. G! c# ]% y. ?# a' M
} else {
+ q* n: \0 f+ F* ]
. v/ a# U0 F- G$ K2 y7 {# d
// 如果已经握手，直接接受数据，并处理+ s2 X; I1 P: P# ?* u
$buffer = $this->decode($buffer);
) t2 S+ q- T, s) J2 k
//process($socket, $buffer); ) k! d4 {5 L, Y- S a
echo "send file\n";/ J7 e6 p" l7 r2 m
}
; \- J/ w# A. W$ Q7 A2 {; p e
}
4 H! ?! O! }$ @6 a
}
, R' R2 r8 \0 T6 D% x1 W
}# N6 Q" l! n$ E
}- c' ]8 F- k' f
; n8 Q# ?' S+ e# U7 {" b
function dohandshake($socket, $req)2 s2 S4 h" b( o1 t
{
X7 E4 t* h9 F& `1 f# a7 l( r
// 获取加密key3 Y2 Z3 a; h! S! |# G3 x
$acceptKey = $this->encry($req);
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" .
" n5 [8 e7 l( x( C5 }0 o
"Upgrade: websocket\r\n" .
* O8 Z: C a6 D1 Z
"Connection: Upgrade\r\n" .
' a. {" K# a# B, {2 @
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" .( O) @# [# u$ J5 O6 [
"\r\n";
* P$ W/ O+ [6 T1 @
4 }$ e& S1 {+ I1 C6 c# F
echo "dohandshake ".$upgrade.chr(0); 1 V O; p5 O1 P* s6 k" x3 G, u* x6 [7 {
// 写入socket1 i! d0 q0 D, Y! e/ P* d
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0)));; V, P* L9 r# q' }5 }0 E
// 标记握手已经成功，下次接受数据采用数据帧格式" l' A j3 J7 }+ C9 v( ~
$this->handshake = true;
6 N0 n6 j2 Y* n- x7 L3 A B6 h' x
}
8 P8 J O' Z$ A1 q* h
# G4 T+ l" Y" K* N# K6 J, d) v- d
7 p+ K: s( w/ U E9 G8 ?0 Y# a
function encry($req)+ Z$ p8 h9 r% [3 E& c
{
% Z5 `: e* Q2 c, \
$key = $this->getKey($req);
4 Z+ z8 {9 n. b5 J4 \ I" e
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";
% m& }2 T+ d) ?) U; N
* D6 s" y( G3 N$ d' Q; z; n- l8 R
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));9 t f' \0 N( j9 U: Z* }$ S2 ]- j
}0 r* V7 p' }* j* J
( g8 N5 ?0 h0 U( ?
function getKey($req)
$ h" K8 X' G6 P, r6 h; K) A. A
{6 F/ P0 [8 v! |7 Z9 [( ^6 Q% W! ~
$key = null;
2 [# B- ?7 ?! c+ O, `
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) {
3 Z5 g1 c/ f4 k& C( |' C
$key = $match[1];
+ ^- N: {1 h8 L' [8 r/ Y" T
}1 T/ u" s3 V/ g9 X; Q. f& R1 k
return $key;
1 S8 y9 s1 E* l) J) Z5 E; ~
}- ?6 ?( i# g) |6 u$ {8 R9 `9 j* a
" `9 g' U3 e$ L
// 解析数据帧
) A1 ~; U8 _9 B; P' |
function decode($buffer)
9 Q' w, r9 R3 U% C' s
{3 [6 Y! \2 C. `7 g3 ~- u
$len = $masks = $data = $decoded = null;( L( c- R w/ K; N) j2 K
$len = ord($buffer[1]) & 127;6 r1 k! ~1 I# S" o1 {* u' l' b" g
* g B0 J9 O, H% v! ?$ ?
if ($len === 126) {7 C; s6 v" R2 I. H( {9 z; z4 z) @1 g
$masks = substr($buffer, 4, 4);7 Y7 `- D2 }$ J Z
$data = substr($buffer, 8);
_1 a5 m1 D& e1 Y$ n
} else if ($len === 127) {, _+ Z- I: _! M0 L& X- v& U
$masks = substr($buffer, 10, 4);7 [3 e9 D% Y2 _- F6 ~2 P
$data = substr($buffer, 14);% s1 z9 Q9 f' N
} else {
4 T8 d- p- H' Y$ n3 e2 Q6 S
$masks = substr($buffer, 2, 4);8 m! R7 g% m4 ^1 U8 ?
$data = substr($buffer, 6);
; V7 ~$ B9 q. a& Z# v
} S- J! _1 d' B0 x# R
for ($index = 0; $index < strlen($data); $index++) {
- \: E: y# [# x4 z( {
$decoded .= $data[$index] ^ $masks[$index % 4];
2 k" w0 Z" n: z* \* C. ? a
}
! e8 V' i1 x( \& \2 C% G3 f& I3 A. E
return $decoded;
( G; k- `' f% O: D' l# m! w
}" y& C& D' }3 Y& I
: o b, `' O9 j) Q8 c6 K
// 返回帧信息处理# [- U' z( i0 }4 O& T$ O
function frame($s)
/ ]" R! U7 b* ? @
{
) U. _7 V7 b8 J) Q+ a
$a = str_split($s, 125);2 J5 `* ^. T7 N0 z- ]' K& {" G
if (count($a) == 1) {
2 [7 _( O# S) t" N" e$ D# t+ s
return "\x81" . chr(strlen($a[0])) . $a[0];
! }: _0 T0 G$ O( W, P
}
8 c/ x# L$ E1 F% m, ? \" C
$ns = "";7 |% F+ L2 B% l7 ]; c0 l
foreach ($a as $o) {
6 W `0 ^) a% Y
$ns .= "\x81" . chr(strlen($o)) . $o;
. e7 ^4 @& o, k; p0 [
}1 s4 a; Y$ q; d
return $ns;9 `' H1 U/ F+ Q$ p7 s3 C
}
# s! [9 q X/ j1 F$ N
" j# U8 X I6 E6 B( Q* q* f* b0 t
// 返回数据8 X) r, a6 V) G
function send($client, $msg)& q# B f2 l! U' \
{! M1 {7 Z6 e) z* b$ B, K) ^( X
$msg = $this->frame($msg);' s% f: M& m9 m
socket_write($client, $msg, strlen($msg));" }# N( {. p- H% a& U
}2 r4 V2 c: \* n0 Q
}+ R- J3 i! m. L: C! E
2 ], v- z$ _8 x: }# M( y/ G9 Z$ ^8 @
测试 $ws = new WS("127.0.0.1",2000);' Y, m9 z9 o/ \
/ `& O! U7 \! q# q7 E

复制代码

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

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