o
    5hz                     @   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   r/var/www/html/construction_image-detection-poc/venv/lib/python3.10/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   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(      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%   r7   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   r8   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*   r5   u     Tu   ¯ )r   r   r   r8   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   r8   r   r   r9   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/   r0   r3   r2   r   u	    xxll)r   r8   r   r   _inner_tag_octetr   r   r   r   r   testObjDecode7   s   zDerObjectTests.testObjDecode7c                 C   s"   t d}| ||td d S )Nr,   r5   )r   r   r8   r   r   r   r   r   testObjDecode8   s   zDerObjectTests.testObjDecode8N)__name__
__module____qualname__r   r!   r#   r'   r.   r4   r:   r<   r>   rA   rB   rE   rG   rH   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)   rM      rN   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,                                	         rN   r   r   r   r   testEncode2   sX   
	
zDerIntegerTests.testEncode2c                 C   rP   )N   ÿiu   iu   þ¨¿rN   r   r   r   r   testEncode3   rU   zDerIntegerTests.testEncode3c                 C   s$   t ddd}| | td d S )N4   r"   r0      £4rN   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 )NrQ   r   rM   r)   rS   rR   r   r8   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 )NrW   rV   rY   rZ   r[   r,   rX   rf   r   r   r   r   testDecode2   sV   	
zDerIntegerTests.testDecode2c                 C   sB   t  }|td | |jd |td | |jd d S )Nr^   r]   u    i rf   r   r   r   r   testDecode3  s
   zDerIntegerTests.testDecode3c                 C   re   )Nz r)   u   ÿÿr]   r   r   rf   r   r   r   r   testDecode5  ri   zDerIntegerTests.testDecode5c                 C   s*   t dd}|td | |jd d S )Nr"   r0   ra   r`   rf   rb   r   r   r   testDecode6)  s   
zDerIntegerTests.testDecode6c                 C       t  }| ||td d S )NrS   )r   r   r8   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   r8   r   r   rb   r   r   r   testStrict16  s
   

zDerIntegerTests.testStrict1c                 C      t  }| t|jtd d S )Nu   )r   r   r   r8   r   r   r   r   r   testErrDecode1@     zDerIntegerTests.testErrDecode1N)rI   rJ   rK   rO   rT   r\   r_   rd   rh   rj   rk   rl   rm   ro   rr   rt   r   r   r   r   rL      s    

rL   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   rO   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   rT   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)   r]   z0)r
   rz   r   lenr   r   r   r   r   r   r\   [  s   
zDerSequenceTests.testEncode2c                 C   s(   t  }|d | | td d S )NrV   u   0r
   rz   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,   rX      0rY   rZ   r[   r~   r   r   r   r   rd   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 )NrV         0 ÿFr,   u	   0þr)   	      r"   r]   z0		)
r
   rz   r   r   r   r|   ry   r{   rx   r}   r   r   r   r   testEncode6  s(   




zDerSequenceTests.testEncode6c                 C   sD   t  }|d |td | | td | |  d S )NrV   z0   0	0)r
   rz   r   r   r   rx   ry   r   r   r   r   testEncode7  s
   
zDerSequenceTests.testEncode7c                 C   sF   t  }|d |t dg | | td | |  d S )NrV   r/   r   )r
   rz   r   r   r   rx   ry   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   rw   r)   r
   r8   r   r   r}   r   r   r   r   rh     s   zDerSequenceTests.testDecode1c                 C   s8   t  }|td | t|d | |d d d S )Nz0r)   r   rR   r   r   r   r   r   rj     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   rY   rZ   r[   r)   r   r,   rX   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   rV   r)   r   r   r   r   r   r   rm     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   rV   r)      $¶cr,    r   r   r   r   r   ro     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
   r8   r   r   r}   r{   rx   ry   r   r   r   r   testDecode8  s   zDerSequenceTests.testDecode8c                 C   rn   )Nr   )r
   r   r8   r   r   r   r   r   testDecode9  rp   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   r8   r   r   r   r   r   rt     s   zDerSequenceTests.testErrDecode1c                 C   rs   )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"   )nr_elements)r,   r"   r)   )r   r/   r
   r   r8   r   r   )r   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)only_ints_expectedFr   )r   der_bin1der_bin2r   r   r   test_expected_only_integers*  s   z,DerSequenceTests.test_expected_only_integersN)rI   rJ   rK   rO   rT   r\   r_   rd   r   r   r   r   rh   rj   r   rm   ro   r   r   rt   r   r   r   r   r   r   r   r   rv   F  s,    
		rv   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   rO   7     zDerOctetStringTests.testInit1c                 C   s<   t  }| | td td|_| | td d S )N r6   )r	   r   r   r   r   r   r   r   r   rT   ;  s   
zDerOctetStringTests.testEncode1c                 C   J   t  }|td | |jtd |td | |jtd d S )Nr   r   r   r6   )r	   r8   r   r   r   r   r   r   r   rh   E  
   zDerOctetStringTests.testDecode1c                 C   rn   )Nr   )r	   r   r8   r   r   r   r   r   rj   N  rp   zDerOctetStringTests.testDecode2c                 C   rs   )Nu   ÿ)r	   r   r   r8   r   r   r   r   r   rt   S  ru   z"DerOctetStringTests.testErrDecode1N)rI   rJ   rK   rO   rT   rh   rj   rt   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   rT   Z  s   zDerNullTests.testEncode1c                 C   rn   r   )r   r   r8   r   r   r   r   r   rh   `  rp   zDerNullTests.testDecode1N)rI   rJ   rK   rT   rh   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   rO   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   rg   r   r   r   r   rT   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   r\   v  s   zDerObjectIdTests.testEncode2c                 C   "   t  }|d | |jd d S )Nr   r   r   r8   r   rg   r   r   r   r   rh     s   
zDerObjectIdTests.testDecode1c                 C   s   t  }| ||d d S )Nr   )r   r   r8   r   r   r   r   rj     s   zDerObjectIdTests.testDecode2c                 C   r   )Ns   	*H z1.2.840.113549.1.0.1r   r   r   r   r   rk        
zDerObjectIdTests.testDecode3c                 C   r   )Nr   r   r   r   r   r   r   r     r   zDerObjectIdTests.testDecode4N)
rI   rJ   rK   rO   rT   r\   rh   rj   rk   r   r   r   r   r   r   e  s    	r   c                   @   r   )DerBitStringTestsc                 C   r   )Nr   u    ÿ)r   r   r   r   r   r   r   r   rO     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 r6    )r   r   r   r   rg   r   r   r   r   rT     s   
zDerBitStringTests.testEncode1c                 C   r   )N r   r   r6   )r   r8   r   r   rg   r   r   r   r   rh     r   zDerBitStringTests.testDecode1c                 C   rn   )Nr   )r   r   r8   r   r   r   r   r   rj     rp   zDerBitStringTests.testDecode2N)rI   rJ   rK   rO   r   rT   rh   rj   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   rO     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   rT     s
   
zDerSetOfTests.testEncode1c                 C   sR   t  }|d |d | | td t ddg}| | td d S )NrV   r   u   1 ÿ)r   r   r   r   r   r   r   r   r   r\     s   

zDerSetOfTests.testEncode2c                 C   s(   t  }|d | t|jtd d S )NrV   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   rd     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   r8   r   r   r}   listr   r   r   r   rh     s   zDerSetOfTests.testDecode1c                 C   sL   t  }|td | t|d t|}| d|v  | d|v  d S )N   1 ÿr,   rV   r   )r   r8   r   r   r}   r   r|   )r   r    lr   r   r   rj     s   zDerSetOfTests.testDecode2c                 C   rs   )Nr   r   r   r   r8   r   r   r   r   r   rk     s   
zDerSetOfTests.testDecode3c                 C   rn   )Nr   )r   r   r8   r   r   r   r   r   r      s   zDerSetOfTests.testDecode4c                 C   rs   )Nu   1 ÿªr   r   r   r   r   rt     s   
zDerSetOfTests.testErrDecode1N)rI   rJ   rK   rO   rT   r\   r_   rd   rh   rj   rk   r   rt   r   r   r   r   r     s    

	
	r   c                   @   r   )DerBooleanTestsc                 C   r   )NF    r   r   r   r   r   r   r   rT     r   zDerBooleanTests.testEncode1c                 C   r   )NT   r   r   r   r   r   r\     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/   r0       r   r   r   r   r   rd     r   zDerBooleanTests.testEncode4c                 C   r   )Nr   Fr   r8   r   rg   r   r   r   r   rh   "  r   zDerBooleanTests.testDecode1c                 C   r   )Nr   Tr   r   r   r   r   rj   '  r   zDerBooleanTests.testDecode2c                 C   &   t dd}|d | |jd d S )Nr   r   r   Fr   r   r   r   r   rk   ,     

zDerBooleanTests.testDecode3c                 C   r   )Nr/   r0   r   Fr   r   r   r   r   r   1  r   zDerBooleanTests.testDecode4c                 C      t  }| t|jd d S )Ns    r   r   r   r8   r   r   r   r   testErrorDecode16     z DerBooleanTests.testErrorDecode1c                 C   r   )Ns    r   r   r   r   r   testErrorDecode2;  r   z DerBooleanTests.testErrorDecode2N)rI   rJ   rK   rT   r\   r_   rd   rh   rj   rk   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)Crypto.SelfTest.st_commonr   r   rL   rv   r   r   r   r   r   r   )configr   	listTestsr   r   r   	get_testsA  s   r   __main__c                   C   s   t t S )N)unittest	TestSuiter   r   r   r   r   <lambda>P  s    r   suite)defaultTest)__doc__r   Crypto.Util.py3compatCrypto.Util.asn1r   r   r   r   r   r   r	   r
   r   TestCaser   rL   rv   r   r   r   r   r   r   r   rI   r   mainr   r   r   r   <module>   s(   !,x # p#1'R2