Websocket和PHP Socket编程

本来是搜一些html5 websocket资料看的，结果被引去看了php的socket编程。下面是一些简单的例子，在命令行运行php脚本就行

[命令行运行PHP]PHP中有一个php.exe文件，可以用命令执行PHP脚本。如：D:/php.exe -f F:/test.php ; 可以使用php.exe -h查看更多参数 ：

server端：

1. <?php
2. /**
3. * 服务器端代码
4. *
5. */
6. //确保在连接客户端时不会超时
7. set_time_limit(0);
8. //设置IP和端口号
9. $address = "localhost";
10. $port = 1234; //调试的时候，可以多换端口来测试程序！
11. //创建一个SOCKET
12. if (($sock = socket_create(AF_INET, SOCK_STREAM, SOL_TCP)) === false)
13. {
14. echo "socket_create() 失败的原因是:" . socket_strerror(socket_last_error()) . "/n";
15. die;
16. }
17. //阻塞模式
18. if (socket_set_block($sock) == false)
19. {
20. echo "socket_set_block() 失败的原因是:" . socket_strerror(socket_last_error()) . "/n";
21. die;
22. }
23. //绑定到socket端口
24. if (socket_bind($sock, $address, $port) == false)
25. {
26. echo "socket_bind() 失败的原因是:" . socket_strerror(socket_last_error()) . "/n";
27. die;
28. }
29. //开始监听
30. if (socket_listen($sock, 4) == false)
31. {
32. echo "socket_listen() 失败的原因是:" . socket_strerror(socket_last_error()) . "/n";
33. die;
34. }
35. do
36. {
37. if (($msgsock = socket_accept($sock)) === false)
38. {
39. echo "socket_accept() failed: reason: " . socket_strerror(socket_last_error()) . "/n";
40. die;
41. }
42. //发到客户端
43. $msg = "welcome /n";
44. if (socket_write($msgsock, $msg, strlen($msg)) === false)
45. {
46. echo "socket_write() failed: reason: " . socket_strerror(socket_last_error()) ."/n";
47. die;
48. }
49. echo "读取客户端发来的信息/n";
50. $buf = socket_read($msgsock, 8192);
51. echo "收到的信息: $buf   /n";
52. socket_close($msgsock);
53. } while (true);
54. socket_close($sock);
55. ?>

client端：

1. <?php
2. /**
3. * 客户端代码
4. */
5. error_reporting(0);
6. set_time_limit(0);
7. echo " TCP/IP Connection /n";
8. $service_port = 10001;
9. $address = '127.0.0.1';
10. $socket = socket_create(AF_INET, SOCK_STREAM, SOL_TCP);
11. if ($socket === false)
12. {
13. die;
14. }
15. else
16. {
17. echo "OK";
18. }
19. echo "试图连接 ";
20. if (socket_connect($socket, $address, $service_port) == false)
21. {
22. $error = socket_strerror(socket_last_error());
23. echo "socket_connect() failed./n","Reason: {$error} /n";
24. die;
25. }
26. else
27. {
28. echo "连接OK/n";
29. }
30. $in   = "Hello World/r/n";
31. if (socket_write($socket, $in, strlen($in)) === false)
32. {
33. echo "socket_write() failed: reason: " . socket_strerror(socket_last_error()) ."/n";
34. die;
35. }
36. else
37. {
38. echo "发送到服务器信息成功！/n","发送的内容为: $in  /n";
39. }
40. $out  = "";
41. while ($out = socket_read($socket, 8192))
42. {
43. echo "接受的内容为: ".$out;
44. }
45. echo "关闭SOCKET…/n";
46. socket_close($socket);
47. echo "关闭OK/n";
48. ?>

再看websocket协议，是HTTP协议升级来的。看其消息头：

所以server端需要解析一下，并返回握手的协议内容：

在网上找到解析的相关代码 phpwebsocket - url:   http://code.google.com/p/phpwebsocket/

1. // Usage: $master=new WebSocket("localhost",12345);
2. class WebSocket{
3. var $master;
4. var $sockets = array();
5. var $users   = array();
6. var $debug   = false;
7. function __construct($address,$port){
8. error_reporting(E_ALL);
9. set_time_limit(0);
10. ob_implicit_flush();
11. $this->master=socket_create(AF_INET, SOCK_STREAM, SOL_TCP)     or die("socket_create() failed");
12. socket_set_option($this->master, SOL_SOCKET, SO_REUSEADDR, 1)  or die("socket_option() failed");
13. socket_bind($this->master, $address, $port)                    or die("socket_bind() failed");
14. socket_listen($this->master,20)                                or die("socket_listen() failed");
15. $this->sockets[] = $this->master;
16. $this->say("Server Started : ".date('Y-m-d H:i:s'));
17. $this->say("Listening on   : ".$address." port ".$port);
18. $this->say("Master socket  : ".$this->master."/n");
19. while(true){
20. $changed = $this->sockets;
21. socket_select($changed,$write=NULL,$except=NULL,NULL);
22. foreach($changed as $socket){
23. if($socket==$this->master){
24. $client=socket_accept($this->master);
25. if($client<0){ $this->log("socket_accept() failed"); continue; }
26. else{ $this->connect($client); }
27. }
28. else{
29. $bytes = @socket_recv($socket,$buffer,2048,0);
30. if($bytes==0){ $this->disconnect($socket); }
31. else{
32. $user = $this->getuserbysocket($socket);
33. if(!$user->handshake){ $this->dohandshake($user,$buffer); }
34. else{ $this->process($user,$this->unwrap($buffer)); }
35. }
36. }
37. }
38. }
39. }
40. function process($user,$msg){
41. /* Extend and modify this method to suit your needs */
42. /* Basic usage is to echo incoming messages back to client */
43. $this->send($user->socket,$msg);
44. }
45. function send($client,$msg){
46. $this->say("> ".$msg);
47. $msg = $this->wrap($msg);
48. socket_write($client,$msg,strlen($msg));
49. $this->say("! ".strlen($msg));
50. }
51. function connect($socket){
52. $user = new User();
53. $user->id = uniqid();
54. $user->socket = $socket;
55. array_push($this->users,$user);
56. array_push($this->sockets,$socket);
57. $this->log($socket." CONNECTED!");
58. $this->log(date("d/n/Y ")."at ".date("H:i:s T"));
59. }
60. function disconnect($socket){
61. $found=null;
62. $n=count($this->users);
63. for($i=0;$i<$n;$i++){
64. if($this->users[$i]->socket==$socket){ $found=$i; break; }
65. }
66. if(!is_null($found)){ array_splice($this->users,$found,1); }
67. $index=array_search($socket,$this->sockets);
68. socket_close($socket);
69. $this->log($socket." DISCONNECTED!");
70. if($index>=0){ array_splice($this->sockets,$index,1); }
71. }
72. function dohandshake($user,$buffer){
73. $this->log("/nRequesting handshake...");
74. $this->log($buffer);
75. list($resource,$host,$origin,$key1,$key2,$l8b) = $this->getheaders($buffer);
76. $this->log("Handshaking...");
77. //$port = explode(":",$host);
78. //$port = $port[1];
79. //$this->log($origin."/r/n".$host);
80. $upgrade  = "HTTP/1.1 101 WebSocket Protocol Handshake/r/n" .
81. "Upgrade: WebSocket/r/n" .
82. "Connection: Upgrade/r/n" .
83. //"WebSocket-Origin: " . $origin . "/r/n" .
84. //"WebSocket-Location: ws://" . $host . $resource . "/r/n" .
85. "Sec-WebSocket-Origin: " . $origin . "/r/n" .
86. "Sec-WebSocket-Location: ws://" . $host . $resource . "/r/n" .
87. //"Sec-WebSocket-Protocol: icbmgame/r/n" . //Client doesn't send this
88. "/r/n" .
89. $this->calcKey($key1,$key2,$l8b) . "/r/n";// .
90. //"/r/n";
91. socket_write($user->socket,$upgrade.chr(0),strlen($upgrade.chr(0)));
92. $user->handshake=true;
93. $this->log($upgrade);
94. $this->log("Done handshaking...");
95. return true;
96. }
97. function calcKey($key1,$key2,$l8b){
98. //Get the numbers
99. preg_match_all('/([/d]+)/', $key1, $key1_num);
100. preg_match_all('/([/d]+)/', $key2, $key2_num);
101. //Number crunching [/bad pun]
102. $this->log("Key1: " . $key1_num = implode($key1_num[0]) );
103. $this->log("Key2: " . $key2_num = implode($key2_num[0]) );
104. //Count spaces
105. preg_match_all('/([ ]+)/', $key1, $key1_spc);
106. preg_match_all('/([ ]+)/', $key2, $key2_spc);
107. //How many spaces did it find?
108. $this->log("Key1 Spaces: " . $key1_spc = strlen(implode($key1_spc[0])) );
109. $this->log("Key2 Spaces: " . $key2_spc = strlen(implode($key2_spc[0])) );
110. if($key1_spc==0|$key2_spc==0){ $this->log("Invalid key");return; }
111. //Some math
112. $key1_sec = pack("N",$key1_num / $key1_spc); //Get the 32bit secret key, minus the other thing
113. $key2_sec = pack("N",$key2_num / $key2_spc);
114. //This needs checking, I'm not completely sure it should be a binary string
115. return md5($key1_sec.$key2_sec.$l8b,1); //The result, I think
116. }
117. function getheaders($req){
118. $r=$h=$o=null;
119. if(preg_match("/GET (.*) HTTP/"               ,$req,$match)){ $r=$match[1]; }
120. if(preg_match("/Host: (.*)/r/n/"              ,$req,$match)){ $h=$match[1]; }
121. if(preg_match("/Origin: (.*)/r/n/"            ,$req,$match)){ $o=$match[1]; }
122. if(preg_match("/Sec-WebSocket-Key1: (.*)/r/n/",$req,$match)){ $this->log("Sec Key1: ".$sk1=$match[1]); }
123. if(preg_match("/Sec-WebSocket-Key2: (.*)/r/n/",$req,$match)){ $this->log("Sec Key2: ".$sk2=$match[1]); }
124. if($match=substr($req,-8))                                                                  { $this->log("Last 8 bytes: ".$l8b=$match); }
125. return array($r,$h,$o,$sk1,$sk2,$l8b);
126. }
127. function getuserbysocket($socket){
128. $found=null;
129. foreach($this->users as $user){
130. if($user->socket==$socket){ $found=$user; break; }
131. }
132. return $found;
133. }
134. function     say($msg=""){ echo $msg."/n"; }
135. function     log($msg=""){ if($this->debug){ echo $msg."/n"; } }
136. function    wrap($msg=""){ return chr(0).$msg.chr(255); }
137. function  unwrap($msg=""){ return substr($msg,1,strlen($msg)-2); }
138. }
139. class User{
140. var $id;
141. var $socket;
142. var $handshake;
143. }

继承类：可以自己按需写，这里我添加了几行代码，sendAll()等，很方便就改成了一个即时的网页版聊天室。

1. // Run from command prompt > php -q chatbot.demo.php
2. include "websocket.class.php";
3. // Extended basic WebSocket as ChatBot
4. class ChatBot extends WebSocket{
5. function process($user,$msg){
6. if (isset($user->first)) {
7. $this->send($user->socket,'');
8. $user->first = true;
9. }
10. $this->say("< ".$msg);
11. switch($msg){
12. case "hello" : $this->send($user->socket,"hello human");                       break;
13. case "hi"    : $this->send($user->socket,"zup human");                         break;
14. case "name"  : $this->send($user->socket,"my name is Multivac, silly I know"); break;
15. case "age"   : $this->send($user->socket,"I am older than time itself");       break;
16. case "date"  : $this->send($user->socket,"today is ".date("Y.m.d"));           break;
17. case "time"  : $this->send($user->socket,"server time is ".date("H:i:s"));     break;
18. case "thanks": $this->send($user->socket,"you're welcome");                    break;
19. case "bye"   : $this->send($user->socket,"bye");                               break;
20. //default      : $this->send($user->socket,$msg." not understood");              break;
21. default      : $this->sendAll($user, $msg);              break;
22. }
23. }
24. function sendAll($currentUser, $msg){
25. $usersList = $this->users;
26. foreach ($usersList as $user){
27. if ($user !== $currentUser) // 自己发送的消息就不再接收一次了
28. $this->send($user->socket, $msg);
29. }
30. }
31. }
32. $master = new ChatBot("localhost",12345);

客户端代码：

<html>

<head>

<title>WebSocket</title>

<style>

html,body{font:normal 0.9em arial,helvetica;}

#log {width:440px; height:200px; border:1px solid #7F9DB9; overflow:auto;}

#msg {width:330px;}

</style>

<script>

var socket;

function init(){

var host = "ws://localhost:12345/websocket/server.php";

try{

socket = new WebSocket(host);

log('WebSocket - status '+socket.readyState);

socket.onopen    = function(msg){ log("Welcome - status "+this.readyState); };

socket.onmessage = function(msg){ log("Received: "+msg.data); };

socket.onclose   = function(msg){ log("Disconnected - status "+this.readyState); };

}

catch(ex){ log(ex); }

$("msg").focus();

}

function send(){

var txt,msg;

txt = $("msg");

msg = txt.value;

if(!msg){ alert("Message can not be empty"); return; }

txt.value="";

txt.focus();

try{ socket.send(msg); log('Sent: '+msg); } catch(ex){ log(ex); }

}

function quit(){

log("Goodbye!");

socket.close();

socket=null;

}

// Utilities

function $(id){ return document.getElementById(id); }

function log(msg){ $("log").innerHTML+="<br>"+msg; }

function onkey(event){ if(event.keyCode==13){ send(); } }

</script>

</head>

<body onload="init()">

<h3>WebSocket v2.00</h3>

<div id="log"></div>

<input id="msg" type="textbox" onkeypress="onkey(event)"/>

<button onclick="send()">Send</button>

<button onclick="quit()">Quit</button>

<div>Commands: hello, hi, name, age, date, time, thanks, bye</div>

</body>

</html>

PS:

*　　这个websocket的类文件可能有一点问题，客户端握手后应该接收的第一条信息都丢失了，没细看代码，以后再检查吧。