소켓 프로그래밍을 이용한 파일 전송

소켓 프로그래밍을 공부하던 도중 파일 전송을 해주는 프로그램을 만들어 보았습니다.

사용방법은 간단합니다.

==================

파일 보낼 곳

C:\> <프로그램명> <PORT> <보낼파일>

파일 받는 곳

C:\> <프로그램명> <IP> <PORT>

=================

위와 같이 하면 끝납니다.

소스코드를 보시겠습니다.

파일 보낼 곳.. 코드입니다.

#include <stdio.h> #include <stdlib.h> #include <string.h> #include <WinSock2.h> #include <process.h> #pragma comment(lib,"ws2_32.lib") void ErrorHandling(char* message); int main(int argc,char **argv) { FILE* file; WSADATA wsaData; SOCKADDR_IN servAddr; SOCKADDR_IN clntAddr; SOCKET hservSock; SOCKET hclntSock; int clntAddr_len; int File_len; char buf[30]; char File_Name[MAX_PATH]; if(argc != 3) { printf("Usage : %s <port> <File Path> \n",argv[0]); exit(1); } if(WSAStartup(MAKEWORD(2,2),&wsaData) != 0) ErrorHandling("WSAStartup() error! \n"); hservSock = socket(PF_INET,SOCK_STREAM,IPPROTO_TCP); if( hservSock == INVALID_SOCKET ) ErrorHandling("socket() error! \n"); memset(&servAddr,0,sizeof(servAddr)); servAddr.sin_family = AF_INET; servAddr.sin_addr.s_addr = htonl(INADDR_ANY); servAddr.sin_port = htons(atoi(argv[1])); if(bind(hservSock,(SOCKADDR*)&servAddr,sizeof(SOCKADDR_IN)) == SOCKET_ERROR) ErrorHandling("bind() error! \n"); if(listen(hservSock,5) == -1 ) ErrorHandling("listen error! \n"); clntAddr_len = sizeof(clntAddr); hclntSock = accept(hservSock,(SOCKADDR*)&clntAddr,&clntAddr_len); if(hclntSock == INVALID_SOCKET) ErrorHandling("accept() error! \n"); char* p = NULL; p = strrchr(argv[2],'\\'); strcpy(File_Name,p+1); send(hclntSock,File_Name,sizeof(File_Name),0); Sleep(100); file = fopen(argv[2],"rb"); if(file == NULL) ErrorHandling("fopen() error! \n"); while(1) { File_len = fread(buf,sizeof(char),30,file); send(hclntSock,buf,File_len,0); if(feof(file)) break; } if(shutdown(hclntSock,SD_SEND) == SOCKET_ERROR) ErrorHandling("shutdown() error!"); fclose(file); closesocket(hclntSock); closesocket(hservSock); WSACleanup(); return 0; } void ErrorHandling(char* message) { printf("%s \n",message); exit(1); }

52번 줄에서 먼저 파일 명을 전달 합니다. 그러면 받는 쪽에선 해당 파일을 명을 가지고 파일을 만듭니다.

그 후 64번째 줄에서 파일 내용을 byte단위로 전달하는 것입니다.

파일명 그대로 전달을 하고 싶었기에 이렇게 구현해 보았습니다.

이번에는 파일을 받는 곳 코드입니다.

#include <stdio.h> #include <stdlib.h> #include <string.h> #include <WinSock2.h> #pragma comment(lib,"ws2_32.lib") int main(int argc, char** argv) { FILE* file; WSADATA wsaData; SOCKET hSocket; SOCKADDR_IN servAddr; char File_Name[MAX_PATH]; char File_Path[MAX_PATH] = "C:\\"; char Data[30]; int File_Name_Len; int Data_Len; if(argc != 3) { printf("Usage : %s <IP> <port> \n",argv[0]); exit(1); } if(WSAStartup(MAKEWORD(2,2),&wsaData) != 0) printf("WSAStartup() error! \n"); hSocket = socket(PF_INET,SOCK_STREAM,IPPROTO_TCP); if(hSocket == INVALID_SOCKET) printf("socket() error! \n"); 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) printf("connect error! \n"); File_Name_Len = recv(hSocket,File_Name,MAX_PATH,0); strcat(File_Path,File_Name); file = fopen(File_Path,"wb"); Sleep(100); while((Data_Len = recv(hSocket,Data,30,0)) != 0) { fwrite(Data,sizeof(char),Data_Len,file); } fclose(file); closesocket(hSocket); WSACleanup(); return 0; }

그냥 파일 만들고 받는 내용을 파일에 넣어주고 있는게 전부입니다.

테스트 결과 잘 돌아가더군요.