def shutdown(self):
"""
attempt to restore everything and clean up. returns whether a full reset or
not occurred
"""
# try shutting down & resetting radio (if it failed we may not be able to)
clean = True
try:
# stop the tuner
try:
# call shutdown and join timing out if necessary
self._tuner.shutdown()
self._tuner.join(max(self._ds)*2)
except:
clean = False
else:
self._tuner = None
# reset the device
try:
iw.devdel(self._vnic)
iw.phyadd(self._phy,self._nic)
if self._spoofed:
iwt.ifconfig(self._nic,'down')
iwt.resethwaddr(self._nic)
iwt.ifconfig(self._nic,'up')
except iw.IWException:
clean = False
# close socket and connection
if self._s: self._s.close()
self._conn.close()
except:
clean = False
return clean
def _setup(self,conf):
"""
1) sets radio properties as specified in conf
2) prepares specified nic for monitoring and binds it
3) creates a scan list and compile statistics
"""
# if the nic specified in conf is present, set it
if conf['nic'] not in iwt.wifaces():
raise RuntimeError("%s:iwtools.wifaces:not found" % conf['role'])
self._nic = conf['nic']
self._role = conf['role']
# get the phy and associated interfaces
try:
(self._phy,ifaces) = iw.dev(self._nic)
self._mac = ifaces[0]['addr']
except (KeyError, IndexError):
raise RuntimeError("%s:iw.dev:error getting interfaces" % self._role)
except iw.IWException as e:
raise RuntimeError("%s:iw.dev:failed to get phy <%s>" % (self._role,e))
# get properties (the below will return None rather than throw exception)
self._chs = iw.chget(self._phy)
self._std = iwt.iwconfig(self._nic,'Standards')
self._txpwr = iwt.iwconfig(self._nic,'Tx-Power')
self._driver = iwt.getdriver(self._nic)
self._chipset = iwt.getchipset(self._driver)
# spoof the mac address ??
if conf['spoofed']:
mac = None if conf['spoofed'].lower() == 'random' else conf['spoofed']
try:
iwt.ifconfig(self._nic,'down')
self._spoofed = iwt.sethwaddr(self._nic,mac)
except iwt.IWToolsException as e:
raise RuntimeError("%s:iwtools.sethwaddr:%s" % (self._role,e))
# delete all associated interfaces - we want full control
for iface in ifaces: iw.devdel(iface['nic'])
# determine virtual interface name
ns = []
for wiface in iwt.wifaces():
cs = wiface.split('dyskt')
try:
if len(cs) > 1: ns.append(int(cs[1]))
except ValueError:
pass
n = 0 if not 0 in ns else max(ns)+1
self._vnic = "dyskt%d" % n
# sniffing interface
try:
iw.phyadd(self._phy,self._vnic,'monitor') # create a monitor,
iwt.ifconfig(self._vnic,'up') # and turn it on
except iw.IWException as e:
# never added virtual nic, restore nic
errMsg = "%s:iw.phyadd:%s" % (self._role,e)
try:
iwt.ifconfig(self._nic,'up')
except iwt.IWToolsException:
errMsg += " Failed to restore %s" % self._nic
raise RuntimeError(errMsg)
except iwt.IWToolsException as e:
# failed to 'raise' virtual nic, remove vnic and add nic
errMsg = "%s:iwtools.ifconfig:%s" % (self._role,e)
try:
iw.phyadd(self._phy,self._nic,'managed')
iw.devdel(self._vnic)
iwt.ifconfig(self._nic,'up')
except (iw.IWException,iwt.IWToolsException):
errMsg += " Failed to restore %s" % self._nic
raise RuntimeError(errMsg)
# wrap remaining in a try block, we must attempt to restore card and
# release the socket after any failures ATT
self._s = None
try:
# bind socket w/ a timeout of 5 just in case there is no wireless traffic
self._s = socket.socket(socket.AF_PACKET,
socket.SOCK_RAW,
socket.htons(0x0003))
self._s.settimeout(5)
self._s.bind((self._vnic,0x0003))
uptime = time.time()
# read in antenna details and radio description
if conf['antennas']['num'] > 0:
self._antenna['num'] = conf['antennas']['num']
self._antenna['type'] = conf['antennas']['type']
self._antenna['gain'] = conf['antennas']['gain']
self._antenna['loss'] = conf['antennas']['loss']
self._antenna['x'] = [v[0] for v in conf['antennas']['xyz']]
self._antenna['y'] = [v[1] for v in conf['antennas']['xyz']]
self._antenna['z'] = [v[2] for v in conf['antennas']['xyz']]
self._desc = conf['desc']
# compile initial scan pattern from config
scan = [t for t in conf['scan'] if str(t[0]) in self._chs and not t in conf['pass']]
# sum hop times with side effect of removing any invalid channel tuples
# i.e. Ch 14, Width HT40+ and channels the card cannot tune to
i = 0
hop = 0
while scan:
try:
t = time.time()
iw.chset(self._vnic,str(scan[i][0]),scan[i][1])
hop += (time.time() - t)
i += 1
except iw.IWException as e:
# if invalid argument, drop the channel otherwise reraise error
if iw.ecode(str(e)) == iw.IW_INVALIDARG:
del scan[i]
else:
raise
except IndexError:
# all channels checked
break
if not scan: raise ValueError("Empty scan pattern")
# calculate avg hop time, and interval time
# NOTE: these are not currently used but may be useful later
hop /= len(scan)
interval = len(scan) * conf['dwell'] + len(scan) * hop
# create list of dwell times
ds = [conf['dwell']] * len(scan)
# get start ch & set the initial channel
try:
ch_i = scan.index(conf['scan_start'])
except ValueError:
ch_i = 0
iw.chset(self._vnic,str(scan[ch_i][0]),scan[ch_i][1])
# initialize tuner thread
self._qT = Queue()
self._tuner = Tuner(self._qT,self._conn,self._vnic,scan,ds,ch_i,conf['paused'])
# notify RTO we are good
self._icomms.put((self._vnic,uptime,'!UP!',self.radio))
except socket.error as e:
try:
iw.devdel(self._vnic)
iw.phyadd(self._phy,self._nic)
iwt.ifconfig(self._nic,'up')
except (iw.IWException,iwt.IWToolsException):
pass
if self._s:
self._s.shutdown(socket.SHUT_RD)
self._s.close()
raise RuntimeError("%s:Raw Socket:%s" % (self._role,e))
except iw.IWException as e:
try:
iw.devdel(self._vnic)
iw.phyadd(self._phy,self._nic)
iwt.ifconfig(self._nic,'up')
except (iw.IWException,iwt.IWToolsException):
pass
if self._s:
self._s.shutdown(socket.SHUT_RD)
self._s.close()
raise RuntimeError("%s:iw.chset:Failed to set channel: %s" % (self._role,e))
except (ValueError,TypeError) as e:
try:
iw.devdel(self._vnic)
iw.phyadd(self._phy,self._nic)
iwt.ifconfig(self._nic,'up')
except (iw.IWException,iwt.IWToolsException):
pass
if self._s:
self._s.shutdown(socket.SHUT_RD)
self._s.close()
raise RuntimeError("%s:config:%s" % (self._role,e))
except Exception as e:
# blanket exception
try:
iw.devdel(self._vnic)
iw.phyadd(self._phy,self._nic)
iwt.ifconfig(self._nic,'up')
except (iw.IWException,iwt.IWToolsException):
pass
if self._s:
self._s.shutdown(socket.SHUT_RD)
self._s.close()
raise RuntimeError("%s:Unknown:%s" % (self._role,e))
def _setup(self, conf):
"""
1) sets radio properties as specified in conf
2) prepares specified nic for monitoring and binds it
3) creates a scan list and compile statistics
"""
# if the nic specified in conf is present, set it
if conf['nic'] not in iwt.wifaces():
raise RuntimeError("%s:iwtools.wifaces:not found" % conf['role'])
self._nic = conf['nic']
self._role = conf['role']
# get the phy and associated interfaces
try:
(self._phy, ifaces) = iw.dev(self._nic)
self._mac = ifaces[0]['addr']
except (KeyError, IndexError):
raise RuntimeError("%s:iw.dev:error getting interfaces" %
self._role)
except iw.IWException as e:
raise RuntimeError("%s:iw.dev:failed to get phy <%s>" %
(self._role, e))
# get properties (the below will return None rather than throw exception)
self._chs = iw.chget(self._phy)
self._std = iwt.iwconfig(self._nic, 'Standards')
self._txpwr = iwt.iwconfig(self._nic, 'Tx-Power')
self._driver = iwt.getdriver(self._nic)
self._chipset = iwt.getchipset(self._driver)
# spoof the mac address ??
if conf['spoofed']:
mac = None if conf['spoofed'].lower(
) == 'random' else conf['spoofed']
try:
iwt.ifconfig(self._nic, 'down')
self._spoofed = iwt.sethwaddr(self._nic, mac)
except iwt.IWToolsException as e:
raise RuntimeError("%s:iwtools.sethwaddr:%s" % (self._role, e))
# delete all associated interfaces - we want full control
for iface in ifaces:
iw.devdel(iface['nic'])
# determine virtual interface name
ns = []
for wiface in iwt.wifaces():
cs = wiface.split('dyskt')
try:
if len(cs) > 1: ns.append(int(cs[1]))
except ValueError:
pass
n = 0 if not 0 in ns else max(ns) + 1
self._vnic = "dyskt%d" % n
# sniffing interface
try:
iw.phyadd(self._phy, self._vnic, 'monitor') # create a monitor,
iwt.ifconfig(self._vnic, 'up') # and turn it on
except iw.IWException as e:
# never added virtual nic, restore nic
errMsg = "%s:iw.phyadd:%s" % (self._role, e)
try:
iwt.ifconfig(self._nic, 'up')
except iwt.IWToolsException:
errMsg += " Failed to restore %s" % self._nic
raise RuntimeError(errMsg)
except iwt.IWToolsException as e:
# failed to 'raise' virtual nic, remove vnic and add nic
errMsg = "%s:iwtools.ifconfig:%s" % (self._role, e)
try:
iw.phyadd(self._phy, self._nic, 'managed')
iw.devdel(self._vnic)
iwt.ifconfig(self._nic, 'up')
except (iw.IWException, iwt.IWToolsException):
errMsg += " Failed to restore %s" % self._nic
raise RuntimeError(errMsg)
# wrap remaining in a try block, we must attempt to restore card and
# release the socket after any failures ATT
self._s = None
try:
# bind socket w/ a timeout of 5 just in case there is no wireless traffic
self._s = socket.socket(socket.AF_PACKET, socket.SOCK_RAW,
socket.htons(0x0003))
self._s.settimeout(5)
self._s.bind((self._vnic, 0x0003))
uptime = time.time()
# read in antenna details and radio description
if conf['antennas']['num'] > 0:
self._antenna['num'] = conf['antennas']['num']
self._antenna['type'] = conf['antennas']['type']
self._antenna['gain'] = conf['antennas']['gain']
self._antenna['loss'] = conf['antennas']['loss']
self._antenna['x'] = [v[0] for v in conf['antennas']['xyz']]
self._antenna['y'] = [v[1] for v in conf['antennas']['xyz']]
self._antenna['z'] = [v[2] for v in conf['antennas']['xyz']]
self._desc = conf['desc']
# compile initial scan pattern from config
scan = [
t for t in conf['scan']
if str(t[0]) in self._chs and not t in conf['pass']
]
# sum hop times with side effect of removing any invalid channel tuples
# i.e. Ch 14, Width HT40+ and channels the card cannot tune to
i = 0
hop = 0
while scan:
try:
t = time.time()
iw.chset(self._vnic, str(scan[i][0]), scan[i][1])
hop += (time.time() - t)
i += 1
except iw.IWException as e:
# if invalid argument, drop the channel otherwise reraise error
if iw.ecode(str(e)) == iw.IW_INVALIDARG:
del scan[i]
else:
raise
except IndexError:
# all channels checked
break
if not scan: raise ValueError("Empty scan pattern")
# calculate avg hop time, and interval time
# NOTE: these are not currently used but may be useful later
hop /= len(scan)
interval = len(scan) * conf['dwell'] + len(scan) * hop
# create list of dwell times
ds = [conf['dwell']] * len(scan)
# get start ch & set the initial channel
try:
ch_i = scan.index(conf['scan_start'])
except ValueError:
ch_i = 0
iw.chset(self._vnic, str(scan[ch_i][0]), scan[ch_i][1])
# initialize tuner thread
self._qT = Queue()
self._tuner = Tuner(self._qT, self._conn, self._vnic, scan, ds,
ch_i, conf['paused'])
# notify RTO we are good
self._icomms.put((self._vnic, uptime, '!UP!', self.radio))
except socket.error as e:
try:
iw.devdel(self._vnic)
iw.phyadd(self._phy, self._nic)
iwt.ifconfig(self._nic, 'up')
except (iw.IWException, iwt.IWToolsException):
pass
if self._s:
self._s.shutdown(socket.SHUT_RD)
self._s.close()
raise RuntimeError("%s:Raw Socket:%s" % (self._role, e))
except iw.IWException as e:
try:
iw.devdel(self._vnic)
iw.phyadd(self._phy, self._nic)
iwt.ifconfig(self._nic, 'up')
except (iw.IWException, iwt.IWToolsException):
pass
if self._s:
self._s.shutdown(socket.SHUT_RD)
self._s.close()
raise RuntimeError("%s:iw.chset:Failed to set channel: %s" %
(self._role, e))
except (ValueError, TypeError) as e:
try:
iw.devdel(self._vnic)
iw.phyadd(self._phy, self._nic)
iwt.ifconfig(self._nic, 'up')
except (iw.IWException, iwt.IWToolsException):
pass
if self._s:
self._s.shutdown(socket.SHUT_RD)
self._s.close()
raise RuntimeError("%s:config:%s" % (self._role, e))
except Exception as e:
# blanket exception
try:
iw.devdel(self._vnic)
iw.phyadd(self._phy, self._nic)
iwt.ifconfig(self._nic, 'up')
except (iw.IWException, iwt.IWToolsException):
pass
if self._s:
self._s.shutdown(socket.SHUT_RD)
self._s.close()
raise RuntimeError("%s:Unknown:%s" % (self._role, e))