用C写的一个扫描器源码[非阻塞]

以非阻塞方式来实现对IP和端口的连接测试，非阻塞的socket我一直不会用，这段代码有空可以拿来学习学习。

#include <afxext.h>
#include <winsock.h>
// 编译时需使用的库
#pragma comment(lib,"wsock32.lib")
// select()成员定义
#define ZERO (fd_set *)0
// 变量定义
int maxth, scanok, scannum;
int portip, hoststart, hoststop, startport, endport;  //定义了开始I和结束P地址，开始和结束端口
long searchnum, searched;
void usage(char *);    // 定义显示使用方法函数
void playx(int);    // 定义状态提示函数
void setip2(char *);   // 定义设置IP函数
void customport(char *, char *, char *); // 定义自定义扫描端口函数
void portscannow(int);   // 定义端口扫描扫描

// 主程序
int main(int argc, char *argv[])
{
WSADATA wsadata;
// 清屏
system("cls.exe");
// 显示版本信息
printf("\r\n==================== HUC Command Line PortScanner V0.2 ====================");
printf("\r\n=============== By Lion, Welcome to  ===============\r\n\n");

// 检查输入
if ((argc < 3) || (argc > 4))
{
// 显示帮助提示
usage(argv[0]);
return -1;
}

// 检测是否为port扫描
if(!(stricmp(strlwr(argv[1]), "-p") == 0))
{
usage(argv[0]);
return -1;
}

// 程序初始化
if (WSAStartup(MAKEWORD(1,1), &wsadata) != 0)  //如果初始化错误
{
printf("\r\nWsatartup error");           //出错信息
return -1;
}

// 端口扫描参数转换
// 如果参数为三个
if (argc == 3)
{
// 直接设置IP
setip2(argv[2]);
}
// 如果参数为四个
else
if (argc == 4)
{
// 进入定制端口扫描处理
customport(argv[0], argv[2], argv[3]);
}
// 参数过多显示帮助
else
{
usage(argv[0]);
return -1;
}

// 扫描端口开始
portscannow(argc);

WSACleanup();
return 0;
}

// 帮助提示函数
void usage(char * prog)
{
printf("Usage: %s ", prog);
printf("\r\n\n :");
printf("\r\n -p [ Port|StartPort-EndPort ] < HostName|IP|StartIP-EndIP >");
printf("\r\n\n Example: ");
printf("\r\n %s -p 192.168.0.1", prog);
printf("\r\n %s -p 192.168.0.1-192.168.0.254", prog);
printf("\r\n %s -p 21-80 192.168.0.1", prog);
printf("\r\n %s -p 21-80 192.168.0.1-192.168.0.254\r\n", prog);
return;
}

// 进度提示
void playx(int play = 0)
{
// 进度条
char *plays[12]=
{
" | ",
" / ",
" - ",
" \\ ",
" | ",
" / ",
" - ",
" \\ ",
" | ",
" / ",
" - ",
" \\ ",
};

if (searchnum != 0)
{
for (int i = 0 ; i <= 3; i ++)
{
  printf(" =%s= %d%s Completed.  \r", plays , searched * 100 / (searchnum + 1), "%");
  Sleep(5);
}
}
else
{
  printf(" =%s=\r", plays[play]); //显示进度
  Sleep(10);
}
}

// 设置IP
void setip2(char *cp)
{
int host;
struct hostent *testhost;
char *startip = "", *endip = "";

// 判断是否为 192.168.0.1-192.168.0.254 形式的输入
if (strstr(cp, "-") && strlen(cp) > 15 && strlen(cp) < 32)
{
// 提取出结束IP
endip = strchr(cp, '-') + 1;
// 提取出开始IP
strncpy(startip, cp, strlen(cp) - strlen(strchr(cp, '-')));

// 给控制要扫描IP段的变量赋值
hoststart = ntohl(inet_addr(startip));
hoststop = ntohl(inet_addr(endip));
}
else
{
// 取得输入的主机地址
testhost = gethostbyname(startip);

// 如果地址不存在
if(!testhost)
{
  WSACleanup( );
  printf("\r\nCan't get ip of: %s", cp);
  exit(-1);
}
// 给控制要扫描IP段的变量赋值
memcpy(&host, testhost->h_addr, 4);
hoststop = hoststart = ntohl(host);
}
}

// 测试线程是否已满
void TestThread(int thread = 200)
{
for (;;)
{
playx();

// 测试线程是否已满
if (maxth > thread)
  Sleep(100);
else break;
}
return;
}

// 等待线程结束函数
void WaitThreadEnd()
{
// 延时
Sleep(6000);

// 显示等待提示
printf("\r     \r\n");
printf(" Wait ( %d )Thread end...\r\n", maxth);

for(;;)
{
// 判断所有线程是否已经结束
if (maxth > 0)
{
  // 延时等待线程序结束
  Sleep(100);
  playx();
  continue;
}
else break;
}
printf("\r\n");
return;
}

// 定制端口扫描参数定义
void customport(char *cp, char *cp2, char *cp3)
{
int intport;
char *checker;

// 处理要扫描的端口
// 扫描开始端口变量赋值
startport = atoi(cp2);
// 扫描结束端口变量赋值
endport = atoi(cp2);

// 判断是否 21-80 形式
if (strstr(cp2,"-"))
{
intport = atoi(checker = strchr(cp2, '-') + 1);
if (intport > 0 && intport < 65536)
  // 扫描结束端口变量赋值
  endport = intport;
}

// 端口大小判断
if (startport < 0 || startport > 65536 || endport < 0 || endport > 65535)
{
usage(cp);
exit(-1);
}
// 处理ip地址
setip2(cp3);
}

// 端口扫描函数
UINT portscan(LPVOID port)
{
int addr = portip; // 取得要扫描的地址
int sock;
struct fd_set mask;
struct timeval timeout;
struct sockaddr_in server;
unsigned long flag = 1;

// 创建一个sock
sock = socket(AF_INET, SOCK_STREAM, 0);

// 创建sock失败处理
if (sock == INVALID_SOCKET)
{
printf("\r\nSock Error:%s", WSAGetLastError());
maxth --;
return -1;
}

// 给sock成员赋值
server.sin_family=AF_INET;
server.sin_addr.s_addr = htonl(addr); // 要扫描的地址
server.sin_port = htons(port); // 要扫描的端口

// 显示进度
playx();

// 调用ioctlsocket()设置套接字为非阻塞模式
if (ioctlsocket(sock, FIONBIO, &flag) != 0)
{
// 设置失败处理
printf("\r\nSock Error:%s", WSAGetLastError());
closesocket(sock);
maxth --;
return -1;
}

// 调用connect()连接远程主机端口
connect(sock, (struct sockaddr*)&server, sizeof(server));

timeout.tv_sec = 18; // 超时限制为18秒
timeout.tv_usec = 0;
FD_ZERO(&mask);  // 清空集合mask
FD_SET(sock, &mask); // 将sock放入集合mask中

// 用select() 处理扫描结果
switch(select(sock + 1, ZERO, &mask, ZERO, &timeout))
{
case -1:
{
  printf("\r\nSelect() error");
  maxth --;
  return -1;
}

// sock超时处理
case 0:
{
  maxth --;
  closesocket(sock);
  return -1;
}

default:
if(FD_ISSET(sock, &mask))
{
  // 禁止sock发送和接受数据
  shutdown(sock, 0);

  // 设置输出结果格式
  printf(" [Found:] %s Port: %d open.\r\n", inet_ntoa(server.sin_addr), ntohs(server.sin_port));

  // 关闭sock
  closesocket(sock);
  scanok ++;
  maxth --;
  return 1;
}
}
return 0;
}

// 扫描开始主函数
void portscannow(int xp)
{
int sport;
char *timenow, timebuf[32];

// 定义默认扫描的端口
char *ports[32]={
"21",
"22",
"23",
"25",
"53",
"79",
"80",
"110",
"111",
"113",
"123",
"135",
"139",
"143",
"443",
"512",
"513",
"514",
"515",
"540",
"1080",
"1433",
"1521",
"1524",
"3306",
"3389",
"5631",
"6000",
"6112",
"8000",
"8080",
"12345"   //这里你也可以自己定义要扫描的端口
};

// 显示扫描开始的时间
timenow = _strtime(timebuf);
printf("\r\nPortScan Start Time: %s\r\n\n",timenow);

// 计数器初始化.
maxth = 0;
scanok = 0;
scannum = 0;
searched = 0;

// 计算要扫描的端口数量
searchnum = hoststop - hoststart +1;
if(xp == 3)
searchnum = searchnum * 32;
if(xp == 4)
searchnum = searchnum * (endport - startport +1);

// 端口扫描开始
for (portip = hoststart; portip <= hoststop; portip ++, scannum ++)
{
// *.*.*.0和*.*.*.255 地址处理
if ((portip % 256) == 0 || (portip % 256) == 255)
{
  if(xp == 3)
  searchnum = searchnum - 32;
  if(xp == 4)
  searchnum = searchnum - (endport - startport +1);
  scannum --;
  playx();
  continue;
}

if(i > 11) i = 0;
// 默认端口扫描
// scan 192.168.0.1
// scan 192.168.0.1-192.168.0.254
if (xp == 3)
{
  for (sport = 0; sport < 32; sport ++, maxth ++, searched ++)
  {
  // 测试当前线程是否大于180
  TestThread(180);
  // 产生新的线程处理端口扫描
  CWinThread * pthread = AfxBeginThread(portscan,LPVOID(atoi((char*)ports[sport])));
  //延时
  Sleep(120);
  }
}

// 自定义端口扫描
// scan -p 21 192.168.0.1
// scan -p 21-80 192.168.0.1-192.168.0.254
if (xp == 4)
{ // 计算要扫描的端口
  sport = endport - startport;
  if(sport > 500 )
  {
  // 扫描自定义的端口
  for(sport = startport; sport <= endport; sport ++, maxth ++, searched ++)
  {
   TestThread(2000);
   // 产生新的线程处理端口扫描
   CWinThread * pthread = AfxBeginThread(portscan, LPVOID(sport));
   // 延时
   Sleep(10);
  }
  }
  else
  {
  // 扫描自定义的端口
  for(sport = startport; sport <= endport; sport ++, maxth ++, searched ++)
  {
   // 测试当前线程是否大于250
   TestThread(250);
   // 产生新的线程处理端口扫描
   CWinThread * pthread = AfxBeginThread(portscan, LPVOID(sport));
   // 延时
   Sleep(100);
   playx();
  }
  }
}
}

// 等待所有的线程结束
WaitThreadEnd();

// 显示端口扫描结束时间
timenow = _strtime(timebuf);
printf("\r\nPortScan End Time: %s", timenow);
printf("\r\nScan %d Hosts completed. Open %d Ports!\r\n", scannum, scanok);
}

再来一个连接测试代码：


/********************************************/
/****   作者：:夕君                **/
/****   时间：2004.04.04                     **/
/****   北京金万维科技 **/
/*******************************************/
/*此函数实现判断m_server的m_port端口是否可以连上，超时限制为nTimeOut秒*/
BOOL ConnectTest(char * m_server,int m_port)
{

        struct hostent* host = NULL;
        struct sockaddr_in saddr;
        unsigned int s = 0;
        BOOL  ret;
        time_t start;
        int error;
        host = gethostbyname (m_server);
        if (host==NULL)return  FALSE;

        saddr.sin_family = AF_INET;
        saddr.sin_port = htons(m_port);
        saddr.sin_addr = *((struct in_addr*)host->h_addr);


        if( (s=socket(AF_INET, SOCK_STREAM, 0))<0){
                return FALSE;
        }


        fcntl(s,F_SETFL, O_NONBLOCK);

        if(connect(s,(struct sockaddr*)&saddr, sizeof(saddr)) == -1) {
                if (errno == EINPROGRESS){// it is in the connect process
                        struct timeval tv;
                        fd_set writefds;
                        tv.tv_sec = m_nTimeOut;
                        tv.tv_usec = 0;
                        FD_ZERO(&writefds);
                        FD_SET(s, &writefds);
                        if(select(s+1,NULL,&writefds,NULL,&tv)>0){
                                int len=sizeof(int);
                               //下面的一句一定要，主要针对防火墙
                                getsockopt(s, SOL_SOCKET, SO_ERROR, &error, &len);
                                if(error==0) ret=TRUE;
                                else ret=FALSE;
                        }else   ret=FALSE;//timeout or error happen
                }else ret=FALSE;
        }
        else    ret=TRUE;

        close(s);
        return ret;


}