blob: 085d58562ed4f01f150c45e42dd4ac5582e513d4 [file] [log] [blame]
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001/* Socket module */
2
3/*
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004
Guido van Rossum6574b3e1991-06-25 21:36:08 +00005This module provides an interface to Berkeley socket IPC.
6
7Limitations:
8
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00009- Only AF_INET, AF_INET6 and AF_UNIX address families are supported in a
Christian Heimes043d6f62008-01-07 17:19:16 +000010 portable manner, though AF_PACKET, AF_NETLINK and AF_TIPC are supported
11 under Linux.
Guido van Rossumc4fcfa32002-06-07 03:19:37 +000012- No read/write operations (use sendall/recv or makefile instead).
13- Additional restrictions apply on some non-Unix platforms (compensated
14 for by socket.py).
Guido van Rossum6574b3e1991-06-25 21:36:08 +000015
Guido van Rossum27e177d1995-03-16 15:43:47 +000016Module interface:
Guido van Rossum6574b3e1991-06-25 21:36:08 +000017
Guido van Rossum27e177d1995-03-16 15:43:47 +000018- socket.error: exception raised for socket specific errors
Martin v. Löwis2d8d4272001-07-21 18:05:31 +000019- socket.gaierror: exception raised for getaddrinfo/getnameinfo errors,
20 a subclass of socket.error
21- socket.herror: exception raised for gethostby* errors,
22 a subclass of socket.error
Guido van Rossum30a685f1991-06-27 15:51:29 +000023- socket.gethostbyname(hostname) --> host IP address (string: 'dd.dd.dd.dd')
Guido van Rossum3bbc62e1995-01-02 19:30:30 +000024- socket.gethostbyaddr(IP address) --> (hostname, [alias, ...], [IP addr, ...])
Guido van Rossum27e177d1995-03-16 15:43:47 +000025- socket.gethostname() --> host name (string: 'spam' or 'spam.domain.com')
Guido van Rossum25405c71996-12-19 16:42:52 +000026- socket.getprotobyname(protocolname) --> protocol number
Barry Warsaw11b91a02004-06-28 00:50:43 +000027- socket.getservbyname(servicename[, protocolname]) --> port number
28- socket.getservbyport(portnumber[, protocolname]) --> service name
Guido van Rossum7d0a8262007-05-21 23:13:11 +000029- socket.socket([family[, type [, proto, fileno]]]) --> new socket object
30 (fileno specifies a pre-existing socket file descriptor)
Dave Cole331708b2004-08-09 04:51:41 +000031- socket.socketpair([family[, type [, proto]]]) --> (socket, socket)
Guido van Rossum006bf911996-06-12 04:04:55 +000032- socket.ntohs(16 bit value) --> new int object
33- socket.ntohl(32 bit value) --> new int object
34- socket.htons(16 bit value) --> new int object
35- socket.htonl(32 bit value) --> new int object
Martin v. Löwis2d8d4272001-07-21 18:05:31 +000036- socket.getaddrinfo(host, port [, family, socktype, proto, flags])
37 --> List of (family, socktype, proto, canonname, sockaddr)
38- socket.getnameinfo(sockaddr, flags) --> (host, port)
Guido van Rossum27e177d1995-03-16 15:43:47 +000039- socket.AF_INET, socket.SOCK_STREAM, etc.: constants from <socket.h>
Guido van Rossum47dfa4a2003-04-25 05:48:32 +000040- socket.has_ipv6: boolean value indicating if IPv6 is supported
Guido van Rossum5c9eb211999-08-20 18:21:51 +000041- socket.inet_aton(IP address) -> 32-bit packed IP representation
42- socket.inet_ntoa(packed IP) -> IP address string
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +000043- socket.getdefaulttimeout() -> None | float
44- socket.setdefaulttimeout(None | float)
Guido van Rossum6574b3e1991-06-25 21:36:08 +000045- an Internet socket address is a pair (hostname, port)
46 where hostname can be anything recognized by gethostbyname()
47 (including the dd.dd.dd.dd notation) and port is in host byte order
48- where a hostname is returned, the dd.dd.dd.dd notation is used
Guido van Rossum27e177d1995-03-16 15:43:47 +000049- a UNIX domain socket address is a string specifying the pathname
Jeremy Hylton22308652001-02-02 03:23:09 +000050- an AF_PACKET socket address is a tuple containing a string
51 specifying the ethernet interface and an integer specifying
52 the Ethernet protocol number to be received. For example:
Jeremy Hyltondbfb6622001-02-02 19:55:17 +000053 ("eth0",0x1234). Optional 3rd,4th,5th elements in the tuple
Andrew M. Kuchlingb8e17172004-07-10 23:39:35 +000054 specify packet-type and ha-type/addr.
Christian Heimes043d6f62008-01-07 17:19:16 +000055- an AF_TIPC socket address is expressed as
56 (addr_type, v1, v2, v3 [, scope]); where addr_type can be one of:
57 TIPC_ADDR_NAMESEQ, TIPC_ADDR_NAME, and TIPC_ADDR_ID;
58 and scope can be one of:
59 TIPC_ZONE_SCOPE, TIPC_CLUSTER_SCOPE, and TIPC_NODE_SCOPE.
60 The meaning of v1, v2 and v3 depends on the value of addr_type:
61 if addr_type is TIPC_ADDR_NAME:
62 v1 is the server type
63 v2 is the port identifier
64 v3 is ignored
65 if addr_type is TIPC_ADDR_NAMESEQ:
66 v1 is the server type
67 v2 is the lower port number
68 v3 is the upper port number
69 if addr_type is TIPC_ADDR_ID:
70 v1 is the node
71 v2 is the ref
72 v3 is ignored
73
Guido van Rossum6574b3e1991-06-25 21:36:08 +000074
Guido van Rossumc4fcfa32002-06-07 03:19:37 +000075Local naming conventions:
Guido van Rossum6574b3e1991-06-25 21:36:08 +000076
Guido van Rossumc4fcfa32002-06-07 03:19:37 +000077- names starting with sock_ are socket object methods
78- names starting with socket_ are module-level functions
79- names starting with PySocket are exported through socketmodule.h
Guido van Rossum30a685f1991-06-27 15:51:29 +000080
Guido van Rossum6574b3e1991-06-25 21:36:08 +000081*/
82
Thomas Wouters477c8d52006-05-27 19:21:47 +000083#ifdef __APPLE__
84 /*
85 * inet_aton is not available on OSX 10.3, yet we want to use a binary
86 * that was build on 10.4 or later to work on that release, weak linking
87 * comes to the rescue.
88 */
89# pragma weak inet_aton
90#endif
91
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000092#include "Python.h"
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000093#include "structmember.h"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000094
Guido van Rossum47dfa4a2003-04-25 05:48:32 +000095#undef MAX
96#define MAX(x, y) ((x) < (y) ? (y) : (x))
97
Guido van Rossumc4fcfa32002-06-07 03:19:37 +000098/* Socket object documentation */
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000099PyDoc_STRVAR(sock_doc,
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000100"socket([family[, type[, proto]]]) -> socket object\n\
101\n\
102Open a socket of the given type. The family argument specifies the\n\
103address family; it defaults to AF_INET. The type argument specifies\n\
104whether this is a stream (SOCK_STREAM, this is the default)\n\
105or datagram (SOCK_DGRAM) socket. The protocol argument defaults to 0,\n\
106specifying the default protocol. Keyword arguments are accepted.\n\
107\n\
108A socket object represents one endpoint of a network connection.\n\
109\n\
110Methods of socket objects (keyword arguments not allowed):\n\
111\n\
Guido van Rossum39eb8fa2007-11-16 01:24:05 +0000112_accept() -- accept connection, returning new socket fd and client address\n\
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000113bind(addr) -- bind the socket to a local address\n\
114close() -- close the socket\n\
115connect(addr) -- connect the socket to a remote address\n\
116connect_ex(addr) -- connect, return an error code instead of an exception\n\
Guido van Rossum39eb8fa2007-11-16 01:24:05 +0000117_dup() -- return a new socket fd duplicated from fileno()\n\
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000118fileno() -- return underlying file descriptor\n\
119getpeername() -- return remote address [*]\n\
120getsockname() -- return local address\n\
121getsockopt(level, optname[, buflen]) -- get socket options\n\
122gettimeout() -- return timeout or None\n\
123listen(n) -- start listening for incoming connections\n\
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000124recv(buflen[, flags]) -- receive data\n\
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000125recv_into(buffer[, nbytes[, flags]]) -- receive data (into a buffer)\n\
Thomas Wouters477c8d52006-05-27 19:21:47 +0000126recvfrom(buflen[, flags]) -- receive data and sender\'s address\n\
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000127recvfrom_into(buffer[, nbytes, [, flags])\n\
Thomas Wouters477c8d52006-05-27 19:21:47 +0000128 -- receive data and sender\'s address (into a buffer)\n\
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000129sendall(data[, flags]) -- send all data\n\
130send(data[, flags]) -- send data, may not send all of it\n\
131sendto(data[, flags], addr) -- send data to a given address\n\
132setblocking(0 | 1) -- set or clear the blocking I/O flag\n\
133setsockopt(level, optname, value) -- set socket options\n\
134settimeout(None | float) -- set or clear the timeout\n\
135shutdown(how) -- shut down traffic in one or both directions\n\
136\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000137 [*] not available on all platforms!");
Guido van Rossum3baaa131999-03-22 21:44:51 +0000138
Walter Dörwaldf0dfc7a2003-10-20 14:01:56 +0000139/* XXX This is a terrible mess of platform-dependent preprocessor hacks.
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000140 I hope some day someone can clean this up please... */
141
Guido van Rossum9376b741999-09-15 22:01:40 +0000142/* Hacks for gethostbyname_r(). On some non-Linux platforms, the configure
143 script doesn't get this right, so we hardcode some platform checks below.
144 On the other hand, not all Linux versions agree, so there the settings
145 computed by the configure script are needed! */
146
147#ifndef linux
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000148# undef HAVE_GETHOSTBYNAME_R_3_ARG
149# undef HAVE_GETHOSTBYNAME_R_5_ARG
150# undef HAVE_GETHOSTBYNAME_R_6_ARG
Guido van Rossum9376b741999-09-15 22:01:40 +0000151#endif
Guido van Rossume7de2061999-03-24 17:24:33 +0000152
Guido van Rossum7a122991999-04-13 04:07:32 +0000153#ifndef WITH_THREAD
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000154# undef HAVE_GETHOSTBYNAME_R
Guido van Rossum7a122991999-04-13 04:07:32 +0000155#endif
156
Guido van Rossume7de2061999-03-24 17:24:33 +0000157#ifdef HAVE_GETHOSTBYNAME_R
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000158# if defined(_AIX) || defined(__osf__)
159# define HAVE_GETHOSTBYNAME_R_3_ARG
160# elif defined(__sun) || defined(__sgi)
161# define HAVE_GETHOSTBYNAME_R_5_ARG
162# elif defined(linux)
Guido van Rossum9376b741999-09-15 22:01:40 +0000163/* Rely on the configure script */
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000164# else
165# undef HAVE_GETHOSTBYNAME_R
166# endif
Guido van Rossume7de2061999-03-24 17:24:33 +0000167#endif
168
Guido van Rossum3eede5a2002-06-07 02:08:35 +0000169#if !defined(HAVE_GETHOSTBYNAME_R) && defined(WITH_THREAD) && \
170 !defined(MS_WINDOWS)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000171# define USE_GETHOSTBYNAME_LOCK
Guido van Rossum3baaa131999-03-22 21:44:51 +0000172#endif
173
Hye-Shik Chang9ceebd52005-09-24 14:58:47 +0000174/* To use __FreeBSD_version */
175#ifdef HAVE_SYS_PARAM_H
176#include <sys/param.h>
177#endif
Just van Rossum1040d2c2003-05-09 07:53:18 +0000178/* On systems on which getaddrinfo() is believed to not be thread-safe,
Just van Rossum09aecd72003-05-09 08:03:44 +0000179 (this includes the getaddrinfo emulation) protect access with a lock. */
Hye-Shik Chang9ceebd52005-09-24 14:58:47 +0000180#if defined(WITH_THREAD) && (defined(__APPLE__) || \
181 (defined(__FreeBSD__) && __FreeBSD_version+0 < 503000) || \
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000182 defined(__OpenBSD__) || defined(__NetBSD__) || \
183 defined(__VMS) || !defined(HAVE_GETADDRINFO))
Just van Rossum1040d2c2003-05-09 07:53:18 +0000184#define USE_GETADDRINFO_LOCK
185#endif
186
187#ifdef USE_GETADDRINFO_LOCK
188#define ACQUIRE_GETADDRINFO_LOCK PyThread_acquire_lock(netdb_lock, 1);
189#define RELEASE_GETADDRINFO_LOCK PyThread_release_lock(netdb_lock);
190#else
191#define ACQUIRE_GETADDRINFO_LOCK
192#define RELEASE_GETADDRINFO_LOCK
193#endif
194
195#if defined(USE_GETHOSTBYNAME_LOCK) || defined(USE_GETADDRINFO_LOCK)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000196# include "pythread.h"
Guido van Rossum4f199ea1998-04-09 20:56:35 +0000197#endif
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000198
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000199#if defined(PYCC_VACPP)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000200# include <types.h>
201# include <io.h>
202# include <sys/ioctl.h>
203# include <utils.h>
204# include <ctype.h>
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000205#endif
206
Martin v. Löwis9e437302002-12-06 12:57:26 +0000207#if defined(__VMS)
Martin v. Löwis79acb9e2002-12-06 12:48:53 +0000208# include <ioctl.h>
209#endif
210
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000211#if defined(PYOS_OS2)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000212# define INCL_DOS
213# define INCL_DOSERRORS
214# define INCL_NOPMAPI
215# include <os2.h>
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000216#endif
217
Jeremy Hyltonfb509a32003-07-17 16:58:48 +0000218#if defined(__sgi) && _COMPILER_VERSION>700 && !_SGIAPI
Martin v. Löwis8eb92a02002-09-19 08:03:21 +0000219/* make sure that the reentrant (gethostbyaddr_r etc)
220 functions are declared correctly if compiling with
221 MIPSPro 7.x in ANSI C mode (default) */
Jeremy Hyltonfb509a32003-07-17 16:58:48 +0000222
Thomas Wouters477c8d52006-05-27 19:21:47 +0000223/* XXX Using _SGIAPI is the wrong thing,
Jeremy Hyltonfb509a32003-07-17 16:58:48 +0000224 but I don't know what the right thing is. */
Trent Mick8ea5bdf2004-09-13 17:48:41 +0000225#undef _SGIAPI /* to avoid warning */
Martin v. Löwis8eb92a02002-09-19 08:03:21 +0000226#define _SGIAPI 1
Jeremy Hyltonfb509a32003-07-17 16:58:48 +0000227
Trent Mick8ea5bdf2004-09-13 17:48:41 +0000228#undef _XOPEN_SOURCE
229#include <sys/socket.h>
230#include <sys/types.h>
231#include <netinet/in.h>
232#ifdef _SS_ALIGNSIZE
233#define HAVE_GETADDRINFO 1
234#define HAVE_GETNAMEINFO 1
235#endif
236
Jeremy Hyltonfb509a32003-07-17 16:58:48 +0000237#define HAVE_INET_PTON
238#include <netdb.h>
Martin v. Löwis8eb92a02002-09-19 08:03:21 +0000239#endif
240
Thomas Wouters477c8d52006-05-27 19:21:47 +0000241/* Irix 6.5 fails to define this variable at all. This is needed
242 for both GCC and SGI's compiler. I'd say that the SGI headers
Georg Brandldbd83392006-02-20 09:42:33 +0000243 are just busted. Same thing for Solaris. */
244#if (defined(__sgi) || defined(sun)) && !defined(INET_ADDRSTRLEN)
Anthony Baxterbab23cf2003-10-04 08:00:49 +0000245#define INET_ADDRSTRLEN 16
246#endif
247
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000248/* Generic includes */
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000249#ifdef HAVE_SYS_TYPES_H
Guido van Rossumb6775db1994-08-01 11:34:53 +0000250#include <sys/types.h>
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000251#endif
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000252
Marc-André Lemburg976ade62002-02-16 18:47:07 +0000253/* Generic socket object definitions and includes */
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000254#define PySocket_BUILDING_SOCKET
Marc-André Lemburgbb8b78b2002-02-16 18:44:52 +0000255#include "socketmodule.h"
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000256
257/* Addressing includes */
258
Guido van Rossum6f489d91996-06-28 20:15:15 +0000259#ifndef MS_WINDOWS
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000260
261/* Non-MS WINDOWS includes */
262# include <netdb.h>
Guido van Rossum5c9eb211999-08-20 18:21:51 +0000263
Guido van Rossum9376b741999-09-15 22:01:40 +0000264/* Headers needed for inet_ntoa() and inet_addr() */
Skip Montanaroeb33e5a2007-08-17 12:57:41 +0000265# if defined(PYOS_OS2) && defined(PYCC_VACPP)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000266# include <netdb.h>
Tim Peters603c6832001-11-05 02:45:59 +0000267typedef size_t socklen_t;
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000268# else
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000269# include <arpa/inet.h>
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000270# endif
Guido van Rossum5c9eb211999-08-20 18:21:51 +0000271
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000272# include <fcntl.h>
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000273
Guido van Rossume2ae77b2001-10-24 20:42:55 +0000274#else
Guido van Rossum48a680c2001-03-02 06:34:14 +0000275
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000276/* MS_WINDOWS includes */
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000277# ifdef HAVE_FCNTL_H
278# include <fcntl.h>
279# endif
Guido van Rossum48a680c2001-03-02 06:34:14 +0000280
Jeremy Hylton22308652001-02-02 03:23:09 +0000281#endif
282
Skip Montanaro7befb992004-02-10 16:50:21 +0000283#include <stddef.h>
Martin v. Löwisa45ecae2001-06-24 21:28:42 +0000284
285#ifndef offsetof
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000286# define offsetof(type, member) ((size_t)(&((type *)0)->member))
Martin v. Löwisa45ecae2001-06-24 21:28:42 +0000287#endif
288
Neal Norwitz39d22e52002-11-02 19:55:21 +0000289#ifndef O_NONBLOCK
290# define O_NONBLOCK O_NDELAY
Guido van Rossum3bbc62e1995-01-02 19:30:30 +0000291#endif
292
Trent Micka708d6e2004-09-07 17:48:26 +0000293/* include Python's addrinfo.h unless it causes trouble */
294#if defined(__sgi) && _COMPILER_VERSION>700 && defined(_SS_ALIGNSIZE)
295 /* Do not include addinfo.h on some newer IRIX versions.
296 * _SS_ALIGNSIZE is defined in sys/socket.h by 6.5.21,
297 * for example, but not by 6.5.10.
298 */
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000299#elif defined(_MSC_VER) && _MSC_VER>1201
Trent Micka708d6e2004-09-07 17:48:26 +0000300 /* Do not include addrinfo.h for MSVC7 or greater. 'addrinfo' and
301 * EAI_* constants are defined in (the already included) ws2tcpip.h.
302 */
303#else
304# include "addrinfo.h"
305#endif
Jason Tishlerc246cb72004-08-09 13:25:59 +0000306
Martin v. Löwisb9ab1592001-06-24 21:18:26 +0000307#ifndef HAVE_INET_PTON
Christian Heimes96e7b3d2007-11-20 06:51:17 +0000308#if !defined(NTDDI_VERSION) || (NTDDI_VERSION < NTDDI_LONGHORN)
Guido van Rossum3eede5a2002-06-07 02:08:35 +0000309int inet_pton(int af, const char *src, void *dst);
Martin v. Löwisc925b1532001-07-21 09:42:15 +0000310const char *inet_ntop(int af, const void *src, char *dst, socklen_t size);
Martin v. Löwisb9ab1592001-06-24 21:18:26 +0000311#endif
Christian Heimesb6150692007-11-15 23:37:07 +0000312#endif
Martin v. Löwisb9ab1592001-06-24 21:18:26 +0000313
Martin v. Löwisae26dc22001-08-03 10:02:29 +0000314#ifdef __APPLE__
315/* On OS X, getaddrinfo returns no error indication of lookup
316 failure, so we must use the emulation instead of the libinfo
317 implementation. Unfortunately, performing an autoconf test
318 for this bug would require DNS access for the machine performing
319 the configuration, which is not acceptable. Therefore, we
320 determine the bug just by checking for __APPLE__. If this bug
321 gets ever fixed, perhaps checking for sys/version.h would be
322 appropriate, which is 10/0 on the system with the bug. */
Jack Jansen84262fb2002-07-02 14:40:42 +0000323#ifndef HAVE_GETNAMEINFO
324/* This bug seems to be fixed in Jaguar. Ths easiest way I could
325 Find to check for Jaguar is that it has getnameinfo(), which
326 older releases don't have */
Martin v. Löwisae26dc22001-08-03 10:02:29 +0000327#undef HAVE_GETADDRINFO
Martin v. Löwisae26dc22001-08-03 10:02:29 +0000328#endif
Thomas Wouters477c8d52006-05-27 19:21:47 +0000329
330#ifdef HAVE_INET_ATON
331#define USE_INET_ATON_WEAKLINK
332#endif
333
Jack Jansen84262fb2002-07-02 14:40:42 +0000334#endif
Martin v. Löwisae26dc22001-08-03 10:02:29 +0000335
Martin v. Löwis01dfdb32001-06-23 16:30:13 +0000336/* I know this is a bad practice, but it is the easiest... */
Martin v. Löwisae26dc22001-08-03 10:02:29 +0000337#if !defined(HAVE_GETADDRINFO)
Martin v. Löwisfccac2e2003-05-01 05:20:46 +0000338/* avoid clashes with the C library definition of the symbol. */
339#define getaddrinfo fake_getaddrinfo
340#define gai_strerror fake_gai_strerror
341#define freeaddrinfo fake_freeaddrinfo
Martin v. Löwis01dfdb32001-06-23 16:30:13 +0000342#include "getaddrinfo.c"
343#endif
Martin v. Löwisae26dc22001-08-03 10:02:29 +0000344#if !defined(HAVE_GETNAMEINFO)
Martin v. Löwisfccac2e2003-05-01 05:20:46 +0000345#define getnameinfo fake_getnameinfo
Martin v. Löwis01dfdb32001-06-23 16:30:13 +0000346#include "getnameinfo.c"
347#endif
348
Guido van Rossum39eb8fa2007-11-16 01:24:05 +0000349#ifdef MS_WINDOWS
350/* On Windows a socket is really a handle not an fd */
351static SOCKET
352dup_socket(SOCKET handle)
353{
354 HANDLE newhandle;
355
356 if (!DuplicateHandle(GetCurrentProcess(), (HANDLE)handle,
357 GetCurrentProcess(), &newhandle,
358 0, FALSE, DUPLICATE_SAME_ACCESS))
359 {
360 WSASetLastError(GetLastError());
361 return INVALID_SOCKET;
362 }
363 return (SOCKET)newhandle;
364}
Guido van Rossum2dd8ddd2000-04-21 20:33:00 +0000365#define SOCKETCLOSE closesocket
Guido van Rossum39eb8fa2007-11-16 01:24:05 +0000366#else
367/* On Unix we can use dup to duplicate the file descriptor of a socket*/
368#define dup_socket(fd) dup(fd)
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +0000369#endif
370
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000371#ifdef MS_WIN32
Guido van Rossum3eede5a2002-06-07 02:08:35 +0000372#define EAFNOSUPPORT WSAEAFNOSUPPORT
373#define snprintf _snprintf
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000374#endif
Fred Drakea04eaad2000-06-30 02:46:07 +0000375
Andrew MacIntyreba43e872002-03-03 03:03:52 +0000376#if defined(PYOS_OS2) && !defined(PYCC_GCC)
Guido van Rossum2dd8ddd2000-04-21 20:33:00 +0000377#define SOCKETCLOSE soclose
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000378#define NO_DUP /* Sockets are Not Actual File Handles under OS/2 */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000379#endif
380
Guido van Rossum2dd8ddd2000-04-21 20:33:00 +0000381#ifndef SOCKETCLOSE
382#define SOCKETCLOSE close
383#endif
384
Jesse Noller32d68c22009-03-31 18:48:42 +0000385#if defined(HAVE_BLUETOOTH_H) || defined(HAVE_BLUETOOTH_BLUETOOTH_H) && !defined(__NetBSD__)
Hye-Shik Chang81268602004-02-02 06:05:24 +0000386#define USE_BLUETOOTH 1
387#if defined(__FreeBSD__)
388#define BTPROTO_L2CAP BLUETOOTH_PROTO_L2CAP
389#define BTPROTO_RFCOMM BLUETOOTH_PROTO_RFCOMM
Thomas Wouterscf297e42007-02-23 15:07:44 +0000390#define BTPROTO_HCI BLUETOOTH_PROTO_HCI
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000391#define SOL_HCI SOL_HCI_RAW
392#define HCI_FILTER SO_HCI_RAW_FILTER
Hye-Shik Chang81268602004-02-02 06:05:24 +0000393#define sockaddr_l2 sockaddr_l2cap
394#define sockaddr_rc sockaddr_rfcomm
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000395#define hci_dev hci_node
Hye-Shik Chang81268602004-02-02 06:05:24 +0000396#define _BT_L2_MEMB(sa, memb) ((sa)->l2cap_##memb)
397#define _BT_RC_MEMB(sa, memb) ((sa)->rfcomm_##memb)
Thomas Wouterscf297e42007-02-23 15:07:44 +0000398#define _BT_HCI_MEMB(sa, memb) ((sa)->hci_##memb)
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000399#elif defined(__NetBSD__)
400#define sockaddr_l2 sockaddr_bt
401#define sockaddr_rc sockaddr_bt
Thomas Wouterscf297e42007-02-23 15:07:44 +0000402#define sockaddr_hci sockaddr_bt
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000403#define sockaddr_sco sockaddr_bt
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000404#define _BT_L2_MEMB(sa, memb) ((sa)->bt_##memb)
405#define _BT_RC_MEMB(sa, memb) ((sa)->bt_##memb)
Thomas Wouterscf297e42007-02-23 15:07:44 +0000406#define _BT_HCI_MEMB(sa, memb) ((sa)->bt_##memb)
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000407#define _BT_SCO_MEMB(sa, memb) ((sa)->bt_##memb)
Hye-Shik Chang81268602004-02-02 06:05:24 +0000408#else
Hye-Shik Chang81268602004-02-02 06:05:24 +0000409#define _BT_L2_MEMB(sa, memb) ((sa)->l2_##memb)
410#define _BT_RC_MEMB(sa, memb) ((sa)->rc_##memb)
Thomas Wouterscf297e42007-02-23 15:07:44 +0000411#define _BT_HCI_MEMB(sa, memb) ((sa)->hci_##memb)
Hye-Shik Chang81268602004-02-02 06:05:24 +0000412#define _BT_SCO_MEMB(sa, memb) ((sa)->sco_##memb)
413#endif
414#endif
415
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000416#ifdef __VMS
417/* TCP/IP Services for VMS uses a maximum send/recv buffer length */
418#define SEGMENT_SIZE (32 * 1024 -1)
419#endif
420
Thomas Wouters89f507f2006-12-13 04:49:30 +0000421#define SAS2SA(x) ((struct sockaddr *)(x))
422
Martin v. Löwise9416172003-05-03 10:12:45 +0000423/*
424 * Constants for getnameinfo()
425 */
426#if !defined(NI_MAXHOST)
427#define NI_MAXHOST 1025
428#endif
429#if !defined(NI_MAXSERV)
430#define NI_MAXSERV 32
431#endif
432
Guido van Rossum7d0a8262007-05-21 23:13:11 +0000433#ifndef INVALID_SOCKET /* MS defines this */
434#define INVALID_SOCKET (-1)
435#endif
436
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000437/* XXX There's a problem here: *static* functions are not supposed to have
438 a Py prefix (or use CapitalizedWords). Later... */
439
Guido van Rossum30a685f1991-06-27 15:51:29 +0000440/* Global variable holding the exception type for errors detected
441 by this module (but not argument type or memory errors, etc.). */
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000442static PyObject *socket_error;
443static PyObject *socket_herror;
444static PyObject *socket_gaierror;
Raymond Hettingeref7343c2003-06-29 03:08:05 +0000445static PyObject *socket_timeout;
Guido van Rossum30a685f1991-06-27 15:51:29 +0000446
Tim Peters643a7fc2002-02-17 04:13:21 +0000447/* A forward reference to the socket type object.
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000448 The sock_type variable contains pointers to various functions,
449 some of which call new_sockobject(), which uses sock_type, so
Tim Peters643a7fc2002-02-17 04:13:21 +0000450 there has to be a circular reference. */
Jeremy Hylton938ace62002-07-17 16:30:39 +0000451static PyTypeObject sock_type;
Guido van Rossum48a680c2001-03-02 06:34:14 +0000452
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000453#if defined(HAVE_POLL_H)
454#include <poll.h>
455#elif defined(HAVE_SYS_POLL_H)
456#include <sys/poll.h>
457#endif
458
Martin v. Löwisf84d1b92006-02-11 09:27:05 +0000459#ifdef Py_SOCKET_FD_CAN_BE_GE_FD_SETSIZE
460/* Platform can select file descriptors beyond FD_SETSIZE */
461#define IS_SELECTABLE(s) 1
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000462#elif defined(HAVE_POLL)
463/* Instead of select(), we'll use poll() since poll() works on any fd. */
464#define IS_SELECTABLE(s) 1
465/* Can we call select() with this socket without a buffer overrun? */
Martin v. Löwisf84d1b92006-02-11 09:27:05 +0000466#else
467/* POSIX says selecting file descriptors beyond FD_SETSIZE
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000468 has undefined behaviour. If there's no timeout left, we don't have to
469 call select, so it's a safe, little white lie. */
470#define IS_SELECTABLE(s) ((s)->sock_fd < FD_SETSIZE || s->sock_timeout <= 0.0)
Martin v. Löwisf84d1b92006-02-11 09:27:05 +0000471#endif
Neal Norwitz082b2df2006-02-07 07:04:46 +0000472
473static PyObject*
474select_error(void)
475{
476 PyErr_SetString(socket_error, "unable to select on socket");
477 return NULL;
478}
479
Guido van Rossum30a685f1991-06-27 15:51:29 +0000480/* Convenience function to raise an error according to errno
481 and return a NULL pointer from a function. */
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000482
Guido van Rossum73624e91994-10-10 17:59:00 +0000483static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000484set_error(void)
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000485{
Guido van Rossum8d665e61996-06-26 18:22:49 +0000486#ifdef MS_WINDOWS
Mark Hammond46a733d2000-07-24 01:45:11 +0000487 int err_no = WSAGetLastError();
Guido van Rossum3d392eb2007-11-16 00:35:22 +0000488 /* PyErr_SetExcFromWindowsErr() invokes FormatMessage() which
489 recognizes the error codes used by both GetLastError() and
490 WSAGetLastError */
491 if (err_no)
492 return PyErr_SetExcFromWindowsErr(socket_error, err_no);
Guido van Rossum14ed0b21994-09-29 09:50:09 +0000493#endif
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000494
Andrew MacIntyreba43e872002-03-03 03:03:52 +0000495#if defined(PYOS_OS2) && !defined(PYCC_GCC)
Guido van Rossum3eede5a2002-06-07 02:08:35 +0000496 if (sock_errno() != NO_ERROR) {
497 APIRET rc;
498 ULONG msglen;
499 char outbuf[100];
500 int myerrorcode = sock_errno();
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000501
Guido van Rossum3eede5a2002-06-07 02:08:35 +0000502 /* Retrieve socket-related error message from MPTN.MSG file */
503 rc = DosGetMessage(NULL, 0, outbuf, sizeof(outbuf),
504 myerrorcode - SOCBASEERR + 26,
505 "mptn.msg",
506 &msglen);
507 if (rc == NO_ERROR) {
508 PyObject *v;
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000509
Guido van Rossum3eede5a2002-06-07 02:08:35 +0000510 /* OS/2 doesn't guarantee a terminator */
511 outbuf[msglen] = '\0';
512 if (strlen(outbuf) > 0) {
513 /* If non-empty msg, trim CRLF */
514 char *lastc = &outbuf[ strlen(outbuf)-1 ];
Neal Norwitz30b5c5d2005-12-19 06:05:18 +0000515 while (lastc > outbuf &&
516 isspace(Py_CHARMASK(*lastc))) {
Guido van Rossum3eede5a2002-06-07 02:08:35 +0000517 /* Trim trailing whitespace (CRLF) */
518 *lastc-- = '\0';
519 }
520 }
521 v = Py_BuildValue("(is)", myerrorcode, outbuf);
522 if (v != NULL) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000523 PyErr_SetObject(socket_error, v);
Guido van Rossum3eede5a2002-06-07 02:08:35 +0000524 Py_DECREF(v);
525 }
526 return NULL;
527 }
528 }
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000529#endif
530
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000531 return PyErr_SetFromErrno(socket_error);
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000532}
533
Guido van Rossum30a685f1991-06-27 15:51:29 +0000534
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000535static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000536set_herror(int h_error)
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000537{
538 PyObject *v;
539
540#ifdef HAVE_HSTRERROR
541 v = Py_BuildValue("(is)", h_error, (char *)hstrerror(h_error));
542#else
543 v = Py_BuildValue("(is)", h_error, "host not found");
544#endif
545 if (v != NULL) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000546 PyErr_SetObject(socket_herror, v);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000547 Py_DECREF(v);
548 }
549
550 return NULL;
551}
552
553
554static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000555set_gaierror(int error)
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000556{
557 PyObject *v;
558
Martin v. Löwis272cb402002-03-01 08:31:07 +0000559#ifdef EAI_SYSTEM
560 /* EAI_SYSTEM is not available on Windows XP. */
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000561 if (error == EAI_SYSTEM)
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000562 return set_error();
Martin v. Löwis272cb402002-03-01 08:31:07 +0000563#endif
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000564
Martin v. Löwisf95dd0a2001-08-15 17:14:33 +0000565#ifdef HAVE_GAI_STRERROR
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000566 v = Py_BuildValue("(is)", error, gai_strerror(error));
Martin v. Löwisf95dd0a2001-08-15 17:14:33 +0000567#else
568 v = Py_BuildValue("(is)", error, "getaddrinfo failed");
569#endif
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000570 if (v != NULL) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000571 PyErr_SetObject(socket_gaierror, v);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000572 Py_DECREF(v);
573 }
574
575 return NULL;
576}
577
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000578#ifdef __VMS
579/* Function to send in segments */
580static int
581sendsegmented(int sock_fd, char *buf, int len, int flags)
582{
583 int n = 0;
584 int remaining = len;
585
586 while (remaining > 0) {
587 unsigned int segment;
588
589 segment = (remaining >= SEGMENT_SIZE ? SEGMENT_SIZE : remaining);
590 n = send(sock_fd, buf, segment, flags);
591 if (n < 0) {
592 return n;
593 }
594 remaining -= segment;
595 buf += segment;
596 } /* end while */
597
598 return len;
599}
600#endif
601
Guido van Rossum3eede5a2002-06-07 02:08:35 +0000602/* Function to perform the setting of socket blocking mode
603 internally. block = (1 | 0). */
Guido van Rossum67f7a382002-06-06 21:08:16 +0000604static int
605internal_setblocking(PySocketSockObject *s, int block)
606{
Guido van Rossum67f7a382002-06-06 21:08:16 +0000607#ifndef MS_WINDOWS
608 int delay_flag;
609#endif
Guido van Rossum67f7a382002-06-06 21:08:16 +0000610
611 Py_BEGIN_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +0000612#ifndef MS_WINDOWS
613#if defined(PYOS_OS2) && !defined(PYCC_GCC)
614 block = !block;
615 ioctl(s->sock_fd, FIONBIO, (caddr_t)&block, sizeof(block));
Martin v. Löwis79acb9e2002-12-06 12:48:53 +0000616#elif defined(__VMS)
617 block = !block;
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000618 ioctl(s->sock_fd, FIONBIO, (unsigned int *)&block);
619#else /* !PYOS_OS2 && !__VMS */
Guido van Rossum67f7a382002-06-06 21:08:16 +0000620 delay_flag = fcntl(s->sock_fd, F_GETFL, 0);
621 if (block)
Neal Norwitz39d22e52002-11-02 19:55:21 +0000622 delay_flag &= (~O_NONBLOCK);
Guido van Rossum67f7a382002-06-06 21:08:16 +0000623 else
Neal Norwitz39d22e52002-11-02 19:55:21 +0000624 delay_flag |= O_NONBLOCK;
Guido van Rossum67f7a382002-06-06 21:08:16 +0000625 fcntl(s->sock_fd, F_SETFL, delay_flag);
626#endif /* !PYOS_OS2 */
627#else /* MS_WINDOWS */
628 block = !block;
629 ioctlsocket(s->sock_fd, FIONBIO, (u_long*)&block);
630#endif /* MS_WINDOWS */
Guido van Rossum67f7a382002-06-06 21:08:16 +0000631 Py_END_ALLOW_THREADS
632
633 /* Since these don't return anything */
634 return 1;
635}
636
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000637/* Do a select()/poll() on the socket, if necessary (sock_timeout > 0).
Guido van Rossum11ba0942002-06-13 15:07:44 +0000638 The argument writing indicates the direction.
Raymond Hettingeref7343c2003-06-29 03:08:05 +0000639 This does not raise an exception; we'll let our caller do that
640 after they've reacquired the interpreter lock.
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000641 Returns 1 on timeout, -1 on error, 0 otherwise. */
Raymond Hettingeref7343c2003-06-29 03:08:05 +0000642static int
Guido van Rossumb9e916a2002-06-07 01:42:47 +0000643internal_select(PySocketSockObject *s, int writing)
Guido van Rossum67f7a382002-06-06 21:08:16 +0000644{
Raymond Hettingeref7343c2003-06-29 03:08:05 +0000645 int n;
Guido van Rossum11ba0942002-06-13 15:07:44 +0000646
Guido van Rossumad654902002-07-19 12:44:59 +0000647 /* Nothing to do unless we're in timeout mode (not non-blocking) */
Guido van Rossum11ba0942002-06-13 15:07:44 +0000648 if (s->sock_timeout <= 0.0)
Raymond Hettingeref7343c2003-06-29 03:08:05 +0000649 return 0;
Guido van Rossum67f7a382002-06-06 21:08:16 +0000650
Guido van Rossumad654902002-07-19 12:44:59 +0000651 /* Guard against closed socket */
652 if (s->sock_fd < 0)
Raymond Hettingeref7343c2003-06-29 03:08:05 +0000653 return 0;
Guido van Rossumad654902002-07-19 12:44:59 +0000654
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000655 /* Prefer poll, if available, since you can poll() any fd
656 * which can't be done with select(). */
657#ifdef HAVE_POLL
658 {
659 struct pollfd pollfd;
660 int timeout;
Guido van Rossum67f7a382002-06-06 21:08:16 +0000661
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000662 pollfd.fd = s->sock_fd;
663 pollfd.events = writing ? POLLOUT : POLLIN;
664
665 /* s->sock_timeout is in seconds, timeout in ms */
Guido van Rossum39eb8fa2007-11-16 01:24:05 +0000666 timeout = (int)(s->sock_timeout * 1000 + 0.5);
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000667 n = poll(&pollfd, 1, timeout);
668 }
669#else
670 {
671 /* Construct the arguments to select */
672 fd_set fds;
673 struct timeval tv;
674 tv.tv_sec = (int)s->sock_timeout;
675 tv.tv_usec = (int)((s->sock_timeout - tv.tv_sec) * 1e6);
676 FD_ZERO(&fds);
677 FD_SET(s->sock_fd, &fds);
678
679 /* See if the socket is ready */
680 if (writing)
681 n = select(s->sock_fd+1, NULL, &fds, NULL, &tv);
682 else
683 n = select(s->sock_fd+1, &fds, NULL, NULL, &tv);
684 }
685#endif
Guido van Rossum39eb8fa2007-11-16 01:24:05 +0000686
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000687 if (n < 0)
688 return -1;
Raymond Hettingeref7343c2003-06-29 03:08:05 +0000689 if (n == 0)
690 return 1;
691 return 0;
Guido van Rossum67f7a382002-06-06 21:08:16 +0000692}
693
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000694/* Initialize a new socket object. */
695
Tim Petersa12b4cf2002-07-18 22:38:44 +0000696static double defaulttimeout = -1.0; /* Default timeout for new sockets */
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +0000697
Martin v. Löwis1a214512008-06-11 05:26:20 +0000698static void
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000699init_sockobject(PySocketSockObject *s,
700 SOCKET_T fd, int family, int type, int proto)
701{
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000702 s->sock_fd = fd;
703 s->sock_family = family;
704 s->sock_type = type;
705 s->sock_proto = proto;
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +0000706 s->sock_timeout = defaulttimeout;
Guido van Rossum67f7a382002-06-06 21:08:16 +0000707
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000708 s->errorhandler = &set_error;
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +0000709
710 if (defaulttimeout >= 0.0)
711 internal_setblocking(s, 0);
712
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000713}
714
715
Guido van Rossum30a685f1991-06-27 15:51:29 +0000716/* Create a new socket object.
717 This just creates the object and initializes it.
718 If the creation fails, return NULL and set an exception (implicit
719 in NEWOBJ()). */
720
Guido van Rossum73624e91994-10-10 17:59:00 +0000721static PySocketSockObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000722new_sockobject(SOCKET_T fd, int family, int type, int proto)
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000723{
Guido van Rossum73624e91994-10-10 17:59:00 +0000724 PySocketSockObject *s;
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000725 s = (PySocketSockObject *)
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000726 PyType_GenericNew(&sock_type, NULL, NULL);
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000727 if (s != NULL)
728 init_sockobject(s, fd, family, type, proto);
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000729 return s;
730}
731
Guido van Rossum30a685f1991-06-27 15:51:29 +0000732
Guido van Rossum48a680c2001-03-02 06:34:14 +0000733/* Lock to allow python interpreter to continue, but only allow one
Just van Rossum1040d2c2003-05-09 07:53:18 +0000734 thread to be in gethostbyname or getaddrinfo */
735#if defined(USE_GETHOSTBYNAME_LOCK) || defined(USE_GETADDRINFO_LOCK)
736PyThread_type_lock netdb_lock;
Guido van Rossum4f199ea1998-04-09 20:56:35 +0000737#endif
738
739
Guido van Rossum30a685f1991-06-27 15:51:29 +0000740/* Convert a string specifying a host name or one of a few symbolic
741 names to a numeric IP address. This usually calls gethostbyname()
742 to do the work; the names "" and "<broadcast>" are special.
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000743 Return the length (IPv4 should be 4 bytes), or negative if
Guido van Rossum30a685f1991-06-27 15:51:29 +0000744 an error occurred; then an exception is raised. */
745
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000746static int
Martin v. Löwisddc6f472002-07-28 16:10:31 +0000747setipaddr(char *name, struct sockaddr *addr_ret, size_t addr_ret_size, int af)
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000748{
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000749 struct addrinfo hints, *res;
750 int error;
Anthony Baxter0e85f9d2003-05-02 15:40:46 +0000751 int d1, d2, d3, d4;
752 char ch;
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000753
Guido van Rossuma376cc51996-12-05 23:43:35 +0000754 memset((void *) addr_ret, '\0', sizeof(*addr_ret));
Guido van Rossum30a685f1991-06-27 15:51:29 +0000755 if (name[0] == '\0') {
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000756 int siz;
757 memset(&hints, 0, sizeof(hints));
758 hints.ai_family = af;
759 hints.ai_socktype = SOCK_DGRAM; /*dummy*/
760 hints.ai_flags = AI_PASSIVE;
Just van Rossum1040d2c2003-05-09 07:53:18 +0000761 Py_BEGIN_ALLOW_THREADS
762 ACQUIRE_GETADDRINFO_LOCK
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000763 error = getaddrinfo(NULL, "0", &hints, &res);
Just van Rossum1040d2c2003-05-09 07:53:18 +0000764 Py_END_ALLOW_THREADS
765 /* We assume that those thread-unsafe getaddrinfo() versions
766 *are* safe regarding their return value, ie. that a
767 subsequent call to getaddrinfo() does not destroy the
768 outcome of the first call. */
769 RELEASE_GETADDRINFO_LOCK
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000770 if (error) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000771 set_gaierror(error);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000772 return -1;
773 }
774 switch (res->ai_family) {
775 case AF_INET:
776 siz = 4;
777 break;
Martin v. Löwis44ddbde2001-12-02 10:15:37 +0000778#ifdef ENABLE_IPV6
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000779 case AF_INET6:
780 siz = 16;
781 break;
782#endif
783 default:
784 freeaddrinfo(res);
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000785 PyErr_SetString(socket_error,
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000786 "unsupported address family");
787 return -1;
788 }
789 if (res->ai_next) {
Martin v. Löwisf0b11d22001-11-07 08:31:03 +0000790 freeaddrinfo(res);
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000791 PyErr_SetString(socket_error,
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000792 "wildcard resolved to multiple address");
793 return -1;
794 }
Martin v. Löwisddc6f472002-07-28 16:10:31 +0000795 if (res->ai_addrlen < addr_ret_size)
796 addr_ret_size = res->ai_addrlen;
797 memcpy(addr_ret, res->ai_addr, addr_ret_size);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000798 freeaddrinfo(res);
799 return siz;
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000800 }
Guido van Rossum30a685f1991-06-27 15:51:29 +0000801 if (name[0] == '<' && strcmp(name, "<broadcast>") == 0) {
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000802 struct sockaddr_in *sin;
Martin v. Löwiscf8f47e2002-12-11 13:10:57 +0000803 if (af != AF_INET && af != AF_UNSPEC) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000804 PyErr_SetString(socket_error,
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000805 "address family mismatched");
806 return -1;
807 }
808 sin = (struct sockaddr_in *)addr_ret;
809 memset((void *) sin, '\0', sizeof(*sin));
810 sin->sin_family = AF_INET;
811#ifdef HAVE_SOCKADDR_SA_LEN
812 sin->sin_len = sizeof(*sin);
813#endif
814 sin->sin_addr.s_addr = INADDR_BROADCAST;
815 return sizeof(sin->sin_addr);
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000816 }
Anthony Baxter0e85f9d2003-05-02 15:40:46 +0000817 if (sscanf(name, "%d.%d.%d.%d%c", &d1, &d2, &d3, &d4, &ch) == 4 &&
818 0 <= d1 && d1 <= 255 && 0 <= d2 && d2 <= 255 &&
819 0 <= d3 && d3 <= 255 && 0 <= d4 && d4 <= 255) {
820 struct sockaddr_in *sin;
821 sin = (struct sockaddr_in *)addr_ret;
822 sin->sin_addr.s_addr = htonl(
823 ((long) d1 << 24) | ((long) d2 << 16) |
824 ((long) d3 << 8) | ((long) d4 << 0));
825 sin->sin_family = AF_INET;
826#ifdef HAVE_SOCKADDR_SA_LEN
827 sin->sin_len = sizeof(*sin);
828#endif
829 return 4;
830 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000831 memset(&hints, 0, sizeof(hints));
832 hints.ai_family = af;
Just van Rossum1040d2c2003-05-09 07:53:18 +0000833 Py_BEGIN_ALLOW_THREADS
834 ACQUIRE_GETADDRINFO_LOCK
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000835 error = getaddrinfo(name, NULL, &hints, &res);
Martin v. Löwis7c4b5fa2001-10-25 09:04:03 +0000836#if defined(__digital__) && defined(__unix__)
Guido van Rossum3eede5a2002-06-07 02:08:35 +0000837 if (error == EAI_NONAME && af == AF_UNSPEC) {
838 /* On Tru64 V5.1, numeric-to-addr conversion fails
839 if no address family is given. Assume IPv4 for now.*/
840 hints.ai_family = AF_INET;
841 error = getaddrinfo(name, NULL, &hints, &res);
842 }
Martin v. Löwis7c4b5fa2001-10-25 09:04:03 +0000843#endif
Just van Rossum1040d2c2003-05-09 07:53:18 +0000844 Py_END_ALLOW_THREADS
845 RELEASE_GETADDRINFO_LOCK /* see comment in setipaddr() */
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000846 if (error) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000847 set_gaierror(error);
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000848 return -1;
849 }
Martin v. Löwisddc6f472002-07-28 16:10:31 +0000850 if (res->ai_addrlen < addr_ret_size)
851 addr_ret_size = res->ai_addrlen;
852 memcpy((char *) addr_ret, res->ai_addr, addr_ret_size);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000853 freeaddrinfo(res);
854 switch (addr_ret->sa_family) {
855 case AF_INET:
856 return 4;
Martin v. Löwis44ddbde2001-12-02 10:15:37 +0000857#ifdef ENABLE_IPV6
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000858 case AF_INET6:
859 return 16;
Guido van Rossum955becc1999-03-22 20:14:53 +0000860#endif
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000861 default:
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000862 PyErr_SetString(socket_error, "unknown address family");
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000863 return -1;
864 }
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000865}
866
Guido van Rossum30a685f1991-06-27 15:51:29 +0000867
Guido van Rossum30a685f1991-06-27 15:51:29 +0000868/* Create a string object representing an IP address.
869 This is always a string of the form 'dd.dd.dd.dd' (with variable
870 size numbers). */
871
Guido van Rossum73624e91994-10-10 17:59:00 +0000872static PyObject *
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000873makeipaddr(struct sockaddr *addr, int addrlen)
Guido van Rossum30a685f1991-06-27 15:51:29 +0000874{
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000875 char buf[NI_MAXHOST];
876 int error;
877
878 error = getnameinfo(addr, addrlen, buf, sizeof(buf), NULL, 0,
879 NI_NUMERICHOST);
880 if (error) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000881 set_gaierror(error);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000882 return NULL;
883 }
Guido van Rossum32c4ac02007-08-15 03:56:40 +0000884 return PyUnicode_FromString(buf);
Guido van Rossum30a685f1991-06-27 15:51:29 +0000885}
886
887
Martin v. Löwis558d9bf2004-06-03 09:24:42 +0000888#ifdef USE_BLUETOOTH
889/* Convert a string representation of a Bluetooth address into a numeric
890 address. Returns the length (6), or raises an exception and returns -1 if
891 an error occurred. */
892
893static int
894setbdaddr(char *name, bdaddr_t *bdaddr)
895{
896 unsigned int b0, b1, b2, b3, b4, b5;
897 char ch;
898 int n;
899
900 n = sscanf(name, "%X:%X:%X:%X:%X:%X%c",
901 &b5, &b4, &b3, &b2, &b1, &b0, &ch);
902 if (n == 6 && (b0 | b1 | b2 | b3 | b4 | b5) < 256) {
903 bdaddr->b[0] = b0;
904 bdaddr->b[1] = b1;
905 bdaddr->b[2] = b2;
906 bdaddr->b[3] = b3;
907 bdaddr->b[4] = b4;
908 bdaddr->b[5] = b5;
909 return 6;
910 } else {
911 PyErr_SetString(socket_error, "bad bluetooth address");
912 return -1;
913 }
914}
915
916/* Create a string representation of the Bluetooth address. This is always a
917 string of the form 'XX:XX:XX:XX:XX:XX' where XX is a two digit hexadecimal
918 value (zero padded if necessary). */
919
920static PyObject *
921makebdaddr(bdaddr_t *bdaddr)
922{
923 char buf[(6 * 2) + 5 + 1];
924
925 sprintf(buf, "%02X:%02X:%02X:%02X:%02X:%02X",
926 bdaddr->b[5], bdaddr->b[4], bdaddr->b[3],
927 bdaddr->b[2], bdaddr->b[1], bdaddr->b[0]);
Guido van Rossum32c4ac02007-08-15 03:56:40 +0000928 return PyUnicode_FromString(buf);
Martin v. Löwis558d9bf2004-06-03 09:24:42 +0000929}
930#endif
931
932
Guido van Rossum30a685f1991-06-27 15:51:29 +0000933/* Create an object representing the given socket address,
934 suitable for passing it back to bind(), connect() etc.
935 The family field of the sockaddr structure is inspected
936 to determine what kind of address it really is. */
937
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000938/*ARGSUSED*/
Guido van Rossum73624e91994-10-10 17:59:00 +0000939static PyObject *
Martin v. Löwis558d9bf2004-06-03 09:24:42 +0000940makesockaddr(int sockfd, struct sockaddr *addr, int addrlen, int proto)
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000941{
Guido van Rossum25bec8c1992-08-05 19:00:45 +0000942 if (addrlen == 0) {
943 /* No address -- may be recvfrom() from known socket */
Guido van Rossum73624e91994-10-10 17:59:00 +0000944 Py_INCREF(Py_None);
945 return Py_None;
Guido van Rossum25bec8c1992-08-05 19:00:45 +0000946 }
947
Guido van Rossum30a685f1991-06-27 15:51:29 +0000948 switch (addr->sa_family) {
949
950 case AF_INET:
951 {
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000952 struct sockaddr_in *a;
953 PyObject *addrobj = makeipaddr(addr, sizeof(*a));
Barry Warsaw752300b1997-01-03 17:18:10 +0000954 PyObject *ret = NULL;
955 if (addrobj) {
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000956 a = (struct sockaddr_in *)addr;
Barry Warsaw752300b1997-01-03 17:18:10 +0000957 ret = Py_BuildValue("Oi", addrobj, ntohs(a->sin_port));
958 Py_DECREF(addrobj);
959 }
Guido van Rossum6f5afc91993-02-05 09:46:15 +0000960 return ret;
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000961 }
Guido van Rossum30a685f1991-06-27 15:51:29 +0000962
Andrew MacIntyred12dfbb2004-04-04 07:13:49 +0000963#if defined(AF_UNIX)
Guido van Rossum30a685f1991-06-27 15:51:29 +0000964 case AF_UNIX:
965 {
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000966 struct sockaddr_un *a = (struct sockaddr_un *) addr;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000967#ifdef linux
968 if (a->sun_path[0] == 0) { /* Linux abstract namespace */
Christian Heimesbbffeb62008-01-24 09:42:52 +0000969 addrlen -= offsetof(struct sockaddr_un, sun_path);
Christian Heimes72b710a2008-05-26 13:28:38 +0000970 return PyBytes_FromStringAndSize(a->sun_path, addrlen);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000971 }
972 else
973#endif /* linux */
974 {
975 /* regular NULL-terminated string */
Guido van Rossum32c4ac02007-08-15 03:56:40 +0000976 return PyUnicode_FromString(a->sun_path);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000977 }
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000978 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000979#endif /* AF_UNIX */
980
Martin v. Löwis11017b12006-01-14 18:12:57 +0000981#if defined(AF_NETLINK)
982 case AF_NETLINK:
983 {
984 struct sockaddr_nl *a = (struct sockaddr_nl *) addr;
Martin v. Löwisd0560052006-01-15 07:49:20 +0000985 return Py_BuildValue("II", a->nl_pid, a->nl_groups);
Martin v. Löwis11017b12006-01-14 18:12:57 +0000986 }
987#endif /* AF_NETLINK */
988
Martin v. Löwis44ddbde2001-12-02 10:15:37 +0000989#ifdef ENABLE_IPV6
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000990 case AF_INET6:
991 {
992 struct sockaddr_in6 *a;
993 PyObject *addrobj = makeipaddr(addr, sizeof(*a));
994 PyObject *ret = NULL;
995 if (addrobj) {
996 a = (struct sockaddr_in6 *)addr;
Guido van Rossum3eede5a2002-06-07 02:08:35 +0000997 ret = Py_BuildValue("Oiii",
998 addrobj,
999 ntohs(a->sin6_port),
1000 a->sin6_flowinfo,
1001 a->sin6_scope_id);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00001002 Py_DECREF(addrobj);
1003 }
1004 return ret;
1005 }
Jeremy Hylton22308652001-02-02 03:23:09 +00001006#endif
Guido van Rossum30a685f1991-06-27 15:51:29 +00001007
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001008#ifdef USE_BLUETOOTH
1009 case AF_BLUETOOTH:
1010 switch (proto) {
1011
1012 case BTPROTO_L2CAP:
1013 {
1014 struct sockaddr_l2 *a = (struct sockaddr_l2 *) addr;
1015 PyObject *addrobj = makebdaddr(&_BT_L2_MEMB(a, bdaddr));
1016 PyObject *ret = NULL;
1017 if (addrobj) {
1018 ret = Py_BuildValue("Oi",
1019 addrobj,
1020 _BT_L2_MEMB(a, psm));
1021 Py_DECREF(addrobj);
1022 }
1023 return ret;
1024 }
1025
1026 case BTPROTO_RFCOMM:
1027 {
1028 struct sockaddr_rc *a = (struct sockaddr_rc *) addr;
1029 PyObject *addrobj = makebdaddr(&_BT_RC_MEMB(a, bdaddr));
1030 PyObject *ret = NULL;
1031 if (addrobj) {
1032 ret = Py_BuildValue("Oi",
1033 addrobj,
1034 _BT_RC_MEMB(a, channel));
1035 Py_DECREF(addrobj);
1036 }
1037 return ret;
1038 }
1039
Thomas Wouterscf297e42007-02-23 15:07:44 +00001040 case BTPROTO_HCI:
1041 {
1042 struct sockaddr_hci *a = (struct sockaddr_hci *) addr;
1043 PyObject *ret = NULL;
1044 ret = Py_BuildValue("i", _BT_HCI_MEMB(a, dev));
1045 return ret;
1046 }
1047
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001048#if !defined(__FreeBSD__)
1049 case BTPROTO_SCO:
1050 {
1051 struct sockaddr_sco *a = (struct sockaddr_sco *) addr;
1052 return makebdaddr(&_BT_SCO_MEMB(a, bdaddr));
1053 }
1054#endif
1055
Antoine Pitroub6375f12010-02-04 20:23:24 +00001056 default:
1057 PyErr_SetString(PyExc_ValueError,
1058 "Unknown Bluetooth protocol");
1059 return NULL;
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001060 }
1061#endif
1062
Martin v. Löwis1ba3fd52001-08-10 20:29:40 +00001063#ifdef HAVE_NETPACKET_PACKET_H
Jeremy Hylton22308652001-02-02 03:23:09 +00001064 case AF_PACKET:
1065 {
1066 struct sockaddr_ll *a = (struct sockaddr_ll *)addr;
1067 char *ifname = "";
1068 struct ifreq ifr;
Jeremy Hyltondbfb6622001-02-02 19:55:17 +00001069 /* need to look up interface name give index */
1070 if (a->sll_ifindex) {
Jeremy Hylton22308652001-02-02 03:23:09 +00001071 ifr.ifr_ifindex = a->sll_ifindex;
Jeremy Hyltondbfb6622001-02-02 19:55:17 +00001072 if (ioctl(sockfd, SIOCGIFNAME, &ifr) == 0)
Jeremy Hylton22308652001-02-02 03:23:09 +00001073 ifname = ifr.ifr_name;
Jeremy Hylton22308652001-02-02 03:23:09 +00001074 }
Guido van Rossum32c4ac02007-08-15 03:56:40 +00001075 return Py_BuildValue("shbhy#",
Guido van Rossum3eede5a2002-06-07 02:08:35 +00001076 ifname,
1077 ntohs(a->sll_protocol),
1078 a->sll_pkttype,
1079 a->sll_hatype,
1080 a->sll_addr,
1081 a->sll_halen);
Jeremy Hylton22308652001-02-02 03:23:09 +00001082 }
1083#endif
Guido van Rossum48a680c2001-03-02 06:34:14 +00001084
Christian Heimes043d6f62008-01-07 17:19:16 +00001085#ifdef HAVE_LINUX_TIPC_H
1086 case AF_TIPC:
1087 {
1088 struct sockaddr_tipc *a = (struct sockaddr_tipc *) addr;
1089 if (a->addrtype == TIPC_ADDR_NAMESEQ) {
1090 return Py_BuildValue("IIIII",
1091 a->addrtype,
1092 a->addr.nameseq.type,
1093 a->addr.nameseq.lower,
1094 a->addr.nameseq.upper,
1095 a->scope);
1096 } else if (a->addrtype == TIPC_ADDR_NAME) {
1097 return Py_BuildValue("IIIII",
1098 a->addrtype,
1099 a->addr.name.name.type,
1100 a->addr.name.name.instance,
1101 a->addr.name.name.instance,
1102 a->scope);
1103 } else if (a->addrtype == TIPC_ADDR_ID) {
1104 return Py_BuildValue("IIIII",
1105 a->addrtype,
1106 a->addr.id.node,
1107 a->addr.id.ref,
1108 0,
1109 a->scope);
1110 } else {
Antoine Pitroub6375f12010-02-04 20:23:24 +00001111 PyErr_SetString(PyExc_ValueError,
Christian Heimes043d6f62008-01-07 17:19:16 +00001112 "Invalid address type");
1113 return NULL;
1114 }
1115 }
1116#endif
1117
Guido van Rossum30a685f1991-06-27 15:51:29 +00001118 /* More cases here... */
1119
1120 default:
Guido van Rossumaa948df1997-05-07 17:41:48 +00001121 /* If we don't know the address family, don't raise an
Guido van Rossum9ce8e382007-08-09 21:06:37 +00001122 exception -- return it as an (int, bytes) tuple. */
1123 return Py_BuildValue("iy#",
Guido van Rossumaa948df1997-05-07 17:41:48 +00001124 addr->sa_family,
1125 addr->sa_data,
1126 sizeof(addr->sa_data));
Guido van Rossum25bec8c1992-08-05 19:00:45 +00001127
Guido van Rossum30a685f1991-06-27 15:51:29 +00001128 }
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001129}
1130
Guido van Rossum30a685f1991-06-27 15:51:29 +00001131
1132/* Parse a socket address argument according to the socket object's
1133 address family. Return 1 if the address was in the proper format,
1134 0 of not. The address is returned through addr_ret, its length
1135 through len_ret. */
1136
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001137static int
Guido van Rossum48a680c2001-03-02 06:34:14 +00001138getsockaddrarg(PySocketSockObject *s, PyObject *args,
Thomas Wouters89f507f2006-12-13 04:49:30 +00001139 struct sockaddr *addr_ret, int *len_ret)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001140{
Guido van Rossum30a685f1991-06-27 15:51:29 +00001141 switch (s->sock_family) {
1142
Andrew MacIntyred12dfbb2004-04-04 07:13:49 +00001143#if defined(AF_UNIX)
Guido van Rossum30a685f1991-06-27 15:51:29 +00001144 case AF_UNIX:
1145 {
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00001146 struct sockaddr_un* addr;
Guido van Rossumff4949e1992-08-05 19:58:53 +00001147 char *path;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001148 int len;
Guido van Rossume22905a2007-08-27 23:09:25 +00001149 if (!PyArg_Parse(args, "s#", &path, &len))
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001150 return 0;
Thomas Wouters89f507f2006-12-13 04:49:30 +00001151
1152 addr = (struct sockaddr_un*)addr_ret;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001153#ifdef linux
1154 if (len > 0 && path[0] == 0) {
1155 /* Linux abstract namespace extension */
1156 if (len > sizeof addr->sun_path) {
1157 PyErr_SetString(socket_error,
1158 "AF_UNIX path too long");
1159 return 0;
1160 }
1161 }
1162 else
1163#endif /* linux */
1164 {
1165 /* regular NULL-terminated string */
1166 if (len >= sizeof addr->sun_path) {
1167 PyErr_SetString(socket_error,
1168 "AF_UNIX path too long");
1169 return 0;
1170 }
1171 addr->sun_path[len] = 0;
Guido van Rossum30a685f1991-06-27 15:51:29 +00001172 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00001173 addr->sun_family = s->sock_family;
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00001174 memcpy(addr->sun_path, path, len);
Andrew MacIntyredaedf212004-04-11 12:03:57 +00001175#if defined(PYOS_OS2)
1176 *len_ret = sizeof(*addr);
1177#else
Christian Heimesbbffeb62008-01-24 09:42:52 +00001178 *len_ret = len + offsetof(struct sockaddr_un, sun_path);
Andrew MacIntyredaedf212004-04-11 12:03:57 +00001179#endif
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001180 return 1;
1181 }
Guido van Rossumb6775db1994-08-01 11:34:53 +00001182#endif /* AF_UNIX */
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001183
Martin v. Löwis11017b12006-01-14 18:12:57 +00001184#if defined(AF_NETLINK)
1185 case AF_NETLINK:
1186 {
1187 struct sockaddr_nl* addr;
1188 int pid, groups;
Thomas Wouters89f507f2006-12-13 04:49:30 +00001189 addr = (struct sockaddr_nl *)addr_ret;
Martin v. Löwis11017b12006-01-14 18:12:57 +00001190 if (!PyTuple_Check(args)) {
1191 PyErr_Format(
1192 PyExc_TypeError,
1193 "getsockaddrarg: "
1194 "AF_NETLINK address must be tuple, not %.500s",
Christian Heimes90aa7642007-12-19 02:45:37 +00001195 Py_TYPE(args)->tp_name);
Martin v. Löwis11017b12006-01-14 18:12:57 +00001196 return 0;
1197 }
1198 if (!PyArg_ParseTuple(args, "II:getsockaddrarg", &pid, &groups))
1199 return 0;
1200 addr->nl_family = AF_NETLINK;
1201 addr->nl_pid = pid;
1202 addr->nl_groups = groups;
Martin v. Löwis11017b12006-01-14 18:12:57 +00001203 *len_ret = sizeof(*addr);
1204 return 1;
1205 }
1206#endif
1207
Guido van Rossum30a685f1991-06-27 15:51:29 +00001208 case AF_INET:
1209 {
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00001210 struct sockaddr_in* addr;
Guido van Rossumff4949e1992-08-05 19:58:53 +00001211 char *host;
Martin v. Löwis5db099a2003-08-07 11:55:15 +00001212 int port, result;
Guido van Rossume4dad902000-12-01 13:13:11 +00001213 if (!PyTuple_Check(args)) {
Guido van Rossum3eede5a2002-06-07 02:08:35 +00001214 PyErr_Format(
1215 PyExc_TypeError,
1216 "getsockaddrarg: "
1217 "AF_INET address must be tuple, not %.500s",
Christian Heimes90aa7642007-12-19 02:45:37 +00001218 Py_TYPE(args)->tp_name);
Guido van Rossume4dad902000-12-01 13:13:11 +00001219 return 0;
1220 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00001221 if (!PyArg_ParseTuple(args, "eti:getsockaddrarg",
Martin v. Löwis2548c732003-04-18 10:39:54 +00001222 "idna", &host, &port))
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001223 return 0;
Thomas Wouters89f507f2006-12-13 04:49:30 +00001224 addr=(struct sockaddr_in*)addr_ret;
Thomas Wouters477c8d52006-05-27 19:21:47 +00001225 result = setipaddr(host, (struct sockaddr *)addr,
Martin v. Löwis5db099a2003-08-07 11:55:15 +00001226 sizeof(*addr), AF_INET);
1227 PyMem_Free(host);
1228 if (result < 0)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001229 return 0;
Benjamin Peterson5c6d7872009-02-06 02:40:07 +00001230 if (port < 0 || port > 0xffff) {
1231 PyErr_SetString(
1232 PyExc_OverflowError,
1233 "getsockaddrarg: port must be 0-65535.");
1234 return 0;
1235 }
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00001236 addr->sin_family = AF_INET;
Guido van Rossum644a12b1997-04-09 19:24:53 +00001237 addr->sin_port = htons((short)port);
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00001238 *len_ret = sizeof *addr;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001239 return 1;
1240 }
1241
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00001242#ifdef ENABLE_IPV6
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00001243 case AF_INET6:
1244 {
1245 struct sockaddr_in6* addr;
1246 char *host;
Martin v. Löwis5db099a2003-08-07 11:55:15 +00001247 int port, flowinfo, scope_id, result;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00001248 flowinfo = scope_id = 0;
Thomas Wouters477c8d52006-05-27 19:21:47 +00001249 if (!PyTuple_Check(args)) {
1250 PyErr_Format(
1251 PyExc_TypeError,
1252 "getsockaddrarg: "
1253 "AF_INET6 address must be tuple, not %.500s",
Christian Heimes90aa7642007-12-19 02:45:37 +00001254 Py_TYPE(args)->tp_name);
Thomas Wouters477c8d52006-05-27 19:21:47 +00001255 return 0;
1256 }
1257 if (!PyArg_ParseTuple(args, "eti|ii",
Martin v. Löwis2548c732003-04-18 10:39:54 +00001258 "idna", &host, &port, &flowinfo,
Guido van Rossum3eede5a2002-06-07 02:08:35 +00001259 &scope_id)) {
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00001260 return 0;
1261 }
Thomas Wouters89f507f2006-12-13 04:49:30 +00001262 addr = (struct sockaddr_in6*)addr_ret;
Thomas Wouters477c8d52006-05-27 19:21:47 +00001263 result = setipaddr(host, (struct sockaddr *)addr,
Martin v. Löwis5db099a2003-08-07 11:55:15 +00001264 sizeof(*addr), AF_INET6);
1265 PyMem_Free(host);
1266 if (result < 0)
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00001267 return 0;
Benjamin Peterson5c6d7872009-02-06 02:40:07 +00001268 if (port < 0 || port > 0xffff) {
1269 PyErr_SetString(
1270 PyExc_OverflowError,
1271 "getsockaddrarg: port must be 0-65535.");
1272 return 0;
1273 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00001274 addr->sin6_family = s->sock_family;
1275 addr->sin6_port = htons((short)port);
1276 addr->sin6_flowinfo = flowinfo;
1277 addr->sin6_scope_id = scope_id;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00001278 *len_ret = sizeof *addr;
1279 return 1;
1280 }
1281#endif
1282
Hye-Shik Chang81268602004-02-02 06:05:24 +00001283#ifdef USE_BLUETOOTH
Martin v. Löwis12af0482004-01-31 12:34:17 +00001284 case AF_BLUETOOTH:
1285 {
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001286 switch (s->sock_proto) {
1287 case BTPROTO_L2CAP:
Martin v. Löwis12af0482004-01-31 12:34:17 +00001288 {
Thomas Wouters89f507f2006-12-13 04:49:30 +00001289 struct sockaddr_l2 *addr;
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001290 char *straddr;
Martin v. Löwis12af0482004-01-31 12:34:17 +00001291
Thomas Wouters89f507f2006-12-13 04:49:30 +00001292 addr = (struct sockaddr_l2 *)addr_ret;
Antoine Pitroubf043492010-04-29 10:11:46 +00001293 memset(addr, 0, sizeof(struct sockaddr_l2));
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001294 _BT_L2_MEMB(addr, family) = AF_BLUETOOTH;
1295 if (!PyArg_ParseTuple(args, "si", &straddr,
1296 &_BT_L2_MEMB(addr, psm))) {
1297 PyErr_SetString(socket_error, "getsockaddrarg: "
1298 "wrong format");
Martin v. Löwis12af0482004-01-31 12:34:17 +00001299 return 0;
1300 }
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001301 if (setbdaddr(straddr, &_BT_L2_MEMB(addr, bdaddr)) < 0)
1302 return 0;
1303
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001304 *len_ret = sizeof *addr;
1305 return 1;
1306 }
1307 case BTPROTO_RFCOMM:
1308 {
Thomas Wouters89f507f2006-12-13 04:49:30 +00001309 struct sockaddr_rc *addr;
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001310 char *straddr;
1311
Thomas Wouters89f507f2006-12-13 04:49:30 +00001312 addr = (struct sockaddr_rc *)addr_ret;
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001313 _BT_RC_MEMB(addr, family) = AF_BLUETOOTH;
1314 if (!PyArg_ParseTuple(args, "si", &straddr,
1315 &_BT_RC_MEMB(addr, channel))) {
1316 PyErr_SetString(socket_error, "getsockaddrarg: "
1317 "wrong format");
1318 return 0;
1319 }
1320 if (setbdaddr(straddr, &_BT_RC_MEMB(addr, bdaddr)) < 0)
1321 return 0;
1322
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001323 *len_ret = sizeof *addr;
1324 return 1;
1325 }
Thomas Wouterscf297e42007-02-23 15:07:44 +00001326 case BTPROTO_HCI:
1327 {
1328 struct sockaddr_hci *addr = (struct sockaddr_hci *)addr_ret;
1329 _BT_HCI_MEMB(addr, family) = AF_BLUETOOTH;
1330 if (!PyArg_ParseTuple(args, "i", &_BT_HCI_MEMB(addr, dev))) {
1331 PyErr_SetString(socket_error, "getsockaddrarg: "
1332 "wrong format");
1333 return 0;
1334 }
1335 *len_ret = sizeof *addr;
1336 return 1;
1337 }
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001338#if !defined(__FreeBSD__)
1339 case BTPROTO_SCO:
1340 {
Thomas Wouters89f507f2006-12-13 04:49:30 +00001341 struct sockaddr_sco *addr;
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001342 char *straddr;
1343
Thomas Wouters89f507f2006-12-13 04:49:30 +00001344 addr = (struct sockaddr_sco *)addr_ret;
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001345 _BT_SCO_MEMB(addr, family) = AF_BLUETOOTH;
Christian Heimes72b710a2008-05-26 13:28:38 +00001346 if (!PyBytes_Check(args)) {
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001347 PyErr_SetString(socket_error, "getsockaddrarg: "
1348 "wrong format");
1349 return 0;
1350 }
Christian Heimes72b710a2008-05-26 13:28:38 +00001351 straddr = PyBytes_AS_STRING(args);
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001352 if (setbdaddr(straddr, &_BT_SCO_MEMB(addr, bdaddr)) < 0)
1353 return 0;
1354
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001355 *len_ret = sizeof *addr;
1356 return 1;
1357 }
1358#endif
1359 default:
1360 PyErr_SetString(socket_error, "getsockaddrarg: unknown Bluetooth protocol");
1361 return 0;
Martin v. Löwis12af0482004-01-31 12:34:17 +00001362 }
1363 }
1364#endif
1365
Martin v. Löwis1ba3fd52001-08-10 20:29:40 +00001366#ifdef HAVE_NETPACKET_PACKET_H
Jeremy Hylton22308652001-02-02 03:23:09 +00001367 case AF_PACKET:
1368 {
1369 struct sockaddr_ll* addr;
1370 struct ifreq ifr;
1371 char *interfaceName;
1372 int protoNumber;
1373 int hatype = 0;
1374 int pkttype = 0;
Andrew M. Kuchlingb8e17172004-07-10 23:39:35 +00001375 char *haddr = NULL;
1376 unsigned int halen = 0;
Guido van Rossum48a680c2001-03-02 06:34:14 +00001377
Thomas Wouters477c8d52006-05-27 19:21:47 +00001378 if (!PyTuple_Check(args)) {
1379 PyErr_Format(
1380 PyExc_TypeError,
1381 "getsockaddrarg: "
1382 "AF_PACKET address must be tuple, not %.500s",
Christian Heimes90aa7642007-12-19 02:45:37 +00001383 Py_TYPE(args)->tp_name);
Thomas Wouters477c8d52006-05-27 19:21:47 +00001384 return 0;
1385 }
Guido van Rossum8a392d72007-11-21 22:09:45 +00001386 if (!PyArg_ParseTuple(args, "si|iiy#", &interfaceName,
Andrew M. Kuchlingb8e17172004-07-10 23:39:35 +00001387 &protoNumber, &pkttype, &hatype,
1388 &haddr, &halen))
Jeremy Hylton22308652001-02-02 03:23:09 +00001389 return 0;
1390 strncpy(ifr.ifr_name, interfaceName, sizeof(ifr.ifr_name));
1391 ifr.ifr_name[(sizeof(ifr.ifr_name))-1] = '\0';
Jeremy Hyltondbfb6622001-02-02 19:55:17 +00001392 if (ioctl(s->sock_fd, SIOCGIFINDEX, &ifr) < 0) {
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00001393 s->errorhandler();
Jeremy Hylton22308652001-02-02 03:23:09 +00001394 return 0;
Jeremy Hyltondbfb6622001-02-02 19:55:17 +00001395 }
Andrew M. Kuchlingb8e17172004-07-10 23:39:35 +00001396 if (halen > 8) {
1397 PyErr_SetString(PyExc_ValueError,
1398 "Hardware address must be 8 bytes or less");
1399 return 0;
1400 }
Benjamin Peterson5c6d7872009-02-06 02:40:07 +00001401 if (protoNumber < 0 || protoNumber > 0xffff) {
1402 PyErr_SetString(
1403 PyExc_OverflowError,
1404 "getsockaddrarg: protoNumber must be 0-65535.");
1405 return 0;
1406 }
Thomas Wouters89f507f2006-12-13 04:49:30 +00001407 addr = (struct sockaddr_ll*)addr_ret;
1408 addr->sll_family = AF_PACKET;
1409 addr->sll_protocol = htons((short)protoNumber);
1410 addr->sll_ifindex = ifr.ifr_ifindex;
1411 addr->sll_pkttype = pkttype;
1412 addr->sll_hatype = hatype;
Andrew M. Kuchlingb8e17172004-07-10 23:39:35 +00001413 if (halen != 0) {
1414 memcpy(&addr->sll_addr, haddr, halen);
1415 }
1416 addr->sll_halen = halen;
Jeremy Hylton22308652001-02-02 03:23:09 +00001417 *len_ret = sizeof *addr;
1418 return 1;
1419 }
Guido van Rossum48a680c2001-03-02 06:34:14 +00001420#endif
1421
Christian Heimes043d6f62008-01-07 17:19:16 +00001422#ifdef HAVE_LINUX_TIPC_H
1423 case AF_TIPC:
1424 {
1425 unsigned int atype, v1, v2, v3;
1426 unsigned int scope = TIPC_CLUSTER_SCOPE;
1427 struct sockaddr_tipc *addr;
1428
1429 if (!PyTuple_Check(args)) {
1430 PyErr_Format(
1431 PyExc_TypeError,
1432 "getsockaddrarg: "
1433 "AF_TIPC address must be tuple, not %.500s",
1434 Py_TYPE(args)->tp_name);
1435 return 0;
1436 }
1437
1438 if (!PyArg_ParseTuple(args,
1439 "IIII|I;Invalid TIPC address format",
1440 &atype, &v1, &v2, &v3, &scope))
1441 return 0;
1442
1443 addr = (struct sockaddr_tipc *) addr_ret;
1444 memset(addr, 0, sizeof(struct sockaddr_tipc));
1445
1446 addr->family = AF_TIPC;
1447 addr->scope = scope;
1448 addr->addrtype = atype;
1449
1450 if (atype == TIPC_ADDR_NAMESEQ) {
1451 addr->addr.nameseq.type = v1;
1452 addr->addr.nameseq.lower = v2;
1453 addr->addr.nameseq.upper = v3;
1454 } else if (atype == TIPC_ADDR_NAME) {
1455 addr->addr.name.name.type = v1;
1456 addr->addr.name.name.instance = v2;
1457 } else if (atype == TIPC_ADDR_ID) {
1458 addr->addr.id.node = v1;
1459 addr->addr.id.ref = v2;
1460 } else {
1461 /* Shouldn't happen */
1462 PyErr_SetString(PyExc_TypeError, "Invalid address type");
1463 return 0;
1464 }
1465
1466 *len_ret = sizeof(*addr);
1467
1468 return 1;
1469 }
1470#endif
1471
Guido van Rossum30a685f1991-06-27 15:51:29 +00001472 /* More cases here... */
1473
1474 default:
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001475 PyErr_SetString(socket_error, "getsockaddrarg: bad family");
Guido van Rossum30a685f1991-06-27 15:51:29 +00001476 return 0;
1477
1478 }
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001479}
1480
Guido van Rossum30a685f1991-06-27 15:51:29 +00001481
Guido van Rossum48a680c2001-03-02 06:34:14 +00001482/* Get the address length according to the socket object's address family.
Guido van Rossum710e1df1992-06-12 10:39:36 +00001483 Return 1 if the family is known, 0 otherwise. The length is returned
1484 through len_ret. */
1485
1486static int
Peter Schneider-Kamp618e25d2000-07-11 23:00:12 +00001487getsockaddrlen(PySocketSockObject *s, socklen_t *len_ret)
Guido van Rossum710e1df1992-06-12 10:39:36 +00001488{
1489 switch (s->sock_family) {
1490
Andrew MacIntyred12dfbb2004-04-04 07:13:49 +00001491#if defined(AF_UNIX)
Guido van Rossum710e1df1992-06-12 10:39:36 +00001492 case AF_UNIX:
1493 {
1494 *len_ret = sizeof (struct sockaddr_un);
1495 return 1;
1496 }
Guido van Rossumb6775db1994-08-01 11:34:53 +00001497#endif /* AF_UNIX */
Martin v. Löwis11017b12006-01-14 18:12:57 +00001498#if defined(AF_NETLINK)
1499 case AF_NETLINK:
1500 {
1501 *len_ret = sizeof (struct sockaddr_nl);
1502 return 1;
1503 }
1504#endif
Guido van Rossum710e1df1992-06-12 10:39:36 +00001505
1506 case AF_INET:
1507 {
1508 *len_ret = sizeof (struct sockaddr_in);
1509 return 1;
1510 }
1511
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00001512#ifdef ENABLE_IPV6
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00001513 case AF_INET6:
1514 {
1515 *len_ret = sizeof (struct sockaddr_in6);
1516 return 1;
1517 }
1518#endif
1519
Hye-Shik Chang81268602004-02-02 06:05:24 +00001520#ifdef USE_BLUETOOTH
Martin v. Löwis12af0482004-01-31 12:34:17 +00001521 case AF_BLUETOOTH:
1522 {
1523 switch(s->sock_proto)
1524 {
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001525
1526 case BTPROTO_L2CAP:
1527 *len_ret = sizeof (struct sockaddr_l2);
1528 return 1;
1529 case BTPROTO_RFCOMM:
1530 *len_ret = sizeof (struct sockaddr_rc);
1531 return 1;
Thomas Wouterscf297e42007-02-23 15:07:44 +00001532 case BTPROTO_HCI:
1533 *len_ret = sizeof (struct sockaddr_hci);
1534 return 1;
Hye-Shik Chang81268602004-02-02 06:05:24 +00001535#if !defined(__FreeBSD__)
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001536 case BTPROTO_SCO:
1537 *len_ret = sizeof (struct sockaddr_sco);
1538 return 1;
Hye-Shik Chang81268602004-02-02 06:05:24 +00001539#endif
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001540 default:
1541 PyErr_SetString(socket_error, "getsockaddrlen: "
1542 "unknown BT protocol");
1543 return 0;
1544
Martin v. Löwis12af0482004-01-31 12:34:17 +00001545 }
1546 }
1547#endif
1548
Martin v. Löwis1ba3fd52001-08-10 20:29:40 +00001549#ifdef HAVE_NETPACKET_PACKET_H
Jeremy Hylton22308652001-02-02 03:23:09 +00001550 case AF_PACKET:
1551 {
1552 *len_ret = sizeof (struct sockaddr_ll);
1553 return 1;
1554 }
1555#endif
Guido van Rossum48a680c2001-03-02 06:34:14 +00001556
Christian Heimes043d6f62008-01-07 17:19:16 +00001557#ifdef HAVE_LINUX_TIPC_H
1558 case AF_TIPC:
1559 {
1560 *len_ret = sizeof (struct sockaddr_tipc);
1561 return 1;
1562 }
1563#endif
1564
Guido van Rossum710e1df1992-06-12 10:39:36 +00001565 /* More cases here... */
1566
1567 default:
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001568 PyErr_SetString(socket_error, "getsockaddrlen: bad family");
Guido van Rossum710e1df1992-06-12 10:39:36 +00001569 return 0;
1570
1571 }
1572}
1573
1574
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00001575/* s._accept() -> (fd, address) */
Guido van Rossum30a685f1991-06-27 15:51:29 +00001576
Guido van Rossum73624e91994-10-10 17:59:00 +00001577static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001578sock_accept(PySocketSockObject *s)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001579{
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001580 sock_addr_t addrbuf;
Guido van Rossum7d0a8262007-05-21 23:13:11 +00001581 SOCKET_T newfd = INVALID_SOCKET;
Guido van Rossumff3ab422000-04-24 15:16:03 +00001582 socklen_t addrlen;
Barry Warsaw752300b1997-01-03 17:18:10 +00001583 PyObject *sock = NULL;
1584 PyObject *addr = NULL;
1585 PyObject *res = NULL;
Raymond Hettingeref7343c2003-06-29 03:08:05 +00001586 int timeout;
Barry Warsaw752300b1997-01-03 17:18:10 +00001587
Guido van Rossum710e1df1992-06-12 10:39:36 +00001588 if (!getsockaddrlen(s, &addrlen))
1589 return NULL;
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001590 memset(&addrbuf, 0, addrlen);
Guido van Rossum67f7a382002-06-06 21:08:16 +00001591
Neal Norwitz082b2df2006-02-07 07:04:46 +00001592 if (!IS_SELECTABLE(s))
1593 return select_error();
1594
Guido van Rossum73624e91994-10-10 17:59:00 +00001595 Py_BEGIN_ALLOW_THREADS
Raymond Hettingeref7343c2003-06-29 03:08:05 +00001596 timeout = internal_select(s, 0);
1597 if (!timeout)
Thomas Wouters89f507f2006-12-13 04:49:30 +00001598 newfd = accept(s->sock_fd, SAS2SA(&addrbuf), &addrlen);
Guido van Rossum73624e91994-10-10 17:59:00 +00001599 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00001600
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001601 if (timeout == 1) {
Raymond Hettingeref7343c2003-06-29 03:08:05 +00001602 PyErr_SetString(socket_timeout, "timed out");
1603 return NULL;
1604 }
1605
Fred Drakea04eaad2000-06-30 02:46:07 +00001606 if (newfd == INVALID_SOCKET)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00001607 return s->errorhandler();
Barry Warsaw752300b1997-01-03 17:18:10 +00001608
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00001609 sock = PyLong_FromSocket_t(newfd);
Barry Warsaw752300b1997-01-03 17:18:10 +00001610 if (sock == NULL) {
Guido van Rossum2dd8ddd2000-04-21 20:33:00 +00001611 SOCKETCLOSE(newfd);
Barry Warsaw752300b1997-01-03 17:18:10 +00001612 goto finally;
1613 }
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00001614
Thomas Wouters89f507f2006-12-13 04:49:30 +00001615 addr = makesockaddr(s->sock_fd, SAS2SA(&addrbuf),
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001616 addrlen, s->sock_proto);
Jeremy Hyltondbfb6622001-02-02 19:55:17 +00001617 if (addr == NULL)
Barry Warsaw752300b1997-01-03 17:18:10 +00001618 goto finally;
1619
Raymond Hettinger8ae46892003-10-12 19:09:37 +00001620 res = PyTuple_Pack(2, sock, addr);
Barry Warsaw752300b1997-01-03 17:18:10 +00001621
Guido van Rossum67f7a382002-06-06 21:08:16 +00001622finally:
Guido van Rossum73624e91994-10-10 17:59:00 +00001623 Py_XDECREF(sock);
1624 Py_XDECREF(addr);
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001625 return res;
1626}
1627
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001628PyDoc_STRVAR(accept_doc,
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00001629"_accept() -> (integer, address info)\n\
Guido van Rossum82a5c661998-07-07 20:45:43 +00001630\n\
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00001631Wait for an incoming connection. Return a new socket file descriptor\n\
1632representing the connection, and the address of the client.\n\
1633For IP sockets, the address info is a pair (hostaddr, port).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001634
Guido van Rossum11ba0942002-06-13 15:07:44 +00001635/* s.setblocking(flag) method. Argument:
1636 False -- non-blocking mode; same as settimeout(0)
1637 True -- blocking mode; same as settimeout(None)
1638*/
Guido van Rossume4485b01994-09-07 14:32:49 +00001639
Guido van Rossum73624e91994-10-10 17:59:00 +00001640static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001641sock_setblocking(PySocketSockObject *s, PyObject *arg)
Guido van Rossume4485b01994-09-07 14:32:49 +00001642{
1643 int block;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001644
Christian Heimes217cfd12007-12-02 14:31:20 +00001645 block = PyLong_AsLong(arg);
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00001646 if (block == -1 && PyErr_Occurred())
Guido van Rossume4485b01994-09-07 14:32:49 +00001647 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001648
Guido van Rossum11ba0942002-06-13 15:07:44 +00001649 s->sock_timeout = block ? -1.0 : 0.0;
Guido van Rossume8008f02002-06-07 03:36:20 +00001650 internal_setblocking(s, block);
Guido van Rossume4485b01994-09-07 14:32:49 +00001651
Guido van Rossum73624e91994-10-10 17:59:00 +00001652 Py_INCREF(Py_None);
1653 return Py_None;
Guido van Rossume4485b01994-09-07 14:32:49 +00001654}
Guido van Rossume4485b01994-09-07 14:32:49 +00001655
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001656PyDoc_STRVAR(setblocking_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001657"setblocking(flag)\n\
1658\n\
1659Set the socket to blocking (flag is true) or non-blocking (false).\n\
Guido van Rossum11ba0942002-06-13 15:07:44 +00001660setblocking(True) is equivalent to settimeout(None);\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001661setblocking(False) is equivalent to settimeout(0.0).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001662
Guido van Rossum11ba0942002-06-13 15:07:44 +00001663/* s.settimeout(timeout) method. Argument:
1664 None -- no timeout, blocking mode; same as setblocking(True)
1665 0.0 -- non-blocking mode; same as setblocking(False)
1666 > 0 -- timeout mode; operations time out after timeout seconds
1667 < 0 -- illegal; raises an exception
1668*/
Guido van Rossum67f7a382002-06-06 21:08:16 +00001669static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001670sock_settimeout(PySocketSockObject *s, PyObject *arg)
Guido van Rossum67f7a382002-06-06 21:08:16 +00001671{
Guido van Rossum11ba0942002-06-13 15:07:44 +00001672 double timeout;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001673
1674 if (arg == Py_None)
Guido van Rossum11ba0942002-06-13 15:07:44 +00001675 timeout = -1.0;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001676 else {
Guido van Rossum11ba0942002-06-13 15:07:44 +00001677 timeout = PyFloat_AsDouble(arg);
1678 if (timeout < 0.0) {
Guido van Rossum67f7a382002-06-06 21:08:16 +00001679 if (!PyErr_Occurred())
1680 PyErr_SetString(PyExc_ValueError,
Guido van Rossum11ba0942002-06-13 15:07:44 +00001681 "Timeout value out of range");
Guido van Rossum67f7a382002-06-06 21:08:16 +00001682 return NULL;
1683 }
1684 }
1685
Guido van Rossum11ba0942002-06-13 15:07:44 +00001686 s->sock_timeout = timeout;
1687 internal_setblocking(s, timeout < 0.0);
Guido van Rossum67f7a382002-06-06 21:08:16 +00001688
1689 Py_INCREF(Py_None);
1690 return Py_None;
1691}
1692
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001693PyDoc_STRVAR(settimeout_doc,
Guido van Rossum3eede5a2002-06-07 02:08:35 +00001694"settimeout(timeout)\n\
Guido van Rossum67f7a382002-06-06 21:08:16 +00001695\n\
Guido van Rossum11ba0942002-06-13 15:07:44 +00001696Set a timeout on socket operations. 'timeout' can be a float,\n\
1697giving in seconds, or None. Setting a timeout of None disables\n\
1698the timeout feature and is equivalent to setblocking(1).\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001699Setting a timeout of zero is the same as setblocking(0).");
Guido van Rossum67f7a382002-06-06 21:08:16 +00001700
Guido van Rossum3eede5a2002-06-07 02:08:35 +00001701/* s.gettimeout() method.
1702 Returns the timeout associated with a socket. */
Guido van Rossum67f7a382002-06-06 21:08:16 +00001703static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001704sock_gettimeout(PySocketSockObject *s)
Guido van Rossum67f7a382002-06-06 21:08:16 +00001705{
1706 if (s->sock_timeout < 0.0) {
1707 Py_INCREF(Py_None);
1708 return Py_None;
1709 }
1710 else
1711 return PyFloat_FromDouble(s->sock_timeout);
1712}
1713
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001714PyDoc_STRVAR(gettimeout_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00001715"gettimeout() -> timeout\n\
Guido van Rossum67f7a382002-06-06 21:08:16 +00001716\n\
1717Returns the timeout in floating seconds associated with socket \n\
1718operations. A timeout of None indicates that timeouts on socket \n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001719operations are disabled.");
Guido van Rossume4485b01994-09-07 14:32:49 +00001720
Guido van Rossumaee08791992-09-08 09:05:33 +00001721/* s.setsockopt() method.
1722 With an integer third argument, sets an integer option.
1723 With a string third argument, sets an option from a buffer;
1724 use optional built-in module 'struct' to encode the string. */
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001725
Guido van Rossum73624e91994-10-10 17:59:00 +00001726static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001727sock_setsockopt(PySocketSockObject *s, PyObject *args)
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001728{
1729 int level;
1730 int optname;
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001731 int res;
Guido van Rossumaee08791992-09-08 09:05:33 +00001732 char *buf;
1733 int buflen;
1734 int flag;
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001735
Guido van Rossum27fc3c02000-03-24 20:56:56 +00001736 if (PyArg_ParseTuple(args, "iii:setsockopt",
1737 &level, &optname, &flag)) {
Guido van Rossumaee08791992-09-08 09:05:33 +00001738 buf = (char *) &flag;
1739 buflen = sizeof flag;
1740 }
1741 else {
Guido van Rossum73624e91994-10-10 17:59:00 +00001742 PyErr_Clear();
Guido van Rossum8a392d72007-11-21 22:09:45 +00001743 if (!PyArg_ParseTuple(args, "iiy#:setsockopt",
Guido van Rossum27fc3c02000-03-24 20:56:56 +00001744 &level, &optname, &buf, &buflen))
Guido van Rossumaee08791992-09-08 09:05:33 +00001745 return NULL;
1746 }
Thomas Wouters334fb892000-07-25 12:56:38 +00001747 res = setsockopt(s->sock_fd, level, optname, (void *)buf, buflen);
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001748 if (res < 0)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00001749 return s->errorhandler();
Guido van Rossum73624e91994-10-10 17:59:00 +00001750 Py_INCREF(Py_None);
1751 return Py_None;
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001752}
1753
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001754PyDoc_STRVAR(setsockopt_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001755"setsockopt(level, option, value)\n\
1756\n\
1757Set a socket option. See the Unix manual for level and option.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001758The value argument can either be an integer or a string.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001759
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001760
Guido van Rossumaee08791992-09-08 09:05:33 +00001761/* s.getsockopt() method.
1762 With two arguments, retrieves an integer option.
1763 With a third integer argument, retrieves a string buffer of that size;
1764 use optional built-in module 'struct' to decode the string. */
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001765
Guido van Rossum73624e91994-10-10 17:59:00 +00001766static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001767sock_getsockopt(PySocketSockObject *s, PyObject *args)
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001768{
1769 int level;
1770 int optname;
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001771 int res;
Guido van Rossum73624e91994-10-10 17:59:00 +00001772 PyObject *buf;
Guido van Rossumff3ab422000-04-24 15:16:03 +00001773 socklen_t buflen = 0;
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001774
Guido van Rossum27fc3c02000-03-24 20:56:56 +00001775 if (!PyArg_ParseTuple(args, "ii|i:getsockopt",
1776 &level, &optname, &buflen))
Guido van Rossumbe32c891996-06-20 16:25:29 +00001777 return NULL;
Guido van Rossum48a680c2001-03-02 06:34:14 +00001778
Guido van Rossumbe32c891996-06-20 16:25:29 +00001779 if (buflen == 0) {
Guido van Rossumaee08791992-09-08 09:05:33 +00001780 int flag = 0;
Guido van Rossumff3ab422000-04-24 15:16:03 +00001781 socklen_t flagsize = sizeof flag;
Guido van Rossumb376a4a1993-11-23 17:53:17 +00001782 res = getsockopt(s->sock_fd, level, optname,
Thomas Wouters334fb892000-07-25 12:56:38 +00001783 (void *)&flag, &flagsize);
Guido van Rossumaee08791992-09-08 09:05:33 +00001784 if (res < 0)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00001785 return s->errorhandler();
Christian Heimes217cfd12007-12-02 14:31:20 +00001786 return PyLong_FromLong(flag);
Guido van Rossumaee08791992-09-08 09:05:33 +00001787 }
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001788#ifdef __VMS
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001789 /* socklen_t is unsigned so no negative test is needed,
1790 test buflen == 0 is previously done */
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001791 if (buflen > 1024) {
1792#else
Guido van Rossumaee08791992-09-08 09:05:33 +00001793 if (buflen <= 0 || buflen > 1024) {
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001794#endif
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001795 PyErr_SetString(socket_error,
Barry Warsaw752300b1997-01-03 17:18:10 +00001796 "getsockopt buflen out of range");
Guido van Rossumaee08791992-09-08 09:05:33 +00001797 return NULL;
1798 }
Christian Heimes72b710a2008-05-26 13:28:38 +00001799 buf = PyBytes_FromStringAndSize((char *)NULL, buflen);
Guido van Rossumaee08791992-09-08 09:05:33 +00001800 if (buf == NULL)
1801 return NULL;
Guido van Rossumb376a4a1993-11-23 17:53:17 +00001802 res = getsockopt(s->sock_fd, level, optname,
Christian Heimes72b710a2008-05-26 13:28:38 +00001803 (void *)PyBytes_AS_STRING(buf), &buflen);
Guido van Rossumaee08791992-09-08 09:05:33 +00001804 if (res < 0) {
Guido van Rossum73624e91994-10-10 17:59:00 +00001805 Py_DECREF(buf);
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00001806 return s->errorhandler();
Guido van Rossumaee08791992-09-08 09:05:33 +00001807 }
Christian Heimes72b710a2008-05-26 13:28:38 +00001808 _PyBytes_Resize(&buf, buflen);
Guido van Rossumaee08791992-09-08 09:05:33 +00001809 return buf;
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001810}
1811
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001812PyDoc_STRVAR(getsockopt_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001813"getsockopt(level, option[, buffersize]) -> value\n\
1814\n\
1815Get a socket option. See the Unix manual for level and option.\n\
1816If a nonzero buffersize argument is given, the return value is a\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001817string of that length; otherwise it is an integer.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001818
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001819
Fred Drake728819a2000-07-01 03:40:12 +00001820/* s.bind(sockaddr) method */
Guido van Rossum30a685f1991-06-27 15:51:29 +00001821
Guido van Rossum73624e91994-10-10 17:59:00 +00001822static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001823sock_bind(PySocketSockObject *s, PyObject *addro)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001824{
Thomas Wouters89f507f2006-12-13 04:49:30 +00001825 sock_addr_t addrbuf;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001826 int addrlen;
Guido van Rossumff4949e1992-08-05 19:58:53 +00001827 int res;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00001828
Thomas Wouters89f507f2006-12-13 04:49:30 +00001829 if (!getsockaddrarg(s, addro, SAS2SA(&addrbuf), &addrlen))
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001830 return NULL;
Guido van Rossum73624e91994-10-10 17:59:00 +00001831 Py_BEGIN_ALLOW_THREADS
Thomas Wouters89f507f2006-12-13 04:49:30 +00001832 res = bind(s->sock_fd, SAS2SA(&addrbuf), addrlen);
Guido van Rossum73624e91994-10-10 17:59:00 +00001833 Py_END_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +00001834 if (res < 0)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00001835 return s->errorhandler();
Guido van Rossum73624e91994-10-10 17:59:00 +00001836 Py_INCREF(Py_None);
1837 return Py_None;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001838}
1839
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001840PyDoc_STRVAR(bind_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001841"bind(address)\n\
1842\n\
1843Bind the socket to a local address. For IP sockets, the address is a\n\
Jeremy Hylton22308652001-02-02 03:23:09 +00001844pair (host, port); the host must refer to the local host. For raw packet\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001845sockets the address is a tuple (ifname, proto [,pkttype [,hatype]])");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001846
Guido van Rossum30a685f1991-06-27 15:51:29 +00001847
1848/* s.close() method.
1849 Set the file descriptor to -1 so operations tried subsequently
1850 will surely fail. */
1851
Guido van Rossum73624e91994-10-10 17:59:00 +00001852static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001853sock_close(PySocketSockObject *s)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001854{
Guido van Rossum20d3fc02000-12-18 22:23:44 +00001855 SOCKET_T fd;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00001856
Guido van Rossum20d3fc02000-12-18 22:23:44 +00001857 if ((fd = s->sock_fd) != -1) {
1858 s->sock_fd = -1;
Guido van Rossum08481461996-10-12 14:07:22 +00001859 Py_BEGIN_ALLOW_THREADS
Guido van Rossum20d3fc02000-12-18 22:23:44 +00001860 (void) SOCKETCLOSE(fd);
Guido van Rossum08481461996-10-12 14:07:22 +00001861 Py_END_ALLOW_THREADS
1862 }
Guido van Rossum73624e91994-10-10 17:59:00 +00001863 Py_INCREF(Py_None);
1864 return Py_None;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001865}
1866
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001867PyDoc_STRVAR(close_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001868"close()\n\
1869\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001870Close the socket. It cannot be used after this call.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001871
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001872static int
Raymond Hettingeref7343c2003-06-29 03:08:05 +00001873internal_connect(PySocketSockObject *s, struct sockaddr *addr, int addrlen,
1874 int *timeoutp)
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001875{
Raymond Hettingeref7343c2003-06-29 03:08:05 +00001876 int res, timeout;
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001877
Raymond Hettingeref7343c2003-06-29 03:08:05 +00001878 timeout = 0;
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001879 res = connect(s->sock_fd, addr, addrlen);
1880
1881#ifdef MS_WINDOWS
1882
1883 if (s->sock_timeout > 0.0) {
Neal Norwitz082b2df2006-02-07 07:04:46 +00001884 if (res < 0 && WSAGetLastError() == WSAEWOULDBLOCK &&
1885 IS_SELECTABLE(s)) {
Guido van Rossumb76bdf82003-02-19 17:50:16 +00001886 /* This is a mess. Best solution: trust select */
1887 fd_set fds;
Mark Hammonda57ec932004-08-03 05:06:26 +00001888 fd_set fds_exc;
Guido van Rossumb76bdf82003-02-19 17:50:16 +00001889 struct timeval tv;
1890 tv.tv_sec = (int)s->sock_timeout;
1891 tv.tv_usec = (int)((s->sock_timeout - tv.tv_sec) * 1e6);
1892 FD_ZERO(&fds);
1893 FD_SET(s->sock_fd, &fds);
Mark Hammonda57ec932004-08-03 05:06:26 +00001894 FD_ZERO(&fds_exc);
1895 FD_SET(s->sock_fd, &fds_exc);
1896 res = select(s->sock_fd+1, NULL, &fds, &fds_exc, &tv);
Raymond Hettingeref7343c2003-06-29 03:08:05 +00001897 if (res == 0) {
Guido van Rossumb76bdf82003-02-19 17:50:16 +00001898 res = WSAEWOULDBLOCK;
Raymond Hettingeref7343c2003-06-29 03:08:05 +00001899 timeout = 1;
Mark Hammonda57ec932004-08-03 05:06:26 +00001900 } else if (res > 0) {
1901 if (FD_ISSET(s->sock_fd, &fds))
Sean Reifscheider54cf12b2007-09-17 17:55:36 +00001902 /* The socket is in the writable set - this
Mark Hammonda57ec932004-08-03 05:06:26 +00001903 means connected */
1904 res = 0;
1905 else {
1906 /* As per MS docs, we need to call getsockopt()
1907 to get the underlying error */
1908 int res_size = sizeof res;
1909 /* It must be in the exception set */
1910 assert(FD_ISSET(s->sock_fd, &fds_exc));
Thomas Wouters477c8d52006-05-27 19:21:47 +00001911 if (0 == getsockopt(s->sock_fd, SOL_SOCKET, SO_ERROR,
Mark Hammonda57ec932004-08-03 05:06:26 +00001912 (char *)&res, &res_size))
1913 /* getsockopt also clears WSAGetLastError,
1914 so reset it back. */
1915 WSASetLastError(res);
1916 else
1917 res = WSAGetLastError();
1918 }
1919 }
Guido van Rossumb76bdf82003-02-19 17:50:16 +00001920 /* else if (res < 0) an error occurred */
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001921 }
1922 }
1923
1924 if (res < 0)
1925 res = WSAGetLastError();
1926
1927#else
1928
1929 if (s->sock_timeout > 0.0) {
Christian Heimesfdab48e2008-01-20 09:06:41 +00001930 if (res < 0 && errno == EINPROGRESS && IS_SELECTABLE(s)) {
1931 timeout = internal_select(s, 1);
1932 if (timeout == 0) {
1933 /* Bug #1019808: in case of an EINPROGRESS,
1934 use getsockopt(SO_ERROR) to get the real
1935 error. */
1936 socklen_t res_size = sizeof res;
1937 (void)getsockopt(s->sock_fd, SOL_SOCKET,
1938 SO_ERROR, &res, &res_size);
1939 if (res == EISCONN)
1940 res = 0;
1941 errno = res;
1942 }
1943 else if (timeout == -1) {
1944 res = errno; /* had error */
1945 }
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001946 else
1947 res = EWOULDBLOCK; /* timed out */
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001948 }
1949 }
1950
1951 if (res < 0)
1952 res = errno;
1953
1954#endif
Raymond Hettingeref7343c2003-06-29 03:08:05 +00001955 *timeoutp = timeout;
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001956
1957 return res;
1958}
Guido van Rossum30a685f1991-06-27 15:51:29 +00001959
Fred Drake728819a2000-07-01 03:40:12 +00001960/* s.connect(sockaddr) method */
Guido van Rossum30a685f1991-06-27 15:51:29 +00001961
Guido van Rossum73624e91994-10-10 17:59:00 +00001962static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001963sock_connect(PySocketSockObject *s, PyObject *addro)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001964{
Thomas Wouters89f507f2006-12-13 04:49:30 +00001965 sock_addr_t addrbuf;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001966 int addrlen;
Guido van Rossumff4949e1992-08-05 19:58:53 +00001967 int res;
Raymond Hettingeref7343c2003-06-29 03:08:05 +00001968 int timeout;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00001969
Thomas Wouters89f507f2006-12-13 04:49:30 +00001970 if (!getsockaddrarg(s, addro, SAS2SA(&addrbuf), &addrlen))
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001971 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001972
Guido van Rossum73624e91994-10-10 17:59:00 +00001973 Py_BEGIN_ALLOW_THREADS
Thomas Wouters89f507f2006-12-13 04:49:30 +00001974 res = internal_connect(s, SAS2SA(&addrbuf), addrlen, &timeout);
Guido van Rossum11ba0942002-06-13 15:07:44 +00001975 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00001976
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001977 if (timeout == 1) {
Raymond Hettingeref7343c2003-06-29 03:08:05 +00001978 PyErr_SetString(socket_timeout, "timed out");
1979 return NULL;
1980 }
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001981 if (res != 0)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00001982 return s->errorhandler();
Guido van Rossum73624e91994-10-10 17:59:00 +00001983 Py_INCREF(Py_None);
1984 return Py_None;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001985}
1986
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001987PyDoc_STRVAR(connect_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001988"connect(address)\n\
1989\n\
1990Connect the socket to a remote address. For IP sockets, the address\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001991is a pair (host, port).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001992
Guido van Rossum30a685f1991-06-27 15:51:29 +00001993
Fred Drake728819a2000-07-01 03:40:12 +00001994/* s.connect_ex(sockaddr) method */
Guido van Rossumfc4255d1997-11-19 18:57:13 +00001995
1996static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001997sock_connect_ex(PySocketSockObject *s, PyObject *addro)
Guido van Rossumfc4255d1997-11-19 18:57:13 +00001998{
Thomas Wouters89f507f2006-12-13 04:49:30 +00001999 sock_addr_t addrbuf;
Guido van Rossumfc4255d1997-11-19 18:57:13 +00002000 int addrlen;
2001 int res;
Raymond Hettingeref7343c2003-06-29 03:08:05 +00002002 int timeout;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00002003
Thomas Wouters89f507f2006-12-13 04:49:30 +00002004 if (!getsockaddrarg(s, addro, SAS2SA(&addrbuf), &addrlen))
Guido van Rossumfc4255d1997-11-19 18:57:13 +00002005 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002006
Guido van Rossumfc4255d1997-11-19 18:57:13 +00002007 Py_BEGIN_ALLOW_THREADS
Thomas Wouters89f507f2006-12-13 04:49:30 +00002008 res = internal_connect(s, SAS2SA(&addrbuf), addrlen, &timeout);
Guido van Rossumfc4255d1997-11-19 18:57:13 +00002009 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00002010
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002011 /* Signals are not errors (though they may raise exceptions). Adapted
2012 from PyErr_SetFromErrnoWithFilenameObject(). */
2013#ifdef EINTR
2014 if (res == EINTR && PyErr_CheckSignals())
2015 return NULL;
2016#endif
2017
Christian Heimes217cfd12007-12-02 14:31:20 +00002018 return PyLong_FromLong((long) res);
Guido van Rossumfc4255d1997-11-19 18:57:13 +00002019}
2020
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002021PyDoc_STRVAR(connect_ex_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00002022"connect_ex(address) -> errno\n\
Guido van Rossum82a5c661998-07-07 20:45:43 +00002023\n\
2024This is like connect(address), but returns an error code (the errno value)\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002025instead of raising an exception when an error occurs.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002026
Guido van Rossumfc4255d1997-11-19 18:57:13 +00002027
Guido van Rossumed233a51992-06-23 09:07:03 +00002028/* s.fileno() method */
2029
Guido van Rossum73624e91994-10-10 17:59:00 +00002030static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002031sock_fileno(PySocketSockObject *s)
Guido van Rossumed233a51992-06-23 09:07:03 +00002032{
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00002033 return PyLong_FromSocket_t(s->sock_fd);
Guido van Rossumed233a51992-06-23 09:07:03 +00002034}
2035
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002036PyDoc_STRVAR(fileno_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002037"fileno() -> integer\n\
2038\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002039Return the integer file descriptor of the socket.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002040
Guido van Rossumed233a51992-06-23 09:07:03 +00002041
Guido van Rossumc89705d1992-11-26 08:54:07 +00002042/* s.getsockname() method */
2043
Guido van Rossum73624e91994-10-10 17:59:00 +00002044static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002045sock_getsockname(PySocketSockObject *s)
Guido van Rossumc89705d1992-11-26 08:54:07 +00002046{
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00002047 sock_addr_t addrbuf;
Guido van Rossumff3ab422000-04-24 15:16:03 +00002048 int res;
2049 socklen_t addrlen;
2050
Guido van Rossumc89705d1992-11-26 08:54:07 +00002051 if (!getsockaddrlen(s, &addrlen))
2052 return NULL;
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00002053 memset(&addrbuf, 0, addrlen);
Guido van Rossum73624e91994-10-10 17:59:00 +00002054 Py_BEGIN_ALLOW_THREADS
Thomas Wouters89f507f2006-12-13 04:49:30 +00002055 res = getsockname(s->sock_fd, SAS2SA(&addrbuf), &addrlen);
Guido van Rossum73624e91994-10-10 17:59:00 +00002056 Py_END_ALLOW_THREADS
Guido van Rossumc89705d1992-11-26 08:54:07 +00002057 if (res < 0)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00002058 return s->errorhandler();
Thomas Wouters89f507f2006-12-13 04:49:30 +00002059 return makesockaddr(s->sock_fd, SAS2SA(&addrbuf), addrlen,
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00002060 s->sock_proto);
Guido van Rossumc89705d1992-11-26 08:54:07 +00002061}
2062
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002063PyDoc_STRVAR(getsockname_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002064"getsockname() -> address info\n\
2065\n\
2066Return the address of the local endpoint. For IP sockets, the address\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002067info is a pair (hostaddr, port).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002068
Guido van Rossumc89705d1992-11-26 08:54:07 +00002069
Guido van Rossumb6775db1994-08-01 11:34:53 +00002070#ifdef HAVE_GETPEERNAME /* Cray APP doesn't have this :-( */
Guido van Rossumc89705d1992-11-26 08:54:07 +00002071/* s.getpeername() method */
2072
Guido van Rossum73624e91994-10-10 17:59:00 +00002073static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002074sock_getpeername(PySocketSockObject *s)
Guido van Rossumc89705d1992-11-26 08:54:07 +00002075{
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00002076 sock_addr_t addrbuf;
Guido van Rossumff3ab422000-04-24 15:16:03 +00002077 int res;
2078 socklen_t addrlen;
2079
Guido van Rossumc89705d1992-11-26 08:54:07 +00002080 if (!getsockaddrlen(s, &addrlen))
2081 return NULL;
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00002082 memset(&addrbuf, 0, addrlen);
Guido van Rossum73624e91994-10-10 17:59:00 +00002083 Py_BEGIN_ALLOW_THREADS
Thomas Wouters89f507f2006-12-13 04:49:30 +00002084 res = getpeername(s->sock_fd, SAS2SA(&addrbuf), &addrlen);
Guido van Rossum73624e91994-10-10 17:59:00 +00002085 Py_END_ALLOW_THREADS
Guido van Rossumc89705d1992-11-26 08:54:07 +00002086 if (res < 0)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00002087 return s->errorhandler();
Thomas Wouters89f507f2006-12-13 04:49:30 +00002088 return makesockaddr(s->sock_fd, SAS2SA(&addrbuf), addrlen,
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00002089 s->sock_proto);
Guido van Rossumc89705d1992-11-26 08:54:07 +00002090}
Guido van Rossum82a5c661998-07-07 20:45:43 +00002091
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002092PyDoc_STRVAR(getpeername_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002093"getpeername() -> address info\n\
2094\n\
2095Return the address of the remote endpoint. For IP sockets, the address\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002096info is a pair (hostaddr, port).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002097
Guido van Rossumb6775db1994-08-01 11:34:53 +00002098#endif /* HAVE_GETPEERNAME */
Guido van Rossumc89705d1992-11-26 08:54:07 +00002099
2100
Guido van Rossum30a685f1991-06-27 15:51:29 +00002101/* s.listen(n) method */
2102
Guido van Rossum73624e91994-10-10 17:59:00 +00002103static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002104sock_listen(PySocketSockObject *s, PyObject *arg)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002105{
2106 int backlog;
Guido van Rossumff4949e1992-08-05 19:58:53 +00002107 int res;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00002108
Christian Heimes217cfd12007-12-02 14:31:20 +00002109 backlog = PyLong_AsLong(arg);
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00002110 if (backlog == -1 && PyErr_Occurred())
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002111 return NULL;
Guido van Rossum73624e91994-10-10 17:59:00 +00002112 Py_BEGIN_ALLOW_THREADS
Guido van Rossumb6775db1994-08-01 11:34:53 +00002113 if (backlog < 1)
2114 backlog = 1;
Guido van Rossumff4949e1992-08-05 19:58:53 +00002115 res = listen(s->sock_fd, backlog);
Guido van Rossum73624e91994-10-10 17:59:00 +00002116 Py_END_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +00002117 if (res < 0)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00002118 return s->errorhandler();
Guido van Rossum73624e91994-10-10 17:59:00 +00002119 Py_INCREF(Py_None);
2120 return Py_None;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002121}
2122
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002123PyDoc_STRVAR(listen_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002124"listen(backlog)\n\
2125\n\
2126Enable a server to accept connections. The backlog argument must be at\n\
2127least 1; it specifies the number of unaccepted connection that the system\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002128will allow before refusing new connections.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002129
2130
Thomas Wouters477c8d52006-05-27 19:21:47 +00002131/*
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002132 * This is the guts of the recv() and recv_into() methods, which reads into a
Thomas Wouters902d6eb2007-01-09 23:18:33 +00002133 * char buffer. If you have any inc/dec ref to do to the objects that contain
Thomas Wouters477c8d52006-05-27 19:21:47 +00002134 * the buffer, do it in the caller. This function returns the number of bytes
2135 * succesfully read. If there was an error, it returns -1. Note that it is
2136 * also possible that we return a number of bytes smaller than the request
2137 * bytes.
2138 */
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002139static ssize_t
Thomas Wouters477c8d52006-05-27 19:21:47 +00002140sock_recv_guts(PySocketSockObject *s, char* cbuf, int len, int flags)
2141{
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002142 ssize_t outlen = -1;
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002143 int timeout;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002144#ifdef __VMS
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002145 int remaining;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002146 char *read_buf;
2147#endif
2148
2149 if (!IS_SELECTABLE(s)) {
2150 select_error();
2151 return -1;
2152 }
Guido van Rossumc2de7c02007-08-03 22:27:51 +00002153 if (len == 0) {
2154 /* If 0 bytes were requested, do nothing. */
2155 return 0;
2156 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002157
2158#ifndef __VMS
2159 Py_BEGIN_ALLOW_THREADS
2160 timeout = internal_select(s, 0);
2161 if (!timeout)
2162 outlen = recv(s->sock_fd, cbuf, len, flags);
2163 Py_END_ALLOW_THREADS
2164
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002165 if (timeout == 1) {
Thomas Wouters477c8d52006-05-27 19:21:47 +00002166 PyErr_SetString(socket_timeout, "timed out");
2167 return -1;
2168 }
2169 if (outlen < 0) {
2170 /* Note: the call to errorhandler() ALWAYS indirectly returned
2171 NULL, so ignore its return value */
2172 s->errorhandler();
2173 return -1;
2174 }
2175#else
2176 read_buf = cbuf;
2177 remaining = len;
2178 while (remaining != 0) {
2179 unsigned int segment;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002180 int nread = -1;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002181
2182 segment = remaining /SEGMENT_SIZE;
2183 if (segment != 0) {
2184 segment = SEGMENT_SIZE;
2185 }
2186 else {
2187 segment = remaining;
2188 }
2189
2190 Py_BEGIN_ALLOW_THREADS
2191 timeout = internal_select(s, 0);
2192 if (!timeout)
2193 nread = recv(s->sock_fd, read_buf, segment, flags);
2194 Py_END_ALLOW_THREADS
2195
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002196 if (timeout == 1) {
Thomas Wouters477c8d52006-05-27 19:21:47 +00002197 PyErr_SetString(socket_timeout, "timed out");
2198 return -1;
2199 }
2200 if (nread < 0) {
2201 s->errorhandler();
2202 return -1;
2203 }
2204 if (nread != remaining) {
2205 read_buf += nread;
2206 break;
2207 }
2208
2209 remaining -= segment;
2210 read_buf += segment;
2211 }
2212 outlen = read_buf - cbuf;
2213#endif /* !__VMS */
2214
2215 return outlen;
2216}
2217
Guido van Rossum48a680c2001-03-02 06:34:14 +00002218
Guido van Rossumeb6b33a1993-05-25 09:38:27 +00002219/* s.recv(nbytes [,flags]) method */
Guido van Rossum30a685f1991-06-27 15:51:29 +00002220
Guido van Rossum73624e91994-10-10 17:59:00 +00002221static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002222sock_recv(PySocketSockObject *s, PyObject *args)
Guido van Rossum30a685f1991-06-27 15:51:29 +00002223{
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002224 int recvlen, flags = 0;
2225 ssize_t outlen;
Guido van Rossum73624e91994-10-10 17:59:00 +00002226 PyObject *buf;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002227
Thomas Wouters477c8d52006-05-27 19:21:47 +00002228 if (!PyArg_ParseTuple(args, "i|i:recv", &recvlen, &flags))
Guido van Rossumbe32c891996-06-20 16:25:29 +00002229 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002230
Thomas Wouters477c8d52006-05-27 19:21:47 +00002231 if (recvlen < 0) {
Martin v. Löwisfba64e12001-11-19 10:41:26 +00002232 PyErr_SetString(PyExc_ValueError,
Guido van Rossum11ba0942002-06-13 15:07:44 +00002233 "negative buffersize in recv");
Martin v. Löwisfba64e12001-11-19 10:41:26 +00002234 return NULL;
2235 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00002236
Thomas Wouters477c8d52006-05-27 19:21:47 +00002237 /* Allocate a new string. */
Christian Heimes72b710a2008-05-26 13:28:38 +00002238 buf = PyBytes_FromStringAndSize((char *) 0, recvlen);
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002239 if (buf == NULL)
2240 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002241
Thomas Wouters477c8d52006-05-27 19:21:47 +00002242 /* Call the guts */
Christian Heimes72b710a2008-05-26 13:28:38 +00002243 outlen = sock_recv_guts(s, PyBytes_AS_STRING(buf), recvlen, flags);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002244 if (outlen < 0) {
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002245 /* An error occurred, release the string and return an
Thomas Wouters477c8d52006-05-27 19:21:47 +00002246 error. */
Raymond Hettingeref7343c2003-06-29 03:08:05 +00002247 Py_DECREF(buf);
Raymond Hettingeref7343c2003-06-29 03:08:05 +00002248 return NULL;
2249 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002250 if (outlen != recvlen) {
2251 /* We did not read as many bytes as we anticipated, resize the
Kurt B. Kaisere4050372007-08-22 21:38:31 +00002252 string if possible and be successful. */
Christian Heimes72b710a2008-05-26 13:28:38 +00002253 _PyBytes_Resize(&buf, outlen);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002254 }
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00002255
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002256 return buf;
2257}
2258
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002259PyDoc_STRVAR(recv_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002260"recv(buffersize[, flags]) -> data\n\
2261\n\
2262Receive up to buffersize bytes from the socket. For the optional flags\n\
2263argument, see the Unix manual. When no data is available, block until\n\
2264at least one byte is available or until the remote end is closed. When\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002265the remote end is closed and all data is read, return the empty string.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002266
Guido van Rossum30a685f1991-06-27 15:51:29 +00002267
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002268/* s.recv_into(buffer, [nbytes [,flags]]) method */
Guido van Rossum30a685f1991-06-27 15:51:29 +00002269
Thomas Wouters477c8d52006-05-27 19:21:47 +00002270static PyObject*
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002271sock_recv_into(PySocketSockObject *s, PyObject *args, PyObject *kwds)
Thomas Wouters477c8d52006-05-27 19:21:47 +00002272{
2273 static char *kwlist[] = {"buffer", "nbytes", "flags", 0};
2274
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002275 int recvlen = 0, flags = 0;
2276 ssize_t readlen;
Martin v. Löwis423be952008-08-13 15:53:07 +00002277 Py_buffer pbuf;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002278 char *buf;
2279 int buflen;
2280
2281 /* Get the buffer's memory */
Martin v. Löwis423be952008-08-13 15:53:07 +00002282 if (!PyArg_ParseTupleAndKeywords(args, kwds, "w*|ii:recv_into", kwlist,
2283 &pbuf, &recvlen, &flags))
Thomas Wouters477c8d52006-05-27 19:21:47 +00002284 return NULL;
Martin v. Löwis423be952008-08-13 15:53:07 +00002285 buf = pbuf.buf;
2286 buflen = pbuf.len;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002287
2288 if (recvlen < 0) {
Martin v. Löwis423be952008-08-13 15:53:07 +00002289 PyBuffer_Release(&pbuf);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002290 PyErr_SetString(PyExc_ValueError,
Guido van Rossumd8faa362007-04-27 19:54:29 +00002291 "negative buffersize in recv_into");
Thomas Wouters477c8d52006-05-27 19:21:47 +00002292 return NULL;
2293 }
2294 if (recvlen == 0) {
2295 /* If nbytes was not specified, use the buffer's length */
2296 recvlen = buflen;
2297 }
2298
2299 /* Check if the buffer is large enough */
2300 if (buflen < recvlen) {
Martin v. Löwis423be952008-08-13 15:53:07 +00002301 PyBuffer_Release(&pbuf);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002302 PyErr_SetString(PyExc_ValueError,
2303 "buffer too small for requested bytes");
2304 return NULL;
2305 }
2306
2307 /* Call the guts */
2308 readlen = sock_recv_guts(s, buf, recvlen, flags);
2309 if (readlen < 0) {
2310 /* Return an error. */
Martin v. Löwis423be952008-08-13 15:53:07 +00002311 PyBuffer_Release(&pbuf);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002312 return NULL;
2313 }
2314
Martin v. Löwis423be952008-08-13 15:53:07 +00002315 PyBuffer_Release(&pbuf);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002316 /* Return the number of bytes read. Note that we do not do anything
2317 special here in the case that readlen < recvlen. */
Christian Heimes217cfd12007-12-02 14:31:20 +00002318 return PyLong_FromSsize_t(readlen);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002319}
2320
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002321PyDoc_STRVAR(recv_into_doc,
2322"recv_into(buffer, [nbytes[, flags]]) -> nbytes_read\n\
Thomas Wouters477c8d52006-05-27 19:21:47 +00002323\n\
2324A version of recv() that stores its data into a buffer rather than creating \n\
2325a new string. Receive up to buffersize bytes from the socket. If buffersize \n\
2326is not specified (or 0), receive up to the size available in the given buffer.\n\
2327\n\
2328See recv() for documentation about the flags.");
2329
2330
2331/*
Christian Heimes99170a52007-12-19 02:07:34 +00002332 * This is the guts of the recvfrom() and recvfrom_into() methods, which reads
2333 * into a char buffer. If you have any inc/def ref to do to the objects that
2334 * contain the buffer, do it in the caller. This function returns the number
2335 * of bytes succesfully read. If there was an error, it returns -1. Note
2336 * that it is also possible that we return a number of bytes smaller than the
2337 * request bytes.
Thomas Wouters477c8d52006-05-27 19:21:47 +00002338 *
2339 * 'addr' is a return value for the address object. Note that you must decref
2340 * it yourself.
2341 */
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002342static ssize_t
Thomas Wouters477c8d52006-05-27 19:21:47 +00002343sock_recvfrom_guts(PySocketSockObject *s, char* cbuf, int len, int flags,
2344 PyObject** addr)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002345{
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00002346 sock_addr_t addrbuf;
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002347 int timeout;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002348 ssize_t n = -1;
Guido van Rossumff3ab422000-04-24 15:16:03 +00002349 socklen_t addrlen;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002350
Thomas Wouters477c8d52006-05-27 19:21:47 +00002351 *addr = NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002352
Guido van Rossum18c9a4f1993-05-25 12:16:29 +00002353 if (!getsockaddrlen(s, &addrlen))
Thomas Wouters477c8d52006-05-27 19:21:47 +00002354 return -1;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002355
Thomas Wouters477c8d52006-05-27 19:21:47 +00002356 if (!IS_SELECTABLE(s)) {
2357 select_error();
2358 return -1;
2359 }
Neal Norwitz082b2df2006-02-07 07:04:46 +00002360
Guido van Rossum73624e91994-10-10 17:59:00 +00002361 Py_BEGIN_ALLOW_THREADS
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00002362 memset(&addrbuf, 0, addrlen);
Raymond Hettingeref7343c2003-06-29 03:08:05 +00002363 timeout = internal_select(s, 0);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002364 if (!timeout) {
Guido van Rossum8d665e61996-06-26 18:22:49 +00002365#ifndef MS_WINDOWS
Andrew MacIntyreba43e872002-03-03 03:03:52 +00002366#if defined(PYOS_OS2) && !defined(PYCC_GCC)
Thomas Wouters477c8d52006-05-27 19:21:47 +00002367 n = recvfrom(s->sock_fd, cbuf, len, flags,
Thomas Wouters89f507f2006-12-13 04:49:30 +00002368 SAS2SA(&addrbuf), &addrlen);
Guido van Rossum32c575d1997-12-02 20:37:32 +00002369#else
Thomas Wouters477c8d52006-05-27 19:21:47 +00002370 n = recvfrom(s->sock_fd, cbuf, len, flags,
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002371 (void *) &addrbuf, &addrlen);
Guido van Rossum32c575d1997-12-02 20:37:32 +00002372#endif
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00002373#else
Thomas Wouters477c8d52006-05-27 19:21:47 +00002374 n = recvfrom(s->sock_fd, cbuf, len, flags,
Thomas Wouters89f507f2006-12-13 04:49:30 +00002375 SAS2SA(&addrbuf), &addrlen);
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00002376#endif
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002377 }
Guido van Rossum73624e91994-10-10 17:59:00 +00002378 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00002379
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002380 if (timeout == 1) {
Raymond Hettingeref7343c2003-06-29 03:08:05 +00002381 PyErr_SetString(socket_timeout, "timed out");
Thomas Wouters477c8d52006-05-27 19:21:47 +00002382 return -1;
Raymond Hettingeref7343c2003-06-29 03:08:05 +00002383 }
Guido van Rossum7c53b771995-09-13 18:39:47 +00002384 if (n < 0) {
Thomas Wouters477c8d52006-05-27 19:21:47 +00002385 s->errorhandler();
2386 return -1;
Guido van Rossum7c53b771995-09-13 18:39:47 +00002387 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00002388
Thomas Wouters89f507f2006-12-13 04:49:30 +00002389 if (!(*addr = makesockaddr(s->sock_fd, SAS2SA(&addrbuf),
Thomas Wouters477c8d52006-05-27 19:21:47 +00002390 addrlen, s->sock_proto)))
2391 return -1;
2392
2393 return n;
2394}
2395
2396/* s.recvfrom(nbytes [,flags]) method */
2397
2398static PyObject *
2399sock_recvfrom(PySocketSockObject *s, PyObject *args)
2400{
2401 PyObject *buf = NULL;
2402 PyObject *addr = NULL;
2403 PyObject *ret = NULL;
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002404 int recvlen, flags = 0;
2405 ssize_t outlen;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002406
2407 if (!PyArg_ParseTuple(args, "i|i:recvfrom", &recvlen, &flags))
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002408 return NULL;
Guido van Rossum48a680c2001-03-02 06:34:14 +00002409
Guido van Rossumd8faa362007-04-27 19:54:29 +00002410 if (recvlen < 0) {
2411 PyErr_SetString(PyExc_ValueError,
2412 "negative buffersize in recvfrom");
2413 return NULL;
2414 }
2415
Christian Heimes72b710a2008-05-26 13:28:38 +00002416 buf = PyBytes_FromStringAndSize((char *) 0, recvlen);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002417 if (buf == NULL)
2418 return NULL;
2419
Christian Heimes72b710a2008-05-26 13:28:38 +00002420 outlen = sock_recvfrom_guts(s, PyBytes_AS_STRING(buf),
Thomas Wouters477c8d52006-05-27 19:21:47 +00002421 recvlen, flags, &addr);
2422 if (outlen < 0) {
Barry Warsaw752300b1997-01-03 17:18:10 +00002423 goto finally;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002424 }
2425
2426 if (outlen != recvlen) {
2427 /* We did not read as many bytes as we anticipated, resize the
2428 string if possible and be succesful. */
Christian Heimes72b710a2008-05-26 13:28:38 +00002429 if (_PyBytes_Resize(&buf, outlen) < 0)
Thomas Wouters477c8d52006-05-27 19:21:47 +00002430 /* Oopsy, not so succesful after all. */
2431 goto finally;
2432 }
Barry Warsaw752300b1997-01-03 17:18:10 +00002433
Raymond Hettinger8ae46892003-10-12 19:09:37 +00002434 ret = PyTuple_Pack(2, buf, addr);
Guido van Rossum67f7a382002-06-06 21:08:16 +00002435
2436finally:
Guido van Rossum73624e91994-10-10 17:59:00 +00002437 Py_XDECREF(buf);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002438 Py_XDECREF(addr);
Guido van Rossum6f5afc91993-02-05 09:46:15 +00002439 return ret;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002440}
2441
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002442PyDoc_STRVAR(recvfrom_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002443"recvfrom(buffersize[, flags]) -> (data, address info)\n\
2444\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002445Like recv(buffersize, flags) but also return the sender's address info.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002446
Thomas Wouters477c8d52006-05-27 19:21:47 +00002447
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002448/* s.recvfrom_into(buffer[, nbytes [,flags]]) method */
Thomas Wouters477c8d52006-05-27 19:21:47 +00002449
2450static PyObject *
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002451sock_recvfrom_into(PySocketSockObject *s, PyObject *args, PyObject* kwds)
Thomas Wouters477c8d52006-05-27 19:21:47 +00002452{
2453 static char *kwlist[] = {"buffer", "nbytes", "flags", 0};
2454
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002455 int recvlen = 0, flags = 0;
2456 ssize_t readlen;
Martin v. Löwis423be952008-08-13 15:53:07 +00002457 Py_buffer pbuf;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002458 char *buf;
2459 int buflen;
2460
2461 PyObject *addr = NULL;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002462
Martin v. Löwis423be952008-08-13 15:53:07 +00002463 if (!PyArg_ParseTupleAndKeywords(args, kwds, "w*|ii:recvfrom_into",
2464 kwlist, &pbuf,
Guido van Rossumd8faa362007-04-27 19:54:29 +00002465 &recvlen, &flags))
Thomas Wouters477c8d52006-05-27 19:21:47 +00002466 return NULL;
Martin v. Löwis423be952008-08-13 15:53:07 +00002467 buf = pbuf.buf;
2468 buflen = pbuf.len;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002469 assert(buf != 0 && buflen > 0);
2470
2471 if (recvlen < 0) {
Martin v. Löwis423be952008-08-13 15:53:07 +00002472 PyBuffer_Release(&pbuf);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002473 PyErr_SetString(PyExc_ValueError,
Guido van Rossumd8faa362007-04-27 19:54:29 +00002474 "negative buffersize in recvfrom_into");
Thomas Wouters477c8d52006-05-27 19:21:47 +00002475 return NULL;
2476 }
2477 if (recvlen == 0) {
2478 /* If nbytes was not specified, use the buffer's length */
2479 recvlen = buflen;
2480 }
2481
2482 readlen = sock_recvfrom_guts(s, buf, recvlen, flags, &addr);
2483 if (readlen < 0) {
Martin v. Löwis423be952008-08-13 15:53:07 +00002484 PyBuffer_Release(&pbuf);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002485 /* Return an error */
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002486 Py_XDECREF(addr);
2487 return NULL;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002488 }
2489
Martin v. Löwis423be952008-08-13 15:53:07 +00002490 PyBuffer_Release(&pbuf);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002491 /* Return the number of bytes read and the address. Note that we do
2492 not do anything special here in the case that readlen < recvlen. */
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002493 return Py_BuildValue("lN", readlen, addr);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002494}
2495
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002496PyDoc_STRVAR(recvfrom_into_doc,
2497"recvfrom_into(buffer[, nbytes[, flags]]) -> (nbytes, address info)\n\
Thomas Wouters477c8d52006-05-27 19:21:47 +00002498\n\
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002499Like recv_into(buffer[, nbytes[, flags]]) but also return the sender's address info.");
Thomas Wouters477c8d52006-05-27 19:21:47 +00002500
2501
Guido van Rossumeb6b33a1993-05-25 09:38:27 +00002502/* s.send(data [,flags]) method */
Guido van Rossum30a685f1991-06-27 15:51:29 +00002503
Guido van Rossum73624e91994-10-10 17:59:00 +00002504static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002505sock_send(PySocketSockObject *s, PyObject *args)
Guido van Rossum30a685f1991-06-27 15:51:29 +00002506{
Guido van Rossumff4949e1992-08-05 19:58:53 +00002507 char *buf;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002508 int len, n = -1, flags = 0, timeout;
Martin v. Löwis423be952008-08-13 15:53:07 +00002509 Py_buffer pbuf;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002510
Martin v. Löwis423be952008-08-13 15:53:07 +00002511 if (!PyArg_ParseTuple(args, "y*|i:send", &pbuf, &flags))
Guido van Rossumbe32c891996-06-20 16:25:29 +00002512 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002513
Martin v. Löwis423be952008-08-13 15:53:07 +00002514 if (!IS_SELECTABLE(s)) {
2515 PyBuffer_Release(&pbuf);
Neal Norwitz082b2df2006-02-07 07:04:46 +00002516 return select_error();
Martin v. Löwis423be952008-08-13 15:53:07 +00002517 }
2518 buf = pbuf.buf;
2519 len = pbuf.len;
Neal Norwitz082b2df2006-02-07 07:04:46 +00002520
Guido van Rossum73624e91994-10-10 17:59:00 +00002521 Py_BEGIN_ALLOW_THREADS
Raymond Hettingeref7343c2003-06-29 03:08:05 +00002522 timeout = internal_select(s, 1);
2523 if (!timeout)
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002524#ifdef __VMS
2525 n = sendsegmented(s->sock_fd, buf, len, flags);
2526#else
Raymond Hettingeref7343c2003-06-29 03:08:05 +00002527 n = send(s->sock_fd, buf, len, flags);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002528#endif
Guido van Rossum73624e91994-10-10 17:59:00 +00002529 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00002530
Martin v. Löwis423be952008-08-13 15:53:07 +00002531 PyBuffer_Release(&pbuf);
2532
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002533 if (timeout == 1) {
Raymond Hettingeref7343c2003-06-29 03:08:05 +00002534 PyErr_SetString(socket_timeout, "timed out");
2535 return NULL;
2536 }
Guido van Rossum30a685f1991-06-27 15:51:29 +00002537 if (n < 0)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00002538 return s->errorhandler();
Christian Heimes217cfd12007-12-02 14:31:20 +00002539 return PyLong_FromLong((long)n);
Guido van Rossum30a685f1991-06-27 15:51:29 +00002540}
2541
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002542PyDoc_STRVAR(send_doc,
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002543"send(data[, flags]) -> count\n\
Guido van Rossum82a5c661998-07-07 20:45:43 +00002544\n\
2545Send a data string to the socket. For the optional flags\n\
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002546argument, see the Unix manual. Return the number of bytes\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002547sent; this may be less than len(data) if the network is busy.");
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002548
2549
2550/* s.sendall(data [,flags]) method */
2551
2552static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002553sock_sendall(PySocketSockObject *s, PyObject *args)
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002554{
2555 char *buf;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002556 int len, n = -1, flags = 0, timeout;
Martin v. Löwis423be952008-08-13 15:53:07 +00002557 Py_buffer pbuf;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002558
Martin v. Löwis423be952008-08-13 15:53:07 +00002559 if (!PyArg_ParseTuple(args, "y*|i:sendall", &pbuf, &flags))
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002560 return NULL;
Martin v. Löwis423be952008-08-13 15:53:07 +00002561 buf = pbuf.buf;
2562 len = pbuf.len;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002563
Martin v. Löwis423be952008-08-13 15:53:07 +00002564 if (!IS_SELECTABLE(s)) {
2565 PyBuffer_Release(&pbuf);
Neal Norwitz082b2df2006-02-07 07:04:46 +00002566 return select_error();
Martin v. Löwis423be952008-08-13 15:53:07 +00002567 }
Neal Norwitz082b2df2006-02-07 07:04:46 +00002568
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002569 Py_BEGIN_ALLOW_THREADS
2570 do {
Raymond Hettingeref7343c2003-06-29 03:08:05 +00002571 timeout = internal_select(s, 1);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002572 n = -1;
Raymond Hettingeref7343c2003-06-29 03:08:05 +00002573 if (timeout)
2574 break;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002575#ifdef __VMS
2576 n = sendsegmented(s->sock_fd, buf, len, flags);
2577#else
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002578 n = send(s->sock_fd, buf, len, flags);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002579#endif
Gregory P. Smithb585a0f2010-01-04 03:29:50 +00002580 if (n < 0) {
2581#ifdef EINTR
2582 /* We must handle EINTR here as there is no way for
2583 * the caller to know how much was sent otherwise. */
2584 if (errno == EINTR) {
2585 /* Run signal handlers. If an exception was
2586 * raised, abort and leave this socket in
2587 * an unknown state. */
2588 if (PyErr_CheckSignals())
2589 return NULL;
2590 continue;
2591 }
2592#endif
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002593 break;
Gregory P. Smithb585a0f2010-01-04 03:29:50 +00002594 }
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002595 buf += n;
2596 len -= n;
2597 } while (len > 0);
2598 Py_END_ALLOW_THREADS
Martin v. Löwis423be952008-08-13 15:53:07 +00002599 PyBuffer_Release(&pbuf);
Guido van Rossum67f7a382002-06-06 21:08:16 +00002600
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002601 if (timeout == 1) {
Raymond Hettingeref7343c2003-06-29 03:08:05 +00002602 PyErr_SetString(socket_timeout, "timed out");
2603 return NULL;
2604 }
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002605 if (n < 0)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00002606 return s->errorhandler();
Guido van Rossum67f7a382002-06-06 21:08:16 +00002607
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002608 Py_INCREF(Py_None);
2609 return Py_None;
2610}
2611
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002612PyDoc_STRVAR(sendall_doc,
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002613"sendall(data[, flags])\n\
2614\n\
2615Send a data string to the socket. For the optional flags\n\
2616argument, see the Unix manual. This calls send() repeatedly\n\
2617until all data is sent. If an error occurs, it's impossible\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002618to tell how much data has been sent.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002619
Guido van Rossum30a685f1991-06-27 15:51:29 +00002620
Guido van Rossumeb6b33a1993-05-25 09:38:27 +00002621/* s.sendto(data, [flags,] sockaddr) method */
Guido van Rossum30a685f1991-06-27 15:51:29 +00002622
Guido van Rossum73624e91994-10-10 17:59:00 +00002623static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002624sock_sendto(PySocketSockObject *s, PyObject *args)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002625{
Martin v. Löwis423be952008-08-13 15:53:07 +00002626 Py_buffer pbuf;
Guido van Rossum73624e91994-10-10 17:59:00 +00002627 PyObject *addro;
Guido van Rossumff4949e1992-08-05 19:58:53 +00002628 char *buf;
Martin v. Löwis423be952008-08-13 15:53:07 +00002629 Py_ssize_t len;
Thomas Wouters89f507f2006-12-13 04:49:30 +00002630 sock_addr_t addrbuf;
Martin v. Löwis423be952008-08-13 15:53:07 +00002631 int addrlen, n = -1, flags, timeout;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002632
Guido van Rossumeb6b33a1993-05-25 09:38:27 +00002633 flags = 0;
Martin v. Löwis423be952008-08-13 15:53:07 +00002634 if (!PyArg_ParseTuple(args, "y*O:sendto", &pbuf, &addro)) {
Guido van Rossum73624e91994-10-10 17:59:00 +00002635 PyErr_Clear();
Martin v. Löwis423be952008-08-13 15:53:07 +00002636 if (!PyArg_ParseTuple(args, "y*iO:sendto",
2637 &pbuf, &flags, &addro))
Guido van Rossumeb6b33a1993-05-25 09:38:27 +00002638 return NULL;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002639 }
Martin v. Löwis423be952008-08-13 15:53:07 +00002640 buf = pbuf.buf;
2641 len = pbuf.len;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002642
Martin v. Löwis423be952008-08-13 15:53:07 +00002643 if (!IS_SELECTABLE(s)) {
2644 PyBuffer_Release(&pbuf);
Neal Norwitz082b2df2006-02-07 07:04:46 +00002645 return select_error();
Martin v. Löwis423be952008-08-13 15:53:07 +00002646 }
Neal Norwitz082b2df2006-02-07 07:04:46 +00002647
Martin v. Löwis423be952008-08-13 15:53:07 +00002648 if (!getsockaddrarg(s, addro, SAS2SA(&addrbuf), &addrlen)) {
2649 PyBuffer_Release(&pbuf);
Thomas Wouters89f507f2006-12-13 04:49:30 +00002650 return NULL;
Martin v. Löwis423be952008-08-13 15:53:07 +00002651 }
Thomas Wouters89f507f2006-12-13 04:49:30 +00002652
Guido van Rossum73624e91994-10-10 17:59:00 +00002653 Py_BEGIN_ALLOW_THREADS
Raymond Hettingeref7343c2003-06-29 03:08:05 +00002654 timeout = internal_select(s, 1);
2655 if (!timeout)
Thomas Wouters89f507f2006-12-13 04:49:30 +00002656 n = sendto(s->sock_fd, buf, len, flags, SAS2SA(&addrbuf), addrlen);
Guido van Rossum73624e91994-10-10 17:59:00 +00002657 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00002658
Martin v. Löwis423be952008-08-13 15:53:07 +00002659 PyBuffer_Release(&pbuf);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002660 if (timeout == 1) {
Raymond Hettingeref7343c2003-06-29 03:08:05 +00002661 PyErr_SetString(socket_timeout, "timed out");
2662 return NULL;
2663 }
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002664 if (n < 0)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00002665 return s->errorhandler();
Christian Heimes217cfd12007-12-02 14:31:20 +00002666 return PyLong_FromLong((long)n);
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002667}
2668
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002669PyDoc_STRVAR(sendto_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00002670"sendto(data[, flags], address) -> count\n\
Guido van Rossum82a5c661998-07-07 20:45:43 +00002671\n\
2672Like send(data, flags) but allows specifying the destination address.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002673For IP sockets, the address is a pair (hostaddr, port).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002674
Guido van Rossum30a685f1991-06-27 15:51:29 +00002675
2676/* s.shutdown(how) method */
2677
Guido van Rossum73624e91994-10-10 17:59:00 +00002678static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002679sock_shutdown(PySocketSockObject *s, PyObject *arg)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002680{
2681 int how;
Guido van Rossumff4949e1992-08-05 19:58:53 +00002682 int res;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00002683
Christian Heimes217cfd12007-12-02 14:31:20 +00002684 how = PyLong_AsLong(arg);
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00002685 if (how == -1 && PyErr_Occurred())
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002686 return NULL;
Guido van Rossum73624e91994-10-10 17:59:00 +00002687 Py_BEGIN_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +00002688 res = shutdown(s->sock_fd, how);
Guido van Rossum73624e91994-10-10 17:59:00 +00002689 Py_END_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +00002690 if (res < 0)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00002691 return s->errorhandler();
Guido van Rossum73624e91994-10-10 17:59:00 +00002692 Py_INCREF(Py_None);
2693 return Py_None;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002694}
2695
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002696PyDoc_STRVAR(shutdown_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002697"shutdown(flag)\n\
2698\n\
Martin v. Löwis94681fc2003-11-27 19:40:22 +00002699Shut down the reading side of the socket (flag == SHUT_RD), the writing side\n\
2700of the socket (flag == SHUT_WR), or both ends (flag == SHUT_RDWR).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002701
Amaury Forgeot d'Arc3d17a5c2008-06-13 01:09:34 +00002702#if defined(MS_WINDOWS) && defined(SIO_RCVALL)
Christian Heimesfaf2f632008-01-06 16:59:19 +00002703static PyObject*
2704sock_ioctl(PySocketSockObject *s, PyObject *arg)
2705{
2706 unsigned long cmd = SIO_RCVALL;
Kristján Valur Jónsson847ec752009-09-27 21:10:38 +00002707 PyObject *argO;
2708 DWORD recv;
Christian Heimesfaf2f632008-01-06 16:59:19 +00002709
Kristján Valur Jónsson847ec752009-09-27 21:10:38 +00002710 if (!PyArg_ParseTuple(arg, "kO:ioctl", &cmd, &argO))
Christian Heimesfaf2f632008-01-06 16:59:19 +00002711 return NULL;
2712
Kristján Valur Jónsson847ec752009-09-27 21:10:38 +00002713 switch (cmd) {
2714 case SIO_RCVALL: {
2715 unsigned int option = RCVALL_ON;
2716 if (!PyArg_ParseTuple(arg, "kI:ioctl", &cmd, &option))
2717 return NULL;
2718 if (WSAIoctl(s->sock_fd, cmd, &option, sizeof(option),
2719 NULL, 0, &recv, NULL, NULL) == SOCKET_ERROR) {
2720 return set_error();
2721 }
2722 return PyLong_FromUnsignedLong(recv); }
2723 case SIO_KEEPALIVE_VALS: {
2724 struct tcp_keepalive ka;
2725 if (!PyArg_ParseTuple(arg, "k(kkk):ioctl", &cmd,
2726 &ka.onoff, &ka.keepalivetime, &ka.keepaliveinterval))
2727 return NULL;
2728 if (WSAIoctl(s->sock_fd, cmd, &ka, sizeof(ka),
2729 NULL, 0, &recv, NULL, NULL) == SOCKET_ERROR) {
2730 return set_error();
2731 }
2732 return PyLong_FromUnsignedLong(recv); }
2733 default:
2734 PyErr_Format(PyExc_ValueError, "invalid ioctl command %d", cmd);
2735 return NULL;
Christian Heimesfaf2f632008-01-06 16:59:19 +00002736 }
Christian Heimesfaf2f632008-01-06 16:59:19 +00002737}
2738PyDoc_STRVAR(sock_ioctl_doc,
2739"ioctl(cmd, option) -> long\n\
2740\n\
Kristján Valur Jónsson847ec752009-09-27 21:10:38 +00002741Control the socket with WSAIoctl syscall. Currently supported 'cmd' values are\n\
2742SIO_RCVALL: 'option' must be one of the socket.RCVALL_* constants.\n\
2743SIO_KEEPALIVE_VALS: 'option' is a tuple of (onoff, timeout, interval).");
Christian Heimesfaf2f632008-01-06 16:59:19 +00002744
2745#endif
Guido van Rossum30a685f1991-06-27 15:51:29 +00002746
2747/* List of methods for socket objects */
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002748
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002749static PyMethodDef sock_methods[] = {
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00002750 {"_accept", (PyCFunction)sock_accept, METH_NOARGS,
Thomas Wouters477c8d52006-05-27 19:21:47 +00002751 accept_doc},
2752 {"bind", (PyCFunction)sock_bind, METH_O,
2753 bind_doc},
2754 {"close", (PyCFunction)sock_close, METH_NOARGS,
2755 close_doc},
2756 {"connect", (PyCFunction)sock_connect, METH_O,
2757 connect_doc},
2758 {"connect_ex", (PyCFunction)sock_connect_ex, METH_O,
2759 connect_ex_doc},
Thomas Wouters477c8d52006-05-27 19:21:47 +00002760 {"fileno", (PyCFunction)sock_fileno, METH_NOARGS,
2761 fileno_doc},
Guido van Rossumb6775db1994-08-01 11:34:53 +00002762#ifdef HAVE_GETPEERNAME
Thomas Wouters477c8d52006-05-27 19:21:47 +00002763 {"getpeername", (PyCFunction)sock_getpeername,
2764 METH_NOARGS, getpeername_doc},
Guido van Rossum9575a441993-04-07 14:06:14 +00002765#endif
Thomas Wouters477c8d52006-05-27 19:21:47 +00002766 {"getsockname", (PyCFunction)sock_getsockname,
2767 METH_NOARGS, getsockname_doc},
2768 {"getsockopt", (PyCFunction)sock_getsockopt, METH_VARARGS,
2769 getsockopt_doc},
Amaury Forgeot d'Arc3d17a5c2008-06-13 01:09:34 +00002770#if defined(MS_WINDOWS) && defined(SIO_RCVALL)
Christian Heimesfaf2f632008-01-06 16:59:19 +00002771 {"ioctl", (PyCFunction)sock_ioctl, METH_VARARGS,
2772 sock_ioctl_doc},
2773#endif
Thomas Wouters477c8d52006-05-27 19:21:47 +00002774 {"listen", (PyCFunction)sock_listen, METH_O,
2775 listen_doc},
Thomas Wouters477c8d52006-05-27 19:21:47 +00002776 {"recv", (PyCFunction)sock_recv, METH_VARARGS,
2777 recv_doc},
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002778 {"recv_into", (PyCFunction)sock_recv_into, METH_VARARGS | METH_KEYWORDS,
2779 recv_into_doc},
Thomas Wouters477c8d52006-05-27 19:21:47 +00002780 {"recvfrom", (PyCFunction)sock_recvfrom, METH_VARARGS,
2781 recvfrom_doc},
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002782 {"recvfrom_into", (PyCFunction)sock_recvfrom_into, METH_VARARGS | METH_KEYWORDS,
2783 recvfrom_into_doc},
Thomas Wouters477c8d52006-05-27 19:21:47 +00002784 {"send", (PyCFunction)sock_send, METH_VARARGS,
2785 send_doc},
2786 {"sendall", (PyCFunction)sock_sendall, METH_VARARGS,
2787 sendall_doc},
2788 {"sendto", (PyCFunction)sock_sendto, METH_VARARGS,
2789 sendto_doc},
2790 {"setblocking", (PyCFunction)sock_setblocking, METH_O,
2791 setblocking_doc},
2792 {"settimeout", (PyCFunction)sock_settimeout, METH_O,
2793 settimeout_doc},
2794 {"gettimeout", (PyCFunction)sock_gettimeout, METH_NOARGS,
2795 gettimeout_doc},
2796 {"setsockopt", (PyCFunction)sock_setsockopt, METH_VARARGS,
2797 setsockopt_doc},
2798 {"shutdown", (PyCFunction)sock_shutdown, METH_O,
2799 shutdown_doc},
Guido van Rossumb6775db1994-08-01 11:34:53 +00002800 {NULL, NULL} /* sentinel */
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002801};
2802
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002803/* SockObject members */
2804static PyMemberDef sock_memberlist[] = {
2805 {"family", T_INT, offsetof(PySocketSockObject, sock_family), READONLY, "the socket family"},
2806 {"type", T_INT, offsetof(PySocketSockObject, sock_type), READONLY, "the socket type"},
2807 {"proto", T_INT, offsetof(PySocketSockObject, sock_proto), READONLY, "the socket protocol"},
2808 {"timeout", T_DOUBLE, offsetof(PySocketSockObject, sock_timeout), READONLY, "the socket timeout"},
2809 {0},
2810};
Guido van Rossum30a685f1991-06-27 15:51:29 +00002811
Guido van Rossum73624e91994-10-10 17:59:00 +00002812/* Deallocate a socket object in response to the last Py_DECREF().
Guido van Rossum30a685f1991-06-27 15:51:29 +00002813 First close the file description. */
2814
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002815static void
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002816sock_dealloc(PySocketSockObject *s)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002817{
Guido van Rossumfa972c92000-04-10 12:45:45 +00002818 if (s->sock_fd != -1)
Guido van Rossum2dd8ddd2000-04-21 20:33:00 +00002819 (void) SOCKETCLOSE(s->sock_fd);
Christian Heimes90aa7642007-12-19 02:45:37 +00002820 Py_TYPE(s)->tp_free((PyObject *)s);
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002821}
2822
Guido van Rossum30a685f1991-06-27 15:51:29 +00002823
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00002824static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002825sock_repr(PySocketSockObject *s)
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00002826{
Fred Drakea04eaad2000-06-30 02:46:07 +00002827#if SIZEOF_SOCKET_T > SIZEOF_LONG
2828 if (s->sock_fd > LONG_MAX) {
2829 /* this can occur on Win64, and actually there is a special
2830 ugly printf formatter for decimal pointer length integer
2831 printing, only bother if necessary*/
2832 PyErr_SetString(PyExc_OverflowError,
Guido van Rossum3eede5a2002-06-07 02:08:35 +00002833 "no printf formatter to display "
2834 "the socket descriptor in decimal");
Fred Drakea04eaad2000-06-30 02:46:07 +00002835 return NULL;
2836 }
2837#endif
Walter Dörwalda7eb93e2007-06-05 13:41:53 +00002838 return PyUnicode_FromFormat(
Guido van Rossum7d0a8262007-05-21 23:13:11 +00002839 "<socket object, fd=%ld, family=%d, type=%d, proto=%d>",
Guido van Rossum3eede5a2002-06-07 02:08:35 +00002840 (long)s->sock_fd, s->sock_family,
2841 s->sock_type,
2842 s->sock_proto);
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00002843}
2844
2845
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002846/* Create a new, uninitialized socket object. */
2847
2848static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002849sock_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002850{
2851 PyObject *new;
2852
2853 new = type->tp_alloc(type, 0);
Guido van Rossume1c478f2002-06-06 20:08:25 +00002854 if (new != NULL) {
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002855 ((PySocketSockObject *)new)->sock_fd = -1;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002856 ((PySocketSockObject *)new)->sock_timeout = -1.0;
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002857 ((PySocketSockObject *)new)->errorhandler = &set_error;
Guido van Rossume1c478f2002-06-06 20:08:25 +00002858 }
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002859 return new;
2860}
2861
2862
2863/* Initialize a new socket object. */
2864
2865/*ARGSUSED*/
2866static int
Andrew MacIntyre7aec4a22002-06-13 11:53:52 +00002867sock_initobj(PyObject *self, PyObject *args, PyObject *kwds)
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002868{
2869 PySocketSockObject *s = (PySocketSockObject *)self;
Guido van Rossum7d0a8262007-05-21 23:13:11 +00002870 PyObject *fdobj = NULL;
2871 SOCKET_T fd = INVALID_SOCKET;
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002872 int family = AF_INET, type = SOCK_STREAM, proto = 0;
Guido van Rossum7d0a8262007-05-21 23:13:11 +00002873 static char *keywords[] = {"family", "type", "proto", "fileno", 0};
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002874
2875 if (!PyArg_ParseTupleAndKeywords(args, kwds,
Guido van Rossum7d0a8262007-05-21 23:13:11 +00002876 "|iiiO:socket", keywords,
2877 &family, &type, &proto, &fdobj))
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002878 return -1;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002879
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00002880 if (fdobj != NULL && fdobj != Py_None) {
2881 fd = PyLong_AsSocket_t(fdobj);
Guido van Rossum7d0a8262007-05-21 23:13:11 +00002882 if (fd == (SOCKET_T)(-1) && PyErr_Occurred())
2883 return -1;
2884 if (fd == INVALID_SOCKET) {
2885 PyErr_SetString(PyExc_ValueError,
2886 "can't use invalid socket value");
2887 return -1;
2888 }
2889 }
2890 else {
2891 Py_BEGIN_ALLOW_THREADS
2892 fd = socket(family, type, proto);
2893 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00002894
Guido van Rossum7d0a8262007-05-21 23:13:11 +00002895 if (fd == INVALID_SOCKET) {
2896 set_error();
2897 return -1;
2898 }
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002899 }
2900 init_sockobject(s, fd, family, type, proto);
Guido van Rossum67f7a382002-06-06 21:08:16 +00002901
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002902 return 0;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002903
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002904}
2905
2906
Guido van Rossumb6775db1994-08-01 11:34:53 +00002907/* Type object for socket objects. */
Guido van Rossum30a685f1991-06-27 15:51:29 +00002908
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002909static PyTypeObject sock_type = {
Martin v. Löwis9f2e3462007-07-21 17:22:18 +00002910 PyVarObject_HEAD_INIT(0, 0) /* Must fill in type value later */
Guido van Rossumca6dfa52001-10-28 12:31:33 +00002911 "_socket.socket", /* tp_name */
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002912 sizeof(PySocketSockObject), /* tp_basicsize */
2913 0, /* tp_itemsize */
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002914 (destructor)sock_dealloc, /* tp_dealloc */
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002915 0, /* tp_print */
2916 0, /* tp_getattr */
2917 0, /* tp_setattr */
Mark Dickinsone94c6792009-02-02 20:36:42 +00002918 0, /* tp_reserved */
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002919 (reprfunc)sock_repr, /* tp_repr */
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002920 0, /* tp_as_number */
2921 0, /* tp_as_sequence */
2922 0, /* tp_as_mapping */
2923 0, /* tp_hash */
2924 0, /* tp_call */
2925 0, /* tp_str */
Jason Tishlerfb8595d2003-01-06 12:41:26 +00002926 PyObject_GenericGetAttr, /* tp_getattro */
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002927 0, /* tp_setattro */
2928 0, /* tp_as_buffer */
2929 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE, /* tp_flags */
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002930 sock_doc, /* tp_doc */
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002931 0, /* tp_traverse */
2932 0, /* tp_clear */
2933 0, /* tp_richcompare */
2934 0, /* tp_weaklistoffset */
2935 0, /* tp_iter */
2936 0, /* tp_iternext */
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002937 sock_methods, /* tp_methods */
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002938 sock_memberlist, /* tp_members */
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002939 0, /* tp_getset */
2940 0, /* tp_base */
2941 0, /* tp_dict */
2942 0, /* tp_descr_get */
2943 0, /* tp_descr_set */
2944 0, /* tp_dictoffset */
Andrew MacIntyre7aec4a22002-06-13 11:53:52 +00002945 sock_initobj, /* tp_init */
Jason Tishlerfb8595d2003-01-06 12:41:26 +00002946 PyType_GenericAlloc, /* tp_alloc */
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002947 sock_new, /* tp_new */
Jason Tishlerfb8595d2003-01-06 12:41:26 +00002948 PyObject_Del, /* tp_free */
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002949};
2950
Guido van Rossum30a685f1991-06-27 15:51:29 +00002951
Guido van Rossum81194471991-07-27 21:42:02 +00002952/* Python interface to gethostname(). */
2953
2954/*ARGSUSED*/
Guido van Rossum73624e91994-10-10 17:59:00 +00002955static PyObject *
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00002956socket_gethostname(PyObject *self, PyObject *unused)
Guido van Rossum81194471991-07-27 21:42:02 +00002957{
2958 char buf[1024];
Guido van Rossumff4949e1992-08-05 19:58:53 +00002959 int res;
Guido van Rossum73624e91994-10-10 17:59:00 +00002960 Py_BEGIN_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +00002961 res = gethostname(buf, (int) sizeof buf - 1);
Guido van Rossum73624e91994-10-10 17:59:00 +00002962 Py_END_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +00002963 if (res < 0)
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002964 return set_error();
Guido van Rossum81194471991-07-27 21:42:02 +00002965 buf[sizeof buf - 1] = '\0';
Guido van Rossum32c4ac02007-08-15 03:56:40 +00002966 return PyUnicode_FromString(buf);
Guido van Rossum81194471991-07-27 21:42:02 +00002967}
Guido van Rossumff4949e1992-08-05 19:58:53 +00002968
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002969PyDoc_STRVAR(gethostname_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002970"gethostname() -> string\n\
2971\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002972Return the current host name.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002973
Guido van Rossumff4949e1992-08-05 19:58:53 +00002974
Guido van Rossum30a685f1991-06-27 15:51:29 +00002975/* Python interface to gethostbyname(name). */
2976
2977/*ARGSUSED*/
Guido van Rossum73624e91994-10-10 17:59:00 +00002978static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002979socket_gethostbyname(PyObject *self, PyObject *args)
Guido van Rossum30a685f1991-06-27 15:51:29 +00002980{
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00002981 char *name;
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00002982 sock_addr_t addrbuf;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002983
Guido van Rossum27fc3c02000-03-24 20:56:56 +00002984 if (!PyArg_ParseTuple(args, "s:gethostbyname", &name))
Guido van Rossum30a685f1991-06-27 15:51:29 +00002985 return NULL;
Thomas Wouters89f507f2006-12-13 04:49:30 +00002986 if (setipaddr(name, SAS2SA(&addrbuf), sizeof(addrbuf), AF_INET) < 0)
Guido van Rossum30a685f1991-06-27 15:51:29 +00002987 return NULL;
Thomas Wouters89f507f2006-12-13 04:49:30 +00002988 return makeipaddr(SAS2SA(&addrbuf), sizeof(struct sockaddr_in));
Guido van Rossum30a685f1991-06-27 15:51:29 +00002989}
2990
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002991PyDoc_STRVAR(gethostbyname_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002992"gethostbyname(host) -> address\n\
2993\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002994Return the IP address (a string of the form '255.255.255.255') for a host.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002995
2996
Guido van Rossum7d896ab1998-08-04 22:16:43 +00002997/* Convenience function common to gethostbyname_ex and gethostbyaddr */
2998
2999static PyObject *
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003000gethost_common(struct hostent *h, struct sockaddr *addr, int alen, int af)
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003001{
3002 char **pch;
3003 PyObject *rtn_tuple = (PyObject *)NULL;
3004 PyObject *name_list = (PyObject *)NULL;
3005 PyObject *addr_list = (PyObject *)NULL;
3006 PyObject *tmp;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003007
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003008 if (h == NULL) {
Guido van Rossum48a680c2001-03-02 06:34:14 +00003009 /* Let's get real error message to return */
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003010 set_herror(h_errno);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003011 return NULL;
3012 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00003013
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003014 if (h->h_addrtype != af) {
Guido van Rossum3eede5a2002-06-07 02:08:35 +00003015 /* Let's get real error message to return */
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003016 PyErr_SetString(socket_error,
Guido van Rossum3eede5a2002-06-07 02:08:35 +00003017 (char *)strerror(EAFNOSUPPORT));
Christian Heimesada8c3b2008-03-18 18:26:33 +00003018
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003019 return NULL;
3020 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00003021
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003022 switch (af) {
Guido van Rossum67f7a382002-06-06 21:08:16 +00003023
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003024 case AF_INET:
3025 if (alen < sizeof(struct sockaddr_in))
3026 return NULL;
3027 break;
Guido van Rossum67f7a382002-06-06 21:08:16 +00003028
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00003029#ifdef ENABLE_IPV6
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003030 case AF_INET6:
3031 if (alen < sizeof(struct sockaddr_in6))
3032 return NULL;
3033 break;
3034#endif
Guido van Rossum67f7a382002-06-06 21:08:16 +00003035
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003036 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00003037
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003038 if ((name_list = PyList_New(0)) == NULL)
3039 goto err;
Guido van Rossum67f7a382002-06-06 21:08:16 +00003040
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003041 if ((addr_list = PyList_New(0)) == NULL)
3042 goto err;
Guido van Rossum67f7a382002-06-06 21:08:16 +00003043
Thomas Wouters00ee7ba2006-08-21 19:07:27 +00003044 /* SF #1511317: h_aliases can be NULL */
3045 if (h->h_aliases) {
3046 for (pch = h->h_aliases; *pch != NULL; pch++) {
3047 int status;
Guido van Rossum32c4ac02007-08-15 03:56:40 +00003048 tmp = PyUnicode_FromString(*pch);
Thomas Wouters00ee7ba2006-08-21 19:07:27 +00003049 if (tmp == NULL)
3050 goto err;
Guido van Rossum67f7a382002-06-06 21:08:16 +00003051
Thomas Wouters00ee7ba2006-08-21 19:07:27 +00003052 status = PyList_Append(name_list, tmp);
3053 Py_DECREF(tmp);
Guido van Rossum67f7a382002-06-06 21:08:16 +00003054
Thomas Wouters00ee7ba2006-08-21 19:07:27 +00003055 if (status)
3056 goto err;
3057 }
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003058 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00003059
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003060 for (pch = h->h_addr_list; *pch != NULL; pch++) {
3061 int status;
Guido van Rossum67f7a382002-06-06 21:08:16 +00003062
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003063 switch (af) {
Guido van Rossum67f7a382002-06-06 21:08:16 +00003064
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003065 case AF_INET:
3066 {
3067 struct sockaddr_in sin;
3068 memset(&sin, 0, sizeof(sin));
3069 sin.sin_family = af;
3070#ifdef HAVE_SOCKADDR_SA_LEN
3071 sin.sin_len = sizeof(sin);
3072#endif
3073 memcpy(&sin.sin_addr, *pch, sizeof(sin.sin_addr));
3074 tmp = makeipaddr((struct sockaddr *)&sin, sizeof(sin));
Guido van Rossum67f7a382002-06-06 21:08:16 +00003075
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003076 if (pch == h->h_addr_list && alen >= sizeof(sin))
3077 memcpy((char *) addr, &sin, sizeof(sin));
3078 break;
3079 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00003080
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00003081#ifdef ENABLE_IPV6
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003082 case AF_INET6:
3083 {
3084 struct sockaddr_in6 sin6;
3085 memset(&sin6, 0, sizeof(sin6));
3086 sin6.sin6_family = af;
3087#ifdef HAVE_SOCKADDR_SA_LEN
3088 sin6.sin6_len = sizeof(sin6);
3089#endif
3090 memcpy(&sin6.sin6_addr, *pch, sizeof(sin6.sin6_addr));
3091 tmp = makeipaddr((struct sockaddr *)&sin6,
3092 sizeof(sin6));
Guido van Rossum67f7a382002-06-06 21:08:16 +00003093
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003094 if (pch == h->h_addr_list && alen >= sizeof(sin6))
3095 memcpy((char *) addr, &sin6, sizeof(sin6));
3096 break;
3097 }
3098#endif
Guido van Rossum67f7a382002-06-06 21:08:16 +00003099
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003100 default: /* can't happen */
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003101 PyErr_SetString(socket_error,
Guido van Rossum3eede5a2002-06-07 02:08:35 +00003102 "unsupported address family");
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003103 return NULL;
3104 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00003105
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003106 if (tmp == NULL)
3107 goto err;
Guido van Rossum67f7a382002-06-06 21:08:16 +00003108
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003109 status = PyList_Append(addr_list, tmp);
3110 Py_DECREF(tmp);
Guido van Rossum67f7a382002-06-06 21:08:16 +00003111
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003112 if (status)
3113 goto err;
3114 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00003115
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003116 rtn_tuple = Py_BuildValue("sOO", h->h_name, name_list, addr_list);
Guido van Rossum67f7a382002-06-06 21:08:16 +00003117
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003118 err:
3119 Py_XDECREF(name_list);
3120 Py_XDECREF(addr_list);
3121 return rtn_tuple;
3122}
3123
3124
3125/* Python interface to gethostbyname_ex(name). */
3126
3127/*ARGSUSED*/
3128static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003129socket_gethostbyname_ex(PyObject *self, PyObject *args)
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003130{
3131 char *name;
3132 struct hostent *h;
Martin v. Löwisc16f3bd2003-05-03 09:14:54 +00003133#ifdef ENABLE_IPV6
3134 struct sockaddr_storage addr;
3135#else
3136 struct sockaddr_in addr;
3137#endif
Martin v. Löwis9db2f572001-07-23 01:30:10 +00003138 struct sockaddr *sa;
Guido van Rossum955becc1999-03-22 20:14:53 +00003139 PyObject *ret;
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003140#ifdef HAVE_GETHOSTBYNAME_R
3141 struct hostent hp_allocated;
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003142#ifdef HAVE_GETHOSTBYNAME_R_3_ARG
3143 struct hostent_data data;
3144#else
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003145 char buf[16384];
3146 int buf_len = (sizeof buf) - 1;
3147 int errnop;
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003148#endif
3149#if defined(HAVE_GETHOSTBYNAME_R_3_ARG) || defined(HAVE_GETHOSTBYNAME_R_6_ARG)
Guido van Rossume9cd07b1999-03-15 21:40:14 +00003150 int result;
3151#endif
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003152#endif /* HAVE_GETHOSTBYNAME_R */
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003153
Guido van Rossum27fc3c02000-03-24 20:56:56 +00003154 if (!PyArg_ParseTuple(args, "s:gethostbyname_ex", &name))
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003155 return NULL;
Martin v. Löwiscf8f47e2002-12-11 13:10:57 +00003156 if (setipaddr(name, (struct sockaddr *)&addr, sizeof(addr), AF_INET) < 0)
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003157 return NULL;
3158 Py_BEGIN_ALLOW_THREADS
3159#ifdef HAVE_GETHOSTBYNAME_R
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003160#if defined(HAVE_GETHOSTBYNAME_R_6_ARG)
Guido van Rossum3eede5a2002-06-07 02:08:35 +00003161 result = gethostbyname_r(name, &hp_allocated, buf, buf_len,
3162 &h, &errnop);
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003163#elif defined(HAVE_GETHOSTBYNAME_R_5_ARG)
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003164 h = gethostbyname_r(name, &hp_allocated, buf, buf_len, &errnop);
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003165#else /* HAVE_GETHOSTBYNAME_R_3_ARG */
Guido van Rossume7de2061999-03-24 17:24:33 +00003166 memset((void *) &data, '\0', sizeof(data));
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003167 result = gethostbyname_r(name, &hp_allocated, &data);
3168 h = (result != 0) ? NULL : &hp_allocated;
Guido van Rossume9cd07b1999-03-15 21:40:14 +00003169#endif
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003170#else /* not HAVE_GETHOSTBYNAME_R */
Guido van Rossum3baaa131999-03-22 21:44:51 +00003171#ifdef USE_GETHOSTBYNAME_LOCK
Just van Rossum1040d2c2003-05-09 07:53:18 +00003172 PyThread_acquire_lock(netdb_lock, 1);
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003173#endif
3174 h = gethostbyname(name);
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003175#endif /* HAVE_GETHOSTBYNAME_R */
3176 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00003177 /* Some C libraries would require addr.__ss_family instead of
Guido van Rossum3eede5a2002-06-07 02:08:35 +00003178 addr.ss_family.
3179 Therefore, we cast the sockaddr_storage into sockaddr to
3180 access sa_family. */
Martin v. Löwis9db2f572001-07-23 01:30:10 +00003181 sa = (struct sockaddr*)&addr;
Guido van Rossum67f7a382002-06-06 21:08:16 +00003182 ret = gethost_common(h, (struct sockaddr *)&addr, sizeof(addr),
Guido van Rossum3eede5a2002-06-07 02:08:35 +00003183 sa->sa_family);
Guido van Rossum3baaa131999-03-22 21:44:51 +00003184#ifdef USE_GETHOSTBYNAME_LOCK
Just van Rossum1040d2c2003-05-09 07:53:18 +00003185 PyThread_release_lock(netdb_lock);
Guido van Rossum955becc1999-03-22 20:14:53 +00003186#endif
3187 return ret;
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003188}
3189
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003190PyDoc_STRVAR(ghbn_ex_doc,
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003191"gethostbyname_ex(host) -> (name, aliaslist, addresslist)\n\
3192\n\
3193Return the true host name, a list of aliases, and a list of IP addresses,\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003194for a host. The host argument is a string giving a host name or IP number.");
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003195
3196
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00003197/* Python interface to gethostbyaddr(IP). */
3198
3199/*ARGSUSED*/
3200static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003201socket_gethostbyaddr(PyObject *self, PyObject *args)
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00003202{
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00003203#ifdef ENABLE_IPV6
Guido van Rossum3eede5a2002-06-07 02:08:35 +00003204 struct sockaddr_storage addr;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003205#else
Guido van Rossum48a680c2001-03-02 06:34:14 +00003206 struct sockaddr_in addr;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003207#endif
3208 struct sockaddr *sa = (struct sockaddr *)&addr;
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00003209 char *ip_num;
3210 struct hostent *h;
Guido van Rossum3baaa131999-03-22 21:44:51 +00003211 PyObject *ret;
Guido van Rossum4f199ea1998-04-09 20:56:35 +00003212#ifdef HAVE_GETHOSTBYNAME_R
3213 struct hostent hp_allocated;
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003214#ifdef HAVE_GETHOSTBYNAME_R_3_ARG
3215 struct hostent_data data;
3216#else
Benjamin Peterson058e31e2009-01-16 03:54:08 +00003217 /* glibcs up to 2.10 assume that the buf argument to
3218 gethostbyaddr_r is 8-byte aligned, which at least llvm-gcc
3219 does not ensure. The attribute below instructs the compiler
3220 to maintain this alignment. */
3221 char buf[16384] Py_ALIGNED(8);
Guido van Rossum4f199ea1998-04-09 20:56:35 +00003222 int buf_len = (sizeof buf) - 1;
3223 int errnop;
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003224#endif
3225#if defined(HAVE_GETHOSTBYNAME_R_3_ARG) || defined(HAVE_GETHOSTBYNAME_R_6_ARG)
Guido van Rossume9cd07b1999-03-15 21:40:14 +00003226 int result;
3227#endif
Guido van Rossum4f199ea1998-04-09 20:56:35 +00003228#endif /* HAVE_GETHOSTBYNAME_R */
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003229 char *ap;
3230 int al;
3231 int af;
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00003232
Guido van Rossum27fc3c02000-03-24 20:56:56 +00003233 if (!PyArg_ParseTuple(args, "s:gethostbyaddr", &ip_num))
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00003234 return NULL;
Martin v. Löwiscf8f47e2002-12-11 13:10:57 +00003235 af = AF_UNSPEC;
Martin v. Löwisddc6f472002-07-28 16:10:31 +00003236 if (setipaddr(ip_num, sa, sizeof(addr), af) < 0)
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00003237 return NULL;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003238 af = sa->sa_family;
3239 ap = NULL;
3240 al = 0;
3241 switch (af) {
3242 case AF_INET:
3243 ap = (char *)&((struct sockaddr_in *)sa)->sin_addr;
3244 al = sizeof(((struct sockaddr_in *)sa)->sin_addr);
3245 break;
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00003246#ifdef ENABLE_IPV6
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003247 case AF_INET6:
3248 ap = (char *)&((struct sockaddr_in6 *)sa)->sin6_addr;
3249 al = sizeof(((struct sockaddr_in6 *)sa)->sin6_addr);
3250 break;
3251#endif
3252 default:
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003253 PyErr_SetString(socket_error, "unsupported address family");
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003254 return NULL;
3255 }
Guido van Rossum4f199ea1998-04-09 20:56:35 +00003256 Py_BEGIN_ALLOW_THREADS
3257#ifdef HAVE_GETHOSTBYNAME_R
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003258#if defined(HAVE_GETHOSTBYNAME_R_6_ARG)
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003259 result = gethostbyaddr_r(ap, al, af,
3260 &hp_allocated, buf, buf_len,
Guido van Rossume9cd07b1999-03-15 21:40:14 +00003261 &h, &errnop);
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003262#elif defined(HAVE_GETHOSTBYNAME_R_5_ARG)
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003263 h = gethostbyaddr_r(ap, al, af,
Guido van Rossum4f199ea1998-04-09 20:56:35 +00003264 &hp_allocated, buf, buf_len, &errnop);
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003265#else /* HAVE_GETHOSTBYNAME_R_3_ARG */
Guido van Rossume7de2061999-03-24 17:24:33 +00003266 memset((void *) &data, '\0', sizeof(data));
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003267 result = gethostbyaddr_r(ap, al, af, &hp_allocated, &data);
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003268 h = (result != 0) ? NULL : &hp_allocated;
Guido van Rossume9cd07b1999-03-15 21:40:14 +00003269#endif
Guido van Rossum4f199ea1998-04-09 20:56:35 +00003270#else /* not HAVE_GETHOSTBYNAME_R */
Guido van Rossum3baaa131999-03-22 21:44:51 +00003271#ifdef USE_GETHOSTBYNAME_LOCK
Just van Rossum1040d2c2003-05-09 07:53:18 +00003272 PyThread_acquire_lock(netdb_lock, 1);
Guido van Rossum4f199ea1998-04-09 20:56:35 +00003273#endif
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003274 h = gethostbyaddr(ap, al, af);
Guido van Rossum4f199ea1998-04-09 20:56:35 +00003275#endif /* HAVE_GETHOSTBYNAME_R */
3276 Py_END_ALLOW_THREADS
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003277 ret = gethost_common(h, (struct sockaddr *)&addr, sizeof(addr), af);
Guido van Rossum3baaa131999-03-22 21:44:51 +00003278#ifdef USE_GETHOSTBYNAME_LOCK
Just van Rossum1040d2c2003-05-09 07:53:18 +00003279 PyThread_release_lock(netdb_lock);
Guido van Rossum3baaa131999-03-22 21:44:51 +00003280#endif
3281 return ret;
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00003282}
3283
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003284PyDoc_STRVAR(gethostbyaddr_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003285"gethostbyaddr(host) -> (name, aliaslist, addresslist)\n\
3286\n\
3287Return the true host name, a list of aliases, and a list of IP addresses,\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003288for a host. The host argument is a string giving a host name or IP number.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003289
Guido van Rossum30a685f1991-06-27 15:51:29 +00003290
3291/* Python interface to getservbyname(name).
3292 This only returns the port number, since the other info is already
3293 known or not useful (like the list of aliases). */
3294
3295/*ARGSUSED*/
Guido van Rossum73624e91994-10-10 17:59:00 +00003296static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003297socket_getservbyname(PyObject *self, PyObject *args)
Guido van Rossum30a685f1991-06-27 15:51:29 +00003298{
Barry Warsaw11b91a02004-06-28 00:50:43 +00003299 char *name, *proto=NULL;
Guido van Rossum30a685f1991-06-27 15:51:29 +00003300 struct servent *sp;
Barry Warsaw11b91a02004-06-28 00:50:43 +00003301 if (!PyArg_ParseTuple(args, "s|s:getservbyname", &name, &proto))
Guido van Rossum30a685f1991-06-27 15:51:29 +00003302 return NULL;
Guido van Rossum73624e91994-10-10 17:59:00 +00003303 Py_BEGIN_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +00003304 sp = getservbyname(name, proto);
Guido van Rossum73624e91994-10-10 17:59:00 +00003305 Py_END_ALLOW_THREADS
Guido van Rossum30a685f1991-06-27 15:51:29 +00003306 if (sp == NULL) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003307 PyErr_SetString(socket_error, "service/proto not found");
Guido van Rossum30a685f1991-06-27 15:51:29 +00003308 return NULL;
3309 }
Christian Heimes217cfd12007-12-02 14:31:20 +00003310 return PyLong_FromLong((long) ntohs(sp->s_port));
Guido van Rossum30a685f1991-06-27 15:51:29 +00003311}
3312
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003313PyDoc_STRVAR(getservbyname_doc,
Barry Warsaw11b91a02004-06-28 00:50:43 +00003314"getservbyname(servicename[, protocolname]) -> integer\n\
Guido van Rossum82a5c661998-07-07 20:45:43 +00003315\n\
3316Return a port number from a service name and protocol name.\n\
Barry Warsaw11b91a02004-06-28 00:50:43 +00003317The optional protocol name, if given, should be 'tcp' or 'udp',\n\
3318otherwise any protocol will match.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003319
Guido van Rossum30a685f1991-06-27 15:51:29 +00003320
Barry Warsaw11b91a02004-06-28 00:50:43 +00003321/* Python interface to getservbyport(port).
3322 This only returns the service name, since the other info is already
3323 known or not useful (like the list of aliases). */
3324
3325/*ARGSUSED*/
3326static PyObject *
3327socket_getservbyport(PyObject *self, PyObject *args)
3328{
Benjamin Peterson5c6d7872009-02-06 02:40:07 +00003329 int port;
Barry Warsaw11b91a02004-06-28 00:50:43 +00003330 char *proto=NULL;
3331 struct servent *sp;
Benjamin Peterson5c6d7872009-02-06 02:40:07 +00003332 if (!PyArg_ParseTuple(args, "i|s:getservbyport", &port, &proto))
Barry Warsaw11b91a02004-06-28 00:50:43 +00003333 return NULL;
Benjamin Peterson5c6d7872009-02-06 02:40:07 +00003334 if (port < 0 || port > 0xffff) {
3335 PyErr_SetString(
3336 PyExc_OverflowError,
3337 "getservbyport: port must be 0-65535.");
3338 return NULL;
3339 }
Barry Warsaw11b91a02004-06-28 00:50:43 +00003340 Py_BEGIN_ALLOW_THREADS
Benjamin Peterson5c6d7872009-02-06 02:40:07 +00003341 sp = getservbyport(htons((short)port), proto);
Barry Warsaw11b91a02004-06-28 00:50:43 +00003342 Py_END_ALLOW_THREADS
3343 if (sp == NULL) {
3344 PyErr_SetString(socket_error, "port/proto not found");
3345 return NULL;
3346 }
Guido van Rossum32c4ac02007-08-15 03:56:40 +00003347 return PyUnicode_FromString(sp->s_name);
Barry Warsaw11b91a02004-06-28 00:50:43 +00003348}
3349
3350PyDoc_STRVAR(getservbyport_doc,
3351"getservbyport(port[, protocolname]) -> string\n\
3352\n\
3353Return the service name from a port number and protocol name.\n\
3354The optional protocol name, if given, should be 'tcp' or 'udp',\n\
3355otherwise any protocol will match.");
3356
Guido van Rossum3901d851996-12-19 16:35:04 +00003357/* Python interface to getprotobyname(name).
3358 This only returns the protocol number, since the other info is
3359 already known or not useful (like the list of aliases). */
3360
3361/*ARGSUSED*/
3362static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003363socket_getprotobyname(PyObject *self, PyObject *args)
Guido van Rossum3901d851996-12-19 16:35:04 +00003364{
3365 char *name;
3366 struct protoent *sp;
Guido van Rossum27fc3c02000-03-24 20:56:56 +00003367 if (!PyArg_ParseTuple(args, "s:getprotobyname", &name))
Guido van Rossum3901d851996-12-19 16:35:04 +00003368 return NULL;
3369 Py_BEGIN_ALLOW_THREADS
3370 sp = getprotobyname(name);
3371 Py_END_ALLOW_THREADS
3372 if (sp == NULL) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003373 PyErr_SetString(socket_error, "protocol not found");
Guido van Rossum3901d851996-12-19 16:35:04 +00003374 return NULL;
3375 }
Christian Heimes217cfd12007-12-02 14:31:20 +00003376 return PyLong_FromLong((long) sp->p_proto);
Guido van Rossum3901d851996-12-19 16:35:04 +00003377}
3378
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003379PyDoc_STRVAR(getprotobyname_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003380"getprotobyname(name) -> integer\n\
3381\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003382Return the protocol number for the named protocol. (Rarely used.)");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003383
Guido van Rossum3901d851996-12-19 16:35:04 +00003384
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00003385#ifndef NO_DUP
3386/* dup() function for socket fds */
3387
3388static PyObject *
3389socket_dup(PyObject *self, PyObject *fdobj)
3390{
3391 SOCKET_T fd, newfd;
3392 PyObject *newfdobj;
3393
3394
3395 fd = PyLong_AsSocket_t(fdobj);
3396 if (fd == (SOCKET_T)(-1) && PyErr_Occurred())
3397 return NULL;
3398
3399 newfd = dup_socket(fd);
3400 if (newfd == INVALID_SOCKET)
3401 return set_error();
3402
3403 newfdobj = PyLong_FromSocket_t(newfd);
3404 if (newfdobj == NULL)
3405 SOCKETCLOSE(newfd);
3406 return newfdobj;
3407}
3408
3409PyDoc_STRVAR(dup_doc,
3410"dup(integer) -> integer\n\
3411\n\
3412Duplicate an integer socket file descriptor. This is like os.dup(), but for\n\
3413sockets; on some platforms os.dup() won't work for socket file descriptors.");
3414#endif
3415
3416
Dave Cole331708b2004-08-09 04:51:41 +00003417#ifdef HAVE_SOCKETPAIR
3418/* Create a pair of sockets using the socketpair() function.
Dave Cole07fda7e2004-08-23 05:16:23 +00003419 Arguments as for socket() except the default family is AF_UNIX if
Dave Colee8bbfe42004-08-26 00:51:16 +00003420 defined on the platform; otherwise, the default is AF_INET. */
Dave Cole331708b2004-08-09 04:51:41 +00003421
3422/*ARGSUSED*/
3423static PyObject *
3424socket_socketpair(PyObject *self, PyObject *args)
3425{
3426 PySocketSockObject *s0 = NULL, *s1 = NULL;
3427 SOCKET_T sv[2];
3428 int family, type = SOCK_STREAM, proto = 0;
3429 PyObject *res = NULL;
3430
3431#if defined(AF_UNIX)
3432 family = AF_UNIX;
3433#else
3434 family = AF_INET;
3435#endif
3436 if (!PyArg_ParseTuple(args, "|iii:socketpair",
3437 &family, &type, &proto))
3438 return NULL;
3439 /* Create a pair of socket fds */
3440 if (socketpair(family, type, proto, sv) < 0)
3441 return set_error();
Dave Cole331708b2004-08-09 04:51:41 +00003442 s0 = new_sockobject(sv[0], family, type, proto);
3443 if (s0 == NULL)
3444 goto finally;
3445 s1 = new_sockobject(sv[1], family, type, proto);
3446 if (s1 == NULL)
3447 goto finally;
3448 res = PyTuple_Pack(2, s0, s1);
3449
3450finally:
3451 if (res == NULL) {
3452 if (s0 == NULL)
3453 SOCKETCLOSE(sv[0]);
3454 if (s1 == NULL)
3455 SOCKETCLOSE(sv[1]);
3456 }
3457 Py_XDECREF(s0);
3458 Py_XDECREF(s1);
3459 return res;
3460}
3461
3462PyDoc_STRVAR(socketpair_doc,
3463"socketpair([family[, type[, proto]]]) -> (socket object, socket object)\n\
3464\n\
3465Create a pair of socket objects from the sockets returned by the platform\n\
3466socketpair() function.\n\
Dave Cole07fda7e2004-08-23 05:16:23 +00003467The arguments are the same as for socket() except the default family is\n\
Dave Colee8bbfe42004-08-26 00:51:16 +00003468AF_UNIX if defined on the platform; otherwise, the default is AF_INET.");
Dave Cole331708b2004-08-09 04:51:41 +00003469
3470#endif /* HAVE_SOCKETPAIR */
3471
3472
Guido van Rossum006bf911996-06-12 04:04:55 +00003473static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003474socket_ntohs(PyObject *self, PyObject *args)
Guido van Rossum006bf911996-06-12 04:04:55 +00003475{
3476 int x1, x2;
3477
Guido van Rossum27fc3c02000-03-24 20:56:56 +00003478 if (!PyArg_ParseTuple(args, "i:ntohs", &x1)) {
Guido van Rossum006bf911996-06-12 04:04:55 +00003479 return NULL;
3480 }
Guido van Rossum018919a2007-01-15 00:07:32 +00003481 if (x1 < 0) {
3482 PyErr_SetString(PyExc_OverflowError,
3483 "can't convert negative number to unsigned long");
3484 return NULL;
3485 }
3486 x2 = (unsigned int)ntohs((unsigned short)x1);
Christian Heimes217cfd12007-12-02 14:31:20 +00003487 return PyLong_FromLong(x2);
Guido van Rossum006bf911996-06-12 04:04:55 +00003488}
3489
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003490PyDoc_STRVAR(ntohs_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003491"ntohs(integer) -> integer\n\
3492\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003493Convert a 16-bit integer from network to host byte order.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003494
3495
Guido van Rossum006bf911996-06-12 04:04:55 +00003496static PyObject *
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003497socket_ntohl(PyObject *self, PyObject *arg)
Guido van Rossum006bf911996-06-12 04:04:55 +00003498{
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003499 unsigned long x;
Guido van Rossum006bf911996-06-12 04:04:55 +00003500
Guido van Rossumddefaf32007-01-14 03:31:43 +00003501 if (PyLong_Check(arg)) {
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003502 x = PyLong_AsUnsignedLong(arg);
Jeremy Hylton825e47b2002-07-25 16:37:51 +00003503 if (x == (unsigned long) -1 && PyErr_Occurred())
3504 return NULL;
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003505#if SIZEOF_LONG > 4
3506 {
3507 unsigned long y;
3508 /* only want the trailing 32 bits */
3509 y = x & 0xFFFFFFFFUL;
3510 if (y ^ x)
3511 return PyErr_Format(PyExc_OverflowError,
3512 "long int larger than 32 bits");
3513 x = y;
Tim Peters58141872002-08-06 22:25:02 +00003514 }
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003515#endif
3516 }
3517 else
Tim Peters58141872002-08-06 22:25:02 +00003518 return PyErr_Format(PyExc_TypeError,
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003519 "expected int/long, %s found",
Christian Heimes90aa7642007-12-19 02:45:37 +00003520 Py_TYPE(arg)->tp_name);
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003521 if (x == (unsigned long) -1 && PyErr_Occurred())
3522 return NULL;
Guido van Rossum018919a2007-01-15 00:07:32 +00003523 return PyLong_FromUnsignedLong(ntohl(x));
Guido van Rossum006bf911996-06-12 04:04:55 +00003524}
3525
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003526PyDoc_STRVAR(ntohl_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003527"ntohl(integer) -> integer\n\
3528\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003529Convert a 32-bit integer from network to host byte order.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003530
3531
Guido van Rossum006bf911996-06-12 04:04:55 +00003532static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003533socket_htons(PyObject *self, PyObject *args)
Guido van Rossum006bf911996-06-12 04:04:55 +00003534{
Hye-Shik Changa7b673f2003-12-17 09:40:23 +00003535 int x1, x2;
Guido van Rossum006bf911996-06-12 04:04:55 +00003536
Guido van Rossum27fc3c02000-03-24 20:56:56 +00003537 if (!PyArg_ParseTuple(args, "i:htons", &x1)) {
Guido van Rossum006bf911996-06-12 04:04:55 +00003538 return NULL;
3539 }
Guido van Rossum018919a2007-01-15 00:07:32 +00003540 if (x1 < 0) {
3541 PyErr_SetString(PyExc_OverflowError,
3542 "can't convert negative number to unsigned long");
3543 return NULL;
3544 }
3545 x2 = (unsigned int)htons((unsigned short)x1);
Christian Heimes217cfd12007-12-02 14:31:20 +00003546 return PyLong_FromLong(x2);
Guido van Rossum006bf911996-06-12 04:04:55 +00003547}
3548
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003549PyDoc_STRVAR(htons_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003550"htons(integer) -> integer\n\
3551\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003552Convert a 16-bit integer from host to network byte order.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003553
3554
Guido van Rossum006bf911996-06-12 04:04:55 +00003555static PyObject *
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003556socket_htonl(PyObject *self, PyObject *arg)
Guido van Rossum006bf911996-06-12 04:04:55 +00003557{
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003558 unsigned long x;
Guido van Rossum006bf911996-06-12 04:04:55 +00003559
Guido van Rossumddefaf32007-01-14 03:31:43 +00003560 if (PyLong_Check(arg)) {
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003561 x = PyLong_AsUnsignedLong(arg);
Jeremy Hylton825e47b2002-07-25 16:37:51 +00003562 if (x == (unsigned long) -1 && PyErr_Occurred())
3563 return NULL;
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003564#if SIZEOF_LONG > 4
3565 {
3566 unsigned long y;
3567 /* only want the trailing 32 bits */
3568 y = x & 0xFFFFFFFFUL;
3569 if (y ^ x)
3570 return PyErr_Format(PyExc_OverflowError,
3571 "long int larger than 32 bits");
3572 x = y;
Tim Peters58141872002-08-06 22:25:02 +00003573 }
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003574#endif
3575 }
3576 else
Tim Peters58141872002-08-06 22:25:02 +00003577 return PyErr_Format(PyExc_TypeError,
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003578 "expected int/long, %s found",
Christian Heimes90aa7642007-12-19 02:45:37 +00003579 Py_TYPE(arg)->tp_name);
Guido van Rossum018919a2007-01-15 00:07:32 +00003580 return PyLong_FromUnsignedLong(htonl((unsigned long)x));
Guido van Rossum006bf911996-06-12 04:04:55 +00003581}
3582
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003583PyDoc_STRVAR(htonl_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003584"htonl(integer) -> integer\n\
3585\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003586Convert a 32-bit integer from host to network byte order.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003587
Guido van Rossum3eede5a2002-06-07 02:08:35 +00003588/* socket.inet_aton() and socket.inet_ntoa() functions. */
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003589
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003590PyDoc_STRVAR(inet_aton_doc,
Guido van Rossum7d0a8262007-05-21 23:13:11 +00003591"inet_aton(string) -> bytes giving packed 32-bit IP representation\n\
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003592\n\
Guido van Rossumc6a164b1999-08-20 19:11:27 +00003593Convert an IP address in string format (123.45.67.89) to the 32-bit packed\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003594binary format used in low-level network functions.");
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003595
3596static PyObject*
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003597socket_inet_aton(PyObject *self, PyObject *args)
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003598{
Guido van Rossuma2e48551999-09-09 15:42:59 +00003599#ifndef INADDR_NONE
3600#define INADDR_NONE (-1)
3601#endif
Neal Norwitz88f115b2003-02-13 02:15:42 +00003602#ifdef HAVE_INET_ATON
Guido van Rossumad05cdf2003-02-12 23:08:22 +00003603 struct in_addr buf;
Thomas Wouters477c8d52006-05-27 19:21:47 +00003604#endif
3605
3606#if !defined(HAVE_INET_ATON) || defined(USE_INET_ATON_WEAKLINK)
Benjamin Petersonf91df042009-02-13 02:50:59 +00003607#if (SIZEOF_INT != 4)
3608#error "Not sure if in_addr_t exists and int is not 32-bits."
3609#endif
Tim Peters1df9fdd2003-02-13 03:13:40 +00003610 /* Have to use inet_addr() instead */
Benjamin Petersonf91df042009-02-13 02:50:59 +00003611 unsigned int packed_addr;
Tim Peters1df9fdd2003-02-13 03:13:40 +00003612#endif
3613 char *ip_addr;
Guido van Rossumc6a164b1999-08-20 19:11:27 +00003614
Tim Peters1df9fdd2003-02-13 03:13:40 +00003615 if (!PyArg_ParseTuple(args, "s:inet_aton", &ip_addr))
Guido van Rossumc6a164b1999-08-20 19:11:27 +00003616 return NULL;
Guido van Rossumad05cdf2003-02-12 23:08:22 +00003617
Tim Peters1df9fdd2003-02-13 03:13:40 +00003618
3619#ifdef HAVE_INET_ATON
Thomas Wouters477c8d52006-05-27 19:21:47 +00003620
3621#ifdef USE_INET_ATON_WEAKLINK
3622 if (inet_aton != NULL) {
3623#endif
Guido van Rossumad05cdf2003-02-12 23:08:22 +00003624 if (inet_aton(ip_addr, &buf))
Christian Heimes72b710a2008-05-26 13:28:38 +00003625 return PyBytes_FromStringAndSize((char *)(&buf),
Guido van Rossumad05cdf2003-02-12 23:08:22 +00003626 sizeof(buf));
3627
3628 PyErr_SetString(socket_error,
3629 "illegal IP address string passed to inet_aton");
3630 return NULL;
3631
Thomas Wouters477c8d52006-05-27 19:21:47 +00003632#ifdef USE_INET_ATON_WEAKLINK
3633 } else {
3634#endif
3635
3636#endif
3637
3638#if !defined(HAVE_INET_ATON) || defined(USE_INET_ATON_WEAKLINK)
3639
Georg Brandld2e3ba72005-08-26 08:34:00 +00003640 /* special-case this address as inet_addr might return INADDR_NONE
3641 * for this */
3642 if (strcmp(ip_addr, "255.255.255.255") == 0) {
3643 packed_addr = 0xFFFFFFFF;
3644 } else {
Thomas Wouters477c8d52006-05-27 19:21:47 +00003645
Georg Brandld2e3ba72005-08-26 08:34:00 +00003646 packed_addr = inet_addr(ip_addr);
Guido van Rossumc6a164b1999-08-20 19:11:27 +00003647
Georg Brandld2e3ba72005-08-26 08:34:00 +00003648 if (packed_addr == INADDR_NONE) { /* invalid address */
3649 PyErr_SetString(socket_error,
3650 "illegal IP address string passed to inet_aton");
3651 return NULL;
3652 }
Guido van Rossumc6a164b1999-08-20 19:11:27 +00003653 }
Christian Heimes72b710a2008-05-26 13:28:38 +00003654 return PyBytes_FromStringAndSize((char *) &packed_addr,
Guido van Rossum8a392d72007-11-21 22:09:45 +00003655 sizeof(packed_addr));
Thomas Wouters477c8d52006-05-27 19:21:47 +00003656
3657#ifdef USE_INET_ATON_WEAKLINK
3658 }
3659#endif
3660
Guido van Rossumad05cdf2003-02-12 23:08:22 +00003661#endif
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003662}
3663
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003664PyDoc_STRVAR(inet_ntoa_doc,
Fred Drakee0661342000-03-07 14:05:16 +00003665"inet_ntoa(packed_ip) -> ip_address_string\n\
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003666\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003667Convert an IP address from 32-bit packed binary format to string format");
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003668
3669static PyObject*
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003670socket_inet_ntoa(PyObject *self, PyObject *args)
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003671{
3672 char *packed_str;
Guido van Rossumc6a164b1999-08-20 19:11:27 +00003673 int addr_len;
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003674 struct in_addr packed_addr;
3675
Guido van Rossum32c4ac02007-08-15 03:56:40 +00003676 if (!PyArg_ParseTuple(args, "y#:inet_ntoa", &packed_str, &addr_len)) {
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003677 return NULL;
3678 }
Guido van Rossum48a680c2001-03-02 06:34:14 +00003679
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003680 if (addr_len != sizeof(packed_addr)) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003681 PyErr_SetString(socket_error,
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003682 "packed IP wrong length for inet_ntoa");
3683 return NULL;
3684 }
3685
3686 memcpy(&packed_addr, packed_str, addr_len);
3687
Guido van Rossum32c4ac02007-08-15 03:56:40 +00003688 return PyUnicode_FromString(inet_ntoa(packed_addr));
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003689}
Guido van Rossum82a5c661998-07-07 20:45:43 +00003690
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003691#ifdef HAVE_INET_PTON
3692
3693PyDoc_STRVAR(inet_pton_doc,
3694"inet_pton(af, ip) -> packed IP address string\n\
3695\n\
3696Convert an IP address from string format to a packed string suitable\n\
3697for use with low-level network functions.");
3698
3699static PyObject *
3700socket_inet_pton(PyObject *self, PyObject *args)
3701{
3702 int af;
3703 char* ip;
3704 int retval;
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003705#ifdef ENABLE_IPV6
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003706 char packed[MAX(sizeof(struct in_addr), sizeof(struct in6_addr))];
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003707#else
3708 char packed[sizeof(struct in_addr)];
3709#endif
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003710 if (!PyArg_ParseTuple(args, "is:inet_pton", &af, &ip)) {
3711 return NULL;
3712 }
3713
Martin v. Löwis04697e82004-06-02 12:35:29 +00003714#if !defined(ENABLE_IPV6) && defined(AF_INET6)
Martin v. Löwis10649092003-08-05 06:25:06 +00003715 if(af == AF_INET6) {
3716 PyErr_SetString(socket_error,
3717 "can't use AF_INET6, IPv6 is disabled");
3718 return NULL;
3719 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00003720#endif
Martin v. Löwis10649092003-08-05 06:25:06 +00003721
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003722 retval = inet_pton(af, ip, packed);
3723 if (retval < 0) {
3724 PyErr_SetFromErrno(socket_error);
3725 return NULL;
3726 } else if (retval == 0) {
3727 PyErr_SetString(socket_error,
3728 "illegal IP address string passed to inet_pton");
3729 return NULL;
3730 } else if (af == AF_INET) {
Christian Heimes72b710a2008-05-26 13:28:38 +00003731 return PyBytes_FromStringAndSize(packed,
Guido van Rossum8a392d72007-11-21 22:09:45 +00003732 sizeof(struct in_addr));
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003733#ifdef ENABLE_IPV6
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003734 } else if (af == AF_INET6) {
Christian Heimes72b710a2008-05-26 13:28:38 +00003735 return PyBytes_FromStringAndSize(packed,
Guido van Rossum8a392d72007-11-21 22:09:45 +00003736 sizeof(struct in6_addr));
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003737#endif
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003738 } else {
3739 PyErr_SetString(socket_error, "unknown address family");
3740 return NULL;
3741 }
3742}
Thomas Wouters477c8d52006-05-27 19:21:47 +00003743
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003744PyDoc_STRVAR(inet_ntop_doc,
3745"inet_ntop(af, packed_ip) -> string formatted IP address\n\
3746\n\
3747Convert a packed IP address of the given family to string format.");
3748
3749static PyObject *
3750socket_inet_ntop(PyObject *self, PyObject *args)
3751{
3752 int af;
3753 char* packed;
3754 int len;
3755 const char* retval;
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003756#ifdef ENABLE_IPV6
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003757 char ip[MAX(INET_ADDRSTRLEN, INET6_ADDRSTRLEN) + 1];
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003758#else
3759 char ip[INET_ADDRSTRLEN + 1];
3760#endif
Thomas Wouters477c8d52006-05-27 19:21:47 +00003761
Guido van Rossum32c4ac02007-08-15 03:56:40 +00003762 /* Guarantee NUL-termination for PyUnicode_FromString() below */
Jeremy Hylton80961f32004-11-07 14:24:25 +00003763 memset((void *) &ip[0], '\0', sizeof(ip));
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003764
Guido van Rossum32c4ac02007-08-15 03:56:40 +00003765 if (!PyArg_ParseTuple(args, "iy#:inet_ntop", &af, &packed, &len)) {
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003766 return NULL;
3767 }
3768
3769 if (af == AF_INET) {
3770 if (len != sizeof(struct in_addr)) {
3771 PyErr_SetString(PyExc_ValueError,
3772 "invalid length of packed IP address string");
3773 return NULL;
3774 }
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003775#ifdef ENABLE_IPV6
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003776 } else if (af == AF_INET6) {
3777 if (len != sizeof(struct in6_addr)) {
3778 PyErr_SetString(PyExc_ValueError,
3779 "invalid length of packed IP address string");
3780 return NULL;
3781 }
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003782#endif
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003783 } else {
3784 PyErr_Format(PyExc_ValueError,
3785 "unknown address family %d", af);
3786 return NULL;
3787 }
3788
3789 retval = inet_ntop(af, packed, ip, sizeof(ip));
3790 if (!retval) {
3791 PyErr_SetFromErrno(socket_error);
3792 return NULL;
3793 } else {
Guido van Rossum32c4ac02007-08-15 03:56:40 +00003794 return PyUnicode_FromString(retval);
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003795 }
3796
3797 /* NOTREACHED */
3798 PyErr_SetString(PyExc_RuntimeError, "invalid handling of inet_ntop");
3799 return NULL;
3800}
3801
3802#endif /* HAVE_INET_PTON */
3803
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003804/* Python interface to getaddrinfo(host, port). */
3805
3806/*ARGSUSED*/
3807static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003808socket_getaddrinfo(PyObject *self, PyObject *args)
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003809{
Martin v. Löwisf0b11d22001-11-07 08:31:03 +00003810 struct addrinfo hints, *res;
3811 struct addrinfo *res0 = NULL;
Martin v. Löwis2548c732003-04-18 10:39:54 +00003812 PyObject *hobj = NULL;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003813 PyObject *pobj = (PyObject *)NULL;
Martin v. Löwisf65b1a12001-08-12 09:28:40 +00003814 char pbuf[30];
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003815 char *hptr, *pptr;
3816 int family, socktype, protocol, flags;
3817 int error;
3818 PyObject *all = (PyObject *)NULL;
Martin v. Löwis2548c732003-04-18 10:39:54 +00003819 PyObject *idna = NULL;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003820
3821 family = socktype = protocol = flags = 0;
Martin v. Löwiscf8f47e2002-12-11 13:10:57 +00003822 family = AF_UNSPEC;
Martin v. Löwis2548c732003-04-18 10:39:54 +00003823 if (!PyArg_ParseTuple(args, "OO|iiii:getaddrinfo",
3824 &hobj, &pobj, &family, &socktype,
3825 &protocol, &flags)) {
3826 return NULL;
3827 }
3828 if (hobj == Py_None) {
3829 hptr = NULL;
3830 } else if (PyUnicode_Check(hobj)) {
3831 idna = PyObject_CallMethod(hobj, "encode", "s", "idna");
3832 if (!idna)
3833 return NULL;
Christian Heimes72b710a2008-05-26 13:28:38 +00003834 assert(PyBytes_Check(idna));
3835 hptr = PyBytes_AS_STRING(idna);
3836 } else if (PyBytes_Check(hobj)) {
3837 hptr = PyBytes_AsString(hobj);
Martin v. Löwis2548c732003-04-18 10:39:54 +00003838 } else {
Thomas Wouters477c8d52006-05-27 19:21:47 +00003839 PyErr_SetString(PyExc_TypeError,
Martin v. Löwis2548c732003-04-18 10:39:54 +00003840 "getaddrinfo() argument 1 must be string or None");
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003841 return NULL;
3842 }
Martin v. Löwisd1a1d1e2007-12-04 22:10:37 +00003843 if (PyLong_CheckExact(pobj)) {
3844 long value = PyLong_AsLong(pobj);
3845 if (value == -1 && PyErr_Occurred())
3846 goto err;
3847 PyOS_snprintf(pbuf, sizeof(pbuf), "%ld", value);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003848 pptr = pbuf;
Guido van Rossum32c4ac02007-08-15 03:56:40 +00003849 } else if (PyUnicode_Check(pobj)) {
Marc-André Lemburg4cc0f242008-08-07 18:54:33 +00003850 pptr = _PyUnicode_AsString(pobj);
Christian Heimes72b710a2008-05-26 13:28:38 +00003851 } else if (PyBytes_Check(pobj)) {
3852 pptr = PyBytes_AsString(pobj);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003853 } else if (pobj == Py_None) {
3854 pptr = (char *)NULL;
3855 } else {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003856 PyErr_SetString(socket_error, "Int or String expected");
Martin v. Löwis5db099a2003-08-07 11:55:15 +00003857 goto err;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003858 }
3859 memset(&hints, 0, sizeof(hints));
3860 hints.ai_family = family;
3861 hints.ai_socktype = socktype;
3862 hints.ai_protocol = protocol;
3863 hints.ai_flags = flags;
Just van Rossum1040d2c2003-05-09 07:53:18 +00003864 Py_BEGIN_ALLOW_THREADS
3865 ACQUIRE_GETADDRINFO_LOCK
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003866 error = getaddrinfo(hptr, pptr, &hints, &res0);
Just van Rossum1040d2c2003-05-09 07:53:18 +00003867 Py_END_ALLOW_THREADS
3868 RELEASE_GETADDRINFO_LOCK /* see comment in setipaddr() */
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003869 if (error) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003870 set_gaierror(error);
Martin v. Löwis5db099a2003-08-07 11:55:15 +00003871 goto err;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003872 }
3873
3874 if ((all = PyList_New(0)) == NULL)
3875 goto err;
3876 for (res = res0; res; res = res->ai_next) {
Guido van Rossum9ce8e382007-08-09 21:06:37 +00003877 PyObject *single;
Guido van Rossum716aac02001-10-12 18:59:27 +00003878 PyObject *addr =
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00003879 makesockaddr(-1, res->ai_addr, res->ai_addrlen, protocol);
Guido van Rossum716aac02001-10-12 18:59:27 +00003880 if (addr == NULL)
3881 goto err;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003882 single = Py_BuildValue("iiisO", res->ai_family,
3883 res->ai_socktype, res->ai_protocol,
3884 res->ai_canonname ? res->ai_canonname : "",
Guido van Rossum716aac02001-10-12 18:59:27 +00003885 addr);
3886 Py_DECREF(addr);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003887 if (single == NULL)
3888 goto err;
3889
3890 if (PyList_Append(all, single))
3891 goto err;
3892 Py_XDECREF(single);
3893 }
Martin v. Löwis2548c732003-04-18 10:39:54 +00003894 Py_XDECREF(idna);
Neal Norwitz90128ba2002-08-09 03:37:42 +00003895 if (res0)
3896 freeaddrinfo(res0);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003897 return all;
3898 err:
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003899 Py_XDECREF(all);
Martin v. Löwis2548c732003-04-18 10:39:54 +00003900 Py_XDECREF(idna);
Martin v. Löwisf0b11d22001-11-07 08:31:03 +00003901 if (res0)
3902 freeaddrinfo(res0);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003903 return (PyObject *)NULL;
3904}
3905
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003906PyDoc_STRVAR(getaddrinfo_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00003907"getaddrinfo(host, port [, family, socktype, proto, flags])\n\
3908 -> list of (family, socktype, proto, canonname, sockaddr)\n\
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003909\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003910Resolve host and port into addrinfo struct.");
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003911
3912/* Python interface to getnameinfo(sa, flags). */
3913
3914/*ARGSUSED*/
3915static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003916socket_getnameinfo(PyObject *self, PyObject *args)
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003917{
3918 PyObject *sa = (PyObject *)NULL;
3919 int flags;
3920 char *hostp;
Martin v. Löwis06b1d212001-11-02 23:34:52 +00003921 int port, flowinfo, scope_id;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003922 char hbuf[NI_MAXHOST], pbuf[NI_MAXSERV];
3923 struct addrinfo hints, *res = NULL;
3924 int error;
3925 PyObject *ret = (PyObject *)NULL;
3926
3927 flags = flowinfo = scope_id = 0;
Martin v. Löwis06b1d212001-11-02 23:34:52 +00003928 if (!PyArg_ParseTuple(args, "Oi:getnameinfo", &sa, &flags))
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003929 return NULL;
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +00003930 if (!PyTuple_Check(sa)) {
3931 PyErr_SetString(PyExc_TypeError,
3932 "getnameinfo() argument 1 must be a tuple");
3933 return NULL;
3934 }
3935 if (!PyArg_ParseTuple(sa, "si|ii",
3936 &hostp, &port, &flowinfo, &scope_id))
Martin v. Löwis06b1d212001-11-02 23:34:52 +00003937 return NULL;
Marc-André Lemburg03d1b182001-07-31 18:05:33 +00003938 PyOS_snprintf(pbuf, sizeof(pbuf), "%d", port);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003939 memset(&hints, 0, sizeof(hints));
Martin v. Löwiscf8f47e2002-12-11 13:10:57 +00003940 hints.ai_family = AF_UNSPEC;
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00003941 hints.ai_socktype = SOCK_DGRAM; /* make numeric port happy */
Just van Rossum1040d2c2003-05-09 07:53:18 +00003942 Py_BEGIN_ALLOW_THREADS
3943 ACQUIRE_GETADDRINFO_LOCK
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003944 error = getaddrinfo(hostp, pbuf, &hints, &res);
Just van Rossum1040d2c2003-05-09 07:53:18 +00003945 Py_END_ALLOW_THREADS
3946 RELEASE_GETADDRINFO_LOCK /* see comment in setipaddr() */
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003947 if (error) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003948 set_gaierror(error);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003949 goto fail;
3950 }
3951 if (res->ai_next) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003952 PyErr_SetString(socket_error,
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003953 "sockaddr resolved to multiple addresses");
3954 goto fail;
3955 }
3956 switch (res->ai_family) {
3957 case AF_INET:
3958 {
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +00003959 if (PyTuple_GET_SIZE(sa) != 2) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003960 PyErr_SetString(socket_error,
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003961 "IPv4 sockaddr must be 2 tuple");
3962 goto fail;
3963 }
3964 break;
3965 }
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00003966#ifdef ENABLE_IPV6
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003967 case AF_INET6:
3968 {
3969 struct sockaddr_in6 *sin6;
3970 sin6 = (struct sockaddr_in6 *)res->ai_addr;
3971 sin6->sin6_flowinfo = flowinfo;
3972 sin6->sin6_scope_id = scope_id;
3973 break;
3974 }
3975#endif
3976 }
3977 error = getnameinfo(res->ai_addr, res->ai_addrlen,
3978 hbuf, sizeof(hbuf), pbuf, sizeof(pbuf), flags);
3979 if (error) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003980 set_gaierror(error);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003981 goto fail;
3982 }
3983 ret = Py_BuildValue("ss", hbuf, pbuf);
3984
3985fail:
3986 if (res)
3987 freeaddrinfo(res);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003988 return ret;
3989}
3990
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003991PyDoc_STRVAR(getnameinfo_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00003992"getnameinfo(sockaddr, flags) --> (host, port)\n\
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003993\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003994Get host and port for a sockaddr.");
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003995
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00003996
3997/* Python API to getting and setting the default timeout value. */
3998
3999static PyObject *
4000socket_getdefaulttimeout(PyObject *self)
4001{
4002 if (defaulttimeout < 0.0) {
4003 Py_INCREF(Py_None);
4004 return Py_None;
4005 }
4006 else
4007 return PyFloat_FromDouble(defaulttimeout);
4008}
4009
4010PyDoc_STRVAR(getdefaulttimeout_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00004011"getdefaulttimeout() -> timeout\n\
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004012\n\
4013Returns the default timeout in floating seconds for new socket objects.\n\
4014A value of None indicates that new socket objects have no timeout.\n\
4015When the socket module is first imported, the default is None.");
4016
4017static PyObject *
4018socket_setdefaulttimeout(PyObject *self, PyObject *arg)
4019{
4020 double timeout;
4021
4022 if (arg == Py_None)
4023 timeout = -1.0;
4024 else {
4025 timeout = PyFloat_AsDouble(arg);
4026 if (timeout < 0.0) {
4027 if (!PyErr_Occurred())
4028 PyErr_SetString(PyExc_ValueError,
4029 "Timeout value out of range");
4030 return NULL;
4031 }
4032 }
4033
4034 defaulttimeout = timeout;
4035
4036 Py_INCREF(Py_None);
4037 return Py_None;
4038}
4039
4040PyDoc_STRVAR(setdefaulttimeout_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00004041"setdefaulttimeout(timeout)\n\
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004042\n\
4043Set the default timeout in floating seconds for new socket objects.\n\
4044A value of None indicates that new socket objects have no timeout.\n\
4045When the socket module is first imported, the default is None.");
4046
4047
Guido van Rossum30a685f1991-06-27 15:51:29 +00004048/* List of functions exported by this module. */
4049
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004050static PyMethodDef socket_methods[] = {
4051 {"gethostbyname", socket_gethostbyname,
Andrew M. Kuchlinge365fb82000-08-03 02:06:16 +00004052 METH_VARARGS, gethostbyname_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004053 {"gethostbyname_ex", socket_gethostbyname_ex,
Andrew M. Kuchlinge365fb82000-08-03 02:06:16 +00004054 METH_VARARGS, ghbn_ex_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004055 {"gethostbyaddr", socket_gethostbyaddr,
Andrew M. Kuchlinge365fb82000-08-03 02:06:16 +00004056 METH_VARARGS, gethostbyaddr_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004057 {"gethostname", socket_gethostname,
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00004058 METH_NOARGS, gethostname_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004059 {"getservbyname", socket_getservbyname,
Andrew M. Kuchlinge365fb82000-08-03 02:06:16 +00004060 METH_VARARGS, getservbyname_doc},
Barry Warsaw11b91a02004-06-28 00:50:43 +00004061 {"getservbyport", socket_getservbyport,
4062 METH_VARARGS, getservbyport_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004063 {"getprotobyname", socket_getprotobyname,
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00004064 METH_VARARGS, getprotobyname_doc},
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00004065#ifndef NO_DUP
4066 {"dup", socket_dup,
4067 METH_O, dup_doc},
4068#endif
Dave Cole331708b2004-08-09 04:51:41 +00004069#ifdef HAVE_SOCKETPAIR
4070 {"socketpair", socket_socketpair,
4071 METH_VARARGS, socketpair_doc},
4072#endif
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004073 {"ntohs", socket_ntohs,
Andrew M. Kuchlinge365fb82000-08-03 02:06:16 +00004074 METH_VARARGS, ntohs_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004075 {"ntohl", socket_ntohl,
Jeremy Hyltonc075e192002-07-25 16:01:12 +00004076 METH_O, ntohl_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004077 {"htons", socket_htons,
Andrew M. Kuchlinge365fb82000-08-03 02:06:16 +00004078 METH_VARARGS, htons_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004079 {"htonl", socket_htonl,
Jeremy Hyltonc075e192002-07-25 16:01:12 +00004080 METH_O, htonl_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004081 {"inet_aton", socket_inet_aton,
Andrew M. Kuchlinge365fb82000-08-03 02:06:16 +00004082 METH_VARARGS, inet_aton_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004083 {"inet_ntoa", socket_inet_ntoa,
Andrew M. Kuchlinge365fb82000-08-03 02:06:16 +00004084 METH_VARARGS, inet_ntoa_doc},
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00004085#ifdef HAVE_INET_PTON
4086 {"inet_pton", socket_inet_pton,
4087 METH_VARARGS, inet_pton_doc},
4088 {"inet_ntop", socket_inet_ntop,
4089 METH_VARARGS, inet_ntop_doc},
4090#endif
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004091 {"getaddrinfo", socket_getaddrinfo,
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004092 METH_VARARGS, getaddrinfo_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004093 {"getnameinfo", socket_getnameinfo,
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004094 METH_VARARGS, getnameinfo_doc},
Guido van Rossum1693ba82002-07-18 21:11:26 +00004095 {"getdefaulttimeout", (PyCFunction)socket_getdefaulttimeout,
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004096 METH_NOARGS, getdefaulttimeout_doc},
4097 {"setdefaulttimeout", socket_setdefaulttimeout,
4098 METH_O, setdefaulttimeout_doc},
Guido van Rossum30a685f1991-06-27 15:51:29 +00004099 {NULL, NULL} /* Sentinel */
Guido van Rossum6574b3e1991-06-25 21:36:08 +00004100};
4101
Guido van Rossum30a685f1991-06-27 15:51:29 +00004102
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004103#ifdef MS_WINDOWS
4104#define OS_INIT_DEFINED
4105
4106/* Additional initialization and cleanup for Windows */
Guido van Rossumbe32c891996-06-20 16:25:29 +00004107
4108static void
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004109os_cleanup(void)
Guido van Rossumbe32c891996-06-20 16:25:29 +00004110{
4111 WSACleanup();
Guido van Rossumbe32c891996-06-20 16:25:29 +00004112}
4113
4114static int
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004115os_init(void)
Guido van Rossumbe32c891996-06-20 16:25:29 +00004116{
4117 WSADATA WSAData;
4118 int ret;
Guido van Rossumbe32c891996-06-20 16:25:29 +00004119 ret = WSAStartup(0x0101, &WSAData);
4120 switch (ret) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004121 case 0: /* No error */
Mark Hammond14350ab2003-07-09 04:57:46 +00004122 Py_AtExit(os_cleanup);
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004123 return 1; /* Success */
Guido van Rossumbe32c891996-06-20 16:25:29 +00004124 case WSASYSNOTREADY:
4125 PyErr_SetString(PyExc_ImportError,
4126 "WSAStartup failed: network not ready");
4127 break;
4128 case WSAVERNOTSUPPORTED:
4129 case WSAEINVAL:
Guido van Rossum3eede5a2002-06-07 02:08:35 +00004130 PyErr_SetString(
4131 PyExc_ImportError,
4132 "WSAStartup failed: requested version not supported");
Guido van Rossumbe32c891996-06-20 16:25:29 +00004133 break;
4134 default:
Walter Dörwald0bbd8ab2007-06-05 13:49:43 +00004135 PyErr_Format(PyExc_ImportError, "WSAStartup failed: error code %d", ret);
Guido van Rossumbe32c891996-06-20 16:25:29 +00004136 break;
4137 }
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004138 return 0; /* Failure */
Guido van Rossumbe32c891996-06-20 16:25:29 +00004139}
4140
Guido van Rossum8d665e61996-06-26 18:22:49 +00004141#endif /* MS_WINDOWS */
Guido van Rossumbe32c891996-06-20 16:25:29 +00004142
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004143
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004144#ifdef PYOS_OS2
4145#define OS_INIT_DEFINED
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004146
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004147/* Additional initialization for OS/2 */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004148
4149static int
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004150os_init(void)
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004151{
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004152#ifndef PYCC_GCC
Guido van Rossum3eede5a2002-06-07 02:08:35 +00004153 int rc = sock_init();
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004154
Guido van Rossum3eede5a2002-06-07 02:08:35 +00004155 if (rc == 0) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004156 return 1; /* Success */
Guido van Rossum3eede5a2002-06-07 02:08:35 +00004157 }
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004158
Walter Dörwald0bbd8ab2007-06-05 13:49:43 +00004159 PyErr_Format(PyExc_ImportError, "OS/2 TCP/IP Error# %d", sock_errno());
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004160
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004161 return 0; /* Failure */
Andrew MacIntyreba43e872002-03-03 03:03:52 +00004162#else
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004163 /* No need to initialise sockets with GCC/EMX */
4164 return 1; /* Success */
Andrew MacIntyreba43e872002-03-03 03:03:52 +00004165#endif
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004166}
4167
4168#endif /* PYOS_OS2 */
4169
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004170
4171#ifndef OS_INIT_DEFINED
4172static int
4173os_init(void)
4174{
4175 return 1; /* Success */
4176}
4177#endif
4178
4179
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00004180/* C API table - always add new things to the end for binary
4181 compatibility. */
4182static
4183PySocketModule_APIObject PySocketModuleAPI =
4184{
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004185 &sock_type,
Brett Cannon06c34792004-03-23 23:16:54 +00004186 NULL
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00004187};
4188
Guido van Rossum3eede5a2002-06-07 02:08:35 +00004189
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004190/* Initialize the _socket module.
Guido van Rossum3eede5a2002-06-07 02:08:35 +00004191
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004192 This module is actually called "_socket", and there's a wrapper
Guido van Rossum7d0a8262007-05-21 23:13:11 +00004193 "socket.py" which implements some additional functionality.
4194 The import of "_socket" may fail with an ImportError exception if
4195 os-specific initialization fails. On Windows, this does WINSOCK
4196 initialization. When WINSOCK is initialized succesfully, a call to
4197 WSACleanup() is scheduled to be made at exit time.
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004198*/
Guido van Rossum3eede5a2002-06-07 02:08:35 +00004199
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004200PyDoc_STRVAR(socket_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00004201"Implementation module for socket operations.\n\
4202\n\
4203See the socket module for documentation.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00004204
Martin v. Löwis1a214512008-06-11 05:26:20 +00004205static struct PyModuleDef socketmodule = {
4206 PyModuleDef_HEAD_INIT,
4207 PySocket_MODULE_NAME,
4208 socket_doc,
4209 -1,
4210 socket_methods,
4211 NULL,
4212 NULL,
4213 NULL,
4214 NULL
4215};
4216
Mark Hammondfe51c6d2002-08-02 02:27:13 +00004217PyMODINIT_FUNC
Martin v. Löwis1a214512008-06-11 05:26:20 +00004218PyInit__socket(void)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00004219{
Christian Heimes25bb7832008-01-11 16:17:00 +00004220 PyObject *m, *has_ipv6;
Fred Drake4baedc12002-04-01 14:53:37 +00004221
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004222 if (!os_init())
Martin v. Löwis1a214512008-06-11 05:26:20 +00004223 return NULL;
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004224
Christian Heimes90aa7642007-12-19 02:45:37 +00004225 Py_TYPE(&sock_type) = &PyType_Type;
Martin v. Löwis1a214512008-06-11 05:26:20 +00004226 m = PyModule_Create(&socketmodule);
Neal Norwitz1ac754f2006-01-19 06:09:39 +00004227 if (m == NULL)
Martin v. Löwis1a214512008-06-11 05:26:20 +00004228 return NULL;
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004229
Gregory P. Smithc111d9f2007-09-09 23:55:55 +00004230 socket_error = PyErr_NewException("socket.error",
4231 PyExc_IOError, NULL);
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004232 if (socket_error == NULL)
Martin v. Löwis1a214512008-06-11 05:26:20 +00004233 return NULL;
Brett Cannon06c34792004-03-23 23:16:54 +00004234 PySocketModuleAPI.error = socket_error;
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004235 Py_INCREF(socket_error);
4236 PyModule_AddObject(m, "error", socket_error);
4237 socket_herror = PyErr_NewException("socket.herror",
4238 socket_error, NULL);
4239 if (socket_herror == NULL)
Martin v. Löwis1a214512008-06-11 05:26:20 +00004240 return NULL;
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004241 Py_INCREF(socket_herror);
4242 PyModule_AddObject(m, "herror", socket_herror);
4243 socket_gaierror = PyErr_NewException("socket.gaierror", socket_error,
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004244 NULL);
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004245 if (socket_gaierror == NULL)
Martin v. Löwis1a214512008-06-11 05:26:20 +00004246 return NULL;
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004247 Py_INCREF(socket_gaierror);
4248 PyModule_AddObject(m, "gaierror", socket_gaierror);
Raymond Hettingeref7343c2003-06-29 03:08:05 +00004249 socket_timeout = PyErr_NewException("socket.timeout",
4250 socket_error, NULL);
4251 if (socket_timeout == NULL)
Martin v. Löwis1a214512008-06-11 05:26:20 +00004252 return NULL;
Raymond Hettingeref7343c2003-06-29 03:08:05 +00004253 Py_INCREF(socket_timeout);
4254 PyModule_AddObject(m, "timeout", socket_timeout);
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004255 Py_INCREF((PyObject *)&sock_type);
Fred Drake4baedc12002-04-01 14:53:37 +00004256 if (PyModule_AddObject(m, "SocketType",
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004257 (PyObject *)&sock_type) != 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00004258 return NULL;
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004259 Py_INCREF((PyObject *)&sock_type);
Fred Drake4baedc12002-04-01 14:53:37 +00004260 if (PyModule_AddObject(m, "socket",
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004261 (PyObject *)&sock_type) != 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00004262 return NULL;
Guido van Rossum09be4091999-08-09 14:40:40 +00004263
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00004264#ifdef ENABLE_IPV6
4265 has_ipv6 = Py_True;
4266#else
4267 has_ipv6 = Py_False;
4268#endif
4269 Py_INCREF(has_ipv6);
4270 PyModule_AddObject(m, "has_ipv6", has_ipv6);
4271
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00004272 /* Export C API */
Fred Drake4baedc12002-04-01 14:53:37 +00004273 if (PyModule_AddObject(m, PySocket_CAPI_NAME,
Benjamin Petersonb173f782009-05-05 22:31:58 +00004274 PyCapsule_New(&PySocketModuleAPI, PySocket_CAPSULE_NAME, NULL)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00004275 ) != 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00004276 return NULL;
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00004277
Guido van Rossum09be4091999-08-09 14:40:40 +00004278 /* Address families (we only support AF_INET and AF_UNIX) */
4279#ifdef AF_UNSPEC
Fred Drake4baedc12002-04-01 14:53:37 +00004280 PyModule_AddIntConstant(m, "AF_UNSPEC", AF_UNSPEC);
Guido van Rossum09be4091999-08-09 14:40:40 +00004281#endif
Fred Drake4baedc12002-04-01 14:53:37 +00004282 PyModule_AddIntConstant(m, "AF_INET", AF_INET);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004283#ifdef AF_INET6
Fred Drake4baedc12002-04-01 14:53:37 +00004284 PyModule_AddIntConstant(m, "AF_INET6", AF_INET6);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004285#endif /* AF_INET6 */
Andrew MacIntyred12dfbb2004-04-04 07:13:49 +00004286#if defined(AF_UNIX)
Fred Drake4baedc12002-04-01 14:53:37 +00004287 PyModule_AddIntConstant(m, "AF_UNIX", AF_UNIX);
Guido van Rossumb6775db1994-08-01 11:34:53 +00004288#endif /* AF_UNIX */
Guido van Rossum09be4091999-08-09 14:40:40 +00004289#ifdef AF_AX25
Fred Drake4baedc12002-04-01 14:53:37 +00004290 /* Amateur Radio AX.25 */
4291 PyModule_AddIntConstant(m, "AF_AX25", AF_AX25);
Guido van Rossum09be4091999-08-09 14:40:40 +00004292#endif
4293#ifdef AF_IPX
Fred Drake4baedc12002-04-01 14:53:37 +00004294 PyModule_AddIntConstant(m, "AF_IPX", AF_IPX); /* Novell IPX */
Guido van Rossum09be4091999-08-09 14:40:40 +00004295#endif
4296#ifdef AF_APPLETALK
Fred Drake4baedc12002-04-01 14:53:37 +00004297 /* Appletalk DDP */
4298 PyModule_AddIntConstant(m, "AF_APPLETALK", AF_APPLETALK);
Guido van Rossum09be4091999-08-09 14:40:40 +00004299#endif
4300#ifdef AF_NETROM
Fred Drake4baedc12002-04-01 14:53:37 +00004301 /* Amateur radio NetROM */
4302 PyModule_AddIntConstant(m, "AF_NETROM", AF_NETROM);
Guido van Rossum09be4091999-08-09 14:40:40 +00004303#endif
4304#ifdef AF_BRIDGE
Fred Drake4baedc12002-04-01 14:53:37 +00004305 /* Multiprotocol bridge */
4306 PyModule_AddIntConstant(m, "AF_BRIDGE", AF_BRIDGE);
Guido van Rossum09be4091999-08-09 14:40:40 +00004307#endif
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004308#ifdef AF_ATMPVC
4309 /* ATM PVCs */
4310 PyModule_AddIntConstant(m, "AF_ATMPVC", AF_ATMPVC);
4311#endif
Guido van Rossum09be4091999-08-09 14:40:40 +00004312#ifdef AF_AAL5
Fred Drake4baedc12002-04-01 14:53:37 +00004313 /* Reserved for Werner's ATM */
4314 PyModule_AddIntConstant(m, "AF_AAL5", AF_AAL5);
Guido van Rossum09be4091999-08-09 14:40:40 +00004315#endif
4316#ifdef AF_X25
Fred Drake4baedc12002-04-01 14:53:37 +00004317 /* Reserved for X.25 project */
4318 PyModule_AddIntConstant(m, "AF_X25", AF_X25);
Guido van Rossum09be4091999-08-09 14:40:40 +00004319#endif
4320#ifdef AF_INET6
Fred Drake4baedc12002-04-01 14:53:37 +00004321 PyModule_AddIntConstant(m, "AF_INET6", AF_INET6); /* IP version 6 */
Guido van Rossum09be4091999-08-09 14:40:40 +00004322#endif
4323#ifdef AF_ROSE
Fred Drake4baedc12002-04-01 14:53:37 +00004324 /* Amateur Radio X.25 PLP */
4325 PyModule_AddIntConstant(m, "AF_ROSE", AF_ROSE);
Guido van Rossum09be4091999-08-09 14:40:40 +00004326#endif
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004327#ifdef AF_DECnet
4328 /* Reserved for DECnet project */
4329 PyModule_AddIntConstant(m, "AF_DECnet", AF_DECnet);
4330#endif
4331#ifdef AF_NETBEUI
4332 /* Reserved for 802.2LLC project */
4333 PyModule_AddIntConstant(m, "AF_NETBEUI", AF_NETBEUI);
4334#endif
4335#ifdef AF_SECURITY
4336 /* Security callback pseudo AF */
4337 PyModule_AddIntConstant(m, "AF_SECURITY", AF_SECURITY);
4338#endif
4339#ifdef AF_KEY
4340 /* PF_KEY key management API */
4341 PyModule_AddIntConstant(m, "AF_KEY", AF_KEY);
4342#endif
4343#ifdef AF_NETLINK
4344 /* */
4345 PyModule_AddIntConstant(m, "AF_NETLINK", AF_NETLINK);
Martin v. Löwis11017b12006-01-14 18:12:57 +00004346 PyModule_AddIntConstant(m, "NETLINK_ROUTE", NETLINK_ROUTE);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004347#ifdef NETLINK_SKIP
Martin v. Löwis11017b12006-01-14 18:12:57 +00004348 PyModule_AddIntConstant(m, "NETLINK_SKIP", NETLINK_SKIP);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004349#endif
4350#ifdef NETLINK_W1
4351 PyModule_AddIntConstant(m, "NETLINK_W1", NETLINK_W1);
4352#endif
Martin v. Löwis11017b12006-01-14 18:12:57 +00004353 PyModule_AddIntConstant(m, "NETLINK_USERSOCK", NETLINK_USERSOCK);
4354 PyModule_AddIntConstant(m, "NETLINK_FIREWALL", NETLINK_FIREWALL);
Guido van Rossum668a94a2006-02-21 01:07:27 +00004355#ifdef NETLINK_TCPDIAG
Martin v. Löwis11017b12006-01-14 18:12:57 +00004356 PyModule_AddIntConstant(m, "NETLINK_TCPDIAG", NETLINK_TCPDIAG);
Guido van Rossum668a94a2006-02-21 01:07:27 +00004357#endif
4358#ifdef NETLINK_NFLOG
Martin v. Löwis11017b12006-01-14 18:12:57 +00004359 PyModule_AddIntConstant(m, "NETLINK_NFLOG", NETLINK_NFLOG);
Guido van Rossum668a94a2006-02-21 01:07:27 +00004360#endif
Neal Norwitz65851662006-01-16 04:31:40 +00004361#ifdef NETLINK_XFRM
Martin v. Löwis11017b12006-01-14 18:12:57 +00004362 PyModule_AddIntConstant(m, "NETLINK_XFRM", NETLINK_XFRM);
Neal Norwitz65851662006-01-16 04:31:40 +00004363#endif
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004364#ifdef NETLINK_ARPD
Martin v. Löwis11017b12006-01-14 18:12:57 +00004365 PyModule_AddIntConstant(m, "NETLINK_ARPD", NETLINK_ARPD);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004366#endif
4367#ifdef NETLINK_ROUTE6
Martin v. Löwis11017b12006-01-14 18:12:57 +00004368 PyModule_AddIntConstant(m, "NETLINK_ROUTE6", NETLINK_ROUTE6);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004369#endif
Martin v. Löwis11017b12006-01-14 18:12:57 +00004370 PyModule_AddIntConstant(m, "NETLINK_IP6_FW", NETLINK_IP6_FW);
Thomas Wouterscf297e42007-02-23 15:07:44 +00004371#ifdef NETLINK_DNRTMSG
Martin v. Löwis11017b12006-01-14 18:12:57 +00004372 PyModule_AddIntConstant(m, "NETLINK_DNRTMSG", NETLINK_DNRTMSG);
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00004373#endif
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004374#ifdef NETLINK_TAPBASE
Martin v. Löwis11017b12006-01-14 18:12:57 +00004375 PyModule_AddIntConstant(m, "NETLINK_TAPBASE", NETLINK_TAPBASE);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004376#endif
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004377#endif /* AF_NETLINK */
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004378#ifdef AF_ROUTE
4379 /* Alias to emulate 4.4BSD */
4380 PyModule_AddIntConstant(m, "AF_ROUTE", AF_ROUTE);
4381#endif
4382#ifdef AF_ASH
4383 /* Ash */
4384 PyModule_AddIntConstant(m, "AF_ASH", AF_ASH);
4385#endif
4386#ifdef AF_ECONET
4387 /* Acorn Econet */
4388 PyModule_AddIntConstant(m, "AF_ECONET", AF_ECONET);
4389#endif
4390#ifdef AF_ATMSVC
4391 /* ATM SVCs */
4392 PyModule_AddIntConstant(m, "AF_ATMSVC", AF_ATMSVC);
4393#endif
4394#ifdef AF_SNA
4395 /* Linux SNA Project (nutters!) */
4396 PyModule_AddIntConstant(m, "AF_SNA", AF_SNA);
4397#endif
4398#ifdef AF_IRDA
4399 /* IRDA sockets */
4400 PyModule_AddIntConstant(m, "AF_IRDA", AF_IRDA);
4401#endif
4402#ifdef AF_PPPOX
4403 /* PPPoX sockets */
4404 PyModule_AddIntConstant(m, "AF_PPPOX", AF_PPPOX);
4405#endif
4406#ifdef AF_WANPIPE
4407 /* Wanpipe API Sockets */
4408 PyModule_AddIntConstant(m, "AF_WANPIPE", AF_WANPIPE);
4409#endif
4410#ifdef AF_LLC
4411 /* Linux LLC */
4412 PyModule_AddIntConstant(m, "AF_LLC", AF_LLC);
4413#endif
Martin v. Löwis12af0482004-01-31 12:34:17 +00004414
Hye-Shik Chang81268602004-02-02 06:05:24 +00004415#ifdef USE_BLUETOOTH
Martin v. Löwis12af0482004-01-31 12:34:17 +00004416 PyModule_AddIntConstant(m, "AF_BLUETOOTH", AF_BLUETOOTH);
4417 PyModule_AddIntConstant(m, "BTPROTO_L2CAP", BTPROTO_L2CAP);
Thomas Wouterscf297e42007-02-23 15:07:44 +00004418 PyModule_AddIntConstant(m, "BTPROTO_HCI", BTPROTO_HCI);
4419 PyModule_AddIntConstant(m, "SOL_HCI", SOL_HCI);
Thomas Wouterscf297e42007-02-23 15:07:44 +00004420 PyModule_AddIntConstant(m, "HCI_FILTER", HCI_FILTER);
Hye-Shik Chang81268602004-02-02 06:05:24 +00004421#if !defined(__FreeBSD__)
Guido van Rossumcd16bf62007-06-13 18:07:49 +00004422 PyModule_AddIntConstant(m, "HCI_TIME_STAMP", HCI_TIME_STAMP);
4423 PyModule_AddIntConstant(m, "HCI_DATA_DIR", HCI_DATA_DIR);
Martin v. Löwis12af0482004-01-31 12:34:17 +00004424 PyModule_AddIntConstant(m, "BTPROTO_SCO", BTPROTO_SCO);
Hye-Shik Chang81268602004-02-02 06:05:24 +00004425#endif
Martin v. Löwis12af0482004-01-31 12:34:17 +00004426 PyModule_AddIntConstant(m, "BTPROTO_RFCOMM", BTPROTO_RFCOMM);
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00004427 PyModule_AddStringConstant(m, "BDADDR_ANY", "00:00:00:00:00:00");
4428 PyModule_AddStringConstant(m, "BDADDR_LOCAL", "00:00:00:FF:FF:FF");
Martin v. Löwis12af0482004-01-31 12:34:17 +00004429#endif
4430
Martin v. Löwis1ba3fd52001-08-10 20:29:40 +00004431#ifdef HAVE_NETPACKET_PACKET_H
Fred Drake4baedc12002-04-01 14:53:37 +00004432 PyModule_AddIntConstant(m, "AF_PACKET", AF_PACKET);
4433 PyModule_AddIntConstant(m, "PF_PACKET", PF_PACKET);
4434 PyModule_AddIntConstant(m, "PACKET_HOST", PACKET_HOST);
4435 PyModule_AddIntConstant(m, "PACKET_BROADCAST", PACKET_BROADCAST);
4436 PyModule_AddIntConstant(m, "PACKET_MULTICAST", PACKET_MULTICAST);
4437 PyModule_AddIntConstant(m, "PACKET_OTHERHOST", PACKET_OTHERHOST);
4438 PyModule_AddIntConstant(m, "PACKET_OUTGOING", PACKET_OUTGOING);
4439 PyModule_AddIntConstant(m, "PACKET_LOOPBACK", PACKET_LOOPBACK);
4440 PyModule_AddIntConstant(m, "PACKET_FASTROUTE", PACKET_FASTROUTE);
Guido van Rossum48a680c2001-03-02 06:34:14 +00004441#endif
Guido van Rossum09be4091999-08-09 14:40:40 +00004442
Christian Heimes043d6f62008-01-07 17:19:16 +00004443#ifdef HAVE_LINUX_TIPC_H
4444 PyModule_AddIntConstant(m, "AF_TIPC", AF_TIPC);
4445
4446 /* for addresses */
4447 PyModule_AddIntConstant(m, "TIPC_ADDR_NAMESEQ", TIPC_ADDR_NAMESEQ);
4448 PyModule_AddIntConstant(m, "TIPC_ADDR_NAME", TIPC_ADDR_NAME);
4449 PyModule_AddIntConstant(m, "TIPC_ADDR_ID", TIPC_ADDR_ID);
4450
4451 PyModule_AddIntConstant(m, "TIPC_ZONE_SCOPE", TIPC_ZONE_SCOPE);
4452 PyModule_AddIntConstant(m, "TIPC_CLUSTER_SCOPE", TIPC_CLUSTER_SCOPE);
4453 PyModule_AddIntConstant(m, "TIPC_NODE_SCOPE", TIPC_NODE_SCOPE);
4454
4455 /* for setsockopt() */
4456 PyModule_AddIntConstant(m, "SOL_TIPC", SOL_TIPC);
4457 PyModule_AddIntConstant(m, "TIPC_IMPORTANCE", TIPC_IMPORTANCE);
4458 PyModule_AddIntConstant(m, "TIPC_SRC_DROPPABLE", TIPC_SRC_DROPPABLE);
4459 PyModule_AddIntConstant(m, "TIPC_DEST_DROPPABLE",
4460 TIPC_DEST_DROPPABLE);
4461 PyModule_AddIntConstant(m, "TIPC_CONN_TIMEOUT", TIPC_CONN_TIMEOUT);
4462
4463 PyModule_AddIntConstant(m, "TIPC_LOW_IMPORTANCE",
4464 TIPC_LOW_IMPORTANCE);
4465 PyModule_AddIntConstant(m, "TIPC_MEDIUM_IMPORTANCE",
4466 TIPC_MEDIUM_IMPORTANCE);
4467 PyModule_AddIntConstant(m, "TIPC_HIGH_IMPORTANCE",
4468 TIPC_HIGH_IMPORTANCE);
4469 PyModule_AddIntConstant(m, "TIPC_CRITICAL_IMPORTANCE",
4470 TIPC_CRITICAL_IMPORTANCE);
4471
4472 /* for subscriptions */
4473 PyModule_AddIntConstant(m, "TIPC_SUB_PORTS", TIPC_SUB_PORTS);
4474 PyModule_AddIntConstant(m, "TIPC_SUB_SERVICE", TIPC_SUB_SERVICE);
Christian Heimes25bb7832008-01-11 16:17:00 +00004475#ifdef TIPC_SUB_CANCEL
4476 /* doesn't seem to be available everywhere */
Christian Heimes043d6f62008-01-07 17:19:16 +00004477 PyModule_AddIntConstant(m, "TIPC_SUB_CANCEL", TIPC_SUB_CANCEL);
Christian Heimes25bb7832008-01-11 16:17:00 +00004478#endif
Christian Heimes043d6f62008-01-07 17:19:16 +00004479 PyModule_AddIntConstant(m, "TIPC_WAIT_FOREVER", TIPC_WAIT_FOREVER);
4480 PyModule_AddIntConstant(m, "TIPC_PUBLISHED", TIPC_PUBLISHED);
4481 PyModule_AddIntConstant(m, "TIPC_WITHDRAWN", TIPC_WITHDRAWN);
4482 PyModule_AddIntConstant(m, "TIPC_SUBSCR_TIMEOUT", TIPC_SUBSCR_TIMEOUT);
4483 PyModule_AddIntConstant(m, "TIPC_CFG_SRV", TIPC_CFG_SRV);
4484 PyModule_AddIntConstant(m, "TIPC_TOP_SRV", TIPC_TOP_SRV);
4485#endif
4486
Guido van Rossum09be4091999-08-09 14:40:40 +00004487 /* Socket types */
Fred Drake4baedc12002-04-01 14:53:37 +00004488 PyModule_AddIntConstant(m, "SOCK_STREAM", SOCK_STREAM);
4489 PyModule_AddIntConstant(m, "SOCK_DGRAM", SOCK_DGRAM);
Guido van Rossumbcc20741998-08-04 22:53:56 +00004490/* We have incomplete socket support. */
Fred Drake4baedc12002-04-01 14:53:37 +00004491 PyModule_AddIntConstant(m, "SOCK_RAW", SOCK_RAW);
4492 PyModule_AddIntConstant(m, "SOCK_SEQPACKET", SOCK_SEQPACKET);
Martin v. Löwiscf8f47e2002-12-11 13:10:57 +00004493#if defined(SOCK_RDM)
Fred Drake4baedc12002-04-01 14:53:37 +00004494 PyModule_AddIntConstant(m, "SOCK_RDM", SOCK_RDM);
Guido van Rossumbcc20741998-08-04 22:53:56 +00004495#endif
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004496
4497#ifdef SO_DEBUG
Fred Drake4baedc12002-04-01 14:53:37 +00004498 PyModule_AddIntConstant(m, "SO_DEBUG", SO_DEBUG);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004499#endif
4500#ifdef SO_ACCEPTCONN
Fred Drake4baedc12002-04-01 14:53:37 +00004501 PyModule_AddIntConstant(m, "SO_ACCEPTCONN", SO_ACCEPTCONN);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004502#endif
4503#ifdef SO_REUSEADDR
Fred Drake4baedc12002-04-01 14:53:37 +00004504 PyModule_AddIntConstant(m, "SO_REUSEADDR", SO_REUSEADDR);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004505#endif
Andrew M. Kuchling42851ab2004-07-10 14:19:21 +00004506#ifdef SO_EXCLUSIVEADDRUSE
4507 PyModule_AddIntConstant(m, "SO_EXCLUSIVEADDRUSE", SO_EXCLUSIVEADDRUSE);
4508#endif
4509
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004510#ifdef SO_KEEPALIVE
Fred Drake4baedc12002-04-01 14:53:37 +00004511 PyModule_AddIntConstant(m, "SO_KEEPALIVE", SO_KEEPALIVE);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004512#endif
4513#ifdef SO_DONTROUTE
Fred Drake4baedc12002-04-01 14:53:37 +00004514 PyModule_AddIntConstant(m, "SO_DONTROUTE", SO_DONTROUTE);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004515#endif
4516#ifdef SO_BROADCAST
Fred Drake4baedc12002-04-01 14:53:37 +00004517 PyModule_AddIntConstant(m, "SO_BROADCAST", SO_BROADCAST);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004518#endif
4519#ifdef SO_USELOOPBACK
Fred Drake4baedc12002-04-01 14:53:37 +00004520 PyModule_AddIntConstant(m, "SO_USELOOPBACK", SO_USELOOPBACK);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004521#endif
4522#ifdef SO_LINGER
Fred Drake4baedc12002-04-01 14:53:37 +00004523 PyModule_AddIntConstant(m, "SO_LINGER", SO_LINGER);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004524#endif
4525#ifdef SO_OOBINLINE
Fred Drake4baedc12002-04-01 14:53:37 +00004526 PyModule_AddIntConstant(m, "SO_OOBINLINE", SO_OOBINLINE);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004527#endif
4528#ifdef SO_REUSEPORT
Fred Drake4baedc12002-04-01 14:53:37 +00004529 PyModule_AddIntConstant(m, "SO_REUSEPORT", SO_REUSEPORT);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004530#endif
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004531#ifdef SO_SNDBUF
Fred Drake4baedc12002-04-01 14:53:37 +00004532 PyModule_AddIntConstant(m, "SO_SNDBUF", SO_SNDBUF);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004533#endif
4534#ifdef SO_RCVBUF
Fred Drake4baedc12002-04-01 14:53:37 +00004535 PyModule_AddIntConstant(m, "SO_RCVBUF", SO_RCVBUF);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004536#endif
4537#ifdef SO_SNDLOWAT
Fred Drake4baedc12002-04-01 14:53:37 +00004538 PyModule_AddIntConstant(m, "SO_SNDLOWAT", SO_SNDLOWAT);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004539#endif
4540#ifdef SO_RCVLOWAT
Fred Drake4baedc12002-04-01 14:53:37 +00004541 PyModule_AddIntConstant(m, "SO_RCVLOWAT", SO_RCVLOWAT);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004542#endif
4543#ifdef SO_SNDTIMEO
Fred Drake4baedc12002-04-01 14:53:37 +00004544 PyModule_AddIntConstant(m, "SO_SNDTIMEO", SO_SNDTIMEO);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004545#endif
4546#ifdef SO_RCVTIMEO
Fred Drake4baedc12002-04-01 14:53:37 +00004547 PyModule_AddIntConstant(m, "SO_RCVTIMEO", SO_RCVTIMEO);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004548#endif
4549#ifdef SO_ERROR
Fred Drake4baedc12002-04-01 14:53:37 +00004550 PyModule_AddIntConstant(m, "SO_ERROR", SO_ERROR);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004551#endif
4552#ifdef SO_TYPE
Fred Drake4baedc12002-04-01 14:53:37 +00004553 PyModule_AddIntConstant(m, "SO_TYPE", SO_TYPE);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004554#endif
Larry Hastingsf0f37952010-04-02 11:47:10 +00004555#ifdef SO_SETFIB
4556 PyModule_AddIntConstant(m, "SO_SETFIB", SO_SETFIB);
4557#endif
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004558
4559 /* Maximum number of connections for "listen" */
4560#ifdef SOMAXCONN
Fred Drake4baedc12002-04-01 14:53:37 +00004561 PyModule_AddIntConstant(m, "SOMAXCONN", SOMAXCONN);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004562#else
Guido van Rossum3eede5a2002-06-07 02:08:35 +00004563 PyModule_AddIntConstant(m, "SOMAXCONN", 5); /* Common value */
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004564#endif
4565
4566 /* Flags for send, recv */
4567#ifdef MSG_OOB
Fred Drake4baedc12002-04-01 14:53:37 +00004568 PyModule_AddIntConstant(m, "MSG_OOB", MSG_OOB);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004569#endif
4570#ifdef MSG_PEEK
Fred Drake4baedc12002-04-01 14:53:37 +00004571 PyModule_AddIntConstant(m, "MSG_PEEK", MSG_PEEK);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004572#endif
4573#ifdef MSG_DONTROUTE
Fred Drake4baedc12002-04-01 14:53:37 +00004574 PyModule_AddIntConstant(m, "MSG_DONTROUTE", MSG_DONTROUTE);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004575#endif
Guido van Rossum2c8bcb82000-04-25 21:34:53 +00004576#ifdef MSG_DONTWAIT
Fred Drake4baedc12002-04-01 14:53:37 +00004577 PyModule_AddIntConstant(m, "MSG_DONTWAIT", MSG_DONTWAIT);
Guido van Rossum2c8bcb82000-04-25 21:34:53 +00004578#endif
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004579#ifdef MSG_EOR
Fred Drake4baedc12002-04-01 14:53:37 +00004580 PyModule_AddIntConstant(m, "MSG_EOR", MSG_EOR);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004581#endif
4582#ifdef MSG_TRUNC
Fred Drake4baedc12002-04-01 14:53:37 +00004583 PyModule_AddIntConstant(m, "MSG_TRUNC", MSG_TRUNC);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004584#endif
4585#ifdef MSG_CTRUNC
Fred Drake4baedc12002-04-01 14:53:37 +00004586 PyModule_AddIntConstant(m, "MSG_CTRUNC", MSG_CTRUNC);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004587#endif
4588#ifdef MSG_WAITALL
Fred Drake4baedc12002-04-01 14:53:37 +00004589 PyModule_AddIntConstant(m, "MSG_WAITALL", MSG_WAITALL);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004590#endif
4591#ifdef MSG_BTAG
Fred Drake4baedc12002-04-01 14:53:37 +00004592 PyModule_AddIntConstant(m, "MSG_BTAG", MSG_BTAG);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004593#endif
4594#ifdef MSG_ETAG
Fred Drake4baedc12002-04-01 14:53:37 +00004595 PyModule_AddIntConstant(m, "MSG_ETAG", MSG_ETAG);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004596#endif
4597
4598 /* Protocol level and numbers, usable for [gs]etsockopt */
4599#ifdef SOL_SOCKET
Fred Drake4baedc12002-04-01 14:53:37 +00004600 PyModule_AddIntConstant(m, "SOL_SOCKET", SOL_SOCKET);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004601#endif
Guido van Rossum09be4091999-08-09 14:40:40 +00004602#ifdef SOL_IP
Fred Drake4baedc12002-04-01 14:53:37 +00004603 PyModule_AddIntConstant(m, "SOL_IP", SOL_IP);
Guido van Rossum09be4091999-08-09 14:40:40 +00004604#else
Fred Drake4baedc12002-04-01 14:53:37 +00004605 PyModule_AddIntConstant(m, "SOL_IP", 0);
Guido van Rossum09be4091999-08-09 14:40:40 +00004606#endif
4607#ifdef SOL_IPX
Fred Drake4baedc12002-04-01 14:53:37 +00004608 PyModule_AddIntConstant(m, "SOL_IPX", SOL_IPX);
Guido van Rossum09be4091999-08-09 14:40:40 +00004609#endif
4610#ifdef SOL_AX25
Fred Drake4baedc12002-04-01 14:53:37 +00004611 PyModule_AddIntConstant(m, "SOL_AX25", SOL_AX25);
Guido van Rossum09be4091999-08-09 14:40:40 +00004612#endif
4613#ifdef SOL_ATALK
Fred Drake4baedc12002-04-01 14:53:37 +00004614 PyModule_AddIntConstant(m, "SOL_ATALK", SOL_ATALK);
Guido van Rossum09be4091999-08-09 14:40:40 +00004615#endif
4616#ifdef SOL_NETROM
Fred Drake4baedc12002-04-01 14:53:37 +00004617 PyModule_AddIntConstant(m, "SOL_NETROM", SOL_NETROM);
Guido van Rossum09be4091999-08-09 14:40:40 +00004618#endif
4619#ifdef SOL_ROSE
Fred Drake4baedc12002-04-01 14:53:37 +00004620 PyModule_AddIntConstant(m, "SOL_ROSE", SOL_ROSE);
Guido van Rossum09be4091999-08-09 14:40:40 +00004621#endif
4622#ifdef SOL_TCP
Fred Drake4baedc12002-04-01 14:53:37 +00004623 PyModule_AddIntConstant(m, "SOL_TCP", SOL_TCP);
Guido van Rossum09be4091999-08-09 14:40:40 +00004624#else
Fred Drake4baedc12002-04-01 14:53:37 +00004625 PyModule_AddIntConstant(m, "SOL_TCP", 6);
Guido van Rossum09be4091999-08-09 14:40:40 +00004626#endif
4627#ifdef SOL_UDP
Fred Drake4baedc12002-04-01 14:53:37 +00004628 PyModule_AddIntConstant(m, "SOL_UDP", SOL_UDP);
Guido van Rossum09be4091999-08-09 14:40:40 +00004629#else
Fred Drake4baedc12002-04-01 14:53:37 +00004630 PyModule_AddIntConstant(m, "SOL_UDP", 17);
Guido van Rossum09be4091999-08-09 14:40:40 +00004631#endif
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004632#ifdef IPPROTO_IP
Fred Drake4baedc12002-04-01 14:53:37 +00004633 PyModule_AddIntConstant(m, "IPPROTO_IP", IPPROTO_IP);
Guido van Rossum578de301998-05-28 20:18:18 +00004634#else
Fred Drake4baedc12002-04-01 14:53:37 +00004635 PyModule_AddIntConstant(m, "IPPROTO_IP", 0);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004636#endif
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004637#ifdef IPPROTO_HOPOPTS
Fred Drake4baedc12002-04-01 14:53:37 +00004638 PyModule_AddIntConstant(m, "IPPROTO_HOPOPTS", IPPROTO_HOPOPTS);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004639#endif
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004640#ifdef IPPROTO_ICMP
Fred Drake4baedc12002-04-01 14:53:37 +00004641 PyModule_AddIntConstant(m, "IPPROTO_ICMP", IPPROTO_ICMP);
Guido van Rossum578de301998-05-28 20:18:18 +00004642#else
Fred Drake4baedc12002-04-01 14:53:37 +00004643 PyModule_AddIntConstant(m, "IPPROTO_ICMP", 1);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004644#endif
4645#ifdef IPPROTO_IGMP
Fred Drake4baedc12002-04-01 14:53:37 +00004646 PyModule_AddIntConstant(m, "IPPROTO_IGMP", IPPROTO_IGMP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004647#endif
4648#ifdef IPPROTO_GGP
Fred Drake4baedc12002-04-01 14:53:37 +00004649 PyModule_AddIntConstant(m, "IPPROTO_GGP", IPPROTO_GGP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004650#endif
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004651#ifdef IPPROTO_IPV4
Fred Drake4baedc12002-04-01 14:53:37 +00004652 PyModule_AddIntConstant(m, "IPPROTO_IPV4", IPPROTO_IPV4);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004653#endif
Martin v. Löwisa0f17342003-10-03 13:56:20 +00004654#ifdef IPPROTO_IPV6
4655 PyModule_AddIntConstant(m, "IPPROTO_IPV6", IPPROTO_IPV6);
4656#endif
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004657#ifdef IPPROTO_IPIP
Fred Drake4baedc12002-04-01 14:53:37 +00004658 PyModule_AddIntConstant(m, "IPPROTO_IPIP", IPPROTO_IPIP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004659#endif
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004660#ifdef IPPROTO_TCP
Fred Drake4baedc12002-04-01 14:53:37 +00004661 PyModule_AddIntConstant(m, "IPPROTO_TCP", IPPROTO_TCP);
Guido van Rossum578de301998-05-28 20:18:18 +00004662#else
Fred Drake4baedc12002-04-01 14:53:37 +00004663 PyModule_AddIntConstant(m, "IPPROTO_TCP", 6);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004664#endif
4665#ifdef IPPROTO_EGP
Fred Drake4baedc12002-04-01 14:53:37 +00004666 PyModule_AddIntConstant(m, "IPPROTO_EGP", IPPROTO_EGP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004667#endif
4668#ifdef IPPROTO_PUP
Fred Drake4baedc12002-04-01 14:53:37 +00004669 PyModule_AddIntConstant(m, "IPPROTO_PUP", IPPROTO_PUP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004670#endif
4671#ifdef IPPROTO_UDP
Fred Drake4baedc12002-04-01 14:53:37 +00004672 PyModule_AddIntConstant(m, "IPPROTO_UDP", IPPROTO_UDP);
Guido van Rossum578de301998-05-28 20:18:18 +00004673#else
Fred Drake4baedc12002-04-01 14:53:37 +00004674 PyModule_AddIntConstant(m, "IPPROTO_UDP", 17);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004675#endif
4676#ifdef IPPROTO_IDP
Fred Drake4baedc12002-04-01 14:53:37 +00004677 PyModule_AddIntConstant(m, "IPPROTO_IDP", IPPROTO_IDP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004678#endif
4679#ifdef IPPROTO_HELLO
Fred Drake4baedc12002-04-01 14:53:37 +00004680 PyModule_AddIntConstant(m, "IPPROTO_HELLO", IPPROTO_HELLO);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004681#endif
4682#ifdef IPPROTO_ND
Fred Drake4baedc12002-04-01 14:53:37 +00004683 PyModule_AddIntConstant(m, "IPPROTO_ND", IPPROTO_ND);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004684#endif
4685#ifdef IPPROTO_TP
Fred Drake4baedc12002-04-01 14:53:37 +00004686 PyModule_AddIntConstant(m, "IPPROTO_TP", IPPROTO_TP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004687#endif
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004688#ifdef IPPROTO_IPV6
Fred Drake4baedc12002-04-01 14:53:37 +00004689 PyModule_AddIntConstant(m, "IPPROTO_IPV6", IPPROTO_IPV6);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004690#endif
4691#ifdef IPPROTO_ROUTING
Fred Drake4baedc12002-04-01 14:53:37 +00004692 PyModule_AddIntConstant(m, "IPPROTO_ROUTING", IPPROTO_ROUTING);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004693#endif
4694#ifdef IPPROTO_FRAGMENT
Fred Drake4baedc12002-04-01 14:53:37 +00004695 PyModule_AddIntConstant(m, "IPPROTO_FRAGMENT", IPPROTO_FRAGMENT);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004696#endif
4697#ifdef IPPROTO_RSVP
Fred Drake4baedc12002-04-01 14:53:37 +00004698 PyModule_AddIntConstant(m, "IPPROTO_RSVP", IPPROTO_RSVP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004699#endif
4700#ifdef IPPROTO_GRE
Fred Drake4baedc12002-04-01 14:53:37 +00004701 PyModule_AddIntConstant(m, "IPPROTO_GRE", IPPROTO_GRE);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004702#endif
4703#ifdef IPPROTO_ESP
Fred Drake4baedc12002-04-01 14:53:37 +00004704 PyModule_AddIntConstant(m, "IPPROTO_ESP", IPPROTO_ESP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004705#endif
4706#ifdef IPPROTO_AH
Fred Drake4baedc12002-04-01 14:53:37 +00004707 PyModule_AddIntConstant(m, "IPPROTO_AH", IPPROTO_AH);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004708#endif
4709#ifdef IPPROTO_MOBILE
Fred Drake4baedc12002-04-01 14:53:37 +00004710 PyModule_AddIntConstant(m, "IPPROTO_MOBILE", IPPROTO_MOBILE);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004711#endif
4712#ifdef IPPROTO_ICMPV6
Fred Drake4baedc12002-04-01 14:53:37 +00004713 PyModule_AddIntConstant(m, "IPPROTO_ICMPV6", IPPROTO_ICMPV6);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004714#endif
4715#ifdef IPPROTO_NONE
Fred Drake4baedc12002-04-01 14:53:37 +00004716 PyModule_AddIntConstant(m, "IPPROTO_NONE", IPPROTO_NONE);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004717#endif
4718#ifdef IPPROTO_DSTOPTS
Fred Drake4baedc12002-04-01 14:53:37 +00004719 PyModule_AddIntConstant(m, "IPPROTO_DSTOPTS", IPPROTO_DSTOPTS);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004720#endif
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004721#ifdef IPPROTO_XTP
Fred Drake4baedc12002-04-01 14:53:37 +00004722 PyModule_AddIntConstant(m, "IPPROTO_XTP", IPPROTO_XTP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004723#endif
4724#ifdef IPPROTO_EON
Fred Drake4baedc12002-04-01 14:53:37 +00004725 PyModule_AddIntConstant(m, "IPPROTO_EON", IPPROTO_EON);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004726#endif
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004727#ifdef IPPROTO_PIM
Fred Drake4baedc12002-04-01 14:53:37 +00004728 PyModule_AddIntConstant(m, "IPPROTO_PIM", IPPROTO_PIM);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004729#endif
4730#ifdef IPPROTO_IPCOMP
Fred Drake4baedc12002-04-01 14:53:37 +00004731 PyModule_AddIntConstant(m, "IPPROTO_IPCOMP", IPPROTO_IPCOMP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004732#endif
4733#ifdef IPPROTO_VRRP
Fred Drake4baedc12002-04-01 14:53:37 +00004734 PyModule_AddIntConstant(m, "IPPROTO_VRRP", IPPROTO_VRRP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004735#endif
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004736#ifdef IPPROTO_BIP
Fred Drake4baedc12002-04-01 14:53:37 +00004737 PyModule_AddIntConstant(m, "IPPROTO_BIP", IPPROTO_BIP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004738#endif
4739/**/
4740#ifdef IPPROTO_RAW
Fred Drake4baedc12002-04-01 14:53:37 +00004741 PyModule_AddIntConstant(m, "IPPROTO_RAW", IPPROTO_RAW);
Guido van Rossum578de301998-05-28 20:18:18 +00004742#else
Fred Drake4baedc12002-04-01 14:53:37 +00004743 PyModule_AddIntConstant(m, "IPPROTO_RAW", 255);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004744#endif
4745#ifdef IPPROTO_MAX
Fred Drake4baedc12002-04-01 14:53:37 +00004746 PyModule_AddIntConstant(m, "IPPROTO_MAX", IPPROTO_MAX);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004747#endif
4748
4749 /* Some port configuration */
4750#ifdef IPPORT_RESERVED
Fred Drake4baedc12002-04-01 14:53:37 +00004751 PyModule_AddIntConstant(m, "IPPORT_RESERVED", IPPORT_RESERVED);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004752#else
Fred Drake4baedc12002-04-01 14:53:37 +00004753 PyModule_AddIntConstant(m, "IPPORT_RESERVED", 1024);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004754#endif
4755#ifdef IPPORT_USERRESERVED
Fred Drake4baedc12002-04-01 14:53:37 +00004756 PyModule_AddIntConstant(m, "IPPORT_USERRESERVED", IPPORT_USERRESERVED);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004757#else
Fred Drake4baedc12002-04-01 14:53:37 +00004758 PyModule_AddIntConstant(m, "IPPORT_USERRESERVED", 5000);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004759#endif
4760
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004761 /* Some reserved IP v.4 addresses */
4762#ifdef INADDR_ANY
Fred Drake4baedc12002-04-01 14:53:37 +00004763 PyModule_AddIntConstant(m, "INADDR_ANY", INADDR_ANY);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004764#else
Fred Drake4baedc12002-04-01 14:53:37 +00004765 PyModule_AddIntConstant(m, "INADDR_ANY", 0x00000000);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004766#endif
4767#ifdef INADDR_BROADCAST
Fred Drake4baedc12002-04-01 14:53:37 +00004768 PyModule_AddIntConstant(m, "INADDR_BROADCAST", INADDR_BROADCAST);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004769#else
Fred Drake4baedc12002-04-01 14:53:37 +00004770 PyModule_AddIntConstant(m, "INADDR_BROADCAST", 0xffffffff);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004771#endif
4772#ifdef INADDR_LOOPBACK
Fred Drake4baedc12002-04-01 14:53:37 +00004773 PyModule_AddIntConstant(m, "INADDR_LOOPBACK", INADDR_LOOPBACK);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004774#else
Fred Drake4baedc12002-04-01 14:53:37 +00004775 PyModule_AddIntConstant(m, "INADDR_LOOPBACK", 0x7F000001);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004776#endif
4777#ifdef INADDR_UNSPEC_GROUP
Fred Drake4baedc12002-04-01 14:53:37 +00004778 PyModule_AddIntConstant(m, "INADDR_UNSPEC_GROUP", INADDR_UNSPEC_GROUP);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004779#else
Fred Drake4baedc12002-04-01 14:53:37 +00004780 PyModule_AddIntConstant(m, "INADDR_UNSPEC_GROUP", 0xe0000000);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004781#endif
4782#ifdef INADDR_ALLHOSTS_GROUP
Fred Drake4baedc12002-04-01 14:53:37 +00004783 PyModule_AddIntConstant(m, "INADDR_ALLHOSTS_GROUP",
4784 INADDR_ALLHOSTS_GROUP);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004785#else
Fred Drake4baedc12002-04-01 14:53:37 +00004786 PyModule_AddIntConstant(m, "INADDR_ALLHOSTS_GROUP", 0xe0000001);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004787#endif
4788#ifdef INADDR_MAX_LOCAL_GROUP
Fred Drake4baedc12002-04-01 14:53:37 +00004789 PyModule_AddIntConstant(m, "INADDR_MAX_LOCAL_GROUP",
4790 INADDR_MAX_LOCAL_GROUP);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004791#else
Fred Drake4baedc12002-04-01 14:53:37 +00004792 PyModule_AddIntConstant(m, "INADDR_MAX_LOCAL_GROUP", 0xe00000ff);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004793#endif
4794#ifdef INADDR_NONE
Fred Drake4baedc12002-04-01 14:53:37 +00004795 PyModule_AddIntConstant(m, "INADDR_NONE", INADDR_NONE);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004796#else
Fred Drake4baedc12002-04-01 14:53:37 +00004797 PyModule_AddIntConstant(m, "INADDR_NONE", 0xffffffff);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004798#endif
4799
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004800 /* IPv4 [gs]etsockopt options */
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004801#ifdef IP_OPTIONS
Fred Drake4baedc12002-04-01 14:53:37 +00004802 PyModule_AddIntConstant(m, "IP_OPTIONS", IP_OPTIONS);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004803#endif
4804#ifdef IP_HDRINCL
Fred Drake4baedc12002-04-01 14:53:37 +00004805 PyModule_AddIntConstant(m, "IP_HDRINCL", IP_HDRINCL);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004806#endif
4807#ifdef IP_TOS
Fred Drake4baedc12002-04-01 14:53:37 +00004808 PyModule_AddIntConstant(m, "IP_TOS", IP_TOS);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004809#endif
4810#ifdef IP_TTL
Fred Drake4baedc12002-04-01 14:53:37 +00004811 PyModule_AddIntConstant(m, "IP_TTL", IP_TTL);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004812#endif
4813#ifdef IP_RECVOPTS
Fred Drake4baedc12002-04-01 14:53:37 +00004814 PyModule_AddIntConstant(m, "IP_RECVOPTS", IP_RECVOPTS);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004815#endif
4816#ifdef IP_RECVRETOPTS
Fred Drake4baedc12002-04-01 14:53:37 +00004817 PyModule_AddIntConstant(m, "IP_RECVRETOPTS", IP_RECVRETOPTS);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004818#endif
4819#ifdef IP_RECVDSTADDR
Fred Drake4baedc12002-04-01 14:53:37 +00004820 PyModule_AddIntConstant(m, "IP_RECVDSTADDR", IP_RECVDSTADDR);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004821#endif
4822#ifdef IP_RETOPTS
Fred Drake4baedc12002-04-01 14:53:37 +00004823 PyModule_AddIntConstant(m, "IP_RETOPTS", IP_RETOPTS);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004824#endif
4825#ifdef IP_MULTICAST_IF
Fred Drake4baedc12002-04-01 14:53:37 +00004826 PyModule_AddIntConstant(m, "IP_MULTICAST_IF", IP_MULTICAST_IF);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004827#endif
4828#ifdef IP_MULTICAST_TTL
Fred Drake4baedc12002-04-01 14:53:37 +00004829 PyModule_AddIntConstant(m, "IP_MULTICAST_TTL", IP_MULTICAST_TTL);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004830#endif
4831#ifdef IP_MULTICAST_LOOP
Fred Drake4baedc12002-04-01 14:53:37 +00004832 PyModule_AddIntConstant(m, "IP_MULTICAST_LOOP", IP_MULTICAST_LOOP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004833#endif
4834#ifdef IP_ADD_MEMBERSHIP
Fred Drake4baedc12002-04-01 14:53:37 +00004835 PyModule_AddIntConstant(m, "IP_ADD_MEMBERSHIP", IP_ADD_MEMBERSHIP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004836#endif
4837#ifdef IP_DROP_MEMBERSHIP
Fred Drake4baedc12002-04-01 14:53:37 +00004838 PyModule_AddIntConstant(m, "IP_DROP_MEMBERSHIP", IP_DROP_MEMBERSHIP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004839#endif
Guido van Rossum09be4091999-08-09 14:40:40 +00004840#ifdef IP_DEFAULT_MULTICAST_TTL
Fred Drake4baedc12002-04-01 14:53:37 +00004841 PyModule_AddIntConstant(m, "IP_DEFAULT_MULTICAST_TTL",
4842 IP_DEFAULT_MULTICAST_TTL);
Guido van Rossum09be4091999-08-09 14:40:40 +00004843#endif
4844#ifdef IP_DEFAULT_MULTICAST_LOOP
Fred Drake4baedc12002-04-01 14:53:37 +00004845 PyModule_AddIntConstant(m, "IP_DEFAULT_MULTICAST_LOOP",
4846 IP_DEFAULT_MULTICAST_LOOP);
Guido van Rossum09be4091999-08-09 14:40:40 +00004847#endif
4848#ifdef IP_MAX_MEMBERSHIPS
Fred Drake4baedc12002-04-01 14:53:37 +00004849 PyModule_AddIntConstant(m, "IP_MAX_MEMBERSHIPS", IP_MAX_MEMBERSHIPS);
Guido van Rossum09be4091999-08-09 14:40:40 +00004850#endif
4851
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004852 /* IPv6 [gs]etsockopt options, defined in RFC2553 */
4853#ifdef IPV6_JOIN_GROUP
Fred Drake4baedc12002-04-01 14:53:37 +00004854 PyModule_AddIntConstant(m, "IPV6_JOIN_GROUP", IPV6_JOIN_GROUP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004855#endif
4856#ifdef IPV6_LEAVE_GROUP
Fred Drake4baedc12002-04-01 14:53:37 +00004857 PyModule_AddIntConstant(m, "IPV6_LEAVE_GROUP", IPV6_LEAVE_GROUP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004858#endif
4859#ifdef IPV6_MULTICAST_HOPS
Fred Drake4baedc12002-04-01 14:53:37 +00004860 PyModule_AddIntConstant(m, "IPV6_MULTICAST_HOPS", IPV6_MULTICAST_HOPS);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004861#endif
4862#ifdef IPV6_MULTICAST_IF
Fred Drake4baedc12002-04-01 14:53:37 +00004863 PyModule_AddIntConstant(m, "IPV6_MULTICAST_IF", IPV6_MULTICAST_IF);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004864#endif
4865#ifdef IPV6_MULTICAST_LOOP
Fred Drake4baedc12002-04-01 14:53:37 +00004866 PyModule_AddIntConstant(m, "IPV6_MULTICAST_LOOP", IPV6_MULTICAST_LOOP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004867#endif
4868#ifdef IPV6_UNICAST_HOPS
Fred Drake4baedc12002-04-01 14:53:37 +00004869 PyModule_AddIntConstant(m, "IPV6_UNICAST_HOPS", IPV6_UNICAST_HOPS);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004870#endif
Martin v. Löwisda91d022003-12-30 11:14:01 +00004871 /* Additional IPV6 socket options, defined in RFC 3493 */
4872#ifdef IPV6_V6ONLY
4873 PyModule_AddIntConstant(m, "IPV6_V6ONLY", IPV6_V6ONLY);
4874#endif
4875 /* Advanced IPV6 socket options, from RFC 3542 */
4876#ifdef IPV6_CHECKSUM
4877 PyModule_AddIntConstant(m, "IPV6_CHECKSUM", IPV6_CHECKSUM);
4878#endif
4879#ifdef IPV6_DONTFRAG
4880 PyModule_AddIntConstant(m, "IPV6_DONTFRAG", IPV6_DONTFRAG);
4881#endif
4882#ifdef IPV6_DSTOPTS
4883 PyModule_AddIntConstant(m, "IPV6_DSTOPTS", IPV6_DSTOPTS);
4884#endif
4885#ifdef IPV6_HOPLIMIT
4886 PyModule_AddIntConstant(m, "IPV6_HOPLIMIT", IPV6_HOPLIMIT);
4887#endif
4888#ifdef IPV6_HOPOPTS
4889 PyModule_AddIntConstant(m, "IPV6_HOPOPTS", IPV6_HOPOPTS);
4890#endif
4891#ifdef IPV6_NEXTHOP
4892 PyModule_AddIntConstant(m, "IPV6_NEXTHOP", IPV6_NEXTHOP);
4893#endif
4894#ifdef IPV6_PATHMTU
4895 PyModule_AddIntConstant(m, "IPV6_PATHMTU", IPV6_PATHMTU);
4896#endif
4897#ifdef IPV6_PKTINFO
4898 PyModule_AddIntConstant(m, "IPV6_PKTINFO", IPV6_PKTINFO);
4899#endif
4900#ifdef IPV6_RECVDSTOPTS
4901 PyModule_AddIntConstant(m, "IPV6_RECVDSTOPTS", IPV6_RECVDSTOPTS);
4902#endif
4903#ifdef IPV6_RECVHOPLIMIT
4904 PyModule_AddIntConstant(m, "IPV6_RECVHOPLIMIT", IPV6_RECVHOPLIMIT);
4905#endif
4906#ifdef IPV6_RECVHOPOPTS
4907 PyModule_AddIntConstant(m, "IPV6_RECVHOPOPTS", IPV6_RECVHOPOPTS);
4908#endif
4909#ifdef IPV6_RECVPKTINFO
4910 PyModule_AddIntConstant(m, "IPV6_RECVPKTINFO", IPV6_RECVPKTINFO);
4911#endif
4912#ifdef IPV6_RECVRTHDR
4913 PyModule_AddIntConstant(m, "IPV6_RECVRTHDR", IPV6_RECVRTHDR);
4914#endif
4915#ifdef IPV6_RECVTCLASS
4916 PyModule_AddIntConstant(m, "IPV6_RECVTCLASS", IPV6_RECVTCLASS);
4917#endif
4918#ifdef IPV6_RTHDR
4919 PyModule_AddIntConstant(m, "IPV6_RTHDR", IPV6_RTHDR);
4920#endif
4921#ifdef IPV6_RTHDRDSTOPTS
4922 PyModule_AddIntConstant(m, "IPV6_RTHDRDSTOPTS", IPV6_RTHDRDSTOPTS);
4923#endif
4924#ifdef IPV6_RTHDR_TYPE_0
4925 PyModule_AddIntConstant(m, "IPV6_RTHDR_TYPE_0", IPV6_RTHDR_TYPE_0);
4926#endif
4927#ifdef IPV6_RECVPATHMTU
4928 PyModule_AddIntConstant(m, "IPV6_RECVPATHMTU", IPV6_RECVPATHMTU);
4929#endif
4930#ifdef IPV6_TCLASS
4931 PyModule_AddIntConstant(m, "IPV6_TCLASS", IPV6_TCLASS);
4932#endif
4933#ifdef IPV6_USE_MIN_MTU
4934 PyModule_AddIntConstant(m, "IPV6_USE_MIN_MTU", IPV6_USE_MIN_MTU);
4935#endif
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004936
Guido van Rossum09be4091999-08-09 14:40:40 +00004937 /* TCP options */
4938#ifdef TCP_NODELAY
Fred Drake4baedc12002-04-01 14:53:37 +00004939 PyModule_AddIntConstant(m, "TCP_NODELAY", TCP_NODELAY);
Guido van Rossum09be4091999-08-09 14:40:40 +00004940#endif
4941#ifdef TCP_MAXSEG
Fred Drake4baedc12002-04-01 14:53:37 +00004942 PyModule_AddIntConstant(m, "TCP_MAXSEG", TCP_MAXSEG);
Guido van Rossum09be4091999-08-09 14:40:40 +00004943#endif
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00004944#ifdef TCP_CORK
Fred Drake4baedc12002-04-01 14:53:37 +00004945 PyModule_AddIntConstant(m, "TCP_CORK", TCP_CORK);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00004946#endif
4947#ifdef TCP_KEEPIDLE
Fred Drake4baedc12002-04-01 14:53:37 +00004948 PyModule_AddIntConstant(m, "TCP_KEEPIDLE", TCP_KEEPIDLE);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00004949#endif
4950#ifdef TCP_KEEPINTVL
Fred Drake4baedc12002-04-01 14:53:37 +00004951 PyModule_AddIntConstant(m, "TCP_KEEPINTVL", TCP_KEEPINTVL);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00004952#endif
4953#ifdef TCP_KEEPCNT
Fred Drake4baedc12002-04-01 14:53:37 +00004954 PyModule_AddIntConstant(m, "TCP_KEEPCNT", TCP_KEEPCNT);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00004955#endif
4956#ifdef TCP_SYNCNT
Fred Drake4baedc12002-04-01 14:53:37 +00004957 PyModule_AddIntConstant(m, "TCP_SYNCNT", TCP_SYNCNT);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00004958#endif
4959#ifdef TCP_LINGER2
Fred Drake4baedc12002-04-01 14:53:37 +00004960 PyModule_AddIntConstant(m, "TCP_LINGER2", TCP_LINGER2);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00004961#endif
4962#ifdef TCP_DEFER_ACCEPT
Fred Drake4baedc12002-04-01 14:53:37 +00004963 PyModule_AddIntConstant(m, "TCP_DEFER_ACCEPT", TCP_DEFER_ACCEPT);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00004964#endif
4965#ifdef TCP_WINDOW_CLAMP
Fred Drake4baedc12002-04-01 14:53:37 +00004966 PyModule_AddIntConstant(m, "TCP_WINDOW_CLAMP", TCP_WINDOW_CLAMP);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00004967#endif
4968#ifdef TCP_INFO
Fred Drake4baedc12002-04-01 14:53:37 +00004969 PyModule_AddIntConstant(m, "TCP_INFO", TCP_INFO);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00004970#endif
4971#ifdef TCP_QUICKACK
Fred Drake4baedc12002-04-01 14:53:37 +00004972 PyModule_AddIntConstant(m, "TCP_QUICKACK", TCP_QUICKACK);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00004973#endif
4974
Guido van Rossum09be4091999-08-09 14:40:40 +00004975
4976 /* IPX options */
4977#ifdef IPX_TYPE
Fred Drake4baedc12002-04-01 14:53:37 +00004978 PyModule_AddIntConstant(m, "IPX_TYPE", IPX_TYPE);
Guido van Rossum09be4091999-08-09 14:40:40 +00004979#endif
Guido van Rossum4f199ea1998-04-09 20:56:35 +00004980
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004981 /* get{addr,name}info parameters */
4982#ifdef EAI_ADDRFAMILY
Fred Drake4baedc12002-04-01 14:53:37 +00004983 PyModule_AddIntConstant(m, "EAI_ADDRFAMILY", EAI_ADDRFAMILY);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004984#endif
4985#ifdef EAI_AGAIN
Fred Drake4baedc12002-04-01 14:53:37 +00004986 PyModule_AddIntConstant(m, "EAI_AGAIN", EAI_AGAIN);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004987#endif
4988#ifdef EAI_BADFLAGS
Fred Drake4baedc12002-04-01 14:53:37 +00004989 PyModule_AddIntConstant(m, "EAI_BADFLAGS", EAI_BADFLAGS);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004990#endif
4991#ifdef EAI_FAIL
Fred Drake4baedc12002-04-01 14:53:37 +00004992 PyModule_AddIntConstant(m, "EAI_FAIL", EAI_FAIL);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004993#endif
4994#ifdef EAI_FAMILY
Fred Drake4baedc12002-04-01 14:53:37 +00004995 PyModule_AddIntConstant(m, "EAI_FAMILY", EAI_FAMILY);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004996#endif
4997#ifdef EAI_MEMORY
Fred Drake4baedc12002-04-01 14:53:37 +00004998 PyModule_AddIntConstant(m, "EAI_MEMORY", EAI_MEMORY);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004999#endif
5000#ifdef EAI_NODATA
Fred Drake4baedc12002-04-01 14:53:37 +00005001 PyModule_AddIntConstant(m, "EAI_NODATA", EAI_NODATA);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005002#endif
5003#ifdef EAI_NONAME
Fred Drake4baedc12002-04-01 14:53:37 +00005004 PyModule_AddIntConstant(m, "EAI_NONAME", EAI_NONAME);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005005#endif
Martin v. Löwisda91d022003-12-30 11:14:01 +00005006#ifdef EAI_OVERFLOW
5007 PyModule_AddIntConstant(m, "EAI_OVERFLOW", EAI_OVERFLOW);
5008#endif
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005009#ifdef EAI_SERVICE
Fred Drake4baedc12002-04-01 14:53:37 +00005010 PyModule_AddIntConstant(m, "EAI_SERVICE", EAI_SERVICE);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005011#endif
5012#ifdef EAI_SOCKTYPE
Fred Drake4baedc12002-04-01 14:53:37 +00005013 PyModule_AddIntConstant(m, "EAI_SOCKTYPE", EAI_SOCKTYPE);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005014#endif
5015#ifdef EAI_SYSTEM
Fred Drake4baedc12002-04-01 14:53:37 +00005016 PyModule_AddIntConstant(m, "EAI_SYSTEM", EAI_SYSTEM);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005017#endif
5018#ifdef EAI_BADHINTS
Fred Drake4baedc12002-04-01 14:53:37 +00005019 PyModule_AddIntConstant(m, "EAI_BADHINTS", EAI_BADHINTS);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005020#endif
5021#ifdef EAI_PROTOCOL
Fred Drake4baedc12002-04-01 14:53:37 +00005022 PyModule_AddIntConstant(m, "EAI_PROTOCOL", EAI_PROTOCOL);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005023#endif
5024#ifdef EAI_MAX
Fred Drake4baedc12002-04-01 14:53:37 +00005025 PyModule_AddIntConstant(m, "EAI_MAX", EAI_MAX);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005026#endif
5027#ifdef AI_PASSIVE
Fred Drake4baedc12002-04-01 14:53:37 +00005028 PyModule_AddIntConstant(m, "AI_PASSIVE", AI_PASSIVE);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005029#endif
5030#ifdef AI_CANONNAME
Fred Drake4baedc12002-04-01 14:53:37 +00005031 PyModule_AddIntConstant(m, "AI_CANONNAME", AI_CANONNAME);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005032#endif
5033#ifdef AI_NUMERICHOST
Fred Drake4baedc12002-04-01 14:53:37 +00005034 PyModule_AddIntConstant(m, "AI_NUMERICHOST", AI_NUMERICHOST);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005035#endif
Martin v. Löwisda91d022003-12-30 11:14:01 +00005036#ifdef AI_NUMERICSERV
5037 PyModule_AddIntConstant(m, "AI_NUMERICSERV", AI_NUMERICSERV);
5038#endif
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005039#ifdef AI_MASK
Fred Drake4baedc12002-04-01 14:53:37 +00005040 PyModule_AddIntConstant(m, "AI_MASK", AI_MASK);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005041#endif
5042#ifdef AI_ALL
Fred Drake4baedc12002-04-01 14:53:37 +00005043 PyModule_AddIntConstant(m, "AI_ALL", AI_ALL);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005044#endif
5045#ifdef AI_V4MAPPED_CFG
Fred Drake4baedc12002-04-01 14:53:37 +00005046 PyModule_AddIntConstant(m, "AI_V4MAPPED_CFG", AI_V4MAPPED_CFG);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005047#endif
5048#ifdef AI_ADDRCONFIG
Fred Drake4baedc12002-04-01 14:53:37 +00005049 PyModule_AddIntConstant(m, "AI_ADDRCONFIG", AI_ADDRCONFIG);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005050#endif
5051#ifdef AI_V4MAPPED
Fred Drake4baedc12002-04-01 14:53:37 +00005052 PyModule_AddIntConstant(m, "AI_V4MAPPED", AI_V4MAPPED);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005053#endif
5054#ifdef AI_DEFAULT
Fred Drake4baedc12002-04-01 14:53:37 +00005055 PyModule_AddIntConstant(m, "AI_DEFAULT", AI_DEFAULT);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005056#endif
5057#ifdef NI_MAXHOST
Fred Drake4baedc12002-04-01 14:53:37 +00005058 PyModule_AddIntConstant(m, "NI_MAXHOST", NI_MAXHOST);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005059#endif
5060#ifdef NI_MAXSERV
Fred Drake4baedc12002-04-01 14:53:37 +00005061 PyModule_AddIntConstant(m, "NI_MAXSERV", NI_MAXSERV);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005062#endif
5063#ifdef NI_NOFQDN
Fred Drake4baedc12002-04-01 14:53:37 +00005064 PyModule_AddIntConstant(m, "NI_NOFQDN", NI_NOFQDN);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005065#endif
5066#ifdef NI_NUMERICHOST
Fred Drake4baedc12002-04-01 14:53:37 +00005067 PyModule_AddIntConstant(m, "NI_NUMERICHOST", NI_NUMERICHOST);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005068#endif
5069#ifdef NI_NAMEREQD
Fred Drake4baedc12002-04-01 14:53:37 +00005070 PyModule_AddIntConstant(m, "NI_NAMEREQD", NI_NAMEREQD);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005071#endif
5072#ifdef NI_NUMERICSERV
Fred Drake4baedc12002-04-01 14:53:37 +00005073 PyModule_AddIntConstant(m, "NI_NUMERICSERV", NI_NUMERICSERV);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005074#endif
5075#ifdef NI_DGRAM
Fred Drake4baedc12002-04-01 14:53:37 +00005076 PyModule_AddIntConstant(m, "NI_DGRAM", NI_DGRAM);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005077#endif
5078
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005079 /* shutdown() parameters */
5080#ifdef SHUT_RD
5081 PyModule_AddIntConstant(m, "SHUT_RD", SHUT_RD);
5082#elif defined(SD_RECEIVE)
5083 PyModule_AddIntConstant(m, "SHUT_RD", SD_RECEIVE);
5084#else
5085 PyModule_AddIntConstant(m, "SHUT_RD", 0);
5086#endif
5087#ifdef SHUT_WR
5088 PyModule_AddIntConstant(m, "SHUT_WR", SHUT_WR);
5089#elif defined(SD_SEND)
5090 PyModule_AddIntConstant(m, "SHUT_WR", SD_SEND);
5091#else
5092 PyModule_AddIntConstant(m, "SHUT_WR", 1);
5093#endif
5094#ifdef SHUT_RDWR
5095 PyModule_AddIntConstant(m, "SHUT_RDWR", SHUT_RDWR);
5096#elif defined(SD_BOTH)
5097 PyModule_AddIntConstant(m, "SHUT_RDWR", SD_BOTH);
5098#else
5099 PyModule_AddIntConstant(m, "SHUT_RDWR", 2);
5100#endif
5101
Christian Heimesfaf2f632008-01-06 16:59:19 +00005102#ifdef SIO_RCVALL
Christian Heimesd32ed6f2008-01-14 18:49:24 +00005103 {
Kristján Valur Jónsson847ec752009-09-27 21:10:38 +00005104 DWORD codes[] = {SIO_RCVALL, SIO_KEEPALIVE_VALS};
5105 const char *names[] = {"SIO_RCVALL", "SIO_KEEPALIVE_VALS"};
5106 int i;
5107 for(i = 0; i<sizeof(codes)/sizeof(*codes); ++i) {
5108 PyObject *tmp;
5109 tmp = PyLong_FromUnsignedLong(codes[i]);
5110 if (tmp == NULL)
5111 return NULL;
5112 PyModule_AddObject(m, names[i], tmp);
5113 }
Christian Heimesd32ed6f2008-01-14 18:49:24 +00005114 }
Christian Heimesfaf2f632008-01-06 16:59:19 +00005115 PyModule_AddIntConstant(m, "RCVALL_OFF", RCVALL_OFF);
5116 PyModule_AddIntConstant(m, "RCVALL_ON", RCVALL_ON);
5117 PyModule_AddIntConstant(m, "RCVALL_SOCKETLEVELONLY", RCVALL_SOCKETLEVELONLY);
Amaury Forgeot d'Arc762681b2008-06-12 23:03:41 +00005118#ifdef RCVALL_IPLEVEL
Christian Heimesfaf2f632008-01-06 16:59:19 +00005119 PyModule_AddIntConstant(m, "RCVALL_IPLEVEL", RCVALL_IPLEVEL);
Amaury Forgeot d'Arc762681b2008-06-12 23:03:41 +00005120#endif
5121#ifdef RCVALL_MAX
Christian Heimesfaf2f632008-01-06 16:59:19 +00005122 PyModule_AddIntConstant(m, "RCVALL_MAX", RCVALL_MAX);
Amaury Forgeot d'Arc762681b2008-06-12 23:03:41 +00005123#endif
Christian Heimesfaf2f632008-01-06 16:59:19 +00005124#endif /* _MSTCPIP_ */
5125
Guido van Rossum4f199ea1998-04-09 20:56:35 +00005126 /* Initialize gethostbyname lock */
Just van Rossum1040d2c2003-05-09 07:53:18 +00005127#if defined(USE_GETHOSTBYNAME_LOCK) || defined(USE_GETADDRINFO_LOCK)
5128 netdb_lock = PyThread_allocate_lock();
Guido van Rossum4f199ea1998-04-09 20:56:35 +00005129#endif
Martin v. Löwis1a214512008-06-11 05:26:20 +00005130 return m;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00005131}
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005132
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00005133
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005134#ifndef HAVE_INET_PTON
Christian Heimes96e7b3d2007-11-20 06:51:17 +00005135#if !defined(NTDDI_VERSION) || (NTDDI_VERSION < NTDDI_LONGHORN)
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00005136
5137/* Simplistic emulation code for inet_pton that only works for IPv4 */
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00005138/* These are not exposed because they do not set errno properly */
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00005139
Guido van Rossum3eede5a2002-06-07 02:08:35 +00005140int
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00005141inet_pton(int af, const char *src, void *dst)
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005142{
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00005143 if (af == AF_INET) {
Benjamin Petersonf91df042009-02-13 02:50:59 +00005144#if (SIZEOF_INT != 4)
5145#error "Not sure if in_addr_t exists and int is not 32-bits."
5146#endif
5147 unsigned int packed_addr;
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005148 packed_addr = inet_addr(src);
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005149 if (packed_addr == INADDR_NONE)
5150 return 0;
5151 memcpy(dst, &packed_addr, 4);
5152 return 1;
5153 }
5154 /* Should set errno to EAFNOSUPPORT */
5155 return -1;
5156}
5157
Martin v. Löwisc925b1532001-07-21 09:42:15 +00005158const char *
5159inet_ntop(int af, const void *src, char *dst, socklen_t size)
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005160{
5161 if (af == AF_INET) {
5162 struct in_addr packed_addr;
5163 if (size < 16)
5164 /* Should set errno to ENOSPC. */
5165 return NULL;
5166 memcpy(&packed_addr, src, sizeof(packed_addr));
5167 return strncpy(dst, inet_ntoa(packed_addr), size);
5168 }
5169 /* Should set errno to EAFNOSUPPORT */
5170 return NULL;
5171}
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00005172
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005173#endif
Christian Heimesb6150692007-11-15 23:37:07 +00005174#endif