初始化项目
This commit is contained in:
440
BLV_Studio/Control/DHCP/DhcpServer.vb
Normal file
440
BLV_Studio/Control/DHCP/DhcpServer.vb
Normal file
@@ -0,0 +1,440 @@
|
||||
Imports System.Management
|
||||
Imports System.Net
|
||||
Imports System.Net.NetworkInformation
|
||||
Imports System.Net.Sockets
|
||||
Imports System.Text
|
||||
Imports ARSoft.Tools.Net.Dns
|
||||
Imports DotNetProjects.DhcpServer
|
||||
Imports DotNetProjects.DhcpServer.DHCPRequest
|
||||
|
||||
Public Class DhcpServer
|
||||
Public Event OupPrintfDHCPlog（IPadd As String， mac As String, msgtype As Integer）
|
||||
Public Event IsOpenwifi（Iswifi As Boolean）
|
||||
Enum IPStateType
|
||||
NotAllocated = 1 ' 未分配
|
||||
Assigned ' 待分配
|
||||
TakeUp ' 占用
|
||||
Overdue ' 过期
|
||||
End Enum
|
||||
Public test As DotNetProjects.DhcpServer.DHCPServer
|
||||
Public _IndexIPDic As Dictionary(Of String, dhcpInfo)
|
||||
|
||||
|
||||
Sub New(Optional IPPoolMax As Integer = 10)
|
||||
_IndexIPDic = New Dictionary(Of String, dhcpInfo)
|
||||
sendReply = New DHCPReplyOptions
|
||||
iscnt = False
|
||||
isnumber = 0
|
||||
'GetMACAddress()
|
||||
init(IPPoolMax)
|
||||
End Sub
|
||||
|
||||
Public Sub init(Optional IPPoolMax As Integer = 10)
|
||||
Dim isNetworkInterfaces As NetworkInterface() = System.Net.NetworkInformation.NetworkInterface.GetAllNetworkInterfaces()
|
||||
Dim ipProperties As IPInterfaceProperties
|
||||
Dim gateways As UnicastIPAddressInformationCollection
|
||||
Dim loadip As IPAddress = IPAddress.Parse("192.168.1.10")
|
||||
Dim mac As String = String.Empty
|
||||
For Each inten In isNetworkInterfaces
|
||||
If inten.NetworkInterfaceType = NetworkInterfaceType.Ethernet Then
|
||||
'SetIPAddress({192, 168, 1, 10}, {255, 255, 255, 0}, {192, 168, 1, 1}, {8, 8, 8, 8}, inten)
|
||||
'SetManagementBaseObject(inten)
|
||||
ipProperties = inten.GetIPProperties
|
||||
gateways = ipProperties.UnicastAddresses
|
||||
For Each tmp In gateways
|
||||
If tmp.Address.AddressFamily = AddressFamily.InterNetwork Then
|
||||
'loadip = tmp.Address
|
||||
mac = inten.GetPhysicalAddress.ToString
|
||||
End If
|
||||
Next
|
||||
test = New DotNetProjects.DhcpServer.DHCPServer(loadip)
|
||||
test.SendDhcpAnswerNetworkInterface = inten
|
||||
test.BroadcastAddress = IPAddress.Parse("255.255.255.255")
|
||||
SetsendReply(loadip.ToString)
|
||||
InitIPPool(loadip.ToString, IPPoolMax, mac)
|
||||
Try
|
||||
DHCPRuning()
|
||||
Catch ex As Exception
|
||||
'MsgBox($"{loadip.ToString }_{inten.Name }{ex}")
|
||||
init(IPPoolMax)
|
||||
End Try
|
||||
|
||||
Exit For
|
||||
End If
|
||||
Next
|
||||
|
||||
End Sub
|
||||
|
||||
|
||||
|
||||
Public Function GetMACAddress() As String
|
||||
Dim mac As String = String.Empty
|
||||
For Each nic In NetworkInterface.GetAllNetworkInterfaces
|
||||
Console.WriteLine(nic.GetPhysicalAddress.ToString)
|
||||
Next
|
||||
|
||||
End Function
|
||||
|
||||
|
||||
Public Sub DHCPRuning()
|
||||
|
||||
'InitIPPool(inten.GetIPProperties, IPPoolMax)
|
||||
AddHandler test.OnDataReceived, AddressOf UserDHCPDataReceivedEventHandler
|
||||
AddHandler test.OnDiscover, AddressOf UserDHCPDataReceivedEventHandler
|
||||
AddHandler test.OnRequest, AddressOf UserDHCPDataReceivedEventHandler
|
||||
AddHandler test.OnDecline, AddressOf UserDHCPDataReceivedEventHandler
|
||||
AddHandler test.OnReleased, AddressOf UserDHCPDataReceivedEventHandler
|
||||
AddHandler test.OnInform, AddressOf UserDHCPDataReceivedEventHandler
|
||||
test.Start()
|
||||
|
||||
End Sub
|
||||
Public Sub SetsendReply(txt_LoadIP As String)
|
||||
sendReply.SubnetMask = IPAddress.Parse("255.255.0.0")
|
||||
sendReply.ServerIpAddress = IPAddress.Parse(txt_LoadIP) '本机电脑iP
|
||||
sendReply.IPAddressLeaseTime = 86400
|
||||
sendReply.RenewalTimeValue_T1 = 86400
|
||||
sendReply.RebindingTimeValue_T2 = 86400
|
||||
sendReply.DomainName = "demo.pi"
|
||||
sendReply.ServerIdentifier = IPAddress.Parse(txt_LoadIP) '本机电脑iP
|
||||
sendReply.RouterIP = IPAddress.Parse("192.168.1.1")
|
||||
sendReply.DomainNameServers = GetDomainNameServers("8.8.8.8,8.4.4.4")
|
||||
'sendReply.LogServerIP = IPAddress.Parse(options_120.Text)
|
||||
IPfile = "/pxelinux.0"
|
||||
End Sub
|
||||
|
||||
|
||||
|
||||
|
||||
Public Function GetDomainNameServers(theip As String) As IPAddress()
|
||||
Dim ipstr2() As String = theip.Split(",")
|
||||
Dim result As New List(Of IPAddress)
|
||||
For Each index2 In ipstr2
|
||||
result.Add(IPAddress.Parse(index2))
|
||||
Next
|
||||
Return result.ToArray
|
||||
|
||||
End Function
|
||||
|
||||
|
||||
Public Function SetManagementBaseObject(isNetworkInterfaces As NetworkInterface) As ManagementBaseObject
|
||||
Dim mos As ManagementObjectSearcher = New ManagementObjectSearcher("select * from Win32_NetworkAdapterConfiguration")
|
||||
For Each mo As ManagementObject In mos.Get()
|
||||
If mo("Description").ToString().Equals(isNetworkInterfaces.Description) Then
|
||||
Dim newIP As ManagementBaseObject = mo.GetMethodParameters("EnableStatic")
|
||||
Dim newGateway As ManagementBaseObject = mo.GetMethodParameters("SetGateways")
|
||||
Dim newDNS As ManagementBaseObject = mo.GetMethodParameters("SetDNSServerSearchOrder")
|
||||
newIP.Item("IPAddress") = New String() {192, 168, 1, 10}
|
||||
newIP.Item("SubnetMask") = New String() {255, 255, 255, 0}
|
||||
newGateway.Item（"DefaultIPGateway"） = New String() {192， 168， 1， 1}
|
||||
' newDNS.Item("DNSServerSearchOrder") = New String[] { info.DNS };
|
||||
Dim setIP As ManagementBaseObject = mo.InvokeMethod("EnableStatic", newIP, Nothing)
|
||||
Dim setGateways As ManagementBaseObject = mo.InvokeMethod("SetGateways", newGateway, Nothing)
|
||||
Dim setDNS As ManagementBaseObject = mo.InvokeMethod("SetDNSServerSearchOrder", newDNS, Nothing)
|
||||
'mo.InvokeMethod("SetDNSServerSearchOrder", Nothing);
|
||||
'//DNS的地址清
|
||||
' mo.InvokeMethod("EnableDHCP", Nothing)
|
||||
'//开启DHCP
|
||||
End If
|
||||
Next
|
||||
|
||||
End Function
|
||||
|
||||
|
||||
|
||||
Public Shared Sub SetIPAddress(ByVal ip As String(), ByVal submask As String(), ByVal getway As String(), ByVal dns As String()， isNetworkInterfaces As NetworkInterface)
|
||||
Dim wmi As ManagementClass = New ManagementClass("Win32_NetworkAdapterConfiguration")
|
||||
Dim moc As ManagementObjectCollection = wmi.GetInstances()
|
||||
Dim inPar As ManagementBaseObject = Nothing
|
||||
Dim outPar As ManagementBaseObject = Nothing
|
||||
|
||||
For Each mo As ManagementObject In moc
|
||||
'/如果没有启用IP设置的网络设备则跳过
|
||||
Console.WriteLine($"{mo("Description").ToString()}")
|
||||
If Not mo("Description").ToString().Equals(isNetworkInterfaces.Description) Then Continue For
|
||||
' 设置IP地址和掩码
|
||||
If ip IsNot Nothing AndAlso submask IsNot Nothing Then
|
||||
inPar = mo.GetMethodParameters("EnableStatic")
|
||||
inPar("IPAddress") = ip
|
||||
inPar("SubnetMask") = submask
|
||||
outPar = mo.InvokeMethod("EnableStatic", inPar, Nothing)
|
||||
End If
|
||||
' //设置网关地址
|
||||
If getway IsNot Nothing Then
|
||||
inPar = mo.GetMethodParameters("SetGateways")
|
||||
inPar("DefaultIPGateway") = getway
|
||||
outPar = mo.InvokeMethod("SetGateways", inPar, Nothing)
|
||||
End If
|
||||
' //设置DNS地址
|
||||
If dns IsNot Nothing Then
|
||||
inPar = mo.GetMethodParameters("SetDNSServerSearchOrder")
|
||||
inPar("DNSServerSearchOrder") = dns
|
||||
outPar = mo.InvokeMethod("SetDNSServerSearchOrder", inPar, Nothing)
|
||||
End If
|
||||
Next
|
||||
End Sub
|
||||
|
||||
Public Sub ClockDHCPServer()
|
||||
If IsNothing(test) Then Return
|
||||
test.BroadcastAddress = IPAddress.Parse("0.0.0.0")
|
||||
test.Dispose()
|
||||
' test.BroadcastAddress = IPAddress.Parse("0.0.0.0")
|
||||
End Sub
|
||||
|
||||
Public Sub UserDHCPDataReceivedEventHandler(dhcpRequest As DHCPRequest)
|
||||
AnalytischeGegevens(dhcpRequest) ' dhcpRequest.SendDHCPReply(,,)
|
||||
End Sub
|
||||
Public iscnt As Boolean
|
||||
Public isnumber As Integer
|
||||
Public Sub AnalytischeGegevens(dhcpRequest As DHCPRequest)
|
||||
|
||||
Select Case dhcpRequest.GetMsgType
|
||||
|
||||
Case DHCPMsgType.DHCPACK
|
||||
|
||||
Case DHCPMsgType.DHCPDECLINE
|
||||
|
||||
Case DHCPMsgType.DHCPDISCOVER '上线
|
||||
|
||||
'Dim ipinfo As NetworkInterface = test.SendDhcpAnswerNetworkInterface
|
||||
'If iscnt Then
|
||||
' RaiseEvent IsOpenwifi(False)
|
||||
' iscnt = False
|
||||
'End If
|
||||
|
||||
|
||||
|
||||
Dhcp_OFFER(dhcpRequest)
|
||||
Case DHCPMsgType.DHCPINFORM
|
||||
|
||||
Case DHCPMsgType.DHCPNAK
|
||||
|
||||
Case DHCPMsgType.DHCPOFFER
|
||||
|
||||
Case DHCPMsgType.DHCPRELEASE '下线
|
||||
|
||||
Case DHCPMsgType.DHCPREQUEST
|
||||
|
||||
Dhcp_ACK(dhcpRequest)
|
||||
|
||||
End Select
|
||||
|
||||
|
||||
End Sub
|
||||
Public Function GetByteIPToString(buff() As Byte) As String
|
||||
Dim result As String = ""
|
||||
For Each index In buff
|
||||
result = result & index & "."
|
||||
Next
|
||||
result = result.Substring(0, result.Length - 1)
|
||||
|
||||
Return result
|
||||
End Function
|
||||
Public Sub InitIPPool(startIP As String， ipMax As Integer， mac As String)
|
||||
Dim ipstrbuff() As String
|
||||
Dim number As Integer = 0
|
||||
ipstrbuff = startIP.Split(".")
|
||||
Dim cip As String = ""
|
||||
Dim ClientIP(3) As Byte
|
||||
_IndexIPDic.Clear()
|
||||
|
||||
For i = 0 To 3
|
||||
|
||||
Integer.TryParse(ipstrbuff(i), number)
|
||||
ClientIP(i) = number
|
||||
If i < 3 Then
|
||||
cip = cip & ClientIP(i) & "."
|
||||
End If
|
||||
|
||||
|
||||
Next
|
||||
If Not _IndexIPDic.ContainsKey(startIP) Then
|
||||
Dim newinfo As New dhcpInfo(ClientIP)
|
||||
newinfo.ChildState = IPStateType.Assigned
|
||||
newinfo.ChildMac = GetMacStr(mac)
|
||||
_IndexIPDic.Add(startIP, newinfo)
|
||||
|
||||
End If
|
||||
|
||||
For i As Integer = 1 To ipMax
|
||||
If ClientIP(3) + i > 255 Then Return
|
||||
Dim nClientIP(3) As Byte
|
||||
Array.Copy(ClientIP, nClientIP, ClientIP.Length)
|
||||
nClientIP(3) = ClientIP(3) + i
|
||||
If _IndexIPDic.ContainsKey(cip & nClientIP(3).ToString) Then Continue For
|
||||
_IndexIPDic.Add(cip & nClientIP(3).ToString, New dhcpInfo(nClientIP))
|
||||
Next
|
||||
|
||||
End Sub
|
||||
Public sendReply As DHCPReplyOptions
|
||||
Public IPfile As String
|
||||
Public Sub Dhcp_ACK(dhcpRequest As DHCPRequest)
|
||||
|
||||
|
||||
Dim sda As DHCPPacket = dhcpRequest.GetRawPacket
|
||||
Dim RequestedIP As IPAddress = dhcpRequest.GetRequestedIP
|
||||
Dim cmac As String = GetMacStr(sda.chaddr)
|
||||
Dim newinfo As dhcpInfo = CheckIsTakeUp(RequestedIP.ToString, cmac)
|
||||
|
||||
|
||||
Try
|
||||
If IsNothing(newinfo) Then
|
||||
dhcpRequest.SendDHCPReply(DHCPMsgType.DHCPNAK, IPAddress.Any, sendReply)
|
||||
RaiseEvent OupPrintfDHCPlog(RequestedIP.ToString, cmac, 4)
|
||||
Return
|
||||
End If
|
||||
Catch ex As Exception
|
||||
Return
|
||||
End Try
|
||||
|
||||
|
||||
Dim IPstr As String = GetByteIPToString(newinfo.ChilIP)
|
||||
Dim ip As IPAddress = IPAddress.Parse(IPstr)
|
||||
|
||||
sda.op = &H2
|
||||
sda.htype = &H1
|
||||
sda.hlen = &H6
|
||||
sda.hops = &H0
|
||||
sda.flags = {&H0, &H1}
|
||||
sda.giaddr = {&H0, &H0, &H0, &H0}
|
||||
sda.yiaddr = newinfo.ChilIP
|
||||
sda.siaddr = newinfo.ChilIP
|
||||
Dim filebuff() As Byte = Encoding.UTF8.GetBytes(IPfile)
|
||||
|
||||
Array.Copy(filebuff, sda.file, filebuff.Length)
|
||||
|
||||
Try
|
||||
RaiseEvent OupPrintfDHCPlog(IPstr, cmac, 5)
|
||||
dhcpRequest.SendDHCPReply(DHCPMsgType.DHCPACK, ip, sendReply)
|
||||
Catch ex As Exception
|
||||
Console.WriteLine("发送异常：" & ex.ToString)
|
||||
End Try
|
||||
|
||||
End Sub
|
||||
Public Sub Dhcp_OFFER(dhcpRequest As DHCPRequest)
|
||||
|
||||
Dim sda As DHCPPacket = dhcpRequest.GetRawPacket
|
||||
Dim cmac As String = GetMacStr(sda.chaddr)
|
||||
RaiseEvent OupPrintfDHCPlog("", cmac, 1)
|
||||
Dim newinfo As dhcpInfo = GetNewIPInfo(cmac)
|
||||
Dim dic As New Dictionary(Of DHCPOption, Byte())
|
||||
dic.Add(DHCPOption.NetBIOSoverTCPIPNameServer, {8, 8, 8, 8})
|
||||
dic.Add(DHCPOption.NetworkTimeProtocolServers, {&H58, &HBF, &HE4, &H8A})
|
||||
dic.Add(DHCPOption.TFTPServerName, {&H31, &H39, &H32, &H2E, &H31, &H36, &H38, &H2E, &H31, &H2E, &H31, &H30})
|
||||
dic.Add(DHCPOption.DomainName, {&H64, &H65, &H6D, &H6F, &H2E, &H70, &H69})
|
||||
dic.Add(DHCPOption.StaticRoutes, {&H78, &H30, &H42, &H41, &H43, &H31, &H30, &H43, &H30, &H41, &H38, &H30, &H31, &H46, &H45})
|
||||
sendReply.OtherRequestedOptions = dic
|
||||
If IsNothing(newinfo) Then
|
||||
dhcpRequest.SendDHCPReply(DHCPMsgType.DHCPNAK, IPAddress.Any, sendReply)
|
||||
RaiseEvent OupPrintfDHCPlog("", cmac, 2)
|
||||
Return
|
||||
End If
|
||||
|
||||
'Dim IPstr As String = GetByteIPToString(newinfo.ChilIP)
|
||||
'Dim ip As IPAddress = IPAddress.Parse(IPstr)
|
||||
Dim IPstr As String = GetByteIPToString(newinfo.ChilIP)
|
||||
Dim ip As IPAddress = IPAddress.Parse(IPstr)
|
||||
|
||||
|
||||
|
||||
newinfo.ChildMac = cmac
|
||||
|
||||
newinfo.ChildState = IPStateType.Assigned
|
||||
|
||||
sda.op = &H2
|
||||
sda.htype = &H1
|
||||
sda.hlen = &H6
|
||||
sda.hops = &H0
|
||||
sda.flags = {&H0, &H1}
|
||||
sda.giaddr = {&H0, &H0, &H0, &H0}
|
||||
sda.yiaddr = newinfo.ChilIP
|
||||
sda.siaddr = newinfo.ChilIP
|
||||
|
||||
Dim filebuff() As Byte = Encoding.UTF8.GetBytes(IPfile)
|
||||
|
||||
Array.Copy(filebuff, sda.file, filebuff.Length)
|
||||
|
||||
Try
|
||||
dhcpRequest.SendDHCPReply(DHCPMsgType.DHCPOFFER, ip, sendReply, dic)
|
||||
RaiseEvent OupPrintfDHCPlog(IPstr, cmac, 3)
|
||||
Catch ex As Exception
|
||||
Console.WriteLine("发送异常：" & ex.ToString)
|
||||
End Try
|
||||
|
||||
End Sub
|
||||
Public Function GetMacStr(duff As String) As String
|
||||
Dim sresult As String = ""
|
||||
For i As Integer = 0 To 5
|
||||
sresult = sresult & duff.Substring(i * 2, 2) & ":"
|
||||
|
||||
Next
|
||||
Return sresult.Substring(0, sresult.Length - 1)
|
||||
End Function
|
||||
Public Function GetMacStr(duff() As Byte) As String
|
||||
Dim sresult As String = ""
|
||||
Dim locadMac(5) As Byte
|
||||
Array.Copy(duff, locadMac, locadMac.Length)
|
||||
For Each index In locadMac
|
||||
sresult = sresult & Hex(index) & ":"
|
||||
Next
|
||||
Return sresult.Substring(0, sresult.Length - 1)
|
||||
End Function
|
||||
|
||||
Public Function GetNewIPInfo(mac As String) As dhcpInfo
|
||||
Dim newdhcpInfo As dhcpInfo
|
||||
Dim olddhcpInfo As dhcpInfo
|
||||
|
||||
|
||||
For Each index In _IndexIPDic.Values
|
||||
If mac.Equals(index.ChildMac) Then Return index
|
||||
If IsNothing(newdhcpInfo) AndAlso index.ChildState = IPStateType.NotAllocated Then
|
||||
newdhcpInfo = index
|
||||
Exit For
|
||||
End If
|
||||
If IsNothing(olddhcpInfo) AndAlso index.ChildState = IPStateType.Overdue Then
|
||||
olddhcpInfo = index
|
||||
End If
|
||||
Next
|
||||
If Not IsNothing(newdhcpInfo) Then
|
||||
Return newdhcpInfo
|
||||
ElseIf Not IsNothing(olddhcpInfo) Then
|
||||
Return olddhcpInfo
|
||||
Else
|
||||
Return Nothing
|
||||
End If
|
||||
|
||||
End Function
|
||||
|
||||
Public Function CheckIsTakeUp(name As String, mac As String) As dhcpInfo
|
||||
|
||||
If _IndexIPDic.ContainsKey(name) Then
|
||||
Dim Index = _IndexIPDic.Item(name)
|
||||
'ip 被占用 且用户 Mac 地址一致
|
||||
If Index.ChildState <> IPStateType.NotAllocated AndAlso Index.ChildMac.Equals(mac) Then
|
||||
Index.ChildState = IPStateType.Assigned
|
||||
Return Index
|
||||
Else
|
||||
Return Nothing
|
||||
End If
|
||||
End If
|
||||
Return Nothing
|
||||
End Function
|
||||
|
||||
End Class
|
||||
Public Class dhcpInfo
|
||||
Public ChilIP() As Byte
|
||||
|
||||
Public ChildMac As String = ""
|
||||
|
||||
Public ChildTime As Long = 0
|
||||
|
||||
Public ChildState As Integer = 1
|
||||
Sub New(ip() As Byte)
|
||||
ChilIP = ip
|
||||
ChildMac = ""
|
||||
ChildTime = 0
|
||||
ChildState = 1
|
||||
|
||||
|
||||
End Sub
|
||||
|
||||
End Class
|
||||
Reference in New Issue
Block a user