C++中使用TCP傳文件

在兩個文件中都定義文件頭和用到的宏：

#define MAX_SIZE 10
#define ONE_PAGE 4096
struct FileHead
{
    char str[260];
    int size;
};

在客戶端發送接收階段：

    //1.發送文件頭
    char path[260] = {0};
    cout<<"請輸入文件路徑"<<endl;
    cin>>path;
    //截取文件名
    char* ptemp = path;
    while(*ptemp++ != '\0');
    while(*(--ptemp) != '\\');
    ptemp++;

    fstream fs;
    fs.open(path,fstream::in|fstream::binary);
    fs.seekg(0,fstream::end);//以最后的位置為基准不偏移
    int nlen = fs.tellg();//取得文件大小
    fs.seekg(0,fstream::beg);

    FileHead fh;
    fh.size = nlen;
    memcpy(fh.str,ptemp,MAX_PATH);
    nlen = send(sockConnect ,(char*)&fh,sizeof(fh),0);
    //2,.如果接受到的內容為是
    char  szResult[MAX_SIZE] = {0};
    recv(sockConnect,szResult,sizeof(szResult),0);
    char szBuf[ONE_PAGE] ={0};
    if(0 == strcmp(szResult,"是"))
    {
        //讀文件
        while(!fs.eof())
        {
            fs.read(szBuf,ONE_PAGE);
            int len = fs.gcount();
            //if(len == 0 ) break;
            send(sockConnect,szBuf,len,0);
        }
    }
    
    //3.關閉文件流
    fs.close();

在服務器端，接收和發送階段

    char str[1024] = {0};
    int nlen;
    nlen = recv(sockWaiter,(char*)&fh,sizeof(fh),0);
    cout<<"是否要接受文件"<<endl;
    cin>>str;
    send(sockWaiter ,str,sizeof(str),0);
    char szPath[MAX_SIZE] = {0};
    cout<<"請輸入要存儲的路徑"<<endl;
    cin>>szPath;
    char szPathName[MAX_SIZE] = {0};
    sprintf(szPathName,"%s%s",szPath,fh.str);//拼接路徑和文件名
    cout<<szPathName<<endl;
    fstream fs;
    fs.open(szPathName,fstream::out|fstream::binary|fstream::trunc);//以空文件的形式打開
    int FileSize = fh.size;
    int len;
    char content[ONE_PAGE] = {0};
    while(FileSize)
    {
        len = recv(sockWaiter,content,ONE_PAGE,0);
        if(len > 0)
        {
            fs.write(content,len);
            FileSize -= len;
        }
    }
    fs.close();