Socket收发文件

Send分为阻塞和非阻塞，阻塞模式下，如果正常的话，

会直到把你所需要发送的数据发完再返回；

非阻塞，会根据你的socket在底层的可用缓冲区的大小，

来将你的缓冲区当中的数据拷贝过去，有多大缓冲区就拷贝多少，

缓冲区满了就立即返回，这个时候的返回值，

只表示拷贝到缓冲区多少数据，但是并不代表发送多少数据，

同时剩下的部分需要你再次调用send才会再一次拷贝到底层缓冲区。

recv函数

int recv( SOCKET s, char FAR *buf, int len, int flags);

不论是客户还是服务器应用程序都用recv函数从TCP连接的另一端接收数据。该函数的第一个参数指定接收端套接字描述符；

第二个参数指明一个缓冲区，该缓冲区用来存放recv函数接收到的数据；

第三个参数指明buf的长度；

第四个参数一般置0。

这里只描述同步Socket的recv函数的执行流程。当应用程序调用recv函数时，

（1）recv先等待s的发送缓冲中的数据被协议传送完毕，如果协议在传送s的发送缓冲中的数据时出现网络错误，那么recv函数返回SOCKET_ERROR，

（2）如果s的发送缓冲中没有数据或者数据被协议成功发送完毕后，recv先检查套接字s的接收缓冲区，如果s接收缓冲区中没有数据或者协议正在接收数 据，那么recv就一直等待，直到协议把数据接收完毕。当协议把数据接收完毕，recv函数就把s的接收缓冲中的数据copy到buf中（注意协议接收到的数据可能大于buf的长度，所以 在这种情况下要调用几次recv函数才能把s的接收缓冲中的数据copy完。recv函数仅仅是copy数据，真正的接收数据是协议来完成的），

recv函数返回其实际copy的字节数。如果recv在copy时出错，那么它返回SOCKET_ERROR；如果recv函数在等待协议接收数据时网络中断了，那么它返回0。

默认 socket 是阻塞的

解阻塞与非阻塞recv返回值没有区分，都是

<0 出错

=0 连接关闭

>0 接收到数据大小，

特别：返回值<0时并且(errno == EINTR || errno == EWOULDBLOCK || errno == EAGAIN)的情况下认为连接是正常的，继续接收。

只是阻塞模式下recv会阻塞着接收数据，非阻塞模式下如果没有数据会返回，不会阻塞着读，因此需要循环读取）。

返回说明：   

成功执行时，返回接收到的字节数。另一端已关闭则返回0。失败返回-1，errno被设为以下的某个值   

EAGAIN：套接字已标记为非阻塞，而接收操作被阻塞或者接收超时

EBADF：sock不是有效的描述词

ECONNREFUSE：远程主机阻绝网络连接

EFAULT：内存空间访问出错

EINTR：操作被信号中断

EINVAL：参数无效

ENOMEM：内存不足

ENOTCONN：与面向连接关联的套接字尚未被连接上

ENOTSOCK：sock索引的不是套接字

当返回值是0时，为正常关闭连接；

思考：

当对侧没有send，即本侧的套接字s的接收缓冲区无数据，返回值是什么

（EAGAIN，原因为超时，待测）

////////////////////////////////////////////////////////////////////

//Server

//本程序输入路径，能进行大文件的socket传输

//本文的优点是为防止接收时接收不完整，例如发送了4个，分三次才接收到，1，1，2

//后面的接收收到脏数据，写了收发函数，

//保证一定发送接收了多少函数

/////////////////////////////////////////////////////////////////////

#include <winsock2.h>

#include <stdio.h>

#include <string.h>

#include <windows.h>

#include <iostream>

#pragma comment(lib,"wsock32.lib")

#define  MAX_LENGTH 4096

using namespace std;

void InitListenSocket(SOCKET& Connect_Socket)

{

int ret;

WSADATA wsadata;

SOCKADDR_IN Server_Addr;

int Port = 9088;

if ((ret = WSAStartup(MAKEWORD(2,2), &wsadata)) != 0)

{

printf("WSAStartup failed with error %d\n", ret);

return;

}

if ((Connect_Socket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP)) == INVALID_SOCKET)

{

printf("socket failed with error %d\n", WSAGetLastError());

WSACleanup();

return;

}

Server_Addr.sin_family = AF_INET;

Server_Addr.sin_port = htons(Port);    

Server_Addr.sin_addr.s_addr = htonl(INADDR_ANY);

if (bind(Connect_Socket, (SOCKADDR *)&Server_Addr, sizeof(Server_Addr))== SOCKET_ERROR)

{

printf("bind failed with error %d\n", WSAGetLastError());

closesocket(Connect_Socket);

WSACleanup();

return;

}

if (listen(Connect_Socket, 5) == SOCKET_ERROR)

{

printf("listen failed with error %d\n", WSAGetLastError());

closesocket(Connect_Socket);

WSACleanup();

return;

} 

printf("We are awaiting a connection on port %d.\n", Port);

}

BOOL Recv(SOCKET sock,char * buf,long length)

{

int ret=0;

long pos=0;

try{

do{

ret=recv(sock,buf+pos,length-pos,MSG_WAITALL);

if(ret < 0)

{

printf("receive data failed!\n");

return FALSE;

}

printf("        ret: %d, pos :%ld, length: %ld\n",ret,pos,length);

pos+=ret;

}while(pos<length);

if(pos==length)

return TRUE;

else 

return FALSE;

}

catch (_com_error &e)

{

cout << e.Description();

}

}

BOOL Send(SOCKET sock,char * buf,long length)

{

int ret=0;

long pos=0;

do{

ret=send(sock,buf+pos,length-pos,0);

if(ret < 0)

{

printf("send data failed!\n");

return FALSE;

}

printf("        ret: %d,pos :%ld ,length: %ld\n",ret,pos,length);

pos+=ret;

}while(pos<length);

if(pos==length)

return TRUE;

else 

return FALSE;

}

void main(void)

{

SOCKET        Connect_Socket;

SOCKADDR_IN   Client_Addr;

int           Client_Addr_Len;   

int           conn_num = 0;

BOOL          ret=FALSE;

char * recvbuf=(char *)malloc(sizeof(MAX_LENGTH));

char name[256],fname[256];;

int i=0;

long filelen,pos;

unsigned char len=0;

while(true)

{

InitListenSocket(Connect_Socket);

Client_Addr_Len = sizeof(SOCKADDR);

SOCKET sock = accept(Connect_Socket, (SOCKADDR *) &Client_Addr,&Client_Addr_Len);

printf("-----------start recvive----------------\n");

printf("start recv header:\n");

ret = Recv(sock,recvbuf,MAX_LENGTH);  //先接收文件名和文件大小信息

if(!ret)

{

printf("接收头信息失败\n");

continue;

}

memcpy(&len,recvbuf,1);

memcpy(name,recvbuf+1,len);

name[len]='\0';

memcpy(&filelen,recvbuf+256,sizeof(long));

printf("文件总长度为：%ld\n",filelen);

i=len-1;

while(name[i]!='\\')       //获取文件名

i--;

strcpy(fname,&name[i+1]);

printf("正在接收：%s\n",fname);

//打开文件

FILE *fd;

fd = fopen((char*)fname,"w");        //这种方式打开文件会把里面的内容清空，没文件则建立

fclose(fd);

fd = fopen((char*)fname,"ab");       //建立一个文件，二进制方式打开，不然会出现接收端忽略换行符

printf("start recv data:\n");

pos=0;

        ret = FALSE;

do

{   

if(filelen-pos==filelen%MAX_LENGTH)

{

ret=Recv(sock,recvbuf,filelen-pos);

if(!ret)

{

printf("接收数据失败\n");

break;

}

fwrite(recvbuf,sizeof(u_char),filelen-pos,fd);

printf("ret:%d ,pos: %ld, filelength: %ld\n",ret,pos,filelen);

printf("接收完毕!\n");

break;

} 

ret=Recv(sock,recvbuf,MAX_LENGTH);

if(!ret)

{

printf("接收数据失败\n");

break;

}

printf("ret:%d ,pos: %ld, filelength: %ld\n",ret,pos,filelen);

fwrite(recvbuf,sizeof(u_char),MAX_LENGTH,fd);

pos+=MAX_LENGTH;

}while(TRUE);

fclose(fd);

    }

free(recvbuf);

closesocket(sock);

closesocket(Connect_Socket);

WSACleanup();

system("pause");

}

======================================================================================================

////////////////////////////////////////////////////////////////////

//Client

//本程序输入路径，能进行大文件的socket传输

/////////////////////////////////////////////////////////////////////

#include <winsock2.h>

#include <stdio.h>

#include <string>

#define MAX_LENGTH 4096

#pragma comment(lib,"wsock32.lib")

SOCKET InitConnection(SOCKADDR_IN  ServerAddr)

{

WSADATA              wsaData;

SOCKET               s;

int                  Ret;

if ((Ret = WSAStartup(MAKEWORD(2,2), &wsaData)) != 0)

{

printf("WSAStartup failed with error %d\n", Ret);

return 0;

}

if ((s = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP))

== INVALID_SOCKET)

{

printf("socket failed with error %d\n", WSAGetLastError());

WSACleanup();

return 0;

}

if (connect(s, (SOCKADDR *) &ServerAddr, sizeof(ServerAddr)) 

== SOCKET_ERROR)

{

printf("connect failed with error %d\n", WSAGetLastError());

closesocket(s);

WSACleanup();

return 0;

}

else 

{   

return s;

}

}

BOOL Recv(SOCKET sock,char * buf,long length)

{

int ret=0;

long pos=0;

do{

ret=recv(sock,buf+pos,length-pos,0);

if(ret < 0)

{

printf("receive data failed!\n");

return FALSE;

}

printf("        ret: %d, pos :%ld, length: %ld\n",ret,pos,length);

pos+=ret;

}while(pos<length);

if(pos==length)

return TRUE;

else 

return FALSE;

}

BOOL Send(SOCKET sock,char * buf,long length)

{

if(length==0)

   return TRUE;

int ret=0;

long pos=0;

do{

ret=send(sock,buf+pos,length-pos,0);

if(ret < 0)

{

printf("        ret: %d,pos :%ld ,length: %ld\n",ret,pos,length);

printf("send data failed!\n");

return FALSE;

}

printf("        ret: %d,pos :%ld ,length: %ld\n",ret,pos,length);

pos+=ret;

}while(pos<length);

if(pos==length)

return TRUE;

else 

return FALSE;

}

void main()

{

char *sendbuf=(char *)malloc(sizeof(MAX_LENGTH));

long filelen,pos=0;

char fname[256];

int readLen ;

unsigned char len=0; 

   while(true)

{

SOCKADDR_IN          ServerAddr0;

ServerAddr0.sin_family = AF_INET;

ServerAddr0.sin_port = htons(9088);

ServerAddr0.sin_addr.s_addr = inet_addr("127.0.0.1");

SOCKET s=InitConnection( ServerAddr0);

if(s==0)

return ;

printf("已经连接成功服务器!\n");

printf("-----------start send----------------\n");

printf("please input the file path：\n");

gets(fname);

FILE *fd;

if( (fd = fopen((char*)fname,"rb")) == NULL) {//二进制方式打开，不然会出现接收端忽略换行符（回车符不记得）的错误

printf("当前目录没找到文件%s!\n",fname);

}

fseek(fd,0,SEEK_END);

filelen = ftell(fd);                       //取得文件总长度

printf("文件总长度为：%ld字节\n",filelen);

fseek(fd,0,SEEK_SET);

len=strlen(fname);

memcpy(sendbuf,&len,1);

memcpy(sendbuf+1,fname,strlen(fname));

memcpy(sendbuf+256,&filelen,sizeof(long));

BOOL ret=FALSE;

printf("start send header:\n");

ret=Send(s,sendbuf,MAX_LENGTH);

if(!ret)

{

printf("发送头信息失败\n");

continue;

}

printf("start send data:\n");

pos=0;

do{ 

readLen = fread(sendbuf,sizeof(char),MAX_LENGTH,fd); 

pos+=readLen;

ret=Send(s,sendbuf,readLen); 

if(!ret)

{   

printf("发送数据失败!\n");

break;

}

            printf("ret:%d ,pos: %ld, filelength: %ld\n",ret,pos,filelen);

}while(pos<filelen); //循环读取文件 

if(pos==filelen)

printf("发送完成!\n\n");

else 

printf("发送失败！\n\n");

fclose(fd);

}

    free(sendbuf);

closesocket(s);

WSACleanup();

system("pause");

}