????
Your IP : 216.73.216.86
3
�[gS � @ s\ d Z ddlZddlZddlZddlmZ ddlmZ ddlmZ ddlmZ ddlm Z ddl
Z
ddl
m Z dd l
m Z dd
l
m
Z
dd l
mZ dd l
mZ dd
l
mZ ddl
mZ ddl
mZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlm Z ddlm Z ddlm Z ddlm Z ddlm Z ddlm!Z! dd lm"Z" dd lm#Z# dd lm$Z$ dd lm%Z% dd lm&Z& dd!lm'Z' e(ed"e)� �Z*d#d$� Z+e!j,j-d%d&�G d'd(� d(e��Z.G d)d*� d*e.�Z/e!j,j-d%d&�G d+d,� d,e.��Z0e!j,j-d%d&�G d-d.� d.e.��Z1G d/d0� d0e.�Z2e!j,j3ed1d2�G d3d4� d4e.��Z4G d5d6� d6e�Z5dS )7zFTests for psutil.net_connections() and Process.net_connections() APIs.� N)�closing)�AF_INET)�AF_INET6)�
SOCK_DGRAM)� SOCK_STREAM)�FREEBSD)�LINUX)�MACOS)�NETBSD)�OPENBSD)�POSIX)�SUNOS)�WINDOWS)�
supports_ipv6)�PY3)�AF_UNIX)�HAS_NET_CONNECTIONS_UNIX)� SKIP_SYSCONS)�PsutilTestCase)� bind_socket)�bind_unix_socket)�check_connection_ntuple)�create_sockets)�filter_proc_net_connections)�pytest)�
reap_children)�retry_on_failure)�skip_on_access_denied)�tcp_socketpair)�unix_socketpair)�
wait_for_file�SOCK_SEQPACKETc C s$ t j� j| d�}| dkr t|�S |S )N)�kind�all�unix)r# r$ )�psutil�Process�net_connectionsr )r" �cons� r) �I/usr/local/lib64/python3.6/site-packages/psutil/tests/test_connections.py�this_proc_net_connections2 s r+ �serial)�namec @ s& e Zd Zdd� Zdd� Zd dd�ZdS )
�ConnectionTestCasec C s t dd�g kst�d S )Nr# )r" )r+ �AssertionError)�selfr) r) r* �setUp; s zConnectionTestCase.setUpc C s t dd�g kst�d S )Nr# )r" )r+ r/ )r0 r) r) r* �tearDown> s zConnectionTestCase.tearDownr# c sd yt j|d�}W n t jk
r0 tr*dS � Y nX � fdd�|D �}|j� |j� ||ks`t�dS )z�Given a process PID and its list of connections compare
those against system-wide connections retrieved via
psutil.net_connections.
)r" Nc s" g | ]}|j � kr|d d� �qS )N� ���)�pid)�.0�c)r5 r) r* �
<listcomp>Q s zBConnectionTestCase.compare_procsys_connections.<locals>.<listcomp>)r% r' � AccessDeniedr �sortr/ )r0 r5 Z proc_consr" Zsys_consr) )r5 r* �compare_procsys_connectionsB s z.ConnectionTestCase.compare_procsys_connectionsN)r# )�__name__�
__module__� __qualname__r1 r2 r; r) r) r) r* r. 9 s r. c @ s4 e Zd Zejjedd�dd� �Zdd� Zdd� Z d S )
�TestBasicOperationsz
requires root)�reasonc
C s4 t � �$ x tjdd�D ] }t|� qW W d Q R X d S )Nr# )r" )r r% r' r )r0 �connr) r) r* � test_systemX s z TestBasicOperations.test_systemc
C s2 t � �" xtdd�D ] }t|� qW W d Q R X d S )Nr# )r" )r r+ r )r0 rA r) r) r* � test_process^ s z TestBasicOperations.test_processc C sF t jt�� tdd� W d Q R X t jt�� tjdd� W d Q R X d S )Nz???)r" )r Zraises�
ValueErrorr+ r% r' )r0 r) r) r* �test_invalid_kindc s z%TestBasicOperations.test_invalid_kindN)
r< r= r> r �mark�skipifr rB rC rE r) r) r) r* r? W s r? c @ s� e Zd ZdZdd� Zdd� Zdd� Zejj e
� dd �d
d � �Z d d
� Z ejj e
� dd �dd� �Z
ejj e dd �dd� �Zejj e dd �dd� �ZdS )�TestUnconnectedSocketsz;Tests sockets which are open but not connected to anything.c C st t dd�}tdd� |D ��}ts$tr0||j� S t|�dks@t�|d jdkrh||j� j|j� ksht�|d S d S )Nr# )r" c S s g | ]}|j |f�qS r) )�fd)r6 r7 r) r) r* r8 p s z=TestUnconnectedSockets.get_conn_from_sock.<locals>.<listcomp>r3 r r4 )r+ �dictr
r �fileno�lenr/ rI )r0 �sockr( Zsmapr) r) r* �get_conn_from_sockn s
z)TestUnconnectedSockets.get_conn_from_sockc C s� | j |�}t|� |jdkr.|j|j� ks.t�|j|jks>t�|j|jtj tj
�ksXt�|j � }| r|t r|t
|t�r||j� }|jtkr�|dd� }|j|ks�t�|jtkr�tr�tdd�}| jtj� |dd� |S )z�Given a socket, makes sure it matches the one obtained
via psutil. It assumes this process created one connection
only (the one supposed to be checked).
r3 N� r# )r" r4 )rN r rI rK r/ �family�type�
getsockopt�socket�
SOL_SOCKET�SO_TYPE� getsocknamer �
isinstance�bytes�decoder �laddrr r r+ r; �os�getpid)r0 rM rA rZ r( r) r) r* � check_socket{ s
z#TestUnconnectedSockets.check_socketc C sN d}t ttt|d���.}| j|�}|jf ks0t�|jtj ks@t�W d Q R X d S )N� 127.0.0.1r )�addr)r^ r )
r r r r r] �raddrr/ �statusr% � CONN_LISTEN)r0 r_ rM rA r) r) r* � test_tcp_v4� s
z"TestUnconnectedSockets.test_tcp_v4zIPv6 not supported)r@ c C sN d}t ttt|d���.}| j|�}|jf ks0t�|jtj ks@t�W d Q R X d S )N�::1r )r_ )rd r )
r r r r r] r` r/ ra r% rb )r0 r_ rM rA r) r) r* � test_tcp_v6� s
z"TestUnconnectedSockets.test_tcp_v6c C sN d}t ttt|d���.}| j|�}|jf ks0t�|jtj ks@t�W d Q R X d S )N� 127.0.0.1r )r_ )rf r )
r r r r r] r` r/ ra r% � CONN_NONE)r0 r_ rM rA r) r) r* � test_udp_v4� s
z"TestUnconnectedSockets.test_udp_v4c C sN d}t ttt|d���.}| j|�}|jf ks0t�|jtj ks@t�W d Q R X d S )N�::1r )r_ )ri r )
r r r r r] r` r/ ra r% rg )r0 r_ rM rA r) r) r* � test_udp_v6� s
z"TestUnconnectedSockets.test_udp_v6z
POSIX onlyc C sP | j � }tt|td���.}| j|�}|jdks2t�|jtj ksBt�W d Q R X d S )N)rQ � )
�
get_testfnr r r r] r` r/ ra r% rg )r0 �testfnrM rA r) r) r* �
test_unix_tcp� s
z$TestUnconnectedSockets.test_unix_tcpc C sP | j � }tt|td���.}| j|�}|jdks2t�|jtj ksBt�W d Q R X d S )N)rQ rk )
rl r r r r] r` r/ ra r% rg )r0 rm rM rA r) r) r* �
test_unix_udp� s
z$TestUnconnectedSockets.test_unix_udpN)r< r= r> �__doc__rN r] rc r rF rG r re rh rj r rn ro r) r) r) r* rH j s
rH c @ sB e Zd ZdZejjedd�dd� �Zejje dd�dd� �Z
d S )
�TestConnectedSocketzFTest socket pairs which are actually connected to
each other.
zunreliable on SUONS)r@ c
C s� d}t dd�g kst�tt|d�\}}zFt dd�}t|�dksBt�|d jtjksVt�|d jtjksjt�W d |j� |j� X d S ) N� 127.0.0.1r �tcp4)r" )r_ rO r3 )rr r ) r+ r/ r r rL ra r% �CONN_ESTABLISHED�close)r0 r_ �server�clientr( r) r) r* �test_tcp� s
z TestConnectedSocket.test_tcpz
POSIX onlyc
C s
| j � }t|�\}}z�tdd�}|d jo2|d j s>t|��|d joP|d j s\t|��tsdtrrdd� |D �}t|�dks�t�t s�ts�t
s�t r�|d jdks�t�|d jdks�t�||d jp�|d jks�t�n |d jp�|d j|ks�t�W d |j � |j � X d S ) Nr$ )r" r r3 c S s g | ]}|j d kr|�qS )z /var/run/log)r` )r6 r7 r) r) r* r8 � s z1TestConnectedSocket.test_unix.<locals>.<listcomp>rO rk )
rl r r+ rZ r` r/ r
r rL r r
r ru )r0 rm rv rw r( r) r) r* � test_unix� s
z TestConnectedSocket.test_unixN) r< r= r> rp r rF rG r
rx r ry r) r) r) r* rq � s rq c @ s. e Zd Zdd� Zeed�dd� �Zdd� ZdS ) � TestFiltersc
C s� dd� }t � �� |dtttgtttg� |dttgttg� |dtgttg� |dttgtg� |dtgtg� |dtgtg� |d ttgtg� |d
tgtg� |d tgtg� tr�|d tgtttg� W d Q R X d S )
Nc S sj x.t | d�D ] }|j|ks t�|j|ks t�q W tsfx0tj| d�D ] }|j|ksTt�|j|ksBt�qBW d S )N)r" )r+ rP r/ rQ r r% r' )r" �families�typesrA r) r) r* �check s z'TestFilters.test_filters.<locals>.checkr# �inet�inet4�tcprs �tcp6�udp�udp4�udp6r$ )r r r r r r r! r )r0 r} r) r) r* � test_filters s& zTestFilters.test_filters)Zonly_ifc s t � � fdd�}tjd�}tjd�}tjj� jtj� d��}|jt t
�d|d�}|jt t
�d|d�}|jt t �d|d�}|jt t �d|d�}� j |�} t
t|d d
��}
� j |�} t
t|d d
��} t� r�� j |�}
t
t|d d
��}� j |�}t
t|d d
��}nd }
d }d }d }� x�tj� j� D � ]�}|j� }t|�d k�s>t�x�|D ]�}|j| jk�rp|||t
t|
f tjd� n�|j| jk�r�|||t
t| f tjd� nZ|jt|
dd �k�r�|||t t|f tjd� n,|jt|dd �k�rD|||t t|f tjd� �qDW �q W d S )Nc
s� d }t |� |j|kst�|j|ks(t�|j|ks6t�|j|ksDt�|j|ksRt�x:|D ]2} | j| d �}
| |kr~|
g ks�t�qX|
g ksXt�qXW tr�� j | j
|g� d S )
Nr# r~ r �inet6r� rs r� r� r� r� )r" )
r# r~ r r� r� rs r� r� r� r� ) r rP r/ rQ rZ r` ra r' r r; r5 ) �procrA rP rQ rZ r` ra ZkindsZ all_kindsr" r( )r0 r) r* �
check_conn$ s.
z+TestFilters.test_combos.<locals>.check_conna4
import socket, time
s = socket.socket({family}, socket.SOCK_STREAM)
s.bind(('{addr}', 0))
s.listen(5)
with open('{testfn}', 'w') as f:
f.write(str(s.getsockname()[:2]))
[time.sleep(0.1) for x in range(100)]
a
import socket, time
s = socket.socket({family}, socket.SOCK_DGRAM)
s.bind(('{addr}', 0))
with open('{testfn}', 'w') as f:
f.write(str(s.getsockname()[:2]))
[time.sleep(0.1) for x in range(100)]
)�dirz 127.0.0.1)rP r_ rm z::1T)�deleter3 r# r~ r r� rs r� r� r5 r� r� r� )r# r~ r r� rs )r# r~ r r� r� )r# r~ r� r� r� )r# r~ r� r� r� ) r �textwrap�dedentr[ �path�basenamerl �getcwd�format�intr r �pyrun�evalr r r% r& �childrenr' rL r/ r5 r rb r rg �getattr)r0 r� Z tcp_templateZ udp_templateZtestfileZ
tcp4_templateZ
udp4_templateZ
tcp6_templateZ
udp6_templateZ tcp4_procZ tcp4_addrZ udp4_procZ udp4_addrZ tcp6_procZ tcp6_addrZ udp6_procZ udp6_addr�pr( rA r) )r0 r* � test_combos s�
zTestFilters.test_combosc C s� t � ��� tdd�}t|�t� r$dndks.t�x,|D ]$}|jttfksJt�|jt ks4t�q4W tdd�}t|�dksvt�|d jtks�t�|d jt ks�t�t� r�tdd�}t|�dks�t�|d jtks�t�|d jt ks�t�tdd�}t|�t� r�dndk�st�x2|D ]*}|jttfk�s"t�|jt
k�s
t��q
W td d�}t|�dk�sTt�|d jtk�sht�|d jt
k�s|t�t� �r�td
d�}t|�dk�s�t�|d jtk�s�t�|d jt
k�s�t�td d�}t|�t� �r�d ndk�s�t�x6|D ].}|jttfk�st�|jt t
fk�s�t��q�W t� �r�td
d�}t|�dk�sLt�x2|D ]*}|jtk�sft�|jt t
fk�sRt��qRW t �r�t �p�t
�r�tdd�}t|�dk�s�t�x2|D ]*}|jtk�s�t�|jt t
fk�s�t��q�W W d Q R X d S )Nr� )r" rO r3 rs r r� r� r� r� r~ � r� r$ � )r r+ rL r r/ rP r r rQ r r r r r
r )r0 r( rA r) r) r* �
test_count� s\
zTestFilters.test_countN)r< r= r> r� r r r� r� r) r) r) r* rz s rz z
requires root)r@ c @ s& e Zd ZdZdd� Ze� dd� �ZdS )�TestSystemWideConnectionsz Tests for net_connections().c C s� dd� }t � �n ddlm} xZ|j� D ]N\}}|dkr>t r>q&|\}}tj|�}t|�tt|��ksht �||||� q&W W d Q R X d S )Nc S s@ x:| D ]2}|j |kst�|j tkr0|j|ks0t�t|� qW d S )N)rP r/ r rQ r )r( r{ �types_rA r) r) r* r} � s
z0TestSystemWideConnections.test_it.<locals>.checkr )� conn_tmapr$ )
r �psutil._commonr� �itemsr r% r' rL �setr/ )r0 r} r� r" �groupsr{ r� r( r) r) r* �test_it� s
z!TestSystemWideConnections.test_itc s� t � �}t|�}W d Q R X g �d}g }xFt|�D ]:}| j� }|j|� tjd| �}| j|�}�j|j� q0W x|D ] }t |� qtW �fdd�t
j dd�D �} xJ�D ]B� t� fdd�| D ��|ks�t �t
j
� �}
t|
j d��|ks�t �q�W d S )N�
a" import time, os
from psutil.tests import create_sockets
with create_sockets():
with open(r'%s', 'w') as f:
f.write("hello")
[time.sleep(0.1) for x in range(100)]
c s g | ]}|j � kr|�qS r) )r5 )r6 �x)�pidsr) r* r8 s zFTestSystemWideConnections.test_multi_sockets_procs.<locals>.<listcomp>r# )r" c s g | ]}|j � kr|�qS r) )r5 )r6 r� )r5 r) r* r8 ! s )r rL �rangerl �appendr� r� r� r5 r r% r' r/ r& ) r0 ZsocksZexpected�times�fnames�_�fname�srcZsprocZsysconsr� r) )r5 r� r* �test_multi_sockets_procs� s&
z2TestSystemWideConnections.test_multi_sockets_procsN)r< r= r> rp r� r r� r) r) r) r* r� � s r� c @ s e Zd Zdd� ZdS )�TestMiscc C s� g }g }xht t�D ]\}|jd�rtt|�}t|�}|j� sBt|��t|ksNt�||ksZt�|j|� |j|� qW tr�tj tj
t r�tj d S )NZCONN_)
r� r% �
startswithr� �str�isupperr/ r� r
Z CONN_IDLEZ
CONN_BOUNDr ZCONN_DELETE_TCB)r0 �ints�strsr- �numZstr_r) r) r* � test_net_connection_constants' s
z&TestMisc.test_net_connection_constantsN)r< r= r> r� r) r) r) r* r� & s r� )6rp r[ rS r� �
contextlibr r r r r r% r r r r
r r r
r r� r Zpsutil._compatr Z psutil.testsr r r r r r r r r r r r r r r r r� �objectr! r+ rF Z xdist_groupr. r? rH rq rz rG r� r� r) r) r) r* �<module> sb ]7 g?