3
l�_U � @ s� d dl mZmZmZ d dlZd dlmZmZ dZdZ dZ
dZ dZ d Z
d
Zd eB Zd eB Zd
ZdZdZG dd� de�Zdd� Zdd� ZdS )� )�absolute_import�division�print_functionN)�int_from_bytes� int_to_bytes� � � � � � � � � � � � c @ sl e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� ZdS )� DERReaderc C s t |�| _d S )N)�
memoryview�data)�selfr � r � /usr/lib64/python3.6/_der.py�__init__$ s zDERReader.__init__c C s | S )Nr )r r r r � __enter__' s zDERReader.__enter__c C s |d kr| j � d S )N)� check_empty)r �exc_type� exc_value�tbr r r �__exit__* s zDERReader.__exit__c C s t | j�dkS )Nr )�lenr )r r r r �is_empty. s zDERReader.is_emptyc C s | j � std��d S )Nz Invalid DER input: trailing data)r! �
ValueError)r r r r r 1 s zDERReader.check_emptyc C s8 t | j�dk rtd��tj| jd�}| jdd � | _|S )N� z$Invalid DER input: insufficient datar )r r r" �six�
indexbytes)r �retr r r � read_byte5 s
zDERReader.read_bytec C s8 t | j�|k rtd��| jd |� }| j|d � | _|S )Nz$Invalid DER input: insufficient data)r r r" )r �nr& r r r �
read_bytes< s
zDERReader.read_bytesc C s� | j � }|d@ dkr td��| j � }|d@ dkr6|}nb|dM }|dkrNtd��d}x4t|�D ](}|dK }|| j � O }|dkr\td��q\W |dk r�td��| j|�}|t|�fS ) N� z-Invalid DER input: unexpected high tag numberr r � z?Invalid DER input: indefinite length form is not allowed in DER� z3Invalid DER input: length was not minimally-encoded)r' r" �ranger) r )r �tagZ length_byte�length�i�bodyr r r �read_any_elementC s,
zDERReader.read_any_elementc C s | j � \}}||kr td��|S )Nz!Invalid DER input: unexpected tag)r2 r" )r � expected_tagr. r1 r r r � read_elementg s zDERReader.read_elementc C s | � | j |�S Q R X d S )N)r4 )r r3 r r r �read_single_elementm s z DERReader.read_single_elementc C s. t | j�dkr*tj| jd�|kr*| j|�S d S )Nr )r r r$ r% r4 )r r3 r r r �read_optional_elementq s
z DERReader.read_optional_elementc C s| t | j�dkrtd��tj| jd�}|d@ dkr8td��t | j�dkrptj| jd�}|dkrp|d@ dkrptd��t| jd�S )Nr z)Invalid DER input: empty integer contentsr z'Negative DER integers are not supportedr# z0Invalid DER input: integer not minimally-encodedZbig)r r r" r$ r% r )r �first�secondr r r �
as_integerv s zDERReader.as_integerN)�__name__�
__module__� __qualname__r r r r! r r' r) r2 r4 r5 r6 r9 r r r r r # s $r c C s> t | tj�std��| dk r$td��| j� d d }t| |�S )NzValue must be an integerr z#Negative integers are not supportedr, r# )�
isinstancer$ Z
integer_typesr" �
bit_lengthr )�xr( r r r �encode_der_integer� s r@ c G s� d}x|D ]}|t |�7 }q
W tj| �g}|dk rD|jtj|�� n*t|�}|jtjdt |�B �� |j|� |j|� dj|�S )Nr r � )r r$ Zint2byte�appendr �extend�join)r. Zchildrenr/ ZchildZchunksZ length_bytesr r r �
encode_der� s
rE )Z
__future__r r r r$ Zcryptography.utilsr r Z CONSTRUCTEDZCONTEXT_SPECIFICZINTEGERZ
BIT_STRINGZ OCTET_STRINGZNULLZOBJECT_IDENTIFIERZSEQUENCEZSETZPRINTABLE_STRINGZUTC_TIMEZGENERALIZED_TIME�objectr r@ rE r r r r � s" d