OwlCyberSecurity - MANAGER
Edit File: ImageFile.cpython-38.pyc
U tCf R � @ s� d dl Z d dlZd dlZddlmZ ddlmZ dZdZdZ dd d d d d �Z dd� Z dd� Z G dd� dej�Z G dd� de �ZG dd� d�Zd dd�Zdd� ZG d d � d �ZG d d � d �ZdS )!� N� )�Image)�isPathi i Fzimage buffer overrun errorzdecoding errorz unknown errorzbad configurationzout of memory error)���������i����i����c C sL zt j�| �}W n tk r. t�| �}Y nX |s<d| }t|d ��d S )Nzdecoder error %dz when reading image file)r �coreZgetcodecstatus�AttributeError�ERRORS�get�OSError)�error�message� r �//usr/lib/python3/dist-packages/PIL/ImageFile.py� raise_ioerror4 s r c C s | d S )N� r )�tr r r � _tilesortC s r c sR e Zd ZdZd� fdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z � Z S )� ImageFilez*Base class for image file format handlers.Nc s� t � �� d| _d | _d | _d| _d| _t| _t |�rPt |d�| _ || _ d| _ n|| _ || _ d | _ zdz | �� W n6 tttttjfk r� } z t|��W 5 d }~X Y nX | jr�| jd dkr�td��W n& tk r� | j r�| j �� � Y nX d S )Nr r r �rbTz not identified by this driver)�super�__init__� _min_frame�custom_mimetype�tile�readonly� decoderconfig�MAXBLOCK�decodermaxblockr �open�fp�filename� _exclusive_fp�_open� IndexError� TypeError�KeyError�EOFError�structr � SyntaxError�mode�size� BaseException�close)�selfr! r" �v�� __class__r r r P s> � zImageFile.__init__c C s, | j r | j S | jd k r(tj�| j�� �S d S �N)r �formatr ZMIMEr �upper�r/ r r r �get_format_mimetype} s z ImageFile.get_format_mimetypec C s | j r| j�� d| _dS )zCheck file integrityN)r# r! r. r6 r r r �verify� s zImageFile.verifyc C s� t j �| �}| jdkr td��| js(|S d| _| jo@t| j�dk}|oPttd� }d}z| j }d}W n t k r� | j j }Y nX z| j }d}W n t k r� | j j}Y nX |�r�| jd \}}}} |dk�r�t| �dk�r�| d | jk�r�| d t jk�r�z�tt jd ��rLt j�| j�| _| j�|� | j�| j| j| d | d �| _nTddl} t| jd �� } | j| �� d| jd �| _W 5 Q R X t j�| j| j|||| �| _d}| j�r�d| j_W n" t tt fk �r� d| _Y nX | � � d } | j�s4| jj t d� z | j } W n t k �r d} Y nX | jD �]\}}}} t �!| j|| | j"�}z�||� |�$| j|� |j%�r�|�&| j � |�'d�\}} n�| }z|| j(�}W n2 t)t*j+fk �r� t,�r�Y �q"ntd��Y nX |�s�t,�rސq"ntdt|� ��|| }|�'|�\}} |dk �r�q"||d� }�q�W 5 |�#� X �q(g | _|| _-| �.� | j/�rb| j0�rb| j �1� d| _ | j�s�t,�s�| dk �r�t2| � t j �| �S )z"Load image data based on tile listNzcannot load this imager Zpypy_version_infor F�raw� �mapr �r)�accessr ��key� zimage file is truncatedz0image file is truncated (%d bytes not processed))3r �loadr r r; r" �len�hasattr�sys� load_readr r! �read� load_seek�seekr+ Z _MAPMODESr Z readimager, �im�mmapr �filenoZ ACCESS_READZ map_bufferZpaletteZdirty� ImportError� load_prepare�sortr Z tile_prefix� _getdecoderr �cleanup�setimage�pulls_fd�setfd�decoder r% r) r �LOAD_TRUNCATED_IMAGESr �load_endr# Z!_close_exclusive_fp_after_loadingr. r )r/ ZpixelZuse_mmapr rF rH Z decoder_name�extents�offset�argsrJ r! Zerr_code�prefix�decoderZstatus�b�s�nr r r rA � s� � � � � � � � � �� zImageFile.loadc C sP | j r"| j j| jks"| j j| jkr6tj�| j| j�| _ | jdkrLtj�| � d S )N�P)rI r+ r, r r �newrA r6 r r r rM s " zImageFile.load_preparec C s d S r3 r r6 r r r rV s zImageFile.load_endc C sB || j k s.t| d�r | jd ks6|| j| j kr6td��| �� |kS )N� _n_framesz attempt to seek outside sequence)r rC ra Zn_framesr( �tell)r/ �framer r r � _seek_check& s ���� zImageFile._seek_check)NN) �__name__� __module__� __qualname__�__doc__r r7 r8 rA rM rV rd � __classcell__r r r1 r r M s - r c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) � StubImageFilez� Base class for stub image loaders. A stub loader is an image loader that can identify files of a certain format, but relies on external code to load the file. c C s t d��d S )Nz+StubImageFile subclass must implement _open��NotImplementedErrorr6 r r r r$ = s zStubImageFile._openc C sH | � � }|d kr td| j ��|�| �}|d k s4t�|j| _|j| _d S )Nz#cannot find loader for this %s file)�_loadr r4 rA �AssertionErrorr2 �__dict__)r/ �loader�imager r r rA @ s zStubImageFile.loadc C s t d��dS )z (Hook) Find actual image loader.z+StubImageFile subclass must implement _loadNrk r6 r r r rm J s zStubImageFile._loadN)re rf rg rh r$ rA rm r r r r rj 5 s rj c @ sP e Zd ZdZdZdZdZdZdZdZ dd� Z dd� Z dd � Z d d � Z d d � ZdS )�Parserzj Incremental image parser. This class implements the standard feed/close consumer interface. Nr c C s | j dkstd��dS )z� (Consumer) Reset the parser. Note that you can only call this method immediately after you've created a parser; parser instances cannot be reused. Nzcannot reuse parsers)�datarn r6 r r r �reset\ s z Parser.resetc C s� | j r dS | jdkr || _n | j| | _| jr�| jdkrztt| j�| j�}| j|d� | _| j| | _| jdksv| jszdS | j�| j�\}}|dk r�d| _d| _ |dk r�d| _t|� ndS | j|d� | _n�| jr�n�z&t � | j��}t � |�}W 5 Q R X W n t k �r Y n�X t|d��p(t|d�}|�s@t|j�dk�rHd| _nv|�� |jd \}}} } g |_t �|j|| |j�| _| j�|j|� | | _| jt| j�k�r�| j| jd� | _d| _|| _dS )z� (Consumer) Feed data to the parser. :param data: A string buffer. :exception IOError: If the parser failed to parse the image file. Nr r rG rE )�finishedrs r[ rX �minrB rT rq r �io�BytesIOr r r rC r rM rO r+ r rQ rI ) r/ rs �skipr^ �er! rI �flag�d�o�ar r r �feedd sR z Parser.feedc C s | S r3 r r6 r r r � __enter__� s zParser.__enter__c G s | � � d S r3 )r. �r/ rY r r r �__exit__� s zParser.__exit__c C sz | j r*| �d� d | _| _ | js*td��| js8td��| jrtt�| j��$}zt � |�| _W 5 | j�� X W 5 Q R X | jS )a (Consumer) Close the stream. :returns: An image object. :exception IOError: If the parser failed to parse the image file either because it cannot be identified or cannot be decoded. r@ Nzimage was incompletezcannot parse this image) r[ r rs ru r rq rw rx rA r r )r/ r! r r r r. � s z Parser.close)re rf rg rh Z incrementalrq rs r[ rX ru rt r r� r� r. r r r r rr O s Prr c C s� | � � t| d�sd| _|jtd� tt|| jd d �}|tj krP|� � dS z|� � }|� � W n� t t jfk �r& |D ]�\}}}}t�| j||| j�}|dkr�|�|� |�| j|� |jr�|�|� |�� \} } n$|�|�\} } } |�| � | rސqq�| dk �rtd| ��|�� q�Y n�X |D ]�\}}}}t�| j||| j�}|dk�r`|�|� |�| j|� |j�r�|�|� |�� \} } n |� ||�} | dk �r�td| ��|�� �q,t|d��r�|� � dS ) z�Helper to save image based on tile list :param im: Image object. :param fp: File object. :param tile: Tile list. :param bufsize: Optional buffer size � encoderconfigr r> r � Nz(encoder error %d when writing image file�flush) rA rC r� rN r �maxr r, rD �stdoutr� rK r rw �UnsupportedOperationr Z _getencoderr+ rH rQ rI Z pushes_fdrS Zencode_to_pyfd�encode�writer rP Zencode_to_file) rI r! r �bufsizeZfhrz r\ r} r~ �lr] r| r r r �_save� sR r� c C s� |dkr dS |t kr6| �|�}t|�|k r2td��|S g }|dkrp| �t|t ��}|sXqp|�|� |t|�8 }q:tdd� |D ��|k r�td��d�|�S )a� Reads large blocks in a safe way. Unlike fp.read(n), this function doesn't trust the user. If the requested size is larger than SAFEBLOCK, the file is read block by block. :param fp: File handle. Must implement a <b>read</b> method. :param size: Number of bytes to read. :returns: A string containing <i>size</i> bytes of data. Raises an OSError if the file is truncated and the read can not be completed r r@ zTruncated File Readc s s | ]}t |�V qd S r3 )rB )�.0r| r r r � <genexpr>3 s z _safe_read.<locals>.<genexpr>)� SAFEBLOCKrF rB r rv �append�sum�join)r! r, rs �blockr r r � _safe_read s" r� c @ s e Zd Zdd� Zdd� ZdS )� PyCodecStatec C s d| _ d| _d| _d| _d S )Nr )�xsize�ysize�xoff�yoffr6 r r r r ; s zPyCodecState.__init__c C s | j | j| j | j | j| j fS r3 )r� r� r� r� r6 r r r rW A s zPyCodecState.extentsN)re rf rg r rW r r r r r� : s r� c @ s\ e Zd ZdZdZdd� Zdd� Zedd� �Zd d � Z d d � Z d d� Z ddd�Z ddd�Z dS )� PyDecoderz� Python implementation of a format decoder. Override this class and add the decoding logic in the `decode` method. See :ref:`Writing Your Own File Decoder in Python<file-decoders-py>` Fc G s( d | _ t� | _d | _|| _| �|� d S r3 )rI r� �state�fdr+ �init)r/ r+ rY r r r r O s zPyDecoder.__init__c C s || _ dS )z� Override to perform decoder specific initialization :param args: Array of args items from the tile entry :returns: None N)rY r� r r r r� V s zPyDecoder.initc C s | j S r3 )� _pulls_fdr6 r r r rR _ s zPyDecoder.pulls_fdc C s t � �dS )a' Override to perform the decoding process. :param buffer: A bytes object with the data to be decoded. :returns: A tuple of (bytes consumed, errcode). If finished with decoding return <0 for the bytes consumed. Err codes are from `ERRORS` Nrk )r/ �bufferr r r rT c s zPyDecoder.decodec C s dS )zV Override to perform decoder specific cleanup :returns: None Nr r6 r r r rP n s zPyDecoder.cleanupc C s || _ dS )z� Called from ImageFile to set the python file-like object :param fd: A python file-like object :returns: None N)r� )r/ r� r r r rS v s zPyDecoder.setfdNc C s� || _ |r|\}}}}n d\}}}}|dkrJ|dkrJ| j j\| j_| j_n(|| j_|| j_|| | j_|| | j_| jjdks�| jjdkr�td��| jj| jj | j jd ks�| jj| jj | j jd kr�td��dS )z� Called from ImageFile to set the core output image for the decoder :param im: A core image object :param extents: a 4 tuple of (x0, y0, x1, y1) defining the rectangle for this tile :returns: None )r r r r r zSize cannot be negativer z Tile cannot extend outside imageN)rI r, r� r� r� r� r� � ValueError)r/ rI rW Zx0Zy0Zx1Zy1r r r rQ s" ��zPyDecoder.setimagec C sd |s | j }t�| j d|�}|�| j| j�� � |�|�}|d dkrLtd��|d dkr`td��dS )a Convenience method to set the internal image from a stream of raw data :param data: Bytes to be set :param rawmode: The rawmode to be used for the decoder. If not specified, it will default to the mode of the image :returns: None r9 r znot enough image datar zcannot decode image dataN) r+ r rO rQ rI r� rW rT r� )r/ rs Zrawmoder| r] r r r � set_as_raw� s zPyDecoder.set_as_raw)N)N)re rf rg rh r� r r� �propertyrR rT rP rS rQ r� r r r r r� E s #r� )r )rw r) rD � r Z_utilr r r� rU r r r r rj rr r� r� r� r� r r r r �<module> s0 � i >"