--- /dev/null
+#
+# 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, os
+
+# constants
+HWSIM_CMD_CREATE_RADIO = 4
+HWSIM_CMD_DESTROY_RADIO = 5
+
+HWSIM_ATTR_CHANNELS = 9
+HWSIM_ATTR_RADIO_ID = 10
+HWSIM_ATTR_SUPPORT_P2P_DEVICE = 14
+HWSIM_ATTR_USE_CHANCTX = 15
+
+# 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,
+ use_p2p_device=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 use_p2p_device:
+ attrs.append(netlink.FlagAttr(HWSIM_ATTR_SUPPORT_P2P_DEVICE))
+
+ 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):
+ attrs = [netlink.U32Attr(HWSIM_ATTR_RADIO_ID, 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)
+
+class HWSimRadio(object):
+ def __init__(self, n_channels=None, use_chanctx=False,
+ use_p2p_device=False):
+ self._controller = HWSimController()
+ self._n_channels = n_channels
+ self._use_chanctx = use_chanctx
+ self._use_p2p_dev = use_p2p_device
+
+ def __enter__(self):
+ self._radio_id = self._controller.create_radio(
+ n_channels=self._n_channels,
+ use_chanctx=self._use_chanctx,
+ use_p2p_device=self._use_p2p_dev)
+ if self._radio_id < 0:
+ raise Exception("Failed to create radio (err:%d)" % self._radio_id)
+ try:
+ iface = os.listdir('/sys/class/mac80211_hwsim/hwsim%d/net/' % self._radio_id)[0]
+ except Exception,e:
+ self._controller.destroy_radio(self._radio_id)
+ raise e
+ return self._radio_id, iface
+
+ def __exit__(self, type, value, traceback):
+ self._controller.destroy_radio(self._radio_id)
+
+
+def create(args):
+ print 'Created radio %d' % c.create_radio(n_channels=args.channels,
+ use_chanctx=args.chanctx)
+
+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.set_defaults(func=create)
+
+ parser_destroy = subparsers.add_parser('destroy', help='destroy a radio')
+ parser_destroy.add_argument('radio', metavar='<radio>', type=int,
+ default=0,
+ help='The number 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)