PHP 简单实现webSocket

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

1）客户端实现

<html>
/ G/ q! U( y" G7 j9 G. u
<head>. f# G0 x, v5 E" M+ g2 h+ q
<meta charset="UTF-8">
( g2 u- y$ t7 J6 f! q4 O
<title>Web sockets test</title>
, Y* S& q* W% R' ` ]
<script src="jquery-min.js" type="text/javascript"></script>
" Q6 E" S# R. f* F5 F8 p. G
<script type="text/javascript">
; g& S: u9 X h9 e3 V1 j
var ws;
8 C" A1 _( v, c( Y
function ToggleConnectionClicked() {
& O/ b5 ^( y7 Z5 v7 R3 a
try {% i4 {. G6 K7 d) E3 j* Z" f C o+ k. e
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器
+ Z: p4 N8 T" q! A" N3 J) @5 _. j8 F
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};
- p G7 z: J0 F
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);};/ G5 ?2 g; o2 A5 E$ y' \$ Y
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);};8 I& f" S' O& x1 O, ?: n
ws.onerror = function(event){alert("WebSocket异常！");};4 N0 v; q7 D/ m0 `! {3 k1 c0 c' b0 a
} catch (ex) {
- |2 c0 m1 G2 G! v: G
alert(ex.message); + ^2 p/ }9 n; l/ J
}
& U- I% t9 K/ R
};9 D; ^6 R9 A0 K2 w; P" _$ c: _8 U
9 P3 y1 ~3 l1 p3 N9 z5 r5 m
function SendData() {
6 B0 G/ k \# z5 l
try{& I" |8 \4 w ~8 I( u. a
var content = document.getElementById("content").value;: F' j3 |" s" H; P* W
if(content){
7 @. ^3 |% E) H0 F8 P0 Q
ws.send(content);
2 S! S5 D$ _3 I$ @. w' [
}
. J6 A5 a/ W1 q5 K0 @5 O* n$ \
# H7 ~0 g! p# `' ~6 h
}catch(ex){/ h9 |0 h# p" [7 r2 |
alert(ex.message);
j. ?; H/ ^ J
}3 @: `- l% v0 M( X- z8 l" B% @
};
* \. `( S( V. k* Y3 d0 i$ _6 D! @. {
9 W v0 U6 z6 R9 X( {
function seestate(){8 B% J- p, F1 ]) ?6 \6 s
alert(ws.readyState);
; N/ A& U2 {9 `2 M8 _
}- R \( r: j- D5 Q j3 P, c
0 ~3 A+ c3 t9 P7 @* l0 I
</script>+ P2 z3 v9 ^# j6 s$ p5 c! ?- e' i: o
</head>- M% W( K) h/ D+ u1 J$ S5 Y
<body>
+ R0 q9 b2 N8 c7 W
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br />7 q8 n/ w% p" G, c5 z3 F4 k
<textarea id="content" ></textarea>; i. x1 z$ Q$ F& }1 [
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br />* W8 m; W1 w+ l+ Y. \ r) k
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br />
/ l, ~) c& u" T' i; s. y) q
% C% U' C( h. ?' V) ?8 W9 W' F
</body>% N' n ]" L5 w: l( ~
</html>0 r1 Y9 V) [) J

复制代码

2）服务器端实现

class WS {6 m! z# @0 r8 x3 a, q E
var $master; // 连接 server 的 client
* P- e+ k" V2 c' K4 ~1 G
var $sockets = array(); // 不同状态的 socket 管理3 g0 b# Y. l) ?9 t! |
var $handshake = false; // 判断是否握手
3 o% t* w+ D# _/ f: r2 x) i; Y
function __construct($address, $port){
) F! Z5 V6 h, |7 p: S9 L0 S! d
// 建立一个 socket 套接字/ X4 E% c# l/ v8 J# T! S7 w
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP) $ g/ ^& A0 s3 i! s
or die("socket_create() failed");
# k* K G" d% G5 `4 g7 P8 E$ K4 C8 `
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1) : ]1 I; D# A" p) K
or die("socket_option() failed");
$ { \& E- V+ F" E
socket_bind($this->master, $address, $port) 6 U E0 z5 Q6 s+ X/ U
or die("socket_bind() failed");- w. H" ~ L: d8 ?6 n2 C
socket_listen($this->master, 2)
: ^5 w+ o. ?" a4 z1 {. x; ?
or die("socket_listen() failed");6 H; h, c! @% m3 h3 J c
9 S5 u- Z+ S- B) `% k r
$this->sockets[] = $this->master;8 A9 ~0 [5 U% Q3 Z# t+ ]9 O
% I u" ^6 `7 b/ {
// debug
" Z5 @$ d' R2 ?/ T$ L7 @. @
echo("Master socket : ".$this->master."\n");! B$ s6 g! N9 k: x* d
4 ?. H1 ?9 k* c+ M {- a7 E" F5 m
while(true) {( c( H% s( I' F M( S" H! Z( A4 ]: W
//自动选择来消息的 socket 如果是握手自动选择主机
) z7 E6 A/ B, g6 c+ a
$write = NULL;
9 M' @7 C. }! g3 ?
$except = NULL;* I3 B" U4 X9 J/ k% X% {& K
socket_select($this->sockets, $write, $except, NULL);8 e* w0 @$ q* S
; z! }6 |6 g' d2 v* c, P# z
foreach ($this->sockets as $socket) {
1 X9 g4 a2 V% p% Y) e2 e
//连接主机的 client
, u7 ^" x4 C" L6 D$ ]" u
if ($socket == $this->master){& U/ h7 ?3 c8 r# S3 z/ E7 P( k3 e+ u
$client = socket_accept($this->master);
4 j, _9 i% \' P; w: H+ [6 S8 i
if ($client < 0) {
! @! K5 M D( T& M) V) \* m0 S4 A
// debug5 @% O4 M8 ?& t% C
echo "socket_accept() failed";( H1 o3 p3 H: |
continue;0 G) v# Z. {4 l. f) T* S
} else {
; @3 m" v$ S9 Z3 S3 E
//connect($client);& T; x; r& N7 c3 g9 P
array_push($this->sockets, $client);" [4 @* D! I. J0 q. c) P# s
echo "connect client\n";
! L- j Z9 W1 r+ ^/ U! Q; J
}* L5 ]/ v% H; [, M
} else {2 Q s" w9 [6 ~7 x* _, B- r
$bytes = @socket_recv($socket,$buffer,2048,0);- u! p0 R% h" a) e4 T; f
print_r($buffer);
: M3 G- w' {+ s( J# F
if($bytes == 0) return;
7 I s2 ]3 e. b, }. s# J5 `% |
if (!$this->handshake) {- d k; b6 m/ `* Q
// 如果没有握手，先握手回应% L8 a R% o" W8 E# ]" A
$this->doHandShake($socket, $buffer);
, q# P& t& E/ c" h$ ~9 c
echo "shakeHands\n";
$ F- f9 m4 P9 S5 {; h* Z" q
} else {) [! E! X) w6 V) f i: q
) b4 D7 H& S+ K+ w3 K, W' Y# d* t( Q
// 如果已经握手，直接接受数据，并处理8 A0 ~) l' G, M' N8 O) j2 m
$buffer = $this->decode($buffer);) M" u `0 ^9 h' G- c0 m
//process($socket, $buffer); ( z9 P3 d7 c6 Y: @4 z- {
echo "send file\n";. E5 ^! T) I( s1 R4 e
}
0 X2 d1 {% E8 Q6 k& t
}
9 A& O" o) K% D1 E
}9 b, r7 @# V* j) K/ S
}
% x8 e( ?: |5 O# F6 G# a$ W' n6 t( l
}
9 q; {( b9 P% V; @' J2 h0 w
function dohandshake($socket, $req)
! ~" ~) B4 T9 c4 {, o W0 R/ V
{
* F5 s' H5 f' M V% B, u0 x
// 获取加密key9 T# b) J7 d9 N F; t6 W6 k) M
$acceptKey = $this->encry($req);, ?# @) N2 [3 V$ V1 [+ m) _4 ]
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" ." l4 x7 F4 A, F5 O( h0 ^
"Upgrade: websocket\r\n" .
( l$ K' z$ P! y' \0 C0 K1 T3 e
"Connection: Upgrade\r\n" .. K8 t w. n+ S! R A1 R
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" .5 F4 A0 O0 q; G$ n' c9 X# N9 }
"\r\n";# p a$ h5 i! |+ t# N8 |
" l" E# S# Z3 u7 `% i7 I0 G9 G5 O
echo "dohandshake ".$upgrade.chr(0);
* u' I1 S/ I+ i
// 写入socket3 q3 O- R, h4 s7 I' V& o, u0 y9 r7 y' O
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0)));3 t, v5 x- t4 p
// 标记握手已经成功，下次接受数据采用数据帧格式$ @$ b9 U$ D( E$ }9 |
$this->handshake = true;
& f. ~; J. s% l' b- ~
}/ q& X- a! P2 ?7 s+ k$ r
0 q# e) M/ q% b+ s- z
4 U* W" J2 Z( i3 t. J
function encry($req)
5 y: j, v" g3 O) j Z
{. N5 ~5 x7 L( Q2 O) _$ c D/ T; `' ?
$key = $this->getKey($req);
( A) q8 G6 u% r- B
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";
8 `& F/ v4 T' N& v, J/ F- o
, {; i9 P, ^% v. C
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));: B# w. E$ X/ L. a4 \
}
, T7 C- _1 s2 T+ V/ i. \
! n1 z9 s- d1 W7 w" K7 y1 _
function getKey($req)
" R P: O1 H4 j7 C7 F. o
{
) c' H6 o4 [5 A2 h1 v" z
$key = null;
' \' i7 l' u1 N: g3 X
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) {
. c; N# g6 n/ Y4 l
$key = $match[1];
% @5 h) e, y$ q( V
}
: L4 ~: L6 d' m' q/ s# W8 W8 }
return $key;& O; q7 O, C" c. Z) h& `& u* G4 i
}
3 T: g1 B3 o! y! G7 G
8 M2 q" d# X W- w. `( P' m
// 解析数据帧; ^! q. c9 I) h
function decode($buffer) - L' @7 |! H* h% V" O/ K- J, S& q; n
{
# A' P9 m$ Z3 E$ t& S
$len = $masks = $data = $decoded = null;/ _4 E( w: H. `
$len = ord($buffer[1]) & 127;
* `7 J4 d! d1 q: U
5 Q! N3 u* g/ L
if ($len === 126) {
; M; R6 g7 h; W- K1 _
$masks = substr($buffer, 4, 4);/ a' W$ a$ |- k
$data = substr($buffer, 8);/ R3 s5 p7 M( O! o
} else if ($len === 127) {
2 s& Q: h x/ e9 y
$masks = substr($buffer, 10, 4);
# J8 J" p" k$ f" }3 [ J7 d
$data = substr($buffer, 14);
7 C2 S3 H1 D# V6 [3 ^4 x9 _
} else {
" C8 S+ S8 r7 c, j% W
$masks = substr($buffer, 2, 4);. V+ y1 X, M" J1 T
$data = substr($buffer, 6);. i/ M; U% }, o) f6 |
}
# U, ~" L) ]/ L/ [% H' W$ W) J
for ($index = 0; $index < strlen($data); $index++) {% a$ I) z* A6 v+ t8 m9 Y; X0 R
$decoded .= $data[$index] ^ $masks[$index % 4];
2 _) O! Q) L$ _# G
}
" i- w* B1 o; ~' k# t
return $decoded;5 ?- `: {; n" W
}
/ M+ U3 R5 t8 J R9 L- g- A
6 n' X6 G. q4 j) O% i
// 返回帧信息处理
, k o5 W, c( N7 ]2 o7 L
function frame($s) 2 n) }3 _4 K$ D) f' B h
{. a& _+ G: e1 J8 f# B
$a = str_split($s, 125);
9 ]' T% {7 ^4 p6 {
if (count($a) == 1) {/ B% j. [/ q% v1 S3 W
return "\x81" . chr(strlen($a[0])) . $a[0];
, I$ u: @9 {( x" ?1 \) @) A8 G
}
& I* t: f% R+ E, r4 L9 B4 s
$ns = "";
5 f& O4 N }9 O
foreach ($a as $o) {1 v* P2 y9 v4 K' m0 d2 q$ }
$ns .= "\x81" . chr(strlen($o)) . $o;% x) _6 _ M$ J3 X
} t" W; B; q) G
return $ns;) i. [. \* I. k/ }# h' j; _) \/ \
}2 Q# u% _2 h( J, [5 a
, Y0 _4 }7 l2 B( y, E
// 返回数据; D( C$ l: ^0 X
function send($client, $msg)( ?- ?+ d' D9 p( \: X6 z9 q
{
8 ]$ E- I% w5 u: C9 m' ~! ~
$msg = $this->frame($msg);$ ~) k+ b- Z4 P, G# s! J( n1 q# @
socket_write($client, $msg, strlen($msg));
7 ^. b, l' O- M' L9 n9 R% b' W
}& \# m( V: ^5 q1 P3 i* C- A
}
; g. M4 @4 u$ K3 Y5 d) s
6 U" X$ g! F7 k- c
测试 $ws = new WS("127.0.0.1",2000);4 T1 j0 C1 [. n" D& @
: Y. q! x! Y% ?/ z# O( l

复制代码

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

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

浏览过的版块