TCP 파일 수신 클라이언트<윈도우기반>

#include<stdio.h>

#include<stdlib.h>

#include<string.h>

#include<winsock2.h>

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

 

#define BUFSIZE 30

void ErrorHandling(char *message);

 

int main(int argc,char **argv)

{

           WSADATA wsaData;

           SOCKET hSocket;

 

 

           char buf[BUFSIZE];

           FILE* fp;

           SOCKADDR_IN servAddr;

           int len;

 

           if (argc!=3)

                     {

                     printf("Usage : %s <port>\n", argv[0]);

                     exit(1);

                     }

 

          

           fp = fopen("receive.dat", "w"); // 파일 생성

           if (fp == NULL)

                     ErrorHandling("File open error");

 

           if (WSAStartup(MAKEWORD(2, 2), &wsaData)!= 0)

                     ErrorHandling("WSAStartup() error");

          

/*서버 접속을 위한 소켓 생성*/

           hSocket=socket(PF_INET,SOCK_STREAM,0);

           if(hSocket == INVALID_SOCKET)

                     ErrorHandling("socket() error");

          

           memset(&servAddr,0,sizeof(servAddr));

           servAddr.sin_family=AF_INET;

           servAddr.sin_addr.s_addr=inet_addr(argv[1]);

           servAddr.sin_port=htons(atoi(argv[2]));

 

           if ( connect(hSocket, (SOCKADDR*) &servAddr,sizeof(servAddr))==SOCKET_ERROR)

                     ErrorHandling("connect() error");

 

/*데이터를 전송받아서 파일에 저장한다*/

           while((len=recv(hSocket, buf, BUFSIZE, 0)) !=0)

                     {

                                fwrite(buf,sizeof(char), len, fp);

                     }

/*전송해 준 것에 대한 감사의 메시지 전달*/

          

           send(hSocket, "Thank You\n",10,0);

 

           fclose(fp);

           closesocket(hSocket);

           WSACleanup();

           return 0;

}

 

 

void ErrorHandling(char *message)

{

           fputs(message, stderr);

           fputc('\n',stderr);

           exit(1);

}

파일 전송 서버와 파일 수신 클라이언트 2개가 한쌍임.