MultiPort Listening Tool .. for Webhacking.kr - 31번 ( C )

살짝의 주석 첨가..

#include <stdio.h>
#include <Windows.h>
 
#pragma warning(disable:4996)
#pragma comment(lib, "ws2_32.lib")
 
#define PORT 10000
#define DELAY 20
 
DWORD WINAPI ThreadProc(LPVOID lpParam);    // Thread Proc
VOID Error_Handling(LPSTR msg);                // Error Handling
 
char Msg[256];        // Received Message
HANDLE h[101];        // Thread Handle Array
CRITICAL_SECTION cs;
 
int main()
{
    // 윈속 초기화
    WSADATA WsaData;
    WSAStartup(MAKEWORD(2, 2), &WsaData);
 
    InitializeCriticalSection(&cs);
 
    // 쓰레드 생성
    DWORD i;
    for (i = 0; i <= 100; i++)
    {
        h[i] = CreateThread(NULL, 0, ThreadProc, (LPVOID)i, 0, NULL);
        Sleep(1);
    }
    
    printf("Wait for Connect ....");
 
    // 쓰레드 대기
    DWORD Return = WaitForMultipleObjects(MAXIMUM_WAIT_OBJECTS, h, FALSE, INFINITE);
 
    return 0;
}
 
DWORD WINAPI ThreadProc(LPVOID lpParam)
{
    // 쓰레드 동기화
    EnterCriticalSection(&cs);
 
    CHAR buf[100];
    DWORD i = (int)lpParam;
 
    // 동기화 해제
    LeaveCriticalSection(&cs);
 
    // 소켓 생성
    SOCKET Sock, Clnt;
    Sock = socket(AF_INET, SOCK_STREAM, 0);
    if (Sock == INVALID_SOCKET)
    {
        sprintf(buf, "%d Thread : socket Error", i);
        MessageBox(NULL, buf, "ERROR", MB_OK);
        return 0;
    }
 
    // 주소 할당
    SOCKADDR_IN Serv_Addr, Clnt_Addr;
    Serv_Addr.sin_addr.s_addr = htonl(INADDR_ANY);
    Serv_Addr.sin_family = AF_INET;
    Serv_Addr.sin_port = htons(PORT + i);
 
    if (bind(Sock, (SOCKADDR*)&Serv_Addr, sizeof(SOCKADDR)) == SOCKET_ERROR)
    {
        sprintf(buf, "%d Thread : bind Error", i);
        MessageBox(NULL, buf, "ERROR", MB_OK);
        return 0;
    }
 
    // 포트 대기
    if (listen(Sock, 5) == SOCKET_ERROR)
    {
        sprintf(buf, "%d Thread : listen Error", i);
        MessageBox(NULL, buf, "ERROR", MB_OK);
        return 0;
    }
    printf("%d 번 포트 대기\n", PORT + i);
 
    // 연결 요청 대기
    DWORD len = sizeof(SOCKADDR);
    Clnt = accept(Sock, (SOCKADDR*)&Clnt_Addr, (int*)&len);
    if (Clnt == INVALID_SOCKET)
    {
        sprintf(buf, "%d Thread : accept Error", i);
        MessageBox(NULL, buf, "ERROR", MB_OK);
        return 0;
    }
    system("cls");
 
    // 연결 시
    printf("\r%d 번 포트 연결 !!\n", PORT + i);
    printf("원활한 수신을 위해 5초 대기...\n");
    for (int i = 5; i > 0; i--)
    {
        printf("\r%d....", i);
        Sleep(1000);
    }
    printf("\r           \n");
 
    // 데이터 수신
    len = recv(Clnt, (char*)buf, 100, 0);
    buf[len] = NULL;
 
    closesocket(Sock);
 
    strcpy(Msg, (const char*)buf);
 
    printf("Received String\n");
    printf("%s\n", Msg);
 
    system("pause");
    system("cls");
 
    // 남은 쓰레드 종료
    for (int j = 0; j <= 100; j++)
    {
        if (j == i)
            continue;
        DWORD ExitCode;
        GetExitCodeThread(h[j], &ExitCode);
        TerminateJobObject(h[j], ExitCode);
        printf("%d 번 쓰레드 종료\n", j);
        Sleep(DELAY);
    }
    exit(1);
}
 
VOID Error_Handling(LPSTR msg)
{
    MessageBox(NULL, msg, "Error", MB_OK);
    exit(1);
}