관리-도구
편집 파일: criu.cpython-39.pyc
a s�1j�" � @ s� d dl Z d dlZd dlZd dlZd dlZd dlmZ dZG dd� d�Z G dd� de �Z G dd� de �Z G d d � d e �Z G d d � d e �ZG d d� de�ZG dd� de�ZG dd� d�ZdS )� 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� ��self�daemonr r �//usr/lib/python3.9/site-packages/pycriu/criu.py�connect 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_SK�COMM_FD�COMM_BIN� comm_type�comm�skr 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 s^ t � t jt j�| _z| j�| j� W n2 ty> td��Y n tyV td��Y n0 | jS )NzSocket file not found.zService not running.)�socket�AF_UNIX�SOCK_SEQPACKETr r r �FileNotFoundError�ConnectionRefusedErrorr r r r r / s z_criu_comm_sk.connectc C s | j �� d S r �r �closer 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 r )r r r �r �fdr r r r E s z_criu_comm_fd.__init__c C s t �| jt jt j�| _| jS r )r Zfromfdr r r r r r r r r I s �z_criu_comm_fd.connectc C s | j �� d S r r r r r r r O s z_criu_comm_fd.disconnectNr! 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 r )r r r �swrkr )r Zbin_pathr r r r X s z_criu_comm_bin.__init__c s t �t jt j�� t�� d tj�}t�� d tj|tjB � t�� d tj�}t�� d tj|tj @ � |�_t � � }|dkrƇ �fdd�}|r�t � � }|dkr�t � � |� q�t � d� q�|� n|r�t � |d� � d �� t�d� d �d��d �_� d �_�jS )Nr r c sh t �d� t �d� t �d� � d �t�dt �� �� t ��j�jdd� d �� g� t � d� d S )Nr r r �ir& z%d) �osr �send�struct�pack�getpid�execvpr �fileno�_exitr �Zcssr r r � exec_criul s ��z)_criu_comm_bin.connect.<locals>.exec_criur' � )r Z socketpairr r �fcntlZF_GETFDZF_SETFDZ FD_CLOEXECr r( �fork�setsidr/ �waitpidr r* �unpack�recvr& r )r r �flags�pr1 r r0 r r ^ s, z_criu_comm_bin.connectc C s" | j �� | js t�| jd� d S )Nr )r r r r( r6 r& r r r r r � s z_criu_comm_bin.disconnectNr! 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 r )�_strr r r r �__str__� s zCRIUException.__str__)r r r r �typr< r= r r r r r; � s r; c @ s e Zd ZdZdd� ZdS )�CRIUExceptionInternalzC Exception class for handling and storing internal errors. c C s" || _ dtj�| j �|f | _d S )Nz!%s failed with internal error: %s)r> �rpc� criu_req_type�Namer< )r r> �sr r r r � s �z CRIUExceptionInternal.__init__N)r r r r r r r r r r? � s r? c @ s e Zd ZdZdd� Zdd� ZdS )�CRIUExceptionExternalzC Exception class for handling and storing criu RPC errors. c C s || _ || _|| _| �� | _d S r )r> �resp_typ�errno�_gen_error_strr< )r Zreq_typrE rF r r r r � s z CRIUExceptionExternal.__init__c C s� dt j�| j�f }| 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@ rA rB r> rE rF ZEBADRQC�DUMPZESRCH�RESTOREZEEXIST)r rC r r r rG � s z$CRIUExceptionExternal._gen_error_strN)r r r r r rG r r r r rD � s rD c @ st e Zd ZdZdZdZdd� Zefdd�Zdd� 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" | � d� tjdd�| _d | _d S )NrJ ���)Z images_dir_fd)� use_binaryr@ Z criu_opts�optsr 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" rN 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% rN )r Zbin_namer r r rL � s zcriu.use_binaryc C s� d}|j tjkr |j�d�s d}zx| js6| j�|�}n| j}|jrH|| _|� |� � � |� t |� dt jt jB ���}|js�| j�� t�� }|�|� W n4 ty� } z t|j t|���W Y d}~n d}~0 0 |S )zA As simple as send request and receive response. F�pidTr N)�typer@ rH rM ZHasFieldr rN r � keep_openr) ZSerializeToStringr8 �lenr Z MSG_TRUNCZMSG_PEEKr Z criu_respZParseFromString� Exceptionr? �str)r �reqr rC �buf�resp�er r r �_send_req_and_recv_resp� s$ &z criu._send_req_and_recv_respc C sD t �� }t j|_|j�| j� | �|�}|js@t|j|j|j ��dS )zB Checks whether the kernel support is up-to-date. N) r@ �criu_reqZCHECKrR rM � MergeFromr[ �successrD �cr_errno�r rW rY r r r �check s z criu.checkc C sF t �� }t j|_|j�| j� | �|�}|js@t|j|j|j ��|j S �zC Checkpoint a process/tree identified by opts.pid. ) r@ r\ rH rR rM r] r[ r^ rD r_ �dumpr` r r r rc s z criu.dumpc C sF t �� }t j|_|j�| j� | �|�}|js@t|j|j|j ��|j S rb ) r@ r\ ZPRE_DUMPrR rM r] r[ r^ rD r_ rc r` r r r �pre_dump+ s z criu.pre_dumpc C sF t �� }t j|_|j�| j� | �|�}|js@t|j|j|j ��|j S )z) Restore a process/tree. ) r@ r\ rI rR rM r] r[ r^ rD r_ �restorer` r r r re : s z criu.restorec C sL t �� }t j|_|j�| j� d|_| �|�}|jsFt |j|j|j ��|j S )NT) r@ r\ ZPAGE_SERVER_CHLDrR rM r] rS r[ r^ rD r_ Zpsr` r r r �page_server_chldI s zcriu.page_server_chldc C s> t �� }t j|_||_| �|�}|js8t|j|j|j��|j S r ) r@ r\ ZWAIT_PIDrR rQ r[ r^ rD r_ �status)r rQ rW rY r r r �wait_pidV s z criu.wait_pid)r r r r rM rN r �CR_DEFAULT_SERVICE_ADDRESSrO rP rL r[ ra rc rd re rf rh r r r r rJ � s # rJ )r rF r3 r( r* Zpycriu.rpc_pb2Zrpc_pb2r@ ri r r r" r% rU r; r? rD rJ r r r r �<module> s A "