调用 close()/closesocket() 函数意味着完全断开连接,即不能发送数据也不能接收数据,这种“生硬”的方式有时候会显得不太“优雅”。
图1:close()/closesocket() 断开连接
上图演示了两台正在进行双向通信的主机。主机A发送完数据后,单方面调用 close()/closesocket() 断开连接,之后主机A、B都不能再接受对方传输的数据。实际上,是完全无法调用与数据收发有关的函数。
一般情况下这不会有问题,但有些特殊时刻,需要只断开一条数据传输通道,而保留另一条。
使用 shutdown() 函数可以达到这个目的,它的原型为:
- int shutdown(int sock, int howto); //Linux
- int shutdown(SOCKET s, int howto); //Windows
sock 为需要断开的套接字,howto 为断开方式。
howto 在 Linux 下有以下取值:
- SHUT_RD:断开输入流。套接字无法接收数据(即使输入缓冲区收到数据也被抹去),无法调用输入相关函数。
- SHUT_WR:断开输出流。套接字无法发送数据,但如果输出缓冲区中还有未传输的数据,则将传递到目标主机。
- SHUT_RDWR:同时断开 I/O 流。相当于分两次调用 shutdown(),其中一次以 SHUT_RD 为参数,另一次以 SHUT_WR 为参数。
howto 在 Windows 下有以下取值:
- SD_RECEIVE:关闭接收操作,也就是断开输入流。
- SD_SEND:关闭发送操作,也就是断开输出流。
- SD_BOTH:同时关闭接收和发送操作。
至于什么时候需要调用 shutdown() 函数,下节我们会以文件传输为例进行讲解。
close()/closesocket()和shutdown()的区别
确切地说,close() / closesocket() 用来关闭套接字,将套接字描述符(或句柄)从内存清除,之后再也不能使用该套接字,与C语言中的 fclose() 类似。应用程序关闭套接字后,与该套接字相关的连接和缓存也失去了意义,TCP协议会自动触发关闭连接的操作。
shutdown() 用来关闭连接,而不是套接字,不管调用多少次 shutdown(),套接字依然存在,直到调用 close() / closesocket() 将套接字从内存清除。
调用 close()/closesocket() 关闭套接字时,或调用 shutdown() 关闭输出流时,都会向对方发送 FIN 包。FIN 包表示数据传输完毕,计算机收到 FIN 包就知道不会再有数据传送过来了。
默认情况下,close()/closesocket() 会立即向网络中发送FIN包,不管输出缓冲区中是否还有数据,而shutdown() 会等输出缓冲区中的数据传输完毕再发送FIN包。也就意味着,调用 close()/closesocket() 将丢失输出缓冲区中的数据,而调用 shutdown() 不会。
使用实例:
这节我们来完成 socket 文件传输程序,这是一个非常实用的例子。要实现的功能为:client 从 server 下载一个文件并保存到本地。
编写这个程序需要注意两个问题:
1) 文件大小不确定,有可能比缓冲区大很多,调用一次 write()/send() 函数不能完成文件内容的发送。接收数据时也会遇到同样的情况。
要解决这个问题,可以使用 while 循环,例如:
- //Server 代码
- int nCount;
- while( (nCount = fread(buffer, 1, BUF_SIZE, fp)) > 0 ){
- send(sock, buffer, nCount, 0);
- }
- //Client 代码
- int nCount;
- while( (nCount = recv(clntSock, buffer, BUF_SIZE, 0)) > 0 ){
- fwrite(buffer, nCount, 1, fp);
- }
对于 Server 端的代码,当读取到文件末尾,fread() 会返回 0,结束循环。
对于 Client 端代码,有一个关键的问题,就是文件传输完毕后让 recv() 返回 0,结束 while 循环。
注意:读取完缓冲区中的数据 recv() 并不会返回 0,而是被阻塞,直到缓冲区中再次有数据。
2) Client 端如何判断文件接收完毕,也就是上面提到的问题——何时结束 while 循环。
最简单的结束 while 循环的方法当然是文件接收完毕后让 recv() 函数返回 0,那么,如何让 recv() 返回 0 呢?recv() 返回 0 的唯一时机就是收到FIN包时。
FIN 包表示数据传输完毕,计算机收到 FIN 包后就知道对方不会再向自己传输数据,当调用 read()/recv() 函数时,如果缓冲区中没有数据,就会返回 0,表示读到了”socket文件的末尾“。
这里我们调用 shutdown() 来发送FIN包:server 端直接调用 close()/closesocket() 会使输出缓冲区中的数据失效,文件内容很有可能没有传输完毕连接就断开了,而调用 shutdown() 会等待输出缓冲区中的数据传输完毕。
本节以Windows为例演示文件传输功能,Linux与此类似,不再赘述。请看下面完整的代码。
服务器端 server.cpp:
- #include <stdio.h>
- #include <stdlib.h>
- #include <winsock2.h>
- #pragma comment (lib, "ws2_32.lib") //加载 ws2_32.dll
- #define BUF_SIZE 1024
- int main(){
- //先检查文件是否存在
- char *filename = "D:\\send.avi"; //文件名
- FILE *fp = fopen(filename, "rb"); //以二进制方式打开文件
- if(fp == NULL){
- printf("Cannot open file, press any key to exit!\n");
- system("pause");
- exit(0);
- }
- WSADATA wsaData;
- WSAStartup( MAKEWORD(2, 2), &wsaData);
- SOCKET servSock = socket(AF_INET, SOCK_STREAM, 0);
- sockaddr_in sockAddr;
- memset(&sockAddr, 0, sizeof(sockAddr));
- sockAddr.sin_family = PF_INET;
- sockAddr.sin_addr.s_addr = inet_addr("127.0.0.1");
- sockAddr.sin_port = htons(1234);
- bind(servSock, (SOCKADDR*)&sockAddr, sizeof(SOCKADDR));
- listen(servSock, 20);
- SOCKADDR clntAddr;
- int nSize = sizeof(SOCKADDR);
- SOCKET clntSock = accept(servSock, (SOCKADDR*)&clntAddr, &nSize);
- //循环发送数据,直到文件结尾
- char buffer[BUF_SIZE] = {0}; //缓冲区
- int nCount;
- while( (nCount = fread(buffer, 1, BUF_SIZE, fp)) > 0 ){
- send(clntSock, buffer, nCount, 0);
- }
- shutdown(clntSock, SD_SEND); //文件读取完毕,断开输出流,向客户端发送FIN包
- recv(clntSock, buffer, BUF_SIZE, 0); //阻塞,等待客户端接收完毕
- fclose(fp);
- closesocket(clntSock);
- closesocket(servSock);
- WSACleanup();
- system("pause");
- return 0;
- }
客户端代码:
- #include <stdio.h>
- #include <stdlib.h>
- #include <WinSock2.h>
- #pragma comment(lib, "ws2_32.lib")
- #define BUF_SIZE 1024
- int main(){
- //先输入文件名,看文件是否能创建成功
- char filename[100] = {0}; //文件名
- printf("Input filename to save: ");
- gets(filename);
- FILE *fp = fopen(filename, "wb"); //以二进制方式打开(创建)文件
- if(fp == NULL){
- printf("Cannot open file, press any key to exit!\n");
- system("pause");
- exit(0);
- }
- WSADATA wsaData;
- WSAStartup(MAKEWORD(2, 2), &wsaData);
- SOCKET sock = socket(PF_INET, SOCK_STREAM, IPPROTO_TCP);
- sockaddr_in sockAddr;
- memset(&sockAddr, 0, sizeof(sockAddr));
- sockAddr.sin_family = PF_INET;
- sockAddr.sin_addr.s_addr = inet_addr("127.0.0.1");
- sockAddr.sin_port = htons(1234);
- connect(sock, (SOCKADDR*)&sockAddr, sizeof(SOCKADDR));
- //循环接收数据,直到文件传输完毕
- char buffer[BUF_SIZE] = {0}; //文件缓冲区
- int nCount;
- while( (nCount = recv(sock, buffer, BUF_SIZE, 0)) > 0 ){
- fwrite(buffer, nCount, 1, fp);
- }
- puts("File transfer success!");
- //文件接收完毕后直接关闭套接字,无需调用shutdown()
- fclose(fp);
- closesocket(sock);
- WSACleanup();
- system("pause");
- return 0;
- }
在D盘中准备好send.avi文件,先运行 server,再运行 client:
Input filename to save: D:\\recv.avi↙
//稍等片刻后
File transfer success!
打开D盘就可以看到 recv.avi,大小和 send.avi 相同,可以正常播放。
注意 server.cpp 第42行代码,recv() 并没有接收到 client 端的数据,当 client 端调用 closesocket() 后,server 端会收到FIN包,recv() 就会返回,后面的代码继续执行。