socket 파일 전송 기능 구현

이 절 에서 우 리 는 socket 파일 전송 프로그램 을 완성 합 니 다. 이것 은 매우 실 용적 인 예 입 니 다.구현 할 기능 은 클 라 이언 트 가 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  
      #include  
      #include  
      #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!
"); 
       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  
     #include  
     #include  
     #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!
"); 
      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()     ，         。

socket 파일 전송 기능 구현

좋은 웹페이지 즐겨찾기