Socket网络编程--小小网盘程序(1)
这个系列是准备讲基于Linux Socket进行文件传输。简单的文件传输就是客户端可以上传文件,可以从服务器端下载文件。就这么两个功能如果再加上身份验证,就成了FTP服务器了,如果对用户的操作再加上一些功能(如分享),就可以作为一个最简单的网盘了。想想是不是有点小激动啊。
我这一小节就不讲那么高级的东西,就先了解文件怎么传输,我们以前的聊天程序传输数据都是一次发送就完成本次的发送,因为一个sendBuf是足够的。但是对于二进制文件来说,文件的大小就不一定了,有可能很大,所以我们的Buf是不知道要多少的。所以要传输大文件,就要分多次传输,然后在目的地进行合并。这样就可以实现大文件传输了。传输的方法有两种,一种是串行一种是并行。我接下来要实现的代码是使用串行传输的,因为比较简单,而并行传输,比较复杂,要传输几个控制信号,保证数据合并重整后不会出错。并行可以使用多进(线)程。优点是可以同时传输多个文件,串行是只能一次传输一个文件。我们看一下网盘,大多是可以下载多个文件(多任务)是吧?应该用的就是并行了。如果只是从原理上讲串行的传输速度肯定是比并行的快,因为并行还要控制信号要传,而且还要重新合并整合成一个文件。既然这样那为什么所有的网盘都是使用多任务下载呢?(多个任务同时下载,每个任务同时还有多个端口在接收数据)。是因为我们所处的网络环境问题。我们的网络环境并不是那么的稳定,如果只是用一个端口进行串行传输,那么如果网络突然中断或者什么原因,导致服务器与客户端暂时连接断开。那么我们之前传的数据就要重传了。这就是为什么今天下载个任务到80%,然后明天还可以接着下载了。如果是一次性串行传输就不行了。你就会问,那在串行传输再加个控制信号不就行了?那这样的话,与并行区别就不大了。
废话说了一大堆,到了真正要写代码的时候我还是使用串行来写,因为比较容易实现。(求原谅!)
实现客户端向服务器发送一个指定的二进制文件
client.cpp (为什么用cpp了,因为上次用c,然后代码越写越多,结果很多变量都写在开头,找起来不是很方便,所以用cpp了,哎,用c++就是为了用这个,会不会被骂?)
1 #include <netinet/in.h> // sockaddr_in 2 #include <sys/types.h> //socket 3 #include <sys/socket.h> //socket 4 #include <netdb.h> //gethostbyname 5 #include <unistd.h> //close 6 #include <stdio.h> 7 #include <stdlib.h> 8 #include <string.h> 9 #include <time.h> 10 #include <arpa/inet.h> //inet_addr 11 12 #define SERVVER_PORT 12138 13 #define LISTEN_QUEUE 20 14 #define BUFFER_SIZE 1024 15 16 struct Addr 17 { 18 char host[64]; 19 int port; 20 }; 21 22 int file_push(struct Addr addr,char *filenames) 23 { 24 struct sockaddr_in servAddr; 25 struct hostent * host; 26 int sockfd; 27 FILE *fp; 28 29 host=gethostbyname(addr.host); 30 servAddr.sin_family=AF_INET; 31 servAddr.sin_addr=*((struct in_addr *)host->h_addr); 32 //servAddr.sin_addr.s_addr=inet_addr("127.0.0.1"); 33 servAddr.sin_port=htons(addr.port); 34 if(host==NULL) 35 { 36 perror("获取IP地址失败"); 37 exit(-1); 38 } 39 if((sockfd=socket(AF_INET,SOCK_STREAM,0))==-1) 40 { 41 perror("socket创建失败"); 42 exit(-1); 43 } 44 45 if(connect(sockfd,(struct sockaddr *)&servAddr,sizeof(struct sockaddr_in))== -1) 46 { 47 perror("connect 失败"); 48 exit(-1); 49 } 50 51 //打开文件 52 if((fp=fopen(filenames,"rb"))==NULL) 53 { 54 perror("文件打开失败"); 55 exit(-1); 56 } 57 char buffer[BUFFER_SIZE]; 58 bzero(buffer,BUFFER_SIZE); 59 printf("正在传输文件"); 60 int len=0; 61 //不断的读取文件直到文件结束 62 while((len=fread(buffer,1,BUFFER_SIZE,fp))>0) 63 { 64 if(send(sockfd,buffer,len,0)<0) 65 { 66 perror("发送数据失败"); 67 exit(-1); 68 } 69 bzero(buffer,BUFFER_SIZE); 70 printf(".");//1K打印一个点//如果要实现百分比,就要计算文件大小,然后再处理即可 71 } 72 73 fclose(fp);//关闭文件流 74 close(sockfd);//关闭socket连接 75 76 return 0; 77 } 78 79 int main(int argc,char *argv[]) 80 { 81 char orderbuf[BUFFER_SIZE]; 82 char orderch[BUFFER_SIZE]; 83 struct Addr addr; 84 85 strcpy(addr.host,argv[1]); 86 addr.port=atoi(argv[2]); 87 while(1) 88 { 89 printf("\n请输入文件名:"); 90 fgets(orderbuf,BUFFER_SIZE,stdin); 91 orderbuf[strlen(orderbuf)-1]=0;//去掉获取到的回车符 92 //printf("%s\n",orderbuf); 93 file_push(addr,orderbuf); 94 } 95 96 return 0; 97 }
server.cpp
1 #include <netinet/in.h> 2 #include <sys/types.h> 3 #include <sys/socket.h> 4 #include <stdlib.h> 5 #include <time.h> 6 #include <string.h> 7 #include <unistd.h> 8 #include <stdio.h> 9 #include <arpa/inet.h> //inet_ntoa 10 11 #define SERVER_PORT 12138 12 #define LISTEN_QUEUE 20 13 #define BUFFER_SIZE 1024 14 15 void print_time(char *ch);//打印时间 16 17 int main(int argc,char *argv[]) 18 { 19 struct sockaddr_in server_addr; 20 bzero(&server_addr,sizeof(server_addr)); 21 server_addr.sin_family=AF_INET; 22 server_addr.sin_addr.s_addr=htons(INADDR_ANY); 23 server_addr.sin_port=htons(SERVER_PORT); 24 25 //创建套接字 26 int sockfd=socket(AF_INET,SOCK_STREAM,0); 27 if(sockfd<0) 28 { 29 perror("创建套接字失败"); 30 exit(-1); 31 } 32 33 if(bind(sockfd,(struct sockaddr *)&server_addr,sizeof(server_addr))==-1) 34 { 35 perror("bind 失败"); 36 exit(-1); 37 } 38 39 if(listen(sockfd,LISTEN_QUEUE)) 40 { 41 perror("listen 失败"); 42 exit(-1); 43 } 44 45 while(1) 46 { 47 pid_t pid; 48 struct sockaddr_in client_addr; 49 socklen_t length=sizeof(client_addr); 50 int clientfd=accept(sockfd,(struct sockaddr *)&client_addr,&length); 51 if(clientfd==-1) 52 { 53 perror("accept 失败"); 54 continue; 55 } 56 else 57 { 58 printf("客户端%s:%d连接成功\n",inet_ntoa(client_addr.sin_addr),ntohs(client_addr.sin_port)); 59 pid=fork(); 60 if(pid<0) 61 { 62 perror("创建进程失败"); 63 } 64 else if(pid==0)/*child*/ 65 { 66 char buffer[BUFFER_SIZE]; 67 int data_len; 68 FILE * fp=NULL; 69 bzero(buffer,BUFFER_SIZE); 70 if((fp=fopen("data","wb"))==NULL) 71 { 72 perror("文件打开失败"); 73 exit(-1); 74 } 75 //循环接收数据 76 int size=0;//表示有多少个块 77 while(data_len=recv(clientfd,buffer,BUFFER_SIZE,0))//data_len为0时结束,是因为当客户端没有再发送数据过来时是接收0的,也表示该文件传输完毕了。 78 { 79 if(data_len<0) 80 { 81 perror("接收数据错误"); 82 exit(-1); 83 } 84 size++; 85 if(size==1) 86 { 87 printf("正在接收来自%s:%d的文件\n",inet_ntoa(client_addr.sin_addr),ntohs(client_addr.sin_port)); 88 } 89 else 90 { 91 printf("."); 92 } 93 //向文件中写入 94 int write_len=fwrite(buffer,sizeof(char),data_len,fp);//向文件中写入,默认文件打开时会有一个文件指针进行写入。如果是并行传输就要修改这个文件指针了,还是有点麻烦的,如果是串行的,我们都不用管了,多方便。 95 if(write_len>data_len) 96 { 97 perror("写入数据错误"); 98 exit(-1); 99 } 100 bzero(buffer,BUFFER_SIZE); 101 } 102 if(size>0) 103 { 104 printf("\n%s:%d的文件传送完毕\n",inet_ntoa(client_addr.sin_addr),ntohs(client_addr.sin_port)); 105 } 106 else 107 { 108 printf("\n%s:%d的文件传送失败\n",inet_ntoa(client_addr.sin_addr),ntohs(client_addr.sin_port));//如果传过来的文件大小为0,也是会出现这个错误 109 } 110 fclose(fp); 111 //rename("data","asdf");//这里可以修改文件的名字。保存到服务器的话可以随便改个名字,防止文件名重复 112 exit(0); 113 } 114 else /*pather*/ 115 { 116 ; 117 } 118 } 119 close(clientfd); 120 } 121 122 return 0; 123 } 124 125 void print_time(char *ch) 126 { 127 time_t now; 128 struct tm * stm; 129 time(&now); 130 stm=localtime(&now); 131 sprintf(ch,"%02d:%02d:%02d\n",stm->tm_hour,stm->tm_min,stm->tm_sec); 132 return ; 133 }
两部分的代码都不难理解,关键的代码就是那个while循环了。这个就不给截图了,运行是没有问题。
这篇杜鑫先生的回答很不错,可以看一下: http://www.zhihu.com/question/21591490
作者:无脑仔的小明 出处:http://www.cnblogs.com/wunaozai/ 本文版权归作者和博客园共有,欢迎转载,但未经作者同意必须保留此段声明,且在文章页面明显位置给出原文连接,否则保留追究法律责任的权利。 如果文中有什么错误,欢迎指出。以免更多的人被误导。有需要沟通的,可以站内私信,文章留言,或者关注“无脑仔的小明”公众号私信我。一定尽力回答。 |