#!/usr/bin/python # This program is free software; you can redistribute it and/or modify # it under the terms of the GNU General Public License as published by # the Free Software Foundation; either version 2 of the License, or # (at your option) any later version. # # This program is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU Library General Public License for more details. # # You should have received a copy of the GNU General Public License # along with this program; if not, write to the Free Software # Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. # Copyright 2002 Duke University import os import clientStuff import rpm import string import sys import archwork import rpmUtils import types from i18n import _ class nevral: def __init__(self): self.rpmbyname = {} self.rpmbynamearch = {} self.localrpmpath = {} self.localhdrpath = {} def add(self,(name,epoch,ver,rel,arch,rpmloc,serverid),state): # if self.rpmbyname.haskey(name): # ((e,v,r,a,l,i),state) = self._get_data(name) # goodarch = clientStuff.betterarch(arch,a) # if goodarch != None: # if goodarch == arch: # self.rpmbyname[name]=((epoch,ver,rel,arch,rpmloc,serverid), state) self.rpmbyname[name]=((epoch,ver,rel,arch,rpmloc,serverid), state) self.rpmbynamearch[(name,arch)]=((epoch,ver,rel,arch,rpmloc,serverid), state) def delete(self, name, arch=None): if self.exists(name, arch): if self.rpmbyname.has_key(name): del self.rpmbyname[name] if self.rpmbynamearch.has_key((name, arch)): del self.rpmbynamearch[(name, arch)] else: if arch is None: errolog(2, _('No Package %s') % (name)) else: errorlog(2, _('No Package %s, %s') %(name, arch)) def _get_data(self, name, arch=None): if arch != None: # search by name and arch if self.rpmbynamearch and self.rpmbynamearch.has_key((name, arch)): return self.rpmbynamearch[(name, arch)] else: return ((None,None,None,None,None,None),None) else: # search by name only if self.rpmbyname and self.rpmbyname.has_key(name): return self.rpmbyname[name] else: return ((None,None,None,None,None,None),None) def getHeader(self, name, arch=None): # FIXME - this should return a header object from rpmUtils ((e,v,r,a,l,i),state) = self._get_data(name, arch) if state == None: errorlog(0, _('Header for pkg %s not found') % (name)) #FIXME: should return none and/or raise Exception sys.exit(1) return None else: if l == 'in_rpm_db': # we're in the rpmdb - get the header from there hindexes = ts.dbMatch('name', name) for hdr in hindexes: return hdr else: # we're in a .hdr file pkghdr = clientStuff.readHeader(self.localHdrPath(name, arch)) if pkghdr == None: errorlog(0, _('Bad Header for pkg %s.%s trying to get headers for the nevral - exiting') % (name, arch)) # FIXME - should raise exception and be handled elsewhere sys.exit(1) else: return pkghdr def NAkeys(self): keys = self.rpmbynamearch.keys() return keys def Nkeys(self): keys = self.rpmbyname.keys() return keys def hdrfn(self, name, arch=None): ((e,v,r,a,l,i),state) = self._get_data(name, arch) if state == None: return None else: return '%s-%s-%s-%s.%s.hdr' % (name, e, v, r, a) def rpmlocation(self, name, arch=None): ((e,v,r,a,l,i),state) = self._get_data(name, arch) if state == None: return None else: return l def serverid(self, name, arch=None): ((e,v,r,a,l,i),state) = self._get_data(name, arch) if state == None: return None else: return i def evr(self, name, arch=None): ((e,v,r,a,l,i),state) = self._get_data(name, arch) if state == None: return (None, None, None) else: return (e, v, r) def exists(self, name, arch=None): ((e,v,r,a,l,i),state) = self._get_data(name, arch) if state == None: return 0 else: return 1 def state(self, name, arch=None): ((e,v,r,a,l,i),state) = self._get_data(name, arch) if state == None: return None else: return state def nafromloc(self, loc): keys = self.rpmbynamearch.keys() for (name, arch) in keys: ((e,v,r,a,l,i),state) = self._get_data(name, arch) if state == None: return None else: if l == loc: return (name,arch) def remoteHdrUrl(self, name, arch=None): ((e,v,r,a,l,i),state) = self._get_data(name, arch) if state == None: return None if l == 'in_rpm_db': return l hdrfn = self.hdrfn(name,arch) base = conf.baseURL(i) return base + '/headers/' + hdrfn def localHdrPath(self, name, arch=None): ((e,v,r,a,l,i),state) = self._get_data(name, arch) if state == None: return None if l == 'in_rpm_db': return l if self.localhdrpath.has_key((name, arch)): return self.localhdrpath[(name, arch)] else: hdrfn = self.hdrfn(name,arch) if conf.serverhdrdir.has_key(i): base = conf.serverhdrdir[i] else: errorlog(0, _('asking for package %s.%s - does not exist in nevral - bailing out - check rpmdb for errors') % (name, arch)) sys.exit(1) log(7, 'localhdrpath= %s for %s %s' % (base + '/' + hdrfn, name, arch)) return base + '/' + hdrfn def setlocalhdrpath(self, name, arch, path): ((e,v,r,a,l,i),state) = self._get_data(name, arch) self.localhdrpath[(name, arch)] = path def remoteRpmUrl(self, name, arch=None): ((e,v,r,a,l,i),state) = self._get_data(name, arch) if state == None: return None if l == 'in_rpm_db': return l base = conf.baseURL(i) return base +'/'+ l def localRpmPath(self, name, arch=None): ((e,v,r,a,l,i),state) = self._get_data(name, arch) if state == None: return None if l == 'in_rpm_db': return l if self.localrpmpath.has_key((name, arch)): return self.localrpmpath[(name, arch)] else: rpmfn = os.path.basename(l) base = conf.serverpkgdir[i] return base + '/' + rpmfn def setlocalrpmpath(self, name, arch, path): ((e,v,r,a,l,i),state) = self._get_data(name, arch) self.localrpmpath[(name, arch)] = path def setPkgState(self, name, arch, newstate): ((e,v,r,a,l,i),state) = self._get_data(name, arch) self.add((name,e,v,r,arch,l,i), newstate) def bestArchsByVersion(self, name): """returns a list of archs that have the highest version for name""" returnarchs = [] archs = archwork.availablearchs(self, name) currentarch = archs[0] for arch in archs[1:]: rc = rpmUtils.compareEVR(self.evr(name, currentarch), self.evr(name, arch)) if rc < 0: currentarch = arch elif rc == 0: pass elif rc > 0: pass (best_e, best_v, best_r) = self.evr(name, currentarch) log(3, _('Best version for %s is %s:%s-%s') % (name, best_e, best_v, best_r)) for arch in archs: rc = rpmUtils.compareEVR(self.evr(name, arch), (best_e, best_v, best_r)) if rc == 0: returnarchs.append(arch) elif rc > 0: log(4, 'What the hell, we just determined it was the bestversion') log(7, returnarchs) return returnarchs def populateTs(self, addavailable = 1, localrpm=0): if localrpm: rpmlocationMethod = self.localRpmPath else: rpmlocationMethod = self.rpmlocation _ts = rpmUtils.Rpm_Ts_Work(conf.installroot) for (name, arch) in self.NAkeys(): if self.state(name, arch) in ['u','ud','iu']: log(4,'Updating: %s, %s' % (name, arch)) rpmloc = rpmlocationMethod(name, arch) pkghdr = self.getHeader(name, arch) provides = rpmUtils.getProvides(pkghdr) if name in conf.installonlypkgs or 'kernel-modules' in provides: bestarchlist = self.bestArchsByVersion(name) bestarch = archwork.bestarch(bestarchlist) if arch == bestarch: log(3, _('Found best arch for install only pkg %s') %(arch)) _ts.addInstall(pkghdr,(pkghdr,rpmloc),'i') self.setPkgState(name, arch, 'i') else: log(3, ('Removing dumb arch for install only pkg: %s') %(arch)) if addavailable: _ts.addInstall(pkghdr,(pkghdr,rpmloc),'a') self.setPkgState(name, arch, 'a') else: log(5, 'Not an install only pkg, adding to ts') _ts.addInstall(pkghdr,(pkghdr,rpmloc),'u') elif self.state(name,arch) == 'i': log(4, 'Installing: %s.%s' % (name, arch)) rpmloc = rpmlocationMethod(name, arch) pkghdr = self.getHeader(name, arch) _ts.addInstall(pkghdr,(pkghdr,rpmloc),'i') elif self.state(name,arch) == 'a': if addavailable: log(7, 'Adding %s into \'a\' state' % name) rpmloc = rpmlocationMethod(name, arch) pkghdr = self.getHeader(name, arch) _ts.addInstall(pkghdr,(pkghdr,rpmloc),'a') else: pass elif self.state(name,arch) == 'e' or self.state(name,arch) == 'ed': log(4, 'Erasing: %s.%s' % (name,arch)) mi = _ts.dbMatch(rpm.RPMTAG_NAME, name) mi.pattern(rpm.RPMTAG_ARCH, rpm.RPMMIRE_DEFAULT, arch) for idx in mi: instance = mi.instance() _ts.addErase(instance) return _ts def resolvedeps(self, rpmDBInfo): #self == tsnevral #populate ts #depcheck #parse deps, if they exist, change nevral pkg states #die if: # no suggestions # conflicts #return 0 and a message if all is fine #return 1 and a list of error messages if shit breaks CheckDeps = 1 conflicts = 0 unresolvable = 0 # this does a quick dep check with adding all the archs # keeps mem usage small in the easy/quick case _ts = self.populateTs(addavailable = 0) deps = _ts.check() if not deps: log(5, 'Quick Check only') return (0, 'Success - deps resolved') del deps del _ts log(5, 'Long Check') depscopy = [] unresolveableloop = 0 while CheckDeps==1 or conflicts != 1 or unresolvable != 1: errors=[] _ts = self.populateTs(addavailable = 1) deps = _ts.check() if deps == depscopy: unresolveableloop = unresolveableloop + 1 log(5, 'looping count = %d' % unresolveableloop) if unresolveableloop >= 3: errors.append('Unable to satisfy dependencies') for ((name, version, release), (reqname, reqversion), flags, suggest, sense) in deps: errors.append('Package %s needs %s, this is not available.' % (name, rpmUtils.formatRequire(reqname, reqversion, flags))) else: unresolveableloop = 0 depscopy = deps CheckDeps = 0 if not deps: return (0, 'Success - deps resolved') log (3, '# of Deps = %d' % len(deps)) for ((name, version, release), (reqname, reqversion), flags, suggest, sense) in deps: log (4, 'debug dep: %s req %s - %s - %s' % (name, reqname, reqversion, sense)) if sense == rpm.RPMDEP_SENSE_REQUIRES: if suggest: (header, sugname) = suggest log(4, '%s wants %s' % (name, sugname)) (name, arch) = self.nafromloc(sugname) archlist = self.bestArchsByVersion(name) bestarch = archwork.bestarch(archlist) pkghdr = self.getHeader(name, bestarch) provides = rpmUtils.getProvides(pkghdr) if reqname in provides: log(3, _('bestarch = %s for %s') % (bestarch, name)) else: log(3, _('bestarch %s does not provide resetting to arch %s') % (bestarch, arch)) bestarch = arch self.setPkgState(name, bestarch, 'ud') log(4, 'Got dep: %s, %s' % (name,bestarch)) CheckDeps = 1 else: log(5, 'No suggestion for %s needing %s' % (name, reqname)) if self.exists(reqname): if self.state(reqname) in ('e', 'ed'): # this is probably an erase depedency archlist = archwork.availablearchs(rpmDBInfo,name) arch = archwork.bestarch(archlist) ((e, v, r, a, l, i), s)=rpmDBInfo._get_data(name,arch) self.add((name,e,v,r,a,l,i),'ed') log(4, 'Got Erase Dep: %s, %s' %(name,a)) else: archlist = self.bestArchsByVersion(reqname) if len(archlist) > 0: arch = archwork.bestarch(archlist) if self.state(reqname, arch) not in ['ud','u','i']: self.setPkgState(reqname, arch, 'ud') log(4, 'Got Extra Dep: %s, %s' %(reqname,arch)) else: log(4, '%s already to be installed/upgraded, trying to upgrade the requiring pkg' % (reqname)) if self.exists(name): archlist = self.bestArchsByVersion(name) if len(archlist) > 0: arch = archwork.bestarch(archlist) self.setPkgState(name, arch, 'ud') log(4, 'Upgrading %s, %s' % (name, arch)) else: unresolvable = 1 log(4, 'unresolvable - %s needs %s' % (name, rpmUtils.formatRequire(reqname, reqversion, flags))) if clientStuff.nameInExcludes(reqname): errors.append('Package %s needs %s that has been excluded.' % (name, reqname)) else: errors.append('Package %s needs %s, this is not available.)' % (name, rpmUtils.formatRequire(reqname, reqversion, flags))) CheckDeps=1 else: # this is horribly ugly but I have to find some way to see if what it needed is provided # by what we are removing - if it is then remove it -otherwise its a real dep problem - move along if reqname[0] == '/': whatprovides = _ts.dbMatch('basenames', reqname) else: whatprovides = _ts.dbMatch('provides', reqname) if whatprovides and whatprovides.count() != 0: log(5, 'Found some provides for %s' % (reqname)) for provhdr in whatprovides: if self.state(provhdr[rpm.RPMTAG_NAME],provhdr[rpm.RPMTAG_ARCH]) in ['e','ed']: ((e,v,r,a,l,i),s)=rpmDBInfo._get_data(name) self.add((name,e,v,r,a,l,i),'ed') log(4, 'Got Erase Dep: %s, %s' %(name, a)) CheckDeps=1 else: # help us obi-wan - you're our only hope! log(4, 'Cannot find a resolution attempting update out of loop on %s' % (name)) if self.exists(name): archlist = self.bestArchsByVersion(name) if len(archlist) > 0: arch = archwork.bestarch(archlist) self.setPkgState(name, arch, 'ud') log(4, 'Upgrading %s, %s' % (name, arch)) else: # it's as if a thousand dependencies cried out and were suddenly silenced! unresolvable = 1 if clientStuff.nameInExcludes(reqname): errors.append('Package %s needs %s that has been excluded.' % (name, reqname)) else: errors.append('Package %s needs %s, this is not available.' % (name, rpmUtils.formatRequire(reqname, reqversion, flags))) else: # help us obi-wan - you're our only hope! log(4, 'Cannot find a resolution attempting update out of loop on %s' % (name)) if self.exists(name): archlist = self.bestArchsByVersion(name) if len(archlist) > 0: arch = archwork.bestarch(archlist) self.setPkgState(name, arch, 'ud') log(4, 'Upgrading %s, %s' % (name, arch)) else: # it's as if a thousand dependencies cried out and were suddenly silenced! unresolvable = 1 if clientStuff.nameInExcludes(reqname): errors.append('Package %s needs %s that has been excluded.' % (name, reqname)) else: errors.append('Package %s needs %s, this is not available.' % (name, rpmUtils.formatRequire(reqname, reqversion, flags))) elif sense == rpm.RPMDEP_SENSE_CONFLICTS: # much more shit should happen here. specifically: # if you have a conflict b/t two pkgs, try to upgrade the reqname pkg. - see if that solves the problem # also something like a "this isn't our fault and we can't help it, continue on" should happen.like in anaconda # more even than the below should happen here - but its getting closer - I need to flesh out all the horrible # states it could be in. log(4, 'conflict: %s %s %s' % (name, reqname, reqversion)) if rpmDBInfo.exists(reqname) and self.exists(reqname) and self.state(reqname) not in ('i','iu','u','ud'): archlist = archwork.availablearchs(rpmDBInfo,reqname) arch = archwork.bestarch(archlist) (e1, v1, r1) = rpmDBInfo.evr(reqname,arch) (e2, v2, r2) = self.evr(reqname,arch) rc = rpmUtils.compareEVR((e1,v1,r1), (e2,v2,r2)) if rc<0: log(4, 'conflict: setting %s to upgrade' % (reqname)) self.setPkgState(reqname, arch, 'ud') CheckDeps=1 else: errors.append('conflict between %s and %s' % (name, reqname)) conflicts=1 else: errors.append('conflict between %s and %s' % (name, reqname)) conflicts=1 log(4, 'Restarting Dependency Loop') log.write(2, '.') sys.stdout.flush() del _ts del deps if len(errors) > 0: return(1, errors)