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�rn d S )Nz/run/systemd/system)�os�path�existsr r r r � test_booted& s r! c C s. | j d�j}tj|� tj|tjtjB �}d S )Nz test.fifo)�join�strpath�posix�mkfifor �open�O_RDONLY�
O_NONBLOCK)�tmpdirr �fdr r r �
test__is_fifo. s
r+ c C s2 | j d�}|jd� |j}tj|tjtjB �}d S )Nz test.fifo�boo)r" �writer# r r& r' r( )r) �filer r* r r r �test__is_fifo_file6 s
r/ c C s< | j d�j}tjt�� W d Q R X tjt�� W d Q R X d S )Nz test.fifo)r" r# r �raisesr )r) r r r r �test__is_fifo_bad_fd? s
r1 c C s: | j d�j}tj|� tj|tjtjB �}tj|d�}d S )Nz test.fifo�r) r" r# r$ r% r r& r' r( �fdopen)r) r r* r. r r r � test_is_fifoH s
r4 c C s> | j d�}|jd� |j}tj|tjtjB �}tj|d�}d S )Nz test.fifor, r2 )r" r- r# r r& r' r( r3 )r) r. r r* r r r �test_is_fifo_fileS s
r5 c C s< | j d�j}tjt�� W d Q R X tjt�� W d Q R X d S )Nz test.fifo)r" r# r r0 r )r) r r r r �test_is_fifo_bad_fd_ s
r6 c C s. yt | �S tk
r( } zdS d }~X nX d S )NF)r
r )�arg�errorr r r �
is_mq_wrapperh s r9 c C s. yt | �S tk
r( } zdS d }~X nX d S )NF)r r )r7 r8 r r r �_is_mq_wrapperp s r: c C sf t tj��R} xJ| D ]B}t� � W d Q R X |j� }t� � W d Q R X t� � W d Q R X qW W d Q R X d S )N)r r �AF_UNIXr �fileno)r �sockr* r r r �test_no_mismatchx s
r> c C sb t tj��N} xF| D ]>}x&||j� fD ]}t� � W d Q R X q$W t� � W d Q R X qW W d Q R X d S )N)r r r; r< r )r r= r7 r r r �test_is_socket� s
r? c C C s` t jtjtj����B} | jd� | j� \}}dj|�}�xd D �]}x�| | j� fD ]�}t� � W d Q R X t� � W d Q R X t� � W d Q R X t� � W d Q R X t� � W d Q R X t j
t �� t |ddd� W d Q R X t� � W d Q R X t� � W d Q R X qRW t j
t �� t
|ddd� W d Q R X t� � W d Q R X t� � W d Q R X | jd� q>W W d Q R X d S )
N� 127.0.0.1r z:{}r i@� )Zflowinfo� )r@ r )r r )�
contextlib�closingr ZAF_INET�bindZ getsockname�formatr< r r r0 �
ValueErrorr r Zlisten)r= ZaddrZportZ listeningr7 r r r �test_is_socket_sockaddr� s:
rG c C s0 t tj�� } x| D ] }|j� }qW W d Q R X d S )N)r r r; r< )r r= r* r r r �test__is_socket� s
rH c
C s> t tj��*} x"| D ]}x||j� fD ]}q$W qW W d Q R X d S )N)r r r; r< )r r= r7 r r r �test_is_socket_unix� s
rI c C s0 t tj�� } x| D ] }|j� }qW W d Q R X d S )N)r r r; r< )r r= r* r r r �test__is_socket_unix� s
rJ c C s t jd� t jd� d S )N�
LISTEN_FDS�
LISTEN_PID)r �unsetenvr r r r �test_listen_fds_no_fds� s
rN c C s dt jd< tt j� �t jd< d S )N�3rK rL )r �environ�str�getpidr r r r �test_listen_fds� s
rS c C s dt jd< tt j� �t jd< d S )N�1rK rL )r rP rQ rR r r r r � test_listen_fds_default_unset� s
rU c C s t � � W d Q R X d S )N)r r r r r �test_notify_no_socket s
rV � 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)�fdsr � )�pid)r[ rY )r )r rP r r0 �connection_errorr r rR r r r r �test_notify_bad_socket s
r] 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< t� � W d Q R X d S )Nr zfailed to bind socket (%s)� SO_PASSCRED� r rX )r" r# r r; Z
SOCK_DGRAMrD r8 r Zxfail�getattrZ
setsockoptZ
SOL_SOCKETr rP r )r) r r= r r^ r r r �test_notify_with_socket s
ra )rW )3�sysr r$ r rB r Zsystemd.daemonr r r r r r r r r
r r r
r r r r �contextmanagerr r r! r+ r/ r1 r4 r5 r6 r9 r: r> r? rG rH rI rJ rN rS rU rV � version_info�ConnectionRefusedErrorr\ r r] ra r r r r � sB D
&