????
Your IP : 216.73.217.84
a
?9j�1 � @ sZ d dl Z d dlZd dlZd dlZd dlZd dlmZmZmZ d dl m
Z d dl Z de
e �vrle 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(g d�Z)g d�Z*g d�Z+g d�Z,g d�Z-g d�Z.e)e*e+e,e.e-d�Z/ddd d d �iZ0d d d!d"d#d$�d%d&d'd(d#d)�d*d+d,d-d.d/d0d1d2d3d4�
d5d6d7d8d9�d:�Z1d;d<� e2e0�3� �D �Z4d=d<� e2e1�3� �D �Z5e0e4fe1e5fd>�Z6d?d@� Z7dAdB� Z8dCZ9dDdE� Z:dFdG� Z;dHdI� Z<dJdK� Z=dLdM� Z>dNdO� Z?dPe>iZ@dPe?iZAdQdR� ZBdSdT� ZCdUdV� ZDd`dXdY�ZEdadZd[�ZFd\d]� ZGd^d_� ZHdS )b� N)� IPv4Address� IPv6Address�
ip_address)�FieldDescriptor� encodebytesc C s | � � jtj jS �N)�
GetOptions�
Extensions�opts_pb2�criu�hex��field� r �9/usr/lib/python3.9/site-packages/pycriu/images/pb2dict.py�_marked_as_hex1 s r c C s | � � jtj jS r )r r r
r Zipaddr
r r r �
_marked_as_ip5 s r c C s | � � jtj jS r )r r r
r �flagsr
r r r �_marked_as_flags9 s r c C s | � � jtj jS r )r r r
r �devr
r r r �_marked_as_dev= s r c C s | � � jtj jS r )r r r
r Zodevr
r r r �_marked_as_odevA s r c C s | � � jtj jS r )r r r
r �dictr
r r r �_marked_as_dictE s r c C s | � � jtj jS r )r r r
r �convr
r r r � _custom_convI s r ))Z PROT_READ� )Z
PROT_WRITE� )Z PROT_EXEC� ))Z
MAP_SHAREDr )Z MAP_PRIVATEr )Z
MAP_DROPPABLE� )ZMAP_ANON� )Z
MAP_GROWSDOWN� ))Z
VMA_AREA_NONEr )ZVMA_AREA_REGULARr )ZVMA_AREA_STACKr )ZVMA_AREA_VSYSCALLr )Z
VMA_AREA_VDSOr )Z
VMA_AREA_HEAPr )ZVMA_FILE_PRIVATE�@ )ZVMA_FILE_SHARED� )ZVMA_ANON_SHAREDr! )ZVMA_ANON_PRIVATE� )ZVMA_AREA_SYSVIPC� )ZVMA_AREA_SOCKET� )Z
VMA_AREA_VVAR� )ZVMA_AREA_AIORING� )ZVMA_AREA_MEMFD� @ )ZVMA_AREA_SHSTK� � )ZVMA_AREA_UPROBES� )Z
VMA_UNSUPPl ))�O_WRONLYr )�O_RDWRr )�O_CREATr" )�O_EXCLr# )�O_NOCTTYr! )�O_TRUNCr$ )�O_APPENDr% )�
O_NONBLOCKr& )�O_DSYNCr' )ZFASYNCr( )�O_DIRECTr) )� O_LARGEFILEr* )� O_DIRECTORYi )�
O_NOFOLLOWr+ )� O_NOATIMEi )� O_CLOEXECi ))Z F_SEAL_SEALr )Z
F_SEAL_SHRINKr )Z F_SEAL_GROWr )Z F_SEAL_WRITEr )ZF_SEAL_FUTURE_WRITE� ))Z PE_PARENTr )ZPE_LAZYr )Z
PE_PRESENTr )z mmap.protz
mmap.flagsz mmap.statusz rfile.flagsz
pmap.flagsz seals.flagsZ
task_stateZAliveZZombieZStopped)r � � ZUNIXZINETZINET6ZNETLINKZPACKET)r r �
r; � ZSTREAMZDGRAMZRAWZ SEQPACKET)r r r< � r> Z ESTABLISHEDZSYN_SENTZSYN_RECVZ FIN_WAIT1Z FIN_WAIT2Z TIME_WAITZCLOSEZ
CLOSE_WAITZLAST_ACKZLISTEN)
r r r< r r@ r= � r � r> ZIPZTCPZUDPZUDPLITE)r r= r? � )Zfamily�type�state�protoc C s( i | ] \}}|d d� t |�� �D ��qS )c S s i | ]\}}||�qS r r ��.0Zk2Zv2r r r �
<dictcomp>� s ��<dictcomp>.<dictcomp>��list�items�rH �k�vr r r rI � s �
�rI c C s( i | ] \}}|d d� t |�� �D ��qS )c S s i | ]\}}||�qS r r rG r r r rI � s �rJ rK rN r r r rI � s �
�)�gen�skc sP dd� � fdd�|D �D �}� t dd� |D �� M � � rF|�d� � d�|�S )Nc S s g | ] }|d �qS �r r �rH �xr r r �
<listcomp>� � z map_flags.<locals>.<listcomp>c s g | ]}� |d @ r|�qS �r r rT ��valuer r rV � rW c S s g | ] }|d �qS rX r rT r r r rV � rW �0x%xz | )�sum�append�join)rZ � flags_map�bsr rY r � map_flags� s
ra c s: | dkr dS t |�� t� fdd�dd� | �d�D �D ��S )N� r c s" g | ]}t t� �||��d ��qS rS )�int�str�getrT �Zbdr r rV � s �z unmap_flags.<locals>.<listcomp>c S s g | ] }|� � �qS r )�striprT r r r rV � rW �|)r r\ �split)rZ r_ r rf r � unmap_flags� s �rj � c C s@ t | �r dt�|�t�|�f S d|t? |dt> d @ f S d S )Nz%d:%dr )r �os�major�minor�kern_minorbits)r rZ r r r �
decode_dev� s
�rp c C sH dd� |� d�D �}t| �r0t�|d |d �S |d t> |d B S d S )Nc S s g | ] }t |��qS r )rc rT r r r rV � rW z encode_dev.<locals>.<listcomp>�:r r )ri r rl �makedevro )r rZ r r r r �
encode_dev� s rs c C s t �| ��� S r )�base64r �decoderY r r r �
encode_base64� s rv c C s t �t�| ��S r )rt � decodebytesrd �encoderY r r r �
decode_base64 s ry c C s
t �| �S r )�quopri� encodestringrY r r r � encode_unix s r| c C s
t �| �S r )rz � decodestringrY r r r � decode_unix s r~ Z unix_namec C s t | �}|rt| S tS d S r )r rx rv �r �cr r r �
get_bytes_enc s r� c C s t | �}|rt| S tS d S r )r ru ry r r r r �
get_bytes_dec s r� c C s dt v rt}nttf}t| |�S )N�
basestring)� __builtins__r� rd �bytes�
isinstance)rZ Z string_typesr r r � is_string s r� Fc C s( |s t | �}| jtjkr$t|||�S | jtjkr<t| �|�S | jtjkrZ| jj �
|d �j S | jt v �rt | j }|�r|t
u �r|r�d| S t| �r�t| |�S t| �}|r�z t| }W n ty� d| Y S 0 t||�S t| �}|�rt| d | j �
|||��S ||�S td| j | jf ��d S )Nr[ r �!Field(%s) has unsupported type %d)r rD �FD� TYPE_MESSAGE�pb2dict�
TYPE_BYTESr� � TYPE_ENUM� enum_typeZvalues_by_numberre �name� _basic_castrc r rp r �
flags_maps� Exceptionra r � dict_maps)r rZ �pretty�is_hex�castr r_ �dctr r r �
_pb2dict_cast) s:
�r� c
C s* |r t �� ni }| �� D �]
\}}|jtjkr�g }|r�t|�r�t|�dkrbt� |d �}t
|�}nPdt� |d �d> t� |d �d> t� |d �d> t� |d � }t |�}|� |j
� q�|D ]}|� t||||�� q�nt||||�}z|�� ||j< W q ttf�y" |||j< Y q0 q|S )z^
Convert protobuf msg to dictionary.
Takes a protobuf message and returns a dict.
r r �` r" r r r< )� collections� OrderedDictZ
ListFields�labelr� �LABEL_REPEATEDr �len�socketZntohlr r r] �
compressedr� ru r� �UnicodeDecodeError�AttributeError) �pbr� r� �dr rZ Zd_valrP �addrr r r r� Q s2
�� �r� c C s
| j tjkrt| �|�S | j tjkr6| jj�|d �jS | j t v r�t | j }|t
u r�t |�r�t | �rlt
| |�S t| �}|r�z t| }W n ty� Y n 0 t||�S t| �}|r�t| d | j �|d �}|d u r�||d�}|S ||d�S ||�S ntd| j| j f ��d S )Nr r r� )rD r� r� r� r� r� Zvalues_by_namere �numberr� rc r� r rs r r� r� rj r r� r� )r rZ r� r r_ r� �retr r r �
_dict2pb_casts s6
�r� c C s� |j jD �]�}|j| vrq| |j }|jtjk�rDt||jd�}t|d ��rt|��rt |d �}|j
dkr�|� t �
t|��� q|j
dkr�t|�}|� t �
|d? d@ �� |� t �
|d? d@ �� |� t �
|d? d@ �� |� t �
|d? d@ �� qtd |j
��q|D ]4}|jtjk�r.t||�� � n|� t||�� �q q|jtjk�rxt||j��� t|t||jd�� qt||jt||�� q|S )
zd
Convert dictionary to protobuf msg.
Takes dict and protobuf message to be merged into.
Nr r r= r� l �� r" r z Unknown IP address version %d)Z
DESCRIPTOR�fieldsr� r� r� r� �getattrr� r r �versionr] r� Zhtonlrc r� rD r� �dict2pb�addr� Z SetInParent�setattr)r� r� r rZ Zpb_val�valZivalrP r r r r� � s:
�r� )FF)FF)Irt r� rl rz r� � ipaddressr r r Zgoogle.protobuf.descriptorr r� r
�dirr{ r r} rw Z TYPE_FIXED64rc Z TYPE_FIXED32Z
TYPE_SFIXED64Z
TYPE_SFIXED32Z
TYPE_INT64Z TYPE_UINT64Z TYPE_SINT64Z
TYPE_INT32Z TYPE_UINT32Z TYPE_SINT32Z TYPE_BOOL�boolZ TYPE_STRINGrd r� r r r r r r r Z
mmap_prot_mapZmmap_flags_mapZmmap_status_mapZrfile_flags_mapZseals_flags_mapZpmap_flags_mapr� Zgen_mapsZsk_mapsrL rM Z gen_rmapsZsk_rmapsr� ra rj ro rp rs rv ry r| r~ rx ru r� r� r� r� r� r� r� r r r r �<module> s� ��
��
���
��#
�
��
(
"(