File: //proc/self/root/usr/local/lib/python3.9/site-packages/amqp/__pycache__/sasl.cpython-39.pyc
a
X>hb � @ s� d Z ddlZddlZddlmZ ddlmZ G dd� d�ZG dd� de�ZG d d
� d
e�Z d d � Z
e
� Z G d
d� de�Z G dd� de�Z
dS )z(SASL mechanisms for AMQP authentication.� N)�BytesIO)� _write_tablec @ s$ e Zd ZdZedd� �Zdd� ZdS )�SASLz�The base class for all amqp SASL authentication mechanisms.
You should sub-class this if you're implementing your own authentication.
c C s t �dS )z2Return a bytes containing the SASL mechanism name.N��NotImplementedError��self� r �3/usr/local/lib/python3.9/site-packages/amqp/sasl.py� mechanism s zSASL.mechanismc C s t �dS )z@Return the first response to a SASL challenge as a bytes object.Nr �r �
connectionr r r
�start s z
SASL.startN)�__name__�
__module__� __qualname__�__doc__�propertyr r r r r r
r
s
r c @ s( e Zd ZdZdZdd� ZdZdd� ZdS ) �PLAINzbPLAIN SASL authentication mechanism.
See https://tools.ietf.org/html/rfc4616 for details
s PLAINc C s || | _ | _d S �N��username�password�r r r r r r
�__init__" s zPLAIN.__init__r c C s^ | j d u s| jd u rtS t� }|�d� |�| j �d�� |�d� |�| j�d�� |�� S )N� zutf-8)r r �NotImplementedr �write�encode�getvalue�r r
Zlogin_responser r r
r * s
z PLAIN.startN�r r r r r r � __slots__r r r r r
r s
r c @ s( e Zd ZdZdZdd� ZdZdd� ZdS ) �AMQPLAINzhAMQPLAIN SASL authentication mechanism.
This is a non-standard mechanism used by AMQP servers.
s AMQPLAINc C s || | _ | _d S r r r r r r
r = s zAMQPLAIN.__init__r c C sF | j d u s| jd u rtS t� }t| j | jd�|jg � |�� dd � S )N)s LOGINs PASSWORD� )r r r r r r r r r r r
r E s �zAMQPLAIN.startNr! r r r r
r# 5 s
r# c sX zdd l � dd l� W n& ty: G dd� dt�} | Y S 0 G � fdd�dt�}|S d S )Nr c @ s& e Zd ZdZdZd dd�Zdd� ZdS )
z)_get_gssapi_mechanism.<locals>.FakeGSSAPIz7A no-op SASL mechanism for when gssapi isn't available.N� amqpFc S s |s t d��d S )Nz?You need to install the `gssapi` module for GSSAPI SASL supportr �r � client_name�service�rdns� fail_softr r r
r Y s �z2_get_gssapi_mechanism.<locals>.FakeGSSAPI.__init__c S s t S r )r r r r r
r ` s z/_get_gssapi_mechanism.<locals>.FakeGSSAPI.start)Nr% FF�r r r r r r r r r r r
�
FakeGSSAPIT s �
r, c s6 e Zd ZdZdZd
dd�ZdZd d
� Z� fd d �ZdS )z%_get_gssapi_mechanism.<locals>.GSSAPIzsGSSAPI SASL authentication mechanism.
See https://tools.ietf.org/html/rfc4752 for details
s GSSAPINr% Fc S s4 |rt |t�s|�d�}|| _|| _|| _|| _d S )N�ascii)�
isinstance�bytesr r' r* r( r) r& r r r
r l s
z._get_gssapi_mechanism.<locals>.GSSAPI.__init__)r' r* r( r) c S s^ |j j}| jr>|jtjtjfv r>|�� }t�|d �\}}}n|j j }t
|t �sZ|� d�}|S )Nr r- )
� transport�sockr) �family�socket�AF_INET�AF_INET6� getpeername�
gethostbyaddr�hostr. r/ r )r r
r1 �peer�hostname�_r r r
� get_hostname| s �
z2_get_gssapi_mechanism.<locals>.GSSAPI.get_hostnamec s� z`| j r � j� �| j �d�}nd }| �|�}� �d�| j|g�� jj�}� j||d�}|� d �W S � j
j j y� | j
r�t Y S � Y n0 d S )N)�name� @)r= �creds)r' � Credentials�Namer< �joinr( ZNameTypeZhostbased_serviceZSecurityContext�step�raw�miscZGSSErrorr* r )r r
r? r: r= �context��gssapir r
r � s
�
� z+_get_gssapi_mechanism.<locals>.GSSAPI.start)Nr% FF) r r r r r r r" r< r r rG r r
�GSSAPId s �
rI )rH Zgssapi.raw.misc� ImportErrorr )r, rI r rG r
�_get_gssapi_mechanismO s
5rK c @ s e Zd ZdZdZdd� ZdS )�EXTERNALz�EXTERNAL SASL mechanism.
Enables external authentication, i.e. not handled through this protocol.
Only passes 'EXTERNAL' as authentication mechanism, but no further
authentication data.
s EXTERNALc C s dS )N� r r r r r
r � s zEXTERNAL.startN)r r r r r r r r r r
rL � s rL c @ s$ e Zd ZdZdZdd� Zdd� ZdS )�RAWz�A generic custom SASL mechanism.
This mechanism takes a mechanism name and response to send to the server,
so can be used for simple custom authentication schemes.
Nc C s: t |t�sJ �t |t�s J �|| | _| _t�dt� d S )NznPassing login_method and login_response to Connection is deprecated. Please implement a SASL subclass instead.)r. r/ r �response�warnings�warn�DeprecationWarning)r r rO r r r
r � s �z RAW.__init__c C s | j S r )rO r r r r
r � s z RAW.startr+ r r r r
rN � s rN )r r3 rP �ior Zamqp.serializationr r r r# rK rI rL rN r r r r
�<module> s M