PHP 简单实现webSocket

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

1）客户端实现

<html>! m- o- \9 @! \+ L5 Q
<head>! N: `" J. K# [8 I0 V/ v
<meta charset="UTF-8">
+ Y& @/ L2 B2 j* i0 a# M( Z% Y
<title>Web sockets test</title>
4 \2 \6 [# b& I
<script src="jquery-min.js" type="text/javascript"></script>
* P3 x& O0 p7 Q; s
<script type="text/javascript">" i* J7 r3 `# }
var ws;
* G1 Y; j v6 D7 _& a' n
function ToggleConnectionClicked() { ' j+ w8 D2 }% R6 s& b& z$ G I
try {
1 V. C" Z4 d+ y2 y
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器
* |/ @ a3 ?/ {+ ~9 y
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};+ N# Q/ c. p, ?; x3 K
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);};
2 A; |: Q( G) x- w4 e
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);};8 j2 |9 w0 C0 r* M
ws.onerror = function(event){alert("WebSocket异常！");};5 ~" u" Q' z; g
} catch (ex) {
1 Y# i9 p7 R2 W# j/ u
alert(ex.message); ! B8 ?, ~: z$ `
}* S; }- g/ M# h n a1 ?
};( D7 e. _$ f, G
6 v& c7 i8 C* `' E; m
function SendData() {
try{) X6 T% K0 J" G9 v% R! a1 w M4 \
var content = document.getElementById("content").value;3 ?$ w4 h: j( R, c
if(content){/ S4 D7 D& I9 v: E
ws.send(content);
n2 c; A1 {- X! r
}! ?3 @; ]) _5 ]( G! E
5 m2 y. w3 h* L' \
}catch(ex){
. s' N1 D7 k! T5 X& C
alert(ex.message);8 G0 r* _8 x% m! b7 Y
}8 t- Z2 q2 Q, y4 W7 G6 @: |
};
9 H; ?1 v. u6 o6 o) p
: U3 ^5 k! |& ]9 @8 e
function seestate(){
! l" @3 I8 A, ~2 ~
alert(ws.readyState);
+ b6 d5 D7 F9 ?9 a
}: o U: M! \2 D1 J' p T* z8 G2 A
! W" @' Y6 Q4 A' \
</script>
, x# k; F4 S: a1 c
</head>' h1 u% o9 w* C5 @" @+ x+ e
<body>* V! g' y3 H2 ~1 v, W
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br />
8 t, ?# `7 R6 Q7 k5 t
<textarea id="content" ></textarea>" a: w( ^2 T4 D: @' {$ N
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br />4 c5 a" ?2 K5 ^. r% G1 J; a/ R( ]
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br />/ d c( ^$ Q% t4 G5 y# I
* T L$ Y+ p' d! N- ]
</body>( P; `$ T- u& Q+ S( j% A
</html>% Y- V6 v, _9 p% r& j1 B# v0 x: \5 l

复制代码

2）服务器端实现

class WS {# \% u1 ]% V( }- _& z z/ D
var $master; // 连接 server 的 client
. u; r1 a6 Z4 r. c8 Q! H
var $sockets = array(); // 不同状态的 socket 管理8 N S- b* Q! i. U) v
var $handshake = false; // 判断是否握手: _- v0 O5 Y: z1 Z, }/ {% O+ f# U
w/ I8 G0 G8 O2 R
function __construct($address, $port){
# i# _: I' O7 C3 ~5 A, l0 b
// 建立一个 socket 套接字
. y' q& o$ |9 }1 W
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP)
. u* n5 {1 H, q# h9 i9 o8 d1 W
or die("socket_create() failed");2 |4 y5 C5 l$ p# B0 w7 `) o
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1)
. f7 Z0 }+ D7 I8 ]0 A! l
or die("socket_option() failed");
" _$ x- p* ?0 M. A) q' G* K
socket_bind($this->master, $address, $port)
/ `1 F9 P [: c1 u
or die("socket_bind() failed");0 W! ]* [ j4 d: V+ ~1 _: ?7 U
socket_listen($this->master, 2)
3 h) K# A* z8 S# B8 ]5 j7 W
or die("socket_listen() failed");: f# t) d `5 N: d
5 g% ~; s" U7 X/ T
$this->sockets[] = $this->master;$ ^' \" h0 E0 h4 X; W
. \9 K4 U3 L% P4 M8 w& n
// debug( d; n% z$ _1 k$ G
echo("Master socket : ".$this->master."\n");' s7 A/ V" {% n( s4 G
i, p1 U5 R5 ?/ ^; z& |! i
while(true) {) D; x2 W8 l6 \
//自动选择来消息的 socket 如果是握手自动选择主机( c2 M: A& H! V) Y; y* K; c F. q
$write = NULL;+ r" l. k% P p. Q. m" n8 [, S6 N6 Q
$except = NULL;+ R! X& P, G; g9 D
socket_select($this->sockets, $write, $except, NULL);8 ^( A. \9 ]. ?! ~2 q8 b% A5 H
5 z% q2 J7 [: k9 v- ^
foreach ($this->sockets as $socket) {
. Y7 ^4 F* _, o
//连接主机的 client
0 i5 v! O8 m Z6 w2 h$ x. d" ?
if ($socket == $this->master){' f- ], t! l& `5 b# D) [! U
$client = socket_accept($this->master);
0 D9 T6 J0 o; J# P% ^: N6 v* I; B
if ($client < 0) {" ]5 Z, R( l- C
// debug( b0 d/ G/ B6 K8 y
echo "socket_accept() failed";' N& Q0 W. f) \* K: e# D
continue;
9 ^" T/ U- J/ t6 g# c8 i
} else {9 B: u& D h% w, @9 Z5 X
//connect($client);
) w' v5 ^5 E; f6 G% D% X
array_push($this->sockets, $client);
4 p! F) Y8 Q/ e+ e
echo "connect client\n";
% t5 l. B0 b& s! `
}
. G* U! ^9 x& o
} else {
+ s+ H2 `2 }9 U3 _4 t, p* t' p
$bytes = @socket_recv($socket,$buffer,2048,0);. ?9 t1 |3 U' c& C) B6 v
print_r($buffer);
2 M2 t9 U4 D/ ~5 \9 P* }0 s
if($bytes == 0) return;" M* x8 \% {% V' z9 A& c
if (!$this->handshake) {
' }+ L; S: k q: d4 b6 n
// 如果没有握手，先握手回应6 T# d6 m$ Q$ w/ \' c+ Z: _
$this->doHandShake($socket, $buffer);
- Y& @6 p- F c5 [) J" P9 i
echo "shakeHands\n";' @! Y" \% |3 b
} else {0 G7 F+ ~' k' k9 Y
! Q; ?+ f7 y+ ^' M, o2 f Q" n
// 如果已经握手，直接接受数据，并处理3 r4 g. x1 u6 m0 A8 p F1 L2 j
$buffer = $this->decode($buffer);
( Q' T* T x5 I' @8 V/ d
//process($socket, $buffer);
. |9 ?* `- y7 m) d/ v+ ~5 a
echo "send file\n";+ Z' C3 r, y: {1 r! v
}
; q5 `! J; F! l
}4 _3 p1 ]8 d) H8 @
}
C+ ? C8 b1 T3 Q2 q) t
}4 w& _* r* K% }/ a; F0 C7 `9 [
}1 q2 b8 S; N: n- U/ c* B1 M" o
. t: Z0 ^" Q _2 d8 e
function dohandshake($socket, $req)5 T$ Y, X- o( H! k" A' r$ ?
{+ q5 C7 b* G! E! e- _
// 获取加密key
/ O, v1 _' {7 F4 B0 I7 z! `- b: s% n
$acceptKey = $this->encry($req);! u% G$ z0 F2 U+ m! L) `
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" .: c {+ ?. ^/ `! f% K
"Upgrade: websocket\r\n" .9 f( s6 |' T2 b+ P6 L3 A5 D4 F& p. T
"Connection: Upgrade\r\n" .. Z$ J0 Q% f V. [+ |
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" .
9 y2 H* v0 a3 R
"\r\n";; j/ } O6 p0 i# q# T `
. p" m: p" W* h
echo "dohandshake ".$upgrade.chr(0);
% a/ Z1 R; b+ e. b1 b1 n8 M/ D' |
// 写入socket! a! I4 T( _5 f+ y3 R
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0)));
$ \0 z( Z7 r3 ?* v
// 标记握手已经成功，下次接受数据采用数据帧格式6 F$ B6 }/ T% Y) J
$this->handshake = true;
, }- g2 A$ h. B% B5 o* z
}2 j. Z- Z9 X ^2 }( \8 M9 N; m6 K1 C
& o$ p6 D9 z9 z9 n. W5 R
i' W9 Y" _8 |# i$ P0 P) Q' @2 j
function encry($req)
- B4 i* u' Y% N0 @) k1 Q9 [( E
{/ Z1 s! }% c" [5 j) q0 R' R
$key = $this->getKey($req);
: C, U7 ]) r$ Y9 h: g% i
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";
/ \5 `, o7 [& |0 m# a4 o8 o
% Q z2 |+ s3 U# A$ I0 D7 ^+ {
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));
1 X8 J% X' v% }( W) l y9 |
}
4 a. _ j, u( S0 B! U/ h) Q
2 G/ B. H2 s8 B& _
function getKey($req)
u- N- `) `- k& d L+ |
{
9 S3 @9 Z, Y- a4 F( M
$key = null;
0 C% x7 B% I% M- Q+ F
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) { 8 ]0 F% i- E/ H3 a/ ~
$key = $match[1];
$ A( h$ L) G5 k. B
}; |( x6 x% ?# @- c
return $key;* \( F& {% W. D% `7 m3 ?
}# A. {& Y [9 B2 o: d
5 y( @# s3 } u
// 解析数据帧1 p) L0 B; v7 s2 }( W
function decode($buffer)
" o6 R6 f1 T. d% w
{- V& U& t# v9 _! t; G: w& H
$len = $masks = $data = $decoded = null;/ \( n9 y1 e# g, S2 Z, t) Q
$len = ord($buffer[1]) & 127;
/ n! ], }2 O& W8 X
6 W6 Z3 |5 ]* J5 S0 J7 d$ M% Y
if ($len === 126) {
- u2 x5 S1 z2 |: h* ]
$masks = substr($buffer, 4, 4);
7 m6 v* u5 j! k6 A# _) F t
$data = substr($buffer, 8);' f5 C# q' e1 S* S% Q
} else if ($len === 127) {: W4 h3 d3 G* X8 F! \8 C
$masks = substr($buffer, 10, 4);
0 x4 S4 [ I* p; l
$data = substr($buffer, 14);8 g4 y0 N% c T* b0 E( T6 ?
} else {# o% o7 `0 \6 C! y6 _8 R
$masks = substr($buffer, 2, 4);
0 r; C4 x& n, c4 t1 z9 Y
$data = substr($buffer, 6);4 d8 | ]* f, w! C
}, n; M$ }, T( k* s# g% x
for ($index = 0; $index < strlen($data); $index++) {$ u4 u+ M0 ]7 V, o1 z' o
$decoded .= $data[$index] ^ $masks[$index % 4];
7 M! s, {$ \9 B& ~6 k5 z/ \; N7 \% r
}
/ t2 A' K8 x9 t* q( j2 y- M, e% I# c
return $decoded;! M$ L% Q& ?* O% F$ q- u: o7 I! @+ Z
}; r: s6 h( N W. j3 L
M2 G, @( h7 Z1 o% L0 R
// 返回帧信息处理5 u0 b0 B! A" n6 [
function frame($s)
- q( G" {) C# A4 A) Y2 |& E& Y" I# d
{: {6 r) c; `% S1 A, u. G
$a = str_split($s, 125);& ?3 T S' D, b. _6 ]$ u; Z& R- p
if (count($a) == 1) {0 q, m4 e. A9 @1 Y) _
return "\x81" . chr(strlen($a[0])) . $a[0];& x/ i' c3 _& Z5 {+ }9 i. ?/ \
}
: G9 P8 c6 _* O
$ns = "";
- B+ j/ U+ o7 \
foreach ($a as $o) {
9 o$ ~9 W9 e ~' t l
$ns .= "\x81" . chr(strlen($o)) . $o;( Q) c& z1 p2 s
}$ z W b! N; |0 i; w
return $ns;7 o1 l/ N% U/ R
}3 _9 R0 z' h! F+ q4 D- p% g, S1 _* x
/ I- x# B" Z' ^* C9 E
// 返回数据
) Q( u1 o: ~- }
function send($client, $msg)! ?/ t9 k# t8 M, i" G
{
1 B1 G8 A7 q: Y
$msg = $this->frame($msg);* A" d0 Y( m# v3 p* U9 y! `
socket_write($client, $msg, strlen($msg));
! X9 m* d% [# F/ a/ N( M
}) S) L& |* v F" O% m
}
! ^" K) e# D3 Y: g, j" z) l/ }
% s( V! d8 }" d. ~
测试 $ws = new WS("127.0.0.1",2000);
* a- m( e5 p( Q( o4 ?& j; B+ c
d! O5 I$ O1 |5 s

复制代码

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

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