
    ¿igt1                         d Z dZddlZddlZddlmZ ddlT ddlZddlmZm	Z	m
Z
  G d dej                  Zi fd	Zed
k    rd Z ej        d           dS dS )z,Self-test suite for Cryptodome.PublicKey.RSAz$Id$    N)PicklingError)*)list_test_casesa2b_hexb2a_hexc                       e Zd ZdZdZdZdZdZd Zd Z	d Z
d	 Zd
 Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd ZdS )RSATesta  
           eb 7a 19 ac e9 e3 00 63 50 e3 29 50 4b 45 e2
        ca 82 31 0b 26 dc d8 7d 5c 68 f1 ee a8 f5 52 67
        c3 1b 2e 8b b4 25 1f 84 d7 e0 b2 c0 46 26 f5 af
        f9 3e dc fb 25 c9 c2 b3 ff 8a e1 0e 83 9a 2d db
        4c dc fe 4f f4 77 28 b4 a1 b7 c1 36 2b aa d2 9a
        b4 8d 28 69 d5 02 41 21 43 58 11 59 1b e3 92 f9
        82 fb 3e 87 d0 95 ae b4 04 48 db 97 2f 3a c1 4f
        7b c2 75 19 52 81 ce 32 d2 f1 b7 6d 4d 35 3e 2d
    a  
        12 53 e0 4d c0 a5 39 7b b4 4a 7a b8 7e 9b f2 a0
        39 a3 3d 1e 99 6f c8 2a 94 cc d3 00 74 c9 5d f7
        63 72 20 17 06 9e 52 68 da 5d 1c 0b 4f 87 2c f6
        53 c1 1d f8 23 14 a6 79 68 df ea e2 8d ef 04 bb
        6d 84 b1 c3 1d 65 4a 19 70 e5 78 3b d6 eb 96 a0
        24 c2 ca 2f 4a 90 fe 9f 2e f5 c9 c1 40 e5 bb 48
        da 95 36 ad 87 00 c8 4f c9 13 0a de a7 4e 55 8d
        51 a7 4d df 85 d8 b5 0d e9 68 38 d6 06 3e 09 55
    a  
        bb f8 2f 09 06 82 ce 9c 23 38 ac 2b 9d a8 71 f7
        36 8d 07 ee d4 10 43 a4 40 d6 b6 f0 74 54 f5 1f
        b8 df ba af 03 5c 02 ab 61 ea 48 ce eb 6f cd 48
        76 ed 52 0d 60 e1 ec 46 19 71 9d 8a 5b 8b 80 7f
        af b8 e0 a3 df c7 37 72 3e e6 b4 b7 d9 3a 25 84
        ee 6a 64 9d 06 09 53 74 88 34 b2 45 45 98 39 4e
        e0 aa b1 2d 7b 61 a5 1f 52 7a 9a 41 f6 c1 68 7f
        e2 53 72 98 ca 2a 8f 59 46 f8 e5 fd 09 1d bd cb
       z
        c9 7f b1 f0 27 f4 53 f6 34 12 33 ea aa d1 d9 35
        3f 6c 42 d0 88 66 b1 d0 5a 0f 20 35 02 8b 9d 86
        98 40 b4 16 66 b4 2e 92 ea 0d a3 b4 32 04 b5 cf
        ce 33 52 52 4d 04 16 a5 a4 41 e7 00 af 46 15 03
    c                    ddl ma ddlma ddlmam} t          t          | j                            | _	        t          t          | j
                            | _        | j	        | j        z  | _         || j        | j        dz
  | j        dz
  z            | _         || j        | j                  | _        t          | _        d S )Nr   )RSA)Random)bytes_to_longinverse   )Cryptodome.PublicKeyr   
Cryptodomer   Cryptodome.Util.numberr   r   r   modulusnprime_factorpqedursa)selfr   s     b/var/www/html/afkarena/venv/lib/python3.11/site-packages/Cryptodome/SelfTest/PublicKey/test_RSA.pysetUpzRSATest.setUpY   s    ,,,,,,%%%%%%AAAAAAAAwt|4455wt'899:: 46!$&(TVAX!677((    c                 
   | j                             d          }|                     |           |                     |           |                                }|                     |           |                     |           dS )z7RSA (default implementation) generated key (1 argument)   N)r   generate_check_private_key_exercise_primitive
public_key_check_public_key_exercise_public_primitiver   rsaObjpubs      r   test_generate_1argzRSATest.test_generate_1argh   s~    ""4(('''  (((!!s###''/////r    c                 D   | j                             dt                                          j                  }|                     |           |                     |           |                                }|                     |           | 	                    |           dS )z8RSA (default implementation) generated key (2 arguments)r"   N)
r   r#   r   newreadr$   r%   r&   r'   r(   r)   s      r   test_generate_2argzRSATest.test_generate_2argq   s    ""4):;;'''  (((!!s###''/////r    c                 ~   | j                             dt                                          j        d          }|                     |           |                     |           |                                }|                     |           | 	                    |           | 
                    d|j                   d S )Nr"   i  )r   )r   r#   r   r.   r/   r$   r%   r&   r'   r(   assertEqualr   r)   s      r   test_generate_3argszRSATest.test_generate_3argsz   s    ""4):U"CC'''  (((!!s###''///vx(((((r    c                     | j                             | j        | j        f          }|                     |           |                     |           dS )z6RSA (default implementation) constructed key (2-tuple)N)r   	constructr   r   r'   _check_encryptionr   r+   s     r   test_construct_2tuplezRSATest.test_construct_2tuple   sN    h  $&$&!122s###s#####r    c                     | j                             | j        | j        | j        f          }|                     |           |                     |           dS )z6RSA (default implementation) constructed key (3-tuple)N)r   r5   r   r   r   r6   _check_decryptionr   r*   s     r   test_construct_3tuplezRSATest.test_construct_3tuple   sR    ##TVTVTV$<==v&&&v&&&&&r    c                     | j                             | j        | j        | j        | j        f          }|                     |           |                     |           dS )z6RSA (default implementation) constructed key (4-tuple)N)r   r5   r   r   r   r   r6   r:   r;   s     r   test_construct_4tuplezRSATest.test_construct_4tuple   sV    ##TVTVTVTV$DEEv&&&v&&&&&r    c                     | j                             | j        | j        | j        | j        | j        f          }|                     |           |                     |           | 	                    |           dS )z6RSA (default implementation) constructed key (5-tuple)N)
r   r5   r   r   r   r   r   r$   r6   r:   r;   s     r   test_construct_5tuplezRSATest.test_construct_5tuple   sn    ##TVTVTVTVTV$LMM'''v&&&v&&&&&r    c                     | j                             | j        | j        | j        | j        | j        | j        f          }|                     |           | 	                    |           | 
                    |           dS )z6RSA (default implementation) constructed key (6-tuple)N)r   r5   r   r   r   r   r   r   r$   r6   r:   r;   s     r   test_construct_6tuplezRSATest.test_construct_6tuple   sr    ##TVTVTVTVTVTV$TUU'''v&&&v&&&&&r    c                     | j         df}|                     t          | j        j        |           | j         dz   | j        f}|                     t          | j        j        |           d S Nr   )r   assertRaises
ValueErrorr   r5   r   r   tups     r   test_construct_bad_key2zRSATest.test_construct_bad_key2   sa    vqk*dh&8#>>> vax *dh&8#>>>>>r    c                     | j         | j        | j        dz   f}|                     t          | j        j        |           d S rD   )r   r   r   rE   rF   r   r5   rG   s     r   test_construct_bad_key3zRSATest.test_construct_bad_key3   s;    vtvtvax(*dh&8#>>>>>r    c                    | j         | j        | j        | j        | j        f}|                     t
          | j        j        |           | j        | j        z  | j        | j        | j        f}|                     t
          | j        j        |           | j        | j        z  d| j        | j        f}|                     t
          | j        j        |           d S )N   )	r   r   r   r   rE   rF   r   r5   r   rG   s     r   test_construct_bad_key5zRSATest.test_construct_bad_key5   s    vtvtvtvtv6*dh&8#>>>vdf}dfdfdf5*dh&8#>>>vdf}a0*dh&8#>>>>>r    c           	      V   | j         | j        | j        | j        | j        df}|                     t          | j        j        |           ddl	m
} | j         | j        | j        | j        | j         || j        | j                  f}|                     t          | j        j        |           d S )N
   r   )r   )r   r   r   r   r   rE   rF   r   r5   r   r   )r   rH   r   s      r   test_construct_bad_key6zRSATest.test_construct_bad_key6   s    vtvtvtvtvr:*dh&8#>>>222222vtvtvtvtvwwtvtv7N7NO*dh&8#>>>>>r    c                    | j                             | j        | j        | j        g          }|                     |j        | j        k    p|j        | j        k               |                     |j        | j        k    p|j        | j        k               |                     |j        |j        z  | j        k               |                     t          | j         j        | j        | j        | j        dz
  g           d S rD   )
r   r5   r   r   r   
assertTruer   r   rE   rF   r;   s     r   test_factoringzRSATest.test_factoring   s    ##TVTVTV$<==$&(<FHdf,<===$&(<FHdf,<===)TV3444*dh&8464646RS8:TUUUUUr    c                     | j                             | j        | j        | j        | j        | j        f          }t          |           d S N)r   r5   r   r   r   r   r   reprr;   s     r   	test_reprzRSATest.test_repr   s;    ##TVTVTVTVTV$LMMVr    c                     | j                             d          }|                     t          t          j        |           dS )zRSA keys are unpickabler"   N)r   r#   rE   r   pickledumps)r   rsa_keys     r   test_serializationzRSATest.test_serialization   s9     (##D))-w?????r    c                    | j                             d          }|                     t          |j        |j                   |                     t          |j        |j                   |                     t          |j        |j                   |                     t          |j        d           |                     t          |j        d           |                     t          |j        d           d S )Nr"   )r   r#   rE   rF   _decryptr   _decrypt_to_bytes_encrypt)r   rsa_objs     r   test_raw_rsa_boundaryzRSATest.test_raw_rsa_boundary   s     (##D))*g&6	BBB*g&?KKK*g&6	BBB*g&6;;;*g&?DDD*g&6;;;;;r    c                     | j                             | j        | j        f          }|                     |                                d           |                     |                                d           d S )Nr"      )r   r5   r   r   r2   size_in_bitssize_in_bytesr7   s     r   	test_sizezRSATest.test_size   sj    h  $&$&!122))++T222**,,c22222r    c                 X   ddl m} |                     d|                                           |                     |j        |j        |j        z             t           ||j        dz
                                |j        dz
                      }|                     d|j	        |j
        z  |z             |                     d|j        |j        z  |j        z             |                     d|j        dk               |                     d|j        dk               |                     d|j
        dk               |                     d|j	        dk               |                     |j        |j                   |                     d|j        |j        z  |j        z             d S )Nr   )Integerr   )Cryptodome.Math.Numbersrk   r2   has_privater   r   r   intlcmr   r   r   invpinvq)r   r*   rk   ro   s       r   r$   zRSATest._check_private_key   s   333333 	F..00111 	68fh#6777''&(1*%%))&(1*5566FHvx/#5666FHvx/&(:;;;FHqL)))FHqL)))FHqL)))FHqL)))6;///FHv{2VX=>>>>>r    c                    t          | j                  }|                     d|                                           |                     |j        |j                   |                     |j        |j                   |                     dt          |d                     |                     dt          |d                     |                     dt          |d                     |                     dt          |d                     |                     d|j        dk               |                     t          |j	        t          |                     |                     t          |j        t          |                     |                     |                                |                                k    d           |                     |                                |                                k    d           |                     |                                |                                           d S )	Nr   r   r   r   r   r   TF)r   
ciphertextr2   rm   r   r   hasattrrE   	TypeErrorr`   r   ra   r&   	publickey)r   r*   rs   s      r   r'   zRSATest._check_public_key   s   T_--
 	F..00111 	68,,,68,,, 	GFC00111GFC00111GFC00111GFC00111 	FHqL))) 	)V_j))	+ 	+ 	+)V%=j))	+ 	+ 	+ 	**,,0A0A0C0CCDIII**,,0A0A0C0CCEJJJ))++V->->-@-@AAAAAr    c                     t          t          | j                            }|                    |          }|                    |          }|                     ||           d S rV   )r   r   rs   r`   rb   r2   )r   r*   rs   	plaintextnew_ciphertext2s        r   r%   zRSATest._exercise_primitive  s]     #74?#;#;<<
 OOJ//	 !//)44_55555r    c                 r    t          | j                  }|                    t          |                    }d S rV   )r   rx   rb   r   )r   r*   rx   ry   s       r   r(   z"RSATest._exercise_public_primitive$  s0    DN++	 !//-	*B*BCCr    c                     t          | j                  }t          | j                  }|                    t	          |                    }|                     t	          |          |           d S rV   )r   rx   rs   rb   r   r2   )r   r*   rx   rs   ry   s        r   r6   zRSATest._check_encryption*  s_    DN++	T_--
 !//-	*B*BCCz22ODDDDDr    c                     t          t          | j                            }t          t          | j                            }|                    |          }|                     ||           d S rV   )r   r   rx   rs   r`   r2   )r   r*   rx   rs   new_plaintexts        r   r:   zRSATest._check_decryption2  s_    !'$."9"9::	"74?#;#;<<
 
33M22222r    N) __name__
__module____qualname__rx   rs   r   r   r   r   r,   r0   r3   r8   r<   r>   r@   rB   rI   rK   rN   rQ   rT   rX   r]   rd   ri   r$   r'   r%   r(   r6   r:    r    r   r	   r	   %   s       	I	J	G 	AL  0 0 00 0 0) ) )$ $ $' ' '' ' '' ' '' ' '? ? ?? ? ?? ? ?? ? ?V V V  @ @ @< < <3 3 3
? ? ?&B B B>6 6 6D D DE E E3 3 3 3 3r    r	   c                 8    g }|t          t                    z  }|S rV   )r   r	   )configtestss     r   	get_testsr   ;  s    E	_W%%%ELr    __main__c                  B    t          j        t                                S rV   )unittest	TestSuiter   r   r    r   <lambda>r   A  s    H&y{{33 r    suite)defaultTest)__doc____revision__osrZ   r   Cryptodome.Util.py3compatr   Cryptodome.SelfTest.st_commonr   r   r   TestCaser	   r   r~   r   mainr   r    r   <module>r      s   2 3 2 				              ' ' ' '  K K K K K K K K K KS3 S3 S3 S3 S3h S3 S3 S3l     
 z33EHMg&&&&&& r    