PHP 简单实现webSocket

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

1）客户端实现

<html>
/ f# D7 e# Z9 o- | |0 \/ Z& V* b; L
<head>
5 m- ~' K9 A+ m" }
<meta charset="UTF-8">
- w& O; ~1 p8 K
<title>Web sockets test</title>% F4 Z" U5 N3 J
<script src="jquery-min.js" type="text/javascript"></script>% i) F$ m( e5 A% H+ Z" Y; ~; o
<script type="text/javascript">- H# q8 Y6 K+ s! b+ {
var ws;
% e) T* Q7 e& ?. R3 I4 i% a3 U
function ToggleConnectionClicked() { , y: c4 n7 G( J
try {
" F4 ?7 e' ~# |5 m6 U% a
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器 6 m5 [5 Y, _* J: ?$ Y
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};& u- h& X( ] }/ u# F
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);};
# Q# U" T+ h' }& h, I3 N
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);};" W3 l/ [2 g. y8 Z- h l) Q
ws.onerror = function(event){alert("WebSocket异常！");};# [: m [$ L3 v. ]2 U. z
} catch (ex) {
0 l- C% i) d; j6 h
alert(ex.message);
R# o( g- L, e6 |. X/ h8 @
}
3 L* Z: p k8 @5 F9 c* k+ |8 ~
};8 n0 Q$ {3 p0 h6 S
; R; I7 F& S1 }- ]4 @
function SendData() {
7 p) q9 v8 B) {& S2 }' O4 S
try{
: g3 e+ z, Q& }) H2 Z% L
var content = document.getElementById("content").value;) c6 o, b7 l$ X- U; T
if(content){
0 s1 i# ?+ R* `) R) B8 q; f. w2 b7 Y
ws.send(content);
) ^" a3 c2 P! ^: o3 [0 p+ h: H
}5 H# ?' U" T& b0 [# D. U8 O
4 E, o% R% L2 C q- x
}catch(ex){% L* t% Z9 o$ f9 o( e8 p
alert(ex.message);
& |, V5 z, |6 j- v) ^
} ^* M; I1 k3 v7 c# P
};3 K1 W+ Q0 t9 n3 A K
9 V5 @! t* ~4 }, M+ z7 N
function seestate(){
! q" ?, {* k( ^/ z0 ^
alert(ws.readyState);
" f9 I0 v( g: `
}
' v' s( y+ _4 v# E
+ o/ j1 q6 P* V$ W
</script>/ q' V% R6 L' ]% ^5 b3 p( T
</head># q& n/ v: n& w- N, S% @3 _
<body>
9 W+ s+ t2 L9 j( U3 s2 O
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br />
1 y6 l$ p( I F$ ?' J0 V; n! J5 C7 e
<textarea id="content" ></textarea>7 A6 f( \6 p2 M0 u6 e* D6 [
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br />
6 k. _/ O) R( W4 {
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br />( t) s! \6 `! x" @' a; p" [
' C; q- y$ @; t3 `- v/ f
</body>
& P' G6 l! C& |! N
</html>
4 o5 e. n* r0 a9 i1 S" r2 I

复制代码

2）服务器端实现

class WS {1 O( v8 z J# T; F) O6 c
var $master; // 连接 server 的 client6 X; e9 k' k3 y
var $sockets = array(); // 不同状态的 socket 管理5 B" T( e2 m1 b/ O
var $handshake = false; // 判断是否握手/ ?& c$ p5 P5 n" [! q
6 H, F: ]: s1 d$ ~$ s0 y' e
function __construct($address, $port){ Y6 |! S8 {" p- c4 m( c6 c
// 建立一个 socket 套接字
2 c: X; j& i7 H4 C7 }0 k
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP) 9 M6 l. A/ y( S/ _8 g$ o3 }& P3 ~
or die("socket_create() failed");
; Q; r' t* z+ g; E) ~" c
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1) 4 x/ [) c* ~- _; q
or die("socket_option() failed");& k+ u' ^ t3 E0 _( H
socket_bind($this->master, $address, $port)
4 B, j) v$ K8 v6 q5 g% @8 V9 s
or die("socket_bind() failed");
/ c0 i5 k* y0 E& Y9 p/ @" a
socket_listen($this->master, 2) , v6 U+ I; _" i; Y" f! d y
or die("socket_listen() failed");* b/ p# [2 ^" g' M
5 Y" i* V: Y8 A% v0 p
$this->sockets[] = $this->master;5 T. f5 y5 b6 i" r% U/ l
- Z6 j: z0 m0 s
// debug. I7 ?( h& D0 ]
echo("Master socket : ".$this->master."\n");8 l8 e2 D; u; s& f. l
# H6 A1 I4 n6 r" h9 X
while(true) {3 ^6 Q' c" o- e) V# t
//自动选择来消息的 socket 如果是握手自动选择主机
! q% i) `6 p X3 L/ Q! ]6 A* h
$write = NULL; z0 e" F' J0 t/ ?9 q1 c7 n
$except = NULL;
! D! I2 F; |1 Y+ ~1 H
socket_select($this->sockets, $write, $except, NULL);) j0 h& m( a& c
, q7 b @1 \5 I, i' o, K) k
foreach ($this->sockets as $socket) {& h b+ o7 n' }2 x$ Z0 p+ D
//连接主机的 client ' \& F5 _ R6 n3 w
if ($socket == $this->master){: O' C; L( @) b6 |% r$ l8 Q/ S2 N
$client = socket_accept($this->master);
/ V9 i. m, Y, S! j
if ($client < 0) {
- U2 {& g/ O; Z5 P+ t
// debug
. D8 ]% f |, V+ S" |7 f8 P
echo "socket_accept() failed";5 f- ^. f2 k% H
continue;
6 O. T8 p- _ l/ Y" F, h
} else {
7 g8 R8 S. p0 v& m, U
//connect($client);
8 [$ L, w0 x3 Z5 T9 i \5 a
array_push($this->sockets, $client);
* r; |$ u% {" _# h7 l: l3 I
echo "connect client\n";$ [- V2 ]9 N6 w- n
}
* ^7 U% i& u" ~% l4 X/ I
} else { E/ Q; m' { v* r# \) x$ O
$bytes = @socket_recv($socket,$buffer,2048,0);% o3 R8 ]+ M5 G- g( y, Q& Y
print_r($buffer);
7 j, i; M0 v% R8 g
if($bytes == 0) return;' F1 q* e3 N0 G; O# K' w3 O. R' n
if (!$this->handshake) {
3 y" M9 a) H4 v) r
// 如果没有握手，先握手回应
5 W. n8 a q! ]+ y
$this->doHandShake($socket, $buffer);
; @/ t4 s5 [) l# S
echo "shakeHands\n";( L' b2 W# }+ u' ?$ I
} else {. u& a$ p, z1 b5 E3 C& z
+ r) a* W7 A- i; h' z3 s* H
// 如果已经握手，直接接受数据，并处理
3 x+ h7 l) Q3 _
$buffer = $this->decode($buffer);
3 {: x3 f- _/ t8 y! E. i2 f
//process($socket, $buffer);
( _- D% D/ M3 w. B- |; u
echo "send file\n";6 B4 j% N- h3 Y1 a0 K$ C
}
6 ~# U5 @, p" m' g& F
}& l0 ~9 E7 T G% f
}
7 ^2 X, `* J; P4 `& `6 B" t
}
( D9 K7 g' g& J1 y
}
; r2 K; b( r8 r8 W+ s' G1 H! b
/ _9 y$ Y; d* \, k" ~/ n
function dohandshake($socket, $req)0 ^7 a# R4 _! }
{
! ]9 @7 m( ^5 a
// 获取加密key
8 D& _. L3 ^/ ]* D
$acceptKey = $this->encry($req);5 k9 N* [+ P: X
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" .
" J1 v0 F! D" T! P
"Upgrade: websocket\r\n" .' W% h8 U4 m V* I2 ^; p8 s! Z9 ^3 ]
"Connection: Upgrade\r\n" .% f! N U% A7 A8 k! O" {* C; u7 ^
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" .1 F6 E1 \7 V! \, _% x$ A
"\r\n";7 U) b" M7 p8 o8 r( F! a; c
" X: M8 U" O# b" P7 X
echo "dohandshake ".$upgrade.chr(0);
) A* E! Y! K" I3 Z
// 写入socket
: b' f- l3 Y1 [7 {' ~
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0)));1 E4 D% a( N* D& D4 j
// 标记握手已经成功，下次接受数据采用数据帧格式
7 W8 ]) d, _/ Q4 {
$this->handshake = true;
6 Q0 k$ j5 D9 ?
}! _9 @ n" G4 ]$ A( N
2 B% P1 I# E6 @" B; o. d
% Q1 d" Z1 P' G3 q8 Z8 O* Q
function encry($req)
2 F! F, K1 ?* b4 N6 w5 I+ R( m0 ~* {
{
) K' @- b0 |, f+ T
$key = $this->getKey($req);. e) q4 c& \2 N" K+ R& j4 ]+ ]5 [
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";, }& }& r! B! J8 n2 p- G6 z9 k
% v& I/ c) M& W% g
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));
/ y3 B$ q( M5 c2 U
}9 _$ E4 w$ L5 D5 h7 i
/ I9 p3 m/ H2 X, V- C1 k3 S! }
function getKey($req)
- A7 o; M' F+ S0 o' E- D
{8 \4 b/ J; G5 k) H6 V
$key = null;" q1 K' o7 \8 @% E
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) { : Y' I% _# A4 {' u4 q0 }3 ]7 t3 N$ ^
$key = $match[1];
0 s+ ?' _! E( t& Y- @7 k
}, j8 v* `% S- n6 c
return $key;
- P4 t5 e9 I$ I% H* n7 t( c! {
}
+ G) W0 @1 X9 _* V: S# M
/ [. I7 M9 T0 k
// 解析数据帧
: |$ l% e1 F/ B& w/ Q5 {
function decode($buffer)
, I# j8 p. ?6 l6 j( P: @- E+ c: t+ ?
{" ]/ T4 L( o5 t8 J
$len = $masks = $data = $decoded = null;
" F/ t3 C( o' l; }/ c. y- [
$len = ord($buffer[1]) & 127;: I$ m6 I5 _+ [ m
0 L8 s6 r' ?. p1 e
if ($len === 126) {
, f M, U. i* g: |1 C: ~
$masks = substr($buffer, 4, 4);% h9 E$ n/ i- h4 {- ?# a
$data = substr($buffer, 8);' j/ @ P$ v4 V9 Y$ E' U
} else if ($len === 127) {
0 V7 \3 ^, j1 X" P
$masks = substr($buffer, 10, 4);
* ?0 a5 e) P x+ K: v+ r) D
$data = substr($buffer, 14);
; s& a' k. {: J3 m" c9 y$ c5 M
} else {* D4 D# y3 m' z5 V. D( b
$masks = substr($buffer, 2, 4);' o" P) h3 g( N- d$ Y
$data = substr($buffer, 6);8 K2 `# Z# s9 i9 s
}* m9 x1 k E. s
for ($index = 0; $index < strlen($data); $index++) {
l2 A; W/ P* e- J3 I- k
$decoded .= $data[$index] ^ $masks[$index % 4];/ C3 g- T y) m9 t( _5 Z+ ^& }
}; v! u$ Z0 d/ l4 `
return $decoded;
9 @7 D4 |9 E) `# i
}
$ G4 X0 i. F9 _' W( C* y
/ i% e, }( }$ r0 `% i& O, j% X
// 返回帧信息处理
; \' s0 j! f- g s( E- h
function frame($s)
! d1 l7 L0 ]8 U8 U
{4 |) `, M) O8 c1 e* @& B
$a = str_split($s, 125);: g+ ~9 e5 _: a0 @
if (count($a) == 1) {: X3 H: _0 u2 K" f1 P# r
return "\x81" . chr(strlen($a[0])) . $a[0];
[( @8 I1 h {( `6 d$ r
}
2 ^, {# c# D: A! O
$ns = "";* V! \& o, ?+ S& [
foreach ($a as $o) {/ S9 r- ]' ~. B9 H, R
$ns .= "\x81" . chr(strlen($o)) . $o;! ^6 `( D9 O0 V. V
}: Y4 e$ [3 v/ Q$ f2 h' t1 J* ?$ P
return $ns;. W! {+ ~4 x P, \+ [' R7 O$ U# ]" D5 y
}
( Z* o1 {) W q% z8 B
$ z z( n5 T2 n; R
// 返回数据
( r8 Q' Y" J; h' e5 K5 v9 p; w
function send($client, $msg)
3 H( C5 S3 z6 n% l
{
. T% x6 _7 b! `0 t( y
$msg = $this->frame($msg);
3 v6 v3 f# _- d3 ~0 b1 i
socket_write($client, $msg, strlen($msg));) s9 L' H( ^2 [) F
}
) P' [- I, N: Q3 v+ e" k* ~ L
}$ W" {$ `6 R8 ?' k$ x# S
* ^" V, D4 q. e4 Y( F
测试 $ws = new WS("127.0.0.1",2000);! K1 l2 I& l/ b! y
: v: e) c5 B8 n6 f7 U1 J3 L! g- a

复制代码

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

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