# Copyright (c) 2012 The Chromium OS Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

from autotest_lib.server import site_linux_cros_router

class LinuxVMRouter(site_linux_cros_router.LinuxCrosRouter):
    """
    Linux/mac80211-style WiFi Router support for WiFiTest class.

    As compared to LinuxCrosRouter, LinuxVMRouter is specialized for
    routers using the virtual radios provided by mac80211_hwsim.
    """

    def __init__(self, host, params, defssid):
        host.run("rmmod mac80211_hwsim", ignore_status=True)
        host.run("modprobe mac80211_hwsim radios=3")

        site_linux_cros_router.LinuxCrosRouter.__init__(
            self, host, params, defssid)

        # Override LinuxSystem's phy assignment.
        self.phys_for_frequency = {}
        phy_infos = host.run("%s list" % self.cmd_iw).stdout.splitlines()
        phy_list = \
            [phy_info.split()[1] for phy_info in phy_infos \
               if phy_info.startswith('Wiphy')]

        # In the VM case, the client runs on the same node as the router.
        # Since LinuxRouter.__init__ removes all interfaces, we need to
        # set up a device for the client. [quiche.20110823]
        client_phy = phy_list[0]
        host.run("%s phy %s interface add client0 type managed" %
                 (self.cmd_iw, client_phy))

        # Both remaining virtual devices are dual-band. Arbitrarily
        # assign one to 2.4 GHz, and the other to 5 GHz.
        self.phydev2 = self.phydev2 or phy_list[1]
        self.phydev5 = self.phydev5 or phy_list[2]

    def _pre_start_hook(self, config):
        # Override default LinuxCrosRouter behavior of stopping supplicant
        # instance because the client is running on the same machine as the
        # router.
        pass