iwd/autotests/util/hostapd.py

#!/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')