閒暇時寫的一個基於SOCKET的文件傳輸程序

//Server----Receive
#include <cstdlib> 
#include <iostream> 
#include "CSocket.h" 
#define SIZE (1024*1024) 

struct pack_head 
{   
    int nfileSize; 
    char szfileName[32];     
}; 

int main(int argc, char *argv[]) 
{ 
    CSocket::InitLib(); 
    CSocket sock, trsock; 
    sock.InitSock(); 
    sock.Bind(); 
    sock.Listen();
     
Redo:      
    printf("Now wait for connection........\n"); 
    sock.Accept(trsock); 
    printf("Connected now........\n"); 
    
    pack_head strupk; 
    memset(&strupk, 0, sizeof(pack_head)); 
    trsock.Recv((char*)&strupk, sizeof(pack_head)); 
    printf("%s\n%d\n", strupk.szfileName, strupk.nfileSize); 
    assert(strupk.nfileSize > 0); 
    char path[32] = "E:\\"; strcat(path, strupk.szfileName); 
    
    FILE *fp = fopen(path, "wb"); 
    assert(fp != NULL); 
    
    int FileSize = strupk.nfileSize; 
    int nXiabiao = 0; 
    int nBytesRecvd = 0;
    char *pBuff = (char*)malloc(SIZE); 
    memset(pBuff, 0, SIZE);     
    printf("Receiving...................................");
    
    while (nBytesRecvd < FileSize) 
    { 
         int k = trsock.Recv(pBuff+nXiabiao, SIZE); 
         
         if (k <= 0) 
         { 
              trsock.Close();
              fclose(fp);
              goto Redo;
         } 
         
         nXiabiao += k; 
         nBytesRecvd += k;
         
         int i = int(((double)nBytesRecvd/FileSize)*100);
         printf("%d%%", i);
         if (i <= 9)
              printf("\b\b");
         else if (i != 100)
              printf("\b\b\b");
         else
              printf("\n");
                    
         if (nXiabiao == SIZE || nBytesRecvd == FileSize) 
         { 
              fflush(fp);
              fwrite(pBuff, 1, nXiabiao, fp);   
              memset(pBuff, 0, SIZE); 
              nXiabiao = 0;               
         } 
    }   
    
    fclose(fp);
    free(pBuff);
    system("PAUSE"); 
    return EXIT_SUCCESS; 
}




//Client------Send  
#include <cstdlib>  
#include <iostream>  
#include "CSocket.h"  
#define SIZE (1024*1024)   
  
struct pack_head   
{     
    int nfileSize;   
    char szfileName[32];       
};   
       
int main(int argc, char *argv[])  
{  
    CSocket::InitLib();  
    CSocket c;  
    c.InitSock();  
      
    while (c.Connect() == FALSE)  
    {  
        Sleep(1000*2);;              
    }  
      
    printf("Connected !\n");  
      
    char fpath[128] = "";  
    char fname[20] = "";  
  
    printf("Input file path:");        
    gets(fpath);  
    printf("Input file name:");  
    gets(fname);  
      
    FILE* fp = fopen(fpath, "rb");     
    assert(fp != NULL);     
      
    fseek(fp, 0, SEEK_END);  
    int filesize = ftell(fp);  
    fseek(fp, 0, SEEK_SET);  
      
    printf("filesize:\t%d\n", filesize);  
      
    pack_head s;   
    memset(&s, 0, sizeof(pack_head));  
    memcpy(s.szfileName, fname, sizeof(fname));  
    s.nfileSize = filesize;  
      
    c.Send((char*)&s, sizeof(s));   
    char *pBuff = (char*)malloc(SIZE);   
    memset(pBuff, 0, SIZE);   
    int nBytesSend = 0;  
    printf("Sending...................................");  
      
    while(!feof(fp))  
    {                   
         int read = fread(pBuff, 1, SIZE, fp);   
         {  
             int k = c.Send(pBuff, read);  
             if (k <= 0)  
             {  
                 c.Close();  
                 fclose(fp);  
                 break;  
             }   
             else  
             {      
                 nBytesSend += k;   
                 int i = int(((double)nBytesSend/filesize)*100);  
                 printf("%d%%", i);  
                 if (i <= 9)  
                     printf("\b\b");  
                 else if (i != 100)  
                     printf("\b\b\b");  
                 else  
                     printf("\n");  
             }      
                               
         }  
         memset(pBuff, 0, SIZE);  
         Sleep(10);            
    }          
    
    fclose(fp);  
    free(pBuff);  
    system("PAUSE");  
    return EXIT_SUCCESS;  

}