PHP 简单实现webSocket

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

1）客户端实现

<html>
5 z' j- A( x U" y/ \5 d
<head>
3 H/ Q6 Q) l- `- M& E0 x
<meta charset="UTF-8">) l# A7 s* [$ X3 e J/ `5 y) ` @, ?
<title>Web sockets test</title>5 Q: V) e5 _/ Y7 I; G0 d" p8 n1 z8 [( R3 ]
<script src="jquery-min.js" type="text/javascript"></script>
9 W4 e j" v- Z+ S' c# q( K
<script type="text/javascript">0 [6 ]8 u5 } W7 H
var ws;
( Q* C( @; d& o1 y3 C7 N
function ToggleConnectionClicked() { # Q- |, X6 P. c6 C: m. ]. \9 U) ?5 M
try {
2 A# O; W8 }$ ?% P
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器 8 v4 K X' j/ b/ E {2 t, X
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};
9 n' g5 A, G; [. L) h
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);};8 r$ y4 S& R3 i. N" J) `
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);};
* J/ A! f# j6 T: q
ws.onerror = function(event){alert("WebSocket异常！");};
. |& p3 x; ~7 A2 m
} catch (ex) {
) @6 C. |' j/ k9 c- ~1 U, _
alert(ex.message); ' ^" m: r5 k1 V7 T
}4 g- k% ` C e6 ^2 o8 b, n+ Z
};$ z N( R7 C% @
8 E( M7 P. t, H, n
function SendData() {
% F+ a, }1 @5 M f7 i
try{. l% j+ O3 E7 F
var content = document.getElementById("content").value;6 V: ]! _4 ?2 e2 A% i- x
if(content){
3 f! ]* h# F" K8 q+ }
ws.send(content);2 j% v- u2 _1 K* h( z M
}4 S m" f) g7 `
, d" L9 o1 ]: A. ], i ^
}catch(ex){. D' I" w# [6 S5 q1 W0 D H8 m
alert(ex.message);3 D! |( `) r/ k3 h
}
5 `+ D) _+ \4 A
};
( G* ^3 ~5 q2 p9 R
% u& U" f3 t- N
function seestate(){
: U, ?5 V; [# m* K
alert(ws.readyState);
& R8 J8 X' w4 }8 ?' i
}4 Q' M' ]" z4 z- _" M' A0 \; l8 Q
7 } C) B. o* a; b
</script>
' Z: D& f0 L5 @
</head>, X* T! u/ T! {6 p$ K8 T z2 f
<body>
: Z0 z. t0 S; k
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br />0 ^! O4 |' r* O* p, q! u: d
<textarea id="content" ></textarea>
, b+ z A! C% r- f
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br />2 o M7 I+ z4 c
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br />8 p/ d, p0 N& R; Z; R
$ Y! h5 Y- J! D
</body>
0 L4 F8 O- t" ?& P
</html>, @2 h( |2 O" J* y& J E F

复制代码

2）服务器端实现

class WS {
5 T, W4 n- F a, U! C+ ~
var $master; // 连接 server 的 client- D! }. U) }2 Q% `
var $sockets = array(); // 不同状态的 socket 管理* w$ G( J+ S" T3 @1 W0 ^3 R5 N
var $handshake = false; // 判断是否握手
5 t( Y, D/ V# L+ D: r2 ]4 b
1 ?: S8 z5 |! ~- L1 W% X9 x, i
function __construct($address, $port){- @$ [0 _: |( D
// 建立一个 socket 套接字0 F2 T1 d6 F5 N/ v& R' E/ a
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP)
/ I+ J8 O- y1 |6 M4 F% i
or die("socket_create() failed");
. L3 y. z5 l0 p1 E. K
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1)
0 q: y9 D; e9 ^- L
or die("socket_option() failed");% J/ B( D) ^8 X1 ^, D
socket_bind($this->master, $address, $port)
% s% E! w Y/ x& B+ U D+ }/ ]
or die("socket_bind() failed");
4 ?! q5 w- w; H0 }5 V
socket_listen($this->master, 2) ' _" o: r+ ^5 Y" Q6 o5 n
or die("socket_listen() failed");
& b+ m% `' l0 ~0 q. H7 f/ C7 Q
8 i# x$ Y7 c$ J9 h! c4 ?
$this->sockets[] = $this->master;$ j3 R8 G4 V! H' j2 G7 f
6 Y3 U2 U' V* T" [4 n0 N
// debug
9 e9 F- `! O7 y
echo("Master socket : ".$this->master."\n");7 }9 p: c/ V* v _& O$ H. A8 L/ S: L
, w7 }) l ^& U( [" M$ `# F
while(true) {
, R1 a- F" o' T0 n
//自动选择来消息的 socket 如果是握手自动选择主机
% }) ]0 _- I Z
$write = NULL;
) b. u9 h! ]+ X7 D/ G
$except = NULL;5 R; i- t6 o/ D0 [
socket_select($this->sockets, $write, $except, NULL);# ]1 i6 F W8 `7 u3 L
& H" T; s4 A& |6 p" |1 ~$ L
foreach ($this->sockets as $socket) {
9 @& {" h1 b0 \6 j- _
//连接主机的 client * | O4 M, C3 m' G$ B% k
if ($socket == $this->master){% U. o3 Q* z7 ^* x4 D
$client = socket_accept($this->master);. g9 E9 N% ]+ B4 ?. P
if ($client < 0) {
5 L; L) h" h4 ]2 M& u$ w
// debug9 o" k3 x$ I& }, G, ?- n5 H8 J
echo "socket_accept() failed";
2 M' \2 q( }3 H: G3 s n# S4 S6 h
continue;7 Z) X, Z* C6 v5 O' Q% U
} else {
& j5 y o5 h; C% s' i
//connect($client);
x3 f0 p* w f& \( h
array_push($this->sockets, $client);
5 s0 E3 U @" z% y: W4 n
echo "connect client\n";
1 `: A" g: A; S* N+ S7 o$ F
}
+ b. v, w$ @0 B/ d0 r0 H) o
} else {
4 N, S0 u4 n; S( U0 T$ R! l
$bytes = @socket_recv($socket,$buffer,2048,0);8 d- @, y9 k9 ~5 e
print_r($buffer);
3 n2 w, W7 t5 A; H2 y
if($bytes == 0) return;0 s8 M( r6 b( K" Q+ y5 |! K' b
if (!$this->handshake) {
+ e6 y+ [+ {/ V
// 如果没有握手，先握手回应) J9 F$ {5 e; m6 U
$this->doHandShake($socket, $buffer);! M% O+ x5 P r0 S( z
echo "shakeHands\n";
& M& C/ E" G8 X$ y0 L2 a
} else {& j0 Z# d" y# b* x
! a* O- m8 g8 {9 W' `
// 如果已经握手，直接接受数据，并处理% ?, d. ~$ f3 @) {1 A) X
$buffer = $this->decode($buffer);7 b" N B) J( e9 I
//process($socket, $buffer);
- A6 c& G% F+ p9 V) z" ~) I. s
echo "send file\n";
A/ c8 Y) l2 {6 v* `+ |
}
) v" u' ?2 G3 O. W
}
6 H+ C& `/ m* N Q8 Z& z3 c
}
+ j; m H( ?' M$ S
}( u) i! J+ a' W
}$ ~: a' s9 J/ h/ z
% I# v( H( I) z& M. l, R# g0 c4 ?
function dohandshake($socket, $req)' r b& H3 J- j8 g5 g( c
{ S/ M6 f1 V( O8 H
// 获取加密key4 m4 e/ D2 E& o6 x
$acceptKey = $this->encry($req);1 `# T2 ~8 G% E: M
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" .5 Z% c. ]! {: {
"Upgrade: websocket\r\n" .& y5 r) o5 _5 Y' v. b/ r
"Connection: Upgrade\r\n" .: q4 R" F9 Y3 i% o" r: J: d8 @ F
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" .
7 r2 l5 @; i8 E3 K% l
"\r\n";
2 h6 Q W K# D- U
& f$ d/ N% Q# L8 @
echo "dohandshake ".$upgrade.chr(0); 8 ^1 W" b- `3 K9 C. J/ H
// 写入socket
8 C) N- Y; g' R9 a2 b
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0)));
& e2 v. x3 e0 T+ b0 M! `) @3 z
// 标记握手已经成功，下次接受数据采用数据帧格式- Z( L# x$ X4 H# @& `
$this->handshake = true;
$ D2 `8 ^; j/ @& { n& d
}
: n2 Z8 V0 J4 I! I
. K/ g$ `9 H* T5 l4 W( b4 Q$ W
; ^2 e" ? {+ c' l- d
function encry($req)
. {8 h, X' W8 ?) h& Z( Z& `6 C
{( x9 {" A& n" U- n4 m4 o1 {
$key = $this->getKey($req);! W0 R s- p5 c# w( _* p. E
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";: I3 ^" U! a& |, J. N& r
5 M/ e" M& B6 i3 m, B( ^( P M6 t
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));
0 ]1 U6 b4 b2 A8 G: k
}
# f/ @" g* W4 \5 M
& s \9 p6 m' w) i( Z9 `: w
function getKey($req)
, L" c/ B8 [+ D6 G P
{
2 E9 S3 e6 J- W( L' Y+ W
$key = null;
: _5 h/ Z! {; K% ?
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) { , j6 h! r% a) W. Q$ B
$key = $match[1];
+ P0 t/ W) ~5 {
}
* s/ H1 e8 o' \0 {
return $key;9 J( a! [1 W( H, S1 a) T& G. [
}; S+ X/ ~$ E% G5 @) M3 p; [
" ]* ]% s/ n8 s# C( K
// 解析数据帧
0 J% u- r1 c" e ?; f# |2 X9 y
function decode($buffer) / f" L$ [1 F# C. _- J$ x% {* H1 N& ^8 v
{
! M* A- a2 P- j! C! _( D: Q2 \$ [3 @
$len = $masks = $data = $decoded = null;
* M5 z6 [5 J7 o
$len = ord($buffer[1]) & 127;
% O3 a/ H8 @& v4 |- R
5 A$ M* O1 I3 ?* B' f
if ($len === 126) {
% m! L$ B! a3 l5 q# A+ A4 R
$masks = substr($buffer, 4, 4);
3 t. a! O! ?* h0 X- n+ V/ A( |( t
$data = substr($buffer, 8);, [) O n( B3 k! \
} else if ($len === 127) {
: K. Z# l$ v' f8 j9 x& d$ @: o
$masks = substr($buffer, 10, 4);
+ |& W" W, u5 D. l" h! L
$data = substr($buffer, 14);* M Y( P" ]" t/ U* l% u0 n
} else {& ]/ j) l" O, [7 j7 W$ M/ g
$masks = substr($buffer, 2, 4);
% R6 l, r4 U$ n' s. M% q
$data = substr($buffer, 6);9 S) e' e! H2 F( h
}
4 v7 ]/ w: b3 K; G; U3 p: l3 R' _
for ($index = 0; $index < strlen($data); $index++) {
* B# a" i" T( G; M$ M1 A. Q
$decoded .= $data[$index] ^ $masks[$index % 4];/ ]& w; E. ~& z. n6 E, M6 w' |! _' c
}
1 [: ?7 P9 h; ~0 r
return $decoded;1 m: B7 f1 D/ J3 O% l2 y( m: \: y
}/ c) H( \0 p8 ^3 S E
$ F: I* O8 Y L" k" z
// 返回帧信息处理
4 M+ D0 x [; T
function frame($s) 8 i& W/ ?+ w$ G* f9 E
{
; F: X( k* ] |$ U% G0 ]
$a = str_split($s, 125);
- J( z$ B" z8 l- }1 x% V U
if (count($a) == 1) {7 v0 H, b5 N0 Z4 e
return "\x81" . chr(strlen($a[0])) . $a[0];
1 }5 I4 R( _* U1 M6 |! s) b
}- F8 E. H+ P8 q; x
$ns = "";; Z! o9 A$ K6 U' f" R& m
foreach ($a as $o) {8 P" s% g J% T5 o1 a) u* V
$ns .= "\x81" . chr(strlen($o)) . $o;
/ u7 u4 S! i7 j
}4 V9 C5 x U2 T: H% V1 } Q6 p) |
return $ns;# ^8 g0 [' A+ H% p. l: w
}/ Q2 S- K8 ]( ]" s
0 Y2 e2 v T) J
// 返回数据
k* A0 l4 L: z! W/ A
function send($client, $msg)4 J7 v. r. U: S6 T
{# }& ? n1 J5 K1 C
$msg = $this->frame($msg);
. v3 R; q7 Q2 o3 Z, }& g9 z
socket_write($client, $msg, strlen($msg));. w Q$ }! G: o) @+ U
}9 a/ V3 L4 ~4 J3 m4 l* f) U. z. J9 D
}
4 u2 |2 O6 e! J' \ t
) p B8 X+ \% x8 Y# N- T2 d
测试 $ws = new WS("127.0.0.1",2000);
. i+ l, j1 o) f2 B6 K5 Q! n
0 s }* t9 V6 D

复制代码

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

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

浏览过的版块