**** CubicPower OpenStack Study ****
# vim: tabstop=4 shiftwidth=4 softtabstop=4
# Copyright (c) 2011 OpenStack Foundation
# All Rights Reserved.
#
#    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.
"""Common Policy Engine Implementation"""
import logging
import urllib
import urllib2
from cinder.openstack.common.gettextutils import _
from cinder.openstack.common import jsonutils
LOG = logging.getLogger(__name__)
_BRAIN = None
**** CubicPower OpenStack Study ****
def set_brain(brain):
    """Set the brain used by enforce().
    Defaults use Brain() if not set.
    """
    global _BRAIN
    _BRAIN = brain
**** CubicPower OpenStack Study ****
def reset():
    """Clear the brain used by enforce()."""
    global _BRAIN
    _BRAIN = None
**** CubicPower OpenStack Study ****
def enforce(match_list, target_dict, credentials_dict, exc=None,
            *args, **kwargs):
    """Enforces authorization of some rules against credentials.
    :param match_list: nested tuples of data to match against
        The basic brain supports three types of match lists:
            1) rules
                looks like: ``('rule:compute:get_instance',)``
                Retrieves the named rule from the rules dict and recursively
                checks against the contents of the rule.
            2) roles
                looks like: ``('role:compute:admin',)``
                Matches if the specified role is in credentials_dict['roles'].
            3) generic
                looks like: ``('tenant_id:%(tenant_id)s',)``
                Substitutes values from the target dict into the match using
                the % operator and matches them against the creds dict.
        Combining rules:
            The brain returns True if any of the outer tuple of rules
            match and also True if all of the inner tuples match. You
            can use this to perform simple boolean logic.  For
            example, the following rule would return True if the creds
            contain the role 'admin' OR the if the tenant_id matches
            the target dict AND the the creds contains the role
            'compute_sysadmin':
            ::
                {
                    "rule:combined": (
                        'role:admin',
                        ('tenant_id:%(tenant_id)s', 'role:compute_sysadmin')
                    )
                }
        Note that rule and role are reserved words in the credentials match, so
        you can't match against properties with those names. Custom brains may
        also add new reserved words. For example, the HttpBrain adds http as a
        reserved word.
    :param target_dict: dict of object properties
      Target dicts contain as much information as we can about the object being
      operated on.
    :param credentials_dict: dict of actor properties
      Credentials dicts contain as much information as we can about the user
      performing the action.
    :param exc: exception to raise
      Class of the exception to raise if the check fails.  Any remaining
      arguments passed to enforce() (both positional and keyword arguments)
      will be passed to the exception class.  If exc is not provided, returns
      False.
    :return: True if the policy allows the action
    :return: False if the policy does not allow the action and exc is not set
    """
    global _BRAIN
    if not _BRAIN:
        _BRAIN = Brain()
    if not _BRAIN.check(match_list, target_dict, credentials_dict):
        if exc:
            raise exc(*args, **kwargs)
        return False
    return True
**** CubicPower OpenStack Study ****
class Brain(object):
    """Implements policy checking."""
    _checks = {}
    @classmethod
    
**** CubicPower OpenStack Study ****
    def _register(cls, name, func):
        cls._checks[name] = func
    @classmethod
**** CubicPower OpenStack Study ****
    def load_json(cls, data, default_rule=None):
        """Init a brain using json instead of a rules dictionary."""
        rules_dict = jsonutils.loads(data)
        return cls(rules=rules_dict, default_rule=default_rule)
**** CubicPower OpenStack Study ****
    def __init__(self, rules=None, default_rule=None):
        if self.__class__ != Brain:
            LOG.warning(_("Inheritance-based rules are deprecated; use "
                          "the default brain instead of %s.") %
                        self.__class__.__name__)
        self.rules = rules or {}
        self.default_rule = default_rule
**** CubicPower OpenStack Study ****
    def add_rule(self, key, match):
        self.rules[key] = match
**** CubicPower OpenStack Study ****
    def _check(self, match, target_dict, cred_dict):
        try:
            match_kind, match_value = match.split(':', 1)
        except Exception:
            LOG.exception(_("Failed to understand rule %(match)r") % locals())
            # If the rule is invalid, fail closed
            return False
        func = None
        try:
            old_func = getattr(self, '_check_%s' % match_kind)
        except AttributeError:
            func = self._checks.get(match_kind, self._checks.get(None, None))
        else:
            LOG.warning(_("Inheritance-based rules are deprecated; update "
                          "_check_%s") % match_kind)
            func = lambda brain, kind, value, target, cred: old_func(value,
                                                                     target,
                                                                     cred)
        if not func:
            LOG.error(_("No handler for matches of kind %s") % match_kind)
            # Fail closed
            return False
        return func(self, match_kind, match_value, target_dict, cred_dict)
**** CubicPower OpenStack Study ****
    def check(self, match_list, target_dict, cred_dict):
        """Checks authorization of some rules against credentials.
        Detailed description of the check with examples in policy.enforce().
        :param match_list: nested tuples of data to match against
        :param target_dict: dict of object properties
        :param credentials_dict: dict of actor properties
        :returns: True if the check passes
        """
        if not match_list:
            return True
        for and_list in match_list:
            if isinstance(and_list, basestring):
                and_list = (and_list,)
            if all([self._check(item, target_dict, cred_dict)
                    for item in and_list]):
                return True
        return False
**** CubicPower OpenStack Study ****
class HttpBrain(Brain):
    """A brain that can check external urls for policy.
    Posts json blobs for target and credentials.
    Note that this brain is deprecated; the http check is registered
    by 
**** CubicPower OpenStack Study ****
    def decorator(func):
        # Register the function
        Brain._register(name, func)
        return func
    # If the function is given, do the registration
    if func:
        return decorator(func)
    return decorator
@register("rule")
def _check_rule(brain, match_kind, match, target_dict, cred_dict):
    """Recursively checks credentials based on the brains rules."""
    try:
        new_match_list = brain.rules[match]
    except KeyError:
        if brain.default_rule and match != brain.default_rule:
            new_match_list = ('rule:%s' % brain.default_rule,)
        else:
            return False
    return brain.check(new_match_list, target_dict, cred_dict)
@register("role")
def _check_role(brain, match_kind, match, target_dict, cred_dict):
    """Check that there is a matching role in the cred dict."""
    return match.lower() in [x.lower() for x in cred_dict['roles']]
@register('http')
def _check_http(brain, match_kind, match, target_dict, cred_dict):
    """Check http: rules by calling to a remote server.
    This example implementation simply verifies that the response is
    exactly 'True'. A custom brain using response codes could easily
    be implemented.
    """
    url = 'http:' + (match % target_dict)
    data = {'target': jsonutils.dumps(target_dict),
            'credentials': jsonutils.dumps(cred_dict)}
    post_data = urllib.urlencode(data)
    f = urllib2.urlopen(url, post_data)
    return f.read() == "True"
@register(None)
def _check_generic(brain, match_kind, match, target_dict, cred_dict):
    """Check an individual match.
    Matches look like:
        tenant:%(tenant_id)s
        role:compute:admin
    """
    # TODO(termie): do dict inspection via dot syntax
    match = match % target_dict
    if match_kind in cred_dict:
        return match == unicode(cred_dict[match_kind])
    return False
**** CubicPower OpenStack Study ****
def _check_rule(brain, match_kind, match, target_dict, cred_dict):
    """Recursively checks credentials based on the brains rules."""
    try:
        new_match_list = brain.rules[match]
    except KeyError:
        if brain.default_rule and match != brain.default_rule:
            new_match_list = ('rule:%s' % brain.default_rule,)
        else:
            return False
    return brain.check(new_match_list, target_dict, cred_dict)
@register("role")
**** CubicPower OpenStack Study ****
def _check_role(brain, match_kind, match, target_dict, cred_dict):
    """Check that there is a matching role in the cred dict."""
    return match.lower() in [x.lower() for x in cred_dict['roles']]
@register('http')
**** CubicPower OpenStack Study ****
def _check_http(brain, match_kind, match, target_dict, cred_dict):
    """Check http: rules by calling to a remote server.
    This example implementation simply verifies that the response is
    exactly 'True'. A custom brain using response codes could easily
    be implemented.
    """
    url = 'http:' + (match % target_dict)
    data = {'target': jsonutils.dumps(target_dict),
            'credentials': jsonutils.dumps(cred_dict)}
    post_data = urllib.urlencode(data)
    f = urllib2.urlopen(url, post_data)
    return f.read() == "True"
@register(None)
**** CubicPower OpenStack Study ****
def _check_generic(brain, match_kind, match, target_dict, cred_dict):
    """Check an individual match.
    Matches look like:
        tenant:%(tenant_id)s
        role:compute:admin
    """
    # TODO(termie): do dict inspection via dot syntax
    match = match % target_dict
    if match_kind in cred_dict:
        return match == unicode(cred_dict[match_kind])
    return False