zoukankan      html  css  js  c++  java
  • 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("
    请输入文件名:");
    90         fgets(orderbuf,BUFFER_SIZE,stdin);
    91         orderbuf[strlen(orderbuf)-1]=0;//去掉获取到的回车符
    92         //printf("%s
    ",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连接成功
    ",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的文件
    ",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("
    %s:%d的文件传送完毕
    ",inet_ntoa(client_addr.sin_addr),ntohs(client_addr.sin_port));
    105                 }
    106                 else
    107                 {
    108                     printf("
    %s:%d的文件传送失败
    ",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
    ",stm->tm_hour,stm->tm_min,stm->tm_sec);
    132     return ;
    133 }

      两部分的代码都不难理解,关键的代码就是那个while循环了。这个就不给截图了,运行是没有问题。

      这篇杜鑫先生的回答很不错,可以看一下: http://www.zhihu.com/question/21591490

      本文地址: http://www.cnblogs.com/wunaozai/p/3886588.html

  • 相关阅读:
    读入输出优化模板
    HDU-2647 Reward(拓扑排序)
    HDU-2647 Reward(拓扑排序)
    HDU-2647 Reward(拓扑排序)
    HDU-2647 Reward(拓扑排序)
    Using KafkaBolt to write to a kafka topic
    Using KafkaBolt to write to a kafka topic
    Using KafkaBolt to write to a kafka topic
    Using KafkaBolt to write to a kafka topic
    getElementById() 获取指定ID的第一个元素
  • 原文地址:https://www.cnblogs.com/wunaozai/p/3886588.html
Copyright © 2011-2022 走看看