145 lines
5.2 KiB
Python
145 lines
5.2 KiB
Python
# Copyright 2014 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 posixpath
|
|
import re
|
|
|
|
from oauthlib import oauth1
|
|
from oslo_config import cfg
|
|
from oslo_log import log as oslo_logging
|
|
from six.moves.urllib import error
|
|
from six.moves.urllib import request
|
|
|
|
from cloudbaseinit.metadata.services import base
|
|
from cloudbaseinit.utils import x509constants
|
|
|
|
MAAS_OPTS = [
|
|
cfg.StrOpt("metadata_base_url", default=None,
|
|
help="The base URL for MaaS metadata",
|
|
deprecated_name="maas_metadata_url",
|
|
deprecated_group="DEFAULT"),
|
|
cfg.StrOpt("oauth_consumer_key", default="",
|
|
help="The MaaS OAuth consumer key",
|
|
deprecated_name="maas_oauth_consumer_key",
|
|
deprecated_group="DEFAULT"),
|
|
cfg.StrOpt("oauth_consumer_secret", default="",
|
|
help="The MaaS OAuth consumer secret",
|
|
deprecated_name="maas_oauth_consumer_secret",
|
|
deprecated_group="DEFAULT"),
|
|
cfg.StrOpt("oauth_token_key", default="",
|
|
help="The MaaS OAuth token key",
|
|
deprecated_name="maas_oauth_token_key",
|
|
deprecated_group="DEFAULT"),
|
|
cfg.StrOpt("oauth_token_secret", default="",
|
|
help="The MaaS OAuth token secret",
|
|
deprecated_name="maas_oauth_token_secret",
|
|
deprecated_group="DEFAULT"),
|
|
]
|
|
|
|
CONF = cfg.CONF
|
|
CONF.register_group(cfg.OptGroup("maas"))
|
|
CONF.register_opts(MAAS_OPTS, "maas")
|
|
|
|
LOG = oslo_logging.getLogger(__name__)
|
|
|
|
|
|
class _Realm(str):
|
|
# There's a bug in oauthlib which ignores empty realm strings,
|
|
# by checking that the given realm is always True.
|
|
# This string class always returns True in a boolean context,
|
|
# making sure that an empty realm can be used by oauthlib.
|
|
def __bool__(self):
|
|
return True
|
|
|
|
__nonzero__ = __bool__
|
|
|
|
|
|
class MaaSHttpService(base.BaseMetadataService):
|
|
_METADATA_2012_03_01 = '2012-03-01'
|
|
|
|
def __init__(self):
|
|
super(MaaSHttpService, self).__init__()
|
|
self._enable_retry = True
|
|
self._metadata_version = self._METADATA_2012_03_01
|
|
|
|
def load(self):
|
|
super(MaaSHttpService, self).load()
|
|
|
|
if not CONF.maas.metadata_base_url:
|
|
LOG.debug('MaaS metadata url not set')
|
|
else:
|
|
try:
|
|
self._get_cache_data('%s/meta-data/' % self._metadata_version)
|
|
return True
|
|
except Exception as ex:
|
|
LOG.exception(ex)
|
|
LOG.debug('Metadata not found at URL \'%s\'' %
|
|
CONF.maas.metadata_base_url)
|
|
return False
|
|
|
|
def _get_response(self, req):
|
|
try:
|
|
return request.urlopen(req)
|
|
except error.HTTPError as ex:
|
|
if ex.code == 404:
|
|
raise base.NotExistingMetadataException()
|
|
else:
|
|
raise
|
|
|
|
def _get_oauth_headers(self, url):
|
|
client = oauth1.Client(
|
|
CONF.maas.oauth_consumer_key,
|
|
client_secret=CONF.maas.oauth_consumer_secret,
|
|
resource_owner_key=CONF.maas.oauth_token_key,
|
|
resource_owner_secret=CONF.maas.oauth_token_secret,
|
|
signature_method=oauth1.SIGNATURE_PLAINTEXT)
|
|
realm = _Realm("")
|
|
headers = client.sign(url, realm=realm)[1]
|
|
return headers
|
|
|
|
def _get_data(self, path):
|
|
norm_path = posixpath.join(CONF.maas.metadata_base_url, path)
|
|
oauth_headers = self._get_oauth_headers(norm_path)
|
|
|
|
LOG.debug('Getting metadata from: %(norm_path)s',
|
|
{'norm_path': norm_path})
|
|
req = request.Request(norm_path, headers=oauth_headers)
|
|
response = self._get_response(req)
|
|
return response.read()
|
|
|
|
def get_host_name(self):
|
|
return self._get_cache_data('%s/meta-data/local-hostname' %
|
|
self._metadata_version, decode=True)
|
|
|
|
def get_instance_id(self):
|
|
return self._get_cache_data('%s/meta-data/instance-id' %
|
|
self._metadata_version, decode=True)
|
|
|
|
def get_public_keys(self):
|
|
return self._get_cache_data('%s/meta-data/public-keys' %
|
|
self._metadata_version,
|
|
decode=True).splitlines()
|
|
|
|
def get_client_auth_certs(self):
|
|
certs_data = self._get_cache_data('%s/meta-data/x509' %
|
|
self._metadata_version,
|
|
decode=True)
|
|
pattern = r"{begin}[\s\S]+?{end}".format(
|
|
begin=x509constants.PEM_HEADER,
|
|
end=x509constants.PEM_FOOTER)
|
|
return re.findall(pattern, certs_data)
|
|
|
|
def get_user_data(self):
|
|
return self._get_cache_data('%s/user-data' % self._metadata_version)
|