socket文件传输功能的实现

原创 2016年06月02日 08:03:59
这节我们来完成 socket 文件传输程序,这是一个非常实用的例子。要实现的功能为:client 从 server 下载一个文件并保存到本地。

编写这个程序需要注意两个问题:
1) 文件大小不确定,有可能比缓冲区大很多,调用一次 write()/send() 函数不能完成文件内容的发送。接收数据时也会遇到同样的情况。

要解决这个问题,可以使用 while 循环,例如:
  1. //Server 代码
  2. int nCount;
  3. while( (nCount = fread(buffer, 1, BUF_SIZE, fp)) > 0 ){
  4. send(sock, buffer, nCount, 0);
  5. }
  6. //Client 代码
  7. int nCount;
  8. while( (nCount = recv(clntSock, buffer, BUF_SIZE, 0)) > 0 ){
  9. fwrite(buffer, nCount, 1, fp);
  10. }
对于 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:
  1. #include <stdio.h>
  2. #include <stdlib.h>
  3. #include <winsock2.h>
  4. #pragma comment (lib, "ws2_32.lib") //加载 ws2_32.dll
  5. #define BUF_SIZE 1024
  6. int main(){
  7. //先检查文件是否存在
  8. char *filename = "D:\\send.avi"; //文件名
  9. FILE *fp = fopen(filename, "rb"); //以二进制方式打开文件
  10. if(fp == NULL){
  11. printf("Cannot open file, press any key to exit!\n");
  12. system("pause");
  13. exit(0);
  14. }
  15. WSADATA wsaData;
  16. WSAStartup( MAKEWORD(2, 2), &wsaData);
  17. SOCKET servSock = socket(AF_INET, SOCK_STREAM, 0);
  18. sockaddr_in sockAddr;
  19. memset(&sockAddr, 0, sizeof(sockAddr));
  20. sockAddr.sin_family = PF_INET;
  21. sockAddr.sin_addr.s_addr = inet_addr("127.0.0.1");
  22. sockAddr.sin_port = htons(1234);
  23. bind(servSock, (SOCKADDR*)&sockAddr, sizeof(SOCKADDR));
  24. listen(servSock, 20);
  25. SOCKADDR clntAddr;
  26. int nSize = sizeof(SOCKADDR);
  27. SOCKET clntSock = accept(servSock, (SOCKADDR*)&clntAddr, &nSize);
  28. //循环发送数据,直到文件结尾
  29. char buffer[BUF_SIZE] = {0}; //缓冲区
  30. int nCount;
  31. while( (nCount = fread(buffer, 1, BUF_SIZE, fp)) > 0 ){
  32. send(clntSock, buffer, nCount, 0);
  33. }
  34. shutdown(clntSock, SD_SEND); //文件读取完毕,断开输出流,向客户端发送FIN包
  35. recv(clntSock, buffer, BUF_SIZE, 0); //阻塞,等待客户端接收完毕
  36. fclose(fp);
  37. closesocket(clntSock);
  38. closesocket(servSock);
  39. WSACleanup();
  40. system("pause");
  41. return 0;
  42. }

客户端代码:
  1. #include <stdio.h>
  2. #include <stdlib.h>
  3. #include <WinSock2.h>
  4. #pragma comment(lib, "ws2_32.lib")
  5. #define BUF_SIZE 1024
  6. int main(){
  7. //先输入文件名,看文件是否能创建成功
  8. char filename[100] = {0}; //文件名
  9. printf("Input filename to save: ");
  10. gets(filename);
  11. FILE *fp = fopen(filename, "wb"); //以二进制方式打开(创建)文件
  12. if(fp == NULL){
  13. printf("Cannot open file, press any key to exit!\n");
  14. system("pause");
  15. exit(0);
  16. }
  17. WSADATA wsaData;
  18. WSAStartup(MAKEWORD(2, 2), &wsaData);
  19. SOCKET sock = socket(PF_INET, SOCK_STREAM, IPPROTO_TCP);
  20. sockaddr_in sockAddr;
  21. memset(&sockAddr, 0, sizeof(sockAddr));
  22. sockAddr.sin_family = PF_INET;
  23. sockAddr.sin_addr.s_addr = inet_addr("127.0.0.1");
  24. sockAddr.sin_port = htons(1234);
  25. connect(sock, (SOCKADDR*)&sockAddr, sizeof(SOCKADDR));
  26. //循环接收数据,直到文件传输完毕
  27. char buffer[BUF_SIZE] = {0}; //文件缓冲区
  28. int nCount;
  29. while( (nCount = recv(sock, buffer, BUF_SIZE, 0)) > 0 ){
  30. fwrite(buffer, nCount, 1, fp);
  31. }
  32. puts("File transfer success!");
  33. //文件接收完毕后直接关闭套接字,无需调用shutdown()
  34. fclose(fp);
  35. closesocket(sock);
  36. WSACleanup();
  37. system("pause");
  38. return 0;
  39. }
在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() 就会返回,后面的代码继续执行。
版权声明:本文为芝麻软件工作室原创文章,未经芝麻软件工作室允许不得转载。

相关文章推荐

socket文件传输示例

Server端代码: /*************************************************************************    > File Na...

Windows用socket实现文件传输

这里只写客户端代码。编译前请加上libws2_32.a链接库,不然会导致链接失败。 客户端: #include #include #pragma comment(lib,"ws2_32.lib")...
  • e_one
  • e_one
  • 2016年05月14日 17:04
  • 14217

基于Socket的文件传输(使用CSocket类)

本软件使用MFC采用面向对象的方法实现了基于Socket的文件传输。这是原来研究生课程的结课作业,实现了Socket的发送和接收,以及读取ini配置文件等操作。 以下是当时结课作业 的正文:   ...

利用Socket实现多客户端传输对象和传输文件实现

第一次洗博客,纯属自己纪念,主要来源是慕课网的Socket通信课程,实现课后任务多客户端传输对象,自己遇到的最大问题是忘记给User类实现序列化接口。客户端实现:package com.imooc.t...

Linux下使用socket传输文件的C语言简单实现

服务器程序和客户端程序应当分别运行在两台计算机上。 在运行服务器端的计算机终端执行:./file_server 在运行客户端的计算终端上执行:./file_client   ipaddr_serv...

Java Socket实现文件传输

最近学Socket学上瘾了,就写了一个简单的文件传输程序。 客户端设计思路:客户端与服务端建立连接,选择客户端本地文件,先将文件名及大小等属性发送给服务端,再将文件通过流的方式传输给服务端。传输的...

socket实现文件传输功能

要实现的功能为:client 从 server 下载一个文件并保存到本地。 编写这个程序需要注意两个问题: 1) 文件大小不确定,有可能比缓冲区大很多,调用一次 write()/send() 函数不...

Socket编程一实现简易的聊天功能以及文件传输

干程序是一件枯燥重复的事,每当感到内心浮躁的时候,我就会找小说来看。我从小就喜爱看武侠小说,一直有着武侠梦。从金庸,古龙,梁羽生系列到凤歌(昆仑),孙晓(英雄志)以及萧鼎的(诛仙)让我领略着不一样的江...

socket实现大型文件传输

最近需要做网络传输的项目,需要实现较大文件的传输。在网上收集了不少资料,但是各有各的做法,尤其是关于文件自动接收这一块不太清楚。 经过图书馆查阅后还是找到了一种解决办法,虽然做的不太专业,但是思路比较...

linux read/write和fread/fwrite有什么区别

1,fread是带缓冲的,read不带缓冲. 2,fopen是标准c里定义的,open是POSIX中定义的. 3,fread可以读一个结构.read在linux/unix中读二进制与普通文件没有区...
内容举报
返回顶部
收藏助手
不良信息举报
您举报文章:socket文件传输功能的实现
举报原因:
原因补充:

(最多只允许输入30个字)