# Copyright 2014-2015, Tresys Technology, LLC
#
# SPDX-License-Identifier: LGPL-2.1-only
#
import logging
import re
from typing import Iterable, Optional
from .descriptors import CriteriaSetDescriptor
from .mixins import MatchAlias, MatchName
from .policyrep import Type
from .query import PolicyQuery
from .util import match_regex_or_set
class TypeQuery(MatchAlias, MatchName, PolicyQuery):
"""
Query SELinux policy types.
Parameter:
policy The policy to query.
Keyword Parameters/Class attributes:
name The type name to match.
name_regex If true, regular expression matching
will be used on the type names.
alias The alias name to match.
alias_regex If true, regular expression matching
will be used on the alias names.
attrs The attribute to match.
attrs_equal If true, only types with attribute sets
that are equal to the criteria will
match. Otherwise, any intersection
will match.
attrs_regex If true, regular expression matching
will be used on the attribute names instead
of set logic.
permissive The permissive state to match. If this
is None, the state is not matched.
"""
attrs = CriteriaSetDescriptor("attrs_regex", "lookup_typeattr")
attrs_regex: bool = False
attrs_equal: bool = False
_permissive: Optional[bool] = None
@property
def permissive(self) -> Optional[bool]:
return self._permissive
@permissive.setter
def permissive(self, value) -> None:
if value is None:
self._permissive = None
else:
self._permissive = bool(value)
def __init__(self, policy, **kwargs) -> None:
super(TypeQuery, self).__init__(policy, **kwargs)
self.log = logging.getLogger(__name__)
def results(self) -> Iterable[Type]:
"""Generator which yields all matching types."""
self.log.info("Generating type results from {0.policy}".format(self))
self._match_name_debug(self.log)
self._match_alias_debug(self.log)
self.log.debug("Attrs: {0.attrs!r}, regex: {0.attrs_regex}, "
"eq: {0.attrs_equal}".format(self))
self.log.debug("Permissive: {0.permissive}".format(self))
for t in self.policy.types():
if not self._match_name(t):
continue
if not self._match_alias(t):
continue
if self.attrs and not match_regex_or_set(
set(t.attributes()),
self.attrs,
self.attrs_equal,
self.attrs_regex):
continue
if self.permissive is not None and t.ispermissive != self.permissive:
continue
yield t
