本文并非解释什么是非阻塞socket,也不是介绍socket API的用法, 取而代替的是让你感受实际工作中的代码编写。虽然很简陋,但你可以通过man手册与其它资源非富你的代码。请注意本教程所说的主题,如果细说,内容可以达到一本书内容,你会发现本教程很有用。
本教程内容如下:
1. 改变一个阻塞的socket为非阻塞模式。
2. select模型
3. FD宏
4. 读写函数
5. 写一个非阻塞socket代码片
6. 整个代码
7.下一步
如果你在此有许多问题,那么恭喜你,在初级阶段,任何人都没有剥夺你发现问题的权利。关于你所发现的问题,请不要犹豫email我。
你可以自由发表本教程到任何WWW或FTP网部,但无论如何也要保持原教程的原型。这样我将会非常感谢你。
1.改变一个阻塞的socket为非阻塞模式
简单的几行代码就可以创建一个socket 并连接,看起来如此简单。(你可以自己加入错误处理)
s = socket(AF_INET, SOCK_STREAM, 0); memset(&sin, 0, sizeof(struct sockaddr_in)); sin.sin_family = AF_INET; sin.sin_port = htons(port); sin.sin_addr.s_addr = inet_addr(hstname); if(sin.sin_addr.s_addr == INADDR_NONE) { connect(s, (struct sockaddr *)&sin, sizeof(sin))
有很多种方法可以设置socket为非阻塞模式, 我在unix下常用的方法如下:
int x; x=fcntl(s,F_GETFL,0); fcntl(s,F_SETFL,x | O_NONBLOCK);
到现在为此, 这个socket已为非阻塞模式了,我们可以把焦点放在如何用它了。 但是在接着写代码之前,我们要看看我们将要用到的命令。
2.选择模型
select这个方法用来检测一个socket是否有数据到来或是是否有准备好的数据要发送。声明如下:
select(s, &read_flags, &write_flags, &exec_flags, timer);
s socket的句柄
read_flags 读描述字集合。检查socket上是否有数据可读。
write_flags 写描述字集合。检查socket上是否已有数据可发送。
exec_flags 错误描述字集合。(本教程这儿不介绍)
timer 等待某个条件为真时超时时间。
在本教程中,我们将如下用:
select(s, &read_flags, &write_flags, NULL, timer);
exec_flags参数设为null,因为在我们的程序中我们不需要关心exec事件。(解释它,超出了本教程的范围)
3.FD宏
在select方法中的描述字集合是用来检测socket上发生的读取事件的,它用法如下:
FD_ZERO(s, &write_flags) sets all associated flags in the socket to 0 FD_SET(s, &write_flags) used to set a socket for checking FD_CLR (s, &write_flags) used to clear a socket from being checked FD_ISSET(s, &write_flags) used to query as to if the socket is ready for reading or writing.
如何用,我们在后面的代码中展示。
4.读取方法
你应知道如何用下面的两个方法,但是在非阻塞模式下,它们的用法有一点点不同。
write(s,buffer,sizeof(buffer)) send the text in "buffer" read(s,buffer,sizeof(buffer)) read available data into "buffer"
当一个socket用非阻塞模式时,当调用这两个方法的时候,它们将立即返回,比如,如果没有数据的时候,它们将不会阻塞等待数据,而是返回一个错误。从现在开始,我们就要看代码了。
5.写一个非阻塞socket代码片
首先声明要用到的变量:
fd_set read_flags,write_flags; // the flag sets to be used struct timeval waitd; // the max wait time for an event char buffer[8196]; // input holding buffer int stat; // holds return value for select();
我们程序运行的大部份时间都花费在不断调用select(它将花费我们大部份CPU时间),至到有数据准备好读或取。此时,timer就体现了它的意义。它决定select将等待多久。下面就是用法,请仔细分析:
// Insert Code to create a socket while(1) // put program in an infinite loop of reading and writing data { waitd.tv_sec = 1; // Make select wait up to 1 second for data waitd.tv_usec = 0; // and 0 milliseconds. FD_ZERO(&read_flags); // Zero the flags ready for using FD_ZERO(&write_flags); // Set the sockets read flag, so when select is called it examines // the read status of available data. FD_SET(thefd, &read_flags); // If there is data in the output buffer to be sent then we // need to also set the write flag to check the write status // of the socket if(strlen(outbuff)!=0) FD_SET(thefd, &write_flags); // Now call select stat=select(s+1, &read_flags,&write_flags,(fd_set*)0,&waitv); if(stat < 0) { // If select breaks then pause for 5 seconds sleep(5); // then continue continue; } // Now select will have modified the flag sets to tell us // what actions can be performed // Check if data is available to read if (FD_ISSET(thefd, &read_flags)) { FD_CLR(thefd, &read_flags); // here is where you use the read(). // If read returns an error then the socket // must be dead so you must close it. } //Check if the socket is prepared to accept data if (FD_ISSET(thefd, &write_flags)) { FD_CLR(thefd, &write_flags); // this means the socket is ready for you to use write() } // Now here you can put in any of the precedures that you want // to happen every 1 second or so. // now the loop repeats over again
补充:请确保只有在你有数据发送的情况下才设置write_flag这个描述字集合,因为socket一量创建总是可写的。也就是说,如果你设置了这个参数,select将不会等待,而是马上返回并一直循环,它将抢占CPU99%的利用率,这是不允许的。
6.整个代码
最后利用我们所学,写一个简单的客户端。当然用非阻塞模式写一个客户端有点大采小用,这儿我们只是为了展示用法。更多示例请看第7节内容。
#include
7.下一步
其它更多的示例代码从此教程中分离,以zip文件的方式给出。为了更好的理解所学, 你最好参考一些结构更复杂,技术更强的代码:
http://users.cybernex.net.au/jj/sock.zip