[ 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.
import sys
sys.path.insert(0,'/usr/share/yum-cli')
import yum
from yum.misc import setup_locale
from utils import YumUtilBase
import logging
import os
import os.path
try:
from yum.misc import find_unfinished_transactions, find_ts_remaining
except ImportError:
import glob
def find_unfinished_transactions(yumlibpath='/var/lib/yum'):
"""returns a list of the timestamps from the filenames of the unfinished
transactions remaining in the yumlibpath specified.
"""
timestamps = []
tsallg = '%s/%s' % (yumlibpath, 'transaction-all*')
#tsdoneg = '%s/%s' % (yumlibpath, 'transaction-done*') # not used remove ?
tsalls = glob.glob(tsallg)
#tsdones = glob.glob(tsdoneg) # not used remove ?
for fn in tsalls:
trans = os.path.basename(fn)
timestamp = trans.replace('transaction-all.','')
timestamps.append(timestamp)
timestamps.sort()
return timestamps
def find_ts_remaining(timestamp, yumlibpath='/var/lib/yum'):
"""this function takes the timestamp of the transaction to look at and
the path to the yum lib dir (defaults to /var/lib/yum)
returns a list of tuples(action, pkgspec) for the unfinished transaction
elements. Returns an empty list if none.
"""
to_complete_items = []
tsallpath = '%s/%s.%s' % (yumlibpath, 'transaction-all', timestamp)
tsdonepath = '%s/%s.%s' % (yumlibpath,'transaction-done', timestamp)
tsdone_items = []
if not os.path.exists(tsallpath):
# something is wrong, here, probably need to raise _something_
return to_complete_items
if os.path.exists(tsdonepath):
tsdone_fo = open(tsdonepath, 'r')
tsdone_items = tsdone_fo.readlines()
tsdone_fo.close()
tsall_fo = open(tsallpath, 'r')
tsall_items = tsall_fo.readlines()
tsall_fo.close()
for item in tsdone_items:
# this probably shouldn't happen but it's worth catching anyway
if item not in tsall_items:
continue
tsall_items.remove(item)
for item in tsall_items:
item = item.replace('\n', '')
if item == '':
continue
(action, pkgspec) = item.split()
to_complete_items.append((action, pkgspec))
return to_complete_items
class YumCompleteTransaction(YumUtilBase):
NAME = 'yum-complete-transaction'
VERSION = '1.0'
USAGE = """
yum-complete-transaction: completes unfinished yum transactions which occur due to error, failure
or act of $deity
usage: yum-complete-transaction
"""
def __init__(self):
YumUtilBase.__init__(self,
YumCompleteTransaction.NAME,
YumCompleteTransaction.VERSION,
YumCompleteTransaction.USAGE)
self.logger = logging.getLogger("yum.verbose.cli.yumcompletets")
# Add util commandline options to the yum-cli ones
self.optparser = self.getOptionParser()
if hasattr(self, 'getOptionGroup'):
self.optparser_grp = self.getOptionGroup()
else:
self.optparser_grp = self.optparser
self.addCmdOptions()
try: self.main()
finally: self.unlock()
def clean_up_ts_files(self, timestamp, path, disable=False):
# clean up the transactions
tsdone = '%s/transaction-done.%s' % (path, timestamp)
tsall = '%s/transaction-all.%s' % (path, timestamp)
results = []
for f in [tsall, tsdone]:
if os.path.exists(f):
if disable:
disable_f = f + '.disabled'
os.rename(f, disable_f)
results.append(disable_f)
else:
os.unlink(f)
return results
def addCmdOptions(self):
self.optparser_grp.add_option("--cleanup-only", default=False,
action="store_true", dest="cleanup",
help='Do not complete the transaction just clean up transaction journals')
def main(self):
# Parse the commandline option and setup the basics.
try:
opts = self.doUtilConfigSetup()
except yum.Errors.RepoError, e:
self.logger.error("Cannot handle specific enablerepo/disablerepo options.")
sys.exit(50)
if self.conf.uid != 0:
self.logger.error("Error: You must be root to run yum-complete-transaction.")
sys.exit(1)
# Setup yum (Ts, RPM db, Repo & Sack)
self.doUtilYumSetup()
# Do the real action
# get the list of transactions remaining
# list the number of them
# take the most recent one
# populate the ts
# run it
self.run_with_package_names.add('yum-utils')
times = []
for thistime in find_unfinished_transactions(self.conf.persistdir):
if thistime.endswith('disabled'):
continue
# XXX maybe a check here for transactions that are just too old to try and complete?
times.append(thistime)
if not times:
print "No unfinished transactions left."
sys.exit()
if opts.cleanup:
print "Cleaning up unfinished transaction journals"
# nuke ts files in yumlibpath
for timestamp in times:
print "Cleaning up %s" % timestamp
self.clean_up_ts_files(timestamp, self.conf.persistdir)
sys.exit()
timestamp = times[-1]
print "There are %d outstanding transactions to complete. Finishing the most recent one" % len(times)
try:
remaining = find_ts_remaining(timestamp, yumlibpath=self.conf.persistdir)
except yum.Errors.MiscError, e:
self.logger.error("Error: %s" % e)
self.logger.error("Please report this error to: %s" % self.conf.bugtracker_url)
sys.exit(1)
print "The remaining transaction had %d elements left to run" % len(remaining)
for (action, pkgspec) in remaining:
if action == 'install':
try:
self.install(pattern=pkgspec)
except yum.Errors.InstallError, e:
pass
if action == 'erase':
(e, m, u) = self.rpmdb.matchPackageNames([pkgspec])
for pkg in e:
self.remove(po=pkg)
current_count = len(self.tsInfo)
if hasattr(self, 'doUtilBuildTransaction'):
errc = self.doUtilBuildTransaction(unfinished_transactions_check=False)
if errc:
sys.exit(errc)
else:
try:
self.buildTransaction(unfinished_transactions_check=False)
except yum.Errors.YumBaseError, e:
self.logger.critical("Error building transaction: %s" % e)
sys.exit(1)
if current_count != len(self.tsInfo):
print '\n\nTransaction size changed - this means we are not doing the\n' \
'same transaction as we were before. Aborting and disabling\n' \
'this transaction.'
print '\nYou could try running: package-cleanup --problems\n' \
' package-cleanup --dupes\n' \
' rpm -Va --nofiles --nodigest'
filelist = self.clean_up_ts_files(timestamp, self.conf.persistdir, disable=True)
print '\nTransaction files renamed to:\n %s' % '\n '.join(filelist)
sys.exit()
if len(self.tsInfo) < 1:
print 'Nothing in the unfinished transaction to cleanup.'
print "Cleaning up completed transaction file"
self.clean_up_ts_files(timestamp, self.conf.persistdir)
sys.exit()
else:
try:
if self.doUtilTransaction() == 0:
print "Cleaning up completed transaction file"
self.clean_up_ts_files(timestamp, self.conf.persistdir)
sys.exit()
else:
print "Not removing old transaction files"
sys.exit()
except yum.Errors.YumBaseError,e:
print "Error: %s" % str(e)
sys.exit(1)
if __name__ == '__main__':
setup_locale()
util = YumCompleteTransaction()
SEA-GHOST - SHELL CODING BY SEA-GHOST