[ SEA-GHOST MINI SHELL]
#!/usr/bin/python -tt
# 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 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., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
## (c) 2008 Red Hat. Written by skvidal@fedoraproject.org
## james@fedoraproject.org
import os
import sys
import gzip
import tempfile
from optparse import OptionParser
import yum
import rpmUtils.miscutils
sections = ['%%%%SYSTEM INFO\n', '%%%%YUM INFO\n',
'%%%%RPMDB PROBLEMS\n', '%%%%RPMDB\n',
'%%%%REPOS\n']
def cmd_line():
parser = OptionParser()
parser.set_usage("yum-debug-restore [options]")
parser.add_option("-C", "--cache", action="store_true",
help="run from cache only")
parser.add_option("-c", dest="conffile", help="config file location")
parser.add_option("--enablerepo", action="append", dest="enablerepos",
help="specify additional repoids to query, can be specified multiple times")
parser.add_option("--disablerepo", action="append", dest="disablerepos",
help="specify repoids to disable, can be specified multiple times")
parser.add_option("-y", dest="assumeyes", action="store_true",
help="answer yes for all questions")
parser.add_option("--skip-broken", action="store_true",
help="skip packages with depsolving problems")
parser.add_option("--output", action="store_true",
help="output the yum shell commands")
parser.add_option("--shell",
help="output the yum shell commands to a file")
parser.add_option("--install-latest", action="store_true",
help="install the latest instead of specific versions")
parser.add_option("--ignore-arch", action="store_true",
help="ignore arch of packages, so you can dump on .i386 and restore on .x86_64")
parser.add_option("--filter-types",
help="Limit to: install, remove, update, downgrade")
(opts, args) = parser.parse_args()
if not args:
parser.print_usage()
sys.exit(1)
return (opts, args)
class OtherRpmDB:
def __init__(self, fn):
self.pkgtups = []
if fn.endswith(".gz"):
fo = gzip.GzipFile(fn)
else:
fo = open(fn)
if fo.readline() != 'yum-debug-dump version 1\n':
print >>sys.stderr, "Bad yum debug file:", fn
sys.exit(1)
skip = sections[:-1]
for line in fo:
if skip: # Header stuff
if line == skip[0]:
skip.pop(0)
continue
if not line or line[0] != ' ':
break
pkgtup = rpmUtils.miscutils.splitFilename(line.strip())
n,v,r,e,a = pkgtup # grrr...
pkgtup = (n,a,e,v,r)
self.pkgtups.append(pkgtup)
def naevr2str(n,a,e,v,r):
if a is None: # Assume epoch doesn't change without release changing
return "%s-%s-%s" % (n,v,r)
if e in (None, '', '0'):
return "%s-%s-%s.%s" % (n,v,r,a)
return "%s-%s:%s-%s.%s" % (n,e,v,r,a)
def pkgtup2str(pkgtup):
n,a,e,v,r = pkgtup
return naevr2str(n,a,e,v,r)
def pkg_data2list(yb, opkgtups, opkgmaps, install_latest, ignore_arch):
ret = []
npkgtups = set()
npkgmaps = {}
installonly = set(yb.conf.installonlypkgs)
for po in sorted(yb.rpmdb.returnPackages()):
arch = po.arch
if ignore_arch:
arch = None
if False: pass
elif po.name in installonly:
if not po.pkgtup in opkgtups:
ret.append(("remove", pkgtup2str(po.pkgtup)))
elif (po.name, arch) not in opkgmaps:
ret.append(("remove", str(po)))
elif po.pkgtup not in opkgtups:
n,a,e,v,r = opkgmaps[(po.name, arch)]
pinstEVR = yum.packages.PackageEVR(e, v, r)
if po.EVR == pinstEVR:
assert ignore_arch and po.arch != a
elif po.EVR < pinstEVR:
ret.append(("upgrade", naevr2str(n,arch,e,v,r)))
else:
ret.append(("downgrade", naevr2str(n,arch,e,v,r)))
npkgtups.add(po.pkgtup)
npkgmaps[(po.name, po.arch)] = po
if ignore_arch:
npkgmaps[(po.name, None)] = po
for name, arch in sorted(opkgmaps):
if name in installonly:
continue # done separately
if ignore_arch and arch is not None:
continue
if (name, arch) in npkgmaps:
continue
if install_latest and ignore_arch:
ret.append(("install", name))
elif install_latest:
ret.append(("install", "%s.%s" % (name, arch)))
else:
ret.append(("install", pkgtup2str(opkgmaps[(name, arch)])))
for pkgtup in opkgtups:
if pkgtup[0] in installonly and not pkgtup in npkgtups:
ret.append(("install", pkgtup2str(pkgtup)))
return ret
def main():
(opts, args) = cmd_line()
yb = yum.YumBase()
yb.preconf.init_plugins = True
if opts.conffile:
yb.preconf.fn = opts.conffile
yb.conf
if opts.cache:
yb.conf.cache = True
if opts.disablerepos:
for repo_match in opts.disablerepos:
for repo in yb.repos.findRepos(repo_match):
repo.disable()
if opts.enablerepos:
for repo_match in opts.enablerepos:
for repo in yb.repos.findRepos(repo_match):
repo.enable()
xtra_args = []
if opts.skip_broken:
xtra_args.append('--skip-broken')
if opts.assumeyes:
xtra_args.append('-y')
fn = args[0]
print "Reading from: %s" % fn
orpmdb = OtherRpmDB(fn)
opkgmaps = {}
for pkgtup in orpmdb.pkgtups:
opkgmaps[(pkgtup[0], pkgtup[1])] = pkgtup
if opts.ignore_arch:
n,a,e,v,r = pkgtup
opkgmaps[(pkgtup[0], None)] = n,None,e,v,r
if opts.output:
fo = sys.stdout
elif opts.shell:
try:
fo = open(opts.shell, "wb")
except OSError, e:
print >>sys.stderr, "open(%s): %s" % (opts.shell, e)
sys.exit(1)
else:
fo = tempfile.NamedTemporaryFile()
fT = None
if opts.filter_types:
fT = set(opts.filter_types.replace(",", " ").split())
counts = {}
for T, pkg in pkg_data2list(yb, set(orpmdb.pkgtups), opkgmaps,
opts.install_latest, opts.ignore_arch):
if fT is not None and T not in fT:
continue
counts[T] = counts.get(T, 0) + 1
try:
print >>fo, "%-9s %s" % (T, pkg)
except IOError:
if opts.output: # mainly due to |
sys.exit(1)
raise
if opts.output:
sys.exit(0)
print >>fo, "run"
fo.flush()
if opts.shell:
if counts:
print "Statistics:"
for T in sorted(counts):
print " %9s %6u" % (T, counts[T])
print "Done"
sys.exit(0)
# Want to do the transaction, hacky method
xtra_args.append('--setopt=installonly_limit=0')
os.system("yum shell %s %s" % (" ".join(xtra_args), fo.name))
if __name__ == "__main__":
main()
SEA-GHOST - SHELL CODING BY SEA-GHOST