# Each device within this is itself a dictionary of device properties
devices = {}
# list of supported DPDK drivers
-dpdk_drivers = [ "igb_uio", "vfio-pci" ]
+dpdk_drivers = [ "igb_uio", "vfio-pci", "uio_pci_generic" ]
+
+# command-line arg flags
+b_flag = None
+status_flag = False
+force_flag = False
+args = []
def usage():
'''Print usage information for the program'''
mod["Found"] = True
# check if we have at least one loaded module
- if True not in [mod["Found"] for mod in mods]:
- print "Error - no supported modules are loaded"
- sys.exit(1)
+ if True not in [mod["Found"] for mod in mods] and b_flag is not None:
+ if b_flag in dpdk_drivers:
+ print "Error - no supported modules(DPDK driver) are loaded"
+ sys.exit(1)
+ else:
+ print "Warning - no supported modules(DPDK driver) are loaded"
# change DPDK driver list to only contain drivers that are loaded
dpdk_drivers = [mod["Name"] for mod in mods if mod["Found"]]
def bind_all(dev_list, driver, force=False):
"""Unbind method, takes a list of device locations"""
+ global devices
+
dev_list = map(dev_id_from_dev_name, dev_list)
+
for d in dev_list:
bind_one(d, driver, force)
+ # when binding devices to a generic driver (i.e. one that doesn't have a
+ # PCI ID table), some devices that are not bound to any other driver could
+ # be bound even if no one has asked them to. hence, we check the list of
+ # drivers again, and see if some of the previously-unbound devices were
+ # erroneously bound.
+ for d in devices.keys():
+ # skip devices that were already bound or that we know should be bound
+ if "Driver_str" in devices[d] or d in dev_list:
+ continue
+
+ # update information about this device
+ devices[d] = dict(devices[d].items() +
+ get_pci_device_details(d).items())
+
+ # check if updated information indicates that the device was bound
+ if "Driver_str" in devices[d]:
+ unbind_one(d, force)
+
def display_devices(title, dev_list, extra_params = None):
'''Displays to the user the details of a list of devices given in "dev_list"
The "extra_params" parameter, if given, should contain a string with
def parse_args():
'''Parses the command-line arguments given by the user and takes the
appropriate action for each'''
- b_flag = None
- status_flag = False
- force_flag = False
+ global b_flag
+ global status_flag
+ global force_flag
+ global args
if len(sys.argv) <= 1:
usage()
sys.exit(0)
else:
b_flag = arg
+def do_arg_actions():
+ '''do the actual action requested by the user'''
+ global b_flag
+ global status_flag
+ global force_flag
+ global args
+
if b_flag is None and not status_flag:
print "Error: No action specified for devices. Please give a -b or -u option"
print "Run '%s --usage' for further information" % sys.argv[0]
def main():
'''program main function'''
+ parse_args()
check_modules()
get_nic_details()
- parse_args()
+ do_arg_actions()
if __name__ == "__main__":
main()