annotate Lib/socket.py @ 57303:4aa4c19dfaed legacy-trunk

#1757126: fix typo with the cyrillic_asian alias
author Philip Jenvey <pjenvey@underboss.org>
date Tue, 17 Nov 2009 02:42:26 +0000
parents 9e718d8f71e8
children 72e3e85bac8b
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
14007
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
1 # Wrapper module for _socket, providing some additional facilities
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
2 # implemented in Python.
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
3
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
4 """\
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
5 This module provides socket operations and some related functions.
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
6 On Unix, it supports IP (Internet Protocol) and Unix domain sockets.
15872
3685ef2a55a4 Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 14935
diff changeset
7 On other systems, it only supports IP. Functions specific for a
14935
13488539f02b Mention in the module's doc string that other functions of the socket API
Martin v. Löwis <martin@v.loewis.de>
parents: 14682
diff changeset
8 socket are available as methods of the socket object.
14007
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
9
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
10 Functions:
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
11
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
12 socket() -- create a new socket object
33063
90da70986407 Patch #1003700: Add socketpair function to socket module.
Dave Cole <djc@object-craft.com.au>
parents: 32160
diff changeset
13 socketpair() -- create a pair of new socket objects [*]
14007
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
14 fromfd() -- create a socket object from an open file descriptor [*]
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
15 gethostname() -- return the current hostname
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
16 gethostbyname() -- map a hostname to its IP number
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
17 gethostbyaddr() -- map an IP number or hostname to DNS info
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
18 getservbyname() -- map a service name and a protocol name to a port number
54512
f0f84af15957 Typo in socket.py. Thanks Pablo Torres Navarrete.
Mark Dickinson <dickinsm@gmail.com>
parents: 53047
diff changeset
19 getprotobyname() -- map a protocol name (e.g. 'tcp') to a number
14007
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
20 ntohs(), ntohl() -- convert 16, 32 bit int from network to host byte order
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
21 htons(), htonl() -- convert 16, 32 bit int from host to network byte order
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
22 inet_aton() -- convert IP addr string (123.45.67.89) to 32-bit packed format
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
23 inet_ntoa() -- convert 32-bit packed format IP to string (123.45.67.89)
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
24 ssl() -- secure socket layer support (only available if configured)
24293
640370f08e2a Add default timeout functionality. This adds setdefaulttimeout() and
Guido van Rossum <guido@python.org>
parents: 23844
diff changeset
25 socket.getdefaulttimeout() -- get the default timeout value
640370f08e2a Add default timeout functionality. This adds setdefaulttimeout() and
Guido van Rossum <guido@python.org>
parents: 23844
diff changeset
26 socket.setdefaulttimeout() -- set the default timeout value
41532
4b9fa8092e82 Whitespace normalization. Ugh, we really need to do this more often.
Neal Norwitz <nnorwitz@gmail.com>
parents: 41288
diff changeset
27 create_connection() -- connects to an address, with an optional timeout
14007
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
28
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
29 [*] not available on all platforms!
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
30
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
31 Special objects:
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
32
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
33 SocketType -- type object for socket objects
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
34 error -- exception raised for I/O errors
28818
57e61588571b Patch by Jp Calderone:
Guido van Rossum <guido@python.org>
parents: 28443
diff changeset
35 has_ipv6 -- boolean value indicating if IPv6 is supported
14007
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
36
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
37 Integer constants:
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
38
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
39 AF_INET, AF_UNIX -- socket domains (first argument to socket() call)
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
40 SOCK_STREAM, SOCK_DGRAM, SOCK_RAW -- socket types (second argument)
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
41
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
42 Many other constants may be defined; these may be used in calls to
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
43 the setsockopt() and getsockopt() methods.
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
44 """
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
45
21971
fce7278d46b0 Repair so that importing socket doesn't blow up on platforms that lack
Tim Peters <tim.peters@gmail.com>
parents: 21962
diff changeset
46 import _socket
14007
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
47 from _socket import *
53047
7e229ca0b92a remove unpleasant exec
Chris Withers <chris@simplistix.co.uk>
parents: 52936
diff changeset
48 from functools import partial
56521
9e718d8f71e8 Fix py3k warnings in the socket module
Antoine Pitrou <solipsis@pitrou.net>
parents: 55627
diff changeset
49 from types import MethodType
21971
fce7278d46b0 Repair so that importing socket doesn't blow up on platforms that lack
Tim Peters <tim.peters@gmail.com>
parents: 21962
diff changeset
50
21959
3957d155f8c5 Break SSL support out of _socket module and place it into a new
Marc-André Lemburg <mal@egenix.com>
parents: 21473
diff changeset
51 try:
24663
b265e76bab88 Extend __all__ with the exports list of the _ssl module.
Guido van Rossum <guido@python.org>
parents: 24662
diff changeset
52 import _ssl
21959
3957d155f8c5 Break SSL support out of _socket module and place it into a new
Marc-André Lemburg <mal@egenix.com>
parents: 21473
diff changeset
53 except ImportError:
43008
4771639fc049 This contains a number of things:
Bill Janssen <janssen@parc.com>
parents: 42828
diff changeset
54 # no SSL support
24661
2f411f2d791a Another refactoring. Changed 'socket' from being a factory function
Guido van Rossum <guido@python.org>
parents: 24656
diff changeset
55 pass
43008
4771639fc049 This contains a number of things:
Bill Janssen <janssen@parc.com>
parents: 42828
diff changeset
56 else:
4771639fc049 This contains a number of things:
Bill Janssen <janssen@parc.com>
parents: 42828
diff changeset
57 def ssl(sock, keyfile=None, certfile=None):
4771639fc049 This contains a number of things:
Bill Janssen <janssen@parc.com>
parents: 42828
diff changeset
58 # we do an internal import here because the ssl
4771639fc049 This contains a number of things:
Bill Janssen <janssen@parc.com>
parents: 42828
diff changeset
59 # module imports the socket module
4771639fc049 This contains a number of things:
Bill Janssen <janssen@parc.com>
parents: 42828
diff changeset
60 import ssl as _realssl
43325
193020bf52dc More work on SSL support.
Bill Janssen <janssen@parc.com>
parents: 43008
diff changeset
61 warnings.warn("socket.ssl() is deprecated. Use ssl.wrap_socket() instead.",
43008
4771639fc049 This contains a number of things:
Bill Janssen <janssen@parc.com>
parents: 42828
diff changeset
62 DeprecationWarning, stacklevel=2)
4771639fc049 This contains a number of things:
Bill Janssen <janssen@parc.com>
parents: 42828
diff changeset
63 return _realssl.sslwrap_simple(sock, keyfile, certfile)
14007
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
64
43008
4771639fc049 This contains a number of things:
Bill Janssen <janssen@parc.com>
parents: 42828
diff changeset
65 # we need to import the same constants we used to...
43325
193020bf52dc More work on SSL support.
Bill Janssen <janssen@parc.com>
parents: 43008
diff changeset
66 from _ssl import SSLError as sslerror
43008
4771639fc049 This contains a number of things:
Bill Janssen <janssen@parc.com>
parents: 42828
diff changeset
67 from _ssl import \
4771639fc049 This contains a number of things:
Bill Janssen <janssen@parc.com>
parents: 42828
diff changeset
68 RAND_add, \
4771639fc049 This contains a number of things:
Bill Janssen <janssen@parc.com>
parents: 42828
diff changeset
69 RAND_egd, \
4771639fc049 This contains a number of things:
Bill Janssen <janssen@parc.com>
parents: 42828
diff changeset
70 RAND_status, \
4771639fc049 This contains a number of things:
Bill Janssen <janssen@parc.com>
parents: 42828
diff changeset
71 SSL_ERROR_ZERO_RETURN, \
4771639fc049 This contains a number of things:
Bill Janssen <janssen@parc.com>
parents: 42828
diff changeset
72 SSL_ERROR_WANT_READ, \
4771639fc049 This contains a number of things:
Bill Janssen <janssen@parc.com>
parents: 42828
diff changeset
73 SSL_ERROR_WANT_WRITE, \
4771639fc049 This contains a number of things:
Bill Janssen <janssen@parc.com>
parents: 42828
diff changeset
74 SSL_ERROR_WANT_X509_LOOKUP, \
4771639fc049 This contains a number of things:
Bill Janssen <janssen@parc.com>
parents: 42828
diff changeset
75 SSL_ERROR_SYSCALL, \
4771639fc049 This contains a number of things:
Bill Janssen <janssen@parc.com>
parents: 42828
diff changeset
76 SSL_ERROR_SSL, \
4771639fc049 This contains a number of things:
Bill Janssen <janssen@parc.com>
parents: 42828
diff changeset
77 SSL_ERROR_WANT_CONNECT, \
4771639fc049 This contains a number of things:
Bill Janssen <janssen@parc.com>
parents: 42828
diff changeset
78 SSL_ERROR_EOF, \
4771639fc049 This contains a number of things:
Bill Janssen <janssen@parc.com>
parents: 42828
diff changeset
79 SSL_ERROR_INVALID_ERROR_CODE
4771639fc049 This contains a number of things:
Bill Janssen <janssen@parc.com>
parents: 42828
diff changeset
80
4771639fc049 This contains a number of things:
Bill Janssen <janssen@parc.com>
parents: 42828
diff changeset
81 import os, sys, warnings
14007
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
82
28890
39f4b97ce828 Extract the errno value to use from the errno module if possible.
Fred Drake <fdrake@acm.org>
parents: 28888
diff changeset
83 try:
46623
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
84 from cStringIO import StringIO
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
85 except ImportError:
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
86 from StringIO import StringIO
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
87
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
88 try:
55627
fbf8b791ce81 Fix issue1628205: Socket file objects returned by socket.socket.makefile() now
Gregory P. Smith <greg@mad-scientist.com>
parents: 54519
diff changeset
89 import errno
28890
39f4b97ce828 Extract the errno value to use from the errno module if possible.
Fred Drake <fdrake@acm.org>
parents: 28888
diff changeset
90 except ImportError:
55627
fbf8b791ce81 Fix issue1628205: Socket file objects returned by socket.socket.makefile() now
Gregory P. Smith <greg@mad-scientist.com>
parents: 54519
diff changeset
91 errno = None
fbf8b791ce81 Fix issue1628205: Socket file objects returned by socket.socket.makefile() now
Gregory P. Smith <greg@mad-scientist.com>
parents: 54519
diff changeset
92 EBADF = getattr(errno, 'EBADF', 9)
fbf8b791ce81 Fix issue1628205: Socket file objects returned by socket.socket.makefile() now
Gregory P. Smith <greg@mad-scientist.com>
parents: 54519
diff changeset
93 EINTR = getattr(errno, 'EINTR', 4)
28890
39f4b97ce828 Extract the errno value to use from the errno module if possible.
Fred Drake <fdrake@acm.org>
parents: 28888
diff changeset
94
52936
5b01fe513f92 Merged revisions 71329 via svnmerge from
Benjamin Peterson <benjamin@python.org>
parents: 48195
diff changeset
95 __all__ = ["getfqdn", "create_connection"]
16613
f68dbffff927 bunch more __all__ lists
Skip Montanaro <skip@pobox.com>
parents: 15872
diff changeset
96 __all__.extend(os._get_exports_list(_socket))
43008
4771639fc049 This contains a number of things:
Bill Janssen <janssen@parc.com>
parents: 42828
diff changeset
97
16613
f68dbffff927 bunch more __all__ lists
Skip Montanaro <skip@pobox.com>
parents: 15872
diff changeset
98
24661
2f411f2d791a Another refactoring. Changed 'socket' from being a factory function
Guido van Rossum <guido@python.org>
parents: 24656
diff changeset
99 _realsocket = socket
14007
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
100
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
101 # WSA error codes
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
102 if sys.platform.lower().startswith("win"):
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
103 errorTab = {}
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
104 errorTab[10004] = "The operation was interrupted."
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
105 errorTab[10009] = "A bad file handle was passed."
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
106 errorTab[10013] = "Permission denied."
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
107 errorTab[10014] = "A fault occurred on the network??" # WSAEFAULT
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
108 errorTab[10022] = "An invalid operation was attempted."
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
109 errorTab[10035] = "The socket operation would block"
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
110 errorTab[10036] = "A blocking operation is already in progress."
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
111 errorTab[10048] = "The network address is in use."
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
112 errorTab[10054] = "The connection has been reset."
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
113 errorTab[10058] = "The network has been shut down."
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
114 errorTab[10060] = "The operation timed out."
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
115 errorTab[10061] = "Connection refused."
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
116 errorTab[10063] = "The name is too long."
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
117 errorTab[10064] = "The host is down."
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
118 errorTab[10065] = "The host is unreachable."
17071
b45ae8c467aa add errorTab to __all__ on win*
Skip Montanaro <skip@pobox.com>
parents: 16949
diff changeset
119 __all__.append("errorTab")
24663
b265e76bab88 Extend __all__ with the exports list of the _ssl module.
Guido van Rossum <guido@python.org>
parents: 24662
diff changeset
120
14007
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
121
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
122
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
123 def getfqdn(name=''):
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
124 """Get fully qualified domain name from name.
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
125
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
126 An empty argument is interpreted as meaning the local host.
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
127
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
128 First the hostname returned by gethostbyaddr() is checked, then
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
129 possibly existing aliases. In case no FQDN is available, hostname
34733
e83309d43a82 Fix test for socket.getfqdn() to also include the name returned by
Brett Cannon <bcannon@gmail.com>
parents: 33063
diff changeset
130 from gethostname() is returned.
14007
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
131 """
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
132 name = name.strip()
14014
3d68d5bccd40 updated occurences of fqdn algorithm (closes patch #101197)
Peter Schneider-Kamp <nowonder@nowonder.de>
parents: 14007
diff changeset
133 if not name or name == '0.0.0.0':
14007
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
134 name = gethostname()
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
135 try:
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
136 hostname, aliases, ipaddrs = gethostbyaddr(name)
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
137 except error:
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
138 pass
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
139 else:
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
140 aliases.insert(0, hostname)
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
141 for name in aliases:
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
142 if '.' in name:
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
143 break
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
144 else:
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
145 name = hostname
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
146 return name
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
147
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
148
19364
b738428238f1 When the socket is closed, don't just assign 0 to self._sock.
Guido van Rossum <guido@python.org>
parents: 17245
diff changeset
149 _socketmethods = (
b738428238f1 When the socket is closed, don't just assign 0 to self._sock.
Guido van Rossum <guido@python.org>
parents: 17245
diff changeset
150 'bind', 'connect', 'connect_ex', 'fileno', 'listen',
b738428238f1 When the socket is closed, don't just assign 0 to self._sock.
Guido van Rossum <guido@python.org>
parents: 17245
diff changeset
151 'getpeername', 'getsockname', 'getsockopt', 'setsockopt',
28888
d3164a304598 Regain throughput lost with the _socketobject wrapper. The
Skip Montanaro <skip@pobox.com>
parents: 28818
diff changeset
152 'sendall', 'setblocking',
44478
73bc002252bf socket.ioctl is only available on Windows
Christian Heimes <christian@cheimes.de>
parents: 44474
diff changeset
153 'settimeout', 'gettimeout', 'shutdown')
73bc002252bf socket.ioctl is only available on Windows
Christian Heimes <christian@cheimes.de>
parents: 44474
diff changeset
154
73bc002252bf socket.ioctl is only available on Windows
Christian Heimes <christian@cheimes.de>
parents: 44474
diff changeset
155 if os.name == "nt":
73bc002252bf socket.ioctl is only available on Windows
Christian Heimes <christian@cheimes.de>
parents: 44474
diff changeset
156 _socketmethods = _socketmethods + ('ioctl',)
19364
b738428238f1 When the socket is closed, don't just assign 0 to self._sock.
Guido van Rossum <guido@python.org>
parents: 17245
diff changeset
157
29066
90342f395721 Patch #734231: Update RiscOS support. In particular, correct
Martin v. Löwis <martin@v.loewis.de>
parents: 28890
diff changeset
158 if sys.platform == "riscos":
90342f395721 Patch #734231: Update RiscOS support. In particular, correct
Martin v. Löwis <martin@v.loewis.de>
parents: 28890
diff changeset
159 _socketmethods = _socketmethods + ('sleeptaskw',)
90342f395721 Patch #734231: Update RiscOS support. In particular, correct
Martin v. Löwis <martin@v.loewis.de>
parents: 28890
diff changeset
160
38976
c547f6e91c58 Release all forwarded functions in .close. Fixes #1513223.
Martin v. Löwis <martin@v.loewis.de>
parents: 38580
diff changeset
161 # All the method names that must be delegated to either the real socket
c547f6e91c58 Release all forwarded functions in .close. Fixes #1513223.
Martin v. Löwis <martin@v.loewis.de>
parents: 38580
diff changeset
162 # object or the _closedsocket object.
c547f6e91c58 Release all forwarded functions in .close. Fixes #1513223.
Martin v. Löwis <martin@v.loewis.de>
parents: 38580
diff changeset
163 _delegate_methods = ("recv", "recvfrom", "recv_into", "recvfrom_into",
c547f6e91c58 Release all forwarded functions in .close. Fixes #1513223.
Martin v. Löwis <martin@v.loewis.de>
parents: 38580
diff changeset
164 "send", "sendto")
c547f6e91c58 Release all forwarded functions in .close. Fixes #1513223.
Martin v. Löwis <martin@v.loewis.de>
parents: 38580
diff changeset
165
24666
4fbf534211b9 The _socketobject class has no need for a __del__ method: all it did was
Guido van Rossum <guido@python.org>
parents: 24665
diff changeset
166 class _closedsocket(object):
4fbf534211b9 The _socketobject class has no need for a __del__ method: all it did was
Guido van Rossum <guido@python.org>
parents: 24665
diff changeset
167 __slots__ = []
28888
d3164a304598 Regain throughput lost with the _socketobject wrapper. The
Skip Montanaro <skip@pobox.com>
parents: 28818
diff changeset
168 def _dummy(*args):
28890
39f4b97ce828 Extract the errno value to use from the errno module if possible.
Fred Drake <fdrake@acm.org>
parents: 28888
diff changeset
169 raise error(EBADF, 'Bad file descriptor')
38976
c547f6e91c58 Release all forwarded functions in .close. Fixes #1513223.
Martin v. Löwis <martin@v.loewis.de>
parents: 38580
diff changeset
170 # All _delegate_methods must also be initialized here.
c547f6e91c58 Release all forwarded functions in .close. Fixes #1513223.
Martin v. Löwis <martin@v.loewis.de>
parents: 38580
diff changeset
171 send = recv = recv_into = sendto = recvfrom = recvfrom_into = _dummy
c547f6e91c58 Release all forwarded functions in .close. Fixes #1513223.
Martin v. Löwis <martin@v.loewis.de>
parents: 38580
diff changeset
172 __getattr__ = _dummy
24666
4fbf534211b9 The _socketobject class has no need for a __del__ method: all it did was
Guido van Rossum <guido@python.org>
parents: 24665
diff changeset
173
42385
e9916a4e06a4 Bug #978833: Close https sockets by releasing the _ssl object.
Martin v. Löwis <martin@v.loewis.de>
parents: 41532
diff changeset
174 # Wrapper around platform socket objects. This implements
e9916a4e06a4 Bug #978833: Close https sockets by releasing the _ssl object.
Martin v. Löwis <martin@v.loewis.de>
parents: 41532
diff changeset
175 # a platform-independent dup() functionality. The
e9916a4e06a4 Bug #978833: Close https sockets by releasing the _ssl object.
Martin v. Löwis <martin@v.loewis.de>
parents: 41532
diff changeset
176 # implementation currently relies on reference counting
e9916a4e06a4 Bug #978833: Close https sockets by releasing the _ssl object.
Martin v. Löwis <martin@v.loewis.de>
parents: 41532
diff changeset
177 # to close the underlying socket object.
24661
2f411f2d791a Another refactoring. Changed 'socket' from being a factory function
Guido van Rossum <guido@python.org>
parents: 24656
diff changeset
178 class _socketobject(object):
2f411f2d791a Another refactoring. Changed 'socket' from being a factory function
Guido van Rossum <guido@python.org>
parents: 24656
diff changeset
179
2f411f2d791a Another refactoring. Changed 'socket' from being a factory function
Guido van Rossum <guido@python.org>
parents: 24656
diff changeset
180 __doc__ = _realsocket.__doc__
14007
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
181
38976
c547f6e91c58 Release all forwarded functions in .close. Fixes #1513223.
Martin v. Löwis <martin@v.loewis.de>
parents: 38580
diff changeset
182 __slots__ = ["_sock", "__weakref__"] + list(_delegate_methods)
24661
2f411f2d791a Another refactoring. Changed 'socket' from being a factory function
Guido van Rossum <guido@python.org>
parents: 24656
diff changeset
183
2f411f2d791a Another refactoring. Changed 'socket' from being a factory function
Guido van Rossum <guido@python.org>
parents: 24656
diff changeset
184 def __init__(self, family=AF_INET, type=SOCK_STREAM, proto=0, _sock=None):
2f411f2d791a Another refactoring. Changed 'socket' from being a factory function
Guido van Rossum <guido@python.org>
parents: 24656
diff changeset
185 if _sock is None:
24662
a61893d7cef2 Oops, stupid tabs. Sorry again.
Guido van Rossum <guido@python.org>
parents: 24661
diff changeset
186 _sock = _realsocket(family, type, proto)
24661
2f411f2d791a Another refactoring. Changed 'socket' from being a factory function
Guido van Rossum <guido@python.org>
parents: 24656
diff changeset
187 self._sock = _sock
38976
c547f6e91c58 Release all forwarded functions in .close. Fixes #1513223.
Martin v. Löwis <martin@v.loewis.de>
parents: 38580
diff changeset
188 for method in _delegate_methods:
c547f6e91c58 Release all forwarded functions in .close. Fixes #1513223.
Martin v. Löwis <martin@v.loewis.de>
parents: 38580
diff changeset
189 setattr(self, method, getattr(_sock, method))
14007
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
190
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
191 def close(self):
24666
4fbf534211b9 The _socketobject class has no need for a __del__ method: all it did was
Guido van Rossum <guido@python.org>
parents: 24665
diff changeset
192 self._sock = _closedsocket()
38976
c547f6e91c58 Release all forwarded functions in .close. Fixes #1513223.
Martin v. Löwis <martin@v.loewis.de>
parents: 38580
diff changeset
193 dummy = self._sock._dummy
c547f6e91c58 Release all forwarded functions in .close. Fixes #1513223.
Martin v. Löwis <martin@v.loewis.de>
parents: 38580
diff changeset
194 for method in _delegate_methods:
c547f6e91c58 Release all forwarded functions in .close. Fixes #1513223.
Martin v. Löwis <martin@v.loewis.de>
parents: 38580
diff changeset
195 setattr(self, method, dummy)
24661
2f411f2d791a Another refactoring. Changed 'socket' from being a factory function
Guido van Rossum <guido@python.org>
parents: 24656
diff changeset
196 close.__doc__ = _realsocket.close.__doc__
14007
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
197
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
198 def accept(self):
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
199 sock, addr = self._sock.accept()
24661
2f411f2d791a Another refactoring. Changed 'socket' from being a factory function
Guido van Rossum <guido@python.org>
parents: 24656
diff changeset
200 return _socketobject(_sock=sock), addr
2f411f2d791a Another refactoring. Changed 'socket' from being a factory function
Guido van Rossum <guido@python.org>
parents: 24656
diff changeset
201 accept.__doc__ = _realsocket.accept.__doc__
14007
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
202
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
203 def dup(self):
24661
2f411f2d791a Another refactoring. Changed 'socket' from being a factory function
Guido van Rossum <guido@python.org>
parents: 24656
diff changeset
204 """dup() -> socket object
2f411f2d791a Another refactoring. Changed 'socket' from being a factory function
Guido van Rossum <guido@python.org>
parents: 24656
diff changeset
205
2f411f2d791a Another refactoring. Changed 'socket' from being a factory function
Guido van Rossum <guido@python.org>
parents: 24656
diff changeset
206 Return a new socket object connected to the same system resource."""
2f411f2d791a Another refactoring. Changed 'socket' from being a factory function
Guido van Rossum <guido@python.org>
parents: 24656
diff changeset
207 return _socketobject(_sock=self._sock)
14007
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
208
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
209 def makefile(self, mode='r', bufsize=-1):
24661
2f411f2d791a Another refactoring. Changed 'socket' from being a factory function
Guido van Rossum <guido@python.org>
parents: 24656
diff changeset
210 """makefile([mode[, bufsize]]) -> file object
2f411f2d791a Another refactoring. Changed 'socket' from being a factory function
Guido van Rossum <guido@python.org>
parents: 24656
diff changeset
211
2f411f2d791a Another refactoring. Changed 'socket' from being a factory function
Guido van Rossum <guido@python.org>
parents: 24656
diff changeset
212 Return a regular file object corresponding to the socket. The mode
2f411f2d791a Another refactoring. Changed 'socket' from being a factory function
Guido van Rossum <guido@python.org>
parents: 24656
diff changeset
213 and bufsize arguments are as for the built-in open() function."""
14007
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
214 return _fileobject(self._sock, mode, bufsize)
37136
71c5694e3423 Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 37123
diff changeset
215
37123
251ecec04f1b Correct API design mistake from rev. 43126: make socket attributes readonly properties.
Georg Brandl <georg@python.org>
parents: 37069
diff changeset
216 family = property(lambda self: self._sock.family, doc="the socket family")
251ecec04f1b Correct API design mistake from rev. 43126: make socket attributes readonly properties.
Georg Brandl <georg@python.org>
parents: 37069
diff changeset
217 type = property(lambda self: self._sock.type, doc="the socket type")
251ecec04f1b Correct API design mistake from rev. 43126: make socket attributes readonly properties.
Georg Brandl <georg@python.org>
parents: 37069
diff changeset
218 proto = property(lambda self: self._sock.proto, doc="the socket protocol")
37069
dd49847a60d3 RFE #567972: Socket objects' family, type and proto properties are
Georg Brandl <georg@python.org>
parents: 34733
diff changeset
219
53047
7e229ca0b92a remove unpleasant exec
Chris Withers <chris@simplistix.co.uk>
parents: 52936
diff changeset
220 def meth(name,self,*args):
7e229ca0b92a remove unpleasant exec
Chris Withers <chris@simplistix.co.uk>
parents: 52936
diff changeset
221 return getattr(self._sock,name)(*args)
7e229ca0b92a remove unpleasant exec
Chris Withers <chris@simplistix.co.uk>
parents: 52936
diff changeset
222
7e229ca0b92a remove unpleasant exec
Chris Withers <chris@simplistix.co.uk>
parents: 52936
diff changeset
223 for _m in _socketmethods:
7e229ca0b92a remove unpleasant exec
Chris Withers <chris@simplistix.co.uk>
parents: 52936
diff changeset
224 p = partial(meth,_m)
7e229ca0b92a remove unpleasant exec
Chris Withers <chris@simplistix.co.uk>
parents: 52936
diff changeset
225 p.__name__ = _m
7e229ca0b92a remove unpleasant exec
Chris Withers <chris@simplistix.co.uk>
parents: 52936
diff changeset
226 p.__doc__ = getattr(_realsocket,_m).__doc__
56521
9e718d8f71e8 Fix py3k warnings in the socket module
Antoine Pitrou <solipsis@pitrou.net>
parents: 55627
diff changeset
227 m = MethodType(p,None,_socketobject)
53047
7e229ca0b92a remove unpleasant exec
Chris Withers <chris@simplistix.co.uk>
parents: 52936
diff changeset
228 setattr(_socketobject,_m,m)
14007
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
229
28443
161b8bd8a226 The socket module now always uses the _socketobject wrapper class, even on
Skip Montanaro <skip@pobox.com>
parents: 24944
diff changeset
230 socket = SocketType = _socketobject
14007
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
231
24661
2f411f2d791a Another refactoring. Changed 'socket' from being a factory function
Guido van Rossum <guido@python.org>
parents: 24656
diff changeset
232 class _fileobject(object):
24656
e61b118e9cab Major restructuring of _fileobject. Hopefully several things now work
Guido van Rossum <guido@python.org>
parents: 24649
diff changeset
233 """Faux file object attached to a socket object."""
e61b118e9cab Major restructuring of _fileobject. Hopefully several things now work
Guido van Rossum <guido@python.org>
parents: 24649
diff changeset
234
e61b118e9cab Major restructuring of _fileobject. Hopefully several things now work
Guido van Rossum <guido@python.org>
parents: 24649
diff changeset
235 default_bufsize = 8192
24661
2f411f2d791a Another refactoring. Changed 'socket' from being a factory function
Guido van Rossum <guido@python.org>
parents: 24656
diff changeset
236 name = "<socket>"
2f411f2d791a Another refactoring. Changed 'socket' from being a factory function
Guido van Rossum <guido@python.org>
parents: 24656
diff changeset
237
2f411f2d791a Another refactoring. Changed 'socket' from being a factory function
Guido van Rossum <guido@python.org>
parents: 24656
diff changeset
238 __slots__ = ["mode", "bufsize", "softspace",
2f411f2d791a Another refactoring. Changed 'socket' from being a factory function
Guido van Rossum <guido@python.org>
parents: 24656
diff changeset
239 # "closed" is a property, see below
54519
e80e25e6a187 http://bugs.python.org/issue6117
Kristján Valur Jónsson <kristjan@ccpgames.com>
parents: 54512
diff changeset
240 "_sock", "_rbufsize", "_wbufsize", "_rbuf", "_wbuf", "_wbuf_len",
40684
2fa6e82b1f17 Patch #1627441: close sockets properly in urllib2.
Georg Brandl <georg@python.org>
parents: 39190
diff changeset
241 "_close"]
14007
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
242
40684
2fa6e82b1f17 Patch #1627441: close sockets properly in urllib2.
Georg Brandl <georg@python.org>
parents: 39190
diff changeset
243 def __init__(self, sock, mode='rb', bufsize=-1, close=False):
14007
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
244 self._sock = sock
24661
2f411f2d791a Another refactoring. Changed 'socket' from being a factory function
Guido van Rossum <guido@python.org>
parents: 24656
diff changeset
245 self.mode = mode # Not actually used in this version
24656
e61b118e9cab Major restructuring of _fileobject. Hopefully several things now work
Guido van Rossum <guido@python.org>
parents: 24649
diff changeset
246 if bufsize < 0:
e61b118e9cab Major restructuring of _fileobject. Hopefully several things now work
Guido van Rossum <guido@python.org>
parents: 24649
diff changeset
247 bufsize = self.default_bufsize
24662
a61893d7cef2 Oops, stupid tabs. Sorry again.
Guido van Rossum <guido@python.org>
parents: 24661
diff changeset
248 self.bufsize = bufsize
a61893d7cef2 Oops, stupid tabs. Sorry again.
Guido van Rossum <guido@python.org>
parents: 24661
diff changeset
249 self.softspace = False
46623
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
250 # _rbufsize is the suggested recv buffer size. It is *strictly*
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
251 # obeyed within readline() for recv calls. If it is larger than
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
252 # default_bufsize it will be used for recv calls within read().
24656
e61b118e9cab Major restructuring of _fileobject. Hopefully several things now work
Guido van Rossum <guido@python.org>
parents: 24649
diff changeset
253 if bufsize == 0:
e61b118e9cab Major restructuring of _fileobject. Hopefully several things now work
Guido van Rossum <guido@python.org>
parents: 24649
diff changeset
254 self._rbufsize = 1
e61b118e9cab Major restructuring of _fileobject. Hopefully several things now work
Guido van Rossum <guido@python.org>
parents: 24649
diff changeset
255 elif bufsize == 1:
e61b118e9cab Major restructuring of _fileobject. Hopefully several things now work
Guido van Rossum <guido@python.org>
parents: 24649
diff changeset
256 self._rbufsize = self.default_bufsize
e61b118e9cab Major restructuring of _fileobject. Hopefully several things now work
Guido van Rossum <guido@python.org>
parents: 24649
diff changeset
257 else:
e61b118e9cab Major restructuring of _fileobject. Hopefully several things now work
Guido van Rossum <guido@python.org>
parents: 24649
diff changeset
258 self._rbufsize = bufsize
14007
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
259 self._wbufsize = bufsize
46623
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
260 # We use StringIO for the read buffer to avoid holding a list
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
261 # of variously sized string objects which have been known to
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
262 # fragment the heap due to how they are malloc()ed and often
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
263 # realloc()ed down much smaller than their original allocation.
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
264 self._rbuf = StringIO()
24664
6a6d02be321c Another refactoring of read() and readline(), this time based on the
Guido van Rossum <guido@python.org>
parents: 24663
diff changeset
265 self._wbuf = [] # A list of strings
54519
e80e25e6a187 http://bugs.python.org/issue6117
Kristján Valur Jónsson <kristjan@ccpgames.com>
parents: 54512
diff changeset
266 self._wbuf_len = 0
40684
2fa6e82b1f17 Patch #1627441: close sockets properly in urllib2.
Georg Brandl <georg@python.org>
parents: 39190
diff changeset
267 self._close = close
14007
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
268
24661
2f411f2d791a Another refactoring. Changed 'socket' from being a factory function
Guido van Rossum <guido@python.org>
parents: 24656
diff changeset
269 def _getclosed(self):
31833
a910b822b12c SF bug 924242: socket._fileobject._getclosed() returns wrong value
Tim Peters <tim.peters@gmail.com>
parents: 29465
diff changeset
270 return self._sock is None
24661
2f411f2d791a Another refactoring. Changed 'socket' from being a factory function
Guido van Rossum <guido@python.org>
parents: 24656
diff changeset
271 closed = property(_getclosed, doc="True if the file is closed")
2f411f2d791a Another refactoring. Changed 'socket' from being a factory function
Guido van Rossum <guido@python.org>
parents: 24656
diff changeset
272
14007
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
273 def close(self):
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
274 try:
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
275 if self._sock:
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
276 self.flush()
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
277 finally:
40684
2fa6e82b1f17 Patch #1627441: close sockets properly in urllib2.
Georg Brandl <georg@python.org>
parents: 39190
diff changeset
278 if self._close:
2fa6e82b1f17 Patch #1627441: close sockets properly in urllib2.
Georg Brandl <georg@python.org>
parents: 39190
diff changeset
279 self._sock.close()
23715
c2733e8f6612 SF patch 555085 (timeout socket implementation) by Michael Gilfix.
Guido van Rossum <guido@python.org>
parents: 21971
diff changeset
280 self._sock = None
14007
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
281
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
282 def __del__(self):
29311
ed70779f6402 I saw errors from _fileobject.__del__ about missing self._sock. This
Guido van Rossum <guido@python.org>
parents: 29066
diff changeset
283 try:
ed70779f6402 I saw errors from _fileobject.__del__ about missing self._sock. This
Guido van Rossum <guido@python.org>
parents: 29066
diff changeset
284 self.close()
ed70779f6402 I saw errors from _fileobject.__del__ about missing self._sock. This
Guido van Rossum <guido@python.org>
parents: 29066
diff changeset
285 except:
ed70779f6402 I saw errors from _fileobject.__del__ about missing self._sock. This
Guido van Rossum <guido@python.org>
parents: 29066
diff changeset
286 # close() may fail if __init__ didn't complete
ed70779f6402 I saw errors from _fileobject.__del__ about missing self._sock. This
Guido van Rossum <guido@python.org>
parents: 29066
diff changeset
287 pass
14007
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
288
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
289 def flush(self):
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
290 if self._wbuf:
55627
fbf8b791ce81 Fix issue1628205: Socket file objects returned by socket.socket.makefile() now
Gregory P. Smith <greg@mad-scientist.com>
parents: 54519
diff changeset
291 data = "".join(self._wbuf)
24656
e61b118e9cab Major restructuring of _fileobject. Hopefully several things now work
Guido van Rossum <guido@python.org>
parents: 24649
diff changeset
292 self._wbuf = []
54519
e80e25e6a187 http://bugs.python.org/issue6117
Kristján Valur Jónsson <kristjan@ccpgames.com>
parents: 54512
diff changeset
293 self._wbuf_len = 0
55627
fbf8b791ce81 Fix issue1628205: Socket file objects returned by socket.socket.makefile() now
Gregory P. Smith <greg@mad-scientist.com>
parents: 54519
diff changeset
294 buffer_size = max(self._rbufsize, self.default_bufsize)
fbf8b791ce81 Fix issue1628205: Socket file objects returned by socket.socket.makefile() now
Gregory P. Smith <greg@mad-scientist.com>
parents: 54519
diff changeset
295 data_size = len(data)
fbf8b791ce81 Fix issue1628205: Socket file objects returned by socket.socket.makefile() now
Gregory P. Smith <greg@mad-scientist.com>
parents: 54519
diff changeset
296 write_offset = 0
56521
9e718d8f71e8 Fix py3k warnings in the socket module
Antoine Pitrou <solipsis@pitrou.net>
parents: 55627
diff changeset
297 view = memoryview(data)
55627
fbf8b791ce81 Fix issue1628205: Socket file objects returned by socket.socket.makefile() now
Gregory P. Smith <greg@mad-scientist.com>
parents: 54519
diff changeset
298 try:
fbf8b791ce81 Fix issue1628205: Socket file objects returned by socket.socket.makefile() now
Gregory P. Smith <greg@mad-scientist.com>
parents: 54519
diff changeset
299 while write_offset < data_size:
56521
9e718d8f71e8 Fix py3k warnings in the socket module
Antoine Pitrou <solipsis@pitrou.net>
parents: 55627
diff changeset
300 self._sock.sendall(view[write_offset:write_offset+buffer_size])
55627
fbf8b791ce81 Fix issue1628205: Socket file objects returned by socket.socket.makefile() now
Gregory P. Smith <greg@mad-scientist.com>
parents: 54519
diff changeset
301 write_offset += buffer_size
fbf8b791ce81 Fix issue1628205: Socket file objects returned by socket.socket.makefile() now
Gregory P. Smith <greg@mad-scientist.com>
parents: 54519
diff changeset
302 finally:
fbf8b791ce81 Fix issue1628205: Socket file objects returned by socket.socket.makefile() now
Gregory P. Smith <greg@mad-scientist.com>
parents: 54519
diff changeset
303 if write_offset < data_size:
fbf8b791ce81 Fix issue1628205: Socket file objects returned by socket.socket.makefile() now
Gregory P. Smith <greg@mad-scientist.com>
parents: 54519
diff changeset
304 remainder = data[write_offset:]
56521
9e718d8f71e8 Fix py3k warnings in the socket module
Antoine Pitrou <solipsis@pitrou.net>
parents: 55627
diff changeset
305 del view, data # explicit free
55627
fbf8b791ce81 Fix issue1628205: Socket file objects returned by socket.socket.makefile() now
Gregory P. Smith <greg@mad-scientist.com>
parents: 54519
diff changeset
306 self._wbuf.append(remainder)
fbf8b791ce81 Fix issue1628205: Socket file objects returned by socket.socket.makefile() now
Gregory P. Smith <greg@mad-scientist.com>
parents: 54519
diff changeset
307 self._wbuf_len = len(remainder)
14007
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
308
23844
bd37eb4daf3b Cleanup a little
Neal Norwitz <nnorwitz@gmail.com>
parents: 23715
diff changeset
309 def fileno(self):
14007
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
310 return self._sock.fileno()
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
311
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
312 def write(self, data):
24656
e61b118e9cab Major restructuring of _fileobject. Hopefully several things now work
Guido van Rossum <guido@python.org>
parents: 24649
diff changeset
313 data = str(data) # XXX Should really reject non-string non-buffers
e61b118e9cab Major restructuring of _fileobject. Hopefully several things now work
Guido van Rossum <guido@python.org>
parents: 24649
diff changeset
314 if not data:
e61b118e9cab Major restructuring of _fileobject. Hopefully several things now work
Guido van Rossum <guido@python.org>
parents: 24649
diff changeset
315 return
e61b118e9cab Major restructuring of _fileobject. Hopefully several things now work
Guido van Rossum <guido@python.org>
parents: 24649
diff changeset
316 self._wbuf.append(data)
54519
e80e25e6a187 http://bugs.python.org/issue6117
Kristján Valur Jónsson <kristjan@ccpgames.com>
parents: 54512
diff changeset
317 self._wbuf_len += len(data)
24656
e61b118e9cab Major restructuring of _fileobject. Hopefully several things now work
Guido van Rossum <guido@python.org>
parents: 24649
diff changeset
318 if (self._wbufsize == 0 or
e61b118e9cab Major restructuring of _fileobject. Hopefully several things now work
Guido van Rossum <guido@python.org>
parents: 24649
diff changeset
319 self._wbufsize == 1 and '\n' in data or
54519
e80e25e6a187 http://bugs.python.org/issue6117
Kristján Valur Jónsson <kristjan@ccpgames.com>
parents: 54512
diff changeset
320 self._wbuf_len >= self._wbufsize):
23715
c2733e8f6612 SF patch 555085 (timeout socket implementation) by Michael Gilfix.
Guido van Rossum <guido@python.org>
parents: 21971
diff changeset
321 self.flush()
14007
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
322
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
323 def writelines(self, list):
24656
e61b118e9cab Major restructuring of _fileobject. Hopefully several things now work
Guido van Rossum <guido@python.org>
parents: 24649
diff changeset
324 # XXX We could do better here for very long lists
e61b118e9cab Major restructuring of _fileobject. Hopefully several things now work
Guido van Rossum <guido@python.org>
parents: 24649
diff changeset
325 # XXX Should really reject non-string non-buffers
54519
e80e25e6a187 http://bugs.python.org/issue6117
Kristján Valur Jónsson <kristjan@ccpgames.com>
parents: 54512
diff changeset
326 lines = filter(None, map(str, list))
e80e25e6a187 http://bugs.python.org/issue6117
Kristján Valur Jónsson <kristjan@ccpgames.com>
parents: 54512
diff changeset
327 self._wbuf_len += sum(map(len, lines))
e80e25e6a187 http://bugs.python.org/issue6117
Kristján Valur Jónsson <kristjan@ccpgames.com>
parents: 54512
diff changeset
328 self._wbuf.extend(lines)
24656
e61b118e9cab Major restructuring of _fileobject. Hopefully several things now work
Guido van Rossum <guido@python.org>
parents: 24649
diff changeset
329 if (self._wbufsize <= 1 or
54519
e80e25e6a187 http://bugs.python.org/issue6117
Kristján Valur Jónsson <kristjan@ccpgames.com>
parents: 54512
diff changeset
330 self._wbuf_len >= self._wbufsize):
24656
e61b118e9cab Major restructuring of _fileobject. Hopefully several things now work
Guido van Rossum <guido@python.org>
parents: 24649
diff changeset
331 self.flush()
14007
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
332
23715
c2733e8f6612 SF patch 555085 (timeout socket implementation) by Michael Gilfix.
Guido van Rossum <guido@python.org>
parents: 21971
diff changeset
333 def read(self, size=-1):
46623
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
334 # Use max, disallow tiny reads in a loop as they are very inefficient.
46684
2729e977fdd9 Fix a bug introduced in r62627. see issue2760 and issue2632.
Gregory P. Smith <greg@mad-scientist.com>
parents: 46623
diff changeset
335 # We never leave read() with any leftover data from a new recv() call
2729e977fdd9 Fix a bug introduced in r62627. see issue2760 and issue2632.
Gregory P. Smith <greg@mad-scientist.com>
parents: 46623
diff changeset
336 # in our internal buffer.
46623
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
337 rbufsize = max(self._rbufsize, self.default_bufsize)
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
338 # Our use of StringIO rather than lists of string objects returned by
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
339 # recv() minimizes memory usage and fragmentation that occurs when
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
340 # rbufsize is large compared to the typical return value of recv().
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
341 buf = self._rbuf
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
342 buf.seek(0, 2) # seek end
24656
e61b118e9cab Major restructuring of _fileobject. Hopefully several things now work
Guido van Rossum <guido@python.org>
parents: 24649
diff changeset
343 if size < 0:
e61b118e9cab Major restructuring of _fileobject. Hopefully several things now work
Guido van Rossum <guido@python.org>
parents: 24649
diff changeset
344 # Read until EOF
46623
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
345 self._rbuf = StringIO() # reset _rbuf. we consume it via buf.
24664
6a6d02be321c Another refactoring of read() and readline(), this time based on the
Guido van Rossum <guido@python.org>
parents: 24663
diff changeset
346 while True:
55627
fbf8b791ce81 Fix issue1628205: Socket file objects returned by socket.socket.makefile() now
Gregory P. Smith <greg@mad-scientist.com>
parents: 54519
diff changeset
347 try:
fbf8b791ce81 Fix issue1628205: Socket file objects returned by socket.socket.makefile() now
Gregory P. Smith <greg@mad-scientist.com>
parents: 54519
diff changeset
348 data = self._sock.recv(rbufsize)
fbf8b791ce81 Fix issue1628205: Socket file objects returned by socket.socket.makefile() now
Gregory P. Smith <greg@mad-scientist.com>
parents: 54519
diff changeset
349 except error, e:
56521
9e718d8f71e8 Fix py3k warnings in the socket module
Antoine Pitrou <solipsis@pitrou.net>
parents: 55627
diff changeset
350 if e.args[0] == EINTR:
55627
fbf8b791ce81 Fix issue1628205: Socket file objects returned by socket.socket.makefile() now
Gregory P. Smith <greg@mad-scientist.com>
parents: 54519
diff changeset
351 continue
fbf8b791ce81 Fix issue1628205: Socket file objects returned by socket.socket.makefile() now
Gregory P. Smith <greg@mad-scientist.com>
parents: 54519
diff changeset
352 raise
46623
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
353 if not data:
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
354 break
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
355 buf.write(data)
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
356 return buf.getvalue()
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
357 else:
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
358 # Read until size bytes or EOF seen, whichever comes first
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
359 buf_len = buf.tell()
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
360 if buf_len >= size:
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
361 # Already have size bytes in our buffer? Extract and return.
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
362 buf.seek(0)
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
363 rv = buf.read(size)
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
364 self._rbuf = StringIO()
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
365 self._rbuf.write(buf.read())
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
366 return rv
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
367
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
368 self._rbuf = StringIO() # reset _rbuf. we consume it via buf.
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
369 while True:
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
370 left = size - buf_len
46684
2729e977fdd9 Fix a bug introduced in r62627. see issue2760 and issue2632.
Gregory P. Smith <greg@mad-scientist.com>
parents: 46623
diff changeset
371 # recv() will malloc the amount of memory given as its
2729e977fdd9 Fix a bug introduced in r62627. see issue2760 and issue2632.
Gregory P. Smith <greg@mad-scientist.com>
parents: 46623
diff changeset
372 # parameter even though it often returns much less data
2729e977fdd9 Fix a bug introduced in r62627. see issue2760 and issue2632.
Gregory P. Smith <greg@mad-scientist.com>
parents: 46623
diff changeset
373 # than that. The returned data string is short lived
2729e977fdd9 Fix a bug introduced in r62627. see issue2760 and issue2632.
Gregory P. Smith <greg@mad-scientist.com>
parents: 46623
diff changeset
374 # as we copy it into a StringIO and free it. This avoids
2729e977fdd9 Fix a bug introduced in r62627. see issue2760 and issue2632.
Gregory P. Smith <greg@mad-scientist.com>
parents: 46623
diff changeset
375 # fragmentation issues on many platforms.
55627
fbf8b791ce81 Fix issue1628205: Socket file objects returned by socket.socket.makefile() now
Gregory P. Smith <greg@mad-scientist.com>
parents: 54519
diff changeset
376 try:
fbf8b791ce81 Fix issue1628205: Socket file objects returned by socket.socket.makefile() now
Gregory P. Smith <greg@mad-scientist.com>
parents: 54519
diff changeset
377 data = self._sock.recv(left)
fbf8b791ce81 Fix issue1628205: Socket file objects returned by socket.socket.makefile() now
Gregory P. Smith <greg@mad-scientist.com>
parents: 54519
diff changeset
378 except error, e:
56521
9e718d8f71e8 Fix py3k warnings in the socket module
Antoine Pitrou <solipsis@pitrou.net>
parents: 55627
diff changeset
379 if e.args[0] == EINTR:
55627
fbf8b791ce81 Fix issue1628205: Socket file objects returned by socket.socket.makefile() now
Gregory P. Smith <greg@mad-scientist.com>
parents: 54519
diff changeset
380 continue
fbf8b791ce81 Fix issue1628205: Socket file objects returned by socket.socket.makefile() now
Gregory P. Smith <greg@mad-scientist.com>
parents: 54519
diff changeset
381 raise
24656
e61b118e9cab Major restructuring of _fileobject. Hopefully several things now work
Guido van Rossum <guido@python.org>
parents: 24649
diff changeset
382 if not data:
e61b118e9cab Major restructuring of _fileobject. Hopefully several things now work
Guido van Rossum <guido@python.org>
parents: 24649
diff changeset
383 break
46623
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
384 n = len(data)
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
385 if n == size and not buf_len:
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
386 # Shortcut. Avoid buffer data copies when:
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
387 # - We have no data in our buffer.
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
388 # AND
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
389 # - Our call to recv returned exactly the
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
390 # number of bytes we were asked to read.
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
391 return data
46684
2729e977fdd9 Fix a bug introduced in r62627. see issue2760 and issue2632.
Gregory P. Smith <greg@mad-scientist.com>
parents: 46623
diff changeset
392 if n == left:
2729e977fdd9 Fix a bug introduced in r62627. see issue2760 and issue2632.
Gregory P. Smith <greg@mad-scientist.com>
parents: 46623
diff changeset
393 buf.write(data)
46623
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
394 del data # explicit free
24656
e61b118e9cab Major restructuring of _fileobject. Hopefully several things now work
Guido van Rossum <guido@python.org>
parents: 24649
diff changeset
395 break
46684
2729e977fdd9 Fix a bug introduced in r62627. see issue2760 and issue2632.
Gregory P. Smith <greg@mad-scientist.com>
parents: 46623
diff changeset
396 assert n <= left, "recv(%d) returned %d bytes" % (left, n)
46623
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
397 buf.write(data)
24664
6a6d02be321c Another refactoring of read() and readline(), this time based on the
Guido van Rossum <guido@python.org>
parents: 24663
diff changeset
398 buf_len += n
46623
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
399 del data # explicit free
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
400 #assert buf_len == buf.tell()
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
401 return buf.getvalue()
14007
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
402
23715
c2733e8f6612 SF patch 555085 (timeout socket implementation) by Michael Gilfix.
Guido van Rossum <guido@python.org>
parents: 21971
diff changeset
403 def readline(self, size=-1):
46623
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
404 buf = self._rbuf
46684
2729e977fdd9 Fix a bug introduced in r62627. see issue2760 and issue2632.
Gregory P. Smith <greg@mad-scientist.com>
parents: 46623
diff changeset
405 buf.seek(0, 2) # seek end
2729e977fdd9 Fix a bug introduced in r62627. see issue2760 and issue2632.
Gregory P. Smith <greg@mad-scientist.com>
parents: 46623
diff changeset
406 if buf.tell() > 0:
2729e977fdd9 Fix a bug introduced in r62627. see issue2760 and issue2632.
Gregory P. Smith <greg@mad-scientist.com>
parents: 46623
diff changeset
407 # check if we already have it in our buffer
46623
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
408 buf.seek(0)
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
409 bline = buf.readline(size)
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
410 if bline.endswith('\n') or len(bline) == size:
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
411 self._rbuf = StringIO()
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
412 self._rbuf.write(buf.read())
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
413 return bline
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
414 del bline
24664
6a6d02be321c Another refactoring of read() and readline(), this time based on the
Guido van Rossum <guido@python.org>
parents: 24663
diff changeset
415 if size < 0:
6a6d02be321c Another refactoring of read() and readline(), this time based on the
Guido van Rossum <guido@python.org>
parents: 24663
diff changeset
416 # Read until \n or EOF, whichever comes first
24665
d25cf94d20b1 OK, one more hack: speed up the case of readline() in unbuffered mode.
Guido van Rossum <guido@python.org>
parents: 24664
diff changeset
417 if self._rbufsize <= 1:
d25cf94d20b1 OK, one more hack: speed up the case of readline() in unbuffered mode.
Guido van Rossum <guido@python.org>
parents: 24664
diff changeset
418 # Speed up unbuffered case
46684
2729e977fdd9 Fix a bug introduced in r62627. see issue2760 and issue2632.
Gregory P. Smith <greg@mad-scientist.com>
parents: 46623
diff changeset
419 buf.seek(0)
2729e977fdd9 Fix a bug introduced in r62627. see issue2760 and issue2632.
Gregory P. Smith <greg@mad-scientist.com>
parents: 46623
diff changeset
420 buffers = [buf.read()]
2729e977fdd9 Fix a bug introduced in r62627. see issue2760 and issue2632.
Gregory P. Smith <greg@mad-scientist.com>
parents: 46623
diff changeset
421 self._rbuf = StringIO() # reset _rbuf. we consume it via buf.
46623
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
422 data = None
24665
d25cf94d20b1 OK, one more hack: speed up the case of readline() in unbuffered mode.
Guido van Rossum <guido@python.org>
parents: 24664
diff changeset
423 recv = self._sock.recv
55627
fbf8b791ce81 Fix issue1628205: Socket file objects returned by socket.socket.makefile() now
Gregory P. Smith <greg@mad-scientist.com>
parents: 54519
diff changeset
424 while True:
fbf8b791ce81 Fix issue1628205: Socket file objects returned by socket.socket.makefile() now
Gregory P. Smith <greg@mad-scientist.com>
parents: 54519
diff changeset
425 try:
fbf8b791ce81 Fix issue1628205: Socket file objects returned by socket.socket.makefile() now
Gregory P. Smith <greg@mad-scientist.com>
parents: 54519
diff changeset
426 while data != "\n":
fbf8b791ce81 Fix issue1628205: Socket file objects returned by socket.socket.makefile() now
Gregory P. Smith <greg@mad-scientist.com>
parents: 54519
diff changeset
427 data = recv(1)
fbf8b791ce81 Fix issue1628205: Socket file objects returned by socket.socket.makefile() now
Gregory P. Smith <greg@mad-scientist.com>
parents: 54519
diff changeset
428 if not data:
fbf8b791ce81 Fix issue1628205: Socket file objects returned by socket.socket.makefile() now
Gregory P. Smith <greg@mad-scientist.com>
parents: 54519
diff changeset
429 break
fbf8b791ce81 Fix issue1628205: Socket file objects returned by socket.socket.makefile() now
Gregory P. Smith <greg@mad-scientist.com>
parents: 54519
diff changeset
430 buffers.append(data)
fbf8b791ce81 Fix issue1628205: Socket file objects returned by socket.socket.makefile() now
Gregory P. Smith <greg@mad-scientist.com>
parents: 54519
diff changeset
431 except error, e:
fbf8b791ce81 Fix issue1628205: Socket file objects returned by socket.socket.makefile() now
Gregory P. Smith <greg@mad-scientist.com>
parents: 54519
diff changeset
432 # The try..except to catch EINTR was moved outside the
fbf8b791ce81 Fix issue1628205: Socket file objects returned by socket.socket.makefile() now
Gregory P. Smith <greg@mad-scientist.com>
parents: 54519
diff changeset
433 # recv loop to avoid the per byte overhead.
56521
9e718d8f71e8 Fix py3k warnings in the socket module
Antoine Pitrou <solipsis@pitrou.net>
parents: 55627
diff changeset
434 if e.args[0] == EINTR:
55627
fbf8b791ce81 Fix issue1628205: Socket file objects returned by socket.socket.makefile() now
Gregory P. Smith <greg@mad-scientist.com>
parents: 54519
diff changeset
435 continue
fbf8b791ce81 Fix issue1628205: Socket file objects returned by socket.socket.makefile() now
Gregory P. Smith <greg@mad-scientist.com>
parents: 54519
diff changeset
436 raise
fbf8b791ce81 Fix issue1628205: Socket file objects returned by socket.socket.makefile() now
Gregory P. Smith <greg@mad-scientist.com>
parents: 54519
diff changeset
437 break
24665
d25cf94d20b1 OK, one more hack: speed up the case of readline() in unbuffered mode.
Guido van Rossum <guido@python.org>
parents: 24664
diff changeset
438 return "".join(buffers)
46623
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
439
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
440 buf.seek(0, 2) # seek end
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
441 self._rbuf = StringIO() # reset _rbuf. we consume it via buf.
24664
6a6d02be321c Another refactoring of read() and readline(), this time based on the
Guido van Rossum <guido@python.org>
parents: 24663
diff changeset
442 while True:
55627
fbf8b791ce81 Fix issue1628205: Socket file objects returned by socket.socket.makefile() now
Gregory P. Smith <greg@mad-scientist.com>
parents: 54519
diff changeset
443 try:
fbf8b791ce81 Fix issue1628205: Socket file objects returned by socket.socket.makefile() now
Gregory P. Smith <greg@mad-scientist.com>
parents: 54519
diff changeset
444 data = self._sock.recv(self._rbufsize)
fbf8b791ce81 Fix issue1628205: Socket file objects returned by socket.socket.makefile() now
Gregory P. Smith <greg@mad-scientist.com>
parents: 54519
diff changeset
445 except error, e:
56521
9e718d8f71e8 Fix py3k warnings in the socket module
Antoine Pitrou <solipsis@pitrou.net>
parents: 55627
diff changeset
446 if e.args[0] == EINTR:
55627
fbf8b791ce81 Fix issue1628205: Socket file objects returned by socket.socket.makefile() now
Gregory P. Smith <greg@mad-scientist.com>
parents: 54519
diff changeset
447 continue
fbf8b791ce81 Fix issue1628205: Socket file objects returned by socket.socket.makefile() now
Gregory P. Smith <greg@mad-scientist.com>
parents: 54519
diff changeset
448 raise
24664
6a6d02be321c Another refactoring of read() and readline(), this time based on the
Guido van Rossum <guido@python.org>
parents: 24663
diff changeset
449 if not data:
6a6d02be321c Another refactoring of read() and readline(), this time based on the
Guido van Rossum <guido@python.org>
parents: 24663
diff changeset
450 break
6a6d02be321c Another refactoring of read() and readline(), this time based on the
Guido van Rossum <guido@python.org>
parents: 24663
diff changeset
451 nl = data.find('\n')
6a6d02be321c Another refactoring of read() and readline(), this time based on the
Guido van Rossum <guido@python.org>
parents: 24663
diff changeset
452 if nl >= 0:
6a6d02be321c Another refactoring of read() and readline(), this time based on the
Guido van Rossum <guido@python.org>
parents: 24663
diff changeset
453 nl += 1
48195
65901ffffc6a Remove buffer() usage in the socket module by just slicing directly on the
Brett Cannon <bcannon@gmail.com>
parents: 47372
diff changeset
454 buf.write(data[:nl])
65901ffffc6a Remove buffer() usage in the socket module by just slicing directly on the
Brett Cannon <bcannon@gmail.com>
parents: 47372
diff changeset
455 self._rbuf.write(data[nl:])
46623
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
456 del data
24664
6a6d02be321c Another refactoring of read() and readline(), this time based on the
Guido van Rossum <guido@python.org>
parents: 24663
diff changeset
457 break
46623
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
458 buf.write(data)
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
459 return buf.getvalue()
23715
c2733e8f6612 SF patch 555085 (timeout socket implementation) by Michael Gilfix.
Guido van Rossum <guido@python.org>
parents: 21971
diff changeset
460 else:
24664
6a6d02be321c Another refactoring of read() and readline(), this time based on the
Guido van Rossum <guido@python.org>
parents: 24663
diff changeset
461 # Read until size bytes or \n or EOF seen, whichever comes first
46684
2729e977fdd9 Fix a bug introduced in r62627. see issue2760 and issue2632.
Gregory P. Smith <greg@mad-scientist.com>
parents: 46623
diff changeset
462 buf.seek(0, 2) # seek end
46623
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
463 buf_len = buf.tell()
24664
6a6d02be321c Another refactoring of read() and readline(), this time based on the
Guido van Rossum <guido@python.org>
parents: 24663
diff changeset
464 if buf_len >= size:
46623
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
465 buf.seek(0)
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
466 rv = buf.read(size)
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
467 self._rbuf = StringIO()
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
468 self._rbuf.write(buf.read())
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
469 return rv
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
470 self._rbuf = StringIO() # reset _rbuf. we consume it via buf.
24664
6a6d02be321c Another refactoring of read() and readline(), this time based on the
Guido van Rossum <guido@python.org>
parents: 24663
diff changeset
471 while True:
55627
fbf8b791ce81 Fix issue1628205: Socket file objects returned by socket.socket.makefile() now
Gregory P. Smith <greg@mad-scientist.com>
parents: 54519
diff changeset
472 try:
fbf8b791ce81 Fix issue1628205: Socket file objects returned by socket.socket.makefile() now
Gregory P. Smith <greg@mad-scientist.com>
parents: 54519
diff changeset
473 data = self._sock.recv(self._rbufsize)
fbf8b791ce81 Fix issue1628205: Socket file objects returned by socket.socket.makefile() now
Gregory P. Smith <greg@mad-scientist.com>
parents: 54519
diff changeset
474 except error, e:
56521
9e718d8f71e8 Fix py3k warnings in the socket module
Antoine Pitrou <solipsis@pitrou.net>
parents: 55627
diff changeset
475 if e.args[0] == EINTR:
55627
fbf8b791ce81 Fix issue1628205: Socket file objects returned by socket.socket.makefile() now
Gregory P. Smith <greg@mad-scientist.com>
parents: 54519
diff changeset
476 continue
fbf8b791ce81 Fix issue1628205: Socket file objects returned by socket.socket.makefile() now
Gregory P. Smith <greg@mad-scientist.com>
parents: 54519
diff changeset
477 raise
24664
6a6d02be321c Another refactoring of read() and readline(), this time based on the
Guido van Rossum <guido@python.org>
parents: 24663
diff changeset
478 if not data:
6a6d02be321c Another refactoring of read() and readline(), this time based on the
Guido van Rossum <guido@python.org>
parents: 24663
diff changeset
479 break
6a6d02be321c Another refactoring of read() and readline(), this time based on the
Guido van Rossum <guido@python.org>
parents: 24663
diff changeset
480 left = size - buf_len
46623
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
481 # did we just receive a newline?
24664
6a6d02be321c Another refactoring of read() and readline(), this time based on the
Guido van Rossum <guido@python.org>
parents: 24663
diff changeset
482 nl = data.find('\n', 0, left)
6a6d02be321c Another refactoring of read() and readline(), this time based on the
Guido van Rossum <guido@python.org>
parents: 24663
diff changeset
483 if nl >= 0:
6a6d02be321c Another refactoring of read() and readline(), this time based on the
Guido van Rossum <guido@python.org>
parents: 24663
diff changeset
484 nl += 1
46623
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
485 # save the excess data to _rbuf
48195
65901ffffc6a Remove buffer() usage in the socket module by just slicing directly on the
Brett Cannon <bcannon@gmail.com>
parents: 47372
diff changeset
486 self._rbuf.write(data[nl:])
46623
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
487 if buf_len:
48195
65901ffffc6a Remove buffer() usage in the socket module by just slicing directly on the
Brett Cannon <bcannon@gmail.com>
parents: 47372
diff changeset
488 buf.write(data[:nl])
46623
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
489 break
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
490 else:
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
491 # Shortcut. Avoid data copy through buf when returning
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
492 # a substring of our first recv().
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
493 return data[:nl]
24664
6a6d02be321c Another refactoring of read() and readline(), this time based on the
Guido van Rossum <guido@python.org>
parents: 24663
diff changeset
494 n = len(data)
46623
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
495 if n == size and not buf_len:
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
496 # Shortcut. Avoid data copy through buf when
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
497 # returning exactly all of our first recv().
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
498 return data
24664
6a6d02be321c Another refactoring of read() and readline(), this time based on the
Guido van Rossum <guido@python.org>
parents: 24663
diff changeset
499 if n >= left:
48195
65901ffffc6a Remove buffer() usage in the socket module by just slicing directly on the
Brett Cannon <bcannon@gmail.com>
parents: 47372
diff changeset
500 buf.write(data[:left])
65901ffffc6a Remove buffer() usage in the socket module by just slicing directly on the
Brett Cannon <bcannon@gmail.com>
parents: 47372
diff changeset
501 self._rbuf.write(data[left:])
24664
6a6d02be321c Another refactoring of read() and readline(), this time based on the
Guido van Rossum <guido@python.org>
parents: 24663
diff changeset
502 break
46623
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
503 buf.write(data)
24664
6a6d02be321c Another refactoring of read() and readline(), this time based on the
Guido van Rossum <guido@python.org>
parents: 24663
diff changeset
504 buf_len += n
46623
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
505 #assert buf_len == buf.tell()
54606ea9f4c7 This should fix issue2632. A long description of the two competing
Gregory P. Smith <greg@mad-scientist.com>
parents: 45475
diff changeset
506 return buf.getvalue()
23715
c2733e8f6612 SF patch 555085 (timeout socket implementation) by Michael Gilfix.
Guido van Rossum <guido@python.org>
parents: 21971
diff changeset
507
c2733e8f6612 SF patch 555085 (timeout socket implementation) by Michael Gilfix.
Guido van Rossum <guido@python.org>
parents: 21971
diff changeset
508 def readlines(self, sizehint=0):
14682
295126433856 Support sizehint in _fileobject.readlines, as documented.
Martin v. Löwis <martin@v.loewis.de>
parents: 14592
diff changeset
509 total = 0
14007
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
510 list = []
24664
6a6d02be321c Another refactoring of read() and readline(), this time based on the
Guido van Rossum <guido@python.org>
parents: 24663
diff changeset
511 while True:
14007
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
512 line = self.readline()
23844
bd37eb4daf3b Cleanup a little
Neal Norwitz <nnorwitz@gmail.com>
parents: 23715
diff changeset
513 if not line:
bd37eb4daf3b Cleanup a little
Neal Norwitz <nnorwitz@gmail.com>
parents: 23715
diff changeset
514 break
14007
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
515 list.append(line)
14682
295126433856 Support sizehint in _fileobject.readlines, as documented.
Martin v. Löwis <martin@v.loewis.de>
parents: 14592
diff changeset
516 total += len(line)
295126433856 Support sizehint in _fileobject.readlines, as documented.
Martin v. Löwis <martin@v.loewis.de>
parents: 14592
diff changeset
517 if sizehint and total >= sizehint:
295126433856 Support sizehint in _fileobject.readlines, as documented.
Martin v. Löwis <martin@v.loewis.de>
parents: 14592
diff changeset
518 break
14007
eda09f824a9f Revise the wrapper structure for the socket module:
Fred Drake <fdrake@acm.org>
parents:
diff changeset
519 return list
24656
e61b118e9cab Major restructuring of _fileobject. Hopefully several things now work
Guido van Rossum <guido@python.org>
parents: 24649
diff changeset
520
e61b118e9cab Major restructuring of _fileobject. Hopefully several things now work
Guido van Rossum <guido@python.org>
parents: 24649
diff changeset
521 # Iterator protocols
e61b118e9cab Major restructuring of _fileobject. Hopefully several things now work
Guido van Rossum <guido@python.org>
parents: 24649
diff changeset
522
e61b118e9cab Major restructuring of _fileobject. Hopefully several things now work
Guido van Rossum <guido@python.org>
parents: 24649
diff changeset
523 def __iter__(self):
e61b118e9cab Major restructuring of _fileobject. Hopefully several things now work
Guido van Rossum <guido@python.org>
parents: 24649
diff changeset
524 return self
e61b118e9cab Major restructuring of _fileobject. Hopefully several things now work
Guido van Rossum <guido@python.org>
parents: 24649
diff changeset
525
e61b118e9cab Major restructuring of _fileobject. Hopefully several things now work
Guido van Rossum <guido@python.org>
parents: 24649
diff changeset
526 def next(self):
e61b118e9cab Major restructuring of _fileobject. Hopefully several things now work
Guido van Rossum <guido@python.org>
parents: 24649
diff changeset
527 line = self.readline()
e61b118e9cab Major restructuring of _fileobject. Hopefully several things now work
Guido van Rossum <guido@python.org>
parents: 24649
diff changeset
528 if not line:
e61b118e9cab Major restructuring of _fileobject. Hopefully several things now work
Guido van Rossum <guido@python.org>
parents: 24649
diff changeset
529 raise StopIteration
e61b118e9cab Major restructuring of _fileobject. Hopefully several things now work
Guido van Rossum <guido@python.org>
parents: 24649
diff changeset
530 return line
41288
a73aa0b821ca Added a 'create_connect()' function to socket.py, which creates a
Facundo Batista <facundobatista@gmail.com>
parents: 41285
diff changeset
531
47372
e2c5ae30cffe Fixed the semantic of timeout for socket.create_connection and
Facundo Batista <facundobatista@gmail.com>
parents: 46684
diff changeset
532 _GLOBAL_DEFAULT_TIMEOUT = object()
41288
a73aa0b821ca Added a 'create_connect()' function to socket.py, which creates a
Facundo Batista <facundobatista@gmail.com>
parents: 41285
diff changeset
533
47372
e2c5ae30cffe Fixed the semantic of timeout for socket.create_connection and
Facundo Batista <facundobatista@gmail.com>
parents: 46684
diff changeset
534 def create_connection(address, timeout=_GLOBAL_DEFAULT_TIMEOUT):
e2c5ae30cffe Fixed the semantic of timeout for socket.create_connection and
Facundo Batista <facundobatista@gmail.com>
parents: 46684
diff changeset
535 """Connect to *address* and return the socket object.
41288
a73aa0b821ca Added a 'create_connect()' function to socket.py, which creates a
Facundo Batista <facundobatista@gmail.com>
parents: 41285
diff changeset
536
47372
e2c5ae30cffe Fixed the semantic of timeout for socket.create_connection and
Facundo Batista <facundobatista@gmail.com>
parents: 46684
diff changeset
537 Convenience function. Connect to *address* (a 2-tuple ``(host,
e2c5ae30cffe Fixed the semantic of timeout for socket.create_connection and
Facundo Batista <facundobatista@gmail.com>
parents: 46684
diff changeset
538 port)``) and return the socket object. Passing the optional
e2c5ae30cffe Fixed the semantic of timeout for socket.create_connection and
Facundo Batista <facundobatista@gmail.com>
parents: 46684
diff changeset
539 *timeout* parameter will set the timeout on the socket instance
e2c5ae30cffe Fixed the semantic of timeout for socket.create_connection and
Facundo Batista <facundobatista@gmail.com>
parents: 46684
diff changeset
540 before attempting to connect. If no *timeout* is supplied, the
e2c5ae30cffe Fixed the semantic of timeout for socket.create_connection and
Facundo Batista <facundobatista@gmail.com>
parents: 46684
diff changeset
541 global default timeout setting returned by :func:`getdefaulttimeout`
e2c5ae30cffe Fixed the semantic of timeout for socket.create_connection and
Facundo Batista <facundobatista@gmail.com>
parents: 46684
diff changeset
542 is used.
41288
a73aa0b821ca Added a 'create_connect()' function to socket.py, which creates a
Facundo Batista <facundobatista@gmail.com>
parents: 41285
diff changeset
543 """
41532
4b9fa8092e82 Whitespace normalization. Ugh, we really need to do this more often.
Neal Norwitz <nnorwitz@gmail.com>
parents: 41288
diff changeset
544
41288
a73aa0b821ca Added a 'create_connect()' function to socket.py, which creates a
Facundo Batista <facundobatista@gmail.com>
parents: 41285
diff changeset
545 msg = "getaddrinfo returns an empty list"
a73aa0b821ca Added a 'create_connect()' function to socket.py, which creates a
Facundo Batista <facundobatista@gmail.com>
parents: 41285
diff changeset
546 host, port = address
a73aa0b821ca Added a 'create_connect()' function to socket.py, which creates a
Facundo Batista <facundobatista@gmail.com>
parents: 41285
diff changeset
547 for res in getaddrinfo(host, port, 0, SOCK_STREAM):
a73aa0b821ca Added a 'create_connect()' function to socket.py, which creates a
Facundo Batista <facundobatista@gmail.com>
parents: 41285
diff changeset
548 af, socktype, proto, canonname, sa = res
a73aa0b821ca Added a 'create_connect()' function to socket.py, which creates a
Facundo Batista <facundobatista@gmail.com>
parents: 41285
diff changeset
549 sock = None
a73aa0b821ca Added a 'create_connect()' function to socket.py, which creates a
Facundo Batista <facundobatista@gmail.com>
parents: 41285
diff changeset
550 try:
a73aa0b821ca Added a 'create_connect()' function to socket.py, which creates a
Facundo Batista <facundobatista@gmail.com>
parents: 41285
diff changeset
551 sock = socket(af, socktype, proto)
47372
e2c5ae30cffe Fixed the semantic of timeout for socket.create_connection and
Facundo Batista <facundobatista@gmail.com>
parents: 46684
diff changeset
552 if timeout is not _GLOBAL_DEFAULT_TIMEOUT:
41288
a73aa0b821ca Added a 'create_connect()' function to socket.py, which creates a
Facundo Batista <facundobatista@gmail.com>
parents: 41285
diff changeset
553 sock.settimeout(timeout)
a73aa0b821ca Added a 'create_connect()' function to socket.py, which creates a
Facundo Batista <facundobatista@gmail.com>
parents: 41285
diff changeset
554 sock.connect(sa)
a73aa0b821ca Added a 'create_connect()' function to socket.py, which creates a
Facundo Batista <facundobatista@gmail.com>
parents: 41285
diff changeset
555 return sock
41532
4b9fa8092e82 Whitespace normalization. Ugh, we really need to do this more often.
Neal Norwitz <nnorwitz@gmail.com>
parents: 41288
diff changeset
556
41288
a73aa0b821ca Added a 'create_connect()' function to socket.py, which creates a
Facundo Batista <facundobatista@gmail.com>
parents: 41285
diff changeset
557 except error, msg:
a73aa0b821ca Added a 'create_connect()' function to socket.py, which creates a
Facundo Batista <facundobatista@gmail.com>
parents: 41285
diff changeset
558 if sock is not None:
a73aa0b821ca Added a 'create_connect()' function to socket.py, which creates a
Facundo Batista <facundobatista@gmail.com>
parents: 41285
diff changeset
559 sock.close()
a73aa0b821ca Added a 'create_connect()' function to socket.py, which creates a
Facundo Batista <facundobatista@gmail.com>
parents: 41285
diff changeset
560
a73aa0b821ca Added a 'create_connect()' function to socket.py, which creates a
Facundo Batista <facundobatista@gmail.com>
parents: 41285
diff changeset
561 raise error, msg