File: //lib/python3.9/site-packages/pycriu/images/__pycache__/pb2dict.cpython-39.pyc
a
k�x]~. � @ sX d dl mZ d dlZd dlmZmZ d dlmZ d dlZd dl Z d dl
Z
d dl Z d dl Z de
e �vrte je _e je _ejeejeejeejeejeejeejeejeejeej eej e ej e i Z!dd� Z"dd � Z#d
d � Z$d d
� Z%dd� Z&dd� Z'dd� Z(g d�Z)g d�Z*g d�Z+g d�Z,g d�Z-e)e*e+e,e-d�Z.ddd d d �iZ/d d d!d"d#d$�d%d&d'd(d#d)�d*d+d,d-d.d/d0d1d2d3d4�
d5d6d7d8d9�d:�Z0d;d<� e1e/�2� �D �Z3d=d<� e1e0�2� �D �Z4e/e3fe0e4fd>�Z5d?d@� Z6dAdB� Z7dCZ8dDdE� Z9dFdG� Z:dHdI� Z;dJdK� Z<dLdM� Z=dNdO� Z>dPe=iZ?dPe>iZ@dQdR� ZAdSdT� ZBdUdV� ZCd`dXdY�ZDdadZd[�ZEd\d]� ZFd^d_� ZGdS )b� )�FieldDescriptorN)� IPv4Address�
ip_address)� IPv6Address� 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_hex0 s r c C s | � � jtj jS r )r r r
r Zipaddr
r r r �
_marked_as_ip4 s r c C s | � � jtj jS r )r r r
r �flagsr
r r r �_marked_as_flags8 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_odev@ s r c C s | � � jtj jS r )r r r
r �dictr
r r r �_marked_as_dictD s r c C s | � � jtj jS r )r r r
r �convr
r r r � _custom_convH s r ))Z PROT_READ� )Z
PROT_WRITE� )Z PROT_EXEC� ))Z
MAP_SHAREDr )Z MAP_PRIVATEr )ZMAP_ANON� )Z
MAP_GROWSDOWN� ))Z
VMA_AREA_NONEr )ZVMA_AREA_REGULARr )ZVMA_AREA_STACKr )ZVMA_AREA_VSYSCALLr )Z
VMA_AREA_VDSO� )Z
VMA_AREA_HEAPr )ZVMA_FILE_PRIVATE�@ )ZVMA_FILE_SHARED� )ZVMA_ANON_SHAREDr )ZVMA_ANON_PRIVATEi )ZVMA_AREA_SYSVIPC� )ZVMA_AREA_SOCKETi )Z
VMA_AREA_VVARi )ZVMA_AREA_AIORINGi )Z
VMA_UNSUPPl ))�O_WRONLYr )�O_RDWRr )�O_APPENDr$ )�O_DIRECTi @ )� O_LARGEFILEi � ))Z PE_PARENTr )ZPE_LAZYr )Z
PE_PRESENTr )z mmap.protz
mmap.flagsz mmap.statusz rfile.flagsz
pmap.flagsZ
task_stateZAliveZZombieZStopped)r � � ZUNIXZINETZINET6ZNETLINKZPACKET)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�r7 �k�vr r r r8 � s �
�r8 c C s( i | ] \}}|d d� t |�� �D ��qS )c S s i | ]\}}||�qS r r r6 r r r r8 � s �r9 r: r= r r r r8 � s �
�)�gen�skc sP dd� � fdd�|D �D �}� t dd� |D �� M � � rF|�d� � d�|�S )Nc S s g | ] }|d �qS �r r �r7 �xr r r �
<listcomp>� � z map_flags.<locals>.<listcomp>c s g | ]}� |d @ r|�qS �r r rC ��valuer r rE � rF c S s g | ] }|d �qS rG r rC r r r rE � rF �0x%xz | )�sum�append�join)rI � flags_map�bsr rH r � map_flags� s
rP c s: | dkr dS t |�� t� fdd�dd� | �d�D �D ��S )N� r c s" g | ]}t t� �||��d ��qS rB )�int�str�getrC �Zbdr r rE � s �z unmap_flags.<locals>.<listcomp>c S s g | ] }|� � �qS r )�striprC r r r rE � rF �|)r rK �split)rI rN r rU r � unmap_flags� s �rY � 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 rI r r r �
decode_dev� s
�r_ 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 )rR rC r r r rE � rF z encode_dev.<locals>.<listcomp>�:r r )rX r r[ �makedevr^ )r rI r r r r �
encode_dev� s rb c C s
t �| �S r )�base64r rH r r r �
encode_base64� s rd c C s
t �| �S r )rc � decodebytesrH r r r �
decode_base64� s rf c C s
t �| �S r )�quopri� encodestringrH r r r � encode_unix� s ri c C s
t �| �S r )rg � decodestringrH r r r � decode_unix� s rk Z unix_namec C s t | �}|rt| S tS d S r )r �encoderd �r �cr r r �
get_bytes_enc� s ro c C s t | �}|rt| S tS d S r )r �decoderf rm r r r �
get_bytes_dec� s rq c C s dt v rt}nttf}t| |�S )N�
basestring)� __builtins__rr rS �bytes�
isinstance)rI Z string_typesr r r � is_string s rv Fc C s( |s t | �}| jtjkr$t|||�S | jtjkr<t| �|�S | jtjkrZ| jj �
|d �j S | jt v �rt | j }|�r|t
k�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 )NrJ r �!Field(%s) has unsupported type %d)r r3 �FD� TYPE_MESSAGE�pb2dict�
TYPE_BYTESro � TYPE_ENUM� enum_typeZvalues_by_numberrT �name� _basic_castrR r r_ r �
flags_maps� ExceptionrP r � dict_maps)r rI �pretty�is_hex�castr rN �dctr r r �
_pb2dict_cast s:
�r� c C s� |r t �� ni }| �� D ]�\}}|jtjkr�g }|r�t|�r�t|�dkr`t� |d �}t
|�}nPdt� |d �d> t� |d �d> t� |d �d> t� |d � }t |�}|� |j
� q�|D ]}|� t||||�� q�nt||||�}|||j< 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�labelrx �LABEL_REPEATEDr �len�socketZntohlr r rL �
compressedr� r~ ) �pbr� r� �dr rI Zd_valr? Zaddrr r r rz 8 s,
�� � rz c C s
| j tjkrt| �|�S | j tjkr6| jj�|d �jS | j t v r�t | j }|t
kr�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 rw )r3 rx r{ rq r| r} Zvalues_by_namerT Znumberr rR rv r rb r r� r� rY r r� r~ )r rI r� r rN r� �retr r r �
_dict2pb_castW 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� rx r� �getattrrv r r �versionrL r� ZhtonlrR r� r3 ry �dict2pb�addr� Z SetInParent�setattr)r� r� r rI Zpb_val�valZivalr? r r r r� s:
�r� )FF)FF)HZgoogle.protobuf.descriptorr rx r
Z ipaddressr r r r� r� r[ rc rg �dirrh r rj re Z TYPE_FIXED64rR Z TYPE_FIXED32Z
TYPE_SFIXED64Z
TYPE_SFIXED32Z
TYPE_INT64Z TYPE_UINT64Z TYPE_SINT64Z
TYPE_INT32Z TYPE_UINT32Z TYPE_SINT32Z TYPE_BOOL�boolZ TYPE_STRINGrS r r r r r r r r Z
mmap_prot_mapZmmap_flags_mapZmmap_status_mapZrfile_flags_mapZpmap_flags_mapr� Zgen_mapsZsk_mapsr; r< Z gen_rmapsZsk_rmapsr� rP rY r^ r_ rb rd rf ri rk rl rp ro rq rv r� rz r� r� r r r r �<module> s� �� ��
���
��#
�
��
(
(