X-Git-Url: http://git.droids-corp.org/?a=blobdiff_plain;f=tools%2Fdpdk_nic_bind.py;h=14c5311061549ac66383a3022a88396b1a5481e3;hb=948fd64befc3;hp=2483056caeb5a3d0545b4495cf97865f1e0638b0;hpb=1dee76c032906d3937db264341a1495714eee78d;p=dpdk.git diff --git a/tools/dpdk_nic_bind.py b/tools/dpdk_nic_bind.py index 2483056cae..14c5311061 100755 --- a/tools/dpdk_nic_bind.py +++ b/tools/dpdk_nic_bind.py @@ -54,7 +54,7 @@ args = [] def usage(): '''Print usage information for the program''' argv0 = basename(sys.argv[0]) - print """ + print (""" Usage: ------ @@ -68,7 +68,7 @@ Options: --help, --usage: Display usage information and quit - --status: + -s, --status: Print the current status of all known network interfaces. For each device, it displays the PCI domain, bus, slot and function, along with a text description of the device. Depending upon whether the @@ -110,7 +110,7 @@ To unbind 0000:01:00.0 from using any driver To bind 0000:02:00.0 and 0000:02:00.1 to the ixgbe kernel driver %(argv0)s -b ixgbe 02:00.0 02:00.1 - """ % locals() # replace items from local variables + """ % locals()) # replace items from local variables # This is roughly compatible with check_output function in subprocess module # which is only available in python 2.7. @@ -156,27 +156,40 @@ def check_modules(): '''Checks that igb_uio is loaded''' global dpdk_drivers - fd = file("/proc/modules") - loaded_mods = fd.readlines() - fd.close() - # list of supported modules mods = [{"Name" : driver, "Found" : False} for driver in dpdk_drivers] # first check if module is loaded - for line in loaded_mods: + try: + # Get list of sysfs modules (both built-in and dynamically loaded) + sysfs_path = '/sys/module/' + + # Get the list of directories in sysfs_path + sysfs_mods = [os.path.join(sysfs_path, o) for o + in os.listdir(sysfs_path) + if os.path.isdir(os.path.join(sysfs_path, o))] + + # Extract the last element of '/sys/module/abc' in the array + sysfs_mods = [a.split('/')[-1] for a in sysfs_mods] + + # special case for vfio_pci (module is named vfio-pci, + # but its .ko is named vfio_pci) + sysfs_mods = map(lambda a: + a if a != 'vfio_pci' else 'vfio-pci', sysfs_mods) + for mod in mods: - if line.startswith(mod["Name"]): - mod["Found"] = True - # special case for vfio_pci (module is named vfio-pci, - # but its .ko is named vfio_pci) - elif line.replace("_", "-").startswith(mod["Name"]): + if mod["Name"] in sysfs_mods: mod["Found"] = True + except: + pass # check if we have at least one loaded module if True not in [mod["Found"] for mod in mods] and b_flag is not None: - print "Error - no supported modules are loaded" - sys.exit(1) + 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"]] @@ -195,7 +208,7 @@ def get_pci_device_details(dev_id): for line in extra_info: if len(line) == 0: continue - name, value = line.split("\t", 1) + name, value = line.decode().split("\t", 1) name = name.strip(":") + "_str" device[name] = value # check for a unix interface name @@ -231,7 +244,7 @@ def get_nic_details(): dev["Device"] = int(dev["Device"],16) devices[dev["Slot"]] = dict(dev) # use dict to make copy of dev else: - name, value = dev_line.split("\t", 1) + name, value = dev_line.decode().split("\t", 1) dev[name.rstrip(":")] = value # check what is the interface if any for an ssh connection if @@ -240,17 +253,17 @@ def get_nic_details(): route = check_output(["ip", "-o", "route"]) # filter out all lines for 169.254 routes route = "\n".join(filter(lambda ln: not ln.startswith("169.254"), - route.splitlines())) + route.decode().splitlines())) rt_info = route.split() - for i in xrange(len(rt_info) - 1): + for i in range(len(rt_info) - 1): if rt_info[i] == "dev": ssh_if.append(rt_info[i+1]) # based on the basic info, get extended text details for d in devices.keys(): # get additional info and add it to existing data - devices[d] = dict(devices[d].items() + - get_pci_device_details(d).items()) + devices[d] = devices[d].copy() + devices[d].update(get_pci_device_details(d).items()) for _if in ssh_if: if _if in devices[d]["Interface"].split(","): @@ -290,22 +303,22 @@ def dev_id_from_dev_name(dev_name): if dev_name in devices[d]["Interface"].split(","): return devices[d]["Slot"] # if nothing else matches - error - print "Unknown device: %s. " \ - "Please specify device in \"bus:slot.func\" format" % dev_name + print ("Unknown device: %s. " \ + "Please specify device in \"bus:slot.func\" format" % dev_name) sys.exit(1) def unbind_one(dev_id, force): '''Unbind the device identified by "dev_id" from its current driver''' dev = devices[dev_id] if not has_driver(dev_id): - print "%s %s %s is not currently managed by any driver\n" % \ - (dev["Slot"], dev["Device_str"], dev["Interface"]) + print ("%s %s %s is not currently managed by any driver\n" % \ + (dev["Slot"], dev["Device_str"], dev["Interface"])) return # prevent us disconnecting ourselves if dev["Ssh_if"] and not force: - print "Routing table indicates that interface %s is active" \ - ". Skipping unbind" % (dev_id) + print ("Routing table indicates that interface %s is active" \ + ". Skipping unbind" % (dev_id)) return # write to /sys to unbind @@ -313,8 +326,8 @@ def unbind_one(dev_id, force): try: f = open(filename, "a") except: - print "Error: unbind failed for %s - Cannot open %s" % (dev_id, filename) - sys/exit(1) + print ("Error: unbind failed for %s - Cannot open %s" % (dev_id, filename)) + sys.exit(1) f.write(dev_id) f.close() @@ -326,14 +339,14 @@ def bind_one(dev_id, driver, force): # prevent disconnection of our ssh session if dev["Ssh_if"] and not force: - print "Routing table indicates that interface %s is active" \ - ". Not modifying" % (dev_id) + print ("Routing table indicates that interface %s is active" \ + ". Not modifying" % (dev_id)) return # unbind any existing drivers we don't want if has_driver(dev_id): if dev["Driver_str"] == driver: - print "%s already bound to driver %s, skipping\n" % (dev_id, driver) + print ("%s already bound to driver %s, skipping\n" % (dev_id, driver)) return else: saved_driver = dev["Driver_str"] @@ -346,14 +359,14 @@ def bind_one(dev_id, driver, force): try: f = open(filename, "w") except: - print "Error: bind failed for %s - Cannot open %s" % (dev_id, filename) + print ("Error: bind failed for %s - Cannot open %s" % (dev_id, filename)) return try: f.write("%04x %04x" % (dev["Vendor"], dev["Device"])) f.close() except: - print "Error: bind failed for %s - Cannot write new PCI ID to " \ - "driver %s" % (dev_id, driver) + print ("Error: bind failed for %s - Cannot write new PCI ID to " \ + "driver %s" % (dev_id, driver)) return # do the bind by writing to /sys @@ -361,7 +374,7 @@ def bind_one(dev_id, driver, force): try: f = open(filename, "a") except: - print "Error: bind failed for %s - Cannot open %s" % (dev_id, filename) + print ("Error: bind failed for %s - Cannot open %s" % (dev_id, filename)) if saved_driver is not None: # restore any previous driver bind_one(dev_id, saved_driver, force) return @@ -375,7 +388,7 @@ def bind_one(dev_id, driver, force): tmp = get_pci_device_details(dev_id) if "Driver_str" in tmp and tmp["Driver_str"] == driver: return - print "Error: bind failed for %s - Cannot bind to driver %s" % (dev_id, driver) + print ("Error: bind failed for %s - Cannot bind to driver %s" % (dev_id, driver)) if saved_driver is not None: # restore any previous driver bind_one(dev_id, saved_driver, force) return @@ -388,7 +401,7 @@ def unbind_all(dev_list, force=False): unbind_one(d, force) def bind_all(dev_list, driver, force=False): - """Unbind method, takes a list of device locations""" + """Bind method, takes a list of device locations""" global devices dev_list = map(dev_id_from_dev_name, dev_list) @@ -420,8 +433,8 @@ def display_devices(title, dev_list, extra_params = None): %()s fields in it for replacement by the named fields in each device's dictionary.''' strings = [] # this holds the strings to print. We sort before printing - print "\n%s" % title - print "="*len(title) + print ("\n%s" % title) + print ("="*len(title)) if len(dev_list) == 0: strings.append("") else: @@ -433,7 +446,7 @@ def display_devices(title, dev_list, extra_params = None): strings.append("%s '%s'" % (dev["Slot"], dev["Device_str"])) # sort before printing, so that the entries appear in PCI order strings.sort() - print "\n".join(strings) # print one per line + print ("\n".join(strings)) # print one per line def show_status(): '''Function called when the script is passed the "--status" option. Displays @@ -474,25 +487,25 @@ def parse_args(): sys.exit(0) try: - opts, args = getopt.getopt(sys.argv[1:], "b:u", + opts, args = getopt.getopt(sys.argv[1:], "b:us", ["help", "usage", "status", "force", "bind=", "unbind"]) - except getopt.GetoptError, error: - print str(error) - print "Run '%s --usage' for further information" % sys.argv[0] + except (getopt.GetoptError, error): + print (str(error)) + print ("Run '%s --usage' for further information" % sys.argv[0]) sys.exit(1) for opt, arg in opts: if opt == "--help" or opt == "--usage": usage() sys.exit(0) - if opt == "--status": + if opt == "--status" or opt == "-s": status_flag = True if opt == "--force": force_flag = True if opt == "-b" or opt == "-u" or opt == "--bind" or opt == "--unbind": if b_flag is not None: - print "Error - Only one bind or unbind may be specified\n" + print ("Error - Only one bind or unbind may be specified\n") sys.exit(1) if opt == "-u" or opt == "--unbind": b_flag = "none" @@ -507,13 +520,13 @@ def do_arg_actions(): 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] + print ("Error: No action specified for devices. Please give a -b or -u option") + print ("Run '%s --usage' for further information" % sys.argv[0]) sys.exit(1) if b_flag is not None and len(args) == 0: - print "Error: No devices specified." - print "Run '%s --usage' for further information" % sys.argv[0] + print ("Error: No devices specified.") + print ("Run '%s --usage' for further information" % sys.argv[0]) sys.exit(1) if b_flag == "none" or b_flag == "None":