File: //lib/python3.9/site-packages/pycriu/images/__pycache__/images.cpython-39.pyc
a
k�x]�H � < @ s4 d dl Z d dlZd dlZd dlZd dlZddlmZ ddlmZ ddlmZ de e�vrhej
e_ ej e_
dZdZd Zd
d � ZG d d
� d
e�ZG dd� d�ZG dd� d�ZG dd� d�ZG dd� d�ZG dd� d�ZG dd� d�ZG dd� d�ZG d d � d �ZG d d � d �Z eej �eej �eej �eej �eej!�eej"�eej#�e� eej$�eej%�eej&e� �eej'�e� eej(�eej)�eej*�eej+�eej,�eej-�eej.�eej/�eej0�eej1�eej2�eej3�eej4�eej5�eej6�eej7�eej8�eej9�eej:�eej;�eej<�eej=�eej>�eej?�eej@�eejA�eejB�eejC�eejD�eejE�eejF�eejG�eejH�eejI�eejJe� �eejJe� �eejKe� �eejLe � �eejMe� �eejNe� �eejO�eejP�eejQ�eejR�eejS�eejT�d �;ZUd!d"� ZVd.d$d%�ZWd&d'� ZXd/d(d)�ZYd*d+� ZZd,d-� Z[dS )0� N� ��magic)�pb)�pb2dict� encodebytes� � � c C s | d |d B d S �Nr � )�x�yr r �8/usr/lib/python3.9/site-packages/pycriu/images/images.py�round_up> s r c @ s e Zd Zdd� ZdS )�MagicExceptionc C s
|| _ d S �Nr )�selfr r r r �__init__C s zMagicException.__init__N)�__name__�
__module__� __qualname__r r r r r r B s r c @ sF e Zd ZdZddd�Zddd�Zddd �Zd
d � Zd d
� Zdd� Z dS )�
entry_handlerzp
Generic class to handle loading/dumping criu images
entries from/to bin format to/from dict(json).
Nc C s || _ || _dS )z=
Sets payload class and extra handler class.
N)�payload�
extra_handler)r r r r r r r O s zentry_handler.__init__Fc C s� g }i }| � � }|�d�}|dkr$q�t�d|�\}|�|�|�� t�||�}| jr�|r�dd� } | j�||�}
d| |
� |d< n| j�||�|d< |� |� q|S )z�
Convert criu image entries from binary format to dict(json).
Takes a file-like object and returnes a list with entries in
dict(json) format.
r � �ic S sN dD ]@}| dk r<t | �| kr,d| |f S d| |f S | d } qd| S )N)� �K�M�G�T�P�E�Zg �@z%d%sBz%.1f%sBz%.1fYB)�int)ZnumZunitr r r �human_readablen s
z*entry_handler.load.<locals>.human_readablez... <%s>�extra)
r �read�struct�unpack�ParseFromStringr r �skip�load�append) r �f�pretty�
no_payload�entries�entry�pbuff�buf�sizer& Zpl_sizer r r r- V s"
zentry_handler.loadc C s t �|�}| �||�S )zD
Same as load(), but takes a string as an argument.
��io�BytesIOr- �r �sr0 r/ r r r �loads� s
zentry_handler.loadsc C st |D ]j}|� dd�}| �� }t�||� |�� }t|�}|�t�d|�� |�|� | j r|r| j �
|||� qdS )z�
Convert criu image entries from dict(json) format to binary.
Takes a list of entries and a file-like object to write entries
in binary format to.
r' Nr ) �popr r �dict2pb�SerializeToString�len�writer) �packr �dump)r r2 r/ r3 r' r4 �pb_strr6 r r r rC � s
zentry_handler.dumpc C s t �d�}| �||� |�� S )zf
Same as dump(), but doesn't take file-like object and just
returns a string.
r �r8 r9 rC r( �r r2 r/ r r r �dumps� s
zentry_handler.dumpsc C s@ d}|� d�}|dkrq<t�d|�\}|�|d� |d7 }q|S )zI
Counts the number of top-level object in the image file
r r r r r )r( r) r* �seek)r r/ r2 r5 r6 r r r �count� s
zentry_handler.count)N)FF)F)
r r r �__doc__r r- r< rC rG rI r r r r r I s
+
r c @ s< e Zd ZdZddd�Zddd�Zdd� Zd d
� Zd d � Zd
S )�pagemap_handlerz�
Special entry handler for pagemap.img, which is unique in a way
that it has a header of pagemap_head type followed by entries
of pagemap_entry type.
Fc C s^ g }t �� }|�d�}|dkr qZt�d|�\}|�|�|�� |�t�||�� t �� }q |S )Nr r r ) r � pagemap_headr( r) r* r+ r. r �
pagemap_entry)r r/ r0 r1 r2 r4 r5 r6 r r r r- � s
zpagemap_handler.loadc C s t �|�}| �||�S r r7 r: r r r r< � s
zpagemap_handler.loadsc C sV t �� }|D ]D}t�||� |�� }t|�}|�t�d|�� |�|� t � � }q d S )Nr )
r rL r r> r? r@ rA r) rB rM �r r2 r/ r4 �itemrD r6 r r r rC � s
zpagemap_handler.dumpc C s t �d�}| �||� |�� S �Nr rE rF r r r rG � s
zpagemap_handler.dumpsc C s t d ��|�d S r )r rI )r r/ r r r rI � s zpagemap_handler.countN)FF)F) r r r rJ r- r< rC rG rI r r r r rK � s
rK c @ s0 e Zd Zd dd�Zd dd�Zdd� Zdd � Zd
S )
�ghost_file_handlerFc C s g }t �� }|�d�}t�d|�\}|�|�|�� t�||�}|jr�|�|� t � � } |�d�}|dkrlq�t�d|�\}| �|�|�� t�| |�}
|r�|�
| j t j
� nt�|�| j ��|
d< |�|
� qPn0|r�|�
dt j� nt�|�� �|d< |�|� |S )Nr r r r' r )r �ghost_file_entryr( r) r* r+ r �chunksr. �ghost_chunk_entryrH r@ �os�SEEK_CUR�base64r �SEEK_END) r r/ r0 r1 r2 Zgfr5 r6 Zg_entry�gcr3 r r r r- � s0
zghost_file_handler.loadc C s t �|�}| �||�S r r7 r: r r r r<
s
zghost_file_handler.loadsc C s� t �� }|�d�}t�||� |�� }t|�}|�t� d|�� |�|� |j
r�|D ]X}t � � }t�||� |�� }t|�}|�t� d|�� |�|� |�t �
|d �� qTn|�t �
|d �� d S )Nr r r' )r rR r= r r> r? r@ rA r) rB rS rT rW � decodebytesrN r r r rC s"
zghost_file_handler.dumpc C s t �d�}| �||� |�� S rP rE rF r r r rG # s
zghost_file_handler.dumpsN)FF)F)r r r r- r< rC rG r r r r rQ � s
!
rQ c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�pipes_data_extra_handlerc C s |j }|�|�}t�|�S r )�bytesr( rW r �r r/ �ploadr6 �datar r r r- 0 s
z pipes_data_extra_handler.loadc C s t �|�}|�|� d S r �rW rZ rA )r r' r/ r^ r_ r r r rC 5 s
z pipes_data_extra_handler.dumpc C s |� |jtj� |jS r )rH r\ rU rV �r r/ r^ r r r r, 9 s z pipes_data_extra_handler.skipN�r r r r- rC r, r r r r r[ / s r[ c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�sk_queues_extra_handlerc C s |j }|�|�}t�|�S r )�lengthr( rW r r] r r r r- ? s
z sk_queues_extra_handler.loadc C s t �|�}|�|� d S r r` )r r' r/ �_unusedr_ r r r rC D s
z sk_queues_extra_handler.dumpc C s |� |jtj� |jS r )rH rd rU rV ra r r r r, H s z sk_queues_extra_handler.skipNrb r r r r rc > s rc c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�tcp_stream_extra_handlerc C s< i }|� |j�}|� |j�}t�|�|d< t�|�|d< |S �N�inq�outq)r( �inq_len�outq_lenrW r )r r/ r4 �drh ri r r r r- N s z tcp_stream_extra_handler.loadc C s4 t �|d �}t �|d �}|�|� |�|� d S rg r` )r r' r/ re rh ri r r r rC Y s
z tcp_stream_extra_handler.dumpc C s |� dtj� |j|j S )Nr )rH rU rX rj rk )r r/ r4 r r r r, ` s z tcp_stream_extra_handler.skipNrb r r r r rf M s rf c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�ipc_sem_set_handlerc C sd t � |�}t|d }t|t�}t�d�}|jtkr<td��|�|�|�� |� || d� |�
� S )N�nsems�H�Array size mismatchr ) r �
sizeof_u16r �
sizeof_u64�array�itemsize� ExceptionZ
fromstringr( rH �tolist)r r/ r4 r3 r6 �roundedr; r r r r- f s
zipc_sem_set_handler.loadc C s� t � |�}t|d }t|t�}t�d�}|jtkr<td��|�|� t|�|d kr^td��|� |�
� � |� d|| � d S )Nrn ro rp z Number of semaphores mismatch� ) r rq r rr rs rt ru �fromlistr@ rA Ztostring)r r' r/ r4 r3 r6 rw r; r r r rC q s
zipc_sem_set_handler.dumpc C s. t � |�}t|d }|�t|t�tj� |S )Nrn )r rq rH r rr rU rV )r r/ r4 r3 r6 r r r r, ~ s
zipc_sem_set_handler.skipNrb r r r r rm e s
rm c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�ipc_msg_queue_handlerc C s� t � |�}g }td|d �D ]�}|�d�}|dkr6 q�t�d|�\}t�� }|�|�|�� t|j t
�} |�|j �}
|� | |j d� |� t � |�� |� t
�|
�� q |S )Nr �qnumr r r r )r �ranger( r) r* r �ipc_msgr+ r �msizerr rH r. rW r ) r r/ r4 r3 �messagesr
r5 r6 �msgrw r_ r r r r- � s
zipc_msg_queue_handler.loadc C s� t � |�}tdt|�d�D ]�}t�� }t �|| |� |�� }t|�}|�t� d|�� |�|� t
|j t �} t
�||d �}
|�|
d |j � � |�d| |j � qd S )Nr r r r rx )r r| r@ r r} r> r? rA r) rB r r~ rr rW rZ ) r r' r/ r4 r3 r r� Zmsg_strr6 rw r_ r r r rC � s
zipc_msg_queue_handler.dumpc
C s� t � |�}d}td|d �D ]h}|�d�}|dkr6 q�t�d|�\}t�� }|�|�|�� t|j t
�} |� | t j
� |||j 7 }q |S )Nr r{ r r r )r r| r( r) r* r r} r+ r r~ rr rH rU rV )
r r/ r4 r3 Zpl_lenr
r5 r6 r� rw r r r r, � s
zipc_msg_queue_handler.skipNrb r r r r rz � s rz c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�ipc_shm_handlerc C s@ t � |�}|d }|�|�}t|t�}|�|| d� t�|�S )Nr6 r )r r( r �
sizeof_u32rH rW r )r r/ r4 r3 r6 r_ rw r r r r- � s
zipc_shm_handler.loadc C sN t � |�}|d }t�|�}t|t�}|�|d |� � |�d|| � d S )Nr6 rx )r rW rZ r r� rA )r r' r/ r4 r3 r6 r_ rw r r r rC � s
zipc_shm_handler.dumpc C s. t � |�}|d }t|t�}|�|tj� |S )Nr6 )r r r� rH rU rV )r r/ r4 r3 r6 rw r r r r, � s
zipc_shm_handler.skipNrb r r r r r� � s r� );� INVENTORYZCOREZIDSZCREDSZUTSNSZIPC_VARZFSZ
GHOST_FILEZMMZCGROUPZ
TCP_STREAM�STATSZPAGEMAPZPSTREEZ REG_FILESZNS_FILESZ EVENTFD_FILEZEVENTPOLL_FILEZ
EVENTPOLL_TFDZSIGNALFDZTIMERFDZ INOTIFY_FILEZ
INOTIFY_WDZ
FANOTIFY_FILEZ
FANOTIFY_MARKZVMASZPIPESZFIFOZSIGACTZ
NETLINK_SKZ REMAP_FPATHZMNTSZ TTY_FILESZTTY_INFOZTTY_DATAZRLIMITZTUNFILEZ EXT_FILES� IRMAP_CACHEZ
FILE_LOCKSZFDINFOZUNIXSKZINETSKZPACKETSKZITIMERSZ POSIX_TIMERSZNETDEVZ
PIPES_DATAZ FIFO_DATAZ SK_QUEUESZ IPCNS_SHMZ IPCNS_SEMZ IPCNS_MSGZNETNSZUSERNSZSECCOMPZAUTOFSZFILESZCPUINFOc C s� t �d| �d��\}|tjd tjd fv r@t �d| �d��\}ztj| }W n t|��Y n0 z t| }W n td| ��Y n0 ||fS )Nr r �
IMG_COMMON� IMG_SERVICEz&No handler found for image with magic ) r) r* r( r �by_nameZby_valr �handlersru )r/ Z img_magic�m�handlerr r r �
__rhandler s
� r� Fc C s. i }t | �\}}||d< |�| ||�|d< |S )z�
Convert criu image from binary format to dict(json).
Takes a file-like object to read criu image from.
Returns criu image in dict(json) format.
r r2 )r� r- )r/ r0 r1 Zimager� r� r r r r- $ s
r- c C s* i }t | �\}}||d< |�| �|d< |S )Nr rI )r� rI )r/ �resr� r� r r r �info4 s
r� c C s t �| �}t||�S )z-
Same as load(), but takes a string.
r7 )r; r0 r/ r r r r< ? s
r< c C s� | d }t j| d }|dkrX|dv r@|�t�dt jd �� n|�t�dt jd �� |�t�d|�� z t| }W n td��Y n0 |�| d |� d S )
z�
Convert criu image from dict(json) format to binary.
Takes an image in dict(json) format and file-like
object to write to.
r r� )r� r� r r� r� z*No handler found for image with such magicr2 N)r r� rA r) rB r� ru rC )�imgr/ r� Z magic_valr� r r r rC G s rC c C s t �d�}t| |� |�� S )zK
Same as dump(), but takes only an image and returns
a string.
r )r8 r9 rC �getvalue)r� r/ r r r rG b s
rG )FF)F)\r8 rW r) rU rs r r r r �dirZ encodestringr Z decodestringrZ rq r� rr r ru r r rK rQ r[ rc rf rm rz r� Zinventory_entryZ
core_entryZtask_kobj_ids_entryZ creds_entryZ utsns_entryZ
ipc_var_entryZfs_entryZmm_entryZ cgroup_entryZtcp_stream_entryZ stats_entryZ pstree_entryZreg_file_entryZ
ns_file_entryZeventfd_file_entryZeventpoll_file_entryZeventpoll_tfd_entryZsignalfd_entryZ
timerfd_entryZinotify_file_entryZinotify_wd_entryZfanotify_file_entryZfanotify_mark_entryZ vma_entryZ
pipe_entryZ
fifo_entryZsa_entryZnetlink_sk_entryZremap_file_path_entryZ mnt_entryZtty_file_entryZtty_info_entryZtty_data_entryZ rlimit_entryZ
tunfile_entryZext_file_entryZirmap_cache_entryZfile_lock_entryZ fdinfo_entryZ
unix_sk_entryZ
inet_sk_entryZpacket_sock_entryZ itimer_entryZposix_timer_entryZnet_device_entryZpipe_data_entryZsk_packet_entryZ
ipc_shm_entryZ
ipc_sem_entryZ
ipc_msg_entryZ netns_entryZ userns_entryZ
seccomp_entryZ autofs_entryZ
file_entryZ
cpuinfo_entryr� r� r- r� r< rC rG r r r r �<module>( s� o0G 1�� �A