ＶＣ＋＋多线程tcp connect扫描

#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("Port 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 ;
}