3
i" @ s d dl Z d dlZd dlZd dlZd dlZd dljZdZG dd dZ G dd de Z
G dd de Z G d d
d
e Z G d d d e
ZG d
d deZG dd deZG dd dZdS ) Nz./criu_service.socketc @ s8 e Zd ZdZdZdZdZdZdZdZ dd Z
dd Z dS )
_criu_commz/
Base class for communication classes.
r Nc C s dS )z
Connect to criu and return socket object.
daemon -- is for whether or not criu should daemonize if executing criu from binary(comm_bin).
N )selfdaemonr r /usr/lib/python3.6/criu.pyconnect s z_criu_comm.connectc C s dS )z'
Disconnect from criu.
Nr )r r r r
disconnect s z_criu_comm.disconnect) __name__
__module__ __qualname____doc__COMM_SKCOMM_FDCOMM_BIN comm_typecommskr r
r r r r r
s r c @ s( e Zd ZdZdd Zdd Zdd ZdS )
_criu_comm_skz.
Communication class for unix socket.
c C s | j | _|| _d S )N)r r r )r Zsk_pathr r r __init__+ s z_criu_comm_sk.__init__c C sb t j t jt j| _y| jj| j W n6 tk
r@ tdY n tk
rZ tdY nX | jS )NzSocket file not found.zService not running.)socketAF_UNIXSOCK_SEQPACKETr r r FileNotFoundErrorConnectionRefusedError)r r r r r r / s z_criu_comm_sk.connectc C s | j j d S )N)r close)r r r r r
< s z_criu_comm_sk.disconnectN)r r r
r r r r
r r r r r & s
r c @ s( e Zd ZdZdd Zdd Zdd ZdS )
_criu_comm_fdz2
Communication class for file descriptor.
c C s | j | _|| _d S )N)r r r )r fdr r r r E s z_criu_comm_fd.__init__c C s t j| jt jt j| _| jS )N)r Zfromfdr r r r )r r r r r r I s
z_criu_comm_fd.connectc C s | j j d S )N)r r )r r r r r
O s z_criu_comm_fd.disconnectN)r r r
r r r r
r r r r r @ s r c @ s( e Zd ZdZdd Zdd Zdd ZdS ) _criu_comm_binz)
Communication class for binary.
c C s | j | _|| _d | _d | _d S )N)r r r swrkr )r Zbin_pathr r r r X s z_criu_comm_bin.__init__c s t jt jt j tj d tj}tj d tj|tjB tj d tj}tj d tj|tj @ |_t j
}|dkrƇ fdd}|rt j
}|dkrt j | qt j d q| n|rt j
|d d j tjd d jdd _ d _jS )Nr r c sh t jd t jd t jd d jtjdt j t jjjdd d j g t j d d S )Nr r r ir z%d)
osr sendstructpackgetpidexecvpr fileno_exitr )cssr r r exec_criul s
z)_criu_comm_bin.connect..exec_criur! )r Z
socketpairr r fcntlZF_GETFDZF_SETFDZ
FD_CLOEXECr r" forksetsidr) waitpidr r$ unpackrecvr r )r r flagspr+ r )r* r r r ^ s,
z_criu_comm_bin.connectc C s" | j j | js tj| jd d S )Nr )r r r r" r0 r )r r r r r
s
z_criu_comm_bin.disconnectN)r r r
r r r r
r r r r r S s 0r c @ s e Zd ZdZdZdZdd ZdS )
CRIUExceptionz?
Exception class for handling and storing criu errors.
Nc C s | j S )N)_str)r r r r __str__ s zCRIUException.__str__)r r r
r typr6 r7 r r r r r5 s r5 c @ s e Zd ZdZdd ZdS )CRIUExceptionInternalzC
Exception class for handling and storing internal errors.
c C s" || _ dtjj| j |f | _d S )Nz!%s failed with internal error: %s)r8 rpc
criu_req_typeNamer6 )r r8 sr r r r s z CRIUExceptionInternal.__init__N)r r r
r r r r r r r9 s r9 c @ s e Zd ZdZdd Zdd ZdS )CRIUExceptionExternalzC
Exception class for handling and storing criu RPC errors.
c C s || _ || _|| _| j | _d S )N)r8 resp_typerrno_gen_error_strr6 )r Zreq_typr? r@ r r r r s z CRIUExceptionExternal.__init__c C s dt jj| jf }| j| jkr0|d| jf 7 }|d| jf 7 }| jtjkrV|d7 }nL| jt jkrx| jtjkrx|d7 }n*| jt j kr| jtj
kr|d7 }n|d7 }|S )Nz %s failed: z Unexpected response type %d: z Error(%d): z Bad optionszNo process with such pidz)Process with requested pid already existsZUnknown) r: r; r< r8 r? r@ ZEBADRQCDUMPZESRCHRESTOREZEEXIST)r r= r r r rA s
z$CRIUExceptionExternal._gen_error_strN)r r r
r r rA r r r r r> s r> c @ st e Zd ZdZdZdZdd ZefddZdd Z d d
Z
d d Z d
d Z dd Z
dd Zdd Zdd Zdd ZdS )criuz
Call criu through RPC.
Nc C s" | j d tjdd| _d | _d S )NrD r )Z
images_dir_fd)
use_binaryr: Z criu_optsoptsr )r r r r r s
z
criu.__init__c C s t || _dS )zZ
Access criu using unix socket which that belongs to criu service daemon.
N)r _comm)r Zsk_namer r r use_sk s z criu.use_skc C s t || _dS )z0
Access criu using provided fd.
N)r rH )r r r r r use_fd s z criu.use_fdc C s t || _dS )zO
Access criu by execing it using provided path to criu binary.
N)r rH )r Zbin_namer r r rF s zcriu.use_binaryc C s d}|j tjkr"|jjd r"d}yx| js8| jj|}n| j}|jrJ|| _|j |j
|j t |j dt
jt
jB }|js| jj tj }|j| W n2 tk
r } zt|j t|W Y dd}~X nX |S )zA
As simple as send request and receive response.
FpidTr N)typer: rB rG ZHasFieldr rH r keep_openr# ZSerializeToStringr2 lenr Z MSG_TRUNCZMSG_PEEKr
Z criu_respZParseFromString Exceptionr9 str)r reqr r= Zbufresper r r _send_req_and_recv_resp s$
"z criu._send_req_and_recv_respc C sD t j }t j|_|jj| j | j|}|js@t|j|j|j dS )zB
Checks whether the kernel support is up-to-date.
N)
r: criu_reqZCHECKrL rG MergeFromrT successr> cr_errno)r rQ rR r r r check s
z
criu.checkc C sF t j }t j|_|jj| j | j|}|js@t|j|j|j |j
S )zC
Checkpoint a process/tree identified by opts.pid.
) r: rU rB rL rG rV rT rW r> rX dump)r rQ rR r r r rZ s
z criu.dumpc C sF t j }t j|_|jj| j | j|}|js@t|j|j|j |j
S )zC
Checkpoint a process/tree identified by opts.pid.
) r: rU ZPRE_DUMPrL rG rV rT rW r> rX rZ )r rQ rR r r r pre_dump+ s
z
criu.pre_dumpc C sF t j }t j|_|jj| j | j|}|js@t|j|j|j |j
S )z)
Restore a process/tree.
) r: rU rC rL rG rV rT rW r> rX restore)r rQ rR r r r r\ : s
z criu.restorec C sL t j }t j|_|jj| j d|_| j|}|jsFt |j|j|j
|j S )NT) r: rU ZPAGE_SERVER_CHLDrL rG rV rM rT rW r> rX Zps)r rQ rR r r r page_server_chldI s
zcriu.page_server_chldc C s> t j }t j|_||_| j|}|js8t|j|j|j|j S )N)
r: rU ZWAIT_PIDrL rK rT rW r> rX Zstatus)r rK rQ rR r r r wait_pidV s
z
criu.wait_pid)r r r
r rG rH r CR_DEFAULT_SERVICE_ADDRESSrI rJ rF rT rY rZ r[ r\ r] r^ r r r r rD s #
rD )r r@ r- r" r$ Zpycriu.rpc_pb2Zrpc_pb2r: r_ r r r r rO r5 r9 r> rD r r r r s
A "