o

�N�g7K�@s�dZd
dl
Z
d
dlZe�Zzd
dlZd
dlZd
dlZd
dl	Ze�
d�

Wn	ey-


Yn
we
�e
�
Zdd�ZGdd�de�Zdd	�Zdd
d�
Zdd
�ZGdd�d�Zddd�
Z				
ddd�
Zdd�Z
dd�Zdd�Zdd�ZdS) a2

Query and modify an LDAP database (alternative interface)
=========================================================

.. versionadded:: 2016.3.0

This is an alternative to the ``ldap`` interface provided by the
:py:mod:`ldapmod <salt.modules.ldapmod>` execution module.

:depends: - ``ldap`` Python module
�N�ldapc
Cstrd
SdS)z:Only load this module if the Python ldap module is presentTF)
�available_backends�rr�F/opt/saltstack/salt/lib/python3.10/site-packages/salt/modules/ldap3.py�__virtual__ s

rcs"eZ
dZd
Zd�f
dd�	Z�ZS)�	LDAPErrora
Base class of all LDAP exceptions raised by backends.

    This is only used for errors encountered while interacting with
    the LDAP server; usage errors (e.g., invalid backend name) will
    have a different type.

    :ivar cause: backend exception object, if applicable
    Ncst��
|
�

||_dS�
N)�super�__init__�cause)�self�messager�
�	__class__rrr
1s


zLDAPError.__init__r)�__name__�
__module__�__qualname__�__doc__r
�
__classcell__rrrrr's
	rc

Cstd
|��|�|�)z?Convert an ldap backend exception to an LDAPError and raise it.zexception in ldap backend: )
r)
�
errr�_convert_exception6srcCs�|
d
urd
S|
�dd�}|d
urd
S|dkr&|�
|
�dd�|
�dd��
d
S|dkr]ttj|
�dd	����}|
�d
d
�}|d
urBi}||�dg�i|�di��
�
}|�|
�dd�|�
d
Std
|d�
�
)zBind helper.N�method�simple�dn��password�saslZ	mechanismZEXTERNALZcredentials�args�kwargszunsupported bind method "z&"; supported bind methods: simple sasl)�getZ
simple_bind_s�getattrrr�lowerZsasl_interactive_bind_s�
ValueError)�
l�bindrZ
sasl_classZcredsZauthrrr�_bind;s*




 









���r%c

Csd
|�d
��d�
S)a%
Formats a string per Microsoft AD password specifications.
    The string must be enclosed in double quotes and UTF-16 encoded.
    See: https://msdn.microsoft.com/en-us/library/cc223248.aspx

    :param pwd:
       The desired password as a string

    :returns:
        A unicode string
    �
"z	utf-16-le)
�encode)
�pwdrrr�_format_unicode_passwordSsr)c@s$eZ
dZd
d�Zdd�Zdd�ZdS)�_connect_ctxcCs
|
|_dSr)
�
c)rr+rrrr
bs

z_connect_ctx.__init__c


Cs|Srr)
rrrr�	__enter__e�
z_connect_ctx.__enter__c

GsdSrr)r�excrrr�__exit__hr-z_connect_ctx.__exit__N)rrrr
r,r/rrrrr*as
r*c

Cs�
t|t
�r|S|d
ur
i}|�dd�}
|
tv
r tdd|
���
�
|�dd�}z�t�|�
}tj|_|�dd
�}|d
ur<i}i}|�	�D]3\}}|d	vrOd
||<qB|dvrh|�
ttd|���ttd|����
qB|�
ttd|���|�
qB|�d
d�r�|�
�
|�dd�r�|�
tjd�
|�
tjd�
|�dd
�}|d
ur�i}|�	�D]\}}ttd|���}	|�
|	|�
q�t||�dd
��
Wt
|�
Stjy�
}

z
t|
�

WYd
}
~
t
|�
Sd
}
~
w
w)a�Connect and optionally bind to an LDAP server.

    :param connect_spec:
        This can be an LDAP connection object returned by a previous
        call to :py:func:`connect` (in which case the argument is
        simply returned), ``None`` (in which case an empty dict is
        used), or a dict with the following keys:

        * ``'backend'``
            Optional; default depends on which Python LDAP modules are
            installed.  Name of the Python LDAP module to use.  Only
            ``'ldap'`` is supported at the moment.

        * ``'url'``
            Optional; defaults to ``'ldapi:///'``.  URL to the LDAP
            server.

        * ``'bind'``
            Optional; defaults to ``None``.  Describes how to bind an
            identity to the LDAP connection.  If ``None``, an
            anonymous connection is made.  Valid keys:

            * ``'method'``
                Optional; defaults to ``None``.  The authentication
                method to use.  Valid values include but are not
                necessarily limited to ``'simple'``, ``'sasl'``, and
                ``None``.  If ``None``, an anonymous connection is
                made.  Available methods depend on the chosen backend.

            * ``'mechanism'``
                Optional; defaults to ``'EXTERNAL'``.  The SASL
                mechanism to use.  Ignored unless the method is
                ``'sasl'``.  Available methods depend on the chosen
                backend and the server's capabilities.

            * ``'credentials'``
                Optional; defaults to ``None``.  An object specific to
                the chosen SASL mechanism and backend that represents
                the authentication credentials.  Ignored unless the
                method is ``'sasl'``.

                For the ``'ldap'`` backend, this is a dictionary.  If
                ``None``, an empty dict is used.  Keys:

                * ``'args'``
                    Optional; defaults to an empty list.  A list of
                    arguments to pass to the SASL mechanism
                    constructor.  See the SASL mechanism constructor
                    documentation in the ``ldap.sasl`` Python module.

                * ``'kwargs'``
                    Optional; defaults to an empty dict.  A dict of
                    keyword arguments to pass to the SASL mechanism
                    constructor.  See the SASL mechanism constructor
                    documentation in the ``ldap.sasl`` Python module.

            * ``'dn'``
                Optional; defaults to an empty string.  The
                distinguished name to bind.

            * ``'password'``
                Optional; defaults to an empty string.  Password for
                binding.  Ignored if the method is ``'sasl'``.

        * ``'tls'``
            Optional; defaults to ``None``.  A backend-specific object
            containing settings to override default TLS behavior.

            For the ``'ldap'`` backend, this is a dictionary.  Not all
            settings in this dictionary are supported by all versions
            of ``python-ldap`` or the underlying TLS library.  If
            ``None``, an empty dict is used.  Possible keys:

            * ``'starttls'``
                If present, initiate a TLS connection using StartTLS.
                (The value associated with this key is ignored.)

            * ``'cacertdir'``
                Set the path of the directory containing CA
                certificates.

            * ``'cacertfile'``
                Set the pathname of the CA certificate file.

            * ``'certfile'``
                Set the pathname of the certificate file.

            * ``'cipher_suite'``
                Set the allowed cipher suite.

            * ``'crlcheck'``
                Set the CRL evaluation strategy.  Valid values are
                ``'none'``, ``'peer'``, and ``'all'``.

            * ``'crlfile'``
                Set the pathname of the CRL file.

            * ``'dhfile'``
                Set the pathname of the file containing the parameters
                for Diffie-Hellman ephemeral key exchange.

            * ``'keyfile'``
                Set the pathname of the certificate key file.

            * ``'newctx'``
                If present, instruct the underlying TLS library to
                create a new TLS context.  (The value associated with
                this key is ignored.)

            * ``'protocol_min'``
                Set the minimum protocol version.

            * ``'random_file'``
                Set the pathname of the random file when
                ``/dev/random`` and ``/dev/urandom`` are not
                available.

            * ``'require_cert'``
                Set the certificate validation policy.  Valid values
                are ``'never'``, ``'hard'``, ``'demand'``,
                ``'allow'``, and ``'try'``.

        * ``'opts'``
            Optional; defaults to ``None``.  A backend-specific object
            containing options for the backend.

            For the ``'ldap'`` backend, this is a dictionary of
            OpenLDAP options to set.  If ``None``, an empty dict is
            used.  Each key is a the name of an OpenLDAP option
            constant without the ``'LDAP_OPT_'`` prefix, then
            converted to lower case.

    :returns:
        an object representing an LDAP connection that can be used as
        the ``connect_spec`` argument to any of the functions in this
        module (to avoid the overhead of making and terminating
        multiple connections).

        This object should be used as a context manager.  It is safe
        to nest ``with`` statements.

    CLI Example:

    .. code-block:: bash

        salt '*' ldap3.connect "{
            'url': 'ldaps://ldap.example.com/',
            'bind': {
                'method': 'simple',
                'dn': 'cn=admin,dc=example,dc=com',
                'password': 'secret'}
        }"
    NZbackendrz-unsupported backend or required Python modulez unavailable: �urlz	ldapi:///�tls)�starttls�newctxT)ZcrlcheckZrequire_certZ
OPT_X_TLS_r2Fr3r
�optsZOPT_r$)�
isinstancer*rrr"rZ
initializeZVERSION3Zprotocol_version�itemsZ
set_optionr �upperZstart_tls_sZOPT_X_TLS_NEWCTXZ
OPT_REFERRALSr%rr)�connect_specZbackend_namer0r#r1�vars�
k�
vr4�optrrrr�connectls^








��














�









�

��r=�subtree�(objectClass=*)c	
Cs�t|�
}t
td
|���}z|j�|
||||�}Wt	|�
Stjy+


g}Yt	|�
StjyD
}
z
t|�

WYd}~t	|�
Sd}~w
w)a�Search an LDAP database.

    :param connect_spec:
        See the documentation for the ``connect_spec`` parameter for
        :py:func:`connect`.

    :param base:
        Distinguished name of the entry at which to start the search.

    :param scope:
        One of the following:

        * ``'subtree'``
            Search the base and all of its descendants.

        * ``'base'``
            Search only the base itself.

        * ``'onelevel'``
            Search only the base's immediate children.

    :param filterstr:
        String representation of the filter to apply in the search.

    :param attrlist:
        Limit the returned attributes to those in the specified list.
        If ``None``, all attributes of each entry are returned.

    :param attrsonly:
        If non-zero, don't return any attribute values.

    :returns:
        a dict of results.  The dict is empty if there are no results.
        The dict maps each returned entry's distinguished name to a
        dict that maps each of the matching attribute names to a list
        of its values.

    CLI Example:

    .. code-block:: bash

        salt '*' ldap3.search "{
            'url': 'ldaps://ldap.example.com/',
            'bind': {
                'method': 'simple',
                'dn': 'cn=admin,dc=example,dc=com',
                'password': 'secret',
            },
        }" "base='dc=example,dc=com'"
    ZSCOPE_N)
r=r rr7r+Zsearch_sZNO_SUCH_OBJECTrr�dict)	r8�baseZscopeZ	filterstrZattrlistZ	attrsonlyr#�resultsrrrr�search8
s:


�
�

��rCc
Cs�t|�
}d
d�|�
�D�
}t�dt|
�
t|�
�
d|vr'dd�|dD�
|d<tj�|�
}z
|j�	|
|�
WdStj
yO
}
zt|�

WYd}~dSd}~w
w)	a�Add an entry to an LDAP database.

    :param connect_spec:
        See the documentation for the ``connect_spec`` parameter for
        :py:func:`connect`.

    :param dn:
        Distinguished name of the entry.

    :param attributes:
        Non-empty dict mapping each of the new entry's attributes to a
        non-empty iterable of values.

    :returns:
        ``True`` if successful, raises an exception otherwise.

    CLI Example:

    .. code-block:: bash

        salt '*' ldap3.add "{
            'url': 'ldaps://ldap.example.com/',
            'bind': {
                'method': 'simple',
                'password': 'secret',
            },
        }" "dn='dc=example,dc=com'" "attributes={'example': 'values'}"
    c
S�$i|]\}
}|
tj
j�t|�
�
�qSr��salt�utils�datar'�list��.0�attr�valsrrr�
<dictcomp>�
s
�zadd.<locals>.<dictcomp>z#adding entry: dn: %s attributes: %s�
unicodePwdc
S�g|]}
t|
�
�qSr�
r)�rK�
xrrr�
<listcomp>�
s
�zadd.<locals>.<listcomp>NT)r=r6�log�info�reprr�modlistZ
addModlistr+Zadd_srr)r8rZ
attributesr#rXrrrr�add}
s$
�


�

�

��rYc
Cs\t|�
}t
�d
t|
�
�
z	|j�|
�

WdStjy-
}
zt|�

WYd}~dSd}~w
w)a6Delete an entry from an LDAP database.

    :param connect_spec:
        See the documentation for the ``connect_spec`` parameter for
        :py:func:`connect`.

    :param dn:
        Distinguished name of the entry.

    :returns:
        ``True`` if successful, raises an exception otherwise.

    CLI Example:

    .. code-block:: bash

        salt '*' ldap3.delete "{
            'url': 'ldaps://ldap.example.com/',
            'bind': {
                'method': 'simple',
                'password': 'secret'}
        }" dn='cn=admin,dc=example,dc=com'
    zdeleting entry: dn: %sNT)	r=rUrVrWr+Zdelete_srrr)r8rr#rrrr�delete�
s


�

��rZc
Cs�t|�
}d
d�|D�
}t
|�
D]\}}|ddkr+|d|ddd�|dD�
f||<qtjjj|ddd	�}z
|j�|
|�
WdStj	yX
}
zt
|�

WYd
}~dSd
}~w
w)a�Modify an entry in an LDAP database.

    :param connect_spec:
        See the documentation for the ``connect_spec`` parameter for
        :py:func:`connect`.

    :param dn:
        Distinguished name of the entry.

    :param directives:
        Iterable of directives that indicate how to modify the entry.
        Each directive is a tuple of the form ``(op, attr, vals)``,
        where:

        * ``op`` identifies the modification operation to perform.
          One of:

          * ``'add'`` to add one or more values to the attribute

          * ``'delete'`` to delete some or all of the values from the
            attribute.  If no values are specified with this
            operation, all of the attribute's values are deleted.
            Otherwise, only the named values are deleted.

          * ``'replace'`` to replace all of the attribute's values
            with zero or more new values

        * ``attr`` names the attribute to modify

        * ``vals`` is an iterable of values to add or delete

    :returns:
        ``True`` if successful, raises an exception otherwise.

    CLI Example:

    .. code-block:: bash

        salt '*' ldap3.modify "{
            'url': 'ldaps://ldap.example.com/',
            'bind': {
                'method': 'simple',
                'password': 'secret'}
        }" dn='cn=admin,dc=example,dc=com'
        directives="('add', 'example', ['example_val'])"
    c
Ss.g|]\}
}}tt
d|
���|t|�
f�qS)
ZMOD_)r rr7rI)rK�oprLrMrrrrTs��zmodify.<locals>.<listcomp>�
rOr
c
SrPrrQrRrrrrT��T)Zto_strZpreserve_tuplesN)r=�	enumeraterFrGrH�decoder+�modify_srrr)r8rZ
directivesr#rX�idx�modrrrr�modify�
s(/�


��

�

��rdc
Cs�t|�
}d
d�|�
�D�
}dd�|�
�D�
}d|vr%dd�|dD�
|d<tj�||�}z
|j�|
|�
WdStjyN
}
zt|�

WYd}~dSd}~w
w)	a�Modify an entry in an LDAP database.

    This does the same thing as :py:func:`modify`, but with a simpler
    interface.  Instead of taking a list of directives, it takes a
    before and after view of an entry, determines the differences
    between the two, computes the directives, and executes them.

    Any attribute value present in ``before`` but missing in ``after``
    is deleted.  Any attribute value present in ``after`` but missing
    in ``before`` is added.  Any attribute value in the database that
    is not mentioned in either ``before`` or ``after`` is not altered.
    Any attribute value that is present in both ``before`` and
    ``after`` is ignored, regardless of whether that attribute value
    exists in the database.

    :param connect_spec:
        See the documentation for the ``connect_spec`` parameter for
        :py:func:`connect`.

    :param dn:
        Distinguished name of the entry.

    :param before:
        The expected state of the entry before modification.  This is
        a dict mapping each attribute name to an iterable of values.

    :param after:
        The desired state of the entry after modification.  This is a
        dict mapping each attribute name to an iterable of values.

    :returns:
        ``True`` if successful, raises an exception otherwise.

    CLI Example:

    .. code-block:: bash

        salt '*' ldap3.change "{
            'url': 'ldaps://ldap.example.com/',
            'bind': {
                'method': 'simple',
                'password': 'secret'}
        }" dn='cn=admin,dc=example,dc=com'
        before="{'example_value': 'before_val'}"
        after="{'example_value': 'after_val'}"
    c
SrDrrErJrrrrNL�$zchange.<locals>.<dictcomp>c
SrDrrErJrrrrNMrerOc
SrPrrQrRrrrrTPr]zchange.<locals>.<listcomp>NT)	r=r6rrXZ
modifyModlistr+rarr)r8r�beforeZafterr#rXrrrr�changes/


�

��rgr)r>r?Nr
)r�loggingZsalt.utils.datarF�setrrZldap.ldapobjectZldap.modlistZ	ldap.saslrY�ImportError�	getLoggerrrUr�	Exceptionrrr%r)r*r=rCrZrdrgrrrr�<module>
s>






�


P



�E3!H