一个网络扫描程序

1 // ScanNet.cpp: implementation of the ScanNet class.
2 //
3 //////////////////////////////////////////////////////////////////////
4
5 #include "stdafx.h"
6 #include "ty_server.h"
7 #include "ScanNet.h"
8
9 #ifdef _DEBUG
10 #undef THIS_FILE
11 static char THIS_FILE[]=__FILE__;
12 #define new DEBUG_NEW
13 #endif
14
15
16 #define MYSQLIP ("127.0.0.1")
17
18 HANDLE hMutex = NULL;
19 int selfip ; //本机尾ip
20 //////////////////////////////////////////////////////////////////////
21 // Construction/Destruction
22 //////////////////////////////////////////////////////////////////////
23
24 ScanNet::ScanNet()
25 {
26
27 }
28
29 ScanNet::~ScanNet()
30 {
31
32 }
33
34
35 //根据数组组合成字符串
36 char * ScanNet::fmtIPString(int *IP)
37 {
38 char *vResult=new char(16);
39 char *tmp=new char[4];
40
41 _itoa(IP[0],vResult,10);
42 strcat(vResult,".");
43
44 _itoa(IP[1],tmp,10);
45 strcat(vResult,tmp);
46 strcat(vResult,".");
47
48 _itoa(IP[2],tmp,10);
49 strcat(vResult,tmp);
50 strcat(vResult,".");
51
52 _itoa(IP[3],tmp,10);
53 strcat(vResult,tmp);
54 delete tmp;
55 return vResult;
56 }
57
58
59 //根据获取的主机信息，组合为hostmg，然后发送消息
60 void ScanNet::GetHostMsg(char *strip,char *strmac)
61 {
62 // AfxMessageBox("addr: %s mac:%s\n",strip,strmac);
63
64 unsigned char mac[6];
65 char machine_mac[6];
66 unsigned long machine_ip = inet_addr( strip );
67 mac_aton(strmac, mac, ":" );
68 mac_ntoa(mac,machine_mac,"");
69
70 // char adr[100];
71 // sprintf(adr,"addr: %u mac:%s\n",machine_ip,machine_mac);
72 // AfxMessageBox(adr);
73
74 WaitForSingleObject(hMutex,INFINITE);
75
76 yCMysql mysql;
77 if (!mysql.Login (MYSQLIP, "amtty", "", "db"))
78 {
79 mysql.Logout ();
80 DEBUG (gbl.is_debug, gbl.debug_level, gbl.log_file, "ScanNet error : Login db fail\n");
81 return ;
82 }
83
84 char sql[300] = {0};
85 sprintf(sql,"select * from tab_zxtj where ip=%u",machine_ip);
86 if(!mysql.QueryRead(sql))
87 {
88 mysql.Logout();
89 DEBUG (gbl.is_debug, gbl.debug_level, gbl.log_file, "ScanNet error : QueryRead sql (%s) fail\n", sql);
90 return ;
91 }
92
93 if(!mysql.GetRowNum()) //如果在线状态表中无此机器,插入此机器
94 {
95 sprintf(sql,"insert into tab_zxtj(ip,mac) values(%u,\'%s\')",machine_ip,machine_mac); //默认状态位为0,报警位为0
96 if (!mysql.QueryWrite (sql))
97 {
98 mysql.Logout ();
99 DEBUG (gbl.is_debug, gbl.debug_level, gbl.log_file, "ScanNet modify tab_zxtj error : QueryWrite sql (%s) fail\n", sql);
100 return ;
101 }
102 }
103 else
104 {
105 mysql.FreeResult();
106 sprintf(sql,"select * from tab_zxtj where ip=%u and (jqzt = %d or jqzt = %d)",machine_ip,0,3); //第二次扫描时jqzt位还是0或者3，报警
107 if(!mysql.QueryRead(sql))
108 {
109 mysql.Logout();
110 DEBUG (gbl.is_debug, gbl.debug_level, gbl.log_file, "ScanNet error : QueryRead sql (%s) fail\n", sql);
111 return ;
112 }
113 if(mysql.GetRowNum())
114 {
115 mysql.FreeResult();
116 sprintf(sql,"update tab_zxtj set jqzt = %d where ip = %u",3,machine_ip); //jqzt位为3报警
117 if (!mysql.QueryWrite (sql))
118 {
119 mysql.Logout ();
120 DEBUG (gbl.is_debug, gbl.debug_level, gbl.log_file, "ScanNet modify tab_zxtj error : QueryWrite sql (%s) fail\n", sql);
121 return ;
122 }
123
124 }
125 else
126 {
127 sprintf(sql,"update tab_zxtj set jqzt = %d where ip =%u",0,machine_ip);
128 if (!mysql.QueryWrite (sql))
129 {
130 mysql.Logout ();
131 DEBUG (gbl.is_debug, gbl.debug_level, gbl.log_file, "ScanNet modify tab_zxtj error : QueryWrite sql (%s) fail\n", sql);
132 return ;
133 }
134
135 }
136
137 }
138 mysql.Logout ();
139 ReleaseMutex(hMutex);
140 }
141
142 void ScanNet::GetMac(char *strip,char *m_HostMAC)
143 {
144 HRESULT hr;
145 IPAddr ipAddr;
146 ULONG pulMac[2];
147 ULONG ulLen;
148
149 ipAddr = inet_addr( strip );
150 memset( pulMac, 0xff, sizeof(pulMac) );
151 ulLen = 6;
152
153 hr = SendARP( ipAddr, 0, pulMac, &ulLen );
154 // DEBUG ( gbl.is_debug, gbl.debug_level, gbl.log_file, "发送ARP请求结果: %d\n", GetLastError());
155
156 int i,j;
157 PBYTE pbHexMac = (PBYTE) pulMac;
158 for (i=0, j=0; i < 6; ++i)
159 {
160 j += sprintf(m_HostMAC+j, "%02x:", pbHexMac[i]);
161 }
162 m_HostMAC[17] = '\0';
163 // DEBUG ( gbl.is_debug, gbl.debug_level, gbl.log_file, "MAC地址: %s\n", m_HostMAC);
164
165 }
166
167
168 void __cdecl threadproc ( LPVOID pParam )
169 {
170 in_addr s_IP;
171 hostent *host;
172 int *startIP = (int *)pParam;
173 char *strip = ScanNet::fmtIPString(startIP);
174 char *strmac = new char[20];
175 s_IP.S_un.S_addr = inet_addr(strip);
176 host=gethostbyaddr((const char*)&s_IP,sizeof(s_IP),AF_INET);
177 if(host)
178 {
179 ScanNet::GetMac(strip,strmac);
180 ScanNet::GetHostMsg(strip,strmac);
181 }
182 delete strip;
183 delete strmac;
184 }
185
186
187 //实际根据IP地址获取主机信息
188 void ScanNet::GetHostByAddr(int *startIP,int *endIP)
189 {
190 for(; startIP[3] <= endIP[3]; startIP[3]++)
191 {
192 // if(startIP[3] == selfip) //本机ip跳过
193 // {
194 // continue;
195 // }
196 if ( _beginthread(threadproc, 0, (void *)startIP) < 0)
197 {
198 DEBUG (gbl.is_debug, gbl.debug_level, gbl.log_file,"(ScanNet)启动扫描线程失败!\n");
199 continue;
200 }
201 Sleep(30);
202 if(startIP[3] % 30 == 0)
203 {
204 Sleep(10000); //启动30个线程暂停10秒
205 }
206 }
207
208 }
209
210 //确定IP地址的类型,并将IP地址串解析为四维的整数数组
211 void ScanNet::ParasIP(char *Start_IP,char *End_IP,int *int_startip,int *int_endip)
212 {
213 char s_startip[16];
214 char s_endip[16];
215 strcpy(s_startip,Start_IP);
216 strcpy(s_endip,End_IP);
217
218 int Index; //整数数组的索引
219 char *tmp,*tmp1;
220
221 //处理第一个IP地址：
222 Index=0;
223 tmp=s_startip;
224 tmp1=s_startip;
225
226 while(Index<4)
227 {
228 if(*tmp1=='.')
229 {
230 *tmp1=NULL; //'.'已被赋为null
231 int_startip[Index]=atoi(tmp);
232 Index++;
233 tmp1++;
234 tmp=tmp1;
235 }
236 else if(*tmp1==NULL)
237 {
238 selfip = atoi(tmp); //本机ip
239 int_startip[Index]=/*atoi(tmp)*/1; //扫描起点
240 break;
241 }
242 else tmp1++;
243 }
244
245 //处理第二个IP地址
246 Index=0;
247 tmp=s_endip;
248 tmp1=s_endip;
249
250 while(Index<4)
251 {
252 if(*tmp1=='.')
253 {
254 *tmp1=NULL;
255 int_endip[Index]=atoi(tmp);
256 Index++;
257 tmp1++;
258 tmp=tmp1;
259 }
260 else if(*tmp1==NULL)
261 {
262 int_endip[Index]=/*atoi(tmp)*/254; //扫描终点
263 break;
264 }
265 else tmp1++;
266 }
267
268 }
269
270
271 void ScanNet::ScanHostByAddr(char *HostAddrs,char *HostAddre)
272 {
273 int vsip[4]={0,0,0,0};
274 int veip[4]={0,0,0,0};
275 ParasIP(HostAddrs,HostAddre,vsip,veip); //将字符串解析为数组
276 GetHostByAddr(vsip,veip); //实际的扫描程序
277
278 }
279
280 void ScanNet::Work()
281 {
282 //////////////////
283 // 获得主机名.
284 //////////////////
285 char *m_HostIP = new char[19];
286 char hostname[256];
287 int res = gethostname( hostname, sizeof(hostname) );
288 if (res != 0)
289 {
290 AfxMessageBox("Error1");
291 return ;
292 }
293 // DEBUG ( gb.lis_debug, gbl.debug_level, gbl.log_file, "主机名: %s\n", hostname);
294
295 ////////////////
296 // 根据主机名获取主机信息.
297 //
298 hostent* pHostent = gethostbyname(hostname);
299 if (pHostent == NULL)
300 {
301 AfxMessageBox( "Error: 2");
302 return ;
303 }
304 // DEBUG ( gbl.is_debug, gbl.debug_level, gbl.log_file, "主机信息: %s\n", pHostent);
305
306 //////////////////
307 // 解析返回的hostent信息.
308 //
309 hostent& he = *pHostent;
310 sockaddr_in sa;
311 for ( int nAdapter=0; he.h_addr_list[nAdapter]; nAdapter++ )
312 {
313 memcpy ( &sa.sin_addr.s_addr, he.h_addr_list[nAdapter],he.h_length );
314 strcpy( m_HostIP,inet_ntoa(sa.sin_addr) );
315 }
316 DEBUG ( gbl.is_debug, gbl.debug_level, gbl.log_file, "(ScanNet)主机ip: %s\n", m_HostIP);
317 ScanHostByAddr(m_HostIP,m_HostIP);
318
319 }
320
321 /*功能：初始化
322 入参：无
323 出参：无
324 */
325 int ScanNet::Init ()
326 {
327 WSADATA wsaData;
328 int vst=WSAStartup(MAKEWORD(1,1),&wsaData);
329 if(vst == 0)
330 {
331 DEBUG (gbl.is_debug, gbl.debug_level, gbl.log_file, "(ScanNet)WSAStartup初始化成功\n");
332 }
333 else
334 {
335 DEBUG (gbl.is_debug, gbl.debug_level, gbl.log_file, "(ScanNet)WSAStartup初始化失败\n");
336 WSACleanup();
337 }
338 return vst;
339 }
340
341 void ScanNet::UpdateZTTAB()
342 {
343 yCMysql mysql;
344 if (!mysql.Login (MYSQLIP, "", "", "db"))
345 {
346 mysql.Logout ();
347 DEBUG (gbl.is_debug, gbl.debug_level, gbl.log_file, "(ScanNet)UpdateZTTAB error : Login db fail\n");
348 return ;
349 }
350
351 char sql[300] = {0};
352 sprintf(sql,"update tab_zxtj set jqzt =%d",0);
353
354 if (!mysql.QueryWrite (sql))
355 {
356 mysql.Logout ();
357 DEBUG (gbl.is_debug, gbl.debug_level, gbl.log_file, "(ScanNet)UpdateZTTAB modify tab_zxtj error : QueryWrite sql (%s) fail\n", sql);
358 return ;
359 }
360 mysql.Logout ();
361
362 }
363
364
365 //主函数
366 int ScanNet::Start()
367 {
368 HANDLE hMutex = CreateMutex(NULL,FALSE,NULL);
369 // Sleep(5000); //暂停等待数据库启动
370 if( Init()) //初始化失败
371 {
372 return 0;
373 }
374 UpdateZTTAB(); //重启时状态表状态为0
375 while(1)
376 {
377 Work();
378 Sleep(300000);
379 }
380 return 1;
381 }
382