REDROOM

Navigation

Breadcrumbs

/ / lib64 / python3.6 / site-packages / setools

Create

Create File | Create Folder | Upload Files | Run Command

Preview: descriptors.py Size: 7.47 KB

//lib64/python3.6/site-packages/setools/descriptors.py

# Copyright 2015, Tresys Technology, LLC
# Copyright 2016, 2018, Chris PeBenito <pebenito@ieee.org>
#
# 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/>.
#
"""
SETools descriptors.

These classes override how a class's attributes are get/set/deleted.
This is how the @property decorator works.

See https://docs.python.org/3/howto/descriptor.html
for more details.
"""

import re
from abc import ABC, abstractmethod
from collections import defaultdict
from weakref import WeakKeyDictionary

#
# Query criteria descriptors
#
# Implementation note: if the name_regex attribute value
# is changed the criteria must be reset.
#


class CriteriaDescriptor:

    """
    Single item criteria descriptor.

    Keyword Parameters:
    name_regex      The name of instance's regex setting attribute;
                    used as name_regex below.  If unset,
                    regular expressions will never be used.
    lookup_function The name of the SELinuxPolicy lookup function,
                    e.g. lookup_type or lookup_boolean.
    default_value   The default value of the criteria.  The default
                    is None.
    enum_class      The class of enumeration which supports a
                    lookup class method.

    Read-only instance attribute use (obj parameter):
    policy          The instance of SELinuxPolicy
    name_regex      This attribute is read to determine if
                    the criteria should be looked up or
                    compiled into a regex.  If the attribute
                    does not exist, False is assumed.
    """

    def __init__(self, name_regex=None, lookup_function=None, default_value=None, enum_class=None):
        assert name_regex or lookup_function or enum_class, \
            "A simple attribute should be used if there is no regex, lookup function, or enum."
        assert not (lookup_function and enum_class), \
            "Lookup functions and enum classes are mutually exclusive."
        self.regex = name_regex
        self.default_value = default_value
        self.lookup_function = lookup_function
        self.enum_class = enum_class

        # use weak references so instances can be
        # garbage collected, rather than unnecessarily
        # kept around due to this descriptor.
        self.instances = WeakKeyDictionary()

    def __get__(self, obj, objtype=None):
        if obj is None:
            return self

        return self.instances.setdefault(obj, self.default_value)

    def __set__(self, obj, value):
        if not value:
            self.instances[obj] = None
        elif self.regex and getattr(obj, self.regex, False):
            self.instances[obj] = re.compile(value)
        elif self.lookup_function:
            lookup = getattr(obj.policy, self.lookup_function)
            self.instances[obj] = lookup(value)
        elif self.enum_class:
            self.instances[obj] = self.enum_class.lookup(value)
        else:
            self.instances[obj] = value


class CriteriaSetDescriptor(CriteriaDescriptor):

    """Descriptor for a set of criteria."""

    def __set__(self, obj, value):
        if not value:
            self.instances[obj] = None
        elif self.regex and getattr(obj, self.regex, False):
            self.instances[obj] = re.compile(value)
        elif self.lookup_function:
            lookup = getattr(obj.policy, self.lookup_function)
            self.instances[obj] = set(lookup(v) for v in value)
        elif self.enum_class:
            self.instances[obj] = set(self.enum_class.lookup(v) for v in value)
        else:
            self.instances[obj] = set(value)


#
# NetworkX Graph Descriptors
#
# These descriptors are used to simplify all
# of the dictionary use in the NetworkX graph.
#


class NetworkXGraphEdgeDescriptor(ABC):

    """
    Descriptor abstract base class for NetworkX graph edge attributes.

    Parameter:
    name        The edge property name

    Instance class attribute use (obj parameter):
    G           The NetworkX graph
    source      The edge's source node
    target      The edge's target node
    """

    def __init__(self, propname):
        self.name = propname

    def __get__(self, obj, objtype=None):
        if obj is None:
            return self

        try:
            return obj.G[obj.source][obj.target][self.name]
        except KeyError:
            raise AttributeError(self.name)

    @abstractmethod
    def __set__(self, obj, value):
        pass

    @abstractmethod
    def __delete__(self, obj):
        pass


class EdgeAttrDict(NetworkXGraphEdgeDescriptor):

    """A descriptor for edge attributes that are dictionaries."""

    def __set__(self, obj, value):
        # None is a special value to initialize the attribute
        if value is None:
            obj.G[obj.source][obj.target][self.name] = defaultdict(list)
        else:
            raise AttributeError("{0} dictionaries should not be assigned directly".
                                 format(self.name))

    def __delete__(self, obj):
        obj.G[obj.source][obj.target][self.name].clear()


class EdgeAttrIntMax(NetworkXGraphEdgeDescriptor):

    """
    A descriptor for edge attributes that are non-negative integers that always
    keep the max assigned value until re-initialized.
    """

    def __set__(self, obj, value):
        # None is a special value to initialize
        if value is None:
            obj.G[obj.source][obj.target][self.name] = 0
        else:
            current_value = obj.G[obj.source][obj.target][self.name]
            obj.G[obj.source][obj.target][self.name] = max(current_value, value)

    def __delete__(self, obj):
        obj.G[obj.source][obj.target][self.name] = 0


class EdgeAttrList(NetworkXGraphEdgeDescriptor):

    """A descriptor for edge attributes that are lists."""

    def __set__(self, obj, value):
        # None is a special value to initialize
        if value is None:
            obj.G[obj.source][obj.target][self.name] = []
        else:
            raise ValueError("{0} lists should not be assigned directly".format(self.name))

    def __delete__(self, obj):
        obj.G[obj.source][obj.target][self.name].clear()


#
# Permission map descriptors
#
class PermissionMapDescriptor:

    """
    Descriptor for Permission Map mappings.

    Parameter:
    name        The map setting name.
    validator   A callable for validating the setting.

    Instance class attribute use (obj parameter):
    perm_map    The full permission map.
    class_      The mapping's object class
    perm        The mapping's permission
    """

    def __init__(self, propname, validator):
        self.name = propname
        self.validator = validator

    def __get__(self, obj, objtype=None):
        if obj is None:
            return self

        return obj.perm_map[obj.class_][obj.perm][self.name]

    def __set__(self, obj, value):
        obj.perm_map[obj.class_][obj.perm][self.name] = self.validator(value)

    def __delete__(self, obj):
        raise AttributeError

Directory Contents

Dirs: 2 × Files: 40

Sort:

Enable drag-to-move

Name	Size	Perms	Modified	Actions
diff DIR	-	drwxr-xr-x	2025-03-30 04:21:30	Edit Download recursive recursive
__pycache__ DIR	-	drwxr-xr-x	2025-03-30 04:21:30	Edit Download recursive recursive
boolquery.py	2.19 KB	lrw-r--r--	2020-04-01 14:57:49	Edit Download recursive recursive
boundsquery.py	2.35 KB	lrw-r--r--	2020-04-01 14:57:49	Edit Download recursive recursive
categoryquery.py	1.88 KB	lrw-r--r--	2020-04-01 14:57:49	Edit Download recursive recursive
commonquery.py	2.11 KB	lrw-r--r--	2020-04-01 14:57:49	Edit Download recursive recursive
constraintquery.py	5.50 KB	lrw-r--r--	2020-04-01 14:57:49	Edit Download recursive recursive
defaultquery.py	2.80 KB	lrw-r--r--	2020-04-01 14:57:49	Edit Download recursive recursive
descriptors.py	7.47 KB	lrw-r--r--	2020-04-01 14:57:49	Edit Download recursive recursive
devicetreeconquery.py	2.72 KB	lrw-r--r--	2020-04-01 14:57:49	Edit Download recursive recursive
dta.py	21.40 KB	lrw-r--r--	2023-10-15 02:31:39	Edit Download recursive recursive
exception.py	5.97 KB	lrw-r--r--	2020-04-01 14:57:49	Edit Download recursive recursive
fsusequery.py	3.36 KB	lrw-r--r--	2020-04-01 14:57:49	Edit Download recursive recursive
genfsconquery.py	3.68 KB	lrw-r--r--	2020-04-01 14:57:49	Edit Download recursive recursive
ibendportconquery.py	3.52 KB	lrw-r--r--	2020-04-01 14:57:49	Edit Download recursive recursive
ibpkeyconquery.py	5.22 KB	lrw-r--r--	2020-04-01 14:57:49	Edit Download recursive recursive
infoflow.py	14.59 KB	lrw-r--r--	2023-10-15 02:31:39	Edit Download recursive recursive
initsidquery.py	2.79 KB	lrw-r--r--	2020-04-01 14:57:49	Edit Download recursive recursive
iomemconquery.py	4.44 KB	lrw-r--r--	2020-04-01 14:57:49	Edit Download recursive recursive
ioportconquery.py	4.46 KB	lrw-r--r--	2020-04-01 14:57:49	Edit Download recursive recursive
mixins.py	7.14 KB	lrw-r--r--	2020-04-01 14:57:49	Edit Download recursive recursive
mlsrulequery.py	4.59 KB	lrw-r--r--	2020-04-01 14:57:49	Edit Download recursive recursive
netifconquery.py	2.93 KB	lrw-r--r--	2020-04-01 14:57:49	Edit Download recursive recursive
nodeconquery.py	4.16 KB	lrw-r--r--	2020-04-01 14:57:49	Edit Download recursive recursive
objclassquery.py	3.78 KB	lrw-r--r--	2020-04-01 14:57:49	Edit Download recursive recursive
pcideviceconquery.py	3.04 KB	lrw-r--r--	2020-04-01 14:57:49	Edit Download recursive recursive
permmap.py	16.10 KB	lrw-r--r--	2020-04-01 14:57:49	Edit Download recursive recursive
perm_map	87.16 KB	lrw-r--r--	2020-04-01 14:57:49	Edit Download recursive recursive
pirqconquery.py	2.95 KB	lrw-r--r--	2020-04-01 14:57:49	Edit Download recursive recursive
polcapquery.py	1.64 KB	lrw-r--r--	2020-04-01 14:57:49	Edit Download recursive recursive
policyrep.cpython-36m-x86_64-linux-gnu.so	1.87 MB	lrwxr-xr-x	2023-10-15 02:32:32	Edit Download recursive recursive
portconquery.py	5.04 KB	lrw-r--r--	2020-04-01 14:57:49	Edit Download recursive recursive
query.py	1.69 KB	lrw-r--r--	2020-04-01 14:57:49	Edit Download recursive recursive
rbacrulequery.py	5.61 KB	lrw-r--r--	2020-04-01 14:57:49	Edit Download recursive recursive
rolequery.py	2.48 KB	lrw-r--r--	2020-04-01 14:57:49	Edit Download recursive recursive
sensitivityquery.py	2.65 KB	lrw-r--r--	2020-04-01 14:57:49	Edit Download recursive recursive
terulequery.py	9.11 KB	lrw-r--r--	2020-04-01 14:57:49	Edit Download recursive recursive
typeattrquery.py	2.62 KB	lrw-r--r--	2020-04-01 14:57:49	Edit Download recursive recursive
typequery.py	3.43 KB	lrw-r--r--	2020-04-01 14:57:49	Edit Download recursive recursive
userquery.py	4.68 KB	lrw-r--r--	2020-04-01 14:57:49	Edit Download recursive recursive
util.py	5.32 KB	lrw-r--r--	2020-04-01 14:57:49	Edit Download recursive recursive
__init__.py	2.79 KB	lrw-r--r--	2023-10-15 02:31:39	Edit Download recursive recursive