D7net
Home
Console
Upload
information
Create File
Create Folder
About
Tools
:
/
usr
/
lib
/
python3.9
/
site-packages
/
crit
/
__pycache__
/
Filename :
__main__.cpython-39.pyc
back
Copy
a A9jG. � @ 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 tj�� r&tjS tjjS d S )N�in�rb)�open�sys�stdin�isatty�buffer��opts� r �1/usr/lib/python3.9/site-packages/crit/__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 tj�| d |�dd�S )N�dirr )r )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 | d�}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)?��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 sV zt �t| ��}W n* ty< tdtjd� t�d� Y n0 tj � |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 �r1 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 r2 F s r2 c C s || v r| | S | d| d S )NZns_r r )�p�valr r r � get_task_idL s r5 c @ s e Zd Zdd� ZdS )�ps_itemc C s, t |d�| _|d | _|| _|| _g | _d S )N�pid�ppid)r5 r7 r8 r3 �core�kids)�selfr3 r9 r r r �__init__V s zps_item.__init__N)�__name__� __module__� __qualname__r< r r r r r6 U s r6 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& r7 r5 r3 r9 r: �show_ps)r3 r �depthZkidr r r rA ^ s � �� rA 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.imgr7 r z%7s%7s%7s %s)ZPIDZPGIDZSIDZCOMM) r# r$ r% r r5 r6 r7 r8 r: �appendr& rA ) r Zpss�ps_imgr3 r9 �psZpsrr7 �ppr r r � explore_psf s$ � rI c C sl t d u r:ztj�t| d��d a W n ty8 g a Y n0 tt �dkrJd S t D ]}|d |krN| S qNd S )Nz files.imgrD r �id)� files_imgr# r$ r% r � Exception�len)r �ft�fidr/ r r r �ftype_find_in_files� s rP c C sx t | ||�}|r,|d |v r(||d S d S |d d u rRtj�t| |��d |d< |d D ]}|d |krZ| S qZd S )N�fieldr- rD rJ )rP r# r$ r% r )r rN rO r- r/ r r r �ftype_find_in_image� s rR c C s t | ||d�}|r|d p dS )Nz reg-files.imgr z unknown path�rR )r rN rO Zrfr r r � ftype_reg� s rT c C s" t | ||d�}|r d|d p dS )Nz pipes.imgzpipe[%d]Zpipe_idzpipe[?]rS )r rN rO r3 r r r � ftype_pipe� s rU 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]�inoZpeerrS )r rN rO Zux�nr r r � ftype_unix� s rY Zreg)�getr- rQ �pipeZusk)�REG�PIPEZUNIXSKc C s d|d |f S )Nz%s.%d�typr )r rN rO r r r � ftype_gen� s r_ c C sZ |d |d f}t �|d �}|sVt�|d t|d d��}|d | ||d �}|t |< |S )N�typerJ )rZ r^ rZ )� files_cacherZ � file_typesr_ )r �fd�keyr/ rN r r r � get_file_str� s re 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 )NrC rD r7 z ids-%s.imgr Zfiles_idz fdinfo-%d.img�%dz %7d: %src z fs-%d.imgz %7s: %s�cwdr\ Zcwd_id�r` rJ �rootZroot_id)r# r$ r% r r5 r& re )r rF r3 r7 ZidiZfdtZfdirc r r r � explore_fds� s, ����rj c @ s e Zd Zdd� Zdd� ZdS )�vma_idc C s i | _ d| _d S �Nr )� _vma_id__ids� _vma_id__last)r; r r r r< � s zvma_id.__init__c C s4 | j �|d �}|s0| j}| jd7 _|| j |< |S rl )rm rZ rn )r; Ziid�retr r r rZ � s z vma_id.getN)r= r> r? r< rZ r r r r rk � s rk 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 )/NrC rD r7 � mm-%d.imgr rf z %-36s %sZexer\ Z exe_file_idrh �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?]rV r z *�prot�r�-�w�xz %08lx-%08lx�start�endz %-36s%s%s) r# r$ r% r rk r5 r& re rZ ) r rF Zvidsr3 r7 Zmmi�vma�st�fnrz Zastrr r r � explore_mems sZ �� � r� c C s� t j�t| d��}|d D �]�}t|d�}t j�t| d| ��d d d }t j�t| d| ��d }td| � d}d }|d d � D �]}d |d |d f } |t|�k r�|| d |d kr�|d 7 }q�|d |d d> } d} |t|�k �r�|| d | k �r�|| } ||k�r | d7 } nR| d| d | d | d d? f 7 } | d d@ �rl| dt| d| d d�� 7 } |}| dd 7 } |d 7 }q�|d 8 }td| | f � q�qd S ) NrC rD r7 rp r rq zpagemap-%d.imgrf ���r z %lx / %-8dZvaddrZnr_pagesr� � rV r z ~z %08lx / %-8drr ru r@ r\ rs rh z %23sz%-24s%s)r# r$ r% r r5 r& rM re ) r rF r3 r7 rq ZpmsZvmiZpvmiZpmZpstrZpmeZvstrr� r r r � explore_rss9 sN ���� �� r� �rG ZfdsZmemsZrssc C s t | d | � d S )N�what)� explorersr 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�}|�d � |jt d� |jd d d�}|�d � |jd!g d"�d#� |jt d� |jd$d%d�}|�d � |jd&d'd d � |jtd(d d)� t |� � �}|�s�tj�|�� � tj�d*� t�d+� |d, |� d S )-NzCRiu Image Tool)� description�formatter_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 r1 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� r2 zshow info about imager r~ 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_argumentr �add_subparsers� add_parser� set_defaultsr r1 r2 r� �vars� parse_argsr r( r, � format_usager) ) Zdesc�parserZ subparsersZ decode_parserZ encode_parserZ info_parserZx_parserZ show_parserr r r r �mainl s~ ��������� � � r� �__main__)r )#r� r r* r r# rV r r r r r r1 r2 r5 r6 rA rI rK rP rR rT rU rY rb r_ ra re rj rk r� r� r� r� r� r= r r r r �<module> sd ���� 5(�E