# Test cases for DFS
# Copyright (c) 2013-2019, Jouni Malinen <j@w1.fi>
#
# This software may be distributed under the terms of the BSD license.
# See README for more details.

from remotehost import remote_compatible
import os
import subprocess
import time
import logging
logger = logging.getLogger()

import hwsim_utils
import hostapd
from utils import *

def wait_dfs_event(hapd, event, timeout):
    dfs_events = ["DFS-RADAR-DETECTED", "DFS-NEW-CHANNEL",
                  "DFS-CAC-START", "DFS-CAC-COMPLETED",
                  "DFS-NOP-FINISHED", "AP-ENABLED", "AP-CSA-FINISHED"]
    ev = hapd.wait_event(dfs_events, timeout=timeout)
    if not ev:
        raise Exception("DFS event timed out")
    if event and event not in ev:
        raise Exception("Unexpected DFS event: " + ev + " (expected: %s)" % event)
    return ev

def start_dfs_ap(ap, ssid="dfs", ht=True, ht40=False,
                 ht40minus=False, vht80=False, vht20=False, chanlist=None,
                 channel=None, country="FI", rrm_beacon_report=False,
                 chan100=False):
    ifname = ap['ifname']
    logger.info("Starting AP " + ifname + " on DFS channel")
    hapd = hostapd.add_ap(ap, {}, no_enable=True)
    hapd.set("ssid", ssid)
    hapd.set("country_code", country)
    hapd.set("ieee80211d", "1")
    hapd.set("ieee80211h", "1")
    hapd.set("hw_mode", "a")
    if chan100:
        hapd.set("channel", "100")
    else:
        hapd.set("channel", "52")
    if not ht:
        hapd.set("ieee80211n", "0")
    if ht40:
        hapd.set("ht_capab", "[HT40+]")
    elif ht40minus:
        hapd.set("ht_capab", "[HT40-]")
        hapd.set("channel", "56")
    if vht80:
        hapd.set("ieee80211ac", "1")
        hapd.set("vht_oper_chwidth", "1")
        if chan100:
            hapd.set("vht_oper_centr_freq_seg0_idx", "106")
        else:
            hapd.set("vht_oper_centr_freq_seg0_idx", "58")
    if vht20:
        hapd.set("ieee80211ac", "1")
        hapd.set("vht_oper_chwidth", "0")
        hapd.set("vht_oper_centr_freq_seg0_idx", "0")
    if chanlist:
        hapd.set("chanlist", chanlist)
    if channel:
        hapd.set("channel", str(channel))
    if rrm_beacon_report:
        hapd.set("rrm_beacon_report", "1")
    hapd.enable()

    ev = wait_dfs_event(hapd, "DFS-CAC-START", 5)
    if "DFS-CAC-START" not in ev:
        raise Exception("Unexpected DFS event: " + ev)

    state = hapd.get_status_field("state")
    if state != "DFS":
        raise Exception("Unexpected interface state: " + state)

    return hapd

def dfs_simulate_radar(hapd):
    logger.info("Trigger a simulated radar event")
    phyname = hapd.get_driver_status_field("phyname")
    radar_file = '/sys/kernel/debug/ieee80211/' + phyname + '/hwsim/dfs_simulate_radar'
    with open(radar_file, 'w') as f:
        f.write('1')

def test_dfs(dev, apdev):
    """DFS CAC functionality on clear channel"""
    try:
        hapd = None
        hapd = start_dfs_ap(apdev[0], country="US")

        ev = wait_dfs_event(hapd, "DFS-CAC-COMPLETED", 70)
        if "success=1" not in ev:
            raise Exception("CAC failed")
        if "freq=5260" not in ev:
            raise Exception("Unexpected DFS freq result")

        ev = hapd.wait_event(["AP-ENABLED"], timeout=5)
        if not ev:
            raise Exception("AP setup timed out")

        state = hapd.get_status_field("state")
        if state != "ENABLED":
            raise Exception("Unexpected interface state")

        freq = hapd.get_status_field("freq")
        if freq != "5260":
            raise Exception("Unexpected frequency")

        dev[0].connect("dfs", key_mgmt="NONE", timeout=30)
        dev[0].wait_regdom(country_ie=True)
        hwsim_utils.test_connectivity(dev[0], hapd)

        hapd.request("RADAR DETECTED freq=5260 ht_enabled=1 chan_width=1")
        ev = hapd.wait_event(["DFS-RADAR-DETECTED"], timeout=10)
        if ev is None:
            raise Exception("DFS-RADAR-DETECTED event not reported")
        if "freq=5260" not in ev:
            raise Exception("Incorrect frequency in radar detected event: " + ev)
        ev = hapd.wait_event(["DFS-NEW-CHANNEL"], timeout=70)
        if ev is None:
            raise Exception("DFS-NEW-CHANNEL event not reported")
        if "freq=5260" in ev:
            raise Exception("Channel did not change after radar was detected")

        ev = hapd.wait_event(["AP-CSA-FINISHED"], timeout=70)
        if ev is None:
            raise Exception("AP-CSA-FINISHED event not reported")
        if "freq=5260" in ev:
            raise Exception("Channel did not change after radar was detected(2)")
        time.sleep(1)
        hwsim_utils.test_connectivity(dev[0], hapd)
    finally:
        clear_regdom(hapd, dev)

@long_duration_test
def test_dfs_etsi(dev, apdev):
    """DFS and uniform spreading requirement for ETSI"""
    try:
        hapd = None
        hapd = start_dfs_ap(apdev[0])

        ev = wait_dfs_event(hapd, "DFS-CAC-COMPLETED", 70)
        if "success=1" not in ev:
            raise Exception("CAC failed")
        if "freq=5260" not in ev:
            raise Exception("Unexpected DFS freq result")

        ev = hapd.wait_event(["AP-ENABLED"], timeout=5)
        if not ev:
            raise Exception("AP setup timed out")

        state = hapd.get_status_field("state")
        if state != "ENABLED":
            raise Exception("Unexpected interface state")

        freq = hapd.get_status_field("freq")
        if freq != "5260":
            raise Exception("Unexpected frequency")

        hapd.dump_monitor()
        dev[0].dump_monitor()

        dev[0].connect("dfs", key_mgmt="NONE")
        dev[0].wait_regdom(country_ie=True)
        time.sleep(1)
        hwsim_utils.test_connectivity(dev[0], hapd)

        hapd.dump_monitor()
        dev[0].dump_monitor()

        hapd.request("RADAR DETECTED freq=%s ht_enabled=1 chan_width=1" % freq)
        ev = hapd.wait_event(["DFS-RADAR-DETECTED"], timeout=5)
        if ev is None:
            raise Exception("DFS-RADAR-DETECTED event not reported")
        if "freq=%s" % freq not in ev:
            raise Exception("Incorrect frequency in radar detected event: " + ev)
        ev = hapd.wait_event(["DFS-NEW-CHANNEL"], timeout=5)
        if ev is None:
            raise Exception("DFS-NEW-CHANNEL event not reported")
        if "freq=%s" % freq in ev:
            raise Exception("Channel did not change after radar was detected")

        ev = hapd.wait_event(["AP-CSA-FINISHED", "DFS-CAC-START"], timeout=10)
        if ev is None:
            raise Exception("AP-CSA-FINISHED or DFS-CAC-START event not reported")
        if "DFS-CAC-START" in ev:
            # The selected new channel requires CAC
            dev[0].wait_disconnected()
            ev = wait_dfs_event(hapd, "DFS-CAC-COMPLETED", 70)
            if "success=1" not in ev:
                raise Exception("CAC failed")

            ev = hapd.wait_event(["AP-ENABLED"], timeout=5)
            if not ev:
                raise Exception("AP setup timed out")
            ev = hapd.wait_event(["AP-STA-CONNECTED"], timeout=30)
            if not ev:
                raise Exception("STA did not reconnect on new DFS channel")
            dev[0].wait_connected()
        else:
            # The new channel did not require CAC - try again
            if "freq=%s" % freq in ev:
                raise Exception("Channel did not change after radar was detected(2)")
            ev = dev[0].wait_event(["CTRL-EVENT-CHANNEL-SWITCH"], timeout=10)
            if ev is None:
                raise Exception("Channel switch not reported on STA")
            time.sleep(1)

        hapd.dump_monitor()
        dev[0].dump_monitor()
        time.sleep(1)
        hwsim_utils.test_connectivity(dev[0], hapd)
    finally:
        clear_regdom(hapd, dev)

def test_dfs_radar1(dev, apdev):
    """DFS CAC functionality with radar detected during initial CAC"""
    try:
        hapd = None
        hapd = start_dfs_ap(apdev[0])
        time.sleep(1)

        dfs_simulate_radar(hapd)

        ev = wait_dfs_event(hapd, "DFS-CAC-COMPLETED", 5)
        if ev is None:
            raise Exception("Timeout on DFS aborted event")
        if "success=0 freq=5260" not in ev:
            raise Exception("Unexpected DFS aborted event contents: " + ev)

        ev = wait_dfs_event(hapd, "DFS-RADAR-DETECTED", 5)
        if "freq=5260" not in ev:
            raise Exception("Unexpected DFS radar detection freq")

        ev = wait_dfs_event(hapd, "DFS-NEW-CHANNEL", 5)
        if "freq=5260" in ev:
            raise Exception("Unexpected DFS new freq")

        ev = wait_dfs_event(hapd, None, 5)
        if "AP-ENABLED" in ev:
            logger.info("Started AP on non-DFS channel")
        else:
            logger.info("Trying to start AP on another DFS channel")
            if "DFS-CAC-START" not in ev:
                raise Exception("Unexpected DFS event: " + ev)
            if "freq=5260" in ev:
                raise Exception("Unexpected DFS CAC freq")

            ev = wait_dfs_event(hapd, "DFS-CAC-COMPLETED", 70)
            if "success=1" not in ev:
                raise Exception("CAC failed")
            if "freq=5260" in ev:
                raise Exception("Unexpected DFS freq result - radar channel")

            ev = hapd.wait_event(["AP-ENABLED"], timeout=5)
            if not ev:
                raise Exception("AP setup timed out")

            state = hapd.get_status_field("state")
            if state != "ENABLED":
                raise Exception("Unexpected interface state")

            freq = hapd.get_status_field("freq")
            if freq == "5260":
                raise Exception("Unexpected frequency: " + freq)

        dev[0].connect("dfs", key_mgmt="NONE")
        dev[0].wait_regdom(country_ie=True)
    finally:
        clear_regdom(hapd, dev)

def test_dfs_radar2(dev, apdev):
    """DFS CAC functionality with radar detected after initial CAC"""
    try:
        hapd = None
        hapd = start_dfs_ap(apdev[0], ssid="dfs2", ht40=True)

        ev = hapd.wait_event(["AP-ENABLED"], timeout=70)
        if not ev:
            raise Exception("AP2 setup timed out")

        dfs_simulate_radar(hapd)

        ev = wait_dfs_event(hapd, "DFS-RADAR-DETECTED", 5)
        if "freq=5260 ht_enabled=1 chan_offset=1 chan_width=2" not in ev:
            raise Exception("Unexpected DFS radar detection freq from AP2")

        ev = wait_dfs_event(hapd, "DFS-NEW-CHANNEL", 5)
        if "freq=5260" in ev:
            raise Exception("Unexpected DFS new freq for AP2")

        wait_dfs_event(hapd, None, 5)
    finally:
        clear_regdom(hapd, dev)

@remote_compatible
def test_dfs_radar_on_non_dfs_channel(dev, apdev):
    """DFS radar detection test code on non-DFS channel"""
    params = {"ssid": "radar"}
    hapd = hostapd.add_ap(apdev[0], params)

    hapd.request("RADAR DETECTED freq=5260 ht_enabled=1 chan_width=1")
    hapd.request("RADAR DETECTED freq=2412 ht_enabled=1 chan_width=1")

def test_dfs_radar_chanlist(dev, apdev):
    """DFS chanlist when radar is detected"""
    try:
        hapd = None
        hapd = start_dfs_ap(apdev[0], chanlist="40 44")
        time.sleep(1)

        dfs_simulate_radar(hapd)

        ev = wait_dfs_event(hapd, "DFS-CAC-COMPLETED", 5)
        if ev is None:
            raise Exception("Timeout on DFS aborted event")
        if "success=0 freq=5260" not in ev:
            raise Exception("Unexpected DFS aborted event contents: " + ev)

        ev = wait_dfs_event(hapd, "DFS-RADAR-DETECTED", 5)
        if "freq=5260" not in ev:
            raise Exception("Unexpected DFS radar detection freq")

        ev = wait_dfs_event(hapd, "DFS-NEW-CHANNEL", 5)
        if "freq=5200 chan=40" not in ev and "freq=5220 chan=44" not in ev:
            raise Exception("Unexpected DFS new freq: " + ev)

        ev = wait_dfs_event(hapd, None, 5)
        if "AP-ENABLED" not in ev:
            raise Exception("Unexpected DFS event: " + ev)
        dev[0].connect("dfs", key_mgmt="NONE")
        dev[0].wait_regdom(country_ie=True)
    finally:
        clear_regdom(hapd, dev)

def test_dfs_radar_chanlist_vht80(dev, apdev):
    """DFS chanlist when radar is detected and VHT80 configured"""
    try:
        hapd = None
        hapd = start_dfs_ap(apdev[0], chanlist="36", ht40=True, vht80=True)
        time.sleep(1)

        dfs_simulate_radar(hapd)

        ev = wait_dfs_event(hapd, "DFS-CAC-COMPLETED", 5)
        if ev is None:
            raise Exception("Timeout on DFS aborted event")
        if "success=0 freq=5260" not in ev:
            raise Exception("Unexpected DFS aborted event contents: " + ev)

        ev = wait_dfs_event(hapd, "DFS-RADAR-DETECTED", 5)
        if "freq=5260" not in ev:
            raise Exception("Unexpected DFS radar detection freq")

        ev = wait_dfs_event(hapd, "DFS-NEW-CHANNEL", 5)
        if "freq=5180 chan=36 sec_chan=1" not in ev:
            raise Exception("Unexpected DFS new freq: " + ev)

        ev = wait_dfs_event(hapd, None, 5)
        if "AP-ENABLED" not in ev:
            raise Exception("Unexpected DFS event: " + ev)
        dev[0].connect("dfs", key_mgmt="NONE")
        dev[0].wait_regdom(country_ie=True)

        if hapd.get_status_field('vht_oper_centr_freq_seg0_idx') != "42":
            raise Exception("Unexpected seg0 idx")
    finally:
        clear_regdom(hapd, dev)

def test_dfs_radar_chanlist_vht20(dev, apdev):
    """DFS chanlist when radar is detected and VHT40 configured"""
    try:
        hapd = None
        hapd = start_dfs_ap(apdev[0], chanlist="36", vht20=True)
        time.sleep(1)

        dfs_simulate_radar(hapd)

        ev = wait_dfs_event(hapd, "DFS-CAC-COMPLETED", 5)
        if ev is None:
            raise Exception("Timeout on DFS aborted event")
        if "success=0 freq=5260" not in ev:
            raise Exception("Unexpected DFS aborted event contents: " + ev)

        ev = wait_dfs_event(hapd, "DFS-RADAR-DETECTED", 5)
        if "freq=5260" not in ev:
            raise Exception("Unexpected DFS radar detection freq")

        ev = wait_dfs_event(hapd, "DFS-NEW-CHANNEL", 5)
        if "freq=5180 chan=36 sec_chan=0" not in ev:
            raise Exception("Unexpected DFS new freq: " + ev)

        ev = wait_dfs_event(hapd, None, 5)
        if "AP-ENABLED" not in ev:
            raise Exception("Unexpected DFS event: " + ev)
        dev[0].connect("dfs", key_mgmt="NONE")
        dev[0].wait_regdom(country_ie=True)
    finally:
        clear_regdom(hapd, dev)

def test_dfs_radar_no_ht(dev, apdev):
    """DFS chanlist when radar is detected and no HT configured"""
    try:
        hapd = None
        hapd = start_dfs_ap(apdev[0], chanlist="36", ht=False)
        time.sleep(1)

        dfs_simulate_radar(hapd)

        ev = wait_dfs_event(hapd, "DFS-CAC-COMPLETED", 5)
        if ev is None:
            raise Exception("Timeout on DFS aborted event")
        if "success=0 freq=5260" not in ev:
            raise Exception("Unexpected DFS aborted event contents: " + ev)

        ev = wait_dfs_event(hapd, "DFS-RADAR-DETECTED", 5)
        if "freq=5260 ht_enabled=0" not in ev:
            raise Exception("Unexpected DFS radar detection freq: " + ev)

        ev = wait_dfs_event(hapd, "DFS-NEW-CHANNEL", 5)
        if "freq=5180 chan=36 sec_chan=0" not in ev:
            raise Exception("Unexpected DFS new freq: " + ev)

        ev = wait_dfs_event(hapd, None, 5)
        if "AP-ENABLED" not in ev:
            raise Exception("Unexpected DFS event: " + ev)
        dev[0].connect("dfs", key_mgmt="NONE")
        dev[0].wait_regdom(country_ie=True)
    finally:
        clear_regdom(hapd, dev)

def test_dfs_radar_ht40minus(dev, apdev):
    """DFS chanlist when radar is detected and HT40- configured"""
    try:
        hapd = None
        hapd = start_dfs_ap(apdev[0], chanlist="36", ht40minus=True)
        time.sleep(1)

        dfs_simulate_radar(hapd)

        ev = wait_dfs_event(hapd, "DFS-CAC-COMPLETED", 5)
        if ev is None:
            raise Exception("Timeout on DFS aborted event")
        if "success=0 freq=5280 ht_enabled=1 chan_offset=-1" not in ev:
            raise Exception("Unexpected DFS aborted event contents: " + ev)

        ev = wait_dfs_event(hapd, "DFS-RADAR-DETECTED", 5)
        if "freq=5280 ht_enabled=1 chan_offset=-1" not in ev:
            raise Exception("Unexpected DFS radar detection freq: " + ev)

        ev = wait_dfs_event(hapd, "DFS-NEW-CHANNEL", 5)
        if "freq=5180 chan=36 sec_chan=1" not in ev:
            raise Exception("Unexpected DFS new freq: " + ev)

        ev = wait_dfs_event(hapd, None, 5)
        if "AP-ENABLED" not in ev:
            raise Exception("Unexpected DFS event: " + ev)
        dev[0].connect("dfs", key_mgmt="NONE")
        dev[0].wait_regdom(country_ie=True)
        dev[0].request("STA_AUTOCONNECT 0")
    finally:
        clear_regdom(hapd, dev)
        dev[0].request("STA_AUTOCONNECT 1")

@long_duration_test
def test_dfs_ht40_minus(dev, apdev):
    """DFS CAC functionality on channel 104 HT40-"""
    try:
        hapd = None
        hapd = start_dfs_ap(apdev[0], ht40minus=True, channel=104)

        ev = wait_dfs_event(hapd, "DFS-CAC-COMPLETED", 70)
        if "success=1" not in ev:
            raise Exception("CAC failed")
        if "freq=5520" not in ev:
            raise Exception("Unexpected DFS freq result")

        ev = hapd.wait_event(["AP-ENABLED"], timeout=5)
        if not ev:
            raise Exception("AP setup timed out")

        state = hapd.get_status_field("state")
        if state != "ENABLED":
            raise Exception("Unexpected interface state")

        freq = hapd.get_status_field("freq")
        if freq != "5520":
            raise Exception("Unexpected frequency")

        dev[0].connect("dfs", key_mgmt="NONE", scan_freq="5520")
        dev[0].wait_regdom(country_ie=True)
        hwsim_utils.test_connectivity(dev[0], hapd)
    finally:
        clear_regdom(hapd, dev)

def test_dfs_cac_restart_on_enable(dev, apdev):
    """DFS CAC interrupted and restarted"""
    try:
        hapd = None
        hapd = start_dfs_ap(apdev[0])
        time.sleep(0.1)
        subprocess.check_call(['ip', 'link', 'set', 'dev', hapd.ifname, 'down'])
        ev = wait_dfs_event(hapd, "DFS-CAC-COMPLETED", 5)
        if ev is None:
            raise Exception("Timeout on DFS aborted event")
        if "success=0 freq=5260" not in ev:
            raise Exception("Unexpected DFS aborted event contents: " + ev)
        time.sleep(0.1)
        subprocess.check_call(['ip', 'link', 'set', 'dev', hapd.ifname, 'up'])

        ev = wait_dfs_event(hapd, "DFS-CAC-START", 5)
        if "DFS-CAC-START" not in ev:
            raise Exception("Unexpected DFS event: " + ev)
        hapd.disable()

    finally:
        clear_regdom(hapd, dev)

@long_duration_test
def test_dfs_rrm(dev, apdev):
    """DFS with RRM"""
    try:
        hapd = None
        hapd = start_dfs_ap(apdev[0], country="US", rrm_beacon_report=True)

        ev = wait_dfs_event(hapd, "DFS-CAC-COMPLETED", 70)
        if "success=1" not in ev or "freq=5260" not in ev:
            raise Exception("Unexpected DFS freq result")
        ev = hapd.wait_event(["AP-ENABLED"], timeout=5)
        if not ev:
            raise Exception("AP setup timed out")

        dev[0].connect("dfs", key_mgmt="NONE", scan_freq="5260")
        dev[0].wait_regdom(country_ie=True)
        hapd.wait_sta()
        hwsim_utils.test_connectivity(dev[0], hapd)
        addr = dev[0].own_addr()
        token = hapd.request("REQ_BEACON " + addr + " " + "51000000000002ffffffffffff")
        if "FAIL" in token:
            raise Exception("REQ_BEACON failed")
        ev = hapd.wait_event(["BEACON-RESP-RX"], timeout=10)
        if ev is None:
            raise Exception("Beacon report response not received")
    finally:
        clear_regdom(hapd, dev)

@long_duration_test
def test_dfs_radar_vht80_downgrade(dev, apdev):
    """DFS channel bandwidth downgrade from VHT80 to VHT40"""
    try:
        # Start with 80 MHz channel 100 (5500 MHz) to find a radar
        hapd = None
        hapd = start_dfs_ap(apdev[0], chanlist="100-140",
                            ht40=True, vht80=True, chan100=True)
        time.sleep(1)
        dfs_simulate_radar(hapd)

        ev = wait_dfs_event(hapd, "DFS-CAC-COMPLETED", 5)
        if ev is None:
            raise Exception("Timeout on DFS aborted event")
        if "success=0 freq=5500" not in ev:
            raise Exception("Unexpected DFS aborted event contents: " + ev)

        ev = wait_dfs_event(hapd, "DFS-RADAR-DETECTED", 5)
        if "freq=5500" not in ev:
            raise Exception("Unexpected DFS radar detection freq: " + ev)

        # The only other available 80 MHz channel in the chanlist is
        # 116 (5580 MHz), so that will be selected next.
        ev = wait_dfs_event(hapd, "DFS-NEW-CHANNEL", 5)
        if "freq=5580 chan=116 sec_chan=1" not in ev:
            raise Exception("Unexpected DFS new freq: " + ev)

        ev = wait_dfs_event(hapd, None, 5)
        if "DFS-CAC-START" not in ev:
            raise Exception("Unexpected DFS event: " + ev)
        if "freq=5580" not in ev:
            raise Exception("Unexpected DFS CAC freq: " + ev)

        time.sleep(1)
        dfs_simulate_radar(hapd)
        ev = wait_dfs_event(hapd, "DFS-CAC-COMPLETED", 5)
        if ev is None:
            raise Exception("Timeout on DFS aborted event (2)")
        if "success=0 freq=5580" not in ev:
            raise Exception("Unexpected DFS aborted event (2) contents: " + ev)

        ev = wait_dfs_event(hapd, "DFS-RADAR-DETECTED", 5)
        if "freq=5580" not in ev:
            raise Exception("Unexpected DFS radar detection (2) freq: " + ev)

        # No more 80 MHz channels are available, so have to downgrade to 40 MHz
        # channels and the only remaining one is channel 132 (5660 MHz).
        ev = wait_dfs_event(hapd, "DFS-NEW-CHANNEL", 5)
        if "freq=5660 chan=132 sec_chan=1" not in ev:
            raise Exception("Unexpected DFS new freq (2): " + ev)

        ev = wait_dfs_event(hapd, None, 5)
        if "DFS-CAC-START" not in ev:
            raise Exception("Unexpected DFS event: " + ev)
        if "freq=5660" not in ev:
            raise Exception("Unexpected DFS CAC freq (2): " + ev)

        ev = wait_dfs_event(hapd, "DFS-CAC-COMPLETED", 70)
        if "success=1" not in ev:
            raise Exception("CAC failed")
        if "freq=5660" not in ev:
            raise Exception("Unexpected DFS freq result: " + ev)

        ev = wait_dfs_event(hapd, None, 5)
        if "AP-ENABLED" not in ev:
            raise Exception("Unexpected DFS event: " + ev)
        dev[0].connect("dfs", key_mgmt="NONE", scan_freq="5660")
        dev[0].wait_regdom(country_ie=True)
        sig = dev[0].request("SIGNAL_POLL").splitlines()
        if "FREQUENCY=5660" not in sig or "WIDTH=40 MHz" not in sig:
            raise Exception("Unexpected SIGNAL_POLL value: " + str(sig))
    finally:
        clear_regdom(hapd, dev)

@long_duration_test
def test_dfs_chan_switch(dev, apdev):
    """DFS channel switch"""
    try:
        hapd = None
        hapd = start_dfs_ap(apdev[0], country="US")

        ev = wait_dfs_event(hapd, "DFS-CAC-COMPLETED", 70)
        if "success=1" not in ev:
            raise Exception("CAC failed")
        if "freq=5260" not in ev:
            raise Exception("Unexpected DFS freq result")
        ev = hapd.wait_event(["AP-ENABLED"], timeout=5)
        if not ev:
            raise Exception("AP setup timed out")
        freq = hapd.get_status_field("freq")
        if freq != "5260":
            raise Exception("Unexpected frequency")

        dev[0].connect("dfs", key_mgmt="NONE", scan_freq="5260 5280 5180 5500")
        dev[0].wait_regdom(country_ie=True)
        hwsim_utils.test_connectivity(dev[0], hapd)

        if "OK" not in hapd.request("CHAN_SWITCH 5 5280 ht"):
            raise Exception("CHAN_SWITCH failed")
        # This results in BSS going down before restart, so the STA is expected
        # to report disconnection.
        dev[0].wait_disconnected()
        ev = wait_dfs_event(hapd, "DFS-CAC-START", 5)
        if "freq=5280" not in ev:
            raise Exception("Unexpected channel: " + ev)
        ev = wait_dfs_event(hapd, "DFS-CAC-COMPLETED", 70)
        if "success=1" not in ev:
            raise Exception("CAC failed")
        if "freq=5280" not in ev:
            raise Exception("Unexpected DFS freq result")
        ev = hapd.wait_event(["AP-ENABLED"], timeout=5)
        if not ev:
            raise Exception("AP setup timed out")
        freq = hapd.get_status_field("freq")
        if freq != "5280":
            raise Exception("Unexpected frequency")

        dev[0].wait_connected(timeout=30)
        hwsim_utils.test_connectivity(dev[0], hapd)

        # Move to non DFS channel: 36, opclass 115
        if "OK" not in hapd.request("CHAN_SWITCH 5 5180"):
                raise Exception("CHAN_SWITCH failed")
        ev = wait_dfs_event(hapd, "AP-CSA-FINISHED", 5)

        if "freq=5180" not in ev:
            raise Exception("Unexpected channel: " + ev)
        time.sleep(1)
        hwsim_utils.test_connectivity(dev[0], hapd)

        # Move to channel 100, opclass 121
        if "OK" not in hapd.request("CHAN_SWITCH 5 5500 ht"):
            raise Exception("CHAN_SWITCH failed")
        # This results in BSS going down before restart, so the STA is expected
        # to report disconnection.
        dev[0].wait_disconnected()
        ev = wait_dfs_event(hapd, "DFS-CAC-START", 5)
        if "freq=5500" not in ev:
            raise Exception("Unexpected channel: " + ev)
        ev = wait_dfs_event(hapd, "DFS-CAC-COMPLETED", 70)
        if "success=1" not in ev:
            raise Exception("CAC failed")
        if "freq=5500" not in ev:
            raise Exception("Unexpected DFS freq result")
        ev = hapd.wait_event(["AP-ENABLED"], timeout=5)
        if not ev:
            raise Exception("AP setup timed out")
        freq = hapd.get_status_field("freq")
        if freq != "5500":
            raise Exception("Unexpected frequency")

        dev[0].wait_connected(timeout=30)
        hwsim_utils.test_connectivity(dev[0], hapd)
    finally:
        clear_regdom(hapd, dev)

@long_duration_test
def test_dfs_chan_switch_to_vht80(dev, apdev):
    """DFS channel switch to VHT80"""
    run_dfs_chan_switch_to_vht(dev, apdev, vht80=True)

@long_duration_test
def test_dfs_chan_switch_to_vht80p80(dev, apdev):
    """DFS channel switch to VHT80+80"""
    run_dfs_chan_switch_to_vht(dev, apdev, vht80p80=True)

@long_duration_test
def test_dfs_chan_switch_to_vht160(dev, apdev):
    """DFS channel switch to VHT160"""
    run_dfs_chan_switch_to_vht(dev, apdev, vht160=True)

@long_duration_test
def test_dfs_chan_switch_to_vht160b(dev, apdev):
    """DFS channel switch to VHT160 from non-DFS"""
    run_dfs_chan_switch_to_vht(dev, apdev, vht160=True, start_dfs=False)

def run_dfs_chan_switch_to_vht(dev, apdev, vht80=False, vht80p80=False,
                               vht160=False, start_dfs=True):
    try:
        hapd = None
        if start_dfs:
            hapd = start_dfs_ap(apdev[0], country="US")

            ev = wait_dfs_event(hapd, "DFS-CAC-COMPLETED", 70)
            if "success=1" not in ev:
                raise Exception("CAC failed")
            if "freq=5260" not in ev:
                raise Exception("Unexpected DFS freq result")
            ev = hapd.wait_event(["AP-ENABLED"], timeout=5)
            if not ev:
                raise Exception("AP setup timed out")
            freq = hapd.get_status_field("freq")
            if freq != "5260":
                raise Exception("Unexpected frequency")

            dev[0].connect("dfs", key_mgmt="NONE", scan_freq="5260 5180")
        else:
            params = {"ssid": "dfs",
                      "country_code": "US",
                      "ieee80211d": "1",
                      "hw_mode": "a",
                      "channel": "40"}
            hapd = hostapd.add_ap(apdev[0], params)
            dev[0].connect("dfs", key_mgmt="NONE", scan_freq="5260 5180 5200")
        dev[0].wait_regdom(country_ie=True)
        hwsim_utils.test_connectivity(dev[0], hapd)

        if vht160:
            if not start_dfs:
                hapd.set("ieee80211h", "1")
            cmd = "CHAN_SWITCH 5 5180 sec_channel_offset=1 center_freq1=5250 bandwidth=160 ht vht"
        if vht80p80:
            cmd = "CHAN_SWITCH 5 5180 sec_channel_offset=1 center_freq1=5210 center_freq2=5775 bandwidth=80 ht vht"
        if vht80:
            cmd = "CHAN_SWITCH 5 5180 sec_channel_offset=1 center_freq1=5210 bandwidth=80 ht vht"
        if "OK" not in hapd.request(cmd):
            raise Exception("CHAN_SWITCH command failed")
        if vht160:
            # This results in BSS going down before restart, so the STA is
            # expected to report disconnection.
            dev[0].wait_disconnected()
            ev = wait_dfs_event(hapd, "DFS-CAC-START", 5)
            if "freq=5180" not in ev:
                raise Exception("Unexpected channel: " + ev)
            ev = wait_dfs_event(hapd, "DFS-CAC-COMPLETED", 70)
            if "success=1" not in ev:
                raise Exception("CAC failed")
        else:
            ev = wait_dfs_event(hapd, "AP-CSA-FINISHED", 5)
        if "freq=5180" not in ev:
            raise Exception("Unexpected channel: " + ev)
        if vht160:
            ev = hapd.wait_event(["AP-ENABLED"], timeout=5)
            if not ev:
                raise Exception("AP setup timed out")

        freq = hapd.get_status_field("freq")
        if freq != "5180":
            raise Exception("Unexpected frequency")

        status = hapd.get_status()
        logger.info("hostapd STATUS: " + str(status))
        if status["ieee80211n"] != "1":
            raise Exception("Unexpected STATUS ieee80211n value")
        if status["ieee80211ac"] != "1":
            raise Exception("Unexpected STATUS ieee80211ac value")
        if status["secondary_channel"] != "1":
            raise Exception("Unexpected STATUS secondary_channel value")
        if vht80 and status["vht_oper_chwidth"] != "1":
            raise Exception("Unexpected STATUS vht_oper_chwidth value")
        if vht80p80 and status["vht_oper_chwidth"] != "3":
            raise Exception("Unexpected STATUS vht_oper_chwidth value")
        if vht160 and status["vht_oper_chwidth"] != "2":
            raise Exception("Unexpected STATUS vht_oper_chwidth value")
        if vht160:
            if status["vht_oper_centr_freq_seg0_idx"] != "50":
                raise Exception("Unexpected STATUS vht_oper_centr_freq_seg0_idx value")
        else:
            if status["vht_oper_centr_freq_seg0_idx"] != "42":
                raise Exception("Unexpected STATUS vht_oper_centr_freq_seg0_idx value")
        if vht80p80 and status["vht_oper_centr_freq_seg1_idx"] != "155":
            raise Exception("Unexpected STATUS vht_oper_centr_freq_seg1_idx value")
        if "vht_caps_info" not in status:
            raise Exception("Missing vht_caps_info")

        if vht160:
            dev[0].wait_connected(timeout=30)
        else:
            ev = dev[0].wait_event(["CTRL-EVENT-CHANNEL-SWITCH"], timeout=10)
            if ev is None:
                raise Exception("Channel switch not reported")
            if "freq=5180" not in ev:
                raise Exception("Unexpected frequency: " + ev)
        sig = dev[0].request("SIGNAL_POLL").splitlines()
        logger.info("STA0: " + str(sig))
        if "FREQUENCY=5180" not in sig:
            raise Exception("Unexpected channel: " + str(sig))
        if vht160 and "WIDTH=160 MHz" not in sig:
            raise Exception("Unexpected channel[1]: " + str(sig))
        hwsim_utils.test_connectivity(dev[0], hapd)

        dev[1].connect("dfs", key_mgmt="NONE", scan_freq="5180")
        sig = dev[1].request("SIGNAL_POLL").splitlines()
        logger.info("STA1: " + str(sig))
        if "FREQUENCY=5180" not in sig:
            raise Exception("Unexpected channel[1]: " + str(sig))
        if vht80 and "WIDTH=80 MHz" not in sig:
            raise Exception("Unexpected channel[1]: " + str(sig))
        if vht80p80 and "WIDTH=80+80 MHz" not in sig:
            raise Exception("Unexpected channel[1]: " + str(sig))
        if vht160 and "WIDTH=160 MHz" not in sig:
            raise Exception("Unexpected channel[1]: " + str(sig))
    finally:
        clear_regdom(hapd, dev)

@long_duration_test
def test_dfs_no_available_channel(dev, apdev):
    """DFS and no available channel after radar detection"""
    try:
        hapd = None
        hapd = start_dfs_ap(apdev[0], chanlist="56")

        ev = hapd.wait_event(["AP-ENABLED"], timeout=70)
        if not ev:
            raise Exception("AP2 setup timed out")

        dfs_simulate_radar(hapd)
        ev = wait_dfs_event(hapd, "DFS-RADAR-DETECTED", 5)
        if "freq=5260 ht_enabled=1 chan_offset=0 chan_width=1" not in ev:
            raise Exception("Unexpected DFS radar detection freq from AP")

        ev = wait_dfs_event(hapd, "DFS-NEW-CHANNEL", 5)
        if "freq=5280 chan=56" not in ev:
            raise Exception("Unexpected DFS new freq: " + ev)
        ev = wait_dfs_event(hapd, "DFS-CAC-START", 5)
        if "freq=5280" not in ev:
            raise Exception("Unexpected channel: " + ev)
        ev = wait_dfs_event(hapd, "DFS-CAC-COMPLETED", 70)
        if "success=1" not in ev:
            raise Exception("CAC failed")
        if "freq=5280" not in ev:
            raise Exception("Unexpected DFS freq result")
        ev = hapd.wait_event(["AP-ENABLED"], timeout=5)
        if not ev:
            raise Exception("AP setup timed out")

        dfs_simulate_radar(hapd)
        ev = wait_dfs_event(hapd, "DFS-RADAR-DETECTED", 5)
        if "freq=5280 ht_enabled=1 chan_offset=0 chan_width=1" not in ev:
            raise Exception("Unexpected DFS radar detection freq from AP [2]")

        ev = hapd.wait_event(["AP-DISABLED"], timeout=10)
        if ev is None:
            raise Exception("AP was not disabled")
    finally:
        clear_regdom(hapd, dev)

def dfs_chan_switch_precac(dev, apdev, country):
    """DFS channel switch pre CAC"""
    try:
        hapd = None

        # Toggle regulatory - clean all preCAC
        hostapd.cmd_execute(apdev[0], ['iw', 'reg', 'set', 'US'])

        hapd = start_dfs_ap(apdev[0], country=country)

        ev = wait_dfs_event(hapd, "DFS-CAC-COMPLETED", 70)
        if "success=1" not in ev:
            raise Exception("CAC failed")
        if "freq=5260" not in ev:
            raise Exception("Unexpected DFS freq result")
        ev = hapd.wait_event(["AP-ENABLED"], timeout=5)
        if not ev:
            raise Exception("AP setup timed out")
        freq = hapd.get_status_field("freq")
        if freq != "5260":
            raise Exception("Unexpected frequency")

        # TODO add/connect station here
        # Today skip this step while dev[0].connect()
        # for some reason toggle regulatory to US
        # and clean preCAC

        # Back to non DFS channel
        if "OK" not in hapd.request("CHAN_SWITCH 5 5180 ht"):
            raise Exception("CHAN_SWITCH 5180 failed")

        ev = hapd.wait_event(["AP-CSA-FINISHED"], timeout=5)
        if not ev:
            raise Exception("No CSA finished event - 5180")
        freq = hapd.get_status_field("freq")
        if freq != "5180":
            raise Exception("Unexpected frequency")

        # Today cfg80211 first send AP-CSA-FINISHED and next
        # DFS-PRE-CAC-EXPIRED
        ev = hapd.wait_event(["DFS-PRE-CAC-EXPIRED"], timeout=3)
        if not ev and country == 'US':
            raise Exception("US - no CAC-EXPIRED event")

	# Back again to DFS channel (CAC passed)
        if "OK" not in hapd.request("CHAN_SWITCH 5 5260 ht"):
            raise Exception("CHAN_SWITCH 5260 failed")

        if country == 'US':
            # For non EU we should start CAC again
            ev = wait_dfs_event(hapd, "DFS-CAC-START", 5)
            if not ev:
                raise Exception("No DFS CAC start event")
        else:
            # For EU preCAC should be used
            ev = wait_dfs_event(hapd, "AP-CSA-FINISHED", 5)
            if not ev:
                raise Exception("No CSA finished event - 5260")
    finally:
        clear_regdom(hapd, dev)

@long_duration_test
def test_dfs_eu_chan_switch_precac(dev, apdev):
    """DFS channel switch pre CAC - ETSI domain"""
    dfs_chan_switch_precac(dev, apdev, 'PL')

@long_duration_test
def test_dfs_us_chan_switch_precac(dev, apdev):
    """DFS channel switch pre CAC - FCC domain"""
    dfs_chan_switch_precac(dev, apdev, 'US')