Source code for lago.sysprep

# Copyright 2015 Red Hat, Inc.
# 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
# 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
# Refer to the README and COPYING files for full details of the license
import os

import utils
from textwrap import dedent
import tempfile
import logging
LOGGER = logging.getLogger(__name__)
_DOT_SSH = '/root/.ssh'
_AUTHORIZED_KEYS = os.path.join(_DOT_SSH, 'authorized_keys')
_SELINUX_CONF_DIR = '/etc/selinux'
_SELINUX_CONF_PATH = os.path.join(_SELINUX_CONF_DIR, 'config')
_ISCSI_DIR = '/etc/iscsi'

[docs]def set_hostname(hostname): return ('--hostname', hostname)
[docs]def set_root_password(password): return ('--root-password', 'password:%s' % password)
[docs]def _write_file(path, content): return ('--write', '%s:%s' % (path, content))
[docs]def _upload_file(local_path, remote_path): return ('--upload', '%s:%s' % (remote_path, local_path))
[docs]def set_iscsi_initiator_name(name): return ('--mkdir', _ISCSI_DIR, '--chmod', '0755:%s' % _ISCSI_DIR, ) + _write_file( os.path.join(_ISCSI_DIR, 'initiatorname.iscsi'), 'InitiatorName=%s' % name, ) # noqa: E126
[docs]def add_ssh_key(key, with_restorecon_fix=False): extra_options = ('--mkdir', _DOT_SSH, '--chmod', '0700:%s' % _DOT_SSH, ) + _upload_file(_AUTHORIZED_KEYS, key) if (not os.stat(key).st_uid == 0 or not os.stat(key).st_gid == 0): extra_options += ( '--run-command', 'chown root.root %s' % _AUTHORIZED_KEYS, ) if with_restorecon_fix: # Fix for fc23 not relabeling on boot # extra_options += ('--firstboot-command', 'restorecon -R /root/.ssh', ) return extra_options
[docs]def set_selinux_mode(mode): return ( '--mkdir', _SELINUX_CONF_DIR, '--chmod', '0755:%s' % _SELINUX_CONF_DIR, ) + _write_file( _SELINUX_CONF_PATH, ('SELINUX=%s\n' 'SELINUXTYPE=targeted\n') % mode, )
[docs]def _config_net_interface(path, iface, type, bootproto, onboot, hwaddr): iface_path = os.path.join(path, 'ifcfg-{0}'.format(iface)) cfg = dedent( """ HWADDR="{hwaddr}" BOOTPROTO="{bootproto}" ONBOOT="{onboot}" TYPE="{type}" NAME="{iface}" """.format( hwaddr=hwaddr, bootproto=bootproto, onboot=onboot, type=type, iface=iface ) ).lstrip() with tempfile.NamedTemporaryFile(delete=False) as ifcfg_file: ifcfg_file.write(cfg) LOGGER.debug('generated %s for %s:\n%s',, iface_path, cfg) return ('--mkdir', path, '--chmod', '0755:{0}'.format(path) ) + _upload_file(iface_path,
[docs]def config_net_iface_debian(name, mac): iface = dedent( """ auto {name} iface {name} inet6 auto iface {name} inet dhcp hwaddress ether {mac} """.format(name=name, mac=mac) ) return ( _write_file( os.path.join( '/etc/network/interfaces.d', 'ifcfg-{0}.cfg'.format(name) ), iface ) )
[docs]def config_net_iface_loop_debian(): loop_device = dedent( """ auto lo iface lo inet loopback source /etc/network/interfaces.d/*.cfg """ ) return (_write_file('/etc/network/interfaces', loop_device))
[docs]def config_net_ifaces_dhcp(distro, mapping): if distro == 'debian': cmd = [config_net_iface_loop_debian()] cmd.extend( [config_net_iface_debian(name, mac) for name, mac in mapping] ) else: cmd = [config_net_iface_dhcp(name, mac) for name, mac in mapping] return cmd
[docs]def config_net_iface_dhcp( iface, hwaddr, path='/etc/sysconfig/network-scripts' ): return _config_net_interface( path=path, iface=iface, type='Ethernet', bootproto='dhcp', onboot='yes', hwaddr=hwaddr, )
[docs]def edit(filename, expression): editstr = '%s:""%s""' % (filename, expression) return ('--edit', editstr, )
[docs]def update(): return ('--update', '--network', )
[docs]def sysprep(disk, mods, backend='direct'): cmd = ['virt-sysprep', '-a', disk, '--selinux-relabel'] env = os.environ.copy() if 'LIBGUESTFS_BACKEND' not in env: env['LIBGUESTFS_BACKEND'] = backend for mod in mods: cmd.extend(mod) ret = utils.run_command(cmd, env=env) if ret: raise RuntimeError( 'Failed to bootstrap %s\ncommand:%s\nstdout:%s\nstderr:%s' % ( disk, ' '.join('"%s"' % elem for elem in cmd), ret.out, ret.err, ) )