o

�N�g�6�@s
dZd
dl
Z
d
dlZd
dlZd
dlZd
dlZd
dlZd
dlZd
dl	Zd
dl
mZ
ddd�Ze�
e�
Zdd�Zd	d
�Zdd�Zd+dd�
Zd+dd�
Zdd�Zdd�Zd+dd�
Zd+dd�
Zdd�Zdd�Zd,dd�
Zd,d d!�
Zd-d#d$�
Zd.d%d&�
Zd/d'd(�
Zd0d)d*�
Z dS)1a?
Wheel system wrapper for the Salt key system to be used in interactions with
the Salt Master programmatically.

The key module for the wheel system is meant to provide an internal interface
for other Salt systems to interact with the Salt Master. The following usage
examples assume that a WheelClient is available:

.. code-block:: python

    import salt.config
    import salt.wheel
    opts = salt.config.master_config('/etc/salt/master')
    wheel = salt.wheel.WheelClient(opts)

Note that importing and using the ``WheelClient`` must be performed on the same
machine as the Salt Master and as the same user that runs the Salt Master,
unless :conf_master:`external_auth` is configured and the user is authorized
to execute wheel functions.

The function documentation starts with the ``wheel`` reference from the code
sample above and use the :py:class:`WheelClient` functions to show how they can
be called from a Python interpreter.

The wheel key functions can also be called via a ``salt`` command at the CLI
using the :mod:`saltutil execution module <salt.modules.saltutil>`.
�N)
�clean�list�print)�list_�key_strc
C�:tj
�t�
�
}
|
�|�
Wd
�
S1sw



Y
d
S)a�

    List all the keys under a named status. Returns a dictionary.

    match
        The type of keys to list. The ``pre``, ``un``, and ``unaccepted``
        options will list unaccepted/unsigned keys. ``acc`` or ``accepted`` will
        list accepted/signed keys. ``rej`` or ``rejected`` will list rejected keys.
        Finally, ``all`` will list all keys.

    .. code-block:: python

        >>> wheel.cmd('key.list', ['accepted'])
        {'minions': ['minion1', 'minion2', 'minion3']}
    N)�salt�key�get_key�__opts__Zlist_status��match�skey�r�B/opt/saltstack/salt/lib/python3.10/site-packages/salt/wheel/key.pyr0s
$�rc
Cs8tj
�t�
�}|��Wd
�
S1sw



Y
d
S)a�

    List all the keys. Returns a dictionary containing lists of the minions in
    each salt-key category, including ``minions``, ``minions_rejected``,
    ``minions_denied``, etc. Returns a dictionary.

    .. code-block:: python

        >>> wheel.cmd('key.list_all')
        {'local': ['master.pem', 'master.pub'], 'minions_rejected': [],
        'minions_denied': [], 'minions_pre': [],
        'minions': ['minion1', 'minion2', 'minion3']}
    N)rr	r
rZall_keys)
rrrr�list_allCs

$�rc
Cr)z1
    List all the keys based on a glob match
    N)rr	r
r�
name_matchrrrrrTs
$�rFcC�@tj
�t�
�}|j||
|d
�Wd�
S1sw



Y
dS)a
    Accept keys based on a glob match. Returns a dictionary.

    match
        The glob match of keys to accept.

    include_rejected
        To include rejected keys in the match along with pending keys, set this
        to ``True``. Defaults to ``False``.

    include_denied
        To include denied keys in the match along with pending keys, set this
        to ``True``. Defaults to ``False``.

    .. code-block:: python

        >>> wheel.cmd('key.accept', ['minion1'])
        {'minions': ['minion1']}
    )�include_rejected�include_deniedN�rr	r
r�accept�r
rrrrrrr\�


�$�rcCr)a`
    Accept keys based on a dict of keys. Returns a dictionary.

    match
        The dictionary of keys to accept.

    include_rejected
        To include rejected keys in the match along with pending keys, set this
        to ``True``. Defaults to ``False``.

        .. versionadded:: 2016.3.4

    include_denied
        To include denied keys in the match along with pending keys, set this
        to ``True``. Defaults to ``False``.

        .. versionadded:: 2016.3.4

    Example to move a list of keys from the ``minions_pre`` (pending) directory
    to the ``minions`` (accepted) directory:

    .. code-block:: python

        >>> wheel.cmd('key.accept_dict',
        {
            'minions_pre': [
                'jerry',
                'stuart',
                'bob',
            ],
        })
        {'minions': ['jerry', 'stuart', 'bob']}
    )�
match_dictrrNrrrrr�accept_dictvs"



�$�rc
Cr)a-

    Delete keys based on a glob match. Returns a dictionary.

    match
        The glob match of keys to delete.

    .. code-block:: python

        >>> wheel.cmd_async({'fun': 'key.delete', 'match': 'minion1'})
        {'jid': '20160826201244808521', 'tag': 'salt/wheel/20160826201244808521'}
    N�rr	r
rZ
delete_keyrrrr�delete�s
$�rc
Cs<tj
�t�
�}
|
j|d
�
Wd�
S1sw



Y
dS)a�

    Delete keys based on a dict of keys. Returns a dictionary.

    match
        The dictionary of keys to delete.

    .. code-block:: python

        >>> wheel.cmd_async({'fun': 'key.delete_dict',
        'match': {
            'minions': [
                'jerry',
                'stuart',
                'bob',
            ],
        }})
        {'jid': '20160826201244808521', 'tag': 'salt/wheel/20160826201244808521'}
    )
rNrrrrr�delete_dict�s

$�rcCr)aM
    Reject keys based on a glob match. Returns a dictionary.

    match
        The glob match of keys to reject.

    include_accepted
        To include accepted keys in the match along with pending keys, set this
        to ``True``. Defaults to ``False``.

    include_denied
        To include denied keys in the match along with pending keys, set this
        to ``True``. Defaults to ``False``.

    .. code-block:: python

        >>> wheel.cmd_async({'fun': 'key.reject', 'match': 'minion1'})
        {'jid': '20160826201244808521', 'tag': 'salt/wheel/20160826201244808521'}
    )�include_acceptedrN�rr	r
r�reject�r
rrrrrrr!�rr!cCr)a
    Reject keys based on a dict of keys. Returns a dictionary.

    match
        The dictionary of keys to reject.

    include_accepted
        To include accepted keys in the match along with pending keys, set this
        to ``True``. Defaults to ``False``.

        .. versionadded:: 2016.3.4

    include_denied
        To include denied keys in the match along with pending keys, set this
        to ``True``. Defaults to ``False``.

        .. versionadded:: 2016.3.4

    .. code-block:: python

        >>> wheel.cmd_async({'fun': 'key.reject_dict',
        'match': {
            'minions': [
                'jerry',
                'stuart',
                'bob',
            ],
        }})
        {'jid': '20160826201244808521', 'tag': 'salt/wheel/20160826201244808521'}
    )rrrNr r"rrr�reject_dict�s



�$�r#c
Cr)a_

    Return information about the key. Returns a dictionary.

    match
        The key to return information about.

    .. code-block:: python

        >>> wheel.cmd('key.key_str', ['minion1'])
        {'minions': {'minion1': '-----BEGIN PUBLIC KEY-----\nMIIBIjANBgkqhkiG9w0B
        ...
        TWugEQpPt\niQIDAQAB\n-----END PUBLIC KEY-----'}}
    N)rr	r
rrrrrrr
s
$�rcCsdd
}tj
�td|�}
tjj�|
d��}tjj�	|�
��
}Wd�
n1s'w



Y
d||i
i
S)a

    Returns master's public key. Returns a dictionary

    .. code-block:: python

        >>> wheel.cmd('key.master_key_str')
        {'local': {'master.pub': '-----BEGIN PUBLIC KEY-----\nMIIBIjANBgkqhkiG9w0B
        ...
        TWugEQpPt\niQIDAQAB\n-----END PUBLIC KEY-----'}}
    �
master.pub�pki_dir�
rN�local)�os�path�joinrr�utils�files�fopen�stringutils�
to_unicode�read)�keynameZpath_to_pubkey�fp_Zkeyvaluerrr�master_key_str
s


�r3cCsL|
d
urtd}
t
j�t�
�}|�||
�Wd
�
S1sw



Y
d
S)au

    Return the matching key fingerprints. Returns a dictionary.

    match
        The key for with to retrieve the fingerprint.

    hash_type
        The hash algorithm used to calculate the fingerprint

    .. code-block:: python

        >>> wheel.cmd('key.finger', ['minion1'])
        {'minions': {'minion1': '5d:f6:79:43:5e:d4:42:3f:57:b8:45:a8:7e:a4:6e:ca'}}

    N�	hash_type)rrr	r
�finger)r
r4rrrrr5,
s



$�r5c
Cs@d
}
|dur
td}t
jjjtj�td|
�|d�}d|
|i
i
S)a$

    Return the fingerprint of the master's public key

    hash_type
        The hash algorithm used to calculate the fingerprint

    .. code-block:: python

        >>> wheel.cmd('key.finger_master')
        {'local': {'master.pub': '5d:f6:79:43:5e:d4:42:3f:57:b8:45:a8:7e:a4:6e:ca'}}
    r$Nr4r%)
Zsum_typer')rrr+�cryptZ
pem_fingerr(r)r*)r4r1Zfingerprintrrr�
finger_masterC
s


�r7�cCs
|d
urt�
t�d�
�
��}nt�|�
}ddd�}tj�	t
d||
�}d�|d
|�d�
��
}tj
j�|�
�}tj
j�|���
|d<Wd
�
n1sLw



Y
tj
j�|�
�}tj
j�|���
|d	<Wd
�
n1snw



Y
tj
j��rt�|d
�
t�|�

t�|�

|S)a
    Generate a key pair. No keys are stored on the master. A key pair is
    returned as a dict containing pub and priv keys. Returns a dictionary
    containing the ``pub`` and ``priv`` keys with their generated values.

    id\_
        Set a name to generate a key pair for use with salt. If not specified,
        a random name will be specified.

    keysize
        The size of the key pair to generate. The size must be ``2048``, which
        is the default, or greater. If set to a value less than ``2048``, the
        key size will be rounded up to ``2048``.

    .. code-block:: python

        >>> wheel.cmd('key.gen')
        {'pub': '-----BEGIN PUBLIC KEY-----\nMIIBIjANBgkqhkiG9w0BAQEFAAOCAQ8AMIIBC
        ...
        BBPfamX9gGPQTpN9e8HwcZjXQnmg8OrcUl10WHw09SDWLOlnW+ueTWugEQpPt\niQIDAQAB\n
        -----END PUBLIC KEY-----',
        'priv': '-----BEGIN RSA PRIVATE KEY-----\nMIIEpAIBAAKCAQEA42Kf+w9XeZWgguzv
        ...
        QH3/W74X1+WTBlx4R2KGLYBiH+bCCFEQ/Zvcu4Xp4bIOPtRKozEQ==\n
        -----END RSA PRIVATE KEY-----'}

    N� �)�priv�pubr%z{}.pub�
.r;r<�)�hashlib�sha512r(�urandom�	hexdigestr�filenamerr6�gen_keysr�format�rindexr+r,r-r.r/r0�platformZ
is_windows�chmod�remove)�id_�keysize�retr;r<r2rrr�genY
s"







�
�




rMcCs�t�
|�
}t||
�}tj�td
d|�}tj�|�
r|siStj	j
�|d��}|�tj	j
�|d�
�

Wd�
|S1s>w



Y
|S)a�
    Generate a key pair then accept the public key. This function returns the
    key pair in a dict, only the public key is preserved on the master. Returns
    a dictionary.

    id\_
        The name of the minion for which to generate a key pair.

    keysize
        The size of the key pair to generate. The size must be ``2048``, which
        is the default, or greater. If set to a value less than ``2048``, the
        key size will be rounded up to ``2048``.

    force
        If a public key has already been accepted for the given minion on the
        master, then the gen_accept function will return an empty dictionary
        and not create a new key. This is the default behavior. If ``force``
        is set to ``True``, then the minion's previously accepted key will be
        overwritten.

    .. code-block:: python

        >>> wheel.cmd('key.gen_accept', ['foo'])
        {'pub': '-----BEGIN PUBLIC KEY-----\nMIIBIjANBgkqhkiG9w0BAQEFAAOCAQ8AMIIBC
        ...
        BBPfamX9gGPQTpN9e8HwcZjXQnmg8OrcUl10WHw09SDWLOlnW+ueTWugEQpPt\niQIDAQAB\n
        -----END PUBLIC KEY-----',
        'priv': '-----BEGIN RSA PRIVATE KEY-----\nMIIEpAIBAAKCAQEA42Kf+w9XeZWgguzv
        ...
        QH3/W74X1+WTBlx4R2KGLYBiH+bCCFEQ/Zvcu4Xp4bIOPtRKozEQ==\n
        -----END RSA PRIVATE KEY-----'}

    We can now see that the ``foo`` minion's key has been accepted by the master:

    .. code-block:: python

        >>> wheel.cmd('key.list', ['accepted'])
        {'minions': ['foo', 'minion1', 'minion2', 'minion3']}
    r%Zminionszw+r<N)r�idrMr(r)r*r�isfilerr+r,r-�writer.Zto_str)rJrK�forcerLZacc_pathr2rrr�
gen_accept�
s
(







��rRcCs@tj
�t�
�}|�||
||�Wd
�
S1sw



Y
d
S)z,
    Generate minion RSA public keypair
    N)rr	r
rrD)Zkeydirr1rK�userrrrrrD�
s
$�rDcCsBtj
�t�
�}|�||
|||�Wd
�
S1sw



Y
d
S)z.
    Generate master public-key-signature
    N)rr	r
rZgen_keys_signature)r;r<Zsignature_pathZauto_createrKrrrr�
gen_signature�
s
$�rT)FF)
N)Nr8)r8F)NNNN)FN)!�__doc__r?�loggingr(Z
salt.cryptrZsalt.keyZsalt.utils.cryptZsalt.utils.filesZsalt.utils.platformZsalt.utils.sanitizersrZ__func_alias__�	getLogger�__name__�logrrrrrrrr!r#rr3r5r7rMrRrDrTrrrr�<module>
s>







�


*

'



2
2