B
    u9aO                 @   s  d dl Z d dlZd dlZdddddddd	d
dddddddddgZeefZdd ZdBddZdCddZ	dd Z
dd ZeddZeddZdd Zdd ZdZdadad d ZdDd!d	Zd"d
 ZdEd#dZdadad$Zd%ZdFd&d'Zdd ddd(d)dZddd*d+d,dZd-Zda da!da"dGd.dZ#d/d Z$d0Z%e%d1 d2 Z&d3d Z'd4d Z(d5d6 Z)d7d Z*d8d9 Z+d:d Z,d;d< Z-d=d> Z.d?d@ Z/e0dAkre.  dS )H    Nencodedecodeencodebytesdecodebytes	b64encode	b64decode	b32encode	b32decode	b16encode	b16decode	b85encode	b85decode	a85encode	a85decodestandard_b64encodestandard_b64decodeurlsafe_b64encodeurlsafe_b64decodec             C   sx   t | tr2y
| dS  tk
r0   tdY nX t | tr@| S yt|  S  tk
rr   td| j	j
 d Y nX d S )Nasciiz4string argument should contain only ASCII charactersz>argument should be a bytes-like object or ASCII string, not %r)
isinstancestrr   UnicodeEncodeError
ValueErrorbytes_types
memoryviewtobytes	TypeError	__class____name__)s r    /usr/lib/python3.7/base64.py_bytes_from_decode_data"   s    


r"   c             C   s,   t j| dd}|d k	r(|td|S |S )NF)newlines   +/)binascii
b2a_base64	translatebytes	maketrans)r   altcharsencodedr    r    r!   r   3   s    Fc             C   sN   t | } |d k	r*t |}| t|d} |rDtd| sDtdt| S )Ns   +/s   ^[A-Za-z0-9+/]*={0,2}$zNon-base64 digit found)	r"   r&   r'   r(   rematchr$   Error
a2b_base64)r   r)   Zvalidater    r    r!   r   A   s    
c             C   s   t | S )N)r   )r   r    r    r!   r   Z   s    c             C   s   t | S )N)r   )r   r    r    r!   r   a   s    s   +/s   -_c             C   s   t | tS )N)r   r&   _urlsafe_encode_translation)r   r    r    r!   r   o   s    c             C   s   t | } | t} t| S )N)r"   r&   _urlsafe_decode_translationr   )r   r    r    r!   r   x   s    
s    ABCDEFGHIJKLMNOPQRSTUVWXYZ234567c                sB  t d kr,dd tD   fdd D a d  t| tsBt|  } t| d }|rb| dd|   } t }tj	}t }xjt
dt| dD ]V}|| ||d  d}|||d?  ||d	? d
@   ||d? d
@   ||d
@   7 }qW |dkrd|dd < nF|dkrd|dd < n.|dkr$d|dd < n|dkr:d|dd < t|S )Nc             S   s   g | ]}t |fqS r    )r'   ).0ir    r    r!   
<listcomp>   s    zb32encode.<locals>.<listcomp>c                s   g | ]} D ]}|| qqS r    r    )r1   ab)b32tabr    r!   r3      s           r   big      i  
      s   ======i   s   ====   s   ===      =)_b32tab2_b32alphabetr   r   r   r   len	bytearrayint
from_bytesranger'   )r   leftoverr*   rJ   Zb32tab2r2   cr    )r6   r!   r      s4    
*


c          	   C   s  t d krdd ttD a t| } t| d r8td|d k	r^t|}| t	dd| } |rj| 
 } t| }| d} |t|  }t }t }xtdt| dD ]l}| ||d  }d}	y"x|D ]}
|	d	> ||
  }	qW W n  tk
r   td
d Y nX ||	d	d7 }qW |d s(|dkr2td|rz|rz|	d	| K }	|	d	d}dd	|  d }|d | |dd < t|S )Nc             S   s   i | ]\}}||qS r    r    )r1   kvr    r    r!   
<dictcomp>   s    zb32decode.<locals>.<dictcomp>   zIncorrect paddings   01   OrC   r   r7   zNon-base32 digit foundr9   >   r   r=   r@   rB      +   )_b32rev	enumeraterF   r"   rG   r$   r-   r&   r'   r(   upperrstriprH   rK   KeyErrorto_bytes)r   casefoldZmap01lZpadcharsdecodedZb32revr2   ZquantaaccrM   ZlastrL   r    r    r!   r	      s@    



c             C   s   t |  S )N)r$   ZhexlifyrX   )r   r    r    r!   r
      s    c             C   s4   t | } |r|  } td| r*tdt| S )Ns	   [^0-9A-F]zNon-base16 digit found)r"   rX   r+   searchr$   r-   Z	unhexlify)r   r\   r    r    r!   r      s    

s   <~s   ~>c       	         s   t | tst|  } t|  d }|r4| d|  } tdt| d  | } fdd|D }|r|s|d dkr d d	 |d< |d d |  |d< d
|S )NrB   r8   z!%dIc                sP   g | ]H}r|sd n6r$|dkr$dn&|d  |d d    |d   qS )   zi       yi^	 U   i9  r    )r1   Zword)charschars2foldnuls
foldspacesr    r!   r3   !  s   z_85encode.<locals>.<listcomp>rD   ra   r   r7       )	r   r   r   r   rG   structStructunpackjoin)	r5   rd   re   padrf   rg   paddingZwordschunksr    )rd   re   rf   rg   r!   	_85encode  s    
rp   )rg   wrapcolrm   adobec               s   t d kr*dd tddD a dd t D at| t t|d| |rHt   rt|rVdnd fd	dtd
t D }|rt|d d kr|d d| |r t	7   S )Nc             S   s   g | ]}t |fqS r    )r'   )r1   r2   r    r    r!   r3   D  s    za85encode.<locals>.<listcomp>!   v   c             S   s   g | ]}t D ]}|| qqS r    )	_a85chars)r1   r4   r5   r    r    r!   r3   E  s    Tr>   r=   c                s   g | ]} ||  qS r    r    )r1   r2   )resultrq   r    r!   r3   M  s   r   rD   rh      
)
ru   rK   
_a85chars2rp   	_A85STARTmaxrG   appendrl   _A85END)r5   rg   rq   rm   rr   ro   r    )rv   rq   r!   r   /  s"    

s    	
)rg   rr   ignorecharsc         	   C   s  t | } |rH| ts$tdt| tr<| dd } n| d d } tdj	}g }|j
}g }|j
}|j}	x| d D ]}
d|
  krdkrn nt||
 t|dkrpd	}x|D ]}
d
| |
d  }qW y||| W n" tjk
r   tdd Y nX |	  qz|
dkr.|r$td|d qz|rV|
dkrV|rLtd|d qz|
|krdqzqztd|
 qzW d|}dt| }|r|d |  }|S )Nz1Ascii85 encoded byte sequences must end with {!r}r>   z!Is   uuuurs   u   r7   r   rc   zAscii85 overflowz   zz inside Ascii85 5-tuples       y   zy inside Ascii85 5-tuples       zNon-Ascii85 digit found: %crh   rB   )r"   endswithr|   r   format
startswithry   ri   rj   packr{   clearrG   errorrl   )r5   rg   rr   r}   packIr^   Zdecoded_appendZcurrZcurr_appendZ
curr_clearxr_   rv   rn   r    r    r!   r   X  sV    







sU   0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz!#$%&()*+-;<=>?@^_`{|}~c             C   s2   t d kr$dd tD a dd t D at| t t|S )Nc             S   s   g | ]}t |fqS r    )r'   )r1   r2   r    r    r!   r3     s    zb85encode.<locals>.<listcomp>c             S   s   g | ]}t D ]}|| qqS r    )	_b85chars)r1   r4   r5   r    r    r!   r3     s    )r   _b85alphabet
_b85chars2rp   )r5   rm   r    r    r!   r     s    	c       
   	   C   sZ  t d kr0d gd a xttD ]\}}|t |< qW t| } t|  d }| d|  } g }tdj}xtdt| dD ]}| ||d  }d}y"x|D ]}|d t |  }qW W nJ t	k
r   x2t|D ]&\}}t | d krt
d||  d qW  Y nX y||| W qt tjk
r2   t
d| d Y qtX qtW d	|}	|rV|	d |  }	|	S )
N   r7      ~z!Ir   rc   z#bad base85 character at position %dz+base85 overflow in hunk starting at byte %drh   )_b85decrW   r   r"   rG   ri   rj   r   rK   r   r   r{   r   rl   )
r5   r2   rM   rn   outr   chunkr_   jrv   r    r    r!   r     s<    


L   rB   r@   c             C   s`   xZ|  t}|sP x0t|tk rB|  tt| }|s8P ||7 }qW t|}|| qW d S )N)read
MAXBINSIZErG   r$   r%   write)inputoutputr   nsliner    r    r!   r     s    

c             C   s,   x&|   }|sP t|}|| qW d S )N)readliner$   r.   r   )r   r   r   r   r    r    r!   r     s    
c          
   C   s   yt | }W n8 tk
rD } zd| jj }t||W d d }~X Y nX |jdkrjd|j| jjf }t||jdkrd|j| jjf }t|d S )Nz"expected bytes-like object, not %s)rM   r5   Bz-expected single byte elements, not %r from %sr=   z(expected 1-D data, not %d-D data from %s)r   r   r   r   r   ndim)r   merrmsgr    r    r!   _input_type_check  s    

r   c             C   sP   t |  g }x8tdt| tD ]$}| ||t  }|t| qW d|S )Nr   rh   )r   rK   rG   r   r{   r$   r%   rl   )r   piecesr2   r   r    r    r!   r     s    c             C   s   dd l }|dtd t| S )Nr   zAencodestring() is a deprecated alias since 3.1, use encodebytes()r>   )warningswarnDeprecationWarningr   )r   r   r    r    r!   encodestring  s    r   c             C   s   t |  t| S )N)r   r$   r.   )r   r    r    r!   r     s    c             C   s   dd l }|dtd t| S )Nr   zHdecodestring() is a deprecated alias since Python 3.1, use decodebytes()r>   )r   r   r   r   )r   r   r    r    r!   decodestring$  s    r   c        	   
   C   s   dd l } dd l}y|| jdd  d\}}W nP |jk
r~ } z0| j| _t| td| jd   | d W d d }~X Y nX t}xF|D ]>\}}|dkrt}|dkrt	}|dkrt	}|d	krt
  d S qW |r
|d d
kr
t|d d}||| jj W d Q R X n|| jj| jj d S )Nr   r=   Zdeutzusage: %s [-d|-e|-u|-t] [file|-]
        -d, -u: decode
        -e: encode (default)
        -t: encode and decode string 'Aladdin:open sesame'r>   z-ez-dz-uz-t-rb)sysgetoptargvr   stderrstdoutprintexitr   r   testopenbufferstdin)	r   r   Zoptsargsr   funcor4   fr    r    r!   main.  s0         r   c              C   s<   d} t t|  t| }t t| t|}t t| d S )Ns   Aladdin:open sesame)r   reprr   r   )Zs0s1s2r    r    r!   r   H  s    r   __main__)N)NF)FN)F)FFF)F)1r+   ri   r$   __all__r'   rH   r   r"   r   r   r   r   r(   r/   r0   r   r   rF   rE   rV   r   r	   r
   r   ru   rx   ry   r|   rp   r   r   r   r   r   r   r   r   ZMAXLINESIZEr   r   r   r   r   r   r   r   r   r   r   r    r    r    r!   <module>	   sf   

	&
C

)H
-	
	


