125 lines
4.4 KiB
Python
125 lines
4.4 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 time
|
|
import urllib2
|
|
|
|
from oauth import oauth
|
|
from oslo.config import cfg
|
|
|
|
from cloudbaseinit.metadata.services import base
|
|
from cloudbaseinit.openstack.common import log as logging
|
|
from cloudbaseinit.utils.windows import x509
|
|
|
|
opts = [
|
|
cfg.StrOpt('maas_metadata_url', default=None,
|
|
help='The base URL for MaaS metadata'),
|
|
cfg.StrOpt('maas_oauth_consumer_key', default="",
|
|
help='The MaaS OAuth consumer key'),
|
|
cfg.StrOpt('maas_oauth_consumer_secret', default="",
|
|
help='The MaaS OAuth consumer secret'),
|
|
cfg.StrOpt('maas_oauth_token_key', default="",
|
|
help='The MaaS OAuth token key'),
|
|
cfg.StrOpt('maas_oauth_token_secret', default="",
|
|
help='The MaaS OAuth token secret'),
|
|
]
|
|
|
|
CONF = cfg.CONF
|
|
CONF.register_opts(opts)
|
|
|
|
LOG = logging.getLogger(__name__)
|
|
|
|
|
|
class MaaSHttpService(base.BaseMetadataService):
|
|
_METADATA_LATEST = 'latest'
|
|
|
|
def __init__(self):
|
|
super(MaaSHttpService, self).__init__()
|
|
self._enable_retry = True
|
|
self._metadata_version = self._METADATA_LATEST
|
|
|
|
def load(self):
|
|
super(MaaSHttpService, self).load()
|
|
|
|
if not CONF.maas_metadata_url:
|
|
LOG.debug('MaaS metadata url not set')
|
|
else:
|
|
try:
|
|
self._get_data('latest/meta-data/')
|
|
return True
|
|
except Exception, ex:
|
|
LOG.exception(ex)
|
|
LOG.debug('Metadata not found at URL \'%s\'' %
|
|
CONF.maas_metadata_url)
|
|
return False
|
|
|
|
def _get_response(self, req):
|
|
try:
|
|
return urllib2.urlopen(req)
|
|
except urllib2.HTTPError as ex:
|
|
if ex.code == 404:
|
|
raise base.NotExistingMetadataException()
|
|
else:
|
|
raise
|
|
|
|
def _get_oauth_headers(self, url):
|
|
consumer = oauth.OAuthConsumer(CONF.maas_oauth_consumer_key,
|
|
CONF.maas_oauth_consumer_secret)
|
|
token = oauth.OAuthToken(CONF.maas_oauth_token_key,
|
|
CONF.maas_oauth_token_secret)
|
|
|
|
parameters = {'oauth_version': "1.0",
|
|
'oauth_nonce': oauth.generate_nonce(),
|
|
'oauth_timestamp': int(time.time()),
|
|
'oauth_token': token.key,
|
|
'oauth_consumer_key': consumer.key}
|
|
|
|
req = oauth.OAuthRequest(http_url=url, parameters=parameters)
|
|
req.sign_request(oauth.OAuthSignatureMethod_PLAINTEXT(), consumer,
|
|
token)
|
|
|
|
return req.to_header()
|
|
|
|
def _get_data(self, path):
|
|
norm_path = posixpath.join(CONF.maas_metadata_url, path)
|
|
oauth_headers = self._get_oauth_headers(norm_path)
|
|
|
|
LOG.debug('Getting metadata from: %(norm_path)s',
|
|
{'norm_path': norm_path})
|
|
req = urllib2.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)
|
|
|
|
def get_instance_id(self):
|
|
return self._get_cache_data('%s/meta-data/instance-id' %
|
|
self._metadata_version)
|
|
|
|
def _get_list_from_text(self, text, delimiter):
|
|
return [v + delimiter for v in text.split(delimiter)]
|
|
|
|
def get_public_keys(self):
|
|
return self._get_list_from_text(
|
|
self._get_cache_data('%s/meta-data/public-keys' %
|
|
self._metadata_version), "\n")
|
|
|
|
def get_client_auth_certs(self):
|
|
return self._get_list_from_text(
|
|
self._get_cache_data('%s/meta-data/x509' % self._metadata_version),
|
|
"%s\n" % x509.PEM_FOOTER)
|