UDP Broadcast 서버/클라이언트

브로드캐스트(Broadcast)와 관련된 모듈을 구현할 일이 있어 간만에 열혈강의 소켓 프로그래밍 책을 뒤졌다. 내용도 알차고 설명도 잘 되어있고, 실용적인 예제가 많은 좋은 책인 것 같다. 물론 IOCP나 Overapped IO같은 녀석들을 이용해 수백개 수천개의 클라이언트를 관리하는 서버 프로그래밍을 위해서는 좀더 심도 있는 책을 봐야겠지만, 뭐 기껏해야 20개 이내의 클라이언트가 연결될 서버에 거창한 테크닉따위는 필요없으니 .흐  그나저나 출판사가 바뀌었다. 예전엔 프리렉이었던 것 같은데.

TCP/IP 소켓 프로그래밍(2010) 카테고리 컴퓨터/IT 지은이 윤성우 (오렌지미디어, 2009년) 상세보기

아무튼 아래의 코드는 열혈강의 TCP/IP 소켓프로그래밍의 예제 + 인터넷에서 좀 알아본 코드를 합쳐서 살짝 수정한 것이다.

서버는 자기 자신의 IP와 서브넷 마스크를 통해 브로드캐스트할 IP를 정한 다음 정해진 포트로 브로드캐스팅을 하고 클라이언트는 별 것 없다. 포트를 바인딩해서 대기하는 것 말고는 없다. ㅎ

// UDP Broadcast Server(Sender) #include <stdio.h> #include <stdlib.h> #include <string.h> #include <winsock2.h> #pragma comment(lib, "ws2_32.lib") #define BUFSIZE 32 #define TRUE 1 #define FALSE 0 #define BROAD_CAST_PORT 4970 void ErrorHandling(char *message); int main() { WSADATA wsaData; SOCKET hSendSock; int state, so_broadcast = TRUE; char name[256] = "localhost"; // localhost라는 name을 통해 자기 자신의 IP를 얻는다. char buf[BUFSIZE] = "abcdefg"; char* address; struct hostent* pHostEnt; // HOSTENT* pHostEnt; struct in_addr in; // IN_ADDR in; struct sockaddr_in broadAddr; // SOCKADDR_IN broadAddr; DWORD nMyIP, nMySubMask, nTemp1, nTemp2; if (WSAStartup(MAKEWORD(2, 2), &wsaData) != 0) ErrorHandling("WSAStartup() error"); if ((hSendSock = socket(PF_INET, SOCK_DGRAM, 0)) == INVALID_SOCKET) ErrorHandling("socket() error"); gethostname(name,256); // printf("Host Name : %s\n", name); pHostEnt = gethostbyname(name); in.s_addr =((struct in_addr*)pHostEnt->h_addr)->s_addr; address = inet_ntoa(in); // printf("IP Addr : %s\n", address); nMyIP = inet_addr(address); nMySubMask = inet_addr("255.255.255.0"); nTemp1 = nMyIP & nMySubMask; nTemp2 = nMySubMask ^ 0xFFFFFFFF; memset(&broadAddr, 0, sizeof(broadAddr)); broadAddr.sin_family = AF_INET; broadAddr.sin_port = htons(BROAD_CAST_PORT); broadAddr.sin_addr.s_addr = nTemp1 + nTemp2; // 소켓을 브로드캐스트용으로 설정 state = setsockopt(hSendSock, SOL_SOCKET, SO_BROADCAST, (char*)&so_broadcast, sizeof(so_broadcast)); if (state == SOCKET_ERROR) ErrorHandling("setsockopt() error"); // printf("%s\n", inet_ntoa(broadAddr.sin_addr)); sendto(hSendSock, buf, strlen(buf), 0, (SOCKADDR*)&broadAddr, sizeof(broadAddr)); Sleep(2000); closesocket(hSendSock); WSACleanup(); return 0; } void ErrorHandling(char *message) { fputs(message, stderr); fputc('\n', stderr); exit(1); }

// UDP Broadcast Client (Receiver) #include "stdafx.h" #include <stdio.h> #include <stdlib.h> #include <string.h> #include <winsock2.h> #pragma comment(lib, "ws2_32.lib") #define BROAD_CAST_PORT 4970 #define BUFSIZE 32 void ErrorHandling(char *message); int main(int argc, char **argv) { WSADATA wsaData; SOCKET hRecvSock; SOCKADDR_IN addr; int strLen; char buf[BUFSIZE]; if(WSAStartup(MAKEWORD(2, 2), &wsaData) != 0) /* Load Winsock 2.2 DLL */ ErrorHandling("WSAStartup() error!"); hRecvSock=socket(PF_INET, SOCK_DGRAM, 0); /*브로드캐스트를 위한 UDP소켓 생성 */ if(hRecvSock == INVALID_SOCKET) ErrorHandling("socket() error"); memset(&addr, 0, sizeof(addr)); addr.sin_family=AF_INET; addr.sin_addr.s_addr=htonl(INADDR_ANY); addr.sin_port=htons(BROAD_CAST_PORT); /* 브로드캐스트 port 설정 */ if(bind(hRecvSock, (SOCKADDR*) &addr, sizeof(addr))==SOCKET_ERROR) ErrorHandling("bind() error"); while(1){ strLen=recvfrom(hRecvSock, buf, BUFSIZE-1, 0, NULL, 0); if(strLen<0) break; buf[strLen]=0; fputs(buf, stdout); } closesocket(hRecvSock); WSACleanup(); return 0; } void ErrorHandling(char *message) { fputs(message, stderr); fputc('\n', stderr); exit(1); }