????
Your IP : 216.73.217.84
a
?9j5M � C @ s� d dl Z d dlZd dlZd dlZd dlZddlmZ ddlmZ ddlmZ de e�vrhej
e_ ej e_
dZdZd Zd
d � ZG d d
� d
e�Zdd� Zdd� ZG dd� d�ZG dd� d�ZG dd� d�ZG dd� d�ZG dd� d�ZG d d � d �ZG d d � d �Z G d d!� d!�Z G d"d#� d#�Z G d$d%� d%�Z eej �eej!�eej"�eej#�eej$�eej%�eej&�eej'�eej(�e� eej)�eej*�eej+e� �eej,�e� 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�eejJ�eejK�eejL�eejM�eejN�eejOe� �eejOe� �eejPe� �eejQe � �eejRe � �eejSe � �eejT�eejU�eejV�eejW�eejX�eejY�eejZ�eej[�eej\�eej]e � �eej^�d&�BZ_d'd(� Z`d4d*d+�Zad,d-� Zbd5d.d/�Zcd0d1� Zdd2d3� ZedS )6� 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 C s t �t�| ��S )z(A helper function to decode base64 data.)�base64� decodebytes�str�encode)�datar r r �decode_base64_dataG s r c C s | � t�t�|��� dS )z9A helper function to write base64 encoded data to a file.N)�writer r r r )�fr r r r �write_base64_dataL 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 Y s zentry_handler.__init__Fc C s� g }i }| � � }|�d�}t|�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 returns a list with entries in
dict(json) format.
r 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_readablex s
z*entry_handler.load.<locals>.human_readablez... <%s>�extra) r" �read�len�struct�unpack�ParseFromStringr r# �skip�load�append) r r �pretty�
no_payload�entries�entry�pbuff�buf�sizer. Zpl_sizer r r r6 ` s"
zentry_handler.loadc C s t �|�}| �||�S )zD
Same as load(), but takes a string as an argument.
��io�BytesIOr6 �r �sr8 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�SerializeToStringr1 r r2 �packr# �dump)r r: r r; r/ r< �pb_strr> r r r rI � 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% �r@ rA rI r0 �r r: r r r r �dumps� s
zentry_handler.dumpsc C sD d}|� d�}t|�dkr q@t�d|�\}|�|d� |d7 }q|S )zI
Counts the number of top-level object in the image file
r r r$ r )r0 r1 r2 r3 �seek)r r r: r= r> r r r �count� s
zentry_handler.count)N)FF)F)
r r r �__doc__r r6 rD rI rM rO r r r r r! S 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 sb g }t �� }|�d�}t|�dkr$q^t�d|�\}|�|�|�� |�t�||�� t � � }q |S )Nr r r$ )
r � pagemap_headr0 r1 r2 r3 r4 r7 r �
pagemap_entry)r r r8 r9 r: r< r= r> r r r r6 � s
zpagemap_handler.loadc C s t �|�}| �||�S r r? rB r r r rD � s
zpagemap_handler.loadsc C sV t �� }|D ]D}t�||� |�� }t|�}|�t�d|�� |�|� t � � }q d S )Nr$ )
r rR r rF rG r1 r r2 rH rS �r r: r r< �itemrJ r> r r r rI � s
zpagemap_handler.dumpc C s t �d�}| �||� |�� S �Nr% rK rL r r r rM � s
zpagemap_handler.dumpsc C s t d ��|�d S r )r! rO )r r r r r rO � s zpagemap_handler.countN)FF)F) r r r rP r6 rD rI rM rO r r r r rQ � s
rQ 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�}t
|�dkrpq�t�d|�\}| �|�|�� t�| |�}
|r�|� | j
t j
� n t�|�| j
���d�|
d< |�|
� qPn6|r�|� dt j� nt�|�� ��d�|d< |�|� |S )Nr r$ r �utf-8r/ )r �ghost_file_entryr0 r2 r3 r4 r �chunksr7 �ghost_chunk_entryr1 rN �os�SEEK_CURr r �decode�SEEK_END) r r r8 r9 r: Zgfr= r> Zg_entry�gcr; r r r r6 � s0
zghost_file_handler.loadc C s t �|�}| �||�S r r? rB r r r rD s
zghost_file_handler.loadsc C s� t �� }|�d�}t�||� |�� }t|�}|�t� d|�� |�|� |j
r�|D ]R}t � � }t�||� |�� }t|�}|�t� d|�� |�|� t ||d � qTnt ||d � d S )Nr r$ r/ )
r rY rE r rF rG r1 r r2 rH rZ r[ r rT r r r rI s"
zghost_file_handler.dumpc C s t �d�}| �||� |�� S rV rK rL r r r rM - s
zghost_file_handler.dumpsN)FF)F)r r r r6 rD rI rM r r r r rW � s
!
rW c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�pipes_data_extra_handlerc C s |j }|�|�}t�|��d�S �NrX )�bytesr0 r r r^ �r r �ploadr> r r r r r6 : s
z pipes_data_extra_handler.loadc C s t |�}|�|� d S r �r r �r r/ r re r r r r rI ? s z pipes_data_extra_handler.dumpc C s |� |jtj� |jS r �rN rc r\ r] �r r re r r r r5 C s z pipes_data_extra_handler.skipN�r r r r6 rI r5 r r r r ra 9 s ra c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�sk_queues_extra_handlerc C s |j }|�|�}t�|��d�S rb )�lengthr0 r r r^ rd r r r r6 I s
z sk_queues_extra_handler.loadc C s t |�}|�|� d S r rf )r r/ r �_unusedr r r r rI N s z sk_queues_extra_handler.dumpc C s |� |jtj� |jS r )rN rl r\ r] ri r r r r5 R s z sk_queues_extra_handler.skipNrj r r r r rk H s rk c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�tcp_stream_extra_handlerc C sH i }|� |j�}|� |j�}t�|��d�|d< t�|��d�|d< |S )NrX �inq�outq)r0 �inq_len�outq_lenr r r^ )r r r< �dro rp r r r r6 X s z tcp_stream_extra_handler.loadc C s0 t |d �}t |d �}|�|� |�|� d S )Nro rp rf )r r/ r rm ro rp r r r rI c s
z tcp_stream_extra_handler.dumpc C s |� dtj� |j|j S )Nr )rN r\ r_ rq rr )r r r< r r r r5 j s z tcp_stream_extra_handler.skipNrj r r r r rn W s rn c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�bpfmap_data_extra_handlerc C s& |j |j }|�|�}t�|��d�S rb )Z
keys_bytesZ values_bytesr0 r r r^ rd r r r r6 p s
z bpfmap_data_extra_handler.loadc C s t �|�}|�|� d S r )r r r rg r r r rI u s
z bpfmap_data_extra_handler.dumpc C s |� |jtj� |jS r rh ri r r r r5 y s z bpfmap_data_extra_handler.skipNrj r r r r rt o s rt c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�ipc_sem_set_handlerc C sP t � |�}t|d }t|t�}| �� }|�|�|�� |�|| d� |�� S )N�nsemsr ) r �
sizeof_u16r �
sizeof_u64�_get_sem_arrayZ frombytesr0 rN �tolist)r r r< r; r> �roundedrC r r r r6 s
zipc_sem_set_handler.loadc C sn t � |�}t|d }t|t�}| �� }|�|� t|�|d krJtd��|�|� � � |�d|| � d S )Nrv z Number of semaphores mismatch� )
r rw r rx ry �fromlistr1 � Exceptionr �tobytes)r r/ r r< r; r> r{ rC r r r rI � s
zipc_sem_set_handler.dumpc C s. t � |�}t|d }|�t|t�tj� |S )Nrv )r rw rN r rx r\ r] )r r r< r; r> r r r r5 � s
zipc_sem_set_handler.skipc C s t � d�}|jtkr td��|S )N�HzArray size mismatch)�array�itemsizerw r~ )r rC r r r ry � s
z"ipc_sem_set_handler._get_sem_arrayN)r r r r6 rI r5 ry r r r r ru ~ s ru c @ s. e Zd Zdd� Zdd� Zdd� Zd dd �Zd
S ) �ipc_msg_queue_handlerc C s | � ||�\}}|S r ��_read_messages)r r r< �messages�_r r r r6 � s zipc_msg_queue_handler.loadc
C s� t dt|�d�D ]�}t�� }t�|| |� |�� }t|�}|�t� d|�� |�|� t
|j t �}t
||d �} |�| d |j � � |�d||j � qd S )Nr r r$ r r| )�ranger1 r �ipc_msgr rF rG r r2 rH r �msizerx r )
r r/ r r< r$ �msgZmsg_strr> r{ r r r r rI � s
zipc_msg_queue_handler.dumpc C s | j ||dd�\}}|S )NT)� skip_datar� )r r r< r� �pl_lenr r r r5 � s zipc_msg_queue_handler.skipFc
C s� t � |�}g }d}td|d �D ]�}|�d�}t|�dkr> q�t�d|�\} t�� }
|
�|�| �� t |
j
t �} || |
j
7 }|r�|� | t
j� q |�|
j
�} |� | |
j
d� |�t � |
�� |�t�| ��d�� q ||fS )Nr Zqnumr r$ r rX )r r� r0 r1 r2 r3 r r� r4 r r� rx rN r\ r] r7 r r r^ )
r r r< r� r; r� r� r
r= r> r� r{ r r r r r� � s&
z$ipc_msg_queue_handler._read_messagesN)F)r r r r6 rI r5 r� r r r r r� � s
r� c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�ipc_shm_handlerc C sF t � |�}|d }|�|�}t|t�}|�|| d� t�|��d�S )Nr> r rX )r r0 r �
sizeof_u32rN r r r^ )r r r< r; r> r r{ r r r r6 � s
zipc_shm_handler.loadc C sN t � |�}|d }t�|�}t|t�}|�|d |� � |�d|| � d S )Nr> r| )r r r r r� r )r r/ r r< r; r> r r{ r r r rI � s
zipc_shm_handler.dumpc C s. t � |�}|d }t|t�}|�|tj� |S )Nr> )r r r� rN r\ r] )r r r< r; r> r{ r r r r5 � s
zipc_shm_handler.skipNrj r r r r r� � s r� )B� INVENTORYZCOREZIDSZCREDSZUTSNSZTIMENSZPIDNSZIPC_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_MSGZNETNSZUSERNSZSECCOMPZAUTOFSZFILESZCPUINFOZ
MEMFD_FILEZ MEMFD_INODEZ BPFMAP_FILEZ BPFMAP_DATAZAPPARMORc C s� t �d| �d��\}|tjd tjd fv r@t �d| �d��\}ztj| }W n tyh t|��Y n0 z t| }W n ty� td| ��Y n0 ||fS )Nr$ r �
IMG_COMMON� IMG_SERVICEz&No handler found for image with magic ) r2 r3 r0 r �by_nameZby_valr~ r �handlers)r Z img_magic�m�handlerr r r �
__rhandler1 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 r: )r� r6 )r r8 r9 Zimager� r� r r r r6 E s
r6 c C s* i }t | �\}}||d< |�| �|d< |S )Nr rO )r� rO )r �resr� r� r r r �infoU s
r� c C s t �| �}t||�S )z-
Same as load(), but takes a string.
r? )rC r8 r r r r rD ` s
rD c C s� | d }t j| d }|dkrX|dv r@|�t�dt jd �� n|�t�dt jd �� |�t�d|�� z t| }W n ty� 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 magicr: N)r r� r r2 rH r� r~ rI )�imgr r� Z magic_valr� r r r rI h s rI c C s t �d�}t| |� |�� S )zK
Same as dump(), but takes only an image and returns
a string.
� )r@ rA rI �getvalue)r� r r r r rM � s
rM )FF)F)fr@ r r2 r\ r� r% r r r �dirZ encodestringr Z decodestringr rw r� rx r r~ r r r r! rQ rW ra rk rn rt ru r� r� Zinventory_entryZ
core_entryZtask_kobj_ids_entryZ creds_entryZ utsns_entryZ timens_entryZ pidns_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_entryZmemfd_file_entryZmemfd_inode_entryZbpfmap_file_entryZbpfmap_data_entryZapparmor_entryr� r� r6 r� rD rI rM r r r r �<module>( s� o0G"/�� ��I