Current File : //usr/lib/python2.7/site-packages/pycriu/images/images.pyc
�
k�x]c @ s� d d l Z d d l Z d d l Z d d l Z d d l Z d d l m Z d d l m Z d d l m Z d e e � k r� e j
e _ e j e _
n d Z d Z d Z d
� Z d e f d � � YZ d
d` d � � YZ d da d � � YZ d db d � � YZ d dc d � � YZ d dd d � � YZ d de d � � YZ d df d � � YZ d dg d � � YZ d dh d � � YZ i; e e j � d 6e e j � d 6e e j � d! 6e e j � d" 6e e j! � d# 6e e j" � d$ 6e e j# � d% 6e � d& 6e e j$ � d' 6e e j% � d( 6e e j& e � � d) 6e e j' � d* 6e � d+ 6e e j( � d, 6e e j) � d- 6e e j* � d. 6e e j+ � d/ 6e e j, � d0 6e e j- � d1 6e e j. � d2 6e e j/ � d3 6e e j0 � d4 6e e j1 � d5 6e e j2 � d6 6e e j3 � d7 6e e j4 � d8 6e e j5 � d9 6e e j6 � d: 6e e j7 � d; 6e e j8 � d< 6e e j9 � d= 6e e j: � d> 6e e j; � d? 6e e j< � d@ 6e e j= � dA 6e e j> � dB 6e e j? � dC 6e e j@ � dD 6e e jA � dE 6e e jB � dF 6e e jC � dG 6e e jD � dH 6e e jE � dI 6e e jF � dJ 6e e jG � dK 6e e jH � dL 6e e jI � dM 6e e jJ e � � dN 6e e jJ e � � dO 6e e jK e � � dP 6e e jL e � � dQ 6e e jM e � � dR 6e e jN e � � dS 6e e jO � dT 6e e jP � dU 6e e jQ � dV 6e e jR � dW 6e e jS � dX 6e e jT � dY 6ZU dZ � ZV eW eW d[ � ZX d\ � ZY eW d] � ZZ d^ � Z[ d_ � Z\ d S(i i����Ni ( t magic( t pb( t pb2dictt encodebytesi i i c C s | d | d Bd S( Ni ( ( t xt y( ( s8 /usr/lib/python2.7/site-packages/pycriu/images/images.pyt round_up> s t MagicExceptionc B s e Z d � Z RS( c C s
| | _ d S( N( R ( t selfR ( ( s8 /usr/lib/python2.7/site-packages/pycriu/images/images.pyt __init__C s ( t __name__t
__module__R ( ( ( s8 /usr/lib/python2.7/site-packages/pycriu/images/images.pyR B s t
entry_handlerc B sP e Z d Z d d � Z e e d � Z e d � Z d � Z d � Z d � Z
RS( sp
Generic class to handle loading/dumping criu images
entries from/to bin format to/from dict(json).
c C s | | _ | | _ d S( s=
Sets payload class and extra handler class.
N( t payloadt
extra_handler( R R
R ( ( s8 /usr/lib/python2.7/site-packages/pycriu/images/images.pyR O s c C s� g } x� t r� i } | j � } | j d � } | d k r@ Pn t j d | � \ } | j | j | � � t j | | � } | j r� | r� d � } | j j | | � }
d | |
� | d <q� | j j | | � | d <n | j
| � q W| S( s�
Convert criu image entries from binary format to dict(json).
Takes a file-like object and returnes a list with entries in
dict(json) format.
i t t ic S sx xm d d d d d d d d g D]M } | d k rb t | � | k rQ d
| | f Sd | | f Sn | d :} q Wd | S(
NR t Kt Mt Gt Tt Pt Et Zg �@s %d%sBs %.1f%sBs %.1fYB( t int( t numt unit( ( s8 /usr/lib/python2.7/site-packages/pycriu/images/images.pyt human_readablen s % s ... <%s>t extra( t TrueR
t readt structt unpackt ParseFromStringR R t skipt loadt append( R t ft prettyt
no_payloadt entriest entryt pbufft buft sizeR t pl_size( ( s8 /usr/lib/python2.7/site-packages/pycriu/images/images.pyR# V s$
c C s t j | � } | j | | � S( sD
Same as load(), but takes a string as an argument.
( t iot BytesIOR# ( R t sR& R% ( ( s8 /usr/lib/python2.7/site-packages/pycriu/images/images.pyt loads� s c C s� x� | D]� } | j d d � } | j � } t j | | � | j � } t | � } | j t j d | � � | j | � | j
r | r | j
j | | | � q q Wd S( s�
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 R N( t popt NoneR
R t dict2pbt SerializeToStringt lent writeR t packR t dump( R R( R% R) R R* t pb_strR, ( ( s8 /usr/lib/python2.7/site-packages/pycriu/images/images.pyR9 � s
c C s) t j d � } | j | | � | j � S( sf
Same as dump(), but doesn't take file-like object and just
returns a string.
R ( R. R/ R9 R ( R R( R% ( ( s8 /usr/lib/python2.7/site-packages/pycriu/images/images.pyt dumps� s c C se d } xX t r` | j d � } | d k r. Pn t j d | � \ } | j | d � | d 7} q W| S( sI
Counts the number of top-level object in the image file
i i R R i ( R R R R t seek( R R% R( R+ R, ( ( s8 /usr/lib/python2.7/site-packages/pycriu/images/images.pyt count� s N( R
R t __doc__R3 R t FalseR# R1 R9 R; R= ( ( ( s8 /usr/lib/python2.7/site-packages/pycriu/images/images.pyR I s + t pagemap_handlerc B sD e Z d Z e e d � Z e d � Z d � Z d � Z d � Z RS( s�
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.
c C s� g } t j � } xy t r� | j d � } | d k r: Pn t j d | � \ } | j | j | � � | j t j | | � � t j � } q W| S( Ni R R (
R t pagemap_headR R R R R! R$ R t
pagemap_entry( R R% R& R' R( R* R+ R, ( ( s8 /usr/lib/python2.7/site-packages/pycriu/images/images.pyR# � s c C s t j | � } | j | | � S( N( R. R/ R# ( R R0 R& R% ( ( s8 /usr/lib/python2.7/site-packages/pycriu/images/images.pyR1 � s c C s{ t j � } xh | D]` } t j | | � | j � } t | � } | j t j d | � � | j | � t j � } q Wd S( NR (
R RA R R4 R5 R6 R7 R R8 RB ( R R( R% R* t itemR: R, ( ( s8 /usr/lib/python2.7/site-packages/pycriu/images/images.pyR9 � s
c C s) t j d � } | j | | � | j � S( NR ( R. R/ R9 R ( R R( R% ( ( s8 /usr/lib/python2.7/site-packages/pycriu/images/images.pyR; � s c C s t d � j | � d S( Ni ( R R3 R= ( R R% ( ( s8 /usr/lib/python2.7/site-packages/pycriu/images/images.pyR= � s ( R
R R> R? R# R1 R9 R; R= ( ( ( s8 /usr/lib/python2.7/site-packages/pycriu/images/images.pyR@ � s t ghost_file_handlerc B s5 e Z e e d � Z e d � Z d � Z d � Z RS( c C s} g } t j � } | j d � } t j d | � \ } | j | j | � � t j | | � } | j r7| j | � xt r3t j
� } | j d � } | d k r� Pn t j d | � \ } | j | j | � � t j | | � }
| r| j | j t
j � n t j | j | j � � |
d <| j |
� qw WnB | rS| j d t
j � n t j | j � � | d <| j | � | S( Ni R R R i ( R t ghost_file_entryR R R R! R t chunksR$ R t ghost_chunk_entryR< R6 t ost SEEK_CURt base64R t SEEK_END( R R% R& R' R( t gfR+ R, t g_entryt gcR) ( ( s8 /usr/lib/python2.7/site-packages/pycriu/images/images.pyR# � s2
c C s t j | � } | j | | � S( N( R. R/ R# ( R R0 R& R% ( ( s8 /usr/lib/python2.7/site-packages/pycriu/images/images.pyR1
s c C s t j � } | j d � } t j | | � | j � } t | � } | j t j d | � � | j | � | j
r� x� | D]z } t j � } t j | | � | j � } t | � } | j t j d | � � | j | � | j t j
| d � � qy Wn | j t j
| d � � d S( Ni R R ( R RE R2 R R4 R5 R6 R7 R R8 RF RG RJ t decodebytes( R R( R% R* RC R: R, ( ( s8 /usr/lib/python2.7/site-packages/pycriu/images/images.pyR9 s"
!c C s) t j d � } | j | | � | j � S( NR ( R. R/ R9 R ( R R( R% ( ( s8 /usr/lib/python2.7/site-packages/pycriu/images/images.pyR; # s ( R
R R? R# R1 R9 R; ( ( ( s8 /usr/lib/python2.7/site-packages/pycriu/images/images.pyRD � s ! t pipes_data_extra_handlerc B s# e Z d � Z d � Z d � Z RS( c C s% | j } | j | � } t j | � S( N( t bytesR RJ R ( R R% t ploadR, t data( ( s8 /usr/lib/python2.7/site-packages/pycriu/images/images.pyR# 0 s c C s t j | � } | j | � d S( N( RJ RO R7 ( R R R% RR RS ( ( s8 /usr/lib/python2.7/site-packages/pycriu/images/images.pyR9 5 s c C s | j | j t j � | j S( N( R< RQ RH RI ( R R% RR ( ( s8 /usr/lib/python2.7/site-packages/pycriu/images/images.pyR" 9 s ( R
R R# R9 R" ( ( ( s8 /usr/lib/python2.7/site-packages/pycriu/images/images.pyRP / s t sk_queues_extra_handlerc B s# e Z d � Z d � Z d � Z RS( c C s% | j } | j | � } t j | � S( N( t lengthR RJ R ( R R% RR R, RS ( ( s8 /usr/lib/python2.7/site-packages/pycriu/images/images.pyR# ? s c C s t j | � } | j | � d S( N( RJ RO R7 ( R R R% t _unusedRS ( ( s8 /usr/lib/python2.7/site-packages/pycriu/images/images.pyR9 D s c C s | j | j t j � | j S( N( R< RU RH RI ( R R% RR ( ( s8 /usr/lib/python2.7/site-packages/pycriu/images/images.pyR" H s ( R
R R# R9 R" ( ( ( s8 /usr/lib/python2.7/site-packages/pycriu/images/images.pyRT > s t tcp_stream_extra_handlerc B s# e Z d � Z d � Z d � Z RS( c C sT i } | j | j � } | j | j � } t j | � | d <t j | � | d <| S( Nt inqt outq( R t inq_lent outq_lenRJ R ( R R% R* t dRX RY ( ( s8 /usr/lib/python2.7/site-packages/pycriu/images/images.pyR# N s c C sD t j | d � } t j | d � } | j | � | j | � d S( NRX RY ( RJ RO R7 ( R R R% RV RX RY ( ( s8 /usr/lib/python2.7/site-packages/pycriu/images/images.pyR9 Y s
c C s! | j d t j � | j | j S( Ni ( R< RH RK RZ R[ ( R R% R* ( ( s8 /usr/lib/python2.7/site-packages/pycriu/images/images.pyR" ` s ( R
R R# R9 R" ( ( ( s8 /usr/lib/python2.7/site-packages/pycriu/images/images.pyRW M s t ipc_sem_set_handlerc B s# e Z d � Z d � Z d � Z RS( c C s� t j | � } t | d } t | t � } t j d � } | j t k rY t d � � n | j | j | � � | j | | d � | j
� S( Nt nsemst Hs Array size mismatchi ( R t
sizeof_u16R t
sizeof_u64t arrayt itemsizet Exceptiont
fromstringR R< t tolist( R R% R* R) R, t roundedR0 ( ( s8 /usr/lib/python2.7/site-packages/pycriu/images/images.pyR# f s c C s� t j | � } t | d } t | t � } t j d � } | j t k rY t d � � n | j | � t | � | d k r� t d � � n | j | j
� � | j d | | � d S( NR^ R_ s Array size mismatchs Number of semaphores mismatchs ( R R` R Ra Rb Rc Rd t fromlistR6 R7 t tostring( R R R% R* R) R, Rg R0 ( ( s8 /usr/lib/python2.7/site-packages/pycriu/images/images.pyR9 q s
c C s= t j | � } t | d } | j t | t � t j � | S( NR^ ( R R` R< R Ra RH RI ( R R% R* R) R, ( ( s8 /usr/lib/python2.7/site-packages/pycriu/images/images.pyR" ~ s ( R
R R# R9 R" ( ( ( s8 /usr/lib/python2.7/site-packages/pycriu/images/images.pyR] e s
t ipc_msg_queue_handlerc B s# e Z d � Z d � Z d � Z RS( c C s� t j | � } g } x� t d | d � D]� } | j d � } | d k rN Pn t j d | � \ } t j � } | j | j | � � t | j t
� } | j | j � }
| j | | j d � | j t j | � � | j t
j |
� � q) W| S( Ni t qnumi R R i ( R t rangeR R R R t ipc_msgR! R t msizeRa R< R$ RJ R ( R R% R* R) t messagesR R+ R, t msgRg RS ( ( s8 /usr/lib/python2.7/site-packages/pycriu/images/images.pyR# � s c C s� t j | � } x� t d t | � d � D]� } t j � } t j | | | � | j � } t | � } | j t j d | � � | j | � t
| j t � } t
j | | d � }
| j |
| j � | j d | | j � q( Wd S( Ni i R i s ( R Rl R6 R Rm R4 R5 R7 R R8 R Rn Ra RJ RO ( R R R% R* R) R Rp t msg_strR, Rg RS ( ( s8 /usr/lib/python2.7/site-packages/pycriu/images/images.pyR9 � s
c
C s� t j | � } d } x� t d | d � D]� } | j d � } | d k rN Pn t j d | � \ } t j � } | j | j | � � t | j t
� } | j | t j
� | | | j 7} q) W| S( Ni Rk i R R ( R Rl R R R R Rm R! R Rn Ra R< RH RI (
R R% R* R) t pl_lenR R+ R, Rp Rg ( ( s8 /usr/lib/python2.7/site-packages/pycriu/images/images.pyR" � s ( R
R R# R9 R" ( ( ( s8 /usr/lib/python2.7/site-packages/pycriu/images/images.pyRj � s t ipc_shm_handlerc B s# e Z d � Z d � Z d � Z RS( c C sX t j | � } | d } | j | � } t | t � } | j | | d � t j | � S( NR, i ( R R R t
sizeof_u32R< RJ R ( R R% R* R) R, RS Rg ( ( s8 /usr/lib/python2.7/site-packages/pycriu/images/images.pyR# � s
c C sa t j | � } | d } t j | � } t | t � } | j | | � | j d | | � d S( NR, s ( R RJ RO R Rt R7 ( R R R% R* R) R, RS Rg ( ( s8 /usr/lib/python2.7/site-packages/pycriu/images/images.pyR9 � s
c C s? t j | � } | d } t | t � } | j | t j � | S( NR, ( R R Rt R< RH RI ( R R% R* R) R, Rg ( ( s8 /usr/lib/python2.7/site-packages/pycriu/images/images.pyR" � s
( R
R R# R9 R" ( ( ( s8 /usr/lib/python2.7/site-packages/pycriu/images/images.pyRs � s t INVENTORYt COREt IDSt CREDSt UTSNSt IPC_VARt FSt
GHOST_FILEt MMt CGROUPt
TCP_STREAMt STATSt PAGEMAPt PSTREEt REG_FILESt NS_FILESt EVENTFD_FILEt EVENTPOLL_FILEt
EVENTPOLL_TFDt SIGNALFDt TIMERFDt INOTIFY_FILEt
INOTIFY_WDt
FANOTIFY_FILEt
FANOTIFY_MARKt VMASt PIPESt FIFOt SIGACTt
NETLINK_SKt REMAP_FPATHt MNTSt TTY_FILESt TTY_INFOt TTY_DATAt RLIMITt TUNFILEt EXT_FILESt IRMAP_CACHEt
FILE_LOCKSt FDINFOt UNIXSKt INETSKt PACKETSKt ITIMERSt POSIX_TIMERSt NETDEVt
PIPES_DATAt FIFO_DATAt SK_QUEUESt IPCNS_SHMt IPCNS_SEMt IPCNS_MSGt NETNSt USERNSt SECCOMPt AUTOFSt FILESt CPUINFOc C s� t j d | j d � � \ } | t j d t j d f k r_ t j d | j d � � \ } n y t j | } Wn t | � � n Xy t | } Wn t d | � � n X| | f S( NR i t
IMG_COMMONt IMG_SERVICEs& No handler found for image with magic ( R R R R t by_namet by_valR t handlersRd ( R% t img_magict mt handler( ( s8 /usr/lib/python2.7/site-packages/pycriu/images/images.pyt
__rhandler s
!c C s? i } t | � \ } } | | d <| j | | | � | d <| S( s�
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 R( ( R� R# ( R% R& R' t imageR� R� ( ( s8 /usr/lib/python2.7/site-packages/pycriu/images/images.pyR# $ s
c C s9 i } t | � \ } } | | d <| j | � | d <| S( NR R= ( R� R= ( R% t resR� R� ( ( s8 /usr/lib/python2.7/site-packages/pycriu/images/images.pyt info4 s
c C s t j | � } t | | � S( s-
Same as load(), but takes a string.
( R. R/ R# ( R0 R& R% ( ( s8 /usr/lib/python2.7/site-packages/pycriu/images/images.pyR1 ? s c C s� | d } t j | d } | d k ry | d k rV | j t j d t j d � � qy | j t j d t j d � � n | j t j d | � � y t | } Wn t d � � n X| j | d | � d
S( s�
Convert criu image from dict(json) format to binary.
Takes an image in dict(json) format and file-like
object to write to.
R Ru R� R� R R� R� s* No handler found for image with such magicR( N( R� R� ( R R� R7 R R8 R� Rd R9 ( t imgR% R� t magic_valR� ( ( s8 /usr/lib/python2.7/site-packages/pycriu/images/images.pyR9 G s
##c C s& t j d � } t | | � | j � S( sK
Same as dump(), but takes only an image and returns
a string.
R ( R. R/ R9 t getvalue( R� R% ( ( s8 /usr/lib/python2.7/site-packages/pycriu/images/images.pyR; b s
( ( ( ( ( ( ( ( ( (] R. RJ R RH Rb R R R R t dirt encodestringR t decodestringRO R` Rt Ra R Rd R R R@ RD RP RT RW R] Rj Rs t inventory_entryt
core_entryt task_kobj_ids_entryt creds_entryt utsns_entryt
ipc_var_entryt fs_entryt mm_entryt cgroup_entryt tcp_stream_entryt stats_entryt pstree_entryt reg_file_entryt
ns_file_entryt eventfd_file_entryt eventpoll_file_entryt eventpoll_tfd_entryt signalfd_entryt
timerfd_entryt inotify_file_entryt inotify_wd_entryt fanotify_file_entryt fanotify_mark_entryt vma_entryt
pipe_entryt
fifo_entryt sa_entryt netlink_sk_entryt remap_file_path_entryt mnt_entryt tty_file_entryt tty_info_entryt tty_data_entryt rlimit_entryt
tunfile_entryt ext_file_entryt irmap_cache_entryt file_lock_entryt fdinfo_entryt
unix_sk_entryt
inet_sk_entryt packet_sock_entryt itimer_entryt posix_timer_entryt net_device_entryt pipe_data_entryt sk_packet_entryt
ipc_shm_entryt
ipc_sem_entryt
ipc_msg_entryt netns_entryt userns_entryt
seccomp_entryt autofs_entryt
file_entryt
cpuinfo_entryR� R� R? R# R� R1 R9 R; ( ( ( s8 /usr/lib/python2.7/site-packages/pycriu/images/images.pyt <module>( s� o0G 1
Mini Shell