예제를 따라 치고 있는데 말이야.
#include <winsock2.h>
#include <stdio.h>
#include <stdlib.h>
#pragma comment (lib, "ws2_32.lib")
const int BUFSIZE = 512;
struct SOCKETINFO
{
SOCKET sock;
int recvbytes;
int sendbytes;
WSAOVERLAPPED overlapped;
WSABUF wsabuf;
char buf [BUFSIZE+1];
};
SOCKETINFO * SocketInfoArr[WSA_MAXIMUM_WAIT_EVENTS];
WSAEVENT EventArr[WSA_MAXIMUM_WAIT_EVENTS];
int nTotalSockets = 0;
CRITICAL_SECTION cs;
DWORD WINAPI WorkerThread(LPVOID pParam);
BOOL AddSocketInfo(SOCKET sock);
void RemoveSocketInfo(int nIndex);
void err_display(char * msg);
void err_quit(char * msg);
int main()
{
int retval;
InitializeCriticalSection(&cs);
WSADATA wsa;
if(WSAStartup(MAKEWORD(2,2), &wsa) != 0)
return -1;
SOCKET listen_sock, client_sock;
SOCKADDR_IN serveraddr, clientaddr;
int addrlen = sizeof(clientaddr);
listen_sock = socket(AF_INET, SOCK_STREAM, 0);
if(listen_sock == INVALID_SOCKET)
err_quit("socket()");
ZeroMemory(&serveraddr,sizeof(serveraddr));
serveraddr.sin_addr.s_addr = htonl(INADDR_ANY);
serveraddr.sin_family = AF_INET;
serveraddr.sin_port = htons(9990);
retval = bind(listen_sock, (SOCKADDR*) &serveraddr, sizeof(serveraddr));
if(retval == SOCKET_ERROR)
err_quit("bind()");
retval = listen(listen_sock, SOMAXCONN);
if(retval == SOCKET_ERROR)
err_quit("listen()");
//특별한 일을 하는 이벤트(?) ......아놔 . ㅋㅋㅋ
WSAEVENT hEvent = WSACreateEvent();
if(hEvent == WSA_INVALID_EVENT)
err_quit("WSACreateEvent()");
EventArr[nTotalSockets++] = hEvent;
DWORD dwThreadId;
HANDLE hThread = CreateThread(NULL, 0, WorkerThread, NULL, 0, &dwThreadId);
if(hThread == NULL)
err_quit("CreateThread()");
while(1){
client_sock = accept(listen_sock, (SOCKADDR*) &clientaddr, &addrlen);
if(client_sock == INVALID_SOCKET){
err_display("accept()");
continue;
}
if(AddSocketInfo(client_sock) == FALSE){
closesocket(client_sock);
continue;
}
printf("[TCP 서버] 클라이언트 접속 : %s %d \\n",inet_ntoa(clientaddr.sin_addr), ntohs(clientaddr.sin_port));
SOCKETINFO * ptr = SocketInfoArr[nTotalSockets-1];
DWORD recvbytes;
DWORD flags = 0;
retval = WSARecv(ptr->sock, &(ptr->wsabuf), 1, &recvbytes, &flags, &(ptr->overlapped), NULL);
if(retval == SOCKET_ERROR){
if(WSAGetLastError() != WSA_IO_PENDING){
err_display("WSARecv()");
RemoveSocketInfo(nTotalSocket-1);
continue;
}
}
if(WSASetEvent(EventArr[0]) == FALSE){
err_display("WSASetEvent()");
break;
}
}
WSACleanup();
DeleteCriticalSection(&cs);
return 0;
}
DWORD WINAPI WorkerThread(LPVOID pParam)
{
int retval;
while(1){
DWORD index = WSAWaitForMultipleEvents(nTotalSockets, EventArr, FALSE, WSA_INFINITE, FALSE);
if(index == WSA_WAIT_FAILED){
err_display("WSAWaitForMultipleEvents()");
continue;
}
index -= WSA_WAIT_EVENT_0;
WSAResetEvent(EventArr[index]);
if(index == 0)
continue;
SOCKETINFO * ptr = SocketInfoArr[index];
SOCKADDR_IN clientaddr;
int addrlen = sizeof(clientaddr);
getpeername(ptr->sock, (SOCKADDR*) &clientaddr, &addrlen);
DWORD cbTrans, flags;
retval = WSAGetOverlappedResult(ptr->sock, &(ptr->overlapped), &cbTrans, FALSE, &flags);
if(retval == FALSE || cbTrans == 0){
if(retval == FALSE)
err_display("WSAGetOverlappedResult()");
RemoveSocketInfo(index);
printf("[TCP서버] 클라이언트 종료 : %s %d\\n", inet_ntoa(clientaddr.sin_addr), ntohs(clientaddr.sin_port));
continue;
}
if(ptr->recvbytes == 0){
ptr->recvbytes = cbTrans;
ptr->sendbytes = 0;
ptr->buf[ptr->recvbytes] = \'\\0\';
printf("[TCP/%s:%d] %s \\n", inet_ntoa(clientaddr.sin_addr), ntohs(clientaddr.sin_port), ptr->buf);
}
else
ptr->sendbytes += cbTrans;
if(ptr->recvbytes > ptr->sendbytes){
ZeroMemory(&(ptr->overlapped), sizeof(ptr->overlapped));
ptr->overlapped.hEvent = EventArr[index];
ptr->wsabuf.buf = ptr->buf + ptr->sendbytes;
ptr->wsabuf.len = ptr->recvbytes - ptr->sendbytes;
DWORD sendbytes;
retval = WSASend(ptr->sock, &(ptr->wsabuf), 1, &sendbytes, 0 , &(ptr->overlapped), NULL);
if(retval == SOCKET_ERROR){
if(WSAGetLastError() != WSA_IO_PENDING){
err_display("WSASend()");
}
continue;
}
}
else{
ptr->recvbytes = 0;
ZeroMemory(&(ptr->overlapped), sizeof(ptr->overlapped));
ptr->overlapped.hEvent = EventArr[index];
ptr->wsabuf.buf = ptr->buf;
ptr->wsabuf.len = BUFSIZE;
DWORD recvbytes;
flags = 0;
retval = WSARecv(ptr->sock, &(ptr->wsabuf), 1, &recvbytes, &flags, &(ptr->overlapped), NULL);
if(retval == SOCKET_ERROR){
if(WSAGetLastError() != WSA_IO_PENDING){
err_display("WSARecv()");
}
continue;
}
}
}
}
BOOL AddSocketInfo(SOCKET sock)
{
EnterCriticalSection(&cs);
if(nTotalSockets >= WSA_MAXIMUM_WAIT_EVENTS){
printf("[오류] 접속 최대수 초과 접속 불가\\n");
LeaveCriticalSection(&cs);
return FALSE;
}
SOCKETINFO * ptr = new SOCKETINFO;
if(ptr == NULL){
printf("[오류] 메모리 부족!!\\n");
delete ptr;
LeaveCriticalSection(&cs);
return FALSE;
}
WSAEVENT hEvent = WSACreateEvent();
if(hEvent == WSA_INVALID_EVENT){
err_display("WSACreatEvent()");
delete ptr;
LeaveCriticalSection(&cs);
return FALSE;
}
::ZeroMemory(&(ptr->overlapped), sizeof(ptr->overlapped));
ptr->sock = sock;
ptr->recvbytes = 0;
ptr->sendbytes = 0;
ptr->wsabuf.buf = ptr->buf;
ptr->wsabuf.len = BUFSIZE;
SocketInfoArr[nTotalSockets] = ptr;
EventArr[nTotalSockets] = hEvent;
ptr->overlapped.hEvent = EventArr[nTotalSockets];
nTotalSockets++;
LeaveCriticalSection(&cs);
return TRUE:
}
void RemoveSocketInfo(int nIndex)
{
EnterCriticalSection(&cs);
SOCKETINFO * ptr = SocketInfoArr[nIndex];
closesocket(ptr->sock);
delete ptr;
WSACloseEvent(EventArray[index]);
for(int i=index;i<nTotalSocket;i++){
SocketInfoArr[i] = SocketInfoArr[i+1];
EventArr[i] = EventArr[i+1];
}
nTotalSocket--;
LeaveCriticalSection(&cs);
}
댓글 영역
획득법
① NFT 발행
작성한 게시물을 NFT로 발행하면 일주일 동안 사용할 수 있습니다. (최초 1회)
② NFT 구매
다른 이용자의 NFT를 구매하면 한 달 동안 사용할 수 있습니다. (구매 시마다 갱신)
사용법
디시콘에서지갑연결시 바로 사용 가능합니다.