blob: 9b6d18448a1ea679634a2ea99c6774ca6fad6a6b [file] [log] [blame]
Benjamin Peterson90f5ba52010-03-11 22:53:45 +00001#!/usr/bin/env python3
Barry Warsawcf3d4b51997-01-03 20:03:32 +00002
Guido van Rossum24e4af82002-06-12 19:18:08 +00003import unittest
Benjamin Petersonee8712c2008-05-20 21:35:26 +00004from test import support
Barry Warsawcf3d4b51997-01-03 20:03:32 +00005
Christian Heimes5e696852008-04-09 08:37:03 +00006import errno
Gregory P. Smithaafdca82010-01-04 04:50:36 +00007import io
Barry Warsawcf3d4b51997-01-03 20:03:32 +00008import socket
Guido van Rossum24e4af82002-06-12 19:18:08 +00009import select
Nick Coghlan96fe56a2011-08-22 11:55:57 +100010import tempfile
11import _testcapi
Christian Heimesbbe741d2008-03-28 10:53:29 +000012import time
13import traceback
Alexandre Vassalottif260e442008-05-11 19:59:59 +000014import queue
Jack Jansen522e7692002-09-06 21:57:50 +000015import sys
Christian Heimesfaf2f632008-01-06 16:59:19 +000016import os
17import array
Antoine Pitrouacd0fda2010-10-23 17:01:44 +000018import platform
Antoine Pitrou4b92b5f2010-09-07 21:05:49 +000019import contextlib
Raymond Hettinger027bb632004-05-31 03:09:25 +000020from weakref import proxy
Thomas Wouters0e3f5912006-08-11 14:57:12 +000021import signal
Antoine Pitrou6d7df632010-09-27 17:52:25 +000022import math
Antoine Pitrou6d58d642011-03-20 23:56:36 +010023import pickle
Charles-François Natali47413c12011-10-06 19:47:44 +020024import struct
Antoine Pitroub1c54962010-10-14 15:05:38 +000025try:
26 import fcntl
27except ImportError:
28 fcntl = False
Kristján Valur Jónsson10f383a2012-04-07 11:23:31 +000029try:
30 import multiprocessing
31except ImportError:
32 multiprocessing = False
Barry Warsawcf3d4b51997-01-03 20:03:32 +000033
Benjamin Petersonee8712c2008-05-20 21:35:26 +000034HOST = support.HOST
Marc-André Lemburg8f36af72011-02-25 15:42:01 +000035MSG = 'Michael Gilfix was here\u1234\r\n'.encode('utf-8') ## test unicode string and carriage return
Barry Warsawcf3d4b51997-01-03 20:03:32 +000036
Victor Stinner45df8202010-04-28 22:31:17 +000037try:
38 import _thread as thread
39 import threading
40except ImportError:
41 thread = None
42 threading = None
43
Charles-François Natali47413c12011-10-06 19:47:44 +020044def _have_socket_can():
45 """Check whether CAN sockets are supported on this host."""
46 try:
47 s = socket.socket(socket.PF_CAN, socket.SOCK_RAW, socket.CAN_RAW)
Andrew Svetlov0832af62012-12-18 23:10:48 +020048 except (AttributeError, OSError):
Charles-François Natali47413c12011-10-06 19:47:44 +020049 return False
50 else:
51 s.close()
52 return True
53
Charles-François Natali10b8cf42011-11-10 19:21:37 +010054def _have_socket_rds():
55 """Check whether RDS sockets are supported on this host."""
56 try:
57 s = socket.socket(socket.PF_RDS, socket.SOCK_SEQPACKET, 0)
58 except (AttributeError, OSError):
59 return False
60 else:
61 s.close()
62 return True
63
Charles-François Natali47413c12011-10-06 19:47:44 +020064HAVE_SOCKET_CAN = _have_socket_can()
65
Charles-François Natali10b8cf42011-11-10 19:21:37 +010066HAVE_SOCKET_RDS = _have_socket_rds()
67
Nick Coghlan96fe56a2011-08-22 11:55:57 +100068# Size in bytes of the int type
69SIZEOF_INT = array.array("i").itemsize
70
Guido van Rossum24e4af82002-06-12 19:18:08 +000071class SocketTCPTest(unittest.TestCase):
Barry Warsawcf3d4b51997-01-03 20:03:32 +000072
Guido van Rossum24e4af82002-06-12 19:18:08 +000073 def setUp(self):
74 self.serv = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
Benjamin Petersonee8712c2008-05-20 21:35:26 +000075 self.port = support.bind_port(self.serv)
Guido van Rossum24e4af82002-06-12 19:18:08 +000076 self.serv.listen(1)
Barry Warsawcf3d4b51997-01-03 20:03:32 +000077
Guido van Rossum24e4af82002-06-12 19:18:08 +000078 def tearDown(self):
79 self.serv.close()
80 self.serv = None
Barry Warsawcf3d4b51997-01-03 20:03:32 +000081
Guido van Rossum24e4af82002-06-12 19:18:08 +000082class SocketUDPTest(unittest.TestCase):
83
84 def setUp(self):
85 self.serv = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
Benjamin Petersonee8712c2008-05-20 21:35:26 +000086 self.port = support.bind_port(self.serv)
Guido van Rossum24e4af82002-06-12 19:18:08 +000087
88 def tearDown(self):
89 self.serv.close()
90 self.serv = None
91
Nick Coghlan96fe56a2011-08-22 11:55:57 +100092class ThreadSafeCleanupTestCase(unittest.TestCase):
93 """Subclass of unittest.TestCase with thread-safe cleanup methods.
94
95 This subclass protects the addCleanup() and doCleanups() methods
96 with a recursive lock.
97 """
98
99 if threading:
100 def __init__(self, *args, **kwargs):
101 super().__init__(*args, **kwargs)
102 self._cleanup_lock = threading.RLock()
103
104 def addCleanup(self, *args, **kwargs):
105 with self._cleanup_lock:
106 return super().addCleanup(*args, **kwargs)
107
108 def doCleanups(self, *args, **kwargs):
109 with self._cleanup_lock:
110 return super().doCleanups(*args, **kwargs)
111
Charles-François Natali47413c12011-10-06 19:47:44 +0200112class SocketCANTest(unittest.TestCase):
113
114 """To be able to run this test, a `vcan0` CAN interface can be created with
115 the following commands:
116 # modprobe vcan
117 # ip link add dev vcan0 type vcan
118 # ifconfig vcan0 up
119 """
120 interface = 'vcan0'
121 bufsize = 128
122
Charles-François Natali773e42d2013-02-05 19:42:01 +0100123 """The CAN frame structure is defined in <linux/can.h>:
124
125 struct can_frame {
126 canid_t can_id; /* 32 bit CAN_ID + EFF/RTR/ERR flags */
127 __u8 can_dlc; /* data length code: 0 .. 8 */
128 __u8 data[8] __attribute__((aligned(8)));
129 };
130 """
131 can_frame_fmt = "=IB3x8s"
132 can_frame_size = struct.calcsize(can_frame_fmt)
133
134 """The Broadcast Management Command frame structure is defined
135 in <linux/can/bcm.h>:
136
137 struct bcm_msg_head {
138 __u32 opcode;
139 __u32 flags;
140 __u32 count;
141 struct timeval ival1, ival2;
142 canid_t can_id;
143 __u32 nframes;
144 struct can_frame frames[0];
145 }
146
147 `bcm_msg_head` must be 8 bytes aligned because of the `frames` member (see
148 `struct can_frame` definition). Must use native not standard types for packing.
149 """
150 bcm_cmd_msg_fmt = "@3I4l2I"
151 bcm_cmd_msg_fmt += "x" * (struct.calcsize(bcm_cmd_msg_fmt) % 8)
152
Charles-François Natali47413c12011-10-06 19:47:44 +0200153 def setUp(self):
154 self.s = socket.socket(socket.PF_CAN, socket.SOCK_RAW, socket.CAN_RAW)
Antoine Pitrou86fe8602011-10-10 20:15:59 +0200155 self.addCleanup(self.s.close)
Charles-François Natali47413c12011-10-06 19:47:44 +0200156 try:
157 self.s.bind((self.interface,))
Andrew Svetlov0832af62012-12-18 23:10:48 +0200158 except OSError:
Charles-François Natali47413c12011-10-06 19:47:44 +0200159 self.skipTest('network interface `%s` does not exist' %
160 self.interface)
Charles-François Natali47413c12011-10-06 19:47:44 +0200161
Charles-François Natali10b8cf42011-11-10 19:21:37 +0100162
163class SocketRDSTest(unittest.TestCase):
164
165 """To be able to run this test, the `rds` kernel module must be loaded:
166 # modprobe rds
167 """
168 bufsize = 8192
169
170 def setUp(self):
171 self.serv = socket.socket(socket.PF_RDS, socket.SOCK_SEQPACKET, 0)
172 self.addCleanup(self.serv.close)
173 try:
174 self.port = support.bind_port(self.serv)
175 except OSError:
176 self.skipTest('unable to bind RDS socket')
177
178
Guido van Rossum24e4af82002-06-12 19:18:08 +0000179class ThreadableTest:
Guido van Rossum83ccb4e2002-06-18 18:35:13 +0000180 """Threadable Test class
181
182 The ThreadableTest class makes it easy to create a threaded
183 client/server pair from an existing unit test. To create a
184 new threaded class from an existing unit test, use multiple
185 inheritance:
186
187 class NewClass (OldClass, ThreadableTest):
188 pass
189
190 This class defines two new fixture functions with obvious
191 purposes for overriding:
192
193 clientSetUp ()
194 clientTearDown ()
195
196 Any new test functions within the class must then define
197 tests in pairs, where the test name is preceeded with a
198 '_' to indicate the client portion of the test. Ex:
199
200 def testFoo(self):
201 # Server portion
202
203 def _testFoo(self):
204 # Client portion
205
206 Any exceptions raised by the clients during their tests
207 are caught and transferred to the main thread to alert
208 the testing framework.
209
210 Note, the server setup function cannot call any blocking
211 functions that rely on the client thread during setup,
Guido van Rossumd8faa362007-04-27 19:54:29 +0000212 unless serverExplicitReady() is called just before
Guido van Rossum83ccb4e2002-06-18 18:35:13 +0000213 the blocking call (such as in setting up a client/server
214 connection and performing the accept() in setUp().
215 """
Guido van Rossum24e4af82002-06-12 19:18:08 +0000216
217 def __init__(self):
218 # Swap the true setup function
219 self.__setUp = self.setUp
220 self.__tearDown = self.tearDown
221 self.setUp = self._setUp
222 self.tearDown = self._tearDown
223
Guido van Rossum83ccb4e2002-06-18 18:35:13 +0000224 def serverExplicitReady(self):
225 """This method allows the server to explicitly indicate that
226 it wants the client thread to proceed. This is useful if the
227 server is about to execute a blocking routine that is
228 dependent upon the client thread during its setup routine."""
229 self.server_ready.set()
230
Guido van Rossum24e4af82002-06-12 19:18:08 +0000231 def _setUp(self):
Guido van Rossum83ccb4e2002-06-18 18:35:13 +0000232 self.server_ready = threading.Event()
233 self.client_ready = threading.Event()
Guido van Rossum24e4af82002-06-12 19:18:08 +0000234 self.done = threading.Event()
Alexandre Vassalottif260e442008-05-11 19:59:59 +0000235 self.queue = queue.Queue(1)
Antoine Pitrou86fe8602011-10-10 20:15:59 +0200236 self.server_crashed = False
Guido van Rossum24e4af82002-06-12 19:18:08 +0000237
238 # Do some munging to start the client test.
Guido van Rossum11ba0942002-06-13 15:07:44 +0000239 methodname = self.id()
240 i = methodname.rfind('.')
241 methodname = methodname[i+1:]
242 test_method = getattr(self, '_' + methodname)
Guido van Rossumab659962002-06-12 21:29:43 +0000243 self.client_thread = thread.start_new_thread(
244 self.clientRun, (test_method,))
Guido van Rossum24e4af82002-06-12 19:18:08 +0000245
Antoine Pitrou86fe8602011-10-10 20:15:59 +0200246 try:
247 self.__setUp()
248 except:
249 self.server_crashed = True
250 raise
251 finally:
Guido van Rossum83ccb4e2002-06-18 18:35:13 +0000252 self.server_ready.set()
253 self.client_ready.wait()
Guido van Rossum24e4af82002-06-12 19:18:08 +0000254
255 def _tearDown(self):
256 self.__tearDown()
257 self.done.wait()
258
Raymond Hettingerda3caed2008-01-14 21:39:24 +0000259 if self.queue.qsize():
Antoine Pitrou98b46702010-09-18 22:59:00 +0000260 exc = self.queue.get()
261 raise exc
Guido van Rossum24e4af82002-06-12 19:18:08 +0000262
263 def clientRun(self, test_func):
Guido van Rossum83ccb4e2002-06-18 18:35:13 +0000264 self.server_ready.wait()
Guido van Rossum24e4af82002-06-12 19:18:08 +0000265 self.clientSetUp()
Charles-François Natalia49ed762011-09-21 22:03:58 +0200266 self.client_ready.set()
Antoine Pitrou86fe8602011-10-10 20:15:59 +0200267 if self.server_crashed:
268 self.clientTearDown()
269 return
Guido van Rossumd59da4b2007-05-22 18:11:13 +0000270 if not hasattr(test_func, '__call__'):
Collin Winter3add4d72007-08-29 23:37:32 +0000271 raise TypeError("test_func must be a callable function")
Guido van Rossum24e4af82002-06-12 19:18:08 +0000272 try:
273 test_func()
Antoine Pitrou98b46702010-09-18 22:59:00 +0000274 except BaseException as e:
275 self.queue.put(e)
276 finally:
277 self.clientTearDown()
Guido van Rossum24e4af82002-06-12 19:18:08 +0000278
279 def clientSetUp(self):
Collin Winter3add4d72007-08-29 23:37:32 +0000280 raise NotImplementedError("clientSetUp must be implemented.")
Guido van Rossum24e4af82002-06-12 19:18:08 +0000281
282 def clientTearDown(self):
283 self.done.set()
284 thread.exit()
285
286class ThreadedTCPSocketTest(SocketTCPTest, ThreadableTest):
287
288 def __init__(self, methodName='runTest'):
289 SocketTCPTest.__init__(self, methodName=methodName)
290 ThreadableTest.__init__(self)
291
292 def clientSetUp(self):
293 self.cli = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
294
295 def clientTearDown(self):
296 self.cli.close()
297 self.cli = None
298 ThreadableTest.clientTearDown(self)
299
300class ThreadedUDPSocketTest(SocketUDPTest, ThreadableTest):
301
302 def __init__(self, methodName='runTest'):
303 SocketUDPTest.__init__(self, methodName=methodName)
304 ThreadableTest.__init__(self)
305
306 def clientSetUp(self):
307 self.cli = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
308
Brian Curtin3beb38f2010-11-04 03:41:43 +0000309 def clientTearDown(self):
310 self.cli.close()
311 self.cli = None
312 ThreadableTest.clientTearDown(self)
313
Charles-François Natali47413c12011-10-06 19:47:44 +0200314class ThreadedCANSocketTest(SocketCANTest, ThreadableTest):
315
316 def __init__(self, methodName='runTest'):
317 SocketCANTest.__init__(self, methodName=methodName)
318 ThreadableTest.__init__(self)
319
320 def clientSetUp(self):
321 self.cli = socket.socket(socket.PF_CAN, socket.SOCK_RAW, socket.CAN_RAW)
322 try:
323 self.cli.bind((self.interface,))
Andrew Svetlov0832af62012-12-18 23:10:48 +0200324 except OSError:
Antoine Pitrou86fe8602011-10-10 20:15:59 +0200325 # skipTest should not be called here, and will be called in the
326 # server instead
327 pass
Charles-François Natali47413c12011-10-06 19:47:44 +0200328
329 def clientTearDown(self):
330 self.cli.close()
331 self.cli = None
332 ThreadableTest.clientTearDown(self)
333
Charles-François Natali10b8cf42011-11-10 19:21:37 +0100334class ThreadedRDSSocketTest(SocketRDSTest, ThreadableTest):
335
336 def __init__(self, methodName='runTest'):
337 SocketRDSTest.__init__(self, methodName=methodName)
338 ThreadableTest.__init__(self)
Charles-François Natali10b8cf42011-11-10 19:21:37 +0100339
340 def clientSetUp(self):
341 self.cli = socket.socket(socket.PF_RDS, socket.SOCK_SEQPACKET, 0)
342 try:
343 # RDS sockets must be bound explicitly to send or receive data
344 self.cli.bind((HOST, 0))
345 self.cli_addr = self.cli.getsockname()
346 except OSError:
347 # skipTest should not be called here, and will be called in the
348 # server instead
349 pass
350
351 def clientTearDown(self):
352 self.cli.close()
353 self.cli = None
354 ThreadableTest.clientTearDown(self)
355
Guido van Rossum24e4af82002-06-12 19:18:08 +0000356class SocketConnectedTest(ThreadedTCPSocketTest):
Jeremy Hylton5accbdb2007-08-03 20:40:09 +0000357 """Socket tests for client-server connection.
358
359 self.cli_conn is a client socket connected to the server. The
360 setUp() method guarantees that it is connected to the server.
361 """
Guido van Rossum24e4af82002-06-12 19:18:08 +0000362
363 def __init__(self, methodName='runTest'):
364 ThreadedTCPSocketTest.__init__(self, methodName=methodName)
365
366 def setUp(self):
367 ThreadedTCPSocketTest.setUp(self)
Guido van Rossum83ccb4e2002-06-18 18:35:13 +0000368 # Indicate explicitly we're ready for the client thread to
369 # proceed and then perform the blocking call to accept
370 self.serverExplicitReady()
Guido van Rossum24e4af82002-06-12 19:18:08 +0000371 conn, addr = self.serv.accept()
372 self.cli_conn = conn
373
374 def tearDown(self):
375 self.cli_conn.close()
376 self.cli_conn = None
377 ThreadedTCPSocketTest.tearDown(self)
378
379 def clientSetUp(self):
380 ThreadedTCPSocketTest.clientSetUp(self)
Christian Heimes5e696852008-04-09 08:37:03 +0000381 self.cli.connect((HOST, self.port))
Guido van Rossum24e4af82002-06-12 19:18:08 +0000382 self.serv_conn = self.cli
383
384 def clientTearDown(self):
385 self.serv_conn.close()
386 self.serv_conn = None
387 ThreadedTCPSocketTest.clientTearDown(self)
388
Dave Cole331708b2004-08-09 04:51:41 +0000389class SocketPairTest(unittest.TestCase, ThreadableTest):
390
391 def __init__(self, methodName='runTest'):
392 unittest.TestCase.__init__(self, methodName=methodName)
393 ThreadableTest.__init__(self)
394
395 def setUp(self):
396 self.serv, self.cli = socket.socketpair()
397
398 def tearDown(self):
399 self.serv.close()
400 self.serv = None
401
402 def clientSetUp(self):
403 pass
404
405 def clientTearDown(self):
406 self.cli.close()
407 self.cli = None
408 ThreadableTest.clientTearDown(self)
409
Tim Peters494aaee2004-08-09 18:54:11 +0000410
Nick Coghlan96fe56a2011-08-22 11:55:57 +1000411# The following classes are used by the sendmsg()/recvmsg() tests.
412# Combining, for instance, ConnectedStreamTestMixin and TCPTestBase
413# gives a drop-in replacement for SocketConnectedTest, but different
414# address families can be used, and the attributes serv_addr and
415# cli_addr will be set to the addresses of the endpoints.
416
417class SocketTestBase(unittest.TestCase):
418 """A base class for socket tests.
419
420 Subclasses must provide methods newSocket() to return a new socket
421 and bindSock(sock) to bind it to an unused address.
422
423 Creates a socket self.serv and sets self.serv_addr to its address.
424 """
425
426 def setUp(self):
427 self.serv = self.newSocket()
428 self.bindServer()
429
430 def bindServer(self):
431 """Bind server socket and set self.serv_addr to its address."""
432 self.bindSock(self.serv)
433 self.serv_addr = self.serv.getsockname()
434
435 def tearDown(self):
436 self.serv.close()
437 self.serv = None
438
439
440class SocketListeningTestMixin(SocketTestBase):
441 """Mixin to listen on the server socket."""
442
443 def setUp(self):
444 super().setUp()
445 self.serv.listen(1)
446
447
448class ThreadedSocketTestMixin(ThreadSafeCleanupTestCase, SocketTestBase,
449 ThreadableTest):
450 """Mixin to add client socket and allow client/server tests.
451
452 Client socket is self.cli and its address is self.cli_addr. See
453 ThreadableTest for usage information.
454 """
455
456 def __init__(self, *args, **kwargs):
457 super().__init__(*args, **kwargs)
458 ThreadableTest.__init__(self)
459
460 def clientSetUp(self):
461 self.cli = self.newClientSocket()
462 self.bindClient()
463
464 def newClientSocket(self):
465 """Return a new socket for use as client."""
466 return self.newSocket()
467
468 def bindClient(self):
469 """Bind client socket and set self.cli_addr to its address."""
470 self.bindSock(self.cli)
471 self.cli_addr = self.cli.getsockname()
472
473 def clientTearDown(self):
474 self.cli.close()
475 self.cli = None
476 ThreadableTest.clientTearDown(self)
477
478
479class ConnectedStreamTestMixin(SocketListeningTestMixin,
480 ThreadedSocketTestMixin):
481 """Mixin to allow client/server stream tests with connected client.
482
483 Server's socket representing connection to client is self.cli_conn
484 and client's connection to server is self.serv_conn. (Based on
485 SocketConnectedTest.)
486 """
487
488 def setUp(self):
489 super().setUp()
490 # Indicate explicitly we're ready for the client thread to
491 # proceed and then perform the blocking call to accept
492 self.serverExplicitReady()
493 conn, addr = self.serv.accept()
494 self.cli_conn = conn
495
496 def tearDown(self):
497 self.cli_conn.close()
498 self.cli_conn = None
499 super().tearDown()
500
501 def clientSetUp(self):
502 super().clientSetUp()
503 self.cli.connect(self.serv_addr)
504 self.serv_conn = self.cli
505
506 def clientTearDown(self):
507 self.serv_conn.close()
508 self.serv_conn = None
509 super().clientTearDown()
510
511
512class UnixSocketTestBase(SocketTestBase):
513 """Base class for Unix-domain socket tests."""
514
515 # This class is used for file descriptor passing tests, so we
516 # create the sockets in a private directory so that other users
517 # can't send anything that might be problematic for a privileged
518 # user running the tests.
519
520 def setUp(self):
521 self.dir_path = tempfile.mkdtemp()
522 self.addCleanup(os.rmdir, self.dir_path)
523 super().setUp()
524
525 def bindSock(self, sock):
526 path = tempfile.mktemp(dir=self.dir_path)
527 sock.bind(path)
528 self.addCleanup(support.unlink, path)
529
530class UnixStreamBase(UnixSocketTestBase):
531 """Base class for Unix-domain SOCK_STREAM tests."""
532
533 def newSocket(self):
534 return socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
535
536
537class InetTestBase(SocketTestBase):
538 """Base class for IPv4 socket tests."""
539
540 host = HOST
541
542 def setUp(self):
543 super().setUp()
544 self.port = self.serv_addr[1]
545
546 def bindSock(self, sock):
547 support.bind_port(sock, host=self.host)
548
549class TCPTestBase(InetTestBase):
550 """Base class for TCP-over-IPv4 tests."""
551
552 def newSocket(self):
553 return socket.socket(socket.AF_INET, socket.SOCK_STREAM)
554
555class UDPTestBase(InetTestBase):
556 """Base class for UDP-over-IPv4 tests."""
557
558 def newSocket(self):
559 return socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
560
561class SCTPStreamBase(InetTestBase):
562 """Base class for SCTP tests in one-to-one (SOCK_STREAM) mode."""
563
564 def newSocket(self):
565 return socket.socket(socket.AF_INET, socket.SOCK_STREAM,
566 socket.IPPROTO_SCTP)
567
568
569class Inet6TestBase(InetTestBase):
570 """Base class for IPv6 socket tests."""
571
572 # Don't use "localhost" here - it may not have an IPv6 address
573 # assigned to it by default (e.g. in /etc/hosts), and if someone
574 # has assigned it an IPv4-mapped address, then it's unlikely to
575 # work with the full IPv6 API.
576 host = "::1"
577
578class UDP6TestBase(Inet6TestBase):
579 """Base class for UDP-over-IPv6 tests."""
580
581 def newSocket(self):
582 return socket.socket(socket.AF_INET6, socket.SOCK_DGRAM)
583
584
585# Test-skipping decorators for use with ThreadableTest.
586
587def skipWithClientIf(condition, reason):
588 """Skip decorated test if condition is true, add client_skip decorator.
589
590 If the decorated object is not a class, sets its attribute
591 "client_skip" to a decorator which will return an empty function
592 if the test is to be skipped, or the original function if it is
593 not. This can be used to avoid running the client part of a
594 skipped test when using ThreadableTest.
595 """
596 def client_pass(*args, **kwargs):
597 pass
598 def skipdec(obj):
599 retval = unittest.skip(reason)(obj)
600 if not isinstance(obj, type):
601 retval.client_skip = lambda f: client_pass
602 return retval
603 def noskipdec(obj):
604 if not (isinstance(obj, type) or hasattr(obj, "client_skip")):
605 obj.client_skip = lambda f: f
606 return obj
607 return skipdec if condition else noskipdec
608
609
610def requireAttrs(obj, *attributes):
611 """Skip decorated test if obj is missing any of the given attributes.
612
613 Sets client_skip attribute as skipWithClientIf() does.
614 """
615 missing = [name for name in attributes if not hasattr(obj, name)]
616 return skipWithClientIf(
617 missing, "don't have " + ", ".join(name for name in missing))
618
619
620def requireSocket(*args):
621 """Skip decorated test if a socket cannot be created with given arguments.
622
623 When an argument is given as a string, will use the value of that
624 attribute of the socket module, or skip the test if it doesn't
625 exist. Sets client_skip attribute as skipWithClientIf() does.
626 """
627 err = None
628 missing = [obj for obj in args if
629 isinstance(obj, str) and not hasattr(socket, obj)]
630 if missing:
631 err = "don't have " + ", ".join(name for name in missing)
632 else:
633 callargs = [getattr(socket, obj) if isinstance(obj, str) else obj
634 for obj in args]
635 try:
636 s = socket.socket(*callargs)
Andrew Svetlov0832af62012-12-18 23:10:48 +0200637 except OSError as e:
Nick Coghlan96fe56a2011-08-22 11:55:57 +1000638 # XXX: check errno?
639 err = str(e)
640 else:
641 s.close()
642 return skipWithClientIf(
643 err is not None,
644 "can't create socket({0}): {1}".format(
645 ", ".join(str(o) for o in args), err))
646
647
Guido van Rossum24e4af82002-06-12 19:18:08 +0000648#######################################################################
649## Begin Tests
650
651class GeneralModuleTests(unittest.TestCase):
652
Walter Dörwalda7eb93e2007-06-05 13:41:53 +0000653 def test_repr(self):
654 s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
Giampaolo Rodola'50331cb2013-04-10 15:49:47 +0200655 with s:
656 self.assertIn('fd=%i' % s.fileno(), repr(s))
657 self.assertIn('family=%i' % socket.AF_INET, repr(s))
658 self.assertIn('type=%i' % socket.SOCK_STREAM, repr(s))
659 self.assertIn('proto=0', repr(s))
660 self.assertIn('laddr', repr(s))
661 self.assertNotIn('raddr', repr(s))
662 s.bind(('127.0.0.1', 0))
663 self.assertIn(str(s.getsockname()), repr(s))
664 self.assertIn('[closed]', repr(s))
665 self.assertNotIn('laddr', repr(s))
Walter Dörwalda7eb93e2007-06-05 13:41:53 +0000666
Raymond Hettinger027bb632004-05-31 03:09:25 +0000667 def test_weakref(self):
668 s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
669 p = proxy(s)
670 self.assertEqual(p.fileno(), s.fileno())
671 s.close()
672 s = None
673 try:
674 p.fileno()
675 except ReferenceError:
676 pass
677 else:
678 self.fail('Socket proxy still exists')
679
Guido van Rossum24e4af82002-06-12 19:18:08 +0000680 def testSocketError(self):
Guido van Rossum8c943832002-08-08 01:00:28 +0000681 # Testing socket module exceptions
Ezio Melotti845f1522011-05-08 20:23:57 +0300682 msg = "Error raising socket exception (%s)."
Andrew Svetlov0832af62012-12-18 23:10:48 +0200683 with self.assertRaises(OSError, msg=msg % 'OSError'):
684 raise OSError
685 with self.assertRaises(OSError, msg=msg % 'socket.herror'):
Guido van Rossum24e4af82002-06-12 19:18:08 +0000686 raise socket.herror
Andrew Svetlov0832af62012-12-18 23:10:48 +0200687 with self.assertRaises(OSError, msg=msg % 'socket.gaierror'):
Guido van Rossum24e4af82002-06-12 19:18:08 +0000688 raise socket.gaierror
Guido van Rossum24e4af82002-06-12 19:18:08 +0000689
Ezio Melotti63e42302011-05-07 19:47:48 +0300690 def testSendtoErrors(self):
691 # Testing that sendto doens't masks failures. See #10169.
692 s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
693 self.addCleanup(s.close)
694 s.bind(('', 0))
695 sockname = s.getsockname()
696 # 2 args
Ezio Melotti4c1aebd2011-05-07 19:50:28 +0300697 with self.assertRaises(TypeError) as cm:
Ezio Melotti63e42302011-05-07 19:47:48 +0300698 s.sendto('\u2620', sockname)
Ezio Melotti4c1aebd2011-05-07 19:50:28 +0300699 self.assertEqual(str(cm.exception),
700 "'str' does not support the buffer interface")
701 with self.assertRaises(TypeError) as cm:
Ezio Melotti63e42302011-05-07 19:47:48 +0300702 s.sendto(5j, sockname)
Ezio Melotti4c1aebd2011-05-07 19:50:28 +0300703 self.assertEqual(str(cm.exception),
704 "'complex' does not support the buffer interface")
705 with self.assertRaises(TypeError) as cm:
Ezio Melotti63e42302011-05-07 19:47:48 +0300706 s.sendto(b'foo', None)
Ezio Melotti4c1aebd2011-05-07 19:50:28 +0300707 self.assertIn('not NoneType',str(cm.exception))
Ezio Melotti63e42302011-05-07 19:47:48 +0300708 # 3 args
Ezio Melotti4c1aebd2011-05-07 19:50:28 +0300709 with self.assertRaises(TypeError) as cm:
Ezio Melotti63e42302011-05-07 19:47:48 +0300710 s.sendto('\u2620', 0, sockname)
Ezio Melotti4c1aebd2011-05-07 19:50:28 +0300711 self.assertEqual(str(cm.exception),
712 "'str' does not support the buffer interface")
713 with self.assertRaises(TypeError) as cm:
Ezio Melotti63e42302011-05-07 19:47:48 +0300714 s.sendto(5j, 0, sockname)
Ezio Melotti4c1aebd2011-05-07 19:50:28 +0300715 self.assertEqual(str(cm.exception),
716 "'complex' does not support the buffer interface")
717 with self.assertRaises(TypeError) as cm:
Ezio Melotti63e42302011-05-07 19:47:48 +0300718 s.sendto(b'foo', 0, None)
Ezio Melotti4c1aebd2011-05-07 19:50:28 +0300719 self.assertIn('not NoneType', str(cm.exception))
720 with self.assertRaises(TypeError) as cm:
Ezio Melotti63e42302011-05-07 19:47:48 +0300721 s.sendto(b'foo', 'bar', sockname)
Ezio Melotti4c1aebd2011-05-07 19:50:28 +0300722 self.assertIn('an integer is required', str(cm.exception))
723 with self.assertRaises(TypeError) as cm:
Ezio Melotti63e42302011-05-07 19:47:48 +0300724 s.sendto(b'foo', None, None)
Ezio Melotti4c1aebd2011-05-07 19:50:28 +0300725 self.assertIn('an integer is required', str(cm.exception))
Ezio Melotti63e42302011-05-07 19:47:48 +0300726 # wrong number of args
Ezio Melotti4c1aebd2011-05-07 19:50:28 +0300727 with self.assertRaises(TypeError) as cm:
Ezio Melotti63e42302011-05-07 19:47:48 +0300728 s.sendto(b'foo')
Ezio Melotti4c1aebd2011-05-07 19:50:28 +0300729 self.assertIn('(1 given)', str(cm.exception))
730 with self.assertRaises(TypeError) as cm:
Ezio Melotti63e42302011-05-07 19:47:48 +0300731 s.sendto(b'foo', 0, sockname, 4)
Ezio Melotti4c1aebd2011-05-07 19:50:28 +0300732 self.assertIn('(4 given)', str(cm.exception))
Ezio Melotti63e42302011-05-07 19:47:48 +0300733
Guido van Rossum24e4af82002-06-12 19:18:08 +0000734 def testCrucialConstants(self):
Guido van Rossum8c943832002-08-08 01:00:28 +0000735 # Testing for mission critical constants
Guido van Rossum24e4af82002-06-12 19:18:08 +0000736 socket.AF_INET
737 socket.SOCK_STREAM
738 socket.SOCK_DGRAM
739 socket.SOCK_RAW
740 socket.SOCK_RDM
741 socket.SOCK_SEQPACKET
742 socket.SOL_SOCKET
743 socket.SO_REUSEADDR
744
Guido van Rossum654c11e2002-06-13 20:24:17 +0000745 def testHostnameRes(self):
Guido van Rossum8c943832002-08-08 01:00:28 +0000746 # Testing hostname resolution mechanisms
Guido van Rossum654c11e2002-06-13 20:24:17 +0000747 hostname = socket.gethostname()
Guido van Rossum71e02942002-12-26 16:55:15 +0000748 try:
749 ip = socket.gethostbyname(hostname)
Andrew Svetlov0832af62012-12-18 23:10:48 +0200750 except OSError:
Guido van Rossum71e02942002-12-26 16:55:15 +0000751 # Probably name lookup wasn't set up right; skip this test
752 return
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000753 self.assertTrue(ip.find('.') >= 0, "Error resolving host to ip.")
Guido van Rossum9647b522002-12-26 17:04:45 +0000754 try:
755 hname, aliases, ipaddrs = socket.gethostbyaddr(ip)
Andrew Svetlov0832af62012-12-18 23:10:48 +0200756 except OSError:
Guido van Rossum9647b522002-12-26 17:04:45 +0000757 # Probably a similar problem as above; skip this test
758 return
Brett Cannon01668a12005-03-11 00:04:17 +0000759 all_host_names = [hostname, hname] + aliases
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000760 fqhn = socket.getfqdn(ip)
Guido van Rossum654c11e2002-06-13 20:24:17 +0000761 if not fqhn in all_host_names:
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000762 self.fail("Error testing host resolution mechanisms. (fqdn: %s, all: %s)" % (fqhn, repr(all_host_names)))
Barry Warsawcf3d4b51997-01-03 20:03:32 +0000763
Antoine Pitrou061cfb52011-02-28 22:25:22 +0000764 @unittest.skipUnless(hasattr(socket, 'sethostname'), "test needs socket.sethostname()")
765 @unittest.skipUnless(hasattr(socket, 'gethostname'), "test needs socket.gethostname()")
766 def test_sethostname(self):
767 oldhn = socket.gethostname()
768 try:
769 socket.sethostname('new')
Andrew Svetlov0832af62012-12-18 23:10:48 +0200770 except OSError as e:
Antoine Pitrou061cfb52011-02-28 22:25:22 +0000771 if e.errno == errno.EPERM:
772 self.skipTest("test should be run as root")
773 else:
774 raise
775 try:
776 # running test as root!
777 self.assertEqual(socket.gethostname(), 'new')
778 # Should work with bytes objects too
779 socket.sethostname(b'bar')
780 self.assertEqual(socket.gethostname(), 'bar')
781 finally:
782 socket.sethostname(oldhn)
783
Gregory P. Smith5ed2e772011-05-15 00:26:45 -0700784 @unittest.skipUnless(hasattr(socket, 'if_nameindex'),
785 'socket.if_nameindex() not available.')
786 def testInterfaceNameIndex(self):
787 interfaces = socket.if_nameindex()
788 for index, name in interfaces:
Charles-François Natali60713592011-05-20 16:55:06 +0200789 self.assertIsInstance(index, int)
790 self.assertIsInstance(name, str)
Gregory P. Smith5ed2e772011-05-15 00:26:45 -0700791 # interface indices are non-zero integers
792 self.assertGreater(index, 0)
Charles-François Natali60713592011-05-20 16:55:06 +0200793 _index = socket.if_nametoindex(name)
794 self.assertIsInstance(_index, int)
795 self.assertEqual(index, _index)
796 _name = socket.if_indextoname(index)
797 self.assertIsInstance(_name, str)
798 self.assertEqual(name, _name)
799
800 @unittest.skipUnless(hasattr(socket, 'if_nameindex'),
801 'socket.if_nameindex() not available.')
802 def testInvalidInterfaceNameIndex(self):
803 # test nonexistent interface index/name
Andrew Svetlov0832af62012-12-18 23:10:48 +0200804 self.assertRaises(OSError, socket.if_indextoname, 0)
805 self.assertRaises(OSError, socket.if_nametoindex, '_DEADBEEF')
Charles-François Natali60713592011-05-20 16:55:06 +0200806 # test with invalid values
807 self.assertRaises(TypeError, socket.if_nametoindex, 0)
808 self.assertRaises(TypeError, socket.if_indextoname, '_DEADBEEF')
Gregory P. Smith5ed2e772011-05-15 00:26:45 -0700809
Guido van Rossum284a2cf2002-06-12 21:19:40 +0000810 def testRefCountGetNameInfo(self):
Guido van Rossum8c943832002-08-08 01:00:28 +0000811 # Testing reference count for getnameinfo
Guido van Rossum284a2cf2002-06-12 21:19:40 +0000812 if hasattr(sys, "getrefcount"):
Guido van Rossum24e4af82002-06-12 19:18:08 +0000813 try:
814 # On some versions, this loses a reference
815 orig = sys.getrefcount(__name__)
816 socket.getnameinfo(__name__,0)
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +0000817 except TypeError:
Guido van Rossumb053cd82006-08-24 03:53:23 +0000818 if sys.getrefcount(__name__) != orig:
Guido van Rossum24e4af82002-06-12 19:18:08 +0000819 self.fail("socket.getnameinfo loses a reference")
Barry Warsawcf3d4b51997-01-03 20:03:32 +0000820
Guido van Rossum24e4af82002-06-12 19:18:08 +0000821 def testInterpreterCrash(self):
Guido van Rossum8c943832002-08-08 01:00:28 +0000822 # Making sure getnameinfo doesn't crash the interpreter
Guido van Rossum24e4af82002-06-12 19:18:08 +0000823 try:
824 # On some versions, this crashes the interpreter.
825 socket.getnameinfo(('x', 0, 0, 0), 0)
Andrew Svetlov0832af62012-12-18 23:10:48 +0200826 except OSError:
Guido van Rossum24e4af82002-06-12 19:18:08 +0000827 pass
Barry Warsawcf3d4b51997-01-03 20:03:32 +0000828
Guido van Rossumc0a0e082002-09-16 01:30:03 +0000829 def testNtoH(self):
Guido van Rossuma2627af2002-09-14 00:58:46 +0000830 # This just checks that htons etc. are their own inverse,
831 # when looking at the lower 16 or 32 bits.
832 sizes = {socket.htonl: 32, socket.ntohl: 32,
833 socket.htons: 16, socket.ntohs: 16}
834 for func, size in sizes.items():
Guido van Rossume2a383d2007-01-15 16:59:06 +0000835 mask = (1<<size) - 1
Guido van Rossuma2627af2002-09-14 00:58:46 +0000836 for i in (0, 1, 0xffff, ~0xffff, 2, 0x01234567, 0x76543210):
837 self.assertEqual(i & mask, func(func(i&mask)) & mask)
Jeremy Hyltoncbd5b892002-07-31 15:57:39 +0000838
Guido van Rossuma2627af2002-09-14 00:58:46 +0000839 swapped = func(mask)
840 self.assertEqual(swapped & mask, mask)
Guido van Rossume2a383d2007-01-15 16:59:06 +0000841 self.assertRaises(OverflowError, func, 1<<34)
Jeremy Hyltonc075e192002-07-25 16:01:12 +0000842
Guido van Rossum018919a2007-01-15 00:07:32 +0000843 def testNtoHErrors(self):
Guido van Rossume2a383d2007-01-15 16:59:06 +0000844 good_values = [ 1, 2, 3, 1, 2, 3 ]
845 bad_values = [ -1, -2, -3, -1, -2, -3 ]
Guido van Rossum018919a2007-01-15 00:07:32 +0000846 for k in good_values:
847 socket.ntohl(k)
848 socket.ntohs(k)
849 socket.htonl(k)
850 socket.htons(k)
851 for k in bad_values:
852 self.assertRaises(OverflowError, socket.ntohl, k)
853 self.assertRaises(OverflowError, socket.ntohs, k)
854 self.assertRaises(OverflowError, socket.htonl, k)
855 self.assertRaises(OverflowError, socket.htons, k)
856
Barry Warsaw11b91a02004-06-28 00:50:43 +0000857 def testGetServBy(self):
858 eq = self.assertEqual
859 # Find one service that exists, then check all the related interfaces.
860 # I've ordered this by protocols that have both a tcp and udp
861 # protocol, at least for modern Linuxes.
Victor Stinnere6747472011-08-21 00:39:18 +0200862 if (sys.platform.startswith(('freebsd', 'netbsd'))
863 or sys.platform in ('linux', 'darwin')):
Andrew MacIntyre18bf43c2004-07-12 12:10:30 +0000864 # avoid the 'echo' service on this platform, as there is an
865 # assumption breaking non-standard port/protocol entry
866 services = ('daytime', 'qotd', 'domain')
867 else:
868 services = ('echo', 'daytime', 'domain')
869 for service in services:
Skip Montanarof4433302002-08-02 15:52:30 +0000870 try:
Barry Warsaw11b91a02004-06-28 00:50:43 +0000871 port = socket.getservbyname(service, 'tcp')
Skip Montanarof4433302002-08-02 15:52:30 +0000872 break
Andrew Svetlov0832af62012-12-18 23:10:48 +0200873 except OSError:
Skip Montanarof4433302002-08-02 15:52:30 +0000874 pass
Skip Montanaro05eb4012004-02-10 15:51:15 +0000875 else:
Andrew Svetlov0832af62012-12-18 23:10:48 +0200876 raise OSError
Barry Warsaw11b91a02004-06-28 00:50:43 +0000877 # Try same call with optional protocol omitted
878 port2 = socket.getservbyname(service)
879 eq(port, port2)
Terry Jan Reedyc30b7b12013-03-11 17:57:08 -0400880 # Try udp, but don't barf if it doesn't exist
Barry Warsaw11b91a02004-06-28 00:50:43 +0000881 try:
882 udpport = socket.getservbyname(service, 'udp')
Andrew Svetlov0832af62012-12-18 23:10:48 +0200883 except OSError:
Barry Warsaw11b91a02004-06-28 00:50:43 +0000884 udpport = None
885 else:
886 eq(udpport, port)
887 # Now make sure the lookup by port returns the same service name
888 eq(socket.getservbyport(port2), service)
889 eq(socket.getservbyport(port, 'tcp'), service)
890 if udpport is not None:
891 eq(socket.getservbyport(udpport, 'udp'), service)
Benjamin Peterson5c6d7872009-02-06 02:40:07 +0000892 # Make sure getservbyport does not accept out of range ports.
893 self.assertRaises(OverflowError, socket.getservbyport, -1)
894 self.assertRaises(OverflowError, socket.getservbyport, 65536)
Guido van Rossum24e4af82002-06-12 19:18:08 +0000895
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +0000896 def testDefaultTimeout(self):
Guido van Rossum8c943832002-08-08 01:00:28 +0000897 # Testing default timeout
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +0000898 # The default timeout should initially be None
899 self.assertEqual(socket.getdefaulttimeout(), None)
900 s = socket.socket()
901 self.assertEqual(s.gettimeout(), None)
902 s.close()
903
904 # Set the default timeout to 10, and see if it propagates
905 socket.setdefaulttimeout(10)
906 self.assertEqual(socket.getdefaulttimeout(), 10)
907 s = socket.socket()
908 self.assertEqual(s.gettimeout(), 10)
909 s.close()
910
911 # Reset the default timeout to None, and see if it propagates
912 socket.setdefaulttimeout(None)
913 self.assertEqual(socket.getdefaulttimeout(), None)
914 s = socket.socket()
915 self.assertEqual(s.gettimeout(), None)
916 s.close()
917
918 # Check that setting it to an invalid value raises ValueError
919 self.assertRaises(ValueError, socket.setdefaulttimeout, -1)
920
921 # Check that setting it to an invalid type raises TypeError
922 self.assertRaises(TypeError, socket.setdefaulttimeout, "spam")
923
Benjamin Petersonf91df042009-02-13 02:50:59 +0000924 def testIPv4_inet_aton_fourbytes(self):
925 if not hasattr(socket, 'inet_aton'):
926 return # No inet_aton, nothing to check
927 # Test that issue1008086 and issue767150 are fixed.
928 # It must return 4 bytes.
Ezio Melottib3aedd42010-11-20 19:04:17 +0000929 self.assertEqual(b'\x00'*4, socket.inet_aton('0.0.0.0'))
930 self.assertEqual(b'\xff'*4, socket.inet_aton('255.255.255.255'))
Benjamin Petersonf91df042009-02-13 02:50:59 +0000931
Guido van Rossum47dfa4a2003-04-25 05:48:32 +0000932 def testIPv4toString(self):
Guido van Rossumf4001ee2003-04-25 15:11:23 +0000933 if not hasattr(socket, 'inet_pton'):
934 return # No inet_pton() on this platform
Guido van Rossum47dfa4a2003-04-25 05:48:32 +0000935 from socket import inet_aton as f, inet_pton, AF_INET
936 g = lambda a: inet_pton(AF_INET, a)
937
Antoine Pitrou7ded21e2011-12-19 16:21:21 +0100938 assertInvalid = lambda func,a: self.assertRaises(
Andrew Svetlov0832af62012-12-18 23:10:48 +0200939 (OSError, ValueError), func, a
Antoine Pitrou7ded21e2011-12-19 16:21:21 +0100940 )
941
Ezio Melottib3aedd42010-11-20 19:04:17 +0000942 self.assertEqual(b'\x00\x00\x00\x00', f('0.0.0.0'))
943 self.assertEqual(b'\xff\x00\xff\x00', f('255.0.255.0'))
944 self.assertEqual(b'\xaa\xaa\xaa\xaa', f('170.170.170.170'))
945 self.assertEqual(b'\x01\x02\x03\x04', f('1.2.3.4'))
946 self.assertEqual(b'\xff\xff\xff\xff', f('255.255.255.255'))
Antoine Pitrou7ded21e2011-12-19 16:21:21 +0100947 assertInvalid(f, '0.0.0.')
948 assertInvalid(f, '300.0.0.0')
949 assertInvalid(f, 'a.0.0.0')
950 assertInvalid(f, '1.2.3.4.5')
951 assertInvalid(f, '::1')
Guido van Rossum47dfa4a2003-04-25 05:48:32 +0000952
Ezio Melottib3aedd42010-11-20 19:04:17 +0000953 self.assertEqual(b'\x00\x00\x00\x00', g('0.0.0.0'))
954 self.assertEqual(b'\xff\x00\xff\x00', g('255.0.255.0'))
955 self.assertEqual(b'\xaa\xaa\xaa\xaa', g('170.170.170.170'))
956 self.assertEqual(b'\xff\xff\xff\xff', g('255.255.255.255'))
Antoine Pitrou7ded21e2011-12-19 16:21:21 +0100957 assertInvalid(g, '0.0.0.')
958 assertInvalid(g, '300.0.0.0')
959 assertInvalid(g, 'a.0.0.0')
960 assertInvalid(g, '1.2.3.4.5')
961 assertInvalid(g, '::1')
Tim Petersc2659cf2003-05-12 20:19:37 +0000962
Guido van Rossum47dfa4a2003-04-25 05:48:32 +0000963 def testIPv6toString(self):
Guido van Rossumf4001ee2003-04-25 15:11:23 +0000964 if not hasattr(socket, 'inet_pton'):
965 return # No inet_pton() on this platform
Guido van Rossum47dfa4a2003-04-25 05:48:32 +0000966 try:
967 from socket import inet_pton, AF_INET6, has_ipv6
968 if not has_ipv6:
969 return
970 except ImportError:
971 return
972 f = lambda a: inet_pton(AF_INET6, a)
Antoine Pitrou7ded21e2011-12-19 16:21:21 +0100973 assertInvalid = lambda a: self.assertRaises(
Andrew Svetlov0832af62012-12-18 23:10:48 +0200974 (OSError, ValueError), f, a
Antoine Pitrou7ded21e2011-12-19 16:21:21 +0100975 )
Guido van Rossum47dfa4a2003-04-25 05:48:32 +0000976
Ezio Melottib3aedd42010-11-20 19:04:17 +0000977 self.assertEqual(b'\x00' * 16, f('::'))
978 self.assertEqual(b'\x00' * 16, f('0::0'))
979 self.assertEqual(b'\x00\x01' + b'\x00' * 14, f('1::'))
980 self.assertEqual(
Guido van Rossum67180622007-07-10 07:29:12 +0000981 b'\x45\xef\x76\xcb\x00\x1a\x56\xef\xaf\xeb\x0b\xac\x19\x24\xae\xae',
Guido van Rossum47dfa4a2003-04-25 05:48:32 +0000982 f('45ef:76cb:1a:56ef:afeb:bac:1924:aeae')
983 )
Antoine Pitrou7ded21e2011-12-19 16:21:21 +0100984 self.assertEqual(
985 b'\xad\x42\x0a\xbc' + b'\x00' * 4 + b'\x01\x27\x00\x00\x02\x54\x00\x02',
986 f('ad42:abc::127:0:254:2')
987 )
988 self.assertEqual(b'\x00\x12\x00\x0a' + b'\x00' * 12, f('12:a::'))
989 assertInvalid('0x20::')
990 assertInvalid(':::')
991 assertInvalid('::0::')
992 assertInvalid('1::abc::')
993 assertInvalid('1::abc::def')
994 assertInvalid('1:2:3:4:5:6:')
995 assertInvalid('1:2:3:4:5:6')
996 assertInvalid('1:2:3:4:5:6:7:8:')
997 assertInvalid('1:2:3:4:5:6:7:8:0')
998
999 self.assertEqual(b'\x00' * 12 + b'\xfe\x2a\x17\x40',
1000 f('::254.42.23.64')
1001 )
1002 self.assertEqual(
1003 b'\x00\x42' + b'\x00' * 8 + b'\xa2\x9b\xfe\x2a\x17\x40',
1004 f('42::a29b:254.42.23.64')
1005 )
1006 self.assertEqual(
1007 b'\x00\x42\xa8\xb9\x00\x00\x00\x02\xff\xff\xa2\x9b\xfe\x2a\x17\x40',
1008 f('42:a8b9:0:2:ffff:a29b:254.42.23.64')
1009 )
1010 assertInvalid('255.254.253.252')
1011 assertInvalid('1::260.2.3.0')
1012 assertInvalid('1::0.be.e.0')
1013 assertInvalid('1:2:3:4:5:6:7:1.2.3.4')
1014 assertInvalid('::1.2.3.4:0')
1015 assertInvalid('0.100.200.0:3:4:5:6:7:8')
Tim Petersc2659cf2003-05-12 20:19:37 +00001016
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00001017 def testStringToIPv4(self):
Guido van Rossumf4001ee2003-04-25 15:11:23 +00001018 if not hasattr(socket, 'inet_ntop'):
1019 return # No inet_ntop() on this platform
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00001020 from socket import inet_ntoa as f, inet_ntop, AF_INET
1021 g = lambda a: inet_ntop(AF_INET, a)
Antoine Pitrou7ded21e2011-12-19 16:21:21 +01001022 assertInvalid = lambda func,a: self.assertRaises(
Andrew Svetlov0832af62012-12-18 23:10:48 +02001023 (OSError, ValueError), func, a
Antoine Pitrou7ded21e2011-12-19 16:21:21 +01001024 )
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00001025
Ezio Melottib3aedd42010-11-20 19:04:17 +00001026 self.assertEqual('1.0.1.0', f(b'\x01\x00\x01\x00'))
1027 self.assertEqual('170.85.170.85', f(b'\xaa\x55\xaa\x55'))
1028 self.assertEqual('255.255.255.255', f(b'\xff\xff\xff\xff'))
1029 self.assertEqual('1.2.3.4', f(b'\x01\x02\x03\x04'))
Antoine Pitrou7ded21e2011-12-19 16:21:21 +01001030 assertInvalid(f, b'\x00' * 3)
1031 assertInvalid(f, b'\x00' * 5)
1032 assertInvalid(f, b'\x00' * 16)
Tim Petersc2659cf2003-05-12 20:19:37 +00001033
Ezio Melottib3aedd42010-11-20 19:04:17 +00001034 self.assertEqual('1.0.1.0', g(b'\x01\x00\x01\x00'))
1035 self.assertEqual('170.85.170.85', g(b'\xaa\x55\xaa\x55'))
1036 self.assertEqual('255.255.255.255', g(b'\xff\xff\xff\xff'))
Antoine Pitrou7ded21e2011-12-19 16:21:21 +01001037 assertInvalid(g, b'\x00' * 3)
1038 assertInvalid(g, b'\x00' * 5)
1039 assertInvalid(g, b'\x00' * 16)
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00001040
1041 def testStringToIPv6(self):
Guido van Rossumf4001ee2003-04-25 15:11:23 +00001042 if not hasattr(socket, 'inet_ntop'):
1043 return # No inet_ntop() on this platform
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00001044 try:
1045 from socket import inet_ntop, AF_INET6, has_ipv6
1046 if not has_ipv6:
1047 return
1048 except ImportError:
1049 return
1050 f = lambda a: inet_ntop(AF_INET6, a)
Antoine Pitrou7ded21e2011-12-19 16:21:21 +01001051 assertInvalid = lambda a: self.assertRaises(
Andrew Svetlov0832af62012-12-18 23:10:48 +02001052 (OSError, ValueError), f, a
Antoine Pitrou7ded21e2011-12-19 16:21:21 +01001053 )
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00001054
Ezio Melottib3aedd42010-11-20 19:04:17 +00001055 self.assertEqual('::', f(b'\x00' * 16))
1056 self.assertEqual('::1', f(b'\x00' * 15 + b'\x01'))
1057 self.assertEqual(
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00001058 'aef:b01:506:1001:ffff:9997:55:170',
Guido van Rossum67180622007-07-10 07:29:12 +00001059 f(b'\x0a\xef\x0b\x01\x05\x06\x10\x01\xff\xff\x99\x97\x00\x55\x01\x70')
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00001060 )
1061
Antoine Pitrou7ded21e2011-12-19 16:21:21 +01001062 assertInvalid(b'\x12' * 15)
1063 assertInvalid(b'\x12' * 17)
1064 assertInvalid(b'\x12' * 4)
1065
Guido van Rossumb6cc7d22002-07-19 12:46:46 +00001066 # XXX The following don't test module-level functionality...
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00001067
Benjamin Peterson5c6d7872009-02-06 02:40:07 +00001068 def testSockName(self):
1069 # Testing getsockname()
Antoine Pitrouccc87b52011-05-10 23:37:11 +02001070 port = support.find_unused_port()
Guido van Rossum24e4af82002-06-12 19:18:08 +00001071 sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
Brian Curtin21ebbb72010-11-02 02:07:09 +00001072 self.addCleanup(sock.close)
Christian Heimes5e696852008-04-09 08:37:03 +00001073 sock.bind(("0.0.0.0", port))
Guido van Rossum24e4af82002-06-12 19:18:08 +00001074 name = sock.getsockname()
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001075 # XXX(nnorwitz): http://tinyurl.com/os5jz seems to indicate
1076 # it reasonable to get the host's addr in addition to 0.0.0.0.
1077 # At least for eCos. This is required for the S/390 to pass.
Georg Brandl89197fe2010-11-20 14:16:17 +00001078 try:
1079 my_ip_addr = socket.gethostbyname(socket.gethostname())
Andrew Svetlov0832af62012-12-18 23:10:48 +02001080 except OSError:
Georg Brandl89197fe2010-11-20 14:16:17 +00001081 # Probably name lookup wasn't set up right; skip this test
1082 return
Benjamin Peterson577473f2010-01-19 00:09:57 +00001083 self.assertIn(name[0], ("0.0.0.0", my_ip_addr), '%s invalid' % name[0])
Christian Heimes5e696852008-04-09 08:37:03 +00001084 self.assertEqual(name[1], port)
Guido van Rossum24e4af82002-06-12 19:18:08 +00001085
1086 def testGetSockOpt(self):
Guido van Rossum8c943832002-08-08 01:00:28 +00001087 # Testing getsockopt()
Guido van Rossum24e4af82002-06-12 19:18:08 +00001088 # We know a socket should start without reuse==0
1089 sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
Brian Curtin21ebbb72010-11-02 02:07:09 +00001090 self.addCleanup(sock.close)
Guido van Rossum24e4af82002-06-12 19:18:08 +00001091 reuse = sock.getsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +00001092 self.assertFalse(reuse != 0, "initial mode is reuse")
Guido van Rossum24e4af82002-06-12 19:18:08 +00001093
1094 def testSetSockOpt(self):
Guido van Rossum8c943832002-08-08 01:00:28 +00001095 # Testing setsockopt()
Guido van Rossum24e4af82002-06-12 19:18:08 +00001096 sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
Brian Curtin21ebbb72010-11-02 02:07:09 +00001097 self.addCleanup(sock.close)
Guido van Rossum24e4af82002-06-12 19:18:08 +00001098 sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
1099 reuse = sock.getsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +00001100 self.assertFalse(reuse == 0, "failed to set reuse mode")
Guido van Rossum24e4af82002-06-12 19:18:08 +00001101
Guido van Rossumb6cc7d22002-07-19 12:46:46 +00001102 def testSendAfterClose(self):
Guido van Rossum8c943832002-08-08 01:00:28 +00001103 # testing send() after close() with timeout
Guido van Rossumb6cc7d22002-07-19 12:46:46 +00001104 sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
1105 sock.settimeout(1)
1106 sock.close()
Andrew Svetlov0832af62012-12-18 23:10:48 +02001107 self.assertRaises(OSError, sock.send, b"spam")
Guido van Rossumb6cc7d22002-07-19 12:46:46 +00001108
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001109 def testNewAttributes(self):
1110 # testing .family, .type and .protocol
1111 sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
1112 self.assertEqual(sock.family, socket.AF_INET)
1113 self.assertEqual(sock.type, socket.SOCK_STREAM)
1114 self.assertEqual(sock.proto, 0)
1115 sock.close()
1116
Benjamin Peterson5c6d7872009-02-06 02:40:07 +00001117 def test_getsockaddrarg(self):
1118 host = '0.0.0.0'
Antoine Pitrouccc87b52011-05-10 23:37:11 +02001119 port = support.find_unused_port()
Benjamin Peterson5c6d7872009-02-06 02:40:07 +00001120 big_port = port + 65536
1121 neg_port = port - 65536
1122 sock = socket.socket()
1123 try:
1124 self.assertRaises(OverflowError, sock.bind, (host, big_port))
1125 self.assertRaises(OverflowError, sock.bind, (host, neg_port))
1126 sock.bind((host, port))
1127 finally:
1128 sock.close()
1129
Brian Curtin5ad0bd62010-11-05 15:47:45 +00001130 @unittest.skipUnless(os.name == "nt", "Windows specific")
Christian Heimesfaf2f632008-01-06 16:59:19 +00001131 def test_sock_ioctl(self):
Benjamin Petersonc9c0f202009-06-30 23:06:06 +00001132 self.assertTrue(hasattr(socket.socket, 'ioctl'))
1133 self.assertTrue(hasattr(socket, 'SIO_RCVALL'))
1134 self.assertTrue(hasattr(socket, 'RCVALL_ON'))
1135 self.assertTrue(hasattr(socket, 'RCVALL_OFF'))
Kristján Valur Jónsson847ec752009-09-27 21:10:38 +00001136 self.assertTrue(hasattr(socket, 'SIO_KEEPALIVE_VALS'))
1137 s = socket.socket()
Brian Curtin5ad0bd62010-11-05 15:47:45 +00001138 self.addCleanup(s.close)
Kristján Valur Jónsson847ec752009-09-27 21:10:38 +00001139 self.assertRaises(ValueError, s.ioctl, -1, None)
1140 s.ioctl(socket.SIO_KEEPALIVE_VALS, (1, 100, 100))
Christian Heimesfaf2f632008-01-06 16:59:19 +00001141
Giampaolo Rodolà419f7042010-08-14 16:45:41 +00001142 def testGetaddrinfo(self):
1143 try:
1144 socket.getaddrinfo('localhost', 80)
1145 except socket.gaierror as err:
1146 if err.errno == socket.EAI_SERVICE:
1147 # see http://bugs.python.org/issue1282647
1148 self.skipTest("buggy libc version")
1149 raise
1150 # len of every sequence is supposed to be == 5
1151 for info in socket.getaddrinfo(HOST, None):
1152 self.assertEqual(len(info), 5)
1153 # host can be a domain name, a string representation of an
1154 # IPv4/v6 address or None
1155 socket.getaddrinfo('localhost', 80)
1156 socket.getaddrinfo('127.0.0.1', 80)
1157 socket.getaddrinfo(None, 80)
Antoine Pitrou9c39f3c2011-04-28 19:18:10 +02001158 if support.IPV6_ENABLED:
Giampaolo Rodolà419f7042010-08-14 16:45:41 +00001159 socket.getaddrinfo('::1', 80)
1160 # port can be a string service name such as "http", a numeric
1161 # port number or None
1162 socket.getaddrinfo(HOST, "http")
1163 socket.getaddrinfo(HOST, 80)
1164 socket.getaddrinfo(HOST, None)
1165 # test family and socktype filters
1166 infos = socket.getaddrinfo(HOST, None, socket.AF_INET)
1167 for family, _, _, _, _ in infos:
1168 self.assertEqual(family, socket.AF_INET)
1169 infos = socket.getaddrinfo(HOST, None, 0, socket.SOCK_STREAM)
1170 for _, socktype, _, _, _ in infos:
1171 self.assertEqual(socktype, socket.SOCK_STREAM)
1172 # test proto and flags arguments
Giampaolo Rodolà677d95c2010-08-16 05:08:11 +00001173 socket.getaddrinfo(HOST, None, 0, 0, socket.SOL_TCP)
Giampaolo Rodolà419f7042010-08-14 16:45:41 +00001174 socket.getaddrinfo(HOST, None, 0, 0, 0, socket.AI_PASSIVE)
1175 # a server willing to support both IPv4 and IPv6 will
1176 # usually do this
1177 socket.getaddrinfo(None, 0, socket.AF_UNSPEC, socket.SOCK_STREAM, 0,
1178 socket.AI_PASSIVE)
Giampaolo Rodolàccfb91c2010-08-17 15:30:23 +00001179 # test keyword arguments
1180 a = socket.getaddrinfo(HOST, None)
1181 b = socket.getaddrinfo(host=HOST, port=None)
1182 self.assertEqual(a, b)
1183 a = socket.getaddrinfo(HOST, None, socket.AF_INET)
1184 b = socket.getaddrinfo(HOST, None, family=socket.AF_INET)
1185 self.assertEqual(a, b)
1186 a = socket.getaddrinfo(HOST, None, 0, socket.SOCK_STREAM)
1187 b = socket.getaddrinfo(HOST, None, type=socket.SOCK_STREAM)
1188 self.assertEqual(a, b)
1189 a = socket.getaddrinfo(HOST, None, 0, 0, socket.SOL_TCP)
1190 b = socket.getaddrinfo(HOST, None, proto=socket.SOL_TCP)
1191 self.assertEqual(a, b)
1192 a = socket.getaddrinfo(HOST, None, 0, 0, 0, socket.AI_PASSIVE)
1193 b = socket.getaddrinfo(HOST, None, flags=socket.AI_PASSIVE)
1194 self.assertEqual(a, b)
1195 a = socket.getaddrinfo(None, 0, socket.AF_UNSPEC, socket.SOCK_STREAM, 0,
1196 socket.AI_PASSIVE)
1197 b = socket.getaddrinfo(host=None, port=0, family=socket.AF_UNSPEC,
1198 type=socket.SOCK_STREAM, proto=0,
1199 flags=socket.AI_PASSIVE)
1200 self.assertEqual(a, b)
Alexander Belopolskye239d232010-12-08 23:31:48 +00001201 # Issue #6697.
1202 self.assertRaises(UnicodeEncodeError, socket.getaddrinfo, 'localhost', '\uD800')
Giampaolo Rodolà419f7042010-08-14 16:45:41 +00001203
Martin v. Löwis112c0f32010-08-25 07:38:15 +00001204 def test_getnameinfo(self):
1205 # only IP addresses are allowed
Andrew Svetlov0832af62012-12-18 23:10:48 +02001206 self.assertRaises(OSError, socket.getnameinfo, ('mail.python.org',0), 0)
Martin v. Löwis112c0f32010-08-25 07:38:15 +00001207
Daniel Stutzbach460ff3d2010-09-09 21:17:58 +00001208 @unittest.skipUnless(support.is_resource_enabled('network'),
1209 'network is not enabled')
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00001210 def test_idna(self):
Nadeem Vawda61baebd2012-01-25 08:02:05 +02001211 # Check for internet access before running test (issue #12804).
1212 try:
1213 socket.gethostbyname('python.org')
1214 except socket.gaierror as e:
1215 if e.errno == socket.EAI_NODATA:
1216 self.skipTest('internet access required for this test')
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00001217 # these should all be successful
1218 socket.gethostbyname('испытание.python.org')
1219 socket.gethostbyname_ex('испытание.python.org')
Martin v. Löwis67e91ad2010-08-23 15:27:26 +00001220 socket.getaddrinfo('испытание.python.org',0,socket.AF_UNSPEC,socket.SOCK_STREAM)
1221 # this may not work if the forward lookup choses the IPv6 address, as that doesn't
1222 # have a reverse entry yet
1223 # socket.gethostbyaddr('испытание.python.org')
Christian Heimesfaf2f632008-01-06 16:59:19 +00001224
Antoine Pitrou6d7df632010-09-27 17:52:25 +00001225 def check_sendall_interrupted(self, with_timeout):
1226 # socketpair() is not stricly required, but it makes things easier.
1227 if not hasattr(signal, 'alarm') or not hasattr(socket, 'socketpair'):
1228 self.skipTest("signal.alarm and socket.socketpair required for this test")
1229 # Our signal handlers clobber the C errno by calling a math function
1230 # with an invalid domain value.
1231 def ok_handler(*args):
1232 self.assertRaises(ValueError, math.acosh, 0)
1233 def raising_handler(*args):
1234 self.assertRaises(ValueError, math.acosh, 0)
1235 1 // 0
1236 c, s = socket.socketpair()
1237 old_alarm = signal.signal(signal.SIGALRM, raising_handler)
1238 try:
1239 if with_timeout:
1240 # Just above the one second minimum for signal.alarm
1241 c.settimeout(1.5)
1242 with self.assertRaises(ZeroDivisionError):
1243 signal.alarm(1)
1244 c.sendall(b"x" * (1024**2))
1245 if with_timeout:
1246 signal.signal(signal.SIGALRM, ok_handler)
1247 signal.alarm(1)
1248 self.assertRaises(socket.timeout, c.sendall, b"x" * (1024**2))
1249 finally:
1250 signal.signal(signal.SIGALRM, old_alarm)
1251 c.close()
1252 s.close()
1253
1254 def test_sendall_interrupted(self):
1255 self.check_sendall_interrupted(False)
1256
1257 def test_sendall_interrupted_with_timeout(self):
1258 self.check_sendall_interrupted(True)
1259
Antoine Pitroue033e062010-10-29 10:38:18 +00001260 def test_dealloc_warn(self):
1261 sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
1262 r = repr(sock)
1263 with self.assertWarns(ResourceWarning) as cm:
1264 sock = None
1265 support.gc_collect()
1266 self.assertIn(r, str(cm.warning.args[0]))
1267 # An open socket file object gets dereferenced after the socket
1268 sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
1269 f = sock.makefile('rb')
1270 r = repr(sock)
1271 sock = None
1272 support.gc_collect()
1273 with self.assertWarns(ResourceWarning):
1274 f = None
1275 support.gc_collect()
1276
Victor Stinnerc3a51ec2011-01-04 11:00:45 +00001277 def test_name_closed_socketio(self):
1278 with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as sock:
1279 fp = sock.makefile("rb")
1280 fp.close()
1281 self.assertEqual(repr(fp), "<_io.BufferedReader name=-1>")
1282
Antoine Pitrou1e7ee9d2012-09-14 17:28:10 +02001283 def test_unusable_closed_socketio(self):
1284 with socket.socket() as sock:
1285 fp = sock.makefile("rb", buffering=0)
1286 self.assertTrue(fp.readable())
1287 self.assertFalse(fp.writable())
1288 self.assertFalse(fp.seekable())
1289 fp.close()
1290 self.assertRaises(ValueError, fp.readable)
1291 self.assertRaises(ValueError, fp.writable)
1292 self.assertRaises(ValueError, fp.seekable)
1293
Antoine Pitrou6d58d642011-03-20 23:56:36 +01001294 def test_pickle(self):
1295 sock = socket.socket()
1296 with sock:
1297 for protocol in range(pickle.HIGHEST_PROTOCOL + 1):
1298 self.assertRaises(TypeError, pickle.dumps, sock, protocol)
1299
Serhiy Storchaka78980432013-01-15 01:12:17 +02001300 def test_listen_backlog(self):
1301 for backlog in 0, -1:
1302 srv = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
1303 srv.bind((HOST, 0))
1304 srv.listen(backlog)
1305 srv.close()
1306
1307 # Issue 15989
Antoine Pitrou1be815a2011-05-10 19:16:29 +02001308 srv = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
1309 srv.bind((HOST, 0))
Serhiy Storchaka78980432013-01-15 01:12:17 +02001310 self.assertRaises(OverflowError, srv.listen, _testcapi.INT_MAX + 1)
Antoine Pitrou1be815a2011-05-10 19:16:29 +02001311 srv.close()
1312
Charles-François Natali42663332012-01-02 15:57:30 +01001313 @unittest.skipUnless(support.IPV6_ENABLED, 'IPv6 required for this test.')
Charles-François Natali366999a2012-01-02 15:47:29 +01001314 def test_flowinfo(self):
1315 self.assertRaises(OverflowError, socket.getnameinfo,
1316 ('::1',0, 0xffffffff), 0)
1317 with socket.socket(socket.AF_INET6, socket.SOCK_STREAM) as s:
1318 self.assertRaises(OverflowError, s.bind, ('::1', 0, -10))
1319
Antoine Pitrou6d7df632010-09-27 17:52:25 +00001320
Charles-François Natali47413c12011-10-06 19:47:44 +02001321@unittest.skipUnless(HAVE_SOCKET_CAN, 'SocketCan required for this test.')
1322class BasicCANTest(unittest.TestCase):
1323
1324 def testCrucialConstants(self):
1325 socket.AF_CAN
1326 socket.PF_CAN
1327 socket.CAN_RAW
1328
Charles-François Natali773e42d2013-02-05 19:42:01 +01001329 @unittest.skipUnless(hasattr(socket, "CAN_BCM"),
1330 'socket.CAN_BCM required for this test.')
1331 def testBCMConstants(self):
1332 socket.CAN_BCM
1333
1334 # opcodes
1335 socket.CAN_BCM_TX_SETUP # create (cyclic) transmission task
1336 socket.CAN_BCM_TX_DELETE # remove (cyclic) transmission task
1337 socket.CAN_BCM_TX_READ # read properties of (cyclic) transmission task
1338 socket.CAN_BCM_TX_SEND # send one CAN frame
1339 socket.CAN_BCM_RX_SETUP # create RX content filter subscription
1340 socket.CAN_BCM_RX_DELETE # remove RX content filter subscription
1341 socket.CAN_BCM_RX_READ # read properties of RX content filter subscription
1342 socket.CAN_BCM_TX_STATUS # reply to TX_READ request
1343 socket.CAN_BCM_TX_EXPIRED # notification on performed transmissions (count=0)
1344 socket.CAN_BCM_RX_STATUS # reply to RX_READ request
1345 socket.CAN_BCM_RX_TIMEOUT # cyclic message is absent
1346 socket.CAN_BCM_RX_CHANGED # updated CAN frame (detected content change)
1347
Charles-François Natali47413c12011-10-06 19:47:44 +02001348 def testCreateSocket(self):
1349 with socket.socket(socket.PF_CAN, socket.SOCK_RAW, socket.CAN_RAW) as s:
1350 pass
1351
Charles-François Natali773e42d2013-02-05 19:42:01 +01001352 @unittest.skipUnless(hasattr(socket, "CAN_BCM"),
1353 'socket.CAN_BCM required for this test.')
1354 def testCreateBCMSocket(self):
1355 with socket.socket(socket.PF_CAN, socket.SOCK_DGRAM, socket.CAN_BCM) as s:
1356 pass
1357
Charles-François Natali47413c12011-10-06 19:47:44 +02001358 def testBindAny(self):
1359 with socket.socket(socket.PF_CAN, socket.SOCK_RAW, socket.CAN_RAW) as s:
1360 s.bind(('', ))
1361
1362 def testTooLongInterfaceName(self):
1363 # most systems limit IFNAMSIZ to 16, take 1024 to be sure
1364 with socket.socket(socket.PF_CAN, socket.SOCK_RAW, socket.CAN_RAW) as s:
Andrew Svetlov0832af62012-12-18 23:10:48 +02001365 self.assertRaisesRegex(OSError, 'interface name too long',
Antoine Pitrouffa547e2011-10-11 22:43:37 +02001366 s.bind, ('x' * 1024,))
Charles-François Natali47413c12011-10-06 19:47:44 +02001367
1368 @unittest.skipUnless(hasattr(socket, "CAN_RAW_LOOPBACK"),
1369 'socket.CAN_RAW_LOOPBACK required for this test.')
1370 def testLoopback(self):
1371 with socket.socket(socket.PF_CAN, socket.SOCK_RAW, socket.CAN_RAW) as s:
1372 for loopback in (0, 1):
1373 s.setsockopt(socket.SOL_CAN_RAW, socket.CAN_RAW_LOOPBACK,
1374 loopback)
1375 self.assertEqual(loopback,
1376 s.getsockopt(socket.SOL_CAN_RAW, socket.CAN_RAW_LOOPBACK))
1377
1378 @unittest.skipUnless(hasattr(socket, "CAN_RAW_FILTER"),
1379 'socket.CAN_RAW_FILTER required for this test.')
1380 def testFilter(self):
1381 can_id, can_mask = 0x200, 0x700
1382 can_filter = struct.pack("=II", can_id, can_mask)
1383 with socket.socket(socket.PF_CAN, socket.SOCK_RAW, socket.CAN_RAW) as s:
1384 s.setsockopt(socket.SOL_CAN_RAW, socket.CAN_RAW_FILTER, can_filter)
1385 self.assertEqual(can_filter,
1386 s.getsockopt(socket.SOL_CAN_RAW, socket.CAN_RAW_FILTER, 8))
1387
1388
1389@unittest.skipUnless(HAVE_SOCKET_CAN, 'SocketCan required for this test.')
Charles-François Natali47413c12011-10-06 19:47:44 +02001390class CANTest(ThreadedCANSocketTest):
1391
Charles-François Natali47413c12011-10-06 19:47:44 +02001392 def __init__(self, methodName='runTest'):
1393 ThreadedCANSocketTest.__init__(self, methodName=methodName)
1394
1395 @classmethod
1396 def build_can_frame(cls, can_id, data):
1397 """Build a CAN frame."""
1398 can_dlc = len(data)
1399 data = data.ljust(8, b'\x00')
1400 return struct.pack(cls.can_frame_fmt, can_id, can_dlc, data)
1401
1402 @classmethod
1403 def dissect_can_frame(cls, frame):
1404 """Dissect a CAN frame."""
1405 can_id, can_dlc, data = struct.unpack(cls.can_frame_fmt, frame)
1406 return (can_id, can_dlc, data[:can_dlc])
1407
1408 def testSendFrame(self):
1409 cf, addr = self.s.recvfrom(self.bufsize)
1410 self.assertEqual(self.cf, cf)
1411 self.assertEqual(addr[0], self.interface)
1412 self.assertEqual(addr[1], socket.AF_CAN)
1413
1414 def _testSendFrame(self):
1415 self.cf = self.build_can_frame(0x00, b'\x01\x02\x03\x04\x05')
1416 self.cli.send(self.cf)
1417
1418 def testSendMaxFrame(self):
1419 cf, addr = self.s.recvfrom(self.bufsize)
1420 self.assertEqual(self.cf, cf)
1421
1422 def _testSendMaxFrame(self):
1423 self.cf = self.build_can_frame(0x00, b'\x07' * 8)
1424 self.cli.send(self.cf)
1425
1426 def testSendMultiFrames(self):
1427 cf, addr = self.s.recvfrom(self.bufsize)
1428 self.assertEqual(self.cf1, cf)
1429
1430 cf, addr = self.s.recvfrom(self.bufsize)
1431 self.assertEqual(self.cf2, cf)
1432
1433 def _testSendMultiFrames(self):
1434 self.cf1 = self.build_can_frame(0x07, b'\x44\x33\x22\x11')
1435 self.cli.send(self.cf1)
1436
1437 self.cf2 = self.build_can_frame(0x12, b'\x99\x22\x33')
1438 self.cli.send(self.cf2)
1439
Charles-François Natali773e42d2013-02-05 19:42:01 +01001440 @unittest.skipUnless(hasattr(socket, "CAN_BCM"),
1441 'socket.CAN_BCM required for this test.')
1442 def _testBCM(self):
1443 cf, addr = self.cli.recvfrom(self.bufsize)
1444 self.assertEqual(self.cf, cf)
1445 can_id, can_dlc, data = self.dissect_can_frame(cf)
1446 self.assertEqual(self.can_id, can_id)
1447 self.assertEqual(self.data, data)
1448
1449 @unittest.skipUnless(hasattr(socket, "CAN_BCM"),
1450 'socket.CAN_BCM required for this test.')
1451 def testBCM(self):
1452 bcm = socket.socket(socket.PF_CAN, socket.SOCK_DGRAM, socket.CAN_BCM)
1453 self.addCleanup(bcm.close)
1454 bcm.connect((self.interface,))
1455 self.can_id = 0x123
1456 self.data = bytes([0xc0, 0xff, 0xee])
1457 self.cf = self.build_can_frame(self.can_id, self.data)
1458 opcode = socket.CAN_BCM_TX_SEND
1459 flags = 0
1460 count = 0
1461 ival1_seconds = ival1_usec = ival2_seconds = ival2_usec = 0
1462 bcm_can_id = 0x0222
1463 nframes = 1
1464 assert len(self.cf) == 16
1465 header = struct.pack(self.bcm_cmd_msg_fmt,
1466 opcode,
1467 flags,
1468 count,
1469 ival1_seconds,
1470 ival1_usec,
1471 ival2_seconds,
1472 ival2_usec,
1473 bcm_can_id,
1474 nframes,
1475 )
1476 header_plus_frame = header + self.cf
1477 bytes_sent = bcm.send(header_plus_frame)
1478 self.assertEqual(bytes_sent, len(header_plus_frame))
1479
Charles-François Natali47413c12011-10-06 19:47:44 +02001480
Charles-François Natali10b8cf42011-11-10 19:21:37 +01001481@unittest.skipUnless(HAVE_SOCKET_RDS, 'RDS sockets required for this test.')
1482class BasicRDSTest(unittest.TestCase):
1483
1484 def testCrucialConstants(self):
1485 socket.AF_RDS
1486 socket.PF_RDS
1487
1488 def testCreateSocket(self):
1489 with socket.socket(socket.PF_RDS, socket.SOCK_SEQPACKET, 0) as s:
1490 pass
1491
1492 def testSocketBufferSize(self):
1493 bufsize = 16384
1494 with socket.socket(socket.PF_RDS, socket.SOCK_SEQPACKET, 0) as s:
1495 s.setsockopt(socket.SOL_SOCKET, socket.SO_RCVBUF, bufsize)
1496 s.setsockopt(socket.SOL_SOCKET, socket.SO_SNDBUF, bufsize)
1497
1498
1499@unittest.skipUnless(HAVE_SOCKET_RDS, 'RDS sockets required for this test.')
1500@unittest.skipUnless(thread, 'Threading required for this test.')
1501class RDSTest(ThreadedRDSSocketTest):
1502
1503 def __init__(self, methodName='runTest'):
1504 ThreadedRDSSocketTest.__init__(self, methodName=methodName)
1505
Charles-François Natali240c55f2011-11-10 20:33:36 +01001506 def setUp(self):
1507 super().setUp()
1508 self.evt = threading.Event()
1509
Charles-François Natali10b8cf42011-11-10 19:21:37 +01001510 def testSendAndRecv(self):
1511 data, addr = self.serv.recvfrom(self.bufsize)
1512 self.assertEqual(self.data, data)
1513 self.assertEqual(self.cli_addr, addr)
1514
1515 def _testSendAndRecv(self):
1516 self.data = b'spam'
1517 self.cli.sendto(self.data, 0, (HOST, self.port))
1518
1519 def testPeek(self):
1520 data, addr = self.serv.recvfrom(self.bufsize, socket.MSG_PEEK)
1521 self.assertEqual(self.data, data)
1522 data, addr = self.serv.recvfrom(self.bufsize)
1523 self.assertEqual(self.data, data)
1524
1525 def _testPeek(self):
1526 self.data = b'spam'
1527 self.cli.sendto(self.data, 0, (HOST, self.port))
1528
1529 @requireAttrs(socket.socket, 'recvmsg')
1530 def testSendAndRecvMsg(self):
1531 data, ancdata, msg_flags, addr = self.serv.recvmsg(self.bufsize)
1532 self.assertEqual(self.data, data)
1533
1534 @requireAttrs(socket.socket, 'sendmsg')
1535 def _testSendAndRecvMsg(self):
1536 self.data = b'hello ' * 10
1537 self.cli.sendmsg([self.data], (), 0, (HOST, self.port))
1538
1539 def testSendAndRecvMulti(self):
1540 data, addr = self.serv.recvfrom(self.bufsize)
1541 self.assertEqual(self.data1, data)
1542
1543 data, addr = self.serv.recvfrom(self.bufsize)
1544 self.assertEqual(self.data2, data)
1545
1546 def _testSendAndRecvMulti(self):
1547 self.data1 = b'bacon'
1548 self.cli.sendto(self.data1, 0, (HOST, self.port))
1549
1550 self.data2 = b'egg'
1551 self.cli.sendto(self.data2, 0, (HOST, self.port))
1552
1553 def testSelect(self):
1554 r, w, x = select.select([self.serv], [], [], 3.0)
1555 self.assertIn(self.serv, r)
1556 data, addr = self.serv.recvfrom(self.bufsize)
1557 self.assertEqual(self.data, data)
1558
1559 def _testSelect(self):
1560 self.data = b'select'
1561 self.cli.sendto(self.data, 0, (HOST, self.port))
1562
1563 def testCongestion(self):
1564 # wait until the sender is done
1565 self.evt.wait()
1566
1567 def _testCongestion(self):
1568 # test the behavior in case of congestion
1569 self.data = b'fill'
1570 self.cli.setblocking(False)
1571 try:
1572 # try to lower the receiver's socket buffer size
1573 self.cli.setsockopt(socket.SOL_SOCKET, socket.SO_RCVBUF, 16384)
1574 except OSError:
1575 pass
1576 with self.assertRaises(OSError) as cm:
1577 try:
1578 # fill the receiver's socket buffer
1579 while True:
1580 self.cli.sendto(self.data, 0, (HOST, self.port))
1581 finally:
1582 # signal the receiver we're done
1583 self.evt.set()
1584 # sendto() should have failed with ENOBUFS
1585 self.assertEqual(cm.exception.errno, errno.ENOBUFS)
1586 # and we should have received a congestion notification through poll
1587 r, w, x = select.select([self.serv], [], [], 3.0)
1588 self.assertIn(self.serv, r)
1589
1590
Victor Stinner45df8202010-04-28 22:31:17 +00001591@unittest.skipUnless(thread, 'Threading required for this test.')
Guido van Rossum24e4af82002-06-12 19:18:08 +00001592class BasicTCPTest(SocketConnectedTest):
1593
1594 def __init__(self, methodName='runTest'):
1595 SocketConnectedTest.__init__(self, methodName=methodName)
1596
1597 def testRecv(self):
Guido van Rossum8c943832002-08-08 01:00:28 +00001598 # Testing large receive over TCP
Guido van Rossum24e4af82002-06-12 19:18:08 +00001599 msg = self.cli_conn.recv(1024)
Guido van Rossum76489682002-06-12 20:38:30 +00001600 self.assertEqual(msg, MSG)
Guido van Rossum24e4af82002-06-12 19:18:08 +00001601
1602 def _testRecv(self):
1603 self.serv_conn.send(MSG)
1604
1605 def testOverFlowRecv(self):
Guido van Rossum8c943832002-08-08 01:00:28 +00001606 # Testing receive in chunks over TCP
Guido van Rossum24e4af82002-06-12 19:18:08 +00001607 seg1 = self.cli_conn.recv(len(MSG) - 3)
1608 seg2 = self.cli_conn.recv(1024)
Guido van Rossumab659962002-06-12 21:29:43 +00001609 msg = seg1 + seg2
Guido van Rossum76489682002-06-12 20:38:30 +00001610 self.assertEqual(msg, MSG)
Guido van Rossum24e4af82002-06-12 19:18:08 +00001611
1612 def _testOverFlowRecv(self):
1613 self.serv_conn.send(MSG)
1614
1615 def testRecvFrom(self):
Guido van Rossum8c943832002-08-08 01:00:28 +00001616 # Testing large recvfrom() over TCP
Guido van Rossum24e4af82002-06-12 19:18:08 +00001617 msg, addr = self.cli_conn.recvfrom(1024)
Guido van Rossum76489682002-06-12 20:38:30 +00001618 self.assertEqual(msg, MSG)
Guido van Rossum24e4af82002-06-12 19:18:08 +00001619
1620 def _testRecvFrom(self):
1621 self.serv_conn.send(MSG)
1622
1623 def testOverFlowRecvFrom(self):
Guido van Rossum8c943832002-08-08 01:00:28 +00001624 # Testing recvfrom() in chunks over TCP
Guido van Rossum24e4af82002-06-12 19:18:08 +00001625 seg1, addr = self.cli_conn.recvfrom(len(MSG)-3)
1626 seg2, addr = self.cli_conn.recvfrom(1024)
Guido van Rossumab659962002-06-12 21:29:43 +00001627 msg = seg1 + seg2
Guido van Rossum76489682002-06-12 20:38:30 +00001628 self.assertEqual(msg, MSG)
Guido van Rossum24e4af82002-06-12 19:18:08 +00001629
1630 def _testOverFlowRecvFrom(self):
1631 self.serv_conn.send(MSG)
1632
1633 def testSendAll(self):
Guido van Rossum8c943832002-08-08 01:00:28 +00001634 # Testing sendall() with a 2048 byte string over TCP
Guido van Rossum7d0a8262007-05-21 23:13:11 +00001635 msg = b''
Guido van Rossum24e4af82002-06-12 19:18:08 +00001636 while 1:
1637 read = self.cli_conn.recv(1024)
1638 if not read:
1639 break
Guido van Rossume531e292002-08-08 20:28:34 +00001640 msg += read
Guido van Rossum7d0a8262007-05-21 23:13:11 +00001641 self.assertEqual(msg, b'f' * 2048)
Guido van Rossum24e4af82002-06-12 19:18:08 +00001642
1643 def _testSendAll(self):
Guido van Rossum7d0a8262007-05-21 23:13:11 +00001644 big_chunk = b'f' * 2048
Guido van Rossum24e4af82002-06-12 19:18:08 +00001645 self.serv_conn.sendall(big_chunk)
1646
1647 def testFromFd(self):
Guido van Rossum8c943832002-08-08 01:00:28 +00001648 # Testing fromfd()
Guido van Rossum24e4af82002-06-12 19:18:08 +00001649 fd = self.cli_conn.fileno()
1650 sock = socket.fromfd(fd, socket.AF_INET, socket.SOCK_STREAM)
Brian Curtin21ebbb72010-11-02 02:07:09 +00001651 self.addCleanup(sock.close)
Antoine Pitrou9e0b8642010-09-14 18:00:02 +00001652 self.assertIsInstance(sock, socket.socket)
Guido van Rossum24e4af82002-06-12 19:18:08 +00001653 msg = sock.recv(1024)
Guido van Rossum76489682002-06-12 20:38:30 +00001654 self.assertEqual(msg, MSG)
Guido van Rossum24e4af82002-06-12 19:18:08 +00001655
1656 def _testFromFd(self):
1657 self.serv_conn.send(MSG)
1658
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00001659 def testDup(self):
1660 # Testing dup()
1661 sock = self.cli_conn.dup()
Brian Curtin21ebbb72010-11-02 02:07:09 +00001662 self.addCleanup(sock.close)
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00001663 msg = sock.recv(1024)
1664 self.assertEqual(msg, MSG)
1665
1666 def _testDup(self):
1667 self.serv_conn.send(MSG)
1668
Guido van Rossum24e4af82002-06-12 19:18:08 +00001669 def testShutdown(self):
Guido van Rossum8c943832002-08-08 01:00:28 +00001670 # Testing shutdown()
Guido van Rossum24e4af82002-06-12 19:18:08 +00001671 msg = self.cli_conn.recv(1024)
Guido van Rossum76489682002-06-12 20:38:30 +00001672 self.assertEqual(msg, MSG)
Mark Dickinson29500f62009-01-15 15:36:10 +00001673 # wait for _testShutdown to finish: on OS X, when the server
1674 # closes the connection the client also becomes disconnected,
1675 # and the client's shutdown call will fail. (Issue #4397.)
1676 self.done.wait()
Guido van Rossum24e4af82002-06-12 19:18:08 +00001677
1678 def _testShutdown(self):
1679 self.serv_conn.send(MSG)
Serhiy Storchaka78980432013-01-15 01:12:17 +02001680 # Issue 15989
1681 self.assertRaises(OverflowError, self.serv_conn.shutdown,
1682 _testcapi.INT_MAX + 1)
1683 self.assertRaises(OverflowError, self.serv_conn.shutdown,
1684 2 + (_testcapi.UINT_MAX + 1))
Guido van Rossum24e4af82002-06-12 19:18:08 +00001685 self.serv_conn.shutdown(2)
1686
Antoine Pitrou6e451df2010-08-09 20:39:54 +00001687 def testDetach(self):
1688 # Testing detach()
1689 fileno = self.cli_conn.fileno()
1690 f = self.cli_conn.detach()
1691 self.assertEqual(f, fileno)
1692 # cli_conn cannot be used anymore...
Antoine Pitrou70deb3d2012-04-01 01:00:17 +02001693 self.assertTrue(self.cli_conn._closed)
Andrew Svetlov0832af62012-12-18 23:10:48 +02001694 self.assertRaises(OSError, self.cli_conn.recv, 1024)
Antoine Pitroue43f9d02010-08-08 23:24:50 +00001695 self.cli_conn.close()
Antoine Pitrou6e451df2010-08-09 20:39:54 +00001696 # ...but we can create another socket using the (still open)
1697 # file descriptor
Antoine Pitroue43f9d02010-08-08 23:24:50 +00001698 sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM, fileno=f)
Brian Curtin21ebbb72010-11-02 02:07:09 +00001699 self.addCleanup(sock.close)
Antoine Pitroue43f9d02010-08-08 23:24:50 +00001700 msg = sock.recv(1024)
1701 self.assertEqual(msg, MSG)
1702
Antoine Pitrou6e451df2010-08-09 20:39:54 +00001703 def _testDetach(self):
Antoine Pitroue43f9d02010-08-08 23:24:50 +00001704 self.serv_conn.send(MSG)
1705
Victor Stinner45df8202010-04-28 22:31:17 +00001706@unittest.skipUnless(thread, 'Threading required for this test.')
Guido van Rossum24e4af82002-06-12 19:18:08 +00001707class BasicUDPTest(ThreadedUDPSocketTest):
1708
1709 def __init__(self, methodName='runTest'):
1710 ThreadedUDPSocketTest.__init__(self, methodName=methodName)
1711
1712 def testSendtoAndRecv(self):
Guido van Rossum8c943832002-08-08 01:00:28 +00001713 # Testing sendto() and Recv() over UDP
Guido van Rossum24e4af82002-06-12 19:18:08 +00001714 msg = self.serv.recv(len(MSG))
Guido van Rossum76489682002-06-12 20:38:30 +00001715 self.assertEqual(msg, MSG)
Guido van Rossum24e4af82002-06-12 19:18:08 +00001716
1717 def _testSendtoAndRecv(self):
Christian Heimes5e696852008-04-09 08:37:03 +00001718 self.cli.sendto(MSG, 0, (HOST, self.port))
Guido van Rossum24e4af82002-06-12 19:18:08 +00001719
Guido van Rossum1c938012002-06-12 21:17:20 +00001720 def testRecvFrom(self):
Guido van Rossum8c943832002-08-08 01:00:28 +00001721 # Testing recvfrom() over UDP
Guido van Rossum24e4af82002-06-12 19:18:08 +00001722 msg, addr = self.serv.recvfrom(len(MSG))
Guido van Rossum76489682002-06-12 20:38:30 +00001723 self.assertEqual(msg, MSG)
Guido van Rossum24e4af82002-06-12 19:18:08 +00001724
Guido van Rossum1c938012002-06-12 21:17:20 +00001725 def _testRecvFrom(self):
Christian Heimes5e696852008-04-09 08:37:03 +00001726 self.cli.sendto(MSG, 0, (HOST, self.port))
Guido van Rossum24e4af82002-06-12 19:18:08 +00001727
Guido van Rossumd8faa362007-04-27 19:54:29 +00001728 def testRecvFromNegative(self):
1729 # Negative lengths passed to recvfrom should give ValueError.
1730 self.assertRaises(ValueError, self.serv.recvfrom, -1)
1731
1732 def _testRecvFromNegative(self):
Christian Heimes5e696852008-04-09 08:37:03 +00001733 self.cli.sendto(MSG, 0, (HOST, self.port))
Guido van Rossumd8faa362007-04-27 19:54:29 +00001734
Nick Coghlan96fe56a2011-08-22 11:55:57 +10001735# Tests for the sendmsg()/recvmsg() interface. Where possible, the
1736# same test code is used with different families and types of socket
1737# (e.g. stream, datagram), and tests using recvmsg() are repeated
1738# using recvmsg_into().
1739#
1740# The generic test classes such as SendmsgTests and
1741# RecvmsgGenericTests inherit from SendrecvmsgBase and expect to be
1742# supplied with sockets cli_sock and serv_sock representing the
1743# client's and the server's end of the connection respectively, and
1744# attributes cli_addr and serv_addr holding their (numeric where
1745# appropriate) addresses.
1746#
1747# The final concrete test classes combine these with subclasses of
1748# SocketTestBase which set up client and server sockets of a specific
1749# type, and with subclasses of SendrecvmsgBase such as
1750# SendrecvmsgDgramBase and SendrecvmsgConnectedBase which map these
1751# sockets to cli_sock and serv_sock and override the methods and
1752# attributes of SendrecvmsgBase to fill in destination addresses if
1753# needed when sending, check for specific flags in msg_flags, etc.
1754#
1755# RecvmsgIntoMixin provides a version of doRecvmsg() implemented using
1756# recvmsg_into().
1757
1758# XXX: like the other datagram (UDP) tests in this module, the code
1759# here assumes that datagram delivery on the local machine will be
1760# reliable.
1761
1762class SendrecvmsgBase(ThreadSafeCleanupTestCase):
1763 # Base class for sendmsg()/recvmsg() tests.
1764
1765 # Time in seconds to wait before considering a test failed, or
1766 # None for no timeout. Not all tests actually set a timeout.
1767 fail_timeout = 3.0
1768
1769 def setUp(self):
1770 self.misc_event = threading.Event()
1771 super().setUp()
1772
1773 def sendToServer(self, msg):
1774 # Send msg to the server.
1775 return self.cli_sock.send(msg)
1776
1777 # Tuple of alternative default arguments for sendmsg() when called
1778 # via sendmsgToServer() (e.g. to include a destination address).
1779 sendmsg_to_server_defaults = ()
1780
1781 def sendmsgToServer(self, *args):
1782 # Call sendmsg() on self.cli_sock with the given arguments,
1783 # filling in any arguments which are not supplied with the
1784 # corresponding items of self.sendmsg_to_server_defaults, if
1785 # any.
1786 return self.cli_sock.sendmsg(
1787 *(args + self.sendmsg_to_server_defaults[len(args):]))
1788
1789 def doRecvmsg(self, sock, bufsize, *args):
1790 # Call recvmsg() on sock with given arguments and return its
1791 # result. Should be used for tests which can use either
1792 # recvmsg() or recvmsg_into() - RecvmsgIntoMixin overrides
1793 # this method with one which emulates it using recvmsg_into(),
1794 # thus allowing the same test to be used for both methods.
1795 result = sock.recvmsg(bufsize, *args)
1796 self.registerRecvmsgResult(result)
1797 return result
1798
1799 def registerRecvmsgResult(self, result):
1800 # Called by doRecvmsg() with the return value of recvmsg() or
1801 # recvmsg_into(). Can be overridden to arrange cleanup based
1802 # on the returned ancillary data, for instance.
1803 pass
1804
1805 def checkRecvmsgAddress(self, addr1, addr2):
1806 # Called to compare the received address with the address of
1807 # the peer.
1808 self.assertEqual(addr1, addr2)
1809
1810 # Flags that are normally unset in msg_flags
1811 msg_flags_common_unset = 0
1812 for name in ("MSG_CTRUNC", "MSG_OOB"):
1813 msg_flags_common_unset |= getattr(socket, name, 0)
1814
1815 # Flags that are normally set
1816 msg_flags_common_set = 0
1817
1818 # Flags set when a complete record has been received (e.g. MSG_EOR
1819 # for SCTP)
1820 msg_flags_eor_indicator = 0
1821
1822 # Flags set when a complete record has not been received
1823 # (e.g. MSG_TRUNC for datagram sockets)
1824 msg_flags_non_eor_indicator = 0
1825
1826 def checkFlags(self, flags, eor=None, checkset=0, checkunset=0, ignore=0):
1827 # Method to check the value of msg_flags returned by recvmsg[_into]().
1828 #
1829 # Checks that all bits in msg_flags_common_set attribute are
1830 # set in "flags" and all bits in msg_flags_common_unset are
1831 # unset.
1832 #
1833 # The "eor" argument specifies whether the flags should
1834 # indicate that a full record (or datagram) has been received.
1835 # If "eor" is None, no checks are done; otherwise, checks
1836 # that:
1837 #
1838 # * if "eor" is true, all bits in msg_flags_eor_indicator are
1839 # set and all bits in msg_flags_non_eor_indicator are unset
1840 #
1841 # * if "eor" is false, all bits in msg_flags_non_eor_indicator
1842 # are set and all bits in msg_flags_eor_indicator are unset
1843 #
1844 # If "checkset" and/or "checkunset" are supplied, they require
1845 # the given bits to be set or unset respectively, overriding
1846 # what the attributes require for those bits.
1847 #
1848 # If any bits are set in "ignore", they will not be checked,
1849 # regardless of the other inputs.
1850 #
1851 # Will raise Exception if the inputs require a bit to be both
1852 # set and unset, and it is not ignored.
1853
1854 defaultset = self.msg_flags_common_set
1855 defaultunset = self.msg_flags_common_unset
1856
1857 if eor:
1858 defaultset |= self.msg_flags_eor_indicator
1859 defaultunset |= self.msg_flags_non_eor_indicator
1860 elif eor is not None:
1861 defaultset |= self.msg_flags_non_eor_indicator
1862 defaultunset |= self.msg_flags_eor_indicator
1863
1864 # Function arguments override defaults
1865 defaultset &= ~checkunset
1866 defaultunset &= ~checkset
1867
1868 # Merge arguments with remaining defaults, and check for conflicts
1869 checkset |= defaultset
1870 checkunset |= defaultunset
1871 inboth = checkset & checkunset & ~ignore
1872 if inboth:
1873 raise Exception("contradictory set, unset requirements for flags "
1874 "{0:#x}".format(inboth))
1875
1876 # Compare with given msg_flags value
1877 mask = (checkset | checkunset) & ~ignore
1878 self.assertEqual(flags & mask, checkset & mask)
1879
1880
1881class RecvmsgIntoMixin(SendrecvmsgBase):
1882 # Mixin to implement doRecvmsg() using recvmsg_into().
1883
1884 def doRecvmsg(self, sock, bufsize, *args):
1885 buf = bytearray(bufsize)
1886 result = sock.recvmsg_into([buf], *args)
1887 self.registerRecvmsgResult(result)
1888 self.assertGreaterEqual(result[0], 0)
1889 self.assertLessEqual(result[0], bufsize)
1890 return (bytes(buf[:result[0]]),) + result[1:]
1891
1892
1893class SendrecvmsgDgramFlagsBase(SendrecvmsgBase):
1894 # Defines flags to be checked in msg_flags for datagram sockets.
1895
1896 @property
1897 def msg_flags_non_eor_indicator(self):
1898 return super().msg_flags_non_eor_indicator | socket.MSG_TRUNC
1899
1900
1901class SendrecvmsgSCTPFlagsBase(SendrecvmsgBase):
1902 # Defines flags to be checked in msg_flags for SCTP sockets.
1903
1904 @property
1905 def msg_flags_eor_indicator(self):
1906 return super().msg_flags_eor_indicator | socket.MSG_EOR
1907
1908
1909class SendrecvmsgConnectionlessBase(SendrecvmsgBase):
1910 # Base class for tests on connectionless-mode sockets. Users must
1911 # supply sockets on attributes cli and serv to be mapped to
1912 # cli_sock and serv_sock respectively.
1913
1914 @property
1915 def serv_sock(self):
1916 return self.serv
1917
1918 @property
1919 def cli_sock(self):
1920 return self.cli
1921
1922 @property
1923 def sendmsg_to_server_defaults(self):
1924 return ([], [], 0, self.serv_addr)
1925
1926 def sendToServer(self, msg):
1927 return self.cli_sock.sendto(msg, self.serv_addr)
1928
1929
1930class SendrecvmsgConnectedBase(SendrecvmsgBase):
1931 # Base class for tests on connected sockets. Users must supply
1932 # sockets on attributes serv_conn and cli_conn (representing the
1933 # connections *to* the server and the client), to be mapped to
1934 # cli_sock and serv_sock respectively.
1935
1936 @property
1937 def serv_sock(self):
1938 return self.cli_conn
1939
1940 @property
1941 def cli_sock(self):
1942 return self.serv_conn
1943
1944 def checkRecvmsgAddress(self, addr1, addr2):
1945 # Address is currently "unspecified" for a connected socket,
1946 # so we don't examine it
1947 pass
1948
1949
1950class SendrecvmsgServerTimeoutBase(SendrecvmsgBase):
1951 # Base class to set a timeout on server's socket.
1952
1953 def setUp(self):
1954 super().setUp()
1955 self.serv_sock.settimeout(self.fail_timeout)
1956
1957
1958class SendmsgTests(SendrecvmsgServerTimeoutBase):
1959 # Tests for sendmsg() which can use any socket type and do not
1960 # involve recvmsg() or recvmsg_into().
1961
1962 def testSendmsg(self):
1963 # Send a simple message with sendmsg().
1964 self.assertEqual(self.serv_sock.recv(len(MSG)), MSG)
1965
1966 def _testSendmsg(self):
1967 self.assertEqual(self.sendmsgToServer([MSG]), len(MSG))
1968
1969 def testSendmsgDataGenerator(self):
1970 # Send from buffer obtained from a generator (not a sequence).
1971 self.assertEqual(self.serv_sock.recv(len(MSG)), MSG)
1972
1973 def _testSendmsgDataGenerator(self):
1974 self.assertEqual(self.sendmsgToServer((o for o in [MSG])),
1975 len(MSG))
1976
1977 def testSendmsgAncillaryGenerator(self):
1978 # Gather (empty) ancillary data from a generator.
1979 self.assertEqual(self.serv_sock.recv(len(MSG)), MSG)
1980
1981 def _testSendmsgAncillaryGenerator(self):
1982 self.assertEqual(self.sendmsgToServer([MSG], (o for o in [])),
1983 len(MSG))
1984
1985 def testSendmsgArray(self):
1986 # Send data from an array instead of the usual bytes object.
1987 self.assertEqual(self.serv_sock.recv(len(MSG)), MSG)
1988
1989 def _testSendmsgArray(self):
1990 self.assertEqual(self.sendmsgToServer([array.array("B", MSG)]),
1991 len(MSG))
1992
1993 def testSendmsgGather(self):
1994 # Send message data from more than one buffer (gather write).
1995 self.assertEqual(self.serv_sock.recv(len(MSG)), MSG)
1996
1997 def _testSendmsgGather(self):
1998 self.assertEqual(self.sendmsgToServer([MSG[:3], MSG[3:]]), len(MSG))
1999
2000 def testSendmsgBadArgs(self):
2001 # Check that sendmsg() rejects invalid arguments.
2002 self.assertEqual(self.serv_sock.recv(1000), b"done")
2003
2004 def _testSendmsgBadArgs(self):
2005 self.assertRaises(TypeError, self.cli_sock.sendmsg)
2006 self.assertRaises(TypeError, self.sendmsgToServer,
2007 b"not in an iterable")
2008 self.assertRaises(TypeError, self.sendmsgToServer,
2009 object())
2010 self.assertRaises(TypeError, self.sendmsgToServer,
2011 [object()])
2012 self.assertRaises(TypeError, self.sendmsgToServer,
2013 [MSG, object()])
2014 self.assertRaises(TypeError, self.sendmsgToServer,
2015 [MSG], object())
2016 self.assertRaises(TypeError, self.sendmsgToServer,
2017 [MSG], [], object())
2018 self.assertRaises(TypeError, self.sendmsgToServer,
2019 [MSG], [], 0, object())
2020 self.sendToServer(b"done")
2021
2022 def testSendmsgBadCmsg(self):
2023 # Check that invalid ancillary data items are rejected.
2024 self.assertEqual(self.serv_sock.recv(1000), b"done")
2025
2026 def _testSendmsgBadCmsg(self):
2027 self.assertRaises(TypeError, self.sendmsgToServer,
2028 [MSG], [object()])
2029 self.assertRaises(TypeError, self.sendmsgToServer,
2030 [MSG], [(object(), 0, b"data")])
2031 self.assertRaises(TypeError, self.sendmsgToServer,
2032 [MSG], [(0, object(), b"data")])
2033 self.assertRaises(TypeError, self.sendmsgToServer,
2034 [MSG], [(0, 0, object())])
2035 self.assertRaises(TypeError, self.sendmsgToServer,
2036 [MSG], [(0, 0)])
2037 self.assertRaises(TypeError, self.sendmsgToServer,
2038 [MSG], [(0, 0, b"data", 42)])
2039 self.sendToServer(b"done")
2040
2041 @requireAttrs(socket, "CMSG_SPACE")
2042 def testSendmsgBadMultiCmsg(self):
2043 # Check that invalid ancillary data items are rejected when
2044 # more than one item is present.
2045 self.assertEqual(self.serv_sock.recv(1000), b"done")
2046
2047 @testSendmsgBadMultiCmsg.client_skip
2048 def _testSendmsgBadMultiCmsg(self):
2049 self.assertRaises(TypeError, self.sendmsgToServer,
2050 [MSG], [0, 0, b""])
2051 self.assertRaises(TypeError, self.sendmsgToServer,
2052 [MSG], [(0, 0, b""), object()])
2053 self.sendToServer(b"done")
2054
2055 def testSendmsgExcessCmsgReject(self):
2056 # Check that sendmsg() rejects excess ancillary data items
2057 # when the number that can be sent is limited.
2058 self.assertEqual(self.serv_sock.recv(1000), b"done")
2059
2060 def _testSendmsgExcessCmsgReject(self):
2061 if not hasattr(socket, "CMSG_SPACE"):
2062 # Can only send one item
Andrew Svetlov0832af62012-12-18 23:10:48 +02002063 with self.assertRaises(OSError) as cm:
Nick Coghlan96fe56a2011-08-22 11:55:57 +10002064 self.sendmsgToServer([MSG], [(0, 0, b""), (0, 0, b"")])
2065 self.assertIsNone(cm.exception.errno)
2066 self.sendToServer(b"done")
2067
2068 def testSendmsgAfterClose(self):
2069 # Check that sendmsg() fails on a closed socket.
2070 pass
2071
2072 def _testSendmsgAfterClose(self):
2073 self.cli_sock.close()
Andrew Svetlov0832af62012-12-18 23:10:48 +02002074 self.assertRaises(OSError, self.sendmsgToServer, [MSG])
Nick Coghlan96fe56a2011-08-22 11:55:57 +10002075
2076
2077class SendmsgStreamTests(SendmsgTests):
2078 # Tests for sendmsg() which require a stream socket and do not
2079 # involve recvmsg() or recvmsg_into().
2080
2081 def testSendmsgExplicitNoneAddr(self):
2082 # Check that peer address can be specified as None.
2083 self.assertEqual(self.serv_sock.recv(len(MSG)), MSG)
2084
2085 def _testSendmsgExplicitNoneAddr(self):
2086 self.assertEqual(self.sendmsgToServer([MSG], [], 0, None), len(MSG))
2087
2088 def testSendmsgTimeout(self):
2089 # Check that timeout works with sendmsg().
2090 self.assertEqual(self.serv_sock.recv(512), b"a"*512)
2091 self.assertTrue(self.misc_event.wait(timeout=self.fail_timeout))
2092
2093 def _testSendmsgTimeout(self):
2094 try:
2095 self.cli_sock.settimeout(0.03)
2096 with self.assertRaises(socket.timeout):
2097 while True:
2098 self.sendmsgToServer([b"a"*512])
2099 finally:
2100 self.misc_event.set()
2101
2102 # XXX: would be nice to have more tests for sendmsg flags argument.
2103
2104 # Linux supports MSG_DONTWAIT when sending, but in general, it
2105 # only works when receiving. Could add other platforms if they
2106 # support it too.
2107 @skipWithClientIf(sys.platform not in {"linux2"},
2108 "MSG_DONTWAIT not known to work on this platform when "
2109 "sending")
2110 def testSendmsgDontWait(self):
2111 # Check that MSG_DONTWAIT in flags causes non-blocking behaviour.
2112 self.assertEqual(self.serv_sock.recv(512), b"a"*512)
2113 self.assertTrue(self.misc_event.wait(timeout=self.fail_timeout))
2114
2115 @testSendmsgDontWait.client_skip
2116 def _testSendmsgDontWait(self):
2117 try:
Andrew Svetlov0832af62012-12-18 23:10:48 +02002118 with self.assertRaises(OSError) as cm:
Nick Coghlan96fe56a2011-08-22 11:55:57 +10002119 while True:
2120 self.sendmsgToServer([b"a"*512], [], socket.MSG_DONTWAIT)
2121 self.assertIn(cm.exception.errno,
2122 (errno.EAGAIN, errno.EWOULDBLOCK))
2123 finally:
2124 self.misc_event.set()
2125
2126
2127class SendmsgConnectionlessTests(SendmsgTests):
2128 # Tests for sendmsg() which require a connectionless-mode
2129 # (e.g. datagram) socket, and do not involve recvmsg() or
2130 # recvmsg_into().
2131
2132 def testSendmsgNoDestAddr(self):
2133 # Check that sendmsg() fails when no destination address is
2134 # given for unconnected socket.
2135 pass
2136
2137 def _testSendmsgNoDestAddr(self):
Andrew Svetlov0832af62012-12-18 23:10:48 +02002138 self.assertRaises(OSError, self.cli_sock.sendmsg,
Nick Coghlan96fe56a2011-08-22 11:55:57 +10002139 [MSG])
Andrew Svetlov0832af62012-12-18 23:10:48 +02002140 self.assertRaises(OSError, self.cli_sock.sendmsg,
Nick Coghlan96fe56a2011-08-22 11:55:57 +10002141 [MSG], [], 0, None)
2142
2143
2144class RecvmsgGenericTests(SendrecvmsgBase):
2145 # Tests for recvmsg() which can also be emulated using
2146 # recvmsg_into(), and can use any socket type.
2147
2148 def testRecvmsg(self):
2149 # Receive a simple message with recvmsg[_into]().
2150 msg, ancdata, flags, addr = self.doRecvmsg(self.serv_sock, len(MSG))
2151 self.assertEqual(msg, MSG)
2152 self.checkRecvmsgAddress(addr, self.cli_addr)
2153 self.assertEqual(ancdata, [])
2154 self.checkFlags(flags, eor=True)
2155
2156 def _testRecvmsg(self):
2157 self.sendToServer(MSG)
2158
2159 def testRecvmsgExplicitDefaults(self):
2160 # Test recvmsg[_into]() with default arguments provided explicitly.
2161 msg, ancdata, flags, addr = self.doRecvmsg(self.serv_sock,
2162 len(MSG), 0, 0)
2163 self.assertEqual(msg, MSG)
2164 self.checkRecvmsgAddress(addr, self.cli_addr)
2165 self.assertEqual(ancdata, [])
2166 self.checkFlags(flags, eor=True)
2167
2168 def _testRecvmsgExplicitDefaults(self):
2169 self.sendToServer(MSG)
2170
2171 def testRecvmsgShorter(self):
2172 # Receive a message smaller than buffer.
2173 msg, ancdata, flags, addr = self.doRecvmsg(self.serv_sock,
2174 len(MSG) + 42)
2175 self.assertEqual(msg, MSG)
2176 self.checkRecvmsgAddress(addr, self.cli_addr)
2177 self.assertEqual(ancdata, [])
2178 self.checkFlags(flags, eor=True)
2179
2180 def _testRecvmsgShorter(self):
2181 self.sendToServer(MSG)
2182
Charles-François Natali8619cd72011-10-03 19:43:15 +02002183 # FreeBSD < 8 doesn't always set the MSG_TRUNC flag when a truncated
2184 # datagram is received (issue #13001).
2185 @support.requires_freebsd_version(8)
Nick Coghlan96fe56a2011-08-22 11:55:57 +10002186 def testRecvmsgTrunc(self):
2187 # Receive part of message, check for truncation indicators.
2188 msg, ancdata, flags, addr = self.doRecvmsg(self.serv_sock,
2189 len(MSG) - 3)
2190 self.assertEqual(msg, MSG[:-3])
2191 self.checkRecvmsgAddress(addr, self.cli_addr)
2192 self.assertEqual(ancdata, [])
2193 self.checkFlags(flags, eor=False)
2194
Charles-François Natali8619cd72011-10-03 19:43:15 +02002195 @support.requires_freebsd_version(8)
Nick Coghlan96fe56a2011-08-22 11:55:57 +10002196 def _testRecvmsgTrunc(self):
2197 self.sendToServer(MSG)
2198
2199 def testRecvmsgShortAncillaryBuf(self):
2200 # Test ancillary data buffer too small to hold any ancillary data.
2201 msg, ancdata, flags, addr = self.doRecvmsg(self.serv_sock,
2202 len(MSG), 1)
2203 self.assertEqual(msg, MSG)
2204 self.checkRecvmsgAddress(addr, self.cli_addr)
2205 self.assertEqual(ancdata, [])
2206 self.checkFlags(flags, eor=True)
2207
2208 def _testRecvmsgShortAncillaryBuf(self):
2209 self.sendToServer(MSG)
2210
2211 def testRecvmsgLongAncillaryBuf(self):
2212 # Test large ancillary data buffer.
2213 msg, ancdata, flags, addr = self.doRecvmsg(self.serv_sock,
2214 len(MSG), 10240)
2215 self.assertEqual(msg, MSG)
2216 self.checkRecvmsgAddress(addr, self.cli_addr)
2217 self.assertEqual(ancdata, [])
2218 self.checkFlags(flags, eor=True)
2219
2220 def _testRecvmsgLongAncillaryBuf(self):
2221 self.sendToServer(MSG)
2222
2223 def testRecvmsgAfterClose(self):
2224 # Check that recvmsg[_into]() fails on a closed socket.
2225 self.serv_sock.close()
Andrew Svetlov0832af62012-12-18 23:10:48 +02002226 self.assertRaises(OSError, self.doRecvmsg, self.serv_sock, 1024)
Nick Coghlan96fe56a2011-08-22 11:55:57 +10002227
2228 def _testRecvmsgAfterClose(self):
2229 pass
2230
2231 def testRecvmsgTimeout(self):
2232 # Check that timeout works.
2233 try:
2234 self.serv_sock.settimeout(0.03)
2235 self.assertRaises(socket.timeout,
2236 self.doRecvmsg, self.serv_sock, len(MSG))
2237 finally:
2238 self.misc_event.set()
2239
2240 def _testRecvmsgTimeout(self):
2241 self.assertTrue(self.misc_event.wait(timeout=self.fail_timeout))
2242
2243 @requireAttrs(socket, "MSG_PEEK")
2244 def testRecvmsgPeek(self):
2245 # Check that MSG_PEEK in flags enables examination of pending
2246 # data without consuming it.
2247
2248 # Receive part of data with MSG_PEEK.
2249 msg, ancdata, flags, addr = self.doRecvmsg(self.serv_sock,
2250 len(MSG) - 3, 0,
2251 socket.MSG_PEEK)
2252 self.assertEqual(msg, MSG[:-3])
2253 self.checkRecvmsgAddress(addr, self.cli_addr)
2254 self.assertEqual(ancdata, [])
2255 # Ignoring MSG_TRUNC here (so this test is the same for stream
2256 # and datagram sockets). Some wording in POSIX seems to
2257 # suggest that it needn't be set when peeking, but that may
2258 # just be a slip.
2259 self.checkFlags(flags, eor=False,
2260 ignore=getattr(socket, "MSG_TRUNC", 0))
2261
2262 # Receive all data with MSG_PEEK.
2263 msg, ancdata, flags, addr = self.doRecvmsg(self.serv_sock,
2264 len(MSG), 0,
2265 socket.MSG_PEEK)
2266 self.assertEqual(msg, MSG)
2267 self.checkRecvmsgAddress(addr, self.cli_addr)
2268 self.assertEqual(ancdata, [])
2269 self.checkFlags(flags, eor=True)
2270
2271 # Check that the same data can still be received normally.
2272 msg, ancdata, flags, addr = self.doRecvmsg(self.serv_sock, len(MSG))
2273 self.assertEqual(msg, MSG)
2274 self.checkRecvmsgAddress(addr, self.cli_addr)
2275 self.assertEqual(ancdata, [])
2276 self.checkFlags(flags, eor=True)
2277
2278 @testRecvmsgPeek.client_skip
2279 def _testRecvmsgPeek(self):
2280 self.sendToServer(MSG)
2281
2282 @requireAttrs(socket.socket, "sendmsg")
2283 def testRecvmsgFromSendmsg(self):
2284 # Test receiving with recvmsg[_into]() when message is sent
2285 # using sendmsg().
2286 self.serv_sock.settimeout(self.fail_timeout)
2287 msg, ancdata, flags, addr = self.doRecvmsg(self.serv_sock, len(MSG))
2288 self.assertEqual(msg, MSG)
2289 self.checkRecvmsgAddress(addr, self.cli_addr)
2290 self.assertEqual(ancdata, [])
2291 self.checkFlags(flags, eor=True)
2292
2293 @testRecvmsgFromSendmsg.client_skip
2294 def _testRecvmsgFromSendmsg(self):
2295 self.assertEqual(self.sendmsgToServer([MSG[:3], MSG[3:]]), len(MSG))
2296
2297
2298class RecvmsgGenericStreamTests(RecvmsgGenericTests):
2299 # Tests which require a stream socket and can use either recvmsg()
2300 # or recvmsg_into().
2301
2302 def testRecvmsgEOF(self):
2303 # Receive end-of-stream indicator (b"", peer socket closed).
2304 msg, ancdata, flags, addr = self.doRecvmsg(self.serv_sock, 1024)
2305 self.assertEqual(msg, b"")
2306 self.checkRecvmsgAddress(addr, self.cli_addr)
2307 self.assertEqual(ancdata, [])
2308 self.checkFlags(flags, eor=None) # Might not have end-of-record marker
2309
2310 def _testRecvmsgEOF(self):
2311 self.cli_sock.close()
2312
2313 def testRecvmsgOverflow(self):
2314 # Receive a message in more than one chunk.
2315 seg1, ancdata, flags, addr = self.doRecvmsg(self.serv_sock,
2316 len(MSG) - 3)
2317 self.checkRecvmsgAddress(addr, self.cli_addr)
2318 self.assertEqual(ancdata, [])
2319 self.checkFlags(flags, eor=False)
2320
2321 seg2, ancdata, flags, addr = self.doRecvmsg(self.serv_sock, 1024)
2322 self.checkRecvmsgAddress(addr, self.cli_addr)
2323 self.assertEqual(ancdata, [])
2324 self.checkFlags(flags, eor=True)
2325
2326 msg = seg1 + seg2
2327 self.assertEqual(msg, MSG)
2328
2329 def _testRecvmsgOverflow(self):
2330 self.sendToServer(MSG)
2331
2332
2333class RecvmsgTests(RecvmsgGenericTests):
2334 # Tests for recvmsg() which can use any socket type.
2335
2336 def testRecvmsgBadArgs(self):
2337 # Check that recvmsg() rejects invalid arguments.
2338 self.assertRaises(TypeError, self.serv_sock.recvmsg)
2339 self.assertRaises(ValueError, self.serv_sock.recvmsg,
2340 -1, 0, 0)
2341 self.assertRaises(ValueError, self.serv_sock.recvmsg,
2342 len(MSG), -1, 0)
2343 self.assertRaises(TypeError, self.serv_sock.recvmsg,
2344 [bytearray(10)], 0, 0)
2345 self.assertRaises(TypeError, self.serv_sock.recvmsg,
2346 object(), 0, 0)
2347 self.assertRaises(TypeError, self.serv_sock.recvmsg,
2348 len(MSG), object(), 0)
2349 self.assertRaises(TypeError, self.serv_sock.recvmsg,
2350 len(MSG), 0, object())
2351
2352 msg, ancdata, flags, addr = self.serv_sock.recvmsg(len(MSG), 0, 0)
2353 self.assertEqual(msg, MSG)
2354 self.checkRecvmsgAddress(addr, self.cli_addr)
2355 self.assertEqual(ancdata, [])
2356 self.checkFlags(flags, eor=True)
2357
2358 def _testRecvmsgBadArgs(self):
2359 self.sendToServer(MSG)
2360
2361
2362class RecvmsgIntoTests(RecvmsgIntoMixin, RecvmsgGenericTests):
2363 # Tests for recvmsg_into() which can use any socket type.
2364
2365 def testRecvmsgIntoBadArgs(self):
2366 # Check that recvmsg_into() rejects invalid arguments.
2367 buf = bytearray(len(MSG))
2368 self.assertRaises(TypeError, self.serv_sock.recvmsg_into)
2369 self.assertRaises(TypeError, self.serv_sock.recvmsg_into,
2370 len(MSG), 0, 0)
2371 self.assertRaises(TypeError, self.serv_sock.recvmsg_into,
2372 buf, 0, 0)
2373 self.assertRaises(TypeError, self.serv_sock.recvmsg_into,
2374 [object()], 0, 0)
2375 self.assertRaises(TypeError, self.serv_sock.recvmsg_into,
2376 [b"I'm not writable"], 0, 0)
2377 self.assertRaises(TypeError, self.serv_sock.recvmsg_into,
2378 [buf, object()], 0, 0)
2379 self.assertRaises(ValueError, self.serv_sock.recvmsg_into,
2380 [buf], -1, 0)
2381 self.assertRaises(TypeError, self.serv_sock.recvmsg_into,
2382 [buf], object(), 0)
2383 self.assertRaises(TypeError, self.serv_sock.recvmsg_into,
2384 [buf], 0, object())
2385
2386 nbytes, ancdata, flags, addr = self.serv_sock.recvmsg_into([buf], 0, 0)
2387 self.assertEqual(nbytes, len(MSG))
2388 self.assertEqual(buf, bytearray(MSG))
2389 self.checkRecvmsgAddress(addr, self.cli_addr)
2390 self.assertEqual(ancdata, [])
2391 self.checkFlags(flags, eor=True)
2392
2393 def _testRecvmsgIntoBadArgs(self):
2394 self.sendToServer(MSG)
2395
2396 def testRecvmsgIntoGenerator(self):
2397 # Receive into buffer obtained from a generator (not a sequence).
2398 buf = bytearray(len(MSG))
2399 nbytes, ancdata, flags, addr = self.serv_sock.recvmsg_into(
2400 (o for o in [buf]))
2401 self.assertEqual(nbytes, len(MSG))
2402 self.assertEqual(buf, bytearray(MSG))
2403 self.checkRecvmsgAddress(addr, self.cli_addr)
2404 self.assertEqual(ancdata, [])
2405 self.checkFlags(flags, eor=True)
2406
2407 def _testRecvmsgIntoGenerator(self):
2408 self.sendToServer(MSG)
2409
2410 def testRecvmsgIntoArray(self):
2411 # Receive into an array rather than the usual bytearray.
2412 buf = array.array("B", [0] * len(MSG))
2413 nbytes, ancdata, flags, addr = self.serv_sock.recvmsg_into([buf])
2414 self.assertEqual(nbytes, len(MSG))
Nadeem Vawda3d1706f2011-08-22 09:46:56 +02002415 self.assertEqual(buf.tobytes(), MSG)
Nick Coghlan96fe56a2011-08-22 11:55:57 +10002416 self.checkRecvmsgAddress(addr, self.cli_addr)
2417 self.assertEqual(ancdata, [])
2418 self.checkFlags(flags, eor=True)
2419
2420 def _testRecvmsgIntoArray(self):
2421 self.sendToServer(MSG)
2422
2423 def testRecvmsgIntoScatter(self):
2424 # Receive into multiple buffers (scatter write).
2425 b1 = bytearray(b"----")
2426 b2 = bytearray(b"0123456789")
2427 b3 = bytearray(b"--------------")
2428 nbytes, ancdata, flags, addr = self.serv_sock.recvmsg_into(
2429 [b1, memoryview(b2)[2:9], b3])
2430 self.assertEqual(nbytes, len(b"Mary had a little lamb"))
2431 self.assertEqual(b1, bytearray(b"Mary"))
2432 self.assertEqual(b2, bytearray(b"01 had a 9"))
2433 self.assertEqual(b3, bytearray(b"little lamb---"))
2434 self.checkRecvmsgAddress(addr, self.cli_addr)
2435 self.assertEqual(ancdata, [])
2436 self.checkFlags(flags, eor=True)
2437
2438 def _testRecvmsgIntoScatter(self):
2439 self.sendToServer(b"Mary had a little lamb")
2440
2441
2442class CmsgMacroTests(unittest.TestCase):
2443 # Test the functions CMSG_LEN() and CMSG_SPACE(). Tests
2444 # assumptions used by sendmsg() and recvmsg[_into](), which share
2445 # code with these functions.
2446
2447 # Match the definition in socketmodule.c
2448 socklen_t_limit = min(0x7fffffff, _testcapi.INT_MAX)
2449
2450 @requireAttrs(socket, "CMSG_LEN")
2451 def testCMSG_LEN(self):
2452 # Test CMSG_LEN() with various valid and invalid values,
2453 # checking the assumptions used by recvmsg() and sendmsg().
2454 toobig = self.socklen_t_limit - socket.CMSG_LEN(0) + 1
2455 values = list(range(257)) + list(range(toobig - 257, toobig))
2456
2457 # struct cmsghdr has at least three members, two of which are ints
2458 self.assertGreater(socket.CMSG_LEN(0), array.array("i").itemsize * 2)
2459 for n in values:
2460 ret = socket.CMSG_LEN(n)
2461 # This is how recvmsg() calculates the data size
2462 self.assertEqual(ret - socket.CMSG_LEN(0), n)
2463 self.assertLessEqual(ret, self.socklen_t_limit)
2464
2465 self.assertRaises(OverflowError, socket.CMSG_LEN, -1)
2466 # sendmsg() shares code with these functions, and requires
2467 # that it reject values over the limit.
2468 self.assertRaises(OverflowError, socket.CMSG_LEN, toobig)
2469 self.assertRaises(OverflowError, socket.CMSG_LEN, sys.maxsize)
2470
2471 @requireAttrs(socket, "CMSG_SPACE")
2472 def testCMSG_SPACE(self):
2473 # Test CMSG_SPACE() with various valid and invalid values,
2474 # checking the assumptions used by sendmsg().
2475 toobig = self.socklen_t_limit - socket.CMSG_SPACE(1) + 1
2476 values = list(range(257)) + list(range(toobig - 257, toobig))
2477
2478 last = socket.CMSG_SPACE(0)
2479 # struct cmsghdr has at least three members, two of which are ints
2480 self.assertGreater(last, array.array("i").itemsize * 2)
2481 for n in values:
2482 ret = socket.CMSG_SPACE(n)
2483 self.assertGreaterEqual(ret, last)
2484 self.assertGreaterEqual(ret, socket.CMSG_LEN(n))
2485 self.assertGreaterEqual(ret, n + socket.CMSG_LEN(0))
2486 self.assertLessEqual(ret, self.socklen_t_limit)
2487 last = ret
2488
2489 self.assertRaises(OverflowError, socket.CMSG_SPACE, -1)
2490 # sendmsg() shares code with these functions, and requires
2491 # that it reject values over the limit.
2492 self.assertRaises(OverflowError, socket.CMSG_SPACE, toobig)
2493 self.assertRaises(OverflowError, socket.CMSG_SPACE, sys.maxsize)
2494
2495
2496class SCMRightsTest(SendrecvmsgServerTimeoutBase):
2497 # Tests for file descriptor passing on Unix-domain sockets.
2498
2499 # Invalid file descriptor value that's unlikely to evaluate to a
2500 # real FD even if one of its bytes is replaced with a different
2501 # value (which shouldn't actually happen).
2502 badfd = -0x5555
2503
2504 def newFDs(self, n):
2505 # Return a list of n file descriptors for newly-created files
2506 # containing their list indices as ASCII numbers.
2507 fds = []
2508 for i in range(n):
2509 fd, path = tempfile.mkstemp()
2510 self.addCleanup(os.unlink, path)
2511 self.addCleanup(os.close, fd)
2512 os.write(fd, str(i).encode())
2513 fds.append(fd)
2514 return fds
2515
2516 def checkFDs(self, fds):
2517 # Check that the file descriptors in the given list contain
2518 # their correct list indices as ASCII numbers.
2519 for n, fd in enumerate(fds):
2520 os.lseek(fd, 0, os.SEEK_SET)
2521 self.assertEqual(os.read(fd, 1024), str(n).encode())
2522
2523 def registerRecvmsgResult(self, result):
2524 self.addCleanup(self.closeRecvmsgFDs, result)
2525
2526 def closeRecvmsgFDs(self, recvmsg_result):
2527 # Close all file descriptors specified in the ancillary data
2528 # of the given return value from recvmsg() or recvmsg_into().
2529 for cmsg_level, cmsg_type, cmsg_data in recvmsg_result[1]:
2530 if (cmsg_level == socket.SOL_SOCKET and
2531 cmsg_type == socket.SCM_RIGHTS):
2532 fds = array.array("i")
Nadeem Vawda3d1706f2011-08-22 09:46:56 +02002533 fds.frombytes(cmsg_data[:
Nick Coghlan96fe56a2011-08-22 11:55:57 +10002534 len(cmsg_data) - (len(cmsg_data) % fds.itemsize)])
2535 for fd in fds:
2536 os.close(fd)
2537
2538 def createAndSendFDs(self, n):
2539 # Send n new file descriptors created by newFDs() to the
2540 # server, with the constant MSG as the non-ancillary data.
2541 self.assertEqual(
2542 self.sendmsgToServer([MSG],
2543 [(socket.SOL_SOCKET,
2544 socket.SCM_RIGHTS,
2545 array.array("i", self.newFDs(n)))]),
2546 len(MSG))
2547
2548 def checkRecvmsgFDs(self, numfds, result, maxcmsgs=1, ignoreflags=0):
2549 # Check that constant MSG was received with numfds file
2550 # descriptors in a maximum of maxcmsgs control messages (which
2551 # must contain only complete integers). By default, check
2552 # that MSG_CTRUNC is unset, but ignore any flags in
2553 # ignoreflags.
2554 msg, ancdata, flags, addr = result
2555 self.assertEqual(msg, MSG)
2556 self.checkRecvmsgAddress(addr, self.cli_addr)
2557 self.checkFlags(flags, eor=True, checkunset=socket.MSG_CTRUNC,
2558 ignore=ignoreflags)
2559
2560 self.assertIsInstance(ancdata, list)
2561 self.assertLessEqual(len(ancdata), maxcmsgs)
2562 fds = array.array("i")
2563 for item in ancdata:
2564 self.assertIsInstance(item, tuple)
2565 cmsg_level, cmsg_type, cmsg_data = item
2566 self.assertEqual(cmsg_level, socket.SOL_SOCKET)
2567 self.assertEqual(cmsg_type, socket.SCM_RIGHTS)
2568 self.assertIsInstance(cmsg_data, bytes)
2569 self.assertEqual(len(cmsg_data) % SIZEOF_INT, 0)
Nadeem Vawda3d1706f2011-08-22 09:46:56 +02002570 fds.frombytes(cmsg_data)
Nick Coghlan96fe56a2011-08-22 11:55:57 +10002571
2572 self.assertEqual(len(fds), numfds)
2573 self.checkFDs(fds)
2574
2575 def testFDPassSimple(self):
2576 # Pass a single FD (array read from bytes object).
2577 self.checkRecvmsgFDs(1, self.doRecvmsg(self.serv_sock,
2578 len(MSG), 10240))
2579
2580 def _testFDPassSimple(self):
2581 self.assertEqual(
2582 self.sendmsgToServer(
2583 [MSG],
2584 [(socket.SOL_SOCKET,
2585 socket.SCM_RIGHTS,
Nadeem Vawda3d1706f2011-08-22 09:46:56 +02002586 array.array("i", self.newFDs(1)).tobytes())]),
Nick Coghlan96fe56a2011-08-22 11:55:57 +10002587 len(MSG))
2588
2589 def testMultipleFDPass(self):
2590 # Pass multiple FDs in a single array.
2591 self.checkRecvmsgFDs(4, self.doRecvmsg(self.serv_sock,
2592 len(MSG), 10240))
2593
2594 def _testMultipleFDPass(self):
2595 self.createAndSendFDs(4)
2596
2597 @requireAttrs(socket, "CMSG_SPACE")
2598 def testFDPassCMSG_SPACE(self):
2599 # Test using CMSG_SPACE() to calculate ancillary buffer size.
2600 self.checkRecvmsgFDs(
2601 4, self.doRecvmsg(self.serv_sock, len(MSG),
2602 socket.CMSG_SPACE(4 * SIZEOF_INT)))
2603
2604 @testFDPassCMSG_SPACE.client_skip
2605 def _testFDPassCMSG_SPACE(self):
2606 self.createAndSendFDs(4)
2607
2608 def testFDPassCMSG_LEN(self):
2609 # Test using CMSG_LEN() to calculate ancillary buffer size.
2610 self.checkRecvmsgFDs(1,
2611 self.doRecvmsg(self.serv_sock, len(MSG),
2612 socket.CMSG_LEN(4 * SIZEOF_INT)),
2613 # RFC 3542 says implementations may set
2614 # MSG_CTRUNC if there isn't enough space
2615 # for trailing padding.
2616 ignoreflags=socket.MSG_CTRUNC)
2617
2618 def _testFDPassCMSG_LEN(self):
2619 self.createAndSendFDs(1)
2620
Nick Coghlan2496f332011-09-19 20:26:31 +10002621 # Issue #12958: The following test has problems on Mac OS X
2622 @support.anticipate_failure(sys.platform == "darwin")
Nick Coghlan96fe56a2011-08-22 11:55:57 +10002623 @requireAttrs(socket, "CMSG_SPACE")
2624 def testFDPassSeparate(self):
2625 # Pass two FDs in two separate arrays. Arrays may be combined
2626 # into a single control message by the OS.
2627 self.checkRecvmsgFDs(2,
2628 self.doRecvmsg(self.serv_sock, len(MSG), 10240),
2629 maxcmsgs=2)
2630
2631 @testFDPassSeparate.client_skip
Nick Coghlan2496f332011-09-19 20:26:31 +10002632 @support.anticipate_failure(sys.platform == "darwin")
Nick Coghlan96fe56a2011-08-22 11:55:57 +10002633 def _testFDPassSeparate(self):
2634 fd0, fd1 = self.newFDs(2)
2635 self.assertEqual(
2636 self.sendmsgToServer([MSG], [(socket.SOL_SOCKET,
2637 socket.SCM_RIGHTS,
2638 array.array("i", [fd0])),
2639 (socket.SOL_SOCKET,
2640 socket.SCM_RIGHTS,
2641 array.array("i", [fd1]))]),
2642 len(MSG))
2643
Nick Coghlan2496f332011-09-19 20:26:31 +10002644 # Issue #12958: The following test has problems on Mac OS X
2645 @support.anticipate_failure(sys.platform == "darwin")
Nick Coghlan96fe56a2011-08-22 11:55:57 +10002646 @requireAttrs(socket, "CMSG_SPACE")
2647 def testFDPassSeparateMinSpace(self):
2648 # Pass two FDs in two separate arrays, receiving them into the
2649 # minimum space for two arrays.
2650 self.checkRecvmsgFDs(2,
2651 self.doRecvmsg(self.serv_sock, len(MSG),
2652 socket.CMSG_SPACE(SIZEOF_INT) +
2653 socket.CMSG_LEN(SIZEOF_INT)),
2654 maxcmsgs=2, ignoreflags=socket.MSG_CTRUNC)
2655
2656 @testFDPassSeparateMinSpace.client_skip
Nick Coghlan2496f332011-09-19 20:26:31 +10002657 @support.anticipate_failure(sys.platform == "darwin")
Nick Coghlan96fe56a2011-08-22 11:55:57 +10002658 def _testFDPassSeparateMinSpace(self):
2659 fd0, fd1 = self.newFDs(2)
2660 self.assertEqual(
2661 self.sendmsgToServer([MSG], [(socket.SOL_SOCKET,
2662 socket.SCM_RIGHTS,
2663 array.array("i", [fd0])),
2664 (socket.SOL_SOCKET,
2665 socket.SCM_RIGHTS,
2666 array.array("i", [fd1]))]),
2667 len(MSG))
2668
2669 def sendAncillaryIfPossible(self, msg, ancdata):
2670 # Try to send msg and ancdata to server, but if the system
2671 # call fails, just send msg with no ancillary data.
2672 try:
2673 nbytes = self.sendmsgToServer([msg], ancdata)
Andrew Svetlov0832af62012-12-18 23:10:48 +02002674 except OSError as e:
Nick Coghlan96fe56a2011-08-22 11:55:57 +10002675 # Check that it was the system call that failed
2676 self.assertIsInstance(e.errno, int)
2677 nbytes = self.sendmsgToServer([msg])
2678 self.assertEqual(nbytes, len(msg))
2679
2680 def testFDPassEmpty(self):
2681 # Try to pass an empty FD array. Can receive either no array
2682 # or an empty array.
2683 self.checkRecvmsgFDs(0, self.doRecvmsg(self.serv_sock,
2684 len(MSG), 10240),
2685 ignoreflags=socket.MSG_CTRUNC)
2686
2687 def _testFDPassEmpty(self):
2688 self.sendAncillaryIfPossible(MSG, [(socket.SOL_SOCKET,
2689 socket.SCM_RIGHTS,
2690 b"")])
2691
2692 def testFDPassPartialInt(self):
2693 # Try to pass a truncated FD array.
2694 msg, ancdata, flags, addr = self.doRecvmsg(self.serv_sock,
2695 len(MSG), 10240)
2696 self.assertEqual(msg, MSG)
2697 self.checkRecvmsgAddress(addr, self.cli_addr)
2698 self.checkFlags(flags, eor=True, ignore=socket.MSG_CTRUNC)
2699 self.assertLessEqual(len(ancdata), 1)
2700 for cmsg_level, cmsg_type, cmsg_data in ancdata:
2701 self.assertEqual(cmsg_level, socket.SOL_SOCKET)
2702 self.assertEqual(cmsg_type, socket.SCM_RIGHTS)
2703 self.assertLess(len(cmsg_data), SIZEOF_INT)
2704
2705 def _testFDPassPartialInt(self):
2706 self.sendAncillaryIfPossible(
2707 MSG,
2708 [(socket.SOL_SOCKET,
2709 socket.SCM_RIGHTS,
Nadeem Vawda3d1706f2011-08-22 09:46:56 +02002710 array.array("i", [self.badfd]).tobytes()[:-1])])
Nick Coghlan96fe56a2011-08-22 11:55:57 +10002711
2712 @requireAttrs(socket, "CMSG_SPACE")
2713 def testFDPassPartialIntInMiddle(self):
2714 # Try to pass two FD arrays, the first of which is truncated.
2715 msg, ancdata, flags, addr = self.doRecvmsg(self.serv_sock,
2716 len(MSG), 10240)
2717 self.assertEqual(msg, MSG)
2718 self.checkRecvmsgAddress(addr, self.cli_addr)
2719 self.checkFlags(flags, eor=True, ignore=socket.MSG_CTRUNC)
2720 self.assertLessEqual(len(ancdata), 2)
2721 fds = array.array("i")
2722 # Arrays may have been combined in a single control message
2723 for cmsg_level, cmsg_type, cmsg_data in ancdata:
2724 self.assertEqual(cmsg_level, socket.SOL_SOCKET)
2725 self.assertEqual(cmsg_type, socket.SCM_RIGHTS)
Nadeem Vawda3d1706f2011-08-22 09:46:56 +02002726 fds.frombytes(cmsg_data[:
Nick Coghlan96fe56a2011-08-22 11:55:57 +10002727 len(cmsg_data) - (len(cmsg_data) % fds.itemsize)])
2728 self.assertLessEqual(len(fds), 2)
2729 self.checkFDs(fds)
2730
2731 @testFDPassPartialIntInMiddle.client_skip
2732 def _testFDPassPartialIntInMiddle(self):
2733 fd0, fd1 = self.newFDs(2)
2734 self.sendAncillaryIfPossible(
2735 MSG,
2736 [(socket.SOL_SOCKET,
2737 socket.SCM_RIGHTS,
Nadeem Vawda3d1706f2011-08-22 09:46:56 +02002738 array.array("i", [fd0, self.badfd]).tobytes()[:-1]),
Nick Coghlan96fe56a2011-08-22 11:55:57 +10002739 (socket.SOL_SOCKET,
2740 socket.SCM_RIGHTS,
2741 array.array("i", [fd1]))])
2742
2743 def checkTruncatedHeader(self, result, ignoreflags=0):
2744 # Check that no ancillary data items are returned when data is
2745 # truncated inside the cmsghdr structure.
2746 msg, ancdata, flags, addr = result
2747 self.assertEqual(msg, MSG)
2748 self.checkRecvmsgAddress(addr, self.cli_addr)
2749 self.assertEqual(ancdata, [])
2750 self.checkFlags(flags, eor=True, checkset=socket.MSG_CTRUNC,
2751 ignore=ignoreflags)
2752
2753 def testCmsgTruncNoBufSize(self):
2754 # Check that no ancillary data is received when no buffer size
2755 # is specified.
2756 self.checkTruncatedHeader(self.doRecvmsg(self.serv_sock, len(MSG)),
2757 # BSD seems to set MSG_CTRUNC only
2758 # if an item has been partially
2759 # received.
2760 ignoreflags=socket.MSG_CTRUNC)
2761
2762 def _testCmsgTruncNoBufSize(self):
2763 self.createAndSendFDs(1)
2764
2765 def testCmsgTrunc0(self):
2766 # Check that no ancillary data is received when buffer size is 0.
2767 self.checkTruncatedHeader(self.doRecvmsg(self.serv_sock, len(MSG), 0),
2768 ignoreflags=socket.MSG_CTRUNC)
2769
2770 def _testCmsgTrunc0(self):
2771 self.createAndSendFDs(1)
2772
2773 # Check that no ancillary data is returned for various non-zero
2774 # (but still too small) buffer sizes.
2775
2776 def testCmsgTrunc1(self):
2777 self.checkTruncatedHeader(self.doRecvmsg(self.serv_sock, len(MSG), 1))
2778
2779 def _testCmsgTrunc1(self):
2780 self.createAndSendFDs(1)
2781
2782 def testCmsgTrunc2Int(self):
2783 # The cmsghdr structure has at least three members, two of
2784 # which are ints, so we still shouldn't see any ancillary
2785 # data.
2786 self.checkTruncatedHeader(self.doRecvmsg(self.serv_sock, len(MSG),
2787 SIZEOF_INT * 2))
2788
2789 def _testCmsgTrunc2Int(self):
2790 self.createAndSendFDs(1)
2791
2792 def testCmsgTruncLen0Minus1(self):
2793 self.checkTruncatedHeader(self.doRecvmsg(self.serv_sock, len(MSG),
2794 socket.CMSG_LEN(0) - 1))
2795
2796 def _testCmsgTruncLen0Minus1(self):
2797 self.createAndSendFDs(1)
2798
2799 # The following tests try to truncate the control message in the
2800 # middle of the FD array.
2801
2802 def checkTruncatedArray(self, ancbuf, maxdata, mindata=0):
2803 # Check that file descriptor data is truncated to between
2804 # mindata and maxdata bytes when received with buffer size
2805 # ancbuf, and that any complete file descriptor numbers are
2806 # valid.
2807 msg, ancdata, flags, addr = self.doRecvmsg(self.serv_sock,
2808 len(MSG), ancbuf)
2809 self.assertEqual(msg, MSG)
2810 self.checkRecvmsgAddress(addr, self.cli_addr)
2811 self.checkFlags(flags, eor=True, checkset=socket.MSG_CTRUNC)
2812
2813 if mindata == 0 and ancdata == []:
2814 return
2815 self.assertEqual(len(ancdata), 1)
2816 cmsg_level, cmsg_type, cmsg_data = ancdata[0]
2817 self.assertEqual(cmsg_level, socket.SOL_SOCKET)
2818 self.assertEqual(cmsg_type, socket.SCM_RIGHTS)
2819 self.assertGreaterEqual(len(cmsg_data), mindata)
2820 self.assertLessEqual(len(cmsg_data), maxdata)
2821 fds = array.array("i")
Nadeem Vawda3d1706f2011-08-22 09:46:56 +02002822 fds.frombytes(cmsg_data[:
Nick Coghlan96fe56a2011-08-22 11:55:57 +10002823 len(cmsg_data) - (len(cmsg_data) % fds.itemsize)])
2824 self.checkFDs(fds)
2825
2826 def testCmsgTruncLen0(self):
2827 self.checkTruncatedArray(ancbuf=socket.CMSG_LEN(0), maxdata=0)
2828
2829 def _testCmsgTruncLen0(self):
2830 self.createAndSendFDs(1)
2831
2832 def testCmsgTruncLen0Plus1(self):
2833 self.checkTruncatedArray(ancbuf=socket.CMSG_LEN(0) + 1, maxdata=1)
2834
2835 def _testCmsgTruncLen0Plus1(self):
2836 self.createAndSendFDs(2)
2837
2838 def testCmsgTruncLen1(self):
2839 self.checkTruncatedArray(ancbuf=socket.CMSG_LEN(SIZEOF_INT),
2840 maxdata=SIZEOF_INT)
2841
2842 def _testCmsgTruncLen1(self):
2843 self.createAndSendFDs(2)
2844
2845 def testCmsgTruncLen2Minus1(self):
2846 self.checkTruncatedArray(ancbuf=socket.CMSG_LEN(2 * SIZEOF_INT) - 1,
2847 maxdata=(2 * SIZEOF_INT) - 1)
2848
2849 def _testCmsgTruncLen2Minus1(self):
2850 self.createAndSendFDs(2)
2851
2852
2853class RFC3542AncillaryTest(SendrecvmsgServerTimeoutBase):
2854 # Test sendmsg() and recvmsg[_into]() using the ancillary data
2855 # features of the RFC 3542 Advanced Sockets API for IPv6.
2856 # Currently we can only handle certain data items (e.g. traffic
2857 # class, hop limit, MTU discovery and fragmentation settings)
2858 # without resorting to unportable means such as the struct module,
2859 # but the tests here are aimed at testing the ancillary data
2860 # handling in sendmsg() and recvmsg() rather than the IPv6 API
2861 # itself.
2862
2863 # Test value to use when setting hop limit of packet
2864 hop_limit = 2
2865
2866 # Test value to use when setting traffic class of packet.
2867 # -1 means "use kernel default".
2868 traffic_class = -1
2869
2870 def ancillaryMapping(self, ancdata):
2871 # Given ancillary data list ancdata, return a mapping from
2872 # pairs (cmsg_level, cmsg_type) to corresponding cmsg_data.
2873 # Check that no (level, type) pair appears more than once.
2874 d = {}
2875 for cmsg_level, cmsg_type, cmsg_data in ancdata:
2876 self.assertNotIn((cmsg_level, cmsg_type), d)
2877 d[(cmsg_level, cmsg_type)] = cmsg_data
2878 return d
2879
2880 def checkHopLimit(self, ancbufsize, maxhop=255, ignoreflags=0):
2881 # Receive hop limit into ancbufsize bytes of ancillary data
2882 # space. Check that data is MSG, ancillary data is not
2883 # truncated (but ignore any flags in ignoreflags), and hop
2884 # limit is between 0 and maxhop inclusive.
2885 self.serv_sock.setsockopt(socket.IPPROTO_IPV6,
2886 socket.IPV6_RECVHOPLIMIT, 1)
2887 self.misc_event.set()
2888 msg, ancdata, flags, addr = self.doRecvmsg(self.serv_sock,
2889 len(MSG), ancbufsize)
2890
2891 self.assertEqual(msg, MSG)
2892 self.checkRecvmsgAddress(addr, self.cli_addr)
2893 self.checkFlags(flags, eor=True, checkunset=socket.MSG_CTRUNC,
2894 ignore=ignoreflags)
2895
2896 self.assertEqual(len(ancdata), 1)
2897 self.assertIsInstance(ancdata[0], tuple)
2898 cmsg_level, cmsg_type, cmsg_data = ancdata[0]
2899 self.assertEqual(cmsg_level, socket.IPPROTO_IPV6)
2900 self.assertEqual(cmsg_type, socket.IPV6_HOPLIMIT)
2901 self.assertIsInstance(cmsg_data, bytes)
2902 self.assertEqual(len(cmsg_data), SIZEOF_INT)
2903 a = array.array("i")
Nadeem Vawda3d1706f2011-08-22 09:46:56 +02002904 a.frombytes(cmsg_data)
Nick Coghlan96fe56a2011-08-22 11:55:57 +10002905 self.assertGreaterEqual(a[0], 0)
2906 self.assertLessEqual(a[0], maxhop)
2907
2908 @requireAttrs(socket, "IPV6_RECVHOPLIMIT", "IPV6_HOPLIMIT")
2909 def testRecvHopLimit(self):
2910 # Test receiving the packet hop limit as ancillary data.
2911 self.checkHopLimit(ancbufsize=10240)
2912
2913 @testRecvHopLimit.client_skip
2914 def _testRecvHopLimit(self):
2915 # Need to wait until server has asked to receive ancillary
2916 # data, as implementations are not required to buffer it
2917 # otherwise.
2918 self.assertTrue(self.misc_event.wait(timeout=self.fail_timeout))
2919 self.sendToServer(MSG)
2920
2921 @requireAttrs(socket, "CMSG_SPACE", "IPV6_RECVHOPLIMIT", "IPV6_HOPLIMIT")
2922 def testRecvHopLimitCMSG_SPACE(self):
2923 # Test receiving hop limit, using CMSG_SPACE to calculate buffer size.
2924 self.checkHopLimit(ancbufsize=socket.CMSG_SPACE(SIZEOF_INT))
2925
2926 @testRecvHopLimitCMSG_SPACE.client_skip
2927 def _testRecvHopLimitCMSG_SPACE(self):
2928 self.assertTrue(self.misc_event.wait(timeout=self.fail_timeout))
2929 self.sendToServer(MSG)
2930
2931 # Could test receiving into buffer sized using CMSG_LEN, but RFC
2932 # 3542 says portable applications must provide space for trailing
2933 # padding. Implementations may set MSG_CTRUNC if there isn't
2934 # enough space for the padding.
2935
2936 @requireAttrs(socket.socket, "sendmsg")
2937 @requireAttrs(socket, "IPV6_RECVHOPLIMIT", "IPV6_HOPLIMIT")
2938 def testSetHopLimit(self):
2939 # Test setting hop limit on outgoing packet and receiving it
2940 # at the other end.
2941 self.checkHopLimit(ancbufsize=10240, maxhop=self.hop_limit)
2942
2943 @testSetHopLimit.client_skip
2944 def _testSetHopLimit(self):
2945 self.assertTrue(self.misc_event.wait(timeout=self.fail_timeout))
2946 self.assertEqual(
2947 self.sendmsgToServer([MSG],
2948 [(socket.IPPROTO_IPV6, socket.IPV6_HOPLIMIT,
2949 array.array("i", [self.hop_limit]))]),
2950 len(MSG))
2951
2952 def checkTrafficClassAndHopLimit(self, ancbufsize, maxhop=255,
2953 ignoreflags=0):
2954 # Receive traffic class and hop limit into ancbufsize bytes of
2955 # ancillary data space. Check that data is MSG, ancillary
2956 # data is not truncated (but ignore any flags in ignoreflags),
2957 # and traffic class and hop limit are in range (hop limit no
2958 # more than maxhop).
2959 self.serv_sock.setsockopt(socket.IPPROTO_IPV6,
2960 socket.IPV6_RECVHOPLIMIT, 1)
2961 self.serv_sock.setsockopt(socket.IPPROTO_IPV6,
2962 socket.IPV6_RECVTCLASS, 1)
2963 self.misc_event.set()
2964 msg, ancdata, flags, addr = self.doRecvmsg(self.serv_sock,
2965 len(MSG), ancbufsize)
2966
2967 self.assertEqual(msg, MSG)
2968 self.checkRecvmsgAddress(addr, self.cli_addr)
2969 self.checkFlags(flags, eor=True, checkunset=socket.MSG_CTRUNC,
2970 ignore=ignoreflags)
2971 self.assertEqual(len(ancdata), 2)
2972 ancmap = self.ancillaryMapping(ancdata)
2973
2974 tcdata = ancmap[(socket.IPPROTO_IPV6, socket.IPV6_TCLASS)]
2975 self.assertEqual(len(tcdata), SIZEOF_INT)
2976 a = array.array("i")
Nadeem Vawda3d1706f2011-08-22 09:46:56 +02002977 a.frombytes(tcdata)
Nick Coghlan96fe56a2011-08-22 11:55:57 +10002978 self.assertGreaterEqual(a[0], 0)
2979 self.assertLessEqual(a[0], 255)
2980
2981 hldata = ancmap[(socket.IPPROTO_IPV6, socket.IPV6_HOPLIMIT)]
2982 self.assertEqual(len(hldata), SIZEOF_INT)
2983 a = array.array("i")
Nadeem Vawda3d1706f2011-08-22 09:46:56 +02002984 a.frombytes(hldata)
Nick Coghlan96fe56a2011-08-22 11:55:57 +10002985 self.assertGreaterEqual(a[0], 0)
2986 self.assertLessEqual(a[0], maxhop)
2987
2988 @requireAttrs(socket, "IPV6_RECVHOPLIMIT", "IPV6_HOPLIMIT",
2989 "IPV6_RECVTCLASS", "IPV6_TCLASS")
2990 def testRecvTrafficClassAndHopLimit(self):
2991 # Test receiving traffic class and hop limit as ancillary data.
2992 self.checkTrafficClassAndHopLimit(ancbufsize=10240)
2993
2994 @testRecvTrafficClassAndHopLimit.client_skip
2995 def _testRecvTrafficClassAndHopLimit(self):
2996 self.assertTrue(self.misc_event.wait(timeout=self.fail_timeout))
2997 self.sendToServer(MSG)
2998
2999 @requireAttrs(socket, "CMSG_SPACE", "IPV6_RECVHOPLIMIT", "IPV6_HOPLIMIT",
3000 "IPV6_RECVTCLASS", "IPV6_TCLASS")
3001 def testRecvTrafficClassAndHopLimitCMSG_SPACE(self):
3002 # Test receiving traffic class and hop limit, using
3003 # CMSG_SPACE() to calculate buffer size.
3004 self.checkTrafficClassAndHopLimit(
3005 ancbufsize=socket.CMSG_SPACE(SIZEOF_INT) * 2)
3006
3007 @testRecvTrafficClassAndHopLimitCMSG_SPACE.client_skip
3008 def _testRecvTrafficClassAndHopLimitCMSG_SPACE(self):
3009 self.assertTrue(self.misc_event.wait(timeout=self.fail_timeout))
3010 self.sendToServer(MSG)
3011
3012 @requireAttrs(socket.socket, "sendmsg")
3013 @requireAttrs(socket, "CMSG_SPACE", "IPV6_RECVHOPLIMIT", "IPV6_HOPLIMIT",
3014 "IPV6_RECVTCLASS", "IPV6_TCLASS")
3015 def testSetTrafficClassAndHopLimit(self):
3016 # Test setting traffic class and hop limit on outgoing packet,
3017 # and receiving them at the other end.
3018 self.checkTrafficClassAndHopLimit(ancbufsize=10240,
3019 maxhop=self.hop_limit)
3020
3021 @testSetTrafficClassAndHopLimit.client_skip
3022 def _testSetTrafficClassAndHopLimit(self):
3023 self.assertTrue(self.misc_event.wait(timeout=self.fail_timeout))
3024 self.assertEqual(
3025 self.sendmsgToServer([MSG],
3026 [(socket.IPPROTO_IPV6, socket.IPV6_TCLASS,
3027 array.array("i", [self.traffic_class])),
3028 (socket.IPPROTO_IPV6, socket.IPV6_HOPLIMIT,
3029 array.array("i", [self.hop_limit]))]),
3030 len(MSG))
3031
3032 @requireAttrs(socket.socket, "sendmsg")
3033 @requireAttrs(socket, "CMSG_SPACE", "IPV6_RECVHOPLIMIT", "IPV6_HOPLIMIT",
3034 "IPV6_RECVTCLASS", "IPV6_TCLASS")
3035 def testOddCmsgSize(self):
3036 # Try to send ancillary data with first item one byte too
3037 # long. Fall back to sending with correct size if this fails,
3038 # and check that second item was handled correctly.
3039 self.checkTrafficClassAndHopLimit(ancbufsize=10240,
3040 maxhop=self.hop_limit)
3041
3042 @testOddCmsgSize.client_skip
3043 def _testOddCmsgSize(self):
3044 self.assertTrue(self.misc_event.wait(timeout=self.fail_timeout))
3045 try:
3046 nbytes = self.sendmsgToServer(
3047 [MSG],
3048 [(socket.IPPROTO_IPV6, socket.IPV6_TCLASS,
Nadeem Vawda3d1706f2011-08-22 09:46:56 +02003049 array.array("i", [self.traffic_class]).tobytes() + b"\x00"),
Nick Coghlan96fe56a2011-08-22 11:55:57 +10003050 (socket.IPPROTO_IPV6, socket.IPV6_HOPLIMIT,
3051 array.array("i", [self.hop_limit]))])
Andrew Svetlov0832af62012-12-18 23:10:48 +02003052 except OSError as e:
Nick Coghlan96fe56a2011-08-22 11:55:57 +10003053 self.assertIsInstance(e.errno, int)
3054 nbytes = self.sendmsgToServer(
3055 [MSG],
3056 [(socket.IPPROTO_IPV6, socket.IPV6_TCLASS,
3057 array.array("i", [self.traffic_class])),
3058 (socket.IPPROTO_IPV6, socket.IPV6_HOPLIMIT,
3059 array.array("i", [self.hop_limit]))])
3060 self.assertEqual(nbytes, len(MSG))
3061
3062 # Tests for proper handling of truncated ancillary data
3063
3064 def checkHopLimitTruncatedHeader(self, ancbufsize, ignoreflags=0):
3065 # Receive hop limit into ancbufsize bytes of ancillary data
3066 # space, which should be too small to contain the ancillary
3067 # data header (if ancbufsize is None, pass no second argument
3068 # to recvmsg()). Check that data is MSG, MSG_CTRUNC is set
3069 # (unless included in ignoreflags), and no ancillary data is
3070 # returned.
3071 self.serv_sock.setsockopt(socket.IPPROTO_IPV6,
3072 socket.IPV6_RECVHOPLIMIT, 1)
3073 self.misc_event.set()
3074 args = () if ancbufsize is None else (ancbufsize,)
3075 msg, ancdata, flags, addr = self.doRecvmsg(self.serv_sock,
3076 len(MSG), *args)
3077
3078 self.assertEqual(msg, MSG)
3079 self.checkRecvmsgAddress(addr, self.cli_addr)
3080 self.assertEqual(ancdata, [])
3081 self.checkFlags(flags, eor=True, checkset=socket.MSG_CTRUNC,
3082 ignore=ignoreflags)
3083
3084 @requireAttrs(socket, "IPV6_RECVHOPLIMIT", "IPV6_HOPLIMIT")
3085 def testCmsgTruncNoBufSize(self):
3086 # Check that no ancillary data is received when no ancillary
3087 # buffer size is provided.
3088 self.checkHopLimitTruncatedHeader(ancbufsize=None,
3089 # BSD seems to set
3090 # MSG_CTRUNC only if an item
3091 # has been partially
3092 # received.
3093 ignoreflags=socket.MSG_CTRUNC)
3094
3095 @testCmsgTruncNoBufSize.client_skip
3096 def _testCmsgTruncNoBufSize(self):
3097 self.assertTrue(self.misc_event.wait(timeout=self.fail_timeout))
3098 self.sendToServer(MSG)
3099
3100 @requireAttrs(socket, "IPV6_RECVHOPLIMIT", "IPV6_HOPLIMIT")
3101 def testSingleCmsgTrunc0(self):
3102 # Check that no ancillary data is received when ancillary
3103 # buffer size is zero.
3104 self.checkHopLimitTruncatedHeader(ancbufsize=0,
3105 ignoreflags=socket.MSG_CTRUNC)
3106
3107 @testSingleCmsgTrunc0.client_skip
3108 def _testSingleCmsgTrunc0(self):
3109 self.assertTrue(self.misc_event.wait(timeout=self.fail_timeout))
3110 self.sendToServer(MSG)
3111
3112 # Check that no ancillary data is returned for various non-zero
3113 # (but still too small) buffer sizes.
3114
3115 @requireAttrs(socket, "IPV6_RECVHOPLIMIT", "IPV6_HOPLIMIT")
3116 def testSingleCmsgTrunc1(self):
3117 self.checkHopLimitTruncatedHeader(ancbufsize=1)
3118
3119 @testSingleCmsgTrunc1.client_skip
3120 def _testSingleCmsgTrunc1(self):
3121 self.assertTrue(self.misc_event.wait(timeout=self.fail_timeout))
3122 self.sendToServer(MSG)
3123
3124 @requireAttrs(socket, "IPV6_RECVHOPLIMIT", "IPV6_HOPLIMIT")
3125 def testSingleCmsgTrunc2Int(self):
3126 self.checkHopLimitTruncatedHeader(ancbufsize=2 * SIZEOF_INT)
3127
3128 @testSingleCmsgTrunc2Int.client_skip
3129 def _testSingleCmsgTrunc2Int(self):
3130 self.assertTrue(self.misc_event.wait(timeout=self.fail_timeout))
3131 self.sendToServer(MSG)
3132
3133 @requireAttrs(socket, "IPV6_RECVHOPLIMIT", "IPV6_HOPLIMIT")
3134 def testSingleCmsgTruncLen0Minus1(self):
3135 self.checkHopLimitTruncatedHeader(ancbufsize=socket.CMSG_LEN(0) - 1)
3136
3137 @testSingleCmsgTruncLen0Minus1.client_skip
3138 def _testSingleCmsgTruncLen0Minus1(self):
3139 self.assertTrue(self.misc_event.wait(timeout=self.fail_timeout))
3140 self.sendToServer(MSG)
3141
3142 @requireAttrs(socket, "IPV6_RECVHOPLIMIT", "IPV6_HOPLIMIT")
3143 def testSingleCmsgTruncInData(self):
3144 # Test truncation of a control message inside its associated
3145 # data. The message may be returned with its data truncated,
3146 # or not returned at all.
3147 self.serv_sock.setsockopt(socket.IPPROTO_IPV6,
3148 socket.IPV6_RECVHOPLIMIT, 1)
3149 self.misc_event.set()
3150 msg, ancdata, flags, addr = self.doRecvmsg(
3151 self.serv_sock, len(MSG), socket.CMSG_LEN(SIZEOF_INT) - 1)
3152
3153 self.assertEqual(msg, MSG)
3154 self.checkRecvmsgAddress(addr, self.cli_addr)
3155 self.checkFlags(flags, eor=True, checkset=socket.MSG_CTRUNC)
3156
3157 self.assertLessEqual(len(ancdata), 1)
3158 if ancdata:
3159 cmsg_level, cmsg_type, cmsg_data = ancdata[0]
3160 self.assertEqual(cmsg_level, socket.IPPROTO_IPV6)
3161 self.assertEqual(cmsg_type, socket.IPV6_HOPLIMIT)
3162 self.assertLess(len(cmsg_data), SIZEOF_INT)
3163
3164 @testSingleCmsgTruncInData.client_skip
3165 def _testSingleCmsgTruncInData(self):
3166 self.assertTrue(self.misc_event.wait(timeout=self.fail_timeout))
3167 self.sendToServer(MSG)
3168
3169 def checkTruncatedSecondHeader(self, ancbufsize, ignoreflags=0):
3170 # Receive traffic class and hop limit into ancbufsize bytes of
3171 # ancillary data space, which should be large enough to
3172 # contain the first item, but too small to contain the header
3173 # of the second. Check that data is MSG, MSG_CTRUNC is set
3174 # (unless included in ignoreflags), and only one ancillary
3175 # data item is returned.
3176 self.serv_sock.setsockopt(socket.IPPROTO_IPV6,
3177 socket.IPV6_RECVHOPLIMIT, 1)
3178 self.serv_sock.setsockopt(socket.IPPROTO_IPV6,
3179 socket.IPV6_RECVTCLASS, 1)
3180 self.misc_event.set()
3181 msg, ancdata, flags, addr = self.doRecvmsg(self.serv_sock,
3182 len(MSG), ancbufsize)
3183
3184 self.assertEqual(msg, MSG)
3185 self.checkRecvmsgAddress(addr, self.cli_addr)
3186 self.checkFlags(flags, eor=True, checkset=socket.MSG_CTRUNC,
3187 ignore=ignoreflags)
3188
3189 self.assertEqual(len(ancdata), 1)
3190 cmsg_level, cmsg_type, cmsg_data = ancdata[0]
3191 self.assertEqual(cmsg_level, socket.IPPROTO_IPV6)
3192 self.assertIn(cmsg_type, {socket.IPV6_TCLASS, socket.IPV6_HOPLIMIT})
3193 self.assertEqual(len(cmsg_data), SIZEOF_INT)
3194 a = array.array("i")
Nadeem Vawda3d1706f2011-08-22 09:46:56 +02003195 a.frombytes(cmsg_data)
Nick Coghlan96fe56a2011-08-22 11:55:57 +10003196 self.assertGreaterEqual(a[0], 0)
3197 self.assertLessEqual(a[0], 255)
3198
3199 # Try the above test with various buffer sizes.
3200
3201 @requireAttrs(socket, "CMSG_SPACE", "IPV6_RECVHOPLIMIT", "IPV6_HOPLIMIT",
3202 "IPV6_RECVTCLASS", "IPV6_TCLASS")
3203 def testSecondCmsgTrunc0(self):
3204 self.checkTruncatedSecondHeader(socket.CMSG_SPACE(SIZEOF_INT),
3205 ignoreflags=socket.MSG_CTRUNC)
3206
3207 @testSecondCmsgTrunc0.client_skip
3208 def _testSecondCmsgTrunc0(self):
3209 self.assertTrue(self.misc_event.wait(timeout=self.fail_timeout))
3210 self.sendToServer(MSG)
3211
3212 @requireAttrs(socket, "CMSG_SPACE", "IPV6_RECVHOPLIMIT", "IPV6_HOPLIMIT",
3213 "IPV6_RECVTCLASS", "IPV6_TCLASS")
3214 def testSecondCmsgTrunc1(self):
3215 self.checkTruncatedSecondHeader(socket.CMSG_SPACE(SIZEOF_INT) + 1)
3216
3217 @testSecondCmsgTrunc1.client_skip
3218 def _testSecondCmsgTrunc1(self):
3219 self.assertTrue(self.misc_event.wait(timeout=self.fail_timeout))
3220 self.sendToServer(MSG)
3221
3222 @requireAttrs(socket, "CMSG_SPACE", "IPV6_RECVHOPLIMIT", "IPV6_HOPLIMIT",
3223 "IPV6_RECVTCLASS", "IPV6_TCLASS")
3224 def testSecondCmsgTrunc2Int(self):
3225 self.checkTruncatedSecondHeader(socket.CMSG_SPACE(SIZEOF_INT) +
3226 2 * SIZEOF_INT)
3227
3228 @testSecondCmsgTrunc2Int.client_skip
3229 def _testSecondCmsgTrunc2Int(self):
3230 self.assertTrue(self.misc_event.wait(timeout=self.fail_timeout))
3231 self.sendToServer(MSG)
3232
3233 @requireAttrs(socket, "CMSG_SPACE", "IPV6_RECVHOPLIMIT", "IPV6_HOPLIMIT",
3234 "IPV6_RECVTCLASS", "IPV6_TCLASS")
3235 def testSecondCmsgTruncLen0Minus1(self):
3236 self.checkTruncatedSecondHeader(socket.CMSG_SPACE(SIZEOF_INT) +
3237 socket.CMSG_LEN(0) - 1)
3238
3239 @testSecondCmsgTruncLen0Minus1.client_skip
3240 def _testSecondCmsgTruncLen0Minus1(self):
3241 self.assertTrue(self.misc_event.wait(timeout=self.fail_timeout))
3242 self.sendToServer(MSG)
3243
3244 @requireAttrs(socket, "CMSG_SPACE", "IPV6_RECVHOPLIMIT", "IPV6_HOPLIMIT",
3245 "IPV6_RECVTCLASS", "IPV6_TCLASS")
3246 def testSecomdCmsgTruncInData(self):
3247 # Test truncation of the second of two control messages inside
3248 # its associated data.
3249 self.serv_sock.setsockopt(socket.IPPROTO_IPV6,
3250 socket.IPV6_RECVHOPLIMIT, 1)
3251 self.serv_sock.setsockopt(socket.IPPROTO_IPV6,
3252 socket.IPV6_RECVTCLASS, 1)
3253 self.misc_event.set()
3254 msg, ancdata, flags, addr = self.doRecvmsg(
3255 self.serv_sock, len(MSG),
3256 socket.CMSG_SPACE(SIZEOF_INT) + socket.CMSG_LEN(SIZEOF_INT) - 1)
3257
3258 self.assertEqual(msg, MSG)
3259 self.checkRecvmsgAddress(addr, self.cli_addr)
3260 self.checkFlags(flags, eor=True, checkset=socket.MSG_CTRUNC)
3261
3262 cmsg_types = {socket.IPV6_TCLASS, socket.IPV6_HOPLIMIT}
3263
3264 cmsg_level, cmsg_type, cmsg_data = ancdata.pop(0)
3265 self.assertEqual(cmsg_level, socket.IPPROTO_IPV6)
3266 cmsg_types.remove(cmsg_type)
3267 self.assertEqual(len(cmsg_data), SIZEOF_INT)
3268 a = array.array("i")
Nadeem Vawda3d1706f2011-08-22 09:46:56 +02003269 a.frombytes(cmsg_data)
Nick Coghlan96fe56a2011-08-22 11:55:57 +10003270 self.assertGreaterEqual(a[0], 0)
3271 self.assertLessEqual(a[0], 255)
3272
3273 if ancdata:
3274 cmsg_level, cmsg_type, cmsg_data = ancdata.pop(0)
3275 self.assertEqual(cmsg_level, socket.IPPROTO_IPV6)
3276 cmsg_types.remove(cmsg_type)
3277 self.assertLess(len(cmsg_data), SIZEOF_INT)
3278
3279 self.assertEqual(ancdata, [])
3280
3281 @testSecomdCmsgTruncInData.client_skip
3282 def _testSecomdCmsgTruncInData(self):
3283 self.assertTrue(self.misc_event.wait(timeout=self.fail_timeout))
3284 self.sendToServer(MSG)
3285
3286
3287# Derive concrete test classes for different socket types.
3288
3289class SendrecvmsgUDPTestBase(SendrecvmsgDgramFlagsBase,
3290 SendrecvmsgConnectionlessBase,
3291 ThreadedSocketTestMixin, UDPTestBase):
3292 pass
3293
3294@requireAttrs(socket.socket, "sendmsg")
3295@unittest.skipUnless(thread, 'Threading required for this test.')
3296class SendmsgUDPTest(SendmsgConnectionlessTests, SendrecvmsgUDPTestBase):
3297 pass
3298
3299@requireAttrs(socket.socket, "recvmsg")
3300@unittest.skipUnless(thread, 'Threading required for this test.')
3301class RecvmsgUDPTest(RecvmsgTests, SendrecvmsgUDPTestBase):
3302 pass
3303
3304@requireAttrs(socket.socket, "recvmsg_into")
3305@unittest.skipUnless(thread, 'Threading required for this test.')
3306class RecvmsgIntoUDPTest(RecvmsgIntoTests, SendrecvmsgUDPTestBase):
3307 pass
3308
3309
3310class SendrecvmsgUDP6TestBase(SendrecvmsgDgramFlagsBase,
3311 SendrecvmsgConnectionlessBase,
3312 ThreadedSocketTestMixin, UDP6TestBase):
3313 pass
3314
3315@requireAttrs(socket.socket, "sendmsg")
Ross Lagerwall8d459f02012-07-08 09:53:57 +02003316@unittest.skipUnless(support.IPV6_ENABLED, 'IPv6 required for this test.')
Nick Coghlan96fe56a2011-08-22 11:55:57 +10003317@requireSocket("AF_INET6", "SOCK_DGRAM")
3318@unittest.skipUnless(thread, 'Threading required for this test.')
3319class SendmsgUDP6Test(SendmsgConnectionlessTests, SendrecvmsgUDP6TestBase):
3320 pass
3321
3322@requireAttrs(socket.socket, "recvmsg")
Ross Lagerwall8d459f02012-07-08 09:53:57 +02003323@unittest.skipUnless(support.IPV6_ENABLED, 'IPv6 required for this test.')
Nick Coghlan96fe56a2011-08-22 11:55:57 +10003324@requireSocket("AF_INET6", "SOCK_DGRAM")
3325@unittest.skipUnless(thread, 'Threading required for this test.')
3326class RecvmsgUDP6Test(RecvmsgTests, SendrecvmsgUDP6TestBase):
3327 pass
3328
3329@requireAttrs(socket.socket, "recvmsg_into")
Ross Lagerwall8d459f02012-07-08 09:53:57 +02003330@unittest.skipUnless(support.IPV6_ENABLED, 'IPv6 required for this test.')
Nick Coghlan96fe56a2011-08-22 11:55:57 +10003331@requireSocket("AF_INET6", "SOCK_DGRAM")
3332@unittest.skipUnless(thread, 'Threading required for this test.')
3333class RecvmsgIntoUDP6Test(RecvmsgIntoTests, SendrecvmsgUDP6TestBase):
3334 pass
3335
3336@requireAttrs(socket.socket, "recvmsg")
Ross Lagerwall8d459f02012-07-08 09:53:57 +02003337@unittest.skipUnless(support.IPV6_ENABLED, 'IPv6 required for this test.')
Nick Coghlan96fe56a2011-08-22 11:55:57 +10003338@requireAttrs(socket, "IPPROTO_IPV6")
3339@requireSocket("AF_INET6", "SOCK_DGRAM")
3340@unittest.skipUnless(thread, 'Threading required for this test.')
3341class RecvmsgRFC3542AncillaryUDP6Test(RFC3542AncillaryTest,
3342 SendrecvmsgUDP6TestBase):
3343 pass
3344
3345@requireAttrs(socket.socket, "recvmsg_into")
Ross Lagerwall8d459f02012-07-08 09:53:57 +02003346@unittest.skipUnless(support.IPV6_ENABLED, 'IPv6 required for this test.')
Nick Coghlan96fe56a2011-08-22 11:55:57 +10003347@requireAttrs(socket, "IPPROTO_IPV6")
3348@requireSocket("AF_INET6", "SOCK_DGRAM")
3349@unittest.skipUnless(thread, 'Threading required for this test.')
3350class RecvmsgIntoRFC3542AncillaryUDP6Test(RecvmsgIntoMixin,
3351 RFC3542AncillaryTest,
3352 SendrecvmsgUDP6TestBase):
3353 pass
3354
3355
3356class SendrecvmsgTCPTestBase(SendrecvmsgConnectedBase,
3357 ConnectedStreamTestMixin, TCPTestBase):
3358 pass
3359
3360@requireAttrs(socket.socket, "sendmsg")
3361@unittest.skipUnless(thread, 'Threading required for this test.')
3362class SendmsgTCPTest(SendmsgStreamTests, SendrecvmsgTCPTestBase):
3363 pass
3364
3365@requireAttrs(socket.socket, "recvmsg")
3366@unittest.skipUnless(thread, 'Threading required for this test.')
3367class RecvmsgTCPTest(RecvmsgTests, RecvmsgGenericStreamTests,
3368 SendrecvmsgTCPTestBase):
3369 pass
3370
3371@requireAttrs(socket.socket, "recvmsg_into")
3372@unittest.skipUnless(thread, 'Threading required for this test.')
3373class RecvmsgIntoTCPTest(RecvmsgIntoTests, RecvmsgGenericStreamTests,
3374 SendrecvmsgTCPTestBase):
3375 pass
3376
3377
3378class SendrecvmsgSCTPStreamTestBase(SendrecvmsgSCTPFlagsBase,
3379 SendrecvmsgConnectedBase,
3380 ConnectedStreamTestMixin, SCTPStreamBase):
3381 pass
3382
3383@requireAttrs(socket.socket, "sendmsg")
3384@requireSocket("AF_INET", "SOCK_STREAM", "IPPROTO_SCTP")
3385@unittest.skipUnless(thread, 'Threading required for this test.')
3386class SendmsgSCTPStreamTest(SendmsgStreamTests, SendrecvmsgSCTPStreamTestBase):
3387 pass
3388
3389@requireAttrs(socket.socket, "recvmsg")
3390@requireSocket("AF_INET", "SOCK_STREAM", "IPPROTO_SCTP")
3391@unittest.skipUnless(thread, 'Threading required for this test.')
3392class RecvmsgSCTPStreamTest(RecvmsgTests, RecvmsgGenericStreamTests,
3393 SendrecvmsgSCTPStreamTestBase):
Antoine Pitrou92f8b002012-12-16 16:40:04 +01003394
3395 def testRecvmsgEOF(self):
3396 try:
3397 super(RecvmsgSCTPStreamTest, self).testRecvmsgEOF()
3398 except OSError as e:
3399 if e.errno != errno.ENOTCONN:
3400 raise
3401 self.skipTest("sporadic ENOTCONN (kernel issue?) - see issue #13876")
Nick Coghlan96fe56a2011-08-22 11:55:57 +10003402
3403@requireAttrs(socket.socket, "recvmsg_into")
3404@requireSocket("AF_INET", "SOCK_STREAM", "IPPROTO_SCTP")
3405@unittest.skipUnless(thread, 'Threading required for this test.')
3406class RecvmsgIntoSCTPStreamTest(RecvmsgIntoTests, RecvmsgGenericStreamTests,
3407 SendrecvmsgSCTPStreamTestBase):
Antoine Pitroud5a070f2012-12-18 23:44:36 +01003408
3409 def testRecvmsgEOF(self):
3410 try:
3411 super(RecvmsgIntoSCTPStreamTest, self).testRecvmsgEOF()
3412 except OSError as e:
3413 if e.errno != errno.ENOTCONN:
3414 raise
3415 self.skipTest("sporadic ENOTCONN (kernel issue?) - see issue #13876")
Nick Coghlan96fe56a2011-08-22 11:55:57 +10003416
3417
3418class SendrecvmsgUnixStreamTestBase(SendrecvmsgConnectedBase,
3419 ConnectedStreamTestMixin, UnixStreamBase):
3420 pass
3421
3422@requireAttrs(socket.socket, "sendmsg")
3423@requireAttrs(socket, "AF_UNIX")
3424@unittest.skipUnless(thread, 'Threading required for this test.')
3425class SendmsgUnixStreamTest(SendmsgStreamTests, SendrecvmsgUnixStreamTestBase):
3426 pass
3427
3428@requireAttrs(socket.socket, "recvmsg")
3429@requireAttrs(socket, "AF_UNIX")
3430@unittest.skipUnless(thread, 'Threading required for this test.')
3431class RecvmsgUnixStreamTest(RecvmsgTests, RecvmsgGenericStreamTests,
3432 SendrecvmsgUnixStreamTestBase):
3433 pass
3434
3435@requireAttrs(socket.socket, "recvmsg_into")
3436@requireAttrs(socket, "AF_UNIX")
3437@unittest.skipUnless(thread, 'Threading required for this test.')
3438class RecvmsgIntoUnixStreamTest(RecvmsgIntoTests, RecvmsgGenericStreamTests,
3439 SendrecvmsgUnixStreamTestBase):
3440 pass
3441
3442@requireAttrs(socket.socket, "sendmsg", "recvmsg")
3443@requireAttrs(socket, "AF_UNIX", "SOL_SOCKET", "SCM_RIGHTS")
3444@unittest.skipUnless(thread, 'Threading required for this test.')
3445class RecvmsgSCMRightsStreamTest(SCMRightsTest, SendrecvmsgUnixStreamTestBase):
3446 pass
3447
3448@requireAttrs(socket.socket, "sendmsg", "recvmsg_into")
3449@requireAttrs(socket, "AF_UNIX", "SOL_SOCKET", "SCM_RIGHTS")
3450@unittest.skipUnless(thread, 'Threading required for this test.')
3451class RecvmsgIntoSCMRightsStreamTest(RecvmsgIntoMixin, SCMRightsTest,
3452 SendrecvmsgUnixStreamTestBase):
3453 pass
3454
3455
3456# Test interrupting the interruptible send/receive methods with a
3457# signal when a timeout is set. These tests avoid having multiple
3458# threads alive during the test so that the OS cannot deliver the
3459# signal to the wrong one.
3460
3461class InterruptedTimeoutBase(unittest.TestCase):
3462 # Base class for interrupted send/receive tests. Installs an
3463 # empty handler for SIGALRM and removes it on teardown, along with
3464 # any scheduled alarms.
3465
3466 def setUp(self):
3467 super().setUp()
3468 orig_alrm_handler = signal.signal(signal.SIGALRM,
3469 lambda signum, frame: None)
3470 self.addCleanup(signal.signal, signal.SIGALRM, orig_alrm_handler)
3471 self.addCleanup(self.setAlarm, 0)
3472
3473 # Timeout for socket operations
3474 timeout = 4.0
3475
3476 # Provide setAlarm() method to schedule delivery of SIGALRM after
3477 # given number of seconds, or cancel it if zero, and an
3478 # appropriate time value to use. Use setitimer() if available.
3479 if hasattr(signal, "setitimer"):
3480 alarm_time = 0.05
3481
3482 def setAlarm(self, seconds):
3483 signal.setitimer(signal.ITIMER_REAL, seconds)
3484 else:
3485 # Old systems may deliver the alarm up to one second early
3486 alarm_time = 2
3487
3488 def setAlarm(self, seconds):
3489 signal.alarm(seconds)
3490
3491
3492# Require siginterrupt() in order to ensure that system calls are
3493# interrupted by default.
3494@requireAttrs(signal, "siginterrupt")
3495@unittest.skipUnless(hasattr(signal, "alarm") or hasattr(signal, "setitimer"),
3496 "Don't have signal.alarm or signal.setitimer")
3497class InterruptedRecvTimeoutTest(InterruptedTimeoutBase, UDPTestBase):
3498 # Test interrupting the recv*() methods with signals when a
3499 # timeout is set.
3500
3501 def setUp(self):
3502 super().setUp()
3503 self.serv.settimeout(self.timeout)
3504
3505 def checkInterruptedRecv(self, func, *args, **kwargs):
Andrew Svetlov0832af62012-12-18 23:10:48 +02003506 # Check that func(*args, **kwargs) raises OSError with an
Nick Coghlan96fe56a2011-08-22 11:55:57 +10003507 # errno of EINTR when interrupted by a signal.
3508 self.setAlarm(self.alarm_time)
Andrew Svetlov0832af62012-12-18 23:10:48 +02003509 with self.assertRaises(OSError) as cm:
Nick Coghlan96fe56a2011-08-22 11:55:57 +10003510 func(*args, **kwargs)
3511 self.assertNotIsInstance(cm.exception, socket.timeout)
3512 self.assertEqual(cm.exception.errno, errno.EINTR)
3513
3514 def testInterruptedRecvTimeout(self):
3515 self.checkInterruptedRecv(self.serv.recv, 1024)
3516
3517 def testInterruptedRecvIntoTimeout(self):
3518 self.checkInterruptedRecv(self.serv.recv_into, bytearray(1024))
3519
3520 def testInterruptedRecvfromTimeout(self):
3521 self.checkInterruptedRecv(self.serv.recvfrom, 1024)
3522
3523 def testInterruptedRecvfromIntoTimeout(self):
3524 self.checkInterruptedRecv(self.serv.recvfrom_into, bytearray(1024))
3525
3526 @requireAttrs(socket.socket, "recvmsg")
3527 def testInterruptedRecvmsgTimeout(self):
3528 self.checkInterruptedRecv(self.serv.recvmsg, 1024)
3529
3530 @requireAttrs(socket.socket, "recvmsg_into")
3531 def testInterruptedRecvmsgIntoTimeout(self):
3532 self.checkInterruptedRecv(self.serv.recvmsg_into, [bytearray(1024)])
3533
3534
3535# Require siginterrupt() in order to ensure that system calls are
3536# interrupted by default.
3537@requireAttrs(signal, "siginterrupt")
3538@unittest.skipUnless(hasattr(signal, "alarm") or hasattr(signal, "setitimer"),
3539 "Don't have signal.alarm or signal.setitimer")
3540@unittest.skipUnless(thread, 'Threading required for this test.')
3541class InterruptedSendTimeoutTest(InterruptedTimeoutBase,
3542 ThreadSafeCleanupTestCase,
3543 SocketListeningTestMixin, TCPTestBase):
3544 # Test interrupting the interruptible send*() methods with signals
3545 # when a timeout is set.
3546
3547 def setUp(self):
3548 super().setUp()
3549 self.serv_conn = self.newSocket()
3550 self.addCleanup(self.serv_conn.close)
3551 # Use a thread to complete the connection, but wait for it to
3552 # terminate before running the test, so that there is only one
3553 # thread to accept the signal.
3554 cli_thread = threading.Thread(target=self.doConnect)
3555 cli_thread.start()
3556 self.cli_conn, addr = self.serv.accept()
3557 self.addCleanup(self.cli_conn.close)
3558 cli_thread.join()
3559 self.serv_conn.settimeout(self.timeout)
3560
3561 def doConnect(self):
3562 self.serv_conn.connect(self.serv_addr)
3563
3564 def checkInterruptedSend(self, func, *args, **kwargs):
3565 # Check that func(*args, **kwargs), run in a loop, raises
Andrew Svetlov0832af62012-12-18 23:10:48 +02003566 # OSError with an errno of EINTR when interrupted by a
Nick Coghlan96fe56a2011-08-22 11:55:57 +10003567 # signal.
Andrew Svetlov0832af62012-12-18 23:10:48 +02003568 with self.assertRaises(OSError) as cm:
Nick Coghlan96fe56a2011-08-22 11:55:57 +10003569 while True:
3570 self.setAlarm(self.alarm_time)
3571 func(*args, **kwargs)
3572 self.assertNotIsInstance(cm.exception, socket.timeout)
3573 self.assertEqual(cm.exception.errno, errno.EINTR)
3574
Nick Coghlan2496f332011-09-19 20:26:31 +10003575 # Issue #12958: The following tests have problems on Mac OS X
3576 @support.anticipate_failure(sys.platform == "darwin")
Nick Coghlan96fe56a2011-08-22 11:55:57 +10003577 def testInterruptedSendTimeout(self):
3578 self.checkInterruptedSend(self.serv_conn.send, b"a"*512)
3579
Nick Coghlan2496f332011-09-19 20:26:31 +10003580 @support.anticipate_failure(sys.platform == "darwin")
Nick Coghlan96fe56a2011-08-22 11:55:57 +10003581 def testInterruptedSendtoTimeout(self):
3582 # Passing an actual address here as Python's wrapper for
3583 # sendto() doesn't allow passing a zero-length one; POSIX
3584 # requires that the address is ignored since the socket is
3585 # connection-mode, however.
3586 self.checkInterruptedSend(self.serv_conn.sendto, b"a"*512,
3587 self.serv_addr)
3588
Nick Coghlan2496f332011-09-19 20:26:31 +10003589 @support.anticipate_failure(sys.platform == "darwin")
Nick Coghlan96fe56a2011-08-22 11:55:57 +10003590 @requireAttrs(socket.socket, "sendmsg")
3591 def testInterruptedSendmsgTimeout(self):
3592 self.checkInterruptedSend(self.serv_conn.sendmsg, [b"a"*512])
3593
3594
Victor Stinner45df8202010-04-28 22:31:17 +00003595@unittest.skipUnless(thread, 'Threading required for this test.')
Thomas Wouters0e3f5912006-08-11 14:57:12 +00003596class TCPCloserTest(ThreadedTCPSocketTest):
3597
3598 def testClose(self):
3599 conn, addr = self.serv.accept()
3600 conn.close()
3601
3602 sd = self.cli
3603 read, write, err = select.select([sd], [], [], 1.0)
3604 self.assertEqual(read, [sd])
Guido van Rossum7d0a8262007-05-21 23:13:11 +00003605 self.assertEqual(sd.recv(1), b'')
Thomas Wouters0e3f5912006-08-11 14:57:12 +00003606
Jeremy Hylton5accbdb2007-08-03 20:40:09 +00003607 # Calling close() many times should be safe.
3608 conn.close()
3609 conn.close()
3610
Thomas Wouters0e3f5912006-08-11 14:57:12 +00003611 def _testClose(self):
Christian Heimes5e696852008-04-09 08:37:03 +00003612 self.cli.connect((HOST, self.port))
Thomas Wouters0e3f5912006-08-11 14:57:12 +00003613 time.sleep(1.0)
3614
Victor Stinner45df8202010-04-28 22:31:17 +00003615@unittest.skipUnless(thread, 'Threading required for this test.')
Dave Cole331708b2004-08-09 04:51:41 +00003616class BasicSocketPairTest(SocketPairTest):
3617
3618 def __init__(self, methodName='runTest'):
3619 SocketPairTest.__init__(self, methodName=methodName)
3620
Antoine Pitrou2e55fec2010-09-14 21:24:25 +00003621 def _check_defaults(self, sock):
3622 self.assertIsInstance(sock, socket.socket)
3623 if hasattr(socket, 'AF_UNIX'):
3624 self.assertEqual(sock.family, socket.AF_UNIX)
3625 else:
3626 self.assertEqual(sock.family, socket.AF_INET)
3627 self.assertEqual(sock.type, socket.SOCK_STREAM)
3628 self.assertEqual(sock.proto, 0)
3629
Antoine Pitrou9e0b8642010-09-14 18:00:02 +00003630 def _testDefaults(self):
Antoine Pitrou2e55fec2010-09-14 21:24:25 +00003631 self._check_defaults(self.cli)
Antoine Pitrou9e0b8642010-09-14 18:00:02 +00003632
3633 def testDefaults(self):
Antoine Pitrou2e55fec2010-09-14 21:24:25 +00003634 self._check_defaults(self.serv)
Antoine Pitrou9e0b8642010-09-14 18:00:02 +00003635
Dave Cole331708b2004-08-09 04:51:41 +00003636 def testRecv(self):
3637 msg = self.serv.recv(1024)
3638 self.assertEqual(msg, MSG)
3639
3640 def _testRecv(self):
3641 self.cli.send(MSG)
3642
3643 def testSend(self):
3644 self.serv.send(MSG)
3645
3646 def _testSend(self):
3647 msg = self.cli.recv(1024)
3648 self.assertEqual(msg, MSG)
3649
Victor Stinner45df8202010-04-28 22:31:17 +00003650@unittest.skipUnless(thread, 'Threading required for this test.')
Guido van Rossum24e4af82002-06-12 19:18:08 +00003651class NonBlockingTCPTests(ThreadedTCPSocketTest):
3652
3653 def __init__(self, methodName='runTest'):
3654 ThreadedTCPSocketTest.__init__(self, methodName=methodName)
3655
3656 def testSetBlocking(self):
Guido van Rossum8c943832002-08-08 01:00:28 +00003657 # Testing whether set blocking works
Serhiy Storchaka45cb7f62013-01-15 11:08:30 +02003658 self.serv.setblocking(True)
3659 self.assertIsNone(self.serv.gettimeout())
3660 self.serv.setblocking(False)
3661 self.assertEqual(self.serv.gettimeout(), 0.0)
Guido van Rossum24e4af82002-06-12 19:18:08 +00003662 start = time.time()
3663 try:
3664 self.serv.accept()
Andrew Svetlov0832af62012-12-18 23:10:48 +02003665 except OSError:
Guido van Rossum24e4af82002-06-12 19:18:08 +00003666 pass
3667 end = time.time()
Benjamin Petersonc9c0f202009-06-30 23:06:06 +00003668 self.assertTrue((end - start) < 1.0, "Error setting non-blocking mode.")
Serhiy Storchaka78980432013-01-15 01:12:17 +02003669 # Issue 15989
Serhiy Storchaka45cb7f62013-01-15 11:08:30 +02003670 if _testcapi.UINT_MAX < _testcapi.ULONG_MAX:
3671 self.serv.setblocking(_testcapi.UINT_MAX + 1)
3672 self.assertIsNone(self.serv.gettimeout())
Guido van Rossum24e4af82002-06-12 19:18:08 +00003673
3674 def _testSetBlocking(self):
Barry Warsaw6870bba2001-03-23 17:40:16 +00003675 pass
Barry Warsawcf3d4b51997-01-03 20:03:32 +00003676
Antoine Pitroub1c54962010-10-14 15:05:38 +00003677 if hasattr(socket, "SOCK_NONBLOCK"):
Charles-François Natali239bb962011-06-03 12:55:15 +02003678 @support.requires_linux_version(2, 6, 28)
Antoine Pitroub1c54962010-10-14 15:05:38 +00003679 def testInitNonBlocking(self):
3680 # reinit server socket
3681 self.serv.close()
3682 self.serv = socket.socket(socket.AF_INET, socket.SOCK_STREAM |
Antoine Pitrou1cae8b52010-10-23 17:05:13 +00003683 socket.SOCK_NONBLOCK)
Antoine Pitroub1c54962010-10-14 15:05:38 +00003684 self.port = support.bind_port(self.serv)
3685 self.serv.listen(1)
3686 # actual testing
3687 start = time.time()
3688 try:
3689 self.serv.accept()
Andrew Svetlov0832af62012-12-18 23:10:48 +02003690 except OSError:
Antoine Pitroub1c54962010-10-14 15:05:38 +00003691 pass
3692 end = time.time()
3693 self.assertTrue((end - start) < 1.0, "Error creating with non-blocking mode.")
3694
3695 def _testInitNonBlocking(self):
3696 pass
3697
Antoine Pitrou600232b2011-01-05 21:03:42 +00003698 def testInheritFlags(self):
3699 # Issue #7995: when calling accept() on a listening socket with a
3700 # timeout, the resulting socket should not be non-blocking.
3701 self.serv.settimeout(10)
3702 try:
3703 conn, addr = self.serv.accept()
3704 message = conn.recv(len(MSG))
3705 finally:
3706 conn.close()
3707 self.serv.settimeout(None)
3708
3709 def _testInheritFlags(self):
3710 time.sleep(0.1)
3711 self.cli.connect((HOST, self.port))
3712 time.sleep(0.5)
3713 self.cli.send(MSG)
3714
Guido van Rossum24e4af82002-06-12 19:18:08 +00003715 def testAccept(self):
Guido van Rossum8c943832002-08-08 01:00:28 +00003716 # Testing non-blocking accept
Guido van Rossum24e4af82002-06-12 19:18:08 +00003717 self.serv.setblocking(0)
Guido van Rossum41360a41998-03-26 19:42:58 +00003718 try:
Guido van Rossum24e4af82002-06-12 19:18:08 +00003719 conn, addr = self.serv.accept()
Andrew Svetlov0832af62012-12-18 23:10:48 +02003720 except OSError:
Guido van Rossum24e4af82002-06-12 19:18:08 +00003721 pass
3722 else:
3723 self.fail("Error trying to do non-blocking accept.")
3724 read, write, err = select.select([self.serv], [], [])
3725 if self.serv in read:
3726 conn, addr = self.serv.accept()
Brian Curtin21ebbb72010-11-02 02:07:09 +00003727 conn.close()
Guido van Rossum24e4af82002-06-12 19:18:08 +00003728 else:
3729 self.fail("Error trying to do accept after select.")
Guido van Rossum67f7a382002-06-06 21:08:16 +00003730
Guido van Rossum24e4af82002-06-12 19:18:08 +00003731 def _testAccept(self):
Guido van Rossumb6cc7d22002-07-19 12:46:46 +00003732 time.sleep(0.1)
Christian Heimes5e696852008-04-09 08:37:03 +00003733 self.cli.connect((HOST, self.port))
Guido van Rossum24e4af82002-06-12 19:18:08 +00003734
3735 def testConnect(self):
Guido van Rossum8c943832002-08-08 01:00:28 +00003736 # Testing non-blocking connect
Guido van Rossum24e4af82002-06-12 19:18:08 +00003737 conn, addr = self.serv.accept()
Brian Curtin21ebbb72010-11-02 02:07:09 +00003738 conn.close()
Guido van Rossum24e4af82002-06-12 19:18:08 +00003739
3740 def _testConnect(self):
Guido van Rossum7b8bac12002-06-13 16:07:04 +00003741 self.cli.settimeout(10)
Christian Heimes5e696852008-04-09 08:37:03 +00003742 self.cli.connect((HOST, self.port))
Guido van Rossum24e4af82002-06-12 19:18:08 +00003743
3744 def testRecv(self):
Guido van Rossum8c943832002-08-08 01:00:28 +00003745 # Testing non-blocking recv
Guido van Rossum24e4af82002-06-12 19:18:08 +00003746 conn, addr = self.serv.accept()
3747 conn.setblocking(0)
3748 try:
3749 msg = conn.recv(len(MSG))
Andrew Svetlov0832af62012-12-18 23:10:48 +02003750 except OSError:
Guido van Rossum24e4af82002-06-12 19:18:08 +00003751 pass
3752 else:
3753 self.fail("Error trying to do non-blocking recv.")
3754 read, write, err = select.select([conn], [], [])
3755 if conn in read:
3756 msg = conn.recv(len(MSG))
Brian Curtin21ebbb72010-11-02 02:07:09 +00003757 conn.close()
Guido van Rossum76489682002-06-12 20:38:30 +00003758 self.assertEqual(msg, MSG)
Guido van Rossum24e4af82002-06-12 19:18:08 +00003759 else:
3760 self.fail("Error during select call to non-blocking socket.")
3761
3762 def _testRecv(self):
Christian Heimes5e696852008-04-09 08:37:03 +00003763 self.cli.connect((HOST, self.port))
Guido van Rossumb6cc7d22002-07-19 12:46:46 +00003764 time.sleep(0.1)
Guido van Rossum24e4af82002-06-12 19:18:08 +00003765 self.cli.send(MSG)
3766
Victor Stinner45df8202010-04-28 22:31:17 +00003767@unittest.skipUnless(thread, 'Threading required for this test.')
Guido van Rossum24e4af82002-06-12 19:18:08 +00003768class FileObjectClassTestCase(SocketConnectedTest):
Jeremy Hylton5accbdb2007-08-03 20:40:09 +00003769 """Unit tests for the object returned by socket.makefile()
3770
Antoine Pitrou834bd812010-10-13 16:17:14 +00003771 self.read_file is the io object returned by makefile() on
Jeremy Hylton5accbdb2007-08-03 20:40:09 +00003772 the client connection. You can read from this file to
3773 get output from the server.
3774
Antoine Pitrou834bd812010-10-13 16:17:14 +00003775 self.write_file is the io object returned by makefile() on the
Jeremy Hylton5accbdb2007-08-03 20:40:09 +00003776 server connection. You can write to this file to send output
3777 to the client.
3778 """
Guido van Rossum24e4af82002-06-12 19:18:08 +00003779
Guido van Rossume9f66142002-08-07 15:46:19 +00003780 bufsize = -1 # Use default buffer size
Marc-André Lemburg8f36af72011-02-25 15:42:01 +00003781 encoding = 'utf-8'
Antoine Pitrou834bd812010-10-13 16:17:14 +00003782 errors = 'strict'
3783 newline = None
3784
3785 read_mode = 'rb'
3786 read_msg = MSG
3787 write_mode = 'wb'
3788 write_msg = MSG
Guido van Rossume9f66142002-08-07 15:46:19 +00003789
Guido van Rossum24e4af82002-06-12 19:18:08 +00003790 def __init__(self, methodName='runTest'):
3791 SocketConnectedTest.__init__(self, methodName=methodName)
3792
3793 def setUp(self):
Antoine Pitrou98b46702010-09-18 22:59:00 +00003794 self.evt1, self.evt2, self.serv_finished, self.cli_finished = [
3795 threading.Event() for i in range(4)]
Guido van Rossum24e4af82002-06-12 19:18:08 +00003796 SocketConnectedTest.setUp(self)
Antoine Pitrou834bd812010-10-13 16:17:14 +00003797 self.read_file = self.cli_conn.makefile(
3798 self.read_mode, self.bufsize,
3799 encoding = self.encoding,
3800 errors = self.errors,
3801 newline = self.newline)
Guido van Rossum24e4af82002-06-12 19:18:08 +00003802
3803 def tearDown(self):
Antoine Pitrou98b46702010-09-18 22:59:00 +00003804 self.serv_finished.set()
Antoine Pitrou834bd812010-10-13 16:17:14 +00003805 self.read_file.close()
3806 self.assertTrue(self.read_file.closed)
3807 self.read_file = None
Guido van Rossum24e4af82002-06-12 19:18:08 +00003808 SocketConnectedTest.tearDown(self)
3809
3810 def clientSetUp(self):
3811 SocketConnectedTest.clientSetUp(self)
Antoine Pitrou834bd812010-10-13 16:17:14 +00003812 self.write_file = self.serv_conn.makefile(
3813 self.write_mode, self.bufsize,
3814 encoding = self.encoding,
3815 errors = self.errors,
3816 newline = self.newline)
Guido van Rossum24e4af82002-06-12 19:18:08 +00003817
3818 def clientTearDown(self):
Antoine Pitrou98b46702010-09-18 22:59:00 +00003819 self.cli_finished.set()
Antoine Pitrou834bd812010-10-13 16:17:14 +00003820 self.write_file.close()
3821 self.assertTrue(self.write_file.closed)
3822 self.write_file = None
Guido van Rossum24e4af82002-06-12 19:18:08 +00003823 SocketConnectedTest.clientTearDown(self)
3824
Antoine Pitrou68e5c042011-02-25 23:07:44 +00003825 def testReadAfterTimeout(self):
3826 # Issue #7322: A file object must disallow further reads
3827 # after a timeout has occurred.
3828 self.cli_conn.settimeout(1)
3829 self.read_file.read(3)
3830 # First read raises a timeout
3831 self.assertRaises(socket.timeout, self.read_file.read, 1)
3832 # Second read is disallowed
Andrew Svetlovf7a17b42012-12-25 16:47:37 +02003833 with self.assertRaises(OSError) as ctx:
Antoine Pitrou68e5c042011-02-25 23:07:44 +00003834 self.read_file.read(1)
3835 self.assertIn("cannot read from timed out object", str(ctx.exception))
3836
3837 def _testReadAfterTimeout(self):
3838 self.write_file.write(self.write_msg[0:3])
3839 self.write_file.flush()
3840 self.serv_finished.wait()
3841
Guido van Rossum24e4af82002-06-12 19:18:08 +00003842 def testSmallRead(self):
Guido van Rossum8c943832002-08-08 01:00:28 +00003843 # Performing small file read test
Antoine Pitrou834bd812010-10-13 16:17:14 +00003844 first_seg = self.read_file.read(len(self.read_msg)-3)
3845 second_seg = self.read_file.read(3)
Guido van Rossumab659962002-06-12 21:29:43 +00003846 msg = first_seg + second_seg
Antoine Pitrou834bd812010-10-13 16:17:14 +00003847 self.assertEqual(msg, self.read_msg)
Guido van Rossum24e4af82002-06-12 19:18:08 +00003848
3849 def _testSmallRead(self):
Antoine Pitrou834bd812010-10-13 16:17:14 +00003850 self.write_file.write(self.write_msg)
3851 self.write_file.flush()
Guido van Rossum24e4af82002-06-12 19:18:08 +00003852
Guido van Rossum8c943832002-08-08 01:00:28 +00003853 def testFullRead(self):
3854 # read until EOF
Antoine Pitrou834bd812010-10-13 16:17:14 +00003855 msg = self.read_file.read()
3856 self.assertEqual(msg, self.read_msg)
Guido van Rossum8c943832002-08-08 01:00:28 +00003857
3858 def _testFullRead(self):
Antoine Pitrou834bd812010-10-13 16:17:14 +00003859 self.write_file.write(self.write_msg)
3860 self.write_file.close()
Guido van Rossum8c943832002-08-08 01:00:28 +00003861
Guido van Rossum24e4af82002-06-12 19:18:08 +00003862 def testUnbufferedRead(self):
Guido van Rossum8c943832002-08-08 01:00:28 +00003863 # Performing unbuffered file read test
Antoine Pitrou834bd812010-10-13 16:17:14 +00003864 buf = type(self.read_msg)()
Guido van Rossum24e4af82002-06-12 19:18:08 +00003865 while 1:
Antoine Pitrou834bd812010-10-13 16:17:14 +00003866 char = self.read_file.read(1)
Guido van Rossum8c943832002-08-08 01:00:28 +00003867 if not char:
Guido van Rossum24e4af82002-06-12 19:18:08 +00003868 break
Guido van Rossum8c943832002-08-08 01:00:28 +00003869 buf += char
Antoine Pitrou834bd812010-10-13 16:17:14 +00003870 self.assertEqual(buf, self.read_msg)
Guido van Rossum24e4af82002-06-12 19:18:08 +00003871
3872 def _testUnbufferedRead(self):
Antoine Pitrou834bd812010-10-13 16:17:14 +00003873 self.write_file.write(self.write_msg)
3874 self.write_file.flush()
Guido van Rossum24e4af82002-06-12 19:18:08 +00003875
3876 def testReadline(self):
Guido van Rossum8c943832002-08-08 01:00:28 +00003877 # Performing file readline test
Antoine Pitrou834bd812010-10-13 16:17:14 +00003878 line = self.read_file.readline()
3879 self.assertEqual(line, self.read_msg)
Guido van Rossum24e4af82002-06-12 19:18:08 +00003880
3881 def _testReadline(self):
Antoine Pitrou834bd812010-10-13 16:17:14 +00003882 self.write_file.write(self.write_msg)
3883 self.write_file.flush()
Guido van Rossum24e4af82002-06-12 19:18:08 +00003884
Jeremy Hylton5accbdb2007-08-03 20:40:09 +00003885 def testCloseAfterMakefile(self):
3886 # The file returned by makefile should keep the socket open.
3887 self.cli_conn.close()
3888 # read until EOF
Antoine Pitrou834bd812010-10-13 16:17:14 +00003889 msg = self.read_file.read()
3890 self.assertEqual(msg, self.read_msg)
Jeremy Hylton5accbdb2007-08-03 20:40:09 +00003891
3892 def _testCloseAfterMakefile(self):
Antoine Pitrou834bd812010-10-13 16:17:14 +00003893 self.write_file.write(self.write_msg)
3894 self.write_file.flush()
Jeremy Hylton5accbdb2007-08-03 20:40:09 +00003895
3896 def testMakefileAfterMakefileClose(self):
Antoine Pitrou834bd812010-10-13 16:17:14 +00003897 self.read_file.close()
Jeremy Hylton5accbdb2007-08-03 20:40:09 +00003898 msg = self.cli_conn.recv(len(MSG))
Antoine Pitrou834bd812010-10-13 16:17:14 +00003899 if isinstance(self.read_msg, str):
3900 msg = msg.decode()
3901 self.assertEqual(msg, self.read_msg)
Jeremy Hylton5accbdb2007-08-03 20:40:09 +00003902
3903 def _testMakefileAfterMakefileClose(self):
Antoine Pitrou834bd812010-10-13 16:17:14 +00003904 self.write_file.write(self.write_msg)
3905 self.write_file.flush()
Jeremy Hylton5accbdb2007-08-03 20:40:09 +00003906
Tim Peters116d83c2004-03-28 02:20:45 +00003907 def testClosedAttr(self):
Antoine Pitrou834bd812010-10-13 16:17:14 +00003908 self.assertTrue(not self.read_file.closed)
Tim Peters116d83c2004-03-28 02:20:45 +00003909
3910 def _testClosedAttr(self):
Antoine Pitrou834bd812010-10-13 16:17:14 +00003911 self.assertTrue(not self.write_file.closed)
Tim Peters116d83c2004-03-28 02:20:45 +00003912
Amaury Forgeot d'Arc9d24ff02008-11-20 23:15:52 +00003913 def testAttributes(self):
Antoine Pitrou834bd812010-10-13 16:17:14 +00003914 self.assertEqual(self.read_file.mode, self.read_mode)
3915 self.assertEqual(self.read_file.name, self.cli_conn.fileno())
Amaury Forgeot d'Arc9d24ff02008-11-20 23:15:52 +00003916
3917 def _testAttributes(self):
Antoine Pitrou834bd812010-10-13 16:17:14 +00003918 self.assertEqual(self.write_file.mode, self.write_mode)
3919 self.assertEqual(self.write_file.name, self.serv_conn.fileno())
Amaury Forgeot d'Arc9d24ff02008-11-20 23:15:52 +00003920
Gregory P. Smithde3369f2009-01-12 04:50:11 +00003921 def testRealClose(self):
Antoine Pitrou834bd812010-10-13 16:17:14 +00003922 self.read_file.close()
3923 self.assertRaises(ValueError, self.read_file.fileno)
Gregory P. Smithde3369f2009-01-12 04:50:11 +00003924 self.cli_conn.close()
Andrew Svetlov0832af62012-12-18 23:10:48 +02003925 self.assertRaises(OSError, self.cli_conn.getsockname)
Gregory P. Smithde3369f2009-01-12 04:50:11 +00003926
3927 def _testRealClose(self):
3928 pass
3929
3930
Gregory P. Smithaafdca82010-01-04 04:50:36 +00003931class FileObjectInterruptedTestCase(unittest.TestCase):
3932 """Test that the file object correctly handles EINTR internally."""
3933
3934 class MockSocket(object):
3935 def __init__(self, recv_funcs=()):
3936 # A generator that returns callables that we'll call for each
3937 # call to recv().
3938 self._recv_step = iter(recv_funcs)
3939
3940 def recv_into(self, buffer):
3941 data = next(self._recv_step)()
3942 assert len(buffer) >= len(data)
3943 buffer[:len(data)] = data
3944 return len(data)
3945
3946 def _decref_socketios(self):
3947 pass
3948
3949 def _textiowrap_for_test(self, buffering=-1):
3950 raw = socket.SocketIO(self, "r")
3951 if buffering < 0:
3952 buffering = io.DEFAULT_BUFFER_SIZE
3953 if buffering == 0:
3954 return raw
3955 buffer = io.BufferedReader(raw, buffering)
3956 text = io.TextIOWrapper(buffer, None, None)
3957 text.mode = "rb"
3958 return text
3959
3960 @staticmethod
3961 def _raise_eintr():
Andrew Svetlov0832af62012-12-18 23:10:48 +02003962 raise OSError(errno.EINTR, "interrupted")
Gregory P. Smithaafdca82010-01-04 04:50:36 +00003963
3964 def _textiowrap_mock_socket(self, mock, buffering=-1):
3965 raw = socket.SocketIO(mock, "r")
3966 if buffering < 0:
3967 buffering = io.DEFAULT_BUFFER_SIZE
3968 if buffering == 0:
3969 return raw
3970 buffer = io.BufferedReader(raw, buffering)
3971 text = io.TextIOWrapper(buffer, None, None)
3972 text.mode = "rb"
3973 return text
3974
3975 def _test_readline(self, size=-1, buffering=-1):
3976 mock_sock = self.MockSocket(recv_funcs=[
3977 lambda : b"This is the first line\nAnd the sec",
3978 self._raise_eintr,
3979 lambda : b"ond line is here\n",
3980 lambda : b"",
3981 lambda : b"", # XXX(gps): io library does an extra EOF read
3982 ])
3983 fo = mock_sock._textiowrap_for_test(buffering=buffering)
Ezio Melottib3aedd42010-11-20 19:04:17 +00003984 self.assertEqual(fo.readline(size), "This is the first line\n")
3985 self.assertEqual(fo.readline(size), "And the second line is here\n")
Gregory P. Smithaafdca82010-01-04 04:50:36 +00003986
3987 def _test_read(self, size=-1, buffering=-1):
3988 mock_sock = self.MockSocket(recv_funcs=[
3989 lambda : b"This is the first line\nAnd the sec",
3990 self._raise_eintr,
3991 lambda : b"ond line is here\n",
3992 lambda : b"",
3993 lambda : b"", # XXX(gps): io library does an extra EOF read
3994 ])
3995 expecting = (b"This is the first line\n"
3996 b"And the second line is here\n")
3997 fo = mock_sock._textiowrap_for_test(buffering=buffering)
3998 if buffering == 0:
3999 data = b''
4000 else:
4001 data = ''
Marc-André Lemburg8f36af72011-02-25 15:42:01 +00004002 expecting = expecting.decode('utf-8')
Gregory P. Smithaafdca82010-01-04 04:50:36 +00004003 while len(data) != len(expecting):
4004 part = fo.read(size)
4005 if not part:
4006 break
4007 data += part
Ezio Melottib3aedd42010-11-20 19:04:17 +00004008 self.assertEqual(data, expecting)
Gregory P. Smithaafdca82010-01-04 04:50:36 +00004009
4010 def test_default(self):
4011 self._test_readline()
4012 self._test_readline(size=100)
4013 self._test_read()
4014 self._test_read(size=100)
4015
4016 def test_with_1k_buffer(self):
4017 self._test_readline(buffering=1024)
4018 self._test_readline(size=100, buffering=1024)
4019 self._test_read(buffering=1024)
4020 self._test_read(size=100, buffering=1024)
4021
4022 def _test_readline_no_buffer(self, size=-1):
4023 mock_sock = self.MockSocket(recv_funcs=[
4024 lambda : b"a",
4025 lambda : b"\n",
4026 lambda : b"B",
4027 self._raise_eintr,
4028 lambda : b"b",
4029 lambda : b"",
4030 ])
4031 fo = mock_sock._textiowrap_for_test(buffering=0)
Ezio Melottib3aedd42010-11-20 19:04:17 +00004032 self.assertEqual(fo.readline(size), b"a\n")
4033 self.assertEqual(fo.readline(size), b"Bb")
Gregory P. Smithaafdca82010-01-04 04:50:36 +00004034
4035 def test_no_buffer(self):
4036 self._test_readline_no_buffer()
4037 self._test_readline_no_buffer(size=4)
4038 self._test_read(buffering=0)
4039 self._test_read(size=100, buffering=0)
4040
4041
Guido van Rossume9f66142002-08-07 15:46:19 +00004042class UnbufferedFileObjectClassTestCase(FileObjectClassTestCase):
4043
4044 """Repeat the tests from FileObjectClassTestCase with bufsize==0.
Tim Peters469cdad2002-08-08 20:19:19 +00004045
Guido van Rossume9f66142002-08-07 15:46:19 +00004046 In this case (and in this case only), it should be possible to
4047 create a file object, read a line from it, create another file
4048 object, read another line from it, without loss of data in the
Georg Brandl24420152008-05-26 16:32:26 +00004049 first file object's buffer. Note that http.client relies on this
Guido van Rossume9f66142002-08-07 15:46:19 +00004050 when reading multiple requests from the same socket."""
4051
4052 bufsize = 0 # Use unbuffered mode
4053
4054 def testUnbufferedReadline(self):
Guido van Rossum8c943832002-08-08 01:00:28 +00004055 # Read a line, create a new file object, read another line with it
Antoine Pitrou834bd812010-10-13 16:17:14 +00004056 line = self.read_file.readline() # first line
4057 self.assertEqual(line, b"A. " + self.write_msg) # first line
4058 self.read_file = self.cli_conn.makefile('rb', 0)
4059 line = self.read_file.readline() # second line
4060 self.assertEqual(line, b"B. " + self.write_msg) # second line
Guido van Rossume9f66142002-08-07 15:46:19 +00004061
4062 def _testUnbufferedReadline(self):
Antoine Pitrou834bd812010-10-13 16:17:14 +00004063 self.write_file.write(b"A. " + self.write_msg)
4064 self.write_file.write(b"B. " + self.write_msg)
4065 self.write_file.flush()
Guido van Rossume9f66142002-08-07 15:46:19 +00004066
Gregory P. Smithde3369f2009-01-12 04:50:11 +00004067 def testMakefileClose(self):
4068 # The file returned by makefile should keep the socket open...
4069 self.cli_conn.close()
4070 msg = self.cli_conn.recv(1024)
Antoine Pitrou834bd812010-10-13 16:17:14 +00004071 self.assertEqual(msg, self.read_msg)
Gregory P. Smithde3369f2009-01-12 04:50:11 +00004072 # ...until the file is itself closed
Antoine Pitrou834bd812010-10-13 16:17:14 +00004073 self.read_file.close()
Andrew Svetlov0832af62012-12-18 23:10:48 +02004074 self.assertRaises(OSError, self.cli_conn.recv, 1024)
Gregory P. Smithde3369f2009-01-12 04:50:11 +00004075
4076 def _testMakefileClose(self):
Antoine Pitrou834bd812010-10-13 16:17:14 +00004077 self.write_file.write(self.write_msg)
4078 self.write_file.flush()
Gregory P. Smithde3369f2009-01-12 04:50:11 +00004079
4080 def testMakefileCloseSocketDestroy(self):
4081 refcount_before = sys.getrefcount(self.cli_conn)
Antoine Pitrou834bd812010-10-13 16:17:14 +00004082 self.read_file.close()
Gregory P. Smithde3369f2009-01-12 04:50:11 +00004083 refcount_after = sys.getrefcount(self.cli_conn)
4084 self.assertEqual(refcount_before - 1, refcount_after)
4085
4086 def _testMakefileCloseSocketDestroy(self):
4087 pass
4088
Antoine Pitrou98b46702010-09-18 22:59:00 +00004089 # Non-blocking ops
Antoine Pitrou834bd812010-10-13 16:17:14 +00004090 # NOTE: to set `read_file` as non-blocking, we must call
Antoine Pitrou98b46702010-09-18 22:59:00 +00004091 # `cli_conn.setblocking` and vice-versa (see setUp / clientSetUp).
4092
4093 def testSmallReadNonBlocking(self):
4094 self.cli_conn.setblocking(False)
Antoine Pitrou834bd812010-10-13 16:17:14 +00004095 self.assertEqual(self.read_file.readinto(bytearray(10)), None)
4096 self.assertEqual(self.read_file.read(len(self.read_msg) - 3), None)
Antoine Pitrou98b46702010-09-18 22:59:00 +00004097 self.evt1.set()
4098 self.evt2.wait(1.0)
Antoine Pitrou834bd812010-10-13 16:17:14 +00004099 first_seg = self.read_file.read(len(self.read_msg) - 3)
Antoine Pitrou43bf0452011-04-04 01:21:37 +02004100 if first_seg is None:
4101 # Data not arrived (can happen under Windows), wait a bit
4102 time.sleep(0.5)
4103 first_seg = self.read_file.read(len(self.read_msg) - 3)
Antoine Pitrou98b46702010-09-18 22:59:00 +00004104 buf = bytearray(10)
Antoine Pitrou834bd812010-10-13 16:17:14 +00004105 n = self.read_file.readinto(buf)
Antoine Pitrou98b46702010-09-18 22:59:00 +00004106 self.assertEqual(n, 3)
4107 msg = first_seg + buf[:n]
Antoine Pitrou834bd812010-10-13 16:17:14 +00004108 self.assertEqual(msg, self.read_msg)
4109 self.assertEqual(self.read_file.readinto(bytearray(16)), None)
4110 self.assertEqual(self.read_file.read(1), None)
Antoine Pitrou98b46702010-09-18 22:59:00 +00004111
4112 def _testSmallReadNonBlocking(self):
4113 self.evt1.wait(1.0)
Antoine Pitrou834bd812010-10-13 16:17:14 +00004114 self.write_file.write(self.write_msg)
4115 self.write_file.flush()
Antoine Pitrou98b46702010-09-18 22:59:00 +00004116 self.evt2.set()
4117 # Avoid cloding the socket before the server test has finished,
4118 # otherwise system recv() will return 0 instead of EWOULDBLOCK.
4119 self.serv_finished.wait(5.0)
4120
4121 def testWriteNonBlocking(self):
4122 self.cli_finished.wait(5.0)
4123 # The client thread can't skip directly - the SkipTest exception
4124 # would appear as a failure.
4125 if self.serv_skipped:
4126 self.skipTest(self.serv_skipped)
4127
4128 def _testWriteNonBlocking(self):
4129 self.serv_skipped = None
4130 self.serv_conn.setblocking(False)
4131 # Try to saturate the socket buffer pipe with repeated large writes.
4132 BIG = b"x" * (1024 ** 2)
4133 LIMIT = 10
4134 # The first write() succeeds since a chunk of data can be buffered
Antoine Pitrou834bd812010-10-13 16:17:14 +00004135 n = self.write_file.write(BIG)
Antoine Pitrou98b46702010-09-18 22:59:00 +00004136 self.assertGreater(n, 0)
4137 for i in range(LIMIT):
Antoine Pitrou834bd812010-10-13 16:17:14 +00004138 n = self.write_file.write(BIG)
Antoine Pitrou98b46702010-09-18 22:59:00 +00004139 if n is None:
4140 # Succeeded
4141 break
4142 self.assertGreater(n, 0)
4143 else:
4144 # Let us know that this test didn't manage to establish
4145 # the expected conditions. This is not a failure in itself but,
4146 # if it happens repeatedly, the test should be fixed.
4147 self.serv_skipped = "failed to saturate the socket buffer"
4148
Gregory P. Smithde3369f2009-01-12 04:50:11 +00004149
Guido van Rossum8c943832002-08-08 01:00:28 +00004150class LineBufferedFileObjectClassTestCase(FileObjectClassTestCase):
4151
4152 bufsize = 1 # Default-buffered for reading; line-buffered for writing
4153
4154
4155class SmallBufferedFileObjectClassTestCase(FileObjectClassTestCase):
4156
4157 bufsize = 2 # Exercise the buffering code
Guido van Rossume9f66142002-08-07 15:46:19 +00004158
Thomas Woutersb2137042007-02-01 18:02:27 +00004159
Antoine Pitrou834bd812010-10-13 16:17:14 +00004160class UnicodeReadFileObjectClassTestCase(FileObjectClassTestCase):
4161 """Tests for socket.makefile() in text mode (rather than binary)"""
4162
4163 read_mode = 'r'
Marc-André Lemburg8f36af72011-02-25 15:42:01 +00004164 read_msg = MSG.decode('utf-8')
Antoine Pitrou834bd812010-10-13 16:17:14 +00004165 write_mode = 'wb'
4166 write_msg = MSG
4167 newline = ''
4168
4169
4170class UnicodeWriteFileObjectClassTestCase(FileObjectClassTestCase):
4171 """Tests for socket.makefile() in text mode (rather than binary)"""
4172
4173 read_mode = 'rb'
4174 read_msg = MSG
4175 write_mode = 'w'
Marc-André Lemburg8f36af72011-02-25 15:42:01 +00004176 write_msg = MSG.decode('utf-8')
Antoine Pitrou834bd812010-10-13 16:17:14 +00004177 newline = ''
4178
4179
4180class UnicodeReadWriteFileObjectClassTestCase(FileObjectClassTestCase):
4181 """Tests for socket.makefile() in text mode (rather than binary)"""
4182
4183 read_mode = 'r'
Marc-André Lemburg8f36af72011-02-25 15:42:01 +00004184 read_msg = MSG.decode('utf-8')
Antoine Pitrou834bd812010-10-13 16:17:14 +00004185 write_mode = 'w'
Marc-André Lemburg8f36af72011-02-25 15:42:01 +00004186 write_msg = MSG.decode('utf-8')
Antoine Pitrou834bd812010-10-13 16:17:14 +00004187 newline = ''
4188
4189
Guido van Rossumd8faa362007-04-27 19:54:29 +00004190class NetworkConnectionTest(object):
4191 """Prove network connection."""
Guido van Rossum7d0a8262007-05-21 23:13:11 +00004192
Guido van Rossumd8faa362007-04-27 19:54:29 +00004193 def clientSetUp(self):
Christian Heimes5e696852008-04-09 08:37:03 +00004194 # We're inherited below by BasicTCPTest2, which also inherits
4195 # BasicTCPTest, which defines self.port referenced below.
4196 self.cli = socket.create_connection((HOST, self.port))
Guido van Rossumd8faa362007-04-27 19:54:29 +00004197 self.serv_conn = self.cli
4198
4199class BasicTCPTest2(NetworkConnectionTest, BasicTCPTest):
4200 """Tests that NetworkConnection does not break existing TCP functionality.
4201 """
4202
4203class NetworkConnectionNoServer(unittest.TestCase):
Guido van Rossum7d0a8262007-05-21 23:13:11 +00004204
Antoine Pitrou4b92b5f2010-09-07 21:05:49 +00004205 class MockSocket(socket.socket):
4206 def connect(self, *args):
4207 raise socket.timeout('timed out')
4208
4209 @contextlib.contextmanager
4210 def mocked_socket_module(self):
4211 """Return a socket which times out on connect"""
4212 old_socket = socket.socket
4213 socket.socket = self.MockSocket
4214 try:
4215 yield
4216 finally:
4217 socket.socket = old_socket
4218
4219 def test_connect(self):
Benjamin Petersonee8712c2008-05-20 21:35:26 +00004220 port = support.find_unused_port()
Antoine Pitrou4b92b5f2010-09-07 21:05:49 +00004221 cli = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
Brian Curtin21ebbb72010-11-02 02:07:09 +00004222 self.addCleanup(cli.close)
Andrew Svetlov0832af62012-12-18 23:10:48 +02004223 with self.assertRaises(OSError) as cm:
Antoine Pitrou4b92b5f2010-09-07 21:05:49 +00004224 cli.connect((HOST, port))
4225 self.assertEqual(cm.exception.errno, errno.ECONNREFUSED)
4226
4227 def test_create_connection(self):
4228 # Issue #9792: errors raised by create_connection() should have
4229 # a proper errno attribute.
4230 port = support.find_unused_port()
Andrew Svetlov0832af62012-12-18 23:10:48 +02004231 with self.assertRaises(OSError) as cm:
Antoine Pitrou4b92b5f2010-09-07 21:05:49 +00004232 socket.create_connection((HOST, port))
Trent Nelson45bb6132012-10-17 06:15:15 -04004233
4234 # Issue #16257: create_connection() calls getaddrinfo() against
4235 # 'localhost'. This may result in an IPV6 addr being returned
4236 # as well as an IPV4 one:
4237 # >>> socket.getaddrinfo('localhost', port, 0, SOCK_STREAM)
4238 # >>> [(2, 2, 0, '', ('127.0.0.1', 41230)),
4239 # (26, 2, 0, '', ('::1', 41230, 0, 0))]
4240 #
4241 # create_connection() enumerates through all the addresses returned
4242 # and if it doesn't successfully bind to any of them, it propagates
4243 # the last exception it encountered.
4244 #
4245 # On Solaris, ENETUNREACH is returned in this circumstance instead
4246 # of ECONNREFUSED. So, if that errno exists, add it to our list of
4247 # expected errnos.
4248 expected_errnos = [ errno.ECONNREFUSED, ]
4249 if hasattr(errno, 'ENETUNREACH'):
4250 expected_errnos.append(errno.ENETUNREACH)
4251
4252 self.assertIn(cm.exception.errno, expected_errnos)
Antoine Pitrou4b92b5f2010-09-07 21:05:49 +00004253
4254 def test_create_connection_timeout(self):
4255 # Issue #9792: create_connection() should not recast timeout errors
4256 # as generic socket errors.
4257 with self.mocked_socket_module():
4258 with self.assertRaises(socket.timeout):
4259 socket.create_connection((HOST, 1234))
4260
Guido van Rossumd8faa362007-04-27 19:54:29 +00004261
Victor Stinner45df8202010-04-28 22:31:17 +00004262@unittest.skipUnless(thread, 'Threading required for this test.')
Guido van Rossumd8faa362007-04-27 19:54:29 +00004263class NetworkConnectionAttributesTest(SocketTCPTest, ThreadableTest):
4264
4265 def __init__(self, methodName='runTest'):
4266 SocketTCPTest.__init__(self, methodName=methodName)
4267 ThreadableTest.__init__(self)
4268
4269 def clientSetUp(self):
Gregory P. Smithb4066372010-01-03 03:28:29 +00004270 self.source_port = support.find_unused_port()
Guido van Rossumd8faa362007-04-27 19:54:29 +00004271
4272 def clientTearDown(self):
4273 self.cli.close()
4274 self.cli = None
4275 ThreadableTest.clientTearDown(self)
4276
4277 def _justAccept(self):
4278 conn, addr = self.serv.accept()
Brian Curtin21ebbb72010-11-02 02:07:09 +00004279 conn.close()
Guido van Rossumd8faa362007-04-27 19:54:29 +00004280
4281 testFamily = _justAccept
4282 def _testFamily(self):
Christian Heimes5e696852008-04-09 08:37:03 +00004283 self.cli = socket.create_connection((HOST, self.port), timeout=30)
Brian Curtin21ebbb72010-11-02 02:07:09 +00004284 self.addCleanup(self.cli.close)
Guido van Rossumd8faa362007-04-27 19:54:29 +00004285 self.assertEqual(self.cli.family, 2)
4286
Gregory P. Smitha81d8812010-01-03 15:09:32 +00004287 testSourceAddress = _justAccept
4288 def _testSourceAddress(self):
Gregory P. Smithb4066372010-01-03 03:28:29 +00004289 self.cli = socket.create_connection((HOST, self.port), timeout=30,
4290 source_address=('', self.source_port))
Brian Curtin21ebbb72010-11-02 02:07:09 +00004291 self.addCleanup(self.cli.close)
Gregory P. Smithb4066372010-01-03 03:28:29 +00004292 self.assertEqual(self.cli.getsockname()[1], self.source_port)
Gregory P. Smitha81d8812010-01-03 15:09:32 +00004293 # The port number being used is sufficient to show that the bind()
4294 # call happened.
Gregory P. Smithb4066372010-01-03 03:28:29 +00004295
Guido van Rossumd8faa362007-04-27 19:54:29 +00004296 testTimeoutDefault = _justAccept
4297 def _testTimeoutDefault(self):
Georg Brandlf78e02b2008-06-10 17:40:04 +00004298 # passing no explicit timeout uses socket's global default
Benjamin Petersonc9c0f202009-06-30 23:06:06 +00004299 self.assertTrue(socket.getdefaulttimeout() is None)
Georg Brandlf78e02b2008-06-10 17:40:04 +00004300 socket.setdefaulttimeout(42)
4301 try:
4302 self.cli = socket.create_connection((HOST, self.port))
Brian Curtin21ebbb72010-11-02 02:07:09 +00004303 self.addCleanup(self.cli.close)
Georg Brandlf78e02b2008-06-10 17:40:04 +00004304 finally:
4305 socket.setdefaulttimeout(None)
Ezio Melottib3aedd42010-11-20 19:04:17 +00004306 self.assertEqual(self.cli.gettimeout(), 42)
Georg Brandlf78e02b2008-06-10 17:40:04 +00004307
4308 testTimeoutNone = _justAccept
4309 def _testTimeoutNone(self):
4310 # None timeout means the same as sock.settimeout(None)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +00004311 self.assertTrue(socket.getdefaulttimeout() is None)
Georg Brandlf78e02b2008-06-10 17:40:04 +00004312 socket.setdefaulttimeout(30)
4313 try:
4314 self.cli = socket.create_connection((HOST, self.port), timeout=None)
Brian Curtin21ebbb72010-11-02 02:07:09 +00004315 self.addCleanup(self.cli.close)
Georg Brandlf78e02b2008-06-10 17:40:04 +00004316 finally:
4317 socket.setdefaulttimeout(None)
4318 self.assertEqual(self.cli.gettimeout(), None)
Guido van Rossumd8faa362007-04-27 19:54:29 +00004319
4320 testTimeoutValueNamed = _justAccept
4321 def _testTimeoutValueNamed(self):
Christian Heimes5e696852008-04-09 08:37:03 +00004322 self.cli = socket.create_connection((HOST, self.port), timeout=30)
Guido van Rossumd8faa362007-04-27 19:54:29 +00004323 self.assertEqual(self.cli.gettimeout(), 30)
4324
4325 testTimeoutValueNonamed = _justAccept
4326 def _testTimeoutValueNonamed(self):
Christian Heimes5e696852008-04-09 08:37:03 +00004327 self.cli = socket.create_connection((HOST, self.port), 30)
Brian Curtin21ebbb72010-11-02 02:07:09 +00004328 self.addCleanup(self.cli.close)
Guido van Rossumd8faa362007-04-27 19:54:29 +00004329 self.assertEqual(self.cli.gettimeout(), 30)
4330
Victor Stinner45df8202010-04-28 22:31:17 +00004331@unittest.skipUnless(thread, 'Threading required for this test.')
Guido van Rossumd8faa362007-04-27 19:54:29 +00004332class NetworkConnectionBehaviourTest(SocketTCPTest, ThreadableTest):
4333
4334 def __init__(self, methodName='runTest'):
4335 SocketTCPTest.__init__(self, methodName=methodName)
4336 ThreadableTest.__init__(self)
4337
4338 def clientSetUp(self):
4339 pass
4340
4341 def clientTearDown(self):
4342 self.cli.close()
4343 self.cli = None
4344 ThreadableTest.clientTearDown(self)
4345
4346 def testInsideTimeout(self):
4347 conn, addr = self.serv.accept()
Brian Curtin21ebbb72010-11-02 02:07:09 +00004348 self.addCleanup(conn.close)
Guido van Rossumd8faa362007-04-27 19:54:29 +00004349 time.sleep(3)
Guido van Rossum7d0a8262007-05-21 23:13:11 +00004350 conn.send(b"done!")
Guido van Rossumd8faa362007-04-27 19:54:29 +00004351 testOutsideTimeout = testInsideTimeout
4352
4353 def _testInsideTimeout(self):
Christian Heimes5e696852008-04-09 08:37:03 +00004354 self.cli = sock = socket.create_connection((HOST, self.port))
Guido van Rossumd8faa362007-04-27 19:54:29 +00004355 data = sock.recv(5)
Guido van Rossum7d0a8262007-05-21 23:13:11 +00004356 self.assertEqual(data, b"done!")
Guido van Rossumd8faa362007-04-27 19:54:29 +00004357
4358 def _testOutsideTimeout(self):
Christian Heimes5e696852008-04-09 08:37:03 +00004359 self.cli = sock = socket.create_connection((HOST, self.port), timeout=1)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +00004360 self.assertRaises(socket.timeout, lambda: sock.recv(5))
Guido van Rossumd8faa362007-04-27 19:54:29 +00004361
4362
Raymond Hettinger11a35f52003-06-29 04:40:22 +00004363class TCPTimeoutTest(SocketTCPTest):
4364
4365 def testTCPTimeout(self):
4366 def raise_timeout(*args, **kwargs):
4367 self.serv.settimeout(1.0)
4368 self.serv.accept()
Benjamin Petersonc9c0f202009-06-30 23:06:06 +00004369 self.assertRaises(socket.timeout, raise_timeout,
Raymond Hettinger11a35f52003-06-29 04:40:22 +00004370 "Error generating a timeout exception (TCP)")
4371
4372 def testTimeoutZero(self):
4373 ok = False
4374 try:
4375 self.serv.settimeout(0.0)
4376 foo = self.serv.accept()
4377 except socket.timeout:
4378 self.fail("caught timeout instead of error (TCP)")
Andrew Svetlov0832af62012-12-18 23:10:48 +02004379 except OSError:
Raymond Hettinger11a35f52003-06-29 04:40:22 +00004380 ok = True
4381 except:
4382 self.fail("caught unexpected exception (TCP)")
4383 if not ok:
4384 self.fail("accept() returned success when we did not expect it")
4385
Thomas Wouters0e3f5912006-08-11 14:57:12 +00004386 def testInterruptedTimeout(self):
4387 # XXX I don't know how to do this test on MSWindows or any other
4388 # plaform that doesn't support signal.alarm() or os.kill(), though
4389 # the bug should have existed on all platforms.
4390 if not hasattr(signal, "alarm"):
4391 return # can only test on *nix
4392 self.serv.settimeout(5.0) # must be longer than alarm
4393 class Alarm(Exception):
4394 pass
4395 def alarm_handler(signal, frame):
4396 raise Alarm
4397 old_alarm = signal.signal(signal.SIGALRM, alarm_handler)
4398 try:
4399 signal.alarm(2) # POSIX allows alarm to be up to 1 second early
4400 try:
4401 foo = self.serv.accept()
4402 except socket.timeout:
4403 self.fail("caught timeout instead of Alarm")
4404 except Alarm:
4405 pass
4406 except:
Christian Heimesbbe741d2008-03-28 10:53:29 +00004407 self.fail("caught other exception instead of Alarm:"
4408 " %s(%s):\n%s" %
4409 (sys.exc_info()[:2] + (traceback.format_exc(),)))
Thomas Wouters0e3f5912006-08-11 14:57:12 +00004410 else:
4411 self.fail("nothing caught")
Christian Heimesbbe741d2008-03-28 10:53:29 +00004412 finally:
4413 signal.alarm(0) # shut off alarm
Thomas Wouters0e3f5912006-08-11 14:57:12 +00004414 except Alarm:
4415 self.fail("got Alarm in wrong place")
4416 finally:
4417 # no alarm can be pending. Safe to restore old handler.
4418 signal.signal(signal.SIGALRM, old_alarm)
4419
Ezio Melotti7f78ddc2011-11-18 18:33:14 +02004420class UDPTimeoutTest(SocketUDPTest):
Raymond Hettinger11a35f52003-06-29 04:40:22 +00004421
4422 def testUDPTimeout(self):
4423 def raise_timeout(*args, **kwargs):
4424 self.serv.settimeout(1.0)
4425 self.serv.recv(1024)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +00004426 self.assertRaises(socket.timeout, raise_timeout,
Raymond Hettinger11a35f52003-06-29 04:40:22 +00004427 "Error generating a timeout exception (UDP)")
4428
4429 def testTimeoutZero(self):
4430 ok = False
4431 try:
4432 self.serv.settimeout(0.0)
4433 foo = self.serv.recv(1024)
4434 except socket.timeout:
4435 self.fail("caught timeout instead of error (UDP)")
Andrew Svetlov0832af62012-12-18 23:10:48 +02004436 except OSError:
Raymond Hettinger11a35f52003-06-29 04:40:22 +00004437 ok = True
4438 except:
4439 self.fail("caught unexpected exception (UDP)")
4440 if not ok:
4441 self.fail("recv() returned success when we did not expect it")
4442
4443class TestExceptions(unittest.TestCase):
4444
4445 def testExceptionTree(self):
Andrew Svetlov0832af62012-12-18 23:10:48 +02004446 self.assertTrue(issubclass(OSError, Exception))
4447 self.assertTrue(issubclass(socket.herror, OSError))
4448 self.assertTrue(issubclass(socket.gaierror, OSError))
4449 self.assertTrue(issubclass(socket.timeout, OSError))
Raymond Hettinger11a35f52003-06-29 04:40:22 +00004450
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004451class TestLinuxAbstractNamespace(unittest.TestCase):
4452
4453 UNIX_PATH_MAX = 108
4454
4455 def testLinuxAbstractNamespace(self):
Guido van Rossum32c4ac02007-08-15 03:56:40 +00004456 address = b"\x00python-test-hello\x00\xff"
Victor Stinner1d5eb342011-01-03 14:30:46 +00004457 with socket.socket(socket.AF_UNIX, socket.SOCK_STREAM) as s1:
4458 s1.bind(address)
4459 s1.listen(1)
4460 with socket.socket(socket.AF_UNIX, socket.SOCK_STREAM) as s2:
4461 s2.connect(s1.getsockname())
4462 with s1.accept()[0] as s3:
4463 self.assertEqual(s1.getsockname(), address)
4464 self.assertEqual(s2.getpeername(), address)
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004465
4466 def testMaxName(self):
Guido van Rossum32c4ac02007-08-15 03:56:40 +00004467 address = b"\x00" + b"h" * (self.UNIX_PATH_MAX - 1)
Victor Stinner1d5eb342011-01-03 14:30:46 +00004468 with socket.socket(socket.AF_UNIX, socket.SOCK_STREAM) as s:
4469 s.bind(address)
4470 self.assertEqual(s.getsockname(), address)
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004471
4472 def testNameOverflow(self):
4473 address = "\x00" + "h" * self.UNIX_PATH_MAX
Victor Stinner1d5eb342011-01-03 14:30:46 +00004474 with socket.socket(socket.AF_UNIX, socket.SOCK_STREAM) as s:
Andrew Svetlov0832af62012-12-18 23:10:48 +02004475 self.assertRaises(OSError, s.bind, address)
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004476
Antoine Pitrou6ec29e22011-12-16 14:46:36 +01004477 def testStrName(self):
4478 # Check that an abstract name can be passed as a string.
4479 s = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
4480 try:
4481 s.bind("\x00python\x00test\x00")
4482 self.assertEqual(s.getsockname(), b"\x00python\x00test\x00")
4483 finally:
4484 s.close()
4485
4486class TestUnixDomain(unittest.TestCase):
4487
4488 def setUp(self):
4489 self.sock = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
4490
4491 def tearDown(self):
4492 self.sock.close()
4493
4494 def encoded(self, path):
4495 # Return the given path encoded in the file system encoding,
4496 # or skip the test if this is not possible.
4497 try:
4498 return os.fsencode(path)
4499 except UnicodeEncodeError:
4500 self.skipTest(
4501 "Pathname {0!a} cannot be represented in file "
4502 "system encoding {1!r}".format(
4503 path, sys.getfilesystemencoding()))
4504
Antoine Pitrou16374872011-12-16 15:04:12 +01004505 def bind(self, sock, path):
4506 # Bind the socket
4507 try:
4508 sock.bind(path)
4509 except OSError as e:
4510 if str(e) == "AF_UNIX path too long":
4511 self.skipTest(
4512 "Pathname {0!a} is too long to serve as a AF_UNIX path"
4513 .format(path))
4514 else:
4515 raise
4516
Antoine Pitrou6ec29e22011-12-16 14:46:36 +01004517 def testStrAddr(self):
4518 # Test binding to and retrieving a normal string pathname.
4519 path = os.path.abspath(support.TESTFN)
Antoine Pitrou16374872011-12-16 15:04:12 +01004520 self.bind(self.sock, path)
Antoine Pitrou6ec29e22011-12-16 14:46:36 +01004521 self.addCleanup(support.unlink, path)
4522 self.assertEqual(self.sock.getsockname(), path)
4523
4524 def testBytesAddr(self):
4525 # Test binding to a bytes pathname.
4526 path = os.path.abspath(support.TESTFN)
Antoine Pitrou16374872011-12-16 15:04:12 +01004527 self.bind(self.sock, self.encoded(path))
Antoine Pitrou6ec29e22011-12-16 14:46:36 +01004528 self.addCleanup(support.unlink, path)
4529 self.assertEqual(self.sock.getsockname(), path)
4530
4531 def testSurrogateescapeBind(self):
4532 # Test binding to a valid non-ASCII pathname, with the
4533 # non-ASCII bytes supplied using surrogateescape encoding.
4534 path = os.path.abspath(support.TESTFN_UNICODE)
4535 b = self.encoded(path)
Antoine Pitrou16374872011-12-16 15:04:12 +01004536 self.bind(self.sock, b.decode("ascii", "surrogateescape"))
Antoine Pitrou6ec29e22011-12-16 14:46:36 +01004537 self.addCleanup(support.unlink, path)
4538 self.assertEqual(self.sock.getsockname(), path)
4539
4540 def testUnencodableAddr(self):
4541 # Test binding to a pathname that cannot be encoded in the
4542 # file system encoding.
4543 if support.TESTFN_UNENCODABLE is None:
4544 self.skipTest("No unencodable filename available")
4545 path = os.path.abspath(support.TESTFN_UNENCODABLE)
Antoine Pitrou16374872011-12-16 15:04:12 +01004546 self.bind(self.sock, path)
Antoine Pitrou6ec29e22011-12-16 14:46:36 +01004547 self.addCleanup(support.unlink, path)
4548 self.assertEqual(self.sock.getsockname(), path)
Raymond Hettinger11a35f52003-06-29 04:40:22 +00004549
Victor Stinner45df8202010-04-28 22:31:17 +00004550@unittest.skipUnless(thread, 'Threading required for this test.')
Thomas Wouters477c8d52006-05-27 19:21:47 +00004551class BufferIOTest(SocketConnectedTest):
4552 """
4553 Test the buffer versions of socket.recv() and socket.send().
4554 """
4555 def __init__(self, methodName='runTest'):
4556 SocketConnectedTest.__init__(self, methodName=methodName)
4557
Antoine Pitrou25480782010-03-17 22:50:28 +00004558 def testRecvIntoArray(self):
Antoine Pitrou2f89aa62008-08-02 21:02:48 +00004559 buf = bytearray(1024)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00004560 nbytes = self.cli_conn.recv_into(buf)
Thomas Wouters477c8d52006-05-27 19:21:47 +00004561 self.assertEqual(nbytes, len(MSG))
Guido van Rossum7d0a8262007-05-21 23:13:11 +00004562 msg = buf[:len(MSG)]
Thomas Wouters477c8d52006-05-27 19:21:47 +00004563 self.assertEqual(msg, MSG)
4564
Antoine Pitrou25480782010-03-17 22:50:28 +00004565 def _testRecvIntoArray(self):
Guido van Rossumb5b22702007-05-18 18:55:53 +00004566 buf = bytes(MSG)
Thomas Wouters477c8d52006-05-27 19:21:47 +00004567 self.serv_conn.send(buf)
4568
Antoine Pitrou25480782010-03-17 22:50:28 +00004569 def testRecvIntoBytearray(self):
4570 buf = bytearray(1024)
4571 nbytes = self.cli_conn.recv_into(buf)
4572 self.assertEqual(nbytes, len(MSG))
4573 msg = buf[:len(MSG)]
4574 self.assertEqual(msg, MSG)
4575
4576 _testRecvIntoBytearray = _testRecvIntoArray
4577
4578 def testRecvIntoMemoryview(self):
4579 buf = bytearray(1024)
4580 nbytes = self.cli_conn.recv_into(memoryview(buf))
4581 self.assertEqual(nbytes, len(MSG))
4582 msg = buf[:len(MSG)]
4583 self.assertEqual(msg, MSG)
4584
4585 _testRecvIntoMemoryview = _testRecvIntoArray
4586
4587 def testRecvFromIntoArray(self):
Antoine Pitrou2f89aa62008-08-02 21:02:48 +00004588 buf = bytearray(1024)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00004589 nbytes, addr = self.cli_conn.recvfrom_into(buf)
Thomas Wouters477c8d52006-05-27 19:21:47 +00004590 self.assertEqual(nbytes, len(MSG))
Guido van Rossum7d0a8262007-05-21 23:13:11 +00004591 msg = buf[:len(MSG)]
Thomas Wouters477c8d52006-05-27 19:21:47 +00004592 self.assertEqual(msg, MSG)
4593
Antoine Pitrou25480782010-03-17 22:50:28 +00004594 def _testRecvFromIntoArray(self):
Guido van Rossumb5b22702007-05-18 18:55:53 +00004595 buf = bytes(MSG)
Thomas Wouters477c8d52006-05-27 19:21:47 +00004596 self.serv_conn.send(buf)
4597
Antoine Pitrou25480782010-03-17 22:50:28 +00004598 def testRecvFromIntoBytearray(self):
4599 buf = bytearray(1024)
4600 nbytes, addr = self.cli_conn.recvfrom_into(buf)
4601 self.assertEqual(nbytes, len(MSG))
4602 msg = buf[:len(MSG)]
4603 self.assertEqual(msg, MSG)
4604
4605 _testRecvFromIntoBytearray = _testRecvFromIntoArray
4606
4607 def testRecvFromIntoMemoryview(self):
4608 buf = bytearray(1024)
4609 nbytes, addr = self.cli_conn.recvfrom_into(memoryview(buf))
4610 self.assertEqual(nbytes, len(MSG))
4611 msg = buf[:len(MSG)]
4612 self.assertEqual(msg, MSG)
4613
4614 _testRecvFromIntoMemoryview = _testRecvFromIntoArray
4615
Christian Heimes043d6f62008-01-07 17:19:16 +00004616
4617TIPC_STYPE = 2000
4618TIPC_LOWER = 200
4619TIPC_UPPER = 210
4620
4621def isTipcAvailable():
4622 """Check if the TIPC module is loaded
4623
4624 The TIPC module is not loaded automatically on Ubuntu and probably
4625 other Linux distros.
4626 """
4627 if not hasattr(socket, "AF_TIPC"):
4628 return False
4629 if not os.path.isfile("/proc/modules"):
4630 return False
4631 with open("/proc/modules") as f:
4632 for line in f:
4633 if line.startswith("tipc "):
4634 return True
Benjamin Petersonee8712c2008-05-20 21:35:26 +00004635 if support.verbose:
Christian Heimes043d6f62008-01-07 17:19:16 +00004636 print("TIPC module is not loaded, please 'sudo modprobe tipc'")
4637 return False
4638
Antoine Pitrou86aed0b2011-10-02 23:33:19 +02004639class TIPCTest(unittest.TestCase):
Christian Heimes043d6f62008-01-07 17:19:16 +00004640 def testRDM(self):
4641 srv = socket.socket(socket.AF_TIPC, socket.SOCK_RDM)
4642 cli = socket.socket(socket.AF_TIPC, socket.SOCK_RDM)
Antoine Pitrou86aed0b2011-10-02 23:33:19 +02004643 self.addCleanup(srv.close)
4644 self.addCleanup(cli.close)
Christian Heimes043d6f62008-01-07 17:19:16 +00004645
4646 srv.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
4647 srvaddr = (socket.TIPC_ADDR_NAMESEQ, TIPC_STYPE,
4648 TIPC_LOWER, TIPC_UPPER)
4649 srv.bind(srvaddr)
4650
4651 sendaddr = (socket.TIPC_ADDR_NAME, TIPC_STYPE,
4652 TIPC_LOWER + int((TIPC_UPPER - TIPC_LOWER) / 2), 0)
4653 cli.sendto(MSG, sendaddr)
4654
4655 msg, recvaddr = srv.recvfrom(1024)
4656
4657 self.assertEqual(cli.getsockname(), recvaddr)
4658 self.assertEqual(msg, MSG)
4659
4660
Antoine Pitrou86aed0b2011-10-02 23:33:19 +02004661class TIPCThreadableTest(unittest.TestCase, ThreadableTest):
Christian Heimes043d6f62008-01-07 17:19:16 +00004662 def __init__(self, methodName = 'runTest'):
4663 unittest.TestCase.__init__(self, methodName = methodName)
4664 ThreadableTest.__init__(self)
4665
4666 def setUp(self):
4667 self.srv = socket.socket(socket.AF_TIPC, socket.SOCK_STREAM)
Antoine Pitrou86aed0b2011-10-02 23:33:19 +02004668 self.addCleanup(self.srv.close)
Christian Heimes043d6f62008-01-07 17:19:16 +00004669 self.srv.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
4670 srvaddr = (socket.TIPC_ADDR_NAMESEQ, TIPC_STYPE,
4671 TIPC_LOWER, TIPC_UPPER)
4672 self.srv.bind(srvaddr)
4673 self.srv.listen(5)
4674 self.serverExplicitReady()
4675 self.conn, self.connaddr = self.srv.accept()
Antoine Pitrou86aed0b2011-10-02 23:33:19 +02004676 self.addCleanup(self.conn.close)
Christian Heimes043d6f62008-01-07 17:19:16 +00004677
4678 def clientSetUp(self):
4679 # The is a hittable race between serverExplicitReady() and the
4680 # accept() call; sleep a little while to avoid it, otherwise
4681 # we could get an exception
4682 time.sleep(0.1)
4683 self.cli = socket.socket(socket.AF_TIPC, socket.SOCK_STREAM)
Antoine Pitrou86aed0b2011-10-02 23:33:19 +02004684 self.addCleanup(self.cli.close)
Christian Heimes043d6f62008-01-07 17:19:16 +00004685 addr = (socket.TIPC_ADDR_NAME, TIPC_STYPE,
4686 TIPC_LOWER + int((TIPC_UPPER - TIPC_LOWER) / 2), 0)
4687 self.cli.connect(addr)
4688 self.cliaddr = self.cli.getsockname()
4689
4690 def testStream(self):
4691 msg = self.conn.recv(1024)
4692 self.assertEqual(msg, MSG)
4693 self.assertEqual(self.cliaddr, self.connaddr)
4694
4695 def _testStream(self):
4696 self.cli.send(MSG)
4697 self.cli.close()
4698
4699
Giampaolo Rodolàb383dbb2010-09-08 22:44:12 +00004700@unittest.skipUnless(thread, 'Threading required for this test.')
4701class ContextManagersTest(ThreadedTCPSocketTest):
4702
4703 def _testSocketClass(self):
4704 # base test
4705 with socket.socket() as sock:
4706 self.assertFalse(sock._closed)
4707 self.assertTrue(sock._closed)
4708 # close inside with block
4709 with socket.socket() as sock:
4710 sock.close()
4711 self.assertTrue(sock._closed)
4712 # exception inside with block
4713 with socket.socket() as sock:
Andrew Svetlov0832af62012-12-18 23:10:48 +02004714 self.assertRaises(OSError, sock.sendall, b'foo')
Giampaolo Rodolàb383dbb2010-09-08 22:44:12 +00004715 self.assertTrue(sock._closed)
4716
4717 def testCreateConnectionBase(self):
4718 conn, addr = self.serv.accept()
Brian Curtin21ebbb72010-11-02 02:07:09 +00004719 self.addCleanup(conn.close)
Giampaolo Rodolàb383dbb2010-09-08 22:44:12 +00004720 data = conn.recv(1024)
4721 conn.sendall(data)
4722
4723 def _testCreateConnectionBase(self):
4724 address = self.serv.getsockname()
4725 with socket.create_connection(address) as sock:
4726 self.assertFalse(sock._closed)
4727 sock.sendall(b'foo')
4728 self.assertEqual(sock.recv(1024), b'foo')
4729 self.assertTrue(sock._closed)
4730
4731 def testCreateConnectionClose(self):
4732 conn, addr = self.serv.accept()
Brian Curtin21ebbb72010-11-02 02:07:09 +00004733 self.addCleanup(conn.close)
Giampaolo Rodolàb383dbb2010-09-08 22:44:12 +00004734 data = conn.recv(1024)
4735 conn.sendall(data)
4736
4737 def _testCreateConnectionClose(self):
4738 address = self.serv.getsockname()
4739 with socket.create_connection(address) as sock:
4740 sock.close()
4741 self.assertTrue(sock._closed)
Andrew Svetlov0832af62012-12-18 23:10:48 +02004742 self.assertRaises(OSError, sock.sendall, b'foo')
Giampaolo Rodolàb383dbb2010-09-08 22:44:12 +00004743
Antoine Pitrouacd0fda2010-10-23 17:01:44 +00004744
Antoine Pitroub1c54962010-10-14 15:05:38 +00004745@unittest.skipUnless(hasattr(socket, "SOCK_CLOEXEC"),
4746 "SOCK_CLOEXEC not defined")
4747@unittest.skipUnless(fcntl, "module fcntl not available")
4748class CloexecConstantTest(unittest.TestCase):
Charles-François Natali239bb962011-06-03 12:55:15 +02004749 @support.requires_linux_version(2, 6, 28)
Antoine Pitroub1c54962010-10-14 15:05:38 +00004750 def test_SOCK_CLOEXEC(self):
Victor Stinner1d5eb342011-01-03 14:30:46 +00004751 with socket.socket(socket.AF_INET,
4752 socket.SOCK_STREAM | socket.SOCK_CLOEXEC) as s:
4753 self.assertTrue(s.type & socket.SOCK_CLOEXEC)
4754 self.assertTrue(fcntl.fcntl(s, fcntl.F_GETFD) & fcntl.FD_CLOEXEC)
Antoine Pitroub1c54962010-10-14 15:05:38 +00004755
4756
4757@unittest.skipUnless(hasattr(socket, "SOCK_NONBLOCK"),
4758 "SOCK_NONBLOCK not defined")
4759class NonblockConstantTest(unittest.TestCase):
4760 def checkNonblock(self, s, nonblock=True, timeout=0.0):
4761 if nonblock:
4762 self.assertTrue(s.type & socket.SOCK_NONBLOCK)
4763 self.assertEqual(s.gettimeout(), timeout)
4764 else:
4765 self.assertFalse(s.type & socket.SOCK_NONBLOCK)
4766 self.assertEqual(s.gettimeout(), None)
4767
Charles-François Natali239bb962011-06-03 12:55:15 +02004768 @support.requires_linux_version(2, 6, 28)
Antoine Pitroub1c54962010-10-14 15:05:38 +00004769 def test_SOCK_NONBLOCK(self):
4770 # a lot of it seems silly and redundant, but I wanted to test that
4771 # changing back and forth worked ok
Victor Stinner1d5eb342011-01-03 14:30:46 +00004772 with socket.socket(socket.AF_INET,
4773 socket.SOCK_STREAM | socket.SOCK_NONBLOCK) as s:
4774 self.checkNonblock(s)
4775 s.setblocking(1)
4776 self.checkNonblock(s, False)
4777 s.setblocking(0)
4778 self.checkNonblock(s)
4779 s.settimeout(None)
4780 self.checkNonblock(s, False)
4781 s.settimeout(2.0)
4782 self.checkNonblock(s, timeout=2.0)
4783 s.setblocking(1)
4784 self.checkNonblock(s, False)
Antoine Pitroub1c54962010-10-14 15:05:38 +00004785 # defaulttimeout
4786 t = socket.getdefaulttimeout()
4787 socket.setdefaulttimeout(0.0)
Victor Stinner1d5eb342011-01-03 14:30:46 +00004788 with socket.socket() as s:
4789 self.checkNonblock(s)
Antoine Pitroub1c54962010-10-14 15:05:38 +00004790 socket.setdefaulttimeout(None)
Victor Stinner1d5eb342011-01-03 14:30:46 +00004791 with socket.socket() as s:
4792 self.checkNonblock(s, False)
Antoine Pitroub1c54962010-10-14 15:05:38 +00004793 socket.setdefaulttimeout(2.0)
Victor Stinner1d5eb342011-01-03 14:30:46 +00004794 with socket.socket() as s:
4795 self.checkNonblock(s, timeout=2.0)
Antoine Pitroub1c54962010-10-14 15:05:38 +00004796 socket.setdefaulttimeout(None)
Victor Stinner1d5eb342011-01-03 14:30:46 +00004797 with socket.socket() as s:
4798 self.checkNonblock(s, False)
Antoine Pitroub1c54962010-10-14 15:05:38 +00004799 socket.setdefaulttimeout(t)
4800
Giampaolo Rodolàb383dbb2010-09-08 22:44:12 +00004801
Kristján Valur Jónsson10f383a2012-04-07 11:23:31 +00004802@unittest.skipUnless(os.name == "nt", "Windows specific")
4803@unittest.skipUnless(multiprocessing, "need multiprocessing")
4804class TestSocketSharing(SocketTCPTest):
4805 # This must be classmethod and not staticmethod or multiprocessing
4806 # won't be able to bootstrap it.
4807 @classmethod
4808 def remoteProcessServer(cls, q):
4809 # Recreate socket from shared data
4810 sdata = q.get()
4811 message = q.get()
4812
4813 s = socket.fromshare(sdata)
4814 s2, c = s.accept()
4815
4816 # Send the message
4817 s2.sendall(message)
4818 s2.close()
4819 s.close()
4820
4821 def testShare(self):
4822 # Transfer the listening server socket to another process
4823 # and service it from there.
4824
4825 # Create process:
4826 q = multiprocessing.Queue()
4827 p = multiprocessing.Process(target=self.remoteProcessServer, args=(q,))
4828 p.start()
4829
4830 # Get the shared socket data
4831 data = self.serv.share(p.pid)
4832
4833 # Pass the shared socket to the other process
4834 addr = self.serv.getsockname()
4835 self.serv.close()
4836 q.put(data)
4837
4838 # The data that the server will send us
4839 message = b"slapmahfro"
4840 q.put(message)
4841
4842 # Connect
4843 s = socket.create_connection(addr)
4844 # listen for the data
4845 m = []
4846 while True:
4847 data = s.recv(100)
4848 if not data:
4849 break
4850 m.append(data)
4851 s.close()
4852 received = b"".join(m)
4853 self.assertEqual(received, message)
4854 p.join()
4855
4856 def testShareLength(self):
4857 data = self.serv.share(os.getpid())
4858 self.assertRaises(ValueError, socket.fromshare, data[:-1])
4859 self.assertRaises(ValueError, socket.fromshare, data+b"foo")
4860
4861 def compareSockets(self, org, other):
4862 # socket sharing is expected to work only for blocking socket
4863 # since the internal python timout value isn't transfered.
4864 self.assertEqual(org.gettimeout(), None)
4865 self.assertEqual(org.gettimeout(), other.gettimeout())
4866
4867 self.assertEqual(org.family, other.family)
4868 self.assertEqual(org.type, other.type)
4869 # If the user specified "0" for proto, then
4870 # internally windows will have picked the correct value.
4871 # Python introspection on the socket however will still return
4872 # 0. For the shared socket, the python value is recreated
4873 # from the actual value, so it may not compare correctly.
4874 if org.proto != 0:
4875 self.assertEqual(org.proto, other.proto)
4876
4877 def testShareLocal(self):
4878 data = self.serv.share(os.getpid())
4879 s = socket.fromshare(data)
4880 try:
4881 self.compareSockets(self.serv, s)
4882 finally:
4883 s.close()
4884
4885 def testTypes(self):
4886 families = [socket.AF_INET, socket.AF_INET6]
4887 types = [socket.SOCK_STREAM, socket.SOCK_DGRAM]
4888 for f in families:
4889 for t in types:
Kristján Valur Jónsson52a11f12012-04-07 20:38:44 +00004890 try:
4891 source = socket.socket(f, t)
4892 except OSError:
4893 continue # This combination is not supported
Kristján Valur Jónsson10f383a2012-04-07 11:23:31 +00004894 try:
4895 data = source.share(os.getpid())
4896 shared = socket.fromshare(data)
4897 try:
4898 self.compareSockets(source, shared)
4899 finally:
4900 shared.close()
4901 finally:
4902 source.close()
4903
4904
Guido van Rossumb995eb72002-07-31 16:08:40 +00004905def test_main():
Thomas Wouters0e3f5912006-08-11 14:57:12 +00004906 tests = [GeneralModuleTests, BasicTCPTest, TCPCloserTest, TCPTimeoutTest,
Ronald Oussoren94f25282010-05-05 19:11:21 +00004907 TestExceptions, BufferIOTest, BasicTCPTest2, BasicUDPTest, UDPTimeoutTest ]
Walter Dörwald21d3a322003-05-01 17:45:56 +00004908
4909 tests.extend([
4910 NonBlockingTCPTests,
4911 FileObjectClassTestCase,
Gregory P. Smithaafdca82010-01-04 04:50:36 +00004912 FileObjectInterruptedTestCase,
Walter Dörwald21d3a322003-05-01 17:45:56 +00004913 UnbufferedFileObjectClassTestCase,
4914 LineBufferedFileObjectClassTestCase,
Thomas Woutersb2137042007-02-01 18:02:27 +00004915 SmallBufferedFileObjectClassTestCase,
Antoine Pitrou834bd812010-10-13 16:17:14 +00004916 UnicodeReadFileObjectClassTestCase,
4917 UnicodeWriteFileObjectClassTestCase,
4918 UnicodeReadWriteFileObjectClassTestCase,
Guido van Rossumd8faa362007-04-27 19:54:29 +00004919 NetworkConnectionNoServer,
4920 NetworkConnectionAttributesTest,
4921 NetworkConnectionBehaviourTest,
Giampaolo Rodolàb383dbb2010-09-08 22:44:12 +00004922 ContextManagersTest,
Antoine Pitroub1c54962010-10-14 15:05:38 +00004923 CloexecConstantTest,
4924 NonblockConstantTest
Walter Dörwald21d3a322003-05-01 17:45:56 +00004925 ])
Dave Cole331708b2004-08-09 04:51:41 +00004926 if hasattr(socket, "socketpair"):
4927 tests.append(BasicSocketPairTest)
Antoine Pitrou6ec29e22011-12-16 14:46:36 +01004928 if hasattr(socket, "AF_UNIX"):
4929 tests.append(TestUnixDomain)
Victor Stinnere6747472011-08-21 00:39:18 +02004930 if sys.platform == 'linux':
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004931 tests.append(TestLinuxAbstractNamespace)
Christian Heimes043d6f62008-01-07 17:19:16 +00004932 if isTipcAvailable():
4933 tests.append(TIPCTest)
Christian Heimes790c8232008-01-07 21:14:23 +00004934 tests.append(TIPCThreadableTest)
Charles-François Natali47413c12011-10-06 19:47:44 +02004935 tests.extend([BasicCANTest, CANTest])
Charles-François Natali10b8cf42011-11-10 19:21:37 +01004936 tests.extend([BasicRDSTest, RDSTest])
Nick Coghlan96fe56a2011-08-22 11:55:57 +10004937 tests.extend([
4938 CmsgMacroTests,
4939 SendmsgUDPTest,
4940 RecvmsgUDPTest,
4941 RecvmsgIntoUDPTest,
4942 SendmsgUDP6Test,
4943 RecvmsgUDP6Test,
4944 RecvmsgRFC3542AncillaryUDP6Test,
4945 RecvmsgIntoRFC3542AncillaryUDP6Test,
4946 RecvmsgIntoUDP6Test,
4947 SendmsgTCPTest,
4948 RecvmsgTCPTest,
4949 RecvmsgIntoTCPTest,
4950 SendmsgSCTPStreamTest,
4951 RecvmsgSCTPStreamTest,
4952 RecvmsgIntoSCTPStreamTest,
4953 SendmsgUnixStreamTest,
4954 RecvmsgUnixStreamTest,
4955 RecvmsgIntoUnixStreamTest,
4956 RecvmsgSCMRightsStreamTest,
4957 RecvmsgIntoSCMRightsStreamTest,
4958 # These are slow when setitimer() is not available
4959 InterruptedRecvTimeoutTest,
4960 InterruptedSendTimeoutTest,
Kristján Valur Jónsson10f383a2012-04-07 11:23:31 +00004961 TestSocketSharing,
Nick Coghlan96fe56a2011-08-22 11:55:57 +10004962 ])
Thomas Wouters0e3f5912006-08-11 14:57:12 +00004963
Benjamin Petersonee8712c2008-05-20 21:35:26 +00004964 thread_info = support.threading_setup()
4965 support.run_unittest(*tests)
4966 support.threading_cleanup(*thread_info)
Guido van Rossum24e4af82002-06-12 19:18:08 +00004967
4968if __name__ == "__main__":
Guido van Rossumb995eb72002-07-31 16:08:40 +00004969 test_main()