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-d2krBe.Z/ne0Z/d.d/ Z1d0d1 Z2dS )3 N)bootedis_fifo_is_fifo is_socket
_is_socketis_socket_inet_is_socket_inetis_socket_unix_is_socket_unixis_socket_sockaddr_is_socket_sockaddris_mq_is_mq
listen_fdsnotifyc c sH y
d V W n8 t k
rB } z | jtjkr0tj W Y d d } ~ X nX d S )N)OSErrorerrnoZENOSYSpytestskip)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
socketpairclose)Zfamilypairr r r closing_socketpair s
r c C s t jjdrn d S )Nz/run/systemd/system)ospathexistsr r r r test_booted& s r! c C s. | j dj}tj| tj|tjtjB }d S )Nz test.fifo)joinstrpathposixmkfifor openO_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.fifoboo)r" writer# r r& r' r( )r) filer r* r r r test__is_fifo_file6 s
r/ c C s< | j dj}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 dj}tj| tj|tjtjB }tj|d}d S )Nz test.fifor) 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 dj}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 )argerrorr 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 tjR} 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 tjN} 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 jtjtjB} | 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 )
contextlibclosingr ZAF_INETbindZ getsocknameformatr< 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 )N3rK rL )r environstrgetpidr r r r test_listen_fds s
rS c C s dt jd< tt j t jd< d S )N1rK 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 dj}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 ZxfailgetattrZ
setsockoptZ
SOL_SOCKETr rP r )r) r r= r r^ r r r test_notify_with_socket s
ra )rW )3sysr 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_infoConnectionRefusedErrorr\ r r] ra r r r r sB D
&