137 lines
4.8 KiB
Python
137 lines
4.8 KiB
Python
# Copyright 2015 Cloudbase Solutions Srl
|
|
# All Rights Reserved.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
import functools
|
|
import os
|
|
|
|
from nova import exception
|
|
from nova.i18n import _
|
|
from nova import utils
|
|
from os_win import utilsfactory
|
|
from oslo_log import log as logging
|
|
from oslo_utils import importutils
|
|
import six
|
|
|
|
from compute_hyperv.nova import pathutils
|
|
from compute_hyperv.nova import serialconsolehandler
|
|
|
|
LOG = logging.getLogger(__name__)
|
|
|
|
_console_handlers = {}
|
|
|
|
|
|
def instance_synchronized(func):
|
|
@functools.wraps(func)
|
|
def wrapper(self, instance_name, *args, **kwargs):
|
|
@utils.synchronized(instance_name)
|
|
def inner():
|
|
return func(self, instance_name, *args, **kwargs)
|
|
return inner()
|
|
return wrapper
|
|
|
|
|
|
class SerialConsoleOps(object):
|
|
def __init__(self):
|
|
self._vmops_prop = None
|
|
|
|
self._vmutils = utilsfactory.get_vmutils()
|
|
self._pathutils = pathutils.PathUtils()
|
|
|
|
@property
|
|
def _vmops(self):
|
|
# We have to avoid a circular dependency.
|
|
if not self._vmops_prop:
|
|
self._vmops_prop = importutils.import_class(
|
|
'compute_hyperv.nova.vmops.VMOps')()
|
|
return self._vmops_prop
|
|
|
|
@instance_synchronized
|
|
def start_console_handler(self, instance_name):
|
|
if self._vmutils.is_secure_vm(instance_name):
|
|
LOG.warning("Skipping starting serial console handler. "
|
|
"Shielded/Encrypted VM %(instance_name)s "
|
|
"doesn't support serial console.",
|
|
{'instance_name': instance_name})
|
|
return
|
|
|
|
# Cleanup existing workers.
|
|
self.stop_console_handler_unsync(instance_name)
|
|
handler = None
|
|
|
|
try:
|
|
handler = serialconsolehandler.SerialConsoleHandler(
|
|
instance_name)
|
|
handler.start()
|
|
_console_handlers[instance_name] = handler
|
|
except Exception as exc:
|
|
LOG.error('Instance %(instance_name)s serial console handler '
|
|
'could not start. Exception %(exc)s',
|
|
{'instance_name': instance_name, 'exc': exc})
|
|
if handler:
|
|
handler.stop()
|
|
|
|
@instance_synchronized
|
|
def stop_console_handler(self, instance_name):
|
|
self.stop_console_handler_unsync(instance_name)
|
|
|
|
def stop_console_handler_unsync(self, instance_name):
|
|
handler = _console_handlers.get(instance_name)
|
|
if handler:
|
|
LOG.info("Stopping instance %(instance_name)s "
|
|
"serial console handler.",
|
|
{'instance_name': instance_name})
|
|
handler.stop()
|
|
del _console_handlers[instance_name]
|
|
|
|
@instance_synchronized
|
|
def get_serial_console(self, instance_name):
|
|
handler = _console_handlers.get(instance_name)
|
|
if not handler:
|
|
raise exception.ConsoleTypeUnavailable(console_type='serial')
|
|
return handler.get_serial_console()
|
|
|
|
@instance_synchronized
|
|
def get_console_output(self, instance_name):
|
|
if self._vmutils.is_secure_vm(instance_name):
|
|
err = _("Shielded/Encrypted VMs don't support serial console.")
|
|
raise exception.ConsoleNotAvailable(err)
|
|
|
|
console_log_paths = self._pathutils.get_vm_console_log_paths(
|
|
instance_name)
|
|
|
|
try:
|
|
log = b''
|
|
# Start with the oldest console log file.
|
|
for log_path in reversed(console_log_paths):
|
|
if os.path.exists(log_path):
|
|
with open(log_path, 'rb') as fp:
|
|
log += fp.read()
|
|
return log
|
|
except IOError as err:
|
|
raise exception.ConsoleLogOutputException(
|
|
instance_id=instance_name, reason=six.text_type(err))
|
|
|
|
def start_console_handlers(self):
|
|
active_instances = self._vmutils.get_active_instances()
|
|
for instance_name in active_instances:
|
|
instance_uuid = self._vmops.get_instance_uuid(instance_name)
|
|
|
|
if instance_uuid:
|
|
self.start_console_handler(instance_name)
|
|
else:
|
|
LOG.debug("Instance uuid could not be retrieved for "
|
|
"instance %s. Its serial console output will not "
|
|
"be handled.", instance_name)
|