windows环境下使用C++&Socket实现文件传输

server

#include <stdio.h>
#include <iostream>
#include <cstring>
#include <fstream>
#include <winsock2.h>

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

using namespace std;

SOCKET m_Client; 


void SendFile();


int main(int argc, char* argv[])  
{  
    //初始化WSA  
    WORD sockVersion = MAKEWORD(2,2);  
    WSADATA wsaData;  
    if(WSAStartup(sockVersion, &wsaData)!=0)  
    {  
        return 0;  
    }  
  
    //创建套接字  
    SOCKET slisten = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);  
    if(slisten == INVALID_SOCKET)  
    {  
        printf("socket error !");  
        return 0;  
    }  
  
    //绑定IP和端口  
    sockaddr_in sin;  
    sin.sin_family = AF_INET;  
    sin.sin_port = htons(8888);  
    sin.sin_addr.S_un.S_addr = INADDR_ANY;   
    if(bind(slisten, (LPSOCKADDR)&sin, sizeof(sin)) == SOCKET_ERROR)  
    {  
        printf("bind error !");  
    }  
  
    //开始监听  
    if(listen(slisten, 5) == SOCKET_ERROR)  
    {  
        printf("listen error !");  
        return 0;  
    }  
  
    //循环接收数据  
    sockaddr_in remoteAddr;  
    int nAddrlen = sizeof(remoteAddr);  
    char revData[255];   
    while (true)  
    {  
        printf("等待连接...\n");  
        m_Client = accept(slisten, (SOCKADDR *)&remoteAddr, &nAddrlen);  
        if(m_Client == INVALID_SOCKET)  
        {  
            printf("accept error !");  
            continue;  
        }  
        printf("接受到一个连接：%s \r\n", inet_ntoa(remoteAddr.sin_addr));  
        SendFile();
        closesocket(m_Client);  
    }  
      
    closesocket(slisten);  
    WSACleanup();  
    return 0;  
} 
void SendFile(){
    int haveSend = 0;
    const int bufferSize = 1024;
    char buffer[bufferSize] = {0};
    int readLen = 0;
    string srcFileName = "in";
    ifstream srcFile;
    srcFile.open(srcFileName.c_str(),ios::binary);
    if(!srcFile){
        return;
    }
    while(!srcFile.eof()){
        srcFile.read(buffer,bufferSize);
        readLen = srcFile.gcount();
        send(m_Client,buffer,readLen,0);
        haveSend += readLen;    
    }
    srcFile.close();
    cout<<"send: "<<haveSend<<"B"<<endl;
}

client

#include <stdio.h>
#include <iostream>
#include <cstring>
#include <fstream>
#include <winsock2.h>

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

using namespace std;
 
SOCKET m_Client; 

void RecvFile();

int main()
{
    WORD sockVersion = MAKEWORD(2, 2);
    WSADATA data;
    if(WSAStartup(sockVersion, &data)!=0)
    {
        return 0;
    }
    //while(true){
        m_Client = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
        if(m_Client == INVALID_SOCKET)
        {
            printf("invalid socket!");
            return 0;
        }
        
        sockaddr_in serAddr;
        serAddr.sin_family = AF_INET;
        serAddr.sin_port = htons(8888);
        serAddr.sin_addr.S_un.S_addr = inet_addr("127.0.0.1");
        if(connect(m_Client, (sockaddr *)&serAddr, sizeof(serAddr)) == SOCKET_ERROR)
        {  //连接失败 
            printf("connect error !");
            closesocket(m_Client);
            return 0;
        }
        RecvFile();
        closesocket(m_Client);
    //}
    
    
    WSACleanup();
    return 0;
    
}

void RecvFile(){
    cout<<"start recv!"<<endl;
    const int bufferSize = 1024;
    char buffer[bufferSize] = {0};
    int readLen = 0;
    string desFileName = "new_in";
    ofstream desFile;
    desFile.open(desFileName.c_str(), ios::binary);
    if (!desFile)
    {
        return ;
    }
    do 
    {
        readLen = recv(m_Client,buffer,bufferSize, 0);
        if (readLen == 0)
        {
            break;
        }
        else
        {
            desFile.write(buffer, readLen);
        }
    } while(true);
    desFile.close();
}