blob: c0c9c84dc7f51abb48d9f4fbb62fae8f2fdbffbb [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,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000020 a subclass of socket.error
Martin v. Löwis2d8d4272001-07-21 18:05:31 +000021- socket.herror: exception raised for gethostby* errors,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000022 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
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000030 (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])
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000037 --> List of (family, socktype, proto, canonname, sockaddr)
Martin v. Löwis2d8d4272001-07-21 18:05:31 +000038- 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:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000057 TIPC_ADDR_NAMESEQ, TIPC_ADDR_NAME, and TIPC_ADDR_ID;
Christian Heimes043d6f62008-01-07 17:19:16 +000058 and scope can be one of:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000059 TIPC_ZONE_SCOPE, TIPC_CLUSTER_SCOPE, and TIPC_NODE_SCOPE.
Christian Heimes043d6f62008-01-07 17:19:16 +000060 The meaning of v1, v2 and v3 depends on the value of addr_type:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000061 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
Christian Heimes043d6f62008-01-07 17:19:16 +000073
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
Antoine Pitrouf95a1b32010-05-09 15:52:27 +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{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000354 HANDLE newhandle;
Guido van Rossum39eb8fa2007-11-16 01:24:05 +0000355
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000356 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;
Guido van Rossum39eb8fa2007-11-16 01:24:05 +0000364}
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
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000421#define SAS2SA(x) ((struct sockaddr *)(x))
Thomas Wouters89f507f2006-12-13 04:49:30 +0000422
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{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000476 PyErr_SetString(socket_error, "unable to select on socket");
477 return NULL;
Neal Norwitz082b2df2006-02-07 07:04:46 +0000478}
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
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000487 int err_no = WSAGetLastError();
488 /* 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)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +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
Antoine Pitrouf95a1b32010-05-09 15:52:27 +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
Antoine Pitrouf95a1b32010-05-09 15:52:27 +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 ];
515 while (lastc > outbuf &&
516 isspace(Py_CHARMASK(*lastc))) {
517 /* Trim trailing whitespace (CRLF) */
518 *lastc-- = '\0';
519 }
520 }
521 v = Py_BuildValue("(is)", myerrorcode, outbuf);
522 if (v != NULL) {
523 PyErr_SetObject(socket_error, v);
524 Py_DECREF(v);
525 }
526 return NULL;
527 }
528 }
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000529#endif
530
Antoine Pitrouf95a1b32010-05-09 15:52:27 +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{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000538 PyObject *v;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000539
540#ifdef HAVE_HSTRERROR
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000541 v = Py_BuildValue("(is)", h_error, (char *)hstrerror(h_error));
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000542#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000543 v = Py_BuildValue("(is)", h_error, "host not found");
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000544#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000545 if (v != NULL) {
546 PyErr_SetObject(socket_herror, v);
547 Py_DECREF(v);
548 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000549
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000550 return NULL;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000551}
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{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000557 PyObject *v;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000558
Martin v. Löwis272cb402002-03-01 08:31:07 +0000559#ifdef EAI_SYSTEM
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000560 /* EAI_SYSTEM is not available on Windows XP. */
561 if (error == EAI_SYSTEM)
562 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
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000566 v = Py_BuildValue("(is)", error, gai_strerror(error));
Martin v. Löwisf95dd0a2001-08-15 17:14:33 +0000567#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000568 v = Py_BuildValue("(is)", error, "getaddrinfo failed");
Martin v. Löwisf95dd0a2001-08-15 17:14:33 +0000569#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000570 if (v != NULL) {
571 PyErr_SetObject(socket_gaierror, v);
572 Py_DECREF(v);
573 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000574
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000575 return NULL;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000576}
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{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000583 int n = 0;
584 int remaining = len;
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000585
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000586 while (remaining > 0) {
587 unsigned int segment;
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000588
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000589 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 */
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000597
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000598 return len;
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000599}
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
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000608 int delay_flag;
Guido van Rossum67f7a382002-06-06 21:08:16 +0000609#endif
Guido van Rossum67f7a382002-06-06 21:08:16 +0000610
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000611 Py_BEGIN_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +0000612#ifndef MS_WINDOWS
613#if defined(PYOS_OS2) && !defined(PYCC_GCC)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000614 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)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000617 block = !block;
618 ioctl(s->sock_fd, FIONBIO, (unsigned int *)&block);
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000619#else /* !PYOS_OS2 && !__VMS */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000620 delay_flag = fcntl(s->sock_fd, F_GETFL, 0);
621 if (block)
622 delay_flag &= (~O_NONBLOCK);
623 else
624 delay_flag |= O_NONBLOCK;
625 fcntl(s->sock_fd, F_SETFL, delay_flag);
Guido van Rossum67f7a382002-06-06 21:08:16 +0000626#endif /* !PYOS_OS2 */
627#else /* MS_WINDOWS */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000628 block = !block;
629 ioctlsocket(s->sock_fd, FIONBIO, (u_long*)&block);
Guido van Rossum67f7a382002-06-06 21:08:16 +0000630#endif /* MS_WINDOWS */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000631 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +0000632
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000633 /* Since these don't return anything */
634 return 1;
Guido van Rossum67f7a382002-06-06 21:08:16 +0000635}
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{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000645 int n;
Guido van Rossum11ba0942002-06-13 15:07:44 +0000646
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000647 /* Nothing to do unless we're in timeout mode (not non-blocking) */
648 if (s->sock_timeout <= 0.0)
649 return 0;
Guido van Rossum67f7a382002-06-06 21:08:16 +0000650
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000651 /* Guard against closed socket */
652 if (s->sock_fd < 0)
653 return 0;
Guido van Rossumad654902002-07-19 12:44:59 +0000654
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000655 /* Prefer poll, if available, since you can poll() any fd
656 * which can't be done with select(). */
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000657#ifdef HAVE_POLL
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000658 {
659 struct pollfd pollfd;
660 int timeout;
Guido van Rossum67f7a382002-06-06 21:08:16 +0000661
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000662 pollfd.fd = s->sock_fd;
663 pollfd.events = writing ? POLLOUT : POLLIN;
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000664
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000665 /* s->sock_timeout is in seconds, timeout in ms */
666 timeout = (int)(s->sock_timeout * 1000 + 0.5);
667 n = poll(&pollfd, 1, timeout);
668 }
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000669#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000670 {
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);
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000678
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000679 /* See if the socket is ready */
680 if (writing)
Antoine Pitrou19467d22010-08-17 19:33:30 +0000681 n = select(Py_SAFE_DOWNCAST(s->sock_fd+1, SOCKET_T, int),
682 NULL, &fds, NULL, &tv);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000683 else
Antoine Pitrou19467d22010-08-17 19:33:30 +0000684 n = select(Py_SAFE_DOWNCAST(s->sock_fd+1, SOCKET_T, int),
685 &fds, NULL, NULL, &tv);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000686 }
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000687#endif
Guido van Rossum39eb8fa2007-11-16 01:24:05 +0000688
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000689 if (n < 0)
690 return -1;
691 if (n == 0)
692 return 1;
693 return 0;
Guido van Rossum67f7a382002-06-06 21:08:16 +0000694}
695
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000696/* Initialize a new socket object. */
697
Tim Petersa12b4cf2002-07-18 22:38:44 +0000698static double defaulttimeout = -1.0; /* Default timeout for new sockets */
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +0000699
Martin v. Löwis1a214512008-06-11 05:26:20 +0000700static void
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000701init_sockobject(PySocketSockObject *s,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000702 SOCKET_T fd, int family, int type, int proto)
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000703{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000704 s->sock_fd = fd;
705 s->sock_family = family;
706 s->sock_type = type;
707 s->sock_proto = proto;
708 s->sock_timeout = defaulttimeout;
Guido van Rossum67f7a382002-06-06 21:08:16 +0000709
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000710 s->errorhandler = &set_error;
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +0000711
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000712 if (defaulttimeout >= 0.0)
713 internal_setblocking(s, 0);
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +0000714
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000715}
716
717
Guido van Rossum30a685f1991-06-27 15:51:29 +0000718/* Create a new socket object.
719 This just creates the object and initializes it.
720 If the creation fails, return NULL and set an exception (implicit
721 in NEWOBJ()). */
722
Guido van Rossum73624e91994-10-10 17:59:00 +0000723static PySocketSockObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000724new_sockobject(SOCKET_T fd, int family, int type, int proto)
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000725{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000726 PySocketSockObject *s;
727 s = (PySocketSockObject *)
728 PyType_GenericNew(&sock_type, NULL, NULL);
729 if (s != NULL)
730 init_sockobject(s, fd, family, type, proto);
731 return s;
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000732}
733
Guido van Rossum30a685f1991-06-27 15:51:29 +0000734
Guido van Rossum48a680c2001-03-02 06:34:14 +0000735/* Lock to allow python interpreter to continue, but only allow one
Just van Rossum1040d2c2003-05-09 07:53:18 +0000736 thread to be in gethostbyname or getaddrinfo */
737#if defined(USE_GETHOSTBYNAME_LOCK) || defined(USE_GETADDRINFO_LOCK)
738PyThread_type_lock netdb_lock;
Guido van Rossum4f199ea1998-04-09 20:56:35 +0000739#endif
740
741
Guido van Rossum30a685f1991-06-27 15:51:29 +0000742/* Convert a string specifying a host name or one of a few symbolic
743 names to a numeric IP address. This usually calls gethostbyname()
744 to do the work; the names "" and "<broadcast>" are special.
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000745 Return the length (IPv4 should be 4 bytes), or negative if
Guido van Rossum30a685f1991-06-27 15:51:29 +0000746 an error occurred; then an exception is raised. */
747
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000748static int
Martin v. Löwisddc6f472002-07-28 16:10:31 +0000749setipaddr(char *name, struct sockaddr *addr_ret, size_t addr_ret_size, int af)
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000750{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000751 struct addrinfo hints, *res;
752 int error;
753 int d1, d2, d3, d4;
754 char ch;
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000755
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000756 memset((void *) addr_ret, '\0', sizeof(*addr_ret));
757 if (name[0] == '\0') {
758 int siz;
759 memset(&hints, 0, sizeof(hints));
760 hints.ai_family = af;
761 hints.ai_socktype = SOCK_DGRAM; /*dummy*/
762 hints.ai_flags = AI_PASSIVE;
763 Py_BEGIN_ALLOW_THREADS
764 ACQUIRE_GETADDRINFO_LOCK
765 error = getaddrinfo(NULL, "0", &hints, &res);
766 Py_END_ALLOW_THREADS
767 /* We assume that those thread-unsafe getaddrinfo() versions
768 *are* safe regarding their return value, ie. that a
769 subsequent call to getaddrinfo() does not destroy the
770 outcome of the first call. */
771 RELEASE_GETADDRINFO_LOCK
772 if (error) {
773 set_gaierror(error);
774 return -1;
775 }
776 switch (res->ai_family) {
777 case AF_INET:
778 siz = 4;
779 break;
Martin v. Löwis44ddbde2001-12-02 10:15:37 +0000780#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000781 case AF_INET6:
782 siz = 16;
783 break;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000784#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000785 default:
786 freeaddrinfo(res);
787 PyErr_SetString(socket_error,
788 "unsupported address family");
789 return -1;
790 }
791 if (res->ai_next) {
792 freeaddrinfo(res);
793 PyErr_SetString(socket_error,
794 "wildcard resolved to multiple address");
795 return -1;
796 }
797 if (res->ai_addrlen < addr_ret_size)
798 addr_ret_size = res->ai_addrlen;
799 memcpy(addr_ret, res->ai_addr, addr_ret_size);
800 freeaddrinfo(res);
801 return siz;
802 }
803 if (name[0] == '<' && strcmp(name, "<broadcast>") == 0) {
804 struct sockaddr_in *sin;
805 if (af != AF_INET && af != AF_UNSPEC) {
806 PyErr_SetString(socket_error,
807 "address family mismatched");
808 return -1;
809 }
810 sin = (struct sockaddr_in *)addr_ret;
811 memset((void *) sin, '\0', sizeof(*sin));
812 sin->sin_family = AF_INET;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000813#ifdef HAVE_SOCKADDR_SA_LEN
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000814 sin->sin_len = sizeof(*sin);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000815#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000816 sin->sin_addr.s_addr = INADDR_BROADCAST;
817 return sizeof(sin->sin_addr);
818 }
819 if (sscanf(name, "%d.%d.%d.%d%c", &d1, &d2, &d3, &d4, &ch) == 4 &&
820 0 <= d1 && d1 <= 255 && 0 <= d2 && d2 <= 255 &&
821 0 <= d3 && d3 <= 255 && 0 <= d4 && d4 <= 255) {
822 struct sockaddr_in *sin;
823 sin = (struct sockaddr_in *)addr_ret;
824 sin->sin_addr.s_addr = htonl(
825 ((long) d1 << 24) | ((long) d2 << 16) |
826 ((long) d3 << 8) | ((long) d4 << 0));
827 sin->sin_family = AF_INET;
Anthony Baxter0e85f9d2003-05-02 15:40:46 +0000828#ifdef HAVE_SOCKADDR_SA_LEN
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000829 sin->sin_len = sizeof(*sin);
Anthony Baxter0e85f9d2003-05-02 15:40:46 +0000830#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000831 return 4;
832 }
833 memset(&hints, 0, sizeof(hints));
834 hints.ai_family = af;
835 Py_BEGIN_ALLOW_THREADS
836 ACQUIRE_GETADDRINFO_LOCK
837 error = getaddrinfo(name, NULL, &hints, &res);
Martin v. Löwis7c4b5fa2001-10-25 09:04:03 +0000838#if defined(__digital__) && defined(__unix__)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000839 if (error == EAI_NONAME && af == AF_UNSPEC) {
840 /* On Tru64 V5.1, numeric-to-addr conversion fails
841 if no address family is given. Assume IPv4 for now.*/
842 hints.ai_family = AF_INET;
843 error = getaddrinfo(name, NULL, &hints, &res);
844 }
Martin v. Löwis7c4b5fa2001-10-25 09:04:03 +0000845#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000846 Py_END_ALLOW_THREADS
847 RELEASE_GETADDRINFO_LOCK /* see comment in setipaddr() */
848 if (error) {
849 set_gaierror(error);
850 return -1;
851 }
852 if (res->ai_addrlen < addr_ret_size)
853 addr_ret_size = res->ai_addrlen;
854 memcpy((char *) addr_ret, res->ai_addr, addr_ret_size);
855 freeaddrinfo(res);
856 switch (addr_ret->sa_family) {
857 case AF_INET:
858 return 4;
Martin v. Löwis44ddbde2001-12-02 10:15:37 +0000859#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000860 case AF_INET6:
861 return 16;
Guido van Rossum955becc1999-03-22 20:14:53 +0000862#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000863 default:
864 PyErr_SetString(socket_error, "unknown address family");
865 return -1;
866 }
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000867}
868
Guido van Rossum30a685f1991-06-27 15:51:29 +0000869
Guido van Rossum30a685f1991-06-27 15:51:29 +0000870/* Create a string object representing an IP address.
871 This is always a string of the form 'dd.dd.dd.dd' (with variable
872 size numbers). */
873
Guido van Rossum73624e91994-10-10 17:59:00 +0000874static PyObject *
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000875makeipaddr(struct sockaddr *addr, int addrlen)
Guido van Rossum30a685f1991-06-27 15:51:29 +0000876{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000877 char buf[NI_MAXHOST];
878 int error;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000879
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000880 error = getnameinfo(addr, addrlen, buf, sizeof(buf), NULL, 0,
881 NI_NUMERICHOST);
882 if (error) {
883 set_gaierror(error);
884 return NULL;
885 }
886 return PyUnicode_FromString(buf);
Guido van Rossum30a685f1991-06-27 15:51:29 +0000887}
888
889
Martin v. Löwis558d9bf2004-06-03 09:24:42 +0000890#ifdef USE_BLUETOOTH
891/* Convert a string representation of a Bluetooth address into a numeric
892 address. Returns the length (6), or raises an exception and returns -1 if
893 an error occurred. */
894
895static int
896setbdaddr(char *name, bdaddr_t *bdaddr)
897{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000898 unsigned int b0, b1, b2, b3, b4, b5;
899 char ch;
900 int n;
Martin v. Löwis558d9bf2004-06-03 09:24:42 +0000901
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000902 n = sscanf(name, "%X:%X:%X:%X:%X:%X%c",
903 &b5, &b4, &b3, &b2, &b1, &b0, &ch);
904 if (n == 6 && (b0 | b1 | b2 | b3 | b4 | b5) < 256) {
905 bdaddr->b[0] = b0;
906 bdaddr->b[1] = b1;
907 bdaddr->b[2] = b2;
908 bdaddr->b[3] = b3;
909 bdaddr->b[4] = b4;
910 bdaddr->b[5] = b5;
911 return 6;
912 } else {
913 PyErr_SetString(socket_error, "bad bluetooth address");
914 return -1;
915 }
Martin v. Löwis558d9bf2004-06-03 09:24:42 +0000916}
917
918/* Create a string representation of the Bluetooth address. This is always a
919 string of the form 'XX:XX:XX:XX:XX:XX' where XX is a two digit hexadecimal
920 value (zero padded if necessary). */
921
922static PyObject *
923makebdaddr(bdaddr_t *bdaddr)
924{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000925 char buf[(6 * 2) + 5 + 1];
Martin v. Löwis558d9bf2004-06-03 09:24:42 +0000926
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000927 sprintf(buf, "%02X:%02X:%02X:%02X:%02X:%02X",
928 bdaddr->b[5], bdaddr->b[4], bdaddr->b[3],
929 bdaddr->b[2], bdaddr->b[1], bdaddr->b[0]);
930 return PyUnicode_FromString(buf);
Martin v. Löwis558d9bf2004-06-03 09:24:42 +0000931}
932#endif
933
934
Guido van Rossum30a685f1991-06-27 15:51:29 +0000935/* Create an object representing the given socket address,
936 suitable for passing it back to bind(), connect() etc.
937 The family field of the sockaddr structure is inspected
938 to determine what kind of address it really is. */
939
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000940/*ARGSUSED*/
Guido van Rossum73624e91994-10-10 17:59:00 +0000941static PyObject *
Antoine Pitrou19467d22010-08-17 19:33:30 +0000942makesockaddr(SOCKET_T sockfd, struct sockaddr *addr, size_t addrlen, int proto)
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000943{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000944 if (addrlen == 0) {
945 /* No address -- may be recvfrom() from known socket */
946 Py_INCREF(Py_None);
947 return Py_None;
948 }
Guido van Rossum25bec8c1992-08-05 19:00:45 +0000949
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000950 switch (addr->sa_family) {
Guido van Rossum30a685f1991-06-27 15:51:29 +0000951
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000952 case AF_INET:
953 {
954 struct sockaddr_in *a;
955 PyObject *addrobj = makeipaddr(addr, sizeof(*a));
956 PyObject *ret = NULL;
957 if (addrobj) {
958 a = (struct sockaddr_in *)addr;
959 ret = Py_BuildValue("Oi", addrobj, ntohs(a->sin_port));
960 Py_DECREF(addrobj);
961 }
962 return ret;
963 }
Guido van Rossum30a685f1991-06-27 15:51:29 +0000964
Andrew MacIntyred12dfbb2004-04-04 07:13:49 +0000965#if defined(AF_UNIX)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000966 case AF_UNIX:
967 {
968 struct sockaddr_un *a = (struct sockaddr_un *) addr;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000969#ifdef linux
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000970 if (a->sun_path[0] == 0) { /* Linux abstract namespace */
971 addrlen -= offsetof(struct sockaddr_un, sun_path);
972 return PyBytes_FromStringAndSize(a->sun_path, addrlen);
973 }
974 else
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000975#endif /* linux */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000976 {
977 /* regular NULL-terminated string */
978 return PyUnicode_FromString(a->sun_path);
979 }
980 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000981#endif /* AF_UNIX */
982
Martin v. Löwis11017b12006-01-14 18:12:57 +0000983#if defined(AF_NETLINK)
984 case AF_NETLINK:
985 {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000986 struct sockaddr_nl *a = (struct sockaddr_nl *) addr;
987 return Py_BuildValue("II", a->nl_pid, a->nl_groups);
Martin v. Löwis11017b12006-01-14 18:12:57 +0000988 }
989#endif /* AF_NETLINK */
990
Martin v. Löwis44ddbde2001-12-02 10:15:37 +0000991#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000992 case AF_INET6:
993 {
994 struct sockaddr_in6 *a;
995 PyObject *addrobj = makeipaddr(addr, sizeof(*a));
996 PyObject *ret = NULL;
997 if (addrobj) {
998 a = (struct sockaddr_in6 *)addr;
999 ret = Py_BuildValue("Oiii",
1000 addrobj,
1001 ntohs(a->sin6_port),
1002 a->sin6_flowinfo,
1003 a->sin6_scope_id);
1004 Py_DECREF(addrobj);
1005 }
1006 return ret;
1007 }
Jeremy Hylton22308652001-02-02 03:23:09 +00001008#endif
Guido van Rossum30a685f1991-06-27 15:51:29 +00001009
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001010#ifdef USE_BLUETOOTH
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001011 case AF_BLUETOOTH:
1012 switch (proto) {
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001013
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001014 case BTPROTO_L2CAP:
1015 {
1016 struct sockaddr_l2 *a = (struct sockaddr_l2 *) addr;
1017 PyObject *addrobj = makebdaddr(&_BT_L2_MEMB(a, bdaddr));
1018 PyObject *ret = NULL;
1019 if (addrobj) {
1020 ret = Py_BuildValue("Oi",
1021 addrobj,
1022 _BT_L2_MEMB(a, psm));
1023 Py_DECREF(addrobj);
1024 }
1025 return ret;
1026 }
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001027
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001028 case BTPROTO_RFCOMM:
1029 {
1030 struct sockaddr_rc *a = (struct sockaddr_rc *) addr;
1031 PyObject *addrobj = makebdaddr(&_BT_RC_MEMB(a, bdaddr));
1032 PyObject *ret = NULL;
1033 if (addrobj) {
1034 ret = Py_BuildValue("Oi",
1035 addrobj,
1036 _BT_RC_MEMB(a, channel));
1037 Py_DECREF(addrobj);
1038 }
1039 return ret;
1040 }
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001041
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001042 case BTPROTO_HCI:
1043 {
1044 struct sockaddr_hci *a = (struct sockaddr_hci *) addr;
1045 PyObject *ret = NULL;
1046 ret = Py_BuildValue("i", _BT_HCI_MEMB(a, dev));
1047 return ret;
1048 }
Thomas Wouterscf297e42007-02-23 15:07:44 +00001049
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001050#if !defined(__FreeBSD__)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001051 case BTPROTO_SCO:
1052 {
1053 struct sockaddr_sco *a = (struct sockaddr_sco *) addr;
1054 return makebdaddr(&_BT_SCO_MEMB(a, bdaddr));
1055 }
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001056#endif
1057
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001058 default:
1059 PyErr_SetString(PyExc_ValueError,
1060 "Unknown Bluetooth protocol");
1061 return NULL;
1062 }
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001063#endif
1064
Martin v. Löwis1ba3fd52001-08-10 20:29:40 +00001065#ifdef HAVE_NETPACKET_PACKET_H
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001066 case AF_PACKET:
1067 {
1068 struct sockaddr_ll *a = (struct sockaddr_ll *)addr;
1069 char *ifname = "";
1070 struct ifreq ifr;
1071 /* need to look up interface name give index */
1072 if (a->sll_ifindex) {
1073 ifr.ifr_ifindex = a->sll_ifindex;
1074 if (ioctl(sockfd, SIOCGIFNAME, &ifr) == 0)
1075 ifname = ifr.ifr_name;
1076 }
1077 return Py_BuildValue("shbhy#",
1078 ifname,
1079 ntohs(a->sll_protocol),
1080 a->sll_pkttype,
1081 a->sll_hatype,
1082 a->sll_addr,
1083 a->sll_halen);
1084 }
Jeremy Hylton22308652001-02-02 03:23:09 +00001085#endif
Guido van Rossum48a680c2001-03-02 06:34:14 +00001086
Christian Heimes043d6f62008-01-07 17:19:16 +00001087#ifdef HAVE_LINUX_TIPC_H
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001088 case AF_TIPC:
1089 {
1090 struct sockaddr_tipc *a = (struct sockaddr_tipc *) addr;
1091 if (a->addrtype == TIPC_ADDR_NAMESEQ) {
1092 return Py_BuildValue("IIIII",
1093 a->addrtype,
1094 a->addr.nameseq.type,
1095 a->addr.nameseq.lower,
1096 a->addr.nameseq.upper,
1097 a->scope);
1098 } else if (a->addrtype == TIPC_ADDR_NAME) {
1099 return Py_BuildValue("IIIII",
1100 a->addrtype,
1101 a->addr.name.name.type,
1102 a->addr.name.name.instance,
1103 a->addr.name.name.instance,
1104 a->scope);
1105 } else if (a->addrtype == TIPC_ADDR_ID) {
1106 return Py_BuildValue("IIIII",
1107 a->addrtype,
1108 a->addr.id.node,
1109 a->addr.id.ref,
1110 0,
1111 a->scope);
1112 } else {
1113 PyErr_SetString(PyExc_ValueError,
1114 "Invalid address type");
1115 return NULL;
1116 }
1117 }
Christian Heimes043d6f62008-01-07 17:19:16 +00001118#endif
1119
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001120 /* More cases here... */
Guido van Rossum30a685f1991-06-27 15:51:29 +00001121
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001122 default:
1123 /* If we don't know the address family, don't raise an
1124 exception -- return it as an (int, bytes) tuple. */
1125 return Py_BuildValue("iy#",
1126 addr->sa_family,
1127 addr->sa_data,
1128 sizeof(addr->sa_data));
Guido van Rossum25bec8c1992-08-05 19:00:45 +00001129
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001130 }
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001131}
1132
Guido van Rossum30a685f1991-06-27 15:51:29 +00001133
1134/* Parse a socket address argument according to the socket object's
1135 address family. Return 1 if the address was in the proper format,
1136 0 of not. The address is returned through addr_ret, its length
1137 through len_ret. */
1138
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001139static int
Guido van Rossum48a680c2001-03-02 06:34:14 +00001140getsockaddrarg(PySocketSockObject *s, PyObject *args,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001141 struct sockaddr *addr_ret, int *len_ret)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001142{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001143 switch (s->sock_family) {
Guido van Rossum30a685f1991-06-27 15:51:29 +00001144
Andrew MacIntyred12dfbb2004-04-04 07:13:49 +00001145#if defined(AF_UNIX)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001146 case AF_UNIX:
1147 {
1148 struct sockaddr_un* addr;
1149 char *path;
1150 int len;
1151 if (!PyArg_Parse(args, "s#", &path, &len))
1152 return 0;
Thomas Wouters89f507f2006-12-13 04:49:30 +00001153
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001154 addr = (struct sockaddr_un*)addr_ret;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001155#ifdef linux
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001156 if (len > 0 && path[0] == 0) {
1157 /* Linux abstract namespace extension */
1158 if (len > sizeof addr->sun_path) {
1159 PyErr_SetString(socket_error,
1160 "AF_UNIX path too long");
1161 return 0;
1162 }
1163 }
1164 else
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001165#endif /* linux */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001166 {
1167 /* regular NULL-terminated string */
1168 if (len >= sizeof addr->sun_path) {
1169 PyErr_SetString(socket_error,
1170 "AF_UNIX path too long");
1171 return 0;
1172 }
1173 addr->sun_path[len] = 0;
1174 }
1175 addr->sun_family = s->sock_family;
1176 memcpy(addr->sun_path, path, len);
Andrew MacIntyredaedf212004-04-11 12:03:57 +00001177#if defined(PYOS_OS2)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001178 *len_ret = sizeof(*addr);
Andrew MacIntyredaedf212004-04-11 12:03:57 +00001179#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001180 *len_ret = len + offsetof(struct sockaddr_un, sun_path);
Andrew MacIntyredaedf212004-04-11 12:03:57 +00001181#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001182 return 1;
1183 }
Guido van Rossumb6775db1994-08-01 11:34:53 +00001184#endif /* AF_UNIX */
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001185
Martin v. Löwis11017b12006-01-14 18:12:57 +00001186#if defined(AF_NETLINK)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001187 case AF_NETLINK:
1188 {
1189 struct sockaddr_nl* addr;
1190 int pid, groups;
1191 addr = (struct sockaddr_nl *)addr_ret;
1192 if (!PyTuple_Check(args)) {
1193 PyErr_Format(
1194 PyExc_TypeError,
1195 "getsockaddrarg: "
1196 "AF_NETLINK address must be tuple, not %.500s",
1197 Py_TYPE(args)->tp_name);
1198 return 0;
1199 }
1200 if (!PyArg_ParseTuple(args, "II:getsockaddrarg", &pid, &groups))
1201 return 0;
1202 addr->nl_family = AF_NETLINK;
1203 addr->nl_pid = pid;
1204 addr->nl_groups = groups;
1205 *len_ret = sizeof(*addr);
1206 return 1;
1207 }
Martin v. Löwis11017b12006-01-14 18:12:57 +00001208#endif
1209
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001210 case AF_INET:
1211 {
1212 struct sockaddr_in* addr;
1213 char *host;
1214 int port, result;
1215 if (!PyTuple_Check(args)) {
1216 PyErr_Format(
1217 PyExc_TypeError,
1218 "getsockaddrarg: "
1219 "AF_INET address must be tuple, not %.500s",
1220 Py_TYPE(args)->tp_name);
1221 return 0;
1222 }
1223 if (!PyArg_ParseTuple(args, "eti:getsockaddrarg",
1224 "idna", &host, &port))
1225 return 0;
1226 addr=(struct sockaddr_in*)addr_ret;
1227 result = setipaddr(host, (struct sockaddr *)addr,
1228 sizeof(*addr), AF_INET);
1229 PyMem_Free(host);
1230 if (result < 0)
1231 return 0;
1232 if (port < 0 || port > 0xffff) {
1233 PyErr_SetString(
1234 PyExc_OverflowError,
1235 "getsockaddrarg: port must be 0-65535.");
1236 return 0;
1237 }
1238 addr->sin_family = AF_INET;
1239 addr->sin_port = htons((short)port);
1240 *len_ret = sizeof *addr;
1241 return 1;
1242 }
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001243
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00001244#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001245 case AF_INET6:
1246 {
1247 struct sockaddr_in6* addr;
1248 char *host;
1249 int port, flowinfo, scope_id, result;
1250 flowinfo = scope_id = 0;
1251 if (!PyTuple_Check(args)) {
1252 PyErr_Format(
1253 PyExc_TypeError,
1254 "getsockaddrarg: "
1255 "AF_INET6 address must be tuple, not %.500s",
1256 Py_TYPE(args)->tp_name);
1257 return 0;
1258 }
1259 if (!PyArg_ParseTuple(args, "eti|ii",
1260 "idna", &host, &port, &flowinfo,
1261 &scope_id)) {
1262 return 0;
1263 }
1264 addr = (struct sockaddr_in6*)addr_ret;
1265 result = setipaddr(host, (struct sockaddr *)addr,
1266 sizeof(*addr), AF_INET6);
1267 PyMem_Free(host);
1268 if (result < 0)
1269 return 0;
1270 if (port < 0 || port > 0xffff) {
1271 PyErr_SetString(
1272 PyExc_OverflowError,
1273 "getsockaddrarg: port must be 0-65535.");
1274 return 0;
1275 }
1276 addr->sin6_family = s->sock_family;
1277 addr->sin6_port = htons((short)port);
1278 addr->sin6_flowinfo = flowinfo;
1279 addr->sin6_scope_id = scope_id;
1280 *len_ret = sizeof *addr;
1281 return 1;
1282 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00001283#endif
1284
Hye-Shik Chang81268602004-02-02 06:05:24 +00001285#ifdef USE_BLUETOOTH
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001286 case AF_BLUETOOTH:
1287 {
1288 switch (s->sock_proto) {
1289 case BTPROTO_L2CAP:
1290 {
1291 struct sockaddr_l2 *addr;
1292 char *straddr;
Martin v. Löwis12af0482004-01-31 12:34:17 +00001293
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001294 addr = (struct sockaddr_l2 *)addr_ret;
1295 memset(addr, 0, sizeof(struct sockaddr_l2));
1296 _BT_L2_MEMB(addr, family) = AF_BLUETOOTH;
1297 if (!PyArg_ParseTuple(args, "si", &straddr,
1298 &_BT_L2_MEMB(addr, psm))) {
1299 PyErr_SetString(socket_error, "getsockaddrarg: "
1300 "wrong format");
1301 return 0;
1302 }
1303 if (setbdaddr(straddr, &_BT_L2_MEMB(addr, bdaddr)) < 0)
1304 return 0;
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001305
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001306 *len_ret = sizeof *addr;
1307 return 1;
1308 }
1309 case BTPROTO_RFCOMM:
1310 {
1311 struct sockaddr_rc *addr;
1312 char *straddr;
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001313
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001314 addr = (struct sockaddr_rc *)addr_ret;
1315 _BT_RC_MEMB(addr, family) = AF_BLUETOOTH;
1316 if (!PyArg_ParseTuple(args, "si", &straddr,
1317 &_BT_RC_MEMB(addr, channel))) {
1318 PyErr_SetString(socket_error, "getsockaddrarg: "
1319 "wrong format");
1320 return 0;
1321 }
1322 if (setbdaddr(straddr, &_BT_RC_MEMB(addr, bdaddr)) < 0)
1323 return 0;
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001324
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001325 *len_ret = sizeof *addr;
1326 return 1;
1327 }
1328 case BTPROTO_HCI:
1329 {
1330 struct sockaddr_hci *addr = (struct sockaddr_hci *)addr_ret;
1331 _BT_HCI_MEMB(addr, family) = AF_BLUETOOTH;
1332 if (!PyArg_ParseTuple(args, "i", &_BT_HCI_MEMB(addr, dev))) {
1333 PyErr_SetString(socket_error, "getsockaddrarg: "
1334 "wrong format");
1335 return 0;
1336 }
1337 *len_ret = sizeof *addr;
1338 return 1;
1339 }
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001340#if !defined(__FreeBSD__)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001341 case BTPROTO_SCO:
1342 {
1343 struct sockaddr_sco *addr;
1344 char *straddr;
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001345
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001346 addr = (struct sockaddr_sco *)addr_ret;
1347 _BT_SCO_MEMB(addr, family) = AF_BLUETOOTH;
1348 if (!PyBytes_Check(args)) {
1349 PyErr_SetString(socket_error, "getsockaddrarg: "
1350 "wrong format");
1351 return 0;
1352 }
1353 straddr = PyBytes_AS_STRING(args);
1354 if (setbdaddr(straddr, &_BT_SCO_MEMB(addr, bdaddr)) < 0)
1355 return 0;
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001356
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001357 *len_ret = sizeof *addr;
1358 return 1;
1359 }
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001360#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001361 default:
1362 PyErr_SetString(socket_error, "getsockaddrarg: unknown Bluetooth protocol");
1363 return 0;
1364 }
1365 }
Martin v. Löwis12af0482004-01-31 12:34:17 +00001366#endif
1367
Martin v. Löwis1ba3fd52001-08-10 20:29:40 +00001368#ifdef HAVE_NETPACKET_PACKET_H
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001369 case AF_PACKET:
1370 {
1371 struct sockaddr_ll* addr;
1372 struct ifreq ifr;
1373 char *interfaceName;
1374 int protoNumber;
1375 int hatype = 0;
1376 int pkttype = 0;
1377 char *haddr = NULL;
1378 unsigned int halen = 0;
Guido van Rossum48a680c2001-03-02 06:34:14 +00001379
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001380 if (!PyTuple_Check(args)) {
1381 PyErr_Format(
1382 PyExc_TypeError,
1383 "getsockaddrarg: "
1384 "AF_PACKET address must be tuple, not %.500s",
1385 Py_TYPE(args)->tp_name);
1386 return 0;
1387 }
1388 if (!PyArg_ParseTuple(args, "si|iiy#", &interfaceName,
1389 &protoNumber, &pkttype, &hatype,
1390 &haddr, &halen))
1391 return 0;
1392 strncpy(ifr.ifr_name, interfaceName, sizeof(ifr.ifr_name));
1393 ifr.ifr_name[(sizeof(ifr.ifr_name))-1] = '\0';
1394 if (ioctl(s->sock_fd, SIOCGIFINDEX, &ifr) < 0) {
1395 s->errorhandler();
1396 return 0;
1397 }
1398 if (halen > 8) {
1399 PyErr_SetString(PyExc_ValueError,
1400 "Hardware address must be 8 bytes or less");
1401 return 0;
1402 }
1403 if (protoNumber < 0 || protoNumber > 0xffff) {
1404 PyErr_SetString(
1405 PyExc_OverflowError,
1406 "getsockaddrarg: protoNumber must be 0-65535.");
1407 return 0;
1408 }
1409 addr = (struct sockaddr_ll*)addr_ret;
1410 addr->sll_family = AF_PACKET;
1411 addr->sll_protocol = htons((short)protoNumber);
1412 addr->sll_ifindex = ifr.ifr_ifindex;
1413 addr->sll_pkttype = pkttype;
1414 addr->sll_hatype = hatype;
1415 if (halen != 0) {
1416 memcpy(&addr->sll_addr, haddr, halen);
1417 }
1418 addr->sll_halen = halen;
1419 *len_ret = sizeof *addr;
1420 return 1;
1421 }
Guido van Rossum48a680c2001-03-02 06:34:14 +00001422#endif
1423
Christian Heimes043d6f62008-01-07 17:19:16 +00001424#ifdef HAVE_LINUX_TIPC_H
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001425 case AF_TIPC:
1426 {
1427 unsigned int atype, v1, v2, v3;
1428 unsigned int scope = TIPC_CLUSTER_SCOPE;
1429 struct sockaddr_tipc *addr;
Christian Heimes043d6f62008-01-07 17:19:16 +00001430
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001431 if (!PyTuple_Check(args)) {
1432 PyErr_Format(
1433 PyExc_TypeError,
1434 "getsockaddrarg: "
1435 "AF_TIPC address must be tuple, not %.500s",
1436 Py_TYPE(args)->tp_name);
1437 return 0;
1438 }
Christian Heimes043d6f62008-01-07 17:19:16 +00001439
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001440 if (!PyArg_ParseTuple(args,
1441 "IIII|I;Invalid TIPC address format",
1442 &atype, &v1, &v2, &v3, &scope))
1443 return 0;
Christian Heimes043d6f62008-01-07 17:19:16 +00001444
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001445 addr = (struct sockaddr_tipc *) addr_ret;
1446 memset(addr, 0, sizeof(struct sockaddr_tipc));
Christian Heimes043d6f62008-01-07 17:19:16 +00001447
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001448 addr->family = AF_TIPC;
1449 addr->scope = scope;
1450 addr->addrtype = atype;
Christian Heimes043d6f62008-01-07 17:19:16 +00001451
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001452 if (atype == TIPC_ADDR_NAMESEQ) {
1453 addr->addr.nameseq.type = v1;
1454 addr->addr.nameseq.lower = v2;
1455 addr->addr.nameseq.upper = v3;
1456 } else if (atype == TIPC_ADDR_NAME) {
1457 addr->addr.name.name.type = v1;
1458 addr->addr.name.name.instance = v2;
1459 } else if (atype == TIPC_ADDR_ID) {
1460 addr->addr.id.node = v1;
1461 addr->addr.id.ref = v2;
1462 } else {
1463 /* Shouldn't happen */
1464 PyErr_SetString(PyExc_TypeError, "Invalid address type");
1465 return 0;
1466 }
Christian Heimes043d6f62008-01-07 17:19:16 +00001467
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001468 *len_ret = sizeof(*addr);
Christian Heimes043d6f62008-01-07 17:19:16 +00001469
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001470 return 1;
1471 }
Christian Heimes043d6f62008-01-07 17:19:16 +00001472#endif
1473
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001474 /* More cases here... */
Guido van Rossum30a685f1991-06-27 15:51:29 +00001475
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001476 default:
1477 PyErr_SetString(socket_error, "getsockaddrarg: bad family");
1478 return 0;
Guido van Rossum30a685f1991-06-27 15:51:29 +00001479
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001480 }
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001481}
1482
Guido van Rossum30a685f1991-06-27 15:51:29 +00001483
Guido van Rossum48a680c2001-03-02 06:34:14 +00001484/* Get the address length according to the socket object's address family.
Guido van Rossum710e1df1992-06-12 10:39:36 +00001485 Return 1 if the family is known, 0 otherwise. The length is returned
1486 through len_ret. */
1487
1488static int
Peter Schneider-Kamp618e25d2000-07-11 23:00:12 +00001489getsockaddrlen(PySocketSockObject *s, socklen_t *len_ret)
Guido van Rossum710e1df1992-06-12 10:39:36 +00001490{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001491 switch (s->sock_family) {
Guido van Rossum710e1df1992-06-12 10:39:36 +00001492
Andrew MacIntyred12dfbb2004-04-04 07:13:49 +00001493#if defined(AF_UNIX)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001494 case AF_UNIX:
1495 {
1496 *len_ret = sizeof (struct sockaddr_un);
1497 return 1;
1498 }
Guido van Rossumb6775db1994-08-01 11:34:53 +00001499#endif /* AF_UNIX */
Martin v. Löwis11017b12006-01-14 18:12:57 +00001500#if defined(AF_NETLINK)
1501 case AF_NETLINK:
1502 {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001503 *len_ret = sizeof (struct sockaddr_nl);
1504 return 1;
Martin v. Löwis11017b12006-01-14 18:12:57 +00001505 }
1506#endif
Guido van Rossum710e1df1992-06-12 10:39:36 +00001507
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001508 case AF_INET:
1509 {
1510 *len_ret = sizeof (struct sockaddr_in);
1511 return 1;
1512 }
Guido van Rossum710e1df1992-06-12 10:39:36 +00001513
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00001514#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001515 case AF_INET6:
1516 {
1517 *len_ret = sizeof (struct sockaddr_in6);
1518 return 1;
1519 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00001520#endif
1521
Hye-Shik Chang81268602004-02-02 06:05:24 +00001522#ifdef USE_BLUETOOTH
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001523 case AF_BLUETOOTH:
1524 {
1525 switch(s->sock_proto)
1526 {
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001527
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001528 case BTPROTO_L2CAP:
1529 *len_ret = sizeof (struct sockaddr_l2);
1530 return 1;
1531 case BTPROTO_RFCOMM:
1532 *len_ret = sizeof (struct sockaddr_rc);
1533 return 1;
1534 case BTPROTO_HCI:
1535 *len_ret = sizeof (struct sockaddr_hci);
1536 return 1;
Hye-Shik Chang81268602004-02-02 06:05:24 +00001537#if !defined(__FreeBSD__)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001538 case BTPROTO_SCO:
1539 *len_ret = sizeof (struct sockaddr_sco);
1540 return 1;
Hye-Shik Chang81268602004-02-02 06:05:24 +00001541#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001542 default:
1543 PyErr_SetString(socket_error, "getsockaddrlen: "
1544 "unknown BT protocol");
1545 return 0;
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001546
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001547 }
1548 }
Martin v. Löwis12af0482004-01-31 12:34:17 +00001549#endif
1550
Martin v. Löwis1ba3fd52001-08-10 20:29:40 +00001551#ifdef HAVE_NETPACKET_PACKET_H
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001552 case AF_PACKET:
1553 {
1554 *len_ret = sizeof (struct sockaddr_ll);
1555 return 1;
1556 }
Jeremy Hylton22308652001-02-02 03:23:09 +00001557#endif
Guido van Rossum48a680c2001-03-02 06:34:14 +00001558
Christian Heimes043d6f62008-01-07 17:19:16 +00001559#ifdef HAVE_LINUX_TIPC_H
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001560 case AF_TIPC:
1561 {
1562 *len_ret = sizeof (struct sockaddr_tipc);
1563 return 1;
1564 }
Christian Heimes043d6f62008-01-07 17:19:16 +00001565#endif
1566
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001567 /* More cases here... */
Guido van Rossum710e1df1992-06-12 10:39:36 +00001568
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001569 default:
1570 PyErr_SetString(socket_error, "getsockaddrlen: bad family");
1571 return 0;
Guido van Rossum710e1df1992-06-12 10:39:36 +00001572
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001573 }
Guido van Rossum710e1df1992-06-12 10:39:36 +00001574}
1575
1576
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00001577/* s._accept() -> (fd, address) */
Guido van Rossum30a685f1991-06-27 15:51:29 +00001578
Guido van Rossum73624e91994-10-10 17:59:00 +00001579static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001580sock_accept(PySocketSockObject *s)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001581{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001582 sock_addr_t addrbuf;
1583 SOCKET_T newfd = INVALID_SOCKET;
1584 socklen_t addrlen;
1585 PyObject *sock = NULL;
1586 PyObject *addr = NULL;
1587 PyObject *res = NULL;
1588 int timeout;
Barry Warsaw752300b1997-01-03 17:18:10 +00001589
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001590 if (!getsockaddrlen(s, &addrlen))
1591 return NULL;
1592 memset(&addrbuf, 0, addrlen);
Guido van Rossum67f7a382002-06-06 21:08:16 +00001593
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001594 if (!IS_SELECTABLE(s))
1595 return select_error();
Neal Norwitz082b2df2006-02-07 07:04:46 +00001596
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001597 Py_BEGIN_ALLOW_THREADS
1598 timeout = internal_select(s, 0);
1599 if (!timeout)
1600 newfd = accept(s->sock_fd, SAS2SA(&addrbuf), &addrlen);
1601 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00001602
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001603 if (timeout == 1) {
1604 PyErr_SetString(socket_timeout, "timed out");
1605 return NULL;
1606 }
Raymond Hettingeref7343c2003-06-29 03:08:05 +00001607
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001608 if (newfd == INVALID_SOCKET)
1609 return s->errorhandler();
Barry Warsaw752300b1997-01-03 17:18:10 +00001610
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001611 sock = PyLong_FromSocket_t(newfd);
1612 if (sock == NULL) {
1613 SOCKETCLOSE(newfd);
1614 goto finally;
1615 }
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00001616
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001617 addr = makesockaddr(s->sock_fd, SAS2SA(&addrbuf),
1618 addrlen, s->sock_proto);
1619 if (addr == NULL)
1620 goto finally;
Barry Warsaw752300b1997-01-03 17:18:10 +00001621
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001622 res = PyTuple_Pack(2, sock, addr);
Barry Warsaw752300b1997-01-03 17:18:10 +00001623
Guido van Rossum67f7a382002-06-06 21:08:16 +00001624finally:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001625 Py_XDECREF(sock);
1626 Py_XDECREF(addr);
1627 return res;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001628}
1629
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001630PyDoc_STRVAR(accept_doc,
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00001631"_accept() -> (integer, address info)\n\
Guido van Rossum82a5c661998-07-07 20:45:43 +00001632\n\
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00001633Wait for an incoming connection. Return a new socket file descriptor\n\
1634representing the connection, and the address of the client.\n\
1635For IP sockets, the address info is a pair (hostaddr, port).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001636
Guido van Rossum11ba0942002-06-13 15:07:44 +00001637/* s.setblocking(flag) method. Argument:
1638 False -- non-blocking mode; same as settimeout(0)
1639 True -- blocking mode; same as settimeout(None)
1640*/
Guido van Rossume4485b01994-09-07 14:32:49 +00001641
Guido van Rossum73624e91994-10-10 17:59:00 +00001642static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001643sock_setblocking(PySocketSockObject *s, PyObject *arg)
Guido van Rossume4485b01994-09-07 14:32:49 +00001644{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001645 int block;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001646
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001647 block = PyLong_AsLong(arg);
1648 if (block == -1 && PyErr_Occurred())
1649 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001650
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001651 s->sock_timeout = block ? -1.0 : 0.0;
1652 internal_setblocking(s, block);
Guido van Rossume4485b01994-09-07 14:32:49 +00001653
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001654 Py_INCREF(Py_None);
1655 return Py_None;
Guido van Rossume4485b01994-09-07 14:32:49 +00001656}
Guido van Rossume4485b01994-09-07 14:32:49 +00001657
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001658PyDoc_STRVAR(setblocking_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001659"setblocking(flag)\n\
1660\n\
1661Set the socket to blocking (flag is true) or non-blocking (false).\n\
Guido van Rossum11ba0942002-06-13 15:07:44 +00001662setblocking(True) is equivalent to settimeout(None);\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001663setblocking(False) is equivalent to settimeout(0.0).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001664
Guido van Rossum11ba0942002-06-13 15:07:44 +00001665/* s.settimeout(timeout) method. Argument:
1666 None -- no timeout, blocking mode; same as setblocking(True)
1667 0.0 -- non-blocking mode; same as setblocking(False)
1668 > 0 -- timeout mode; operations time out after timeout seconds
1669 < 0 -- illegal; raises an exception
1670*/
Guido van Rossum67f7a382002-06-06 21:08:16 +00001671static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001672sock_settimeout(PySocketSockObject *s, PyObject *arg)
Guido van Rossum67f7a382002-06-06 21:08:16 +00001673{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001674 double timeout;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001675
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001676 if (arg == Py_None)
1677 timeout = -1.0;
1678 else {
1679 timeout = PyFloat_AsDouble(arg);
1680 if (timeout < 0.0) {
1681 if (!PyErr_Occurred())
1682 PyErr_SetString(PyExc_ValueError,
1683 "Timeout value out of range");
1684 return NULL;
1685 }
1686 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00001687
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001688 s->sock_timeout = timeout;
1689 internal_setblocking(s, timeout < 0.0);
Guido van Rossum67f7a382002-06-06 21:08:16 +00001690
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001691 Py_INCREF(Py_None);
1692 return Py_None;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001693}
1694
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001695PyDoc_STRVAR(settimeout_doc,
Guido van Rossum3eede5a2002-06-07 02:08:35 +00001696"settimeout(timeout)\n\
Guido van Rossum67f7a382002-06-06 21:08:16 +00001697\n\
Guido van Rossum11ba0942002-06-13 15:07:44 +00001698Set a timeout on socket operations. 'timeout' can be a float,\n\
1699giving in seconds, or None. Setting a timeout of None disables\n\
1700the timeout feature and is equivalent to setblocking(1).\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001701Setting a timeout of zero is the same as setblocking(0).");
Guido van Rossum67f7a382002-06-06 21:08:16 +00001702
Guido van Rossum3eede5a2002-06-07 02:08:35 +00001703/* s.gettimeout() method.
1704 Returns the timeout associated with a socket. */
Guido van Rossum67f7a382002-06-06 21:08:16 +00001705static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001706sock_gettimeout(PySocketSockObject *s)
Guido van Rossum67f7a382002-06-06 21:08:16 +00001707{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001708 if (s->sock_timeout < 0.0) {
1709 Py_INCREF(Py_None);
1710 return Py_None;
1711 }
1712 else
1713 return PyFloat_FromDouble(s->sock_timeout);
Guido van Rossum67f7a382002-06-06 21:08:16 +00001714}
1715
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001716PyDoc_STRVAR(gettimeout_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00001717"gettimeout() -> timeout\n\
Guido van Rossum67f7a382002-06-06 21:08:16 +00001718\n\
1719Returns the timeout in floating seconds associated with socket \n\
1720operations. A timeout of None indicates that timeouts on socket \n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001721operations are disabled.");
Guido van Rossume4485b01994-09-07 14:32:49 +00001722
Guido van Rossumaee08791992-09-08 09:05:33 +00001723/* s.setsockopt() method.
1724 With an integer third argument, sets an integer option.
1725 With a string third argument, sets an option from a buffer;
1726 use optional built-in module 'struct' to encode the string. */
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001727
Guido van Rossum73624e91994-10-10 17:59:00 +00001728static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001729sock_setsockopt(PySocketSockObject *s, PyObject *args)
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001730{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001731 int level;
1732 int optname;
1733 int res;
1734 char *buf;
1735 int buflen;
1736 int flag;
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001737
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001738 if (PyArg_ParseTuple(args, "iii:setsockopt",
1739 &level, &optname, &flag)) {
1740 buf = (char *) &flag;
1741 buflen = sizeof flag;
1742 }
1743 else {
1744 PyErr_Clear();
1745 if (!PyArg_ParseTuple(args, "iiy#:setsockopt",
1746 &level, &optname, &buf, &buflen))
1747 return NULL;
1748 }
1749 res = setsockopt(s->sock_fd, level, optname, (void *)buf, buflen);
1750 if (res < 0)
1751 return s->errorhandler();
1752 Py_INCREF(Py_None);
1753 return Py_None;
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001754}
1755
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001756PyDoc_STRVAR(setsockopt_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001757"setsockopt(level, option, value)\n\
1758\n\
1759Set a socket option. See the Unix manual for level and option.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001760The value argument can either be an integer or a string.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001761
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001762
Guido van Rossumaee08791992-09-08 09:05:33 +00001763/* s.getsockopt() method.
1764 With two arguments, retrieves an integer option.
1765 With a third integer argument, retrieves a string buffer of that size;
1766 use optional built-in module 'struct' to decode the string. */
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001767
Guido van Rossum73624e91994-10-10 17:59:00 +00001768static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001769sock_getsockopt(PySocketSockObject *s, PyObject *args)
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001770{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001771 int level;
1772 int optname;
1773 int res;
1774 PyObject *buf;
1775 socklen_t buflen = 0;
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001776
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001777 if (!PyArg_ParseTuple(args, "ii|i:getsockopt",
1778 &level, &optname, &buflen))
1779 return NULL;
Guido van Rossum48a680c2001-03-02 06:34:14 +00001780
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001781 if (buflen == 0) {
1782 int flag = 0;
1783 socklen_t flagsize = sizeof flag;
1784 res = getsockopt(s->sock_fd, level, optname,
1785 (void *)&flag, &flagsize);
1786 if (res < 0)
1787 return s->errorhandler();
1788 return PyLong_FromLong(flag);
1789 }
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001790#ifdef __VMS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001791 /* socklen_t is unsigned so no negative test is needed,
1792 test buflen == 0 is previously done */
1793 if (buflen > 1024) {
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001794#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001795 if (buflen <= 0 || buflen > 1024) {
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001796#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001797 PyErr_SetString(socket_error,
1798 "getsockopt buflen out of range");
1799 return NULL;
1800 }
1801 buf = PyBytes_FromStringAndSize((char *)NULL, buflen);
1802 if (buf == NULL)
1803 return NULL;
1804 res = getsockopt(s->sock_fd, level, optname,
1805 (void *)PyBytes_AS_STRING(buf), &buflen);
1806 if (res < 0) {
1807 Py_DECREF(buf);
1808 return s->errorhandler();
1809 }
1810 _PyBytes_Resize(&buf, buflen);
1811 return buf;
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001812}
1813
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001814PyDoc_STRVAR(getsockopt_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001815"getsockopt(level, option[, buffersize]) -> value\n\
1816\n\
1817Get a socket option. See the Unix manual for level and option.\n\
1818If a nonzero buffersize argument is given, the return value is a\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001819string of that length; otherwise it is an integer.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001820
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001821
Fred Drake728819a2000-07-01 03:40:12 +00001822/* s.bind(sockaddr) method */
Guido van Rossum30a685f1991-06-27 15:51:29 +00001823
Guido van Rossum73624e91994-10-10 17:59:00 +00001824static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001825sock_bind(PySocketSockObject *s, PyObject *addro)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001826{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001827 sock_addr_t addrbuf;
1828 int addrlen;
1829 int res;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00001830
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001831 if (!getsockaddrarg(s, addro, SAS2SA(&addrbuf), &addrlen))
1832 return NULL;
1833 Py_BEGIN_ALLOW_THREADS
1834 res = bind(s->sock_fd, SAS2SA(&addrbuf), addrlen);
1835 Py_END_ALLOW_THREADS
1836 if (res < 0)
1837 return s->errorhandler();
1838 Py_INCREF(Py_None);
1839 return Py_None;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001840}
1841
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001842PyDoc_STRVAR(bind_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001843"bind(address)\n\
1844\n\
1845Bind the socket to a local address. For IP sockets, the address is a\n\
Jeremy Hylton22308652001-02-02 03:23:09 +00001846pair (host, port); the host must refer to the local host. For raw packet\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001847sockets the address is a tuple (ifname, proto [,pkttype [,hatype]])");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001848
Guido van Rossum30a685f1991-06-27 15:51:29 +00001849
1850/* s.close() method.
1851 Set the file descriptor to -1 so operations tried subsequently
1852 will surely fail. */
1853
Guido van Rossum73624e91994-10-10 17:59:00 +00001854static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001855sock_close(PySocketSockObject *s)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001856{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001857 SOCKET_T fd;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00001858
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001859 if ((fd = s->sock_fd) != -1) {
1860 s->sock_fd = -1;
1861 Py_BEGIN_ALLOW_THREADS
1862 (void) SOCKETCLOSE(fd);
1863 Py_END_ALLOW_THREADS
1864 }
1865 Py_INCREF(Py_None);
1866 return Py_None;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001867}
1868
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001869PyDoc_STRVAR(close_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001870"close()\n\
1871\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001872Close the socket. It cannot be used after this call.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001873
Antoine Pitroue43f9d02010-08-08 23:24:50 +00001874static PyObject *
Antoine Pitrou6e451df2010-08-09 20:39:54 +00001875sock_detach(PySocketSockObject *s)
Antoine Pitroue43f9d02010-08-08 23:24:50 +00001876{
Antoine Pitrou6e451df2010-08-09 20:39:54 +00001877 SOCKET_T fd = s->sock_fd;
Antoine Pitroue43f9d02010-08-08 23:24:50 +00001878 s->sock_fd = -1;
Antoine Pitrou6e451df2010-08-09 20:39:54 +00001879 return PyLong_FromSocket_t(fd);
Antoine Pitroue43f9d02010-08-08 23:24:50 +00001880}
1881
Antoine Pitrou6e451df2010-08-09 20:39:54 +00001882PyDoc_STRVAR(detach_doc,
1883"detach()\n\
Antoine Pitroue43f9d02010-08-08 23:24:50 +00001884\n\
1885Close the socket object without closing the underlying file descriptor.\
1886The object cannot be used after this call, but the file descriptor\
Antoine Pitrou6e451df2010-08-09 20:39:54 +00001887can be reused for other purposes. The file descriptor is returned.");
Antoine Pitroue43f9d02010-08-08 23:24:50 +00001888
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001889static int
Raymond Hettingeref7343c2003-06-29 03:08:05 +00001890internal_connect(PySocketSockObject *s, struct sockaddr *addr, int addrlen,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001891 int *timeoutp)
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001892{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001893 int res, timeout;
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001894
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001895 timeout = 0;
1896 res = connect(s->sock_fd, addr, addrlen);
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001897
1898#ifdef MS_WINDOWS
1899
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001900 if (s->sock_timeout > 0.0) {
1901 if (res < 0 && WSAGetLastError() == WSAEWOULDBLOCK &&
1902 IS_SELECTABLE(s)) {
1903 /* This is a mess. Best solution: trust select */
1904 fd_set fds;
1905 fd_set fds_exc;
1906 struct timeval tv;
1907 tv.tv_sec = (int)s->sock_timeout;
1908 tv.tv_usec = (int)((s->sock_timeout - tv.tv_sec) * 1e6);
1909 FD_ZERO(&fds);
1910 FD_SET(s->sock_fd, &fds);
1911 FD_ZERO(&fds_exc);
1912 FD_SET(s->sock_fd, &fds_exc);
Antoine Pitrou19467d22010-08-17 19:33:30 +00001913 res = select(Py_SAFE_DOWNCAST(s->sock_fd+1, SOCKET_T, int),
1914 NULL, &fds, &fds_exc, &tv);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001915 if (res == 0) {
1916 res = WSAEWOULDBLOCK;
1917 timeout = 1;
1918 } else if (res > 0) {
1919 if (FD_ISSET(s->sock_fd, &fds))
1920 /* The socket is in the writable set - this
1921 means connected */
1922 res = 0;
1923 else {
1924 /* As per MS docs, we need to call getsockopt()
1925 to get the underlying error */
1926 int res_size = sizeof res;
1927 /* It must be in the exception set */
1928 assert(FD_ISSET(s->sock_fd, &fds_exc));
1929 if (0 == getsockopt(s->sock_fd, SOL_SOCKET, SO_ERROR,
1930 (char *)&res, &res_size))
1931 /* getsockopt also clears WSAGetLastError,
1932 so reset it back. */
1933 WSASetLastError(res);
1934 else
1935 res = WSAGetLastError();
1936 }
1937 }
1938 /* else if (res < 0) an error occurred */
1939 }
1940 }
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001941
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001942 if (res < 0)
1943 res = WSAGetLastError();
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001944
1945#else
1946
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001947 if (s->sock_timeout > 0.0) {
1948 if (res < 0 && errno == EINPROGRESS && IS_SELECTABLE(s)) {
1949 timeout = internal_select(s, 1);
1950 if (timeout == 0) {
1951 /* Bug #1019808: in case of an EINPROGRESS,
1952 use getsockopt(SO_ERROR) to get the real
1953 error. */
1954 socklen_t res_size = sizeof res;
1955 (void)getsockopt(s->sock_fd, SOL_SOCKET,
1956 SO_ERROR, &res, &res_size);
1957 if (res == EISCONN)
1958 res = 0;
1959 errno = res;
1960 }
1961 else if (timeout == -1) {
1962 res = errno; /* had error */
1963 }
1964 else
1965 res = EWOULDBLOCK; /* timed out */
1966 }
1967 }
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001968
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001969 if (res < 0)
1970 res = errno;
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001971
1972#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001973 *timeoutp = timeout;
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001974
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001975 return res;
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001976}
Guido van Rossum30a685f1991-06-27 15:51:29 +00001977
Fred Drake728819a2000-07-01 03:40:12 +00001978/* s.connect(sockaddr) method */
Guido van Rossum30a685f1991-06-27 15:51:29 +00001979
Guido van Rossum73624e91994-10-10 17:59:00 +00001980static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001981sock_connect(PySocketSockObject *s, PyObject *addro)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001982{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001983 sock_addr_t addrbuf;
1984 int addrlen;
1985 int res;
1986 int timeout;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00001987
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001988 if (!getsockaddrarg(s, addro, SAS2SA(&addrbuf), &addrlen))
1989 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001990
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001991 Py_BEGIN_ALLOW_THREADS
1992 res = internal_connect(s, SAS2SA(&addrbuf), addrlen, &timeout);
1993 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00001994
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001995 if (timeout == 1) {
1996 PyErr_SetString(socket_timeout, "timed out");
1997 return NULL;
1998 }
1999 if (res != 0)
2000 return s->errorhandler();
2001 Py_INCREF(Py_None);
2002 return Py_None;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002003}
2004
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002005PyDoc_STRVAR(connect_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002006"connect(address)\n\
2007\n\
2008Connect the socket to a remote address. For IP sockets, the address\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002009is a pair (host, port).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002010
Guido van Rossum30a685f1991-06-27 15:51:29 +00002011
Fred Drake728819a2000-07-01 03:40:12 +00002012/* s.connect_ex(sockaddr) method */
Guido van Rossumfc4255d1997-11-19 18:57:13 +00002013
2014static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002015sock_connect_ex(PySocketSockObject *s, PyObject *addro)
Guido van Rossumfc4255d1997-11-19 18:57:13 +00002016{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002017 sock_addr_t addrbuf;
2018 int addrlen;
2019 int res;
2020 int timeout;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00002021
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002022 if (!getsockaddrarg(s, addro, SAS2SA(&addrbuf), &addrlen))
2023 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002024
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002025 Py_BEGIN_ALLOW_THREADS
2026 res = internal_connect(s, SAS2SA(&addrbuf), addrlen, &timeout);
2027 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00002028
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002029 /* Signals are not errors (though they may raise exceptions). Adapted
2030 from PyErr_SetFromErrnoWithFilenameObject(). */
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002031#ifdef EINTR
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002032 if (res == EINTR && PyErr_CheckSignals())
2033 return NULL;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002034#endif
2035
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002036 return PyLong_FromLong((long) res);
Guido van Rossumfc4255d1997-11-19 18:57:13 +00002037}
2038
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002039PyDoc_STRVAR(connect_ex_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00002040"connect_ex(address) -> errno\n\
Guido van Rossum82a5c661998-07-07 20:45:43 +00002041\n\
2042This is like connect(address), but returns an error code (the errno value)\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002043instead of raising an exception when an error occurs.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002044
Guido van Rossumfc4255d1997-11-19 18:57:13 +00002045
Guido van Rossumed233a51992-06-23 09:07:03 +00002046/* s.fileno() method */
2047
Guido van Rossum73624e91994-10-10 17:59:00 +00002048static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002049sock_fileno(PySocketSockObject *s)
Guido van Rossumed233a51992-06-23 09:07:03 +00002050{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002051 return PyLong_FromSocket_t(s->sock_fd);
Guido van Rossumed233a51992-06-23 09:07:03 +00002052}
2053
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002054PyDoc_STRVAR(fileno_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002055"fileno() -> integer\n\
2056\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002057Return the integer file descriptor of the socket.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002058
Guido van Rossumed233a51992-06-23 09:07:03 +00002059
Guido van Rossumc89705d1992-11-26 08:54:07 +00002060/* s.getsockname() method */
2061
Guido van Rossum73624e91994-10-10 17:59:00 +00002062static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002063sock_getsockname(PySocketSockObject *s)
Guido van Rossumc89705d1992-11-26 08:54:07 +00002064{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002065 sock_addr_t addrbuf;
2066 int res;
2067 socklen_t addrlen;
Guido van Rossumff3ab422000-04-24 15:16:03 +00002068
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002069 if (!getsockaddrlen(s, &addrlen))
2070 return NULL;
2071 memset(&addrbuf, 0, addrlen);
2072 Py_BEGIN_ALLOW_THREADS
2073 res = getsockname(s->sock_fd, SAS2SA(&addrbuf), &addrlen);
2074 Py_END_ALLOW_THREADS
2075 if (res < 0)
2076 return s->errorhandler();
2077 return makesockaddr(s->sock_fd, SAS2SA(&addrbuf), addrlen,
2078 s->sock_proto);
Guido van Rossumc89705d1992-11-26 08:54:07 +00002079}
2080
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002081PyDoc_STRVAR(getsockname_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002082"getsockname() -> address info\n\
2083\n\
2084Return the address of the local endpoint. For IP sockets, the address\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002085info is a pair (hostaddr, port).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002086
Guido van Rossumc89705d1992-11-26 08:54:07 +00002087
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002088#ifdef HAVE_GETPEERNAME /* Cray APP doesn't have this :-( */
Guido van Rossumc89705d1992-11-26 08:54:07 +00002089/* s.getpeername() method */
2090
Guido van Rossum73624e91994-10-10 17:59:00 +00002091static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002092sock_getpeername(PySocketSockObject *s)
Guido van Rossumc89705d1992-11-26 08:54:07 +00002093{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002094 sock_addr_t addrbuf;
2095 int res;
2096 socklen_t addrlen;
Guido van Rossumff3ab422000-04-24 15:16:03 +00002097
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002098 if (!getsockaddrlen(s, &addrlen))
2099 return NULL;
2100 memset(&addrbuf, 0, addrlen);
2101 Py_BEGIN_ALLOW_THREADS
2102 res = getpeername(s->sock_fd, SAS2SA(&addrbuf), &addrlen);
2103 Py_END_ALLOW_THREADS
2104 if (res < 0)
2105 return s->errorhandler();
2106 return makesockaddr(s->sock_fd, SAS2SA(&addrbuf), addrlen,
2107 s->sock_proto);
Guido van Rossumc89705d1992-11-26 08:54:07 +00002108}
Guido van Rossum82a5c661998-07-07 20:45:43 +00002109
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002110PyDoc_STRVAR(getpeername_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002111"getpeername() -> address info\n\
2112\n\
2113Return the address of the remote endpoint. For IP sockets, the address\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002114info is a pair (hostaddr, port).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002115
Guido van Rossumb6775db1994-08-01 11:34:53 +00002116#endif /* HAVE_GETPEERNAME */
Guido van Rossumc89705d1992-11-26 08:54:07 +00002117
2118
Guido van Rossum30a685f1991-06-27 15:51:29 +00002119/* s.listen(n) method */
2120
Guido van Rossum73624e91994-10-10 17:59:00 +00002121static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002122sock_listen(PySocketSockObject *s, PyObject *arg)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002123{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002124 int backlog;
2125 int res;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00002126
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002127 backlog = PyLong_AsLong(arg);
2128 if (backlog == -1 && PyErr_Occurred())
2129 return NULL;
2130 Py_BEGIN_ALLOW_THREADS
2131 if (backlog < 1)
2132 backlog = 1;
2133 res = listen(s->sock_fd, backlog);
2134 Py_END_ALLOW_THREADS
2135 if (res < 0)
2136 return s->errorhandler();
2137 Py_INCREF(Py_None);
2138 return Py_None;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002139}
2140
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002141PyDoc_STRVAR(listen_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002142"listen(backlog)\n\
2143\n\
2144Enable a server to accept connections. The backlog argument must be at\n\
2145least 1; it specifies the number of unaccepted connection that the system\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002146will allow before refusing new connections.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002147
2148
Thomas Wouters477c8d52006-05-27 19:21:47 +00002149/*
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002150 * This is the guts of the recv() and recv_into() methods, which reads into a
Thomas Wouters902d6eb2007-01-09 23:18:33 +00002151 * char buffer. If you have any inc/dec ref to do to the objects that contain
Thomas Wouters477c8d52006-05-27 19:21:47 +00002152 * the buffer, do it in the caller. This function returns the number of bytes
2153 * succesfully read. If there was an error, it returns -1. Note that it is
2154 * also possible that we return a number of bytes smaller than the request
2155 * bytes.
2156 */
Antoine Pitrou19467d22010-08-17 19:33:30 +00002157static Py_ssize_t
2158sock_recv_guts(PySocketSockObject *s, char* cbuf, Py_ssize_t len, int flags)
Thomas Wouters477c8d52006-05-27 19:21:47 +00002159{
Antoine Pitrou19467d22010-08-17 19:33:30 +00002160 Py_ssize_t outlen = -1;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002161 int timeout;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002162#ifdef __VMS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002163 int remaining;
2164 char *read_buf;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002165#endif
2166
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002167 if (!IS_SELECTABLE(s)) {
2168 select_error();
2169 return -1;
2170 }
2171 if (len == 0) {
2172 /* If 0 bytes were requested, do nothing. */
2173 return 0;
2174 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002175
2176#ifndef __VMS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002177 Py_BEGIN_ALLOW_THREADS
2178 timeout = internal_select(s, 0);
2179 if (!timeout)
2180 outlen = recv(s->sock_fd, cbuf, len, flags);
2181 Py_END_ALLOW_THREADS
Thomas Wouters477c8d52006-05-27 19:21:47 +00002182
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002183 if (timeout == 1) {
2184 PyErr_SetString(socket_timeout, "timed out");
2185 return -1;
2186 }
2187 if (outlen < 0) {
2188 /* Note: the call to errorhandler() ALWAYS indirectly returned
2189 NULL, so ignore its return value */
2190 s->errorhandler();
2191 return -1;
2192 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002193#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002194 read_buf = cbuf;
2195 remaining = len;
2196 while (remaining != 0) {
2197 unsigned int segment;
2198 int nread = -1;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002199
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002200 segment = remaining /SEGMENT_SIZE;
2201 if (segment != 0) {
2202 segment = SEGMENT_SIZE;
2203 }
2204 else {
2205 segment = remaining;
2206 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002207
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002208 Py_BEGIN_ALLOW_THREADS
2209 timeout = internal_select(s, 0);
2210 if (!timeout)
2211 nread = recv(s->sock_fd, read_buf, segment, flags);
2212 Py_END_ALLOW_THREADS
Thomas Wouters477c8d52006-05-27 19:21:47 +00002213
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002214 if (timeout == 1) {
2215 PyErr_SetString(socket_timeout, "timed out");
2216 return -1;
2217 }
2218 if (nread < 0) {
2219 s->errorhandler();
2220 return -1;
2221 }
2222 if (nread != remaining) {
2223 read_buf += nread;
2224 break;
2225 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002226
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002227 remaining -= segment;
2228 read_buf += segment;
2229 }
2230 outlen = read_buf - cbuf;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002231#endif /* !__VMS */
2232
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002233 return outlen;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002234}
2235
Guido van Rossum48a680c2001-03-02 06:34:14 +00002236
Guido van Rossumeb6b33a1993-05-25 09:38:27 +00002237/* s.recv(nbytes [,flags]) method */
Guido van Rossum30a685f1991-06-27 15:51:29 +00002238
Guido van Rossum73624e91994-10-10 17:59:00 +00002239static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002240sock_recv(PySocketSockObject *s, PyObject *args)
Guido van Rossum30a685f1991-06-27 15:51:29 +00002241{
Antoine Pitrou19467d22010-08-17 19:33:30 +00002242 Py_ssize_t recvlen, outlen;
2243 int flags = 0;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002244 PyObject *buf;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002245
Antoine Pitrou19467d22010-08-17 19:33:30 +00002246 if (!PyArg_ParseTuple(args, "n|i:recv", &recvlen, &flags))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002247 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002248
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002249 if (recvlen < 0) {
2250 PyErr_SetString(PyExc_ValueError,
2251 "negative buffersize in recv");
2252 return NULL;
2253 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00002254
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002255 /* Allocate a new string. */
2256 buf = PyBytes_FromStringAndSize((char *) 0, recvlen);
2257 if (buf == NULL)
2258 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002259
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002260 /* Call the guts */
2261 outlen = sock_recv_guts(s, PyBytes_AS_STRING(buf), recvlen, flags);
2262 if (outlen < 0) {
2263 /* An error occurred, release the string and return an
2264 error. */
2265 Py_DECREF(buf);
2266 return NULL;
2267 }
2268 if (outlen != recvlen) {
2269 /* We did not read as many bytes as we anticipated, resize the
2270 string if possible and be successful. */
2271 _PyBytes_Resize(&buf, outlen);
2272 }
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00002273
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002274 return buf;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002275}
2276
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002277PyDoc_STRVAR(recv_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002278"recv(buffersize[, flags]) -> data\n\
2279\n\
2280Receive up to buffersize bytes from the socket. For the optional flags\n\
2281argument, see the Unix manual. When no data is available, block until\n\
2282at least one byte is available or until the remote end is closed. When\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002283the remote end is closed and all data is read, return the empty string.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002284
Guido van Rossum30a685f1991-06-27 15:51:29 +00002285
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002286/* s.recv_into(buffer, [nbytes [,flags]]) method */
Guido van Rossum30a685f1991-06-27 15:51:29 +00002287
Thomas Wouters477c8d52006-05-27 19:21:47 +00002288static PyObject*
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002289sock_recv_into(PySocketSockObject *s, PyObject *args, PyObject *kwds)
Thomas Wouters477c8d52006-05-27 19:21:47 +00002290{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002291 static char *kwlist[] = {"buffer", "nbytes", "flags", 0};
Thomas Wouters477c8d52006-05-27 19:21:47 +00002292
Antoine Pitrou19467d22010-08-17 19:33:30 +00002293 int flags = 0;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002294 Py_buffer pbuf;
2295 char *buf;
Antoine Pitrou19467d22010-08-17 19:33:30 +00002296 Py_ssize_t buflen, readlen, recvlen = 0;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002297
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002298 /* Get the buffer's memory */
Antoine Pitrou19467d22010-08-17 19:33:30 +00002299 if (!PyArg_ParseTupleAndKeywords(args, kwds, "w*|ni:recv_into", kwlist,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002300 &pbuf, &recvlen, &flags))
2301 return NULL;
2302 buf = pbuf.buf;
2303 buflen = pbuf.len;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002304
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002305 if (recvlen < 0) {
2306 PyBuffer_Release(&pbuf);
2307 PyErr_SetString(PyExc_ValueError,
2308 "negative buffersize in recv_into");
2309 return NULL;
2310 }
2311 if (recvlen == 0) {
2312 /* If nbytes was not specified, use the buffer's length */
2313 recvlen = buflen;
2314 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002315
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002316 /* Check if the buffer is large enough */
2317 if (buflen < recvlen) {
2318 PyBuffer_Release(&pbuf);
2319 PyErr_SetString(PyExc_ValueError,
2320 "buffer too small for requested bytes");
2321 return NULL;
2322 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002323
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002324 /* Call the guts */
2325 readlen = sock_recv_guts(s, buf, recvlen, flags);
2326 if (readlen < 0) {
2327 /* Return an error. */
2328 PyBuffer_Release(&pbuf);
2329 return NULL;
2330 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002331
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002332 PyBuffer_Release(&pbuf);
2333 /* Return the number of bytes read. Note that we do not do anything
2334 special here in the case that readlen < recvlen. */
2335 return PyLong_FromSsize_t(readlen);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002336}
2337
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002338PyDoc_STRVAR(recv_into_doc,
2339"recv_into(buffer, [nbytes[, flags]]) -> nbytes_read\n\
Thomas Wouters477c8d52006-05-27 19:21:47 +00002340\n\
2341A version of recv() that stores its data into a buffer rather than creating \n\
2342a new string. Receive up to buffersize bytes from the socket. If buffersize \n\
2343is not specified (or 0), receive up to the size available in the given buffer.\n\
2344\n\
2345See recv() for documentation about the flags.");
2346
2347
2348/*
Christian Heimes99170a52007-12-19 02:07:34 +00002349 * This is the guts of the recvfrom() and recvfrom_into() methods, which reads
2350 * into a char buffer. If you have any inc/def ref to do to the objects that
2351 * contain the buffer, do it in the caller. This function returns the number
2352 * of bytes succesfully read. If there was an error, it returns -1. Note
2353 * that it is also possible that we return a number of bytes smaller than the
2354 * request bytes.
Thomas Wouters477c8d52006-05-27 19:21:47 +00002355 *
2356 * 'addr' is a return value for the address object. Note that you must decref
2357 * it yourself.
2358 */
Antoine Pitrou19467d22010-08-17 19:33:30 +00002359static Py_ssize_t
2360sock_recvfrom_guts(PySocketSockObject *s, char* cbuf, Py_ssize_t len, int flags,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002361 PyObject** addr)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002362{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002363 sock_addr_t addrbuf;
2364 int timeout;
Antoine Pitrou19467d22010-08-17 19:33:30 +00002365 Py_ssize_t n = -1;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002366 socklen_t addrlen;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002367
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002368 *addr = NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002369
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002370 if (!getsockaddrlen(s, &addrlen))
2371 return -1;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002372
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002373 if (!IS_SELECTABLE(s)) {
2374 select_error();
2375 return -1;
2376 }
Neal Norwitz082b2df2006-02-07 07:04:46 +00002377
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002378 Py_BEGIN_ALLOW_THREADS
2379 memset(&addrbuf, 0, addrlen);
2380 timeout = internal_select(s, 0);
2381 if (!timeout) {
Guido van Rossum8d665e61996-06-26 18:22:49 +00002382#ifndef MS_WINDOWS
Andrew MacIntyreba43e872002-03-03 03:03:52 +00002383#if defined(PYOS_OS2) && !defined(PYCC_GCC)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002384 n = recvfrom(s->sock_fd, cbuf, len, flags,
2385 SAS2SA(&addrbuf), &addrlen);
Guido van Rossum32c575d1997-12-02 20:37:32 +00002386#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002387 n = recvfrom(s->sock_fd, cbuf, len, flags,
2388 (void *) &addrbuf, &addrlen);
Guido van Rossum32c575d1997-12-02 20:37:32 +00002389#endif
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00002390#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002391 n = recvfrom(s->sock_fd, cbuf, len, flags,
2392 SAS2SA(&addrbuf), &addrlen);
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00002393#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002394 }
2395 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00002396
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002397 if (timeout == 1) {
2398 PyErr_SetString(socket_timeout, "timed out");
2399 return -1;
2400 }
2401 if (n < 0) {
2402 s->errorhandler();
2403 return -1;
2404 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00002405
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002406 if (!(*addr = makesockaddr(s->sock_fd, SAS2SA(&addrbuf),
2407 addrlen, s->sock_proto)))
2408 return -1;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002409
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002410 return n;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002411}
2412
2413/* s.recvfrom(nbytes [,flags]) method */
2414
2415static PyObject *
2416sock_recvfrom(PySocketSockObject *s, PyObject *args)
2417{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002418 PyObject *buf = NULL;
2419 PyObject *addr = NULL;
2420 PyObject *ret = NULL;
Antoine Pitrou19467d22010-08-17 19:33:30 +00002421 int flags = 0;
2422 Py_ssize_t recvlen, outlen;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002423
Antoine Pitrou19467d22010-08-17 19:33:30 +00002424 if (!PyArg_ParseTuple(args, "n|i:recvfrom", &recvlen, &flags))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002425 return NULL;
Guido van Rossum48a680c2001-03-02 06:34:14 +00002426
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002427 if (recvlen < 0) {
2428 PyErr_SetString(PyExc_ValueError,
2429 "negative buffersize in recvfrom");
2430 return NULL;
2431 }
Guido van Rossumd8faa362007-04-27 19:54:29 +00002432
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002433 buf = PyBytes_FromStringAndSize((char *) 0, recvlen);
2434 if (buf == NULL)
2435 return NULL;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002436
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002437 outlen = sock_recvfrom_guts(s, PyBytes_AS_STRING(buf),
2438 recvlen, flags, &addr);
2439 if (outlen < 0) {
2440 goto finally;
2441 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002442
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002443 if (outlen != recvlen) {
2444 /* We did not read as many bytes as we anticipated, resize the
2445 string if possible and be succesful. */
2446 if (_PyBytes_Resize(&buf, outlen) < 0)
2447 /* Oopsy, not so succesful after all. */
2448 goto finally;
2449 }
Barry Warsaw752300b1997-01-03 17:18:10 +00002450
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002451 ret = PyTuple_Pack(2, buf, addr);
Guido van Rossum67f7a382002-06-06 21:08:16 +00002452
2453finally:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002454 Py_XDECREF(buf);
2455 Py_XDECREF(addr);
2456 return ret;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002457}
2458
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002459PyDoc_STRVAR(recvfrom_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002460"recvfrom(buffersize[, flags]) -> (data, address info)\n\
2461\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002462Like recv(buffersize, flags) but also return the sender's address info.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002463
Thomas Wouters477c8d52006-05-27 19:21:47 +00002464
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002465/* s.recvfrom_into(buffer[, nbytes [,flags]]) method */
Thomas Wouters477c8d52006-05-27 19:21:47 +00002466
2467static PyObject *
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002468sock_recvfrom_into(PySocketSockObject *s, PyObject *args, PyObject* kwds)
Thomas Wouters477c8d52006-05-27 19:21:47 +00002469{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002470 static char *kwlist[] = {"buffer", "nbytes", "flags", 0};
Thomas Wouters477c8d52006-05-27 19:21:47 +00002471
Antoine Pitrou19467d22010-08-17 19:33:30 +00002472 int flags = 0;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002473 Py_buffer pbuf;
2474 char *buf;
Antoine Pitrou19467d22010-08-17 19:33:30 +00002475 Py_ssize_t readlen, buflen, recvlen = 0;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002476
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002477 PyObject *addr = NULL;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002478
Antoine Pitrou19467d22010-08-17 19:33:30 +00002479 if (!PyArg_ParseTupleAndKeywords(args, kwds, "w*|ni:recvfrom_into",
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002480 kwlist, &pbuf,
2481 &recvlen, &flags))
2482 return NULL;
2483 buf = pbuf.buf;
2484 buflen = pbuf.len;
2485 assert(buf != 0 && buflen > 0);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002486
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002487 if (recvlen < 0) {
2488 PyBuffer_Release(&pbuf);
2489 PyErr_SetString(PyExc_ValueError,
2490 "negative buffersize in recvfrom_into");
2491 return NULL;
2492 }
2493 if (recvlen == 0) {
2494 /* If nbytes was not specified, use the buffer's length */
2495 recvlen = buflen;
2496 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002497
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002498 readlen = sock_recvfrom_guts(s, buf, recvlen, flags, &addr);
2499 if (readlen < 0) {
2500 PyBuffer_Release(&pbuf);
2501 /* Return an error */
2502 Py_XDECREF(addr);
2503 return NULL;
2504 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002505
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002506 PyBuffer_Release(&pbuf);
2507 /* Return the number of bytes read and the address. Note that we do
2508 not do anything special here in the case that readlen < recvlen. */
Antoine Pitrou19467d22010-08-17 19:33:30 +00002509 return Py_BuildValue("nN", readlen, addr);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002510}
2511
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002512PyDoc_STRVAR(recvfrom_into_doc,
2513"recvfrom_into(buffer[, nbytes[, flags]]) -> (nbytes, address info)\n\
Thomas Wouters477c8d52006-05-27 19:21:47 +00002514\n\
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002515Like recv_into(buffer[, nbytes[, flags]]) but also return the sender's address info.");
Thomas Wouters477c8d52006-05-27 19:21:47 +00002516
2517
Guido van Rossumeb6b33a1993-05-25 09:38:27 +00002518/* s.send(data [,flags]) method */
Guido van Rossum30a685f1991-06-27 15:51:29 +00002519
Guido van Rossum73624e91994-10-10 17:59:00 +00002520static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002521sock_send(PySocketSockObject *s, PyObject *args)
Guido van Rossum30a685f1991-06-27 15:51:29 +00002522{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002523 char *buf;
Antoine Pitrou19467d22010-08-17 19:33:30 +00002524 Py_ssize_t len, n = -1;
2525 int flags = 0, timeout;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002526 Py_buffer pbuf;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002527
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002528 if (!PyArg_ParseTuple(args, "y*|i:send", &pbuf, &flags))
2529 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002530
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002531 if (!IS_SELECTABLE(s)) {
2532 PyBuffer_Release(&pbuf);
2533 return select_error();
2534 }
2535 buf = pbuf.buf;
2536 len = pbuf.len;
Neal Norwitz082b2df2006-02-07 07:04:46 +00002537
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002538 Py_BEGIN_ALLOW_THREADS
2539 timeout = internal_select(s, 1);
2540 if (!timeout)
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002541#ifdef __VMS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002542 n = sendsegmented(s->sock_fd, buf, len, flags);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002543#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002544 n = send(s->sock_fd, buf, len, flags);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002545#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002546 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00002547
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002548 PyBuffer_Release(&pbuf);
Martin v. Löwis423be952008-08-13 15:53:07 +00002549
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002550 if (timeout == 1) {
2551 PyErr_SetString(socket_timeout, "timed out");
2552 return NULL;
2553 }
2554 if (n < 0)
2555 return s->errorhandler();
Antoine Pitrou19467d22010-08-17 19:33:30 +00002556 return PyLong_FromSsize_t(n);
Guido van Rossum30a685f1991-06-27 15:51:29 +00002557}
2558
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002559PyDoc_STRVAR(send_doc,
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002560"send(data[, flags]) -> count\n\
Guido van Rossum82a5c661998-07-07 20:45:43 +00002561\n\
2562Send a data string to the socket. For the optional flags\n\
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002563argument, see the Unix manual. Return the number of bytes\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002564sent; this may be less than len(data) if the network is busy.");
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002565
2566
2567/* s.sendall(data [,flags]) method */
2568
2569static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002570sock_sendall(PySocketSockObject *s, PyObject *args)
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002571{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002572 char *buf;
Antoine Pitrou19467d22010-08-17 19:33:30 +00002573 Py_ssize_t len, n = -1;
2574 int flags = 0, timeout;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002575 Py_buffer pbuf;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002576
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002577 if (!PyArg_ParseTuple(args, "y*|i:sendall", &pbuf, &flags))
2578 return NULL;
2579 buf = pbuf.buf;
2580 len = pbuf.len;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002581
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002582 if (!IS_SELECTABLE(s)) {
2583 PyBuffer_Release(&pbuf);
2584 return select_error();
2585 }
Neal Norwitz082b2df2006-02-07 07:04:46 +00002586
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002587 Py_BEGIN_ALLOW_THREADS
2588 do {
2589 timeout = internal_select(s, 1);
2590 n = -1;
2591 if (timeout)
2592 break;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002593#ifdef __VMS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002594 n = sendsegmented(s->sock_fd, buf, len, flags);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002595#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002596 n = send(s->sock_fd, buf, len, flags);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002597#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002598 if (n < 0) {
Gregory P. Smithb585a0f2010-01-04 03:29:50 +00002599#ifdef EINTR
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002600 /* We must handle EINTR here as there is no way for
2601 * the caller to know how much was sent otherwise. */
2602 if (errno == EINTR) {
2603 /* Run signal handlers. If an exception was
2604 * raised, abort and leave this socket in
2605 * an unknown state. */
2606 if (PyErr_CheckSignals())
2607 return NULL;
2608 continue;
2609 }
Gregory P. Smithb585a0f2010-01-04 03:29:50 +00002610#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002611 break;
2612 }
2613 buf += n;
2614 len -= n;
2615 } while (len > 0);
2616 Py_END_ALLOW_THREADS
2617 PyBuffer_Release(&pbuf);
Guido van Rossum67f7a382002-06-06 21:08:16 +00002618
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002619 if (timeout == 1) {
2620 PyErr_SetString(socket_timeout, "timed out");
2621 return NULL;
2622 }
2623 if (n < 0)
2624 return s->errorhandler();
Guido van Rossum67f7a382002-06-06 21:08:16 +00002625
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002626 Py_INCREF(Py_None);
2627 return Py_None;
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002628}
2629
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002630PyDoc_STRVAR(sendall_doc,
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002631"sendall(data[, flags])\n\
2632\n\
2633Send a data string to the socket. For the optional flags\n\
2634argument, see the Unix manual. This calls send() repeatedly\n\
2635until all data is sent. If an error occurs, it's impossible\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002636to tell how much data has been sent.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002637
Guido van Rossum30a685f1991-06-27 15:51:29 +00002638
Guido van Rossumeb6b33a1993-05-25 09:38:27 +00002639/* s.sendto(data, [flags,] sockaddr) method */
Guido van Rossum30a685f1991-06-27 15:51:29 +00002640
Guido van Rossum73624e91994-10-10 17:59:00 +00002641static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002642sock_sendto(PySocketSockObject *s, PyObject *args)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002643{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002644 Py_buffer pbuf;
2645 PyObject *addro;
2646 char *buf;
2647 Py_ssize_t len;
2648 sock_addr_t addrbuf;
2649 int addrlen, n = -1, flags, timeout;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002650
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002651 flags = 0;
2652 if (!PyArg_ParseTuple(args, "y*O:sendto", &pbuf, &addro)) {
2653 PyErr_Clear();
2654 if (!PyArg_ParseTuple(args, "y*iO:sendto",
2655 &pbuf, &flags, &addro))
2656 return NULL;
2657 }
2658 buf = pbuf.buf;
2659 len = pbuf.len;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002660
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002661 if (!IS_SELECTABLE(s)) {
2662 PyBuffer_Release(&pbuf);
2663 return select_error();
2664 }
Neal Norwitz082b2df2006-02-07 07:04:46 +00002665
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002666 if (!getsockaddrarg(s, addro, SAS2SA(&addrbuf), &addrlen)) {
2667 PyBuffer_Release(&pbuf);
2668 return NULL;
2669 }
Thomas Wouters89f507f2006-12-13 04:49:30 +00002670
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002671 Py_BEGIN_ALLOW_THREADS
2672 timeout = internal_select(s, 1);
2673 if (!timeout)
2674 n = sendto(s->sock_fd, buf, len, flags, SAS2SA(&addrbuf), addrlen);
2675 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00002676
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002677 PyBuffer_Release(&pbuf);
2678 if (timeout == 1) {
2679 PyErr_SetString(socket_timeout, "timed out");
2680 return NULL;
2681 }
2682 if (n < 0)
2683 return s->errorhandler();
Antoine Pitrou19467d22010-08-17 19:33:30 +00002684 return PyLong_FromSsize_t(n);
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002685}
2686
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002687PyDoc_STRVAR(sendto_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00002688"sendto(data[, flags], address) -> count\n\
Guido van Rossum82a5c661998-07-07 20:45:43 +00002689\n\
2690Like send(data, flags) but allows specifying the destination address.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002691For IP sockets, the address is a pair (hostaddr, port).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002692
Guido van Rossum30a685f1991-06-27 15:51:29 +00002693
2694/* s.shutdown(how) method */
2695
Guido van Rossum73624e91994-10-10 17:59:00 +00002696static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002697sock_shutdown(PySocketSockObject *s, PyObject *arg)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002698{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002699 int how;
2700 int res;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00002701
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002702 how = PyLong_AsLong(arg);
2703 if (how == -1 && PyErr_Occurred())
2704 return NULL;
2705 Py_BEGIN_ALLOW_THREADS
2706 res = shutdown(s->sock_fd, how);
2707 Py_END_ALLOW_THREADS
2708 if (res < 0)
2709 return s->errorhandler();
2710 Py_INCREF(Py_None);
2711 return Py_None;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002712}
2713
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002714PyDoc_STRVAR(shutdown_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002715"shutdown(flag)\n\
2716\n\
Martin v. Löwis94681fc2003-11-27 19:40:22 +00002717Shut down the reading side of the socket (flag == SHUT_RD), the writing side\n\
2718of the socket (flag == SHUT_WR), or both ends (flag == SHUT_RDWR).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002719
Amaury Forgeot d'Arc3d17a5c2008-06-13 01:09:34 +00002720#if defined(MS_WINDOWS) && defined(SIO_RCVALL)
Christian Heimesfaf2f632008-01-06 16:59:19 +00002721static PyObject*
2722sock_ioctl(PySocketSockObject *s, PyObject *arg)
2723{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002724 unsigned long cmd = SIO_RCVALL;
2725 PyObject *argO;
2726 DWORD recv;
Christian Heimesfaf2f632008-01-06 16:59:19 +00002727
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002728 if (!PyArg_ParseTuple(arg, "kO:ioctl", &cmd, &argO))
2729 return NULL;
Christian Heimesfaf2f632008-01-06 16:59:19 +00002730
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002731 switch (cmd) {
2732 case SIO_RCVALL: {
2733 unsigned int option = RCVALL_ON;
2734 if (!PyArg_ParseTuple(arg, "kI:ioctl", &cmd, &option))
2735 return NULL;
2736 if (WSAIoctl(s->sock_fd, cmd, &option, sizeof(option),
2737 NULL, 0, &recv, NULL, NULL) == SOCKET_ERROR) {
2738 return set_error();
2739 }
2740 return PyLong_FromUnsignedLong(recv); }
2741 case SIO_KEEPALIVE_VALS: {
2742 struct tcp_keepalive ka;
2743 if (!PyArg_ParseTuple(arg, "k(kkk):ioctl", &cmd,
2744 &ka.onoff, &ka.keepalivetime, &ka.keepaliveinterval))
2745 return NULL;
2746 if (WSAIoctl(s->sock_fd, cmd, &ka, sizeof(ka),
2747 NULL, 0, &recv, NULL, NULL) == SOCKET_ERROR) {
2748 return set_error();
2749 }
2750 return PyLong_FromUnsignedLong(recv); }
2751 default:
2752 PyErr_Format(PyExc_ValueError, "invalid ioctl command %d", cmd);
2753 return NULL;
2754 }
Christian Heimesfaf2f632008-01-06 16:59:19 +00002755}
2756PyDoc_STRVAR(sock_ioctl_doc,
2757"ioctl(cmd, option) -> long\n\
2758\n\
Kristján Valur Jónsson847ec752009-09-27 21:10:38 +00002759Control the socket with WSAIoctl syscall. Currently supported 'cmd' values are\n\
2760SIO_RCVALL: 'option' must be one of the socket.RCVALL_* constants.\n\
2761SIO_KEEPALIVE_VALS: 'option' is a tuple of (onoff, timeout, interval).");
Christian Heimesfaf2f632008-01-06 16:59:19 +00002762
2763#endif
Guido van Rossum30a685f1991-06-27 15:51:29 +00002764
2765/* List of methods for socket objects */
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002766
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002767static PyMethodDef sock_methods[] = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002768 {"_accept", (PyCFunction)sock_accept, METH_NOARGS,
2769 accept_doc},
2770 {"bind", (PyCFunction)sock_bind, METH_O,
2771 bind_doc},
2772 {"close", (PyCFunction)sock_close, METH_NOARGS,
2773 close_doc},
2774 {"connect", (PyCFunction)sock_connect, METH_O,
2775 connect_doc},
2776 {"connect_ex", (PyCFunction)sock_connect_ex, METH_O,
2777 connect_ex_doc},
Antoine Pitrou6e451df2010-08-09 20:39:54 +00002778 {"detach", (PyCFunction)sock_detach, METH_NOARGS,
2779 detach_doc},
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002780 {"fileno", (PyCFunction)sock_fileno, METH_NOARGS,
2781 fileno_doc},
Guido van Rossumb6775db1994-08-01 11:34:53 +00002782#ifdef HAVE_GETPEERNAME
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002783 {"getpeername", (PyCFunction)sock_getpeername,
2784 METH_NOARGS, getpeername_doc},
Guido van Rossum9575a441993-04-07 14:06:14 +00002785#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002786 {"getsockname", (PyCFunction)sock_getsockname,
2787 METH_NOARGS, getsockname_doc},
2788 {"getsockopt", (PyCFunction)sock_getsockopt, METH_VARARGS,
2789 getsockopt_doc},
Amaury Forgeot d'Arc3d17a5c2008-06-13 01:09:34 +00002790#if defined(MS_WINDOWS) && defined(SIO_RCVALL)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002791 {"ioctl", (PyCFunction)sock_ioctl, METH_VARARGS,
2792 sock_ioctl_doc},
Christian Heimesfaf2f632008-01-06 16:59:19 +00002793#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002794 {"listen", (PyCFunction)sock_listen, METH_O,
2795 listen_doc},
2796 {"recv", (PyCFunction)sock_recv, METH_VARARGS,
2797 recv_doc},
2798 {"recv_into", (PyCFunction)sock_recv_into, METH_VARARGS | METH_KEYWORDS,
2799 recv_into_doc},
2800 {"recvfrom", (PyCFunction)sock_recvfrom, METH_VARARGS,
2801 recvfrom_doc},
2802 {"recvfrom_into", (PyCFunction)sock_recvfrom_into, METH_VARARGS | METH_KEYWORDS,
2803 recvfrom_into_doc},
2804 {"send", (PyCFunction)sock_send, METH_VARARGS,
2805 send_doc},
2806 {"sendall", (PyCFunction)sock_sendall, METH_VARARGS,
2807 sendall_doc},
2808 {"sendto", (PyCFunction)sock_sendto, METH_VARARGS,
2809 sendto_doc},
2810 {"setblocking", (PyCFunction)sock_setblocking, METH_O,
2811 setblocking_doc},
2812 {"settimeout", (PyCFunction)sock_settimeout, METH_O,
2813 settimeout_doc},
2814 {"gettimeout", (PyCFunction)sock_gettimeout, METH_NOARGS,
2815 gettimeout_doc},
2816 {"setsockopt", (PyCFunction)sock_setsockopt, METH_VARARGS,
2817 setsockopt_doc},
2818 {"shutdown", (PyCFunction)sock_shutdown, METH_O,
2819 shutdown_doc},
2820 {NULL, NULL} /* sentinel */
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002821};
2822
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002823/* SockObject members */
2824static PyMemberDef sock_memberlist[] = {
2825 {"family", T_INT, offsetof(PySocketSockObject, sock_family), READONLY, "the socket family"},
2826 {"type", T_INT, offsetof(PySocketSockObject, sock_type), READONLY, "the socket type"},
2827 {"proto", T_INT, offsetof(PySocketSockObject, sock_proto), READONLY, "the socket protocol"},
2828 {"timeout", T_DOUBLE, offsetof(PySocketSockObject, sock_timeout), READONLY, "the socket timeout"},
2829 {0},
2830};
Guido van Rossum30a685f1991-06-27 15:51:29 +00002831
Guido van Rossum73624e91994-10-10 17:59:00 +00002832/* Deallocate a socket object in response to the last Py_DECREF().
Guido van Rossum30a685f1991-06-27 15:51:29 +00002833 First close the file description. */
2834
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002835static void
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002836sock_dealloc(PySocketSockObject *s)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002837{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002838 if (s->sock_fd != -1)
2839 (void) SOCKETCLOSE(s->sock_fd);
2840 Py_TYPE(s)->tp_free((PyObject *)s);
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002841}
2842
Guido van Rossum30a685f1991-06-27 15:51:29 +00002843
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00002844static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002845sock_repr(PySocketSockObject *s)
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00002846{
Fred Drakea04eaad2000-06-30 02:46:07 +00002847#if SIZEOF_SOCKET_T > SIZEOF_LONG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002848 if (s->sock_fd > LONG_MAX) {
2849 /* this can occur on Win64, and actually there is a special
2850 ugly printf formatter for decimal pointer length integer
2851 printing, only bother if necessary*/
2852 PyErr_SetString(PyExc_OverflowError,
2853 "no printf formatter to display "
2854 "the socket descriptor in decimal");
2855 return NULL;
2856 }
Fred Drakea04eaad2000-06-30 02:46:07 +00002857#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002858 return PyUnicode_FromFormat(
2859 "<socket object, fd=%ld, family=%d, type=%d, proto=%d>",
2860 (long)s->sock_fd, s->sock_family,
2861 s->sock_type,
2862 s->sock_proto);
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00002863}
2864
2865
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002866/* Create a new, uninitialized socket object. */
2867
2868static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002869sock_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002870{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002871 PyObject *new;
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002872
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002873 new = type->tp_alloc(type, 0);
2874 if (new != NULL) {
2875 ((PySocketSockObject *)new)->sock_fd = -1;
2876 ((PySocketSockObject *)new)->sock_timeout = -1.0;
2877 ((PySocketSockObject *)new)->errorhandler = &set_error;
2878 }
2879 return new;
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002880}
2881
2882
2883/* Initialize a new socket object. */
2884
2885/*ARGSUSED*/
2886static int
Andrew MacIntyre7aec4a22002-06-13 11:53:52 +00002887sock_initobj(PyObject *self, PyObject *args, PyObject *kwds)
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002888{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002889 PySocketSockObject *s = (PySocketSockObject *)self;
2890 PyObject *fdobj = NULL;
2891 SOCKET_T fd = INVALID_SOCKET;
2892 int family = AF_INET, type = SOCK_STREAM, proto = 0;
2893 static char *keywords[] = {"family", "type", "proto", "fileno", 0};
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002894
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002895 if (!PyArg_ParseTupleAndKeywords(args, kwds,
2896 "|iiiO:socket", keywords,
2897 &family, &type, &proto, &fdobj))
2898 return -1;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002899
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002900 if (fdobj != NULL && fdobj != Py_None) {
2901 fd = PyLong_AsSocket_t(fdobj);
2902 if (fd == (SOCKET_T)(-1) && PyErr_Occurred())
2903 return -1;
2904 if (fd == INVALID_SOCKET) {
2905 PyErr_SetString(PyExc_ValueError,
2906 "can't use invalid socket value");
2907 return -1;
2908 }
2909 }
2910 else {
2911 Py_BEGIN_ALLOW_THREADS
2912 fd = socket(family, type, proto);
2913 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00002914
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002915 if (fd == INVALID_SOCKET) {
2916 set_error();
2917 return -1;
2918 }
2919 }
2920 init_sockobject(s, fd, family, type, proto);
Guido van Rossum67f7a382002-06-06 21:08:16 +00002921
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002922 return 0;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002923
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002924}
2925
2926
Guido van Rossumb6775db1994-08-01 11:34:53 +00002927/* Type object for socket objects. */
Guido van Rossum30a685f1991-06-27 15:51:29 +00002928
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002929static PyTypeObject sock_type = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002930 PyVarObject_HEAD_INIT(0, 0) /* Must fill in type value later */
2931 "_socket.socket", /* tp_name */
2932 sizeof(PySocketSockObject), /* tp_basicsize */
2933 0, /* tp_itemsize */
2934 (destructor)sock_dealloc, /* tp_dealloc */
2935 0, /* tp_print */
2936 0, /* tp_getattr */
2937 0, /* tp_setattr */
2938 0, /* tp_reserved */
2939 (reprfunc)sock_repr, /* tp_repr */
2940 0, /* tp_as_number */
2941 0, /* tp_as_sequence */
2942 0, /* tp_as_mapping */
2943 0, /* tp_hash */
2944 0, /* tp_call */
2945 0, /* tp_str */
2946 PyObject_GenericGetAttr, /* tp_getattro */
2947 0, /* tp_setattro */
2948 0, /* tp_as_buffer */
2949 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE, /* tp_flags */
2950 sock_doc, /* tp_doc */
2951 0, /* tp_traverse */
2952 0, /* tp_clear */
2953 0, /* tp_richcompare */
2954 0, /* tp_weaklistoffset */
2955 0, /* tp_iter */
2956 0, /* tp_iternext */
2957 sock_methods, /* tp_methods */
2958 sock_memberlist, /* tp_members */
2959 0, /* tp_getset */
2960 0, /* tp_base */
2961 0, /* tp_dict */
2962 0, /* tp_descr_get */
2963 0, /* tp_descr_set */
2964 0, /* tp_dictoffset */
2965 sock_initobj, /* tp_init */
2966 PyType_GenericAlloc, /* tp_alloc */
2967 sock_new, /* tp_new */
2968 PyObject_Del, /* tp_free */
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002969};
2970
Guido van Rossum30a685f1991-06-27 15:51:29 +00002971
Guido van Rossum81194471991-07-27 21:42:02 +00002972/* Python interface to gethostname(). */
2973
2974/*ARGSUSED*/
Guido van Rossum73624e91994-10-10 17:59:00 +00002975static PyObject *
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00002976socket_gethostname(PyObject *self, PyObject *unused)
Guido van Rossum81194471991-07-27 21:42:02 +00002977{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002978 char buf[1024];
2979 int res;
2980 Py_BEGIN_ALLOW_THREADS
2981 res = gethostname(buf, (int) sizeof buf - 1);
2982 Py_END_ALLOW_THREADS
2983 if (res < 0)
2984 return set_error();
2985 buf[sizeof buf - 1] = '\0';
2986 return PyUnicode_FromString(buf);
Guido van Rossum81194471991-07-27 21:42:02 +00002987}
Guido van Rossumff4949e1992-08-05 19:58:53 +00002988
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002989PyDoc_STRVAR(gethostname_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002990"gethostname() -> string\n\
2991\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002992Return the current host name.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002993
Guido van Rossumff4949e1992-08-05 19:58:53 +00002994
Guido van Rossum30a685f1991-06-27 15:51:29 +00002995/* Python interface to gethostbyname(name). */
2996
2997/*ARGSUSED*/
Guido van Rossum73624e91994-10-10 17:59:00 +00002998static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002999socket_gethostbyname(PyObject *self, PyObject *args)
Guido van Rossum30a685f1991-06-27 15:51:29 +00003000{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003001 char *name;
3002 sock_addr_t addrbuf;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003003
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003004 if (!PyArg_ParseTuple(args, "s:gethostbyname", &name))
3005 return NULL;
3006 if (setipaddr(name, SAS2SA(&addrbuf), sizeof(addrbuf), AF_INET) < 0)
3007 return NULL;
3008 return makeipaddr(SAS2SA(&addrbuf), sizeof(struct sockaddr_in));
Guido van Rossum30a685f1991-06-27 15:51:29 +00003009}
3010
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003011PyDoc_STRVAR(gethostbyname_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003012"gethostbyname(host) -> address\n\
3013\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003014Return the IP address (a string of the form '255.255.255.255') for a host.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003015
3016
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003017/* Convenience function common to gethostbyname_ex and gethostbyaddr */
3018
3019static PyObject *
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003020gethost_common(struct hostent *h, struct sockaddr *addr, int alen, int af)
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003021{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003022 char **pch;
3023 PyObject *rtn_tuple = (PyObject *)NULL;
3024 PyObject *name_list = (PyObject *)NULL;
3025 PyObject *addr_list = (PyObject *)NULL;
3026 PyObject *tmp;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003027
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003028 if (h == NULL) {
3029 /* Let's get real error message to return */
3030 set_herror(h_errno);
3031 return NULL;
3032 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00003033
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003034 if (h->h_addrtype != af) {
3035 /* Let's get real error message to return */
3036 PyErr_SetString(socket_error,
3037 (char *)strerror(EAFNOSUPPORT));
Christian Heimesada8c3b2008-03-18 18:26:33 +00003038
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003039 return NULL;
3040 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00003041
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003042 switch (af) {
Guido van Rossum67f7a382002-06-06 21:08:16 +00003043
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003044 case AF_INET:
3045 if (alen < sizeof(struct sockaddr_in))
3046 return NULL;
3047 break;
Guido van Rossum67f7a382002-06-06 21:08:16 +00003048
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00003049#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003050 case AF_INET6:
3051 if (alen < sizeof(struct sockaddr_in6))
3052 return NULL;
3053 break;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003054#endif
Guido van Rossum67f7a382002-06-06 21:08:16 +00003055
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003056 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00003057
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003058 if ((name_list = PyList_New(0)) == NULL)
3059 goto err;
Guido van Rossum67f7a382002-06-06 21:08:16 +00003060
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003061 if ((addr_list = PyList_New(0)) == NULL)
3062 goto err;
Guido van Rossum67f7a382002-06-06 21:08:16 +00003063
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003064 /* SF #1511317: h_aliases can be NULL */
3065 if (h->h_aliases) {
3066 for (pch = h->h_aliases; *pch != NULL; pch++) {
3067 int status;
3068 tmp = PyUnicode_FromString(*pch);
3069 if (tmp == NULL)
3070 goto err;
Guido van Rossum67f7a382002-06-06 21:08:16 +00003071
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003072 status = PyList_Append(name_list, tmp);
3073 Py_DECREF(tmp);
Guido van Rossum67f7a382002-06-06 21:08:16 +00003074
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003075 if (status)
3076 goto err;
3077 }
3078 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00003079
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003080 for (pch = h->h_addr_list; *pch != NULL; pch++) {
3081 int status;
Guido van Rossum67f7a382002-06-06 21:08:16 +00003082
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003083 switch (af) {
Guido van Rossum67f7a382002-06-06 21:08:16 +00003084
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003085 case AF_INET:
3086 {
3087 struct sockaddr_in sin;
3088 memset(&sin, 0, sizeof(sin));
3089 sin.sin_family = af;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003090#ifdef HAVE_SOCKADDR_SA_LEN
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003091 sin.sin_len = sizeof(sin);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003092#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003093 memcpy(&sin.sin_addr, *pch, sizeof(sin.sin_addr));
3094 tmp = makeipaddr((struct sockaddr *)&sin, sizeof(sin));
Guido van Rossum67f7a382002-06-06 21:08:16 +00003095
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003096 if (pch == h->h_addr_list && alen >= sizeof(sin))
3097 memcpy((char *) addr, &sin, sizeof(sin));
3098 break;
3099 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00003100
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00003101#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003102 case AF_INET6:
3103 {
3104 struct sockaddr_in6 sin6;
3105 memset(&sin6, 0, sizeof(sin6));
3106 sin6.sin6_family = af;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003107#ifdef HAVE_SOCKADDR_SA_LEN
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003108 sin6.sin6_len = sizeof(sin6);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003109#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003110 memcpy(&sin6.sin6_addr, *pch, sizeof(sin6.sin6_addr));
3111 tmp = makeipaddr((struct sockaddr *)&sin6,
3112 sizeof(sin6));
Guido van Rossum67f7a382002-06-06 21:08:16 +00003113
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003114 if (pch == h->h_addr_list && alen >= sizeof(sin6))
3115 memcpy((char *) addr, &sin6, sizeof(sin6));
3116 break;
3117 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003118#endif
Guido van Rossum67f7a382002-06-06 21:08:16 +00003119
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003120 default: /* can't happen */
3121 PyErr_SetString(socket_error,
3122 "unsupported address family");
3123 return NULL;
3124 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00003125
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003126 if (tmp == NULL)
3127 goto err;
Guido van Rossum67f7a382002-06-06 21:08:16 +00003128
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003129 status = PyList_Append(addr_list, tmp);
3130 Py_DECREF(tmp);
Guido van Rossum67f7a382002-06-06 21:08:16 +00003131
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003132 if (status)
3133 goto err;
3134 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00003135
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003136 rtn_tuple = Py_BuildValue("sOO", h->h_name, name_list, addr_list);
Guido van Rossum67f7a382002-06-06 21:08:16 +00003137
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003138 err:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003139 Py_XDECREF(name_list);
3140 Py_XDECREF(addr_list);
3141 return rtn_tuple;
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003142}
3143
3144
3145/* Python interface to gethostbyname_ex(name). */
3146
3147/*ARGSUSED*/
3148static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003149socket_gethostbyname_ex(PyObject *self, PyObject *args)
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003150{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003151 char *name;
3152 struct hostent *h;
Martin v. Löwisc16f3bd2003-05-03 09:14:54 +00003153#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003154 struct sockaddr_storage addr;
Martin v. Löwisc16f3bd2003-05-03 09:14:54 +00003155#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003156 struct sockaddr_in addr;
Martin v. Löwisc16f3bd2003-05-03 09:14:54 +00003157#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003158 struct sockaddr *sa;
3159 PyObject *ret;
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003160#ifdef HAVE_GETHOSTBYNAME_R
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003161 struct hostent hp_allocated;
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003162#ifdef HAVE_GETHOSTBYNAME_R_3_ARG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003163 struct hostent_data data;
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003164#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003165 char buf[16384];
3166 int buf_len = (sizeof buf) - 1;
3167 int errnop;
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003168#endif
3169#if defined(HAVE_GETHOSTBYNAME_R_3_ARG) || defined(HAVE_GETHOSTBYNAME_R_6_ARG)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003170 int result;
Guido van Rossume9cd07b1999-03-15 21:40:14 +00003171#endif
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003172#endif /* HAVE_GETHOSTBYNAME_R */
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003173
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003174 if (!PyArg_ParseTuple(args, "s:gethostbyname_ex", &name))
3175 return NULL;
3176 if (setipaddr(name, (struct sockaddr *)&addr, sizeof(addr), AF_INET) < 0)
3177 return NULL;
3178 Py_BEGIN_ALLOW_THREADS
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003179#ifdef HAVE_GETHOSTBYNAME_R
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003180#if defined(HAVE_GETHOSTBYNAME_R_6_ARG)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003181 result = gethostbyname_r(name, &hp_allocated, buf, buf_len,
3182 &h, &errnop);
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003183#elif defined(HAVE_GETHOSTBYNAME_R_5_ARG)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003184 h = gethostbyname_r(name, &hp_allocated, buf, buf_len, &errnop);
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003185#else /* HAVE_GETHOSTBYNAME_R_3_ARG */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003186 memset((void *) &data, '\0', sizeof(data));
3187 result = gethostbyname_r(name, &hp_allocated, &data);
3188 h = (result != 0) ? NULL : &hp_allocated;
Guido van Rossume9cd07b1999-03-15 21:40:14 +00003189#endif
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003190#else /* not HAVE_GETHOSTBYNAME_R */
Guido van Rossum3baaa131999-03-22 21:44:51 +00003191#ifdef USE_GETHOSTBYNAME_LOCK
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003192 PyThread_acquire_lock(netdb_lock, 1);
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003193#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003194 h = gethostbyname(name);
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003195#endif /* HAVE_GETHOSTBYNAME_R */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003196 Py_END_ALLOW_THREADS
3197 /* Some C libraries would require addr.__ss_family instead of
3198 addr.ss_family.
3199 Therefore, we cast the sockaddr_storage into sockaddr to
3200 access sa_family. */
3201 sa = (struct sockaddr*)&addr;
3202 ret = gethost_common(h, (struct sockaddr *)&addr, sizeof(addr),
3203 sa->sa_family);
Guido van Rossum3baaa131999-03-22 21:44:51 +00003204#ifdef USE_GETHOSTBYNAME_LOCK
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003205 PyThread_release_lock(netdb_lock);
Guido van Rossum955becc1999-03-22 20:14:53 +00003206#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003207 return ret;
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003208}
3209
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003210PyDoc_STRVAR(ghbn_ex_doc,
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003211"gethostbyname_ex(host) -> (name, aliaslist, addresslist)\n\
3212\n\
3213Return the true host name, a list of aliases, and a list of IP addresses,\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003214for a host. The host argument is a string giving a host name or IP number.");
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003215
3216
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00003217/* Python interface to gethostbyaddr(IP). */
3218
3219/*ARGSUSED*/
3220static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003221socket_gethostbyaddr(PyObject *self, PyObject *args)
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00003222{
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00003223#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003224 struct sockaddr_storage addr;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003225#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003226 struct sockaddr_in addr;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003227#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003228 struct sockaddr *sa = (struct sockaddr *)&addr;
3229 char *ip_num;
3230 struct hostent *h;
3231 PyObject *ret;
Guido van Rossum4f199ea1998-04-09 20:56:35 +00003232#ifdef HAVE_GETHOSTBYNAME_R
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003233 struct hostent hp_allocated;
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003234#ifdef HAVE_GETHOSTBYNAME_R_3_ARG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003235 struct hostent_data data;
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003236#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003237 /* glibcs up to 2.10 assume that the buf argument to
3238 gethostbyaddr_r is 8-byte aligned, which at least llvm-gcc
3239 does not ensure. The attribute below instructs the compiler
3240 to maintain this alignment. */
3241 char buf[16384] Py_ALIGNED(8);
3242 int buf_len = (sizeof buf) - 1;
3243 int errnop;
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003244#endif
3245#if defined(HAVE_GETHOSTBYNAME_R_3_ARG) || defined(HAVE_GETHOSTBYNAME_R_6_ARG)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003246 int result;
Guido van Rossume9cd07b1999-03-15 21:40:14 +00003247#endif
Guido van Rossum4f199ea1998-04-09 20:56:35 +00003248#endif /* HAVE_GETHOSTBYNAME_R */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003249 char *ap;
3250 int al;
3251 int af;
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00003252
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003253 if (!PyArg_ParseTuple(args, "s:gethostbyaddr", &ip_num))
3254 return NULL;
3255 af = AF_UNSPEC;
3256 if (setipaddr(ip_num, sa, sizeof(addr), af) < 0)
3257 return NULL;
3258 af = sa->sa_family;
3259 ap = NULL;
3260 al = 0;
3261 switch (af) {
3262 case AF_INET:
3263 ap = (char *)&((struct sockaddr_in *)sa)->sin_addr;
3264 al = sizeof(((struct sockaddr_in *)sa)->sin_addr);
3265 break;
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00003266#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003267 case AF_INET6:
3268 ap = (char *)&((struct sockaddr_in6 *)sa)->sin6_addr;
3269 al = sizeof(((struct sockaddr_in6 *)sa)->sin6_addr);
3270 break;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003271#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003272 default:
3273 PyErr_SetString(socket_error, "unsupported address family");
3274 return NULL;
3275 }
3276 Py_BEGIN_ALLOW_THREADS
Guido van Rossum4f199ea1998-04-09 20:56:35 +00003277#ifdef HAVE_GETHOSTBYNAME_R
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003278#if defined(HAVE_GETHOSTBYNAME_R_6_ARG)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003279 result = gethostbyaddr_r(ap, al, af,
3280 &hp_allocated, buf, buf_len,
3281 &h, &errnop);
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003282#elif defined(HAVE_GETHOSTBYNAME_R_5_ARG)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003283 h = gethostbyaddr_r(ap, al, af,
3284 &hp_allocated, buf, buf_len, &errnop);
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003285#else /* HAVE_GETHOSTBYNAME_R_3_ARG */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003286 memset((void *) &data, '\0', sizeof(data));
3287 result = gethostbyaddr_r(ap, al, af, &hp_allocated, &data);
3288 h = (result != 0) ? NULL : &hp_allocated;
Guido van Rossume9cd07b1999-03-15 21:40:14 +00003289#endif
Guido van Rossum4f199ea1998-04-09 20:56:35 +00003290#else /* not HAVE_GETHOSTBYNAME_R */
Guido van Rossum3baaa131999-03-22 21:44:51 +00003291#ifdef USE_GETHOSTBYNAME_LOCK
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003292 PyThread_acquire_lock(netdb_lock, 1);
Guido van Rossum4f199ea1998-04-09 20:56:35 +00003293#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003294 h = gethostbyaddr(ap, al, af);
Guido van Rossum4f199ea1998-04-09 20:56:35 +00003295#endif /* HAVE_GETHOSTBYNAME_R */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003296 Py_END_ALLOW_THREADS
3297 ret = gethost_common(h, (struct sockaddr *)&addr, sizeof(addr), af);
Guido van Rossum3baaa131999-03-22 21:44:51 +00003298#ifdef USE_GETHOSTBYNAME_LOCK
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003299 PyThread_release_lock(netdb_lock);
Guido van Rossum3baaa131999-03-22 21:44:51 +00003300#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003301 return ret;
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00003302}
3303
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003304PyDoc_STRVAR(gethostbyaddr_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003305"gethostbyaddr(host) -> (name, aliaslist, addresslist)\n\
3306\n\
3307Return the true host name, a list of aliases, and a list of IP addresses,\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003308for a host. The host argument is a string giving a host name or IP number.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003309
Guido van Rossum30a685f1991-06-27 15:51:29 +00003310
3311/* Python interface to getservbyname(name).
3312 This only returns the port number, since the other info is already
3313 known or not useful (like the list of aliases). */
3314
3315/*ARGSUSED*/
Guido van Rossum73624e91994-10-10 17:59:00 +00003316static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003317socket_getservbyname(PyObject *self, PyObject *args)
Guido van Rossum30a685f1991-06-27 15:51:29 +00003318{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003319 char *name, *proto=NULL;
3320 struct servent *sp;
3321 if (!PyArg_ParseTuple(args, "s|s:getservbyname", &name, &proto))
3322 return NULL;
3323 Py_BEGIN_ALLOW_THREADS
3324 sp = getservbyname(name, proto);
3325 Py_END_ALLOW_THREADS
3326 if (sp == NULL) {
3327 PyErr_SetString(socket_error, "service/proto not found");
3328 return NULL;
3329 }
3330 return PyLong_FromLong((long) ntohs(sp->s_port));
Guido van Rossum30a685f1991-06-27 15:51:29 +00003331}
3332
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003333PyDoc_STRVAR(getservbyname_doc,
Barry Warsaw11b91a02004-06-28 00:50:43 +00003334"getservbyname(servicename[, protocolname]) -> integer\n\
Guido van Rossum82a5c661998-07-07 20:45:43 +00003335\n\
3336Return a port number from a service name and protocol name.\n\
Barry Warsaw11b91a02004-06-28 00:50:43 +00003337The optional protocol name, if given, should be 'tcp' or 'udp',\n\
3338otherwise any protocol will match.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003339
Guido van Rossum30a685f1991-06-27 15:51:29 +00003340
Barry Warsaw11b91a02004-06-28 00:50:43 +00003341/* Python interface to getservbyport(port).
3342 This only returns the service name, since the other info is already
3343 known or not useful (like the list of aliases). */
3344
3345/*ARGSUSED*/
3346static PyObject *
3347socket_getservbyport(PyObject *self, PyObject *args)
3348{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003349 int port;
3350 char *proto=NULL;
3351 struct servent *sp;
3352 if (!PyArg_ParseTuple(args, "i|s:getservbyport", &port, &proto))
3353 return NULL;
3354 if (port < 0 || port > 0xffff) {
3355 PyErr_SetString(
3356 PyExc_OverflowError,
3357 "getservbyport: port must be 0-65535.");
3358 return NULL;
3359 }
3360 Py_BEGIN_ALLOW_THREADS
3361 sp = getservbyport(htons((short)port), proto);
3362 Py_END_ALLOW_THREADS
3363 if (sp == NULL) {
3364 PyErr_SetString(socket_error, "port/proto not found");
3365 return NULL;
3366 }
3367 return PyUnicode_FromString(sp->s_name);
Barry Warsaw11b91a02004-06-28 00:50:43 +00003368}
3369
3370PyDoc_STRVAR(getservbyport_doc,
3371"getservbyport(port[, protocolname]) -> string\n\
3372\n\
3373Return the service name from a port number and protocol name.\n\
3374The optional protocol name, if given, should be 'tcp' or 'udp',\n\
3375otherwise any protocol will match.");
3376
Guido van Rossum3901d851996-12-19 16:35:04 +00003377/* Python interface to getprotobyname(name).
3378 This only returns the protocol number, since the other info is
3379 already known or not useful (like the list of aliases). */
3380
3381/*ARGSUSED*/
3382static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003383socket_getprotobyname(PyObject *self, PyObject *args)
Guido van Rossum3901d851996-12-19 16:35:04 +00003384{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003385 char *name;
3386 struct protoent *sp;
3387 if (!PyArg_ParseTuple(args, "s:getprotobyname", &name))
3388 return NULL;
3389 Py_BEGIN_ALLOW_THREADS
3390 sp = getprotobyname(name);
3391 Py_END_ALLOW_THREADS
3392 if (sp == NULL) {
3393 PyErr_SetString(socket_error, "protocol not found");
3394 return NULL;
3395 }
3396 return PyLong_FromLong((long) sp->p_proto);
Guido van Rossum3901d851996-12-19 16:35:04 +00003397}
3398
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003399PyDoc_STRVAR(getprotobyname_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003400"getprotobyname(name) -> integer\n\
3401\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003402Return the protocol number for the named protocol. (Rarely used.)");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003403
Guido van Rossum3901d851996-12-19 16:35:04 +00003404
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00003405#ifndef NO_DUP
3406/* dup() function for socket fds */
3407
3408static PyObject *
3409socket_dup(PyObject *self, PyObject *fdobj)
3410{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003411 SOCKET_T fd, newfd;
3412 PyObject *newfdobj;
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00003413
3414
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003415 fd = PyLong_AsSocket_t(fdobj);
3416 if (fd == (SOCKET_T)(-1) && PyErr_Occurred())
3417 return NULL;
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00003418
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003419 newfd = dup_socket(fd);
3420 if (newfd == INVALID_SOCKET)
3421 return set_error();
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00003422
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003423 newfdobj = PyLong_FromSocket_t(newfd);
3424 if (newfdobj == NULL)
3425 SOCKETCLOSE(newfd);
3426 return newfdobj;
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00003427}
3428
3429PyDoc_STRVAR(dup_doc,
3430"dup(integer) -> integer\n\
3431\n\
3432Duplicate an integer socket file descriptor. This is like os.dup(), but for\n\
3433sockets; on some platforms os.dup() won't work for socket file descriptors.");
3434#endif
3435
3436
Dave Cole331708b2004-08-09 04:51:41 +00003437#ifdef HAVE_SOCKETPAIR
3438/* Create a pair of sockets using the socketpair() function.
Dave Cole07fda7e2004-08-23 05:16:23 +00003439 Arguments as for socket() except the default family is AF_UNIX if
Dave Colee8bbfe42004-08-26 00:51:16 +00003440 defined on the platform; otherwise, the default is AF_INET. */
Dave Cole331708b2004-08-09 04:51:41 +00003441
3442/*ARGSUSED*/
3443static PyObject *
3444socket_socketpair(PyObject *self, PyObject *args)
3445{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003446 PySocketSockObject *s0 = NULL, *s1 = NULL;
3447 SOCKET_T sv[2];
3448 int family, type = SOCK_STREAM, proto = 0;
3449 PyObject *res = NULL;
Dave Cole331708b2004-08-09 04:51:41 +00003450
3451#if defined(AF_UNIX)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003452 family = AF_UNIX;
Dave Cole331708b2004-08-09 04:51:41 +00003453#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003454 family = AF_INET;
Dave Cole331708b2004-08-09 04:51:41 +00003455#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003456 if (!PyArg_ParseTuple(args, "|iii:socketpair",
3457 &family, &type, &proto))
3458 return NULL;
3459 /* Create a pair of socket fds */
3460 if (socketpair(family, type, proto, sv) < 0)
3461 return set_error();
3462 s0 = new_sockobject(sv[0], family, type, proto);
3463 if (s0 == NULL)
3464 goto finally;
3465 s1 = new_sockobject(sv[1], family, type, proto);
3466 if (s1 == NULL)
3467 goto finally;
3468 res = PyTuple_Pack(2, s0, s1);
Dave Cole331708b2004-08-09 04:51:41 +00003469
3470finally:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003471 if (res == NULL) {
3472 if (s0 == NULL)
3473 SOCKETCLOSE(sv[0]);
3474 if (s1 == NULL)
3475 SOCKETCLOSE(sv[1]);
3476 }
3477 Py_XDECREF(s0);
3478 Py_XDECREF(s1);
3479 return res;
Dave Cole331708b2004-08-09 04:51:41 +00003480}
3481
3482PyDoc_STRVAR(socketpair_doc,
3483"socketpair([family[, type[, proto]]]) -> (socket object, socket object)\n\
3484\n\
3485Create a pair of socket objects from the sockets returned by the platform\n\
3486socketpair() function.\n\
Dave Cole07fda7e2004-08-23 05:16:23 +00003487The arguments are the same as for socket() except the default family is\n\
Dave Colee8bbfe42004-08-26 00:51:16 +00003488AF_UNIX if defined on the platform; otherwise, the default is AF_INET.");
Dave Cole331708b2004-08-09 04:51:41 +00003489
3490#endif /* HAVE_SOCKETPAIR */
3491
3492
Guido van Rossum006bf911996-06-12 04:04:55 +00003493static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003494socket_ntohs(PyObject *self, PyObject *args)
Guido van Rossum006bf911996-06-12 04:04:55 +00003495{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003496 int x1, x2;
Guido van Rossum006bf911996-06-12 04:04:55 +00003497
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003498 if (!PyArg_ParseTuple(args, "i:ntohs", &x1)) {
3499 return NULL;
3500 }
3501 if (x1 < 0) {
3502 PyErr_SetString(PyExc_OverflowError,
3503 "can't convert negative number to unsigned long");
3504 return NULL;
3505 }
3506 x2 = (unsigned int)ntohs((unsigned short)x1);
3507 return PyLong_FromLong(x2);
Guido van Rossum006bf911996-06-12 04:04:55 +00003508}
3509
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003510PyDoc_STRVAR(ntohs_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003511"ntohs(integer) -> integer\n\
3512\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003513Convert a 16-bit integer from network to host byte order.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003514
3515
Guido van Rossum006bf911996-06-12 04:04:55 +00003516static PyObject *
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003517socket_ntohl(PyObject *self, PyObject *arg)
Guido van Rossum006bf911996-06-12 04:04:55 +00003518{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003519 unsigned long x;
Guido van Rossum006bf911996-06-12 04:04:55 +00003520
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003521 if (PyLong_Check(arg)) {
3522 x = PyLong_AsUnsignedLong(arg);
3523 if (x == (unsigned long) -1 && PyErr_Occurred())
3524 return NULL;
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003525#if SIZEOF_LONG > 4
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003526 {
3527 unsigned long y;
3528 /* only want the trailing 32 bits */
3529 y = x & 0xFFFFFFFFUL;
3530 if (y ^ x)
3531 return PyErr_Format(PyExc_OverflowError,
3532 "long int larger than 32 bits");
3533 x = y;
3534 }
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003535#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003536 }
3537 else
3538 return PyErr_Format(PyExc_TypeError,
3539 "expected int/long, %s found",
3540 Py_TYPE(arg)->tp_name);
3541 if (x == (unsigned long) -1 && PyErr_Occurred())
3542 return NULL;
3543 return PyLong_FromUnsignedLong(ntohl(x));
Guido van Rossum006bf911996-06-12 04:04:55 +00003544}
3545
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003546PyDoc_STRVAR(ntohl_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003547"ntohl(integer) -> integer\n\
3548\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003549Convert a 32-bit integer from network to host byte order.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003550
3551
Guido van Rossum006bf911996-06-12 04:04:55 +00003552static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003553socket_htons(PyObject *self, PyObject *args)
Guido van Rossum006bf911996-06-12 04:04:55 +00003554{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003555 int x1, x2;
Guido van Rossum006bf911996-06-12 04:04:55 +00003556
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003557 if (!PyArg_ParseTuple(args, "i:htons", &x1)) {
3558 return NULL;
3559 }
3560 if (x1 < 0) {
3561 PyErr_SetString(PyExc_OverflowError,
3562 "can't convert negative number to unsigned long");
3563 return NULL;
3564 }
3565 x2 = (unsigned int)htons((unsigned short)x1);
3566 return PyLong_FromLong(x2);
Guido van Rossum006bf911996-06-12 04:04:55 +00003567}
3568
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003569PyDoc_STRVAR(htons_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003570"htons(integer) -> integer\n\
3571\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003572Convert a 16-bit integer from host to network byte order.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003573
3574
Guido van Rossum006bf911996-06-12 04:04:55 +00003575static PyObject *
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003576socket_htonl(PyObject *self, PyObject *arg)
Guido van Rossum006bf911996-06-12 04:04:55 +00003577{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003578 unsigned long x;
Guido van Rossum006bf911996-06-12 04:04:55 +00003579
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003580 if (PyLong_Check(arg)) {
3581 x = PyLong_AsUnsignedLong(arg);
3582 if (x == (unsigned long) -1 && PyErr_Occurred())
3583 return NULL;
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003584#if SIZEOF_LONG > 4
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003585 {
3586 unsigned long y;
3587 /* only want the trailing 32 bits */
3588 y = x & 0xFFFFFFFFUL;
3589 if (y ^ x)
3590 return PyErr_Format(PyExc_OverflowError,
3591 "long int larger than 32 bits");
3592 x = y;
3593 }
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003594#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003595 }
3596 else
3597 return PyErr_Format(PyExc_TypeError,
3598 "expected int/long, %s found",
3599 Py_TYPE(arg)->tp_name);
3600 return PyLong_FromUnsignedLong(htonl((unsigned long)x));
Guido van Rossum006bf911996-06-12 04:04:55 +00003601}
3602
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003603PyDoc_STRVAR(htonl_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003604"htonl(integer) -> integer\n\
3605\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003606Convert a 32-bit integer from host to network byte order.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003607
Guido van Rossum3eede5a2002-06-07 02:08:35 +00003608/* socket.inet_aton() and socket.inet_ntoa() functions. */
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003609
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003610PyDoc_STRVAR(inet_aton_doc,
Guido van Rossum7d0a8262007-05-21 23:13:11 +00003611"inet_aton(string) -> bytes giving packed 32-bit IP representation\n\
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003612\n\
Guido van Rossumc6a164b1999-08-20 19:11:27 +00003613Convert 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 +00003614binary format used in low-level network functions.");
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003615
3616static PyObject*
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003617socket_inet_aton(PyObject *self, PyObject *args)
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003618{
Guido van Rossuma2e48551999-09-09 15:42:59 +00003619#ifndef INADDR_NONE
3620#define INADDR_NONE (-1)
3621#endif
Neal Norwitz88f115b2003-02-13 02:15:42 +00003622#ifdef HAVE_INET_ATON
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003623 struct in_addr buf;
Thomas Wouters477c8d52006-05-27 19:21:47 +00003624#endif
3625
3626#if !defined(HAVE_INET_ATON) || defined(USE_INET_ATON_WEAKLINK)
Benjamin Petersonf91df042009-02-13 02:50:59 +00003627#if (SIZEOF_INT != 4)
3628#error "Not sure if in_addr_t exists and int is not 32-bits."
3629#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003630 /* Have to use inet_addr() instead */
3631 unsigned int packed_addr;
Tim Peters1df9fdd2003-02-13 03:13:40 +00003632#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003633 char *ip_addr;
Guido van Rossumc6a164b1999-08-20 19:11:27 +00003634
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003635 if (!PyArg_ParseTuple(args, "s:inet_aton", &ip_addr))
3636 return NULL;
Guido van Rossumad05cdf2003-02-12 23:08:22 +00003637
Tim Peters1df9fdd2003-02-13 03:13:40 +00003638
3639#ifdef HAVE_INET_ATON
Thomas Wouters477c8d52006-05-27 19:21:47 +00003640
3641#ifdef USE_INET_ATON_WEAKLINK
3642 if (inet_aton != NULL) {
3643#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003644 if (inet_aton(ip_addr, &buf))
3645 return PyBytes_FromStringAndSize((char *)(&buf),
3646 sizeof(buf));
Guido van Rossumad05cdf2003-02-12 23:08:22 +00003647
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003648 PyErr_SetString(socket_error,
3649 "illegal IP address string passed to inet_aton");
3650 return NULL;
Guido van Rossumad05cdf2003-02-12 23:08:22 +00003651
Thomas Wouters477c8d52006-05-27 19:21:47 +00003652#ifdef USE_INET_ATON_WEAKLINK
3653 } else {
3654#endif
3655
3656#endif
3657
3658#if !defined(HAVE_INET_ATON) || defined(USE_INET_ATON_WEAKLINK)
3659
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003660 /* special-case this address as inet_addr might return INADDR_NONE
3661 * for this */
3662 if (strcmp(ip_addr, "255.255.255.255") == 0) {
3663 packed_addr = 0xFFFFFFFF;
3664 } else {
Thomas Wouters477c8d52006-05-27 19:21:47 +00003665
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003666 packed_addr = inet_addr(ip_addr);
Guido van Rossumc6a164b1999-08-20 19:11:27 +00003667
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003668 if (packed_addr == INADDR_NONE) { /* invalid address */
3669 PyErr_SetString(socket_error,
3670 "illegal IP address string passed to inet_aton");
3671 return NULL;
3672 }
3673 }
3674 return PyBytes_FromStringAndSize((char *) &packed_addr,
3675 sizeof(packed_addr));
Thomas Wouters477c8d52006-05-27 19:21:47 +00003676
3677#ifdef USE_INET_ATON_WEAKLINK
3678 }
3679#endif
3680
Guido van Rossumad05cdf2003-02-12 23:08:22 +00003681#endif
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003682}
3683
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003684PyDoc_STRVAR(inet_ntoa_doc,
Fred Drakee0661342000-03-07 14:05:16 +00003685"inet_ntoa(packed_ip) -> ip_address_string\n\
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003686\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003687Convert an IP address from 32-bit packed binary format to string format");
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003688
3689static PyObject*
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003690socket_inet_ntoa(PyObject *self, PyObject *args)
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003691{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003692 char *packed_str;
3693 int addr_len;
3694 struct in_addr packed_addr;
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003695
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003696 if (!PyArg_ParseTuple(args, "y#:inet_ntoa", &packed_str, &addr_len)) {
3697 return NULL;
3698 }
Guido van Rossum48a680c2001-03-02 06:34:14 +00003699
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003700 if (addr_len != sizeof(packed_addr)) {
3701 PyErr_SetString(socket_error,
3702 "packed IP wrong length for inet_ntoa");
3703 return NULL;
3704 }
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003705
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003706 memcpy(&packed_addr, packed_str, addr_len);
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003707
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003708 return PyUnicode_FromString(inet_ntoa(packed_addr));
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003709}
Guido van Rossum82a5c661998-07-07 20:45:43 +00003710
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003711#ifdef HAVE_INET_PTON
3712
3713PyDoc_STRVAR(inet_pton_doc,
3714"inet_pton(af, ip) -> packed IP address string\n\
3715\n\
3716Convert an IP address from string format to a packed string suitable\n\
3717for use with low-level network functions.");
3718
3719static PyObject *
3720socket_inet_pton(PyObject *self, PyObject *args)
3721{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003722 int af;
3723 char* ip;
3724 int retval;
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003725#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003726 char packed[MAX(sizeof(struct in_addr), sizeof(struct in6_addr))];
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003727#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003728 char packed[sizeof(struct in_addr)];
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003729#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003730 if (!PyArg_ParseTuple(args, "is:inet_pton", &af, &ip)) {
3731 return NULL;
3732 }
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003733
Martin v. Löwis04697e82004-06-02 12:35:29 +00003734#if !defined(ENABLE_IPV6) && defined(AF_INET6)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003735 if(af == AF_INET6) {
3736 PyErr_SetString(socket_error,
3737 "can't use AF_INET6, IPv6 is disabled");
3738 return NULL;
3739 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00003740#endif
Martin v. Löwis10649092003-08-05 06:25:06 +00003741
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003742 retval = inet_pton(af, ip, packed);
3743 if (retval < 0) {
3744 PyErr_SetFromErrno(socket_error);
3745 return NULL;
3746 } else if (retval == 0) {
3747 PyErr_SetString(socket_error,
3748 "illegal IP address string passed to inet_pton");
3749 return NULL;
3750 } else if (af == AF_INET) {
3751 return PyBytes_FromStringAndSize(packed,
3752 sizeof(struct in_addr));
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003753#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003754 } else if (af == AF_INET6) {
3755 return PyBytes_FromStringAndSize(packed,
3756 sizeof(struct in6_addr));
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003757#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003758 } else {
3759 PyErr_SetString(socket_error, "unknown address family");
3760 return NULL;
3761 }
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003762}
Thomas Wouters477c8d52006-05-27 19:21:47 +00003763
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003764PyDoc_STRVAR(inet_ntop_doc,
3765"inet_ntop(af, packed_ip) -> string formatted IP address\n\
3766\n\
3767Convert a packed IP address of the given family to string format.");
3768
3769static PyObject *
3770socket_inet_ntop(PyObject *self, PyObject *args)
3771{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003772 int af;
3773 char* packed;
3774 int len;
3775 const char* retval;
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003776#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003777 char ip[MAX(INET_ADDRSTRLEN, INET6_ADDRSTRLEN) + 1];
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003778#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003779 char ip[INET_ADDRSTRLEN + 1];
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003780#endif
Thomas Wouters477c8d52006-05-27 19:21:47 +00003781
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003782 /* Guarantee NUL-termination for PyUnicode_FromString() below */
3783 memset((void *) &ip[0], '\0', sizeof(ip));
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003784
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003785 if (!PyArg_ParseTuple(args, "iy#:inet_ntop", &af, &packed, &len)) {
3786 return NULL;
3787 }
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003788
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003789 if (af == AF_INET) {
3790 if (len != sizeof(struct in_addr)) {
3791 PyErr_SetString(PyExc_ValueError,
3792 "invalid length of packed IP address string");
3793 return NULL;
3794 }
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003795#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003796 } else if (af == AF_INET6) {
3797 if (len != sizeof(struct in6_addr)) {
3798 PyErr_SetString(PyExc_ValueError,
3799 "invalid length of packed IP address string");
3800 return NULL;
3801 }
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003802#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003803 } else {
3804 PyErr_Format(PyExc_ValueError,
3805 "unknown address family %d", af);
3806 return NULL;
3807 }
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003808
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003809 retval = inet_ntop(af, packed, ip, sizeof(ip));
3810 if (!retval) {
3811 PyErr_SetFromErrno(socket_error);
3812 return NULL;
3813 } else {
3814 return PyUnicode_FromString(retval);
3815 }
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003816
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003817 /* NOTREACHED */
3818 PyErr_SetString(PyExc_RuntimeError, "invalid handling of inet_ntop");
3819 return NULL;
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003820}
3821
3822#endif /* HAVE_INET_PTON */
3823
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003824/* Python interface to getaddrinfo(host, port). */
3825
3826/*ARGSUSED*/
3827static PyObject *
Giampaolo Rodolàccfb91c2010-08-17 15:30:23 +00003828socket_getaddrinfo(PyObject *self, PyObject *args, PyObject* kwargs)
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003829{
Giampaolo Rodolàccfb91c2010-08-17 15:30:23 +00003830 static char* kwnames[] = {"host", "port", "family", "type", "proto",
3831 "flags", 0};
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003832 struct addrinfo hints, *res;
3833 struct addrinfo *res0 = NULL;
3834 PyObject *hobj = NULL;
3835 PyObject *pobj = (PyObject *)NULL;
3836 char pbuf[30];
3837 char *hptr, *pptr;
3838 int family, socktype, protocol, flags;
3839 int error;
3840 PyObject *all = (PyObject *)NULL;
3841 PyObject *idna = NULL;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003842
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003843 family = socktype = protocol = flags = 0;
3844 family = AF_UNSPEC;
Giampaolo Rodolàccfb91c2010-08-17 15:30:23 +00003845 if (!PyArg_ParseTupleAndKeywords(args, kwargs, "OO|iiii:getaddrinfo",
3846 kwnames, &hobj, &pobj, &family, &socktype,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003847 &protocol, &flags)) {
3848 return NULL;
3849 }
3850 if (hobj == Py_None) {
3851 hptr = NULL;
3852 } else if (PyUnicode_Check(hobj)) {
3853 idna = PyObject_CallMethod(hobj, "encode", "s", "idna");
3854 if (!idna)
3855 return NULL;
3856 assert(PyBytes_Check(idna));
3857 hptr = PyBytes_AS_STRING(idna);
3858 } else if (PyBytes_Check(hobj)) {
3859 hptr = PyBytes_AsString(hobj);
3860 } else {
3861 PyErr_SetString(PyExc_TypeError,
3862 "getaddrinfo() argument 1 must be string or None");
3863 return NULL;
3864 }
3865 if (PyLong_CheckExact(pobj)) {
3866 long value = PyLong_AsLong(pobj);
3867 if (value == -1 && PyErr_Occurred())
3868 goto err;
3869 PyOS_snprintf(pbuf, sizeof(pbuf), "%ld", value);
3870 pptr = pbuf;
3871 } else if (PyUnicode_Check(pobj)) {
3872 pptr = _PyUnicode_AsString(pobj);
3873 } else if (PyBytes_Check(pobj)) {
3874 pptr = PyBytes_AsString(pobj);
3875 } else if (pobj == Py_None) {
3876 pptr = (char *)NULL;
3877 } else {
3878 PyErr_SetString(socket_error, "Int or String expected");
3879 goto err;
3880 }
3881 memset(&hints, 0, sizeof(hints));
3882 hints.ai_family = family;
3883 hints.ai_socktype = socktype;
3884 hints.ai_protocol = protocol;
3885 hints.ai_flags = flags;
3886 Py_BEGIN_ALLOW_THREADS
3887 ACQUIRE_GETADDRINFO_LOCK
3888 error = getaddrinfo(hptr, pptr, &hints, &res0);
3889 Py_END_ALLOW_THREADS
3890 RELEASE_GETADDRINFO_LOCK /* see comment in setipaddr() */
3891 if (error) {
3892 set_gaierror(error);
3893 goto err;
3894 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003895
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003896 if ((all = PyList_New(0)) == NULL)
3897 goto err;
3898 for (res = res0; res; res = res->ai_next) {
3899 PyObject *single;
3900 PyObject *addr =
3901 makesockaddr(-1, res->ai_addr, res->ai_addrlen, protocol);
3902 if (addr == NULL)
3903 goto err;
3904 single = Py_BuildValue("iiisO", res->ai_family,
3905 res->ai_socktype, res->ai_protocol,
3906 res->ai_canonname ? res->ai_canonname : "",
3907 addr);
3908 Py_DECREF(addr);
3909 if (single == NULL)
3910 goto err;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003911
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003912 if (PyList_Append(all, single))
3913 goto err;
3914 Py_XDECREF(single);
3915 }
3916 Py_XDECREF(idna);
3917 if (res0)
3918 freeaddrinfo(res0);
3919 return all;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003920 err:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003921 Py_XDECREF(all);
3922 Py_XDECREF(idna);
3923 if (res0)
3924 freeaddrinfo(res0);
3925 return (PyObject *)NULL;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003926}
3927
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003928PyDoc_STRVAR(getaddrinfo_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00003929"getaddrinfo(host, port [, family, socktype, proto, flags])\n\
3930 -> list of (family, socktype, proto, canonname, sockaddr)\n\
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003931\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003932Resolve host and port into addrinfo struct.");
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003933
3934/* Python interface to getnameinfo(sa, flags). */
3935
3936/*ARGSUSED*/
3937static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003938socket_getnameinfo(PyObject *self, PyObject *args)
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003939{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003940 PyObject *sa = (PyObject *)NULL;
3941 int flags;
3942 char *hostp;
3943 int port, flowinfo, scope_id;
3944 char hbuf[NI_MAXHOST], pbuf[NI_MAXSERV];
3945 struct addrinfo hints, *res = NULL;
3946 int error;
3947 PyObject *ret = (PyObject *)NULL;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003948
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003949 flags = flowinfo = scope_id = 0;
3950 if (!PyArg_ParseTuple(args, "Oi:getnameinfo", &sa, &flags))
3951 return NULL;
3952 if (!PyTuple_Check(sa)) {
3953 PyErr_SetString(PyExc_TypeError,
3954 "getnameinfo() argument 1 must be a tuple");
3955 return NULL;
3956 }
3957 if (!PyArg_ParseTuple(sa, "si|ii",
3958 &hostp, &port, &flowinfo, &scope_id))
3959 return NULL;
3960 PyOS_snprintf(pbuf, sizeof(pbuf), "%d", port);
3961 memset(&hints, 0, sizeof(hints));
3962 hints.ai_family = AF_UNSPEC;
3963 hints.ai_socktype = SOCK_DGRAM; /* make numeric port happy */
3964 Py_BEGIN_ALLOW_THREADS
3965 ACQUIRE_GETADDRINFO_LOCK
3966 error = getaddrinfo(hostp, pbuf, &hints, &res);
3967 Py_END_ALLOW_THREADS
3968 RELEASE_GETADDRINFO_LOCK /* see comment in setipaddr() */
3969 if (error) {
3970 set_gaierror(error);
3971 goto fail;
3972 }
3973 if (res->ai_next) {
3974 PyErr_SetString(socket_error,
3975 "sockaddr resolved to multiple addresses");
3976 goto fail;
3977 }
3978 switch (res->ai_family) {
3979 case AF_INET:
3980 {
3981 if (PyTuple_GET_SIZE(sa) != 2) {
3982 PyErr_SetString(socket_error,
3983 "IPv4 sockaddr must be 2 tuple");
3984 goto fail;
3985 }
3986 break;
3987 }
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00003988#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003989 case AF_INET6:
3990 {
3991 struct sockaddr_in6 *sin6;
3992 sin6 = (struct sockaddr_in6 *)res->ai_addr;
3993 sin6->sin6_flowinfo = flowinfo;
3994 sin6->sin6_scope_id = scope_id;
3995 break;
3996 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003997#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003998 }
Antoine Pitrou19467d22010-08-17 19:33:30 +00003999 error = getnameinfo(res->ai_addr, (socklen_t) res->ai_addrlen,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004000 hbuf, sizeof(hbuf), pbuf, sizeof(pbuf), flags);
4001 if (error) {
4002 set_gaierror(error);
4003 goto fail;
4004 }
4005 ret = Py_BuildValue("ss", hbuf, pbuf);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004006
4007fail:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004008 if (res)
4009 freeaddrinfo(res);
4010 return ret;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004011}
4012
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004013PyDoc_STRVAR(getnameinfo_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00004014"getnameinfo(sockaddr, flags) --> (host, port)\n\
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004015\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004016Get host and port for a sockaddr.");
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004017
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004018
4019/* Python API to getting and setting the default timeout value. */
4020
4021static PyObject *
4022socket_getdefaulttimeout(PyObject *self)
4023{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004024 if (defaulttimeout < 0.0) {
4025 Py_INCREF(Py_None);
4026 return Py_None;
4027 }
4028 else
4029 return PyFloat_FromDouble(defaulttimeout);
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004030}
4031
4032PyDoc_STRVAR(getdefaulttimeout_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00004033"getdefaulttimeout() -> timeout\n\
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004034\n\
4035Returns the default timeout in floating seconds for new socket objects.\n\
4036A value of None indicates that new socket objects have no timeout.\n\
4037When the socket module is first imported, the default is None.");
4038
4039static PyObject *
4040socket_setdefaulttimeout(PyObject *self, PyObject *arg)
4041{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004042 double timeout;
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004043
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004044 if (arg == Py_None)
4045 timeout = -1.0;
4046 else {
4047 timeout = PyFloat_AsDouble(arg);
4048 if (timeout < 0.0) {
4049 if (!PyErr_Occurred())
4050 PyErr_SetString(PyExc_ValueError,
4051 "Timeout value out of range");
4052 return NULL;
4053 }
4054 }
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004055
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004056 defaulttimeout = timeout;
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004057
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004058 Py_INCREF(Py_None);
4059 return Py_None;
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004060}
4061
4062PyDoc_STRVAR(setdefaulttimeout_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00004063"setdefaulttimeout(timeout)\n\
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004064\n\
4065Set the default timeout in floating seconds for new socket objects.\n\
4066A value of None indicates that new socket objects have no timeout.\n\
4067When the socket module is first imported, the default is None.");
4068
4069
Guido van Rossum30a685f1991-06-27 15:51:29 +00004070/* List of functions exported by this module. */
4071
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004072static PyMethodDef socket_methods[] = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004073 {"gethostbyname", socket_gethostbyname,
4074 METH_VARARGS, gethostbyname_doc},
4075 {"gethostbyname_ex", socket_gethostbyname_ex,
4076 METH_VARARGS, ghbn_ex_doc},
4077 {"gethostbyaddr", socket_gethostbyaddr,
4078 METH_VARARGS, gethostbyaddr_doc},
4079 {"gethostname", socket_gethostname,
4080 METH_NOARGS, gethostname_doc},
4081 {"getservbyname", socket_getservbyname,
4082 METH_VARARGS, getservbyname_doc},
4083 {"getservbyport", socket_getservbyport,
4084 METH_VARARGS, getservbyport_doc},
4085 {"getprotobyname", socket_getprotobyname,
4086 METH_VARARGS, getprotobyname_doc},
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00004087#ifndef NO_DUP
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004088 {"dup", socket_dup,
4089 METH_O, dup_doc},
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00004090#endif
Dave Cole331708b2004-08-09 04:51:41 +00004091#ifdef HAVE_SOCKETPAIR
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004092 {"socketpair", socket_socketpair,
4093 METH_VARARGS, socketpair_doc},
Dave Cole331708b2004-08-09 04:51:41 +00004094#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004095 {"ntohs", socket_ntohs,
4096 METH_VARARGS, ntohs_doc},
4097 {"ntohl", socket_ntohl,
4098 METH_O, ntohl_doc},
4099 {"htons", socket_htons,
4100 METH_VARARGS, htons_doc},
4101 {"htonl", socket_htonl,
4102 METH_O, htonl_doc},
4103 {"inet_aton", socket_inet_aton,
4104 METH_VARARGS, inet_aton_doc},
4105 {"inet_ntoa", socket_inet_ntoa,
4106 METH_VARARGS, inet_ntoa_doc},
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00004107#ifdef HAVE_INET_PTON
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004108 {"inet_pton", socket_inet_pton,
4109 METH_VARARGS, inet_pton_doc},
4110 {"inet_ntop", socket_inet_ntop,
4111 METH_VARARGS, inet_ntop_doc},
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00004112#endif
Giampaolo Rodolàccfb91c2010-08-17 15:30:23 +00004113 {"getaddrinfo", (PyCFunction)socket_getaddrinfo,
4114 METH_VARARGS | METH_KEYWORDS, getaddrinfo_doc},
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004115 {"getnameinfo", socket_getnameinfo,
4116 METH_VARARGS, getnameinfo_doc},
4117 {"getdefaulttimeout", (PyCFunction)socket_getdefaulttimeout,
4118 METH_NOARGS, getdefaulttimeout_doc},
4119 {"setdefaulttimeout", socket_setdefaulttimeout,
4120 METH_O, setdefaulttimeout_doc},
4121 {NULL, NULL} /* Sentinel */
Guido van Rossum6574b3e1991-06-25 21:36:08 +00004122};
4123
Guido van Rossum30a685f1991-06-27 15:51:29 +00004124
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004125#ifdef MS_WINDOWS
4126#define OS_INIT_DEFINED
4127
4128/* Additional initialization and cleanup for Windows */
Guido van Rossumbe32c891996-06-20 16:25:29 +00004129
4130static void
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004131os_cleanup(void)
Guido van Rossumbe32c891996-06-20 16:25:29 +00004132{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004133 WSACleanup();
Guido van Rossumbe32c891996-06-20 16:25:29 +00004134}
4135
4136static int
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004137os_init(void)
Guido van Rossumbe32c891996-06-20 16:25:29 +00004138{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004139 WSADATA WSAData;
4140 int ret;
4141 ret = WSAStartup(0x0101, &WSAData);
4142 switch (ret) {
4143 case 0: /* No error */
4144 Py_AtExit(os_cleanup);
4145 return 1; /* Success */
4146 case WSASYSNOTREADY:
4147 PyErr_SetString(PyExc_ImportError,
4148 "WSAStartup failed: network not ready");
4149 break;
4150 case WSAVERNOTSUPPORTED:
4151 case WSAEINVAL:
4152 PyErr_SetString(
4153 PyExc_ImportError,
4154 "WSAStartup failed: requested version not supported");
4155 break;
4156 default:
4157 PyErr_Format(PyExc_ImportError, "WSAStartup failed: error code %d", ret);
4158 break;
4159 }
4160 return 0; /* Failure */
Guido van Rossumbe32c891996-06-20 16:25:29 +00004161}
4162
Guido van Rossum8d665e61996-06-26 18:22:49 +00004163#endif /* MS_WINDOWS */
Guido van Rossumbe32c891996-06-20 16:25:29 +00004164
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004165
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004166#ifdef PYOS_OS2
4167#define OS_INIT_DEFINED
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004168
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004169/* Additional initialization for OS/2 */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004170
4171static int
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004172os_init(void)
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004173{
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004174#ifndef PYCC_GCC
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004175 int rc = sock_init();
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004176
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004177 if (rc == 0) {
4178 return 1; /* Success */
4179 }
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004180
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004181 PyErr_Format(PyExc_ImportError, "OS/2 TCP/IP Error# %d", sock_errno());
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004182
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004183 return 0; /* Failure */
Andrew MacIntyreba43e872002-03-03 03:03:52 +00004184#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004185 /* No need to initialise sockets with GCC/EMX */
4186 return 1; /* Success */
Andrew MacIntyreba43e872002-03-03 03:03:52 +00004187#endif
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004188}
4189
4190#endif /* PYOS_OS2 */
4191
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004192
4193#ifndef OS_INIT_DEFINED
4194static int
4195os_init(void)
4196{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004197 return 1; /* Success */
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004198}
4199#endif
4200
4201
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00004202/* C API table - always add new things to the end for binary
4203 compatibility. */
4204static
4205PySocketModule_APIObject PySocketModuleAPI =
4206{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004207 &sock_type,
4208 NULL
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00004209};
4210
Guido van Rossum3eede5a2002-06-07 02:08:35 +00004211
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004212/* Initialize the _socket module.
Guido van Rossum3eede5a2002-06-07 02:08:35 +00004213
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004214 This module is actually called "_socket", and there's a wrapper
Guido van Rossum7d0a8262007-05-21 23:13:11 +00004215 "socket.py" which implements some additional functionality.
4216 The import of "_socket" may fail with an ImportError exception if
4217 os-specific initialization fails. On Windows, this does WINSOCK
4218 initialization. When WINSOCK is initialized succesfully, a call to
4219 WSACleanup() is scheduled to be made at exit time.
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004220*/
Guido van Rossum3eede5a2002-06-07 02:08:35 +00004221
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004222PyDoc_STRVAR(socket_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00004223"Implementation module for socket operations.\n\
4224\n\
4225See the socket module for documentation.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00004226
Martin v. Löwis1a214512008-06-11 05:26:20 +00004227static struct PyModuleDef socketmodule = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004228 PyModuleDef_HEAD_INIT,
4229 PySocket_MODULE_NAME,
4230 socket_doc,
4231 -1,
4232 socket_methods,
4233 NULL,
4234 NULL,
4235 NULL,
4236 NULL
Martin v. Löwis1a214512008-06-11 05:26:20 +00004237};
4238
Mark Hammondfe51c6d2002-08-02 02:27:13 +00004239PyMODINIT_FUNC
Martin v. Löwis1a214512008-06-11 05:26:20 +00004240PyInit__socket(void)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00004241{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004242 PyObject *m, *has_ipv6;
Fred Drake4baedc12002-04-01 14:53:37 +00004243
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004244 if (!os_init())
4245 return NULL;
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004246
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004247 Py_TYPE(&sock_type) = &PyType_Type;
4248 m = PyModule_Create(&socketmodule);
4249 if (m == NULL)
4250 return NULL;
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004251
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004252 socket_error = PyErr_NewException("socket.error",
4253 PyExc_IOError, NULL);
4254 if (socket_error == NULL)
4255 return NULL;
4256 PySocketModuleAPI.error = socket_error;
4257 Py_INCREF(socket_error);
4258 PyModule_AddObject(m, "error", socket_error);
4259 socket_herror = PyErr_NewException("socket.herror",
4260 socket_error, NULL);
4261 if (socket_herror == NULL)
4262 return NULL;
4263 Py_INCREF(socket_herror);
4264 PyModule_AddObject(m, "herror", socket_herror);
4265 socket_gaierror = PyErr_NewException("socket.gaierror", socket_error,
4266 NULL);
4267 if (socket_gaierror == NULL)
4268 return NULL;
4269 Py_INCREF(socket_gaierror);
4270 PyModule_AddObject(m, "gaierror", socket_gaierror);
4271 socket_timeout = PyErr_NewException("socket.timeout",
4272 socket_error, NULL);
4273 if (socket_timeout == NULL)
4274 return NULL;
4275 Py_INCREF(socket_timeout);
4276 PyModule_AddObject(m, "timeout", socket_timeout);
4277 Py_INCREF((PyObject *)&sock_type);
4278 if (PyModule_AddObject(m, "SocketType",
4279 (PyObject *)&sock_type) != 0)
4280 return NULL;
4281 Py_INCREF((PyObject *)&sock_type);
4282 if (PyModule_AddObject(m, "socket",
4283 (PyObject *)&sock_type) != 0)
4284 return NULL;
Guido van Rossum09be4091999-08-09 14:40:40 +00004285
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00004286#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004287 has_ipv6 = Py_True;
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00004288#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004289 has_ipv6 = Py_False;
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00004290#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004291 Py_INCREF(has_ipv6);
4292 PyModule_AddObject(m, "has_ipv6", has_ipv6);
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00004293
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004294 /* Export C API */
4295 if (PyModule_AddObject(m, PySocket_CAPI_NAME,
4296 PyCapsule_New(&PySocketModuleAPI, PySocket_CAPSULE_NAME, NULL)
4297 ) != 0)
4298 return NULL;
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00004299
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004300 /* Address families (we only support AF_INET and AF_UNIX) */
Guido van Rossum09be4091999-08-09 14:40:40 +00004301#ifdef AF_UNSPEC
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004302 PyModule_AddIntConstant(m, "AF_UNSPEC", AF_UNSPEC);
Guido van Rossum09be4091999-08-09 14:40:40 +00004303#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004304 PyModule_AddIntConstant(m, "AF_INET", AF_INET);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004305#ifdef AF_INET6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004306 PyModule_AddIntConstant(m, "AF_INET6", AF_INET6);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004307#endif /* AF_INET6 */
Andrew MacIntyred12dfbb2004-04-04 07:13:49 +00004308#if defined(AF_UNIX)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004309 PyModule_AddIntConstant(m, "AF_UNIX", AF_UNIX);
Guido van Rossumb6775db1994-08-01 11:34:53 +00004310#endif /* AF_UNIX */
Guido van Rossum09be4091999-08-09 14:40:40 +00004311#ifdef AF_AX25
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004312 /* Amateur Radio AX.25 */
4313 PyModule_AddIntConstant(m, "AF_AX25", AF_AX25);
Guido van Rossum09be4091999-08-09 14:40:40 +00004314#endif
4315#ifdef AF_IPX
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004316 PyModule_AddIntConstant(m, "AF_IPX", AF_IPX); /* Novell IPX */
Guido van Rossum09be4091999-08-09 14:40:40 +00004317#endif
4318#ifdef AF_APPLETALK
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004319 /* Appletalk DDP */
4320 PyModule_AddIntConstant(m, "AF_APPLETALK", AF_APPLETALK);
Guido van Rossum09be4091999-08-09 14:40:40 +00004321#endif
4322#ifdef AF_NETROM
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004323 /* Amateur radio NetROM */
4324 PyModule_AddIntConstant(m, "AF_NETROM", AF_NETROM);
Guido van Rossum09be4091999-08-09 14:40:40 +00004325#endif
4326#ifdef AF_BRIDGE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004327 /* Multiprotocol bridge */
4328 PyModule_AddIntConstant(m, "AF_BRIDGE", AF_BRIDGE);
Guido van Rossum09be4091999-08-09 14:40:40 +00004329#endif
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004330#ifdef AF_ATMPVC
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004331 /* ATM PVCs */
4332 PyModule_AddIntConstant(m, "AF_ATMPVC", AF_ATMPVC);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004333#endif
Guido van Rossum09be4091999-08-09 14:40:40 +00004334#ifdef AF_AAL5
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004335 /* Reserved for Werner's ATM */
4336 PyModule_AddIntConstant(m, "AF_AAL5", AF_AAL5);
Guido van Rossum09be4091999-08-09 14:40:40 +00004337#endif
4338#ifdef AF_X25
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004339 /* Reserved for X.25 project */
4340 PyModule_AddIntConstant(m, "AF_X25", AF_X25);
Guido van Rossum09be4091999-08-09 14:40:40 +00004341#endif
4342#ifdef AF_INET6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004343 PyModule_AddIntConstant(m, "AF_INET6", AF_INET6); /* IP version 6 */
Guido van Rossum09be4091999-08-09 14:40:40 +00004344#endif
4345#ifdef AF_ROSE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004346 /* Amateur Radio X.25 PLP */
4347 PyModule_AddIntConstant(m, "AF_ROSE", AF_ROSE);
Guido van Rossum09be4091999-08-09 14:40:40 +00004348#endif
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004349#ifdef AF_DECnet
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004350 /* Reserved for DECnet project */
4351 PyModule_AddIntConstant(m, "AF_DECnet", AF_DECnet);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004352#endif
4353#ifdef AF_NETBEUI
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004354 /* Reserved for 802.2LLC project */
4355 PyModule_AddIntConstant(m, "AF_NETBEUI", AF_NETBEUI);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004356#endif
4357#ifdef AF_SECURITY
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004358 /* Security callback pseudo AF */
4359 PyModule_AddIntConstant(m, "AF_SECURITY", AF_SECURITY);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004360#endif
4361#ifdef AF_KEY
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004362 /* PF_KEY key management API */
4363 PyModule_AddIntConstant(m, "AF_KEY", AF_KEY);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004364#endif
4365#ifdef AF_NETLINK
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004366 /* */
4367 PyModule_AddIntConstant(m, "AF_NETLINK", AF_NETLINK);
4368 PyModule_AddIntConstant(m, "NETLINK_ROUTE", NETLINK_ROUTE);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004369#ifdef NETLINK_SKIP
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004370 PyModule_AddIntConstant(m, "NETLINK_SKIP", NETLINK_SKIP);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004371#endif
4372#ifdef NETLINK_W1
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004373 PyModule_AddIntConstant(m, "NETLINK_W1", NETLINK_W1);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004374#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004375 PyModule_AddIntConstant(m, "NETLINK_USERSOCK", NETLINK_USERSOCK);
4376 PyModule_AddIntConstant(m, "NETLINK_FIREWALL", NETLINK_FIREWALL);
Guido van Rossum668a94a2006-02-21 01:07:27 +00004377#ifdef NETLINK_TCPDIAG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004378 PyModule_AddIntConstant(m, "NETLINK_TCPDIAG", NETLINK_TCPDIAG);
Guido van Rossum668a94a2006-02-21 01:07:27 +00004379#endif
4380#ifdef NETLINK_NFLOG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004381 PyModule_AddIntConstant(m, "NETLINK_NFLOG", NETLINK_NFLOG);
Guido van Rossum668a94a2006-02-21 01:07:27 +00004382#endif
Neal Norwitz65851662006-01-16 04:31:40 +00004383#ifdef NETLINK_XFRM
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004384 PyModule_AddIntConstant(m, "NETLINK_XFRM", NETLINK_XFRM);
Neal Norwitz65851662006-01-16 04:31:40 +00004385#endif
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004386#ifdef NETLINK_ARPD
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004387 PyModule_AddIntConstant(m, "NETLINK_ARPD", NETLINK_ARPD);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004388#endif
4389#ifdef NETLINK_ROUTE6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004390 PyModule_AddIntConstant(m, "NETLINK_ROUTE6", NETLINK_ROUTE6);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004391#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004392 PyModule_AddIntConstant(m, "NETLINK_IP6_FW", NETLINK_IP6_FW);
Thomas Wouterscf297e42007-02-23 15:07:44 +00004393#ifdef NETLINK_DNRTMSG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004394 PyModule_AddIntConstant(m, "NETLINK_DNRTMSG", NETLINK_DNRTMSG);
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00004395#endif
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004396#ifdef NETLINK_TAPBASE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004397 PyModule_AddIntConstant(m, "NETLINK_TAPBASE", NETLINK_TAPBASE);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004398#endif
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004399#endif /* AF_NETLINK */
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004400#ifdef AF_ROUTE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004401 /* Alias to emulate 4.4BSD */
4402 PyModule_AddIntConstant(m, "AF_ROUTE", AF_ROUTE);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004403#endif
4404#ifdef AF_ASH
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004405 /* Ash */
4406 PyModule_AddIntConstant(m, "AF_ASH", AF_ASH);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004407#endif
4408#ifdef AF_ECONET
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004409 /* Acorn Econet */
4410 PyModule_AddIntConstant(m, "AF_ECONET", AF_ECONET);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004411#endif
4412#ifdef AF_ATMSVC
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004413 /* ATM SVCs */
4414 PyModule_AddIntConstant(m, "AF_ATMSVC", AF_ATMSVC);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004415#endif
4416#ifdef AF_SNA
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004417 /* Linux SNA Project (nutters!) */
4418 PyModule_AddIntConstant(m, "AF_SNA", AF_SNA);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004419#endif
4420#ifdef AF_IRDA
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004421 /* IRDA sockets */
4422 PyModule_AddIntConstant(m, "AF_IRDA", AF_IRDA);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004423#endif
4424#ifdef AF_PPPOX
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004425 /* PPPoX sockets */
4426 PyModule_AddIntConstant(m, "AF_PPPOX", AF_PPPOX);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004427#endif
4428#ifdef AF_WANPIPE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004429 /* Wanpipe API Sockets */
4430 PyModule_AddIntConstant(m, "AF_WANPIPE", AF_WANPIPE);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004431#endif
4432#ifdef AF_LLC
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004433 /* Linux LLC */
4434 PyModule_AddIntConstant(m, "AF_LLC", AF_LLC);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004435#endif
Martin v. Löwis12af0482004-01-31 12:34:17 +00004436
Hye-Shik Chang81268602004-02-02 06:05:24 +00004437#ifdef USE_BLUETOOTH
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004438 PyModule_AddIntConstant(m, "AF_BLUETOOTH", AF_BLUETOOTH);
4439 PyModule_AddIntConstant(m, "BTPROTO_L2CAP", BTPROTO_L2CAP);
4440 PyModule_AddIntConstant(m, "BTPROTO_HCI", BTPROTO_HCI);
4441 PyModule_AddIntConstant(m, "SOL_HCI", SOL_HCI);
4442 PyModule_AddIntConstant(m, "HCI_FILTER", HCI_FILTER);
Hye-Shik Chang81268602004-02-02 06:05:24 +00004443#if !defined(__FreeBSD__)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004444 PyModule_AddIntConstant(m, "HCI_TIME_STAMP", HCI_TIME_STAMP);
4445 PyModule_AddIntConstant(m, "HCI_DATA_DIR", HCI_DATA_DIR);
4446 PyModule_AddIntConstant(m, "BTPROTO_SCO", BTPROTO_SCO);
Hye-Shik Chang81268602004-02-02 06:05:24 +00004447#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004448 PyModule_AddIntConstant(m, "BTPROTO_RFCOMM", BTPROTO_RFCOMM);
4449 PyModule_AddStringConstant(m, "BDADDR_ANY", "00:00:00:00:00:00");
4450 PyModule_AddStringConstant(m, "BDADDR_LOCAL", "00:00:00:FF:FF:FF");
Martin v. Löwis12af0482004-01-31 12:34:17 +00004451#endif
4452
Martin v. Löwis1ba3fd52001-08-10 20:29:40 +00004453#ifdef HAVE_NETPACKET_PACKET_H
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004454 PyModule_AddIntConstant(m, "AF_PACKET", AF_PACKET);
4455 PyModule_AddIntConstant(m, "PF_PACKET", PF_PACKET);
4456 PyModule_AddIntConstant(m, "PACKET_HOST", PACKET_HOST);
4457 PyModule_AddIntConstant(m, "PACKET_BROADCAST", PACKET_BROADCAST);
4458 PyModule_AddIntConstant(m, "PACKET_MULTICAST", PACKET_MULTICAST);
4459 PyModule_AddIntConstant(m, "PACKET_OTHERHOST", PACKET_OTHERHOST);
4460 PyModule_AddIntConstant(m, "PACKET_OUTGOING", PACKET_OUTGOING);
4461 PyModule_AddIntConstant(m, "PACKET_LOOPBACK", PACKET_LOOPBACK);
4462 PyModule_AddIntConstant(m, "PACKET_FASTROUTE", PACKET_FASTROUTE);
Guido van Rossum48a680c2001-03-02 06:34:14 +00004463#endif
Guido van Rossum09be4091999-08-09 14:40:40 +00004464
Christian Heimes043d6f62008-01-07 17:19:16 +00004465#ifdef HAVE_LINUX_TIPC_H
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004466 PyModule_AddIntConstant(m, "AF_TIPC", AF_TIPC);
Christian Heimes043d6f62008-01-07 17:19:16 +00004467
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004468 /* for addresses */
4469 PyModule_AddIntConstant(m, "TIPC_ADDR_NAMESEQ", TIPC_ADDR_NAMESEQ);
4470 PyModule_AddIntConstant(m, "TIPC_ADDR_NAME", TIPC_ADDR_NAME);
4471 PyModule_AddIntConstant(m, "TIPC_ADDR_ID", TIPC_ADDR_ID);
Christian Heimes043d6f62008-01-07 17:19:16 +00004472
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004473 PyModule_AddIntConstant(m, "TIPC_ZONE_SCOPE", TIPC_ZONE_SCOPE);
4474 PyModule_AddIntConstant(m, "TIPC_CLUSTER_SCOPE", TIPC_CLUSTER_SCOPE);
4475 PyModule_AddIntConstant(m, "TIPC_NODE_SCOPE", TIPC_NODE_SCOPE);
Christian Heimes043d6f62008-01-07 17:19:16 +00004476
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004477 /* for setsockopt() */
4478 PyModule_AddIntConstant(m, "SOL_TIPC", SOL_TIPC);
4479 PyModule_AddIntConstant(m, "TIPC_IMPORTANCE", TIPC_IMPORTANCE);
4480 PyModule_AddIntConstant(m, "TIPC_SRC_DROPPABLE", TIPC_SRC_DROPPABLE);
4481 PyModule_AddIntConstant(m, "TIPC_DEST_DROPPABLE",
4482 TIPC_DEST_DROPPABLE);
4483 PyModule_AddIntConstant(m, "TIPC_CONN_TIMEOUT", TIPC_CONN_TIMEOUT);
Christian Heimes043d6f62008-01-07 17:19:16 +00004484
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004485 PyModule_AddIntConstant(m, "TIPC_LOW_IMPORTANCE",
4486 TIPC_LOW_IMPORTANCE);
4487 PyModule_AddIntConstant(m, "TIPC_MEDIUM_IMPORTANCE",
4488 TIPC_MEDIUM_IMPORTANCE);
4489 PyModule_AddIntConstant(m, "TIPC_HIGH_IMPORTANCE",
4490 TIPC_HIGH_IMPORTANCE);
4491 PyModule_AddIntConstant(m, "TIPC_CRITICAL_IMPORTANCE",
4492 TIPC_CRITICAL_IMPORTANCE);
Christian Heimes043d6f62008-01-07 17:19:16 +00004493
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004494 /* for subscriptions */
4495 PyModule_AddIntConstant(m, "TIPC_SUB_PORTS", TIPC_SUB_PORTS);
4496 PyModule_AddIntConstant(m, "TIPC_SUB_SERVICE", TIPC_SUB_SERVICE);
Christian Heimes25bb7832008-01-11 16:17:00 +00004497#ifdef TIPC_SUB_CANCEL
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004498 /* doesn't seem to be available everywhere */
4499 PyModule_AddIntConstant(m, "TIPC_SUB_CANCEL", TIPC_SUB_CANCEL);
Christian Heimes25bb7832008-01-11 16:17:00 +00004500#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004501 PyModule_AddIntConstant(m, "TIPC_WAIT_FOREVER", TIPC_WAIT_FOREVER);
4502 PyModule_AddIntConstant(m, "TIPC_PUBLISHED", TIPC_PUBLISHED);
4503 PyModule_AddIntConstant(m, "TIPC_WITHDRAWN", TIPC_WITHDRAWN);
4504 PyModule_AddIntConstant(m, "TIPC_SUBSCR_TIMEOUT", TIPC_SUBSCR_TIMEOUT);
4505 PyModule_AddIntConstant(m, "TIPC_CFG_SRV", TIPC_CFG_SRV);
4506 PyModule_AddIntConstant(m, "TIPC_TOP_SRV", TIPC_TOP_SRV);
Christian Heimes043d6f62008-01-07 17:19:16 +00004507#endif
4508
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004509 /* Socket types */
4510 PyModule_AddIntConstant(m, "SOCK_STREAM", SOCK_STREAM);
4511 PyModule_AddIntConstant(m, "SOCK_DGRAM", SOCK_DGRAM);
Guido van Rossumbcc20741998-08-04 22:53:56 +00004512/* We have incomplete socket support. */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004513 PyModule_AddIntConstant(m, "SOCK_RAW", SOCK_RAW);
4514 PyModule_AddIntConstant(m, "SOCK_SEQPACKET", SOCK_SEQPACKET);
Martin v. Löwiscf8f47e2002-12-11 13:10:57 +00004515#if defined(SOCK_RDM)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004516 PyModule_AddIntConstant(m, "SOCK_RDM", SOCK_RDM);
Guido van Rossumbcc20741998-08-04 22:53:56 +00004517#endif
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004518
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004519#ifdef SO_DEBUG
4520 PyModule_AddIntConstant(m, "SO_DEBUG", SO_DEBUG);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004521#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004522#ifdef SO_ACCEPTCONN
4523 PyModule_AddIntConstant(m, "SO_ACCEPTCONN", SO_ACCEPTCONN);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004524#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004525#ifdef SO_REUSEADDR
4526 PyModule_AddIntConstant(m, "SO_REUSEADDR", SO_REUSEADDR);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004527#endif
Andrew M. Kuchling42851ab2004-07-10 14:19:21 +00004528#ifdef SO_EXCLUSIVEADDRUSE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004529 PyModule_AddIntConstant(m, "SO_EXCLUSIVEADDRUSE", SO_EXCLUSIVEADDRUSE);
Andrew M. Kuchling42851ab2004-07-10 14:19:21 +00004530#endif
4531
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004532#ifdef SO_KEEPALIVE
4533 PyModule_AddIntConstant(m, "SO_KEEPALIVE", SO_KEEPALIVE);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004534#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004535#ifdef SO_DONTROUTE
4536 PyModule_AddIntConstant(m, "SO_DONTROUTE", SO_DONTROUTE);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004537#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004538#ifdef SO_BROADCAST
4539 PyModule_AddIntConstant(m, "SO_BROADCAST", SO_BROADCAST);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004540#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004541#ifdef SO_USELOOPBACK
4542 PyModule_AddIntConstant(m, "SO_USELOOPBACK", SO_USELOOPBACK);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004543#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004544#ifdef SO_LINGER
4545 PyModule_AddIntConstant(m, "SO_LINGER", SO_LINGER);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004546#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004547#ifdef SO_OOBINLINE
4548 PyModule_AddIntConstant(m, "SO_OOBINLINE", SO_OOBINLINE);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004549#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004550#ifdef SO_REUSEPORT
4551 PyModule_AddIntConstant(m, "SO_REUSEPORT", SO_REUSEPORT);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004552#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004553#ifdef SO_SNDBUF
4554 PyModule_AddIntConstant(m, "SO_SNDBUF", SO_SNDBUF);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004555#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004556#ifdef SO_RCVBUF
4557 PyModule_AddIntConstant(m, "SO_RCVBUF", SO_RCVBUF);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004558#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004559#ifdef SO_SNDLOWAT
4560 PyModule_AddIntConstant(m, "SO_SNDLOWAT", SO_SNDLOWAT);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004561#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004562#ifdef SO_RCVLOWAT
4563 PyModule_AddIntConstant(m, "SO_RCVLOWAT", SO_RCVLOWAT);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004564#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004565#ifdef SO_SNDTIMEO
4566 PyModule_AddIntConstant(m, "SO_SNDTIMEO", SO_SNDTIMEO);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004567#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004568#ifdef SO_RCVTIMEO
4569 PyModule_AddIntConstant(m, "SO_RCVTIMEO", SO_RCVTIMEO);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004570#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004571#ifdef SO_ERROR
4572 PyModule_AddIntConstant(m, "SO_ERROR", SO_ERROR);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004573#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004574#ifdef SO_TYPE
4575 PyModule_AddIntConstant(m, "SO_TYPE", SO_TYPE);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004576#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004577#ifdef SO_SETFIB
4578 PyModule_AddIntConstant(m, "SO_SETFIB", SO_SETFIB);
Larry Hastingsf0f37952010-04-02 11:47:10 +00004579#endif
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004580
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004581 /* Maximum number of connections for "listen" */
4582#ifdef SOMAXCONN
4583 PyModule_AddIntConstant(m, "SOMAXCONN", SOMAXCONN);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004584#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004585 PyModule_AddIntConstant(m, "SOMAXCONN", 5); /* Common value */
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004586#endif
4587
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004588 /* Flags for send, recv */
4589#ifdef MSG_OOB
4590 PyModule_AddIntConstant(m, "MSG_OOB", MSG_OOB);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004591#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004592#ifdef MSG_PEEK
4593 PyModule_AddIntConstant(m, "MSG_PEEK", MSG_PEEK);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004594#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004595#ifdef MSG_DONTROUTE
4596 PyModule_AddIntConstant(m, "MSG_DONTROUTE", MSG_DONTROUTE);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004597#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004598#ifdef MSG_DONTWAIT
4599 PyModule_AddIntConstant(m, "MSG_DONTWAIT", MSG_DONTWAIT);
Guido van Rossum2c8bcb82000-04-25 21:34:53 +00004600#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004601#ifdef MSG_EOR
4602 PyModule_AddIntConstant(m, "MSG_EOR", MSG_EOR);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004603#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004604#ifdef MSG_TRUNC
4605 PyModule_AddIntConstant(m, "MSG_TRUNC", MSG_TRUNC);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004606#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004607#ifdef MSG_CTRUNC
4608 PyModule_AddIntConstant(m, "MSG_CTRUNC", MSG_CTRUNC);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004609#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004610#ifdef MSG_WAITALL
4611 PyModule_AddIntConstant(m, "MSG_WAITALL", MSG_WAITALL);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004612#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004613#ifdef MSG_BTAG
4614 PyModule_AddIntConstant(m, "MSG_BTAG", MSG_BTAG);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004615#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004616#ifdef MSG_ETAG
4617 PyModule_AddIntConstant(m, "MSG_ETAG", MSG_ETAG);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004618#endif
4619
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004620 /* Protocol level and numbers, usable for [gs]etsockopt */
4621#ifdef SOL_SOCKET
4622 PyModule_AddIntConstant(m, "SOL_SOCKET", SOL_SOCKET);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004623#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004624#ifdef SOL_IP
4625 PyModule_AddIntConstant(m, "SOL_IP", SOL_IP);
Guido van Rossum09be4091999-08-09 14:40:40 +00004626#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004627 PyModule_AddIntConstant(m, "SOL_IP", 0);
Guido van Rossum09be4091999-08-09 14:40:40 +00004628#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004629#ifdef SOL_IPX
4630 PyModule_AddIntConstant(m, "SOL_IPX", SOL_IPX);
Guido van Rossum09be4091999-08-09 14:40:40 +00004631#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004632#ifdef SOL_AX25
4633 PyModule_AddIntConstant(m, "SOL_AX25", SOL_AX25);
Guido van Rossum09be4091999-08-09 14:40:40 +00004634#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004635#ifdef SOL_ATALK
4636 PyModule_AddIntConstant(m, "SOL_ATALK", SOL_ATALK);
Guido van Rossum09be4091999-08-09 14:40:40 +00004637#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004638#ifdef SOL_NETROM
4639 PyModule_AddIntConstant(m, "SOL_NETROM", SOL_NETROM);
Guido van Rossum09be4091999-08-09 14:40:40 +00004640#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004641#ifdef SOL_ROSE
4642 PyModule_AddIntConstant(m, "SOL_ROSE", SOL_ROSE);
Guido van Rossum09be4091999-08-09 14:40:40 +00004643#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004644#ifdef SOL_TCP
4645 PyModule_AddIntConstant(m, "SOL_TCP", SOL_TCP);
Guido van Rossum09be4091999-08-09 14:40:40 +00004646#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004647 PyModule_AddIntConstant(m, "SOL_TCP", 6);
Guido van Rossum09be4091999-08-09 14:40:40 +00004648#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004649#ifdef SOL_UDP
4650 PyModule_AddIntConstant(m, "SOL_UDP", SOL_UDP);
Guido van Rossum09be4091999-08-09 14:40:40 +00004651#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004652 PyModule_AddIntConstant(m, "SOL_UDP", 17);
Guido van Rossum09be4091999-08-09 14:40:40 +00004653#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004654#ifdef IPPROTO_IP
4655 PyModule_AddIntConstant(m, "IPPROTO_IP", IPPROTO_IP);
Guido van Rossum578de301998-05-28 20:18:18 +00004656#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004657 PyModule_AddIntConstant(m, "IPPROTO_IP", 0);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004658#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004659#ifdef IPPROTO_HOPOPTS
4660 PyModule_AddIntConstant(m, "IPPROTO_HOPOPTS", IPPROTO_HOPOPTS);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004661#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004662#ifdef IPPROTO_ICMP
4663 PyModule_AddIntConstant(m, "IPPROTO_ICMP", IPPROTO_ICMP);
Guido van Rossum578de301998-05-28 20:18:18 +00004664#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004665 PyModule_AddIntConstant(m, "IPPROTO_ICMP", 1);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004666#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004667#ifdef IPPROTO_IGMP
4668 PyModule_AddIntConstant(m, "IPPROTO_IGMP", IPPROTO_IGMP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004669#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004670#ifdef IPPROTO_GGP
4671 PyModule_AddIntConstant(m, "IPPROTO_GGP", IPPROTO_GGP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004672#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004673#ifdef IPPROTO_IPV4
4674 PyModule_AddIntConstant(m, "IPPROTO_IPV4", IPPROTO_IPV4);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004675#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004676#ifdef IPPROTO_IPV6
4677 PyModule_AddIntConstant(m, "IPPROTO_IPV6", IPPROTO_IPV6);
Martin v. Löwisa0f17342003-10-03 13:56:20 +00004678#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004679#ifdef IPPROTO_IPIP
4680 PyModule_AddIntConstant(m, "IPPROTO_IPIP", IPPROTO_IPIP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004681#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004682#ifdef IPPROTO_TCP
4683 PyModule_AddIntConstant(m, "IPPROTO_TCP", IPPROTO_TCP);
Guido van Rossum578de301998-05-28 20:18:18 +00004684#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004685 PyModule_AddIntConstant(m, "IPPROTO_TCP", 6);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004686#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004687#ifdef IPPROTO_EGP
4688 PyModule_AddIntConstant(m, "IPPROTO_EGP", IPPROTO_EGP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004689#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004690#ifdef IPPROTO_PUP
4691 PyModule_AddIntConstant(m, "IPPROTO_PUP", IPPROTO_PUP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004692#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004693#ifdef IPPROTO_UDP
4694 PyModule_AddIntConstant(m, "IPPROTO_UDP", IPPROTO_UDP);
Guido van Rossum578de301998-05-28 20:18:18 +00004695#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004696 PyModule_AddIntConstant(m, "IPPROTO_UDP", 17);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004697#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004698#ifdef IPPROTO_IDP
4699 PyModule_AddIntConstant(m, "IPPROTO_IDP", IPPROTO_IDP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004700#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004701#ifdef IPPROTO_HELLO
4702 PyModule_AddIntConstant(m, "IPPROTO_HELLO", IPPROTO_HELLO);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004703#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004704#ifdef IPPROTO_ND
4705 PyModule_AddIntConstant(m, "IPPROTO_ND", IPPROTO_ND);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004706#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004707#ifdef IPPROTO_TP
4708 PyModule_AddIntConstant(m, "IPPROTO_TP", IPPROTO_TP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004709#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004710#ifdef IPPROTO_IPV6
4711 PyModule_AddIntConstant(m, "IPPROTO_IPV6", IPPROTO_IPV6);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004712#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004713#ifdef IPPROTO_ROUTING
4714 PyModule_AddIntConstant(m, "IPPROTO_ROUTING", IPPROTO_ROUTING);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004715#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004716#ifdef IPPROTO_FRAGMENT
4717 PyModule_AddIntConstant(m, "IPPROTO_FRAGMENT", IPPROTO_FRAGMENT);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004718#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004719#ifdef IPPROTO_RSVP
4720 PyModule_AddIntConstant(m, "IPPROTO_RSVP", IPPROTO_RSVP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004721#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004722#ifdef IPPROTO_GRE
4723 PyModule_AddIntConstant(m, "IPPROTO_GRE", IPPROTO_GRE);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004724#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004725#ifdef IPPROTO_ESP
4726 PyModule_AddIntConstant(m, "IPPROTO_ESP", IPPROTO_ESP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004727#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004728#ifdef IPPROTO_AH
4729 PyModule_AddIntConstant(m, "IPPROTO_AH", IPPROTO_AH);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004730#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004731#ifdef IPPROTO_MOBILE
4732 PyModule_AddIntConstant(m, "IPPROTO_MOBILE", IPPROTO_MOBILE);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004733#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004734#ifdef IPPROTO_ICMPV6
4735 PyModule_AddIntConstant(m, "IPPROTO_ICMPV6", IPPROTO_ICMPV6);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004736#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004737#ifdef IPPROTO_NONE
4738 PyModule_AddIntConstant(m, "IPPROTO_NONE", IPPROTO_NONE);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004739#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004740#ifdef IPPROTO_DSTOPTS
4741 PyModule_AddIntConstant(m, "IPPROTO_DSTOPTS", IPPROTO_DSTOPTS);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004742#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004743#ifdef IPPROTO_XTP
4744 PyModule_AddIntConstant(m, "IPPROTO_XTP", IPPROTO_XTP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004745#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004746#ifdef IPPROTO_EON
4747 PyModule_AddIntConstant(m, "IPPROTO_EON", IPPROTO_EON);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004748#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004749#ifdef IPPROTO_PIM
4750 PyModule_AddIntConstant(m, "IPPROTO_PIM", IPPROTO_PIM);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004751#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004752#ifdef IPPROTO_IPCOMP
4753 PyModule_AddIntConstant(m, "IPPROTO_IPCOMP", IPPROTO_IPCOMP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004754#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004755#ifdef IPPROTO_VRRP
4756 PyModule_AddIntConstant(m, "IPPROTO_VRRP", IPPROTO_VRRP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004757#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004758#ifdef IPPROTO_BIP
4759 PyModule_AddIntConstant(m, "IPPROTO_BIP", IPPROTO_BIP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004760#endif
4761/**/
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004762#ifdef IPPROTO_RAW
4763 PyModule_AddIntConstant(m, "IPPROTO_RAW", IPPROTO_RAW);
Guido van Rossum578de301998-05-28 20:18:18 +00004764#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004765 PyModule_AddIntConstant(m, "IPPROTO_RAW", 255);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004766#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004767#ifdef IPPROTO_MAX
4768 PyModule_AddIntConstant(m, "IPPROTO_MAX", IPPROTO_MAX);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004769#endif
4770
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004771 /* Some port configuration */
4772#ifdef IPPORT_RESERVED
4773 PyModule_AddIntConstant(m, "IPPORT_RESERVED", IPPORT_RESERVED);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004774#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004775 PyModule_AddIntConstant(m, "IPPORT_RESERVED", 1024);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004776#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004777#ifdef IPPORT_USERRESERVED
4778 PyModule_AddIntConstant(m, "IPPORT_USERRESERVED", IPPORT_USERRESERVED);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004779#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004780 PyModule_AddIntConstant(m, "IPPORT_USERRESERVED", 5000);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004781#endif
4782
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004783 /* Some reserved IP v.4 addresses */
4784#ifdef INADDR_ANY
4785 PyModule_AddIntConstant(m, "INADDR_ANY", INADDR_ANY);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004786#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004787 PyModule_AddIntConstant(m, "INADDR_ANY", 0x00000000);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004788#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004789#ifdef INADDR_BROADCAST
4790 PyModule_AddIntConstant(m, "INADDR_BROADCAST", INADDR_BROADCAST);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004791#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004792 PyModule_AddIntConstant(m, "INADDR_BROADCAST", 0xffffffff);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004793#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004794#ifdef INADDR_LOOPBACK
4795 PyModule_AddIntConstant(m, "INADDR_LOOPBACK", INADDR_LOOPBACK);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004796#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004797 PyModule_AddIntConstant(m, "INADDR_LOOPBACK", 0x7F000001);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004798#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004799#ifdef INADDR_UNSPEC_GROUP
4800 PyModule_AddIntConstant(m, "INADDR_UNSPEC_GROUP", INADDR_UNSPEC_GROUP);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004801#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004802 PyModule_AddIntConstant(m, "INADDR_UNSPEC_GROUP", 0xe0000000);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004803#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004804#ifdef INADDR_ALLHOSTS_GROUP
4805 PyModule_AddIntConstant(m, "INADDR_ALLHOSTS_GROUP",
4806 INADDR_ALLHOSTS_GROUP);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004807#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004808 PyModule_AddIntConstant(m, "INADDR_ALLHOSTS_GROUP", 0xe0000001);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004809#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004810#ifdef INADDR_MAX_LOCAL_GROUP
4811 PyModule_AddIntConstant(m, "INADDR_MAX_LOCAL_GROUP",
4812 INADDR_MAX_LOCAL_GROUP);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004813#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004814 PyModule_AddIntConstant(m, "INADDR_MAX_LOCAL_GROUP", 0xe00000ff);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004815#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004816#ifdef INADDR_NONE
4817 PyModule_AddIntConstant(m, "INADDR_NONE", INADDR_NONE);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004818#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004819 PyModule_AddIntConstant(m, "INADDR_NONE", 0xffffffff);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004820#endif
4821
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004822 /* IPv4 [gs]etsockopt options */
4823#ifdef IP_OPTIONS
4824 PyModule_AddIntConstant(m, "IP_OPTIONS", IP_OPTIONS);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004825#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004826#ifdef IP_HDRINCL
4827 PyModule_AddIntConstant(m, "IP_HDRINCL", IP_HDRINCL);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004828#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004829#ifdef IP_TOS
4830 PyModule_AddIntConstant(m, "IP_TOS", IP_TOS);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004831#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004832#ifdef IP_TTL
4833 PyModule_AddIntConstant(m, "IP_TTL", IP_TTL);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004834#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004835#ifdef IP_RECVOPTS
4836 PyModule_AddIntConstant(m, "IP_RECVOPTS", IP_RECVOPTS);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004837#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004838#ifdef IP_RECVRETOPTS
4839 PyModule_AddIntConstant(m, "IP_RECVRETOPTS", IP_RECVRETOPTS);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004840#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004841#ifdef IP_RECVDSTADDR
4842 PyModule_AddIntConstant(m, "IP_RECVDSTADDR", IP_RECVDSTADDR);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004843#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004844#ifdef IP_RETOPTS
4845 PyModule_AddIntConstant(m, "IP_RETOPTS", IP_RETOPTS);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004846#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004847#ifdef IP_MULTICAST_IF
4848 PyModule_AddIntConstant(m, "IP_MULTICAST_IF", IP_MULTICAST_IF);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004849#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004850#ifdef IP_MULTICAST_TTL
4851 PyModule_AddIntConstant(m, "IP_MULTICAST_TTL", IP_MULTICAST_TTL);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004852#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004853#ifdef IP_MULTICAST_LOOP
4854 PyModule_AddIntConstant(m, "IP_MULTICAST_LOOP", IP_MULTICAST_LOOP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004855#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004856#ifdef IP_ADD_MEMBERSHIP
4857 PyModule_AddIntConstant(m, "IP_ADD_MEMBERSHIP", IP_ADD_MEMBERSHIP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004858#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004859#ifdef IP_DROP_MEMBERSHIP
4860 PyModule_AddIntConstant(m, "IP_DROP_MEMBERSHIP", IP_DROP_MEMBERSHIP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004861#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004862#ifdef IP_DEFAULT_MULTICAST_TTL
4863 PyModule_AddIntConstant(m, "IP_DEFAULT_MULTICAST_TTL",
4864 IP_DEFAULT_MULTICAST_TTL);
Guido van Rossum09be4091999-08-09 14:40:40 +00004865#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004866#ifdef IP_DEFAULT_MULTICAST_LOOP
4867 PyModule_AddIntConstant(m, "IP_DEFAULT_MULTICAST_LOOP",
4868 IP_DEFAULT_MULTICAST_LOOP);
Guido van Rossum09be4091999-08-09 14:40:40 +00004869#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004870#ifdef IP_MAX_MEMBERSHIPS
4871 PyModule_AddIntConstant(m, "IP_MAX_MEMBERSHIPS", IP_MAX_MEMBERSHIPS);
Guido van Rossum09be4091999-08-09 14:40:40 +00004872#endif
4873
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004874 /* IPv6 [gs]etsockopt options, defined in RFC2553 */
4875#ifdef IPV6_JOIN_GROUP
4876 PyModule_AddIntConstant(m, "IPV6_JOIN_GROUP", IPV6_JOIN_GROUP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004877#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004878#ifdef IPV6_LEAVE_GROUP
4879 PyModule_AddIntConstant(m, "IPV6_LEAVE_GROUP", IPV6_LEAVE_GROUP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004880#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004881#ifdef IPV6_MULTICAST_HOPS
4882 PyModule_AddIntConstant(m, "IPV6_MULTICAST_HOPS", IPV6_MULTICAST_HOPS);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004883#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004884#ifdef IPV6_MULTICAST_IF
4885 PyModule_AddIntConstant(m, "IPV6_MULTICAST_IF", IPV6_MULTICAST_IF);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004886#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004887#ifdef IPV6_MULTICAST_LOOP
4888 PyModule_AddIntConstant(m, "IPV6_MULTICAST_LOOP", IPV6_MULTICAST_LOOP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004889#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004890#ifdef IPV6_UNICAST_HOPS
4891 PyModule_AddIntConstant(m, "IPV6_UNICAST_HOPS", IPV6_UNICAST_HOPS);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004892#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004893 /* Additional IPV6 socket options, defined in RFC 3493 */
Martin v. Löwisda91d022003-12-30 11:14:01 +00004894#ifdef IPV6_V6ONLY
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004895 PyModule_AddIntConstant(m, "IPV6_V6ONLY", IPV6_V6ONLY);
Martin v. Löwisda91d022003-12-30 11:14:01 +00004896#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004897 /* Advanced IPV6 socket options, from RFC 3542 */
Martin v. Löwisda91d022003-12-30 11:14:01 +00004898#ifdef IPV6_CHECKSUM
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004899 PyModule_AddIntConstant(m, "IPV6_CHECKSUM", IPV6_CHECKSUM);
Martin v. Löwisda91d022003-12-30 11:14:01 +00004900#endif
4901#ifdef IPV6_DONTFRAG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004902 PyModule_AddIntConstant(m, "IPV6_DONTFRAG", IPV6_DONTFRAG);
Martin v. Löwisda91d022003-12-30 11:14:01 +00004903#endif
4904#ifdef IPV6_DSTOPTS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004905 PyModule_AddIntConstant(m, "IPV6_DSTOPTS", IPV6_DSTOPTS);
Martin v. Löwisda91d022003-12-30 11:14:01 +00004906#endif
4907#ifdef IPV6_HOPLIMIT
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004908 PyModule_AddIntConstant(m, "IPV6_HOPLIMIT", IPV6_HOPLIMIT);
Martin v. Löwisda91d022003-12-30 11:14:01 +00004909#endif
4910#ifdef IPV6_HOPOPTS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004911 PyModule_AddIntConstant(m, "IPV6_HOPOPTS", IPV6_HOPOPTS);
Martin v. Löwisda91d022003-12-30 11:14:01 +00004912#endif
4913#ifdef IPV6_NEXTHOP
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004914 PyModule_AddIntConstant(m, "IPV6_NEXTHOP", IPV6_NEXTHOP);
Martin v. Löwisda91d022003-12-30 11:14:01 +00004915#endif
4916#ifdef IPV6_PATHMTU
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004917 PyModule_AddIntConstant(m, "IPV6_PATHMTU", IPV6_PATHMTU);
Martin v. Löwisda91d022003-12-30 11:14:01 +00004918#endif
4919#ifdef IPV6_PKTINFO
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004920 PyModule_AddIntConstant(m, "IPV6_PKTINFO", IPV6_PKTINFO);
Martin v. Löwisda91d022003-12-30 11:14:01 +00004921#endif
4922#ifdef IPV6_RECVDSTOPTS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004923 PyModule_AddIntConstant(m, "IPV6_RECVDSTOPTS", IPV6_RECVDSTOPTS);
Martin v. Löwisda91d022003-12-30 11:14:01 +00004924#endif
4925#ifdef IPV6_RECVHOPLIMIT
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004926 PyModule_AddIntConstant(m, "IPV6_RECVHOPLIMIT", IPV6_RECVHOPLIMIT);
Martin v. Löwisda91d022003-12-30 11:14:01 +00004927#endif
4928#ifdef IPV6_RECVHOPOPTS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004929 PyModule_AddIntConstant(m, "IPV6_RECVHOPOPTS", IPV6_RECVHOPOPTS);
Martin v. Löwisda91d022003-12-30 11:14:01 +00004930#endif
4931#ifdef IPV6_RECVPKTINFO
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004932 PyModule_AddIntConstant(m, "IPV6_RECVPKTINFO", IPV6_RECVPKTINFO);
Martin v. Löwisda91d022003-12-30 11:14:01 +00004933#endif
4934#ifdef IPV6_RECVRTHDR
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004935 PyModule_AddIntConstant(m, "IPV6_RECVRTHDR", IPV6_RECVRTHDR);
Martin v. Löwisda91d022003-12-30 11:14:01 +00004936#endif
4937#ifdef IPV6_RECVTCLASS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004938 PyModule_AddIntConstant(m, "IPV6_RECVTCLASS", IPV6_RECVTCLASS);
Martin v. Löwisda91d022003-12-30 11:14:01 +00004939#endif
4940#ifdef IPV6_RTHDR
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004941 PyModule_AddIntConstant(m, "IPV6_RTHDR", IPV6_RTHDR);
Martin v. Löwisda91d022003-12-30 11:14:01 +00004942#endif
4943#ifdef IPV6_RTHDRDSTOPTS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004944 PyModule_AddIntConstant(m, "IPV6_RTHDRDSTOPTS", IPV6_RTHDRDSTOPTS);
Martin v. Löwisda91d022003-12-30 11:14:01 +00004945#endif
4946#ifdef IPV6_RTHDR_TYPE_0
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004947 PyModule_AddIntConstant(m, "IPV6_RTHDR_TYPE_0", IPV6_RTHDR_TYPE_0);
Martin v. Löwisda91d022003-12-30 11:14:01 +00004948#endif
4949#ifdef IPV6_RECVPATHMTU
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004950 PyModule_AddIntConstant(m, "IPV6_RECVPATHMTU", IPV6_RECVPATHMTU);
Martin v. Löwisda91d022003-12-30 11:14:01 +00004951#endif
4952#ifdef IPV6_TCLASS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004953 PyModule_AddIntConstant(m, "IPV6_TCLASS", IPV6_TCLASS);
Martin v. Löwisda91d022003-12-30 11:14:01 +00004954#endif
4955#ifdef IPV6_USE_MIN_MTU
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004956 PyModule_AddIntConstant(m, "IPV6_USE_MIN_MTU", IPV6_USE_MIN_MTU);
Martin v. Löwisda91d022003-12-30 11:14:01 +00004957#endif
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004958
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004959 /* TCP options */
4960#ifdef TCP_NODELAY
4961 PyModule_AddIntConstant(m, "TCP_NODELAY", TCP_NODELAY);
Guido van Rossum09be4091999-08-09 14:40:40 +00004962#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004963#ifdef TCP_MAXSEG
4964 PyModule_AddIntConstant(m, "TCP_MAXSEG", TCP_MAXSEG);
Guido van Rossum09be4091999-08-09 14:40:40 +00004965#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004966#ifdef TCP_CORK
4967 PyModule_AddIntConstant(m, "TCP_CORK", TCP_CORK);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00004968#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004969#ifdef TCP_KEEPIDLE
4970 PyModule_AddIntConstant(m, "TCP_KEEPIDLE", TCP_KEEPIDLE);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00004971#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004972#ifdef TCP_KEEPINTVL
4973 PyModule_AddIntConstant(m, "TCP_KEEPINTVL", TCP_KEEPINTVL);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00004974#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004975#ifdef TCP_KEEPCNT
4976 PyModule_AddIntConstant(m, "TCP_KEEPCNT", TCP_KEEPCNT);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00004977#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004978#ifdef TCP_SYNCNT
4979 PyModule_AddIntConstant(m, "TCP_SYNCNT", TCP_SYNCNT);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00004980#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004981#ifdef TCP_LINGER2
4982 PyModule_AddIntConstant(m, "TCP_LINGER2", TCP_LINGER2);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00004983#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004984#ifdef TCP_DEFER_ACCEPT
4985 PyModule_AddIntConstant(m, "TCP_DEFER_ACCEPT", TCP_DEFER_ACCEPT);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00004986#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004987#ifdef TCP_WINDOW_CLAMP
4988 PyModule_AddIntConstant(m, "TCP_WINDOW_CLAMP", TCP_WINDOW_CLAMP);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00004989#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004990#ifdef TCP_INFO
4991 PyModule_AddIntConstant(m, "TCP_INFO", TCP_INFO);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00004992#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004993#ifdef TCP_QUICKACK
4994 PyModule_AddIntConstant(m, "TCP_QUICKACK", TCP_QUICKACK);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00004995#endif
4996
Guido van Rossum09be4091999-08-09 14:40:40 +00004997
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004998 /* IPX options */
4999#ifdef IPX_TYPE
5000 PyModule_AddIntConstant(m, "IPX_TYPE", IPX_TYPE);
Guido van Rossum09be4091999-08-09 14:40:40 +00005001#endif
Guido van Rossum4f199ea1998-04-09 20:56:35 +00005002
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005003 /* get{addr,name}info parameters */
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005004#ifdef EAI_ADDRFAMILY
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005005 PyModule_AddIntConstant(m, "EAI_ADDRFAMILY", EAI_ADDRFAMILY);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005006#endif
5007#ifdef EAI_AGAIN
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005008 PyModule_AddIntConstant(m, "EAI_AGAIN", EAI_AGAIN);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005009#endif
5010#ifdef EAI_BADFLAGS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005011 PyModule_AddIntConstant(m, "EAI_BADFLAGS", EAI_BADFLAGS);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005012#endif
5013#ifdef EAI_FAIL
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005014 PyModule_AddIntConstant(m, "EAI_FAIL", EAI_FAIL);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005015#endif
5016#ifdef EAI_FAMILY
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005017 PyModule_AddIntConstant(m, "EAI_FAMILY", EAI_FAMILY);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005018#endif
5019#ifdef EAI_MEMORY
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005020 PyModule_AddIntConstant(m, "EAI_MEMORY", EAI_MEMORY);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005021#endif
5022#ifdef EAI_NODATA
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005023 PyModule_AddIntConstant(m, "EAI_NODATA", EAI_NODATA);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005024#endif
5025#ifdef EAI_NONAME
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005026 PyModule_AddIntConstant(m, "EAI_NONAME", EAI_NONAME);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005027#endif
Martin v. Löwisda91d022003-12-30 11:14:01 +00005028#ifdef EAI_OVERFLOW
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005029 PyModule_AddIntConstant(m, "EAI_OVERFLOW", EAI_OVERFLOW);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005030#endif
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005031#ifdef EAI_SERVICE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005032 PyModule_AddIntConstant(m, "EAI_SERVICE", EAI_SERVICE);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005033#endif
5034#ifdef EAI_SOCKTYPE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005035 PyModule_AddIntConstant(m, "EAI_SOCKTYPE", EAI_SOCKTYPE);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005036#endif
5037#ifdef EAI_SYSTEM
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005038 PyModule_AddIntConstant(m, "EAI_SYSTEM", EAI_SYSTEM);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005039#endif
5040#ifdef EAI_BADHINTS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005041 PyModule_AddIntConstant(m, "EAI_BADHINTS", EAI_BADHINTS);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005042#endif
5043#ifdef EAI_PROTOCOL
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005044 PyModule_AddIntConstant(m, "EAI_PROTOCOL", EAI_PROTOCOL);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005045#endif
5046#ifdef EAI_MAX
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005047 PyModule_AddIntConstant(m, "EAI_MAX", EAI_MAX);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005048#endif
5049#ifdef AI_PASSIVE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005050 PyModule_AddIntConstant(m, "AI_PASSIVE", AI_PASSIVE);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005051#endif
5052#ifdef AI_CANONNAME
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005053 PyModule_AddIntConstant(m, "AI_CANONNAME", AI_CANONNAME);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005054#endif
5055#ifdef AI_NUMERICHOST
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005056 PyModule_AddIntConstant(m, "AI_NUMERICHOST", AI_NUMERICHOST);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005057#endif
Martin v. Löwisda91d022003-12-30 11:14:01 +00005058#ifdef AI_NUMERICSERV
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005059 PyModule_AddIntConstant(m, "AI_NUMERICSERV", AI_NUMERICSERV);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005060#endif
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005061#ifdef AI_MASK
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005062 PyModule_AddIntConstant(m, "AI_MASK", AI_MASK);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005063#endif
5064#ifdef AI_ALL
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005065 PyModule_AddIntConstant(m, "AI_ALL", AI_ALL);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005066#endif
5067#ifdef AI_V4MAPPED_CFG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005068 PyModule_AddIntConstant(m, "AI_V4MAPPED_CFG", AI_V4MAPPED_CFG);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005069#endif
5070#ifdef AI_ADDRCONFIG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005071 PyModule_AddIntConstant(m, "AI_ADDRCONFIG", AI_ADDRCONFIG);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005072#endif
5073#ifdef AI_V4MAPPED
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005074 PyModule_AddIntConstant(m, "AI_V4MAPPED", AI_V4MAPPED);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005075#endif
5076#ifdef AI_DEFAULT
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005077 PyModule_AddIntConstant(m, "AI_DEFAULT", AI_DEFAULT);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005078#endif
5079#ifdef NI_MAXHOST
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005080 PyModule_AddIntConstant(m, "NI_MAXHOST", NI_MAXHOST);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005081#endif
5082#ifdef NI_MAXSERV
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005083 PyModule_AddIntConstant(m, "NI_MAXSERV", NI_MAXSERV);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005084#endif
5085#ifdef NI_NOFQDN
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005086 PyModule_AddIntConstant(m, "NI_NOFQDN", NI_NOFQDN);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005087#endif
5088#ifdef NI_NUMERICHOST
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005089 PyModule_AddIntConstant(m, "NI_NUMERICHOST", NI_NUMERICHOST);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005090#endif
5091#ifdef NI_NAMEREQD
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005092 PyModule_AddIntConstant(m, "NI_NAMEREQD", NI_NAMEREQD);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005093#endif
5094#ifdef NI_NUMERICSERV
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005095 PyModule_AddIntConstant(m, "NI_NUMERICSERV", NI_NUMERICSERV);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005096#endif
5097#ifdef NI_DGRAM
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005098 PyModule_AddIntConstant(m, "NI_DGRAM", NI_DGRAM);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005099#endif
5100
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005101 /* shutdown() parameters */
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005102#ifdef SHUT_RD
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005103 PyModule_AddIntConstant(m, "SHUT_RD", SHUT_RD);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005104#elif defined(SD_RECEIVE)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005105 PyModule_AddIntConstant(m, "SHUT_RD", SD_RECEIVE);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005106#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005107 PyModule_AddIntConstant(m, "SHUT_RD", 0);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005108#endif
5109#ifdef SHUT_WR
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005110 PyModule_AddIntConstant(m, "SHUT_WR", SHUT_WR);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005111#elif defined(SD_SEND)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005112 PyModule_AddIntConstant(m, "SHUT_WR", SD_SEND);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005113#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005114 PyModule_AddIntConstant(m, "SHUT_WR", 1);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005115#endif
5116#ifdef SHUT_RDWR
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005117 PyModule_AddIntConstant(m, "SHUT_RDWR", SHUT_RDWR);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005118#elif defined(SD_BOTH)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005119 PyModule_AddIntConstant(m, "SHUT_RDWR", SD_BOTH);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005120#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005121 PyModule_AddIntConstant(m, "SHUT_RDWR", 2);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005122#endif
5123
Christian Heimesfaf2f632008-01-06 16:59:19 +00005124#ifdef SIO_RCVALL
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005125 {
5126 DWORD codes[] = {SIO_RCVALL, SIO_KEEPALIVE_VALS};
5127 const char *names[] = {"SIO_RCVALL", "SIO_KEEPALIVE_VALS"};
5128 int i;
5129 for(i = 0; i<sizeof(codes)/sizeof(*codes); ++i) {
5130 PyObject *tmp;
5131 tmp = PyLong_FromUnsignedLong(codes[i]);
5132 if (tmp == NULL)
5133 return NULL;
5134 PyModule_AddObject(m, names[i], tmp);
5135 }
5136 }
5137 PyModule_AddIntConstant(m, "RCVALL_OFF", RCVALL_OFF);
5138 PyModule_AddIntConstant(m, "RCVALL_ON", RCVALL_ON);
5139 PyModule_AddIntConstant(m, "RCVALL_SOCKETLEVELONLY", RCVALL_SOCKETLEVELONLY);
Amaury Forgeot d'Arc762681b2008-06-12 23:03:41 +00005140#ifdef RCVALL_IPLEVEL
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005141 PyModule_AddIntConstant(m, "RCVALL_IPLEVEL", RCVALL_IPLEVEL);
Amaury Forgeot d'Arc762681b2008-06-12 23:03:41 +00005142#endif
5143#ifdef RCVALL_MAX
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005144 PyModule_AddIntConstant(m, "RCVALL_MAX", RCVALL_MAX);
Amaury Forgeot d'Arc762681b2008-06-12 23:03:41 +00005145#endif
Christian Heimesfaf2f632008-01-06 16:59:19 +00005146#endif /* _MSTCPIP_ */
5147
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005148 /* Initialize gethostbyname lock */
Just van Rossum1040d2c2003-05-09 07:53:18 +00005149#if defined(USE_GETHOSTBYNAME_LOCK) || defined(USE_GETADDRINFO_LOCK)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005150 netdb_lock = PyThread_allocate_lock();
Guido van Rossum4f199ea1998-04-09 20:56:35 +00005151#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005152 return m;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00005153}
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005154
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00005155
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005156#ifndef HAVE_INET_PTON
Christian Heimes96e7b3d2007-11-20 06:51:17 +00005157#if !defined(NTDDI_VERSION) || (NTDDI_VERSION < NTDDI_LONGHORN)
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00005158
5159/* Simplistic emulation code for inet_pton that only works for IPv4 */
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00005160/* These are not exposed because they do not set errno properly */
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00005161
Guido van Rossum3eede5a2002-06-07 02:08:35 +00005162int
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00005163inet_pton(int af, const char *src, void *dst)
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005164{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005165 if (af == AF_INET) {
Benjamin Petersonf91df042009-02-13 02:50:59 +00005166#if (SIZEOF_INT != 4)
5167#error "Not sure if in_addr_t exists and int is not 32-bits."
5168#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005169 unsigned int packed_addr;
5170 packed_addr = inet_addr(src);
5171 if (packed_addr == INADDR_NONE)
5172 return 0;
5173 memcpy(dst, &packed_addr, 4);
5174 return 1;
5175 }
5176 /* Should set errno to EAFNOSUPPORT */
5177 return -1;
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005178}
5179
Martin v. Löwisc925b1532001-07-21 09:42:15 +00005180const char *
5181inet_ntop(int af, const void *src, char *dst, socklen_t size)
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005182{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005183 if (af == AF_INET) {
5184 struct in_addr packed_addr;
5185 if (size < 16)
5186 /* Should set errno to ENOSPC. */
5187 return NULL;
5188 memcpy(&packed_addr, src, sizeof(packed_addr));
5189 return strncpy(dst, inet_ntoa(packed_addr), size);
5190 }
5191 /* Should set errno to EAFNOSUPPORT */
5192 return NULL;
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005193}
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00005194
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005195#endif
Christian Heimesb6150692007-11-15 23:37:07 +00005196#endif