PHP 简单实现webSocket

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

1）客户端实现

<html>
6 p! y' T* t. U+ B
<head>! f; U. W6 V! m: o+ B* W
<meta charset="UTF-8">; o6 I, e' g' w
<title>Web sockets test</title>; t' _' N# O5 y: P
<script src="jquery-min.js" type="text/javascript"></script>
& r) G5 p# f; m9 @3 ]; {% E+ Z5 R
<script type="text/javascript">
" N. T1 C! _! h* |2 t2 f2 n
var ws;" Z) g/ w9 @+ i" ^
function ToggleConnectionClicked() {
$ _% @" l! j' j
try {: d7 o1 p5 X9 p1 J4 e
ws = new WebSocket("ws://127.0.0.1:2000");//连接服务器 - J8 ^; H1 o& q- }& e
ws.onopen = function(event){alert("已经与服务器建立了连接\r\n当前连接状态："+this.readyState);};3 H4 p$ c1 j7 T& A# e
ws.onmessage = function(event){alert("接收到服务器发送的数据：\r\n"+event.data);};
* A8 y1 i0 v, S; F3 i
ws.onclose = function(event){alert("已经与服务器断开连接\r\n当前连接状态："+this.readyState);}; N0 h. H* F3 ]$ m8 d( e
ws.onerror = function(event){alert("WebSocket异常！");};
; ~/ y4 y$ z7 g1 j
} catch (ex) {; O! Y$ X/ o5 ?
alert(ex.message); - Y% B; F7 {/ m0 S
}1 ~ a$ x9 [9 b9 M# ^+ Z7 R
};
/ y8 Z7 I7 e& x" W$ M& I
2 G, ?& D; R3 O6 h( i/ F$ P; {2 `+ ^8 V
function SendData() {" G( i. u, O: h' _1 W2 H$ k8 K
try{
* B) y; Y) W! b( d6 [
var content = document.getElementById("content").value;" C, u# `/ w! Q& a
if(content){
. I. s* g6 S3 @" b( L( B
ws.send(content);
2 S5 E) q9 V' N( j
}
* N* ]& c c& W' d2 M- P
; J; t7 @! S/ J; t. V+ N
}catch(ex){7 N) |, Q b( y3 }5 W0 d
alert(ex.message);
, D7 G. Y% ?, \
}2 @; t! r' w9 f, n6 f) n! ~
};
" T1 S$ U5 A% r6 Q7 v
/ n, v" [2 \& T) U2 f1 d( q
function seestate(){
1 }) r, n+ f: }' K# }5 B
alert(ws.readyState);/ A) ^2 ^- N# I8 r( e L: p4 i
}
; x& Z ^1 N: ~% |4 C7 ?" Z
% G7 X- n" h) v
</script>; N( o% X4 ]! a7 \, \4 e0 q
</head>
0 `* T0 ~) f% b5 N" c6 `
<body>+ g2 V3 @; U4 S1 S' Y
<button id='ToggleConnection' type="button" onclick='ToggleConnectionClicked();'>连接服务器</button><br /><br />) g0 }6 A% v7 _( V3 T0 x
<textarea id="content" ></textarea>
& Z4 ~7 x" t( M1 g$ }$ {
<button id='ToggleConnection' type="button" onclick='SendData();'>发送我的名字：beston</button><br /><br />$ ]5 G/ D B, ^2 V! c3 [# W9 G* s
<button id='ToggleConnection' type="button" onclick='seestate();'>查看状态</button><br /><br />- ? D+ q1 d) R+ ^7 ?6 t5 d4 U8 ^ a0 f
; |1 |2 J* d; F2 Y/ u6 u
</body>
) Q3 w) L$ l8 w, a8 K4 A# K
</html>6 R ~4 t0 J* V; b/ S1 I

复制代码

2）服务器端实现

class WS {
3 {: _4 _% U3 d8 p5 ^
var $master; // 连接 server 的 client$ O9 ]- m9 K- _' U$ ~" i
var $sockets = array(); // 不同状态的 socket 管理# r4 r6 l6 G# R8 d* u* l' f! `
var $handshake = false; // 判断是否握手+ A5 A' @1 R( m# ~& `4 v* G
; x3 N! I4 F9 x/ k6 T9 W1 c: r
function __construct($address, $port){# V+ ?8 d; ?4 d& e. v* R
// 建立一个 socket 套接字: y" [/ U: S9 A' y( }
$this->master = socket_create(AF_INET, SOCK_STREAM, SOL_TCP) / S U1 T2 `" W
or die("socket_create() failed");
% a7 w: |# J% u" } M+ {
socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1) , A3 g) E& C+ p, ]0 g
or die("socket_option() failed");% W; j7 V# b! S3 h$ ^& V4 l
socket_bind($this->master, $address, $port)
' z1 p& _. I1 c& T5 I/ I% T5 c1 |
or die("socket_bind() failed");$ V% M" M# o& t
socket_listen($this->master, 2) 1 R( _7 H& q0 f3 Q L
or die("socket_listen() failed");
' u3 w6 }5 Q4 S$ ?: L2 |
. z$ i8 y& X3 R5 M" s) u$ X7 W
$this->sockets[] = $this->master;
8 U1 S2 n. U3 t! r( L
* ^" ?2 Q7 a/ m2 B+ O n: k, L
// debug3 {* g6 o9 T6 ^0 g7 r7 y# a
echo("Master socket : ".$this->master."\n");
2 l6 _! ~* f5 G3 L
+ `- w M, O0 _
while(true) {
! P2 v6 k- v) c/ \
//自动选择来消息的 socket 如果是握手自动选择主机
6 }. ^# ?% X4 S4 I: |- k
$write = NULL;
- S/ C1 D J+ [. a
$except = NULL;- E2 Q/ f c3 O- G" O
socket_select($this->sockets, $write, $except, NULL);
5 `; C5 B& |' x) r$ J
* @1 P1 @7 q1 s! y8 k4 Z& v7 q1 J
foreach ($this->sockets as $socket) {
2 \) q& c) R* c9 T9 F
//连接主机的 client
$ `0 a. T* d% B$ K( Z/ ~
if ($socket == $this->master){
# e2 L5 F$ Z- S$ x% M& R: P
$client = socket_accept($this->master);
- u3 X# d/ m4 F, b0 g k% Z- l( D
if ($client < 0) {8 p r$ r% Q' c% {5 i, g
// debug
$ B! [6 x" X5 F5 h& t# n
echo "socket_accept() failed";& S* F- m- }7 x8 K
continue;
, `6 a3 E6 Z9 C! _/ j$ t
} else {0 S7 n; D$ s' V$ C
//connect($client);
# h2 P2 F; l& M8 E C: C+ G
array_push($this->sockets, $client);
' `3 e, g7 k4 z' _) y- a
echo "connect client\n";1 U$ A" F& Q: x- @ c! O
}' N, e( @, [8 M9 ~4 r
} else {! ~2 G! \- X2 @$ C
$bytes = @socket_recv($socket,$buffer,2048,0);. R- p2 t, e/ F! Z
print_r($buffer); ^( ^1 Y% o& b# e# b
if($bytes == 0) return;
% y: |- B( d4 U: N! {; M: x
if (!$this->handshake) {' O. h& I5 M6 b1 g; q; c( X0 C( m7 A
// 如果没有握手，先握手回应5 ~& x( i+ V: P/ h! X% j
$this->doHandShake($socket, $buffer);
3 V* q1 _3 W" O. | N& e) i) ]
echo "shakeHands\n"; a0 B8 n/ G0 C0 o; [" n. b/ m
} else {
) k7 @8 K# d& N# H6 G. e1 P T: o
- g I6 }, [1 u* c( O9 q0 c
// 如果已经握手，直接接受数据，并处理$ j) x0 P" M, L2 P8 ?# r4 B# I
$buffer = $this->decode($buffer);
. u8 q) C( g- s
//process($socket, $buffer); ]' c% t/ N0 S* x1 k9 m
echo "send file\n";
& ~9 M f, v# f8 W& }+ ?
}2 S( t1 D) K& {
}) I" {$ E. ]. M, }. N1 }8 l
}
6 Z/ G; f) B, n1 u; q. }5 ?% E
}) ]0 k0 {- H3 S' s
}1 {, G6 U. D: M, Q
3 e7 x% h+ Q$ Z7 }
function dohandshake($socket, $req)* E% Q) T _; F6 J1 b8 P
{
/ k G/ u6 Y, k
// 获取加密key
- y3 H( _0 I4 n- O
$acceptKey = $this->encry($req);- {) P: H) b0 d' F- g
$upgrade = "HTTP/1.1 101 Switching Protocols\r\n" .' g4 q, ~) h9 U- V( j d8 m
"Upgrade: websocket\r\n" .
( W' a1 e4 b e/ l/ y
"Connection: Upgrade\r\n" .
6 q0 r9 }9 }$ |! v/ `0 ?1 s3 Z
"Sec-WebSocket-Accept: " . $acceptKey . "\r\n" .
# r/ y: l2 i/ U3 N
"\r\n";1 x e- c0 H* A: B
/ H: x: D# A* N" m
echo "dohandshake ".$upgrade.chr(0);
! R# t$ O- m) `* W
// 写入socket
1 Y% p; b2 T( o
socket_write($socket,$upgrade.chr(0), strlen($upgrade.chr(0)));
3 E& w7 q2 s7 F L% z& [$ _
// 标记握手已经成功，下次接受数据采用数据帧格式. H m# p# B/ _; R+ }% x6 a, `
$this->handshake = true;# M# b* l; A* z$ }1 J& }6 C
}3 c: \7 i. K( g2 t* _( m# |: a; C8 |
/ i5 p6 F2 m5 c _2 C' N
$ ^6 \2 H |6 e* l( c
function encry($req)
- O9 [. P7 S3 Q5 A6 }' L m' Z
{
4 M' z# K2 E3 L+ V
$key = $this->getKey($req);. R0 A; s0 {3 b6 M) Z0 d5 ^
$mask = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";
* N8 ^' C9 r2 C9 ?" X0 E
( A% ~* p: @8 N* s' Y+ J- W
return base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));
# D& _- F+ \0 p) d1 T* K" }
}* t* m6 U4 z8 e+ g9 ~
+ r- v: x% v# O* k+ N
function getKey($req)
4 n/ k3 B) t/ e9 e+ {
{' b6 \, w, D. @ {1 f8 I5 B8 m
$key = null;/ r1 P& f. w; j* A4 V) B H
if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) {
( x8 P+ K6 ^4 s, }% R1 B6 P5 p# t
$key = $match[1]; ' ?' I6 R8 s( z1 S7 L/ ?5 S/ l
}
6 m* t0 s) @) Q8 D: d. P
return $key;
( q$ A! A4 _, i# f# s- R
}* w( u' N. _0 F# \ {, m" Q
1 {( f h* g2 x" y& j4 ?
// 解析数据帧
* l8 s& b- r3 S j
function decode($buffer) ! ?8 U0 a. M- z5 K( p: B
{0 i2 ?" `# W( O4 p# f
$len = $masks = $data = $decoded = null;6 j& D& E4 v# P' l6 _$ z9 b% |) f
$len = ord($buffer[1]) & 127;9 w+ B3 G! l# ]% T
6 E u! t5 U2 }! c6 s/ i
if ($len === 126) {
; w8 `& C9 G3 F( [
$masks = substr($buffer, 4, 4);+ o7 R4 ]* y/ Y! D3 G0 R
$data = substr($buffer, 8);7 ~) X1 V0 t g* n* a5 R6 Z8 ^
} else if ($len === 127) {
& l/ ^0 |6 A1 I& q1 s6 W
$masks = substr($buffer, 10, 4);( k* s9 S: O4 B, z
$data = substr($buffer, 14);
, Y0 `6 f, u& h; _/ a( f% ]( e
} else {
/ D4 _$ u3 Q6 T2 [
$masks = substr($buffer, 2, 4);8 r; Y. K' C3 e: ]3 ^ l, Z& ^
$data = substr($buffer, 6);
& f% i$ j y% Z+ t9 E: @" U' F
}
6 ~5 \. L- ~1 P, C$ a# p
for ($index = 0; $index < strlen($data); $index++) {( [0 |+ X$ q) k; V0 ]' U: u
$decoded .= $data[$index] ^ $masks[$index % 4];
# n8 i0 c7 K8 n4 {' R) p) }/ o3 Y* |
}
/ J$ |+ @- k5 Q( Z; b
return $decoded;( }& D; v9 D8 v6 \# V
}
3 |4 P# @5 `/ s, }9 H( ]) t
' G3 X( X' u/ {) t
// 返回帧信息处理3 ^. C/ M( x W0 v
function frame($s)
+ k0 s# [5 o, w# U9 n* p
{9 F. ~$ c) O. s4 G
$a = str_split($s, 125);
; a3 @; q9 l6 B% U0 p% t; d. j8 p; C
if (count($a) == 1) {
* I. ^4 O" P H4 w8 @" c5 T& ?, L
return "\x81" . chr(strlen($a[0])) . $a[0];
* R+ Z% [+ c4 e6 f
}% G1 P8 I7 g% ^% z! }
$ns = "";+ S6 r9 y, h+ l+ u
foreach ($a as $o) {
0 p8 @3 K$ t1 L/ Y7 H
$ns .= "\x81" . chr(strlen($o)) . $o;
8 w3 j: X' R. R. U, m% o8 M
}
# R: {* D) c5 o+ [) u+ V8 c
return $ns;# D3 l; R% f$ b; F
}# e, N8 I( V" |, K8 P9 L
" U% k: q& b- M0 j
// 返回数据- ^/ z+ B/ }% X" O
function send($client, $msg)" { |9 Z F& e
{
! l/ C3 B0 x- a+ ]# X
$msg = $this->frame($msg);6 M) o, B: G( x- r
socket_write($client, $msg, strlen($msg));9 T! V; ?$ F! X) W% ~9 S9 {2 T* _! J
}
6 m! `6 i, Z- v) v+ C% @% P
}
+ \9 A( {4 D) |0 u7 I
; U+ i6 |$ d! d( g
测试 $ws = new WS("127.0.0.1",2000);
& [; l" B6 n; N& ?: A

复制代码

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

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

浏览过的版块