【转帖】多线程tcp扫描器

yogy

/**********************************************************************
gscan.cpp
多线程tcp connect扫描器

E-mail:gxisone@hotmail.com
最后更新：2004/8/5
*********************************************************************/
#include <stdio.h>
#include<winsock2.h>
#include <time.h>
#pragma comment(lib,"ws2_32.lib")

char *host;
int threadnum,maxthread,totalport;
long nowport;


TIMEVAL timeout;
FD_SET    mask;
void usage(char *name)
{
printf("\t===================Portscaner======================\n");
printf("\t============gxisone@hotmail.com     2004/7/6=======\n");
printf("\tusage: %s IP StartPort-EndPort MaxThread(1000)\n",name);
printf("\tExample: %s 192.168.1.1 1-10000 500\n",name);
}


void display(void)  // 定义状态提示函数
{
static int play=0;
// 进度条
char *plays[12]=
{
" | ",
" / ",
" - ",
" \\ ",
" | ",
" / ",
" - ",
" // ",
" | ",
" / ",
" - ",
" // ",
};

printf("=%s=\t%d threads %d %s Completed. \r", plays[play],threadnum,nowport*100/(totalport+1),"%");
play=(play==11)?0:play+1;
Sleep(1);

}




void WaitThreadEnd(void)
{
Sleep(1000);
printf("\nThread ending....\n");
while(threadnum>0)
{
  Sleep(1);
  printf(" =|=\t%d threads \r",threadnum);


}

printf("\nThread ended!\n");
}

DWORD WINAPI ThreadFunc( LPVOID lp ) //线程函数，跟普通的函数没什么两样
{   

int port=*(DWORD*)lp;

    SOCKET sockfd;
    struct sockaddr_in addr;
   
   
    u_long value;
    addr.sin_family =AF_INET;
addr.sin_addr.s_addr =inet_addr(host);
  
  value=1;
   
   

sockfd = socket(AF_INET, SOCK_STREAM, 0);
    if(sockfd==INVALID_SOCKET)
    {   
  printf("Socket error!\n");
  InterlockedExchangeAdd((long *)&threadnum,-1);
        return 0;
    }
    ioctlsocket(sockfd,FIONBIO,&value);
addr.sin_port = htons(port);
   

    connect(sockfd,(struct sockaddr *) &addr, sizeof(addr));

   
    FD_ZERO(&mask);
    FD_SET(sockfd,&mask);
    value=select(0,NULL,&mask,NULL,&timeout);
    if(value==0 || value==-1)
    {   
  closesocket(sockfd);
  Sleep(50);
      InterlockedExchangeAdd((long *)&threadnum,-1);
   return 0;
      
    }
    else
{
  shutdown(sockfd, 0);
  printf("\t\tFound port %d open.\r\n",port);
  closesocket(sockfd);
  Sleep(50);
    InterlockedExchangeAdd((long *)&threadnum,-1);
}
   
   
   
return 0;
}

VOID main( int argc,char **argv )
{

WSADATA ws;
char *p;
int startport,endport;
clock_t start,end;//程序运行的起始和结束时间
float costtime;//程序耗时



if(argc!=4)
{
usage(argv[0]);
return ;
}

   

long lresult;

lresult=WSAStartup(MAKEWORD(2,2), &ws);

p=argv[2];//处理端口参数
    if(strstr(argv[2],"-"))
    {    startport=atoi(argv[2]);
        for(;*p;)
            if(*(p++)=='-')break;
        endport=atoi(p);
        
        if(startport<1 || endport>65535)
        {   
   printf("ort Error!\n");
            return;
        }
    }
host=argv[1];
maxthread=(atoi(argv[3])-1>999)?999:atoi(argv[3])-1;
    usage(argv[0]);

timeout.tv_usec=0;
if(maxthread>500)timeout.tv_sec=2;
else timeout.tv_sec=1;

     start=clock();//开始计时

  totalport=endport-startport;
   
for(int port=startport;port<endport;port++,nowport++,threadnum++)
{
     display();
  while(threadnum>maxthread)
{
Sleep(10);
}
  CreateThread( NULL,0,ThreadFunc,&port, 0,NULL);
}

end=clock();//计时结束
costtime= (float)(end - start) / CLOCKS_PER_SEC;  //转换时间格式
printf("Cost time:%f Sec",costtime);//显示耗时
printf("\n\n");
WaitThreadEnd();
   

WSACleanup();
return ;
}