PHP 简单实现webSocket

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

1）客户端实现

<html>
+ u" Z; l. z4 Y$ S( M
<head>
+ \8 x6 s- D) A. ?8 {
<meta charset="UTF-8">
$ n- x% e( S! h! K+ g
<title>Web sockets test</title>
7 _9 P) p% u: `- b0 F& E
<script src="jquery-min.js" type="text/javascript"></script>6 x$ S! S4 T ]; f2 I3 U
<script type="text/javascript">
+ ^- t6 Y. N, i" L
var ws;. i7 i0 @5 P1 h& T4 X# R# E
function ToggleConnectionClicked() { . d7 P: N' [$ Z- E ]
try {: W3 b/ j+ X: _8 _) z# \
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器
; D6 n: Q: T. D
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};
. F$ I8 {* h4 ?3 U+ o: J
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);};
1 a( n3 W- }9 C4 I" {
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);};
1 \: f% h% j1 k0 W& `
ws.onerror = function(event){alert("WebSocket异常！");};* H% }1 Z9 _7 W3 g
} catch (ex) {$ F5 `" g5 ] Y, t" X3 D: ~
alert(ex.message);
, i5 N& w1 W6 d3 n% y) [
}
6 b5 c& w7 l! r4 c6 r4 J/ ?$ j
};" }, E( z( m- Y3 M
- v3 n8 r, \' Z
function SendData() {
9 ]- R& s% W" Z
try{" i+ ?6 x7 ~+ a1 N x
var content = document.getElementById("content").value;
0 i0 ~, y1 P5 r" _& A" q
if(content){9 S( c) _6 _! c' e
ws.send(content);- a5 }* ~/ W5 Q; \. `
}% M& a( }6 J: G
8 }. r) q: \; c- L5 r
}catch(ex){: b) h# u t7 s% B" p X7 n6 Q
alert(ex.message);
9 J; K, s# p1 D7 D$ `- L
}/ B1 ]1 ?) q( j! Y) q* M8 ~3 \5 X$ d0 N
};
/ g2 I1 \ T% k, N7 x8 N3 `! q
7 y/ V$ f+ ~# k8 J) s7 i# I
function seestate(){
6 g; ]5 n$ g& m( X& Q( p$ }
alert(ws.readyState);
1 D# ]* A- @8 m3 x
}
0 `5 U& `* u! E# X
+ s5 C2 Z X' y8 ^2 p; g
</script>
- o7 b g; m4 Y1 Q; J; P
</head>
+ n8 v" T9 |: Q
<body>6 ?5 a. d$ P1 x& H% G$ J
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br /> _9 T9 ~0 w7 k& ]" M# Y( w
<textarea id="content" ></textarea>) z* N+ K/ r4 _! [
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br />" }0 Q( a9 }/ ]5 s
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br />
B9 q% a& z$ l1 n4 m
: W3 t1 A3 O& Q, D- @4 t* G
</body>7 x9 _& j( v# Z+ J% z% t5 I
</html>
' w( Y, U. y, f7 X) S. u2 _

复制代码

2）服务器端实现

class WS {
" f; l. s8 A; {, B% ^7 s% l
var $master; // 连接 server 的 client
6 P9 R4 z: v2 O8 X5 X" N
var $sockets = array(); // 不同状态的 socket 管理
. T8 T l5 E0 D5 H+ a, f4 s6 R
var $handshake = false; // 判断是否握手
7 K6 Z% V* V$ }; x7 }& I/ ]
+ d: S7 M5 t5 O* P' ]( d+ a
function __construct($address, $port){+ H4 z# H% b# W1 B
// 建立一个 socket 套接字
% S; W. o" J% h4 S/ q% d
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP) & o5 e; F; m; y ^1 C. f, N7 h
or die("socket_create() failed");
/ E. d: P& y5 x% y/ U+ J& H5 J, L: \
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1)
" Q7 H {6 |; v
or die("socket_option() failed");' a% Q% g1 P0 i5 ]4 Z
socket_bind($this->master, $address, $port) & ?% A9 }1 P; b
or die("socket_bind() failed");! q/ U: `+ O9 a3 f' {
socket_listen($this->master, 2)
; ], ?% [4 F }! [7 B
or die("socket_listen() failed");# V8 }2 c. z- C
( L! K$ Z% v; {8 c1 j+ e
$this->sockets[] = $this->master;
2 v3 H: N! K3 M$ q. |/ X
' c# u( {9 r7 Q: Z( {; W3 x1 u
// debug6 F7 }* `2 A$ G5 J& R$ D3 s
echo("Master socket : ".$this->master."\n");0 P+ [- n- @: W( S9 Q; a
1 W- v$ n, A( j" i3 G
while(true) {" K- u" l- Y% H
//自动选择来消息的 socket 如果是握手自动选择主机
( I/ |# K; y" v9 _
$write = NULL;+ ^' L" \2 M+ ?0 Z8 \2 [! F6 D
$except = NULL;
! N8 e% e. t7 \: F! e
socket_select($this->sockets, $write, $except, NULL);2 P! l. D) s' t; z
4 a" x& h L' y. F- p' P
foreach ($this->sockets as $socket) {
; C! P% N4 Q4 [ v- }) U
//连接主机的 client
, E& {, u7 I* a
if ($socket == $this->master){3 h) r! `7 g/ j3 B8 \6 Z) P
$client = socket_accept($this->master);( q* L- ] M# J0 g& H/ _- z
if ($client < 0) {8 U2 w: c% S9 h& [' p7 Q& [: c: K4 w
// debug
3 ^9 }: }) g8 G
echo "socket_accept() failed";
% f( L, t5 v; G
continue;/ Q, o8 {& G' Z4 d# @6 |
} else {
3 n8 M" F! L! `" t: m* }
//connect($client);; c7 {7 L( }4 p( ~5 d" g
array_push($this->sockets, $client);
# A! h: E! x9 h# ~# r& c
echo "connect client\n";/ O* v1 p0 v: f; o
}
" N) o% R% I% W; h$ F
} else {1 G1 a" r# P h4 x6 u" P- V; N
$bytes = @socket_recv($socket,$buffer,2048,0);
+ f- x- _& V' c$ }/ |: [: x1 J
print_r($buffer);( q% N2 r; n3 ^9 F/ N# J
if($bytes == 0) return;" x0 z4 r3 g6 x
if (!$this->handshake) {
2 n2 L* y( Q/ ~& j0 S
// 如果没有握手，先握手回应( N8 {& b) A8 U3 Q" [6 R
$this->doHandShake($socket, $buffer);$ [2 L9 Z# y! X" v; l5 N: Y5 I7 l& w5 b
echo "shakeHands\n";6 S/ i8 ]8 F/ p& [7 A: }
} else {
' {* J, L7 [& b
+ _4 _& H9 B' N7 M' L% I( W% t2 |/ j
// 如果已经握手，直接接受数据，并处理
: X& x0 \1 G7 Y% [* e. R
$buffer = $this->decode($buffer);
2 n* R9 H+ A0 z- f# g9 _
//process($socket, $buffer);
+ l8 D( \5 E' E' T/ ^1 @
echo "send file\n";( N$ g- \: m& m4 X, K% T# B
}' n3 e! H( M+ l3 u
}. ^$ q0 ~* K$ y3 \* W3 Q5 _5 Z
}. G- m3 k3 @% R. x6 |: H2 d
}
2 ]5 ~2 W' y9 O; L2 p
}. ?/ i. ]9 O9 D, n2 a
: {$ N: t2 G; H% e
function dohandshake($socket, $req)
( c) l& {2 H( S% @" p9 ~
{
! f) d7 K% Q" ~7 b9 p/ E* {: e
// 获取加密key4 S1 ?% o/ S- B0 W4 n& H6 e) t! z
$acceptKey = $this->encry($req);
2 D7 O6 g7 m: g7 R
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" .
! C( h9 @% K8 l, ]4 {. T
"Upgrade: websocket\r\n" .
$ ], k# b' l+ o: D8 o
"Connection: Upgrade\r\n" .
+ |0 @. W; u: g( V
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" ./ a# x! R% m# D/ `/ ]+ b$ R# c' N
"\r\n";
% j# L) D0 ~! x& l, Y0 l
( k" P* I1 w% D; Z$ O" F8 [- O2 {4 v
echo "dohandshake ".$upgrade.chr(0); 9 f* U1 p5 A- h% F1 p
// 写入socket
; S% o5 J$ b9 m; L" s! w; C3 a/ D
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0)));
9 \2 i5 k) y9 \! Q9 ^& X1 a
// 标记握手已经成功，下次接受数据采用数据帧格式
1 L2 q8 q k0 X! I
$this->handshake = true;4 v3 R% [$ b3 B' L* U) N3 Q, ^
}
" Y+ Y! a/ E5 Z4 b5 ?
! g+ K$ I6 O6 e7 S0 g
. D( d4 g9 F. x
function encry($req)1 L3 K2 b' m* m3 S5 _+ }& c
{9 ?* k! r" m5 D5 S. o5 }
$key = $this->getKey($req);
( z0 g/ Y6 _4 Z: m. [
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";: f! C' I& m. A0 x
0 v- k( Z) E4 u8 @( v& U6 g0 n2 ^
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));" b1 [. V9 f E3 H: [; t
}
5 `2 c* H6 x, H& U( |- k, w, ?( z! n: D
5 K8 u$ N: ~& U0 X' S6 ?
function getKey($req)
6 p; N2 w& \) ]8 o) _
{& z; S3 i: C8 Z/ S
$key = null;
" H" ~; n7 v1 X$ |+ O
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) {
1 L& h7 m$ ^6 y1 J% N) _
$key = $match[1]; }' U0 w9 r+ @- |
}- F6 E( X1 e d3 g. b0 w# I
return $key;) y$ I# ?; k, D) {
}
6 ?9 f0 b9 H8 q
9 s6 q% n$ e0 a( x9 X# V4 n
// 解析数据帧
6 y* j, n8 L: t. f' H% g
function decode($buffer) " e4 V A* S2 x- a% c
{
9 ^( O1 j' U% Z
$len = $masks = $data = $decoded = null; }- p' E1 P2 o
$len = ord($buffer[1]) & 127;
- s: P# Z7 Y/ I/ z& @) a
& i+ S$ O! Y6 p- \/ V% N) [3 [
if ($len === 126) {
1 r- P9 W$ ]: u: u
$masks = substr($buffer, 4, 4);
' t5 y0 u# J# Z# @4 Y
$data = substr($buffer, 8);
2 u& c. I) e6 D; ]% u0 r! N
} else if ($len === 127) {1 N! c7 M5 U I1 l5 v
$masks = substr($buffer, 10, 4);
8 n% q% e! z" o4 {
$data = substr($buffer, 14);9 w2 f2 s4 E( f
} else {- S* U& x- i8 F4 d! G+ {0 i
$masks = substr($buffer, 2, 4);
' D% ^; q# C$ x, h* p( v$ V8 Z
$data = substr($buffer, 6);/ I" L+ ~4 }. m& D3 b, a- @# H8 S
}* b# `8 v$ j: @" d/ i
for ($index = 0; $index < strlen($data); $index++) {
) m! J( n1 f0 d3 F
$decoded .= $data[$index] ^ $masks[$index % 4];' |3 b, c. d: h6 N, ^' Z( Z' w7 A
}
8 {( h4 R! n- z2 V: d8 B+ x
return $decoded;
/ a: m& k+ [( c6 s' Z" `
}3 y+ W ]/ A Z3 D: z8 `& d9 Z
+ q) s1 q9 M* ?5 I# F, V/ A9 _
// 返回帧信息处理
% Q- N# ^' H4 w0 u2 w: ~# L' K3 K
function frame($s) ' i4 N$ ~; \0 w( R% z
{
+ I ~3 W% B7 D0 |) j# r0 v8 H
$a = str_split($s, 125);
* u- R9 X- K J6 r# t2 E$ q
if (count($a) == 1) {1 p7 I) }, J7 j: z1 q1 E
return "\x81" . chr(strlen($a[0])) . $a[0];0 ]- N+ k* c' Z" k$ p1 ^/ g( m
}7 }6 x* J( i; Z6 a4 o
$ns = "";& O' |! H- [* M. I
foreach ($a as $o) {5 s5 k2 N T1 S* s4 C' ]
$ns .= "\x81" . chr(strlen($o)) . $o;
/ @ v+ F8 l# n- U: b: J& e
}. M" j& S: P+ a" p) Y* i, N
return $ns;
) ?9 ^/ X# E1 J6 z4 K, l
}8 K' M% a; S: A* v# V
% K Y( W) s9 s) I5 R: j* S/ E/ y
// 返回数据
2 E" Z7 A3 ]0 q8 `! G
function send($client, $msg)2 ?7 j8 f% Z9 @$ a* w: ^
{
! e: @1 b! O& h ~8 z: W, R/ c
$msg = $this->frame($msg);) ]$ H2 T2 L+ D u, r* v6 [
socket_write($client, $msg, strlen($msg));
% @! b# S8 X5 K7 `" A# Y4 z
}+ r6 r$ c: Y* E" b& T9 v1 |3 q
}
* ^. D, H) {' q. G% z
, h, }( j3 f) M1 y
测试 $ws = new WS("127.0.0.1",2000);
; G" n3 o7 C; X. v7 P" l/ d
2 ?9 g- R3 [9 c( t/ G- `$ J8 |

复制代码

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

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