socket 程式設計實作檔案傳輸功能！強無敵，網絡通訊的必備知識儲備！

 socket 檔案傳輸程式，這是一個非常實用的例子。

要實作的功能為：client 從 server 下載下傳一個檔案并儲存到本地。

編寫這個程式需要注意兩個問題：

========

1) 檔案大小不确定，有可能比緩沖區大很多，調用一次 write()/send() 函數不能完成檔案内容的發送。接收資料時也會遇到同樣的情況。

要解決這個問題，可以使用 while 循環，例如：

//Server 代碼

int nCount;

while( (nCount = fread(buffer, 1, BUF_SIZE, fp)) > 0 ){

    send(sock, buffer, nCount, 0);

}

//Client 代碼

int nCount;

while( (nCount = recv(clntSock, buffer, BUF_SIZE, 0)) > 0 ){

    fwrite(buffer, nCount, 1, fp);

}      

對于 Server 端的代碼，當讀取到檔案末尾，fread() 會傳回 0，結束循環。

對于 Client 端代碼，有一個關鍵的問題，就是檔案傳輸完畢後讓 recv() 傳回 0，結束 while 循環。

注意：讀取完緩沖區中的資料 recv() 并不會傳回 0，而是被阻塞，直到緩沖區中再次有資料。

2) Client 端如何判斷檔案接收完畢，也就是上面提到的問題——何時結束 while 循環。

最簡單的結束 while 循環的方法當然是檔案接收完畢後讓 recv() 函數傳回 0，那麼，如何讓 recv() 傳回 0 呢？recv() 傳回 0 的唯一時機就是收到FIN包時。

FIN 包表示資料傳輸完畢，計算機收到 FIN 包後就知道對方不會再向自己傳輸資料，當調用 read()/recv() 函數時，如果緩沖區中沒有資料，就會傳回 0，表示讀到了”socket檔案的末尾“。

這裡我們調用 shutdown() 來發送FIN包：server 端直接調用 close()/closesocket() 會使輸出緩沖區中的資料失效，檔案内容很有可能沒有傳輸完畢連接配接就斷開了，而調用 shutdown() 會等待輸出緩沖區中的資料傳輸完畢。

以Windows為例示範檔案傳輸功能，Linux與此類似，不再贅述。請看下面完整的代碼。

伺服器端 server.cpp：

#include <stdio.h>

#include <stdlib.h>

#include <winsock2.h>

#pragma comment (lib, "ws2_32.lib")  //加載 ws2_32.dll

#define BUF_SIZE 1024

int main(){

    //先檢查檔案是否存在

    char *filename = "D:\\send.avi";  //檔案名

    FILE *fp = fopen(filename, "rb");  //以二進制方式打開檔案

    if(fp == NULL){

        printf("Cannot open file, press any key to exit!\n");

        system("pause");

        exit(0);

    }

    WSADATA wsaData;

    WSAStartup( MAKEWORD(2, 2), &wsaData);

    SOCKET servSock = socket(AF_INET, SOCK_STREAM, 0);

    sockaddr_in sockAddr;

    memset(&sockAddr, 0, sizeof(sockAddr));

    sockAddr.sin_family = PF_INET;

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

    sockAddr.sin_port = htons(1234);

    bind(servSock, (SOCKADDR*)&sockAddr, sizeof(SOCKADDR));

    listen(servSock, 20);

    SOCKADDR clntAddr;

    int nSize = sizeof(SOCKADDR);

    SOCKET clntSock = accept(servSock, (SOCKADDR*)&clntAddr, &nSize);

    //循環發送資料，直到檔案結尾

    char buffer[BUF_SIZE] = {0};  //緩沖區

    int nCount;

    while( (nCount = fread(buffer, 1, BUF_SIZE, fp)) > 0 ){

        send(clntSock, buffer, nCount, 0);

    }

    shutdown(clntSock, SD_SEND);  //檔案讀取完畢，斷開輸出流，向用戶端發送FIN包

    recv(clntSock, buffer, BUF_SIZE, 0);  //阻塞，等待用戶端接收完畢

    fclose(fp);

    closesocket(clntSock);

    closesocket(servSock);

    WSACleanup();

    system("pause");

    return 0;

}      

用戶端代碼：

#include <stdio.h>

#include <stdlib.h>

#include <WinSock2.h>

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

#define BUF_SIZE 1024

int main(){

    //先輸入檔案名，看檔案是否能建立成功

    char filename[100] = {0};  //檔案名

    printf("Input filename to save: ");

    gets(filename);

    FILE *fp = fopen(filename, "wb");  //以二進制方式打開（建立）檔案

    if(fp == NULL){

        printf("Cannot open file, press any key to exit!\n");

        system("pause");

        exit(0);

    }

    WSADATA wsaData;

    WSAStartup(MAKEWORD(2, 2), &wsaData);

    SOCKET sock = socket(PF_INET, SOCK_STREAM, IPPROTO_TCP);

    sockaddr_in sockAddr;

    memset(&sockAddr, 0, sizeof(sockAddr));

    sockAddr.sin_family = PF_INET;

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

    sockAddr.sin_port = htons(1234);

    connect(sock, (SOCKADDR*)&sockAddr, sizeof(SOCKADDR));

    //循環接收資料，直到檔案傳輸完畢

    char buffer[BUF_SIZE] = {0};  //檔案緩沖區

    int nCount;

    while( (nCount = recv(sock, buffer, BUF_SIZE, 0)) > 0 ){

        fwrite(buffer, nCount, 1, fp);

    }

    puts("File transfer success!");

    //檔案接收完畢後直接關閉套接字，無需調用shutdown()

    fclose(fp);

    closesocket(sock);

    WSACleanup();

    system("pause");

    return 0;

}
      

在D盤中準備好send.avi檔案，先運作 server，再運作 client：

Input filename to save: D:\\recv.avi↙

//稍等片刻後

File transfer success!

打開D盤就可以看到 recv.avi，大小和 send.avi 相同，可以正常播放。

注意 server.cpp 第42行代碼，recv() 并沒有接收到 client 端的資料，當 client 端調用 closesocket() 後，server 端會收到FIN包，recv() 就會傳回，後面的代碼繼續執行。

如果你還想更深入的學習以及其他知識，不管你是轉行也好，初學也罷，進階也可~

【值得關注】我的 程式設計學習交流俱樂部 ！【點選進入】

C語言入門資料：

C語言必讀書籍：