基于TCP图片文件的传输

#include <winsock2.h> #include <stdio.h> #define SER_PORT 6000 #define MAX_LEN 1024 struct FILE_INFO{ int fileLen; char fileName[MAX_PATH]; }; void main() { WORD wVersionRequested; WSADATA wsaData; int err; //加载套接字库 wVersionRequested = MAKEWORD( 1, 1 ); //第一个1为高字节指定winsock库的副版本， //第二个1为低字节字节指定winsock库的主版本 err = WSAStartup( wVersionRequested, &wsaData ); if ( err != 0 ) { return; } if ( LOBYTE( wsaData.wVersion ) != 1 || HIBYTE( wsaData.wVersion ) != 1 ) { WSACleanup( ); return; } char buf[MAX_LEN+1]={0}; FILE *fp; char filename[MAX_PATH]; //创建用于监听的套接字 SOCKET sockSrv; sockSrv=socket(AF_INET,SOCK_STREAM,0); //绑定套接字 SOCKADDR_IN addrClient; addrClient.sin_family = AF_INET; addrClient.sin_port = htons(SER_PORT); addrClient.sin_addr.s_addr = htonl(INADDR_ANY); //网络字节区,inet_addr("192.168.0.10") if(SOCKET_ERROR == bind(sockSrv,(SOCKADDR *)&addrClient,sizeof(SOCKADDR))) { perror( "Bind error. "); exit(1); } //将套接字设为监听模式，准备接收客户请求 if(SOCKET_ERROR == listen(sockSrv,5)) { perror( "listen() error\n "); exit(1); } //等待客户请求到来 int len = sizeof(SOCKADDR); SOCKET sockConnect= accept(sockSrv, (SOCKADDR*)&addrClient,&len);//其中len必须赋初值 printf("输入新建文件名：\n"); scanf("%s",filename); FILE_INFO fileInfo; recv(sockConnect,(char*)&fileInfo,sizeof(FILE_INFO),0); printf("\n发送文件大小：0x%x字节,\n原文件：%s\n",fileInfo.fileLen, fileInfo.fileName); fp=fopen( filename, "wb"); if(fp == NULL) { printf( "File open error!\n "); exit(1); } int rn; int num=0; while(1) { rn=recv(sockConnect,buf,MAX_LEN,0); if(rn < 0) { printf( "Can 't receive file!\n "); exit(1); } buf[rn]= '\0 '; if(rn != 0) { num+=rn; if(num >= fileInfo.fileLen) { fwrite(buf, 1, fileInfo.fileLen+MAX_LEN-num , fp); //printf("1:%d\n",fileInfo.fileLen+MAX_LEN-num); //printf("最后%d\n", rn); break; } else { fwrite(buf,1,rn,fp); //printf("%d\n",rn); } } else { printf("\nreceive over.\n"); printf("0x%x\n",num); break; } } closesocket(sockConnect); fclose(fp); WSACleanup(); getchar(); }