使用Visual Basic.NET扫描系统服务 2

End Try

End Sub

End Class



‘由于篇幅限制，这里省略了根据IP取得机器信息的类的代码。



Imports System.Threading ‘用于支持多线程

Imports System.Xml ‘用于分析XML格式的参数文件

Imports System.Data ‘用于保存结果到数据库

Module Module1

Public ds As New DataSet()

Public conn1 As SqlClient.SqlConnection ‘数据库连接

Public ipf As String ‘IP列表文件名

Public dbf As String ‘数据库信息文件

Public ThreadCounterStopped As Integer

Public StopThr As New StoppCounter()

Sub Main() ‘程序主程序

Dim machineIP As String

Dim iplistF As New Xml.XmlDocument()

Dim iplist As Xml.XmlNode

Dim ipitem As Xml.XmlNode

Dim DBinfoF As New Xml.XmlDocument()

Dim DBinfo As Xml.XmlNode

Dim LanID As String

Dim i As Integer

Dim timestart As Integer

Dim ThreadCounterStarted As Integer

ThreadCounterStarted = 0

ThreadCounterStopped = 0

Dim server As String

Dim database As String

Dim uid As String

Dim pwd As String

Dim table As String

Dim connstr, connstr1 As String

Dim ServiceName As String

Dim Purgestr As String

Try

DBinfoF.Load(dbf) ‘读取数据库信息文件

Catch nodb As Exception

MsgBox(nodb.Message & "Wrong DB info file name.")

Exit Sub

End Try

Try

iplistF.Load(ipf) ‘读取IP列表文件

Catch noip As Exception

MsgBox(noip.Message & "Wrong IP list file name.")

Exit Sub

End Try

‘分析数据库信息文件

DBinfo = DBinfoF.ChildNodes(0)

server = DBinfo.ChildNodes(0).InnerText

database = DBinfo.ChildNodes(1).InnerText

uid = DBinfo.ChildNodes(2).InnerText

pwd = DBinfo.ChildNodes(3).InnerText

ServiceName = DBinfo.ChildNodes(4).InnerText

table = DBinfo.ChildNodes(4).Attributes(0).Value

‘根据分析所得，构造连接字符串

connstr1 = "server=" & server & ";database=" & database & ";uid=" & uid & ";password=" & pwd

conn1 = New SqlClient.SqlConnection(connstr1) ‘实例化数据库连接

conn1.Open() ‘打开数据库连接

Dim sa As SqlClient.SqlDataAdapter = New SqlClient.SqlDataAdapter("select * from " & table, conn1)

Dim combu As New SqlClient.SqlCommandBuilder(sa)

sa.Fill(ds, table) ‘填充DataSet

ds.Clear() ‘清空旧的数据

Dim IPAddress As String

‘分析IP列表文件

iplist = iplistF.ChildNodes(0)

Dim Ai As Integer

Dim ipexcepCount As Integer

Dim ipexcep As Xml.XmlNode

For Each ipitem In iplist.ChildNodes

Dim Excep(2, 83) As Integer

LanID = ipitem.Attributes(0).Value‘得到网络ID

For i = 2 To 254 ‘从2到254，根据每个网络ID构造IP地址

Ai = 0
以下判断是为了跳过保留地址段

If ipitem.HasChildNodes Then

ipexcepCount = ipitem.ChildNodes.Count

ReDim Excep(2, ipexcepCount - 1)

For Each ipexcep In ipitem.ChildNodes

Excep(0, Ai) = CInt(ipexcep.Attributes(0).Value)

Excep(1, Ai) = CInt(ipexcep.Attributes(1).Value)

Ai = Ai + 1

Next

End If

For Ai = 0 To ipexcepCount - 1

If i ＞= Excep(0, Ai) And i ＜= Excep(1, Ai) Then

Console.WriteLine("跳过保留地址： " & LanID & i.ToString)

GoTo SkipIP

End If

Next

machineIP = LanID & i.ToString ‘IP地址

‘以下触发线程以，得到服务状态

Dim getSt As New GetStatus(machineIP, ServiceName, table)

Dim GetStThread As New Thread(New ThreadStart(AddressOf getSt.GetStausF))

GetStThread.Start()

ThreadCounterStarted = ThreadCounterStarted + 1‘启动线程数加1

Console.WriteLine("线程" & machineIP & " 启动。检测 " & ServiceName)

‘每启动100个线程，程序主线程停止15秒，避免太多线程造成内存溢出

If (ThreadCounterStarted Mod 100) = 0 Then

Console.WriteLine("等待 .......")

Thread.CurrentThread.Sleep(15000)

GC.Collect() 'force garbage collection to aviod outOfMemory when run with long IP list

End If

SkipIP:

Next

Next

Console.WriteLine("Exiting program ...") ‘所有线程都已触发

Finish:

Thread.CurrentThread.Sleep(5000) ‘以下程序等待所有线程结束

GC.Collect()

If ThreadCounterStopped = ThreadCounterStarted And HostInfoThreadCounter.counter = HostInfoThreadCounter.counterSTOP Then‘如果触发线程等于结束线程



Dim row As Data.DataRow

For Each row In ds.Tables(table).Rows

row.Item("SysTime") = Now

Next

Purgestr = "delete " & table

Dim com1 As New SqlClient.SqlCommand(Purgestr, conn1)

com1.ExecuteNonQuery() ‘删除旧记录

sa.InsertCommand = combu.GetInsertCommand

sa.Update(ds, table) ' 将新记录写入数据库

Else

GoTo Finish ' goto finish and wait another 30 seconds

End If

End Sub

　　可以利用如下命令在DOS窗口启动该程序。

Scanservice –i iplist.xml –d dbinfo.xml

　　2．SCANSERVICE 数据库

　　该数据库保存保存程序运行结果，以便用WEB等方式展现出来。以下是建立表的脚本，包含域名，用户名，机器名，IP以及服务状态。

CREATE TABLE [dbo].[Service] ( [IP] [varchar] (50) NULL , [状态] [varchar] (50) NULL , [用户名] [varchar] (50) NULL , [机器名] [varchar] (50) NULL , [MAC地址] [varchar] (50) NULL , [域] [varchar] (50) NULL , [超时] [varchar] (10) NULL , [时间安] [DateTime] (8) NULL , )

　　总结:

　　以上是一个完整的方法，也是比较简单明晰的解决方法，如果要求技巧和性能的话，还有一些地方可以做些改进，比如对线程池的使用。另外还有一些方面需要大家自己完成，比如将数据库中的信息以WEB的方式展现出来。