o
    ek}                     @   s  d Z ddlZddlT ddlmZmZmZmZmZm	Z	m
Z
mZmZ G dd dejZG dd dejZG d	d
 d
ejZG dd dejZG dd dejZG dd dejZG dd dejZG dd dejZG dd dejZi fddZedkrdd Zejdd dS dS )zSelf-tests for Crypto.Util.asn1    N)*)		DerObjectDerSetOf
DerIntegerDerBitStringDerObjectIdDerNullDerOctetStringDerSequence
DerBooleanc                   @   s|   e Zd Zdd Zdd Zdd Zdd Zd	d
 Zdd Zdd Z	dd Z
dd Zdd Zdd Zdd Zdd Zdd ZdS )DerObjectTestsc                 C   s$   |  tttd |  ttd d S )Nu       )assertRaises
ValueErrorr   b)self r   ND:\Projects\ConvertPro\env\Lib\site-packages\Crypto/SelfTest/Util/test_asn1.pytestObjInit1.   s   zDerObjectTests.testObjInit1c                 C   s   t td}| | td td|_| | td | | td t d}td|_| | td t tddd	}| | td
 d S )N EzE   zET)constructed0 )r   r   assertEqualencodepayloadr   derr   r   r   testObjEncode16   s   

zDerObjectTests.testObjEncode1c                 C   s&   t dtd}| | td d S )N   zz)r   r   r   r   r   r   r   r   testObjEncode2G   s   zDerObjectTests.testObjEncode2c                 C   s2   t td}tdd |_| | td d S )Nr   0   u   00000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000r   r   r   r   r   r   r   r   r   testObjEncode3L   s   zDerObjectTests.testObjEncode3c                 C   s\   t dddd}td|_| | td t ddd	d}td|_| | td
 d S )N      T)implicitr   Zppllu   ¡ppll      Fu   ppllr&   r   r   r   r   testObjEncode4R   s   

zDerObjectTests.testObjEncode4c                 C   s.   t ddd}td|_| | td d S )Nr(      Zexplicitxxll	   ¥xxllr&   r   r   r   r   testObjEncode5\   s   
zDerObjectTests.testObjEncode5c                 C   s:   t d}|td | |jtd | |jd d S )Nr+   r   decoder   r   r   
_tag_octetr   r   r   r   testObjDecode1d   s   zDerObjectTests.testObjDecode1c                 C   s>   t d}|td | |jtdd  | |jd d S )Nr+   u   111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111r%   r5   r   r   r   r   testObjDecode2k   s   zDerObjectTests.testObjDecode2c                 C   s<   t d}| t|jtd t d}| t|jtd d S )Nr+   u   ÿzr   r   r   r6   r   r   r   r   r   testObjDecode3r   s   zDerObjectTests.testObjDecode3c                 C   s   t dddd}| t|jtd |td | |jtd t dddd}| t|jtd |td	 | |jtd d S )
Nr+   F   )r   r*   r3   u     Tu   ¯ )r   r   r   r6   r   r   r   r   r   r   r   testObjDecode4z   s   zDerObjectTests.testObjDecode4c                 C   s    t d}| t|jtd d S )Nr+   zr;   r   r   r   r   testObjDecode5   s   zDerObjectTests.testObjDecode5c                 C   s8   t  }|td | |jd | |jtd d S )Nu   ee      )r   r6   r   r   r7   r   r   r   r   r   testObjDecode6   s   zDerObjectTests.testObjDecode6c                 C   sx   t ddd}|td | |jd | |jtd t ddd}|td | |jd | |jtd d S )Nr(   r.   r/   r1   r0   r   u	    xxll)r   r6   r   r   Z_inner_tag_octetr   r   r   r   r   testObjDecode7   s   zDerObjectTests.testObjDecode7c                 C   s"   t d}| ||td d S )Nr+   r3   )r   r   r6   r   r   r   r   r   testObjDecode8   s   zDerObjectTests.testObjDecode8N)__name__
__module____qualname__r   r!   r#   r'   r-   r2   r8   r:   r<   r?   r@   rC   rD   rE   r   r   r   r   r   ,   s    
r   c                   @   st   e Zd Zdd Zdd Zdd Zdd Zd	d
 Zdd Zdd Z	dd Z
dd Zdd Zdd Zdd Zdd ZdS )DerIntegerTestsc                 C   s    t d}| | td d S )Nr)   r   r   r   r   r   r   r   r   	testInit1   s   zDerIntegerTests.testInit1c                 C   X   t d}| | td t d}| | td t d}| | td d S )Nr    r)   rJ      rK   r   r   r   r   testEncode1      zDerIntegerTests.testEncode1c                 C   s   t d}| | td t d}| | td t dd }| | tdtd td td td td td td td td td td td td td td td td td	  d S )
Nr%   u         r+                                	         rK   r   r   r   r   testEncode2   sX   
	
zDerIntegerTests.testEncode2c                 C   rM   )N   ÿiu   iu   þ¨¿rK   r   r   r   r   testEncode3   rR   zDerIntegerTests.testEncode3c                 C   s$   t ddd}| | td d S )N4   r"   r/      £4rK   r   numberr   r   r   testEncode4   s   zDerIntegerTests.testEncode4c                 C   ^   t  }|td | |jd |td | |jd |td | |jd d S )NrN   r   rJ   r)   rP   rO   r   r6   r   r   valuer   r   r   r   testDecode1      zDerIntegerTests.testDecode1c                 C   s   t  }|td | |jd |tdtd td td td td td td td td td td td td td td td td td  | |jdd  d S )NrT   rS   rV   rW   rX   r+   rU   rc   r   r   r   r   testDecode2   sV   	
zDerIntegerTests.testDecode2c                 C   sB   t  }|td | |jd |td | |jd d S )Nr[   rZ   u    i rc   r   r   r   r   testDecode3  s
   zDerIntegerTests.testDecode3c                 C   rb   )Nz r)   u   ÿÿrZ   r   r   rc   r   r   r   r   testDecode5  rf   zDerIntegerTests.testDecode5c                 C   s*   t dd}|td | |jd d S )Nr"   r/   r^   r]   rc   r_   r   r   r   testDecode6)  s   
zDerIntegerTests.testDecode6c                 C       t  }| ||td d S )NrP   )r   r   r6   r   r   r   r   r   testDecode7/     zDerIntegerTests.testDecode7c                 C   sF   t  }|d |d | jt|jddd | jt|jddd d S )Ns    s    T)strict)r   r6   r   r   r_   r   r   r   testStrict16  s
   

zDerIntegerTests.testStrict1c                 C      t  }| t|jtd d S )Nu   )r   r   r   r6   r   r   r   r   r   testErrDecode1@     zDerIntegerTests.testErrDecode1N)rF   rG   rH   rL   rQ   rY   r\   ra   re   rg   rh   ri   rj   rl   ro   rq   r   r   r   r   rI      s    

rI   c                   @   s   e Zd Zdd Zdd Zdd Zdd Zd	d
 Zdd Zdd Z	dd Z
dd Zdd Zdd Zdd Zdd Zdd Zdd Zdd  Zd!d" Zd#d$ Zd%d& Zd'd( Zd)d* Zd+S ),DerSequenceTestsc                 C   s.   t dtdtdg}| | td d S )Nr)   r+   r   z
00 )r
   r   r   r   r   r   r   r   r   rL   H  s   zDerSequenceTests.testInit1c                 C   s   t  }| | td | |  |d | | td | | d | |dd | |  | |d | | td d S )Nr   r   0 r)   F)	r
   r   r   r   assertFalsehasOnlyIntsappendhasInts
assertTruer   r   r   r   rQ   L  s   
zDerSequenceTests.testEncode1c                 C   s   t  }|d d|d< | t|d | |d d | |d d | | td dg|d d < | t|d | |d d | | td d S )Nr   r)   rZ   z0)r
   rw   r   lenr   r   r   r   r   r   rY   [  s   
zDerSequenceTests.testEncode2c                 C   s(   t  }|d | | td d S )NrS   u   0r
   rw   r   r   r   r   r   r   r   r\   j     
zDerSequenceTests.testEncode3c                 C   s   t  }|dd  | | tdtd td td td td td td td td td td td td td td td td td td  d S )Nr+   rU      0rV   rW   rX   r{   r   r   r   r   ra   p  sR   	

zDerSequenceTests.testEncode4c                 C   s2   t  }|d7 }|td7 }| | td d S )Nr)   r   z00 )r
   r   r   r   r   r   r   r   testEncode5  s   zDerSequenceTests.testEncode5c                 C   s*  t  }|d |d | | td | |  | |d t  }|d |d | | td | | d | |dd | |  | |d |d d	d
g|dd < | t	|d | |dd  d	d
g | |dd d	g | | td d S )NrS         0 ÿFr+   u	   0þr)   	      r"   rZ   z0		)
r
   rw   r   r   r   ry   rv   rx   ru   rz   r   r   r   r   testEncode6  s(   




zDerSequenceTests.testEncode6c                 C   sD   t  }|d |td | | td | |  d S )NrS   z0   0	0)r
   rw   r   r   r   ru   rv   r   r   r   r   testEncode7  s
   
zDerSequenceTests.testEncode7c                 C   sF   t  }|d |t dg | | td | |  d S )NrS   r.   r   )r
   rw   r   r   r   ru   rv   r   r   r   r   testEncode8  s
   
zDerSequenceTests.testEncode8c                 C   s   t  }|td | t|d |td | t|d | |d d |td | t|d | |d d d S )Nr   r   rt   r)   r
   r6   r   r   rz   r   r   r   r   re     s   zDerSequenceTests.testDecode1c                 C   s8   t  }|td | t|d | |d d d S )Nz0r)   r   rO   r   r   r   r   r   rg     s   zDerSequenceTests.testDecode2c                 C   s   t  }|tdtd td td td td td td td td td td td td td td td td td td  | t|d | |d dd  d S )	Nr}   rV   rW   rX   r)   r   r+   rU   r   r   r   r   r   testDecode4  sT   
	
zDerSequenceTests.testDecode4c                 C   sH   t  }|td | t|d | |d d | |d d d S )Nr   r+   r   rS   r)   r   r   r   r   r   r   rj     s
   zDerSequenceTests.testDecode6c                 C   s`   t  }|td | t|d | |d d | |d td | |d td d S )	N   0
$¶c r"   r   rS   r)      $¶cr+    r   r   r   r   r   rl     s   zDerSequenceTests.testDecode7c                 C   s   t  }|td | t|d | |d td | |d td | | d | |dd | |  | |d d S )N	   0$¶c r+   r   r   r)   r   F)r
   r6   r   r   rz   rx   ru   rv   r   r   r   r   testDecode8  s   zDerSequenceTests.testDecode8c                 C   rk   )Nr   )r
   r   r6   r   r   r   r   r   testDecode9  rm   zDerSequenceTests.testDecode9c                 C   F   t  }| t|jtd | t|jtd | t|jtd d S )N r>   r$   r
   r   r   r6   r   r   r   r   r   rq     s   zDerSequenceTests.testErrDecode1c                 C   rp   )Nz0  r   r   r   r   r   testErrDecode2  s   zDerSequenceTests.testErrDecode2c                 C   r   )Nz0 u   0u   0r   r   r   r   r   testErrDecode3  s   zDerSequenceTests.testErrDecode3c                 C   s`   t g d }t  j|dd t  j|dd | jtt  j|dd | jtt  j|dd d S )Nr)   r+   r"   r"   )Znr_elements)r+   r"   r)   )r   r.   r
   r   r6   r   r   )r   Zder_binr   r   r   test_expected_nr_elements"  s
   z*DerSequenceTests.test_expected_nr_elementsc                 C   st   t g d }t ddt ddgg }t  j|dd t  j|dd t  j|dd | jtt  j|dd d S )	Nr   r)   r+   r"   r   T)Zonly_ints_expectedFr   )r   Zder_bin1Zder_bin2r   r   r   test_expected_only_integers*  s   z,DerSequenceTests.test_expected_only_integersN)rF   rG   rH   rL   rQ   rY   r\   ra   r~   r   r   r   re   rg   r   rj   rl   r   r   rq   r   r   r   r   r   r   r   r   rs   F  s,    
		rs   c                   @   4   e Zd Zdd Zdd Zdd Zdd Zd	d
 ZdS )DerOctetStringTestsc                 C   $   t td}| | td d S )N   ÿu   ÿ)r	   r   r   r   r   r   r   r   rL   7     zDerOctetStringTests.testInit1c                 C   s<   t  }| | td td|_| | td d S )N r4   )r	   r   r   r   r   r   r   r   r   rQ   ;  s   
zDerOctetStringTests.testEncode1c                 C   J   t  }|td | |jtd |td | |jtd d S )Nr   r   r   r4   )r	   r6   r   r   r   r   r   r   r   re   E  
   zDerOctetStringTests.testDecode1c                 C   rk   )Nr   )r	   r   r6   r   r   r   r   r   rg   N  rm   zDerOctetStringTests.testDecode2c                 C   rp   )Nu   ÿ)r	   r   r   r6   r   r   r   r   r   rq   S  rr   z"DerOctetStringTests.testErrDecode1N)rF   rG   rH   rL   rQ   re   rg   rq   r   r   r   r   r   5  s    
	r   c                   @   s   e Zd Zdd Zdd ZdS )DerNullTestsc                 C   s   t  }| | td d S Nz )r   r   r   r   r   r   r   r   rQ   Z  s   zDerNullTests.testEncode1c                 C   rk   r   )r   r   r6   r   r   r   r   r   re   `  rm   zDerNullTests.testDecode1N)rF   rG   rH   rQ   re   r   r   r   r   r   X  s    r   c                   @   sD   e Zd Zdd Zdd Zdd Zdd Zd	d
 Zdd Zdd Z	dS )DerObjectIdTestsc                 C      t d}| | d d S )Nz1.1s   ))r   r   r   r   r   r   r   rL   g     zDerObjectIdTests.testInit1c                 C   sP   t d}| | d t  }d|_| | d t d}| | d d S )N1.2.840.113549.1.1.1   	*H
2.999.1234   7R)r   r   r   rd   r   r   r   r   rQ   k  s   zDerObjectIdTests.testEncode1c                 C   s0   t d}| t|j t d}| t|j d S )Nz3.4z1.40)r   r   r   r   r   r   r   r   rY   v  s   zDerObjectIdTests.testEncode2c                 C   "   t  }|d | |jd d S )Nr   r   r   r6   r   rd   r   r   r   r   re     s   
zDerObjectIdTests.testDecode1c                 C   s   t  }| ||d d S )Nr   )r   r   r6   r   r   r   r   rg     s   zDerObjectIdTests.testDecode2c                 C   r   )Ns   	*H z1.2.840.113549.1.0.1r   r   r   r   r   rh        
zDerObjectIdTests.testDecode3c                 C   r   )Nr   r   r   r   r   r   r   r     r   zDerObjectIdTests.testDecode4N)
rF   rG   rH   rL   rQ   rY   re   rg   rh   r   r   r   r   r   r   e  s    	r   c                   @   r   )DerBitStringTestsc                 C   r   )Nr   u    ÿ)r   r   r   r   r   r   r   r   rL     r   zDerBitStringTests.testInit1c                 C   s$   t td}| | td d S )Nr)   z )r   r   r   r   r   r   r   r   r   	testInit2  r   zDerBitStringTests.testInit2c                 C   sb   t  }| | td t td}| | td t  }td|_| | td d S )Nz r4    )r   r   r   r   rd   r   r   r   r   rQ     s   
zDerBitStringTests.testEncode1c                 C   r   )N r   r   r4   )r   r6   r   r   rd   r   r   r   r   re     r   zDerBitStringTests.testDecode1c                 C   rk   )Nr   )r   r   r6   r   r   r   r   r   rg     rm   zDerBitStringTests.testDecode2N)rF   rG   rH   rL   r   rQ   re   rg   r   r   r   r   r     s    	r   c                   @   \   e Zd Zdd Zdd Zdd Zdd Zd	d
 Zdd Zdd Z	dd Z
dd Zdd ZdS )DerSetOfTestsc                 C   s,   t tdtdg}| | td d S )Nr)   r+   z1)r   r   r   r   r   r   r   r   r   rL     s   zDerSetOfTests.testInit1c                 C   sP   t  }| | td |d | | td | | td d S )N1 r   1 )r   r   r   r   addr   r   r   r   rQ     s
   
zDerSetOfTests.testEncode1c                 C   sR   t  }|d |d | | td t ddg}| | td d S )NrS   r   u   1 ÿ)r   r   r   r   r   r   r   r   r   rY     s   

zDerSetOfTests.testEncode2c                 C   s(   t  }|d | t|jtd d S )NrS   z   )r   r   r   r   r   r   r   r   r   r\     r|   zDerSetOfTests.testEncode3c                 C   s:   t  }|td |td | | td d S )Nz zz1 )r   r   r   r   r   r   r   r   r   ra     s   zDerSetOfTests.testEncode4c                 C   sX   t  }|td | t|d |td | t|d | t|dg d S )Nr   r   r   r)   )r   r6   r   r   rz   listr   r   r   r   re     s   zDerSetOfTests.testDecode1c                 C   sL   t  }|td | t|d t|}| d|v  | d|v  d S )N   1 ÿr+   rS   r   )r   r6   r   r   rz   r   ry   )r   r    lr   r   r   rg     s   zDerSetOfTests.testDecode2c                 C   rp   )Nr   r   r   r   r6   r   r   r   r   r   rh     s   
zDerSetOfTests.testDecode3c                 C   rk   )Nr   )r   r   r6   r   r   r   r   r   r      s   zDerSetOfTests.testDecode4c                 C   rp   )Nu   1 ÿªr   r   r   r   r   rq     s   
zDerSetOfTests.testErrDecode1N)rF   rG   rH   rL   rQ   rY   r\   ra   re   rg   rh   r   rq   r   r   r   r   r     s    

	
	r   c                   @   r   )DerBooleanTestsc                 C   r   )NF    r   r   r   r   r   r   r   rQ     r   zDerBooleanTests.testEncode1c                 C   r   )NT   r   r   r   r   r   rY     r   zDerBooleanTests.testEncode2c                 C       t ddd}| | d d S )NF   r*       r   r   r   r   r   r\        zDerBooleanTests.testEncode3c                 C   r   )NFr.   r/       r   r   r   r   r   ra     r   zDerBooleanTests.testEncode4c                 C   r   )Nr   Fr   r6   r   rd   r   r   r   r   re   "  r   zDerBooleanTests.testDecode1c                 C   r   )Nr   Tr   r   r   r   r   rg   '  r   zDerBooleanTests.testDecode2c                 C   &   t dd}|d | |jd d S )Nr   r   r   Fr   r   r   r   r   rh   ,     

zDerBooleanTests.testDecode3c                 C   r   )Nr.   r/   r   Fr   r   r   r   r   r   1  r   zDerBooleanTests.testDecode4c                 C      t  }| t|jd d S )Ns    r   r   r   r6   r   r   r   r   testErrorDecode16     z DerBooleanTests.testErrorDecode1c                 C   r   )Ns    r   r   r   r   r   testErrorDecode2;  r   z DerBooleanTests.testErrorDecode2N)rF   rG   rH   rQ   rY   r\   ra   re   rg   rh   r   r   r   r   r   r   r   r     s    r   c                 C   s   ddl m} g }||t7 }||t7 }||t7 }||t7 }||t7 }||t7 }||t7 }||t	7 }||t
7 }|S )Nr   )list_test_cases)ZCrypto.SelfTest.st_commonr   r   rI   rs   r   r   r   r   r   r   )configr   Z	listTestsr   r   r   	get_testsA  s   r   __main__c                   C   s   t t S )N)unittestZ	TestSuiter   r   r   r   r   <lambda>P  s    r   suite)ZdefaultTest)__doc__r   ZCrypto.Util.py3compatZCrypto.Util.asn1r   r   r   r   r   r   r	   r
   r   ZTestCaser   rI   rs   r   r   r   r   r   r   r   rF   r   mainr   r   r   r   <module>   s(   !,x # p#1'R2