=begin = Win32 DNS and DHCP I/F =end require 'win32/registry' module Win32 module Resolv API = Registry::API def self.get_hosts_path path = get_hosts_dir path = File.join(path.gsub(/\\/, File::SEPARATOR), 'hosts') File.exist?(path) ? path : nil end def self.get_resolv_info search, nameserver = get_info if search.empty? search = nil else search.delete("") search.uniq! end if nameserver.empty? nameserver = nil else nameserver.delete("") nameserver.delete("0.0.0.0") nameserver.uniq! end [ search, nameserver ] end getv = Win32API.new('kernel32.dll', 'GetVersionExA', 'P', 'L') info = [ 148, 0, 0, 0, 0 ].pack('V5') + "\0" * 128 getv.call(info) if info.unpack('V5')[4] == 2 # VER_PLATFORM_WIN32_NT #==================================================================== # Windows NT #==================================================================== module_eval <<-'__EOS__', __FILE__, __LINE__+1 TCPIP_NT = 'SYSTEM\CurrentControlSet\Services\Tcpip\Parameters' class << self private def get_hosts_dir Registry::HKEY_LOCAL_MACHINE.open(TCPIP_NT) do |reg| reg.read_s_expand('DataBasePath') end end def get_info search = nil nameserver = [] Registry::HKEY_LOCAL_MACHINE.open(TCPIP_NT) do |reg| begin slist = reg.read_s('SearchList') search = slist.split(/,\s*/) unless slist.empty? rescue Registry::Error end if add_search = search.nil? search = [] begin nvdom = reg.read_s('NV Domain') unless nvdom.empty? @search = [ nvdom ] if reg.read_i('UseDomainNameDevolution') != 0 if /^[\w\d]+\./ =~ nvdom devo = $' end end end rescue Registry::Error end end reg.open('Interfaces') do |reg| reg.each_key do |iface,| reg.open(iface) do |regif| begin [ 'NameServer', 'DhcpNameServer' ].each do |key| ns = regif.read_s(key) unless ns.empty? nameserver.concat(ns.split(/\s+/)) break end end rescue Registry::Error end if add_search begin [ 'Domain', 'DhcpDomain' ].each do |key| dom = regif.read_s(key) unless dom.empty? search.concat(dom.split(/,\s*/)) break end end rescue Registry::Error end end end end end search << devo if add_search and devo end [ search.uniq, nameserver.uniq ] end end __EOS__ else #==================================================================== # Windows 9x #==================================================================== module_eval <<-'__EOS__', __FILE__, __LINE__+1 TCPIP_9X = 'SYSTEM\CurrentControlSet\Services\VxD\MSTCP' DHCP_9X = 'SYSTEM\CurrentControlSet\Services\VxD\DHCP' WINDOWS = 'Software\Microsoft\Windows\CurrentVersion' class << self # private def get_hosts_dir Registry::HKEY_LOCAL_MACHINE.open(WINDOWS) do |reg| reg.read_s_expand('SystemRoot') end end def get_info search = [] nameserver = [] begin Registry::HKEY_LOCAL_MACHINE.open(TCPIP_9X) do |reg| if reg.read_s("EnableDNS") == "1" domain = reg.read_s("Domain") ns = reg.read_s("NameServer") slist = reg.read_s("SearchList") search << domain unless domain.empty? search.concat(slist.split(/,\s*/)) nameserver.concat(ns.split(/,\s*/)) end end rescue Registry::Error end dhcpinfo = get_dhcpinfo search.concat(dhcpinfo[0]) nameserver.concat(dhcpinfo[1]) [ search, nameserver ] end def get_dhcpinfo macaddrs = {} ipaddrs = {} WsControl.get_iflist.each do |index, macaddr, *ipaddr| macaddrs[macaddr] = 1 ipaddr.each { |ipaddr| ipaddrs[ipaddr] = 1 } end iflist = [ macaddrs, ipaddrs ] search = [] nameserver = [] version = -1 Registry::HKEY_LOCAL_MACHINE.open(DHCP_9X) do |reg| begin version = API.unpackdw(reg.read_bin("Version")) rescue Registry::Error end reg.each_key do |key,| catch(:not_used) do reg.open(key) do |regdi| dom, ns = get_dhcpinfo_key(version, regdi, iflist) search << dom if dom nameserver.concat(ns) if ns end end end end [ search, nameserver ] end def get_dhcpinfo_95(reg) dhcp = reg.read_bin("DhcpInfo") [ API.unpackdw(dhcp[4..7]), API.unpackdw(dhcp[8..11]), 1, dhcp[45..50], reg.read_bin("OptionInfo"), ] end def get_dhcpinfo_98(reg) [ API.unpackdw(reg.read_bin("DhcpIPAddress")), API.unpackdw(reg.read_bin("DhcpSubnetMask")), API.unpackdw(reg.read_bin("HardwareType")), reg.read_bin("HardwareAddress"), reg.read_bin("OptionInfo"), ] end def get_dhcpinfo_key(version, reg, iflist) info = case version when 1 get_dhcpinfo_95(reg) when 2 get_dhcpinfo_98(reg) else begin get_dhcpinfo_98(reg) rescue Registry::Error get_dhcpinfo_95(reg) end end ipaddr, netmask, hwtype, macaddr, opt = info throw :not_used unless ipaddr and ipaddr != 0 and netmask and netmask != 0 and macaddr and macaddr.size == 6 and hwtype == 1 and iflist[0][macaddr] and iflist[1][ipaddr] size = opt.size idx = 0 while idx <= size opttype = opt[idx] optsize = opt[idx + 1] optval = opt[idx + 2, optsize] case opttype when 0xFF ## term break when 0x0F ## domain domain = optval.chomp("\0") when 0x06 ## dns nameserver = optval.scan(/..../).collect { |addr| "%d.%d.%d.%d" % addr.unpack('C4') } end idx += optsize + 2 end [ domain, nameserver ] rescue Registry::Error throw :not_used end end module WsControl WsControl = Win32API.new('wsock32.dll', 'WsControl', 'LLPPPP', 'L') WSAGetLastError = Win32API.new('wsock32.dll', 'WSAGetLastError', 'V', 'L') MAX_TDI_ENTITIES = 512 IPPROTO_TCP = 6 WSCTL_TCP_QUERY_INFORMATION = 0 INFO_CLASS_GENERIC = 0x100 INFO_CLASS_PROTOCOL = 0x200 INFO_TYPE_PROVIDER = 0x100 ENTITY_LIST_ID = 0 GENERIC_ENTITY = 0 CL_NL_ENTITY = 0x301 IF_ENTITY = 0x200 ENTITY_TYPE_ID = 1 CL_NL_IP = 0x303 IF_MIB = 0x202 IF_MIB_STATS_ID = 1 IP_MIB_ADDRTABLE_ENTRY_ID = 0x102 def self.wsctl(tei_entity, tei_instance, toi_class, toi_type, toi_id, buffsize) reqinfo = [ ## TDIEntityID tei_entity, tei_instance, ## TDIObjectID toi_class, toi_type, toi_id, ## TCP_REQUEST_INFORMATION_EX "" ].pack('VVVVVa16') reqsize = API.packdw(reqinfo.size) buff = "\0" * buffsize buffsize = API.packdw(buffsize) result = WsControl.call( IPPROTO_TCP, WSCTL_TCP_QUERY_INFORMATION, reqinfo, reqsize, buff, buffsize) if result != 0 raise RuntimeError, "WsControl failed.(#{result})" end [ buff, API.unpackdw(buffsize) ] end private_class_method :wsctl def self.get_iflist # Get TDI Entity List entities, size = wsctl(GENERIC_ENTITY, 0, INFO_CLASS_GENERIC, INFO_TYPE_PROVIDER, ENTITY_LIST_ID, MAX_TDI_ENTITIES * 8) # sizeof(TDIEntityID) entities = entities[0, size]. scan(/.{8}/). collect { |e| e.unpack('VV') } # Get MIB Interface List iflist = [] ifcount = 0 entities.each do |entity, instance| if( (entity & IF_ENTITY)>0 ) ifcount += 1 etype, = wsctl(entity, instance, INFO_CLASS_GENERIC, INFO_TYPE_PROVIDER, ENTITY_TYPE_ID, 4) if( (API.unpackdw(etype) & IF_MIB)==IF_MIB ) ifentry, = wsctl(entity, instance, INFO_CLASS_PROTOCOL, INFO_TYPE_PROVIDER, IF_MIB_STATS_ID, 21 * 4 + 8 + 130) # sizeof(IFEntry) iflist << [ API.unpackdw(ifentry[0,4]), ifentry[20, 6] ] end end end # Get IP Addresses entities.each do |entity, instance| if entity == CL_NL_ENTITY etype, = wsctl(entity, instance, INFO_CLASS_GENERIC, INFO_TYPE_PROVIDER, ENTITY_TYPE_ID, 4) if API.unpackdw(etype) == CL_NL_IP ipentries, = wsctl(entity, instance, INFO_CLASS_PROTOCOL, INFO_TYPE_PROVIDER, IP_MIB_ADDRTABLE_ENTRY_ID, 24 * (ifcount+1)) # sizeof(IPAddrEntry) ipentries.scan(/.{24}/) do |ipentry| ipaddr, index = ipentry.unpack('VV') if ifitem = iflist.assoc(index) ifitem << ipaddr end end end end end iflist end end __EOS__ end #==================================================================== end end