Source code for gssapi.names


import typing as t

from gssapi.raw import names as rname
from gssapi.raw import NameType
from gssapi.raw import named_tuples as tuples
from gssapi.raw import oids as roids
from gssapi import _utils

from collections.abc import MutableMapping, Iterable

rname_rfc6680 = _utils.import_gssapi_extension('rfc6680')
rname_rfc6680_comp_oid = _utils.import_gssapi_extension('rfc6680_comp_oid')


[docs]class Name(rname.Name): """A GSSAPI Name This class represents a GSSAPI name which may be used with and/or returned by other GSSAPI methods. It inherits from the low-level GSSAPI :class:`~gssapi.raw.names.Name` class, and thus may used with both low-level and high-level API methods. This class may be pickled and unpickled, as well as copied. The :class:`str` and :class:`bytes` methods may be used to retrieve the text of the name. Note: Name strings will be automatically converted to and from unicode strings as appropriate. If a method is listed as returning a :class:`str` object, it will return a unicode string. The encoding used will be python-gssapi's current encoding, which defaults to UTF-8. """ __slots__ = ('_attr_obj') def __new__( cls, base: t.Optional[t.Union[rname.Name, bytes, str]] = None, name_type: t.Optional[roids.OID] = None, token: t.Optional[bytes] = None, composite: bool = False, ) -> "Name": if token is not None: if composite: if rname_rfc6680 is None: raise NotImplementedError( "Your GSSAPI implementation does not support RFC 6680 " "(the GSSAPI naming extensions)") if rname_rfc6680_comp_oid is not None: base_name = rname.import_name(token, NameType.composite_export) displ_name = rname.display_name(base_name, name_type=True) if displ_name.name_type == NameType.composite_export: # NB(directxman12): there's a bug in MIT krb5 <= 1.13 # where GSS_C_NT_COMPOSITE_EXPORT doesn't trigger # immediate import logic. However, we can just use # the normal GSS_C_NT_EXPORT_NAME in this case. base_name = rname.import_name(token, NameType.export) else: # NB(directxman12): some older versions of MIT krb5 don't # have support for the GSS_C_NT_COMPOSITE_EXPORT, but do # support composite tokens via GSS_C_NT_EXPORT_NAME. base_name = rname.import_name(token, NameType.export) else: base_name = rname.import_name(token, NameType.export) elif isinstance(base, rname.Name): base_name = base else: if isinstance(base, str): base = base.encode(_utils._get_encoding()) base_name = rname.import_name( base, # type: ignore[arg-type] name_type) return t.cast("Name", super(Name, cls).__new__(cls, base_name)) def __init__( self, base: t.Optional[t.Union[rname.Name, bytes, str]] = None, name_type: t.Optional[roids.OID] = None, token: t.Optional[bytes] = None, composite: bool = False, ) -> None: """ The constructor can be used to "import" a name from a human readable representation, or from a token, and can also be used to convert a low-level :class:`gssapi.raw.names.Name` object into a high-level object. If a :class:`~gssapi.raw.names.Name` object from the low-level API is passed as the `base` argument, it will be converted into a high-level object. If the `token` argument is used, the name will be imported using the token. If the token was exported as a composite token, pass `composite=True`. Otherwise, a new name will be created, using the `base` argument as the human-readable string and the `name_type` argument to denote the name type. Raises: ~gssapi.exceptions.BadNameTypeError ~gssapi.exceptions.BadNameError ~gssapi.exceptions.BadMechanismError """ self._attr_obj: t.Optional[_NameAttributeMapping] if rname_rfc6680 is not None: self._attr_obj = _NameAttributeMapping(self) else: self._attr_obj = None def __str__(self) -> str: return bytes(self).decode(_utils._get_encoding()) def __unicode__(self) -> str: # Python 2 -- someone asked for unicode return self.__bytes__().decode(_utils._get_encoding()) def __bytes__(self) -> bytes: # Python 3 -- someone asked for bytes return rname.display_name(self, name_type=False).name
[docs] def display_as( self, name_type: roids.OID, ) -> str: """ Display this name as the given name type. This method attempts to display the current :class:`Name` using the syntax of the given :class:`~gssapi.raw.types.NameType`, if possible. Warning: In MIT krb5 versions below 1.13.3, this method can segfault if the name was not *originally* created with a `name_type` that was not ``None`` (even in cases when a ``name_type`` is later "added", such as via :meth:`canonicalize`). **Do not use this method unless you are sure the above conditions can never happen in your code.** Warning: In addition to the above warning, current versions of MIT krb5 do not actually fully implement this method, and it may return incorrect results in the case of canonicalized names. :requires-ext:`rfc6680` Args: name_type (~gssapi.OID): the :class:`~gssapi.raw.types.NameType` to use to display the given name Returns: str: the displayed name Raises: ~gssapi.exceptions.OperationUnavailableError """ if rname_rfc6680 is None: raise NotImplementedError("Your GSSAPI implementation does not " "support RFC 6680 (the GSSAPI naming " "extensions)") return rname_rfc6680.display_name_ext(self, name_type).decode( _utils._get_encoding())
@property def name_type(self) -> t.Optional[roids.OID]: """The :class:`~gssapi.raw.types.NameType` of this name""" return rname.display_name(self, name_type=True).name_type def __eq__( self, other: object, ) -> bool: if not isinstance(other, rname.Name): # maybe something else can compare this # to other classes, but we certainly can't return NotImplemented else: return rname.compare_name(self, other) def __ne__( self, other: object, ) -> bool: return not self.__eq__(other) def __repr__(self) -> str: disp_res = rname.display_name(self, name_type=True) return "Name({name!r}, {name_type})".format( name=disp_res.name, name_type=disp_res.name_type)
[docs] def export( self, composite: bool = False, ) -> bytes: """Export this name as a token. This method exports the name into a byte string which can then be imported by using the `token` argument of the constructor. Args: composite (bool): whether or not use to a composite token -- :requires-ext:`rfc6680` Returns: bytes: the exported name in token form Raises: ~gssapi.exceptions.MechanismNameRequiredError ~gssapi.exceptions.BadNameTypeError ~gssapi.exceptions.BadNameError """ if composite: if rname_rfc6680 is None: raise NotImplementedError("Your GSSAPI implementation does " "not support RFC 6680 (the GSSAPI " "naming extensions)") return rname_rfc6680.export_name_composite(self) else: return rname.export_name(self)
[docs] def canonicalize( self, mech: roids.OID ) -> "Name": """Canonicalize a name with respect to a mechanism. This method returns a new :class:`Name` that is canonicalized according to the given mechanism. Args: mech (~gssapi.OID): the :class:`MechType` to use Returns: Name: the canonicalized name Raises: ~gssapi.exceptions.BadMechanismError ~gssapi.exceptions.BadNameTypeError ~gssapi.exceptions.BadNameError """ return type(self)(rname.canonicalize_name(self, mech))
def __copy__(self) -> "Name": return type(self)(rname.duplicate_name(self)) def __deepcopy__( self, memo: t.Dict, ) -> "Name": return type(self)(rname.duplicate_name(self)) def _inquire( self, **kwargs: t.Any, ) -> tuples.InquireNameResult: """Inspect this name for information. This method inspects the name for information. If no keyword arguments are passed, all available information is returned. Otherwise, only the keyword arguments that are passed and set to `True` are returned. Args: mech_name (bool): get whether this is a mechanism name, and, if so, the associated mechanism attrs (bool): get the attributes names for this name Returns: InquireNameResult: the results of the inquiry, with unused fields set to None Raises: ~gssapi.exceptions.GSSError """ if rname_rfc6680 is None: raise NotImplementedError("Your GSSAPI implementation does not " "support RFC 6680 (the GSSAPI naming " "extensions)") if not kwargs: default_val = True else: default_val = False attrs = kwargs.get('attrs', default_val) mech_name = kwargs.get('mech_name', default_val) return rname_rfc6680.inquire_name(self, mech_name=mech_name, attrs=attrs) @property def is_mech_name(self) -> bool: """Whether or not this name is a mechanism name (:requires-ext:`rfc6680`) """ return self._inquire(mech_name=True).is_mech_name @property def mech(self) -> roids.OID: """The mechanism associated with this name (:requires-ext:`rfc6680`) """ return self._inquire(mech_name=True).mech @property def attributes(self) -> t.Optional[MutableMapping]: """The attributes of this name (:requires-ext:`rfc6680`) The attributes are presenting in the form of a :class:`~collections.abc.MutableMapping` (a dict-like object). Retrieved values will always be in the form of :class:`frozenset`. When assigning values, if iterables are used, they be considered to be the set of values for the given attribute. If a non-iterable is used, it will be considered a single value, and automatically wrapped in an iterable. Note: String types (includes :class:`bytes`) are not considered to be iterables in this case. """ if self._attr_obj is None: raise NotImplementedError("Your GSSAPI implementation does not " "support RFC 6680 (the GSSAPI naming " "extensions)") return self._attr_obj
class _NameAttributeMapping(MutableMapping): """Provides dict-like access to RFC 6680 Name attributes.""" def __init__( self, name: Name, ) -> None: self._name = name def __getitem__( self, key: t.Union[bytes, str], ) -> tuples.GetNameAttributeResult: if isinstance(key, str): key = key.encode(_utils._get_encoding()) res = rname_rfc6680.get_name_attribute( # type: ignore[union-attr] self._name, key) res = t.cast(tuples.GetNameAttributeResult, res) return tuples.GetNameAttributeResult(list(res.values), list(res.display_values), res.authenticated, res.complete) def __setitem__( self, key: t.Union[bytes, str], value: t.Union[ tuples.GetNameAttributeResult, t.Tuple[bytes, bool], bytes ], ) -> None: if isinstance(key, str): key = key.encode(_utils._get_encoding()) rname_rfc6680.delete_name_attribute( # type: ignore[union-attr] self._name, key) attr_value: t.List[bytes] if isinstance(value, tuples.GetNameAttributeResult): complete = value.complete attr_value = value.values elif isinstance(value, tuple) and len(value) == 2: complete = t.cast(bool, value[1]) attr_value = [t.cast(bytes, value[0])] else: complete = False if (isinstance(value, (str, bytes)) or not isinstance(value, Iterable)): # NB(directxman12): this allows us to easily assign a single # value, since that's a common case attr_value = [value] rname_rfc6680.set_name_attribute( # type: ignore[union-attr] self._name, key, attr_value, complete=complete) def __delitem__( self, key: t.Union[bytes, str], ) -> None: if isinstance(key, str): key = key.encode(_utils._get_encoding()) rname_rfc6680.delete_name_attribute( # type: ignore[union-attr] self._name, key) def __iter__(self) -> t.Iterator[bytes]: return iter(self._name._inquire(attrs=True).attrs) def __len__(self) -> int: return len(self._name._inquire(attrs=True).attrs)