323 lines
13 KiB
Python
323 lines
13 KiB
Python
# kojid plugin
|
|
|
|
import commands
|
|
import koji
|
|
import ConfigParser
|
|
import os
|
|
import platform
|
|
compat_mode = False
|
|
try:
|
|
import koji.tasks as tasks
|
|
from koji.tasks import scan_mounts
|
|
from koji.util import isSuccess as _isSuccess
|
|
from koji.util import parseStatus as _parseStatus
|
|
from koji.daemon import log_output
|
|
from __main__ import BuildRoot
|
|
except ImportError:
|
|
compat_mode = True
|
|
#old way
|
|
import tasks
|
|
#XXX - stuff we need from kojid
|
|
from __main__ import BuildRoot, log_output, scan_mounts, _isSuccess, _parseStatus
|
|
|
|
|
|
__all__ = ('RunRootTask',)
|
|
|
|
CONFIG_FILE = '/etc/kojid/runroot.conf'
|
|
|
|
|
|
class RunRootTask(tasks.BaseTaskHandler):
|
|
|
|
Methods = ['runroot']
|
|
|
|
_taskWeight = 2.0
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
self._read_config()
|
|
return super(RunRootTask, self).__init__(*args, **kwargs)
|
|
|
|
def _get_path_params(self, path, rw=False):
|
|
found = False
|
|
for mount_data in self.config['paths']:
|
|
if path.startswith(mount_data['mountpoint']):
|
|
found = True
|
|
break
|
|
if not found:
|
|
raise koji.GenericError("bad config: missing corresponding mountpoint")
|
|
options = []
|
|
for o in mount_data['options'].split(','):
|
|
if rw and o == 'ro':
|
|
options.append('rw')
|
|
else:
|
|
options.append(o)
|
|
rel_path = path[len(mount_data['mountpoint']):]
|
|
res = (os.path.join(mount_data['path'], rel_path), path, mount_data['fstype'], ','.join(options))
|
|
return res
|
|
|
|
def _read_config(self):
|
|
cp = ConfigParser.SafeConfigParser()
|
|
cp.read(CONFIG_FILE)
|
|
self.config = {
|
|
'default_mounts': [],
|
|
'safe_roots': [],
|
|
'path_subs': [],
|
|
'paths': [],
|
|
}
|
|
|
|
if cp.has_option('paths', 'default_mounts'):
|
|
self.config['default_mounts'] = cp.get('paths', 'default_mounts').split(',')
|
|
if cp.has_option('paths', 'safe_roots'):
|
|
self.config['safe_roots'] = cp.get('paths', 'safe_roots').split(',')
|
|
if cp.has_option('paths', 'path_subs'):
|
|
self.config['path_subs'] = [x.split(',') for x in cp.get('paths', 'path_subs').split('\n')]
|
|
|
|
count = 0
|
|
while True:
|
|
section_name = 'path%d' % count
|
|
if not cp.has_section(section_name):
|
|
break
|
|
try:
|
|
self.config['paths'].append({
|
|
'mountpoint': cp.get(section_name, 'mountpoint'),
|
|
'path': cp.get(section_name, 'path'),
|
|
'fstype': cp.get(section_name, 'fstype'),
|
|
'options': cp.get(section_name, 'options'),
|
|
})
|
|
except ConfigParser.NoOptionError:
|
|
raise koji.GenericError("bad config: missing options in %s section" % section_name)
|
|
count += 1
|
|
|
|
for path in self.config['default_mounts'] + self.config['safe_roots'] + [x[0] for x in self.config['path_subs']]:
|
|
if not path.startswith('/'):
|
|
raise koji.GenericError("bad config: all paths (default_mounts, safe_roots, path_subs) needs to be absolute: %s" % path)
|
|
|
|
def handler(self, root, arch, command, keep=False, packages=[], mounts=[], repo_id=None, skip_setarch=False, weight=None, upload_logs=None):
|
|
"""Create a buildroot and run a command (as root) inside of it
|
|
|
|
Command may be a string or a list.
|
|
|
|
Returns a message indicating success if the command was successful, and
|
|
raises an error otherwise. Command output will be available in
|
|
runroot.log in the task output directory on the hub.
|
|
|
|
skip_setarch is a rough approximation of an old hack
|
|
|
|
the keep option is not used. keeping for compatibility for now...
|
|
|
|
upload_logs is list of absolute paths which will be uploaded for
|
|
archiving on hub. It always consists of /tmp/runroot.log, but can be
|
|
used for additional logs (pungi.log, etc.)
|
|
"""
|
|
if weight is not None:
|
|
weight = max(weight, 0.5)
|
|
self.session.host.setTaskWeight(self.id, weight)
|
|
#noarch is funny
|
|
if arch == "noarch":
|
|
#we need a buildroot arch. Pick one that:
|
|
# a) this host can handle
|
|
# b) the build tag can support
|
|
# c) is canonical
|
|
host_arches = self.session.host.getHost()['arches']
|
|
if not host_arches:
|
|
raise koji.BuildError, "No arch list for this host"
|
|
tag_arches = self.session.getBuildConfig(root)['arches']
|
|
if not tag_arches:
|
|
raise koji.BuildError, "No arch list for tag: %s" % root
|
|
#index canonical host arches
|
|
host_arches = dict([(koji.canonArch(a),1) for a in host_arches.split()])
|
|
#pick the first suitable match from tag's archlist
|
|
for br_arch in tag_arches.split():
|
|
br_arch = koji.canonArch(br_arch)
|
|
if host_arches.has_key(br_arch):
|
|
#we're done
|
|
break
|
|
else:
|
|
#no overlap
|
|
raise koji.BuildError, "host does not match tag arches: %s (%s)" % (root, tag_arches)
|
|
else:
|
|
br_arch = arch
|
|
if repo_id:
|
|
repo_info = self.session.repoInfo(repo_id, strict=True)
|
|
if repo_info['tag_name'] != root:
|
|
raise koji.BuildError, "build tag (%s) does not match repo tag (%s)" % (root, repo_info['tag_name'])
|
|
if repo_info['state'] not in (koji.REPO_STATES['READY'], koji.REPO_STATES['EXPIRED']):
|
|
raise koji.BuildError, "repos in the %s state may not be used by runroot" % koji.REPO_STATES[repo_info['state']]
|
|
else:
|
|
repo_info = self.session.getRepo(root)
|
|
if not repo_info:
|
|
#wait for it
|
|
task_id = self.session.host.subtask(method='waitrepo',
|
|
arglist=[root, None, None],
|
|
parent=self.id)
|
|
repo_info = self.wait(task_id)[task_id]
|
|
if compat_mode:
|
|
broot = BuildRoot(root, br_arch, self.id, repo_id=repo_info['id'])
|
|
else:
|
|
broot = BuildRoot(self.session, self.options, root, br_arch, self.id, repo_id=repo_info['id'])
|
|
broot.workdir = self.workdir
|
|
broot.init()
|
|
rootdir = broot.rootdir()
|
|
#workaround for rpm oddness
|
|
os.system('rm -f "%s"/var/lib/rpm/__db.*' % rootdir)
|
|
#update buildroot state (so that updateBuildRootList() will work)
|
|
self.session.host.setBuildRootState(broot.id, 'BUILDING')
|
|
try:
|
|
if packages:
|
|
#pkglog = '%s/%s' % (broot.resultdir(), 'packages.log')
|
|
pkgcmd = ['--install'] + packages
|
|
status = broot.mock(pkgcmd)
|
|
self.session.host.updateBuildRootList(broot.id, broot.getPackageList())
|
|
if not _isSuccess(status):
|
|
raise koji.BuildrootError, _parseStatus(status, pkgcmd)
|
|
|
|
if isinstance(command, str):
|
|
cmdstr = command
|
|
else:
|
|
#we were passed an arglist
|
|
#we still have to run this through the shell (for redirection)
|
|
#but we can preserve the list structure precisely with careful escaping
|
|
cmdstr = ' '.join(["'%s'" % arg.replace("'", r"'\''") for arg in command])
|
|
# A nasty hack to put command output into its own file until mock can be
|
|
# patched to do something more reasonable than stuff everything into build.log
|
|
cmdargs = ['/bin/sh', '-c', "{ %s; } < /dev/null 2>&1 | /usr/bin/tee /tmp/runroot.log; exit ${PIPESTATUS[0]}" % cmdstr]
|
|
|
|
# always mount /mnt/redhat (read-only)
|
|
# always mount /mnt/iso (read-only)
|
|
# also need /dev bind mount
|
|
self.do_mounts(rootdir, [self._get_path_params(x) for x in self.config['default_mounts']])
|
|
self.do_extra_mounts(rootdir, mounts)
|
|
mock_cmd = ['chroot']
|
|
if skip_setarch:
|
|
#we can't really skip it, but we can set it to the current one instead of of the chroot one
|
|
myarch = platform.uname()[5]
|
|
mock_cmd.extend(['--arch', myarch])
|
|
mock_cmd.append('--')
|
|
mock_cmd.extend(cmdargs)
|
|
rv = broot.mock(mock_cmd)
|
|
log_paths = ['/tmp/runroot.log']
|
|
if upload_logs is not None:
|
|
log_paths += upload_logs
|
|
for log_path in log_paths:
|
|
self.uploadFile(rootdir + log_path)
|
|
finally:
|
|
# mock should umount its mounts, but it will not handle ours
|
|
self.undo_mounts(rootdir, fatal=False)
|
|
broot.expire()
|
|
if isinstance(command, str):
|
|
cmdlist = command.split()
|
|
else:
|
|
cmdlist = command
|
|
cmdlist = [param for param in cmdlist if '=' not in param]
|
|
if cmdlist:
|
|
cmd = os.path.basename(cmdlist[0])
|
|
else:
|
|
cmd = '(none)'
|
|
if _isSuccess(rv):
|
|
return '%s completed successfully' % cmd
|
|
else:
|
|
raise koji.BuildrootError, _parseStatus(rv, cmd)
|
|
|
|
def do_extra_mounts(self, rootdir, mounts):
|
|
mnts = []
|
|
for mount in mounts:
|
|
mount = os.path.normpath(mount)
|
|
for safe_root in self.config['safe_roots']:
|
|
if mount.startswith(safe_root):
|
|
break
|
|
else:
|
|
#no match
|
|
raise koji.GenericError("read-write mount point is not safe: %s" % mount)
|
|
#normpath should have removed any .. dirs, but just in case...
|
|
if mount.find('/../') != -1:
|
|
raise koji.GenericError("read-write mount point is not safe: %s" % mount)
|
|
|
|
for re, sub in self.config['path_subs']:
|
|
mount = mount.replace(re, sub)
|
|
|
|
mnts.append(self._get_path_params(mount, rw=True))
|
|
self.do_mounts(rootdir, mnts)
|
|
|
|
def do_mounts(self, rootdir, mounts):
|
|
if not mounts:
|
|
return
|
|
self.logger.info('New runroot')
|
|
self.logger.info("Runroot mounts: %s" % mounts)
|
|
fn = '%s/tmp/runroot_mounts' % rootdir
|
|
fslog = file(fn, 'a')
|
|
logfile = "%s/do_mounts.log" % self.workdir
|
|
uploadpath = self.getUploadDir()
|
|
error = None
|
|
for dev,path,type,opts in mounts:
|
|
if not path.startswith('/'):
|
|
raise koji.GenericError("invalid mount point: %s" % path)
|
|
mpoint = "%s%s" % (rootdir,path)
|
|
if opts is None:
|
|
opts = []
|
|
else:
|
|
opts = opts.split(',')
|
|
if 'bind' in opts:
|
|
#make sure dir exists
|
|
if not os.path.isdir(dev):
|
|
error = koji.GenericError("No such directory or mount: %s" % dev)
|
|
break
|
|
type = 'none'
|
|
if path is None:
|
|
#shorthand for "same path"
|
|
path = dev
|
|
if 'bg' in opts:
|
|
error = koji.GenericError("bad config: background mount not allowed")
|
|
break
|
|
opts = ','.join(opts)
|
|
cmd = ['mount', '-t', type, '-o', opts, dev, mpoint]
|
|
self.logger.info("Mount command: %r" % cmd)
|
|
koji.ensuredir(mpoint)
|
|
if compat_mode:
|
|
status = log_output(cmd[0], cmd, logfile, uploadpath, logerror=True, append=True)
|
|
else:
|
|
status = log_output(self.session, cmd[0], cmd, logfile, uploadpath, logerror=True, append=True)
|
|
if not _isSuccess(status):
|
|
error = koji.GenericError("Unable to mount %s: %s" \
|
|
% (mpoint, _parseStatus(status, cmd)))
|
|
break
|
|
fslog.write("%s\n" % mpoint)
|
|
fslog.flush()
|
|
fslog.close()
|
|
if error is not None:
|
|
self.undo_mounts(rootdir, fatal=False)
|
|
raise error
|
|
|
|
def undo_mounts(self, rootdir, fatal=True):
|
|
self.logger.debug("Unmounting runroot mounts")
|
|
mounts = {}
|
|
fn = '%s/tmp/runroot_mounts' % rootdir
|
|
if os.path.exists(fn):
|
|
fslog = file(fn,'r')
|
|
for line in fslog:
|
|
mounts.setdefault(line.strip(), 1)
|
|
fslog.close()
|
|
#also, check /proc/mounts just in case
|
|
for dir in scan_mounts(rootdir):
|
|
mounts.setdefault(dir, 1)
|
|
mounts = mounts.keys()
|
|
# deeper directories first
|
|
mounts.sort()
|
|
mounts.reverse()
|
|
failed = []
|
|
self.logger.info("Unmounting (runroot): %s" % mounts)
|
|
for dir in mounts:
|
|
(rv, output) = commands.getstatusoutput("umount -l '%s'" % dir)
|
|
if rv != 0:
|
|
failed.append("%s: %s" % (dir, output))
|
|
if failed:
|
|
msg = "Unable to unmount: %s" % ', '.join(failed)
|
|
self.logger.warn(msg)
|
|
if fatal:
|
|
raise koji.GenericError, msg
|
|
else:
|
|
# remove the mount list when everything is unmounted
|
|
try:
|
|
os.unlink(fn)
|
|
except OSError:
|
|
pass
|
|
|
|
|