HOME

Mini Shell 1.0
DIR:/proc/self/root/lib/python3.9/site-packages/pycriu/images/__pycache__/
Current File : //proc/self/root/lib/python3.9/site-packages/pycriu/images/__pycache__/images.cpython-39.pyc
a

k�x]�H�<@s4ddlZddlZddlZddlZddlZddlmZddlmZddlmZde	e�vrhej
e_eje_
dZdZd	Zd
d�ZGdd
�d
e�ZGdd�d�ZGdd�d�ZGdd�d�ZGdd�d�ZGdd�d�ZGdd�d�ZGdd�d�ZGdd�d�ZGdd�d�Zeej�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���cCs|d|dBdS�Nr�)�x�yrr�8/usr/lib/python3.9/site-packages/pycriu/images/images.py�round_up>src@seZdZdd�ZdS)�MagicExceptioncCs
||_dS�Nr)�selfrrrr�__init__CszMagicException.__init__N)�__name__�
__module__�__qualname__rrrrrrBsrc@sFeZdZdZddd�Zddd�Zddd	�Zd
d�Zdd
�Zdd�Z	dS)�
entry_handlerzp
    Generic class to handle loading/dumping criu images
    entries from/to bin format to/from dict(json).
    NcCs||_||_dS)z=
        Sets payload class and extra handler class.
        N)�payload�
extra_handler)rrrrrrrOszentry_handler.__init__FcCs�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	��icSsNdD]@}|dkr<t|�|kr,d||fSd||fS|d}qd|S)N)��K�M�G�T�P�E�Zg�@z%d%sBz%.1f%sBz%.1fYB)�int)ZnumZunitrrr�human_readablens
z*entry_handler.load.<locals>.human_readablez... <%s>�extra)
r�read�struct�unpack�ParseFromStringrr�skip�load�append)r�f�pretty�
no_payload�entries�entry�pbuff�buf�sizer&Zpl_sizerrrr-Vs"

zentry_handler.loadcCst�|�}|�||�S)zD
        Same as load(), but takes a string as an argument.
        ��io�BytesIOr-�r�sr0r/rrr�loads�s
zentry_handler.loadscCst|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)�poprr�dict2pb�SerializeToString�len�writer)�packr�dump)rr2r/r3r'r4�pb_strr6rrrrC�s

zentry_handler.dumpcCst�d�}|�||�|��S)zf
        Same as dump(), but doesn't take file-like object and just
        returns a string.
        r�r8r9rCr(�rr2r/rrr�dumps�s
zentry_handler.dumpscCs@d}|�d�}|dkrq<t�d|�\}|�|d�|d7}q|S)zI
        Counts the number of top-level object in the image file
        rr	rrr)r(r)r*�seek)rr/r2r5r6rrr�count�s

zentry_handler.count)N)FF)F)
rrr�__doc__rr-r<rCrGrIrrrrrIs

+
	rc@s<eZdZdZddd�Zddd�Zdd�Zd	d
�Zdd�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.
    FcCs^g}t��}|�d�}|dkr qZt�d|�\}|�|�|��|�t�||��t��}q|S)Nr	rr)	r�pagemap_headr(r)r*r+r.r�
pagemap_entry)rr/r0r1r2r4r5r6rrrr-�s

zpagemap_handler.loadcCst�|�}|�||�Srr7r:rrrr<�s
zpagemap_handler.loadscCsVt��}|D]D}t�||�|��}t|�}|�t�d|��|�|�t�	�}qdS)Nr)
rrLrr>r?r@rAr)rBrM�rr2r/r4�itemrDr6rrrrC�s
zpagemap_handler.dumpcCst�d�}|�||�|��S�NrrErFrrrrG�s
zpagemap_handler.dumpscCstd��|�dSr)rrI)rr/rrrrI�szpagemap_handler.countN)FF)F)	rrrrJr-r<rCrGrIrrrrrK�s

rKc@s0eZdZddd�Zddd�Zdd�Zdd	�Zd
S)
�ghost_file_handlerFcCsg}t��}|�d�}t�d|�\}|�|�|��t�||�}|jr�|�|�t�	�}	|�d�}|dkrlq�t�d|�\}|	�|�|��t�|	|�}
|r�|�
|	jtj
�nt�|�|	j��|
d<|�|
�qPn0|r�|�
dtj�nt�|���|d<|�|�|S)Nr	rrr'r)r�ghost_file_entryr(r)r*r+r�chunksr.�ghost_chunk_entryrHr@�os�SEEK_CUR�base64r�SEEK_END)rr/r0r1r2Zgfr5r6Zg_entry�gcr3rrrr-�s0



zghost_file_handler.loadcCst�|�}|�||�Srr7r:rrrr<
s
zghost_file_handler.loadscCs�t��}|�d�}t�||�|��}t|�}|�t�	d|��|�|�|j
r�|D]X}t��}t�||�|��}t|�}|�t�	d|��|�|�|�t�
|d��qTn|�t�
|d��dS)Nrrr')rrRr=rr>r?r@rAr)rBrSrTrW�decodebytesrNrrrrCs"


zghost_file_handler.dumpcCst�d�}|�||�|��SrPrErFrrrrG#s
zghost_file_handler.dumpsN)FF)F)rrrr-r<rCrGrrrrrQ�s
!
rQc@s$eZdZdd�Zdd�Zdd�ZdS)�pipes_data_extra_handlercCs|j}|�|�}t�|�Sr)�bytesr(rWr�rr/�ploadr6�datarrrr-0s
zpipes_data_extra_handler.loadcCst�|�}|�|�dSr�rWrZrA)rr'r/r^r_rrrrC5s
zpipes_data_extra_handler.dumpcCs|�|jtj�|jSr)rHr\rUrV�rr/r^rrrr,9szpipes_data_extra_handler.skipN�rrrr-rCr,rrrrr[/sr[c@s$eZdZdd�Zdd�Zdd�ZdS)�sk_queues_extra_handlercCs|j}|�|�}t�|�Sr)�lengthr(rWrr]rrrr-?s
zsk_queues_extra_handler.loadcCst�|�}|�|�dSrr`)rr'r/�_unusedr_rrrrCDs
zsk_queues_extra_handler.dumpcCs|�|jtj�|jSr)rHrdrUrVrarrrr,Hszsk_queues_extra_handler.skipNrbrrrrrc>srcc@s$eZdZdd�Zdd�Zdd�ZdS)�tcp_stream_extra_handlercCs<i}|�|j�}|�|j�}t�|�|d<t�|�|d<|S�N�inq�outq)r(�inq_len�outq_lenrWr)rr/r4�drhrirrrr-Nsztcp_stream_extra_handler.loadcCs4t�|d�}t�|d�}|�|�|�|�dSrgr`)rr'r/rerhrirrrrCYs
ztcp_stream_extra_handler.dumpcCs|�dtj�|j|jS)Nr)rHrUrXrjrk)rr/r4rrrr,`sztcp_stream_extra_handler.skipNrbrrrrrfMsrfc@s$eZdZdd�Zdd�Zdd�ZdS)�ipc_sem_set_handlercCsdt�|�}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)rr/r4r3r6�roundedr;rrrr-fs



zipc_sem_set_handler.loadcCs�t�|�}t|d}t|t�}t�d�}|jtkr<td��|�|�t|�|dkr^td��|�	|�
��|�	d||�dS)NrnrorpzNumber of semaphores mismatch�)rrqrrrrsrtru�fromlistr@rAZtostring)rr'r/r4r3r6rwr;rrrrCqs




zipc_sem_set_handler.dumpcCs.t�|�}t|d}|�t|t�tj�|S)Nrn)rrqrHrrrrUrV)rr/r4r3r6rrrr,~s
zipc_sem_set_handler.skipNrbrrrrrmes
rmc@s$eZdZdd�Zdd�Zdd�ZdS)�ipc_msg_queue_handlercCs�t�|�}g}td|d�D]�}|�d�}|dkr6q�t�d|�\}t��}|�|�|��t|j	t
�}	|�|j	�}
|�|	|j	d�|�t�|��|�t
�|
��q|S)Nr�qnumr	rrr)r�ranger(r)r*r�ipc_msgr+r�msizerrrHr.rWr)rr/r4r3�messagesr
r5r6�msgrwr_rrrr-�s

zipc_msg_queue_handler.loadcCs�t�|�}tdt|�d�D]�}t��}t�|||�|��}t|�}|�t�	d|��|�|�t
|jt�}	t
�||d�}
|�|
d|j��|�d|	|j�qdS)Nrrrrrx)rr|r@rr}r>r?rAr)rBrr~rrrWrZ)rr'r/r4r3rr�Zmsg_strr6rwr_rrrrC�s

zipc_msg_queue_handler.dumpc
Cs�t�|�}d}td|d�D]h}|�d�}|dkr6q�t�d|�\}t��}|�|�|��t|j	t
�}	|�|	tj
�|||j	7}q|S)Nrr{r	rr)rr|r(r)r*rr}r+rr~rrrHrUrV)
rr/r4r3Zpl_lenr
r5r6r�rwrrrr,�s

zipc_msg_queue_handler.skipNrbrrrrrz�srzc@s$eZdZdd�Zdd�Zdd�ZdS)�ipc_shm_handlercCs@t�|�}|d}|�|�}t|t�}|�||d�t�|�S)Nr6r)rr(r�
sizeof_u32rHrWr)rr/r4r3r6r_rwrrrr-�s


zipc_shm_handler.loadcCsNt�|�}|d}t�|�}t|t�}|�|d|��|�d||�dS)Nr6rx)rrWrZrr�rA)rr'r/r4r3r6r_rwrrrrC�s


zipc_shm_handler.dumpcCs.t�|�}|d}t|t�}|�|tj�|S)Nr6)rrr�rHrUrV)rr/r4r3r6rwrrrr,�s


zipc_shm_handler.skipNrbrrrrr��sr�);�	INVENTORYZCOREZIDSZCREDSZUTSNSZIPC_VARZFSZ
GHOST_FILEZMMZCGROUPZ
TCP_STREAM�STATSZPAGEMAPZPSTREEZ	REG_FILESZNS_FILESZEVENTFD_FILEZEVENTPOLL_FILEZ
EVENTPOLL_TFDZSIGNALFDZTIMERFDZINOTIFY_FILEZ
INOTIFY_WDZ
FANOTIFY_FILEZ
FANOTIFY_MARKZVMASZPIPESZFIFOZSIGACTZ
NETLINK_SKZREMAP_FPATHZMNTSZ	TTY_FILESZTTY_INFOZTTY_DATAZRLIMITZTUNFILEZ	EXT_FILES�IRMAP_CACHEZ
FILE_LOCKSZFDINFOZUNIXSKZINETSKZPACKETSKZITIMERSZPOSIX_TIMERSZNETDEVZ
PIPES_DATAZ	FIFO_DATAZ	SK_QUEUESZ	IPCNS_SHMZ	IPCNS_SEMZ	IPCNS_MSGZNETNSZUSERNSZSECCOMPZAUTOFSZFILESZCPUINFOcCs�t�d|�d��\}|tjdtjdfvr@t�d|�d��\}ztj|}Wnt|��Yn0zt|}Wntd|��Yn0||fS)Nrr	�
IMG_COMMON�IMG_SERVICEz&No handler found for image with magic )	r)r*r(r�by_nameZby_valr�handlersru)r/Z	img_magic�m�handlerrrr�
__rhandlers
�r�FcCs.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.
    rr2)r�r-)r/r0r1Zimager�r�rrrr-$s
r-cCs*i}t|�\}}||d<|�|�|d<|S)NrrI)r�rI)r/�resr�r�rrr�info4s
r�cCst�|�}t||�S)z-
    Same as load(), but takes a string.
    r7)r;r0r/rrrr<?s
r<cCs�|d}tj|d}|dkrX|dvr@|�t�dtjd��n|�t�dtjd��|�t�d|��zt|}Wntd��Yn0|�|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.
    rr�)r�r�rr�r�z*No handler found for image with such magicr2N)rr�rAr)rBr�rurC)�imgr/r�Z	magic_valr�rrrrCGsrCcCst�d�}t||�|��S)zK
    Same as dump(), but takes only an image and returns
    a string.
    r)r8r9rC�getvalue)r�r/rrrrGbs

rG)FF)F)\r8rWr)rUrsrrrr�dirZencodestringrZdecodestringrZrqr�rrrrurrrKrQr[rcrfrmrzr�Zinventory_entryZ
core_entryZtask_kobj_ids_entryZcreds_entryZutsns_entryZ
ipc_var_entryZfs_entryZmm_entryZcgroup_entryZtcp_stream_entryZstats_entryZpstree_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_entryZrlimit_entryZ
tunfile_entryZext_file_entryZirmap_cache_entryZfile_lock_entryZfdinfo_entryZ
unix_sk_entryZ
inet_sk_entryZpacket_sock_entryZitimer_entryZposix_timer_entryZnet_device_entryZpipe_data_entryZsk_packet_entryZ
ipc_shm_entryZ
ipc_sem_entryZ
ipc_msg_entryZnetns_entryZuserns_entryZ
seccomp_entryZautofs_entryZ
file_entryZ
cpuinfo_entryr�r�r-r�r<rCrGrrrr�<module>(s�o0G 1���A