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 dZd9ddZdd 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 d3Z d4d5 Z d6d7 Z!e"d8krDe! dS ): N ) __version__c C s2 | d rt | d dS tjj r&tjS tjjS d S )Ninrb)opensysstdinisattybuffer)opts r /usr/lib/python3.6/__main__.pyinf s
r c C s8 | d r"d}|rd}t | d |S |r,tjS tjjS d S )Noutzwb+zw+)r r stdoutr
)r decodemoder r r
outf s r c C s t tjj| d |ddS )Ndirr )r )r ospathjoin)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 )
NprettynoplzSUnknown magic %#x.
Maybe you are feeding me an image with raw data(i.e. pages.img)?)filer T)indent
)pycriuimagesloadr ZMagicExceptionprintmagicr stderrexitr jsondumpr write)r r imgexcfr 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 )pvalr r r
get_task_idL s r2 c @ s e Zd Zdd ZdS )ps_itemc C s, t |d| _|d | _|| _|| _g | _d S )Npidppid)r2 r4 r5 r0 corekids)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| jdt| jddd| | 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.imgentriesz core-%d.imgr4 r z%7s%7s%7s %sPIDPGIDSIDCOMM)rA rB rC rD ) r r! r" r r2 r3 r4 r5 r7 appendr# r> ) r Zpssps_imgr0 r6 psZpsrr4 Zppr r r
explore_psf s"
rH c C sn t d kr }
d} x|t|k r|| d |
k r|| } ||kr,| d7 } nR| d| d | d
| d d? f 7 } | d d @ rz| dt| d| d d 7 } |}| dd 7 } |d 7 }qW |d 8 }td| | f qW 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%sr} r~ r )r r! r" r r2 r# rL rb )
r rF r0 r4 rj ZpmsZvmiZpvmiZpmZpstrZpmeZvstrr r r r
explore_rss9 s<
" "
r )rG fdsmemsrssc C s t | d | d S )Nwhat) 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
}|stjj|j tjjd- tjd. |d/ | d S )0NzCRiu Image Tool) descriptionZformatter_classz --versionversion)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 rH rJ rO rQ rR rS rV r_ r\ r^ rb re rf r r r r r r: r r r r
sZ
5(E