3
iG. � @ sH d dl Z d dlZd dlZd dlZd dlZddlmZ dd� Zdd� Zdd � Z d
d � Z
d d
� Z dd� Z dd� Z
G dd� d�Zd9dd�Zdd� Zdadd� Zdd� Zd d � Zd d � Zd d!� Zedd"d#�edd$d#�edd%d#�d&�Zd'd(� Zi Zd)d*� Zd+d,� ZG d-d.� d.�Z d/d0� Z d1d2� Z eee e d3�Z d4d5� Z d6d7� Z!e"d8k�rDe!� dS ):� N� )� __version__c C s2 | d rt | d d�S tjj� r&tjS tjjS d S )N�in�rb)�open�sys�stdin�isatty�buffer)�opts� r � /usr/lib/python3.6/__main__.py�inf s
r c C s8 | d r"d}|rd}t | d |�S |r,tjS tjjS d S )N�outzwb+zw+)r r �stdoutr
)r �decode�moder r r
�outf s r c C s t tjj| d |�dd�S )N�dirr )r )r �os�path�join)r �namer r r
�dinf# s r c C s� d }y t jjt| �| d | d �}W nD t jjk
rh } z$td|j tjd� tj d� W Y d d }~X nX | d rvd}t
| d�}t j |||d� |tj
kr�|jd � d S )
N�pretty�noplzSUnknown magic %#x.
Maybe you are feeding me an image with raw data(i.e. pages.img)?)�filer � T)�indent�
)�pycriu�images�loadr ZMagicException�print�magicr �stderr�exitr �json�dumpr �write)r r �img�exc�fr r r
r ' s
r c
C sX yt jt| ��}W n, tk
r> tdtjd� tjd� Y nX tj j
|t | d�� d S )NzbCannot read JSON.
Maybe you are feeding me an image with protobuf data? Encode expects JSON input.)r r F) r' r" r �UnicodeDecodeErrorr# r r% r& r r! r( r )r r* r r r
�encode; s
r. c C s, t jjt| ��}tj|tjdd� t� d S )Nr )r ) r r! �infor r' r( r r r# )r Zinfsr r r
r/ F s r/ c C s || kr| | S | d| d S )NZns_r r )�p�valr r r
� get_task_idL s r2 c @ s e Zd Zdd� ZdS )�ps_itemc C s, t |d�| _|d | _|| _|| _g | _d S )N�pid�ppid)r2 r4 r5 r0 �core�kids)�selfr0 r6 r r r
�__init__V s
zps_item.__init__N)�__name__�
__module__� __qualname__r9 r r r r
r3 U s r3 c C s^ t d| jt| jd�t| jd�dd| | jd d f � x | jD ]}t|||d � qBW d S ) Nz%7d%7d%7d %s%sZpgidZsid� r ZtcZcommr )r# r4 r2 r0 r6 r7 �show_ps)r0 r �depthZkidr r r
r> ^ s
r> c C s� i }t jjt| d��}xH|d D ]<}t jjt| dt|d� ��}t||d d �}|||j< q W d }x:|D ]2}|| }|jdkr�|}qj||j }|jj |� qjW t
dd � t || � d S ) Nz
pstree.img�entriesz core-%d.imgr4 r z%7s%7s%7s %s�PID�PGID�SID�COMM)rA rB rC rD ) r r! r" r r2 r3 r4 r5 r7 �appendr# r> ) r Zpss�ps_imgr0 r6 �psZpsrr4 Zppr r r
�
explore_psf s"
rH c C sn t d kr<ytjjt| d��d a W n tk
r: g a Y nX tt �dkrLd S x t D ]}|d |krR|S qRW d S )Nz files.imgr@ r �id)� files_imgr r! r" r � Exception�len)r �ft�fidr, r r r
�ftype_find_in_files� s
rO c C sx t | ||�}|r,|d |kr(||d S d S |d d krRtjjt| |��d |d< x |d D ]}|d |kr\|S q\W d S )N�fieldr* r@ rI )rO r r! r" r )r rM rN r* r, r r r
�ftype_find_in_image� s rQ c C s t | ||d�}|r|d p dS )Nz
reg-files.imgr z unknown path)rQ )r rM rN Zrfr r r
� ftype_reg� s rR c C s" t | ||d�}|r d|d p dS )Nz pipes.imgzpipe[%d]Zpipe_idzpipe[?])rQ )r rM rN r0 r r r
�
ftype_pipe� s rS c C sD t | ||d�}|sdS |d r*d|d p,d}d|d |d |f S ) Nz
unixsk.imgzunix[?]r z %s� zunix[%d (%d)%s]ZinoZpeer)rQ )r rM rN Zux�nr r r
�
ftype_unix� s
rV Zreg)�getr* rP �pipeZusk)�REG�PIPEZUNIXSKc C s d|d |f S )Nz%s.%d�typr )r rM rN r r r
� ftype_gen� s r\ c C sZ |d |d f}t j|d �}|sVtj|d t|d d��}|d | ||d �}|t |< |S )N�typerI )rW r[ rW )� files_cacherW �
file_typesr\ )r �fd�keyr, rM r r r
� get_file_str� s rb c C s t jjt| d��}x�|d D ]�}t|d�}t jjt| d| ��}|d d d }t jjt| d| ��}td| � x*|d D ] }td |d
t| |�f � q|W t jjt| d | ��d d }td d
t| d|d d��f � td dt| d|d d��f � q W d S )Nz
pstree.imgr@ r4 z
ids-%s.imgr Zfiles_idz
fdinfo-%d.imgz%dz %7d: %sr` z fs-%d.imgz %7s: %s�cwdrY Zcwd_id)r] rI �rootZroot_id)r r! r" r r2 r# rb )r rF r0 r4 ZidiZfdtZfdir` r r r
� explore_fds� s$
re c @ s e Zd Zdd� Zdd� ZdS )�vma_idc C s i | _ d| _d S )Nr )� _vma_id__ids�
_vma_id__last)r8 r r r
r9 � s zvma_id.__init__c C s4 | j j|d �}|s0| j}| jd7 _|| j |< |S )Nr )rg rW rh )r8 Ziid�retr r r
rW � s
z
vma_id.getN)r: r; r< r9 rW r r r r
rf � s rf c C s& t jjt| d��}t� }�x|d D �]�}t|d�}t jjt| d| ��d d }td| � tddt| d |d
d ��f � �x�|d D �]�}|d
}|d0@ r�dd|j|d � }n�|d1@ r�dd|j|d � }n�|d2@ � r�dd|j|d � }n�|d5@ �rJdt| d |d d �� }|d �r6|d|d 7 }|d6@ �r�|d7 }nH|d7@ �rZd }n8|d8@ �rjd }n(|d9@ �rzd }n|d! d"@ �r�d#}nd$}|d:@ �s�|d%7 }|d& d@ �r�d'�p�d(} | |d& d @ �r�d)�p�d(7 } | |d& d*@ �r�d+�p�d(7 } d,|d- |d. f }
td/|
| |f � q�W q$W d S );Nz
pstree.imgr@ r4 z mm-%d.imgr z%dz %-36s %sZexerY Z exe_file_id)r] rI �vmas�statusr �
r= zips[%lx]�shmid� z
shmem[%lx]� z packet[%lx]� � Zpgoffz + %#lxz (s)z [stack]� z [vsyscall]� z [vdso]�flags� z [stack?]rT z *�prot�r�-�wr �xz %08lx-%08lx�start�endz
%-36s%s%si ru i �@ � � r~ rr r rn r ) r r! r" r rf r2 r# rb rW ) r rF Zvidsr0 r4 Zmmi�vma�st�fnrv Zastrr r r
� explore_mems sT
r� c
C s� t jjt| d��}�x�|d D �]�}t|d�}t jjt| d| ��d d d }t jjt| d| ��d }td| � d}d }�x0|d d � D �] }d
|d |d f } x,|t|�k r�|| d
|d kr�|d 7 }q�W |d |d d> }
d} x�|t|�k �r�|| d |
k �r�|| } ||k�r,| d7 } nR| d| d | d
| d d? f 7 } | d d @ �rz| dt| d| d d�� 7 } |}| dd 7 } |d 7 }q�W |d 8 }td| | f � q�W q W d S ) Nz
pstree.imgr@ r4 z mm-%d.imgr rj zpagemap-%d.imgz%dr z %lx / %-8dZvaddrZnr_pagesr| � rT r{ z ~z
%08lx / %-8drk rp rq r= rY rm )r] rI z
%23sz%-24s%s���r} r~ r )r r! r" r r2 r# rL rb )
r rF r0 r4 rj ZpmsZvmiZpvmiZpmZpstrZpmeZvstrr� r r r
� explore_rss9 s<
" "
r� )rG �fds�mems�rssc C s t | d | � d S )N�what)� explorers)r r r r
�exploreh s r� c C s� d} t j| t jd�}|jddtd� |jdd�}|jdd d�}|jd
d d d
� |jdddd� |jdddd� |jtdd� |jddd�}|jdddd� |jdddd� |jt d� |jdd d�}|jd � |jt
d� |jd d d�}|jd � |jd!d"d#d$d%gd&� |jt d� |jd'd(d�}|jd � |jd)d*d d
� |jtd+d d,� t |j
� �}|�s�tjj|j� � tjjd-� tjd.� |d/ |� d S )0NzCRiu Image Tool)� descriptionZformatter_classz --version�version)�actionr� z-Use crit CMD --help for command-specific help)�helpr z+convert criu image from binary type to jsonz--prettyzIMultiline with indents and some numerical fields in field-specific format�
store_true)r� r� z-iz--inz<criu image in binary format to be decoded (stdin by default)z-oz--outz:where to put criu image in json format (stdout by default)F)�funcr r. z+convert criu image from json type to binaryz:criu image in json format to be encoded (stdin by default)z<where to put criu image in binary format (stdout by default))r� r/ zshow info about imager rz zexplore image dirr r� rG r� r� r� )�choicesZshowz5convert criu image from binary to human-readable jsonz--noplz%do not show entry payload (if exists)T)r� r r z crit: error: too few arguments
r r� )�argparse�ArgumentParserZRawTextHelpFormatter� add_argumentr Zadd_subparsersZ
add_parserZ set_defaultsr r. r/ r� �vars�
parse_argsr r% r) Z format_usager& ) Zdesc�parserZ
subparsersZ
decode_parserZ
encode_parserZ info_parserZx_parserZ show_parserr r r r
�mainl sh
r� �__main__)r )#r� r r' r r rT r r r r r r. r/ r2 r3 r> rH rJ rO rQ rR rS rV r_ r\ r^ rb re rf r� r� r� r� r� r: r r r r
�<module> sZ
5(E