setools/setools/policyrep/role.py
Chris PeBenito c146f0ab47 Restructure exceptions across the entire project.
Group all exception classes into a module for each package.  This is the
first step towards having all exceptions raised by setools libs be a child of
SEToolsException.
2015-04-01 12:10:45 -04:00

83 lines
2.1 KiB
Python

# Copyright 2014, Tresys Technology, LLC
#
# This file is part of SETools.
#
# SETools is free software: you can redistribute it and/or modify
# it under the terms of the GNU Lesser General Public License as
# published by the Free Software Foundation, either version 2.1 of
# the License, or (at your option) any later version.
#
# SETools 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 Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public
# License along with SETools. If not, see
# <http://www.gnu.org/licenses/>.
#
from . import exception
from . import qpol
from . import symbol
from . import typeattr
def role_factory(qpol_policy, name):
"""Factory function for creating Role objects."""
if isinstance(name, qpol.qpol_role_t):
return Role(qpol_policy, name)
try:
symbol = qpol.qpol_role_t(qpol_policy, name)
except ValueError:
raise exception.InvalidRole("{0} is not a valid role".format(name))
return Role(qpol_policy, symbol)
class BaseRole(symbol.PolicySymbol):
"""Role/role attribute base class."""
def expand(self):
raise NotImplementedError
def types(self):
raise NotImplementedError
class Role(BaseRole):
"""A role."""
def expand(self):
"""Generator that expands this into its member roles."""
yield self
def types(self):
"""Generator which yields the role's set of types."""
for type_ in self.qpol_symbol.type_iter(self.policy):
yield typeattr.type_or_attr_factory(self.policy, type_)
def statement(self):
types = list(str(t) for t in self.types())
stmt = "role {0}".format(self)
if (len(types) > 1):
stmt += " types {{ {0} }};".format(' '.join(types))
else:
try:
stmt += " types {0};".format(types[0])
except IndexError:
stmt += ";"
return stmt
class RoleAttribute(BaseRole):
"""A role attribute."""
pass