o
� @ s� d dl Zd dlZd dlZd dlZd dlZdZd ZdZdZ dZ
G dd� d�Z G dd � d �Z G d
d � d e
�ZG d d
� d
�ZG dd� d�ZG dd� de�ZdS )� N� � � � c @ s e Zd ZdZdd� ZdS )�FilePosz`
This class keeps track of the file object representing and current position in a file.
c C s d | _ d| _d S )Nr )�file�pos��self� r �&PyInstaller/loader/pyimod01_archive.py�__init__- s
zFilePos.__init__N)�__name__�
__module__� __qualname__�__doc__r
r r r r r ) s r c @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd S )
� ArchiveFilez~
File class support auto open when access member from file object This class is use to avoid file locking on windows.
c O s || _ || _i | _d S )N)�args�kwargs�_filePos)r
r r r r r r
8 s
zArchiveFile.__init__c C s( t �� }|| jvrt� | j|< | j| S )z�
Return an instance of FilePos for the current thread. This is a crude # re-implementation of threading.local,
which isn't a built-in module # and therefore isn't available.
)�thread� get_identr r )r
�tir r r �local= s
zArchiveFile.localc C s | � � j}|s J �t||�S )zj
Make this class act like a file, by invoking most methods on its underlying file object.
)r r �getattr)r
�namer r r r � __getattr__G s
zArchiveFile.__getattr__c C s8 | � � }|jr J �t| ji | j��|_|j�|j� dS )zC
Open file and seek to pos record from last close.
N)r r �openr r �seekr )r
�fpr r r � __enter__O s
zArchiveFile.__enter__c C s2 | � � }|js J �|j�� |_|j�� d|_dS )z,
Close file and record pos.
N)r r �tellr �close)r
�type�value� tracebackr r r r �__exit__Z s
zArchiveFile.__exit__N) r r r r r
r r r r& r r r r r 4 s
r c @ s e Zd ZdS )�ArchiveReadErrorN)r r r r r r r r' h s r' c @ sV e Zd ZdZdZdZdZdZdZddd�Z d d
� Z
d d � Z d
d� Z dd� Z
dd� ZdS )�
ArchiveReadera
A base class for a repository of python code objects. The extract method is used by imputil.ArchiveImporter to
get code objects by name (fully qualified name), so an end-user "import a.b" becomes:
extract('a.__init__')
extract('a.b')
s PYL � � Nr c C sz d| _ || _|| _ddl}|jj| _|dur;t| jd�| _| j� | � � | �
� W d � dS 1 s4w Y dS dS )zY
Initialize an Archive. If path is omitted, it will be an empty Archive.
Nr �rb) �toc�path�start�_frozen_importlib�_bootstrap_external� MAGIC_NUMBER�pymagicr �lib�
checkmagic�loadtoc)r
r- r. r/ r r r r
y s
"��zArchiveReader.__init__c C sV | j �| j| j � t�d| j �d��\}| j �| j| � tt� | j �� ��| _
dS )z�
Overridable. Default: After magic comes an int (4 byte native) giving the position of the TOC within
self.lib. Default: The TOC is a marshal-able string.
z!i� N) r3 r r. �TOCPOS�struct�unpack�read�dict�marshal�loadsr, )r
�offsetr r r r5 � s zArchiveReader.loadtocc C s& | j �|d�\}}|d u rd S t|�S )N�r N)r, �get�bool)r
r �ispkgr r r r �
is_package� s zArchiveReader.is_packagec C st | j �|d�\}}|du rdS | j� | j�| j| � t�| j�� �}W d � ||fS 1 s1w Y ||fS )a+
Get the object corresponding to name, or None. For use with imputil ArchiveImporter, object is a python code
object. 'name' is the name as specified in an 'import name'. 'import a.b' becomes:
extract('a') (return None because 'a' is not a code object)
extract('a.__init__') (return a code object)
extract('a.b') (return a code object)
Default implementation:
self.toc is a dict
self.toc[name] is pos
self.lib has the code object marshal-ed at pos
r? N)r, r@ r3 r r. r< r= r: )r
r rB r �objr r r �extract� s
��zArchiveReader.extractc C s t | j�� �S )z�
Return a list of the contents Default implementation assumes self.toc is a dict like object. Not required by
ArchiveImporter.
)�listr, �keysr r r r �contents� s zArchiveReader.contentsc C sr | j �| j� | j �t| j��| jkr td| j| jj f ��| j �t| j
��| j
kr1td| j ��| j �d� dS )zj
Overridable. Check to see if the file object self.lib actually has a file we understand.
z!%s is not a valid %s archive filez %s has version mismatch to dllr6 N) r3 r r. r: �len�MAGICr' r- � __class__r r2 r r r r r4 � s zArchiveReader.checkmagic)Nr )r r r r rJ �HDRLENr7 �os� _bincacher
r5 rC rE rH r4 r r r r r( l s
r( c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �Cipherz<
This class is used only to decrypt Python modules.
c C sr dd l }|j}t|�tu sJ �t|�tkr |dt� | _n|�t�| _t| j�tks,J �dd l}|| _t j
d= d S )Nr �tinyaes) �pyimod00_crypto_key�keyr# �strrI �CRYPT_BLOCK_SIZE�zfillrP �_aesmod�sys�modules)r
rQ rR rP r r r r
� s zCipher.__init__c C s | j �| j�� |�S )N)rV �AESrR �encode)r
�ivr r r �__create_cipher� s zCipher.__create_cipherc C s$ | � |d t� �}|�|td � �S )N)�_Cipher__create_cipherrT �CTR_xcrypt_buffer)r
�data�cipherr r r �decrypt� s zCipher.decryptN)r r r r r
r] ra r r r r rO � s
rO c sL e Zd ZdZdZdZejd Zd� fdd� Zdd � Z d
d � Z
d d
� Z � Z S )�ZlibArchiveReadera1
ZlibArchive - an archive with compressed entries. Archive is read from the executable created by PyInstaller.
This archive is used for bundling python modules inside the executable.
NOTE: The whole ZlibArchive (PYZ) is compressed, so it is not necessary to compress individual modules.
s PYZ r* � Nc s� |d u rd}n7|d u r>t t|�d dd�D ]&}|| dkr;z t||d d � �}W n ty2 Y qw |d |� } nqd}t� �||� z dd l}t� | _W d S t y] d | _Y d S w )Nr r ����?)
�rangerI �int�
ValueError�superr
rQ rO r` � ImportError)r
r- r> �irQ �rK r r r
� s* � � �zZlibArchiveReader.__init__c C s, | j �|d�\}}}|d u rd S |ttfv S )N�r Nr )r, r@ � PYZ_TYPE_PKG�PYZ_TYPE_NSPKG�r
r �typr �lengthr r r rC s z ZlibArchiveReader.is_packagec C s( | j �|d�\}}}|d u rd S |tkS )Nrm )r, r@ ro rp r r r �is_pep420_namespace_package s z-ZlibArchiveReader.is_pep420_namespace_packagec
C s� | j �|d�\}}}|d u rd S z$| j� | j�| j| � | j�|�}W d � n1 s.w Y W n tyC t| j� d���w z$| j rN| j �
|�}t � |�}|t
ttfv rdt�|�}W ||fS W ||fS ty{ } ztd| �|�d }~ww )Nrm z� appears to have been moved or deleted since this application was launched. Continouation from this state is impossible. Exiting now.z"PYZ entry '%s' failed to unmarshal)r, r@ r3 r r. r: �FileNotFoundError�
SystemExitr- r` ra �zlib�
decompress�PYZ_TYPE_MODULErn ro r< r= �EOFErrorrj )r
r rq r rr rD �er r r rE s6 ��
��
����zZlibArchiveReader.extract)NN)
r r r r rJ r7 r( rL r
rC rs rE �
__classcell__r r rl r rb � s
rb )�_threadr r< r8 rW rv rT rx rn �
PYZ_TYPE_DATAro r r � RuntimeErrorr' r( rO rb r r r r �<module> s 4`!