| Current File : //usr/lib64/python2.7/site-packages/Crypto/SelfTest/Cipher/common.pyc |
�
���\c @ s_ d Z d Z d d l Z d d l Z d d l m Z m Z d d l Te j d k r_ d � Z n e Z d d d � � YZ
e
d
� Z d e j f d � � YZ
d
e
f d � � YZ d e j f d � � YZ d e j f d � � YZ d e j f d � � YZ d e j f d � � YZ d e j f d � � YZ d e j f d � � YZ d � Z d � Z d S( s& Self-testing for PyCrypto hash moduless $Id$i����N( t a2b_hext b2a_hex( t *i c K s
| j � S( N( t copy( t kwargs( ( sC /usr/lib64/python2.7/site-packages/Crypto/SelfTest/Cipher/common.pyt dict& s t
_NoDefaultc B s e Z RS( ( t __name__t
__module__( ( ( sC /usr/lib64/python2.7/site-packages/Crypto/SelfTest/Cipher/common.pyR + s c C s@ y | | } Wn$ t k
r4 | t k r0 � n | SX| | =| S( sA Get an item from a dictionary, and remove it from the dictionary.( t KeyErrorR ( t dt kt defaultt retval( ( sC /usr/lib64/python2.7/site-packages/Crypto/SelfTest/Cipher/common.pyt _extract, s
t CipherSelfTestc B s/ e Z d � Z d � Z d d � Z d � Z RS( c C sn t j j | � | | _ | j � } t | d � | _ t t | d � � | _ t t | d � � | _ t t | d � � | _
t | d d � | _ t | d d � } t
| � | _ | d k rOt | j d | � | _ t | d d � | _ | j d k rt | j � | _ n t | d d � | _ | j d k rat | j � | _ qan d | _ d | _ | | _ d S(
Nt descriptiont keyt plaintextt
ciphertextt module_namet modet MODE_t ivt encrypted_iv( t unittestt TestCaset __init__t moduleR R R t bR R R t NoneR t strt mode_namet getattrR R R t extra_params( t selfR t paramsR ( ( sC /usr/lib64/python2.7/site-packages/Crypto/SelfTest/Cipher/common.pyR : s* c C s | j S( N( R ( R# ( ( sC /usr/lib64/python2.7/site-packages/Crypto/SelfTest/Cipher/common.pyt shortDescriptionY s i c C s | j j � } t | j d � r0| j | j j k r0d d l m } t | d | j � } t | d i � j � } | j
d � r� t t | d � � | d <n | j
d � r� t t | d � � | d <n | j
d � s d | j j
t | j d d
� � t | j d d
� � | d <n | | � | d <n | j d k r[| j j t | j � | � S| j d k r�| j j t | j � | j | � S| r�| j | j j k r�| j j t | j � | j t | j � | � S| j j t | j � | j t | j � | � Sd S( Nt MODE_CTRi����( t Countert ctr_classt
ctr_paramst prefixt suffixt nbitsi t t counter( R" R t hasattrR R R& t Crypto.UtilR' R t newt has_keyR R t
block_sizet lent getR R R t MODE_OPENPGPR ( R# t
do_decryptionR$ R' R( R) ( ( sC /usr/lib64/python2.7/site-packages/Crypto/SelfTest/Cipher/common.pyt _new\ s& ' C ".c C s[ t | j � } t | j � } t | j � j | � � } t | j d � j | � � } t | j � j | � � } t | j d � j | � � } t | j d � r | j | j j
k r t | j � } | j
| j | | � | j
| j | | � | | } | | } n | j
| j | � | j
| j | � | j
| j | � | j
| j | � d S( Ni R6 ( R R R R R8 t encryptt decryptR/ R R R6 R4 R t assertEqual( R# R R t ct1t pt1t ct2t pt2t eilen( ( sC /usr/lib64/python2.7/site-packages/Crypto/SelfTest/Cipher/common.pyt runTestx s '
( R R R R% R8 RA ( ( ( sC /usr/lib64/python2.7/site-packages/Crypto/SelfTest/Cipher/common.pyR 8 s t CipherStreamingSelfTestc B s e Z d � Z d � Z RS( c C s: | j } | j d k r/ | d | j f 7} n d | f S( Ns in %s modes% %s should behave like a stream cipher( R R R R ( R# t desc( ( sC /usr/lib64/python2.7/site-packages/Crypto/SelfTest/Cipher/common.pyR% � s c C s* t | j � } t | j � } g } | j � } xA t d t | � d � D]' } | j | j | | | d !� � qI Wt t d � j
| � � } | j | j | � g } | j � } xA t d t | � d � D]' } | j | j | | | d !� � q� Wt t d � j
| � � } | j | j | � d S( Ni i R- ( R R R R8 t rangeR4 t appendR9 R R t joinR; ( R# R R t ct3t ciphert it pt3( ( sC /usr/lib64/python2.7/site-packages/Crypto/SelfTest/Cipher/common.pyRA � s % %( R R R% RA ( ( ( sC /usr/lib64/python2.7/site-packages/Crypto/SelfTest/Cipher/common.pyRB � s t CTRSegfaultTestc B s# e Z d � Z d � Z d � Z RS( c C sE t j j | � | | _ t | d � | _ | j d d � | _ d S( NR R ( R R R R R R R5 R R ( R# R R$ ( ( sC /usr/lib64/python2.7/site-packages/Crypto/SelfTest/Cipher/common.pyR � s c C s d | j | j f S( NsN Regression test: %s.new(key, %s.MODE_CTR) should raise TypeError, not segfault( R ( R# ( ( sC /usr/lib64/python2.7/site-packages/Crypto/SelfTest/Cipher/common.pyR% � s c C s/ | j t | j j t | j � | j j � d S( N( t assertRaisest TypeErrorR R1 R R R& ( R# ( ( sC /usr/lib64/python2.7/site-packages/Crypto/SelfTest/Cipher/common.pyRA � s ( R R R R% RA ( ( ( sC /usr/lib64/python2.7/site-packages/Crypto/SelfTest/Cipher/common.pyRK � s t CTRWraparoundTestc B s# e Z d � Z d � Z d � Z RS( c C sE t j j | � | | _ t | d � | _ | j d d � | _ d S( NR R ( R R R R R R R5 R R ( R# R R$ ( ( sC /usr/lib64/python2.7/site-packages/Crypto/SelfTest/Cipher/common.pyR � s c C s d | j f S( Ns] Regression test: %s with MODE_CTR should raise OverflowError on wraparound when shortcut used( R ( R# ( ( sC /usr/lib64/python2.7/site-packages/Crypto/SelfTest/Cipher/common.pyR% � s c
C s� d d l m } x� d D]� } x� d
D]� } | j d | j j d d d | j j d d | d | �} | j j t | j � | j j d
| �} t d � | j j } | j | � | j
t | j | � q$ Wq Wd S( Ni����( R' i i i t
initial_valuel t
little_endiant disable_shortcutR. t ( i i ( i i ( R0 R' R1 R R3 R R R& R R9 RL t
OverflowError( R# R' RQ RP t ctrRH t block( ( sC /usr/lib64/python2.7/site-packages/Crypto/SelfTest/Cipher/common.pyRA � s
=*
( R R R R% RA ( ( ( sC /usr/lib64/python2.7/site-packages/Crypto/SelfTest/Cipher/common.pyRN � s t CFBSegmentSizeTestc B s# e Z d � Z d � Z d � Z RS( c C s= t j j | � | | _ t | d � | _ | d | _ d S( NR R ( R R R R R R R ( R# R R$ ( ( sC /usr/lib64/python2.7/site-packages/Crypto/SelfTest/Cipher/common.pyR � s c C s | j S( N( R ( R# ( ( sC /usr/lib64/python2.7/site-packages/Crypto/SelfTest/Cipher/common.pyR% � s c C s� xH t d d � D]7 } | j t | j j t | j � | j j d | �q W| j j t | j � | j j d | j j d d �d S( sn Regression test: m.new(key, m.MODE_CFB, segment_size=N) should require segment_size to be a multiple of 8 bitsi i t segment_sizeRR N( RD RL t
ValueErrorR R1 R R t MODE_CFBR3 ( R# RI ( ( sC /usr/lib64/python2.7/site-packages/Crypto/SelfTest/Cipher/common.pyRA � s 5( R R R R% RA ( ( ( sC /usr/lib64/python2.7/site-packages/Crypto/SelfTest/Cipher/common.pyRV � s t
RoundtripTestc B s# e Z d � Z d � Z d � Z RS( c C s� d d l m } t j j | � | | _ | j | j � | _ t | d � | _
d t | d � | _ | j d d � | _ d S( Ni����( t RandomR id R R ( t CryptoR[ R R R R t get_random_bytesR3 R R R R R5 R R ( R# R R$ R[ ( ( sC /usr/lib64/python2.7/site-packages/Crypto/SelfTest/Cipher/common.pyR � s c C s d | j f S( Ns8 %s .decrypt() output of .encrypt() should not be garbled( R ( R# ( ( sC /usr/lib64/python2.7/site-packages/Crypto/SelfTest/Cipher/common.pyR% � s c C s� | j j } | j j t | j � | � } | j | j � } | j j t | j � | � } | j | � } | j | j | � | j j } | j j t | j � | | j
� } | j | j � } | | j j d } | | j j d } | j j t | j � | | � } | j | � } | j | j | � x� | j j | j j
| j j f D]� } | j j t | j � | | j
� } | j | j � } | j j t | j � | | j
� } | j | � } | j | j | � qKWd S( Ni ( R t MODE_ECBR1 R R R9 R R: R; R6 R R3 t MODE_CBCRY t MODE_OFB( R# R t encryption_cipherR t decryption_ciphert decrypted_plaintextt eiv_ciphertextt eiv( ( sC /usr/lib64/python2.7/site-packages/Crypto/SelfTest/Cipher/common.pyRA � s( $!($$( R R R R% RA ( ( ( sC /usr/lib64/python2.7/site-packages/Crypto/SelfTest/Cipher/common.pyRZ � s t PGPTestc B s# e Z d � Z d � Z d � Z RS( c C s0 t j j | � | | _ t | d � | _ d S( NR ( R R R R R R ( R# R R$ ( ( sC /usr/lib64/python2.7/site-packages/Crypto/SelfTest/Cipher/common.pyR s c C s d S( NsR MODE_PGP was implemented incorrectly and insecurely. It's completely banished now.( ( R# ( ( sC /usr/lib64/python2.7/site-packages/Crypto/SelfTest/Cipher/common.pyR% s c C s/ | j t | j j t | j � | j j � d S( N( RL RX R R1 R R t MODE_PGP( R# ( ( sC /usr/lib64/python2.7/site-packages/Crypto/SelfTest/Cipher/common.pyRA s ( R R R R% RA ( ( ( sC /usr/lib64/python2.7/site-packages/Crypto/SelfTest/Cipher/common.pyRf
s t IVLengthTestc B s, e Z d � Z d � Z d � Z d � Z RS( c C s0 t j j | � | | _ t | d � | _ d S( NR ( R R R R R R ( R# R R$ ( ( sC /usr/lib64/python2.7/site-packages/Crypto/SelfTest/Cipher/common.pyR s c C s d S( NsT Check that all modes except MODE_ECB and MODE_CTR require an IV of the proper length( ( R# ( ( sC /usr/lib64/python2.7/site-packages/Crypto/SelfTest/Cipher/common.pyR% s c C s | j t | j j t | j � | j j d � | j t | j j t | j � | j j d � | j t | j j t | j � | j j d � | j t | j j t | j � | j j d � | j j t | j � | j j
d � | j j t | j � | j j d d | j �d S( NR- R. (
RL RX R R1 R R R_ RY R` R6 R^ R& t _dummy_counter( R# ( ( sC /usr/lib64/python2.7/site-packages/Crypto/SelfTest/Cipher/common.pyRA # s %c C s d | j j S( NRR ( R R3 ( R# ( ( sC /usr/lib64/python2.7/site-packages/Crypto/SelfTest/Cipher/common.pyRi / s ( R R R R% RA Ri ( ( ( sC /usr/lib64/python2.7/site-packages/Crypto/SelfTest/Cipher/common.pyRh s c C s3 g } d } x t t | � � D] } | | } i d d 6} t | � d k rl | \ | d <| d <| d <n� t | � d k r� | \ | d <| d <| d <| d <n` t | � d
k r� | \ | d <| d <| d <| d <} | j | � n t d t | � f � � | j � } t | d � }
t | d � } t | d � } t | d d � }
| j d d � } | d k rt | d d � n |
d k r�|
} n9 | d k r�| r�d | |
f } n d
| |
| f } d | | d | f } | | d <| | d <| sY| t | | � t | | � t
| | � t | | � t | | � t
| | � g 7} d } n | j t | | � � | d k r�| j t | | � � n | d k r | j d � r | j � } | d c d 7<| j d i � j � } | | d <| d j d � sd | d d <n | j t | | � � q q W| S( Ni t ECBR i R R R i R i s Unsupported tuple size %ds
p=%s, k=%ss p=%s, k=%s, %rs
%s #%d: %si R t CTRR( s (shortcut disabled)R) RQ ( RD R4 t updatet AssertionErrorR R R R5 RK RN RV RZ Rf Rh RE R RB R2 ( R R t test_datat testst extra_tests_addedRI t rowR$ R" t p2t p_keyt p_plaintextt p_ciphertextt
p_descriptiont p_modeR t namet params2t ctr_params2( ( sC /usr/lib64/python2.7/site-packages/Crypto/SelfTest/Cipher/common.pyt make_block_tests2 sd
%%
c C s� g } x�t t | � � D]�} | | } i } t | � d k r_ | \ | d <| d <| d <n� t | � d k r� | \ | d <| d <| d <| d <n` t | � d k r� | \ | d <| d <| d <| d <} | j | � n t d t | � f � � | j � } t | d � } t | d � }
t | d � } t | d d � } | d k rV| }
n, | sod |
| f }
n d
|
| | f }
d | | d |
f } | | d <| | d
<| j t | | � � | j t | | � � q W| S( Ni R R R i R i s Unsupported tuple size %ds
p=%s, k=%ss p=%s, k=%s, %rs
%s #%d: %si R (
RD R4 Rl Rm R R R RE R RB ( R R Rn Ro RI Rq R$ R" Rr Rs Rt Ru Rv R Rx ( ( sC /usr/lib64/python2.7/site-packages/Crypto/SelfTest/Cipher/common.pyt make_stream_testsv s8
%%
( ( t __doc__t __revision__t sysR t binasciiR R t Crypto.Util.py3compatt
hexversionR R R R R RB RK RN RV RZ Rf Rh R{ R| ( ( ( sC /usr/lib64/python2.7/site-packages/Crypto/SelfTest/Cipher/common.pyt <module> s(
X )
D