b !ddlZddlZddlZddlZddlZddlmZGddZGddeZ GddeZ Gdd eZ Gd d e Z Gd d e ZGdde ZGddZdS)Nc6eZdZdZdZdZdZdZdZdZ dZ dZ dS) _criu_commz/ Base class for communication classes. rNcdS)z Connect to criu and return socket object. daemon -- is for whether or not criu should daemonize if executing criu from binary(comm_bin). Nselfdaemons -/usr/lib/python3/dist-packages/pycriu/criu.pyconnectz_criu_comm.connects cdS)z' Disconnect from criu. Nrr s r disconnectz_criu_comm.disconnects  r) __name__ __module__ __qualname____doc__COMM_SKCOMM_FDCOMM_BIN comm_typecommskr rrrr rr s\GGHI D B        rrc$eZdZdZdZdZdZdS) _criu_comm_skz. Communication class for unix socket. c,|j|_||_dSN)rrr)r sk_paths r __init__z_criu_comm_sk.__init__*s rctjtjtj|_|j|j|jSr)socketAF_UNIXSOCK_SEQPACKETrr rr s r r z_criu_comm_sk.connect.s7-0EFF  """wrc8|jdSrrclosers r rz_criu_comm_sk.disconnect4  rNrrrrr!r rrrr rr%sK rrc$eZdZdZdZdZdZdS) _criu_comm_fdz2 Communication class for file descriptor. c,|j|_||_dSr)rrrr fds r r!z_criu_comm_fd.__init__=s rcxtj|jtjtj|_|jSr)r#fromfdrr$r%rr s r r z_criu_comm_fd.connectAs,- 6> & 577wrc8|jdSrr'rs r rz_criu_comm_fd.disconnectGr)rNr*rrr r,r,8sK rr,c$eZdZdZdZdZdZdS)_criu_comm_binz) Communication class for binary. cH|j|_||_d|_d|_dSr)rrrswrkr )r bin_paths r r!z_criu_comm_bin.__init__Ps$   rctjtjtjt jdtj}t jdtj|tjzt jdtj}t jdtj|tjz|_tj }|dkr_fd}|rLtj }|dkrtj |n7tj dn"|n|rtj |ddtjdddd_d_jS)Nrrctjdtjdtjddtjdtjtjjjdddzgtj ddS)Nrrrir6z%d) osr(sendstructpackgetpidexecvrfileno_exit)cssr sr exec_criuz)_criu_comm_bin.connect..exec_criuds   A FKRY[[99:::)VQ02333 rr:)r# socketpairr$r%fcntlF_GETFDF_SETFD FD_CLOEXECr r;forksetsidrBwaitpidr(r=unpackrecvr6r)r r flagsprDrCs` @r r z_criu_comm_bin.connectVs0EFF CFEM22 CFEM553C+CDDD CFEM22 CFEM5E4D3D+DEEE GII 66       GII66IKKKIKKKKHQKKKK  ! 1a    A M#s1v{{1~~66q9 a&wrc~|j|jstj|jddSdS)Nr)rr(r r;rMr6rs r rz_criu_comm_bin.disconnects?  { % Jty! $ $ $ $ $ % %rNr*rrr r4r4KsL ...`%%%%%rr4c eZdZdZdZdZdZdS) CRIUExceptionz? Exception class for handling and storing criu errors. Nc|jSr)_strrs r __str__zCRIUException.__str__s yr)rrrrtyprVrWrrr rTrTs9 C DrrTceZdZdZdZdS)CRIUExceptionInternalzC Exception class for handling and storing internal errors. cp||_tj|jd||_dS)Nz failed with internal error: )rXrpc criu_req_typeNamerV)r rXss r r!zCRIUExceptionInternal.__init__s6   " "48 , , , ,aa1 rN)rrrrr!rrr rZrZs-11111rrZceZdZdZdZdZdS)CRIUExceptionExternalzC Exception class for handling and storing criu RPC errors. cb||_||_||_||_dSr)rXresp_typerrno_gen_error_strrV)r req_typrcrds r r!zCRIUExceptionExternal.__init__s.   '')) rctj|jd}|j|jkr|d|jfzz }|d|jfzz }|jt jkr|dz }|jtjkr|jt jkr|dz }|jtj kr|jt j kr|dz }|dz }|S)Nz failed: zUnexpected response type %d: z Error(%d): z Bad optionszNo process with such pidz)Process with requested pid already existsUnknown) r\r]r^rXrcrdEBADRQCDUMPESRCHRESTOREEEXIST)r r_s r rez$CRIUExceptionExternal._gen_error_strs .33DH==== @ 8t} $ $ 0DM3DD DA ]dj^ ++ : & &  A 8sx  zU[((// 8s{ " "zU\))@@ YrN)rrrrr!rerrr raras<*** rrac\eZdZdZdZdZdZdZdZdZ dZ dZ d Z d Z d Zd Zd ZdS)criuz Call criu through RPC. Ncn|dtj|_d|_dS)Nro) use_binaryr\ criu_optsoptsrrs r r!z criu.__init__s, MOO rc.t||_dS)zZ Access criu using unix socket which that belongs to criu service daemon. N)r_comm)r sk_names r use_skz criu.use_sks#7++ rc.t||_dS)z0 Access criu using provided fd. N)r,rur.s r use_fdz criu.use_fds#2&& rc.t||_dS)zO Access criu by execing it using provided path to criu binary. N)r4ru)r bin_names r rqzcriu.use_binarys$H-- rc d}|jtjkr|jdsd} |js|j|}n|j}|jr||_| | | t| dtjtjz}|js|jtj}||n4#t&$r'}t)|jt+|d}~wwxYw|S)zA As simple as send request and receive response. FpidTrN)typer\rjrsHasFieldrrur keep_openr<SerializeToStringrOlenr# MSG_TRUNCMSG_PEEKr criu_respParseFromString ExceptionrZstr)r reqr r_bufrespes r _send_req_and_recv_respzcriu._send_req_and_recv_resps= 8sx  (9(9%(@(@ F :7 J&&v..G}  FF3((** + + +&&QVVAv'7&/'IJJKKLLC= ( %%'''=??D   % % % % : : :'#a&&99 9 : sC3D)) E3"EEctj}tj|_||}|js t |j|j|jdS)zB Checks whether the kernel support is up-to-date. N)r\criu_reqCHECKr~rsuccessracr_errnor rrs r checkz criu.checksZlnn9++C00| L'$)T]KK K L Lrctj}tj|_|j|j||}|js t|j|j|j |j SzC Checkpoint a process/tree identified by opts.pid. ) r\rrjr~rs MergeFromrrrardumprs r rz criu.dumpsnlnn8 49%%%++C00| L'$)T]KK Kyrctj}tj|_|j|j||}|js t|j|j|j |j Sr) r\rPRE_DUMPr~rsrrrrarrrs r pre_dumpz criu.pre_dump"snlnn< 49%%%++C00| L'$)T]KK Kyrctj}tj|_|j|j||}|js t|j|j|j |j S)z) Restore a process/tree. ) r\rrlr~rsrrrrarrestorers r rz criu.restore1snlnn; 49%%%++C00| L'$)T]KK K|rctj}tj|_|j|jd|_||}|js t|j|j|j |j S)NT) r\rPAGE_SERVER_CHLDr~rsrrrrrarpsrs r page_server_chldzcriu.page_server_chld@sslnn' 49%%% ++C00| L'$)T]KK Kwrctj}tj|_||_||}|js t|j|j|j|j Sr) r\rWAIT_PIDr~r}rrrarstatus)r r}rrs r wait_pidz criu.wait_pidMs[lnn<++C00| L'$)T]KK K{r)rrrrrsrur!rwryrqrrrrrrrrrr roros D E ,,, ''' ... !!!F L L L                 rro)r#rdrGr;r=pycriu.rpc_pb2rpc_pb2r\rrr,r4rrTrZrarorrr rs         2J&J&>%>%>%>%>%Z>%>%>%BI11111M111     M   FRRRRRRRRRRr