Amazing-Mirror/fragattacks
1
0
Fork 0
mirror of https://github.com/vanhoefm/fragattacks.git synced 2025-02-21 11:33:04 -05:00
Code Issues Projects Releases Wiki Activity

tests: Add UDP ctrl_iface support to wpasupplicant.py

Browse Source 
Allow use of a remote host using wpaspy.Ctrl with UDP ctrl_iface
support.

Signed-off-by: Janusz Dziedzic <janusz.dziedzic@tieto.com>
This commit is contained in:
Janusz Dziedzic 2016-03-04 10:20:34 +01:00 committed by Jouni Malinen
parent a2c88a8025
commit 0fd4792b48
1 changed files with 39 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

47
tests/hwsim/wpasupplicant.py
View File

@ -17,18 +17,24 @@ logger = logging.getLogger()
wpas_ctrl = '/var/run/wpa_supplicant' wpas_ctrl = '/var/run/wpa_supplicant'
class WpaSupplicant: class WpaSupplicant:
def __init__(self, ifname=None, global_iface=None): def __init__(self, ifname=None, global_iface=None, hostname=None,
port=9877, global_port=9878):
self.hostname = hostname
self.group_ifname = None self.group_ifname = None
self.gctrl_mon = None self.gctrl_mon = None
if ifname: if ifname:
self.set_ifname(ifname) self.set_ifname(ifname, hostname, port)
else: else:
self.ifname = None self.ifname = None
self.global_iface = global_iface self.global_iface = global_iface
if global_iface: if global_iface:
self.global_ctrl = wpaspy.Ctrl(global_iface) if hostname != None:
self.global_mon = wpaspy.Ctrl(global_iface) self.global_ctrl = wpaspy.Ctrl(hostname, global_port)
self.global_mon = wpaspy.Ctrl(hostname, global_port)
else:
self.global_ctrl = wpaspy.Ctrl(global_iface)
self.global_mon = wpaspy.Ctrl(global_iface)
self.global_mon.attach() self.global_mon.attach()
else: else:
self.global_mon = None self.global_mon = None
@ -40,10 +46,14 @@ class WpaSupplicant:
self.global_ctrl = None self.global_ctrl = None
self.remove_ifname() self.remove_ifname()
def set_ifname(self, ifname): def set_ifname(self, ifname, hostname=None, port=9877):
self.ifname = ifname self.ifname = ifname
self.ctrl = wpaspy.Ctrl(os.path.join(wpas_ctrl, ifname)) if hostname != None:
self.mon = wpaspy.Ctrl(os.path.join(wpas_ctrl, ifname)) self.ctrl = wpaspy.Ctrl(hostname, port)
self.mon = wpaspy.Ctrl(hostname, port)
else:
self.ctrl = wpaspy.Ctrl(os.path.join(wpas_ctrl, ifname))
self.mon = wpaspy.Ctrl(os.path.join(wpas_ctrl, ifname))
self.mon.attach() self.mon.attach()
def remove_ifname(self): def remove_ifname(self):
@ -53,6 +63,26 @@ class WpaSupplicant:
self.ctrl = None self.ctrl = None
self.ifname = None self.ifname = None
def get_ctrl_iface_port(self, ifname):
if self.hostname is None:
return None
res = self.global_request("INTERFACES ctrl")
lines = res.splitlines()
found = False
for line in lines:
words = line.split()
if words[0] == ifname:
found = True
break
if not found:
raise Exception("Could not find UDP port for " + ifname)
res = line.find("ctrl_iface=udp:")
if res == -1:
raise Exception("Wrong ctrl_interface format")
words = line.split(":")
return int(words[1])
def interface_add(self, ifname, config="", driver="nl80211", def interface_add(self, ifname, config="", driver="nl80211",
drv_params=None, br_ifname=None, create=False, drv_params=None, br_ifname=None, create=False,
set_ifname=True, all_params=False, if_type=None): set_ifname=True, all_params=False, if_type=None):
@ -85,7 +115,8 @@ class WpaSupplicant:
if "FAIL" in self.global_request(cmd): if "FAIL" in self.global_request(cmd):
raise Exception("Failed to add a dynamic wpa_supplicant interface") raise Exception("Failed to add a dynamic wpa_supplicant interface")
if not create and set_ifname: if not create and set_ifname:
self.set_ifname(ifname) port = self.get_ctrl_iface_port(ifname)
self.set_ifname(ifname, self.hostname, port)
def interface_remove(self, ifname): def interface_remove(self, ifname):
self.remove_ifname() self.remove_ifname()