3
i5M C @ s d dl Z d dlZd dlZd dlZd dlZddlmZ ddlmZ ddlmZ de ekrhej
e_ ej e_
dZdZd Zd
d ZG d d
d
eZdd Zdd ZG dd dZG dd dZG dd dZG dd dZG dd dZG 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/eej0eej1eej2eej3eej4eej5eej6eej7eej8eej9eej:eej;eej<eej=eej>eej?eej@eejAeejBeejCeejDeejEeejFeejGeejHeejIeejJeejKeejLeejMeejNeejOe eejOe eejPe eejQe eejRe eejSe eejTeejUeejVeejWeejXeejYeejZeej[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 )xyr
r
/usr/lib/python3.6/images.pyround_up> s r c @ s e Zd Zdd ZdS )MagicExceptionc C s
|| _ d S )N)r )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 jtj| S )z(A helper function to decode base64 data.)base64 decodebytesstrencode)datar
r
r
decode_base64_dataG s r c C s | j tjtj| 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 ZdZdddZdddZddd 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 }xi }| j }|jd}t|dkr*P tjd|\}|j|j| tj||}| jr|rdd } | jj||}
d| |
|d< n| jj |||d< |j
| qW |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 sJ x@d
D ]8}| d k r6t | | kr*d
| |f S d | |f S | d } qW d | S )N KMGTPEZg @z%d%sBz%.1f%sBz%.1fYB)r" r# r$ r% r& r' r( r) )int)ZnumZunitr
r
r
human_readablex s
z*entry_handler.load..human_readablez... <%s>extra) r readlenstructunpackParseFromStringr r skiploadappend) r r pretty
no_payloadentriesentrypbuffbufsizer+ Zpl_sizer
r
r
r3 ` s$
zentry_handler.loadc C s t j|}| j||S )zD
Same as load(), but takes a string as an argument.
)ioBytesIOr3 )r sr5 r r
r
r
loads s
zentry_handler.loadsc C sx xr|D ]j}|j dd}| j }tj|| |j }t|}|jtjd| |j| | j r|r| j j
||| qW dS )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 dict2pbSerializeToStringr. r r/ packr dump)r r7 r r8 r, r9 pb_strr; r
r
r
rD s
zentry_handler.dumpc C s t jd}| j|| |j S )zf
Same as dump(), but doesn't take file-like object and just
returns a string.
r" )r< r= rD r- )r r7 r r
r
r
dumps s
zentry_handler.dumpsc C sH d}x>|j d}t|dkr P tjd|\}|j|d |d7 }qW |S )zI
Counts the number of top-level object in the image file
r r r! r )r- r. r/ r0 seek)r r r7 r: r; r
r
r
count s
zentry_handler.count)N)FF)F)
r r r __doc__r r3 r? rD rF rH r
r
r
r
r S s
+
r c @ s< e Zd ZdZdddZdddZdd 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 sf g }t j }xT|jd}t|dkr&P tjd|\}|j|j| |jtj|| t j }qW |S )Nr r r! )
r pagemap_headr- r. r/ r0 r1 r4 r
pagemap_entry)r r r5 r6 r7 r9 r: r; r
r
r
r3 s
zpagemap_handler.loadc C s t j|}| j||S )N)r< r= r3 )r r> r5 r r
r
r
r? s
zpagemap_handler.loadsc C sZ t j }xL|D ]D}tj|| |j }t|}|jtjd| |j| t j }qW d S )Nr! )
r rK r rA rB r. r r/ rC rL )r r7 r r9 itemrE r; r
r
r
rD s
zpagemap_handler.dumpc C s t jd}| j|| |j S )Nr" )r< r= rD r- )r r7 r r
r
r
rF s
zpagemap_handler.dumpsc C s t d j|d S )Nr )r rH )r r r
r
r
rH s zpagemap_handler.countN)FF)F) r r r rI r3 r? rD rF rH r
r
r
r
rJ s
rJ c @ s0 e Zd Zd ddZd ddZdd Zdd Zd
S )
ghost_file_handlerFc C s g }t j }|jd}tjd|\}|j|j| tj||}|jr|j| xt j } |jd}t
|dkrrP tjd|\}| j|j| tj| |}
|r|j | j
t j
n tj|j| j
jd|
d< |j|
qRW n6|r|j dt j ntj|j jd|d< |j| |S )Nr r! r zutf-8r, )r ghost_file_entryr- r/ r0 r1 r chunksr4 ghost_chunk_entryr. rG osSEEK_CURr r decodeSEEK_END) r r r5 r6 r7 Zgfr: r; Zg_entrygcr8 r
r
r
r3 s2
zghost_file_handler.loadc C s t j|}| j||S )N)r< r= r3 )r r> r5 r r
r
r
r? s
zghost_file_handler.loadsc C s t j }|jd}tj|| |j }t|}|jtj d| |j| |j
rxj|D ]R}t j }tj|| |j }t|}|jtj d| |j| t ||d qVW nt ||d d S )Nr r! r, )
r rO r@ r rA rB r. r r/ rC rP rQ r )r r7 r r9 rM rE r; r
r
r
rD s"
zghost_file_handler.dumpc C s t jd}| j|| |j S )Nr" )r< r= rD r- )r r7 r r
r
r
rF - s
zghost_file_handler.dumpsN)FF)F)r r r r3 r? rD rF r
r
r
r
rN s
!
rN c @ s$ e Zd Zdd Zdd Zdd ZdS )pipes_data_extra_handlerc C s |j }|j|}tj|jdS )Nzutf-8)bytesr- r r rT )r r ploadr; r r
r
r
r3 : s
z pipes_data_extra_handler.loadc C s t |}|j| d S )N)r r )r r, r rY r r
r
r
rD ? s z pipes_data_extra_handler.dumpc C s |j |jtj |jS )N)rG rX rR rS )r r rY r
r
r
r2 C s z pipes_data_extra_handler.skipN)r r r r3 rD r2 r
r
r
r
rW 9 s rW c @ s$ e Zd Zdd Zdd Zdd ZdS )sk_queues_extra_handlerc C s |j }|j|}tj|jdS )Nzutf-8)lengthr- r r rT )r r rY r; r r
r
r
r3 I s
z sk_queues_extra_handler.loadc C s t |}|j| d S )N)r r )r r, r _unusedr r
r
r
rD N s z sk_queues_extra_handler.dumpc C s |j |jtj |jS )N)rG r[ rR rS )r r rY r
r
r
r2 R s z sk_queues_extra_handler.skipN)r r r r3 rD r2 r
r
r
r
rZ H s rZ c @ s$ e Zd Zdd Zdd Zdd ZdS )tcp_stream_extra_handlerc C sH i }|j |j}|j |j}tj|jd|d< tj|jd|d< |S )Nzutf-8inqoutq)r- inq_lenoutq_lenr r rT )r r r9 dr^ r_ r
r
r
r3 X s z tcp_stream_extra_handler.loadc C s0 t |d }t |d }|j| |j| d S )Nr^ r_ )r r )r r, r r\ r^ r_ r
r
r
rD c s
z tcp_stream_extra_handler.dumpc C s |j dtj |j|j S )Nr )rG rR rU r` ra )r r r9 r
r
r
r2 j s z tcp_stream_extra_handler.skipN)r r r r3 rD r2 r
r
r
r
r] W s r] c @ s$ e Zd Zdd Zdd Zdd ZdS )bpfmap_data_extra_handlerc C s& |j |j }|j|}tj|jdS )Nzutf-8)Z
keys_bytesZ values_bytesr- r r rT )r r rY r; r r
r
r
r3 p s
z bpfmap_data_extra_handler.loadc C s t j|}|j| d S )N)r r r )r r, r rY r r
r
r
rD u s
z bpfmap_data_extra_handler.dumpc C s |j |jtj |jS )N)rG rX rR rS )r r rY r
r
r
r2 y s z bpfmap_data_extra_handler.skipN)r r r r3 rD r2 r
r
r
r
rc o s rc c @ s, e Zd Zdd Zdd Zdd Zdd Zd S )
ipc_sem_set_handlerc C sP t j |}t|d }t|t}| j }|j|j| |j|| d |j S )Nnsemsr ) r
sizeof_u16r
sizeof_u64_get_sem_arrayZ frombytesr- rG tolist)r r r9 r8 r; roundedr> r
r
r
r3 s
zipc_sem_set_handler.loadc C sn t j |}t|d }t|t}| j }|j| t||d krJtd|j|j |jd|| d S )Nre z Number of semaphores mismatch )
r rf r rg rh fromlistr. Exceptionr tobytes)r r, r r9 r8 r; rj r> r
r
r
rD s
zipc_sem_set_handler.dumpc C s. t j |}t|d }|jt|ttj |S )Nre )r rf rG r rg rR rS )r r r9 r8 r; r
r
r
r2 s
zipc_sem_set_handler.skipc C s t j d}|jtkr td|S )NHzArray size mismatch)arrayitemsizerf rm )r r> r
r
r
rh s
z"ipc_sem_set_handler._get_sem_arrayN)r r r r3 rD r2 rh r
r
r
r
rd ~ s rd c @ s. e Zd Zdd Zdd Zdd Zd dd Zd
S ) ipc_msg_queue_handlerc C s | j ||\}}|S )N)_read_messages)r r r9 messages_r
r
r
r3 s zipc_msg_queue_handler.loadc
C s xt dt|dD ]}tj }tj|| | |j }t|}|jtj d| |j| t
|j t }t
||d } |j| d |j |jd||j qW d S )Nr r r! r rk )ranger. r ipc_msgr rA rB r r/ rC r msizerg r )
r r, r r9 r! msgZmsg_strr; rj r r
r
r
rD s
zipc_msg_queue_handler.dumpc C s | j ||dd\}}|S )NT) skip_data)rs )r r r9 ru pl_lenr
r
r
r2 s zipc_msg_queue_handler.skipFc
C s t j |}g }d}xtd|d D ]}|jd}t|dkr>P tjd|\} tj }
|
j|j| t |
j
t } || |
j
7 }|r|j | t
j q"|j|
j
} |j | |
j
d |jt j |
|jtj| jd q"W ||fS )Nr Zqnumr r! r zutf-8)r rv r- r. r/ r0 r rw r1 r rx rg rG rR rS r4 r r rT )
r r r9 rz r8 rt r{ r r: r; ry rj r r
r
r
rs s&
z$ipc_msg_queue_handler._read_messagesN)F)r r r r3 rD r2 rs r
r
r
r
rr s
rr c @ s$ e Zd Zdd Zdd Zdd ZdS )ipc_shm_handlerc C sF t j |}|d }|j|}t|t}|j|| d tj|jdS )Nr; r zutf-8)r r- r
sizeof_u32rG r r rT )r r r9 r8 r; r rj r
r
r
r3 s
zipc_shm_handler.loadc C sN t j |}|d }tj|}t|t}|j|d | |jd|| d S )Nr; rk )r r r r r} r )r r, r r9 r8 r; r rj r
r
r
rD s
zipc_shm_handler.dumpc C s. t j |}|d }t|t}|j|tj |S )Nr; )r r r} rG rR rS )r r r9 r8 r; rj r
r
r
r2 s
zipc_shm_handler.skipN)r r r r3 rD r2 r
r
r
r
r| s r| )B INVENTORYZCOREZIDSZCREDSZUTSNSZTIMENSZPIDNSZIPC_VARZFSZ
GHOST_FILEZMMZCGROUPZ
TCP_STREAMSTATSZPAGEMAPZPSTREEZ 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 jd| jd\}|tjd tjd fkr@t jd| jd\}ytj| }W n tk
rj t|Y nX y t| }W n tk
r td| Y nX ||fS )Nr! r
IMG_COMMON IMG_SERVICEz&No handler found for image with magic ) r/ r0 r- r by_nameZby_valrm r handlers)r Z img_magicmhandlerr
r
r
__rhandler1 s
r Fc C s. i }t | \}}||d< |j| |||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 r7 )r r3 )r r5 r6 Zimager r r
r
r
r3 E s
r3 c C s* i }t | \}}||d< |j| |d< |S )Nr rH )r rH )r resr r r
r
r
infoU s
r c C s t j| }t||S )z-
Same as load(), but takes a string.
)r< r= r3 )r> r5 r r
r
r
r? ` s
r? c C s | d }t j| d }|dkrX|d kr@|jtjdt jd n|jtjdt jd |jtjd| y t| }W n tk
r tdY nX |j| 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 magicr7 N)r r )r r r r/ rC r rm rD )imgr r Z magic_valr r
r
r
rD h s rD c C s t jd}t| | |j S )zK
Same as dump(), but takes only an image and returns
a string.
)r< r= rD getvalue)r r r
r
r
rF s
rF )FF)F)fr< r r/ rR rp r" r r r dirZ encodestringr Z decodestringr rf r} rg r rm r r r r rJ rN rW rZ r] rc rd rr 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 r3 r r? rD rF r
r
r
r
( s o0G"/