This commit was generated by cvs2svn to compensate for changes in r525,

[bootcd.git] / scripts / rewrite-pcitable.py

diff --git a/scripts/rewrite-pcitable.py b/scripts/rewrite-pcitable.py
new file mode 100755 (executable)
index 0000000..c2dadb5
--- /dev/null
+++ b/scripts/rewrite-pcitable.py
@@ -0,0 +1,112 @@
+#!/usr/bin/env python
+
+"""
+The point of this small utility is to take a file in the format
+of /lib/modules/`uname -r`/modules.pcimap and output a condensed, more
+easily used format for module detection
+
+The output is used by the PlanetLab boot cd (3.0+) and the pl_hwinit script
+to load all the applicable modules by scanning lspci output.
+
+Excepted format of file includes lines of:
+
+# pci_module vendor device subvendor subdevice class class_mask driver_data
+cciss 0x00000e11 0x0000b060 0x00000e11 0x00004070 0x00000000 0x00000000 0x0
+cciss 0x00000e11 0x0000b178 0x00000e11 0x00004070 0x00000000 0x00000000 0x0
+
+Output format, for each line that matches the above lines:
+cciss 0e11:b060 0e11:b178
+"""
+
+import os, sys
+import string
+
+
+def usage():
+    print( "Usage:" )
+    print( "rewrite-pcitable.py <pcitable> [<output>]" )
+    print( "" )
+
+
+if len(sys.argv) < 2:
+    usage()
+    sys.exit(1)
+
+
+pcitable_file_name= sys.argv[1]
+try:
+    pcitable_file= file(pcitable_file_name,"r")
+except IOError:
+    sys.stderr.write( "Unable to open: %s\n" % pcitable_file_name )
+    sys.exit(1)
+
+if len(sys.argv) > 2:
+    output_file_name= sys.argv[2]
+    try:
+        output_file= file(output_file_name,"w")
+    except IOError:
+        sys.stderr.write( "Unable to open %s for writing.\n" % output_file )
+        sys.exit(1)
+else:
+    output_file= sys.stdout
+
+
+line_num= 0
+
+# associative array to store all matches of module -> ['vendor:device',..]
+# entries
+all_modules= {}
+
+for line in pcitable_file:
+    line_num= line_num+1
+    
+    # skip blank lines, or lines that begin with # (comments)
+    line= string.strip(line)
+    if len(line) == 0:
+        continue
+    
+    if line[0] == "#":
+        continue
+
+    line_parts= string.split(line)
+    if line_parts is None or len(line_parts) != 8:
+        sys.stderr.write( "Skipping line %d (incorrect format)\n" % line_num )
+        continue
+
+    # first two parts are always vendor / device id
+    module= line_parts[0]
+    vendor_id= line_parts[1]
+    device_id= line_parts[2]
+    
+
+    # valid vendor and devices are 10 chars (0xXXXXXXXX) and begin with 0x
+    if len(vendor_id) != 10 or len(device_id) != 10:
+        sys.stderr.write( "Skipping line %d (invalid vendor/device id length)\n"
+                          % line_num )
+        continue
+
+    if string.lower(vendor_id[:2]) != "0x" \
+           or string.lower(device_id[:2]) != "0x":
+        sys.stderr.write( "Skipping line %d (invalid vendor/device id format)\n"
+                          % line_num )
+        continue
+
+    # cut down the ids, only need last 4 bytes
+    # start at 6 = (10 total chars - 4 last chars need)
+    vendor_id= string.lower(vendor_id[6:])
+    device_id= string.lower(device_id[6:])
+
+    full_id= "%s:%s" % (vendor_id, device_id)
+
+    if all_modules.has_key(module):
+        all_modules[module].append( full_id )
+    else:
+        all_modules[module]= [full_id,]
+
+for module in all_modules.keys():
+    devices= string.join( all_modules[module], " " )
+    output_file.write( "%s %s\n" % (module,devices) )
+
+    
+output_file.close()
+pcitable_file.close()