File manager

�
k�x]c@s�dd
lZdd
l
Z
dd
lZdd
lZdd
lZdd
ljZddd��YZdef
d��YZ	def
d��YZ
def
d	��YZd
ef
d��YZ
de
f
d
��YZde
f
d��YZddd��YZd
S(i����Nt
_criu_commc
BsDeZ
dZd
ZdZdZdZdZdZ	d�Z
d�ZRS(s/
    Base class for communication classes.
    ii
ic
Csd
S(s�
        Connect to criu and return socket object.
        daemon -- is for whether or not criu should daemonize if executing criu from binary(comm_bin).
        N((tselftdaemon((s//usr/lib/python2.7/site-packages/pycriu/criu.pytconnectsc


Csd
S(s'
        Disconnect from criu.
        N((
R
((s//usr/lib/python2.7/site-packages/pycriu/criu.pyt
disconnectsN(t__name__t
__module__t__doc__tCOMM_SKtCOMM_FDtCOMM_BINtNonet	comm_typetcommtskRR(((s//usr/lib/python2.7/site-packages/pycriu/criu.pyRs





	t
_criu_comm_skc
Bs)eZ
dZd
�Zd�Zd�ZRS(s.
    Communication class for unix socket.
    cCs|j|_
|
|_dS(
N(RRR
(R
tsk_path((s//usr/lib/python2.7/site-packages/pycriu/criu.pyt__init__*s

cCs5tjtj
tj�|_|jj|j�

|jS(
N(tsockettAF_UNIXtSOCK_SEQPACKETRRR
(R
R((s//usr/lib/python2.7/site-packages/pycriu/criu.pyR.s

c


Cs|jj
�
dS(
N(Rtclose(
R
((s//usr/lib/python2.7/site-packages/pycriu/criu.pyR4s
(RRRRRR(((s//usr/lib/python2.7/site-packages/pycriu/criu.pyR%s		t
_criu_comm_fdc
Bs)eZ
dZd
�Zd�Zd�ZRS(s2
    Communication class for file descriptor.
    cCs|j|_
|
|_dS(
N(R	RR
(R
tfd((s//usr/lib/python2.7/site-packages/pycriu/criu.pyR=s

cCs(tj
|jtjtj�|_|jS(
N(RtfromfdR
RRR(R
R((s//usr/lib/python2.7/site-packages/pycriu/criu.pyRAs

c


Cs|jj
�
dS(
N(RR(
R
((s//usr/lib/python2.7/site-packages/pycriu/criu.pyRGs
(RRRRRR(((s//usr/lib/python2.7/site-packages/pycriu/criu.pyR8s		t_criu_comm_binc
Bs)eZ
dZd
�Zd�Zd�ZRS(s)
    Communication class for binary.
    cCs+|j|_
|
|_d|_d|_dS(
N(R
RR
RtswrkR(R
tbin_path((s//usr/lib/python2.7/site-packages/pycriu/criu.pyRPs

	
	
csm
tj
tjtj��tj�d
tj�}tj�d
tj|tjB�
tj�dtj�}tj�dtj|tj@�
|
�
_t	j
�}|dkr
��
fd�}|
r
t	j
�}|dkr�t	j�
|�
q	
t	jd�

q%
|�
n|
r%
t	j
|d�
n�dj�
tjd�d
jd�
�d�
_�d
�
_�
jS(Ni
ics�tj
d
�

tj
d�

tj
d�

�d
jtjdtj���

tj�
j�
jdd�d
j�g�
tj	d�

dS(Nii
it
iRs%d(
tosRtsendtstructtpacktgetpidtexecvR
tfilenot_exit((tcssR
(s//usr/lib/python2.7/site-packages/pycriu/criu.pyt	exec_criuds





#

	

Ri(Rt
socketpairRRtfcntltF_GETFDtF_SETFDt
FD_CLOEXECRRtforktsetsidR$twaitpidRRtunpacktrecvRR(R
Rtflagst
pR&((R%R
s//usr/lib/python2.7/site-packages/pycriu/criu.pyRVs,

!

"	





&

c

Cs0|jj
�
|js,tj|jd
�
ndS(Ni(RRRRR.R(
R
((s//usr/lib/python2.7/site-packages/pycriu/criu.pyR�s


	
(RRRRRR(((s//usr/lib/python2.7/site-packages/pycriu/criu.pyRKs		0t
CRIUExceptionc
Bs#eZ
dZdZdZd
�ZRS(s?
    Exception class for handling and storing criu errors.
    c


Cs|jS(
N(
t_str(
R
((s//usr/lib/python2.7/site-packages/pycriu/criu.pyt__str__�s
N(RRRRttypR4R5(((s//usr/lib/python2.7/site-packages/pycriu/criu.pyR3�s

tCRIUExceptionInternalc
BseZ
dZd
�ZRS(sC
    Exception class for handling and storing internal errors.
    cCs/|
|_d
t
jj|j�
|f|_dS(Ns!%s failed with internal error: %s(R6trpct
criu_req_typetNameR4(R
R6t
s((s//usr/lib/python2.7/site-packages/pycriu/criu.pyR�s
	

(RRRR(((s//usr/lib/python2.7/site-packages/pycriu/criu.pyR7�stCRIUExceptionExternalc
Bs eZ
dZd
�Zd�ZRS(sC
    Exception class for handling and storing criu RPC errors.
    cCs.|
|_||_
||_|j�|_dS(
N(R6tresp_typterrnot_gen_error_strR4(R
treq_typR=R>((s//usr/lib/python2.7/site-packages/pycriu/criu.pyR�s
	
	
	
c
Cs�d
tj
j|j�
f
}
|j|jkrE|
d|jf
7}
n|
d|jf
7}
|jtjkrx|
d7}
n|jtjkr�|jtjkr�|
d7}
q�n|jtj	kr�|jtj
kr�|
d7}
q�n|
d7}
|
S(Ns%s failed: sUnexpected response type %d: sError(%d): sBad optionssNo process with such pids)Process with requested pid already existstUnknown(R8R9R:R6R=R>tEBADRQCtDUMPtESRCHtRESTOREtEEXIST(R
R;((s//usr/lib/python2.7/site-packages/pycriu/criu.pyR?�s








(RRRRR?(((s//usr/lib/python2.7/site-packages/pycriu/criu.pyR<�s	tcriuc
Bs}eZ
dZdZdZd
�Zd�Zd�Zd�Z	d�Z
d�Zd�Zd�Z
d	�Zd
�Zd�ZRS(
s 
    Call criu through RPC.
    c

Cs)|jd
�

t
j�|_d|_dS(NRG(t
use_binaryR8t	criu_optstoptsRR(
R
((s//usr/lib/python2.7/site-packages/pycriu/criu.pyR�s



cCst|
�
|_
d
S(sZ
        Access criu using unix socket which that belongs to criu service daemon.
        N(Rt_comm(R
tsk_name((s//usr/lib/python2.7/site-packages/pycriu/criu.pytuse_sk�scCst|
�
|_
d
S(s0
        Access criu using provided fd.
        N(RRK(R
R((s//usr/lib/python2.7/site-packages/pycriu/criu.pytuse_fd�scCst|
�
|_
d
S(sO
        Access criu by execing it using provided path to criu binary.
        N(RRK(R
tbin_name((s//usr/lib/python2.7/site-packages/pycriu/criu.pyRH�scCs
t}|
j
tjkr4|
jjd
�
r4t}ny�|jsU|jj	|�
}n	|j}|
j
rs||_n|j|
j��

|j
t|j
dtjtjB��
�
}|
j
s�|jj�
ntj�}|j|�

Wn+tk
r

}
t|
j
t|�
��
n
X|S(sA
        As simple as send request and receive response.
        tpidi
(tFalsettypeR8RCRJtHasFieldtTrueRRKRt	keep_openRtSerializeToStringR0tlenRt	MSG_TRUNCtMSG_PEEKRt	criu_resptParseFromStringt	ExceptionR7tstr(R
treqRR;tbuftrespt
e((s//usr/lib/python2.7/site-packages/pycriu/criu.pyt_send_req_and_recv_resp�s$
%
	
	
		
+	



c
CsRtj
�}
tj|
_|j|
�
}|jsNt|
j|j|j��
nd
S(sB
        Checks whether the kernel support is up-to-date.
        N(R8tcriu_reqtCHECKRRRbtsuccessR<tcr_errno(R
R^R`((s//usr/lib/python2.7/site-packages/pycriu/criu.pytcheck
s

	
c
Cshtj
�}
tj|
_|
jj|j�

|j|
�
}|jsat|
j|j|j	��
n|j
S(
sC
        Checkpoint a process/tree identified by opts.pid.
        (R8RcRCRRRJt	MergeFromRbReR<Rftdump(R
R^R`((s//usr/lib/python2.7/site-packages/pycriu/criu.pyRi
s

	
c
Cshtj
�}
tj|
_|
jj|j�

|j|
�
}|jsat|
j|j|j	��
n|j
S(
sC
        Checkpoint a process/tree identified by opts.pid.
        (R8RctPRE_DUMPRRRJRhRbReR<RfRi(R
R^R`((s//usr/lib/python2.7/site-packages/pycriu/criu.pytpre_dump"
s

	
c
Cshtj
�}
tj|
_|
jj|j�

|j|
�
}|jsat|
j|j|j	��
n|j
S(
s)
        Restore a process/tree.
        (R8RcRERRRJRhRbReR<Rftrestore(R
R^R`((s//usr/lib/python2.7/site-packages/pycriu/criu.pyRl1
s

	
c
Csqtj
�}
tj|
_|
jj|j�

t|
_|j|
�
}|j	sjt
|
j|j|j��
n|jS(
N(
R8RctPAGE_SERVER_CHLDRRRJRhRTRURbReR<Rftps(R
R^R`((s//usr/lib/python2.7/site-packages/pycriu/criu.pytpage_server_chld@
s



		
cCs^tj
�}tj|_|
|_|j|�
}|jsWt|j|j|j��
n|j	S(
N(
R8RctWAIT_PIDRRRPRbReR<Rftstatus(R
RPR^R`((s//usr/lib/python2.7/site-packages/pycriu/criu.pytwait_pidM
s


		
N(RRRRRJRKRRMRNRHRbRgRiRkRlRoRr(((s//usr/lib/python2.7/site-packages/pycriu/criu.pyRG�s
					#					
(((RR>R(RRtpycriu.rpc_pb2trpc_pb2R8RRRRR\R3R7R<RG(((s//usr/lib/python2.7/site-packages/pycriu/criu.pyt<module>s



A#