File: //usr/lib/python3.9/site-packages/pycriu/__pycache__/criu.cpython-39.pyc
a
k�x] ! � @ s� d dl Z d dlZd dlZd dlZd dlZd dlmZ G dd� d�ZG dd� de�Z G dd� de�Z
G dd � d e�Z G d
d � d e �Z
G d d
� d
e
�ZG dd� de
�ZG dd� d�ZdS )� Nc @ 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�| _| j�| j� | jS r )�socket�AF_UNIX�SOCK_SEQPACKETr r
r r r r r r
. s z_criu_comm_sk.connectc C s | j �� d S r �r �closer r r r r 4 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 = 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
A s �z_criu_comm_fd.connectc C s | j �� d S r r r r r r r G s z_criu_comm_fd.disconnectNr r r r r r 8 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 P 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�execvr �fileno�_exitr �Zcssr r r � exec_criud 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�pr/ r r. r r
V s,
z_criu_comm_bin.connectc C s" | j �� | js t�| jd� d S )Nr )r r r r& r4 r$ r r r r r � s
z_criu_comm_bin.disconnectNr r r r r r# K 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 r9 � s r9 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_typrC rD r r r r � s z CRIUExceptionExternal.__init__c C s� dt j�| j�f }| j| jkr0|d| jf 7 }|d| jf 7 }| jtjkrT|d7 }| jt jkrt| jtjkrt|d7 }| jt j kr�| jtj
kr�|d7 }|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@ r< rC rD ZEBADRQC�DUMPZESRCH�RESTOREZEEXIST)r rA r r r rE � s z$CRIUExceptionExternal._gen_error_strN)r
r r r r rE r r r r rB � s rB c @ sp e Zd ZdZdZdZdd� Zdd� Zdd� Zd d
� Z d d � Z
d
d� Z dd� Z dd� Z
dd� Zdd� Zdd� ZdS )�criuz
Call criu through RPC.
Nc C s | � d� t�� | _d | _d S )NrH )�
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 rK 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# rK )r Zbin_namer r r rI � 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> rF rJ ZHasFieldr rK r
� keep_openr' ZSerializeToStringr6 �lenr Z MSG_TRUNCZMSG_PEEKr Z criu_respZParseFromString� Exceptionr= �str)r �reqr rA �buf�resp�er r r �_send_req_and_recv_resp� s$
&z criu._send_req_and_recv_respc C s6 t �� }t j|_| �|�}|js2t|j|j|j��dS )zB
Checks whether the kernel support is up-to-date.
N)r> �criu_reqZCHECKrO rX �successrB �cr_errno�r rT rV 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> rY rF rO rJ � MergeFromrX rZ rB r[ �dumpr\ r r r r` s
z criu.dumpc C sF t �� }t j|_|j�| j� | �|�}|js@t|j|j|j ��|j
S r^ ) r> rY ZPRE_DUMPrO rJ r_ rX rZ rB r[ r` 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> rY rG rO rJ r_ rX rZ rB r[ �restorer\ r r r rb 1 s
z criu.restorec C sL t �� }t j|_|j�| j� d|_| �|�}|jsFt |j|j|j
��|j S )NT) r> rY ZPAGE_SERVER_CHLDrO rJ r_ rP rX rZ rB r[ Zpsr\ r r r �page_server_chld@ s
zcriu.page_server_chldc C s> t �� }t j|_||_| �|�}|js8t|j|j|j��|j S r )
r> rY ZWAIT_PIDrO rN rX rZ rB r[ �status)r rN rT rV r r r �wait_pidM s
z
criu.wait_pid)r
r r r rJ rK r rL rM rI rX r] r` ra rb rc re r r r r rH � s #
rH )r rD r1 r& r( Zpycriu.rpc_pb2Zrpc_pb2r> r r r r# rR r9 r= rB rH r r r r �<module> s A #