ZerethShell
File Manager
SQL Manager
/
usr
/
lib
/
python3.6
/
site-packages
/
crit
/
__pycache__
__main__.cpython-36.pyc
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
Kaydet
Vazgeç