3
��X( � @ sZ d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZmZm Z m
Z
m Z m Z m
Z
mZmZmZmZmZmZmZmZ d dlZejdd� �Zejdd� �Zdd� Zd d
� Zd d � Z d
d� Z dd� Z dd� Z dd� Z dd� Z!dd� Z"dd� Z#dd � Z$d d � Z%d d � Z&d!d"� Z'd#d$� Z(d%d&� Z)d'd(� Z*d)d*� Z+d+d,� Z,e j-d2k�rBe.Z/ne0Z/d.d/� Z1d0d1� Z2dS )3� N)�booted�is_fifo�_is_fifo� is_socket�
_is_socket�is_socket_inet�_is_socket_inet�is_socket_unix�_is_socket_unix�is_socket_sockaddr�_is_socket_sockaddr�is_mq�_is_mq�
listen_fds�notifyc c sH y
d V W n8 t k
rB } z | jtjkr0tj� � W Y d d } ~ X nX d S )N)�OSError�errnoZENOSYS�pytest�skip)�e� r �#/usr/lib64/python3.6/test_daemon.py� skip_enosys s
r c c s4 t j| �}z
|V W d |d j� |d j� X d S )Nr � )�socketZ
socketpair�close)Zfamily�pairr r r �closing_socketpair s
r c C s* t jjd�rt� s&t�nt� dks&t�d S )Nz/run/systemd/systemFT> FT)�os�path�existsr �AssertionErrorr r r r � test_booted& s r" c C sJ | j d�j}tj|� tj|tjtjB �}t|d �s8t �t||�sFt �d S )Nz test.fifo)
�join�strpath�posix�mkfifor �open�O_RDONLY�
O_NONBLOCKr r! )�tmpdirr �fdr r r �
test__is_fifo. s
r, c C sR | j d�}|jd� |j}tj|tjtjB �}t|d � s>t�t||� sNt�d S )Nz test.fifo�boo) r# �writer$ r r' r( r) r r! )r* �filer r+ r r r �test__is_fifo_file6 s
r0 c C s\ | j d�j}tjt�� tdd � s(t�W d Q R X tjt�� td|� sNt�W d Q R X d S )Nz test.fifor �����r1 )r# r$ r �raisesr r r! )r* r r r r �test__is_fifo_bad_fd? s
r3 c C sr | j d�j}tj|� tj|tjtjB �}tj|d�}t |d �sDt
�t ||�sRt
�t |d �s`t
�t ||�snt
�d S )Nz test.fifo�r) r# r$ r% r& r r' r( r) �fdopenr r! )r* r r+ r/ r r r � test_is_fifoH s
r6 c C s~ | j d�}|jd� |j}tj|tjtjB �}tj|d�}t|d � sJt �t||� sZt �t|d � sjt �t||� szt �d S )Nz test.fifor- r4 )
r# r. r$ r r' r( r) r5 r r! )r* r/ r r+ r r r �test_is_fifo_fileS s
r7 c C s\ | j d�j}tjt�� tdd � s(t�W d Q R X tjt�� td|� sNt�W d Q R X d S )Nz test.fifor r1 r1 )r# r$ r r2 r r r! )r* r r r r �test_is_fifo_bad_fd_ s
r8 c C s> yt | �S tk
r8 } z|jtjks*t�dS d }~X nX d S )NF)r
r r �EBADFr! )�arg�errorr r r �
is_mq_wrapperh s
r< c C s> yt | �S tk
r8 } z|jtjks*t�dS d }~X nX d S )NF)r r r r9 r! )r: r; r r r �_is_mq_wrapperp s
r= c C s t tj��� } x�| D ]�}t|� s&t�t|� s4t�t|� sBt�t� � t|d� sZt�W d Q R X |j � }t|� szt�t|� s�t�t|� s�t�t� � t|d� s�t�W d Q R X t
|� s�t�t |� s�t�t |� s�t�t� � t
|d� s�t�W d Q R X qW W d Q R X d S )Nz127.0.0.1:2000)r r �AF_UNIXr r! r< r r r �filenor r= r r )r �sockr+ r r r �test_no_mismatchx s$
rA c C sL t tj���6} �x,| D �]"}x�||j� fD ]�}t|�s:t�t|tj�sJt�t|tj� s\t�t|tjtj�spt�t|tjtj� s�t�t � � t
|dtjdd� s�t�W d Q R X q*W t |�s�t�t |tj�s�t�t |tj� s�t�t |tjtj�s�t�t |tjtj� �st�t � � t |dtjdd� �s0t�W d Q R X qW W d Q R X d S )Nz 8.8.8.8:2000r )
r r r> r? r r! �AF_INET� SOCK_STREAM�
SOCK_DGRAMr r r r )r r@ r: r r r �test_is_socket� s" & rE c E C s, t jtjtj����} | jd
� | j� \}}dj|�}�x�d D �]�}�xT| | j� fD �]B}t� � t |dtj
�stt �W d Q R X t� � t |d| tj
�s�t �W d Q R X t� � t |d| |d�s�t �W d Q R X t� � t |d| d d�s�t �W d Q R X t� �" t |d| | d� �st �W d Q R X t j
t�� t |ddd� W d Q R X t� � t |dtj
� �sft �W d Q R X t� � t |dtj� �s�t �W d Q R X qTW t j
t�� t|ddd� W d Q R X t� � t|dtj
� �s�t �W d Q R X t� � t|dtj� �st �W d Q R X | jd � q>W W d Q R X d S )
N� 127.0.0.1r z:{}r )� listeningi@� )Zflowinfoz129.168.11.11:23� )rF r )r r r1 )�
contextlib�closingr rB �bindZ getsockname�formatr? r r rC r! r r2 �
ValueErrorrD r Zlisten)r@ ZaddrZportrG r: r r r �test_is_socket_sockaddr� s:
& $ rN c C s� t tj���} x�| D ]�}|j� }t|�s*t�t|tj�s:t�t|tj� sLt�t|tjtj�s`t�t|tjtj� svt�t|�s�t�t|tj�s�t�t|tj� s�t�t|tjtj�s�t�t|tjtj� st�qW W d Q R X d S )N) r r r> r? r r! rB rC rD )r r@ r+ r r r �test__is_socket� s
rO c
C s~ t tj��j} xb| D ]Z}xT||j� fD ]D}t|�s4t�t|dd� sFt�t|tj�sVt�t|tj� s$t�q$W qW W d Q R X d S )Nz
/no/such/path)r )r r r> r? r r! rC rD )r r@ r: r r r �test_is_socket_unix� s
rP c
C sr t tj��^} xV| D ]N}|j� }t|�s*t�t|ddd� s>t�t|tj�sNt�t|tj� st�qW W d Q R X d S )Nr r z
/no/such/pathr1 )r r r> r? r
r! rC rD )r r@ r+ r r r �test__is_socket_unix� s
rQ c C sF t jd� t jd� t� g ks"t�td�g ks2t�td�g ksBt�d S )N�
LISTEN_FDS�
LISTEN_PIDTF)r �unsetenvr r! r r r r �test_listen_fds_no_fds� s
rU c C sZ dt jd< tt j� �t jd< td�dddgks2t�td�dddgksHt�t� g ksVt�d S ) N�3rR rS F� � � T)r �environ�str�getpidr r! r r r r �test_listen_fds� s
r] c C sP dt jd< tt j� �t jd< td�dgks.t�t� dgks>t�t� g ksLt�d S )N�1rR rS FrW )r rZ r[ r\ r r! r r r r � test_listen_fds_default_unset� s
r_ c C s� t d�dkst�t� � t dg d�dks,t�W d Q R X t dddgd�dksNt�t dtj� d�dksft�t dtj� d d�dks�t�d S )
NzREADY=1Fz FDSTORE=1)�fdsr � )�pid)rb r` )r )r r! r r r\ r r r r �test_notify_no_socket s rc rW c / C s� dt jd< tjt�� td� W d Q R X tjt��$ t� � tdg d� W d Q R X W d Q R X tjt�� tdddgd� W d Q R X tjt�� tdt j� d� W d Q R X tjt�� tdt j� d
d � W d Q R X d S ) Nz /dev/null�
NOTIFY_SOCKETzREADY=1z FDSTORE=1)r` r ra )rb )rb r` )r )r rZ r r2 �connection_errorr r r\ r r r r �test_notify_bad_socket s
rf c C s� | j d�j}tjtjtj�}y|j|� W n2 tjk
r\ } ztjd| � W Y d d }~X nX t tdd�}|j
tj |d� |t j
d< td�s�t�t� � tdg d �s�t�W d Q R X tddd
gd �s�t�tdt j� d �s�t�tdt j� d
d �s�t�d S )Nr zfailed to bind socket (%s)� SO_PASSCRED� r rd zREADY=1z FDSTORE=1)r` ra )rb )rb r` )r )r# r$ r r> rD rK r; r Zxfail�getattrZ
setsockoptZ
SOL_SOCKETr rZ r r! r r\ )r* r r@ r rg r r r �test_notify_with_socket s
rj )rW )3�sysr r% r rI r Zsystemd.daemonr r r r r r r r r
r r r
r r r r �contextmanagerr r r" r, r0 r3 r6 r7 r8 r< r= rA rE rN rO rP rQ rU r] r_ rc � version_info�ConnectionRefusedErrorre r rf rj r r r r � sB D
&