d3fd188841
Added all functionality from upd-instroot and scrubtree. Rewrote most of the old utils/ scripts.
162 lines
4.4 KiB
Python
162 lines
4.4 KiB
Python
import sys
|
|
import os
|
|
import commands
|
|
import re
|
|
import socket
|
|
|
|
import logging
|
|
logging.basicConfig(level=logging.DEBUG)
|
|
logger = logging.getLogger('pylorax')
|
|
|
|
|
|
def genmodinfo(path=None):
|
|
if not path:
|
|
release = os.uname()[2]
|
|
path = os.path.join('/lib/modules', release)
|
|
|
|
mods = {}
|
|
for root, dirs, files in os.walk(path):
|
|
for file in files:
|
|
mods[file] = os.path.join(root, file)
|
|
|
|
modules = { 'scsi_hostadapter': ['block'], 'eth': ['networking'] }
|
|
blacklist = ('floppy', 'scsi_mod', 'libiscsi')
|
|
|
|
list = {}
|
|
for modtype in modules:
|
|
list[modtype] = {}
|
|
for file in modules[modtype]:
|
|
try:
|
|
filename = os.path.join(path, 'modules.%s' % file)
|
|
f = open(filename, 'r')
|
|
except IOError:
|
|
logger.error('cannot open file %s', filename)
|
|
continue
|
|
else:
|
|
lines = f.readlines()
|
|
f.close()
|
|
|
|
for line in lines:
|
|
line = line.strip()
|
|
if line in mods:
|
|
modname, ext = os.path.splitext(line)
|
|
if modname in blacklist:
|
|
logger.info('skipping %s', modname)
|
|
continue
|
|
|
|
outtext = commands.getoutput('modinfo -F description %s' % mods[line])
|
|
desc = outtext.split('\n')[0]
|
|
desc = desc.strip()
|
|
|
|
# XXX why we need to do this?
|
|
desc = desc[:65]
|
|
|
|
if not desc:
|
|
desc = '%s driver' % modname
|
|
modinfo = '%s\n\t%s\n\t"%s"\n' % (modname, modtype, desc)
|
|
list[modtype][modname] = modinfo
|
|
|
|
# XXX i don't think we want to print it out now
|
|
print 'Version 0'
|
|
for type in list:
|
|
modlist = list[type].keys()
|
|
modlist.sort()
|
|
for m in modlist:
|
|
print list[type][m]
|
|
|
|
|
|
# XXX is the input a path to a file?
|
|
def filtermoddeps(input):
|
|
try:
|
|
f = open(input, 'r')
|
|
except IOError:
|
|
logger.error('cannot open file %s', input)
|
|
return
|
|
|
|
# concatenate lines ending with \
|
|
lines = []
|
|
for line in f.readlines():
|
|
if line.endswith('\\\n'):
|
|
line = line[:-2]
|
|
lines.append(line)
|
|
|
|
f.close()
|
|
|
|
lines = ''.join(lines).split('\n')
|
|
|
|
# XXX what's the purpose of these gentlemen?
|
|
lines = filter(lambda line: re.search(':.*ko', line), lines)
|
|
lines = map(lambda line: re.sub('\.ko', '', line), lines)
|
|
lines = map(lambda line: re.sub('/[^: ]*/', '', line), lines)
|
|
lines = map(lambda line: re.sub('\t+', ' ', line), lines)
|
|
|
|
return lines
|
|
|
|
|
|
# XXX what are the inputs for this thing?
|
|
def trimpciids(pciids, *paths):
|
|
vendors = []
|
|
devices = []
|
|
|
|
for file in paths:
|
|
try:
|
|
f = open(file, 'r')
|
|
except IOError:
|
|
logger.error('cannot open file %s', file)
|
|
continue
|
|
else:
|
|
pcitable = f.readlines()
|
|
f.close()
|
|
|
|
for line in pcitable:
|
|
if line.startswith('alias pci:'):
|
|
vend = '0x%s' % line[15:19]
|
|
dev = '0x%s' % line[24:28]
|
|
elif line.startswith('alias pcivideo:'):
|
|
vend = '0x%s' % line[20:24]
|
|
dev = '0x%s' % line[29:33]
|
|
else:
|
|
continue
|
|
|
|
vend = vend.upper()
|
|
dev = dev.upper()
|
|
if vend not in vendors:
|
|
vendors.append(vend)
|
|
if (vend, dev) not in devices:
|
|
devices.append((vend, dev))
|
|
|
|
current_vend = 0
|
|
for line in pciids:
|
|
if line.startswith('#') or line.startswith('\t\t') or line == '\n':
|
|
continue
|
|
|
|
# XXX print or return?
|
|
if not line.startswith('\t'):
|
|
current_vend = '0x%s' % line.split()[0]
|
|
current_vend = current_vend.upper()
|
|
if current_vend in vendors:
|
|
print line,
|
|
continue
|
|
|
|
# XXX print or return?
|
|
dev = '0x%s' % line.split()[0]
|
|
dev = dev.upper()
|
|
if (current_vend, dev) in devices:
|
|
print line,
|
|
|
|
|
|
# XXX what is this whole thing for?
|
|
def geninitrdsz(size, filename):
|
|
size = socket.htonl(size)
|
|
|
|
try:
|
|
f = open(filename, 'w')
|
|
except IOError:
|
|
logger.error('cannot open file %s', filename)
|
|
else:
|
|
f.write(str(size))
|
|
f.close()
|
|
|
|
# XXX is this needed? default mode is 0664
|
|
os.chmod(filename, 0644)
|