180 lines
6.1 KiB
Python
180 lines
6.1 KiB
Python
#
|
|
# The Qubes OS Project, http://www.qubes-os.org
|
|
#
|
|
# Copyright (C) 2010-2015 Joanna Rutkowska <joanna@invisiblethingslab.com>
|
|
# Copyright (C) 2015 Wojtek Porczyk <woju@invisiblethingslab.com>
|
|
#
|
|
# 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.
|
|
#
|
|
|
|
# TODO list available classes
|
|
# TODO list labels (maybe in qvm-prefs)
|
|
# TODO features, devices, tags
|
|
|
|
from __future__ import print_function
|
|
|
|
import argparse
|
|
import os
|
|
import subprocess
|
|
import sys
|
|
|
|
import qubes
|
|
import qubes.tools
|
|
|
|
|
|
parser = qubes.tools.QubesArgumentParser(want_force_root=True)
|
|
|
|
parser.add_argument('--class', '-C', dest='cls',
|
|
default='AppVM',
|
|
help='specify the class of the new domain (default: %(default)s)')
|
|
|
|
parser.add_argument('--property', '--prop',
|
|
action=qubes.tools.PropertyAction,
|
|
help='set domain\'s property, like "internal", "memory" or "vcpus"')
|
|
|
|
parser.add_argument('--pool', '-p',
|
|
action='append',
|
|
metavar='POOL_NAME:VOLUME_NAME',
|
|
help='specify the pool to use for a volume')
|
|
|
|
parser.add_argument('-P',
|
|
metavar='POOL_NAME',
|
|
dest='one_pool',
|
|
default='',
|
|
help='change all volume pools to specified pool')
|
|
|
|
parser.add_argument('--template', '-t',
|
|
action=qubes.tools.SinglePropertyAction,
|
|
help='specify the TemplateVM to use')
|
|
|
|
parser.add_argument('--label', '-l',
|
|
action=qubes.tools.SinglePropertyAction,
|
|
help='specify the label to use for the new domain'
|
|
' (e.g. red, yellow, green, ...)')
|
|
|
|
parser_root = parser.add_mutually_exclusive_group()
|
|
parser_root.add_argument('--root-copy-from', '-r', metavar='FILENAME',
|
|
help='use provided root.img instead of default/empty one'
|
|
' (file will be COPIED)')
|
|
parser_root.add_argument('--root-move-from', '-R', metavar='FILENAME',
|
|
help='use provided root.img instead of default/empty one'
|
|
' (file will be MOVED)')
|
|
parser_root.add_argument('--no-root',
|
|
action='store_true', default=False,
|
|
help=argparse.SUPPRESS)
|
|
|
|
parser.add_argument('name', metavar='VMNAME',
|
|
action=qubes.tools.SinglePropertyAction,
|
|
nargs='?',
|
|
help='name of the domain to create')
|
|
|
|
|
|
def main(args=None):
|
|
args = parser.parse_args(args)
|
|
|
|
pools = {}
|
|
pool = None
|
|
if hasattr(args, 'pools') and args.pools:
|
|
for pool_vol in args.pool:
|
|
try:
|
|
pool_name, volume_name = pool_vol.split(':')
|
|
pools[volume_name] = pool_name
|
|
except ValueError:
|
|
parser.error(
|
|
'Pool argument must be of form: -P pool_name:volume_name')
|
|
if args.one_pool:
|
|
pool = args.one_pool
|
|
|
|
if 'label' not in args.properties:
|
|
parser.error('--label option is mandatory')
|
|
|
|
if 'name' not in args.properties:
|
|
parser.error('VMNAME is mandatory')
|
|
|
|
try:
|
|
args.app.get_label(args.properties['label'])
|
|
except KeyError:
|
|
parser.error('no such label: {!r}; available: {}'.format(
|
|
args.properties['label'],
|
|
', '.join(repr(l.name) for l in args.app.labels)))
|
|
|
|
try:
|
|
cls = args.app.get_vm_class(args.cls)
|
|
except KeyError:
|
|
parser.error('no such domain class: {!r}'.format(args.cls))
|
|
|
|
if 'template' in args.properties and \
|
|
'template' not in (prop.__name__ for prop in cls.property_list()):
|
|
parser.error('this domain class does not support template')
|
|
|
|
vm = args.app.add_new_vm(cls, **args.properties)
|
|
|
|
# pylint: disable=line-too-long
|
|
|
|
# if not options.standalone and any([options.root_copy_from, options.root_move_from]):
|
|
# print >> sys.stderr, "root.img can be specified only for standalone VMs"
|
|
# exit (1)
|
|
|
|
# if options.hvm_template and options.template is not None:
|
|
# print >> sys.stderr, "Template VM cannot be based on another template"
|
|
# exit (1)
|
|
|
|
# if options.root_copy_from is not None and not os.path.exists(options.root_copy_from):
|
|
# print >> sys.stderr, "File specified as root.img does not exists"
|
|
# exit (1)
|
|
|
|
# if options.root_move_from is not None and not os.path.exists(options.root_move_from):
|
|
# print >> sys.stderr, "File specified as root.img does not exists"
|
|
# exit (1)
|
|
|
|
# elif not options.hvm and not options.hvm_template:
|
|
# if qvm_collection.get_default_template() is None:
|
|
# print >> sys.stderr, "No default TemplateVM defined!"
|
|
# exit (1)
|
|
# else:
|
|
# template = qvm_collection.get_default_template()
|
|
# if (options.verbose):
|
|
# print('--> Using default TemplateVM: {0}'.format(template.name))
|
|
|
|
if not args.no_root:
|
|
try:
|
|
vm.create_on_disk(pool, pools)
|
|
|
|
# TODO this is file pool specific. Change it to a more general
|
|
# solution
|
|
root_img_path = vm.volumes['root'].vid
|
|
if args.root_move_from is not None:
|
|
# if (options.verbose):
|
|
# print "--> Replacing root.img with provided file"
|
|
os.unlink(root_img_path)
|
|
os.rename(args.root_move_from, root_img_path)
|
|
elif args.root_copy_from is not None:
|
|
# if (options.verbose):
|
|
# print "--> Replacing root.img with provided file"
|
|
os.unlink(root_img_path)
|
|
# use 'cp' to preserve sparse file
|
|
subprocess.check_call(['cp', args.root_copy_from, root_img_path])
|
|
|
|
except (IOError, OSError) as err:
|
|
parser.error(str(err))
|
|
|
|
args.app.save()
|
|
|
|
return 0
|
|
|
|
|
|
if __name__ == '__main__':
|
|
sys.exit(main())
|