Stephen Finucane
6bb1fd776f
All Python files, except executables, should have a '.py' extension. This wasn't being done for files in the 'plugins/xenserver' folder, likely because it seems unnatural to call a "plugin" with a file extension. Begin working towards a resolution for this by renaming all files to include a '.py' extension. To avoid forcing operators to upgrade the Nova service and per-host plugins at the same time, continue to reference the old, non-'.py' filenames, using symlinks to ensure the references continue to work. Once Ocata has been released, these symlinks can be removed, the API version updated and the reference updated to include the '.py' extension. Change-Id: Icf4cf2424c15276a6c01778fb92f76bbdc9b2068 Closes-bug: #1302831
144 lines
4.5 KiB
Python
Executable File
144 lines
4.5 KiB
Python
Executable File
#!/usr/bin/env python
|
|
|
|
# Copyright (c) 2012 OpenStack Foundation
|
|
# Copyright (c) 2010 Citrix Systems, Inc.
|
|
# Copyright 2010 United States Government as represented by the
|
|
# Administrator of the National Aeronautics and Space Administration.
|
|
# 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.
|
|
|
|
# NOTE: XenServer still only supports Python 2.4 in it's dom0 userspace
|
|
# which means the Nova xenapi plugins must use only Python 2.4 features
|
|
|
|
# TODO(sfinucan): Resolve all 'noqa' items once the above is no longer true
|
|
|
|
"""Handle the manipulation of kernel images."""
|
|
|
|
import errno
|
|
import os
|
|
import shutil
|
|
|
|
import XenAPIPlugin
|
|
|
|
import pluginlib_nova
|
|
|
|
|
|
pluginlib_nova.configure_logging('kernel')
|
|
logging = pluginlib_nova.logging
|
|
exists = pluginlib_nova.exists
|
|
optional = pluginlib_nova.optional
|
|
with_vdi_in_dom0 = pluginlib_nova.with_vdi_in_dom0
|
|
|
|
|
|
KERNEL_DIR = '/boot/guest'
|
|
|
|
|
|
def _copy_vdi(dest, copy_args):
|
|
vdi_uuid = copy_args['vdi_uuid']
|
|
vdi_size = copy_args['vdi_size']
|
|
cached_image = copy_args['cached-image']
|
|
|
|
logging.debug("copying kernel/ramdisk file from %s to /boot/guest/%s",
|
|
dest, vdi_uuid)
|
|
filename = KERNEL_DIR + '/' + vdi_uuid
|
|
|
|
# Make sure KERNEL_DIR exists, otherwise create it
|
|
if not os.path.isdir(KERNEL_DIR):
|
|
logging.debug("Creating directory %s", KERNEL_DIR)
|
|
os.makedirs(KERNEL_DIR)
|
|
|
|
# Read data from /dev/ and write into a file on /boot/guest
|
|
of = open(filename, 'wb')
|
|
f = open(dest, 'rb')
|
|
|
|
# Copy only vdi_size bytes
|
|
data = f.read(vdi_size)
|
|
of.write(data)
|
|
|
|
if cached_image:
|
|
# Create a cache file. If caching is enabled, kernel images do not have
|
|
# to be fetched from glance.
|
|
cached_image = KERNEL_DIR + '/' + cached_image
|
|
logging.debug("copying kernel/ramdisk file from %s to /boot/guest/%s",
|
|
dest, cached_image)
|
|
cache_file = open(cached_image, 'wb')
|
|
cache_file.write(data)
|
|
cache_file.close()
|
|
logging.debug("Done. Filename: %s", cached_image)
|
|
|
|
f.close()
|
|
of.close()
|
|
logging.debug("Done. Filename: %s", filename)
|
|
return filename
|
|
|
|
|
|
def copy_vdi(session, args):
|
|
vdi = exists(args, 'vdi-ref')
|
|
size = exists(args, 'image-size')
|
|
cached_image = optional(args, 'cached-image')
|
|
|
|
# Use the uuid as a filename
|
|
vdi_uuid = session.xenapi.VDI.get_uuid(vdi)
|
|
copy_args = {'vdi_uuid': vdi_uuid,
|
|
'vdi_size': int(size),
|
|
'cached-image': cached_image}
|
|
|
|
filename = with_vdi_in_dom0(session, vdi, False,
|
|
lambda dev:
|
|
_copy_vdi('/dev/%s' % dev, copy_args))
|
|
return filename
|
|
|
|
|
|
def create_kernel_ramdisk(session, args):
|
|
"""Creates a copy of the kernel/ramdisk image if it is present in the
|
|
cache. If the image is not present in the cache, it does nothing.
|
|
"""
|
|
cached_image = exists(args, 'cached-image')
|
|
image_uuid = exists(args, 'new-image-uuid')
|
|
cached_image_filename = KERNEL_DIR + '/' + cached_image
|
|
filename = KERNEL_DIR + '/' + image_uuid
|
|
|
|
if os.path.isfile(cached_image_filename):
|
|
shutil.copyfile(cached_image_filename, filename)
|
|
logging.debug("Done. Filename: %s", filename)
|
|
else:
|
|
filename = ""
|
|
logging.debug("Cached kernel/ramdisk image not found")
|
|
return filename
|
|
|
|
|
|
def _remove_file(filepath):
|
|
try:
|
|
os.remove(filepath)
|
|
except OSError, exc: # noqa
|
|
if exc.errno != errno.ENOENT:
|
|
raise
|
|
|
|
|
|
def remove_kernel_ramdisk(session, args):
|
|
"""Removes kernel and/or ramdisk from dom0's file system."""
|
|
kernel_file = optional(args, 'kernel-file')
|
|
ramdisk_file = optional(args, 'ramdisk-file')
|
|
if kernel_file:
|
|
_remove_file(kernel_file)
|
|
if ramdisk_file:
|
|
_remove_file(ramdisk_file)
|
|
return "ok"
|
|
|
|
|
|
if __name__ == '__main__':
|
|
XenAPIPlugin.dispatch({'copy_vdi': copy_vdi,
|
|
'create_kernel_ramdisk': create_kernel_ramdisk,
|
|
'remove_kernel_ramdisk': remove_kernel_ramdisk})
|