首页 > 编程语言 > 详细

c++ 多线程(2000个)端口扫描(附源码)

时间:2019-05-07 17:13:04      阅读:252      评论:0      收藏:0      [点我收藏+]

程序适应环境与基本介绍

  • 情况:开2000个线程,扫描主机上开启的端口,扫描时间40秒左右。
  • 瓶颈:不管开5000还是更多,都不能大幅加快扫描时间。
  • 瓶颈解决方法:可以使用最常被开放的1000个端口列表进行扫描,网上应该有,社会学+编程。
  • 注意:socket是宝贵的系统资源,不用要关闭;多线程中临界区资源要加锁。
  • 编程环境:vs2008(c98)
  • 代码如下:
#include "stdafx.h"
#include <stdio.h>
#include <Winsock2.h>
#pragma comment(lib,"Ws2_32")//socket要用到的系统库

//目标地址
#define IP "127.0.0.1"
//线程个数
#define THREADCOUNT 2000
DWORD WINAPI ThreadProc(  LPVOID lpParameter);
//端口号
int PortNum=0;
//临界区变量
CRITICAL_SECTION cs;
//线程函数
DWORD WINAPI ThreadProc(  LPVOID lpParameter)
{
    //创建套接字
    SOCKET TryConnect;
    while (1)
    {
        if (PortNum>65535)
        {
            break;
        }
        //进入临界区
        EnterCriticalSection(&cs);
        int tmpport = PortNum;
        PortNum++;
        //DWORD threadID=GetCurrentThreadId();
        //printf("线程%d正在检测端口%d\n",threadID,PortNum);//所有使用临界区资源的代码都要加锁
        //离开临界区
        LeaveCriticalSection(&cs);

        TryConnect = socket(AF_INET,SOCK_STREAM,0);
        if (INVALID_SOCKET == TryConnect)
        {
            printf("Invalid socket.\n");
        }
        //尝试连接
        sockaddr_in addrMe = {0};
        addrMe.sin_family=AF_INET;
        
        addrMe.sin_port = htons(tmpport);
        addrMe.sin_addr.S_un.S_addr = inet_addr(IP);
        int retCon = connect(TryConnect,(sockaddr*)&addrMe,sizeof(sockaddr_in));
        if (SOCKET_ERROR != retCon)
        {
            printf("检测到目标主机开放%d端口\n",tmpport);
        }
        closesocket(TryConnect);//防止开启太多socket连接,导致后面socket分配无效
    }
    return 0;
}
int main(int argc, char* argv[])
{
    //初始化套接字
    WSADATA ws;
    ::WSAStartup(MAKEWORD(2,0),&ws);

    DWORD start = GetTickCount();
    //初始化临界区
    InitializeCriticalSection(&cs);  

    //多线程扫描
    HANDLE hThread[THREADCOUNT];
    for (int i=0;i<THREADCOUNT;i++)
    {
        hThread[i] = CreateThread(NULL,0,(LPTHREAD_START_ROUTINE)ThreadProc,(LPVOID)0,0,NULL);
    }

    //当thread数量超过64的处理
    int tempNumThreads = THREADCOUNT; 
    int tempMax = 0; 
    while( tempNumThreads >= MAXIMUM_WAIT_OBJECTS ) 
    { 
        tempNumThreads -= MAXIMUM_WAIT_OBJECTS; 
        WaitForMultipleObjects( MAXIMUM_WAIT_OBJECTS, &hThread[ tempMax ], false, INFINITE); 
        tempMax += MAXIMUM_WAIT_OBJECTS; 
    } 
    WaitForMultipleObjects( tempNumThreads, &hThread[ tempMax ], false, INFINITE); 

    //删除临界区
    DeleteCriticalSection(&cs);

    DWORD end = GetTickCount();
    printf("use time(s):%f\n", (end-start)/1000.0);
    system("pause");
    return 0;
}

c++ 多线程(2000个)端口扫描(附源码)

原文:https://www.cnblogs.com/woniu-felix/p/10826157.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!