#
# HWSIM generic netlink controller code
# Copyright (c) 2014	Intel Corporation
#
# Author: Johannes Berg <johannes.berg@intel.com>
#
# This software may be distributed under the terms of the BSD license.
# See README for more details.

import netlink

# constants
HWSIM_CMD_CREATE_RADIO		= 4
HWSIM_CMD_DESTROY_RADIO		= 5

HWSIM_ATTR_ADDR_TRANSMITTER     = 2
HWSIM_ATTR_CHANNELS		= 9
HWSIM_ATTR_RADIO_ID		= 10
HWSIM_ATTR_USE_CHANCTX		= 15
HWSIM_ATTR_RADIO_NAME		= 17
HWSIM_ATTR_NO_VDEV		= 18 # Create hwsim radio w/out auto-creating a vdev.

# the controller class
class HWSimController(object):
    def __init__(self):
        self._conn = netlink.Connection(netlink.NETLINK_GENERIC)
        self._fid = netlink.genl_controller.get_family_id('MAC80211_HWSIM')

    def create_radio(self, n_channels=None, use_chanctx=False, hwname=None, no_vdev=False):
        attrs = []
        if n_channels:
            attrs.append(netlink.U32Attr(HWSIM_ATTR_CHANNELS, n_channels))
        if use_chanctx:
            attrs.append(netlink.FlagAttr(HWSIM_ATTR_USE_CHANCTX))
        if hwname:
            attrs.append(netlink.NulStrAttr(HWSIM_ATTR_RADIO_NAME, hwname))
        if no_vdev:
            attrs.append(netlink.FlagAttr(HWSIM_ATTR_NO_VDEV))

        msg = netlink.GenlMessage(self._fid, HWSIM_CMD_CREATE_RADIO,
                                  flags = netlink.NLM_F_REQUEST |
                                          netlink.NLM_F_ACK,
                                  attrs = attrs)
        return msg.send_and_recv(self._conn).ret

    def destroy_radio(self, radio_id):
        try:
            int(radio_id)
            attrs = [netlink.U32Attr(HWSIM_ATTR_RADIO_ID, int(radio_id))]
        except ValueError:
            attrs = [netlink.NulStrAttr(HWSIM_ATTR_RADIO_NAME, radio_id)]
        msg = netlink.GenlMessage(self._fid, HWSIM_CMD_DESTROY_RADIO,
                                  flags = netlink.NLM_F_REQUEST |
                                          netlink.NLM_F_ACK,
                                  attrs = attrs)
        msg.send_and_recv(self._conn)

def create(args):
    print 'Created radio %d' % c.create_radio(n_channels=args.channels,
                                              use_chanctx=args.chanctx,
                                              hwname=args.radio_name,
                                              no_vdev=args.no_vdev)

def destroy(args):
    print c.destroy_radio(args.radio)

if __name__ == '__main__':
    import argparse
    c = HWSimController()

    parser = argparse.ArgumentParser(description='send hwsim control commands')
    subparsers = parser.add_subparsers(help="Commands", dest='command')
    parser_create = subparsers.add_parser('create', help='create a radio')
    parser_create.add_argument('--channels', metavar='<number_of_channels>', type=int,
                               default=0,
                               help='Number of concurrent channels supported ' +
                               'by the radio. If not specified, the number ' +
                               'of channels specified in the ' +
                               'mac80211_hwsim.channels module parameter is ' +
                               'used')
    parser_create.add_argument('--chanctx', action="store_true",
                               help='Use channel contexts, regardless of ' +
                               'whether the number of channels is 1 or ' +
                               'greater. By default channel contexts are ' +
                               'only used if the number of channels is ' +
                               'greater than 1.')
    parser_create.add_argument('--radio_name', metavar='<radio-name>', type=str,
                               default="",
                               help='Optional radio name to request for the new ' +
                               'radio.')
    parser_create.add_argument('--no_vdev', action="store_true",
                               help='Ask kernel to NOT create wlanX interfaces. ' +
                               'User could create them manually, with specific name '
                               'for easier scripting and automated testing.')
    parser_create.set_defaults(func=create)

    parser_destroy = subparsers.add_parser('destroy', help='destroy a radio')
    parser_destroy.add_argument('radio', metavar='<radio>', type=str,
                                default="0",
                                help='The number or name of the radio to be ' +
                                'destroyed (i.e., 0 for phy0, 1 for phy1...)')
    parser_destroy.set_defaults(func=destroy)

    args = parser.parse_args()
    args.func(args)
