forked from uncloud/uncloud
233 lines
9 KiB
Python
Executable file
233 lines
9 KiB
Python
Executable file
# QEMU Manual
|
|
# https://qemu.weilnetz.de/doc/qemu-doc.html
|
|
|
|
# For QEMU Monitor Protocol Commands Information, See
|
|
# https://qemu.weilnetz.de/doc/qemu-doc.html#pcsys_005fmonitor
|
|
|
|
import os
|
|
import subprocess as sp
|
|
import ipaddress
|
|
|
|
from string import Template
|
|
from os.path import join as join_path
|
|
|
|
from ucloud.common.request import RequestEntry, RequestType
|
|
from ucloud.common.vm import VMStatus, declare_stopped
|
|
from ucloud.common.network import create_dev, delete_network_interface
|
|
from ucloud.common.schemas import VMSchema, NetworkSchema
|
|
from ucloud.host import logger
|
|
from ucloud.shared import shared
|
|
from ucloud.settings import settings
|
|
from ucloud.vmm import VMM
|
|
|
|
from marshmallow import ValidationError
|
|
|
|
|
|
def maintenance():
|
|
pass
|
|
|
|
|
|
class VM:
|
|
def __init__(self, vm_entry):
|
|
self.schema = VMSchema()
|
|
self.vmm = VMM()
|
|
self.key = vm_entry.key
|
|
try:
|
|
self.vm = self.schema.loads(vm_entry.value)
|
|
except ValidationError:
|
|
logger.exception('Couldn\'t validate VM Entry', vm_entry.value)
|
|
self.vm = None
|
|
else:
|
|
self.uuid = vm_entry.key.split('/')[-1]
|
|
self.host_key = self.vm['hostname']
|
|
|
|
def get_qemu_args(self):
|
|
command = (
|
|
'-name {owner}_{name}'
|
|
' -drive file={file},format=raw,if=virtio,cache=none'
|
|
' -device virtio-rng-pci'
|
|
' -m {memory} -smp cores={cores},threads={threads}'
|
|
).format(owner=self.vm['owner'], name=self.vm['name'],
|
|
memory=int(self.vm['specs']['ram'].to_MB()), cores=self.vm['specs']['cpu'],
|
|
threads=1, file=shared.storage_handler.qemu_path_string(self.uuid))
|
|
|
|
return command.split(' ')
|
|
|
|
def start(self, destination_host_key=None):
|
|
migration = False
|
|
if destination_host_key:
|
|
migration = True
|
|
|
|
self.create()
|
|
try:
|
|
network_args = self.create_network_dev()
|
|
except Exception as err:
|
|
declare_stopped(self.vm)
|
|
self.vm['log'].append('Cannot Setup Network Properly')
|
|
logger.error('Cannot Setup Network Properly for vm %s', self.uuid, exc_info=err)
|
|
else:
|
|
self.vmm.start(uuid=self.uuid, migration=migration,
|
|
*self.get_qemu_args(), *network_args)
|
|
|
|
status = self.vmm.get_status(self.uuid)
|
|
if status == 'running':
|
|
self.vm['status'] = VMStatus.running
|
|
self.vm['vnc_socket'] = self.vmm.get_vnc(self.uuid)
|
|
elif status == 'inmigrate':
|
|
r = RequestEntry.from_scratch(
|
|
type=RequestType.TransferVM, # Transfer VM
|
|
hostname=self.host_key, # Which VM should get this request. It is source host
|
|
uuid=self.uuid, # uuid of VM
|
|
destination_host_key=destination_host_key, # Where source host transfer VM
|
|
request_prefix=settings['etcd']['request_prefix']
|
|
)
|
|
shared.request_pool.put(r)
|
|
else:
|
|
self.stop()
|
|
declare_stopped(self.vm)
|
|
|
|
self.sync()
|
|
|
|
def stop(self):
|
|
self.vmm.stop(self.uuid)
|
|
self.delete_network_dev()
|
|
declare_stopped(self.vm)
|
|
self.sync()
|
|
|
|
def migrate(self, destination):
|
|
self.vmm.transfer(src_uuid=self.uuid, dest_uuid=self.uuid, host=destination)
|
|
|
|
def create_network_dev(self):
|
|
command = ''
|
|
for network_mac_and_tap in self.vm['network']:
|
|
network_name, mac, tap = network_mac_and_tap
|
|
|
|
_key = os.path.join(settings['etcd']['network_prefix'], self.vm['owner'], network_name)
|
|
network = shared.etcd_client.get(_key, value_in_json=True)
|
|
network_schema = NetworkSchema()
|
|
try:
|
|
network = network_schema.load(network.value)
|
|
except ValidationError:
|
|
continue
|
|
|
|
if network['type'] == "vxlan":
|
|
tap = create_vxlan_br_tap(_id=network['id'],
|
|
_dev=settings['network']['vxlan_phy_dev'],
|
|
tap_id=tap,
|
|
ip=network['ipv6'])
|
|
|
|
all_networks = shared.etcd_client.get_prefix(settings['etcd']['network_prefix'],
|
|
value_in_json=True)
|
|
|
|
if ipaddress.ip_network(network['ipv6']).is_global:
|
|
update_radvd_conf(all_networks)
|
|
|
|
command += '-netdev tap,id=vmnet{net_id},ifname={tap},script=no,downscript=no' \
|
|
' -device virtio-net-pci,netdev=vmnet{net_id},mac={mac}' \
|
|
.format(tap=tap, net_id=network['id'], mac=mac)
|
|
|
|
return command.split(' ')
|
|
|
|
def delete_network_dev(self):
|
|
try:
|
|
for network in self.vm['network']:
|
|
network_name = network[0]
|
|
_ = network[1] # tap_mac
|
|
tap_id = network[2]
|
|
|
|
delete_network_interface('tap{}'.format(tap_id))
|
|
|
|
owners_vms = shared.vm_pool.by_owner(self.vm['owner'])
|
|
owners_running_vms = shared.vm_pool.by_status(VMStatus.running,
|
|
_vms=owners_vms)
|
|
|
|
networks = map(
|
|
lambda n: n[0], map(lambda vm: vm.network, owners_running_vms)
|
|
)
|
|
networks_in_use_by_user_vms = [vm[0] for vm in networks]
|
|
if network_name not in networks_in_use_by_user_vms:
|
|
network_entry = resolve_network(network[0], self.vm['owner'])
|
|
if network_entry:
|
|
network_type = network_entry.value["type"]
|
|
network_id = network_entry.value["id"]
|
|
if network_type == "vxlan":
|
|
delete_network_interface('br{}'.format(network_id))
|
|
delete_network_interface('vxlan{}'.format(network_id))
|
|
except Exception:
|
|
logger.exception("Exception in network interface deletion")
|
|
|
|
def create(self):
|
|
if shared.storage_handler.is_vm_image_exists(self.uuid):
|
|
# File Already exists. No Problem Continue
|
|
logger.debug("Image for vm %s exists", self.uuid)
|
|
else:
|
|
if shared.storage_handler.make_vm_image(src=self.vm['image_uuid'], dest=self.uuid):
|
|
if not shared.storage_handler.resize_vm_image(path=self.uuid,
|
|
size=int(self.vm['specs']['os-ssd'].to_MB())):
|
|
self.vm['status'] = VMStatus.error
|
|
else:
|
|
logger.info("New VM Created")
|
|
|
|
def sync(self):
|
|
shared.etcd_client.put(self.key, self.schema.dump(self.vm), value_in_json=True)
|
|
|
|
def delete(self):
|
|
self.stop()
|
|
|
|
if shared.storage_handler.is_vm_image_exists(self.uuid):
|
|
r_status = shared.storage_handler.delete_vm_image(self.uuid)
|
|
if r_status:
|
|
shared.etcd_client.client.delete(self.key)
|
|
else:
|
|
shared.etcd_client.client.delete(self.key)
|
|
|
|
|
|
def resolve_network(network_name, network_owner):
|
|
network = shared.etcd_client.get(
|
|
join_path(settings['etcd']['network_prefix'], network_owner, network_name), value_in_json=True
|
|
)
|
|
return network
|
|
|
|
|
|
def create_vxlan_br_tap(_id, _dev, tap_id, ip=None):
|
|
network_script_base = os.path.join(os.path.dirname(os.path.dirname(__file__)), 'network')
|
|
vxlan = create_dev(script=os.path.join(network_script_base, 'create-vxlan.sh'),
|
|
_id=_id, dev=_dev)
|
|
if vxlan:
|
|
bridge = create_dev(script=os.path.join(network_script_base, 'create-bridge.sh'),
|
|
_id=_id, dev=vxlan, ip=ip)
|
|
if bridge:
|
|
tap = create_dev(script=os.path.join(network_script_base, 'create-tap.sh'),
|
|
_id=str(tap_id), dev=bridge)
|
|
if tap:
|
|
return tap
|
|
|
|
|
|
def update_radvd_conf(all_networks):
|
|
network_script_base = os.path.join(os.path.dirname(os.path.dirname(__file__)), 'network')
|
|
|
|
networks = {
|
|
net.value['ipv6']: net.value['id']
|
|
for net in all_networks
|
|
if net.value.get('ipv6') and ipaddress.ip_network(net.value.get('ipv6')).is_global
|
|
}
|
|
radvd_template = open(os.path.join(network_script_base,
|
|
'radvd-template.conf'), 'r').read()
|
|
radvd_template = Template(radvd_template)
|
|
|
|
content = [
|
|
radvd_template.safe_substitute(
|
|
bridge='br{}'.format(networks[net]),
|
|
prefix=net
|
|
)
|
|
for net in networks if networks.get(net)
|
|
]
|
|
with open('/etc/radvd.conf', 'w') as radvd_conf:
|
|
radvd_conf.writelines(content)
|
|
try:
|
|
sp.check_output(['systemctl', 'restart', 'radvd'])
|
|
except sp.CalledProcessError:
|
|
try:
|
|
sp.check_output(['service', 'radvd', 'restart'])
|
|
except sp.CalledProcessError as err:
|
|
raise err.__class__('Cannot start/restart radvd service', err.cmd) from err
|