a

�,�h
�@s�dZd
dl
mZmZmZmZ
d
dlZd
dlZd
dlZddl	m
Z

ddlmZ
mZmZ
ejdkrld
dlmZ
nd
d	lmZ
d
d�Zddd
�
Zdd�Zddd�
ZdS)zq
Encoding DER to PEM and decoding PEM to DER. Exports the following items:

 - armor()
 - detect()
 - unarmor()

�)�unicode_literals�division�absolute_import�print_functionN�
)
�unwrap)�	type_name�str_cls�byte_cls)
�)
�StringIO)
�BytesIOc

Cs8t|t
�sttd
t|�
��
�
|�d�
dkp6|�d�
dkS)z�
    Detect if a byte string seems to contain a PEM-encoded block

    :param byte_string:
        A byte string to look through

    :return:
        A boolean, indicating if a PEM-encoded block is contained in the byte
        string
    zC
            byte_string must be a byte string, not %s
            s
-----BEGIN���s
---- BEGIN)�
isinstancer
�	TypeErrorr�
_type_name�find)
Zbyte_string�r�?/opt/nydus/tmp/pip-target-wkfpz8uv/lib/python/asn1crypto/pem.py�detects


�rcCs,
t|
t
�sttd
t|
�
�
�
�
t|t�s:ttdt|�
��
�
|���d�
}t�}|�	d�

|�	|�

|�	d�

|r�|D]<}|�	|�d�
�

|�	d�

|�	||�d�
�

|�	d�

qt|�	d�

t
�|
�
}t|�
}d}||k�
r|�	|||d	��

|�	d�

|d	7}q�|�	d
�

|�	|�

|�	d�

|�
�S)a

    Armors a DER-encoded byte string in PEM

    :param type_name:
        A unicode string that will be capitalized and placed in the header
        and footer of the block. E.g. "CERTIFICATE", "PRIVATE KEY", etc. This
        will appear as "-----BEGIN CERTIFICATE-----" and
        "-----END CERTIFICATE-----".

    :param der_bytes:
        A byte string to be armored

    :param headers:
        An OrderedDict of the header lines to write after the BEGIN line

    :return:
        A byte string of the PEM block
    zA
            der_bytes must be a byte string, not %s
            zD
            type_name must be a unicode string, not %s
            �asciis-----BEGIN s-----
s: �

r
�@s	-----END )rr
rrrr	�upper�encoder
�write�base64�	b64encode�len�getvalue)r�	der_bytes�headers�output�keyZ	b64_bytesZb64_len�
irrr�armor2sD


��


�































r%c

cs0
t|t
�sttd
t|�
��
�
d}
i}d}d}d}d}|�d�
D]�}|dkrLq>|
dkr�t�d|�}|sfq>|�d�
�	d�
}d	}d
}
q>|
d
kr�|�
d�
dkr�d
}
n(|�	d�
}	|	�dd�\}
}|��||
<q>|
d
kr>|dd�dv�
r
t
�|�
}|||fV
d}
i}d}d}d	}q>||7}q>|�
r |�
s,ttd�
�
�
dS)ax
    Convert a PEM-encoded byte string into one or more DER-encoded byte strings

    :param pem_bytes:
        A byte string of the PEM-encoded data

    :raises:
        ValueError - when the pem_bytes do not appear to be PEM-encoded bytes

    :return:
        A generator of 3-element tuples in the format: (object_type, headers,
        der_bytes). The object_type is a unicode string of what is between
        "-----BEGIN " and "-----". Examples include: "CERTIFICATE",
        "PUBLIC KEY", "PRIVATE KEY". The headers is a dict containing any lines
        in the form "Name: Value" that are right after the begin line.
    zA
            pem_bytes must be a byte string, not %s
            Ztrash�NFs1^(?:---- |-----)BEGIN ([A-Z0-9 ]+)(?: ----|-----)rrTr!�
:r�body�
:r
�)s-----s---- z|
            pem_bytes does not appear to contain PEM-encoded data - no
            BEGIN/END combination found
            )rr
rrr�
splitlines�re�match�group�decoder�split�stripr�	b64decode�
ValueError)
�	pem_bytes�stater!Zbase64_dataZobject_typeZfound_startZ	found_end�lineZtype_name_matchZdecoded_line�name�valuer rrr�_unarmorpsX


�



























�r9FcCst|�
}|
st
|�
S|S)
a�
    Convert a PEM-encoded byte string into a DER-encoded byte string

    :param pem_bytes:
        A byte string of the PEM-encoded data

    :param multiple:
        If True, function will return a generator

    :raises:
        ValueError - when the pem_bytes do not appear to be PEM-encoded bytes

    :return:
        A 3-element tuple (object_name, headers, der_bytes). The object_name is
        a unicode string of what is between "-----BEGIN " and "-----". Examples
        include: "CERTIFICATE", "PUBLIC KEY", "PRIVATE KEY". The headers is a
        dict containing any lines in the form "Name: Value" that are right
        after the begin line.
    )r9�next)r4�multiple�	generatorrrr�unarmor�s
r=)
N)
F)�__doc__�
__future__rrrrrr,�sys�_errorsr�_typesrrr	r
�version_info�	cStringIOrr
�iorr%r9r=rrrr�<module>s	





>T