springsuite: planemo/lib/python3.7/site-packages/psutil/tests/test

author	guerler
date	Fri, 31 Jul 2020 00:32:28 -0400
parents
children

rev	line source
1 56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	1 #!/usr/bin/env python3
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	2
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	3 # Copyright (c) 2009, Giampaolo Rodola'. All rights reserved.
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	4 # Use of this source code is governed by a BSD-style license that can be
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	5 # found in the LICENSE file.
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	6
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	7 """Tests for net_connections() and Process.connections() APIs."""
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	8
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	9 import contextlib
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	10 import errno
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	11 import os
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	12 import socket
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	13 import textwrap
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	14 from contextlib import closing
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	15 from socket import AF_INET
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	16 from socket import AF_INET6
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	17 from socket import SOCK_DGRAM
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	18 from socket import SOCK_STREAM
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	19
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	20 import psutil
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	21 from psutil import FREEBSD
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	22 from psutil import LINUX
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	23 from psutil import MACOS
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	24 from psutil import NETBSD
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	25 from psutil import OPENBSD
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	26 from psutil import POSIX
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	27 from psutil import SUNOS
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	28 from psutil import WINDOWS
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	29 from psutil._common import supports_ipv6
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	30 from psutil._compat import PY3
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	31 from psutil.tests import AF_UNIX
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	32 from psutil.tests import bind_socket
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	33 from psutil.tests import bind_unix_socket
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	34 from psutil.tests import check_net_address
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	35 from psutil.tests import CIRRUS
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	36 from psutil.tests import create_sockets
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	37 from psutil.tests import enum
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	38 from psutil.tests import get_free_port
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	39 from psutil.tests import HAS_CONNECTIONS_UNIX
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	40 from psutil.tests import PsutilTestCase
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	41 from psutil.tests import reap_children
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	42 from psutil.tests import retry_on_failure
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	43 from psutil.tests import serialrun
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	44 from psutil.tests import skip_on_access_denied
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	45 from psutil.tests import SKIP_SYSCONS
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	46 from psutil.tests import tcp_socketpair
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	47 from psutil.tests import TRAVIS
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	48 from psutil.tests import unittest
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	49 from psutil.tests import unix_socketpair
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	50 from psutil.tests import wait_for_file
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	51
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	52
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	53 thisproc = psutil.Process()
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	54 SOCK_SEQPACKET = getattr(socket, "SOCK_SEQPACKET", object())
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	55
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	56
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	57 @serialrun
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	58 class _ConnTestCase(PsutilTestCase):
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	59
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	60 def setUp(self):
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	61 if not (NETBSD or FREEBSD):
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	62 # process opens a UNIX socket to /var/log/run.
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	63 cons = thisproc.connections(kind='all')
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	64 assert not cons, cons
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	65
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	66 def tearDown(self):
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	67 if not (FREEBSD or NETBSD):
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	68 # Make sure we closed all resources.
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	69 # NetBSD opens a UNIX socket to /var/log/run.
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	70 cons = thisproc.connections(kind='all')
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	71 assert not cons, cons
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	72
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	73 def compare_procsys_connections(self, pid, proc_cons, kind='all'):
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	74 """Given a process PID and its list of connections compare
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	75 those against system-wide connections retrieved via
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	76 psutil.net_connections.
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	77 """
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	78 try:
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	79 sys_cons = psutil.net_connections(kind=kind)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	80 except psutil.AccessDenied:
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	81 # On MACOS, system-wide connections are retrieved by iterating
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	82 # over all processes
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	83 if MACOS:
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	84 return
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	85 else:
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	86 raise
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	87 # Filter for this proc PID and exlucde PIDs from the tuple.
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	88 sys_cons = [c[:-1] for c in sys_cons if c.pid == pid]
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	89 sys_cons.sort()
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	90 proc_cons.sort()
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	91 self.assertEqual(proc_cons, sys_cons)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	92
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	93 def check_connection_ntuple(self, conn):
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	94 """Check validity of a connection namedtuple."""
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	95 def check_ntuple(conn):
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	96 has_pid = len(conn) == 7
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	97 self.assertIn(len(conn), (6, 7))
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	98 self.assertEqual(conn[0], conn.fd)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	99 self.assertEqual(conn[1], conn.family)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	100 self.assertEqual(conn[2], conn.type)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	101 self.assertEqual(conn[3], conn.laddr)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	102 self.assertEqual(conn[4], conn.raddr)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	103 self.assertEqual(conn[5], conn.status)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	104 if has_pid:
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	105 self.assertEqual(conn[6], conn.pid)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	106
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	107 def check_family(conn):
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	108 self.assertIn(conn.family, (AF_INET, AF_INET6, AF_UNIX))
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	109 if enum is not None:
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	110 assert isinstance(conn.family, enum.IntEnum), conn
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	111 else:
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	112 assert isinstance(conn.family, int), conn
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	113 if conn.family == AF_INET:
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	114 # actually try to bind the local socket; ignore IPv6
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	115 # sockets as their address might be represented as
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	116 # an IPv4-mapped-address (e.g. "::127.0.0.1")
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	117 # and that's rejected by bind()
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	118 s = socket.socket(conn.family, conn.type)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	119 with contextlib.closing(s):
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	120 try:
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	121 s.bind((conn.laddr[0], 0))
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	122 except socket.error as err:
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	123 if err.errno != errno.EADDRNOTAVAIL:
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	124 raise
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	125 elif conn.family == AF_UNIX:
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	126 self.assertEqual(conn.status, psutil.CONN_NONE)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	127
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	128 def check_type(conn):
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	129 # SOCK_SEQPACKET may happen in case of AF_UNIX socks
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	130 self.assertIn(conn.type, (SOCK_STREAM, SOCK_DGRAM, SOCK_SEQPACKET))
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	131 if enum is not None:
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	132 assert isinstance(conn.type, enum.IntEnum), conn
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	133 else:
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	134 assert isinstance(conn.type, int), conn
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	135 if conn.type == SOCK_DGRAM:
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	136 self.assertEqual(conn.status, psutil.CONN_NONE)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	137
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	138 def check_addrs(conn):
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	139 # check IP address and port sanity
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	140 for addr in (conn.laddr, conn.raddr):
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	141 if conn.family in (AF_INET, AF_INET6):
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	142 self.assertIsInstance(addr, tuple)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	143 if not addr:
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	144 continue
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	145 self.assertIsInstance(addr.port, int)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	146 assert 0 <= addr.port <= 65535, addr.port
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	147 check_net_address(addr.ip, conn.family)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	148 elif conn.family == AF_UNIX:
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	149 self.assertIsInstance(addr, str)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	150
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	151 def check_status(conn):
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	152 self.assertIsInstance(conn.status, str)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	153 valids = [getattr(psutil, x) for x in dir(psutil)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	154 if x.startswith('CONN_')]
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	155 self.assertIn(conn.status, valids)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	156 if conn.family in (AF_INET, AF_INET6) and conn.type == SOCK_STREAM:
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	157 self.assertNotEqual(conn.status, psutil.CONN_NONE)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	158 else:
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	159 self.assertEqual(conn.status, psutil.CONN_NONE)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	160
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	161 check_ntuple(conn)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	162 check_family(conn)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	163 check_type(conn)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	164 check_addrs(conn)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	165 check_status(conn)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	166
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	167
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	168 class TestBasicOperations(_ConnTestCase):
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	169
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	170 @unittest.skipIf(SKIP_SYSCONS, "requires root")
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	171 def test_system(self):
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	172 with create_sockets():
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	173 for conn in psutil.net_connections(kind='all'):
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	174 self.check_connection_ntuple(conn)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	175
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	176 def test_process(self):
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	177 with create_sockets():
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	178 for conn in psutil.Process().connections(kind='all'):
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	179 self.check_connection_ntuple(conn)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	180
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	181 def test_invalid_kind(self):
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	182 self.assertRaises(ValueError, thisproc.connections, kind='???')
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	183 self.assertRaises(ValueError, psutil.net_connections, kind='???')
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	184
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	185
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	186 @serialrun
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	187 class TestUnconnectedSockets(_ConnTestCase):
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	188 """Tests sockets which are open but not connected to anything."""
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	189
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	190 def get_conn_from_sock(self, sock):
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	191 cons = thisproc.connections(kind='all')
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	192 smap = dict([(c.fd, c) for c in cons])
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	193 if NETBSD or FREEBSD:
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	194 # NetBSD opens a UNIX socket to /var/log/run
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	195 # so there may be more connections.
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	196 return smap[sock.fileno()]
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	197 else:
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	198 self.assertEqual(len(cons), 1)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	199 if cons[0].fd != -1:
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	200 self.assertEqual(smap[sock.fileno()].fd, sock.fileno())
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	201 return cons[0]
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	202
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	203 def check_socket(self, sock):
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	204 """Given a socket, makes sure it matches the one obtained
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	205 via psutil. It assumes this process created one connection
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	206 only (the one supposed to be checked).
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	207 """
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	208 conn = self.get_conn_from_sock(sock)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	209 self.check_connection_ntuple(conn)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	210
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	211 # fd, family, type
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	212 if conn.fd != -1:
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	213 self.assertEqual(conn.fd, sock.fileno())
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	214 self.assertEqual(conn.family, sock.family)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	215 # see: http://bugs.python.org/issue30204
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	216 self.assertEqual(
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	217 conn.type, sock.getsockopt(socket.SOL_SOCKET, socket.SO_TYPE))
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	218
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	219 # local address
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	220 laddr = sock.getsockname()
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	221 if not laddr and PY3 and isinstance(laddr, bytes):
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	222 # See: http://bugs.python.org/issue30205
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	223 laddr = laddr.decode()
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	224 if sock.family == AF_INET6:
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	225 laddr = laddr[:2]
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	226 if sock.family == AF_UNIX and OPENBSD:
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	227 # No addresses are set for UNIX sockets on OpenBSD.
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	228 pass
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	229 else:
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	230 self.assertEqual(conn.laddr, laddr)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	231
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	232 # XXX Solaris can't retrieve system-wide UNIX sockets
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	233 if sock.family == AF_UNIX and HAS_CONNECTIONS_UNIX:
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	234 cons = thisproc.connections(kind='all')
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	235 self.compare_procsys_connections(os.getpid(), cons, kind='all')
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	236 return conn
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	237
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	238 def test_tcp_v4(self):
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	239 addr = ("127.0.0.1", get_free_port())
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	240 with closing(bind_socket(AF_INET, SOCK_STREAM, addr=addr)) as sock:
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	241 conn = self.check_socket(sock)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	242 assert not conn.raddr
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	243 self.assertEqual(conn.status, psutil.CONN_LISTEN)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	244
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	245 @unittest.skipIf(not supports_ipv6(), "IPv6 not supported")
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	246 def test_tcp_v6(self):
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	247 addr = ("::1", get_free_port())
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	248 with closing(bind_socket(AF_INET6, SOCK_STREAM, addr=addr)) as sock:
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	249 conn = self.check_socket(sock)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	250 assert not conn.raddr
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	251 self.assertEqual(conn.status, psutil.CONN_LISTEN)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	252
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	253 def test_udp_v4(self):
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	254 addr = ("127.0.0.1", get_free_port())
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	255 with closing(bind_socket(AF_INET, SOCK_DGRAM, addr=addr)) as sock:
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	256 conn = self.check_socket(sock)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	257 assert not conn.raddr
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	258 self.assertEqual(conn.status, psutil.CONN_NONE)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	259
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	260 @unittest.skipIf(not supports_ipv6(), "IPv6 not supported")
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	261 def test_udp_v6(self):
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	262 addr = ("::1", get_free_port())
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	263 with closing(bind_socket(AF_INET6, SOCK_DGRAM, addr=addr)) as sock:
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	264 conn = self.check_socket(sock)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	265 assert not conn.raddr
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	266 self.assertEqual(conn.status, psutil.CONN_NONE)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	267
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	268 @unittest.skipIf(not POSIX, 'POSIX only')
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	269 def test_unix_tcp(self):
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	270 testfn = self.get_testfn()
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	271 with closing(bind_unix_socket(testfn, type=SOCK_STREAM)) as sock:
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	272 conn = self.check_socket(sock)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	273 assert not conn.raddr
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	274 self.assertEqual(conn.status, psutil.CONN_NONE)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	275
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	276 @unittest.skipIf(not POSIX, 'POSIX only')
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	277 def test_unix_udp(self):
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	278 testfn = self.get_testfn()
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	279 with closing(bind_unix_socket(testfn, type=SOCK_STREAM)) as sock:
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	280 conn = self.check_socket(sock)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	281 assert not conn.raddr
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	282 self.assertEqual(conn.status, psutil.CONN_NONE)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	283
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	284
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	285 @serialrun
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	286 class TestConnectedSocket(_ConnTestCase):
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	287 """Test socket pairs which are are actually connected to
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	288 each other.
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	289 """
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	290
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	291 # On SunOS, even after we close() it, the server socket stays around
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	292 # in TIME_WAIT state.
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	293 @unittest.skipIf(SUNOS, "unreliable on SUONS")
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	294 def test_tcp(self):
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	295 addr = ("127.0.0.1", get_free_port())
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	296 assert not thisproc.connections(kind='tcp4')
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	297 server, client = tcp_socketpair(AF_INET, addr=addr)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	298 try:
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	299 cons = thisproc.connections(kind='tcp4')
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	300 self.assertEqual(len(cons), 2)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	301 self.assertEqual(cons[0].status, psutil.CONN_ESTABLISHED)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	302 self.assertEqual(cons[1].status, psutil.CONN_ESTABLISHED)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	303 # May not be fast enough to change state so it stays
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	304 # commenteed.
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	305 # client.close()
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	306 # cons = thisproc.connections(kind='all')
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	307 # self.assertEqual(len(cons), 1)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	308 # self.assertEqual(cons[0].status, psutil.CONN_CLOSE_WAIT)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	309 finally:
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	310 server.close()
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	311 client.close()
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	312
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	313 @unittest.skipIf(not POSIX, 'POSIX only')
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	314 def test_unix(self):
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	315 testfn = self.get_testfn()
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	316 server, client = unix_socketpair(testfn)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	317 try:
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	318 cons = thisproc.connections(kind='unix')
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	319 assert not (cons[0].laddr and cons[0].raddr)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	320 assert not (cons[1].laddr and cons[1].raddr)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	321 if NETBSD or FREEBSD:
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	322 # On NetBSD creating a UNIX socket will cause
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	323 # a UNIX connection to /var/run/log.
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	324 cons = [c for c in cons if c.raddr != '/var/run/log']
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	325 if CIRRUS:
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	326 cons = [c for c in cons if c.fd in
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	327 (server.fileno(), client.fileno())]
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	328 self.assertEqual(len(cons), 2, msg=cons)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	329 if LINUX or FREEBSD or SUNOS:
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	330 # remote path is never set
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	331 self.assertEqual(cons[0].raddr, "")
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	332 self.assertEqual(cons[1].raddr, "")
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	333 # one local address should though
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	334 self.assertEqual(testfn, cons[0].laddr or cons[1].laddr)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	335 elif OPENBSD:
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	336 # No addresses whatsoever here.
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	337 for addr in (cons[0].laddr, cons[0].raddr,
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	338 cons[1].laddr, cons[1].raddr):
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	339 self.assertEqual(addr, "")
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	340 else:
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	341 # On other systems either the laddr or raddr
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	342 # of both peers are set.
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	343 self.assertEqual(cons[0].laddr or cons[1].laddr, testfn)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	344 self.assertEqual(cons[0].raddr or cons[1].raddr, testfn)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	345 finally:
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	346 server.close()
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	347 client.close()
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	348
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	349
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	350 class TestFilters(_ConnTestCase):
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	351
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	352 def test_filters(self):
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	353 def check(kind, families, types):
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	354 for conn in thisproc.connections(kind=kind):
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	355 self.assertIn(conn.family, families)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	356 self.assertIn(conn.type, types)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	357 if not SKIP_SYSCONS:
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	358 for conn in psutil.net_connections(kind=kind):
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	359 self.assertIn(conn.family, families)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	360 self.assertIn(conn.type, types)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	361
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	362 with create_sockets():
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	363 check('all',
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	364 [AF_INET, AF_INET6, AF_UNIX],
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	365 [SOCK_STREAM, SOCK_DGRAM, SOCK_SEQPACKET])
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	366 check('inet',
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	367 [AF_INET, AF_INET6],
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	368 [SOCK_STREAM, SOCK_DGRAM])
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	369 check('inet4',
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	370 [AF_INET],
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	371 [SOCK_STREAM, SOCK_DGRAM])
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	372 check('tcp',
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	373 [AF_INET, AF_INET6],
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	374 [SOCK_STREAM])
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	375 check('tcp4',
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	376 [AF_INET],
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	377 [SOCK_STREAM])
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	378 check('tcp6',
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	379 [AF_INET6],
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	380 [SOCK_STREAM])
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	381 check('udp',
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	382 [AF_INET, AF_INET6],
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	383 [SOCK_DGRAM])
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	384 check('udp4',
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	385 [AF_INET],
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	386 [SOCK_DGRAM])
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	387 check('udp6',
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	388 [AF_INET6],
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	389 [SOCK_DGRAM])
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	390 if HAS_CONNECTIONS_UNIX:
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	391 check('unix',
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	392 [AF_UNIX],
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	393 [SOCK_STREAM, SOCK_DGRAM, SOCK_SEQPACKET])
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	394
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	395 @skip_on_access_denied(only_if=MACOS)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	396 def test_combos(self):
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	397 reap_children()
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	398
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	399 def check_conn(proc, conn, family, type, laddr, raddr, status, kinds):
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	400 all_kinds = ("all", "inet", "inet4", "inet6", "tcp", "tcp4",
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	401 "tcp6", "udp", "udp4", "udp6")
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	402 self.check_connection_ntuple(conn)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	403 self.assertEqual(conn.family, family)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	404 self.assertEqual(conn.type, type)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	405 self.assertEqual(conn.laddr, laddr)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	406 self.assertEqual(conn.raddr, raddr)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	407 self.assertEqual(conn.status, status)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	408 for kind in all_kinds:
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	409 cons = proc.connections(kind=kind)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	410 if kind in kinds:
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	411 assert cons
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	412 else:
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	413 assert not cons, cons
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	414 # compare against system-wide connections
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	415 # XXX Solaris can't retrieve system-wide UNIX
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	416 # sockets.
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	417 if HAS_CONNECTIONS_UNIX:
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	418 self.compare_procsys_connections(proc.pid, [conn])
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	419
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	420 tcp_template = textwrap.dedent("""
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	421 import socket, time
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	422 s = socket.socket({family}, socket.SOCK_STREAM)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	423 s.bind(('{addr}', 0))
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	424 s.listen(5)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	425 with open('{testfn}', 'w') as f:
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	426 f.write(str(s.getsockname()[:2]))
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	427 time.sleep(60)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	428 """)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	429
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	430 udp_template = textwrap.dedent("""
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	431 import socket, time
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	432 s = socket.socket({family}, socket.SOCK_DGRAM)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	433 s.bind(('{addr}', 0))
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	434 with open('{testfn}', 'w') as f:
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	435 f.write(str(s.getsockname()[:2]))
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	436 time.sleep(60)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	437 """)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	438
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	439 # must be relative on Windows
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	440 testfile = os.path.basename(self.get_testfn(dir=os.getcwd()))
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	441 tcp4_template = tcp_template.format(
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	442 family=int(AF_INET), addr="127.0.0.1", testfn=testfile)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	443 udp4_template = udp_template.format(
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	444 family=int(AF_INET), addr="127.0.0.1", testfn=testfile)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	445 tcp6_template = tcp_template.format(
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	446 family=int(AF_INET6), addr="::1", testfn=testfile)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	447 udp6_template = udp_template.format(
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	448 family=int(AF_INET6), addr="::1", testfn=testfile)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	449
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	450 # launch various subprocess instantiating a socket of various
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	451 # families and types to enrich psutil results
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	452 tcp4_proc = self.pyrun(tcp4_template)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	453 tcp4_addr = eval(wait_for_file(testfile, delete=True))
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	454 udp4_proc = self.pyrun(udp4_template)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	455 udp4_addr = eval(wait_for_file(testfile, delete=True))
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	456 if supports_ipv6():
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	457 tcp6_proc = self.pyrun(tcp6_template)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	458 tcp6_addr = eval(wait_for_file(testfile, delete=True))
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	459 udp6_proc = self.pyrun(udp6_template)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	460 udp6_addr = eval(wait_for_file(testfile, delete=True))
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	461 else:
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	462 tcp6_proc = None
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	463 udp6_proc = None
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	464 tcp6_addr = None
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	465 udp6_addr = None
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	466
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	467 for p in thisproc.children():
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	468 cons = p.connections()
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	469 self.assertEqual(len(cons), 1)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	470 for conn in cons:
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	471 # TCP v4
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	472 if p.pid == tcp4_proc.pid:
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	473 check_conn(p, conn, AF_INET, SOCK_STREAM, tcp4_addr, (),
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	474 psutil.CONN_LISTEN,
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	475 ("all", "inet", "inet4", "tcp", "tcp4"))
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	476 # UDP v4
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	477 elif p.pid == udp4_proc.pid:
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	478 check_conn(p, conn, AF_INET, SOCK_DGRAM, udp4_addr, (),
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	479 psutil.CONN_NONE,
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	480 ("all", "inet", "inet4", "udp", "udp4"))
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	481 # TCP v6
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	482 elif p.pid == getattr(tcp6_proc, "pid", None):
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	483 check_conn(p, conn, AF_INET6, SOCK_STREAM, tcp6_addr, (),
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	484 psutil.CONN_LISTEN,
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	485 ("all", "inet", "inet6", "tcp", "tcp6"))
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	486 # UDP v6
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	487 elif p.pid == getattr(udp6_proc, "pid", None):
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	488 check_conn(p, conn, AF_INET6, SOCK_DGRAM, udp6_addr, (),
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	489 psutil.CONN_NONE,
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	490 ("all", "inet", "inet6", "udp", "udp6"))
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	491
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	492 def test_count(self):
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	493 with create_sockets():
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	494 # tcp
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	495 cons = thisproc.connections(kind='tcp')
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	496 self.assertEqual(len(cons), 2 if supports_ipv6() else 1)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	497 for conn in cons:
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	498 self.assertIn(conn.family, (AF_INET, AF_INET6))
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	499 self.assertEqual(conn.type, SOCK_STREAM)
56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1" guerler parents: diff changeset	500 # tcp4

1

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

1 #!/usr/bin/env python3

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

2

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

4 # Use of this source code is governed by a BSD-style license that can be

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

5 # found in the LICENSE file.

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

6

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

7 """Tests for net_connections() and Process.connections() APIs."""

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

8

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

9 import contextlib

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

10 import errno

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

11 import os

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

12 import socket

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

13 import textwrap

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

14 from contextlib import closing

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

15 from socket import AF_INET

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

16 from socket import AF_INET6

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

17 from socket import SOCK_DGRAM

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

18 from socket import SOCK_STREAM

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

19

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

20 import psutil

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

21 from psutil import FREEBSD

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

22 from psutil import LINUX

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

23 from psutil import MACOS

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

24 from psutil import NETBSD

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

25 from psutil import OPENBSD

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

26 from psutil import POSIX

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

27 from psutil import SUNOS

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

28 from psutil import WINDOWS

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

29 from psutil._common import supports_ipv6

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

30 from psutil._compat import PY3

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

31 from psutil.tests import AF_UNIX

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

32 from psutil.tests import bind_socket

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

33 from psutil.tests import bind_unix_socket

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

34 from psutil.tests import check_net_address

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

35 from psutil.tests import CIRRUS

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

36 from psutil.tests import create_sockets

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

37 from psutil.tests import enum

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

38 from psutil.tests import get_free_port

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

39 from psutil.tests import HAS_CONNECTIONS_UNIX

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

40 from psutil.tests import PsutilTestCase

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

41 from psutil.tests import reap_children

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

42 from psutil.tests import retry_on_failure

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

43 from psutil.tests import serialrun

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

44 from psutil.tests import skip_on_access_denied

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

45 from psutil.tests import SKIP_SYSCONS

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

46 from psutil.tests import tcp_socketpair

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

47 from psutil.tests import TRAVIS

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

48 from psutil.tests import unittest

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

49 from psutil.tests import unix_socketpair

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

50 from psutil.tests import wait_for_file

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

51

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

52

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

53 thisproc = psutil.Process()

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

54 SOCK_SEQPACKET = getattr(socket, "SOCK_SEQPACKET", object())

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

55

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

56

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

57 @serialrun

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

58 class _ConnTestCase(PsutilTestCase):

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

59

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

60 def setUp(self):

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

61 if not (NETBSD or FREEBSD):

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

62 # process opens a UNIX socket to /var/log/run.

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

63 cons = thisproc.connections(kind='all')

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

64 assert not cons, cons

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

65

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

66 def tearDown(self):

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

67 if not (FREEBSD or NETBSD):

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

68 # Make sure we closed all resources.

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

69 # NetBSD opens a UNIX socket to /var/log/run.

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

70 cons = thisproc.connections(kind='all')

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

71 assert not cons, cons

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

72

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

73 def compare_procsys_connections(self, pid, proc_cons, kind='all'):

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

74 """Given a process PID and its list of connections compare

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

75 those against system-wide connections retrieved via

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

76 psutil.net_connections.

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

77 """

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

78 try:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

79 sys_cons = psutil.net_connections(kind=kind)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

80 except psutil.AccessDenied:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

81 # On MACOS, system-wide connections are retrieved by iterating

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

82 # over all processes

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

83 if MACOS:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

84 return

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

85 else:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

86 raise

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

87 # Filter for this proc PID and exlucde PIDs from the tuple.

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

88 sys_cons = [c[:-1] for c in sys_cons if c.pid == pid]

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

89 sys_cons.sort()

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

90 proc_cons.sort()

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

91 self.assertEqual(proc_cons, sys_cons)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

92

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

93 def check_connection_ntuple(self, conn):

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

94 """Check validity of a connection namedtuple."""

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

95 def check_ntuple(conn):

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

96 has_pid = len(conn) == 7

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

97 self.assertIn(len(conn), (6, 7))

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

98 self.assertEqual(conn[0], conn.fd)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

99 self.assertEqual(conn[1], conn.family)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

100 self.assertEqual(conn[2], conn.type)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

101 self.assertEqual(conn[3], conn.laddr)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

102 self.assertEqual(conn[4], conn.raddr)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

103 self.assertEqual(conn[5], conn.status)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

104 if has_pid:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

105 self.assertEqual(conn[6], conn.pid)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

106

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

107 def check_family(conn):

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

108 self.assertIn(conn.family, (AF_INET, AF_INET6, AF_UNIX))

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

109 if enum is not None:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

110 assert isinstance(conn.family, enum.IntEnum), conn

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

111 else:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

112 assert isinstance(conn.family, int), conn

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

113 if conn.family == AF_INET:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

114 # actually try to bind the local socket; ignore IPv6

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

115 # sockets as their address might be represented as

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

116 # an IPv4-mapped-address (e.g. "::127.0.0.1")

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

117 # and that's rejected by bind()

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

118 s = socket.socket(conn.family, conn.type)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

119 with contextlib.closing(s):

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

120 try:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

121 s.bind((conn.laddr[0], 0))

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

122 except socket.error as err:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

123 if err.errno != errno.EADDRNOTAVAIL:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

124 raise

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

125 elif conn.family == AF_UNIX:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

126 self.assertEqual(conn.status, psutil.CONN_NONE)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

127

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

128 def check_type(conn):

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

129 # SOCK_SEQPACKET may happen in case of AF_UNIX socks

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

130 self.assertIn(conn.type, (SOCK_STREAM, SOCK_DGRAM, SOCK_SEQPACKET))

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

131 if enum is not None:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

132 assert isinstance(conn.type, enum.IntEnum), conn

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

133 else:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

134 assert isinstance(conn.type, int), conn

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

135 if conn.type == SOCK_DGRAM:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

136 self.assertEqual(conn.status, psutil.CONN_NONE)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

137

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

138 def check_addrs(conn):

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

139 # check IP address and port sanity

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

140 for addr in (conn.laddr, conn.raddr):

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

141 if conn.family in (AF_INET, AF_INET6):

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

142 self.assertIsInstance(addr, tuple)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

143 if not addr:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

144 continue

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

145 self.assertIsInstance(addr.port, int)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

146 assert 0 <= addr.port <= 65535, addr.port

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

147 check_net_address(addr.ip, conn.family)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

148 elif conn.family == AF_UNIX:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

149 self.assertIsInstance(addr, str)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

150

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

151 def check_status(conn):

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

152 self.assertIsInstance(conn.status, str)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

153 valids = [getattr(psutil, x) for x in dir(psutil)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

154 if x.startswith('CONN_')]

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

155 self.assertIn(conn.status, valids)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

156 if conn.family in (AF_INET, AF_INET6) and conn.type == SOCK_STREAM:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

157 self.assertNotEqual(conn.status, psutil.CONN_NONE)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

158 else:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

159 self.assertEqual(conn.status, psutil.CONN_NONE)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

160

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

161 check_ntuple(conn)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

162 check_family(conn)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

163 check_type(conn)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

164 check_addrs(conn)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

165 check_status(conn)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

166

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

167

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

168 class TestBasicOperations(_ConnTestCase):

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

169

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

170 @unittest.skipIf(SKIP_SYSCONS, "requires root")

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

171 def test_system(self):

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

172 with create_sockets():

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

173 for conn in psutil.net_connections(kind='all'):

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

174 self.check_connection_ntuple(conn)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

175

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

176 def test_process(self):

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

177 with create_sockets():

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

178 for conn in psutil.Process().connections(kind='all'):

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

179 self.check_connection_ntuple(conn)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

180

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

181 def test_invalid_kind(self):

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

182 self.assertRaises(ValueError, thisproc.connections, kind='???')

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

183 self.assertRaises(ValueError, psutil.net_connections, kind='???')

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

184

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

185

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

186 @serialrun

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

187 class TestUnconnectedSockets(_ConnTestCase):

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

188 """Tests sockets which are open but not connected to anything."""

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

189

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

190 def get_conn_from_sock(self, sock):

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

191 cons = thisproc.connections(kind='all')

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

192 smap = dict([(c.fd, c) for c in cons])

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

193 if NETBSD or FREEBSD:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

194 # NetBSD opens a UNIX socket to /var/log/run

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

195 # so there may be more connections.

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

196 return smap[sock.fileno()]

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

197 else:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

198 self.assertEqual(len(cons), 1)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

199 if cons[0].fd != -1:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

200 self.assertEqual(smap[sock.fileno()].fd, sock.fileno())

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

201 return cons[0]

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

202

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

203 def check_socket(self, sock):

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

204 """Given a socket, makes sure it matches the one obtained

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

205 via psutil. It assumes this process created one connection

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

206 only (the one supposed to be checked).

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

207 """

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

208 conn = self.get_conn_from_sock(sock)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

209 self.check_connection_ntuple(conn)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

210

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

211 # fd, family, type

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

212 if conn.fd != -1:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

213 self.assertEqual(conn.fd, sock.fileno())

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

214 self.assertEqual(conn.family, sock.family)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

215 # see: http://bugs.python.org/issue30204

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

216 self.assertEqual(

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

217 conn.type, sock.getsockopt(socket.SOL_SOCKET, socket.SO_TYPE))

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

218

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

219 # local address

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

220 laddr = sock.getsockname()

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

221 if not laddr and PY3 and isinstance(laddr, bytes):

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

222 # See: http://bugs.python.org/issue30205

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

223 laddr = laddr.decode()

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

224 if sock.family == AF_INET6:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

225 laddr = laddr[:2]

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

226 if sock.family == AF_UNIX and OPENBSD:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

227 # No addresses are set for UNIX sockets on OpenBSD.

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

228 pass

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

229 else:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

230 self.assertEqual(conn.laddr, laddr)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

231

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

232 # XXX Solaris can't retrieve system-wide UNIX sockets

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

233 if sock.family == AF_UNIX and HAS_CONNECTIONS_UNIX:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

234 cons = thisproc.connections(kind='all')

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

235 self.compare_procsys_connections(os.getpid(), cons, kind='all')

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

236 return conn

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

237

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

238 def test_tcp_v4(self):

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

239 addr = ("127.0.0.1", get_free_port())

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

240 with closing(bind_socket(AF_INET, SOCK_STREAM, addr=addr)) as sock:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

241 conn = self.check_socket(sock)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

242 assert not conn.raddr

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

243 self.assertEqual(conn.status, psutil.CONN_LISTEN)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

244

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

245 @unittest.skipIf(not supports_ipv6(), "IPv6 not supported")

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

246 def test_tcp_v6(self):

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

247 addr = ("::1", get_free_port())

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

248 with closing(bind_socket(AF_INET6, SOCK_STREAM, addr=addr)) as sock:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

249 conn = self.check_socket(sock)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

250 assert not conn.raddr

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

251 self.assertEqual(conn.status, psutil.CONN_LISTEN)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

252

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

253 def test_udp_v4(self):

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

254 addr = ("127.0.0.1", get_free_port())

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

255 with closing(bind_socket(AF_INET, SOCK_DGRAM, addr=addr)) as sock:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

256 conn = self.check_socket(sock)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

257 assert not conn.raddr

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

258 self.assertEqual(conn.status, psutil.CONN_NONE)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

259

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

260 @unittest.skipIf(not supports_ipv6(), "IPv6 not supported")

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

261 def test_udp_v6(self):

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

262 addr = ("::1", get_free_port())

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

263 with closing(bind_socket(AF_INET6, SOCK_DGRAM, addr=addr)) as sock:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

264 conn = self.check_socket(sock)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

265 assert not conn.raddr

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

266 self.assertEqual(conn.status, psutil.CONN_NONE)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

267

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

268 @unittest.skipIf(not POSIX, 'POSIX only')

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

269 def test_unix_tcp(self):

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

270 testfn = self.get_testfn()

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

271 with closing(bind_unix_socket(testfn, type=SOCK_STREAM)) as sock:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

272 conn = self.check_socket(sock)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

273 assert not conn.raddr

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

274 self.assertEqual(conn.status, psutil.CONN_NONE)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

275

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

276 @unittest.skipIf(not POSIX, 'POSIX only')

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

277 def test_unix_udp(self):

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

278 testfn = self.get_testfn()

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

279 with closing(bind_unix_socket(testfn, type=SOCK_STREAM)) as sock:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

280 conn = self.check_socket(sock)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

281 assert not conn.raddr

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

282 self.assertEqual(conn.status, psutil.CONN_NONE)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

283

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

284

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

285 @serialrun

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

286 class TestConnectedSocket(_ConnTestCase):

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

287 """Test socket pairs which are are actually connected to

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

288 each other.

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

289 """

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

290

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

291 # On SunOS, even after we close() it, the server socket stays around

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

292 # in TIME_WAIT state.

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

293 @unittest.skipIf(SUNOS, "unreliable on SUONS")

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

294 def test_tcp(self):

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

295 addr = ("127.0.0.1", get_free_port())

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

296 assert not thisproc.connections(kind='tcp4')

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

297 server, client = tcp_socketpair(AF_INET, addr=addr)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

298 try:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

299 cons = thisproc.connections(kind='tcp4')

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

300 self.assertEqual(len(cons), 2)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

301 self.assertEqual(cons[0].status, psutil.CONN_ESTABLISHED)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

302 self.assertEqual(cons[1].status, psutil.CONN_ESTABLISHED)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

303 # May not be fast enough to change state so it stays

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

304 # commenteed.

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

305 # client.close()

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

306 # cons = thisproc.connections(kind='all')

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

307 # self.assertEqual(len(cons), 1)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

308 # self.assertEqual(cons[0].status, psutil.CONN_CLOSE_WAIT)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

309 finally:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

310 server.close()

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

311 client.close()

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

312

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

313 @unittest.skipIf(not POSIX, 'POSIX only')

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

314 def test_unix(self):

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

315 testfn = self.get_testfn()

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

316 server, client = unix_socketpair(testfn)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

317 try:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

318 cons = thisproc.connections(kind='unix')

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

319 assert not (cons[0].laddr and cons[0].raddr)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

320 assert not (cons[1].laddr and cons[1].raddr)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

321 if NETBSD or FREEBSD:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

322 # On NetBSD creating a UNIX socket will cause

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

323 # a UNIX connection to /var/run/log.

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

324 cons = [c for c in cons if c.raddr != '/var/run/log']

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

325 if CIRRUS:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

326 cons = [c for c in cons if c.fd in

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

327 (server.fileno(), client.fileno())]

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

328 self.assertEqual(len(cons), 2, msg=cons)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

329 if LINUX or FREEBSD or SUNOS:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

330 # remote path is never set

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

331 self.assertEqual(cons[0].raddr, "")

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

332 self.assertEqual(cons[1].raddr, "")

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

333 # one local address should though

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

334 self.assertEqual(testfn, cons[0].laddr or cons[1].laddr)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

335 elif OPENBSD:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

336 # No addresses whatsoever here.

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

337 for addr in (cons[0].laddr, cons[0].raddr,

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

338 cons[1].laddr, cons[1].raddr):

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

339 self.assertEqual(addr, "")

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

340 else:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

341 # On other systems either the laddr or raddr

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

342 # of both peers are set.

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

343 self.assertEqual(cons[0].laddr or cons[1].laddr, testfn)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

344 self.assertEqual(cons[0].raddr or cons[1].raddr, testfn)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

345 finally:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

346 server.close()

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

347 client.close()

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

348

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

349

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

350 class TestFilters(_ConnTestCase):

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

351

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

352 def test_filters(self):

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

353 def check(kind, families, types):

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

354 for conn in thisproc.connections(kind=kind):

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

355 self.assertIn(conn.family, families)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

356 self.assertIn(conn.type, types)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

357 if not SKIP_SYSCONS:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

358 for conn in psutil.net_connections(kind=kind):

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

359 self.assertIn(conn.family, families)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

360 self.assertIn(conn.type, types)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

361

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

362 with create_sockets():

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

363 check('all',

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

364 [AF_INET, AF_INET6, AF_UNIX],

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

365 [SOCK_STREAM, SOCK_DGRAM, SOCK_SEQPACKET])

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

366 check('inet',

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

367 [AF_INET, AF_INET6],

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

368 [SOCK_STREAM, SOCK_DGRAM])

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

369 check('inet4',

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

370 [AF_INET],

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

371 [SOCK_STREAM, SOCK_DGRAM])

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

372 check('tcp',

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

373 [AF_INET, AF_INET6],

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

374 [SOCK_STREAM])

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

375 check('tcp4',

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

376 [AF_INET],

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

377 [SOCK_STREAM])

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

378 check('tcp6',

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

379 [AF_INET6],

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

380 [SOCK_STREAM])

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

381 check('udp',

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

382 [AF_INET, AF_INET6],

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

383 [SOCK_DGRAM])

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

384 check('udp4',

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

385 [AF_INET],

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

386 [SOCK_DGRAM])

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

387 check('udp6',

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

388 [AF_INET6],

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

389 [SOCK_DGRAM])

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

390 if HAS_CONNECTIONS_UNIX:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

391 check('unix',

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

392 [AF_UNIX],

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

393 [SOCK_STREAM, SOCK_DGRAM, SOCK_SEQPACKET])

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

394

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

395 @skip_on_access_denied(only_if=MACOS)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

396 def test_combos(self):

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

397 reap_children()

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

398

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

399 def check_conn(proc, conn, family, type, laddr, raddr, status, kinds):

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

400 all_kinds = ("all", "inet", "inet4", "inet6", "tcp", "tcp4",

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

401 "tcp6", "udp", "udp4", "udp6")

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

402 self.check_connection_ntuple(conn)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

403 self.assertEqual(conn.family, family)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

404 self.assertEqual(conn.type, type)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

405 self.assertEqual(conn.laddr, laddr)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

406 self.assertEqual(conn.raddr, raddr)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

407 self.assertEqual(conn.status, status)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

408 for kind in all_kinds:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

409 cons = proc.connections(kind=kind)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

410 if kind in kinds:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

411 assert cons

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

412 else:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

413 assert not cons, cons

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

414 # compare against system-wide connections

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

415 # XXX Solaris can't retrieve system-wide UNIX

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

416 # sockets.

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

417 if HAS_CONNECTIONS_UNIX:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

418 self.compare_procsys_connections(proc.pid, [conn])

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

419

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

420 tcp_template = textwrap.dedent("""

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

421 import socket, time

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

422 s = socket.socket({family}, socket.SOCK_STREAM)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

423 s.bind(('{addr}', 0))

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

424 s.listen(5)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

425 with open('{testfn}', 'w') as f:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

426 f.write(str(s.getsockname()[:2]))

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

427 time.sleep(60)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

428 """)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

429

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

430 udp_template = textwrap.dedent("""

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

431 import socket, time

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

432 s = socket.socket({family}, socket.SOCK_DGRAM)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

433 s.bind(('{addr}', 0))

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

434 with open('{testfn}', 'w') as f:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

435 f.write(str(s.getsockname()[:2]))

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

436 time.sleep(60)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

437 """)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

438

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

439 # must be relative on Windows

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

440 testfile = os.path.basename(self.get_testfn(dir=os.getcwd()))

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

441 tcp4_template = tcp_template.format(

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

442 family=int(AF_INET), addr="127.0.0.1", testfn=testfile)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

443 udp4_template = udp_template.format(

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

444 family=int(AF_INET), addr="127.0.0.1", testfn=testfile)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

445 tcp6_template = tcp_template.format(

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

446 family=int(AF_INET6), addr="::1", testfn=testfile)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

447 udp6_template = udp_template.format(

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

448 family=int(AF_INET6), addr="::1", testfn=testfile)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

449

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

450 # launch various subprocess instantiating a socket of various

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

451 # families and types to enrich psutil results

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

452 tcp4_proc = self.pyrun(tcp4_template)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

453 tcp4_addr = eval(wait_for_file(testfile, delete=True))

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

454 udp4_proc = self.pyrun(udp4_template)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

455 udp4_addr = eval(wait_for_file(testfile, delete=True))

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

456 if supports_ipv6():

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

457 tcp6_proc = self.pyrun(tcp6_template)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

458 tcp6_addr = eval(wait_for_file(testfile, delete=True))

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

459 udp6_proc = self.pyrun(udp6_template)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

460 udp6_addr = eval(wait_for_file(testfile, delete=True))

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

461 else:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

462 tcp6_proc = None

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

463 udp6_proc = None

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

464 tcp6_addr = None

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

465 udp6_addr = None

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

466

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

467 for p in thisproc.children():

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

468 cons = p.connections()

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

469 self.assertEqual(len(cons), 1)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

470 for conn in cons:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

471 # TCP v4

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

472 if p.pid == tcp4_proc.pid:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

473 check_conn(p, conn, AF_INET, SOCK_STREAM, tcp4_addr, (),

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

474 psutil.CONN_LISTEN,

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

475 ("all", "inet", "inet4", "tcp", "tcp4"))

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

476 # UDP v4

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

477 elif p.pid == udp4_proc.pid:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

478 check_conn(p, conn, AF_INET, SOCK_DGRAM, udp4_addr, (),

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

479 psutil.CONN_NONE,

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

480 ("all", "inet", "inet4", "udp", "udp4"))

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

481 # TCP v6

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

482 elif p.pid == getattr(tcp6_proc, "pid", None):

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

483 check_conn(p, conn, AF_INET6, SOCK_STREAM, tcp6_addr, (),

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

484 psutil.CONN_LISTEN,

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

485 ("all", "inet", "inet6", "tcp", "tcp6"))

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

486 # UDP v6

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

487 elif p.pid == getattr(udp6_proc, "pid", None):

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

488 check_conn(p, conn, AF_INET6, SOCK_DGRAM, udp6_addr, (),

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

489 psutil.CONN_NONE,

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

490 ("all", "inet", "inet6", "udp", "udp6"))

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

491

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

492 def test_count(self):

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

493 with create_sockets():

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

494 # tcp

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

495 cons = thisproc.connections(kind='tcp')

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

496 self.assertEqual(len(cons), 2 if supports_ipv6() else 1)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

497 for conn in cons:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

498 self.assertIn(conn.family, (AF_INET, AF_INET6))

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

499 self.assertEqual(conn.type, SOCK_STREAM)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

500 # tcp4

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

501 cons = thisproc.connections(kind='tcp4')

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

502 self.assertEqual(len(cons), 1)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

503 self.assertEqual(cons[0].family, AF_INET)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

504 self.assertEqual(cons[0].type, SOCK_STREAM)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

505 # tcp6

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

506 if supports_ipv6():

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

507 cons = thisproc.connections(kind='tcp6')

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

508 self.assertEqual(len(cons), 1)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

509 self.assertEqual(cons[0].family, AF_INET6)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

510 self.assertEqual(cons[0].type, SOCK_STREAM)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

511 # udp

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

512 cons = thisproc.connections(kind='udp')

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

513 self.assertEqual(len(cons), 2 if supports_ipv6() else 1)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

514 for conn in cons:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

515 self.assertIn(conn.family, (AF_INET, AF_INET6))

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

516 self.assertEqual(conn.type, SOCK_DGRAM)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

517 # udp4

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

518 cons = thisproc.connections(kind='udp4')

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

519 self.assertEqual(len(cons), 1)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

520 self.assertEqual(cons[0].family, AF_INET)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

521 self.assertEqual(cons[0].type, SOCK_DGRAM)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

522 # udp6

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

523 if supports_ipv6():

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

524 cons = thisproc.connections(kind='udp6')

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

525 self.assertEqual(len(cons), 1)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

526 self.assertEqual(cons[0].family, AF_INET6)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

527 self.assertEqual(cons[0].type, SOCK_DGRAM)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

528 # inet

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

529 cons = thisproc.connections(kind='inet')

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

530 self.assertEqual(len(cons), 4 if supports_ipv6() else 2)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

531 for conn in cons:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

532 self.assertIn(conn.family, (AF_INET, AF_INET6))

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

533 self.assertIn(conn.type, (SOCK_STREAM, SOCK_DGRAM))

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

534 # inet6

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

535 if supports_ipv6():

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

536 cons = thisproc.connections(kind='inet6')

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

537 self.assertEqual(len(cons), 2)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

538 for conn in cons:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

539 self.assertEqual(conn.family, AF_INET6)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

540 self.assertIn(conn.type, (SOCK_STREAM, SOCK_DGRAM))

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

541 # Skipped on BSD becayse by default the Python process

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

542 # creates a UNIX socket to '/var/run/log'.

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

543 if HAS_CONNECTIONS_UNIX and not (FREEBSD or NETBSD):

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

544 cons = thisproc.connections(kind='unix')

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

545 self.assertEqual(len(cons), 3)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

546 for conn in cons:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

547 self.assertEqual(conn.family, AF_UNIX)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

548 self.assertIn(conn.type, (SOCK_STREAM, SOCK_DGRAM))

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

549

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

550

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

551 @unittest.skipIf(SKIP_SYSCONS, "requires root")

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

552 class TestSystemWideConnections(_ConnTestCase):

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

553 """Tests for net_connections()."""

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

554

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

555 def test_it(self):

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

556 def check(cons, families, types_):

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

557 for conn in cons:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

558 self.assertIn(conn.family, families, msg=conn)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

559 if conn.family != AF_UNIX:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

560 self.assertIn(conn.type, types_, msg=conn)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

561 self.check_connection_ntuple(conn)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

562

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

563 with create_sockets():

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

564 from psutil._common import conn_tmap

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

565 for kind, groups in conn_tmap.items():

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

566 # XXX: SunOS does not retrieve UNIX sockets.

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

567 if kind == 'unix' and not HAS_CONNECTIONS_UNIX:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

568 continue

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

569 families, types_ = groups

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

570 cons = psutil.net_connections(kind)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

571 self.assertEqual(len(cons), len(set(cons)))

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

572 check(cons, families, types_)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

573

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

574 # See: https://travis-ci.org/giampaolo/psutil/jobs/237566297

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

575 @unittest.skipIf(MACOS and TRAVIS, "unreliable on MACOS + TRAVIS")

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

576 @retry_on_failure()

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

577 def test_multi_sockets_procs(self):

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

578 # Creates multiple sub processes, each creating different

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

579 # sockets. For each process check that proc.connections()

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

580 # and net_connections() return the same results.

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

581 # This is done mainly to check whether net_connections()'s

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

582 # pid is properly set, see:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

583 # https://github.com/giampaolo/psutil/issues/1013

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

584 with create_sockets() as socks:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

585 expected = len(socks)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

586 pids = []

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

587 times = 10

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

588 fnames = []

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

589 for i in range(times):

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

590 fname = self.get_testfn()

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

591 fnames.append(fname)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

592 src = textwrap.dedent("""\

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

593 import time, os

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

594 from psutil.tests import create_sockets

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

595 with create_sockets():

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

596 with open(r'%s', 'w') as f:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

597 f.write("hello")

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

598 time.sleep(60)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

599 """ % fname)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

600 sproc = self.pyrun(src)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

601 pids.append(sproc.pid)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

602

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

603 # sync

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

604 for fname in fnames:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

605 wait_for_file(fname)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

606

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

607 syscons = [x for x in psutil.net_connections(kind='all') if x.pid

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

608 in pids]

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

609 for pid in pids:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

610 self.assertEqual(len([x for x in syscons if x.pid == pid]),

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

611 expected)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

612 p = psutil.Process(pid)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

613 self.assertEqual(len(p.connections('all')), expected)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

614

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

615

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

616 class TestMisc(PsutilTestCase):

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

617

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

618 def test_connection_constants(self):

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

619 ints = []

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

620 strs = []

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

621 for name in dir(psutil):

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

622 if name.startswith('CONN_'):

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

623 num = getattr(psutil, name)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

624 str_ = str(num)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

625 assert str_.isupper(), str_

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

626 self.assertNotIn(str, strs)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

627 self.assertNotIn(num, ints)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

628 ints.append(num)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

629 strs.append(str_)

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

630 if SUNOS:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

631 psutil.CONN_IDLE

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

632 psutil.CONN_BOUND

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

633 if WINDOWS:

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

634 psutil.CONN_DELETE_TCB

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

635

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

636

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

637 if __name__ == '__main__':

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

638 from psutil.tests.runner import run_from_name

56ad4e20f292 "planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

guerler

parents:

diff changeset

639 run_from_name(__file__)

Mercurial > repos > guerler > springsuite

annotate planemo/lib/python3.7/site-packages/psutil/tests/test_connections.py @ 1:56ad4e20f292 draft