관리-도구
편집 파일: images.pyo
� 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