Simple Port Scanner 업그레이드 No.1

Simple Port Scanner의 경우 실행시켜보면 알겠지만,

매우 느리다는 것을 알 수 있다.

그 이유는 기본 TCP 접속을 통해서 해당 포트번호가 열렸는지를 체크할 때 걸리는 Timeout 세팅 때문이다.

Windows의 경우 레지스트리에 세팅이 되어 있는데, 72초라고 하네.

http://serverfault.com/questions/193160/which-is-the-default-tcp-connect-timeout-in-windows

아무튼 이러한 Timeout * 루프 카운트를 하게 되면 엄청난 시간이 걸리게 되는 것은 인지상정!

결국 Timeout + 한번에 한번만 접속 시도하는 Loop 때문에 생기는 문제다.

그럼 이것을 병렬처리하면 당연히 속도가 빠르지 않겠는가?

병렬처리를 위해 우리는 try_tcp_port 함수를 수정해보도록 하자.

... if ((sock = socket (PF_INET, SOCK_STREAM, 0)) < 0) // 소켓 생성 { printf ("Couldn't make socket!\n"); exit(-1); } setsockopt (sock, SOL_SOCKET, SO_RCVTIMEO, (const char*)&timeout, sizeof (timeout)); /* Setting up the sockaddr_in struct with connection details, port, 'family', hostname/IP address */ ...

setsockopt 함수설명은 아래 링크로 대체하겠다.

http://synch3d.com/winsock/setsockopt.html

여기서 사용된 timeout은 timeval 구조체를 사용하였다.

... // 전역변수 struct timeval timeout; int timeout_sec = 3; ... void TCPScanning () { ... printf (BLUE ... timeout.tv_sec = timeout_sec; tmeout.tv_usec = timeout.sec;

자, 이제 여기까지 하고 실행해도 엄청 빠르게 실행되는 것을 볼 수 있다.

그러면 병렬로 처리해서 속도를 한번 더 내볼까?

병렬로 connect하기 위해 thread를 사용해 봅시다~

gcc SimplePortScanner.c -o SimplePortScanner -lpthread

#include <pthread.h> ... int main (int argc, char *argv[]) { ... switch (option) { case 0: TCPScanning (); break; case 1: bParallel = TRUE; TCPScanning (); break; } ... } void TCPScanning () { ... for (i = start_port; i <= end_port; i++) { if (bParallel== TRUE) { pthread_t thread_t; n_threads++; thread_id = pthread_create (&thread_t, NULL, try_tcp_port, (void *)i); if (thread_id < 0) { i--; n_threads--; } pthread_detach (thread_t); } else { try_tcp_port ((void *)i); } } ... } ... int printOption () { printf ("Choose scanning type...\n"); printf ("[0] : TCP Scanning\n"); printf ("[1] : TCP Scanning - Parallel\n"); printf ("> "); scanf ("%d", &option); if (option < 0 || option > 1) return FAILED; else return SUCCESS; }