PHP 简单实现webSocket

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

1）客户端实现

<html>0 V2 _5 b4 l2 ]: t; @
<head>" v' j7 x1 j' n$ A# s0 ~$ `% v
<meta charset="UTF-8">
& s7 n0 m+ U! ^' X! k
<title>Web sockets test</title>" t' h7 r+ P( m: t4 g( G' d1 s9 ~
<script src="jquery-min.js" type="text/javascript"></script>& U/ t, U& ?; v x7 Z! u- l# e
<script type="text/javascript">
7 Q2 a8 P4 e# c& L! }
var ws;( |2 }9 D' G* P) n. n+ C/ F! k
function ToggleConnectionClicked() { $ S8 y5 h& S3 l( K1 I% u" z, e
try {2 K( [( {" o9 M1 U- I* X& r: M
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器
$ x+ _' z6 @: R# u' g0 g7 ~
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};
5 [" W8 Y) P1 n+ J+ s% C2 R
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);};
2 l" b. [. [% M& v# }8 o
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);};
# y( K% H" y; D M5 A
ws.onerror = function(event){alert("WebSocket异常！");};. V+ a3 e2 n, L# D
} catch (ex) {
# t3 L+ L0 c# E. z; c
alert(ex.message); : b0 T- r; b+ ~3 j) q7 {8 r; {8 }* i
}, R; @! _2 d; Y" q0 r9 {. n- N
};
; E6 A; [! E+ j
! n8 X. i. R; y5 z2 X" G6 R/ X2 T
function SendData() {6 T# e$ P* L, E8 L. p
try{* p- Z' W0 O5 Z4 y5 s" k0 A
var content = document.getElementById("content").value;9 H3 ^( o0 B6 c& |0 a/ V7 s
if(content){& Z/ z" ~2 R- j2 j
ws.send(content);0 x x5 r% H: x2 }, I4 w
}
4 q" t% J" w- w1 U1 s
3 }" o* P4 z; U2 l/ G) O1 Y
}catch(ex){
, H( @ W: q5 f6 Y
alert(ex.message);
7 m9 t0 Y, g2 [5 }7 B2 M
}
. W, e1 ? l- a5 W- o* _
};
) x/ w: G! s6 B. f7 L
$ r, ^5 g; {4 V( Q
function seestate(){
7 O+ C2 h5 _& `0 y! W7 v
alert(ws.readyState);
8 v) ]2 I! u! T9 C
}
. v, L7 M% X! q' V/ D0 s0 q, i$ V4 O' B
& L& T( i! |3 k& _& g) _7 m
</script>
9 U, `, d8 v+ p7 i7 V s. V
</head>" H( m; h0 i% E. I n1 @
<body>; v4 S2 k+ g( E$ z. v/ f I
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br />
+ z* q2 `5 r( y0 G! `
<textarea id="content" ></textarea>
/ R# j+ x% Q0 T( \5 _- B& f
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br />
3 ]5 K1 c! o) t8 W
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br />
1 X Q. o* v" i& g$ C* @* v
( V% a8 w8 {4 | T# r) t
</body>
6 V5 ^* \ z- B4 X0 b9 x
</html>" P5 v8 o2 ]3 ]% Y' P. C- S F

复制代码

2）服务器端实现

class WS {
/ v- V+ J/ Q& _$ D2 q4 I
var $master; // 连接 server 的 client
' K8 z- I& X' i' v/ A1 I
var $sockets = array(); // 不同状态的 socket 管理
) A( @% Y4 w8 `, c
var $handshake = false; // 判断是否握手
1 B. k7 C% M% W# V
( M c; W3 i" @9 a0 u; c7 R
function __construct($address, $port){7 M4 v4 Z m% r5 j
// 建立一个 socket 套接字
4 l0 g+ X) N& _& N: |3 S- \* `* W
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP)
' t$ ?) X+ E$ a: T
or die("socket_create() failed");' g. F9 F5 ^. P- T" `. Y3 Q
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1) ' H. `; H/ s; S+ {; j0 E' \. M
or die("socket_option() failed");
9 a, ?$ i; g# P! {% ]( F5 ~! Y
socket_bind($this->master, $address, $port)
! r/ t3 y: ]( V5 i
or die("socket_bind() failed");
/ k- Y- J5 g& I2 I" \' @/ o( @
socket_listen($this->master, 2)
4 v5 {9 |2 [1 Z, q" ~
or die("socket_listen() failed");
6 e; p- |6 j; @
$ P6 k- q3 S4 ]# T4 C
$this->sockets[] = $this->master;7 u! q1 N% t: _+ I3 e0 T& u
1 ?' r$ O8 b7 a1 r4 o# W
// debug
$ Y7 v+ V5 K1 L% N/ B
echo("Master socket : ".$this->master."\n");3 [0 L& V0 Y3 P, J$ Y9 g) K( U
3 n+ B6 x+ ^9 m: Q% J
while(true) {8 m$ a/ A' x: I
//自动选择来消息的 socket 如果是握手自动选择主机& i# W+ T3 s. P$ k5 ?" W
$write = NULL;! N' u" y0 n1 d% s0 N
$except = NULL;
* x9 \) V+ _9 ^" c% b
socket_select($this->sockets, $write, $except, NULL);
9 O& S* v1 K0 Y
) e2 u- `$ `8 r* ~/ o6 E: ^' M
foreach ($this->sockets as $socket) {4 p8 ?0 r# Y, x7 U; W, V/ J
//连接主机的 client
9 Z/ `8 x* X3 u( @ d/ k6 M
if ($socket == $this->master){% ?) X7 X! }2 w1 v% x* t. q1 z! n
$client = socket_accept($this->master);0 C6 ~9 b% F* \" s. w1 S
if ($client < 0) {; c5 X& f3 B0 k) R
// debug
1 k" {6 \5 O' c1 |' J$ U
echo "socket_accept() failed";
+ \8 k/ ?) x, H1 m
continue;
+ y1 t9 t! c. D4 R z
} else {
9 S9 n t7 Q! q' w( j3 Q
//connect($client);, G7 X, p1 j7 b+ p7 `% _: E
array_push($this->sockets, $client);, T4 }6 g+ Z1 k! f) h5 g
echo "connect client\n";" O; J3 w/ k* \, `4 J5 q
}% [+ f) z q9 p
} else {
# r* g- D/ Y. b" o$ ~2 G9 a8 b
$bytes = @socket_recv($socket,$buffer,2048,0);$ L* O& [# k- L' m& S0 Y
print_r($buffer);
/ P1 o9 a2 _( K6 S) F
if($bytes == 0) return;: w1 V& y: J T* P, J8 G
if (!$this->handshake) {
/ M; i+ p: ] @$ R1 ]; M$ e
// 如果没有握手，先握手回应
1 z4 H3 g# P1 _- Z$ v8 \$ \
$this->doHandShake($socket, $buffer);
; c- x& D L' R6 X2 x4 S+ a
echo "shakeHands\n";
+ p6 x: [. k! K+ f2 d" ^' [, E* G
} else {1 \- J, E9 U) |+ R
3 s2 z8 G0 G8 N* B
// 如果已经握手，直接接受数据，并处理
, A* ^! C: p+ C3 G2 ?+ M4 _
$buffer = $this->decode($buffer);
. T5 a% u& A" j R% y
//process($socket, $buffer);
& V6 w" b$ Y& q X e$ B
echo "send file\n";0 s# V% H/ c& b" J" ]" _- D
}
* i, ^0 x4 c3 F, a: p
}* T4 F1 ~6 d2 J0 G
}
# e8 H5 ]2 Z- t+ }# \& M8 f
}/ k% f, |7 ~& B
}
' X1 O( T' Q9 ?' ]* U s
/ u6 R5 V4 ~1 t7 R: U$ L7 m: y
function dohandshake($socket, $req)
$ U; ^' V1 Y2 v: a5 C, c5 _
{
; z# h" A7 P% d* z# Z r( |
// 获取加密key% }3 N( f% Q/ x W. z
$acceptKey = $this->encry($req);, b" u' f$ K, U9 _' `, y4 G
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" ., p v' z. U3 D/ ~8 z2 K0 ]! p
"Upgrade: websocket\r\n" .
7 Z/ @0 b3 V1 ~3 M2 O% y+ M
"Connection: Upgrade\r\n" .9 X) L v5 U1 w7 S" y- e
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" .
# ^* I" N; g& Z$ I: V
"\r\n";
% v) N+ N+ g `5 Y8 j
7 H3 g% F- `1 |1 \& I8 ^# y
echo "dohandshake ".$upgrade.chr(0); T) e( u6 ]0 P/ c/ i6 o
// 写入socket
# O' N5 g! q5 P# X2 ?3 V% c) G
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0)));9 M) H/ t9 R2 }3 ^# {
// 标记握手已经成功，下次接受数据采用数据帧格式
1 n" A$ [( |7 _ h
$this->handshake = true;
/ j; T" f8 V% L8 y1 u5 u
}4 r7 L5 B# D# [* p( Q' E( u
. b/ ?" j% ?4 F X" ?
7 @: S$ I0 s0 m) c- G
function encry($req)9 {- O& h4 y/ C1 p
{
9 Y8 x1 Y( E0 D5 l/ @: Z. s) @( V
$key = $this->getKey($req);
( t/ O; C1 g) J; {' x% Y$ @4 y* L
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11"; p) y [! D7 ?7 o' U0 G
: d- q- y+ _! W
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));& F4 v! [+ x: O0 K" ~7 I1 @
}9 y6 U0 L7 z0 k) X, d% x/ Y
& b3 d* W% |3 r. m( E0 M" O
function getKey($req)
! k* y: o4 N7 N
{1 z, o7 r% F- E3 ^. b
$key = null;8 }5 R! g# t, ?. s0 Y6 S8 E
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) {
- C8 B0 W' o, [+ I
$key = $match[1];
$ G6 F5 {5 ~7 y5 e' `
}
6 f( d/ \/ d8 s
return $key;
0 z# n7 ~9 H, T- A1 }: ]6 l
}
! M5 U* S6 n. B( Y, t S
$ D+ x; [. R0 ]7 o
// 解析数据帧
% z8 }7 `$ C4 f2 X( \, D$ a
function decode($buffer) " e& A4 i; Z: D9 p1 r3 ~3 d* |* F* h
{& a& |" Z) x( I" M/ [. v
$len = $masks = $data = $decoded = null;
* B) J' A% `+ ]) ~# i
$len = ord($buffer[1]) & 127;. S6 ]5 `! s, h) B1 w
6 c p3 |; Y3 a4 E! g0 y7 K
if ($len === 126) {
* Y) N( C- w5 L$ ^6 W1 `0 m
$masks = substr($buffer, 4, 4);9 i. J( G+ P7 R; R
$data = substr($buffer, 8);
9 L% o+ ]7 c* G* @! ]1 U, k' O
} else if ($len === 127) {3 R0 B7 p* w9 X6 m4 K& }* i( a9 H
$masks = substr($buffer, 10, 4);
& A7 n0 x9 }3 }+ y
$data = substr($buffer, 14);& m" I' M7 b0 s8 Z
} else {
1 U# R3 s. ~/ w& I- Y b, z
$masks = substr($buffer, 2, 4);4 o3 w/ y r" i% w: Q0 |
$data = substr($buffer, 6);; r' G, v' f8 R3 H% B4 o# Z
}& X6 P' R& N7 _9 q
for ($index = 0; $index < strlen($data); $index++) {/ }$ @4 K; y. i) \0 q- C
$decoded .= $data[$index] ^ $masks[$index % 4];% Q+ p$ l, ]: W. ] v
}
# d# @1 Q' u6 o. e1 a
return $decoded;9 t3 m8 J6 l, t$ q( |
}
& J4 S( i+ E# }4 r
: @+ @# S! u9 ]$ W' X
// 返回帧信息处理
6 b W3 W' A5 j5 Z# g
function frame($s) 9 P9 a. b* y! [' t
{+ e/ w4 W* o2 H4 D! @/ V, J* q
$a = str_split($s, 125);( H8 O5 R* [- \ U% u
if (count($a) == 1) {
* H3 s" B+ n: \ j2 a- x; y
return "\x81" . chr(strlen($a[0])) . $a[0];' F5 D, ]( K; S& G
}
; ], y2 A" x' z* [( q: L
$ns = "";" k3 z( |& O' Y2 Y% {& y
foreach ($a as $o) {
% H! O* K6 I4 k6 \- G. O
$ns .= "\x81" . chr(strlen($o)) . $o;
) c" ^/ K) X. ^! d
}, q3 H! X8 o# S7 z! H2 V0 m' h3 `
return $ns;
3 w1 F1 j( [: K5 q
}
- m* E; x* |1 R& k8 K
, _5 F4 _# w- W1 w3 U9 A
// 返回数据7 S- w7 h6 ^# ^2 ?9 J0 c
function send($client, $msg)
5 n5 c# n; |! @3 K, l+ r: W/ a
{; o q0 }. g+ e
$msg = $this->frame($msg);
3 M+ T8 S, Q2 ^0 H% A2 W
socket_write($client, $msg, strlen($msg));8 I5 |% x+ J, \; i/ z
}4 V1 M6 d8 y2 i" d- ]* I
}& \: I. b* U# d/ K7 ^5 u* E
& P$ _+ Q1 n/ w4 R( `
测试 $ws = new WS("127.0.0.1",2000);
0 Z( |, U+ O: ~( h; M
/ m1 A4 d8 x: b) f- E0 Q4 i" f

复制代码

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

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