windows下基於TCP協議的大文件傳輸程序

目標

（1）大文件的傳輸。對於比較大的文件，應該進行分包操作，以防止佔用過多的內存，導致文件發送
失敗，實驗中每次最多傳輸1024個字符；
（2）用戶根據參數輸入選擇傳輸的文件和傳輸位置；
（3）發送端和接收端分別顯示文件傳輸相應的信息，包括：對方IP地址，當前已寫（讀）文件長度等；
（4）對於Ctrl+C命令以及服務器提前關閉的特殊情況給出響應（如：顯示信息）

服務器代碼

#define  _WINSOCK_DEPRECATED_NO_WARNINGS  
#define _CRT_SECURE_NO_WARNINGS
#include <winsock2.h>   
#include<iostream>  
#include<string> 
#include<fstream>
#include<sstream>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

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

//using namespace std;  //直接用std好像會出現bug  
using std::cout;

//封裝套接字綁定
int make_socket(const char* address,int port)
{
    //創建套接字    
    SOCKET Sock = socket(AF_INET, SOCK_STREAM, 0);

    //綁定套接字    
    sockaddr_in sockAddr;
    memset(&sockAddr, 0, sizeof(sockAddr));  //每個字節都用0填充    
    sockAddr.sin_family = AF_INET;  //使用IPv4地址    
    sockAddr.sin_addr.s_addr = inet_addr(address);  //回送IP地址   
    sockAddr.sin_port = htons(1234);  //端口    
    bind(Sock, (SOCKADDR*)&sockAddr, sizeof(SOCKADDR));
    return Sock;
}

//通過套接字獲取IP、Port等地址信息
bool GetAddressBySocket(SOCKET m_socket, SOCKADDR_IN &m_address)
{
    memset(&m_address, 0, sizeof(m_address));
    int nAddrLen = sizeof(m_address);
    //根據套接字獲取地址信息
    if (::getpeername(m_socket, (SOCKADDR*)&m_address, &nAddrLen) != 0)
    {
        printf("獲取IP地址失敗");
        return false;
    }
    //讀取IP和Port
    return true;
}

int main() 
{
    //初始化dll  
    WSADATA wsaData;
    WSAStartup(MAKEWORD(2, 2), &wsaData);


    int servSock = make_socket("127.0.0.1", 1234);

    //服務器socket進入監聽狀態    
    listen(servSock, 200);

    //接收客戶端請求    
    SOCKADDR clntAddr;
    int nSize = sizeof(SOCKADDR);
    char buffer[BUF_SIZE] = { 0 };  //緩衝區    

    while (1) 
    {
        SOCKET clntSock = accept(servSock,&clntAddr, &nSize);
        if (clntSock<=0)
        {
            cout << "連接客戶端有誤" << std::endl;
            continue;
        }
        else
        {
            SOCKADDR_IN temp;
            GetAddressBySocket(clntSock, temp);
            cout << "客戶端:" << ::inet_ntoa(temp.sin_addr) << "連接了服務器" << std::endl;
        }

        std::ofstream out("結果.txt", std::ios::app);

        int recv_length = recv(clntSock, buffer, BUF_SIZE, 0);
        int all_lenth = 0;
        while (recv_length>0)
        {
            all_lenth += recv_length;
            out << buffer;
            memset(buffer, 0, BUF_SIZE);  //重置緩衝區   
            cout << "已接受" << all_lenth <<"字節"<< std::endl;
            recv_length = recv(clntSock, buffer, BUF_SIZE, 0);
        }
        out.close();
    }
    //關閉套接字    
    closesocket(servSock);
    //終止 DLL 的使用    
    WSACleanup();
    return 0;
}

客戶端代碼

#define  _WINSOCK_DEPRECATED_NO_WARNINGS  
#define _CRT_SECURE_NO_WARNINGS
#include <stdio.h>    
#include<iostream>  
#include <stdlib.h>    
#include <WinSock2.h> 
#include<fstream>
#include<sstream>

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

using std::cout;
using std::cin;
using std::endl;
//TODO 解決黏包問題

//通過套接字獲取IP、Port等地址信息
bool GetAddressBySocket(SOCKET m_socket, SOCKADDR_IN &m_address)
{
    memset(&m_address, 0, sizeof(m_address));
    int nAddrLen = sizeof(m_address);
    //根據套接字獲取地址信息
    if (::getpeername(m_socket, (SOCKADDR*)&m_address, &nAddrLen) != 0)
    {
        printf("獲取IP地址失敗");
        return false;
    }
    //讀取IP和Port
    return true;
}

//封裝套接字綁定
int make_socket(const char* address, int port)
{
    //創建套接字    
    SOCKET Sock = socket(AF_INET, SOCK_STREAM, 0);

    //綁定套接字    
    sockaddr_in sockAddr;
    memset(&sockAddr, 0, sizeof(sockAddr));  //每個字節都用0填充    
    sockAddr.sin_family = AF_INET;  //使用IPv4地址    
    sockAddr.sin_addr.s_addr = inet_addr(address);  //回送IP地址   
    sockAddr.sin_port = htons(1234);  //端口    
    bind(Sock, (SOCKADDR*)&sockAddr, sizeof(SOCKADDR));
    return Sock;
}

int main() {
    //初始化DLL    
    WSADATA wsaData;
    WSAStartup(MAKEWORD(2, 2), &wsaData);

    //向服務器發起請求    
    sockaddr_in sockAddr;
    memset(&sockAddr, 0, sizeof(sockAddr));  //每個字節都用0填充    
    sockAddr.sin_family = AF_INET;
    sockAddr.sin_addr.s_addr = inet_addr("127.0.0.1");      //回送地址  
    sockAddr.sin_port = htons(1234);


    char bufSend[BUF_SIZE] = { 0 };
    char bufRecv[BUF_SIZE] = { 0 };

    while (true)
    {
        //創建套接字    
        SOCKET sock = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
        connect(sock, (SOCKADDR*)&sockAddr, sizeof(SOCKADDR));

        //獲取服務器地址端口相關
        SOCKADDR_IN address;
        GetAddressBySocket(sock, address);
        cout << "IP: " << ::inet_ntoa(address.sin_addr) << "  PORT: " << ntohs(address.sin_port) << endl;

        //獲取文件位置
        std::ifstream file;
        do
        {
            cout << "請輸入文件名" << std::endl;
            std::string filename = "";
            cin >> filename;
            file.open(filename, std::ifstream::binary);
        } while (!file.is_open());


        //傳輸文件
        //發送完畢標誌
        bool flag = true;
        for (int  i = 0; !file.eof();)
        {
            char buf[1024] = { '\0' };
            memset(buf, 0, sizeof(buf));
            file.read(buf, sizeof(buf) - 1);

            int n = file.gcount();
            int send_char_num= send(sock, buf, n, 0);
            if (send_char_num==0)
            {
                cout << "服務器已連接斷開" << endl;
                flag = false;
                break;
            }
            if (send_char_num<0)
            {
                cout << "發生錯誤，我也不知道是啥錯誤^_^" << endl;
                flag = false;
                break;
            }
            i += send_char_num;
            cout << "已發送：" << i << "字節" << endl;
        }
        if (flag)
        {
            cout << "文件已發送完畢" << endl;
        }
        closesocket(sock);  //關閉套接字  
    }
    WSACleanup();  //終止使用 DLL    
    return 0;
}