File: //usr/lib/python3/dist-packages/pdfminer/__pycache__/pdfdocument.cpython-38.pyc
U
�R�]0k � @ s� d dl Z d dlZd dlZd dlZz d dlZW n ek
rH d dlZY nX z(d dlmZ d dlm Z d dl
m Z W n( ek
r� d Z Z ddl m
Z Y nX ddlmZ ddlmZ dd lmZ dd
lmZ dd l mZ dd lmZ dd
lmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlm Z ddlm Z ddlm Z ddl m Z ddl m!Z! ddl"m#Z# ddl"m$Z$ ddl"m%Z% e�&e'�Z(G dd � d e �Z)G d d � d e�Z*G d d � d e�Z+G d!d"� d"e�Z,G d#d$� d$e,�Z-G d%d&� d&e,�Z.ed'�Z/ed(�Z0ed)�Z1G d*d+� d+e2�Z3G d,d-� d-e3�Z4G d.d/� d/e4�Z5G d0d1� d1e3�Z6G d2d3� d3e2�Z7G d4d5� d5e7�Z8G d6d7� d7e8�Z9G d8d9� d9e2�Z:dS ):� N)�ARC4)�AES)�SHA256� )�arcfour)�PSEOF)� literal_name)�LIT)�KWD)�settings)� PDFException)� PDFTypeError)� PDFStream)�PDFObjectNotFound)� decipher_all)� int_value)� str_value)�
list_value)�
dict_value)� stream_value)�PDFSyntaxError)�PDFStreamParser)�choplist)�nunpack)� decode_textc @ s e Zd ZdS )�PDFNoValidXRefN��__name__�
__module__� __qualname__� r r �6/usr/lib/python3/dist-packages/pdfminer/pdfdocument.pyr , s r c @ s e Zd ZdS )�
PDFNoOutlinesNr r r r r! r" / s r" c @ s e Zd ZdS )�PDFDestinationNotFoundNr r r r r! r# 2 s r# c @ s e Zd ZdS )�PDFEncryptionErrorNr r r r r! r$ 5 s r$ c @ s e Zd ZdS )�PDFPasswordIncorrectNr r r r r! r% 8 s r% c @ s e Zd ZdS )�PDFTextExtractionNotAllowedNr r r r r! r& ; s r& ZObjStmZXRefZCatalogc @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� PDFBaseXRefc C s t �d S �N)�NotImplementedError��selfr r r! � get_trailerH s zPDFBaseXRef.get_trailerc C s g S r( r r* r r r! �
get_objidsK s zPDFBaseXRef.get_objidsc C s t |��d S r( )�KeyError�r+ �objidr r r! �get_posQ s zPDFBaseXRef.get_posN)r r r r, r- r1 r r r r! r' F s r' c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dS )�PDFXRefc C s i | _ i | _d S r( )�offsets�trailerr* r r r! �__init__Y s zPDFXRef.__init__c C s d| j �� S )Nz<PDFXRef: offsets=%r>�r3 �keysr* r r r! �__repr__^ s zPDFXRef.__repr__c C s� z |� � \}}|�� sW q W n tk
r8 td��Y nX |sJtd| ��|�d�rb|�|� �q�|�� �d�}t|�dkr�td||f ��z(tj r�t
t |�\}}nt
t |�\}}W n$ t
k
r� td||f ��Y nX t||| �D ]�}z|� � \}}W n tk
�r td��Y nX |�� �d�}t|�dk�rHtd ||f ��|\}} }
|
d
k�r^q�d tj �rpt |�nt |�t | �f| j|< q�q t�d | j� | �|� d S ) Nz Unexpected EOF - file corrupted?zPremature eof: %r� trailer� � z Trailer not found: %r: line=%rzInvalid line: %r: line=%r� z Invalid XRef format: %r, line=%r� nzxref objects: %r)�nextline�stripr r �
startswith�seek�split�len�six�PY2�map�long�int�
ValueError�ranger3 �log�info� load_trailer) r+ �parser�pos�line�f�start�nobjsr0 �_�gennoZuser r r! �loada sF
,
z PDFXRef.loadc C s� z4|� � \}}|td�ks&tt|���|�� \}}W n6 tk
rj |�d�}|sZtd��|d \}}Y nX | j� t
|�� t � d| j� d S )Nr9 r z Unexpected EOF - file corruptedr z
trailer=%r)
� nexttokenr
�AssertionError�str�
nextobjectr �popr r4 �updater rK �debug)r+ rN rT �kwdZdic�xr r r! rM � s
zPDFXRef.load_trailerc C s | j S r( �r4 r* r r r! r, � s zPDFXRef.get_trailerc C s t �| j�S r( )rD Ziterkeysr3 r* r r r! r- � s zPDFXRef.get_objidsc C s( z | j | W S tk
r" � Y nX d S r( )r3 r. r/ r r r! r1 � s zPDFXRef.get_posN)
r r r r5 r8 rV rM r, r- r1 r r r r! r2 W s 'r2 c @ s& e Zd Zdd� Ze�d�Zdd� ZdS )�PDFXRefFallbackc C s d| j �� S )Nz <PDFXRefFallback: offsets=%r>r6 r* r r r! r8 � s zPDFXRefFallback.__repr__z^(\d+)\s+(\d+)\s+obj\bc C s� |� d� z|�� \}}W n tk
r4 Y �q�Y nX |�d�rf|� |� | �|� t�d| j� �q�tj rv|�
d�}| j � |�}|s�q
|�
� \}}t|�}t|�}d ||f| j|< |� |� |�� \}}t|t�r
|�d�tkr
t|�} z | d }
W n. tk
�r$ tj�r td| ��d}
Y nX t| �� �} g } z | �� \}}| � |� �q8W n tk
�rj Y nX t |
t | �d �}
t |
�D ]"}
| |
d }||
df| j|< �q�q
d S ) Nr r9 � trailer: %rzlatin-1�Type�N�N is not defined: %rr; ) rA r> r r@ rM rK rL r4 rD ZPY3�decode�
PDFOBJ_CUE�match�groupsrH r3 rZ �
isinstancer �get�LITERAL_OBJSTMr r. r �STRICTr r �get_data�append�minrC rJ )r+ rN rO rP �mr0 rU rT �obj�stream�nZparser1�objs�index�objid1r r r! rV � sR
zPDFXRefFallback.loadN)r r r r8 �re�compilerg rV r r r r! ra � s
ra c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )�
PDFXRefStreamc C s( d | _ d | _d | _ | _| _g | _d S r( )�data�entlen�fl1�fl2�fl3�rangesr* r r r! r5 � s
zPDFXRefStream.__init__c C s
d| j S )Nz<PDFXRefStream: ranges=%r>)r� r* r r r! r8 � s zPDFXRefStream.__repr__c C s� |� � \}}|� � \}}|� � \}}|�� \}}t|t�rF|d tk rNtd��|d }|�dd|f�}t|�d dkr~td��| j �
t d|�� |d \| _ | _
| _|�� | _| j | j
| j | _|j| _t�d d
�tt| j ��| j | j
| j� d S ) Nrc zInvalid PDF stream spec.ZSizeZIndexr r; zInvalid index number�Wz&xref stream: objid=%s, fields=%d,%d,%dz, )rW rZ rj r � LITERAL_XREFr rk rC r r� �extendr r} r~ r rn r{ r| �attrsr4 rK rL �joinrF �repr) r+ rN rT r0 rU r^ rs �sizeZ index_arrayr r r! rV � s,
�zPDFXRefStream.loadc C s | j S r( r` r* r r r! r, � s zPDFXRefStream.get_trailerc c sn | j D ]b\}}t|�D ]P}| j| }| j||| j � }t|d | j� d�}|dks\|dkr|| V qqd S )Nr r; )r� rJ r| r{ r r} )r+ rR rS �i�offset�ent�f1r r r! r- s
zPDFXRefStream.get_objidsc
C s� d}| j D ]6\}}||kr8||| k r8||| 7 } qJq
||7 }q
t|��| j| }| j||| j � }t|d | j� d�}t|| j| j| j � �}t|| j| j d � �} |dkr�d || fS |dkr�|| dfS t|��d S )Nr r r; )r� r. r| r{ r r} r~ )
r+ r0 rv rR rS r� r� r� �f2Zf3r r r! r1 s"
zPDFXRefStream.get_posN) r r r r5 r8 rV r, r- r1 r r r r! rz � s
rz c @ s� e Zd ZdZdZd#dd�Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zd d � Zd$d d �Zd!d"� Zd S )%�PDFStandardSecurityHandlers (�N^Nu�Ad NV��.. ��h>�/ ��dSiz)r; r< � c C s || _ || _|| _| �� d S r( )�docid�param�password�init)r+ r� r� r� r r r! r5 + s
z#PDFStandardSecurityHandler.__init__c C s. | � � | j| jkr"td| j ��| �� d S )Nz Unsupported revision: param=%r)� init_params�r�supported_revisionsr$ r� �init_keyr* r r r! r� 2 s
z PDFStandardSecurityHandler.initc C sl t | j�dd��| _t | jd �| _t | jd �| _t| jd �| _t| jd �| _t | j�dd��| _ d S ) N�Vr �R�P�O�UZLength�( )
r r� rk �vr� �pr �o�u�lengthr* r r r! r� 9 s z&PDFStandardSecurityHandler.init_paramsc C s | � | j�| _| jd kr t�d S r( )� authenticater� �keyr% r* r r r! r� B s
z#PDFStandardSecurityHandler.init_keyc C s t | jd@ �S )N� ��boolr� r* r r r! � is_printableH s z'PDFStandardSecurityHandler.is_printablec C s t | jd@ �S )N� r� r* r r r! �
is_modifiableK s z(PDFStandardSecurityHandler.is_modifiablec C s t | jd@ �S )N� r� r* r r r! �is_extractableN s z)PDFStandardSecurityHandler.is_extractablec s� | j dkr t�|��| j�S t�| j�}|�| jd � t�|��|�� �}t dd�D ]2� d�
� fdd�t � |�D ��}t�|��|�}qV||7 }|S d S )Nr; r r � � c 3 s | ]}t �|� A �V qd S r( �rD Zint2byte��.0�c�r� r r! � <genexpr>[ s z7PDFStandardSecurityHandler.compute_u.<locals>.<genexpr>)
r� r �newZencrypt�PASSWORD_PADDING�md5r\ r� �digestrJ r� rD � iterbytes)r+ r� �hash�result�kr r� r! � compute_uQ s
z$PDFStandardSecurityHandler.compute_uc C s� || j d d� }t�|�}|�| j� |�t�d| j�� |�| jd � | jdkrf| j sf|�d� |�
� }d}| jdkr�| j d }t d �D ]}t�|d |� ��
� }q�|d |� S )
N� z<lr r� s ����� r< r� �2 )
r� r� r\ r� �struct�packr� r� r� �encrypt_metadatar� r� rJ )r+ r� r� r� rt rT r r r! �compute_encryption_key` s
z1PDFStandardSecurityHandler.compute_encryption_keyc C s* |� d�}| �|�}|d kr&| �|�}|S )N�latin1)�encode�authenticate_user_password�authenticate_owner_password�r+ r� r� r r r! r� r s
z'PDFStandardSecurityHandler.authenticatec C s | � |�}| �|�r|S d S d S r( )r� �verify_encryption_keyr� r r r! r� y s
z5PDFStandardSecurityHandler.authenticate_user_passwordc C s8 | � |�}| jdkr || jkS |d d� | jd d� kS )Nr; r� )r� r� r� )r+ r� r� r r r! r� � s
z0PDFStandardSecurityHandler.verify_encryption_keyc s� || j d d� }t�|�}| jdkrBtd�D ]}t�|�� �}q.d}| jdkrZ| jd }|�� d |� }| jdkr�t�|��| j �}nF| j }tddd�D ]2� d �
� fd
d �t � |�D ��}t�|��|�}q�| �
|�S ) Nr� r< r� r� r� r; � ���r� c 3 s | ]}t �|� A �V qd S r( r� r� r� r r! r� � s zIPDFStandardSecurityHandler.authenticate_owner_password.<locals>.<genexpr>)r� r� r� rJ r� r� r r� �decryptr� r� rD r� r� )r+ r� r� rT rt r� Z
user_passwordr� r r� r! r� � s
z6PDFStandardSecurityHandler.authenticate_owner_passwordNc C s | � |||�S r( )� decrypt_rc4)r+ r0 rU r{ r� r r r! r� � s z"PDFStandardSecurityHandler.decryptc C sb | j t�d|�d d� t�d|�d d� }t�|�}|�� d tt|�d�� }t�|�� |�S )N�<Lr< r; r� )
r� r� r� r� r� rp rC r r� r� �r+ r0 rU r{ r� r� r r r! r� � s .
z&PDFStandardSecurityHandler.decrypt_rc4)r� )N)r r r r� r� r5 r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r! r� % s"
r� c sB e Zd ZdZ� fdd�Zdd� Zd
dd�Zd d
� Zd d � Z� Z S )� PDFStandardSecurityHandlerV4)r� c s� t t| ��� d| _t| j�d��| _t| jd �| _ t| jd �| _
t | j�dd��| _ | j | j
krtt
d| j ��i | _| j�� D ]:\}}| �t|d ��}|d kr�t
d | j ��|| j|< q�| j| jd
< | j
| jkr�t
d | j ��d S ) N� ZCFZStmFZStrFZEncryptMetadataTz"Unsupported crypt filter: param=%rZCFMz%Unknown crypt filter method: param=%rZIdentityz Undefined crypt filter: param=%r)�superr� r� r� r r� rk Zcfr Zstmf�strfr� r� r$ �cfm�items�get_cfm�decrypt_identity)r+ r� r� rQ �� __class__r r! r� � s$ z(PDFStandardSecurityHandlerV4.init_paramsc C s$ |dkr| j S |dkr | jS d S d S )NZV2ZAESV2)r� �decrypt_aes128�r+ �namer r r! r� � s
z$PDFStandardSecurityHandlerV4.get_cfmNc C sP | j s0|d k r0|�d�}|d k r0t|�dkr0|S |d kr>| j}| j| |||�S )Nrc ZMetadata)r� rk r r� r� )r+ r0 rU r{ r� r� �tr r r! r� � s
z$PDFStandardSecurityHandlerV4.decryptc C s |S r( r �r+ r0 rU r{ r r r! r� � s z-PDFStandardSecurityHandlerV4.decrypt_identityc C s~ | j t�d|�d d� t�d|�d d� d }t�|�}|�� d tt|�d�� }tj|tj |d d� d��
|dd � �S )Nr� r< r; s sAlTr� ��modeZIV) r� r� r� r� r� rp rC r r� �MODE_CBCr� r� r r r! r� � s 2
z+PDFStandardSecurityHandlerV4.decrypt_aes128)NN)
r r r r� r� r� r� r� r� �
__classcell__r r r� r! r� � s
r� c s8 e Zd ZdZ� fdd�Zdd� Zdd� Zdd � Z� ZS )
� PDFStandardSecurityHandlerV5)r� c s� t t| ��� d| _t| jd �| _t| jd �| _| jd d� | _ | jdd� | _
| jdd � | _ | j d d� | _
| j dd� | _| j dd � | _d S )N� ZOEZUEr� r� )r� r� r� r� r r� �oe�uer� �o_hash�o_validation_salt�
o_key_saltr� �u_hash�u_validation_salt�
u_key_saltr* r� r r! r� � s z(PDFStandardSecurityHandlerV5.init_paramsc C s |dkr| j S d S d S )NZAESV3)�decrypt_aes256r� r r r! r� � s z$PDFStandardSecurityHandlerV5.get_cfmc C s� |� d�d d� }t�|�}|�| j� |�| j� |�� | jkr�t�|�}|�| j� |�| j� t j|�� t j
dd�� | j �S t�|�}|�| j
� |�� | jkr�t�|�}|�| j� t j|�� t j
dd�� | j�S d S )Nzutf-8� s r� )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
z)PDFStandardSecurityHandlerV5.authenticatec C s* t j| jt j|d d� d��|dd � �S )Nr� r� )r r� r� r� r� r� r r r! r� s z+PDFStandardSecurityHandlerV5.decrypt_aes256) r r r r� r� r� r� r� r� r r r� r! r� � s
r� c @ s� e Zd ZdZeed�Zedk r6eed< edk r6e ed< d dd �Z
e d
�Z d d d �Z
d
d� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � ZdS )!� PDFDocumentaP PDFDocument object represents a PDF document.
Since a PDF file can be very big, normally it is not loaded at
once. So PDF document has to cooperate with a PDF parser in order to
dynamically import the data as processing goes.
Typical usage:
doc = PDFDocument(parser, password)
obj = doc.getobj(objid)
)r r; Nr� r� r� Tc C sb || _ g | _g | _d| _d| _d| _d| _i | _i | _|| _| j� | � d | _
| _ | _ z | �
|�}| �||| j� W n tk
r� Y nX |r�d|_t� }|�|� | j�|� | jD ]x}|�� }|s�q�d|kr�t|d �t|d �f| _| �|� d|k�r| j�t|d �� d|kr�t|d �| _ �q<q�td��| j�d�tk �r^tj �r^td ��dS )
z1Set the document to use a given PDFParser object.NTZEncryptZIDZInfoZRootz(No /Root object! - Is this really a PDF?rc zCatalog not found!) �caching�xrefsrL �catalog�
encryption�decipher�_parser� _cached_objs� _parsed_objs� set_documentr� r� r� � find_xref�read_xref_fromr �fallbackra rV ro r, r r �_initialize_passwordr rk �LITERAL_CATALOGr rm )r+ rN r� r� r� rO �xrefr4 r r r! r5 sR
�
zPDFDocument.__init__� objc C s� | j \}}t|�d��dkr(td| ��t|�dd��}| j�|�}|d krXtd| ��||||�}|j| _|�� | _|� � | _ |�
� | _
d| j _ d S )NZFilterZStandardzUnknown filter: param=%rr� r zUnknown algorithm: param=%rF)
r� r rk r$ r �security_handler_registryr� r� r� r� r� r� r� )r+ r� r� r� r� �factoryZhandlerr r r! r� S s
z PDFDocument._initialize_passwordc C s� |j | jkr | j|j \}}n$| �|�\}}| jrB||f| j|j < |d | }z || }W n tk
rz td| ��Y nX |S )Nr; zindex too big: %r)r0 r� � _get_objectsr� �
IndexErrorr )r+ rs rv r0 ru rt r� rr r r r! �_getobj_objstmc s zPDFDocument._getobj_objstmc C s� |� d�tk r tjr td| ��z |d }W n* tk
rV tjrNtd| ��d}Y nX t|�� �}|�| � g }z |� � \}}|�
|� qtW n t k
r� Y nX ||fS )Nrc zNot a stream object: %rrd re r ) rk rl r rm r r. r rn r� rZ ro r )r+ rs rt rN ru rT rr r r r! r q s$
zPDFDocument._get_objectsc C s� | j �|� | j �� \}}| j �� \}}| j �� \}}||krzg }|| jk rf| j �� \}}|�|� qB|rz|d }|d }||kr�td||f ��|td�kr�td| ��| j �� \}}|S )N���r� zobjid mismatch: %r=%rr� z Invalid object spec: offset=%r)r� rA rW � KEYWORD_OBJro r r
rZ ) r+ rO r0 rT rw rU r^ r_ rr r r r! �
_getobj_parse� s$
zPDFDocument._getobj_parsec
C s2 |dks t �| jstd��t�d|� || jkr@| j| \}}n�| jD ]�}z|�|�\}}}W n tk
rv Y qFY nX zj|d k r�t| � |��}| �
|||�}n"| � ||�}| j r�t
| j |||�}t|t�r�|�||� W �q
W qF ttfk
r� Y qFY qFX qFt|��t�d||� | j�r.||f| j|< |S )Nr z PDFDocument is not initializedzgetobj: objid=%rzregister: objid=%r: %r)rX r� r rK r] r� r1 r. r �getobjr r r� r rj r Z set_objidr r r r� )r+ r0 rr rU r� Zstrmidrv rs r r r! r � s8
zPDFDocument.getobjc s* d| j krt�� fdd�� � | j d d�S )NZOutlinesc 3 s� t | �} d| kr^d| ks d| kr^tt| d ��}| �d�}| �d�}| �d�}|||||fV d| kr�d| kr�� | d |d �D ]
}|V q�d| kr�� | d |�D ]
}|V q�d S ) NZTitle�AZDestZSEZFirstZLastr ZNext)r r r rk )�entry�level�title�dest�actionZser_ ��searchr r! r � s
z(PDFDocument.get_outlines.<locals>.searchr )r� r" r* r r r! � get_outlines� s
zPDFDocument.get_outlinesc s\ zt | jd �}W n$ ttfk
r6 t� �f��Y nX t |� �}� ��fdd���|�S )N�Namesc s� d| kr,t | d �\}}�|k s(|�k r,d S d| krVt | d �}ttd|��}|� S d| kr�t | d �D ] }�t|��}|rj| S qjt� �f��d S )NZLimitsr r; ZKids)r �dictr r r. )�dZk1Zk2ru �namesr� r� ��catr� �lookupr r! r � s
z'PDFDocument.lookup_name.<locals>.lookup)r r� r
r. )r+ r r� r �d0r r r! � lookup_name� s zPDFDocument.lookup_namec C sb z| � d|�}W nL tk
r\ d| jkr2t|��t| jd �}||krPt|��|| }Y nX |S )NZDests)r r. r� r# r )r+ r� rr r r r r! �get_dest� s
zPDFDocument.get_destc C sd d}|� � D ],}|�� }t�d|� |dkr0 qB|r |}q td��t�d|� tjr\t|�S t |�S )z0Internal function used to locate the first XRef.Nz
find_xref: %rs startxref�Unexpected EOFzxref found: pos=%r)
Z revreadlinesr? rK r] r rL rD rE rG rH )r+ rN �prevrP r r r! r� s zPDFDocument.find_xrefc C s |� |� |�� z|�� \}}W n tk
r> td��Y nX t�d||� t|t�r||� |� |�� t � }|�
|� n"||j kr�|� � t
� }|�
|� |�|� |�� }t�d|� d|kr�t|d �}| �|||� d|k�rt|d �}| �|||� dS )z$Reads XRefs from the given location.r z"read_xref_from: start=%d, token=%rrb ZXRefStmZPrevN)rA �resetrW r r rK rL rj rH rz rV Z KEYWORD_XREFr> r2 ro r, r r� )r+ rN rR r� rO �tokenr� r4 r r r! r� s4
zPDFDocument.read_xref_from)r� TT)r� )r r r �__doc__r� r r r� r r� r5 r
r r� r r r r r r r r� r� r r r r! r� s(
�
1
"r� );rx r� ZloggingrD Zhashlibr� � ImportErrorZ
Crypto.Cipherr r Z Crypto.Hashr r� r Zpsparserr r r r
r Zpdftypesr r
r r r r r r r r Z pdfparserr r Zutilsr r r Z getLoggerr rK r r" r# r$ r% r& rl r� r� �objectr' r2 ra rz r� r� r� r� r r r r! �<module> sj
N8H 3/