mirror of
https://github.com/vanhoefm/fragattacks.git
synced 2024-12-01 11:48:23 -05:00
8a5cc2fe38
The new HWSimRadio context manager allows the following
syntax to create (and appropriately destroy) a new radio:
with HWSimRadio([...]) as (radio_id, iface_name):
[...]
Signed-off-by: Johannes Berg <johannes.berg@intel.com>
115 lines
4.5 KiB
Python
115 lines
4.5 KiB
Python
#
|
|
# 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)
|