PHP 简单实现webSocket

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

1）客户端实现

<html>' C. t( x- \- J4 ]
<head>
+ R+ ~, o) `8 N' g7 G5 C
<meta charset="UTF-8">
& @2 T: N; O5 @3 Y8 {: K7 K( e
<title>Web sockets test</title>
" M; b( [8 w0 m: ^0 y5 m7 N
<script src="jquery-min.js" type="text/javascript"></script>+ H3 H. s3 ~, n' J* L$ B
<script type="text/javascript">
% R8 {' O9 v* a; z" j4 ]
var ws;
' _) o# w( k0 t
function ToggleConnectionClicked() { " s5 K, Z, ? g1 x1 c8 }
try {) A% g; U$ n* b, d- h2 [
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器 L+ W5 W8 A( B4 u& s- T+ a
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};
( }. R9 a5 b6 ]
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);};0 [. f. F- G; z! O( t6 O# R$ Y
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);};
$ w4 M3 Q/ G U
ws.onerror = function(event){alert("WebSocket异常！");};* S2 k% E1 f5 M8 @* T
} catch (ex) {4 x; g3 U2 T& o) ]. L4 q( L
alert(ex.message); . F: N# I" o0 r7 M+ I
}
5 D. s8 Z }4 S0 o6 {5 V
};4 c" d) E2 \ v: ~( y
+ A1 L: w9 {! W0 S4 h6 W
function SendData() {# d% Z$ H' L$ j. F, b+ q
try{, L |! i0 i' n) n/ [
var content = document.getElementById("content").value;: @7 D) ^6 H1 r( p2 Q
if(content){0 ~; f% H% S7 X* K% C6 H" v9 O
ws.send(content);
. d4 r4 R: }& J5 |
}9 W. ^0 |8 H3 [; T9 a0 R
C5 D$ k% z9 N" p7 c
}catch(ex){- x- \7 g3 C6 q* n( s
alert(ex.message);
" i5 j: `% j. v4 x& g
}: L$ y5 T+ q( i3 W% k5 ?
};8 D' t6 e! k# C. A% K4 |1 {+ h6 K
) d. O- D7 l* ^% x: p
function seestate(){
1 h' x7 B. O2 i9 u
alert(ws.readyState);( q7 M: L& Q9 l3 w* r' R2 E8 H
}
; n! r# e6 `: E' M4 s" v8 E
% j& m4 \, K: {
</script>
; N0 G& H3 d( y3 d! N+ Z2 B- I
</head> t7 [; z; ?3 [% R+ q8 ^% }
<body>) s) v+ {" p. s" ]' e& d! C8 ~
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br />
@' |1 V/ u6 U% N
<textarea id="content" ></textarea>9 a) E3 V+ C/ N# R8 P" q; a @5 u
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br />
* M! z6 D5 s% W$ v' U3 a
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br />
q! e& g$ C! o0 ?
% X0 N* ^$ D0 L( P
</body>3 z9 l$ O, H( R5 Q9 q
</html>
: U' L; G" R+ N

复制代码

2）服务器端实现

class WS {
: t5 l/ O' R5 k* `% }
var $master; // 连接 server 的 client0 ]9 p5 j6 |: C4 c0 O9 H
var $sockets = array(); // 不同状态的 socket 管理) _# B2 `) g2 C
var $handshake = false; // 判断是否握手# X" M9 P9 y' T9 A
8 u9 `' j8 G# o0 d1 A& R
function __construct($address, $port){
3 {0 s0 S, [4 B* h, z* h! ~# J/ F. @
// 建立一个 socket 套接字8 g# W, `0 u: c0 Q V( N
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP)
N/ o5 F/ R* r5 c( t
or die("socket_create() failed");$ R; b/ u3 Q: Y' b3 |4 Y
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1) ( I+ `7 i. @4 }) {8 |* a
or die("socket_option() failed");
5 h5 ]. R) r: {6 x: p2 a. l1 l
socket_bind($this->master, $address, $port) 5 t, k3 e. v9 b4 ~
or die("socket_bind() failed");
" k4 `! A" M, N7 ?( H7 ]: }
socket_listen($this->master, 2) # M# L( C2 z- ^2 o$ \6 k- A. z
or die("socket_listen() failed");
4 B0 h7 I% Z9 X1 x, [
8 p, B" W" e9 r) Y+ T
$this->sockets[] = $this->master;5 u$ M0 ]( z) U0 [0 T. k/ N S
" O. Q7 @) ]* _
// debug& i: ~( f$ J6 o% S( Z
echo("Master socket : ".$this->master."\n");
3 Q( a. a- f) N" j5 w+ E9 A! w
6 E/ L+ r& F6 p/ @2 q8 K
while(true) {
5 p+ ]! A- P& a" H I, x- I$ n- a
//自动选择来消息的 socket 如果是握手自动选择主机
4 A9 k$ [8 C: J# ^0 j
$write = NULL;9 P" p. v% B# p, R" ?1 j; [. B. F
$except = NULL;
7 W( c, G% Z9 N( z$ d
socket_select($this->sockets, $write, $except, NULL);1 k! l1 _& _4 F- s
3 Z+ Y+ J9 [ c) Z4 c1 c7 e; m
foreach ($this->sockets as $socket) {, m8 S( I1 v5 D
//连接主机的 client 9 F' b( p( w0 P! |; _6 @; W
if ($socket == $this->master){
+ D! l2 E+ h+ D# Y7 m# k3 {! X
$client = socket_accept($this->master);
6 O7 G% ^9 B( y
if ($client < 0) {
5 o: z) y( N. C9 b5 z5 H: _# f; T
// debug
5 O0 J/ @. Y* d9 h& I# N
echo "socket_accept() failed";
2 y8 @1 }& S4 a
continue;! T& @, R/ E( d; N1 q
} else {1 G) q, T- U/ @) M$ m
//connect($client);
9 J8 |6 o& _) Y: h& v0 I
array_push($this->sockets, $client);
4 d" N A" b8 O/ V+ T
echo "connect client\n";
/ s0 {, d3 t- ~
}3 v% h4 c0 ~0 f l/ t
} else {% h+ `2 P- `! p# Q" W2 Y5 H
$bytes = @socket_recv($socket,$buffer,2048,0);( Y. H1 W0 ~9 P u1 y# x; x
print_r($buffer);
8 R6 @' P o: Q" q
if($bytes == 0) return;
' x7 j/ i8 \* P+ N7 z* O8 R
if (!$this->handshake) {4 F p } ^5 x% P/ J
// 如果没有握手，先握手回应0 Z/ j- M2 W2 G4 `! Q% @* m6 |4 r3 m+ ]
$this->doHandShake($socket, $buffer);* B! C9 d B# C4 }1 b
echo "shakeHands\n";
, u; `* S2 |* _' V, n3 c
} else {
; ]4 {5 g3 R& Y! e0 G9 I
3 b4 Q& G9 N; V, p+ U
// 如果已经握手，直接接受数据，并处理+ E( O+ j% r4 x) b
$buffer = $this->decode($buffer);
( O0 _) Q8 I$ c2 D/ v3 w) A) t& B
//process($socket, $buffer); 9 s3 v2 Z; x0 U4 X4 e9 V
echo "send file\n";
0 c) g: o* u; Y" {1 N& r9 q
}
0 |. u. R$ `" q% t" ^
}
F; m5 L( @5 C0 N
}2 t; d0 f( b4 R2 Z( }8 D( J+ L7 ~! |
}
' A2 r+ S; `3 ^) Z
}
! m& w& s( m9 ]6 ^& Q; ~9 H: G( a
8 h3 k4 W5 }4 L& W
function dohandshake($socket, $req)% O; ~( n7 b7 E6 `# A$ [0 B# r
{! @! b( a3 l2 ^. S4 I
// 获取加密key
+ @ _- O. p+ O# N. x4 W
$acceptKey = $this->encry($req);
: j+ U& j% }9 B! z% I" [
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" .
E7 j O. }/ ?3 K/ j4 Z
"Upgrade: websocket\r\n" .7 O0 ]3 F9 F. N6 _
"Connection: Upgrade\r\n" .# y: H; c2 M& d: M0 F7 S( R8 W
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" .( d( s3 v- F% n) y. a! d
"\r\n";
7 Y, D9 ]- i( n# I
2 F6 S5 A+ }6 W: s$ |( }* u
echo "dohandshake ".$upgrade.chr(0); " ^: r! i: i4 I) G; t8 J* @% q
// 写入socket( W: P1 g; [, I+ s2 y _/ V9 o- g& k
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0)));
9 \& j4 X: E8 v& O* } ^# B
// 标记握手已经成功，下次接受数据采用数据帧格式( l/ d5 m4 }# {2 z( Z5 W0 h
$this->handshake = true;
3 Z. c- P( v+ R# G7 b' G: G7 z
}
: b2 p2 d% ~4 R- h
9 ^# T, R1 v8 C: {. R( w
1 V0 k8 @5 d( r4 r0 R) ?) F
function encry($req)
U1 N1 s/ L9 l8 D4 i1 y1 C
{
1 m, e; q- q5 J& r& e6 E9 n% S* {' D
$key = $this->getKey($req);
$ }/ O2 ]/ p1 i8 V! N
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";
) X2 t% t) i" N
! l, C2 E' R& \7 J$ V% i) n
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));3 {3 w4 s/ J; g! ]; h/ Y
}
* f% w3 T3 E" n% Z0 o) |+ ^7 N8 |
2 n6 \% ~1 j/ T# M; c# l" g; Y( x6 l
function getKey($req) * s. z( I# W% C7 T
{
3 o$ C- N1 m# ?9 r$ d' E2 ] v* Y) j
$key = null;- ]8 O; Z$ k% h% Y2 ]
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) { 8 Q3 T/ B6 ~4 O$ O% W" e5 N* Q7 c& Z
$key = $match[1]; $ B& p2 h+ v/ R1 C
}( G7 O5 k/ y1 p* i
return $key;1 J0 `& M) E, x: G
}
% n/ y; x3 K5 k9 l
# n& l# U# G7 S7 v2 k& Z; N! O9 ^
// 解析数据帧7 p1 y C2 B4 Y0 |( l" o V
function decode($buffer)
0 F, v6 I% i+ T7 ]) p1 x
{
7 F0 G2 E5 X* v: k1 f$ L% f
$len = $masks = $data = $decoded = null;
8 G: b( z( T( c9 |4 _, r9 o
$len = ord($buffer[1]) & 127;- _0 s% L7 f1 e, k5 A7 f
( C8 [% ^) `* s# y" }% {4 Q+ Y
if ($len === 126) {
: j( |* i7 L5 B( n
$masks = substr($buffer, 4, 4);
+ I. ]$ A8 Y+ Z4 A: b" ]( Q0 I5 u
$data = substr($buffer, 8);" i5 v! ^$ H9 w6 d @, {
} else if ($len === 127) {4 V0 @5 ]4 Z4 A5 a" N! n7 F
$masks = substr($buffer, 10, 4);, _) W5 Y t; ?3 D
$data = substr($buffer, 14);
) B. C% k) k) u
} else {0 g8 t: J" [& s9 ~( t" H
$masks = substr($buffer, 2, 4);/ C8 U% v# F: }8 _1 q3 l
$data = substr($buffer, 6);
]% ]7 c9 ?. b2 j4 o, F" H
}% i. G" V7 `' i, x* G5 V
for ($index = 0; $index < strlen($data); $index++) {
3 i$ }; i; w& C) f: _5 H6 ]$ J
$decoded .= $data[$index] ^ $masks[$index % 4];
( l L/ V s: A$ I) f
}
s# X5 o% O4 r) E
return $decoded;
) j$ s1 Q7 q) u4 c# l4 p2 G
}5 e, ~( X$ w$ M8 I
' Y; p) M# X9 I0 ~* N: M
// 返回帧信息处理+ f, M! v( @7 f2 m
function frame($s) u) ^* e. V/ W+ U9 }9 C+ D
{
+ J" x @4 p6 z. f; `% t* q
$a = str_split($s, 125);
0 v% o' i4 p, P5 z* `
if (count($a) == 1) {: l' B3 g3 h& r+ a1 s7 `
return "\x81" . chr(strlen($a[0])) . $a[0];
/ G( u* U) p9 X
}+ j* C$ C; Q1 @* }) k
$ns = "";0 h. ?! j$ l4 ^+ q5 W5 g
foreach ($a as $o) {
8 l7 Q+ {% x+ @" ]
$ns .= "\x81" . chr(strlen($o)) . $o;
$ `/ ]$ K; r; L8 |+ Y' K9 ?9 {8 Q" O
}
: w0 G6 K) f* k
return $ns;
. ] H3 y4 G0 W$ r2 ^# Q1 R" D7 a8 Y
}( {) p0 j' Z% U' A+ u
+ I5 r# p5 ?/ _6 A( F, w1 a
// 返回数据
% d. w/ L5 R& f7 B3 S& B J: K
function send($client, $msg)- w2 i0 s( `. h% w% _, ?
{8 v e$ o" G m T- @* |
$msg = $this->frame($msg);) p$ Y* }( f( D ]8 N3 x
socket_write($client, $msg, strlen($msg));
* A9 T$ q$ M) Y: i4 e) {
}
7 m! {$ _0 M8 ~" }* B
}6 t/ V( `" x( S7 Z, P; Y. Q
+ {+ g& x3 Y0 Q
测试 $ws = new WS("127.0.0.1",2000);
9 h; L2 r; b5 J
0 S. p0 {0 \3 `- ]. V

复制代码

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

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