# vi: ts=4 expandtab # # Copyright (C) 2009-2010 Canonical Ltd. # Copyright (C) 2012 Hewlett-Packard Development Company, L.P. # Copyright (C) 2012 Yahoo! Inc. # # Author: Joe VLcek # # This program is free software: you can redistribute it and/or modify # it under the terms of the GNU General Public License version 3, as # published by the Free Software Foundation. # # This program is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU General Public License for more details. # # You should have received a copy of the GNU General Public License # along with this program. If not, see . import errno import os import os.path import time from cloudinit import log as logging from cloudinit import sources from cloudinit import util from cloudinit.util import ProcessExecutionError LOG = logging.getLogger(__name__) # Needed file paths CLOUD_INFO_FILE = '/etc/sysconfig/cloud-info' # Shell command lists CMD_DMI_SYSTEM = ['/usr/sbin/dmidecode', '--string', 'system-product-name'] CMD_PROBE_FLOPPY = ['/sbin/modprobe', 'floppy'] # Retry times and sleep secs between each try RETRY_TIMES = 3 SLEEP_SECS = 3 META_DATA_NOT_SUPPORTED = { 'block-device-mapping': {}, 'instance-id': 455, 'local-hostname': 'localhost', 'placement': {}, } def read_user_data_callback(mount_dir): ''' Description: This callback will be applied by util.mount_cb() on the mounted file. Deltacloud file name contains deltacloud. Those not using Deltacloud but instead instrumenting the injection, could drop deltacloud from the file name. Input: mount_dir - Mount directory Returns: User Data ''' deltacloud_user_data_file = mount_dir + '/deltacloud-user-data.txt' user_data_file = mount_dir + '/user-data.txt' # First try deltacloud_user_data_file. On failure try user_data_file. try: with open(deltacloud_user_data_file, 'r') as user_data_f: user_data = user_data_f.read().strip() except: try: with open(user_data_file, 'r') as user_data_f: user_data = user_data_f.read().strip() except: util.logexc(LOG, ('Failed accessing user data file.')) return None return user_data class DataSourceAltCloud(sources.DataSource): def __init__(self, sys_cfg, distro, paths): sources.DataSource.__init__(self, sys_cfg, distro, paths) self.seed = None self.supported_seed_starts = ("/", "file://") def __str__(self): mstr = "%s [seed=%s]" % (util.obj_name(self), self.seed) return mstr def get_cloud_type(self): ''' Description: Get the type for the cloud back end this instance is running on by examining the string returned by: dmidecode --string system-product-name On VMWare/vSphere dmidecode returns: RHEV Hypervisor On VMWare/vSphere dmidecode returns: VMware Virtual Platform Input: None Returns: One of the following strings: 'RHEV', 'VSPHERE' or 'UNKNOWN' ''' cmd = CMD_DMI_SYSTEM try: (cmd_out, _err) = util.subp(cmd) except ProcessExecutionError, _err: LOG.debug(('Failed command: %s\n%s') % \ (' '.join(cmd), _err.message)) return 'UNKNOWN' except OSError, _err: LOG.debug(('Failed command: %s\n%s') % \ (' '.join(cmd), _err.message)) return 'UNKNOWN' if cmd_out.upper().startswith('RHEV'): return 'RHEV' if cmd_out.upper().startswith('VMWARE'): return 'VSPHERE' return 'UNKNOWN' def get_data(self): ''' Description: User Data is passed to the launching instance which is used to perform instance configuration. Cloud providers expose the user data differently. It is necessary to determine which cloud provider the current instance is running on to determine how to access the user data. Images built with image factory will contain a CLOUD_INFO_FILE which contains a string identifying the cloud provider. Images not built with Imagefactory will try to determine what the cloud provider is based on system information. ''' LOG.debug('Invoked get_data()') if os.path.exists(CLOUD_INFO_FILE): try: cloud_info = open(CLOUD_INFO_FILE) cloud_type = cloud_info.read().strip().upper() cloud_info.close() except: util.logexc(LOG, 'Unable to access cloud info file.') return False else: cloud_type = self.get_cloud_type() LOG.debug('cloud_type: ' + str(cloud_type)) # Simple retry logic around user_data_() methods tries = RETRY_TIMES sleep_secs = SLEEP_SECS while tries > 0: if 'RHEV' in cloud_type: if self.user_data_rhevm(): return True elif 'VSPHERE' in cloud_type: if self.user_data_vsphere(): return True else: # there was no recognized alternate cloud type. # suggesting this handler should not be used. return False time.sleep(sleep_secs) tries -= 1 sleep_secs *= 3 # Retry loop exhausted return False def user_data_rhevm(self): ''' RHEVM specific userdata read If on RHEV-M the user data will be contained on the floppy device in file To access it: modprobe floppy Leverage util.mount_cb to: mkdir mount /dev/fd0 The call back passed to util.mount_cb will do: read / ''' return_str = None # modprobe floppy try: cmd = CMD_PROBE_FLOPPY (cmd_out, _err) = util.subp(cmd) LOG.debug(('Command: %s\nOutput%s') % (' '.join(cmd), cmd_out)) except ProcessExecutionError, _err: util.logexc(LOG, (('Failed command: %s\n%s') % \ (' '.join(cmd), _err.message))) return False except OSError, _err: util.logexc(LOG, (('Failed command: %s\n%s') % \ (' '.join(cmd), _err.message))) return False floppy_dev = '/dev/fd0' try: return_str = util.mount_cb(floppy_dev, read_user_data_callback) except OSError as err: if err.errno != errno.ENOENT: raise except util.MountFailedError: util.logexc(LOG, ("Failed to mount %s" " when looking for user data"), floppy_dev) self.userdata_raw = return_str self.metadata = META_DATA_NOT_SUPPORTED if return_str: return True else: return False def user_data_vsphere(self): ''' vSphere specific userdata read If on vSphere the user data will be contained on the cdrom device in file To access it: Leverage util.mount_cb to: mkdir mount /dev/fd0 The call back passed to util.mount_cb will do: read / ''' return_str = None cdrom_list = util.find_devs_with('LABEL=CDROM') for cdrom_dev in cdrom_list: try: return_str = util.mount_cb(cdrom_dev, read_user_data_callback) if return_str: break except OSError as err: if err.errno != errno.ENOENT: raise except util.MountFailedError: util.logexc(LOG, ("Failed to mount %s" " when looking for user data"), cdrom_dev) self.userdata_raw = return_str self.metadata = META_DATA_NOT_SUPPORTED if return_str: return True else: return False # Used to match classes to dependencies datasources = [ (DataSourceAltCloud, (sources.DEP_FILESYSTEM, )), ] # Return a list of data sources that match this set of dependencies def get_datasource_list(depends): return sources.list_from_depends(depends, datasources)