PHP 简单实现webSocket

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

1）客户端实现

<html>7 q" G0 b8 o0 V# ^& x
<head>
& a5 |4 h) k% P! g3 s% T
<meta charset="UTF-8">* i+ G( }* Q6 ~
<title>Web sockets test</title>
" S1 O" {! | z( L! R
<script src="jquery-min.js" type="text/javascript"></script>$ u6 A6 O# g3 W1 D @9 m) r# `
<script type="text/javascript">
, g$ ]1 w7 m) M( K/ y( s
var ws;: W3 ?( _$ R' B0 X2 O: r. e
function ToggleConnectionClicked() {
# C3 d! l- @+ X9 p: S2 p, o$ H
try {
" @" X% p% T9 q! S1 x: l; D$ R% w$ k
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器
# h+ m8 y1 s/ g' n* F2 ?
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};* j6 K# E0 m' P) i7 m; s' S) C: [$ M9 c
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);};
9 @/ V' k: n' i1 B5 J
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);};+ I m% f9 T$ F0 z- K
ws.onerror = function(event){alert("WebSocket异常！");};7 Z9 Q% G5 D' R6 x
} catch (ex) {2 a' O3 l% a% N, w. N
alert(ex.message); ! {. k, A1 D3 b) s+ g- X) t+ @
}
0 K9 M3 N" F6 Y8 j( V# e+ L
};+ p# \6 k0 a7 M! J6 G
0 R' A( V+ D }: s6 D3 d
function SendData() {. q+ \0 S0 i9 {3 ]3 D- I, e
try{
( X( ?+ T7 }! T8 P% E2 t: v. q8 V
var content = document.getElementById("content").value;3 U( I. c- Y- P; k7 q6 v% v, N" f
if(content){
- O$ `' G: S5 V+ y: R
ws.send(content);; U2 l8 e3 _( {
}
- \; y7 I! M4 C% h0 M" D
9 @- K; {# \; \1 v+ J( t
}catch(ex){
" v6 s$ _7 V# ]1 A6 R; J
alert(ex.message);
1 G. R" s8 f& Z% ~
}. [- P" f7 Z. l# l. k
}; @9 D0 k2 K, ]' N
8 K) E3 y& p/ p6 k0 B! U) x, r
function seestate(){
) I5 C% A8 g% y& s m! p3 c h9 ~
alert(ws.readyState);
% o8 N! k: c+ H, t1 U4 ~) f
}5 M1 _, Z0 C6 f( q* A+ X
5 s0 {' `( J2 [/ a+ w
</script>2 V6 T4 V# O; k5 u
</head>2 F0 S6 C6 d+ C8 B
<body>
% X0 _6 j* U1 Z5 H; J9 E% l! I
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br />
5 n- \, o/ V/ S4 b) w
<textarea id="content" ></textarea>
, ]( e9 ~$ g! L
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br />
; Q, N. I4 i' q1 u0 O) ?
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br />" d( d* b( p. _4 k% `+ c* g
3 `+ {/ ?' k4 ^6 ^$ P
</body>% |% J( [# v- L; r3 l7 R
</html>
- V8 E: H7 B. |

复制代码

2）服务器端实现

class WS {
, i# a) y3 \9 m! ?1 _& n% p* e. O Q
var $master; // 连接 server 的 client
0 o3 F' V* i4 j& ~
var $sockets = array(); // 不同状态的 socket 管理4 }/ M$ \ T' Q( o
var $handshake = false; // 判断是否握手9 V! L9 A, g0 I* T
+ M' N7 w# R$ b3 T8 d
function __construct($address, $port){
L6 U6 S! t' f9 ~$ _" @
// 建立一个 socket 套接字4 k2 ]* |( @ [
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP)
& @" Y& Y/ i5 S1 a' D. }5 o6 @
or die("socket_create() failed");
3 p$ `9 Q9 J3 \+ \
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1)
1 O! q- n4 t) x6 k" b
or die("socket_option() failed");6 Q. y; O( Y& E- r; z
socket_bind($this->master, $address, $port) ! T5 c7 |1 q) T! U- n- `* M
or die("socket_bind() failed");
+ n7 _& t2 A& k/ U
socket_listen($this->master, 2)
# [3 c a. H' S) W; F
or die("socket_listen() failed");6 ]( g! M7 z: {/ J
1 h& E7 Y8 S# I) L4 C5 d3 v
$this->sockets[] = $this->master;
/ k) ^0 @+ t8 V- B
// debug
6 O9 Y% A {2 ]6 p; i1 c2 b8 h8 O
echo("Master socket : ".$this->master."\n");+ C: I& e7 t) ? G
( P4 |! Y- ]( Z& `2 w
while(true) {
5 W; b( H+ z* j2 t' F8 c9 q
//自动选择来消息的 socket 如果是握手自动选择主机0 F6 Z( S5 k" J% g
$write = NULL;$ |! |8 T/ y! Y* t8 [, ?
$except = NULL;+ z7 O+ A% I6 X9 O4 @$ }
socket_select($this->sockets, $write, $except, NULL);
# a0 c [1 d; x" r- {) r$ }9 t; b- [
" Z7 h3 t6 s, D7 A) D! H% y
foreach ($this->sockets as $socket) {2 r5 R7 f1 \! ?; u' L) B
//连接主机的 client . q" Z* Q" n' r8 P
if ($socket == $this->master){
# g0 I! y% O: V
$client = socket_accept($this->master);) S: D, W- A$ m, O
if ($client < 0) {; q( n0 L8 T1 m" j5 X8 R
// debug
, f! C7 A5 ?3 Q- ~2 k
echo "socket_accept() failed";
* n i# z" `, l3 ?5 C0 H, f4 ]
continue;
" g% k% Q3 U4 a: x7 A
} else {
5 E8 F, f5 c9 u! K( e% `# e9 V
//connect($client);5 ~# b) g; u, _1 l8 U- }
array_push($this->sockets, $client);
. z; V8 V1 u- a4 v9 T8 J$ c
echo "connect client\n";
9 z+ C6 G/ V0 v6 v' y3 ^: c3 S
}8 D0 P! Y M# l4 u
} else {
6 I( T! @- g% P8 j+ Z
$bytes = @socket_recv($socket,$buffer,2048,0);! `) {7 V _8 k# }2 M8 z$ D
print_r($buffer);
/ W+ q+ _5 s) m, r
if($bytes == 0) return;
3 @" }1 p; R4 C/ [
if (!$this->handshake) {$ E2 \5 C3 y ^" G \& t! {
// 如果没有握手，先握手回应% Y0 ~7 V+ m) B% k; H
$this->doHandShake($socket, $buffer);
, f- k' y I1 e* ?6 ^5 Z: x
echo "shakeHands\n";
) I( V4 G8 ^6 ^
} else {
9 C3 z- D1 g9 s: D/ k
1 j5 X! k- v8 a# d) ?0 l5 s
// 如果已经握手，直接接受数据，并处理
' p8 ?' n- S6 C
$buffer = $this->decode($buffer);8 P7 V* O0 _8 V$ z" k
//process($socket, $buffer);
# r; e. {0 f; U w" y
echo "send file\n";5 b) y7 d( d$ W4 U; s
}
2 i. P- B1 \! f7 S
}# w0 f1 d" P7 _$ O$ c) ^
}0 V* \. n7 Z6 H
}
% W0 h" S( L" E- y
}
* x) d$ I2 C: U6 t, U, d
2 a- P+ V' y) L4 F. ]3 l9 J
function dohandshake($socket, $req)% C+ ^3 g X% ~% h0 n4 k* J
{$ Y: v8 o0 v" g9 d$ u2 B, H* |# [
// 获取加密key, |( }$ ~9 ~6 L, v5 Y
$acceptKey = $this->encry($req);
) H+ r+ a! N z, o) y
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" .( P0 _( Q. l: G5 Y" w6 [
"Upgrade: websocket\r\n" .5 ^9 S0 Z/ v, D% ^! c& z
"Connection: Upgrade\r\n" .
( z# W5 e- J& z; w4 I; B
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" .6 w4 @7 ]4 ]& T8 }. U
"\r\n";) W1 [# N9 Z. W: i6 ~9 y4 r* \. B
) K3 y3 ]8 v: d( |( J
echo "dohandshake ".$upgrade.chr(0); & }+ I$ Y7 g; z/ R4 j# C4 c8 Z; K' T/ @
// 写入socket
; @6 C% Z: p" G n- \
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0)));* V6 Z6 g6 C% \# G9 J9 C7 S
// 标记握手已经成功，下次接受数据采用数据帧格式) n9 `/ u' s2 D/ C+ j/ C
$this->handshake = true;+ s$ O/ N" _) U. [! o+ g. G* ~
}
6 j1 z. R8 i. X7 S' A7 n; e
, x; x& P3 p x4 E/ |9 v& Q3 W
4 Q2 g. I0 C! e
function encry($req)6 u1 Q7 O+ M# i7 R6 V/ l
{" M: T4 b& ]" C
$key = $this->getKey($req);
& L0 S# V4 E% R, V% n
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";
: ?( [6 t: Z9 E$ R
4 [; [3 m$ H) e) B; k$ h, P8 ^: [
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));
# d* W2 x3 \3 f" g. `2 z; `
}
2 A' |) Y' u0 B" C+ r0 h; i
* x) U. J$ A- ?+ j- ~
function getKey($req)
) @/ V7 ^: D, R* a- d9 ~' m. D
{
( P3 ]! {& l' b3 z$ ]! b- Y. N
$key = null;* a, o; ]# F1 M- a/ V c) z2 Z% v" B
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) {
, d" T' M( W4 i( z9 m% v
$key = $match[1]; ) x4 ]5 {. [6 w4 z4 ?0 U
}
+ l2 d; n5 _( L- C
return $key;5 C1 b. k0 a- T
}* x* A! P. \/ r
2 X2 v, o! A) w3 w" D
// 解析数据帧
. ~1 t C( k t @3 N0 Z
function decode($buffer) & C1 L/ M( T4 O" f5 H% g0 W
{
# i1 a/ v4 p5 ]7 {; h2 C
$len = $masks = $data = $decoded = null;
9 o5 e! u3 j p, K: t% f$ E
$len = ord($buffer[1]) & 127;
5 r5 }4 C3 {. V' f
0 \. h o+ Y E- ~! h
if ($len === 126) {
5 A8 j( u" N9 {* l3 b$ o- ^
$masks = substr($buffer, 4, 4);
5 K* i( k9 Q, g& S3 C
$data = substr($buffer, 8); K1 f% c/ i. o+ ^* R
} else if ($len === 127) {, N# t' r& T2 A& F' c" T
$masks = substr($buffer, 10, 4);
1 A# w* \- x7 b: I/ w; X- Y) u
$data = substr($buffer, 14);
( I( b- _" Z* i! @
} else {' y, c2 c9 ~- S
$masks = substr($buffer, 2, 4);# X. i7 u7 T7 o8 h. N
$data = substr($buffer, 6);' P1 z/ S9 i; U$ v
}
0 @$ J3 I R; Z3 o
for ($index = 0; $index < strlen($data); $index++) {
8 E3 o" w3 R6 Y9 O! G& U+ D# j; e
$decoded .= $data[$index] ^ $masks[$index % 4];/ l$ Y' H& Q1 k
}9 |: o6 |. Q+ E- ]! A
return $decoded;
4 c( Q, ^: O2 ]5 G) u0 [
}
' V" w N, |4 ]- p7 y7 ?' K; Y
" i0 l2 D# R3 _# l# A$ `
// 返回帧信息处理1 Z4 D0 C/ ~ D
function frame($s) 2 H" P% U8 r5 t' F1 l
{. _1 P/ l/ T0 q2 I
$a = str_split($s, 125);
0 E4 `! Y H; d/ X1 Z
if (count($a) == 1) {
, L$ a8 E- c! {3 _4 M
return "\x81" . chr(strlen($a[0])) . $a[0];
! R( F7 T) M9 L/ x6 H2 e1 Q, z
}# ?; N4 ]# [) p3 K3 C
$ns = "";% x% A: ?! K% X- {
foreach ($a as $o) {4 y& L, W- w4 a+ `
$ns .= "\x81" . chr(strlen($o)) . $o; n! R# j$ _8 G6 s& x) A
}1 O: d) b. r& n) s8 m/ s% ~
return $ns;# l: f ~- d$ D4 k- t+ m
}
" C! c& R7 c' p3 Z* Q
U# W& I9 d6 z8 t# g5 Y3 ?
// 返回数据
! S# m _& @1 u( P
function send($client, $msg)# y, ~! f' V* |4 H9 @+ \4 T
{: q1 j* l0 ^/ Y% x4 E& g- P/ C
$msg = $this->frame($msg);
2 O1 i' z( ~) C& l3 @
socket_write($client, $msg, strlen($msg));
: k/ M0 t2 e. J7 [$ N5 R
}
2 M: j% s2 d. l
}
, U4 m0 y# N$ @
8 N6 Q! Z% V$ Z9 M
测试 $ws = new WS("127.0.0.1",2000);/ I% l2 S/ z |. e. p( y3 u
% C( T8 u: k- o3 h( \9 ^

复制代码

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

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