arno/gns3-server
1
0
Fork 0
mirror of https://github.com/GNS3/gns3-server synced 2024-11-28 11:18:11 +00:00
Code Issues Releases Wiki Activity

Get Windows interface list from the registry if the COM service fails.

Browse Source
This commit is contained in:
Jeremy 2014-12-09 19:17:25 -07:00
parent 20592f72c0
commit 823344bf44
2 changed files with 39 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

44
gns3server/builtins/interfaces.py
View File

@ -35,16 +35,48 @@ def get_windows_interfaces():
""" """
import win32com.client import win32com.client
import pywintypes
locator = win32com.client.Dispatch("WbemScripting.SWbemLocator") locator = win32com.client.Dispatch("WbemScripting.SWbemLocator")
service = locator.ConnectServer(".", "root\cimv2") service = locator.ConnectServer(".", "root\cimv2")
interfaces = [] interfaces = []
# more info on Win32_NetworkAdapter: http://msdn.microsoft.com/en-us/library/aa394216%28v=vs.85%29.aspx try:
for adapter in service.InstancesOf("Win32_NetworkAdapter"): # more info on Win32_NetworkAdapter: http://msdn.microsoft.com/en-us/library/aa394216%28v=vs.85%29.aspx
if adapter.NetConnectionStatus == 2 or adapter.NetConnectionStatus == 7: for adapter in service.InstancesOf("Win32_NetworkAdapter"):
# adapter is connected or media disconnected if adapter.NetConnectionStatus == 2 or adapter.NetConnectionStatus == 7:
npf_interface = "\\Device\\NPF_{guid}".format(guid=adapter.GUID) # adapter is connected or media disconnected
npf_interface = "\\Device\\NPF_{guid}".format(guid=adapter.GUID)
interfaces.append({"id": npf_interface,
"name": adapter.NetConnectionID})
except pywintypes.com_error:
log.warn("could not use the COM service to retrieve interface info, trying using the registry...")
return get_windows_interfaces_from_registry()
return interfaces
def get_windows_interfaces_from_registry():
import winreg
interfaces = []
try:
hkey = winreg.OpenKey(winreg.HKEY_LOCAL_MACHINE, r"SOFTWARE\Microsoft\Windows NT\CurrentVersion\NetworkCards")
for index in range(winreg.QueryInfoKey(hkey)[0]):
network_card_id = winreg.EnumKey(hkey, index)
hkeycard = winreg.OpenKey(hkey, network_card_id)
guid, _ = winreg.QueryValueEx(hkeycard, "ServiceName")
connection = r"SYSTEM\CurrentControlSet\Control\Network\{4D36E972-E325-11CE-BFC1-08002BE10318}" + "\{}\Connection".format(guid)
hkeycon = winreg.OpenKey(winreg.HKEY_LOCAL_MACHINE, connection)
name, _ = winreg.QueryValueEx(hkeycon, "Name")
npf_interface = "\\Device\\NPF_{guid}".format(guid=guid)
interfaces.append({"id": npf_interface, interfaces.append({"id": npf_interface,
"name": adapter.NetConnectionID}) "name": name})
winreg.CloseKey(hkeycon)
winreg.CloseKey(hkeycard)
winreg.CloseKey(hkey)
except OSError as e:
log.error("could not read registry information: {}".format(e))
return interfaces return interfaces

2
gns3server/modules/attic.py
View File

@ -129,7 +129,7 @@ def has_privileged_access(executable):
if os.geteuid() == 0: if os.geteuid() == 0:
# we are root, so we should have privileged access. # we are root, so we should have privileged access.
return True return True
if os.stat(executable).st_mode & stat.S_ISUID: if os.stat(executable).st_mode & stat.S_ISUID or os.stat(executable).st_mode & stat.S_ISGID:
# the executable has set UID bit. # the executable has set UID bit.
return True return True