← Back
Editing: images.cpython-36.pyc
3 i5M � C @ s� d dl Z d dlZd dlZd dlZd dlZddlmZ ddlmZ ddlmZ de e�krhej e_ ej e_ dZdZd Zd d � ZG d d � d e�Zdd� Zdd� ZG dd� d�ZG dd� d�ZG dd� d�ZG dd� d�ZG dd� d�ZG d d � d �ZG d d � d �Z G d d!� d!�Z G d"d#� d#�Z G d$d%� d%�Z eej �eej!�eej"�eej#�eej$�eej%�eej&�eej'�eej(�e� eej)�eej*�eej+e� �eej,�e� eej-�eej.�eej/�eej0�eej1�eej2�eej3�eej4�eej5�eej6�eej7�eej8�eej9�eej:�eej;�eej<�eej=�eej>�eej?�eej@�eejA�eejB�eejC�eejD�eejE�eejF�eejG�eejH�eejI�eejJ�eejK�eejL�eejM�eejN�eejOe� �eejOe� �eejPe� �eejQe � �eejRe � �eejSe � �eejT�eejU�eejV�eejW�eejX�eejY�eejZ�eej[�eej\�eej]e � �eej^�d&�BZ_d'd(� Z`d4d*d+�Zad,d-� Zbd5d.d/�Zcd0d1� Zdd2d3� ZedS )6� N� )�magic)�pb)�pb2dict� encodebytes� � � c C s | d |d B d S )Nr � )�x�yr r � /usr/lib/python3.6/images.py�round_up> s r c @ s e Zd Zdd� ZdS )�MagicExceptionc C s || _ d S )N)r )�selfr r r r �__init__C s zMagicException.__init__N)�__name__� __module__� __qualname__r r r r r r B s r c C s t jtj| ��S )z(A helper function to decode base64 data.)�base64� decodebytes�str�encode)�datar r r �decode_base64_dataG s r c C s | j tjtj|��� dS )z9A helper function to write base64 encoded data to a file.N)�writer r r r )�fr r r r �write_base64_dataL s r c @ sF e Zd ZdZddd�Zddd�Zddd �Zd d � Zd d � Zdd� Z dS )� entry_handlerzp Generic class to handle loading/dumping criu images entries from/to bin format to/from dict(json). Nc C s || _ || _dS )z= Sets payload class and extra handler class. N)�payload� extra_handler)r r r r r r r Y s zentry_handler.__init__Fc C s� g }x�i }| j � }|jd�}t|�dkr*P tjd|�\}|j|j|�� tj||�}| jr�|r�dd� } | jj||�} d| | � |d< n| jj ||�|d< |j |� qW |S )z� Convert criu image entries from binary format to dict(json). Takes a file-like object and returns a list with entries in dict(json) format. r r �ic S sJ x@d D ]8}| d k r6t | �| kr*d | |f S d | |f S | d } qW d | S )N� �K�M�G�T�P�E�Zg �@z%d%sBz%.1f%sBz%.1fYB)r" r# r$ r% r& r' r( r) )�int)ZnumZunitr r r �human_readablex s z*entry_handler.load.<locals>.human_readablez... <%s>�extra) r �read�len�struct�unpack�ParseFromStringr r �skip�load�append) r r �pretty� no_payload�entries�entry�pbuff�buf�sizer+ Zpl_sizer r r r3 ` s$ zentry_handler.loadc C s t j|�}| j||�S )zD Same as load(), but takes a string as an argument. )�io�BytesIOr3 )r �sr5 r r r r �loads� s zentry_handler.loadsc C sx xr|D ]j}|j dd�}| j� }tj||� |j� }t|�}|jtjd|�� |j|� | j r|r| j j |||� qW dS )z� Convert criu image entries from dict(json) format to binary. Takes a list of entries and a file-like object to write entries in binary format to. r, Nr! ) �popr r �dict2pb�SerializeToStringr. r r/ �packr �dump)r r7 r r8 r, r9 �pb_strr; r r r rD � s zentry_handler.dumpc C s t jd�}| j||� |j� S )zf Same as dump(), but doesn't take file-like object and just returns a string. r" )r< r= rD r- )r r7 r r r r �dumps� s zentry_handler.dumpsc C sH d}x>|j d�}t|�dkr P tjd|�\}|j|d� |d7 }qW |S )zI Counts the number of top-level object in the image file r r r! r )r- r. r/ r0 �seek)r r r7 r: r; r r r �count� s zentry_handler.count)N)FF)F) r r r �__doc__r r3 r? rD rF rH r r r r r S s + r c @ s< e Zd ZdZddd�Zddd�Zdd� Zd d � Zd d � Zd S )�pagemap_handlerz� Special entry handler for pagemap.img, which is unique in a way that it has a header of pagemap_head type followed by entries of pagemap_entry type. Fc C sf g }t j� }xT|jd�}t|�dkr&P tjd|�\}|j|j|�� |jtj||�� t j � }qW |S )Nr r r! ) r � pagemap_headr- r. r/ r0 r1 r4 r � pagemap_entry)r r r5 r6 r7 r9 r: r; r r r r3 � s zpagemap_handler.loadc C s t j|�}| j||�S )N)r< r= r3 )r r> r5 r r r r r? � s zpagemap_handler.loadsc C sZ t j� }xL|D ]D}tj||� |j� }t|�}|jtjd|�� |j|� t j � }qW d S )Nr! ) r rK r rA rB r. r r/ rC rL )r r7 r r9 �itemrE r; r r r rD � s zpagemap_handler.dumpc C s t jd�}| j||� |j� S )Nr" )r< r= rD r- )r r7 r r r r rF � s zpagemap_handler.dumpsc C s t d �j|�d S )Nr )r rH )r r r r r rH � s zpagemap_handler.countN)FF)F) r r r rI r3 r? rD rF rH r r r r rJ � s rJ c @ s0 e Zd Zd dd�Zd dd�Zdd� Zdd � Zd S ) �ghost_file_handlerFc C s g }t j� }|jd�}tjd|�\}|j|j|�� tj||�}|jr�|j|� x�t j � } |jd�}t |�dkrrP tjd|�\}| j|j|�� tj| |�} |r�|j | j t j � n tj|j| j ��jd�| d< |j| � qRW n6|r�|j dt j� ntj|j� �jd�|d< |j|� |S )Nr r! r zutf-8r, )r �ghost_file_entryr- r/ r0 r1 r �chunksr4 �ghost_chunk_entryr. rG �os�SEEK_CURr r �decode�SEEK_END) r r r5 r6 r7 Zgfr: r; Zg_entry�gcr8 r r r r3 � s2 zghost_file_handler.loadc C s t j|�}| j||�S )N)r< r= r3 )r r> r5 r r r r r? s zghost_file_handler.loadsc C s� t j� }|jd�}tj||� |j� }t|�}|jtj d|�� |j|� |j r�xj|D ]R}t j � }tj||� |j� }t|�}|jtj d|�� |j|� t ||d � qVW nt ||d � d S )Nr r! r, ) r rO r@ r rA rB r. r r/ rC rP rQ r )r r7 r r9 rM rE r; r r r rD s" zghost_file_handler.dumpc C s t jd�}| j||� |j� S )Nr" )r< r= rD r- )r r7 r r r r rF - s zghost_file_handler.dumpsN)FF)F)r r r r3 r? rD rF r r r r rN � s ! rN c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�pipes_data_extra_handlerc C s |j }|j|�}tj|�jd�S )Nzutf-8)�bytesr- r r rT )r r �ploadr; r r r r r3 : s z pipes_data_extra_handler.loadc C s t |�}|j|� d S )N)r r )r r, r rY r r r r rD ? s z pipes_data_extra_handler.dumpc C s |j |jtj� |jS )N)rG rX rR rS )r r rY r r r r2 C s z pipes_data_extra_handler.skipN)r r r r3 rD r2 r r r r rW 9 s rW c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�sk_queues_extra_handlerc C s |j }|j|�}tj|�jd�S )Nzutf-8)�lengthr- r r rT )r r rY r; r r r r r3 I s z sk_queues_extra_handler.loadc C s t |�}|j|� d S )N)r r )r r, r �_unusedr r r r rD N s z sk_queues_extra_handler.dumpc C s |j |jtj� |jS )N)rG r[ rR rS )r r rY r r r r2 R s z sk_queues_extra_handler.skipN)r r r r3 rD r2 r r r r rZ H s rZ c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�tcp_stream_extra_handlerc C sH i }|j |j�}|j |j�}tj|�jd�|d< tj|�jd�|d< |S )Nzutf-8�inq�outq)r- �inq_len�outq_lenr r rT )r r r9 �dr^ r_ r r r r3 X s z tcp_stream_extra_handler.loadc C s0 t |d �}t |d �}|j|� |j|� d S )Nr^ r_ )r r )r r, r r\ r^ r_ r r r rD c s z tcp_stream_extra_handler.dumpc C s |j dtj� |j|j S )Nr )rG rR rU r` ra )r r r9 r r r r2 j s z tcp_stream_extra_handler.skipN)r r r r3 rD r2 r r r r r] W s r] c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�bpfmap_data_extra_handlerc C s& |j |j }|j|�}tj|�jd�S )Nzutf-8)Z keys_bytesZ values_bytesr- r r rT )r r rY r; r r r r r3 p s z bpfmap_data_extra_handler.loadc C s t j|�}|j|� d S )N)r r r )r r, r rY r r r r rD u s z bpfmap_data_extra_handler.dumpc C s |j |jtj� |jS )N)rG rX rR rS )r r rY r r r r2 y s z bpfmap_data_extra_handler.skipN)r r r r3 rD r2 r r r r rc o s rc c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) �ipc_sem_set_handlerc C sP t j |�}t|d }t|t�}| j� }|j|j|�� |j|| d� |j� S )N�nsemsr ) r � sizeof_u16r � sizeof_u64�_get_sem_arrayZ frombytesr- rG �tolist)r r r9 r8 r; �roundedr> r r r r3 s zipc_sem_set_handler.loadc C sn t j |�}t|d }t|t�}| j� }|j|� t|�|d krJtd��|j|j � � |jd|| � d S )Nre z Number of semaphores mismatch� ) r rf r rg rh �fromlistr. � Exceptionr �tobytes)r r, r r9 r8 r; rj r> r r r rD � s zipc_sem_set_handler.dumpc C s. t j |�}t|d }|jt|t�tj� |S )Nre )r rf rG r rg rR rS )r r r9 r8 r; r r r r2 � s zipc_sem_set_handler.skipc C s t j d�}|jtkr td��|S )N�HzArray size mismatch)�array�itemsizerf rm )r r> r r r rh � s z"ipc_sem_set_handler._get_sem_arrayN)r r r r3 rD r2 rh r r r r rd ~ s rd c @ s. e Zd Zdd� Zdd� Zdd� Zd dd �Zd S ) �ipc_msg_queue_handlerc C s | j ||�\}}|S )N)�_read_messages)r r r9 �messages�_r r r r3 � s zipc_msg_queue_handler.loadc C s� x�t dt|�d�D ]�}tj� }tj|| |� |j� }t|�}|jtj d|�� |j|� t |j t �}t ||d �} |j| d |j � � |jd||j � qW d S )Nr r r! r rk )�ranger. r �ipc_msgr rA rB r r/ rC r �msizerg r ) r r, r r9 r! �msgZmsg_strr; rj r r r r rD � s zipc_msg_queue_handler.dumpc C s | j ||dd�\}}|S )NT)� skip_data)rs )r r r9 ru �pl_lenr r r r2 � s zipc_msg_queue_handler.skipFc C s� t j |�}g }d}x�td|d �D ]�}|jd�}t|�dkr>P tjd|�\} tj� } | j|j| �� t | j t �} || | j 7 }|r�|j | t j� q"|j| j �} |j | | j d� |jt j | �� |jtj| �jd�� q"W ||fS )Nr Zqnumr r! r zutf-8)r rv r- r. r/ r0 r rw r1 r rx rg rG rR rS r4 r r rT ) r r r9 rz r8 rt r{ r r: r; ry rj r r r r rs � s& z$ipc_msg_queue_handler._read_messagesN)F)r r r r3 rD r2 rs r r r r rr � s rr c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�ipc_shm_handlerc C sF t j |�}|d }|j|�}t|t�}|j|| d� tj|�jd�S )Nr; r zutf-8)r r- r � sizeof_u32rG r r rT )r r r9 r8 r; r rj r r r r3 � s zipc_shm_handler.loadc C sN t j |�}|d }tj|�}t|t�}|j|d |� � |jd|| � d S )Nr; rk )r r r r r} r )r r, r r9 r8 r; r rj r r r rD � s zipc_shm_handler.dumpc C s. t j |�}|d }t|t�}|j|tj� |S )Nr; )r r r} rG rR rS )r r r9 r8 r; rj r r r r2 � s zipc_shm_handler.skipN)r r r r3 rD r2 r r r r r| � s r| )B� INVENTORYZCOREZIDSZCREDSZUTSNSZTIMENSZPIDNSZIPC_VARZFSZ GHOST_FILEZMMZCGROUPZ TCP_STREAM�STATSZPAGEMAPZPSTREEZ REG_FILESZNS_FILESZ EVENTFD_FILEZEVENTPOLL_FILEZ EVENTPOLL_TFDZSIGNALFDZTIMERFDZ INOTIFY_FILEZ INOTIFY_WDZ FANOTIFY_FILEZ FANOTIFY_MARKZVMASZPIPESZFIFOZSIGACTZ NETLINK_SKZ REMAP_FPATHZMNTSZ TTY_FILESZTTY_INFOZTTY_DATAZRLIMITZTUNFILEZ EXT_FILES� IRMAP_CACHEZ FILE_LOCKSZFDINFOZUNIXSKZINETSKZPACKETSKZITIMERSZ POSIX_TIMERSZNETDEVZ PIPES_DATAZ FIFO_DATAZ SK_QUEUESZ IPCNS_SHMZ IPCNS_SEMZ IPCNS_MSGZNETNSZUSERNSZSECCOMPZAUTOFSZFILESZCPUINFOZ MEMFD_FILEZ MEMFD_INODEZ BPFMAP_FILEZ BPFMAP_DATAZAPPARMORc C s� t jd| jd��\}|tjd tjd fkr@t jd| jd��\}ytj| }W n tk rj t|��Y nX y t| }W n tk r� td| ��Y nX ||fS )Nr! r � IMG_COMMON� IMG_SERVICEz&No handler found for image with magic ) r/ r0 r- r �by_nameZby_valrm r �handlers)r Z img_magic�m�handlerr r r � __rhandler1 s r� Fc C s. i }t | �\}}||d< |j| ||�|d< |S )z� Convert criu image from binary format to dict(json). Takes a file-like object to read criu image from. Returns criu image in dict(json) format. r r7 )r� r3 )r r5 r6 Zimager� r� r r r r3 E s r3 c C s* i }t | �\}}||d< |j| �|d< |S )Nr rH )r� rH )r �resr� r� r r r �infoU s r� c C s t j| �}t||�S )z- Same as load(), but takes a string. )r< r= r3 )r> r5 r r r r r? ` s r? c C s� | d }t j| d }|dkrX|d kr@|jtjdt jd �� n|jtjdt jd �� |jtjd|�� y t| }W n tk r� td��Y nX |j| d |� d S ) z� Convert criu image from dict(json) format to binary. Takes an image in dict(json) format and file-like object to write to. r r~ r r� r! r� r� z*No handler found for image with such magicr7 N)r r� )r r� r r/ rC r� rm rD )�imgr r� Z magic_valr� r r r rD h s rD c C s t jd�}t| |� |j� S )zK Same as dump(), but takes only an image and returns a string. � )r< r= rD �getvalue)r� r r r r rF � s rF )FF)F)fr< r r/ rR rp r" r r r �dirZ encodestringr Z decodestringr rf r} rg r rm r r r r rJ rN rW rZ r] rc rd rr r| Zinventory_entryZ core_entryZtask_kobj_ids_entryZ creds_entryZ utsns_entryZ timens_entryZ pidns_entryZ ipc_var_entryZfs_entryZmm_entryZ cgroup_entryZtcp_stream_entryZ stats_entryZ pstree_entryZreg_file_entryZ ns_file_entryZeventfd_file_entryZeventpoll_file_entryZeventpoll_tfd_entryZsignalfd_entryZ timerfd_entryZinotify_file_entryZinotify_wd_entryZfanotify_file_entryZfanotify_mark_entryZ vma_entryZ pipe_entryZ fifo_entryZsa_entryZnetlink_sk_entryZremap_file_path_entryZ mnt_entryZtty_file_entryZtty_info_entryZtty_data_entryZ rlimit_entryZ tunfile_entryZext_file_entryZirmap_cache_entryZfile_lock_entryZ fdinfo_entryZ unix_sk_entryZ inet_sk_entryZpacket_sock_entryZ itimer_entryZposix_timer_entryZnet_device_entryZpipe_data_entryZsk_packet_entryZ ipc_shm_entryZ ipc_sem_entryZ ipc_msg_entryZ netns_entryZ userns_entryZ seccomp_entryZ autofs_entryZ file_entryZ cpuinfo_entryZmemfd_file_entryZmemfd_inode_entryZbpfmap_file_entryZbpfmap_data_entryZapparmor_entryr� r� r3 r� r? rD rF r r r r �<module>( s� o0G"/
Save File
Cancel