46 lines
1.4 KiB
Python
46 lines
1.4 KiB
Python
#!/usr/bin/python3
|
|
import os, os.path
|
|
import wiphy
|
|
|
|
hostapd_map = {ifname: intf for wname, wiphy in wiphy.wiphy_map.items()
|
|
for ifname, intf in wiphy.items() if intf.use == 'hostapd'}
|
|
|
|
class HostapdCLI:
|
|
def __init__(self, interface):
|
|
self.ifname = interface.name
|
|
self.ctrl_interface = interface.ctrl_interface
|
|
|
|
socket_path = os.path.dirname(self.ctrl_interface)
|
|
|
|
self.cmdline = 'hostapd_cli -p"' + socket_path + '" -i"' + \
|
|
self.ifname + '"'
|
|
|
|
def wps_push_button(self):
|
|
os.system(self.cmdline + ' wps_pbc')
|
|
|
|
def deauthenticate(self, client_address):
|
|
os.system(self.cmdline + ' deauthenticate ' + client_address)
|
|
|
|
def reload(self):
|
|
# Seemingly all three commands needed for the instance to notice
|
|
# interface's address change
|
|
cmds = 'reload\ndisable\nenable\n'
|
|
proc = os.popen(self.cmdline, mode='w')
|
|
lines = proc.write(cmds)
|
|
proc.close()
|
|
|
|
def list_sta(self):
|
|
proc = os.popen(self.cmdline + ' list_sta')
|
|
lines = proc.read()
|
|
proc.close()
|
|
|
|
return [line for line in lines.split('\n') if line]
|
|
|
|
def set_neighbor(self, addr, ssid, nr):
|
|
os.system(self.cmdline + ' set_neighbor ' + addr + ' ssid=\\""' + ssid +
|
|
'"\\" nr=' + nr)
|
|
|
|
@staticmethod
|
|
def kill_all():
|
|
os.system('killall hostapd')
|