e5a97d7d20
In some cases, the NoCloud config drive is built using `_` instead of `-` characters for the metadata and userdata paths. This commit adds two configuration options to be able to set custom values to those paths: ```ini [nocloud] metadata_file = "meta-data" userdata_file = "user-data" ``` Fixes: https://github.com/cloudbase/cloudbase-init/issues/89 Change-Id: I312aa26ed9be6f22156ac238f456c3906d93760d Signed-off-by: Adrian Vladu <avladu@cloudbasesolutions.com>
96 lines
3.5 KiB
Python
96 lines
3.5 KiB
Python
# Copyright 2012 Cloudbase Solutions Srl
|
|
#
|
|
# 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 os
|
|
import shutil
|
|
|
|
from oslo_log import log as oslo_logging
|
|
|
|
from cloudbaseinit import conf as cloudbaseinit_conf
|
|
from cloudbaseinit import constant
|
|
from cloudbaseinit import exception
|
|
from cloudbaseinit.metadata.services import base
|
|
from cloudbaseinit.metadata.services.osconfigdrive import factory
|
|
|
|
CONF = cloudbaseinit_conf.CONF
|
|
LOG = oslo_logging.getLogger(__name__)
|
|
|
|
CD_TYPES = constant.CD_TYPES
|
|
CD_LOCATIONS = constant.CD_LOCATIONS
|
|
|
|
|
|
class BaseConfigDriveService(base.BaseMetadataService):
|
|
|
|
def __init__(self, drive_label, metadata_file,
|
|
userdata_file='user-data'):
|
|
super(BaseConfigDriveService, self).__init__()
|
|
self._drive_label = drive_label
|
|
self._metadata_file = metadata_file
|
|
self._userdata_file = userdata_file
|
|
self._metadata_path = None
|
|
self._searched_types = set()
|
|
self._searched_locations = set()
|
|
|
|
def _preprocess_options(self):
|
|
self._searched_types = set(CONF.config_drive.types)
|
|
self._searched_locations = set(CONF.config_drive.locations)
|
|
|
|
# Deprecation backward compatibility.
|
|
if CONF.config_drive.raw_hdd:
|
|
self._searched_types.add("iso")
|
|
self._searched_locations.add("hdd")
|
|
if CONF.config_drive.cdrom:
|
|
self._searched_types.add("iso")
|
|
self._searched_locations.add("cdrom")
|
|
if CONF.config_drive.vfat:
|
|
self._searched_types.add("vfat")
|
|
self._searched_locations.add("hdd")
|
|
|
|
# Check for invalid option values.
|
|
if self._searched_types | CD_TYPES != CD_TYPES:
|
|
raise exception.CloudbaseInitException(
|
|
"Invalid Config Drive types %s", self._searched_types)
|
|
if self._searched_locations | CD_LOCATIONS != CD_LOCATIONS:
|
|
raise exception.CloudbaseInitException(
|
|
"Invalid Config Drive locations %s", self._searched_locations)
|
|
|
|
def load(self):
|
|
super(BaseConfigDriveService, self).load()
|
|
|
|
self._preprocess_options()
|
|
self._mgr = factory.get_config_drive_manager()
|
|
found = self._mgr.get_config_drive_files(
|
|
drive_label=self._drive_label,
|
|
metadata_file=self._metadata_file,
|
|
searched_types=self._searched_types,
|
|
searched_locations=self._searched_locations)
|
|
|
|
if found:
|
|
self._metadata_path = self._mgr.target_path
|
|
LOG.debug('Metadata copied to folder: %r', self._metadata_path)
|
|
return found
|
|
|
|
def _get_data(self, path):
|
|
norm_path = os.path.normpath(os.path.join(self._metadata_path, path))
|
|
try:
|
|
with open(norm_path, 'rb') as stream:
|
|
return stream.read()
|
|
except IOError:
|
|
raise base.NotExistingMetadataException()
|
|
|
|
def cleanup(self):
|
|
LOG.debug('Deleting metadata folder: %r', self._mgr.target_path)
|
|
shutil.rmtree(self._mgr.target_path, ignore_errors=True)
|
|
self._metadata_path = None
|