various updates
This commit is contained in:
1
node_agent/utils/__init__.py
Normal file
1
node_agent/utils/__init__.py
Normal file
@ -0,0 +1 @@
|
||||
from .mac import *
|
||||
16
node_agent/utils/mac.py
Normal file
16
node_agent/utils/mac.py
Normal file
@ -0,0 +1,16 @@
|
||||
import random
|
||||
|
||||
|
||||
def random_mac() -> str:
|
||||
"""Retrun random MAC address."""
|
||||
mac = [0x00, 0x16, 0x3e,
|
||||
random.randint(0x00, 0x7f),
|
||||
random.randint(0x00, 0xff),
|
||||
random.randint(0x00, 0xff)]
|
||||
return ':'.join(map(lambda x: "%02x" % x, mac))
|
||||
|
||||
|
||||
def unique_mac() -> str:
|
||||
"""Return non-conflicting MAC address."""
|
||||
# todo: see virtinst.DeviceInterface.generate_mac
|
||||
raise NotImplementedError()
|
||||
@ -1,72 +0,0 @@
|
||||
"""
|
||||
Manage virtual machines.
|
||||
|
||||
Usage: na-vmctl [options] status <machine>
|
||||
na-vmctl [options] is-running <machine>
|
||||
na-vmctl [options] start <machine>
|
||||
na-vmctl [options] shutdown <machine> [-f|--force] [-9|--sigkill]
|
||||
|
||||
Options:
|
||||
-c, --config <file> Config file [default: /etc/node-agent/config.yaml]
|
||||
-l, --loglvl <lvl> Logging level
|
||||
-f, --force Force action. On shutdown calls graceful destroy()
|
||||
-9, --sigkill Send SIGKILL to QEMU process. Not affects without --force
|
||||
"""
|
||||
|
||||
import sys
|
||||
import pathlib
|
||||
import logging
|
||||
|
||||
import libvirt
|
||||
from docopt import docopt
|
||||
|
||||
from ..main import LibvirtSession
|
||||
from ..vm import VirtualMachine, VMError, VMNotFound
|
||||
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
levels = logging.getLevelNamesMapping()
|
||||
|
||||
|
||||
class Color:
|
||||
RED = '\033[31m'
|
||||
GREEN = '\033[32m'
|
||||
YELLOW = '\033[33m'
|
||||
NONE = '\033[0m'
|
||||
|
||||
|
||||
def cli():
|
||||
args = docopt(__doc__)
|
||||
config = pathlib.Path(args['--config']) or None
|
||||
loglvl = None
|
||||
machine = args['<machine>']
|
||||
|
||||
if args['--loglvl']:
|
||||
loglvl = args['--loglvl'].upper()
|
||||
|
||||
if loglvl in levels:
|
||||
logging.basicConfig(level=levels[loglvl])
|
||||
|
||||
with LibvirtSession(config) as session:
|
||||
try:
|
||||
vm = VirtualMachine(session, machine)
|
||||
if args['status']:
|
||||
print(vm.status)
|
||||
if args['is-running']:
|
||||
if vm.is_running:
|
||||
print('running')
|
||||
else:
|
||||
sys.exit(vm.status)
|
||||
if args['start']:
|
||||
vm.start()
|
||||
print(f'{vm.name} started')
|
||||
if args['shutdown']:
|
||||
vm.shutdown(force=args['--force'], sigkill=args['sigkill'])
|
||||
except VMNotFound as nferr:
|
||||
sys.exit(f'{Color.RED}VM {machine} not found.{Color.NONE}')
|
||||
except VMError as vmerr:
|
||||
sys.exit(f'{Color.RED}{vmerr}{Color.NONE}')
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
cli()
|
||||
@ -1,99 +0,0 @@
|
||||
"""
|
||||
Execute shell commands on guest via guest agent.
|
||||
|
||||
Usage: na-vmexec [options] <machine> <command>
|
||||
|
||||
Options:
|
||||
-c, --config <file> Config file [default: /etc/node-agent/config.yaml]
|
||||
-l, --loglvl <lvl> Logging level
|
||||
-s, --shell <shell> Guest shell [default: /bin/sh]
|
||||
-t, --timeout <sec> QEMU timeout in seconds to stop polling command status [default: 60]
|
||||
"""
|
||||
|
||||
import sys
|
||||
import pathlib
|
||||
import logging
|
||||
|
||||
from docopt import docopt
|
||||
|
||||
from ..main import LibvirtSession
|
||||
from ..vm import QemuAgent, QemuAgentError, VMNotFound
|
||||
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
levels = logging.getLevelNamesMapping()
|
||||
|
||||
|
||||
class Color:
|
||||
RED = '\033[31m'
|
||||
GREEN = '\033[32m'
|
||||
YELLOW = '\033[33m'
|
||||
NONE = '\033[0m'
|
||||
|
||||
|
||||
def cli():
|
||||
args = docopt(__doc__)
|
||||
config = pathlib.Path(args['--config']) or None
|
||||
loglvl = None
|
||||
machine = args['<machine>']
|
||||
|
||||
if args['--loglvl']:
|
||||
loglvl = args['--loglvl'].upper()
|
||||
|
||||
if loglvl in levels:
|
||||
logging.basicConfig(level=levels[loglvl])
|
||||
|
||||
with LibvirtSession(config) as session:
|
||||
shell = args['--shell']
|
||||
cmd = args['<command>']
|
||||
|
||||
try:
|
||||
ga = QemuAgent(session, machine)
|
||||
exited, exitcode, stdout, stderr = ga.shellexec(
|
||||
cmd,
|
||||
executable=shell,
|
||||
capture_output=True,
|
||||
decode_output=True,
|
||||
timeout=int(args['--timeout']),
|
||||
)
|
||||
except QemuAgentError as qemuerr:
|
||||
errmsg = f'{Color.RED}{err}{Color.NONE}'
|
||||
if str(err).startswith('Polling command pid='):
|
||||
errmsg = (
|
||||
errmsg + Color.YELLOW
|
||||
+ '\n[NOTE: command may still running]'
|
||||
+ Color.NONE
|
||||
)
|
||||
sys.exit(errmsg)
|
||||
except VMNotFound as err:
|
||||
sys.exit(
|
||||
f'{Color.RED}VM {machine} not found.{Color.NONE}'
|
||||
)
|
||||
|
||||
if not exited:
|
||||
print(
|
||||
Color.YELLOW
|
||||
+'[NOTE: command may still running]'
|
||||
+ Color.NONE,
|
||||
file=sys.stderr
|
||||
)
|
||||
else:
|
||||
if exitcode == 0:
|
||||
exitcolor = Color.GREEN
|
||||
else:
|
||||
exitcolor = Color.RED
|
||||
print(
|
||||
exitcolor
|
||||
+ f'[command exited with exit code {exitcode}]'
|
||||
+ Color.NONE,
|
||||
file=sys.stderr
|
||||
)
|
||||
|
||||
if stderr:
|
||||
print(Color.RED + stderr.strip() + Color.NONE, file=sys.stderr)
|
||||
if stdout:
|
||||
print(Color.GREEN + stdout.strip() + Color.NONE, file=sys.stdout)
|
||||
sys.exit(exitcode)
|
||||
|
||||
if __name__ == '__main__':
|
||||
cli()
|
||||
197
node_agent/utils/xml.py
Normal file
197
node_agent/utils/xml.py
Normal file
@ -0,0 +1,197 @@
|
||||
from pathlib import Path
|
||||
|
||||
from lxml.builder import E
|
||||
from lxml.etree import Element, QName, SubElement, tostring
|
||||
|
||||
from .mac import random_mac
|
||||
|
||||
|
||||
XPATH_DOMAIN_NAME = '/domain/name'
|
||||
XPATH_DOMAIN_TITLE = '/domain/title'
|
||||
XPATH_DOMAIN_DESCRIPTION = '/domain/description'
|
||||
XPATH_DOMAIN_METADATA = '/domain/metadata'
|
||||
XPATH_DOMAIN_MEMORY = '/domain/memory'
|
||||
XPATH_DOMAIN_CURRENT_MEMORY = '/domain/currentMemory'
|
||||
XPATH_DOMAIN_VCPU = '/domain/vcpu'
|
||||
XPATH_DOMAIN_OS = '/domian/os'
|
||||
XPATH_DOMAIN_CPU = '/domain/cpu'
|
||||
|
||||
|
||||
class XMLConstructor:
|
||||
"""
|
||||
The XML constructor. This class builds XML configs for libvirtd.
|
||||
Features:
|
||||
- Generate basic virtual machine XML. See gen_domain_xml()
|
||||
- Generate virtual disk XML. See gen_volume_xml()
|
||||
- Add arbitrary metadata to XML from special structured dict
|
||||
"""
|
||||
|
||||
def __init__(self, xml: str | None = None):
|
||||
self.xml_string = xml
|
||||
self.xml = None
|
||||
|
||||
@property
|
||||
def domain_xml(self):
|
||||
return self.xml
|
||||
|
||||
def gen_domain_xml(self,
|
||||
name: str,
|
||||
title: str,
|
||||
vcpus: int,
|
||||
vcpu_vendor: str,
|
||||
vcpu_model: str,
|
||||
memory: int,
|
||||
volume: Path,
|
||||
vcpu_features: dict | None = None,
|
||||
desc: str = "") -> None:
|
||||
"""
|
||||
Generate default domain XML configuration for virtual machines.
|
||||
See https://lxml.de/tutorial.html#the-e-factory for details.
|
||||
"""
|
||||
self.xml = E.domain(
|
||||
E.name(name),
|
||||
E.title(title),
|
||||
E.description(desc),
|
||||
E.metadata(),
|
||||
E.memory(str(memory), unit='MB'),
|
||||
E.currentMemory(str(memory), unit='MB'),
|
||||
E.vcpu(str(vcpus), placement='static'),
|
||||
E.os(
|
||||
E.type('hvm', arch='x86_64'),
|
||||
E.boot(dev='cdrom'),
|
||||
E.boot(dev='hd'),
|
||||
),
|
||||
E.features(
|
||||
E.acpi(),
|
||||
E.apic(),
|
||||
),
|
||||
E.cpu(
|
||||
E.vendor(vcpu_vendor),
|
||||
E.model(vcpu_model, fallback='forbid'),
|
||||
E.topology(sockets='1', dies='1', cores=str(vcpus),
|
||||
threads='1'),
|
||||
mode='custom',
|
||||
match='exact',
|
||||
check='partial',
|
||||
),
|
||||
E.on_poweroff('destroy'),
|
||||
E.on_reboot('restart'),
|
||||
E.on_crash('restart'),
|
||||
E.pm(
|
||||
E('suspend-to-mem', enabled='no'),
|
||||
E('suspend-to-disk', enabled='no'),
|
||||
),
|
||||
E.devices(
|
||||
E.emulator('/usr/bin/qemu-system-x86_64'),
|
||||
E.disk(
|
||||
E.driver(name='qemu', type='qcow2', cache='writethrough'),
|
||||
E.source(file=volume),
|
||||
E.target(dev='vda', bus='virtio'),
|
||||
type='file',
|
||||
device='disk',
|
||||
),
|
||||
E.interface(
|
||||
E.source(network='default'),
|
||||
E.mac(address=random_mac()),
|
||||
type='network',
|
||||
),
|
||||
E.graphics(
|
||||
E.listen(type='address'),
|
||||
type='vnc', port='-1', autoport='yes'
|
||||
),
|
||||
E.video(
|
||||
E.model(type='vga', vram='16384', heads='1', primary='yes'),
|
||||
E.address(type='pci', domain='0x0000', bus='0x00',
|
||||
slot='0x02', function='0x0'),
|
||||
),
|
||||
),
|
||||
type='kvm',
|
||||
)
|
||||
|
||||
def gen_volume_xml(self,
|
||||
device_name: str,
|
||||
file: Path,
|
||||
bus: str = 'virtio',
|
||||
cache: str = 'writethrough',
|
||||
disktype: str = 'file'):
|
||||
return E.disk(E.driver(name='qemu', type='qcow2', cache=cache),
|
||||
E.source(file=file),
|
||||
E.target(dev=device_name, bus=bus),
|
||||
type=disktype,
|
||||
device='disk')
|
||||
|
||||
def add_volume(self):
|
||||
raise NotImplementedError()
|
||||
|
||||
def add_meta(self, data: dict, namespace: str, nsprefix: str) -> None:
|
||||
"""
|
||||
Add metadata to domain. See:
|
||||
https://libvirt.org/formatdomain.html#general-metadata
|
||||
"""
|
||||
metadata = metadata_old = self.xml.xpath('/domain/metadata')[0]
|
||||
metadata.append(
|
||||
self.construct_xml(
|
||||
data,
|
||||
namespace=namespace,
|
||||
nsprefix=nsprefix,
|
||||
))
|
||||
self.xml.replace(metadata_old, metadata)
|
||||
|
||||
def remove_meta(self, namespace: str):
|
||||
"""Remove metadata by namespace."""
|
||||
raise NotImplementedError()
|
||||
|
||||
def construct_xml(self,
|
||||
tag: dict,
|
||||
namespace: str | None = None,
|
||||
nsprefix: str | None = None,
|
||||
root: Element = None) -> Element:
|
||||
"""
|
||||
Shortly this recursive function transforms dictonary to XML.
|
||||
Return etree.Element built from dict with following structure::
|
||||
|
||||
{
|
||||
'name': 'devices', # tag name
|
||||
'text': '', # optional key
|
||||
'values': { # optional key, must be a dict of key-value pairs
|
||||
'type': 'disk'
|
||||
},
|
||||
children: [] # optional key, must be a list of dicts
|
||||
}
|
||||
|
||||
Child elements must have the same structure. Infinite `children` nesting
|
||||
is allowed.
|
||||
"""
|
||||
use_ns = False
|
||||
if isinstance(namespace, str) and isinstance(nsprefix, str):
|
||||
use_ns = True
|
||||
# Create element
|
||||
if root is None:
|
||||
if use_ns:
|
||||
element = Element(QName(namespace, tag['name']),
|
||||
nsmap={nsprefix: namespace})
|
||||
else:
|
||||
element = Element(tag['name'])
|
||||
else:
|
||||
if use_ns:
|
||||
element = SubElement(root, QName(namespace, tag['name']))
|
||||
else:
|
||||
element = SubElement(root, tag['name'])
|
||||
# Fill up element with content
|
||||
if 'text' in tag.keys():
|
||||
element.text = tag['text']
|
||||
if 'values' in tag.keys():
|
||||
for key in tag['values'].keys():
|
||||
element.set(str(key), str(tag['values'][key]))
|
||||
if 'children' in tag.keys():
|
||||
for child in tag['children']:
|
||||
element.append(
|
||||
self.construct_xml(child,
|
||||
namespace=namespace,
|
||||
nsprefix=nsprefix,
|
||||
root=element))
|
||||
return element
|
||||
|
||||
def to_string(self):
|
||||
return (tostring(self.xml, pretty_print=True,
|
||||
encoding='utf-8').decode().strip())
|
||||
Reference in New Issue
Block a user