Windows多线程端口扫描

还不是很了解多线程，先简单运用下。

/*
2015.5    HT
多线程端口扫描

netstat -an

*/

#include <iostream>
#include <process.h>
#include <winsock2.h>
#pragma comment(lib,"ws2_32.lib")
using namespace std;

// 线程传递的端口参数
typedef struct 
{
    unsigned int min_port;
    unsigned int max_port;
}Port;
Port *lpPort = new Port;
// 线程数
int Thread_NUM;
// 互斥量
HANDLE hMutex;

DWORD WINAPI scan(LPVOID lpParameter)
{
    // 获得参数
    Port* port = (Port*)lpParameter;
    SOCKET sockfd = NULL;
    SOCKADDR_IN sin;
    sin.sin_family = AF_INET;
    sin.sin_addr.S_un.S_addr = inet_addr("127.0.0.1");

    for (unsigned int i = port->min_port; i <= port->max_port; ++i)
    {
        if (sockfd == NULL)
        {
            // 创建套接字
            sockfd = socket(AF_INET, SOCK_STREAM, 0);
            if (sockfd == INVALID_SOCKET)
            {
                cout << "创建socket失败！" << endl;
                return 0;
            }
        }
        sin.sin_port = htons(i);
        // 连接服务器
        int ret = connect(sockfd, (sockaddr *)&sin, sizeof(sockaddr));
        // 请求一个互斥量的访问权
        WaitForSingleObject(hMutex, INFINITE);
        if (ret == SOCKET_ERROR)
        {
            cout << "Port: " << i << "    not open !" << endl;
        }
        else
        {
            cout << "Port: " << i << "    open !" << endl;
            closesocket(sockfd);
            sockfd = NULL;
        }
        // 释放一个互斥量的访问权
        ReleaseMutex(hMutex);
    }

    return 0;
}

int main()
{
    WSADATA wsaData;
    if (WSAStartup(MAKEWORD(2, 2), &wsaData) != 0)
    {
        cout << "找不到 WinSock dll!" << endl;
        return 0;
    }

    hMutex = CreateMutex(NULL, FALSE, NULL);

    cout << "输入起始端口，结束端口号： " ;
    cin >> lpPort->min_port >> lpPort->max_port;
    cout << "输入线程数(小于200)： ";
    cin >> Thread_NUM;
    HANDLE handle[200];
    int Scan_Len = (lpPort->max_port - lpPort->min_port) / Thread_NUM;
    
    for (int i = 0; i < Thread_NUM; ++i)
    {
        Port *NewPort = new Port;
        NewPort->min_port = lpPort->min_port + Scan_Len * i;
        if (i < (Thread_NUM - 1))
            NewPort->max_port = NewPort->min_port + Scan_Len - 1;
        else
        {
            NewPort->max_port = lpPort->max_port;
            handle[i] = (HANDLE)CreateThread(NULL, 0, (LPTHREAD_START_ROUTINE)scan, NewPort, 0, NULL);
            break;
        }
        handle[i] = (HANDLE)CreateThread(NULL, 0, (LPTHREAD_START_ROUTINE)scan, NewPort, 0, NULL);
    }
    WaitForMultipleObjects(Thread_NUM, handle, TRUE, INFINITE);

    CloseHandle(hMutex);
    return 0;
}