[C++]初学多线程写的端口扫描

admin

1. 
   
2. // PortScan.cpp
   
3. //
   
4. //=============================================================================
   
5. // PortScan - 端口扫描程序
   
6. //
   
7. // XXXXXXXXXXXXXXX
   
8. //
   
9. // XXXXXXXXXXXXXXX
   
10. //=============================================================================
    
11. #include <winsock2.h>
    
12. #include <stdio.h>
    
13. #pragma comment(lib,"ws2_32.lib")
    
14. 
    
15. DWORD WINAPI ScanThread(LPVOID lp);
    
16. 
    
17. //保存扫描的主机IP和端口的结构
    
18. typedef struct{
    
19.         char HostIp[17];
    
20.         int ScanPort;
    
21. }HostStruct;
    
22. 
    
23. long lPortOpen = 0; // 统计开放端口数
    
24. 
    
25. //帮助
    
26. void Help(char *argv0)
    
27. {
    
28.         printf("\nPortScan.exe");
    
29.         printf("\nUsage:");
    
30.         printf("\n\t%s 127.0.0.1 1 3389\n",argv0);
    
31.         exit(0);
    
32. }
    
33. 
    
34. //==============================================================================
    
35. // 线程函数
    
36. //==============================================================================
    
37. DWORD WINAPI ScanThread(LPVOID lp)
    
38. {
    
39.         struct sockaddr_in sin;                // sockaddr_in结构
    
40.         HostStruct *lpScanHost=(HostStruct*)lp;  // 将参数传化成HostStruct结构指针
    
41.         SOCKET s = socket(AF_INET,SOCK_STREAM,0);
    
42. 
    
43.         // 给结构成员赋值
    
44.         sin.sin_family = AF_INET;
    
45.         sin.sin_port = htons(lpScanHost->ScanPort);
    
46.         sin.sin_addr.S_un.S_addr = inet_addr(lpScanHost->HostIp);
    
47. 
    
48.         // 建立连接
    
49.         if(connect(s,(struct sockaddr*)&sin,sizeof(sin)) != SOCKET_ERROR)
    
50.         {
    
51.             printf("\n%s:%d\tOPEN",lpScanHost->HostIp,lpScanHost->ScanPort);
    
52.             InterlockedIncrement(&lPortOpen);                // 原子性++         相对        InterlockedDecrement 原子性--
    
53.         }
    
54.         closesocket(s);        // 无论连接是否成功都需要关闭socket
    
55.         return 0;
    
56. }
    
57. 
    
58. int main(int argc, char* argv[])
    
59. {
    
60.         WORD wVersion = MAKEWORD(2,0); //socket的版本
    
61.         WSADATA wsaData;
    
62.         int iFromPort;        //开始端口
    
63.         int iToPort;          //结束端口
    
64.         int iNowPort;          //正在扫描的端口
    
65.         int iPortCount;        //端口总数
    
66.         int badCount=0;        //线程失败数，有时候失败是因为目标设置了最大连接数
    
67. 
    
68.         //如果命令行下参数不是4个，提示正确的用法
    
69.         if(argc != 4)
    
70.             Help(argv[0]);
    
71. 
    
72.         //保存用户输入的要扫描的起始端口和结束端口,由于用户输入的是char型，所以要先转成int型  
    
73.         iFromPort = atoi(argv[2]);
    
74.         iToPort = atoi(argv[3]);
    
75. 
    
76.         //对用户输入的端口进行判断
    
77.         if(iFromPort > iToPort || iToPort >65535 || iFromPort < 0)
    
78.             return -1;
    
79. 
    
80.         if (WSAStartup(wVersion , &wsaData))
    
81.             return -1;
    
82. 
    
83.         //要扫描的端口总数
    
84.         iPortCount = iToPort - iFromPort + 1;
    
85. 
    
86.         // 根据端口数创建线程句柄数
    
87.         HANDLE *hThreads = new HANDLE[iPortCount];
    
88.         ZeroMemory(hThreads,sizeof(HANDLE)*iPortCount);
    
89.         //
    
90.         // 根据端口数创建线程参数结构指针，使用它只是为了记录结构指针，用于主线程释放动态内存
    
91.         //
    
92.         HostStruct** pScanHosts = new HostStruct*[iPortCount];
    
93.         ZeroMemory(pScanHosts,sizeof(HostStruct*)*iPortCount);
    
94. 
    
95.         printf("\n====== Scanning ======\n");
    
96. 
    
97.         //循环连接端口，以判断端口是否开放
    
98.         int iThreadCount = 0; // 成功创建的线程数目计数
    
99.         int iParamCount = 0;  // 线程句柄数组索引
    
100.         for(iParamCount = 0, iNowPort = iFromPort; iNowPort <= iToPort; iParamCount++,iNowPort++)
     
101.         {
     
102.             // 为单独的线程提供单独的线程参数，此参数不能在局部栈上分配
     
103.             pScanHosts[iParamCount] = new HostStruct;
     
104.             strcpy( pScanHosts[iParamCount]->HostIp,argv[1]);
     
105.             pScanHosts[iParamCount]->ScanPort = iNowPort;
     
106.             HANDLE hThread = CreateThread(NULL,NULL,ScanThread,pScanHosts[iParamCount],NULL,NULL);
     
107.             if(hThread != NULL){
     
108.             hThreads[iThreadCount++] = hThread;//只统计正常运行的线程
     
109.             if(iThreadCount%64==0) Sleep(200);                // 每成功创建64个线程，停顿200毫秒(多次测试后的合理取值)，降低CPU
     
110.             }
     
111.             else
     
112.             badCount++;
     
113.         }
     
114. 
     
115.         // 等待线程执行完毕，这一步是必须的，否则上面分配的动态内存会过早释放，线程执行时参数数据将出现异常
     
116.         // WaitForMultipleObjects 所能等待的对象最大不能超过MAXIMUM_WAIT_OBJECTS (64)个，如果超过，进行分组等待
     
117.         //
     
118.         for(int iGroup = 0; iThreadCount>0; iGroup ++, iThreadCount -= MAXIMUM_WAIT_OBJECTS)
     
119.         {
     
120.             WaitForMultipleObjects(
     
121.             iThreadCount>MAXIMUM_WAIT_OBJECTS ? MAXIMUM_WAIT_OBJECTS : iThreadCount,
     
122.             hThreads+iGroup*MAXIMUM_WAIT_OBJECTS,
     
123.             TRUE,
     
124.             256);              // 此参数单位为毫秒，若为INFINITE则代表一直等待，直到线程返回
     
125.         }
     
126. 
     
127.         printf("\n\n======  OK     ======");
     
128.         printf("\n\tBad threads %d : %d",iPortCount,badCount);
     
129.         printf("\n\tDetected open ports %ld\n",lPortOpen);
     
130. 
     
131.         // 释放所有动态分配的内存
     
132.         delete[iPortCount] hThreads;
     
133.         for(int ii=0; ii < iPortCount; ii ++ )
     
134.             delete pScanHosts[ii];
     
135.         delete[iPortCount] pScanHosts;
     
136.         WSACleanup();
     
137.         return 0;
     
138. }
     

复制代码

真是 收益 匪浅

学习了，谢谢分享、、、