CasperSecurity
o
h9Ra( � @ s d Z ddlZddlmZ ddlmZmZmZmZm Z m
Z
m Z ddl m
Z
mZ ddlZddlZddlmZ ddlmZmZ ded efd
d �ZG d d
� d
�ZG dd� d�ZG dd� de�ZG dd� de�ZG dd� dee
�Z G dd� dee
�Z G dd� d�Z e ee ej e ��ej e < dS )zJOSE utilities.� N)�
ModuleType)�Union�Any�Callable�Iterator�Tuple�List�cast)�Hashable�Mapping)�crypto)�ec�rsa�func�returnc C s t t�| ��S �N)� classmethod�abc�abstractmethod)r � r �-/usr/lib/python3/dist-packages/josepy/util.py�abstractclassmethod � r c @ s� e Zd ZdZdeejejf ddfdd�Zde de
fdd �Z ej fd
e
defd d �Zd
e
defdd�Zde
fdd�Zde fdd�ZdS )�ComparableX509z�Wrapper for OpenSSL.crypto.X509** objects that supports __eq__.
:ivar wrapped: Wrapped certificate or certificate request.
:type wrapped: `OpenSSL.crypto.X509` or `OpenSSL.crypto.X509Req`.
�wrappedr Nc C s& t |tj�st |tj�sJ �|| _d S r )�
isinstancer �X509�X509Reqr ��selfr r r r �__init__ s �
zComparableX509.__init__�namec C � t | j|�S r )�getattrr �r r! r r r � __getattr__ � zComparableX509.__getattr__�filetypec C s* t | jtj�rt�|| j�S t�|| j�S )aN Dumps the object into a buffer with the specified encoding.
:param int filetype: The desired encoding. Should be one of
`OpenSSL.crypto.FILETYPE_ASN1`,
`OpenSSL.crypto.FILETYPE_PEM`, or
`OpenSSL.crypto.FILETYPE_TEXT`.
:returns: Encoded X509 object.
:rtype: bytes
)r r r r �dump_certificate�dump_certificate_request)r r' r r r �_dump" s zComparableX509._dump�otherc C s t || j�stS | �� |�� kS r )r � __class__�NotImplementedr* �r r+ r r r �__eq__4 s zComparableX509.__eq__c C s t | j| �� f�S r )�hashr, r* �r r r r �__hash__: � zComparableX509.__hash__c C � d� | jj| j�S �Nz <{0}({1!r})>)�formatr, �__name__r r1 r r r �__repr__= r3 zComparableX509.__repr__)r7 �
__module__� __qualname__�__doc__r r r r r �strr r% �
FILETYPE_ASN1�int�bytesr* �boolr/ r2 r8 r r r r r s r c @ s� e Zd ZU dZeZeg ef ed< de e
j e
j e
je
jf fdd�Zdedefdd �Zd
edefd d �Zdefd
d�Zddd�ZdS )�
ComparableKeyznComparable wrapper for ``cryptography`` keys.
See https://github.com/pyca/cryptography/issues/2122.
r2 r c C s
|| _ d S r )�_wrappedr r r r r I s
zComparableKey.__init__r! r c C r" r )r# rB r$ r r r r% Q r& zComparableKey.__getattr__r+ c C s\ t || j�r| jj|jjurtS t| jd�r | �� |�� kS t| jd�r,| �� |�� kS tS )N�private_numbers�public_numbers)r r, rB r- �hasattrrC rD r. r r r r/ T s zComparableKey.__eq__c C r4 r5 )r6 r, r7 rB r1 r r r r8 ` r3 zComparableKey.__repr__c C s( t | jtjtjf�r | S | �| j�� �S )zGet wrapped public key.)r rB r � RSAPublicKeyWithSerializationr
�'EllipticCurvePublicKeyWithSerializationr, �
public_keyr1 r r r rH c s
�zComparableKey.public_keyN)r rA )r7 r9 r: r; r- r2 r r> �__annotations__r r � RSAPrivateKeyWithSerializationrF r
�(EllipticCurvePrivateKeyWithSerializationrG r r< r r% r@ r/ r8 rH r r r r rA A s
��
� rA c @ � e Zd ZdZdefdd�ZdS )�ComparableRSAKeyz�Wrapper for ``cryptography`` RSA keys.
Wraps around:
- :class:`~cryptography.hazmat.primitives.asymmetric.rsa.RSAPrivateKey`
- :class:`~cryptography.hazmat.primitives.asymmetric.rsa.RSAPublicKey`
r c C st t | jtj�r"| �� }|j}t| j|j|j |j
|j |j |j
|jf�S t | jtj�r7| �� }t| j|j
|jf�S t� �r )r rB r rJ rC rD r0 r, �p�q�dmp1�dmq1�iqmp�n�erF �NotImplementedError�r �priv�pubr r r r2 v s �zComparableRSAKey.__hash__N�r7 r9 r: r; r> r2 r r r r rM l s rM c @ rL )�ComparableECKeyz�Wrapper for ``cryptography`` RSA keys.
Wraps around:
- :class:`~cryptography.hazmat.primitives.asymmetric.ec.EllipticCurvePrivateKey`
- :class:`~cryptography.hazmat.primitives.asymmetric.ec.EllipticCurvePublicKey`
r c C sp t | jtj�r | �� }|j}t| j|jj |j
|j |j f�S t | jtj
�r5| �� }t| j|jj |j
|j f�S t� �r )r rB r
rK rC rD r0 r, �curver! �x�y�
private_valuerG rU rV r r r r2 � s zComparableECKey.__hash__NrY r r r r rZ � s rZ c @ s� e Zd ZU dZdZeedf ed< deddfdd �Z dedd fd
d �Z
d edefd
d�Z de e fdd�Z
defdd�Zdefdd�Zdededdfdd�Zdefdd�ZdS )� ImmutableMapz5Immutable key to value mapping with attribute access.r .� __slots__�kwargsr Nc K s` t |�t | j�kr td�d�| j�|rd�|����d���| jD ] }t�| ||�|�� q!d S )NzA__init__() takes exactly the following arguments: {0} ({1} given)�, �none)�setr` � TypeErrorr6 �join�object� __setattr__�pop)r ra �slotr r r r � s ����
�zImmutableMap.__init__c K s i | �|�}t | �di |��S )zReturn updated map.Nr )�type)r ra �itemsr r r �update� s zImmutableMap.update�keyc C s$ zt | |�W S ty t|��w r )r# �AttributeError�KeyError�r rn r r r � __getitem__� �
�zImmutableMap.__getitem__c C �
t | j�S r )�iterr` r1 r r r �__iter__� �
zImmutableMap.__iter__c C rt r )�lenr` r1 r r r �__len__� rw zImmutableMap.__len__c s t t� fdd�� jD ���S )Nc 3 s � | ]}t � |�V qd S r )r# )�.0rj r1 r r � <genexpr>� s � z(ImmutableMap.__hash__.<locals>.<genexpr>)r0 �tupler` r1 r r1 r r2 � s zImmutableMap.__hash__r! �valuec C � t d���Nzcan't set attribute�ro �r r! r} r r r rh � � zImmutableMap.__setattr__c C s$ d� | jjd�dd� | �� D ���S )Nz{0}({1})rb c s � � | ] \}}d � ||�V qdS �z {0}={1!r}N�r6 �rz rn r} r r r r{ � s
�
�
�z(ImmutableMap.__repr__.<locals>.<genexpr>)r6 r, r7 rf rl r1 r r r r8 � s
�zImmutableMap.__repr__)r7 r9 r: r; r` r r< rI r r rm rr r rv r> ry r2 rh r8 r r r r r_ � s
r_ c @ s� e Zd ZdZdZdededdfdd�Zd edefd
d �Zde e fd d
�Z
de fdd�Z de
e
eef df fdd�Zde fdd�Zdedefdd�Zdededdfdd�Zdefdd �ZdS ) �
frozendictzFrozen dictionary.)�_items�_keys�argsra r Nc O sh |r |s t |�}nt|�dkrt|d t�r|d }nt� �t�| d|� t�| dtt|� � ��� d S )N� r r� r� )
�dictrx r r re rg rh r| �sorted�keys)r r� ra rl r r r r � s
zfrozendict.__init__rn c C s
| j | S r )r� rq r r r rr � rw zfrozendict.__getitem__c C rt r )ru r� r1 r r r rv � rw zfrozendict.__iter__c C rt r )rx r� r1 r r r ry � rw zfrozendict.__len__.c s t � fdd�� jD ��S )Nc 3 s � | ] }|� | fV qd S r r )rz rn r1 r r r{ � s � z+frozendict._sorted_items.<locals>.<genexpr>)r| r� r1 r r1 r �
_sorted_items� s zfrozendict._sorted_itemsc C s t | �� �S r )r0 r� r1 r r r r2 � r& zfrozendict.__hash__r! c C s$ z| j | W S ty t|��w r )r� rp ro r$ r r r r% � rs zfrozendict.__getattr__r} c C r~ r r� r� r r r rh � r� zfrozendict.__setattr__c C s d� d�dd� | �� D ���S )Nzfrozendict({0})rb c s r� r� r� r� r r r r{ � s � � �z&frozendict.__repr__.<locals>.<genexpr>)r6 rf r� r1 r r r r8 � s
�zfrozendict.__repr__)r7 r9 r: r; r` r r r< rr r rv r> ry r r� r2 r% rh r8 r r r r r� � s
r� c @ sn e Zd ZdZdeddfdd�Zdedefdd �Zded
eddfd d �Z deddfd
d�Z
de e fdd�Z dS )�_UtilDeprecationModulez�
Internal class delegating to a module, and displaying warnings when attributes
related to the deprecated "abstractclassmethod" attributes in the josepy.util module.
�moduler Nc C s || j d< d S �N�_module)�__dict__)r r� r r r r � r z _UtilDeprecationModule.__init__�attrc C s$ |dkr t jdtdd� t| j|�S )Nr z�The abstractclassmethod attribute in josepy.util is deprecated and will be removed soon. Please use the built-in decorators @classmethod and @abc.abstractmethod together instead.� )�
stacklevel)�warnings�warn�DeprecationWarningr# r� �r r� r r r r% s
� z"_UtilDeprecationModule.__getattr__r} c C s t | j||� d S r )�setattrr� )r r� r} r r r rh r3 z"_UtilDeprecationModule.__setattr__c C s t | j|� d S r )�delattrr� r� r r r � __delattr__ � z"_UtilDeprecationModule.__delattr__c C s dgt | j� S r� )�dirr� r1 r r r �__dir__ r� z _UtilDeprecationModule.__dir__)
r7 r9 r: r; r r r< r r% rh r� r r� r r r r r� � s r� )!r; r �typesr �typingr r r r r r r �collections.abcr
r �sysr� �OpenSSLr �)cryptography.hazmat.primitives.asymmetricr
r r r r rA rM rZ r_ r� r� �modulesr7 r r r r �<module> s$ $ /+-2