Socket網絡編程--小小網盤程序(1)

　　這個系列是准備講基於Linux Socket進行文件傳輸。簡單的文件傳輸就是客戶端可以上傳文件，可以從服務器端下載文件。就這么兩個功能如果再加上身份驗證，就成了FTP服務器了，如果對用戶的操作再加上一些功能(如分享)，就可以作為一個最簡單的網盤了。想想是不是有點小激動啊。

　　我這一小節就不講那么高級的東西，就先了解文件怎么傳輸，我們以前的聊天程序傳輸數據都是一次發送就完成本次的發送，因為一個sendBuf是足夠的。但是對於二進制文件來說，文件的大小就不一定了，有可能很大，所以我們的Buf是不知道要多少的。所以要傳輸大文件，就要分多次傳輸，然后在目的地進行合並。這樣就可以實現大文件傳輸了。傳輸的方法有兩種，一種是串行一種是並行。我接下來要實現的代碼是使用串行傳輸的，因為比較簡單，而並行傳輸，比較復雜，要傳輸幾個控制信號，保證數據合並重整后不會出錯。並行可以使用多進(線)程。優點是可以同時傳輸多個文件，串行是只能一次傳輸一個文件。我們看一下網盤，大多是可以下載多個文件(多任務)是吧？應該用的就是並行了。如果只是從原理上講串行的傳輸速度肯定是比並行的快，因為並行還要控制信號要傳，而且還要重新合並整合成一個文件。既然這樣那為什么所有的網盤都是使用多任務下載呢？(多個任務同時下載，每個任務同時還有多個端口在接收數據)。是因為我們所處的網絡環境問題。我們的網絡環境並不是那么的穩定，如果只是用一個端口進行串行傳輸，那么如果網絡突然中斷或者什么原因，導致服務器與客戶端暫時連接斷開。那么我們之前傳的數據就要重傳了。這就是為什么今天下載個任務到80%，然后明天還可以接着下載了。如果是一次性串行傳輸就不行了。你就會問，那在串行傳輸再加個控制信號不就行了？那這樣的話，與並行區別就不大了。

　　廢話說了一大堆，到了真正要寫代碼的時候我還是使用串行來寫，因為比較容易實現。(求原諒!)

　　實現客戶端向服務器發送一個指定的二進制文件

　　client.cpp (為什么用cpp了，因為上次用c，然后代碼越寫越多，結果很多變量都寫在開頭，找起來不是很方便，所以用cpp了，哎，用c++就是為了用這個，會不會被罵？)

#include <netinet/in.h> // sockaddr_in
#include <sys/types.h> //socket
#include <sys/socket.h> //socket
#include <netdb.h> //gethostbyname
#include <unistd.h> //close
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <time.h>
#include <arpa/inet.h> //inet_addr

#define SERVVER_PORT 12138
#define LISTEN_QUEUE 20
#define BUFFER_SIZE 1024

struct Addr
{
    char host[64];
    int port;
};

int file_push(struct Addr addr,char *filenames)
{
    struct sockaddr_in servAddr;
    struct hostent * host;
    int sockfd;
    FILE *fp;

    host=gethostbyname(addr.host);
    servAddr.sin_family=AF_INET;
    servAddr.sin_addr=*((struct in_addr *)host->h_addr);
    //servAddr.sin_addr.s_addr=inet_addr("127.0.0.1");
    servAddr.sin_port=htons(addr.port);
    if(host==NULL)
    {
        perror("獲取IP地址失敗");
        exit(-1);
    }
    if((sockfd=socket(AF_INET,SOCK_STREAM,0))==-1)
    {
        perror("socket創建失敗");
        exit(-1);
    }

    if(connect(sockfd,(struct sockaddr *)&servAddr,sizeof(struct sockaddr_in))== -1)
    {
        perror("connect 失敗");
        exit(-1);
    }

    //打開文件
    if((fp=fopen(filenames,"rb"))==NULL)
    {
        perror("文件打開失敗");
        exit(-1);
    }
    char buffer[BUFFER_SIZE];
    bzero(buffer,BUFFER_SIZE);
    printf("正在傳輸文件");
    int len=0;
    //不斷的讀取文件直到文件結束
    while((len=fread(buffer,1,BUFFER_SIZE,fp))>0)
    {
        if(send(sockfd,buffer,len,0)<0)
        {
            perror("發送數據失敗");
            exit(-1);
        }
        bzero(buffer,BUFFER_SIZE);
        printf(".");//1K打印一個點//如果要實現百分比，就要計算文件大小，然后再處理即可
    }

    fclose(fp);//關閉文件流
    close(sockfd);//關閉socket連接

    return 0;
}

int main(int argc,char *argv[])
{
    char orderbuf[BUFFER_SIZE];
    char orderch[BUFFER_SIZE];
    struct Addr addr;

    strcpy(addr.host,argv[1]);
    addr.port=atoi(argv[2]);
    while(1)
    {
        printf("\n請輸入文件名:");
        fgets(orderbuf,BUFFER_SIZE,stdin);
        orderbuf[strlen(orderbuf)-1]=0;//去掉獲取到的回車符
        //printf("%s\n",orderbuf);
        file_push(addr,orderbuf);
    }

    return 0;
}

　　server.cpp

#include <netinet/in.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <stdlib.h>
#include <time.h>
#include <string.h>
#include <unistd.h>
#include <stdio.h>
#include <arpa/inet.h> //inet_ntoa

#define SERVER_PORT 12138
#define LISTEN_QUEUE 20
#define BUFFER_SIZE 1024

void print_time(char *ch);//打印時間

int main(int argc,char *argv[])
{
    struct sockaddr_in server_addr;
    bzero(&server_addr,sizeof(server_addr));
    server_addr.sin_family=AF_INET;
    server_addr.sin_addr.s_addr=htons(INADDR_ANY);
    server_addr.sin_port=htons(SERVER_PORT);

    //創建套接字
    int sockfd=socket(AF_INET,SOCK_STREAM,0);
    if(sockfd<0)
    {
        perror("創建套接字失敗");
        exit(-1);
    }

    if(bind(sockfd,(struct sockaddr *)&server_addr,sizeof(server_addr))==-1)
    {
        perror("bind 失敗");
        exit(-1);
    }

    if(listen(sockfd,LISTEN_QUEUE))
    {
        perror("listen 失敗");
        exit(-1);
    }

    while(1)
    {
        pid_t pid;
        struct sockaddr_in client_addr;
        socklen_t length=sizeof(client_addr);
        int clientfd=accept(sockfd,(struct sockaddr *)&client_addr,&length);
        if(clientfd==-1)
        {
            perror("accept 失敗");
            continue;
        }
        else
        {
            printf("客戶端%s:%d連接成功\n",inet_ntoa(client_addr.sin_addr),ntohs(client_addr.sin_port));
            pid=fork();
            if(pid<0)
            {
                perror("創建進程失敗");
            }
            else if(pid==0)/*child*/
            {
                char buffer[BUFFER_SIZE];
                int data_len;
                FILE * fp=NULL;
                bzero(buffer,BUFFER_SIZE);
                if((fp=fopen("data","wb"))==NULL)
                {
                    perror("文件打開失敗");
                    exit(-1);
                }
                //循環接收數據
                int size=0;//表示有多少個塊
                while(data_len=recv(clientfd,buffer,BUFFER_SIZE,0))//data_len為0時結束，是因為當客戶端沒有再發送數據過來時是接收0的，也表示該文件傳輸完畢了。  78                 {
                    if(data_len<0)
                    {
                        perror("接收數據錯誤");
                        exit(-1);
                    }
                    size++;
                    if(size==1)
                    {
                        printf("正在接收來自%s:%d的文件\n",inet_ntoa(client_addr.sin_addr),ntohs(client_addr.sin_port));
                    }
                    else
                    {
                        printf(".");
                    }
                    //向文件中寫入
                    int write_len=fwrite(buffer,sizeof(char),data_len,fp);//向文件中寫入，默認文件打開時會有一個文件指針進行寫入。如果是並行傳輸就要修改這個文件指針了，還是有點麻煩的，如果是串行的，我們都不用管了，多方便。  95                     if(write_len>data_len)
                    {
                        perror("寫入數據錯誤");
                        exit(-1);
                    }
                    bzero(buffer,BUFFER_SIZE);
                }
                if(size>0)
                {
                    printf("\n%s:%d的文件傳送完畢\n",inet_ntoa(client_addr.sin_addr),ntohs(client_addr.sin_port));
                }
                else
                {
                    printf("\n%s:%d的文件傳送失敗\n",inet_ntoa(client_addr.sin_addr),ntohs(client_addr.sin_port));//如果傳過來的文件大小為0，也是會出現這個錯誤 109                 }
                fclose(fp);
                //rename("data","asdf");//這里可以修改文件的名字。保存到服務器的話可以隨便改個名字，防止文件名重復
                exit(0);
            }
            else /*pather*/
            {
                ;
            }
        }
        close(clientfd);
    }

    return 0;
}

void print_time(char *ch)
{
    time_t now;
    struct tm * stm;
    time(&now);
    stm=localtime(&now);
    sprintf(ch,"%02d:%02d:%02d\n",stm->tm_hour,stm->tm_min,stm->tm_sec);
    return ;
}

　　兩部分的代碼都不難理解，關鍵的代碼就是那個while循環了。這個就不給截圖了，運行是沒有問題。

 

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

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