PHP 简单实现webSocket

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

1）客户端实现

<html>! M1 r" ?7 ^5 Z4 n. l
<head>
2 q% C/ A% p# A' |% ^3 O
<meta charset="UTF-8">! `0 j+ N' f- G" S& n3 S
<title>Web sockets test</title>
% h6 [" F! r- Z `$ ~8 I: Z, r
<script src="jquery-min.js" type="text/javascript"></script>
J1 [6 A+ c0 K8 Q$ g% c* J
<script type="text/javascript">* g& B+ ~2 I% ]7 _5 o
var ws;
; a/ X* T0 a2 n' u; G }! A
function ToggleConnectionClicked() { ?7 k E% F$ P' x% [
try {
: C" u9 B+ V$ q5 \$ N
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器 0 h0 o5 [ g9 t! }% I
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};" _, h8 _9 Y4 [
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);};$ }8 ^5 d3 e5 M' T9 p
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);};
" n* w6 R2 {( `+ f. g
ws.onerror = function(event){alert("WebSocket异常！");};# t" m0 k! k6 @7 L& X: f7 f7 l
} catch (ex) {1 A3 h- v! c0 ?% a: a3 L) I( \
alert(ex.message);
+ j* ? i; \5 P, w% F$ ?
}" @; h6 b& b" a l& Z o! O" ?
};- b8 q' o6 X( Y1 z9 V6 w1 }
. T: @! x. Y& k
function SendData() {
5 G& g( s O6 z5 x3 G
try{' s8 N$ A4 a7 k/ P- n& w" [
var content = document.getElementById("content").value;
) l1 h6 B/ X% R
if(content){! X, n1 R8 V9 V; N8 N
ws.send(content);
. j' x* c1 F3 B2 O% q: c, m
}* j/ M; F. Z4 t2 F" T
+ o; Z# F, F- G# J. m
}catch(ex){" V7 \' d5 m* j: J
alert(ex.message);' L, ?5 ]& a, F+ y. D# \
}9 j7 e) x* C& A* B8 t
};. |0 H4 }! t2 S" Z0 _" W5 `
1 T8 r0 w3 t' ?) X. h* Y
function seestate(){& a- b* O# x' b2 C* w
alert(ws.readyState);6 v! d. V5 \, ^# b1 t7 w
}
* f. S3 e$ [7 R$ W2 _
+ v& A% Z. a1 ^& R; C: V" w2 {6 x1 b: V
</script>8 M G! N& z0 b% I" U. l7 K
</head>
) z4 R @1 [' B( Z
<body>0 p# j q) C7 m
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br />+ ^+ {7 _% p; I6 C! [" j3 m
<textarea id="content" ></textarea>. u# t+ K1 U8 ]% G9 D g2 O
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br />
7 ~9 p7 s, x' ^# l
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br /> x. A/ ]2 Q* Z. P. ]7 j" I
* N6 {( |, P9 R1 F% Z
</body>0 c7 j9 h. H% ] e) T+ ]
</html>
. H* W3 V' A! }

复制代码

2）服务器端实现

class WS {4 q, v" l- F5 n
var $master; // 连接 server 的 client p! d' @6 N+ `
var $sockets = array(); // 不同状态的 socket 管理5 N5 N: z( g: B8 Z* ?* `
var $handshake = false; // 判断是否握手1 i5 V9 S" j. p! w/ `: E
" }, P) G1 [0 S. B& y
function __construct($address, $port){
* p- b7 m! R; n' D- ^5 Q
// 建立一个 socket 套接字+ q8 R+ T2 i! F) a2 X; Y" F
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP)
: l5 o, t/ \$ I
or die("socket_create() failed");
7 |! A9 h2 v0 p1 R
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1) & A3 a ?, s/ n }9 L, \* v
or die("socket_option() failed");7 _0 _/ f0 P/ P* ]. G% l0 {2 S( t
socket_bind($this->master, $address, $port)
' T( ?( u8 N) `1 W' `
or die("socket_bind() failed");& P6 @; Y- i& A. K% ?
socket_listen($this->master, 2)
) K: M2 {- w/ j& ~' l' n
or die("socket_listen() failed");7 @% B8 `" i6 D! k$ k3 @
% O0 _+ B/ \9 y9 A% k+ {8 o
$this->sockets[] = $this->master;, y; I* L% L" _; q z" Y' F
+ Q E& Y+ S; x) P, n: b5 ~7 |
// debug
4 S4 Z9 x7 @$ {; b' v; Y; Y
echo("Master socket : ".$this->master."\n");0 c3 r$ \; d4 v1 d
) A5 U; d! M6 z7 ?
while(true) {
4 g4 v3 l5 B2 e0 S, }$ d% O3 j0 j
//自动选择来消息的 socket 如果是握手自动选择主机
- P5 o9 _- c8 ?/ o0 f8 X
$write = NULL;
% b5 `- m" ^6 m# k3 G$ U$ W
$except = NULL;
* B' Y4 t J; I! y. v) t/ ?, r U
socket_select($this->sockets, $write, $except, NULL);
3 c' h5 s$ B+ h7 q' s
1 N8 R$ E" f6 [8 e- p
foreach ($this->sockets as $socket) {
6 }7 R7 D* i, y
//连接主机的 client : Q+ v; z$ }5 @4 r( p0 M9 y# m
if ($socket == $this->master){( x1 J1 t4 k, _/ X6 c
$client = socket_accept($this->master);
: }7 R1 K* w* w% u L
if ($client < 0) {& y+ ?+ Q9 \0 i4 h. R1 g9 [
// debug/ O$ W) C, e0 E6 `( G9 z
echo "socket_accept() failed";
) h; \4 G* W% Q% R
continue;4 @* A8 G Z6 j7 |
} else {2 e/ B& M$ m+ x
//connect($client);# {- d) x6 U$ g4 o( G# l
array_push($this->sockets, $client);+ E( b/ A/ [7 D
echo "connect client\n";
, d; e$ Y. s$ u) ]! l* {
}5 b1 j4 [! N0 W
} else {" c* {6 t/ h2 `
$bytes = @socket_recv($socket,$buffer,2048,0);
9 ?5 U- `7 o* t8 f" Y6 J$ p4 [8 [
print_r($buffer);9 H7 Y6 T: W; C2 b. g5 X7 e6 S
if($bytes == 0) return;0 J; x/ S- k6 l$ N
if (!$this->handshake) {" Q+ a8 S" V% z# @0 O& l
// 如果没有握手，先握手回应
2 L# f+ U( k; F2 x* R+ ~
$this->doHandShake($socket, $buffer);
% v, Z1 m: x6 e/ l0 ^0 W% k3 s
echo "shakeHands\n";" {' z- v$ |. H0 g
} else {6 x7 H3 L. z- q3 _0 x6 i
0 C5 h, M* Q& [
// 如果已经握手，直接接受数据，并处理" D% k6 V& P0 ]& Y
$buffer = $this->decode($buffer);
. A# K) r+ T- U: Y! h$ ~
//process($socket, $buffer); 2 L4 v5 a& K4 m: V: _
echo "send file\n";) p+ L, Q2 k, N$ P! u* y; x& e+ R
}
6 p9 i: E& r6 a% {# K. m
}) X4 |' k7 D2 K& A0 i6 r
}" I$ Q- n- E! A. I% w8 [/ J! F
}
0 W n" Y* l8 E9 J# i
}
# _; d- c0 ~& F
! y1 u, }+ g, \- T
function dohandshake($socket, $req)% S' p9 R' E- j' O
{
" s9 ~; c) A8 J, p
// 获取加密key
- K2 M: @9 ]" {: W/ a
$acceptKey = $this->encry($req);/ i) {$ i0 \) d+ p, R& t
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" .7 S1 j, M2 e- M! @
"Upgrade: websocket\r\n" .! [2 r! E `. Y1 T
"Connection: Upgrade\r\n" ./ x2 {5 Q* M$ G2 H6 Y
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" .
4 e' ?) s+ T% m( I; K
"\r\n";
2 ?) w* B' W" [% U" q
( L# I9 H( l7 @6 Z/ V/ ?
echo "dohandshake ".$upgrade.chr(0); 6 T! s* ~3 k) h% Z! o+ U. M* o+ A1 I
// 写入socket8 q: `6 g. i. T% D- p p
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0)));) N3 ^- {% h. l; Z. c" @
// 标记握手已经成功，下次接受数据采用数据帧格式2 c+ X; n+ h8 M4 c. i3 P6 `$ E
$this->handshake = true;
; [5 r4 @, M3 Z4 F' J& K v) ~8 S
}
- E: h0 F' r2 \6 _
& w8 Q: }3 I3 P
' H8 t3 d* Y# {
function encry($req)) i$ e3 _# L# n8 W7 u
{6 b2 B! w5 Q( `# t& [* ^0 M/ e/ E9 l
$key = $this->getKey($req);
' ?3 X* }- J; L9 r5 \4 G. d; J# ~
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";& G4 L+ q( S! i+ B& q" m
" B" q, o6 V3 \* s/ ^/ O% y
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));
3 `% {" Q$ g, K) n% ?2 p5 x( l3 D
}
|( k0 c6 B+ J& E; H$ d$ C( {
9 v' r# z% l$ v% W1 B9 `
function getKey($req) 1 o. s7 d* ^3 ^0 P! g; \
{, Z! n8 V% |$ l {" h
$key = null;
. J, c+ B6 G6 q
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) { o: |: W* S5 X$ P6 Y F) p
$key = $match[1];
/ z5 r+ X4 j' E+ c
} m5 R9 h9 F: `# K& [$ [
return $key;
; o# c5 K5 a, q" G) X4 B" f
}
5 N' P* k+ c1 \( B
8 P7 @& p/ m; G r6 F
// 解析数据帧
{3 D& s; R6 E& k0 q
function decode($buffer)
6 V' m3 m$ [. P" Q6 E8 n& X" l
{. {) Q3 ^9 U4 \9 ^! |( J
$len = $masks = $data = $decoded = null;
9 O0 K! h# W0 n0 K2 L1 a; D8 I
$len = ord($buffer[1]) & 127;
4 a5 B) u' h( k
) S5 N# m- f: }- G2 A, h
if ($len === 126) {2 V" b) Y1 j. M+ L, Y: y
$masks = substr($buffer, 4, 4);
" r2 D6 ?2 `2 v; V" R! x
$data = substr($buffer, 8);6 b# c' k# Y, u; H6 k( r4 Q9 t6 ^& }
} else if ($len === 127) {3 D" R7 X) @6 n/ H2 E( X
$masks = substr($buffer, 10, 4);, K9 O( R! H9 W$ [2 Y" }
$data = substr($buffer, 14);9 W& Z( H5 m- D8 m( G! `
} else {
% o9 L8 S: F$ ]1 l; D; D
$masks = substr($buffer, 2, 4);$ \ n/ q2 ]1 x: F1 }# o- M0 H9 o# c
$data = substr($buffer, 6);
' T7 B# _0 P* R: D, L
}3 ]" b7 ~' n3 [8 j( c9 {
for ($index = 0; $index < strlen($data); $index++) {2 h# B# H! O- s ? {# r+ h: x
$decoded .= $data[$index] ^ $masks[$index % 4];( I" f9 \# {$ o
}% Q" }# |. y& W5 D: ^" V! M
return $decoded;" {) n p7 u4 ]% m) x) u# }. ?
}
. L6 i$ ?4 W3 x8 [0 q8 j
$ K; \9 W% k6 ~
// 返回帧信息处理
" M {: d$ I$ M7 f F: U* a5 D$ T
function frame($s) ; S2 Z9 L7 A h9 |
{# v$ o6 \$ E+ J; F6 j
$a = str_split($s, 125);: d. o+ ]1 L1 j' f+ I. U9 u% ?- i
if (count($a) == 1) {
2 b6 W" R+ E: t! m) J! b
return "\x81" . chr(strlen($a[0])) . $a[0];
d: B6 ^: K9 y% T2 u2 k) Z
}/ @+ w6 ^! x% ?! y- Q* ?( B x) t
$ns = "";8 Z4 I' B9 o6 O [
foreach ($a as $o) {* ?7 O {6 v. P: M1 t C" z
$ns .= "\x81" . chr(strlen($o)) . $o;& }3 L0 m7 a. i" L
}+ Q8 b* q( w. K A. [3 T, [/ |
return $ns;
" }. z1 H0 g. ~/ f3 Y
}
2 \" G6 D$ M% m) v
4 O8 f, _! }* m* p, k9 p4 W! ]
// 返回数据$ E+ }+ F: O9 `5 k2 {+ h9 V3 X+ D# U
function send($client, $msg)
* E% }0 e6 W4 [1 C
{1 E4 D1 |; M5 l- z2 P7 a
$msg = $this->frame($msg);
9 k1 l, X4 q5 ?$ @! x" ^
socket_write($client, $msg, strlen($msg));2 k1 x( [8 n- W2 d
}
l) @1 j6 f' |8 e8 w6 Z
}
* M3 A4 f6 H! P9 B3 k6 H
; H( ~( @$ Q1 \: C$ p4 B. G1 I
测试 $ws = new WS("127.0.0.1",2000);2 w# O3 U, p/ S! m
o& H, D9 d) X7 G& ?4 O# j4 F

复制代码

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

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