3
i1 @ s d dl Z d dlZd dlZd dlZd dlZd dlmZmZmZ d dl m
Z d dl Z de
e krle je _e je _e jee jee jee jee jee jee jee jee jee j ee j e e j e i Z!dd Z"dd Z#d d
Z$d d Z%d
d Z&dd Z'dd Z(dddgZ)dddddgZ*ddddddddddddddddddgZ+ddddddddddddddddgZ,dddddgZ-dddgZ.e)e*e+e,e.e-dfZ/dgdhdidjdkiZ0dldmdndodpdqdrdsdtdudpdvdwdxdydzd{d|d}d~ddd
ddddddZ1dd e2e0j3 D Z4dd e2e1j3 D Z5e0e4fe1e5fdZ6dd Z7dd Z8dZ9dd Z:dd Z;dd Zdd Z?de>iZ@de?iZAdd ZBdd ZCdd ZDdddZEdddZFdd ZGdd ZHdS ) N) IPv4Address IPv6Address
ip_address)FieldDescriptor encodebytesc C s | j jtj jS )N)
GetOptions
Extensionsopts_pb2criuhex)field r
/usr/lib/python3.6/pb2dict.py_marked_as_hex1 s r c C s | j jtj jS )N)r r r r
Zipadd)r r
r
r
_marked_as_ip5 s r c C s | j jtj jS )N)r r r r
flags)r r
r
r _marked_as_flags9 s r c C s | j jtj jS )N)r r r r
dev)r r
r
r _marked_as_dev= s r c C s | j jtj jS )N)r r r r
Zodev)r r
r
r _marked_as_odevA s r c C s | j jtj jS )N)r r r r
dict)r r
r
r _marked_as_dictE s r c C s | j jtj jS )N)r r r r
Zconv)r r
r
r _custom_convI s r PROT_READ
PROT_WRITE PROT_EXEC
MAP_SHARED MAP_PRIVATE
MAP_DROPPABLE MAP_ANON
MAP_GROWSDOWN
VMA_AREA_NONEVMA_AREA_REGULARVMA_AREA_STACKVMA_AREA_VSYSCALL
VMA_AREA_VDSO
VMA_AREA_HEAP VMA_FILE_PRIVATE VMA_FILE_SHARED VMA_ANON_SHAREDVMA_ANON_PRIVATE VMA_AREA_SYSVIPC
VMA_AREA_SOCKET
VMA_AREA_VVAR VMA_AREA_AIORING
VMA_AREA_MEMFD VMA_AREA_SHSTK VMA_AREA_UPROBES
VMA_UNSUPP O_WRONLYO_RDWRO_CREAT@ O_EXCL O_NOCTTYO_TRUNC O_APPEND
O_NONBLOCK O_DSYNC FASYNC O_DIRECT @ O_LARGEFILE O_DIRECTORY
O_NOFOLLOW O_NOATIME O_CLOEXEC F_SEAL_SEAL
F_SEAL_SHRINK F_SEAL_GROW F_SEAL_WRITEF_SEAL_FUTURE_WRITE PE_PARENTPE_LAZY
PE_PRESENT)z mmap.protz
mmap.flagsz mmap.statusz rfile.flagsz
pmap.flagsz seals.flagsZ
task_stateZAliveZZombieZStopped)r r, r0 ZUNIXZINETZINET6ZNETLINKZPACKET)r r r7 rh rC ZSTREAMZDGRAMZRAWZ SEQPACKET)r r r, r. r7 Z ESTABLISHEDZSYN_SENTZSYN_RECVZ FIN_WAIT1Z FIN_WAIT2Z TIME_WAITZCLOSEZ
CLOSE_WAITZLAST_ACKZLISTEN)
r r r, r r. r0 r2 r" r5 r7 ZIPZTCPZUDPZUDPLITE)r r0 rC )Zfamilytypestateprotoc C s( i | ] \}}d d t |j D |qS )c S s i | ]\}}||qS r
r
).0k2v2r
r
r
s z.)listitems)rp kvr
r
r rs s rs c C s( i | ] \}}d d t |j D |qS )c S s i | ]\}}||qS r
r
)rp rq rr r
r
r rs s z.)rt ru )rp rv rw r
r
r rs s )genZskc sP dd fdd|D D } t dd |D M rF|jd dj|S )Nc S s g | ] }|d qS )r r
)rp xr
r
r
s z map_flags..c s g | ]} |d @ r|qS )r r
)rp ry )valuer
r rz s c S s g | ] }|d qS )r r
)rp ry r
r
r rz s z0x%xz | )sumappendjoin)r{ flags_mapZbsr
)r{ r map_flags s
r c s: | dkr dS t | t fdddd | jdD D S )N r c s" g | ]}t t j||d qS )r )intstrget)rp ry )bdr
r rz s z unmap_flags..c S s g | ] }|j qS r
)strip)rp ry r
r
r rz s |)r r| split)r{ r r
)r r unmap_flags s
r c C s@ t | r dtj|tj|f S d|t? |dt> d @ f S d S )Nz%d:%dr )r osmajorminorkern_minorbits)r r{ r
r
r
decode_dev s r c C sH dd |j dD }t| r0tj|d |d S |d t> |d B S d S )Nc S s g | ] }t |qS r
)r )rp ry r
r
r rz s z encode_dev..:r r )r r r makedevr )r r{ r r
r
r
encode_dev s r c C s t j| j S )N)base64r decode)r{ r
r
r
encode_base64 s r c C s t jtj| S )N)r decodebytesr encode)r{ r
r
r
decode_base64 s r c C s
t j| S )N)quopri encodestring)r{ r
r
r encode_unix s r c C s
t j| S )N)r decodestring)r{ r
r
r decode_unix s r Z unix_namec C s t | }|rt| S tS d S )N)r r r )r cr
r
r
get_bytes_enc s r c C s t | }|rt| S tS d S )N)r r r )r r r
r
r
get_bytes_dec s r c C s dt krt}nttf}t| |S )N
basestring) __builtins__r r bytes
isinstance)r{ Z string_typesr
r
r is_string s r Fc C s$ |s t | }| jtjkr$t|||S | jtjkr tj |d d> tj |d d> tj |d }t |}|j |j
n"x0|D ]}|j t|||| qW nt||||}y|j ||j< W q ttfk
r, |||j< Y q X q W |S ) z^
Convert protobuf msg to dictionary.
Takes a protobuf message and returns a dict.
r r r$ r, r ` rI r$ ) collections OrderedDictZ
ListFieldslabelr LABEL_REPEATEDr lensocketZntohlr r r} Z
compressedr r r UnicodeDecodeErrorAttributeError) pbr r dr r{ Zd_valrw Zaddrr
r
r r Q s(
8
r c C s | j tjkrt| |S | j tjkr6| jjj|d jS | j t krt | j }|t
krt |rt | rlt
| |S t| }|ry t| }W n tk
r Y n X t||S t| }|rt| d | j j|d }|d kr||d}|S ||dS ||S ntd| j| j f d S )Nr r z!Field(%s) has unsupported type %d)rm r r r r r Zvalues_by_namer Znumberr r r r r r r r r r r r )r r{ r r r r retr
r
r
_dict2pb_casts s4
r c C s x|j jD ]}|j| kr q | |j }|jtjkrLt||jd}t|d oVt|r t |d }|j
dkr|j t j
t| q |j
dk rt|}|j t j
|d ? d@ |j t j
|d ? d@ |j t j
|d
? d@ |j t j
|d? d@ q td
|j
q x|D ]4}|jtjkr4t||j n|j t|| qW q |jtjkrt||jj t|t||jd q t||jt|| q W |S )zd
Convert dictionary to protobuf msg.
Takes dict and protobuf message to be merged into.
Nr r r0 r$ r, l r r z Unknown IP address version %dr rI r$ r )Z
DESCRIPTORZfieldsr r r r getattrr r r versionr} r Zhtonlr r rm r dict2pbaddr Z SetInParentsetattr)r r r r{ Zpb_valvalZivalrw r
r
r r s8
r )r r )r r )r r )r r )r r )r! r" )r# r$ )r% r& r )r' r r )r( r r )r) r r )r* r r" )r+ r" r$ )r- r$ rI )r/ rI rK )r1 rK r& )r3 r& )r4 r )r6 r )r8 r )r: r )r< r @ )r> r )r@ r )rB r )rD r )rF r )rG r )rH rI )rJ rK )rL r& )rM rN )rO rP )rQ rR )rS rT )rU rV )rW rX )rY rZ )r[ r\ )r] r^ )r_ r` )ra rb )rc r )rd r )re r )rf r" )rg rh r )ri r r )rj r r )rk r )FF)FF)Ir r r r r Z ipaddressr r r Zgoogle.protobuf.descriptorr r r dirr r r r Z TYPE_FIXED64r Z TYPE_FIXED32Z
TYPE_SFIXED64Z
TYPE_SFIXED32Z
TYPE_INT64Z TYPE_UINT64Z TYPE_SINT64Z
TYPE_INT32Z TYPE_UINT32Z TYPE_SINT32Z TYPE_BOOLboolZ TYPE_STRINGr r r r r r r r r Z
mmap_prot_mapZmmap_flags_mapZmmap_status_mapZrfile_flags_mapZseals_flags_mapZpmap_flags_mapr Zgen_mapsZsk_mapsrt ru Z gen_rmapsZsk_rmapsr r r r r r r r r r r r r r r r r r r r
r
r
r s
(
"(