hwsim tests: Add support for the chanctx flag when creating radios

Add a new option to set the use_chanctx flag when creating a radio.
While at it, refactor the arguments parsing code to use argparse.

Signed-hostap: Luciano Coelho <luciano.coelho@intel.com>

tests/hwsim/hwsim.py

@@ -15,6 +15,7 @@ HWSIM_CMD_DESTROY_RADIO		= 5
 
 HWSIM_ATTR_CHANNELS		= 9
 HWSIM_ATTR_RADIO_ID		= 10
+HWSIM_ATTR_USE_CHANCTX		= 15
 
 # the controller class
 class HWSimController(object):
@@ -22,10 +23,13 @@ class HWSimController(object):
         self._conn = netlink.Connection(netlink.NETLINK_GENERIC)
         self._fid = netlink.genl_controller.get_family_id('MAC80211_HWSIM')
 
-    def create_radio(self, n_channels=None):
+    def create_radio(self, n_channels=None, use_chanctx=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))
+
         msg = netlink.GenlMessage(self._fid, HWSIM_CMD_CREATE_RADIO,
                                   flags = netlink.NLM_F_REQUEST |
                                           netlink.NLM_F_ACK,
@@ -40,14 +44,41 @@ class HWSimController(object):
                                   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)
+
+def destroy(args):
+    print c.destroy_radio(args.radio)
+
 if __name__ == '__main__':
-    import sys
+    import argparse
     c = HWSimController()
-    if sys.argv[1] == 'create':
+
-        if len(sys.argv) > 2:
+    parser = argparse.ArgumentParser(description='send hwsim control commands')
-            n_channels = int(sys.argv[2])
+    subparsers = parser.add_subparsers(help="Commands", dest='command')
-        else:
+    parser_create = subparsers.add_parser('create', help='create a radio')
-            n_channels = 0
+    parser_create.add_argument('--channels', metavar='<number_of_channels>', type=int,
-        print 'Created radio %d' % c.create_radio(n_channels=n_channels)
+                               default=0,
-    elif sys.argv[1] == 'destroy':
+                               help='Number of concurrent channels supported ' +
-        print c.destroy_radio(int(sys.argv[2]))
+                               '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)

tests/hwsim/netlink.py

@@ -69,6 +69,10 @@ class U8Attr(Attr):
     def __init__(self, attr_type, val):
         Attr.__init__(self, attr_type, "B", val)
 
+class FlagAttr(Attr):
+    def __init__(self, attr_type):
+        Attr.__init__(self, attr_type, "")
+
 class Nested(Attr):
     def __init__(self, attr_type, attrs):
         self.attrs = attrs