grthtrhthjhtyjytjytkergtrhtrjytjerhrfh4:24 29/09/2026a s1j"@sddlZddlZddlZddlZddlZddlmZdZGdddZ Gddde Z Gddde Z Gd d d e Z Gd d d e ZGd ddeZGdddeZGdddZdS)Nz./criu_service.socketc@s8eZdZdZdZdZdZdZdZdZ ddZ dd Z dS) _criu_commz/ Base class for communication classes. rNcCsdS)z Connect to criu and return socket object. daemon -- is for whether or not criu should daemonize if executing criu from binary(comm_bin). Nselfdaemonrr//usr/lib/python3.9/site-packages/pycriu/criu.pyconnectsz_criu_comm.connectcCsdS)z' Disconnect from criu. Nrrrrr disconnectsz_criu_comm.disconnect) __name__ __module__ __qualname____doc__COMM_SKCOMM_FDCOMM_BIN comm_typecommskr r rrrr r src@s(eZdZdZddZddZddZdS) _criu_comm_skz. Communication class for unix socket. cCs|j|_||_dSN)rrr)rZsk_pathrrr __init__+sz_criu_comm_sk.__init__cCs^ttjtj|_z|j|jWn2ty>tdYntyVtdYn0|jS)NzSocket file not found.zService not running.)socketAF_UNIXSOCK_SEQPACKETrr rFileNotFoundErrorConnectionRefusedErrorrrrr r /s   z_criu_comm_sk.connectcCs|jdSrrcloser rrr r <sz_criu_comm_sk.disconnectNr rrrrr r rrrr r&s rc@s(eZdZdZddZddZddZdS) _criu_comm_fdz2 Communication class for file descriptor. cCs|j|_||_dSr)rrrrfdrrr rEsz_criu_comm_fd.__init__cCst|jtjtj|_|jSr)rZfromfdrrrrrrrr r Is z_criu_comm_fd.connectcCs|jdSrrr rrr r Osz_criu_comm_fd.disconnectNr!rrrr r"@sr"c@s(eZdZdZddZddZddZdS) _criu_comm_binz) Communication class for binary. cCs|j|_||_d|_d|_dSr)rrrswrkr)rZbin_pathrrr rXsz_criu_comm_bin.__init__csttjtjtdtj}tdtj|tjBtdtj}tdtj|tj@|_t }|dkrƇfdd}|rt }|dkrt |qt dq|n|rt |ddtdddd_d_jS)Nrrcshtdtdtddtdttjjdddgt ddS)Nrrrir&z%d) osr sendstructpackgetpidexecvprfileno_exitrZcssrrr exec_criuls   z)_criu_comm_bin.connect..exec_criur')rZ socketpairrrfcntlZF_GETFDZF_SETFDZ FD_CLOEXECrr(forksetsidr/waitpidr r*unpackrecvr&r)rrflagspr1rr0r r ^s,     z_criu_comm_bin.connectcCs"|j|jst|jddS)Nr)rr rr(r6r&r rrr r s z_criu_comm_bin.disconnectNr!rrrr r%Ss0r%c@s eZdZdZdZdZddZdS) CRIUExceptionz? Exception class for handling and storing criu errors. NcCs|jSr)_strr rrr __str__szCRIUException.__str__)r rrrtypr<r=rrrr r;sr;c@seZdZdZddZdS)CRIUExceptionInternalzC Exception class for handling and storing internal errors. cCs"||_dtj|j|f|_dS)Nz!%s failed with internal error: %s)r>rpc criu_req_typeNamer<)rr>srrr rszCRIUExceptionInternal.__init__N)r rrrrrrrr r?sr?c@s eZdZdZddZddZdS)CRIUExceptionExternalzC Exception class for handling and storing criu RPC errors. cCs ||_||_||_||_dSr)r>resp_typerrno_gen_error_strr<)rZreq_typrErFrrr rszCRIUExceptionExternal.__init__cCsdtj|jf}|j|jkr0|d|jf7}|d|jf7}|jtjkrV|d7}nL|jtjkrx|jtjkrx|d7}n*|jtj kr|jtj kr|d7}n|d7}|S)Nz %s failed: zUnexpected response type %d: z Error(%d): z Bad optionszNo process with such pidz)Process with requested pid already existsZUnknown) r@rArBr>rErFZEBADRQCDUMPZESRCHRESTOREZEEXIST)rrCrrr rGs     z$CRIUExceptionExternal._gen_error_strN)r rrrrrGrrrr rDsrDc@steZdZdZdZdZddZefddZddZ d d Z d d Z d dZ ddZ ddZddZddZddZdS)criuz Call criu through RPC. NcCs"|dtjdd|_d|_dS)NrJ)Z images_dir_fd) use_binaryr@Z criu_optsoptsrr rrr rs z criu.__init__cCst||_dS)zZ Access criu using unix socket which that belongs to criu service daemon. N)r_comm)rZsk_namerrr use_sksz criu.use_skcCst||_dS)z0 Access criu using provided fd. N)r"rNr#rrr use_fdsz criu.use_fdcCst||_dS)zO Access criu by execing it using provided path to criu binary. N)r%rN)rZbin_namerrr rLszcriu.use_binaryc Csd}|jtjkr |jds d}zx|js6|j|}n|j}|jrH||_| | | t | dt jt jB}|js|jt}||Wn4ty}zt|jt|WYd}~n d}~00|S)zA As simple as send request and receive response. FpidTrN)typer@rHrMZHasFieldrrNr keep_openr)ZSerializeToStringr8lenrZ MSG_TRUNCZMSG_PEEKr Z criu_respZParseFromString Exceptionr?str)rreqrrCbufresperrr _send_req_and_recv_resps$ &zcriu._send_req_and_recv_respcCsDt}tj|_|j|j||}|js@t|j|j|j dS)zB Checks whether the kernel support is up-to-date. N) r@criu_reqZCHECKrRrM MergeFromr[successrDcr_errnorrWrYrrr checks  z criu.checkcCsFt}tj|_|j|j||}|js@t|j|j|j |j SzC Checkpoint a process/tree identified by opts.pid. ) r@r\rHrRrMr]r[r^rDr_dumpr`rrr rcs z criu.dumpcCsFt}tj|_|j|j||}|js@t|j|j|j |j Srb) r@r\ZPRE_DUMPrRrMr]r[r^rDr_rcr`rrr pre_dump+s z criu.pre_dumpcCsFt}tj|_|j|j||}|js@t|j|j|j |j S)z) Restore a process/tree. ) r@r\rIrRrMr]r[r^rDr_restorer`rrr re:s z criu.restorecCsLt}tj|_|j|jd|_||}|jsFt |j|j|j |j S)NT) r@r\ZPAGE_SERVER_CHLDrRrMr]rSr[r^rDr_Zpsr`rrr page_server_chldIs zcriu.page_server_chldcCs>t}tj|_||_||}|js8t|j|j|j|j Sr) r@r\ZWAIT_PIDrRrQr[r^rDr_status)rrQrWrYrrr wait_pidVs z criu.wait_pid)r rrrrMrNrCR_DEFAULT_SERVICE_ADDRESSrOrPrLr[rarcrdrerfrhrrrr rJs #  rJ)rrFr3r(r*Zpycriu.rpc_pb2Zrpc_pb2r@rirrr"r%rUr;r?rDrJrrrr s A  "