201 lines
7.4 KiB
Python
201 lines
7.4 KiB
Python
# vim: tabstop=4 shiftwidth=4 softtabstop=4
|
|
|
|
#
|
|
# 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.
|
|
|
|
from heat.openstack.common import local
|
|
from heat.common import exception
|
|
from heat.common import wsgi
|
|
from heat.openstack.common import cfg
|
|
from heat.openstack.common import importutils
|
|
from heat.common import utils as heat_utils
|
|
import json
|
|
|
|
|
|
def generate_request_id():
|
|
return 'req-' + str(heat_utils.gen_uuid())
|
|
|
|
|
|
class RequestContext(object):
|
|
"""
|
|
Stores information about the security context under which the user
|
|
accesses the system, as well as additional request information.
|
|
"""
|
|
|
|
def __init__(self, auth_token=None, username=None, password=None,
|
|
aws_creds=None, aws_auth_uri=None,
|
|
service_user=None, service_password=None, tenant=None,
|
|
tenant_id=None, auth_url=None, roles=None, is_admin=False,
|
|
read_only=False, show_deleted=False,
|
|
owner_is_tenant=True, overwrite=True, **kwargs):
|
|
"""
|
|
:param overwrite: Set to False to ensure that the greenthread local
|
|
copy of the index is not overwritten.
|
|
|
|
:param kwargs: Extra arguments that might be present, but we ignore
|
|
because they possibly came in from older rpc messages.
|
|
"""
|
|
|
|
self.auth_token = auth_token
|
|
self.username = username
|
|
self.password = password
|
|
self.aws_creds = aws_creds
|
|
self.aws_auth_uri = aws_auth_uri
|
|
self.service_user = service_user
|
|
self.service_password = service_password
|
|
self.tenant = tenant
|
|
self.tenant_id = tenant_id
|
|
self.auth_url = auth_url
|
|
self.roles = roles or []
|
|
self.is_admin = is_admin
|
|
self.read_only = read_only
|
|
self._show_deleted = show_deleted
|
|
self.owner_is_tenant = owner_is_tenant
|
|
if overwrite or not hasattr(local.store, 'context'):
|
|
self.update_store()
|
|
|
|
def update_store(self):
|
|
local.store.context = self
|
|
|
|
def to_dict(self):
|
|
return {'auth_token': self.auth_token,
|
|
'username': self.username,
|
|
'password': self.password,
|
|
'aws_creds': self.aws_creds,
|
|
'aws_auth_uri': self.aws_auth_uri,
|
|
'service_user': self.service_user,
|
|
'service_password': self.service_password,
|
|
'tenant': self.tenant,
|
|
'tenant_id': self.tenant_id,
|
|
'auth_url': self.auth_url,
|
|
'roles': self.roles,
|
|
'is_admin': self.is_admin}
|
|
|
|
@classmethod
|
|
def from_dict(cls, values):
|
|
return cls(**values)
|
|
|
|
@property
|
|
def owner(self):
|
|
"""Return the owner to correlate with an image."""
|
|
return self.tenant if self.owner_is_tenant else self.user
|
|
|
|
@property
|
|
def show_deleted(self):
|
|
"""Admins can see deleted by default"""
|
|
if self._show_deleted or self.is_admin:
|
|
return True
|
|
return False
|
|
|
|
|
|
def get_admin_context(read_deleted="no"):
|
|
return RequestContext(is_admin=True)
|
|
|
|
|
|
class ContextMiddleware(wsgi.Middleware):
|
|
|
|
opts = [
|
|
cfg.BoolOpt('owner_is_tenant', default=True),
|
|
cfg.StrOpt('admin_role', default='admin'),
|
|
]
|
|
|
|
def __init__(self, app, conf, **local_conf):
|
|
self.conf = conf
|
|
self.conf.register_opts(self.opts)
|
|
|
|
# Determine the context class to use
|
|
self.ctxcls = RequestContext
|
|
if 'context_class' in local_conf:
|
|
self.ctxcls = importutils.import_class(local_conf['context_class'])
|
|
|
|
super(ContextMiddleware, self).__init__(app)
|
|
|
|
def make_context(self, *args, **kwargs):
|
|
"""
|
|
Create a context with the given arguments.
|
|
"""
|
|
kwargs.setdefault('owner_is_tenant', self.conf.owner_is_tenant)
|
|
|
|
return self.ctxcls(*args, **kwargs)
|
|
|
|
def process_request(self, req):
|
|
"""
|
|
Extract any authentication information in the request and
|
|
construct an appropriate context from it.
|
|
|
|
A few scenarios exist:
|
|
|
|
1. If X-Auth-Token is passed in, then consult TENANT and ROLE headers
|
|
to determine permissions.
|
|
|
|
2. An X-Auth-Token was passed in, but the Identity-Status is not
|
|
confirmed. For now, just raising a NotAuthenticated exception.
|
|
|
|
3. X-Auth-Token is omitted. If we were using Keystone, then the
|
|
tokenauth middleware would have rejected the request, so we must be
|
|
using NoAuth. In that case, assume that is_admin=True.
|
|
"""
|
|
headers = req.headers
|
|
|
|
try:
|
|
"""
|
|
This sets the username/password to the admin user because you
|
|
need this information in order to perform token authentication.
|
|
The real 'username' is the 'tenant'.
|
|
|
|
We should also check here to see if X-Auth-Token is not set and
|
|
in that case we should assign the user/pass directly as the real
|
|
username/password and token as None. 'tenant' should still be
|
|
the username.
|
|
"""
|
|
|
|
username = None
|
|
password = None
|
|
aws_creds = None
|
|
aws_auth_uri = None
|
|
|
|
if headers.get('X-Auth-EC2-Creds') is not None:
|
|
aws_creds = headers.get('X-Auth-EC2-Creds')
|
|
aws_auth_uri = headers.get('X-Auth-EC2-Url')
|
|
else:
|
|
# XXX: The eval here is a bit scary, it's possible someone
|
|
# could put something malicious in here I would think.
|
|
# I Haven't tested to see if WSGI stuff would escape
|
|
# everything to make this safe. However, I haven't found
|
|
# a better way to do this either.
|
|
if 'KeyStoneCreds' in req.params:
|
|
creds = eval(req.params['KeyStoneCreds'])
|
|
username = creds['username']
|
|
password = creds['password']
|
|
|
|
token = headers.get('X-Auth-Token')
|
|
service_user = headers.get('X-Admin-User')
|
|
service_password = headers.get('X-Admin-Pass')
|
|
tenant = headers.get('X-Tenant')
|
|
tenant_id = headers.get('X-Tenant-Id')
|
|
auth_url = headers.get('X-Auth-Url')
|
|
roles = headers.get('X-Roles')
|
|
except:
|
|
raise exception.NotAuthenticated()
|
|
|
|
req.context = self.make_context(auth_token=token,
|
|
tenant=tenant, tenant_id=tenant_id,
|
|
aws_creds=aws_creds,
|
|
aws_auth_uri=aws_auth_uri,
|
|
username=username,
|
|
password=password,
|
|
service_user=service_user,
|
|
service_password=service_password,
|
|
auth_url=auth_url, roles=roles,
|
|
is_admin=True)
|