Sindbad~EG File Manager
3
i5M � C @ s� d dl Z d dlZd dlZd dlZd dlZddlmZ ddlmZ ddlmZ de e�krhej
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
� /usr/lib/python3.6/images.py�round_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� decodebytes�str�encode)�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 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 }x�i }| j � }|jd�}t|�dkr*P tjd|�\}|j|j|�� tj||�}| jr�|r�dd� } | 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� �K�M�G�T�P�E�Zg �@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.<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
r3 ` s$
zentry_handler.loadc C s t j|�}| j||�S )zD
Same as load(), but takes a string as an argument.
)�io�BytesIOr3 )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 �dict2pb�SerializeToStringr. 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 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 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 dd�Zd dd�Zdd� Zdd � Zd
S )
�ghost_file_handlerFc C s g }t j� }|jd�}tjd|�\}|j|j|�� tj||�}|jr�|j|� x�t 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 �os�SEEK_CURr r �decode�SEEK_END) r r r5 r6 r7 Zgfr: r; Zg_entry�gcr8 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
r�xj|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|�jd�S )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|�jd�S )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-8�inq�outq)r- �inq_len�outq_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|�jd�S )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 )N�nsemsr ) 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|t�tj� |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 )N�HzArray size mismatch)�array�itemsizerf 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� x�t dt|�d�D ]�}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}x�td|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|�jd�S )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_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 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_magic�m�handlerr
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� td��Y 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
�<module>( s� o0G"/
Sindbad File Manager Version 1.0, Coded By Sindbad EG ~ The Terrorists