PHP 简单实现webSocket

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

1）客户端实现

<html>
' a5 V: Y9 p; x
<head>9 B3 H7 k- n2 U" w: c
<meta charset="UTF-8">
) n1 A# Z$ y8 r* a; x1 Y
<title>Web sockets test</title>
, n. K/ i: |$ q- F5 L1 r7 L
<script src="jquery-min.js" type="text/javascript"></script>5 M2 S q, u, q6 v
<script type="text/javascript">
5 ]; e3 b7 x& ^8 R6 m7 ^
var ws;# X6 z; \! {( t4 L! n
function ToggleConnectionClicked() {
8 r$ u' x+ Z! p* `9 i- Q8 E3 N
try {
3 Z! b! q: L7 ` H9 K
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器
$ |4 o( N: K7 x& Y5 I8 T. F5 y# c7 ~
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};
+ |% R5 }4 I6 L
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);};5 D8 R% s( F( F! \: z& U
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);};+ S4 _( @( {6 ], I: K: G F
ws.onerror = function(event){alert("WebSocket异常！");};
! ^1 u) S R/ I5 l9 B" ~2 t7 p2 e9 |
} catch (ex) {
7 K) D0 r% U& o$ R# N+ e, }
alert(ex.message); " x) d, ~( ~/ T
}" e3 z; E; D/ F! m- U1 b
};9 h9 a4 x5 \8 L% | n0 x3 y( K- I0 x
" U9 n* \( L; @& S, L3 o& o& \
function SendData() {; Q# w& ^* k: y: X. G ?
try{" B2 |9 h8 h7 L" x
var content = document.getElementById("content").value;
/ V1 `3 o O- j1 n* E
if(content){
( j% d- X% ? L0 w4 S; V2 l
ws.send(content);
- R4 Q$ T8 n% ^
}* x: s. Q* v3 W+ C0 v' ]
5 r, l* w; V8 m( B. g- _. R
}catch(ex){
. ?) r: y# E1 I( H4 N) W' w r! |3 w
alert(ex.message);; ~& @( L# s Y# @ z. K
}
1 N& Z$ o7 q8 Q, f2 {
};
) S n) a8 l2 [2 ?8 Z
* {8 f, ^" c9 G+ n2 T* M3 ?& d
function seestate(){
6 x* p2 w6 W$ s/ K; S
alert(ws.readyState);2 `; U! ^, W. v; F7 r( P
}
4 [' A: ~# I8 K- {; |& O+ ^+ r; m
& ~3 S, S$ \; u4 C. i6 v8 q
</script>
8 k( J: O! K7 G; Y
</head>+ x5 H' N* v: I: h% U/ C
<body>9 n+ W* C$ ^8 Z5 `' @
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br />( g! K" }9 W) k+ n! s/ g/ ~
<textarea id="content" ></textarea> f+ f0 J& G7 e
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br />
2 |8 R# A% Y% q
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br />
" U+ W$ H2 C0 x* \) i' s( W' D
8 ^% d" O3 j7 {( Z5 l$ n) e! ~
</body>
5 V- ]2 W/ g8 H) C; |9 r9 l
</html>' W0 ~1 o! P" M. Y7 V8 Z" d

复制代码

2）服务器端实现

class WS {
1 x1 u7 v. R. {2 t7 }
var $master; // 连接 server 的 client
- [: v w" p8 L6 N& f: F6 z. G
var $sockets = array(); // 不同状态的 socket 管理7 ~9 G& \+ l5 Q1 s, I/ E
var $handshake = false; // 判断是否握手
) B2 n* U1 B) Z4 k6 F/ x
% l' x4 O: [* U+ M
function __construct($address, $port){
; f7 y2 j) c! |4 s; r) F# r
// 建立一个 socket 套接字. d7 @" B+ q& q7 L& I, l
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP) + D |1 b$ i, b: M$ q( Z4 P
or die("socket_create() failed");7 r& i. y1 b: [8 C S
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1) ) s8 _6 s. B& D: h
or die("socket_option() failed");
' p) Q7 P' X$ g2 n5 G0 H# B
socket_bind($this->master, $address, $port)
& \! k3 l8 x5 l! @4 q
or die("socket_bind() failed");
: @ T; I B; i0 L# U% a
socket_listen($this->master, 2) $ z2 ]) e! d+ a! |# @2 U
or die("socket_listen() failed");
) |1 h8 [; Y9 r
* F8 e! _- m5 K9 \
$this->sockets[] = $this->master;" ^' b! {1 A7 ^) C
* |; E8 n' y: R' B; N! z* N' E
// debug8 S0 ?$ j% j; k# u) G+ R
echo("Master socket : ".$this->master."\n");
( A1 L2 N0 F/ S+ H1 n5 F9 I
" G9 F6 q+ Q- K: N' N- Q
while(true) {7 \+ N* w; A" C
//自动选择来消息的 socket 如果是握手自动选择主机
3 m1 c9 z( k& v9 y7 v; t; _, b
$write = NULL;, ^, m! A" r1 Z& _" n% U4 h
$except = NULL;
^+ v8 C$ m/ b6 E! a# q" _! g
socket_select($this->sockets, $write, $except, NULL);
3 C" e% ]* ^; Z% T- A
6 j$ {8 o v8 v2 r
foreach ($this->sockets as $socket) {
/ A2 B& G% |4 @; g/ N# J3 V
//连接主机的 client
5 A |8 R/ ^. E! j( G$ u
if ($socket == $this->master){/ N1 M7 R5 a: x& }
$client = socket_accept($this->master);: E Q3 M( T o2 E2 V& ^
if ($client < 0) {
4 @- m+ R/ h( [: E
// debug: @0 Q4 e. B/ {; U- y, Z4 w
echo "socket_accept() failed";& R" q$ K/ g/ e- O8 o0 H. ^6 F
continue;# N. U4 W! O. i2 d/ q8 w' ~
} else {" {3 X5 f# z- t; c" [
//connect($client);1 k% s, p8 Q3 u; b2 H) S
array_push($this->sockets, $client);
, ]5 P0 D7 W4 `: \# V2 m6 S
echo "connect client\n";
. ^, _: a! k3 d$ @7 g1 ~ \) S# t8 [
}
6 F. p3 R# \( @; i9 N
} else {5 |' L; Z0 \- z9 _
$bytes = @socket_recv($socket,$buffer,2048,0);" G" Q+ m3 a; m+ z$ c6 {
print_r($buffer);) V4 [' @6 M4 [
if($bytes == 0) return;
$ b' D$ i2 {. m2 J; e( _
if (!$this->handshake) {$ e* p7 I u) ^5 R: @, ~; B* B/ M
// 如果没有握手，先握手回应
3 A1 I# Z' \1 y
$this->doHandShake($socket, $buffer);: C: h+ t: w; C2 |' r: A1 ^
echo "shakeHands\n";
" z1 f# v8 b* [; U2 _0 r2 m
} else {
4 T0 e* v/ e/ v8 H7 M+ A' C
; T9 F- S$ q5 N8 ?+ K8 l0 y2 W
// 如果已经握手，直接接受数据，并处理
( `' l) ?0 e9 m. D5 D% F
$buffer = $this->decode($buffer);* \' N6 s/ [& w- V7 h% Z" R
//process($socket, $buffer);
. E; y' J- i T, C
echo "send file\n";
/ m7 X% b0 s) U& q
}8 p x6 Q" J1 r6 K+ q
}
- |9 W4 J- v* l# N
}# P8 H8 H3 h* h
}
# Z6 k! }+ \4 @) v
}
- R9 m$ l; I. B( |/ `* ]) h
# _- d0 P1 u7 [" o6 J: B
function dohandshake($socket, $req)( w/ c M, e6 j6 K4 x
{
6 A8 C x9 E* k, K
// 获取加密key
% P8 e3 o7 }5 w/ }$ y7 ?
$acceptKey = $this->encry($req);6 l+ d& B. I) d9 M9 X$ H) m
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" .. T- p* I4 r; M5 I4 i+ s8 }) p
"Upgrade: websocket\r\n" .
. I% T) u4 a! z6 a$ Y6 l
"Connection: Upgrade\r\n" .4 S3 `& _3 i S
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" .& q! ~% Q, |% b! F. s2 p7 F
"\r\n";8 ~- P& `4 g& c$ u; f( K
- F9 D* a' @5 m, u* c N
echo "dohandshake ".$upgrade.chr(0);
; T5 l0 H& G9 y: r8 E
// 写入socket
, j0 V. h" E5 Z; ^# [( c0 Z
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0)));
3 V$ U* D6 w/ J" A
// 标记握手已经成功，下次接受数据采用数据帧格式
, Q9 W* I% ]3 x8 l% J6 x( Q8 P
$this->handshake = true;
$ X! c+ F6 b E1 H; y: |4 P
}0 d `4 ?; B6 i0 R3 L
+ l S0 o$ V& [1 s6 J8 N# j$ |
$ _0 ^' J# x$ F1 O( {0 @
function encry($req)0 G) w/ z) R9 i* X3 K
{
+ d9 ?: w# r# A7 g
$key = $this->getKey($req);
3 j4 K% \! z( y4 b- `
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";/ P$ O- u1 Y* o
- R* U) i9 e2 F/ C4 a
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));
q: \' N8 a- x" k/ H" z8 @" r
}
" e7 D( _6 o# q ~
3 g$ @/ W2 G- X9 T9 ^1 `8 [
function getKey($req)
0 H$ E$ L! Q4 s5 [' ~* f V
{# t' x# R) I- |$ d7 |* K
$key = null;
4 m8 s9 `, Q( \! A6 u" A4 g4 r
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) {
6 N! F C& [0 B% m% |* c2 f1 }% t
$key = $match[1]; + V+ h: q" ^! K- M
}
$ x5 c* n2 h) d8 ^# v; R
return $key;
5 G" G- T; T3 `& l ^
}3 D+ Q0 \0 f0 o' g# w, N: d
N+ \2 t! y( C$ M: G1 F5 _% m+ K
// 解析数据帧
. T" R# m2 v& u+ C8 i5 `1 y
function decode($buffer) ! h. E! f' E0 J r/ ]$ V
{
$ h( M/ L8 o* [( q
$len = $masks = $data = $decoded = null;
6 p# Q; c2 {. b$ c; l+ c+ k! j$ B
$len = ord($buffer[1]) & 127;5 T! c; f# b4 E8 g& s' C- I
2 w( R- Z B' Q' s) G, F# F
if ($len === 126) {- G% Y! q) W! S g0 j& X& Z" z, x
$masks = substr($buffer, 4, 4);7 B7 L0 g- g# |9 I2 p1 q
$data = substr($buffer, 8);8 R$ k# U& x; A* }7 g8 @
} else if ($len === 127) {
# ~6 h- J" f/ v* i. {' a5 S
$masks = substr($buffer, 10, 4);) n* L# c N- k3 }
$data = substr($buffer, 14);
" C9 l/ G3 l0 Y1 `. g g
} else {
9 k# g/ Z+ @, e2 W% @
$masks = substr($buffer, 2, 4);
! [( `* k5 J, q) n% z1 ~# O
$data = substr($buffer, 6);
8 S5 g X7 q$ [% W$ e# i1 u
}* D2 V# c1 J3 S, a
for ($index = 0; $index < strlen($data); $index++) {
- u, h+ _9 T6 k! l, _+ o
$decoded .= $data[$index] ^ $masks[$index % 4];
# j0 z2 ~% m* a
}3 m) P1 V8 b! W0 W- U6 @
return $decoded;4 U3 M H; q! s3 `
}
/ x/ ?6 z3 n; O; ?/ y& b$ I s. m) \
9 ?7 k: k6 ], E9 Q: i) m
// 返回帧信息处理
5 W X! g, J8 V
function frame($s)
z3 Z; T. J5 `8 p5 l0 p/ S4 ^
{
! y4 k' Y( }. e3 N# i6 a8 Q( o
$a = str_split($s, 125);
. F9 _; e% o7 _4 ]; x0 a7 l* w
if (count($a) == 1) {" }4 \) k# K- F0 [
return "\x81" . chr(strlen($a[0])) . $a[0];6 n. f' A1 Y# h' u" d0 J
}
M4 i4 z* ?% E! A; l% u8 C8 G5 p
$ns = "";
2 l; L _* O2 F, s! m
foreach ($a as $o) {; L" M, N2 ^/ J: Y' V0 y
$ns .= "\x81" . chr(strlen($o)) . $o;. Z) w5 r4 o, z( [
}
# z( J/ j% s% j/ k! K
return $ns;- k+ M, @1 T. d2 B; ?7 Y2 `9 [2 Y5 ~- t
}! p V5 H$ v# z; z' B! O
: ^5 c8 v% E5 ~9 M* M* n' @; t
// 返回数据
1 U& D, i8 D* }; L5 ~
function send($client, $msg)
1 y8 u& h9 x- G1 \. }
{1 ]# N3 k7 p; W( g
$msg = $this->frame($msg);
. |) p& G; \: B7 @$ F9 d
socket_write($client, $msg, strlen($msg));. F' R% c7 Y1 ]5 _( c. T! A
}
5 N7 s0 y+ s' a2 O' ~' ~' [
}
9 W) @. U/ A3 y! c
% u9 A( f4 G \, i- ?: S
测试 $ws = new WS("127.0.0.1",2000);$ y1 P- m% I. T' @# U& [, F. ~/ L
) E7 j: {3 g) B# [6 G5 }5 r

复制代码

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

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

浏览过的版块