diff options
author | Paul Nasrat <pnasrat@redhat.com> | 2005-11-13 21:44:15 +0000 |
---|---|---|
committer | Paul Nasrat <pnasrat@redhat.com> | 2005-11-13 21:44:15 +0000 |
commit | c5eed570405e7f1a5d9a416e1bde8953fece1850 (patch) | |
tree | 02e52facfc16684b87933771969dd46ffed34c66 | |
parent | 6c086eb5952a0c5e97c6602f3fcc1567b4879c1e (diff) | |
download | anaconda-c5eed570405e7f1a5d9a416e1bde8953fece1850.tar.gz anaconda-c5eed570405e7f1a5d9a416e1bde8953fece1850.tar.xz anaconda-c5eed570405e7f1a5d9a416e1bde8953fece1850.zip |
Generate headers for cd installs
-rw-r--r-- | genheader.py | 137 |
1 files changed, 137 insertions, 0 deletions
diff --git a/genheader.py b/genheader.py new file mode 100644 index 000000000..2797854ac --- /dev/null +++ b/genheader.py @@ -0,0 +1,137 @@ +#!/usr/bin/python + +import os +from struct import * +import rpm + +# Types +RPM_NULL = 0 +RPM_CHAR = 1 +RPM_INT8 = 2 +RPM_INT16 = 3 +RPM_INT32 = 4 +RPM_INT64 = 5 # Unused currently +RPM_STRING = 6 +RPM_BIN = 7 +RPM_STRING_ARRAY = 8 +RPM_I18NSTRING = 9 + +stringTypes = [ RPM_STRING, RPM_STRING_ARRAY, RPM_I18NSTRING ] + +formatTable = { + RPM_CHAR : "!c", + RPM_INT8 : "!B", + RPM_INT16 : "!H", + RPM_INT32 : "!I", + RPM_INT64 : "!Q", + RPM_STRING : "s", + RPM_BIN : "s", + RPM_STRING_ARRAY : "s", + RPM_I18NSTRING : "s" +} + +senseTable = { + "EQ": rpm.RPMSENSE_EQUAL, + "LT": rpm.RPMSENSE_LESS, + "GT": rpm.RPMSENSE_GREATER, + "LE": rpm.RPMSENSE_EQUAL | rpm.RPMSENSE_LESS, + "GE": rpm.RPMSENSE_EQUAL | rpm.RPMSENSE_GREATER +} + +class YumHeader: + def __init__(self,po): + """Partial and dumbed down header generation for cd installation + @param po + @type po: PackageObject""" + self.po = po + self.store = "" + self.offset = 0 + self.indexes = [] + self.tagtbl = { + 'os': (rpm.RPMTAG_OS, RPM_STRING), + 'name': (rpm.RPMTAG_NAME, RPM_STRING), + 'epoch': (rpm.RPMTAG_EPOCH, RPM_INT32), + 'version': (rpm.RPMTAG_VERSION, RPM_STRING), + 'release': (rpm.RPMTAG_RELEASE, RPM_STRING), + 'arch': (rpm.RPMTAG_ARCH, RPM_STRING), + 'providename': (rpm.RPMTAG_PROVIDENAME, RPM_STRING_ARRAY), + 'provideversion': (rpm.RPMTAG_PROVIDEVERSION, RPM_STRING_ARRAY), + 'provideflags': (rpm.RPMTAG_PROVIDEFLAGS, RPM_STRING_ARRAY) + } + if 'epoch' in self.po.simpleItems(): + self.po.simple['epoch'] = int(self.po.simple['epoch']) + + def __format(self, tag, tagtype, value): + if not isinstance(value, (tuple, list)): + return self.__formatSingle(tag, tagtype, value) + else: + count = len(value) + data = "" + for entry in value: + (entrycount, entrydata) = self.__formatSingle(tag, tagtype, value) + data += entrydata + return (count, data) + + def __formatSingle(self, tag, tagtype, value): + format = formatTable[tagtype] + count = 1 + if tagtype == RPM_BIN: + format = "%%d%s" % format + count = len(value) + data = pack(format % (count, value)) + elif tagtype in stringTypes: + # Null terminate + data = pack("%ds" % (len(value) + 1), value) + else: + data = pack(format, value) + return (count, data) + + def __alignTag(self, tagtype): + """Return alignment data for aligning for ttype from offset + self.offset.""" + if tagtype == RPM_INT16: + align = (2 - (self.offset % 2)) % 2 + elif tagtype == RPM_INT32: + align = (4 - (self.offset % 4)) % 4 + elif tagtype == RPM_INT64: + align = (8 - (self.offset % 8)) % 8 + else: + align = 0 + return '\x00' * align + + def convertTag(self, tag): + if self.tagtbl.has_key(tag): + (rpmtag, tagtype) = self.tagtbl[tag] + (count, data) = self.__format(rpmtag, tagtype, self.po.returnSimple(tag)) + pad = self.__alignTag(tagtype) + self.offset += len(pad) + self.indexes.append((rpmtag, tagtype,self.offset, count)) + self.store += pad + data + self.offset += len(data) + + def generateProvides(self): + self.po.simple['provideversion'] = [ "%s-%s" % (self.po.returnSimple('version'), self.po.returnSimple('release')) ] + self.po.simple['providename'] = [self.po.returnSimple['name']] + self.po.simple['provideversion'] = [ "%s-%s" % (self.po.returnSimple('version'), self.po.returnSimple('release')) ] + self.po.returnSimple['provideflags'] = [senseTable["EQ"]] + self.convertTag('providename') + self.convertTag('provideversion') + self.convertTag('provideflags') + self.convertTag('provideflags') + + def str(self): + self.po.simple['os'] = 'linux' + self.convertTag('os') + for tag in ['name','version', 'release', 'arch', 'epoch']: + if tag in self.po.simpleItems(): + self.convertTag(tag) + + magic = '\x8e\xad\xe8' + hdr_start_fmt= '!3sB4xii' + index_fmt = '!4I' + version = 1 + hdr = pack(hdr_start_fmt, magic, version, len(self.indexes), len(self.store)) + for (tag, tagtype, offset, count) in self.indexes: + hdr += pack(index_fmt, tag, tagtype, offset, count) + hdr += self.store + return hdr |