JezK
Edit File: criu.cpython-36.pyc
3 i�" � @ s� d dl Z d dlZd dlZd dlZd dlZd dljZdZG 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.6/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 sb t j t jt j�| _y| jj| j� W n6 tk r@ td��Y n tk rZ td��Y nX | jS )NzSocket file not found.zService not running.)�socket�AF_UNIX�SOCK_SEQPACKETr r r �FileNotFoundError�ConnectionRefusedError)r r r r r r / s z_criu_comm_sk.connectc C s | j j� d S )N)r �close)r 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 )N)r r r )r �fdr r r r E s z_criu_comm_fd.__init__c C s t j| jt jt j�| _| jS )N)r Zfromfdr r r r )r r r r r r I s z_criu_comm_fd.connectc C s | j j� d S )N)r r )r r r r r O s z_criu_comm_fd.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_binz) Communication class for binary. c C s | j | _|| _d | _d | _d S )N)r r r �swrkr )r Zbin_pathr r r r X s z_criu_comm_bin.__init__c s t jt jt j�� tj� d tj�}tj� d tj|tjB � tj� d tj�}tj� d tj|tj @ � |�_t j � }|dkrƇ �fdd�}|r�t j � }|dkr�t j � |� q�t j d� q�|� n|r�t j |d� � d j� tjd� d jd��d �_� d �_�jS )Nr r c sh t jd� t jd� t jd� � d jtjdt j� �� t j�j�jdd� d j� g� t j d� d S )Nr r r �ir z%d) �osr �send�struct�pack�getpid�execvpr �fileno�_exitr )�cssr 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�pr+ r )r* r r r ^ s, z_criu_comm_bin.connectc C s" | j j� | js tj| jd� d S )Nr )r r r r" r0 r )r r r r r � s z_criu_comm_bin.disconnectN)r r r r r r r 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 )N)�_str)r r r r �__str__� s zCRIUException.__str__)r r r r �typr6 r7 r r r r r5 � s r5 c @ s e Zd ZdZdd� ZdS )�CRIUExceptionInternalzC Exception class for handling and storing internal errors. c C s" || _ dtjj| j �|f | _d S )Nz!%s failed with internal error: %s)r8 �rpc� criu_req_type�Namer6 )r r8 �sr r r r � s z CRIUExceptionInternal.__init__N)r r r r r r r r r r9 � s r9 c @ s e Zd ZdZdd� Zdd� ZdS )�CRIUExceptionExternalzC Exception class for handling and storing criu RPC errors. c C s || _ || _|| _| j� | _d S )N)r8 �resp_typ�errno�_gen_error_strr6 )r Zreq_typr? r@ r r r r � s z CRIUExceptionExternal.__init__c C s� dt jj| 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: r; r< r8 r? r@ ZEBADRQC�DUMPZESRCH�RESTOREZEEXIST)r r= r r r rA � s z$CRIUExceptionExternal._gen_error_strN)r r r r r rA r r r r r> � s r> 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" | j d� tjdd�| _d | _d S )NrD r )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 rH )r 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 rH )r Zbin_namer r r rF � s zcriu.use_binaryc C s� d}|j tjkr"|jjd� r"d}yx| js8| jj|�}n| j}|jrJ|| _|j |j � � |j t |j dt jt jB ���}|js�| jj� tj� }|j|� W n2 tk r� } zt|j t|���W Y dd}~X nX |S )zA As simple as send request and receive response. F�pidTr N)�typer: rB rG ZHasFieldr rH r � keep_openr# ZSerializeToStringr2 �lenr Z MSG_TRUNCZMSG_PEEKr Z criu_respZParseFromString� Exceptionr9 �str)r �reqr r= Zbuf�resp�er r r �_send_req_and_recv_resp� s$ "z criu._send_req_and_recv_respc C sD t j� }t j|_|jj| j� | j|�}|js@t|j|j|j ��dS )zB Checks whether the kernel support is up-to-date. N) r: �criu_reqZCHECKrL rG � MergeFromrT �successr> �cr_errno)r rQ rR r r r �check s z criu.checkc C sF t j� }t j|_|jj| j� | j|�}|js@t|j|j|j ��|j S )zC Checkpoint a process/tree identified by opts.pid. ) r: rU rB rL rG rV rT rW r> rX �dump)r rQ rR r r r rZ s z criu.dumpc C sF t j� }t j|_|jj| j� | j|�}|js@t|j|j|j ��|j S )zC Checkpoint a process/tree identified by opts.pid. ) r: rU ZPRE_DUMPrL rG rV rT rW r> rX rZ )r rQ rR r r r �pre_dump+ s z criu.pre_dumpc C sF t j� }t j|_|jj| j� | j|�}|js@t|j|j|j ��|j S )z) Restore a process/tree. ) r: rU rC rL rG rV rT rW r> rX �restore)r rQ rR r r r r\ : s z criu.restorec C sL t j� }t j|_|jj| j� d|_| j|�}|jsFt |j|j|j ��|j S )NT) r: rU ZPAGE_SERVER_CHLDrL rG rV rM rT rW r> rX Zps)r rQ rR r r r �page_server_chldI s zcriu.page_server_chldc C s> t j� }t j|_||_| j|�}|js8t|j|j|j��|j S )N) r: rU ZWAIT_PIDrL rK rT rW r> rX Zstatus)r rK rQ rR r r r �wait_pidV s z criu.wait_pid)r r r r rG rH r �CR_DEFAULT_SERVICE_ADDRESSrI rJ rF rT rY rZ r[ r\ r] r^ r r r r rD � s # rD )r r@ r- r" r$ Zpycriu.rpc_pb2Zrpc_pb2r: r_ r r r r rO r5 r9 r> rD r r r r �<module> s A "