PHP 简单实现webSocket

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

1）客户端实现

<html>
3 l+ x, I# \- b) A! s
<head>
8 F. N) T' M+ W
<meta charset="UTF-8">
! H) E- ?( a2 D* a
<title>Web sockets test</title>) A7 e, \( Z( V3 C- b0 X! u+ t8 e
<script src="jquery-min.js" type="text/javascript"></script>
7 [: w* y$ S' a- A, x6 Z
<script type="text/javascript">
( H' D4 K* w. L0 {9 u5 [3 m8 v# t
var ws;
9 b% W& J: m/ g8 `; d/ i3 a9 R& K. w: B
function ToggleConnectionClicked() { 1 ^* U: v) u) T T% X( Z
try {
/ w2 i8 m( z4 h6 @3 E' b
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器
. L+ `1 K2 z. q$ H- q0 l4 S
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};7 m% H4 T- v6 c2 n4 g+ S) Q
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);};# {* X& r! F1 w% p. b( r1 B
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);};
. }* Q i: W( F' `% p
ws.onerror = function(event){alert("WebSocket异常！");};
/ N6 [7 \) F0 o+ o9 y
} catch (ex) {
1 r+ C; g+ R) k) G
alert(ex.message);
/ y: P( f0 g3 M% ~8 b
}: j- |; c4 D' k5 n+ _+ d
};; t4 i* v- }7 T9 ~1 [( R
0 h' g* @7 ^8 K( k s
function SendData() {$ Y: C- D, X/ {: ?% ~7 J1 J
try{( a7 J) f$ ]( Y) E+ D
var content = document.getElementById("content").value;) N' i2 Q o% s! {0 u: q$ r
if(content){ U _+ x9 G1 ^# t9 B
ws.send(content);
; i( j( z& |3 L0 Q
}- |# {4 W) m5 Q/ _& h
& a0 p1 _% K" m2 i4 c# ]
}catch(ex){
2 u2 q5 N) J A V8 Q) s
alert(ex.message);
$ @( h% U2 G3 X: W: V7 l# R
}
0 }" b/ b! y. B% N! o
};
' W5 E* Y( f% _7 L
8 T3 B' W9 J: I6 I
function seestate(){
& d! R2 t5 x2 Z! u/ R# E7 \/ H
alert(ws.readyState);- b0 P/ O' {8 K: Q" {
}/ F3 Y' h' a* N, c, v5 M( L
; i. _# G# \, r/ L3 B
</script>
$ f5 W. e4 F5 n' X' F$ t# o
</head>! F1 u. y. Q5 a w1 }
<body>- W4 {3 L+ m4 O& G
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br />0 u, r( F/ G+ x1 |) ~
<textarea id="content" ></textarea>
/ }3 ?% S" h( T a: \; P8 [& q
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br />
) P" [* d+ C/ c& T% C0 R
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br />- {+ ~; A( m1 k4 z/ J/ i- Z2 x G
8 [5 i F3 R3 w$ Y
</body>* G! q+ s E. Z( } v& B7 I
</html>5 q2 y4 c/ c2 e! X# f# X/ b

复制代码

2）服务器端实现

class WS {! K1 t- z) P3 a9 Q5 J, `% U
var $master; // 连接 server 的 client
; E+ I! S: w* x! a! w9 b
var $sockets = array(); // 不同状态的 socket 管理
: a% i: @4 A7 |8 s/ F7 B# X \
var $handshake = false; // 判断是否握手1 F" u- L, k% w1 G; ?5 y' [
; n8 j% m2 T2 W8 A/ a
function __construct($address, $port){
7 r% x7 m$ z6 U3 J
// 建立一个 socket 套接字
: o3 {9 w# k+ q+ n1 {* M
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP) 3 Y- C- A8 b, D! p( Y3 }
or die("socket_create() failed");
3 E9 o `/ J% h r% S6 J
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1) ! t- K9 }; O- p7 r' X
or die("socket_option() failed");
1 T1 B; x( R( y3 m
socket_bind($this->master, $address, $port)
- r+ b" f& Z, M. J
or die("socket_bind() failed");; q* R( h$ V) V- R+ b( g8 ]
socket_listen($this->master, 2) 0 Q6 n" r, ]* u
or die("socket_listen() failed");: Q% [' s: l6 y) `# N
. ?$ M) h X f* \
$this->sockets[] = $this->master;6 X9 [2 U: d$ v/ ? w" C0 N+ Q; e
2 a9 h' C3 a+ i! G% B* T$ P+ X0 s
// debug
$ C3 \8 N, z* ] s
echo("Master socket : ".$this->master."\n");$ [, y) |8 b7 E3 D% u
6 c6 t5 i: o) h1 g3 I! M. X
while(true) {
# x8 O3 i+ k' N
//自动选择来消息的 socket 如果是握手自动选择主机, q/ h2 \& r. S
$write = NULL;; S( e& \( U: v8 _2 b3 } F/ k
$except = NULL;& \! S0 q5 z# [. I: S) p' a3 u
socket_select($this->sockets, $write, $except, NULL);
- S: D7 F) O4 _0 Z8 a! u
9 k7 f. X8 J3 r/ p
foreach ($this->sockets as $socket) {: m, a |9 R: K4 w0 O
//连接主机的 client . e7 o) Y& j& @
if ($socket == $this->master){& c5 |( ?% Q7 }8 {- m- i0 E
$client = socket_accept($this->master);. p7 E& y" Q: |
if ($client < 0) {
3 C8 s. h+ p' o) t. k# z% x f
// debug: k' A+ l' ~5 U4 s
echo "socket_accept() failed";7 O% Z( ~7 m; {/ Q+ S
continue;
: ?) t) I% e& S+ c+ s' ~( j9 L
} else {
P( U0 C& h9 |" T# ~8 @
//connect($client);
5 N3 B$ U- y9 d# z) f& [
array_push($this->sockets, $client);* @( D: [4 z/ G4 j( [
echo "connect client\n";
8 ]' u3 ]: D" i/ P& G
}
& k+ s% B% y2 h6 E8 \
} else {: q& q5 s2 Q0 ^1 L
$bytes = @socket_recv($socket,$buffer,2048,0);
5 ^9 P* y/ `: C4 }; K' g! l7 q Q( u
print_r($buffer);
1 s' v _/ J6 e
if($bytes == 0) return;* o; C& l1 N. {0 X# ?
if (!$this->handshake) {
& `" O8 ?( R) Y) C+ Z+ z
// 如果没有握手，先握手回应
8 X1 u& W5 c2 T! I
$this->doHandShake($socket, $buffer);6 b" `% Q1 h6 t- \
echo "shakeHands\n";& X. J3 I" ^& E5 H. Q7 i7 W$ s) q
} else {
, r; F2 e, c- A2 q2 U: m" ^5 n- `" B
1 P- \) O9 c# T" m- w7 w: m
// 如果已经握手，直接接受数据，并处理3 R @& @8 n7 m$ [( c6 \/ o: q) v
$buffer = $this->decode($buffer);
9 i# o2 ]- {0 D7 `5 h4 @$ z
//process($socket, $buffer);
6 W0 v. c8 l' b4 r- \: h
echo "send file\n";8 u0 f, W' |3 U; |" H! M
}5 Q% G! o- J2 X" H0 S
}
) ?0 Q l; C: e3 K b+ z
}
9 H/ y5 V# f, Z
}3 Q ~9 J# ^1 R# f
}& I! u6 u; h) T$ W% h: K! e
, p' r6 D6 F# [9 e7 t4 W
function dohandshake($socket, $req). Z+ [, U, U4 ?! _7 h# q% w# _
{
& e' L: C t. W6 R
// 获取加密key* }. G4 o' a& d/ ]
$acceptKey = $this->encry($req);
5 x/ i, m# k8 j ^* N) z" Z, A
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" .. S% H# u4 ^; t. T7 C
"Upgrade: websocket\r\n" .3 p, ~& K* F, V: G. C+ z
"Connection: Upgrade\r\n" .
3 M+ t; X7 c! ?. Q
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" .1 p1 @- g3 j4 { y; Z, m% ]( i
"\r\n";
- j& [" v8 m" }7 l* g# r
' c- |8 j0 }+ u6 i: C4 P
echo "dohandshake ".$upgrade.chr(0); 3 H; h* u8 R1 B$ T( G* i
// 写入socket6 R2 e( P! y+ Y; @6 t1 J, J3 @
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0)));& a9 y9 u* g: {# I$ B: O8 Y9 Y& ~, P9 t
// 标记握手已经成功，下次接受数据采用数据帧格式
& M! l( j& S8 m% }6 T+ m; {+ J
$this->handshake = true;8 }- s2 f( |' u5 i6 ?8 a# ^
}) P" X/ G. H/ R2 v# `& I$ ]' |7 C
6 ^) f5 A( {- S3 O/ j4 j" u
" X" J3 f4 P+ P/ g# v0 I+ E4 e
function encry($req)2 u" P( y/ N7 C# q' c
{
7 F9 @+ }* P( J g8 k2 u- a
$key = $this->getKey($req);& e5 b5 v7 v- Z% F* r
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";) {) ^ G" o9 y. s. O# j8 B$ z9 l
7 X7 x* h& C5 ~6 v6 f! Y
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));* G* T" Z% x) N! W1 b" G# r
} z5 R9 e# v& |! j
# g4 A$ D5 ]9 s% {/ d$ W
function getKey($req) & ~) w# e! J. b$ N
{
1 u, J. u; K4 }" C* P8 |) e8 `" f4 g
$key = null;( N% G% I6 I& _! _5 B
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) { 3 b% W' t% b+ w: l k
$key = $match[1];
& X, Q+ b, c; X4 d* z5 R
}, G6 J/ i$ X- h- \5 p# a5 }
return $key;
+ u0 B4 l/ H) t$ L, O
}
& {+ i- M* T0 V1 r3 W0 E
4 D$ v( F2 w% x4 z4 V: b
// 解析数据帧 x H5 q+ y) E: {1 Y1 L! c7 h7 D
function decode($buffer)
" Q2 V5 V' R1 U! k( d4 J
{! m9 g& P. T8 R. x* N: J- g9 b
$len = $masks = $data = $decoded = null;' h8 j+ k) b) v# ? }
$len = ord($buffer[1]) & 127;2 L" \8 W& h8 L, ~, O5 Z' o- x
) ?2 _7 _1 }4 p1 A- D& l. h: r
if ($len === 126) {
7 r, u) A; T" _7 g6 u* R+ T% `
$masks = substr($buffer, 4, 4);
6 w/ B/ I; k; i- h. m: f
$data = substr($buffer, 8);( S% X% b: g5 n# W- u- n0 P2 l
} else if ($len === 127) {
: k2 q: W" I; K5 p' B
$masks = substr($buffer, 10, 4);& R! G m& E4 O# p& B( s1 l
$data = substr($buffer, 14);+ ~: ~. e# r2 o& Z* S. w; v
} else {1 I: W. K; r2 F* I+ O, z' a/ I2 Q8 g
$masks = substr($buffer, 2, 4);3 c" T0 V' \% B* H6 Y- s/ K* Y
$data = substr($buffer, 6);9 z" y' O4 ]" ?& I8 P: `3 G8 c
}
' M, B, ]! v% _8 E. `' m' t
for ($index = 0; $index < strlen($data); $index++) {
! c2 C3 p' S; O2 \! |" E0 O
$decoded .= $data[$index] ^ $masks[$index % 4];/ G/ b5 s' [$ A4 `. ]% a
}+ Q7 @6 A' u: [/ G, U: p; F) d
return $decoded;3 T8 S0 W- a- m! D3 ^+ Q; [: J1 [0 o
}
8 U/ a E' a& T( q8 j* K- _+ D
- [' P" H( D+ O ^+ U
// 返回帧信息处理, ^& P3 L0 R7 G( x8 a* m
function frame($s)
% N/ u1 n" b) _3 U9 q
{( O( J4 k J8 v& w
$a = str_split($s, 125);" i4 t Q" c; q1 b* L
if (count($a) == 1) {' Q& { g0 U3 V* D* V0 G5 ~8 P
return "\x81" . chr(strlen($a[0])) . $a[0];8 n' K) O7 K( Z+ f6 H+ |! J4 r7 D
}
. a+ t3 I6 B: [4 {
$ns = "";: p# o# H/ {0 N& Y8 R8 {! c3 n& @
foreach ($a as $o) {3 H \/ D4 J5 `. u- X) P
$ns .= "\x81" . chr(strlen($o)) . $o;
& W2 U- v! {# c! S
}0 P0 u% K/ K7 a6 c' b# A( I) m
return $ns;
9 V2 C* z! N& _6 H
}
7 L$ v- }" h1 [7 F% ^
' {4 a9 A: \3 k
// 返回数据' O* k8 a# Y J+ C
function send($client, $msg)3 Z9 K- B/ a+ ?" F
{+ o5 v9 g! @" x! X6 P
$msg = $this->frame($msg);
5 Y3 `) T' q2 z1 A5 ?$ f1 T( t+ X
socket_write($client, $msg, strlen($msg));
( T s8 K& E- ]7 S& n$ t9 F
}
+ t) p \5 L' X A A
}
' `4 y9 z2 Z2 I8 D7 f. K0 |
/ z2 C, M3 t- \
测试 $ws = new WS("127.0.0.1",2000);
+ B: q9 f% y, C7 n! Y2 a7 B: q
T/ O |3 h- }+ u V- G

复制代码

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

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