OwlCyberSecurity - MANAGER
Edit File: PngImagePlugin.cpython-38.pyc
U g� ^9k � @ s� d dl Z d dlZd dlZd dlZddlmZmZmZ ddlm Z m Z m Z mZmZ e �e�Ze�d�jZdZddd d d d d ddddddddd�Ze�d�ZejZ de Z dd� Z dBdd �Z G d d � d �Z G d d � d e!�Z"G d!d"� d"�Z#G d#d$� d$e �Z$d%d&� Z%G d'd(� d(ej�Z&d)d*d+d,d-d.d/d/d0d1d2d3d4d5d6�Z'd7d8� Z(G d9d:� d:�Z)e(fd;d<�Z*d=d>� Z+e�,e&j-e&e%� e�.e&j-e*� e�/e&j-d?d@g� e�0e&j-dA� dS )C� N� )�Image� ImageFile� ImagePalette)�i8�i16be�i32be�o16be�o32bes \w\w\w\ws �PNG )�1r )�L�L;2)r �L;4)r r )�I�I;16B)�RGBr )r zRGB;16B)�P�P;1)r �P;2)r �P;4)r r )�LAr )�RGBAzLA;16B)r r )r zRGBA;16B))r r )� r )� r )� r )� r )r r )r r )r � )r r )r r )r r )r r )r r )r � )r r s ^�* �*$�@ c C s& t �� }|�| t�}|jr"td��|S )NzDecompressed Data Too Large)�zlibZ decompressobj� decompress�MAX_TEXT_CHUNKZunconsumed_tail� ValueError)�sZdobjZ plaintext� r$ �4/usr/lib/python3/dist-packages/PIL/PngImagePlugin.py�_safe_zlib_decompressT s r& c C s t �| |�d@ S )Nl �� )r Zcrc32)�dataZseedr$ r$ r% �_crc32\ s r( c @ s^ e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z ddd�Z dS )� ChunkStreamc C s || _ g | _d S �N)�fp�queue��selfr+ r$ r$ r% �__init__e s zChunkStream.__init__c C sz d}| j r(| j �� \}}}| j�|� n*| j�d�}|dd� }| j�� }t|�}t|�sptj spt dt |� ��|||fS )z.Fetch a new chunk. Returns header information.Nr r zbroken PNG file (chunk %s)) r, �popr+ �seek�read�tell�i32�is_cidr �LOAD_TRUNCATED_IMAGES� SyntaxError�repr)r. �cid�pos�lengthr# r$ r$ r% r2 j s zChunkStream.readc C s | S r* r$ �r. r$ r$ r% � __enter__} s zChunkStream.__enter__c G s | � � d S r* )�close)r. �argsr$ r$ r% �__exit__� s zChunkStream.__exit__c C s d | _ | _| _d S r* )r, �crcr+ r< r$ r$ r% r> � s zChunkStream.closec C s | j �|||f� d S r* )r, �append�r. r9 r: r; r$ r$ r% �push� s zChunkStream.pushc C s* t �d|||� t| d|�d� �||�S )z"Call the appropriate chunk handlerzSTREAM %r %s %sZchunk_�ascii)�logger�debug�getattr�decoderC r$ r$ r% �call� s zChunkStream.callc C s� t jr*t|d �d? d@ r*| �||� dS z6t|t|��}t| j�d��}||kr^td| ��W n" t j k r� td| ��Y nX dS )zRead and verify checksumr � r Nr z+broken PNG file (bad header checksum in %r)z+broken PNG file (incomplete checksum in %r)) r r6 r �crc_skipr( r4 r+ r2 r7 �struct�error)r. r9 r' Zcrc1Zcrc2r$ r$ r% rA � s zChunkStream.crcc C s | j �d� dS )z3Read checksum. Used if the C module is not presentr N)r+ r2 �r. r9 r' r$ r$ r% rL � s zChunkStream.crc_skip� IENDc C sf g }z| � � \}}}W n tjk r4 td��Y nX ||kr@qb| �|t�| j|�� |�|� q|S )Nztruncated PNG file) r2 rM rN �OSErrorrA r � _safe_readr+ rB )r. ZendchunkZcidsr9 r: r; r$ r$ r% �verify� s zChunkStream.verifyN)rP ) �__name__� __module__� __qualname__r/ r2 r= r@ r> rD rJ rA rL rS r$ r$ r$ r% r) d s r) c @ s e Zd ZdZeddd��ZdS )�iTXtzq Subclass of string to allow iTXt chunks to look like strings while keeping their extra information Nc C s t �| |�}||_||_|S )z� :param cls: the class to use when creating the instance :param text: value for this key :param lang: language code :param tkey: UTF-8 version of the key name )�str�__new__�lang�tkey)�cls�textrZ r[ r. r$ r$ r% rY � s z iTXt.__new__)NN)rT rU rV �__doc__� staticmethodrY r$ r$ r$ r% rW � s rW c @ s4 e Zd ZdZdd� Zdd� Zd dd �Zdd d �Zd S )�PngInfoz< PNG chunk container (for use with save(pnginfo=)) c C s g | _ d S r* )�chunksr< r$ r$ r% r/ � s zPngInfo.__init__c C s | j �||f� dS )z�Appends an arbitrary chunk. Use with caution. :param cid: a byte string, 4 bytes long. :param data: a byte string of the encoded data N)ra rB rO r$ r$ r% �add� s z PngInfo.add� Fc C s� t |t�s|�dd�}t |t�s,|�dd�}t |t�sB|�dd�}t |t�sX|�dd�}|r�| �d|d | d | d t�|� � n$| �d|d | d | d | � dS ) z�Appends an iTXt chunk. :param key: latin-1 encodable text key name :param value: value for this key :param lang: language code :param tkey: UTF-8 version of the key name :param zip: compression flag �latin-1�strict�utf-8� iTXts � s N)� isinstance�bytes�encoderb r �compress)r. �key�valuerZ r[ �zipr$ r$ r% �add_itxt� s �zPngInfo.add_itxtc C s� t |t�r"| j|||j|j|d�S t |t�sbz|�dd�}W n$ tk r` | j|||d� Y S X t |t�sx|�dd�}|r�| �d|d t � |� � n| �d|d | � dS ) z�Appends a text chunk. :param key: latin-1 encodable text key name :param value: value for this key, text or an :py:class:`PIL.PngImagePlugin.iTXt` instance :param zip: compression flag )ro rd re � zTXt� � tEXtrh N) ri rW rp rZ r[ rj rk � UnicodeErrorrb r rl )r. rm rn ro r$ r$ r% �add_text s zPngInfo.add_textN)rc rc F)F)rT rU rV r^ r/ rb rp ru r$ r$ r$ r% r` � s r` c s� e Zd Z� fdd�Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Z� ZS )#� PngStreamc s@ t � �|� i | _i | _d| _d | _d | _d | _d | _d| _ d S )N�r r r ) �superr/ �im_info�im_text�im_size�im_mode�im_tile� im_palette�im_custom_mimetype� text_memoryr- �� __class__r$ r% r/ " s zPngStream.__init__c C s* | j |7 _ | j tkr&td| j ��d S )Nz7Too much memory used in text chunks: %s>MAX_TEXT_MEMORY)r� �MAX_TEXT_MEMORYr" )r. Zchunklenr$ r$ r% �check_text_memory0 s ��zPngStream.check_text_memoryc C s� t �| j|�}|�d�}t�d|d |� � t�dt|| �� t|| �}|dkr`td| ��zt||d d � �}W n: t k r� t j r�d }n� Y n t j k r� d }Y nX || j d< |S )Nrh ziCCP profile name %rzCompression method %sr z+Unknown compression method %s in iCCP chunkr � icc_profile)r rR r+ �findrF rG r r7 r& r" r6 r rN ry )r. r: r; r# �i� comp_methodr� r$ r$ r% � chunk_iCCP8 s& � zPngStream.chunk_iCCPc C s� t �| j|�}t|�t|dd � �f| _z(tt|d �t|d �f \| _| _W n t k rd Y nX t|d �r|d| j d< t|d �r�t d��|S ) Nr r � � r � interlace� zunknown filter category) r rR r+ r4 r{ �_MODESr r| � im_rawmode� Exceptionry r7 �r. r: r; r# r$ r$ r% � chunk_IHDRU s ( zPngStream.chunk_IHDRc C s& dd| j || jfg| _|| _t�d S )Nro rw )r{ r� r} Zim_idat�EOFError�r. r: r; r$ r$ r% � chunk_IDATd s zPngStream.chunk_IDATc C s t �d S r* )r� r� r$ r$ r% � chunk_IENDk s zPngStream.chunk_IENDc C s& t �| j|�}| jdkr"d|f| _|S )Nr r )r rR r+ r| r~ r� r$ r$ r% � chunk_PLTEp s zPngStream.chunk_PLTEc C s� t �| j|�}| jdkrLt�|�r@|�d�}|dkrJ|| jd< q�|| jd< nP| jdkrft|�| jd< n6| jdkr�t|�t|dd � �t|dd � �f| jd< |S ) Nr rh r � transparency�r r r r r r ) r rR r+ r| �_simple_palette�matchr� ry �i16)r. r: r; r# r� r$ r$ r% � chunk_tRNSx s ,zPngStream.chunk_tRNSc C s$ t �| j|�}t|�d | jd< |S )N� j�@Zgamma)r rR r+ r4 ry r� r$ r$ r% � chunk_gAMA� s zPngStream.chunk_gAMAc C sB t �| j|�}t�dt|�d |�}tdd� |D ��| jd< |S )Nz>%dIr c s s | ]}|d V qdS )r� Nr$ )�.0Zeltr$ r$ r% � <genexpr>� s z'PngStream.chunk_cHRM.<locals>.<genexpr>Z chromaticity)r rR r+ rM Zunpack�len�tuplery )r. r: r; r# Zraw_valsr$ r$ r% � chunk_cHRM� s zPngStream.chunk_cHRMc C s t �| j|�}t|�| jd< |S )NZsrgb)r rR r+ r ry r� r$ r$ r% � chunk_sRGB� s zPngStream.chunk_sRGBc C s� t �| j|�}t|�t|dd � � }}t|d �}|dkrht|d d �t|d d �f}|| jd< n|dkr~||f| jd< |S ) Nr r r � F%u�?� �?�dpir Zaspect)r rR r+ r4 r �intry )r. r: r; r# Zpx�pyZunitr� r$ r$ r% � chunk_pHYs� s zPngStream.chunk_pHYsc C s� t �| j|�}z|�dd�\}}W n tk r> |}d}Y nX |r~|�dd�}|�dd�}| | j|< | j|< | �t |�� |S )Nrh r � rd re �replace) r rR r+ �splitr" rI ry rz r� r� )r. r: r; r# �k�vr$ r$ r% � chunk_tEXt� s zPngStream.chunk_tEXtc C s� t �| j|�}z|�dd�\}}W n tk r> |}d}Y nX |rRt|d �}nd}|dkrjtd| ��zt|dd � �}W n: tk r� t jr�d}n� Y n t j k r� d}Y nX |r�|� dd�}|� dd�}| | j |< | j |< | �t|�� |S ) Nrh r r� r z+Unknown compression method %s in zTXt chunkrd re r� )r rR r+ r� r" r r7 r& r6 r rN rI ry rz r� r� )r. r: r; r# r� r� r� r$ r$ r% � chunk_zTXt� s6 � zPngStream.chunk_zTXtc C s� t �| j|� }}z|�dd�\}}W n tk r> | Y S X t|�dk rP|S t|d �t|d �|dd � }}}z|�dd�\}} } W n tk r� | Y S X |dk�r|dk�r z t| �} W n> tk r� t jr�| Y S � Y n t j k �r | Y S X n|S z4|� dd�}|� dd�}| � dd�} | � dd�} W n t k �r\ | Y S X t | || � | j|< | j|< | �t| �� |S )Nrh r r r rd re rf )r rR r+ r� r" r� r r& r6 r rN rI rt rW ry rz r� ) r. r: r; �rr# r� Zcf�cmrZ Ztkr� r$ r$ r% � chunk_iTXt� sB ( zPngStream.chunk_iTXtc C s t �| j|�}d| | jd< |S )N� Exif �exif)r rR r+ ry r� r$ r$ r% � chunk_eXIf s zPngStream.chunk_eXIfc C s t �| j|�}d| _|S )Nz image/apng)r rR r+ r r� r$ r$ r% � chunk_acTL s zPngStream.chunk_acTL)rT rU rV r/ r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� � __classcell__r$ r$ r� r% rv ! s" $)rv c C s | d d� t kS )Nr )�_MAGIC)�prefixr$ r$ r% �_accept$ s r� c @ sX e Zd ZdZdZdd� Zedd� �Zdd� Zd d � Z d d � Z d d� Z dd� Z dd� Z dS )� PngImageFileZPNGzPortable network graphicsc C s | j �d�tkrtd��t| j �| _| j�� \}}}z| j�|||�}W nH tk r` Y q�Y n2 tk r� t � d|||� t � | j |�}Y nX | j� ||� q$| jj| _| jj| _| jj| _d | _| jj| _| jj| _| jjr�| jj\}}t�||�| _ || _ d S )Nr znot a PNG file�%r %s %s (unknown)) r+ r2 r� r7 rv �pngrJ r� �AttributeErrorrF rG r rR rA r| �moder{ Z_sizery �info�_textr} �tiler Zcustom_mimetyper~ r �raw�palette�_PngImageFile__prepare_idat)r. r9 r: r; r# �rawmoder' r$ r$ r% �_open1 s, zPngImageFile._openc C s | j d kr| �� | j S r* )r� �loadr< r$ r$ r% r] ` s zPngImageFile.textc C sZ | j dkrtd��| j �| jd d d � | j�� | j�� | jrP| j �� d| _ dS )zVerify PNG fileNz)verify must be called directly after openr r r )r+ � RuntimeErrorr1 r� r� rS r> Z _exclusive_fpr< r$ r$ r% rS i s zPngImageFile.verifyc C s0 | j �d�r| jd | _| j| _tj�| � dS )z"internal: prepare to read PNG filer� )r N)r� �getZ decoderconfigr� �_PngImageFile__idatr � load_preparer< r$ r$ r% r� y s zPngImageFile.load_preparec C s~ | j dkrJ| j�d� | j�� \}}}|dkrB| j�|||� dS || _ q |dkrZ| j }n t|| j �}| j | | _ | j�|�S )z internal: read more image datar r )� IDATs DDATr� )r� r+ r2 r� rD �min)r. Z read_bytesr9 r: r; r$ r$ r% � load_read� s zPngImageFile.load_readc C s� | j �d� z| j�� \}}}W n tjtfk r> Y q�Y nX |dkrJq�z| j�|||� W q tk rv Y q�Y q tk r� t � | j |� Y q t k r� t � d|||� t � | j |� Y q X q | jj| _| j�� d| _dS )z%internal: finished reading image datar rP r� N)r+ r2 r� rM rN r7 rJ �UnicodeDecodeErrorr� r rR r� rF rG rz r� r> rC r$ r$ r% �load_end� s& zPngImageFile.load_endc C s, d| j kr| �� d| j kr d S t| �� �S �Nr� )r� r� �dict�getexifr< r$ r$ r% �_getexif� s zPngImageFile._getexifc C s d| j kr| �� tj�| �S r� )r� r� r r� r< r$ r$ r% r� � s zPngImageFile.getexifN)rT rU rV �formatZformat_descriptionr� �propertyr] rS r� r� r� r� r� r$ r$ r$ r% r� , s / r� )r � )�L;1r� )r s )r s )r s )r s )r s )r s )r s )r s )r s )r s )r s )r r� r r r r r zI;16r r r r r r c G sJ d� |�}| �tt|��| � | �|� t|t|��}| �t|�� dS )z'Write a PNG chunk (including CRC field)r� N)�join�write�o32r� r( �r+ r9 r' rA r$ r$ r% �putchunk� s r� c @ s e Zd Zdd� Zdd� ZdS )�_idatc C s || _ || _d S r* )r+ �chunk)r. r+ r� r$ r$ r% r/ � s z_idat.__init__c C s | � | jd|� d S )Nr� )r� r+ �r. r' r$ r$ r% r� � s z _idat.writeN)rT rU rV r/ r� r$ r$ r$ r% r� � s r� c C s� | j }|dkr�d| jkr(d| jd > }n.| jrRttt| j�� d �d d�d�}nd}|dkrdd}n |dkrrd}n|dkr�d}nd }|d kr�d ||f }| j�d d �| j�d d�| j�dd�| j�dd�f| _zt | \}}W n t k r� t d| ��Y nX |� t � ||dt| jd �t| jd �|ddd� dddddg}| j�d| j�d��} | �r�d } | d t�| � } ||d | � |�d� | j�d �} | �r�d d!d"d#g} | jD ]B\}} ||k�r�|�|� |||| � n|| k�r�|||| � �q�| j dk�rDd| d }| j�d$�d |� }t|�|k �r8|d7 }�q ||d%|� | j�d&| j�d&d ��}|�sl|dk�rP| j dk�r�d| }t|t��r�||d'|d |� � n0tdtd(|��}d)| d }||d'|d |� � nz| j d*k�rtdtd+|��}||d't|�� nL| j d$k�r:|\}}}||d't|�t|� t|� � nd&| jk�r�t d,��nF| j dk�r�| j�� d-k�r�| j�d-d.�}d| }||d'|d |� � | j�d/�}|�r�||d0tt|d d1 d2 ��tt|d d1 d2 ��d3� | �r d4d5g}| jD ]*\}} ||k�r�|�|� |||| � �q�| j�d6| j�d6��}|�rxt|t j ��rT|� d �}|� d7��rl|d8d � }||d9|� t �!| t"||�d:d;| j d|fg� ||d<d� t#|d=��r�|�$� d S )>Nr �bitsr r � r r r r z%s;%d�optimizeFZcompress_level���Z compress_typeZ dictionaryr� zcannot write mode %s as PNGs IHDRr rh s cHRMs gAMAs sBITs sRGBs tIMEr� s ICC Profilerr s iCCPZpnginfos sPLTrg rs rq r s PLTEr� s tRNS� � �r� i�� z%cannot use transparency for this moder �Ar� s pHYsr� r� � s bKGDs hISTr� r� r s eXIfro rw rP �flush)%r� � encoderinfor� �maxr� r� Zgetdatar� Z encoderconfig� _OUTMODES�KeyErrorrQ r� r� r� �sizer� r rl �removera �imZ getpaletteri rj �o16Zgetpalettemoder� r ZExif�tobytes� startswithr �_saver� �hasattrr� )r� r+ �filenamer� r� Zcolorsr� r� ra Zicc�namer' r� Zchunks_multiple_allowedr9 Zpalette_byte_numberZ palette_bytesr� Z alpha_bytesZalphaZredZgreenZbluer� r� r$ r$ r% r� � s� $ � � " � $ r� c K sB G dd� d�}dd� }|� }z|| _ t| |d|� W 5 | ` X |jS )z4Return a list of PNG chunks representing this image.c @ s e Zd Zg Zdd� Zdd� ZdS )z getchunks.<locals>.collectorc S s d S r* r$ r� r$ r$ r% r� � s z"getchunks.<locals>.collector.writec S s | j �|� d S r* )r' rB )r. r� r$ r$ r% rB � s z#getchunks.<locals>.collector.appendN)rT rU rV r' r� rB r$ r$ r$ r% � collector� s r� c W s0 d� |�}tt|t|���}| �|||f� d S )Nr� )r� r� r( rB r� r$ r$ r% rB � s zgetchunks.<locals>.appendN)r� r� r' )r� Zparamsr� rB r+ r$ r$ r% � getchunks� s r� z.pngz.apngz image/png)r )1Zlogging�rerM r rc r r r Z_binaryr r r� r r4 r r� r r� Z getLoggerrT rF �compiler� r5 r� r� r� Z SAFEBLOCKr! r� r& r( r) rX rW r` rv r� r� r� r� r� r� r� Z register_openr� Z register_saveZregister_extensionsZ register_mimer$ r$ r$ r% �<module>" s| � XO � "