18140a7787
Previously this had 'local' as the default datasource mode, meaning that user-data code such as boot hooks and such would not be guaranteed to have network access. That would be out of sync with the expectation on other platforms where the default is 'network up'. The user can still specify 'dsmode' as local if necessary and the local datasource will claim itself found.
105 lines
3.4 KiB
Python
105 lines
3.4 KiB
Python
# vi: ts=4 expandtab
|
|
#
|
|
# Copyright (C) 2014 CloudSigma
|
|
#
|
|
# Author: Kiril Vladimiroff <kiril.vladimiroff@cloudsigma.com>
|
|
#
|
|
# 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 <http://www.gnu.org/licenses/>.
|
|
import re
|
|
|
|
from cloudinit import log as logging
|
|
from cloudinit import sources
|
|
from cloudinit import util
|
|
from cloudinit.cs_utils import Cepko
|
|
|
|
LOG = logging.getLogger(__name__)
|
|
|
|
VALID_DSMODES = ("local", "net", "disabled")
|
|
|
|
|
|
class DataSourceCloudSigma(sources.DataSource):
|
|
"""
|
|
Uses cepko in order to gather the server context from the VM.
|
|
|
|
For more information about CloudSigma's Server Context:
|
|
http://cloudsigma-docs.readthedocs.org/en/latest/server_context.html
|
|
"""
|
|
def __init__(self, sys_cfg, distro, paths):
|
|
self.dsmode = 'local'
|
|
self.cepko = Cepko()
|
|
self.ssh_public_key = ''
|
|
sources.DataSource.__init__(self, sys_cfg, distro, paths)
|
|
|
|
def get_data(self):
|
|
"""
|
|
Metadata is the whole server context and /meta/cloud-config is used
|
|
as userdata.
|
|
"""
|
|
dsmode = None
|
|
try:
|
|
server_context = self.cepko.all().result
|
|
server_meta = server_context['meta']
|
|
except:
|
|
util.logexc(LOG, "Failed reading from the serial port")
|
|
return False
|
|
|
|
dsmode = server_meta.get('cloudinit-dsmode', self.dsmode)
|
|
if dsmode not in VALID_DSMODES:
|
|
LOG.warn("Invalid dsmode %s, assuming default of 'net'", dsmode)
|
|
dsmode = 'net'
|
|
if dsmode == "disabled" or dsmode != self.dsmode:
|
|
return False
|
|
|
|
self.userdata_raw = server_meta.get('cloudinit-user-data', "")
|
|
self.metadata = server_context
|
|
self.ssh_public_key = server_meta['ssh_public_key']
|
|
|
|
return True
|
|
|
|
def get_hostname(self, fqdn=False, resolve_ip=False):
|
|
"""
|
|
Cleans up and uses the server's name if the latter is set. Otherwise
|
|
the first part from uuid is being used.
|
|
"""
|
|
if re.match(r'^[A-Za-z0-9 -_\.]+$', self.metadata['name']):
|
|
return self.metadata['name'][:61]
|
|
else:
|
|
return self.metadata['uuid'].split('-')[0]
|
|
|
|
def get_public_ssh_keys(self):
|
|
return [self.ssh_public_key]
|
|
|
|
def get_instance_id(self):
|
|
return self.metadata['uuid']
|
|
|
|
|
|
class DataSourceCloudSigmaNet(DataSourceCloudSigma):
|
|
def __init__(self, sys_cfg, distro, paths):
|
|
DataSourceCloudSigma.__init__(self, sys_cfg, distro, paths)
|
|
self.dsmode = 'net'
|
|
|
|
|
|
# Used to match classes to dependencies. Since this datasource uses the serial
|
|
# port network is not really required, so it's okay to load without it, too.
|
|
datasources = [
|
|
(DataSourceCloudSigma, (sources.DEP_FILESYSTEM)),
|
|
(DataSourceCloudSigmaNet, (sources.DEP_FILESYSTEM, sources.DEP_NETWORK)),
|
|
]
|
|
|
|
|
|
def get_datasource_list(depends):
|
|
"""
|
|
Return a list of data sources that match this set of dependencies
|
|
"""
|
|
return sources.list_from_depends(depends, datasources)
|