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