프로그래밍 개요

TCP로 서버 클라이언트 통신 부분을 구현해 본 소스이다. TCP는 Connection Oriented Protocol 로써 먼저 연결을 한 뒤, 메시지를 송·수신 하게 된다. 말이 귀찮으니 그냥 코드 보면서 알아보자.

// winsock2.h를 include 해주어야 합니다. 또한 [Project>Setting>Link] 에서 Object/Library 에 ws2_32.lib 를 써주어야 됩니다. #include <winsock2.h> #include <stdio.h> #include <stdlib.h> #define DEFAULT_PORT 5150 #define DEFAULT_BUFFER 4096 int iPort = DEFAULT_PORT; BOOL bInterface = FALSE, BOOL bRecvOnly = FALSE; char szAddress[128]; void CheckArgs(int, char**); // main에 들어오는 인자를 처리하는 부분 void usage(void);                // 잘못된 인자일 경우 사용방법을 프린트하고 종료 DWORD WINAPI ClientThread(LPVOID); // 데이타 송수신 하는 쓰레드 int main(int argc, char **argv) {     WSADATA wsd;     SOCKET sListen, sClient;     int nAddrSize;     HANDLE hThread;     DWORD dwThreadId;     struct sockaddr_in local, client;     CheckArgs(argc, argv);     // WSAStartup() 부분은 winsock2 라이브러리를를 현재 컴퓨터에서 사용할수 있는지의 여부를 확인 하는 부분입니다. MAKEWORD(2, 2)는 버전 번호 입니다.     if(WSAStartup(MAKEWORD(2, 2), &wsd) != 0) {         printf("Winsock를 로드하는데 실패했습니다\n");         return 1;     }     // 클라이언트의 접속을 대기하는 소켓을 생성한다.     sListen = socket(AF_INET, SOCK_STREAM, IPPROTO_IP);     if(sListen == SOCKET_ERROR) {         printf("Socket 생성 실패, Error번호:%d\n", WSAGetLastError());         return 1;     }     if(bInterface) {         // inet_addr() 함수는 10진수로 4자리로된 IP주소를 Bin-인디언의 8byte방식으로 바꾼다. 그러나 잘못된 IP주소가 있는 경우는 INADDR_NONE 을 리턴하게 된다. 그리고 따로 IP주소를 설정하지 않으면 INADDR_ANY 를 주소에 넣게 되며 이는 현재 컴퓨터에서 쓸수 있는 모든 IP를 쓴다는 뜻이다. (현재 설정된 포트로 들어오는 데이타는 전부 받게 된다.)         local.sin_addr.S_un.S_addr = inet_addr(szAddress);         if(local.sin_addr.S_un.S_addr == INADDR_NONE)             usage();     }     else         local.sin_addr.S_un.S_addr = htonl(INADDR_ANY);     local.sin_family = AF_INET; // AF_INET 는IPV4 어드레스 체계에서 쓴다.     // 포트 번호 역시 Big-인디언으로 바꿔서 들어가야 하므로 htons함수를 사용해서 변환했다.     local.sin_port = htons(iPort);     if(bind(sListen, (struct sockaddr *)&local, sizeof(local)) == SOCKET_ERROR) {         printf("bind 실패, Error번호:%d\n", WSAGetLastError());         return 1;     }     // bind 를 하고 난후 listen 을 함으로써, 클라이언트의 접속을 대기하게 된다.     listen(sListen, 8);     while(1) {         nAddrSize = sizeof(client);         sClient = accept(sListen, (struct sockaddr *)&client, &nAddrSize);         // 여기까지 수행되고 프로그램은 대기상태로 빠진다. 클라이언트로 부터 접속 요청이 들어와야 다음라인으로 넘어가게 된다. 클라이언트로 부터 접속요청이 오면 accept가 소켓의 식별자를 리턴하게 된다.         if(sClient == INVALID_SOCKET) {             printf("accept 실패, Error번호:%d\n",WSAGetLastError());             break;         }         printf("Accepted Client: IP=%s, Port=%d\n",                 inet_ntoa(client.sin_addr), ntohs(client.sin_port));           hThread = CreateThread(NULL, 0, ClientThread, (LPVOID)sClient, 0, &dwThreadId);         if(hThread == NULL) {             printf("CreateThread 실패, Error번호:%d\n", GetLastError());             break;         }         // ClientThread 함수를 수행하는 쓰레드가 하나 생성되어진다.         CloseHandle(hThread);     }     closesocket(sListen);     WSACleanup();     return 0; } void CheckArgs(int argc, char** argv) {     for(int i = 1; i < argc; i++)     {         if((argv[i][0] == '-') || (argv[i][0] == '/'))         {             switch(tolower(argv[i][1]))             {             case 'P':                 iPort = atoi(&argv[i][3]);                 break;            case 'i':                 bInterface = TRUE;                 if(strlen(argv[i]) > 3)                     strcpy(szAddress, &argv[i][3]);                 break;             case 'o':                 bRecvOnly = TRUE;                 break;                 default :                 usage();                 break;             }         }         else             usage();     } } /************************************************************/ /* Function : usage                                         */ /*                                                          */ /* Description                                              */ /*  사용방법을 화면에 보여주고 종료한다.                      */ /************************************************************/ void usage(void) {     printf("usage: server [-p:x] [-i:IP] [-o]\n\n");     printf("       -p:x      서버에서 사용할 포트 번호, 안쓰면 기본으로 5150을 쓴다.\n");     printf("       -i:IP     서버에 사용할 IP주소, 안쓰면 기본IP로 사용한다.\n");     printf("       -o        데이타를 되돌려 보내지 않는다.(not echo)\n\n");     ExitProcess(1); } DWORD WINAPI ClientThread(LPVOID lpParam) {     SOCKET sock = (SOCKET)lpParam;     char szBuff[DEFAULT_BUFFER];     int ret, nLeft, idx;     while(1) {         // 블로킹 방식의 recv()함수 사용         ret = recv(sock, szBuff, DEFAULT_BUFFER, 0);         // 위의 recv는 sock 소켓에서 수신되는 데이타를 받게 되며, 수신된 데이타는 szBuff에 들어간다. 그리고 한번에 최대 DEFAULT_BUFFER 갯수만큼 수신한다. 그리고 recv는 현재 수신된 데이타의 byte수를 리턴하게 된다.         if(ret == 0) { // 0 일 경우는 close를 요청했을 경우이다. 종료하면 된다.             break;         }         else if(ret == SOCKET_ERROR) {             printf("recv함수 에러, Error번호:%d\n", WSAGetLastError());             break;         }         szBuff[ret] = '\0'; // 수신 데이타의 제일끝에 NULL 값을 넣는다.         printf("RECV: '%s'\n",szBuff);         if(!bRecvOnly)         {             nLeft = ret;             idx = 0;             // 데이타를 전부 보낼때까지 반복한다.             while(nLeft > 0) {                 ret = send(sock, &szBuff[idx], nLeft, 0);                 if(ret = 0)                     break;                 else if(ret == SOCKET_ERROR) {                     printf("send함수 에러, Error번호:%d\n", WSAGetLastError());                     break;                 }                 nLeft -= ret;                 idx += ret;             }         }     }     return 0; }