Client 모델
: 하나 이상의 소켓 처리시 overlapped I/O 모델이나 WSAEventSelect 모델이 성능상 유리.
윈도우 기반일 경우 WSAAsyncSelct 모델이 개발 편의성 제공.
Server 모델
: 동시에 몇개의 소켓 처리시 overlapped I/O 모델이 성능상 유리.
대량의 I/O 처리시 IOCP 모델이 성능상 유리.
|
소켓 모델의 장단점
SPI(Service Provider Interface) : http://msdn.microsoft.com/en-us/library/ms741418.aspx (스펙)
윈도우 소켓은 소켓 인터페이스와 트랜스포트 계층(OSI 7계층) 사이에 SPI 계층이 있다.
SPI는 본질적으로 DLL이며 윈도우 시스템의 모든 소켓 송수신의 데이터 모니터링, hook이 가능하며
다른 소켓 어플리케이션을 통제할 수 있다.
winpcap 같은 라이브러리를 통해서도 가능하나 SPI는 ms에서 제공하는 표준적인 방법이다.
2MSL (2 * maximum segment lifetime)
TCP는 TcpTimedWaitDelay에 지정되어 있는 시간이 지나지 않을 경우
연결을 해제하지도 않고 그 연결에 사용된 자원을 재사용하지 않는다.
이 시간을 TIME_WAIT 또는 2MSL 상태라 한다.
Port
[0, 1023] : Well-known ports. 표준으로 예약된 포트 영역
[1024, 49151] : Registered ports. 어플리케이션에서 사용하는 포트 영역
[49152, 65535] : Dynamic ports. 필요에 따라 할당되는 포트 영역
socket 생성 옵션
TCP : PF_INET, SOCK_STREAM, IPPROTO_TCP
UDP : PF_INET, SOCK_DGRAM, IPPROTO_UDP
socket 기본 함수
socket : socket 생성
bind : 주소 bind
listen : 대기중인 socket임을 지정 (LISTENING 상태)
accept : client의 접속 요청을 받음
connect : server로 연결
send : 데이터 보냄
recv : 데이터 받음
closesocket : socket 닫음
**. 샘플 소스는 Visual Studio C++ 2008 Unicode Console Project 로 작성됨.
TCP
UDP
SOCKET mode
blocking mode
: default. 입출력 작업 완료시 까지 block. recv() 함수 처럼 대기하는 경우.
non-blocking mode
: 입출력 작업 완료와 상관없이 무조건 리턴.
ioctlsocket() 함수로 non-blocking mode 지정.
ioctlsocket() 사용시 소스가 복잡해짐.
또한 함수 내부 loop문의 CPU 자원 과소비로 인해 소켓 모델 사용 권장.
Windows SOCKET Model
1. blocking (single-thread 기반)
2. select (single-thread 기반)
3. WSAAsyncSelect (single-thread 기반)
4. WSAEventSelect (multi-thread 기반)
5. Ovelapped (multi-thread 기반)
6. IOCP (multi-thread 기반)
1. blocking model
일반적인 send(), recv() 형식의 SOCKET 통신.
2. select model
UNIX, Windows 모두 지원되는 모델.
많지 않은 socket 처리시 장점을 갖는 모델.
사건/이벤트 발생시 함수 호출.
single-thread 기반에서 여러 socket들의 non-blocking 구현이 가능하다.
int select(int nfds, fd_set* readfds, fd_set* writefds, fd_set* exceptfds, const struct timeval* timeout);
1st 인자 : 윈도우 소켓에서 사용되지 않음. 0으로 설정.
2nd 인자 : 읽기 사건(fd_set 포인터). 미사용시 NULL.
데이터 수신/수동적 종료시 recv/recvfrom 함수 호출.
대기 소켓에 읽기사건 발생시 accept 함수 호출.
3rd 인자 : 쓰기 사건(fd_set 포인터). 미사용시 NULL.
다음 데이터 전송 가능시 발생.
non-blocking 모드에서 connect 함수 호출시 접속 성공을 알리기 위해 발생.
4th 인자 : 예외 사건(fd_set 포인터). 미사용시 NULL.
OOB 도착시 발생하며 recv/recvfrom 함수를 호출하여 OOB를 읽어야 함.
non-blocking 모드에서 connect 함수 호출 후 접속 실패시 발생.
5th 인자 : 최대 대기 시간(timeval 포인터). NULL 일 경우 무한 대기
return :
0 : 타임아웃일 경우
-1 (SOCKET_ERROR) : 에러 발생. WSAGetLastError()로 확인 필요.
양수 : 이벤트가 발생한 소켓 개수. 개수만큼 소켓을 찾아 처리 필요.
**. 특징
: 읽기/쓰기/예외 사건 발생시 fd_set 구조체에 해당 소켓만 남기고 모두 제거한다.
대기할수 있는 소켓 개수는 64 (최대 1024)
내부적으로 loop를 통해 socket을 체크하므로 많은 CPU 자원 소비.
select() 함수 자체가 blocking 을 유발.
struct fd_set 사용 및 매크로
fd_set fds;
FD_ZERO(&fds); //모든 socket을 비움.
FD_SET(sock, &fds); //socket 추가. socket이 이미 들어있거나 FD_SETSIZE(64)보다 크면 미동작.
FD_CLR(sock, &fds); //socket 제거.
FD_ISSET(sock, &fds); //socket이 들어있으면 nonzero, 없으면 zero 리턴.
struct timeval 사용
long tv_sec : 초단위 설정
long tv_usec : 마이크로초(1/100만 초) 설정
Server.cpp
#include <iostream>
#include <winsock2.h>
#include <windows.h>
using namespace std;
#pragma comment(lib, "ws2_32.LIB")
#pragma comment(lib, "Kernel32.lib")
void printError(TCHAR *errMsg)
{
TCHAR* lpOSMsg;
int errNum = WSAGetLastError();
FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_IGNORE_INSERTS,
NULL,
errNum,
MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
(LPTSTR)&lpOSMsg,
0,
NULL);
wcout << TEXT("[ERROR] ") << errMsg << TEXT(", [") << errNum << TEXT("] ") << lpOSMsg << endl;
LocalFree(lpOSMsg);
}
int main()
{
setlocale(LC_ALL, "");
WSADATA wsaData;
if (WSAStartup(MAKEWORD(2, 2), &wsaData) == SOCKET_ERROR)
{
printError(TEXT("윈도우 소켓 초기화 에러"));
return -1;
}
wcout << TEXT("server WinSock 초기화 : ver ") << (int)LOBYTE(wsaData.wVersion) << "." << (int)HIBYTE(wsaData.wVersion) << endl;
SOCKET sock = socket(PF_INET, SOCK_STREAM, IPPROTO_TCP); //socket
try
{
if (sock == INVALID_SOCKET)
throw TEXT("대기 소켓 객체 생성 실패");
struct sockaddr_in addr;
addr.sin_family = AF_INET; //TCP, UDP..
addr.sin_port = htons(1024);
addr.sin_addr.s_addr = inet_addr("127.0.0.1"); //htonl(INADDR_ANY); //inet_addr("127.0.0.1");
bool isReuse = true;
if (setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, (const char*)&isReuse, sizeof(isReuse)) == SOCKET_ERROR) throw TEXT("setsockopt error");
if (bind(sock, (struct sockaddr*)&addr, sizeof(addr)) == SOCKET_ERROR) throw TEXT("bind error");
if (listen(sock, SOMAXCONN) == SOCKET_ERROR) throw TEXT("listen error");
fd_set fds;
FD_ZERO(&fds);
FD_SET(sock, &fds);
int cliSockCount = 0;
SOCKET cliSockArr[FD_SETSIZE-1];
while (true)
{
fd_set readfds = fds;
wcout << TEXT("fds 소켓 개수 : ") << fds.fd_count << endl;
int ret = select(0, &readfds, NULL, NULL, NULL);
if (ret == SOCKET_ERROR) throw TEXT("select error");
int eventSockCount = ret;
//서버 socket이 읽기 사건 발생시 accept 수행 및 client socket 등록
if (FD_ISSET(sock, &readfds)) //서버 소켓에서 읽기 event가 발생했을 경우
{
struct sockaddr_in cliaddr;
int addrlen = sizeof(cliaddr);
SOCKET clisock = accept(sock, (struct sockaddr*) &cliaddr, &addrlen);
if (cliSockCount < FD_SETSIZE -1)
{
//wcout << TEXT("[") << inet_ntoa(cliaddr.sin_addr) << TEXT("] 로 부터 접속") << endl;
cliSockArr[cliSockCount++] = clisock;
FD_SET(clisock, &fds);
}
else //select() 의 대기 최대 소켓 개수 초과시
{
struct linger ling; //TIME_WAIT 상태 회피
ling.l_onoff = 1; //nonzero : socket 유지, zero : 즉시 종료
ling.l_linger = 0; //socket 유지 시간(sec)
setsockopt(sock, SOL_SOCKET, SO_LINGER, (const char*)&ling, sizeof(ling));
closesocket(clisock);
}
eventSockCount--;
}
//event가 발생한 socket 에 대해 recv/send 수행.
for (int i=0; i<cliSockCount && 0<eventSockCount; i++)
{
if (!FD_ISSET(cliSockArr[i], &readfds)) continue;
eventSockCount--;
TCHAR msg[100];
int nReadBytes = recv(cliSockArr[i], (char*)msg, sizeof(msg), 0);
if (nReadBytes == 0 || nReadBytes == SOCKET_ERROR) //연결이 닫혔거나(0), 에러(SOCK_ERROR)면..
{
closesocket(cliSockArr[i]);
FD_CLR(cliSockArr[i], &fds);
cliSockArr[i] = cliSockArr[cliSockCount - 1];
cliSockCount--;
continue;
}
else
{
//wcout << TEXT("받은 데이터 : ") << msg << TEXT(", cliSock : ") << cliSockArr[i] << endl;
send(cliSockArr[i], (const char*)msg, nReadBytes, 0);
}
}
}
}
catch (TCHAR* errMsg)
printError(errMsg);
if (sock != INVALID_SOCKET)
{
struct linger ling;
ling.l_onoff = 1;
ling.l_linger = 0;
setsockopt(sock, SOL_SOCKET, SO_LINGER, (const char*)&ling, sizeof(ling));
closesocket(sock);
}
WSACleanup();
return 0;
}
|
Client.cpp
#include <winsock2.h>
#include <iostream>
#include <windows.h>
using namespace std;
#pragma comment(lib, "ws2_32.LIB")
void printError(TCHAR *errMsg)
{
TCHAR* lpOSMsg;
int errNum = WSAGetLastError();
FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_IGNORE_INSERTS,
NULL,
errNum,
MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
(LPTSTR)&lpOSMsg,
0,
NULL);
wcout << TEXT("[ERROR] ") << errMsg << TEXT(", [") << errNum << TEXT("] ") << lpOSMsg << endl;
LocalFree(lpOSMsg);
}
int main()
{
setlocale(LC_ALL, "");
WSADATA wsaData;
if (WSAStartup(MAKEWORD(2, 2), &wsaData) == SOCKET_ERROR)
{
printError(TEXT("윈도우 소켓 초기화 에러"));
return -1;
}
wcout << TEXT("client WinSock 초기화 : ver ") << (int)LOBYTE(wsaData.wVersion) << "." << (int)HIBYTE(wsaData.wVersion) << endl;
while(true)
{
int n;
wcout << TEXT("client 수 입력 : "); wcin >> n;
SOCKET* arrSock = new SOCKET[n];
for (int i=0; i<n; i++)
{
SOCKET sock = socket(PF_INET, SOCK_STREAM, IPPROTO_TCP);
arrSock[i] = sock;
try
{
if (sock == INVALID_SOCKET) throw TEXT("대기 소켓 객체 생성 실패");
struct sockaddr_in addr;
addr.sin_family = AF_INET; //TCP, UDP..
addr.sin_port = htons(1024);
addr.sin_addr.s_addr = inet_addr("127.0.0.1"); //htonl(INADDR_ANY); //inet_addr("127.0.0.1");
if (connect(sock, (struct sockaddr*) &addr, sizeof(addr)) == SOCKET_ERROR) throw TEXT("connect error");
while (true)
{
TCHAR msg[100] = {0};
wsprintf(msg, TEXT("hi~"));
int nSendBytes = send(sock, (const char*)msg, sizeof(msg), 0);
if (nSendBytes == SOCKET_ERROR) throw TEXT("send error");
ZeroMemory(msg, sizeof(msg));
int nRecvBytes = recv(sock, (char*)msg, sizeof(msg), 0);
if (nRecvBytes == SOCKET_ERROR) throw TEXT("recv error");
//wcout << TEXT("recv data : ") << msg << endl;
break;
}
}
catch (TCHAR* errMsg)
printError(errMsg);
}
wcout << TEXT("현재 client 수 : ") << n << endl;
wcout << TEXT("enter : 계속 진행") << endl;
wcin.ignore();
wcin.get();
for (int i=0; i<n; i++)
{
if (arrSock[i] != INVALID_SOCKET)
{
//테스트 용으로 client의 TIME_WAIT를 없애기 위한 linger
struct linger ling;
ling.l_onoff = 1;
ling.l_linger = 0;
setsockopt(arrSock[i], SOL_SOCKET, SO_LINGER, (const char*)&ling, sizeof(ling));
closesocket(arrSock[i]);
}
}
delete arrSock;
}
WSACleanup();
return 0;
}
|
3. WSAAsyncSelect model
3-1. WinAPI WSAAsyncSelect model
int WSAAsyncSelect(SOCKET s, HWNDhWnd, unsigned int wMsg, long lEvent);
**. 설명
항상 non-blocking.
socket에 event 발생시 특정 윈도우로 메세지 전송.
내부적으로 ioctlsocket 함수 호출.
**. 인자
1st 인자 : server socket
2nd 인자 : 윈도우 핸들
3rd 인자 : 윈도우로 보낼 메세지.
사용자 메세지 정의 방법 :
WM_USER + 0x01; //0x0400 + 0x01
WM_APP + 0x01; //0x8000 + 0x01
RegisterWindowMessage("__socket_notify__"); //0xC000~0xFFFF 사이에 자동 지정
4th 인자 : 처리할 socket event 종류 지정. (|로 조합가능)
FD_ACCEPT - clinet 접속시 (후에 accept() 처리 필요)
FD_READ - 데이터 수신시 (후에 recv/recvfrom 처리 필요)
FD_WRITE - 다음 데이터 송신 가능시 (후에 send/sendto 처리 필요)
FD_OOB - OOB 데이터 수신시 (후에 MSG_OOB flag로 recv 처리 필요)
FD_CLOSE - socket 닫힐시 (후에 closesocket() 처리 필요)
FD_CONNECT- client가 connect()로 접속 요청 성공시 (후에 처리 불필요)
return : 성공 = zero, 실패 = SOCKET_ERROR(-1)
**. 메세지 처리
LRESULT CALLBACK WndProc(HWND HwND, UINT message, WPARAM wParam, LPARAM lParam)
{
/*
hWnd : WSAAyncSelect()의 2nd 인자인 윈도우 핸들
messgae : WSAAyncSelect()의 3rd 인자인 메세지
wParam : WSAAyncSelect()의 1st 인자인 SOCKET (캐스팅 필요)
lParam : WSAGETSELECTEVENT(lParam) - WSAAyncSelect()의 4th 인자인 event 종류
WSAGETSELECTERROR(lParam) - 에러 코드
#define WSAGETSELECTEVENT(lParam) LOWORD(lParam)
#define WSAGETSELECTERROR(lParam) HIWORD(lParam)
*/
}
실행결과
Server.cpp
#include <winsock2.h>
#include <windows.h>
#include <list>
#pragma comment(lib, "ws2_32.lib")
using namespace std;
#define ID_LIST (WM_USER + 0x01)
#define ID_BUTTON (WM_USER + 0x02)
#define WM_SOCKET_NOTIFY (WM_USER + 0x03)
HWND hList;
HWND hBtn;
SOCKET gSock = INVALID_SOCKET;
list<SOCKET> gClisock;
void MoveCenter(HWND hWnd);
void printError(TCHAR *errMsg);
bool CreateSock(HWND hWnd);
void fdAccept(HWND hWnd);
void fdRead(SOCKET clisock);
void fdClose(HWND hWnd, SOCKET clisock);
LRESULT CALLBACK WndProc(HWND,UINT,WPARAM,LPARAM);
HINSTANCE g_hInst;
HWND hWndMain;
LPCTSTR lpClassName=TEXT("main");
LPCTSTR lpTitleName=TEXT("Tomato Soap");
int APIENTRY WinMain(HINSTANCE hInstance,HINSTANCE hPrevInstance,LPSTR lpszCmdParam,int nCmdShow)
{
HWND hWnd;
MSG Message;
WNDCLASS WndClass;
g_hInst=hInstance;
WndClass.cbClsExtra=0;
WndClass.cbWndExtra=0;
WndClass.hbrBackground=(HBRUSH)(COLOR_WINDOW+1);
WndClass.hCursor=LoadCursor(NULL,IDC_ARROW);
WndClass.hIcon=LoadIcon(NULL,IDI_APPLICATION);
WndClass.hInstance=hInstance;
WndClass.lpfnWndProc=(WNDPROC)WndProc;
WndClass.lpszClassName=lpClassName;
WndClass.lpszMenuName=NULL;
WndClass.style=CS_HREDRAW | CS_VREDRAW;
RegisterClass(&WndClass);
hWnd = CreateWindow(lpClassName,lpTitleName,WS_OVERLAPPEDWINDOW,
CW_USEDEFAULT,CW_USEDEFAULT,500,500,
NULL,(HMENU)NULL,hInstance,NULL);
ShowWindow(hWnd,nCmdShow);
while (GetMessage(&Message,0,0,0)) {
TranslateMessage(&Message);
DispatchMessage(&Message);
}
UnregisterClass(lpClassName, hInstance);
return (int)Message.wParam;
}
LRESULT CALLBACK WndProc(HWND hWnd,UINT iMessage,WPARAM wParam,LPARAM lParam)
{
switch(iMessage) {
case WM_CREATE:
hWndMain=hWnd;
MoveCenter(hWnd);
hList = CreateWindow(TEXT("LISTBOX"), NULL, WS_CHILD | WS_VISIBLE | WS_BORDER | WS_VSCROLL | LBS_NOINTEGRALHEIGHT, 10, 10, 400, 400, hWnd, (HMENU)ID_LIST, g_hInst, NULL);
hBtn = CreateWindow(TEXT("BUTTON"), TEXT("close"), WS_CHILD | WS_VISIBLE | BS_PUSHBUTTON | BS_NOTIFY, 10, 410, 100, 30, hWnd, (HMENU)ID_BUTTON, g_hInst, NULL);
if (!CreateSock(hWnd)) SendMessage(hWnd, WM_DESTROY, NULL, NULL);
return 0;
case WM_COMMAND:
switch (LOWORD(wParam))
{
case ID_BUTTON:
switch (HIWORD(wParam))
{
case BN_CLICKED:
SendMessage(hWnd, WM_DESTROY, NULL, NULL);
break;
}
break;
case ID_LIST:
break;
}
return 0;
case WM_DESTROY:
if (gSock != INVALID_SOCKET)
closesocket(gSock);
WSACleanup();
PostQuitMessage(0);
return 0;
case WM_SOCKET_NOTIFY:
switch (WSAGETSELECTEVENT(lParam))
{
case FD_ACCEPT:
fdAccept(hWnd);
break;
case FD_READ:
fdRead((SOCKET)wParam);
break;
case FD_CLOSE:
fdClose(hWnd, (SOCKET)wParam);
break;
}
return 0;
}
return(DefWindowProc(hWnd,iMessage,wParam,lParam));
}
void MoveCenter(HWND hWnd)
{
RECT rectDesktop, rectWindow;
GetWindowRect(GetDesktopWindow(),&rectDesktop) ;
GetWindowRect(hWnd,&rectWindow) ;
LONG width, height;
width = rectWindow.right - rectWindow.left ;
height = rectWindow.bottom - rectWindow.top ;
LONG x, y;
x = LONG((rectDesktop.right - width) / 2) ;
y = LONG((rectDesktop.bottom - height) / 2) ;
MoveWindow(hWnd, x, y, width, height, TRUE) ;
}
void printError(TCHAR *errMsg)
{
TCHAR* lpOSMsg;
FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_IGNORE_INSERTS,
NULL,
WSAGetLastError(),
MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
(LPTSTR)&lpOSMsg,
0,
NULL);
TCHAR msg[100];
wsprintf(msg, TEXT("%s :: %s"), errMsg, lpOSMsg);
SendMessage(hList, LB_ADDSTRING, 0, (LPARAM)msg);
MessageBox(hWndMain, msg, TEXT("error"), MB_OK);
LocalFree(lpOSMsg);
SendMessage(hWndMain, WM_DESTROY, NULL, NULL);
}
bool CreateSock(HWND hWnd)
{
TCHAR msg[100];
WSADATA wsaData;
if (WSAStartup(MAKEWORD(2, 2), &wsaData) == SOCKET_ERROR)
{
wsprintf(msg, TEXT("server WinSock 초기화 error"));
SendMessage(hList, LB_ADDSTRING, 0, (LPARAM)msg);
SendMessage(hWnd, WM_DESTROY, NULL, NULL);
}
wsprintf(msg, TEXT("server WinSock 초기화 : ver %d.%d"), (int)LOBYTE(wsaData.wVersion), (int)HIBYTE(wsaData.wVersion));
SendMessage(hList, LB_ADDSTRING, 0, (LPARAM)msg);
SOCKET sock = socket(PF_INET, SOCK_STREAM, IPPROTO_TCP); //socket
try
{
if (sock == INVALID_SOCKET) throw TEXT("대기 소켓 객체 생성 실패");
struct sockaddr_in addr;
addr.sin_family = AF_INET; //TCP, UDP..
addr.sin_port = htons(1024);
addr.sin_addr.s_addr = inet_addr("127.0.0.1"); //htonl(INADDR_ANY); //inet_addr("127.0.0.1");
//TIME_WAIT인 경우 socket 재사용
bool isReuse = true;
if (setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, (const char*)&isReuse, sizeof(isReuse)) == SOCKET_ERROR)
throw TEXT("setsockopt error");
if (bind(sock, (struct sockaddr*)&addr, sizeof(addr)) == SOCKET_ERROR)
throw TEXT("bind error");
if (listen(sock, SOMAXCONN) == SOCKET_ERROR)
throw TEXT("listen error");
if (WSAAsyncSelect(sock, hWnd, WM_SOCKET_NOTIFY, FD_ACCEPT) == SOCKET_ERROR)
throw TEXT("WSAAsyncSelect error");
gSock = sock;
}
catch (TCHAR* errMsg)
{
printError(errMsg);
if (sock != INVALID_SOCKET)
{
struct linger ling; //server에서의 TIME_WAIT 방지를 위한 조치
ling.l_onoff = 1;
ling.l_linger = 0;
setsockopt(sock, SOL_SOCKET, SO_LINGER, (const char*)&ling, sizeof(ling));
closesocket(sock);
}
return false;
}
return true;
}
void fdAccept(HWND hWnd)
{
SOCKET sock = gSock;
struct sockaddr_in cliaddr;
int addrlen = sizeof(cliaddr);
SOCKET clisock= accept(sock, (struct sockaddr*)&cliaddr, &addrlen);
WSAAsyncSelect(clisock, hWnd, WM_SOCKET_NOTIFY, FD_READ | FD_CLOSE);
gClisock.push_back(clisock);
TCHAR cliip[20];
MultiByteToWideChar( CP_ACP, MB_COMPOSITE, inet_ntoa(cliaddr.sin_addr), -1, cliip, lstrlen(cliip));
TCHAR msg[100];
wsprintf(msg, TEXT("접속수:%d, 접속ip: %s"), gClisock.size(), cliip);
int n = (int)SendMessage(hList, LB_ADDSTRING, NULL, (LPARAM)msg); //리스트에 문자 추가 후 해당 index를 리턴.
SendMessage(hList, LB_SETTOPINDEX, n, 0); //리스트에 해당 index가 항상 보여지도록 scrolling함
}
void fdRead(SOCKET clisock)
{
TCHAR msg[1024] = {0};
int nReadbytes = recv(clisock, (char*)msg, sizeof(msg), 0);
if (nReadbytes == SOCKET_ERROR) return;
for (list<SOCKET>::iterator it=gClisock.begin(); it!=gClisock.end(); it++)
send(*it, (const char*)msg, nReadbytes, 0);
}
void fdClose(HWND hWnd, SOCKET clisock)
{
closesocket(clisock);
gClisock.remove(clisock);
TCHAR msg[100];
wsprintf(msg, TEXT("접속수:%d"), gClisock.size());
int n = (int)SendMessage(hList, LB_ADDSTRING, NULL, (LPARAM)msg); //리스트에 문자 추가 후 해당 index를 리턴.
SendMessage(hList, LB_SETTOPINDEX, n, 0); //리스트에 해당 index가 항상 보여지도록 scrolling함
}
|
Client.cpp
#include <windows.h>
#include "resource.h"
#pragma comment(lib, "ws2_32.lib")
#define ID_LIST (WM_USER + 0x01)
#define ID_CLOSEBUTTON (WM_USER + 0x02)
#define WM_SOCKET_NOTIFY (WM_USER + 0x03)
#define ID_MSGEDIT (WM_USER + 0x04)
#define ID_SENDBUTTON (WM_USER + 0x05)
HWND hList;
HWND hCloseBtn;
HWND hMsgEdit;
HWND hSendBtn;
SOCKET gSock = INVALID_SOCKET;
TCHAR gUserID[128]; //사용자 이름
void printError(TCHAR *errMsg);
void MoveCenter(HWND hWnd);
bool CreateSock(HWND hWnd);
void fdRead(HWND hWnd);
void fdSend(HWND hWnd);
BOOL CALLBACK LoginDlgProc(HWND hWnd,UINT iMessage,WPARAM wParam,LPARAM lParam);
LRESULT CALLBACK WndProc(HWND hWnd,UINT iMessage,WPARAM wParam,LPARAM lParam);
LRESULT CALLBACK EditProc(HWND hWnd,UINT iMessage,WPARAM wParam,LPARAM lParam);
HINSTANCE g_hInst;
HWND hWndMain;
LPCTSTR lpClassName=TEXT("main");
LPCTSTR lpTitleName=TEXT("Tomato Soap");
//Login 다이알로그 메세지 proc
BOOL CALLBACK LoginDlgProc(HWND hWnd,UINT iMessage,WPARAM wParam,LPARAM lParam)
{
HWND hEdit = NULL;
switch (iMessage)
{
case WM_INITDIALOG:
MoveCenter(hWnd);
return TRUE;
case WM_COMMAND:
switch (LOWORD(wParam))
{
case IDOK:
GetDlgItemText(hWnd, IDC_EDIT1, gUserID, sizeof(gUserID));
EndDialog(hWnd, IDOK);
break;
case IDCANCEL:
EndDialog(hWnd, IDCANCEL);
break;
}
return TRUE;
}
return FALSE;
}
int APIENTRY WinMain(HINSTANCE hInstance,HINSTANCE hPrevInstance,LPSTR lpszCmdParam,int nCmdShow)
{
//다이알 로그호출하여 user id를 받아 gUserID에 저장.
INT_PTR nLoginID = DialogBox(hInstance, MAKEINTRESOURCE(IDD_DIALOG1), HWND_DESKTOP, LoginDlgProc);
if (nLoginID == IDCANCEL) return 0;
HWND hWnd;
MSG Message;
WNDCLASS WndClass;
g_hInst=hInstance;
WndClass.cbClsExtra=0;
WndClass.cbWndExtra=0;
WndClass.hbrBackground=(HBRUSH)(COLOR_WINDOW+1);
WndClass.hCursor=LoadCursor(NULL,IDC_ARROW);
WndClass.hIcon=LoadIcon(NULL,IDI_APPLICATION);
WndClass.hInstance=hInstance;
WndClass.lpfnWndProc=(WNDPROC)WndProc;
WndClass.lpszClassName=lpClassName;
WndClass.lpszMenuName=NULL;
WndClass.style=CS_HREDRAW | CS_VREDRAW;
RegisterClass(&WndClass);
hWnd = CreateWindow(lpClassName,lpTitleName,WS_OVERLAPPEDWINDOW,
CW_USEDEFAULT,CW_USEDEFAULT,500,500,
NULL,(HMENU)NULL,hInstance,NULL);
ShowWindow(hWnd,nCmdShow);
while (GetMessage(&Message,0,0,0)) {
TranslateMessage(&Message);
DispatchMessage(&Message);
}
UnregisterClass(lpClassName, hInstance);
return (int)Message.wParam;
}
WNDPROC EditProcOrg = NULL;
LRESULT CALLBACK WndProc(HWND hWnd,UINT iMessage,WPARAM wParam,LPARAM lParam)
{
switch(iMessage) {
case WM_CREATE:
hWndMain=hWnd;
MoveCenter(hWnd);
hList = CreateWindow(TEXT("LISTBOX"), NULL, WS_CHILD | WS_VISIBLE | WS_BORDER | WS_VSCROLL | LBS_NOINTEGRALHEIGHT, 10, 10, 400, 400, hWnd, (HMENU)ID_LIST, g_hInst, NULL);
hMsgEdit = CreateWindow(TEXT("EDIT"), NULL, WS_CHILD | WS_VISIBLE| WS_BORDER, 10, 410, 300, 20, hWnd, (HMENU)ID_MSGEDIT, g_hInst, NULL);
hSendBtn = CreateWindow(TEXT("BUTTON"), TEXT("send"), WS_CHILD | WS_VISIBLE, 310, 410, 50, 20, hWnd, (HMENU)ID_SENDBUTTON, g_hInst, NULL);
hCloseBtn = CreateWindow(TEXT("BUTTON"), TEXT("close"), WS_CHILD | WS_VISIBLE, 360, 410, 50, 20, hWnd, (HMENU)ID_CLOSEBUTTON, g_hInst, NULL);
SetFocus(hMsgEdit);
//채팅 메세지 입력시 엔터 감지 위한 에디터 컨트롤 서브 클래싱
EditProcOrg = (WNDPROC)SetWindowLong(hMsgEdit, GWL_WNDPROC, (LONG)EditProc);
if (!CreateSock(hWnd)) SendMessage(hWnd, WM_DESTROY, NULL, NULL);
return 0;
case WM_DESTROY:
if (gSock != INVALID_SOCKET)
closesocket(gSock);
WSACleanup();
PostQuitMessage(0);
return 0;
case WM_COMMAND:
switch (LOWORD(wParam))
{
case ID_CLOSEBUTTON:
switch (HIWORD(wParam))
{
case BN_CLICKED:
SendMessage(hWnd, WM_DESTROY, NULL, NULL);
break;
}
break;
case ID_SENDBUTTON:
fdSend(hWnd);
break;
}
return 0;
case WM_SOCKET_NOTIFY:
switch (WSAGETSELECTEVENT(lParam))
{
case FD_READ:
fdRead(hWnd);
break;
}
return 0;
}
return(DefWindowProc(hWnd,iMessage,wParam,lParam));
}
LRESULT CALLBACK EditProc(HWND hWnd,UINT iMessage,WPARAM wParam,LPARAM lParam)
{
switch (iMessage)
{
case WM_CHAR:
if (wParam == VK_RETURN) return 0; //엔터시 삑~ 소리 없애기 위한 방법
break;
case WM_KEYDOWN:
if (wParam == VK_RETURN)
SendMessage(hWndMain, WM_COMMAND, ID_SENDBUTTON, NULL);
break;
}
return(EditProcOrg(hWnd,iMessage,wParam,lParam));
}
void printError(TCHAR *errMsg)
{
TCHAR* lpOSMsg;
FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_IGNORE_INSERTS,
NULL,
WSAGetLastError(),
MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
(LPTSTR)&lpOSMsg,
0,
NULL);
TCHAR msg[100];
wsprintf(msg, TEXT("%s :: %s"), errMsg, lpOSMsg);
SendMessage(hList, LB_ADDSTRING, 0, (LPARAM)msg);
MessageBox(hWndMain, msg, TEXT("error"), MB_OK);
LocalFree(lpOSMsg);
SendMessage(hWndMain, WM_DESTROY, NULL, NULL);
}
void MoveCenter(HWND hWnd)
{
RECT rectDesktop, rectWindow;
GetWindowRect(GetDesktopWindow(),&rectDesktop) ;
GetWindowRect(hWnd,&rectWindow) ;
LONG width, height;
width = rectWindow.right - rectWindow.left ;
height = rectWindow.bottom - rectWindow.top ;
LONG x, y;
x = LONG((rectDesktop.right - width) / 2) ;
y = LONG((rectDesktop.bottom - height) / 2) ;
MoveWindow(hWnd, x, y, width, height, TRUE) ;
}
bool CreateSock(HWND hWnd)
{
TCHAR msg[100];
WSADATA wsaData;
if (WSAStartup(MAKEWORD(2, 2), &wsaData) == SOCKET_ERROR)
{
wsprintf(msg, TEXT("client WinSock 초기화 error"));
SendMessage(hList, LB_ADDSTRING, 0, (LPARAM)msg);
SendMessage(hWnd, WM_DESTROY, NULL, NULL);
}
wsprintf(msg, TEXT("client WinSock 초기화 : ver %d.%d"), (int)LOBYTE(wsaData.wVersion), (int)HIBYTE(wsaData.wVersion));
SendMessage(hList, LB_ADDSTRING, 0, (LPARAM)msg);
SOCKET sock = socket(PF_INET, SOCK_STREAM, IPPROTO_TCP); //socket
try
{
if (sock == INVALID_SOCKET) throw TEXT("대기 소켓 객체 생성 실패");
struct sockaddr_in addr;
addr.sin_family = AF_INET; //TCP, UDP..
addr.sin_port = htons(1024);
addr.sin_addr.s_addr = inet_addr("127.0.0.1"); //htonl(INADDR_ANY); //inet_addr("127.0.0.1");
if (connect(sock, (struct sockaddr*) &addr, sizeof(addr)) == SOCKET_ERROR)
throw TEXT("connect error");
if (WSAAsyncSelect(sock, hWnd, WM_SOCKET_NOTIFY, FD_READ) == SOCKET_ERROR)
throw TEXT("WSAAsyncSelect error");
gSock = sock;
}
catch (TCHAR* errMsg)
{
printError(errMsg);
if (sock != INVALID_SOCKET) closesocket(sock);
return false;
}
return true;
}
void fdRead(HWND hWnd)
{
TCHAR msg[128+256];
int nReadbytes = recv(gSock, (char*)msg, sizeof(msg), 0);
if (nReadbytes == SOCKET_ERROR) return;
int n = (int)SendMessage(hList, LB_ADDSTRING, NULL, (LPARAM)msg); //리스트에 문자 추가 후 해당 index를 리턴.
SendMessage(hList, LB_SETTOPINDEX, n, 0); //리스트에 해당 index가 항상 보여지도록 scrolling함
}
void fdSend(HWND hWnd)
{
TCHAR msg[256];
GetWindowText(hMsgEdit, msg, sizeof(msg)/sizeof(TCHAR));
SetWindowText(hMsgEdit, TEXT(""));
TCHAR buf[128+256];
wsprintf(buf, TEXT("%s : %s"), gUserID, msg);
if (send(gSock, (const char*) buf, sizeof(buf), 0) == SOCKET_ERROR)
printError(TEXT("send error"));
}
|
3-2. MFC WSAAsyncSelect model : CSocket
**. MFC project 생성시 "Advenced Features > Windows sockets"를 선택한다.
주요 클래스/함수
CSocket : CAsyncSocket 클래스를 상속한 blocking방식(함수동작방식) 클래스.
CAsyncSocket : WSAAsyncSelect 및 SOCKET을 객체화한 클래스.
CSocketFile : CSocket을 파일처럼 사용할 수 있게 해줌.
CArchive : CSocketFile과 연결되어 직렬화 방식으로 어플리케이션 프로토콜 구현을 도움.
AfxSocketInit() : WSAStartup()와 자동 WSACleanup()의 역할을 수행.
CAsyncSocket 주요 멤버
m_hSocket : 실제 socket handle
Create : SOCKET 생성. ip/port 를 인자로 줄 경우 내부에서 Bind 수행.
Socket : 객체 내의 socket handle 생성
Bind : ip/port 바인드
Close : socket handle 닫음
Connect : 서버에 접속 수행
Listen : 접송 요청 대기 모드
Receive : 데이터 수신(TCP)
ReceiveFrom : 데이터 수신(UDP)
Send : 데이터 송신(TCP)
SendTo : 데이터 송신(UDP)
Accept : 클라이언트의 접속된 소켓을 얻어 새로운 소켓 생성
AsyncSelect : 내부적으로 WSAAsyncSelect 함수 호출로 비동기 이벤트 설정
IOCtl : 소켓모드 설정 또는 소켓 버퍼의 데이터 크기 조회용
Attach : 외부의 소켓 핸들을 CAsyncSocket 객체에 병합
Detach : CAsyncSocket 객체에 병합된 소켓핸들을 분리
FromHandle : MFC 내부의 소켓 핸들 테이블을 검색하여 CAsyncSocket 객체 포인터 반환
GetLastError : 마지막 에러코드 조회
GetPeerName : 상대편 host의 주소 정보 조회
GetSockName : 소켓의 ip/port 조회
GetSockOpt : 소켓 옵션 상태 조회
SetSockOpt : 소켓 옵션 상태 설정
SOCKET EVENT 및 대응되는 가상 함수
FD_READ - OnReceive()
FD_WRITE - OnSend()
FD_CLOSE - OnClose()
FD_CONNECT - OnConnect()
FD_ACCEPT - OnAccept()
FD_OOB - OnOutOfBandData()
실행결과 및 소스
CSocket_Client.zip
..
CServerSock gSock;
BOOL CServerDlg::OnInitDialog()
{
..
if (!AfxSocketInit())
{
printError(TEXT("AfxSocketInit error"));
EndDialog(-1);
}
if (gSock.Create(1024) && gSock.Listen())
{
gSock.AsyncSelect(FD_ACCEPT);
}
else
{
printError(TEXT("create listen error"));
gSock.Close();
EndDialog(-1);
}
..
//CSocket을 상속받아 OnAccept()를 오버라이딩.
void CServerSock::OnAccept(int nErrorCode)
{
CClientSock* cliSock = new CClientSock;
..
Accept(*cliSock);
..
CString clientIp;
UINT clientPort;
cliSock->GetPeerName(clientIp, clientPort); //client ip/port 조회
..
cliSock->AsyncSelect(FD_READ | FD_CLOSE); //client의 메세지 수신 및 종료 evnet 감시
CSocket::OnAccept(nErrorCode);
}
//CSocket을 상속받아 OnAccept()를 오버라이딩.
void CClientSock::OnReceive(int nErrorCode)
{
TCHAR msg[1024] = {0};
int nReadbytes = Receive(msg, sizeof(msg)-1); //데이터 수신
if (nReadbytes == SOCKET_ERROR) return;
..
for (list<CClientSock*>::iterator it=cliSockList->begin(); it!=cliSockList->end(); it++)
((CClientSock*)*it)->Send(msg, nReadbytes); //데이터 송신
CSocket::OnReceive(nErrorCode);
}
//CSocket을 상속받아 OnClose() 오버라이딩
void CClientSock::OnClose(int nErrorCode)
{
//this->Close();
cliSockList->remove(this);
CSocket::OnClose(nErrorCode);
}
.. |
CSocket_Client 주요부분
..
CClientSock gSock;
int CChatDlg::OnCreate(LPCREATESTRUCT lpCreateStruct)
{
..
struct sockaddr_in addr;
addr.sin_family = AF_INET; //TCP, UDP..
addr.sin_port = htons(1024);
addr.sin_addr.s_addr = inet_addr("127.0.0.1"); //htonl(INADDR_ANY); //inet_addr("127.0.0.1");
if (!AfxSocketInit())
{
printError(TEXT("AfxSocketInit error"));
EndDialog(-1);
}
if (gSock.Create() && gSock.Connect((const SOCKADDR*)&addr, sizeof(addr)))
gSock.AsyncSelect(FD_READ);
else
{
printError(TEXT("create or connect error"));
gSock.Close();
EndDialog(-1);
}
void CChatDlg::OnBnClickedSendButton()
{
CString msg;
GetDlgItemText(IDC_MSG_EDIT, msg);
CString str = userid + TEXT(" : ") + msg;
gSock.Send((LPCTSTR)str, str.GetLength() * sizeof(TCHAR)); //데이터 송신
..
//CSocket 을 상속한 후 OnReceive() 오버라이딩
void CClientSock::OnReceive(int nErrorCode)
{
TCHAR buf[1024] = {0};
Receive(buf, sizeof(buf)-1); //데이터 수신
..
int n = msgList->AddString(buf);
msgList->SetTopIndex(n);
CSocket::OnReceive(nErrorCode);
}
|
4. WSAEventSelect model
WSAEventSelect
thread 사용을 전제로 하며 WSAAsyncSelect 모델과 비슷함.
이벤트 커널 객체 사용. non-blocking 소켓 사용.
주요 함수
WSAEVENT hEvent = WSACreateEvent(); //반드시 수동 non-signal 상태로 생성
HANDLE hEvent = CreateEvent(NULL, TRUE, FALSE, NULL); 과 동일.
int WSAEventSelect(SOCKET s, WSAEVENT hEventObject, long lNetworkEvents);
EX) WSAEventSelect(sock, hEvt, FD_READ | FD_CLOSE);
DWORD WSAWiatForMultipleEvents(DWORD cEvents, const WSAEVENT* lphEvents,
BOOL fWaitAll, DWORD dwTimeout, BOOL fAlertable);
/*
1st 인자 : 대기 이벤트 개수(최대 WSA_MAXIMUM_WAIT_EVENTS(64))
2nd 인자 : 이벤트 핸들 배열 포인터(WSA_MAXIMUM_WAIT_EVENTS(64) 크기)
3rd 인자 : FALSE - 이벤트가 하나라도 신호 상태가 되면 실행
TRUE - 이벤트가 모두 신호 상태가 되면 실행
4th 인자 : 최대 대기 시간(ms). 무한 대기 INFINITE
5th 인자 : APC(asynchronous procedure call) 사용 여부.
TRUE - APC 사용하는 경우는 Overlapped 모델 뿐임.
FALSE - APC 사용 안함.
return :
case 1 - 신호상태(return 값 = WSA_WAIT_EVENT_0)
case 2 - 타임아웃(return 값 = WAS_WAIT_TIMEOUT)
case 3 - APC가 실행되고 반환된 경우(return 값 = WAIT_IO_COMPLETION)
*/
WaitFroMultipleObjectsEx/WaitForSingleObject/WaitForMultipleObjects 함수와 동일한 기능.
int WSAEnumNetworkEvents(SOCKETK s, WSAEVENT hEventObject, LPWSANETWORKEVENTS lpNetworkEvents);
event 종류 검사.
/*
1st : event 종류 검사할 소켓
2nd : 1st 소켓과 연결된 이벤트 커널 객체 핸들
3rd : 발생한 event 종류를 담는 struct
typedef struct _WSANETWORKEVENTS {
long lNetworkEvents; //소켓 event
int iErrorCode[FD_MAX_EVENTS]; //에러 코드
} WSANETWORKEVENTS, *LPWSANETWORKEVENTS;
EX) WSANETWORKEVENTS netEvt;
WSAEnumNetworkEvents(sock, hEvt, &netEvt);
if (netEvt.lNetworkEvents & FD_READ)
//수신
else if (netEvt.lNetworkEvents & FD_CLOSE)
//수동 종료
EX) FD_READ : FD_READ_BIT : 0
FD_WRITE : FD_WRITE_BIT : 1
FD_OBB : FD_OBB_BIT : 2
FD_ACCEPT : FD_ACCEPT_BIT : 3
FD_CONNECT : FD_CONNECT_BIT : 4
FD_CLOSE : FD_CLOSE_BIT : 5
return : 성공 - zero, 실패 : SOCKET_ERROR(-1)
*/
BOOL WSACloseEvent(WSAEVENT hEvent);
CloseHandle() 과 동일.
Server.cpp
#include <winsock2.h>
#include <windows.h>
#include <process.h>
#pragma comment(lib, "ws2_32.lib")
#define ID_LIST (WM_USER + 0x01)
#define ID_BUTTON (WM_USER + 0x02)
#define WM_SOCKET_NOTIFY (WM_USER + 0x03)
HWND hList;
HWND hBtn;
#define MAX_CLIENT (WSA_MAXIMUM_WAIT_EVENTS -1 -1)
UINT WINAPI ThreadProc(LPVOID lpParameter);
void MoveCenter(HWND hWnd);
void printError(TCHAR *errMsg);
void InnerThreadProc(void);
SOCKET gSock = INVALID_SOCKET;
SOCKET gClientSock[MAX_CLIENT];
WSAEVENT g_hEvent[MAX_CLIENT+2]; //처음 2개는 스레드 탈출, 대기 소켓을 위한 이벤트
WSAEVENT* g_phEvtSock = g_hEvent+2;
long g_lSockCnt = 0;
long g_lThreadCnt = 0;
void OnAccept(void);
void OnRead(SOCKET sock);
void OnClose(int i);
LRESULT CALLBACK WndProc(HWND,UINT,WPARAM,LPARAM);
HINSTANCE g_hInst;
HWND hWndMain;
LPCTSTR lpClassName=TEXT("main");
LPCTSTR lpTitleName=TEXT("Tomato Soap");
int APIENTRY WinMain(HINSTANCE hInstance,HINSTANCE hPrevInstance,LPSTR lpszCmdParam,int nCmdShow)
{
HWND hWnd;
MSG Message;
WNDCLASS WndClass;
g_hInst=hInstance;
WndClass.cbClsExtra=0;
WndClass.cbWndExtra=0;
WndClass.hbrBackground=(HBRUSH)(COLOR_WINDOW+1);
WndClass.hCursor=LoadCursor(NULL,IDC_ARROW);
WndClass.hIcon=LoadIcon(NULL,IDI_APPLICATION);
WndClass.hInstance=hInstance;
WndClass.lpfnWndProc=(WNDPROC)WndProc;
WndClass.lpszClassName=lpClassName;
WndClass.lpszMenuName=NULL;
WndClass.style=CS_HREDRAW | CS_VREDRAW;
RegisterClass(&WndClass);
hWnd = CreateWindow(lpClassName,lpTitleName,WS_OVERLAPPEDWINDOW,
CW_USEDEFAULT,CW_USEDEFAULT,500,500,
NULL,(HMENU)NULL,hInstance,NULL);
ShowWindow(hWnd,nCmdShow);
while (GetMessage(&Message,0,0,0)) {
TranslateMessage(&Message);
DispatchMessage(&Message);
}
UnregisterClass(lpClassName, hInstance);
return (int)Message.wParam;
}
LRESULT CALLBACK WndProc(HWND hWnd,UINT iMessage,WPARAM wParam,LPARAM lParam)
{
TCHAR msg[100];
HANDLE hThread;
switch(iMessage) {
case WM_CREATE:
hWndMain=hWnd;
MoveCenter(hWnd);
hList = CreateWindow(TEXT("LISTBOX"), NULL, WS_CHILD | WS_VISIBLE | WS_BORDER | WS_VSCROLL | LBS_NOINTEGRALHEIGHT, 10, 10, 400, 400, hWnd, (HMENU)ID_LIST, g_hInst, NULL);
hBtn = CreateWindow(TEXT("BUTTON"), TEXT("close"), WS_CHILD | WS_VISIBLE | BS_PUSHBUTTON | BS_NOTIFY, 10, 410, 100, 30, hWnd, (HMENU)ID_BUTTON, g_hInst, NULL);
WSADATA wsaData;
if (WSAStartup(MAKEWORD(2, 2), &wsaData) == SOCKET_ERROR)
{
wsprintf(msg, TEXT("server WinSock 초기화 error"));
SendMessage(hList, LB_ADDSTRING, 0, (LPARAM)msg);
SendMessage(hWnd, WM_DESTROY, NULL, NULL);
}
wsprintf(msg, TEXT("server WinSock 초기화 : ver %d.%d"), (int)LOBYTE(wsaData.wVersion), (int)HIBYTE(wsaData.wVersion));
SendMessage(hList, LB_ADDSTRING, 0, (LPARAM)msg);
g_hEvent[0] = WSACreateEvent();
if (g_hEvent[0] == NULL)
{
printError(TEXT("WSACreateEvent error"));
SendMessage(hWnd, WM_DESTROY, 0, 0);
}
hThread = (HANDLE)_beginthreadex(NULL, 0, ThreadProc, NULL, 0, NULL);
CloseHandle(hThread);
return 0;
case WM_COMMAND:
switch (LOWORD(wParam))
{
case ID_BUTTON:
switch (HIWORD(wParam))
{
case BN_CLICKED:
SendMessage(hWnd, WM_DESTROY, NULL, NULL);
break;
}
break;
case ID_LIST:
break;
}
return 0;
case WM_DESTROY:
SetEvent(g_hEvent[0]);
while (0<g_lThreadCnt);
Sleep(100);
WSACloseEvent(g_hEvent[0]);
WSACleanup();
PostQuitMessage(0);
return 0;
}
return(DefWindowProc(hWnd,iMessage,wParam,lParam));
}
void MoveCenter(HWND hWnd)
{
RECT rectDesktop, rectWindow;
GetWindowRect(GetDesktopWindow(),&rectDesktop) ;
GetWindowRect(hWnd,&rectWindow) ;
LONG width, height;
width = rectWindow.right - rectWindow.left ;
height = rectWindow.bottom - rectWindow.top ;
LONG x, y;
x = LONG((rectDesktop.right - width) / 2) ;
y = LONG((rectDesktop.bottom - height) / 2) ;
MoveWindow(hWnd, x, y, width, height, TRUE) ;
}
void printError(TCHAR *errMsg)
{
TCHAR* lpOSMsg;
FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_IGNORE_INSERTS,
NULL,
WSAGetLastError(),
MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
(LPTSTR)&lpOSMsg,
0,
NULL);
TCHAR msg[100];
wsprintf(msg, TEXT("%s :: %s"), errMsg, lpOSMsg);
SendMessage(hList, LB_ADDSTRING, 0, (LPARAM)msg);
MessageBox(hWndMain, msg, TEXT("error"), MB_OK);
LocalFree(lpOSMsg);
SendMessage(hWndMain, WM_DESTROY, NULL, NULL);
}
UINT WINAPI ThreadProc(LPVOID lpParameter)
{
InterlockedIncrement(&g_lThreadCnt);
__try
{
__try
{
InnerThreadProc();
}
__except (EXCEPTION_EXECUTE_HANDLER)
{
}
}
__finally
{
closesocket(gSock);
WSACloseEvent(g_hEvent[1]);
for (int i=0; i<g_lSockCnt; i++)
{
closesocket(gClientSock[i]);
WSACloseEvent(g_phEvtSock[i+2]);
}
InterlockedDecrement(&g_lThreadCnt);
}
return 0;
}
void InnerThreadProc(void)
{
try
{
gSock = socket(PF_INET, SOCK_STREAM, IPPROTO_TCP);
if (gSock == INVALID_SOCKET) throw TEXT("대기 소켓 객체 생성 실패");
g_hEvent[1] = WSACreateEvent();
if (g_hEvent[1] == WSA_INVALID_EVENT) throw TEXT("WSACreateEvent error");
if (WSAEventSelect(gSock, g_hEvent[1], FD_ACCEPT) == SOCKET_ERROR)
throw TEXT("WSAEventSelect error");
struct sockaddr_in addr;
addr.sin_family = AF_INET; //TCP, UDP..
addr.sin_port = htons(1024);
addr.sin_addr.s_addr = inet_addr("127.0.0.1"); //htonl(INADDR_ANY); //inet_addr("127.0.0.1");
//TIME_WAIT인 경우 socket 재사용
bool isReuse = true;
if (setsockopt(gSock, SOL_SOCKET, SO_REUSEADDR, (const char*)&isReuse, sizeof(isReuse)) == SOCKET_ERROR)
throw TEXT("setsockopt error");
if (bind(gSock, (struct sockaddr*)&addr, sizeof(addr)) == SOCKET_ERROR)
throw TEXT("bind error");
if (listen(gSock, SOMAXCONN) == SOCKET_ERROR)
throw TEXT("listen error");
while (true)
{
DWORD dwRet = WSAWaitForMultipleEvents(2+g_lSockCnt, g_hEvent, FALSE, INFINITE, FALSE);
int n = dwRet - WSA_WAIT_EVENT_0;
if (n == 0) break;
SOCKET sock = (2 <=n ? gClientSock[n-2] : gSock);
WSANETWORKEVENTS netEvt;
WSAEnumNetworkEvents(sock, g_hEvent[n], &netEvt);
if (netEvt.lNetworkEvents & FD_ACCEPT)
OnAccept();
if (netEvt.lNetworkEvents & FD_READ)
OnRead(sock);
if (netEvt.lNetworkEvents & FD_CLOSE)
OnClose(n-2);
}
}
catch (TCHAR * errMsg)
{
printError(errMsg);
}
}
void OnAccept(void)
{
sockaddr_in cliaddr;
int addrlen = sizeof(cliaddr);
SOCKET commsock = accept(gSock, (sockaddr*)&cliaddr, &addrlen);
if (g_lSockCnt < MAX_CLIENT)
{
WSAEVENT hEvtSock = WSACreateEvent();
if (hEvtSock != WSA_INVALID_EVENT)
{
WSAEventSelect(commsock, hEvtSock, FD_READ | FD_CLOSE);
g_phEvtSock[g_lSockCnt] = hEvtSock;
gClientSock[g_lSockCnt] = commsock;
g_lSockCnt++;
TCHAR cliip[20];
MultiByteToWideChar( CP_ACP, MB_COMPOSITE, inet_ntoa(cliaddr.sin_addr), -1, cliip, lstrlen(cliip));
TCHAR msg[100];
wsprintf(msg, TEXT("접속수:%d, 접속ip: %s"), g_lSockCnt, cliip);
int n = (int)SendMessage(hList, LB_ADDSTRING, NULL, (LPARAM)msg); //리스트에 문자 추가 후 해당 index를 리턴.
SendMessage(hList, LB_SETTOPINDEX, n, 0); //리스트에 해당 index가 항상 보여지도록 scrolling함
return;
}
}
struct linger ling; //server에서의 TIME_WAIT 방지를 위한 조치
ling.l_onoff = 1;
ling.l_linger = 0;
setsockopt(commsock, SOL_SOCKET, SO_LINGER, (const char*)&ling, sizeof(ling));
closesocket(commsock);
}
void OnRead(SOCKET sock)
{
TCHAR msg[1024] = {0};
int nReadbytes = recv(sock, (char*)msg, sizeof(msg), 0);
if (nReadbytes == SOCKET_ERROR) return;
for (int i=0; i<g_lSockCnt; i++)
send(gClientSock[i], (const char*)msg, nReadbytes, 0);
}
void OnClose(int i)
{
closesocket(gClientSock[i]);
WSACloseEvent(g_phEvtSock[i]);
g_phEvtSock[i] = g_phEvtSock[g_lSockCnt-1];
gClientSock[i] = gClientSock[g_lSockCnt-1];
g_lSockCnt--;
TCHAR msg[100];
wsprintf(msg, TEXT("접속수:%d"), g_lSockCnt);
int n = (int)SendMessage(hList, LB_ADDSTRING, NULL, (LPARAM)msg); //리스트에 문자 추가 후 해당 index를 리턴.
SendMessage(hList, LB_SETTOPINDEX, n, 0); //리스트에 해당 index가 항상 보여지도록 scrolling함
}
|
'C++' 카테고리의 다른 글
| 콘솔로 한글 출력이 안되는 경우 setlocale(LC_ALL, ""); (0) | 2010.10.15 |
|---|---|
| Dialog 창에서 엔터나 ESC 일때 창 닫히는 현상 막기 (0) | 2010.10.13 |
| 윈도우 창을 모니터 화면 가운데로 이동하기 (0) | 2010.10.13 |
| Random 난수 생성하기 (0) | 2010.10.13 |
| Windows Thread 와 Synchronization(동기화) (0) | 2010.10.06 |
| 자료구조 (0) | 2010.09.30 |
| VC++ 수행시간 체크 (1) | 2010.09.15 |
| IME(Input Method Editor) (0) | 2010.09.11 |