File: //usr/lib64/python3.6/site-packages/borg/crypto/__pycache__/file_integrity.cpython-36.pyc
3
y��b � @ s� d dl Z d dlZd dlZd dlZd dlmZ ddlmZ ddlm Z ddl
m Z e � ZG dd� d�Z G d d
� d
e �Z
G d d � d e
�ZG d
d� de
�ZejeejeiZG dd� de�ZG dd� de �ZG dd� de�ZdS )� N)�compare_digest� )�IntegrityError)�
create_logger)�StreamingXXH64c @ sT e Zd Zdd� Zdd� Zdd� Zejfdd�Zd d
� Z dd d
�Z
dd� Z dd� Z d S )�FileLikeWrapperc C s | j j� | S )N)�fd� __enter__)�self� r �&/usr/lib64/python3.6/file_integrity.pyr s
zFileLikeWrapper.__enter__c C s | j j|||� d S )N)r �__exit__)r
�exc_type�exc_val�exc_tbr r r r
s zFileLikeWrapper.__exit__c C s
| j j� S )N)r �tell)r
r r r r s zFileLikeWrapper.tellc C s | j j||�S )N)r �seek)r
�offset�whencer r r r s zFileLikeWrapper.seekc C s | j j|�S )N)r �write)r
�datar r r r s zFileLikeWrapper.writeNc C s | j j|�S )N)r �read)r
�nr r r r s zFileLikeWrapper.readc C s | j j� d S )N)r �flush)r
r r r r " s zFileLikeWrapper.flushc C s
| j j� S )N)r �fileno)r
r r r r % s zFileLikeWrapper.fileno)N)
�__name__�
__module__� __qualname__r r
r �io�SEEK_SETr r r r r r r r r r s
r c sj e Zd ZdZdZdZdd� Z� fdd�Z� fdd�Zd� fd d
� Z d d � Z
e d
�dd�Z ddd�Z
� ZS )�FileHashingWrappera'
Wrapper for file-like objects that computes a hash on-the-fly while reading/writing.
WARNING: Seeks should only be used to query the size of the file, not
to skip data, because skipped data isn't read and not hashed into the digest.
Similarly skipping while writing to create sparse files is also not supported.
Data has to be read/written in a symmetric fashion, otherwise different
digests will be generated.
Note: When used as a context manager read/write operations outside the enclosed scope
are illegal.
Nc C s || _ || _| j� | _d S )N)r �writing�FACTORY�hash)r
�
backing_fdr r r r �__init__< s zFileHashingWrapper.__init__c s$ |d kr| j � t� j|||� d S )N)� hash_length�superr
)r
r r r )� __class__r r r
A s zFileHashingWrapper.__exit__c s t � j|�}| jj|� |S )zI
Write *data* to backing file and update internal state.
)r' r r# �update)r
r r )r( r r r F s zFileHashingWrapper.writec s t � j|�}| jj|� |S )zf
Read *data* from backing file (*n* has the usual meaning) and update internal state.
)r' r r# r) )r
r r )r( r r r N s zFileHashingWrapper.readc C s
| j j� S )zn
Return current digest bytes as hex-string.
Note: this can be called multiple times.
)r# � hexdigest)r
r r r r* V s z FileHashingWrapper.hexdigest)r c C s | j j|� d S )N)r# r) )r
r r r r r) ^ s zFileHashingWrapper.updateFc C s. |r| j dtj� | jjt| j� �j� � d S )Nr )r r �SEEK_ENDr# r) �strr �encode)r
� seek_to_endr r r r& a s z FileHashingWrapper.hash_length)N)F)r r r �__doc__� ALGORITHMr" r% r
r r r* �bytesr) r& �
__classcell__r r )r( r r ) s r c @ s e Zd ZdZejZdS )�SHA512FileHashingWrapperZSHA512N)r r r r0 �hashlibZsha512r" r r r r r3 h s r3 c @ s e Zd ZdZeZdS )�XXH64FileHashingWrapperZXXH64N)r r r r0 r r" r r r r r5 m s r5 c @ s e Zd ZdZdS )�FileIntegrityErrorz File failed integrity check: {}N)r r r r/ r r r r r6 x s r6 c @ s\ e Zd Zddd�Zdd� Zddd�Zeeed�d d
��Zdd d
�Z dd� Z
ed�dd�Z dS )�IntegrityCheckedFileNc
C s| || _ || _|rdnd}|p$t||�| _i | _t}|sX| j||�}|rX|\} | _t| }|| j|d� | _| _ | j
|� d S )N�wb�rb)r$ r ) �pathr! �openZfile_fd�digestsr5 �load_integrity_data�SUPPORTED_ALGORITHMSr �hasher�
hash_filename)
r
r: r �filename� override_fd�integrity_data�modeZhash_clsZalgorithm_and_digests� algorithmr r r r% } s
z IntegrityCheckedFile.__init__c C s |d k r| j ||�S d S )N)�parse_integrity_data)r
r: rC r r r r= � s z(IntegrityCheckedFile.load_integrity_datac C s> t jj|p| j�}| jjdt|� j� � | jj|j� � d S )Nz%10d)�osr: �basenamer? r) �lenr- )r
rA r r r r@ � s z"IntegrityCheckedFile.hash_filename)r: r c C s� yDt j|�}|d }|tkr.tjd||� d S |d }|d ||fS tttfk
r� } z tjd||� t|��W Y d d }~X nX d S )NrE z3Cannot verify integrity of %s: Unknown algorithm %rr< �finalz)Could not parse integrity data for %s: %s) �json�loadsr> �logger�warning�
ValueError� TypeError�KeyErrorr6 )�clsr: r rC rE r< �er r r rF � s
z)IntegrityCheckedFile.parse_integrity_dataFc C s� | j r| j rd S | jjdt|� j� � | jj|j� � | jj|d� | jj� }| j rf|| j|< n&| jr�t| jj |d�|� r�t
| j ��d S )Nz%10d)r. � ) r! r< r? r) rI r- r& r* r �getr6 r: )r
Zpartname�is_finalZdigestr r r � hash_part� s
z IntegrityCheckedFile.hash_partc C sn |d k }|s| j ddd� | jj|||� |r2d S | jrV| jtj| jj| jd��� n| jrjt j
d| j � d S )NrJ T)rV )rE r< zVerified integrity of %s) rW r? r
r! �store_integrity_datarK �dumpsr0 r< rM �debugr: )r
r r r Z exceptionr r r r
� s z IntegrityCheckedFile.__exit__)r c C s
|| _ d S )N)rC )r
r r r r rX � s z)IntegrityCheckedFile.store_integrity_data)NNN)N)F) r r r r% r= r@ � classmethodr, rF rW r
rX r r r r r7 | s
r7 c sL e Zd Zd
� fdd� Zdd� Zedd� �Zedd � �Ze d
�d d �Z
� Z S )� DetachedIntegrityCheckedFileNc sH t � j||||� |p tjj|�}tjj|�}| jtjj||��| _d S )N) r' r% rG r: rH �dirname�integrity_file_path�join�output_integrity_file)r
r: r rA rB Z
output_dir)r( r r r% � s z%DetachedIntegrityCheckedFile.__init__c C s | st d��| j| j�S )NzCCannot pass explicit integrity_data to DetachedIntegrityCheckedFile)�AssertionError�read_integrity_filer: )r
r: rC r r r r= � s z0DetachedIntegrityCheckedFile.load_integrity_datac C s | d S )Nz
.integrityr )r: r r r r^ � s z0DetachedIntegrityCheckedFile.integrity_file_pathc C s� y,t | j|�d��}| j||j� �S Q R X W nV tk
rL tjd|� Y n8 tk
r� } z tjd||� t |��W Y d d }~X nX d S )N�rz No integrity file found for %sz(Could not read integrity file for %s: %s)
r; r^ rF r �FileNotFoundErrorrM �info�OSErrorrN r6 )rR r: r rS r r r rb � s z0DetachedIntegrityCheckedFile.read_integrity_file)r c C s& t | jd��}|j|� W d Q R X d S )N�w)r; r` r )r
r r r r r rX � s z1DetachedIntegrityCheckedFile.store_integrity_data)NN) r r r r% r= � staticmethodr^ r[ rb r, rX r2 r r )r( r r\ � s
r\ )r4 r rK rG Zhmacr Zhelpersr rM r Zalgorithms.checksumsr r r r3 r5 r0 r> r6 r7 r\ r r r r �<module> s" ?
U