File: //lib/python3.9/site-packages/pycriu/__pycache__/cli.cpython-39.pyc
a
k�x]�* � @ sH d dl mZ d dlZd dlZd dlZd dlZd dlZdd� Zdd� Zdd� Z d d
� Z
d d � Z d
d� Z dd� Z
G dd� d�Zd8dd�Zdd� Zdadd� Zdd� Zdd � Zd d � Zd d � Zedd!d"�edd#d"�edd$d"�d%�Zd&d'� Zi Zd(d)� Zd*d+� ZG d,d-� d-�Z d.d/� Z d0d1� Z eee e d2�Z d3d4� Z d5d6� Z!e"d7k�rDe!� dS )9� )�print_functionNc C s | d rt | d d�S tjS d S )N�in�rb)�open�sys�stdin��opts� r
�./usr/lib/python3.9/site-packages/pycriu/cli.py�inf
s r c C s | d rt | d d�S tjS d S )N�outzw+)r r �stdoutr r
r
r �outf s r c C s t tj�| d |��S )N�dir)r �os�path�join)r �namer
r
r �dinf s r c
C s� d }z t j�t| �| d | d �}W nF t jjyj } z*td|j tjd� t� d� W Y d }~n
d }~0 0 | d rxd}t
| �}t j |||d� |tj
kr�|�d� d S ) N�pretty�noplzSUnknown magic %#x.
Maybe you are feeding me an image with raw data(i.e. pages.img)?)�file� � ��indent�
)�pycriu�images�loadr ZMagicException�print�magicr �stderr�exitr �json�dumpr �write)r r �img�exc�fr
r
r �decode s ��
r+ c C s$ t �t| ��}tj�|t| �� d S )N)r% r r r r r& r )r r( r
r
r �encode0 s r, c C s, t j�t| ��}tj|tjdd� t� d S )Nr r ) r r �infor r% r& r r r! )r Zinfsr
r
r r- 5 s r- c C s || v r| | S | d| d S )NZns_r r
)�p�valr
r
r � get_task_id; s r0 c @ s e Zd Zdd� ZdS )�ps_itemc C s, t |d�| _|d | _|| _|| _g | _d S )N�pid�ppid)r0 r2 r3 r. �core�kids)�selfr. r4 r
r
r �__init__E s
zps_item.__init__N)�__name__�
__module__� __qualname__r7 r
r
r
r r1 D s r1 c C sZ t d| jt| jd�t| jd�dd| | jd d f � | jD ]}t|||d � q@d S ) Nz%7d%7d%7d %s%sZpgidZsid� r ZtcZcommr )r! r2 r0 r. r4 r5 �show_ps)r. r �depthZkidr
r
r r< M s � ��
r< c C s� i }t j�t| d��}|d D ]<}t j�t| dt|d� ��}t||d d �}|||j< q d }|D ]2}|| }|jdkr�|}qd||j }|j� |� qdt
dd � t || � d S )N�
pstree.img�entriesz core-%d.imgr2 r z%7s%7s%7s %s)ZPIDZPGIDZSIDZCOMM) r r r r r0 r1 r2 r3 r5 �appendr! r< ) r Zpss�ps_imgr. r4 �psZpsrr2 �ppr
r
r �
explore_psU s$ �
rD c C sf t d u r4ztj�t| d��d a W n g a Y n0 tt �dkrDd S t D ]}|d |krH| S qHd S )Nz files.imgr? r �id)� files_imgr r r r �len)r �ft�fidr* r
r
r �ftype_find_in_filesp s
rJ c C sh t | ||�}|r ||d S |d d krBtj�t| |��d |d< |d D ]}|d |krJ| S qJd S )N�fieldr( r? rE )rJ r r r r )r rH rI r( r* r
r
r �ftype_find_in_image� s
rL c C s t | ||d�}|r|d p dS )Nz
reg-files.imgr z unknown path�rL )r rH rI Zrfr
r
r � ftype_reg� s rN c C s" t | ||d�}|r d|d p dS )Nz pipes.imgzpipe[%d]Zpipe_idzpipe[?]rM )r rH rI r. r
r
r �
ftype_pipe� s rO 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]�inoZpeerrM )r rH rI Zux�nr
r
r �
ftype_unix� s
rS Zreg)�getr( rK �pipeZusk)�REG�PIPEZUNIXSKc C s d|d |f S )Nz%s.%d�typr
)r rH rI r
r
r � ftype_gen� s rY c C sZ |d |d f}t �|d �}|sVt�|d t|d d��}|d | ||d �}|t |< |S )N�typerE )rT rX rT )� files_cacherT �
file_typesrY )r �fd�keyr* rH r
r
r � get_file_str� s r_ c C s� t j�t| d��}|d D ]�}t|d�}t j�t| d| ��}|d d d }t j�t| d| ��}td| � |d D ] }td |d
t| |�f � qxt j�t| d | ��d d }td d
t| d|d d��f � td dt| d|d d��f � qd S )Nr> r? r2 z
ids-%s.imgr Zfiles_idz
fdinfo-%d.img�%dz %7d: %sr] z fs-%d.imgz %7s: %s�cwdrV Zcwd_id�rZ rE �rootZroot_id�r r r r r0 r! r_ )r rA r. r2 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)r6 r
r
r r7 � s zvma_id.__init__c C s4 | j �|d �}|s0| j}| jd7 _|| j |< |S rg )rh rT ri )r6 Ziid�retr
r
r rT � s
z
vma_id.getN)r8 r9 r: r7 rT r
r
r
r rf � s rf c C s t j�t| d��}t� }|d D �]�}t|d�}t j�t| d| ��d d }td| � tddt| d |d
d ��f � |d D �]�}|d
}|d@ r�dd|�|d � }n�|d@ r�dd|�|d � }n�|d@ r�dd|�|d � }n�|d@ �r@dt| d |d d �� }|d �r,|d|d 7 }|d@ �r�|d7 }nH|d@ �rPd }n8|d @ �r`d }n(|d @ �rpd }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�q d S )/Nr> r? r2 � mm-%d.imgr r` z %-36s %sZexerV Z exe_file_idrb �vmas�statusi r; zips[%lx]�shmid� z
shmem[%lx]i z packet[%lx]� Zpgoffz + %#lx� z (s)� z [stack]r z [vsyscall]� z [vdso]�flagsz [stack?]rP r z *�prot�r�-�w�xz %08lx-%08lx�start�endz
%-36s%s%s) r r r r rf r0 r! r_ rT ) r rA Zvidsr. r2 Zmmi�vma�st�fnru Zastrr
r
r � explore_mems� sZ
��
�
r c
C s� t j�t| d��}|d D �]h}t|d�}t j�t| d| ��d d d }t j�t| d| ��d }td| � d}d }|d
d � D ]�}d |d |d
f } || d |d kr�|d
7 }q�|d |d
d> }
d} || d |
k �rj|| } ||k�r| d7 } nR| d| d | d | d d? f 7 } | d d@ �rP| dt| d| d d�� 7 } |}| dd 7 } |d
7 }q�|d
8 }td| | f � q�qd S ) Nr> r? r2 rk r rl zpagemap-%d.imgr` ���r z %lx / %-8dZvaddrZnr_pagesr{ � rP rz z ~z
%08lx / %-8drm rp r; rV rn rb z
%23sz%-24s%srd )
r rA r. r2 rl ZpmsZvmiZpvmiZpmZpstrZpmeZvstrr| r
r
r � explore_rss% sN
����
��
r� �rB ZfdsZmemsZrssc C s t | d | � d S )N�what)� explorersr r
r
r �exploreT s r� c C s� d} t j| t jd�}|jdd�}|jddd�}|jddd d
� |jd d d
d� |jdddd� |jtdd� |jddd�}|jd d dd� |jdddd� |jtd� |jddd�}|�d� |jt d� |jdd d�}|�d � |jd g d �d � |jt
d� |jd!d"d�}|�d� |jd#d$d d
� |jtd%d d&� t |� � �}|�spt
j�|�� � t
j�d'� t
�d(� |d) |� d S )*NzCRiu Image Tool)� description�formatter_classz-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� �actionz-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 ry zexplore image dirr 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�ArgumentParser�RawTextHelpFormatter�add_subparsers�
add_parser� add_argument� set_defaultsr+ r, r- r� �vars�
parse_argsr r# r' � format_usager$ ) Zdesc�parserZ
subparsersZ
decode_parserZ
encode_parserZ info_parserZx_parserZ show_parserr r
r
r �mainX s| ���������
�
�
r� �__main__)r )#Z
__future__r r� r r% r r r r r r+ r, r- r0 r1 r< rD rF rJ rL rN rO rS r\ rY r[ r_ re rf r r� r� r� r� r8 r
r
r
r �<module> sd
���� 5(�E