blob: cd56ee31c1914753c6fb191210c7cacc49baf225 [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{
Daniel Stutzbach6c765282010-09-03 12:38:33 +0000354 WSAPROTOCOL_INFO info;
Guido van Rossum39eb8fa2007-11-16 01:24:05 +0000355
Daniel Stutzbach6c765282010-09-03 12:38:33 +0000356 if (WSADuplicateSocket(handle, GetCurrentProcessId(), &info))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000357 return INVALID_SOCKET;
Daniel Stutzbach6c765282010-09-03 12:38:33 +0000358
359 return WSASocket(FROM_PROTOCOL_INFO, FROM_PROTOCOL_INFO,
360 FROM_PROTOCOL_INFO, &info, 0, 0);
Guido van Rossum39eb8fa2007-11-16 01:24:05 +0000361}
Guido van Rossum2dd8ddd2000-04-21 20:33:00 +0000362#define SOCKETCLOSE closesocket
Guido van Rossum39eb8fa2007-11-16 01:24:05 +0000363#else
364/* On Unix we can use dup to duplicate the file descriptor of a socket*/
365#define dup_socket(fd) dup(fd)
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +0000366#endif
367
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000368#ifdef MS_WIN32
Guido van Rossum3eede5a2002-06-07 02:08:35 +0000369#define EAFNOSUPPORT WSAEAFNOSUPPORT
370#define snprintf _snprintf
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000371#endif
Fred Drakea04eaad2000-06-30 02:46:07 +0000372
Andrew MacIntyreba43e872002-03-03 03:03:52 +0000373#if defined(PYOS_OS2) && !defined(PYCC_GCC)
Guido van Rossum2dd8ddd2000-04-21 20:33:00 +0000374#define SOCKETCLOSE soclose
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000375#define NO_DUP /* Sockets are Not Actual File Handles under OS/2 */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000376#endif
377
Guido van Rossum2dd8ddd2000-04-21 20:33:00 +0000378#ifndef SOCKETCLOSE
379#define SOCKETCLOSE close
380#endif
381
Jesse Noller32d68c22009-03-31 18:48:42 +0000382#if defined(HAVE_BLUETOOTH_H) || defined(HAVE_BLUETOOTH_BLUETOOTH_H) && !defined(__NetBSD__)
Hye-Shik Chang81268602004-02-02 06:05:24 +0000383#define USE_BLUETOOTH 1
384#if defined(__FreeBSD__)
385#define BTPROTO_L2CAP BLUETOOTH_PROTO_L2CAP
386#define BTPROTO_RFCOMM BLUETOOTH_PROTO_RFCOMM
Thomas Wouterscf297e42007-02-23 15:07:44 +0000387#define BTPROTO_HCI BLUETOOTH_PROTO_HCI
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000388#define SOL_HCI SOL_HCI_RAW
389#define HCI_FILTER SO_HCI_RAW_FILTER
Hye-Shik Chang81268602004-02-02 06:05:24 +0000390#define sockaddr_l2 sockaddr_l2cap
391#define sockaddr_rc sockaddr_rfcomm
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000392#define hci_dev hci_node
Hye-Shik Chang81268602004-02-02 06:05:24 +0000393#define _BT_L2_MEMB(sa, memb) ((sa)->l2cap_##memb)
394#define _BT_RC_MEMB(sa, memb) ((sa)->rfcomm_##memb)
Thomas Wouterscf297e42007-02-23 15:07:44 +0000395#define _BT_HCI_MEMB(sa, memb) ((sa)->hci_##memb)
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000396#elif defined(__NetBSD__)
397#define sockaddr_l2 sockaddr_bt
398#define sockaddr_rc sockaddr_bt
Thomas Wouterscf297e42007-02-23 15:07:44 +0000399#define sockaddr_hci sockaddr_bt
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000400#define sockaddr_sco sockaddr_bt
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000401#define _BT_L2_MEMB(sa, memb) ((sa)->bt_##memb)
402#define _BT_RC_MEMB(sa, memb) ((sa)->bt_##memb)
Thomas Wouterscf297e42007-02-23 15:07:44 +0000403#define _BT_HCI_MEMB(sa, memb) ((sa)->bt_##memb)
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000404#define _BT_SCO_MEMB(sa, memb) ((sa)->bt_##memb)
Hye-Shik Chang81268602004-02-02 06:05:24 +0000405#else
Hye-Shik Chang81268602004-02-02 06:05:24 +0000406#define _BT_L2_MEMB(sa, memb) ((sa)->l2_##memb)
407#define _BT_RC_MEMB(sa, memb) ((sa)->rc_##memb)
Thomas Wouterscf297e42007-02-23 15:07:44 +0000408#define _BT_HCI_MEMB(sa, memb) ((sa)->hci_##memb)
Hye-Shik Chang81268602004-02-02 06:05:24 +0000409#define _BT_SCO_MEMB(sa, memb) ((sa)->sco_##memb)
410#endif
411#endif
412
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000413#ifdef __VMS
414/* TCP/IP Services for VMS uses a maximum send/recv buffer length */
415#define SEGMENT_SIZE (32 * 1024 -1)
416#endif
417
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000418#define SAS2SA(x) ((struct sockaddr *)(x))
Thomas Wouters89f507f2006-12-13 04:49:30 +0000419
Martin v. Löwise9416172003-05-03 10:12:45 +0000420/*
421 * Constants for getnameinfo()
422 */
423#if !defined(NI_MAXHOST)
424#define NI_MAXHOST 1025
425#endif
426#if !defined(NI_MAXSERV)
427#define NI_MAXSERV 32
428#endif
429
Guido van Rossum7d0a8262007-05-21 23:13:11 +0000430#ifndef INVALID_SOCKET /* MS defines this */
431#define INVALID_SOCKET (-1)
432#endif
433
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000434/* XXX There's a problem here: *static* functions are not supposed to have
435 a Py prefix (or use CapitalizedWords). Later... */
436
Guido van Rossum30a685f1991-06-27 15:51:29 +0000437/* Global variable holding the exception type for errors detected
438 by this module (but not argument type or memory errors, etc.). */
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000439static PyObject *socket_error;
440static PyObject *socket_herror;
441static PyObject *socket_gaierror;
Raymond Hettingeref7343c2003-06-29 03:08:05 +0000442static PyObject *socket_timeout;
Guido van Rossum30a685f1991-06-27 15:51:29 +0000443
Tim Peters643a7fc2002-02-17 04:13:21 +0000444/* A forward reference to the socket type object.
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000445 The sock_type variable contains pointers to various functions,
446 some of which call new_sockobject(), which uses sock_type, so
Tim Peters643a7fc2002-02-17 04:13:21 +0000447 there has to be a circular reference. */
Jeremy Hylton938ace62002-07-17 16:30:39 +0000448static PyTypeObject sock_type;
Guido van Rossum48a680c2001-03-02 06:34:14 +0000449
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000450#if defined(HAVE_POLL_H)
451#include <poll.h>
452#elif defined(HAVE_SYS_POLL_H)
453#include <sys/poll.h>
454#endif
455
Martin v. Löwisf84d1b92006-02-11 09:27:05 +0000456#ifdef Py_SOCKET_FD_CAN_BE_GE_FD_SETSIZE
457/* Platform can select file descriptors beyond FD_SETSIZE */
458#define IS_SELECTABLE(s) 1
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000459#elif defined(HAVE_POLL)
460/* Instead of select(), we'll use poll() since poll() works on any fd. */
461#define IS_SELECTABLE(s) 1
462/* Can we call select() with this socket without a buffer overrun? */
Martin v. Löwisf84d1b92006-02-11 09:27:05 +0000463#else
464/* POSIX says selecting file descriptors beyond FD_SETSIZE
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000465 has undefined behaviour. If there's no timeout left, we don't have to
466 call select, so it's a safe, little white lie. */
467#define IS_SELECTABLE(s) ((s)->sock_fd < FD_SETSIZE || s->sock_timeout <= 0.0)
Martin v. Löwisf84d1b92006-02-11 09:27:05 +0000468#endif
Neal Norwitz082b2df2006-02-07 07:04:46 +0000469
470static PyObject*
471select_error(void)
472{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000473 PyErr_SetString(socket_error, "unable to select on socket");
474 return NULL;
Neal Norwitz082b2df2006-02-07 07:04:46 +0000475}
476
Guido van Rossum30a685f1991-06-27 15:51:29 +0000477/* Convenience function to raise an error according to errno
478 and return a NULL pointer from a function. */
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000479
Guido van Rossum73624e91994-10-10 17:59:00 +0000480static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000481set_error(void)
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000482{
Guido van Rossum8d665e61996-06-26 18:22:49 +0000483#ifdef MS_WINDOWS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000484 int err_no = WSAGetLastError();
485 /* PyErr_SetExcFromWindowsErr() invokes FormatMessage() which
486 recognizes the error codes used by both GetLastError() and
487 WSAGetLastError */
488 if (err_no)
489 return PyErr_SetExcFromWindowsErr(socket_error, err_no);
Guido van Rossum14ed0b21994-09-29 09:50:09 +0000490#endif
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000491
Andrew MacIntyreba43e872002-03-03 03:03:52 +0000492#if defined(PYOS_OS2) && !defined(PYCC_GCC)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000493 if (sock_errno() != NO_ERROR) {
494 APIRET rc;
495 ULONG msglen;
496 char outbuf[100];
497 int myerrorcode = sock_errno();
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000498
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000499 /* Retrieve socket-related error message from MPTN.MSG file */
500 rc = DosGetMessage(NULL, 0, outbuf, sizeof(outbuf),
501 myerrorcode - SOCBASEERR + 26,
502 "mptn.msg",
503 &msglen);
504 if (rc == NO_ERROR) {
505 PyObject *v;
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000506
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000507 /* OS/2 doesn't guarantee a terminator */
508 outbuf[msglen] = '\0';
509 if (strlen(outbuf) > 0) {
510 /* If non-empty msg, trim CRLF */
511 char *lastc = &outbuf[ strlen(outbuf)-1 ];
512 while (lastc > outbuf &&
513 isspace(Py_CHARMASK(*lastc))) {
514 /* Trim trailing whitespace (CRLF) */
515 *lastc-- = '\0';
516 }
517 }
518 v = Py_BuildValue("(is)", myerrorcode, outbuf);
519 if (v != NULL) {
520 PyErr_SetObject(socket_error, v);
521 Py_DECREF(v);
522 }
523 return NULL;
524 }
525 }
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000526#endif
527
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000528 return PyErr_SetFromErrno(socket_error);
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000529}
530
Guido van Rossum30a685f1991-06-27 15:51:29 +0000531
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000532static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000533set_herror(int h_error)
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000534{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000535 PyObject *v;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000536
537#ifdef HAVE_HSTRERROR
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000538 v = Py_BuildValue("(is)", h_error, (char *)hstrerror(h_error));
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000539#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000540 v = Py_BuildValue("(is)", h_error, "host not found");
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000541#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000542 if (v != NULL) {
543 PyErr_SetObject(socket_herror, v);
544 Py_DECREF(v);
545 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000546
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000547 return NULL;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000548}
549
550
551static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000552set_gaierror(int error)
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000553{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000554 PyObject *v;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000555
Martin v. Löwis272cb402002-03-01 08:31:07 +0000556#ifdef EAI_SYSTEM
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000557 /* EAI_SYSTEM is not available on Windows XP. */
558 if (error == EAI_SYSTEM)
559 return set_error();
Martin v. Löwis272cb402002-03-01 08:31:07 +0000560#endif
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000561
Martin v. Löwisf95dd0a2001-08-15 17:14:33 +0000562#ifdef HAVE_GAI_STRERROR
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000563 v = Py_BuildValue("(is)", error, gai_strerror(error));
Martin v. Löwisf95dd0a2001-08-15 17:14:33 +0000564#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000565 v = Py_BuildValue("(is)", error, "getaddrinfo failed");
Martin v. Löwisf95dd0a2001-08-15 17:14:33 +0000566#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000567 if (v != NULL) {
568 PyErr_SetObject(socket_gaierror, v);
569 Py_DECREF(v);
570 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000571
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000572 return NULL;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000573}
574
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000575#ifdef __VMS
576/* Function to send in segments */
577static int
578sendsegmented(int sock_fd, char *buf, int len, int flags)
579{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000580 int n = 0;
581 int remaining = len;
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000582
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000583 while (remaining > 0) {
584 unsigned int segment;
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000585
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000586 segment = (remaining >= SEGMENT_SIZE ? SEGMENT_SIZE : remaining);
587 n = send(sock_fd, buf, segment, flags);
588 if (n < 0) {
589 return n;
590 }
591 remaining -= segment;
592 buf += segment;
593 } /* end while */
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000594
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000595 return len;
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000596}
597#endif
598
Guido van Rossum3eede5a2002-06-07 02:08:35 +0000599/* Function to perform the setting of socket blocking mode
600 internally. block = (1 | 0). */
Guido van Rossum67f7a382002-06-06 21:08:16 +0000601static int
602internal_setblocking(PySocketSockObject *s, int block)
603{
Guido van Rossum67f7a382002-06-06 21:08:16 +0000604#ifndef MS_WINDOWS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000605 int delay_flag;
Guido van Rossum67f7a382002-06-06 21:08:16 +0000606#endif
Guido van Rossum67f7a382002-06-06 21:08:16 +0000607
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000608 Py_BEGIN_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +0000609#ifndef MS_WINDOWS
610#if defined(PYOS_OS2) && !defined(PYCC_GCC)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000611 block = !block;
612 ioctl(s->sock_fd, FIONBIO, (caddr_t)&block, sizeof(block));
Martin v. Löwis79acb9e2002-12-06 12:48:53 +0000613#elif defined(__VMS)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000614 block = !block;
615 ioctl(s->sock_fd, FIONBIO, (unsigned int *)&block);
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000616#else /* !PYOS_OS2 && !__VMS */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000617 delay_flag = fcntl(s->sock_fd, F_GETFL, 0);
618 if (block)
619 delay_flag &= (~O_NONBLOCK);
620 else
621 delay_flag |= O_NONBLOCK;
622 fcntl(s->sock_fd, F_SETFL, delay_flag);
Guido van Rossum67f7a382002-06-06 21:08:16 +0000623#endif /* !PYOS_OS2 */
624#else /* MS_WINDOWS */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000625 block = !block;
626 ioctlsocket(s->sock_fd, FIONBIO, (u_long*)&block);
Guido van Rossum67f7a382002-06-06 21:08:16 +0000627#endif /* MS_WINDOWS */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000628 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +0000629
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000630 /* Since these don't return anything */
631 return 1;
Guido van Rossum67f7a382002-06-06 21:08:16 +0000632}
633
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000634/* Do a select()/poll() on the socket, if necessary (sock_timeout > 0).
Guido van Rossum11ba0942002-06-13 15:07:44 +0000635 The argument writing indicates the direction.
Raymond Hettingeref7343c2003-06-29 03:08:05 +0000636 This does not raise an exception; we'll let our caller do that
637 after they've reacquired the interpreter lock.
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000638 Returns 1 on timeout, -1 on error, 0 otherwise. */
Raymond Hettingeref7343c2003-06-29 03:08:05 +0000639static int
Guido van Rossumb9e916a2002-06-07 01:42:47 +0000640internal_select(PySocketSockObject *s, int writing)
Guido van Rossum67f7a382002-06-06 21:08:16 +0000641{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000642 int n;
Guido van Rossum11ba0942002-06-13 15:07:44 +0000643
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000644 /* Nothing to do unless we're in timeout mode (not non-blocking) */
645 if (s->sock_timeout <= 0.0)
646 return 0;
Guido van Rossum67f7a382002-06-06 21:08:16 +0000647
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000648 /* Guard against closed socket */
649 if (s->sock_fd < 0)
650 return 0;
Guido van Rossumad654902002-07-19 12:44:59 +0000651
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000652 /* Prefer poll, if available, since you can poll() any fd
653 * which can't be done with select(). */
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000654#ifdef HAVE_POLL
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000655 {
656 struct pollfd pollfd;
657 int timeout;
Guido van Rossum67f7a382002-06-06 21:08:16 +0000658
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000659 pollfd.fd = s->sock_fd;
660 pollfd.events = writing ? POLLOUT : POLLIN;
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000661
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000662 /* s->sock_timeout is in seconds, timeout in ms */
663 timeout = (int)(s->sock_timeout * 1000 + 0.5);
664 n = poll(&pollfd, 1, timeout);
665 }
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000666#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000667 {
668 /* Construct the arguments to select */
669 fd_set fds;
670 struct timeval tv;
671 tv.tv_sec = (int)s->sock_timeout;
672 tv.tv_usec = (int)((s->sock_timeout - tv.tv_sec) * 1e6);
673 FD_ZERO(&fds);
674 FD_SET(s->sock_fd, &fds);
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000675
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000676 /* See if the socket is ready */
677 if (writing)
Antoine Pitrou19467d22010-08-17 19:33:30 +0000678 n = select(Py_SAFE_DOWNCAST(s->sock_fd+1, SOCKET_T, int),
679 NULL, &fds, NULL, &tv);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000680 else
Antoine Pitrou19467d22010-08-17 19:33:30 +0000681 n = select(Py_SAFE_DOWNCAST(s->sock_fd+1, SOCKET_T, int),
682 &fds, NULL, NULL, &tv);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000683 }
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000684#endif
Guido van Rossum39eb8fa2007-11-16 01:24:05 +0000685
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000686 if (n < 0)
687 return -1;
688 if (n == 0)
689 return 1;
690 return 0;
Guido van Rossum67f7a382002-06-06 21:08:16 +0000691}
692
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000693/* Initialize a new socket object. */
694
Tim Petersa12b4cf2002-07-18 22:38:44 +0000695static double defaulttimeout = -1.0; /* Default timeout for new sockets */
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +0000696
Martin v. Löwis1a214512008-06-11 05:26:20 +0000697static void
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000698init_sockobject(PySocketSockObject *s,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000699 SOCKET_T fd, int family, int type, int proto)
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000700{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000701 s->sock_fd = fd;
702 s->sock_family = family;
703 s->sock_type = type;
704 s->sock_proto = proto;
705 s->sock_timeout = defaulttimeout;
Guido van Rossum67f7a382002-06-06 21:08:16 +0000706
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000707 s->errorhandler = &set_error;
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +0000708
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000709 if (defaulttimeout >= 0.0)
710 internal_setblocking(s, 0);
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +0000711
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000712}
713
714
Guido van Rossum30a685f1991-06-27 15:51:29 +0000715/* Create a new socket object.
716 This just creates the object and initializes it.
717 If the creation fails, return NULL and set an exception (implicit
718 in NEWOBJ()). */
719
Guido van Rossum73624e91994-10-10 17:59:00 +0000720static PySocketSockObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000721new_sockobject(SOCKET_T fd, int family, int type, int proto)
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000722{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000723 PySocketSockObject *s;
724 s = (PySocketSockObject *)
725 PyType_GenericNew(&sock_type, NULL, NULL);
726 if (s != NULL)
727 init_sockobject(s, fd, family, type, proto);
728 return s;
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000729}
730
Guido van Rossum30a685f1991-06-27 15:51:29 +0000731
Guido van Rossum48a680c2001-03-02 06:34:14 +0000732/* Lock to allow python interpreter to continue, but only allow one
Just van Rossum1040d2c2003-05-09 07:53:18 +0000733 thread to be in gethostbyname or getaddrinfo */
734#if defined(USE_GETHOSTBYNAME_LOCK) || defined(USE_GETADDRINFO_LOCK)
735PyThread_type_lock netdb_lock;
Guido van Rossum4f199ea1998-04-09 20:56:35 +0000736#endif
737
738
Guido van Rossum30a685f1991-06-27 15:51:29 +0000739/* Convert a string specifying a host name or one of a few symbolic
740 names to a numeric IP address. This usually calls gethostbyname()
741 to do the work; the names "" and "<broadcast>" are special.
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000742 Return the length (IPv4 should be 4 bytes), or negative if
Guido van Rossum30a685f1991-06-27 15:51:29 +0000743 an error occurred; then an exception is raised. */
744
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000745static int
Martin v. Löwisddc6f472002-07-28 16:10:31 +0000746setipaddr(char *name, struct sockaddr *addr_ret, size_t addr_ret_size, int af)
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000747{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000748 struct addrinfo hints, *res;
749 int error;
750 int d1, d2, d3, d4;
751 char ch;
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000752
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000753 memset((void *) addr_ret, '\0', sizeof(*addr_ret));
754 if (name[0] == '\0') {
755 int siz;
756 memset(&hints, 0, sizeof(hints));
757 hints.ai_family = af;
758 hints.ai_socktype = SOCK_DGRAM; /*dummy*/
759 hints.ai_flags = AI_PASSIVE;
760 Py_BEGIN_ALLOW_THREADS
761 ACQUIRE_GETADDRINFO_LOCK
762 error = getaddrinfo(NULL, "0", &hints, &res);
763 Py_END_ALLOW_THREADS
764 /* We assume that those thread-unsafe getaddrinfo() versions
765 *are* safe regarding their return value, ie. that a
766 subsequent call to getaddrinfo() does not destroy the
767 outcome of the first call. */
768 RELEASE_GETADDRINFO_LOCK
769 if (error) {
770 set_gaierror(error);
771 return -1;
772 }
773 switch (res->ai_family) {
774 case AF_INET:
775 siz = 4;
776 break;
Martin v. Löwis44ddbde2001-12-02 10:15:37 +0000777#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000778 case AF_INET6:
779 siz = 16;
780 break;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000781#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000782 default:
783 freeaddrinfo(res);
784 PyErr_SetString(socket_error,
785 "unsupported address family");
786 return -1;
787 }
788 if (res->ai_next) {
789 freeaddrinfo(res);
790 PyErr_SetString(socket_error,
791 "wildcard resolved to multiple address");
792 return -1;
793 }
794 if (res->ai_addrlen < addr_ret_size)
795 addr_ret_size = res->ai_addrlen;
796 memcpy(addr_ret, res->ai_addr, addr_ret_size);
797 freeaddrinfo(res);
798 return siz;
799 }
800 if (name[0] == '<' && strcmp(name, "<broadcast>") == 0) {
801 struct sockaddr_in *sin;
802 if (af != AF_INET && af != AF_UNSPEC) {
803 PyErr_SetString(socket_error,
804 "address family mismatched");
805 return -1;
806 }
807 sin = (struct sockaddr_in *)addr_ret;
808 memset((void *) sin, '\0', sizeof(*sin));
809 sin->sin_family = AF_INET;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000810#ifdef HAVE_SOCKADDR_SA_LEN
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000811 sin->sin_len = sizeof(*sin);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000812#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000813 sin->sin_addr.s_addr = INADDR_BROADCAST;
814 return sizeof(sin->sin_addr);
815 }
816 if (sscanf(name, "%d.%d.%d.%d%c", &d1, &d2, &d3, &d4, &ch) == 4 &&
817 0 <= d1 && d1 <= 255 && 0 <= d2 && d2 <= 255 &&
818 0 <= d3 && d3 <= 255 && 0 <= d4 && d4 <= 255) {
819 struct sockaddr_in *sin;
820 sin = (struct sockaddr_in *)addr_ret;
821 sin->sin_addr.s_addr = htonl(
822 ((long) d1 << 24) | ((long) d2 << 16) |
823 ((long) d3 << 8) | ((long) d4 << 0));
824 sin->sin_family = AF_INET;
Anthony Baxter0e85f9d2003-05-02 15:40:46 +0000825#ifdef HAVE_SOCKADDR_SA_LEN
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000826 sin->sin_len = sizeof(*sin);
Anthony Baxter0e85f9d2003-05-02 15:40:46 +0000827#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000828 return 4;
829 }
830 memset(&hints, 0, sizeof(hints));
831 hints.ai_family = af;
832 Py_BEGIN_ALLOW_THREADS
833 ACQUIRE_GETADDRINFO_LOCK
834 error = getaddrinfo(name, NULL, &hints, &res);
Martin v. Löwis7c4b5fa2001-10-25 09:04:03 +0000835#if defined(__digital__) && defined(__unix__)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000836 if (error == EAI_NONAME && af == AF_UNSPEC) {
837 /* On Tru64 V5.1, numeric-to-addr conversion fails
838 if no address family is given. Assume IPv4 for now.*/
839 hints.ai_family = AF_INET;
840 error = getaddrinfo(name, NULL, &hints, &res);
841 }
Martin v. Löwis7c4b5fa2001-10-25 09:04:03 +0000842#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000843 Py_END_ALLOW_THREADS
844 RELEASE_GETADDRINFO_LOCK /* see comment in setipaddr() */
845 if (error) {
846 set_gaierror(error);
847 return -1;
848 }
849 if (res->ai_addrlen < addr_ret_size)
850 addr_ret_size = res->ai_addrlen;
851 memcpy((char *) addr_ret, res->ai_addr, addr_ret_size);
852 freeaddrinfo(res);
853 switch (addr_ret->sa_family) {
854 case AF_INET:
855 return 4;
Martin v. Löwis44ddbde2001-12-02 10:15:37 +0000856#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000857 case AF_INET6:
858 return 16;
Guido van Rossum955becc1999-03-22 20:14:53 +0000859#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000860 default:
861 PyErr_SetString(socket_error, "unknown address family");
862 return -1;
863 }
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000864}
865
Guido van Rossum30a685f1991-06-27 15:51:29 +0000866
Guido van Rossum30a685f1991-06-27 15:51:29 +0000867/* Create a string object representing an IP address.
868 This is always a string of the form 'dd.dd.dd.dd' (with variable
869 size numbers). */
870
Guido van Rossum73624e91994-10-10 17:59:00 +0000871static PyObject *
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000872makeipaddr(struct sockaddr *addr, int addrlen)
Guido van Rossum30a685f1991-06-27 15:51:29 +0000873{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000874 char buf[NI_MAXHOST];
875 int error;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000876
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000877 error = getnameinfo(addr, addrlen, buf, sizeof(buf), NULL, 0,
878 NI_NUMERICHOST);
879 if (error) {
880 set_gaierror(error);
881 return NULL;
882 }
883 return PyUnicode_FromString(buf);
Guido van Rossum30a685f1991-06-27 15:51:29 +0000884}
885
886
Martin v. Löwis558d9bf2004-06-03 09:24:42 +0000887#ifdef USE_BLUETOOTH
888/* Convert a string representation of a Bluetooth address into a numeric
889 address. Returns the length (6), or raises an exception and returns -1 if
890 an error occurred. */
891
892static int
893setbdaddr(char *name, bdaddr_t *bdaddr)
894{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000895 unsigned int b0, b1, b2, b3, b4, b5;
896 char ch;
897 int n;
Martin v. Löwis558d9bf2004-06-03 09:24:42 +0000898
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000899 n = sscanf(name, "%X:%X:%X:%X:%X:%X%c",
900 &b5, &b4, &b3, &b2, &b1, &b0, &ch);
901 if (n == 6 && (b0 | b1 | b2 | b3 | b4 | b5) < 256) {
902 bdaddr->b[0] = b0;
903 bdaddr->b[1] = b1;
904 bdaddr->b[2] = b2;
905 bdaddr->b[3] = b3;
906 bdaddr->b[4] = b4;
907 bdaddr->b[5] = b5;
908 return 6;
909 } else {
910 PyErr_SetString(socket_error, "bad bluetooth address");
911 return -1;
912 }
Martin v. Löwis558d9bf2004-06-03 09:24:42 +0000913}
914
915/* Create a string representation of the Bluetooth address. This is always a
916 string of the form 'XX:XX:XX:XX:XX:XX' where XX is a two digit hexadecimal
917 value (zero padded if necessary). */
918
919static PyObject *
920makebdaddr(bdaddr_t *bdaddr)
921{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000922 char buf[(6 * 2) + 5 + 1];
Martin v. Löwis558d9bf2004-06-03 09:24:42 +0000923
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000924 sprintf(buf, "%02X:%02X:%02X:%02X:%02X:%02X",
925 bdaddr->b[5], bdaddr->b[4], bdaddr->b[3],
926 bdaddr->b[2], bdaddr->b[1], bdaddr->b[0]);
927 return PyUnicode_FromString(buf);
Martin v. Löwis558d9bf2004-06-03 09:24:42 +0000928}
929#endif
930
931
Guido van Rossum30a685f1991-06-27 15:51:29 +0000932/* Create an object representing the given socket address,
933 suitable for passing it back to bind(), connect() etc.
934 The family field of the sockaddr structure is inspected
935 to determine what kind of address it really is. */
936
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000937/*ARGSUSED*/
Guido van Rossum73624e91994-10-10 17:59:00 +0000938static PyObject *
Antoine Pitrou19467d22010-08-17 19:33:30 +0000939makesockaddr(SOCKET_T sockfd, struct sockaddr *addr, size_t addrlen, int proto)
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000940{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000941 if (addrlen == 0) {
942 /* No address -- may be recvfrom() from known socket */
943 Py_INCREF(Py_None);
944 return Py_None;
945 }
Guido van Rossum25bec8c1992-08-05 19:00:45 +0000946
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000947 switch (addr->sa_family) {
Guido van Rossum30a685f1991-06-27 15:51:29 +0000948
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000949 case AF_INET:
950 {
951 struct sockaddr_in *a;
952 PyObject *addrobj = makeipaddr(addr, sizeof(*a));
953 PyObject *ret = NULL;
954 if (addrobj) {
955 a = (struct sockaddr_in *)addr;
956 ret = Py_BuildValue("Oi", addrobj, ntohs(a->sin_port));
957 Py_DECREF(addrobj);
958 }
959 return ret;
960 }
Guido van Rossum30a685f1991-06-27 15:51:29 +0000961
Andrew MacIntyred12dfbb2004-04-04 07:13:49 +0000962#if defined(AF_UNIX)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000963 case AF_UNIX:
964 {
965 struct sockaddr_un *a = (struct sockaddr_un *) addr;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000966#ifdef linux
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000967 if (a->sun_path[0] == 0) { /* Linux abstract namespace */
968 addrlen -= offsetof(struct sockaddr_un, sun_path);
969 return PyBytes_FromStringAndSize(a->sun_path, addrlen);
970 }
971 else
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000972#endif /* linux */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000973 {
974 /* regular NULL-terminated string */
975 return PyUnicode_FromString(a->sun_path);
976 }
977 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000978#endif /* AF_UNIX */
979
Martin v. Löwis11017b12006-01-14 18:12:57 +0000980#if defined(AF_NETLINK)
981 case AF_NETLINK:
982 {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000983 struct sockaddr_nl *a = (struct sockaddr_nl *) addr;
984 return Py_BuildValue("II", a->nl_pid, a->nl_groups);
Martin v. Löwis11017b12006-01-14 18:12:57 +0000985 }
986#endif /* AF_NETLINK */
987
Martin v. Löwis44ddbde2001-12-02 10:15:37 +0000988#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000989 case AF_INET6:
990 {
991 struct sockaddr_in6 *a;
992 PyObject *addrobj = makeipaddr(addr, sizeof(*a));
993 PyObject *ret = NULL;
994 if (addrobj) {
995 a = (struct sockaddr_in6 *)addr;
996 ret = Py_BuildValue("Oiii",
997 addrobj,
998 ntohs(a->sin6_port),
999 a->sin6_flowinfo,
1000 a->sin6_scope_id);
1001 Py_DECREF(addrobj);
1002 }
1003 return ret;
1004 }
Jeremy Hylton22308652001-02-02 03:23:09 +00001005#endif
Guido van Rossum30a685f1991-06-27 15:51:29 +00001006
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001007#ifdef USE_BLUETOOTH
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001008 case AF_BLUETOOTH:
1009 switch (proto) {
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001010
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001011 case BTPROTO_L2CAP:
1012 {
1013 struct sockaddr_l2 *a = (struct sockaddr_l2 *) addr;
1014 PyObject *addrobj = makebdaddr(&_BT_L2_MEMB(a, bdaddr));
1015 PyObject *ret = NULL;
1016 if (addrobj) {
1017 ret = Py_BuildValue("Oi",
1018 addrobj,
1019 _BT_L2_MEMB(a, psm));
1020 Py_DECREF(addrobj);
1021 }
1022 return ret;
1023 }
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001024
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001025 case BTPROTO_RFCOMM:
1026 {
1027 struct sockaddr_rc *a = (struct sockaddr_rc *) addr;
1028 PyObject *addrobj = makebdaddr(&_BT_RC_MEMB(a, bdaddr));
1029 PyObject *ret = NULL;
1030 if (addrobj) {
1031 ret = Py_BuildValue("Oi",
1032 addrobj,
1033 _BT_RC_MEMB(a, channel));
1034 Py_DECREF(addrobj);
1035 }
1036 return ret;
1037 }
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001038
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001039 case BTPROTO_HCI:
1040 {
1041 struct sockaddr_hci *a = (struct sockaddr_hci *) addr;
1042 PyObject *ret = NULL;
1043 ret = Py_BuildValue("i", _BT_HCI_MEMB(a, dev));
1044 return ret;
1045 }
Thomas Wouterscf297e42007-02-23 15:07:44 +00001046
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001047#if !defined(__FreeBSD__)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001048 case BTPROTO_SCO:
1049 {
1050 struct sockaddr_sco *a = (struct sockaddr_sco *) addr;
1051 return makebdaddr(&_BT_SCO_MEMB(a, bdaddr));
1052 }
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001053#endif
1054
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001055 default:
1056 PyErr_SetString(PyExc_ValueError,
1057 "Unknown Bluetooth protocol");
1058 return NULL;
1059 }
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001060#endif
1061
Martin v. Löwis1ba3fd52001-08-10 20:29:40 +00001062#ifdef HAVE_NETPACKET_PACKET_H
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001063 case AF_PACKET:
1064 {
1065 struct sockaddr_ll *a = (struct sockaddr_ll *)addr;
1066 char *ifname = "";
1067 struct ifreq ifr;
1068 /* need to look up interface name give index */
1069 if (a->sll_ifindex) {
1070 ifr.ifr_ifindex = a->sll_ifindex;
1071 if (ioctl(sockfd, SIOCGIFNAME, &ifr) == 0)
1072 ifname = ifr.ifr_name;
1073 }
1074 return Py_BuildValue("shbhy#",
1075 ifname,
1076 ntohs(a->sll_protocol),
1077 a->sll_pkttype,
1078 a->sll_hatype,
1079 a->sll_addr,
1080 a->sll_halen);
1081 }
Jeremy Hylton22308652001-02-02 03:23:09 +00001082#endif
Guido van Rossum48a680c2001-03-02 06:34:14 +00001083
Christian Heimes043d6f62008-01-07 17:19:16 +00001084#ifdef HAVE_LINUX_TIPC_H
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001085 case AF_TIPC:
1086 {
1087 struct sockaddr_tipc *a = (struct sockaddr_tipc *) addr;
1088 if (a->addrtype == TIPC_ADDR_NAMESEQ) {
1089 return Py_BuildValue("IIIII",
1090 a->addrtype,
1091 a->addr.nameseq.type,
1092 a->addr.nameseq.lower,
1093 a->addr.nameseq.upper,
1094 a->scope);
1095 } else if (a->addrtype == TIPC_ADDR_NAME) {
1096 return Py_BuildValue("IIIII",
1097 a->addrtype,
1098 a->addr.name.name.type,
1099 a->addr.name.name.instance,
1100 a->addr.name.name.instance,
1101 a->scope);
1102 } else if (a->addrtype == TIPC_ADDR_ID) {
1103 return Py_BuildValue("IIIII",
1104 a->addrtype,
1105 a->addr.id.node,
1106 a->addr.id.ref,
1107 0,
1108 a->scope);
1109 } else {
1110 PyErr_SetString(PyExc_ValueError,
1111 "Invalid address type");
1112 return NULL;
1113 }
1114 }
Christian Heimes043d6f62008-01-07 17:19:16 +00001115#endif
1116
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001117 /* More cases here... */
Guido van Rossum30a685f1991-06-27 15:51:29 +00001118
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001119 default:
1120 /* If we don't know the address family, don't raise an
1121 exception -- return it as an (int, bytes) tuple. */
1122 return Py_BuildValue("iy#",
1123 addr->sa_family,
1124 addr->sa_data,
1125 sizeof(addr->sa_data));
Guido van Rossum25bec8c1992-08-05 19:00:45 +00001126
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001127 }
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001128}
1129
Guido van Rossum30a685f1991-06-27 15:51:29 +00001130
1131/* Parse a socket address argument according to the socket object's
1132 address family. Return 1 if the address was in the proper format,
1133 0 of not. The address is returned through addr_ret, its length
1134 through len_ret. */
1135
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001136static int
Guido van Rossum48a680c2001-03-02 06:34:14 +00001137getsockaddrarg(PySocketSockObject *s, PyObject *args,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001138 struct sockaddr *addr_ret, int *len_ret)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001139{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001140 switch (s->sock_family) {
Guido van Rossum30a685f1991-06-27 15:51:29 +00001141
Andrew MacIntyred12dfbb2004-04-04 07:13:49 +00001142#if defined(AF_UNIX)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001143 case AF_UNIX:
1144 {
1145 struct sockaddr_un* addr;
1146 char *path;
1147 int len;
1148 if (!PyArg_Parse(args, "s#", &path, &len))
1149 return 0;
Thomas Wouters89f507f2006-12-13 04:49:30 +00001150
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001151 addr = (struct sockaddr_un*)addr_ret;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001152#ifdef linux
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001153 if (len > 0 && path[0] == 0) {
1154 /* Linux abstract namespace extension */
1155 if (len > sizeof addr->sun_path) {
1156 PyErr_SetString(socket_error,
1157 "AF_UNIX path too long");
1158 return 0;
1159 }
1160 }
1161 else
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001162#endif /* linux */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001163 {
1164 /* regular NULL-terminated string */
1165 if (len >= sizeof addr->sun_path) {
1166 PyErr_SetString(socket_error,
1167 "AF_UNIX path too long");
1168 return 0;
1169 }
1170 addr->sun_path[len] = 0;
1171 }
1172 addr->sun_family = s->sock_family;
1173 memcpy(addr->sun_path, path, len);
Andrew MacIntyredaedf212004-04-11 12:03:57 +00001174#if defined(PYOS_OS2)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001175 *len_ret = sizeof(*addr);
Andrew MacIntyredaedf212004-04-11 12:03:57 +00001176#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001177 *len_ret = len + offsetof(struct sockaddr_un, sun_path);
Andrew MacIntyredaedf212004-04-11 12:03:57 +00001178#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001179 return 1;
1180 }
Guido van Rossumb6775db1994-08-01 11:34:53 +00001181#endif /* AF_UNIX */
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001182
Martin v. Löwis11017b12006-01-14 18:12:57 +00001183#if defined(AF_NETLINK)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001184 case AF_NETLINK:
1185 {
1186 struct sockaddr_nl* addr;
1187 int pid, groups;
1188 addr = (struct sockaddr_nl *)addr_ret;
1189 if (!PyTuple_Check(args)) {
1190 PyErr_Format(
1191 PyExc_TypeError,
1192 "getsockaddrarg: "
1193 "AF_NETLINK address must be tuple, not %.500s",
1194 Py_TYPE(args)->tp_name);
1195 return 0;
1196 }
1197 if (!PyArg_ParseTuple(args, "II:getsockaddrarg", &pid, &groups))
1198 return 0;
1199 addr->nl_family = AF_NETLINK;
1200 addr->nl_pid = pid;
1201 addr->nl_groups = groups;
1202 *len_ret = sizeof(*addr);
1203 return 1;
1204 }
Martin v. Löwis11017b12006-01-14 18:12:57 +00001205#endif
1206
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001207 case AF_INET:
1208 {
1209 struct sockaddr_in* addr;
1210 char *host;
1211 int port, result;
1212 if (!PyTuple_Check(args)) {
1213 PyErr_Format(
1214 PyExc_TypeError,
1215 "getsockaddrarg: "
1216 "AF_INET address must be tuple, not %.500s",
1217 Py_TYPE(args)->tp_name);
1218 return 0;
1219 }
1220 if (!PyArg_ParseTuple(args, "eti:getsockaddrarg",
1221 "idna", &host, &port))
1222 return 0;
1223 addr=(struct sockaddr_in*)addr_ret;
1224 result = setipaddr(host, (struct sockaddr *)addr,
1225 sizeof(*addr), AF_INET);
1226 PyMem_Free(host);
1227 if (result < 0)
1228 return 0;
1229 if (port < 0 || port > 0xffff) {
1230 PyErr_SetString(
1231 PyExc_OverflowError,
1232 "getsockaddrarg: port must be 0-65535.");
1233 return 0;
1234 }
1235 addr->sin_family = AF_INET;
1236 addr->sin_port = htons((short)port);
1237 *len_ret = sizeof *addr;
1238 return 1;
1239 }
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001240
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00001241#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001242 case AF_INET6:
1243 {
1244 struct sockaddr_in6* addr;
1245 char *host;
1246 int port, flowinfo, scope_id, result;
1247 flowinfo = scope_id = 0;
1248 if (!PyTuple_Check(args)) {
1249 PyErr_Format(
1250 PyExc_TypeError,
1251 "getsockaddrarg: "
1252 "AF_INET6 address must be tuple, not %.500s",
1253 Py_TYPE(args)->tp_name);
1254 return 0;
1255 }
1256 if (!PyArg_ParseTuple(args, "eti|ii",
1257 "idna", &host, &port, &flowinfo,
1258 &scope_id)) {
1259 return 0;
1260 }
1261 addr = (struct sockaddr_in6*)addr_ret;
1262 result = setipaddr(host, (struct sockaddr *)addr,
1263 sizeof(*addr), AF_INET6);
1264 PyMem_Free(host);
1265 if (result < 0)
1266 return 0;
1267 if (port < 0 || port > 0xffff) {
1268 PyErr_SetString(
1269 PyExc_OverflowError,
1270 "getsockaddrarg: port must be 0-65535.");
1271 return 0;
1272 }
1273 addr->sin6_family = s->sock_family;
1274 addr->sin6_port = htons((short)port);
1275 addr->sin6_flowinfo = flowinfo;
1276 addr->sin6_scope_id = scope_id;
1277 *len_ret = sizeof *addr;
1278 return 1;
1279 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00001280#endif
1281
Hye-Shik Chang81268602004-02-02 06:05:24 +00001282#ifdef USE_BLUETOOTH
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001283 case AF_BLUETOOTH:
1284 {
1285 switch (s->sock_proto) {
1286 case BTPROTO_L2CAP:
1287 {
1288 struct sockaddr_l2 *addr;
1289 char *straddr;
Martin v. Löwis12af0482004-01-31 12:34:17 +00001290
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001291 addr = (struct sockaddr_l2 *)addr_ret;
1292 memset(addr, 0, sizeof(struct sockaddr_l2));
1293 _BT_L2_MEMB(addr, family) = AF_BLUETOOTH;
1294 if (!PyArg_ParseTuple(args, "si", &straddr,
1295 &_BT_L2_MEMB(addr, psm))) {
1296 PyErr_SetString(socket_error, "getsockaddrarg: "
1297 "wrong format");
1298 return 0;
1299 }
1300 if (setbdaddr(straddr, &_BT_L2_MEMB(addr, bdaddr)) < 0)
1301 return 0;
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001302
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001303 *len_ret = sizeof *addr;
1304 return 1;
1305 }
1306 case BTPROTO_RFCOMM:
1307 {
1308 struct sockaddr_rc *addr;
1309 char *straddr;
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001310
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001311 addr = (struct sockaddr_rc *)addr_ret;
1312 _BT_RC_MEMB(addr, family) = AF_BLUETOOTH;
1313 if (!PyArg_ParseTuple(args, "si", &straddr,
1314 &_BT_RC_MEMB(addr, channel))) {
1315 PyErr_SetString(socket_error, "getsockaddrarg: "
1316 "wrong format");
1317 return 0;
1318 }
1319 if (setbdaddr(straddr, &_BT_RC_MEMB(addr, bdaddr)) < 0)
1320 return 0;
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001321
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001322 *len_ret = sizeof *addr;
1323 return 1;
1324 }
1325 case BTPROTO_HCI:
1326 {
1327 struct sockaddr_hci *addr = (struct sockaddr_hci *)addr_ret;
1328 _BT_HCI_MEMB(addr, family) = AF_BLUETOOTH;
1329 if (!PyArg_ParseTuple(args, "i", &_BT_HCI_MEMB(addr, dev))) {
1330 PyErr_SetString(socket_error, "getsockaddrarg: "
1331 "wrong format");
1332 return 0;
1333 }
1334 *len_ret = sizeof *addr;
1335 return 1;
1336 }
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001337#if !defined(__FreeBSD__)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001338 case BTPROTO_SCO:
1339 {
1340 struct sockaddr_sco *addr;
1341 char *straddr;
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001342
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001343 addr = (struct sockaddr_sco *)addr_ret;
1344 _BT_SCO_MEMB(addr, family) = AF_BLUETOOTH;
1345 if (!PyBytes_Check(args)) {
1346 PyErr_SetString(socket_error, "getsockaddrarg: "
1347 "wrong format");
1348 return 0;
1349 }
1350 straddr = PyBytes_AS_STRING(args);
1351 if (setbdaddr(straddr, &_BT_SCO_MEMB(addr, bdaddr)) < 0)
1352 return 0;
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001353
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001354 *len_ret = sizeof *addr;
1355 return 1;
1356 }
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001357#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001358 default:
1359 PyErr_SetString(socket_error, "getsockaddrarg: unknown Bluetooth protocol");
1360 return 0;
1361 }
1362 }
Martin v. Löwis12af0482004-01-31 12:34:17 +00001363#endif
1364
Martin v. Löwis1ba3fd52001-08-10 20:29:40 +00001365#ifdef HAVE_NETPACKET_PACKET_H
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001366 case AF_PACKET:
1367 {
1368 struct sockaddr_ll* addr;
1369 struct ifreq ifr;
1370 char *interfaceName;
1371 int protoNumber;
1372 int hatype = 0;
1373 int pkttype = 0;
1374 char *haddr = NULL;
1375 unsigned int halen = 0;
Guido van Rossum48a680c2001-03-02 06:34:14 +00001376
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001377 if (!PyTuple_Check(args)) {
1378 PyErr_Format(
1379 PyExc_TypeError,
1380 "getsockaddrarg: "
1381 "AF_PACKET address must be tuple, not %.500s",
1382 Py_TYPE(args)->tp_name);
1383 return 0;
1384 }
1385 if (!PyArg_ParseTuple(args, "si|iiy#", &interfaceName,
1386 &protoNumber, &pkttype, &hatype,
1387 &haddr, &halen))
1388 return 0;
1389 strncpy(ifr.ifr_name, interfaceName, sizeof(ifr.ifr_name));
1390 ifr.ifr_name[(sizeof(ifr.ifr_name))-1] = '\0';
1391 if (ioctl(s->sock_fd, SIOCGIFINDEX, &ifr) < 0) {
1392 s->errorhandler();
1393 return 0;
1394 }
1395 if (halen > 8) {
1396 PyErr_SetString(PyExc_ValueError,
1397 "Hardware address must be 8 bytes or less");
1398 return 0;
1399 }
1400 if (protoNumber < 0 || protoNumber > 0xffff) {
1401 PyErr_SetString(
1402 PyExc_OverflowError,
1403 "getsockaddrarg: protoNumber must be 0-65535.");
1404 return 0;
1405 }
1406 addr = (struct sockaddr_ll*)addr_ret;
1407 addr->sll_family = AF_PACKET;
1408 addr->sll_protocol = htons((short)protoNumber);
1409 addr->sll_ifindex = ifr.ifr_ifindex;
1410 addr->sll_pkttype = pkttype;
1411 addr->sll_hatype = hatype;
1412 if (halen != 0) {
1413 memcpy(&addr->sll_addr, haddr, halen);
1414 }
1415 addr->sll_halen = halen;
1416 *len_ret = sizeof *addr;
1417 return 1;
1418 }
Guido van Rossum48a680c2001-03-02 06:34:14 +00001419#endif
1420
Christian Heimes043d6f62008-01-07 17:19:16 +00001421#ifdef HAVE_LINUX_TIPC_H
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001422 case AF_TIPC:
1423 {
1424 unsigned int atype, v1, v2, v3;
1425 unsigned int scope = TIPC_CLUSTER_SCOPE;
1426 struct sockaddr_tipc *addr;
Christian Heimes043d6f62008-01-07 17:19:16 +00001427
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001428 if (!PyTuple_Check(args)) {
1429 PyErr_Format(
1430 PyExc_TypeError,
1431 "getsockaddrarg: "
1432 "AF_TIPC address must be tuple, not %.500s",
1433 Py_TYPE(args)->tp_name);
1434 return 0;
1435 }
Christian Heimes043d6f62008-01-07 17:19:16 +00001436
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001437 if (!PyArg_ParseTuple(args,
1438 "IIII|I;Invalid TIPC address format",
1439 &atype, &v1, &v2, &v3, &scope))
1440 return 0;
Christian Heimes043d6f62008-01-07 17:19:16 +00001441
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001442 addr = (struct sockaddr_tipc *) addr_ret;
1443 memset(addr, 0, sizeof(struct sockaddr_tipc));
Christian Heimes043d6f62008-01-07 17:19:16 +00001444
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001445 addr->family = AF_TIPC;
1446 addr->scope = scope;
1447 addr->addrtype = atype;
Christian Heimes043d6f62008-01-07 17:19:16 +00001448
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001449 if (atype == TIPC_ADDR_NAMESEQ) {
1450 addr->addr.nameseq.type = v1;
1451 addr->addr.nameseq.lower = v2;
1452 addr->addr.nameseq.upper = v3;
1453 } else if (atype == TIPC_ADDR_NAME) {
1454 addr->addr.name.name.type = v1;
1455 addr->addr.name.name.instance = v2;
1456 } else if (atype == TIPC_ADDR_ID) {
1457 addr->addr.id.node = v1;
1458 addr->addr.id.ref = v2;
1459 } else {
1460 /* Shouldn't happen */
1461 PyErr_SetString(PyExc_TypeError, "Invalid address type");
1462 return 0;
1463 }
Christian Heimes043d6f62008-01-07 17:19:16 +00001464
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001465 *len_ret = sizeof(*addr);
Christian Heimes043d6f62008-01-07 17:19:16 +00001466
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001467 return 1;
1468 }
Christian Heimes043d6f62008-01-07 17:19:16 +00001469#endif
1470
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001471 /* More cases here... */
Guido van Rossum30a685f1991-06-27 15:51:29 +00001472
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001473 default:
1474 PyErr_SetString(socket_error, "getsockaddrarg: bad family");
1475 return 0;
Guido van Rossum30a685f1991-06-27 15:51:29 +00001476
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001477 }
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001478}
1479
Guido van Rossum30a685f1991-06-27 15:51:29 +00001480
Guido van Rossum48a680c2001-03-02 06:34:14 +00001481/* Get the address length according to the socket object's address family.
Guido van Rossum710e1df1992-06-12 10:39:36 +00001482 Return 1 if the family is known, 0 otherwise. The length is returned
1483 through len_ret. */
1484
1485static int
Peter Schneider-Kamp618e25d2000-07-11 23:00:12 +00001486getsockaddrlen(PySocketSockObject *s, socklen_t *len_ret)
Guido van Rossum710e1df1992-06-12 10:39:36 +00001487{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001488 switch (s->sock_family) {
Guido van Rossum710e1df1992-06-12 10:39:36 +00001489
Andrew MacIntyred12dfbb2004-04-04 07:13:49 +00001490#if defined(AF_UNIX)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001491 case AF_UNIX:
1492 {
1493 *len_ret = sizeof (struct sockaddr_un);
1494 return 1;
1495 }
Guido van Rossumb6775db1994-08-01 11:34:53 +00001496#endif /* AF_UNIX */
Martin v. Löwis11017b12006-01-14 18:12:57 +00001497#if defined(AF_NETLINK)
1498 case AF_NETLINK:
1499 {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001500 *len_ret = sizeof (struct sockaddr_nl);
1501 return 1;
Martin v. Löwis11017b12006-01-14 18:12:57 +00001502 }
1503#endif
Guido van Rossum710e1df1992-06-12 10:39:36 +00001504
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001505 case AF_INET:
1506 {
1507 *len_ret = sizeof (struct sockaddr_in);
1508 return 1;
1509 }
Guido van Rossum710e1df1992-06-12 10:39:36 +00001510
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00001511#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001512 case AF_INET6:
1513 {
1514 *len_ret = sizeof (struct sockaddr_in6);
1515 return 1;
1516 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00001517#endif
1518
Hye-Shik Chang81268602004-02-02 06:05:24 +00001519#ifdef USE_BLUETOOTH
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001520 case AF_BLUETOOTH:
1521 {
1522 switch(s->sock_proto)
1523 {
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001524
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001525 case BTPROTO_L2CAP:
1526 *len_ret = sizeof (struct sockaddr_l2);
1527 return 1;
1528 case BTPROTO_RFCOMM:
1529 *len_ret = sizeof (struct sockaddr_rc);
1530 return 1;
1531 case BTPROTO_HCI:
1532 *len_ret = sizeof (struct sockaddr_hci);
1533 return 1;
Hye-Shik Chang81268602004-02-02 06:05:24 +00001534#if !defined(__FreeBSD__)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001535 case BTPROTO_SCO:
1536 *len_ret = sizeof (struct sockaddr_sco);
1537 return 1;
Hye-Shik Chang81268602004-02-02 06:05:24 +00001538#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001539 default:
1540 PyErr_SetString(socket_error, "getsockaddrlen: "
1541 "unknown BT protocol");
1542 return 0;
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001543
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001544 }
1545 }
Martin v. Löwis12af0482004-01-31 12:34:17 +00001546#endif
1547
Martin v. Löwis1ba3fd52001-08-10 20:29:40 +00001548#ifdef HAVE_NETPACKET_PACKET_H
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001549 case AF_PACKET:
1550 {
1551 *len_ret = sizeof (struct sockaddr_ll);
1552 return 1;
1553 }
Jeremy Hylton22308652001-02-02 03:23:09 +00001554#endif
Guido van Rossum48a680c2001-03-02 06:34:14 +00001555
Christian Heimes043d6f62008-01-07 17:19:16 +00001556#ifdef HAVE_LINUX_TIPC_H
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001557 case AF_TIPC:
1558 {
1559 *len_ret = sizeof (struct sockaddr_tipc);
1560 return 1;
1561 }
Christian Heimes043d6f62008-01-07 17:19:16 +00001562#endif
1563
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001564 /* More cases here... */
Guido van Rossum710e1df1992-06-12 10:39:36 +00001565
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001566 default:
1567 PyErr_SetString(socket_error, "getsockaddrlen: bad family");
1568 return 0;
Guido van Rossum710e1df1992-06-12 10:39:36 +00001569
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001570 }
Guido van Rossum710e1df1992-06-12 10:39:36 +00001571}
1572
1573
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00001574/* s._accept() -> (fd, address) */
Guido van Rossum30a685f1991-06-27 15:51:29 +00001575
Guido van Rossum73624e91994-10-10 17:59:00 +00001576static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001577sock_accept(PySocketSockObject *s)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001578{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001579 sock_addr_t addrbuf;
1580 SOCKET_T newfd = INVALID_SOCKET;
1581 socklen_t addrlen;
1582 PyObject *sock = NULL;
1583 PyObject *addr = NULL;
1584 PyObject *res = NULL;
1585 int timeout;
Barry Warsaw752300b1997-01-03 17:18:10 +00001586
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001587 if (!getsockaddrlen(s, &addrlen))
1588 return NULL;
1589 memset(&addrbuf, 0, addrlen);
Guido van Rossum67f7a382002-06-06 21:08:16 +00001590
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001591 if (!IS_SELECTABLE(s))
1592 return select_error();
Neal Norwitz082b2df2006-02-07 07:04:46 +00001593
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001594 Py_BEGIN_ALLOW_THREADS
1595 timeout = internal_select(s, 0);
1596 if (!timeout)
1597 newfd = accept(s->sock_fd, SAS2SA(&addrbuf), &addrlen);
1598 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00001599
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001600 if (timeout == 1) {
1601 PyErr_SetString(socket_timeout, "timed out");
1602 return NULL;
1603 }
Raymond Hettingeref7343c2003-06-29 03:08:05 +00001604
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001605 if (newfd == INVALID_SOCKET)
1606 return s->errorhandler();
Barry Warsaw752300b1997-01-03 17:18:10 +00001607
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001608 sock = PyLong_FromSocket_t(newfd);
1609 if (sock == NULL) {
1610 SOCKETCLOSE(newfd);
1611 goto finally;
1612 }
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00001613
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001614 addr = makesockaddr(s->sock_fd, SAS2SA(&addrbuf),
1615 addrlen, s->sock_proto);
1616 if (addr == NULL)
1617 goto finally;
Barry Warsaw752300b1997-01-03 17:18:10 +00001618
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001619 res = PyTuple_Pack(2, sock, addr);
Barry Warsaw752300b1997-01-03 17:18:10 +00001620
Guido van Rossum67f7a382002-06-06 21:08:16 +00001621finally:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001622 Py_XDECREF(sock);
1623 Py_XDECREF(addr);
1624 return res;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001625}
1626
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001627PyDoc_STRVAR(accept_doc,
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00001628"_accept() -> (integer, address info)\n\
Guido van Rossum82a5c661998-07-07 20:45:43 +00001629\n\
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00001630Wait for an incoming connection. Return a new socket file descriptor\n\
1631representing the connection, and the address of the client.\n\
1632For IP sockets, the address info is a pair (hostaddr, port).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001633
Guido van Rossum11ba0942002-06-13 15:07:44 +00001634/* s.setblocking(flag) method. Argument:
1635 False -- non-blocking mode; same as settimeout(0)
1636 True -- blocking mode; same as settimeout(None)
1637*/
Guido van Rossume4485b01994-09-07 14:32:49 +00001638
Guido van Rossum73624e91994-10-10 17:59:00 +00001639static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001640sock_setblocking(PySocketSockObject *s, PyObject *arg)
Guido van Rossume4485b01994-09-07 14:32:49 +00001641{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001642 int block;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001643
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001644 block = PyLong_AsLong(arg);
1645 if (block == -1 && PyErr_Occurred())
1646 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001647
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001648 s->sock_timeout = block ? -1.0 : 0.0;
1649 internal_setblocking(s, block);
Guido van Rossume4485b01994-09-07 14:32:49 +00001650
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001651 Py_INCREF(Py_None);
1652 return Py_None;
Guido van Rossume4485b01994-09-07 14:32:49 +00001653}
Guido van Rossume4485b01994-09-07 14:32:49 +00001654
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001655PyDoc_STRVAR(setblocking_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001656"setblocking(flag)\n\
1657\n\
1658Set the socket to blocking (flag is true) or non-blocking (false).\n\
Guido van Rossum11ba0942002-06-13 15:07:44 +00001659setblocking(True) is equivalent to settimeout(None);\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001660setblocking(False) is equivalent to settimeout(0.0).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001661
Guido van Rossum11ba0942002-06-13 15:07:44 +00001662/* s.settimeout(timeout) method. Argument:
1663 None -- no timeout, blocking mode; same as setblocking(True)
1664 0.0 -- non-blocking mode; same as setblocking(False)
1665 > 0 -- timeout mode; operations time out after timeout seconds
1666 < 0 -- illegal; raises an exception
1667*/
Guido van Rossum67f7a382002-06-06 21:08:16 +00001668static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001669sock_settimeout(PySocketSockObject *s, PyObject *arg)
Guido van Rossum67f7a382002-06-06 21:08:16 +00001670{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001671 double timeout;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001672
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001673 if (arg == Py_None)
1674 timeout = -1.0;
1675 else {
1676 timeout = PyFloat_AsDouble(arg);
1677 if (timeout < 0.0) {
1678 if (!PyErr_Occurred())
1679 PyErr_SetString(PyExc_ValueError,
1680 "Timeout value out of range");
1681 return NULL;
1682 }
1683 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00001684
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001685 s->sock_timeout = timeout;
1686 internal_setblocking(s, timeout < 0.0);
Guido van Rossum67f7a382002-06-06 21:08:16 +00001687
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001688 Py_INCREF(Py_None);
1689 return Py_None;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001690}
1691
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001692PyDoc_STRVAR(settimeout_doc,
Guido van Rossum3eede5a2002-06-07 02:08:35 +00001693"settimeout(timeout)\n\
Guido van Rossum67f7a382002-06-06 21:08:16 +00001694\n\
Guido van Rossum11ba0942002-06-13 15:07:44 +00001695Set a timeout on socket operations. 'timeout' can be a float,\n\
1696giving in seconds, or None. Setting a timeout of None disables\n\
1697the timeout feature and is equivalent to setblocking(1).\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001698Setting a timeout of zero is the same as setblocking(0).");
Guido van Rossum67f7a382002-06-06 21:08:16 +00001699
Guido van Rossum3eede5a2002-06-07 02:08:35 +00001700/* s.gettimeout() method.
1701 Returns the timeout associated with a socket. */
Guido van Rossum67f7a382002-06-06 21:08:16 +00001702static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001703sock_gettimeout(PySocketSockObject *s)
Guido van Rossum67f7a382002-06-06 21:08:16 +00001704{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001705 if (s->sock_timeout < 0.0) {
1706 Py_INCREF(Py_None);
1707 return Py_None;
1708 }
1709 else
1710 return PyFloat_FromDouble(s->sock_timeout);
Guido van Rossum67f7a382002-06-06 21:08:16 +00001711}
1712
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001713PyDoc_STRVAR(gettimeout_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00001714"gettimeout() -> timeout\n\
Guido van Rossum67f7a382002-06-06 21:08:16 +00001715\n\
1716Returns the timeout in floating seconds associated with socket \n\
1717operations. A timeout of None indicates that timeouts on socket \n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001718operations are disabled.");
Guido van Rossume4485b01994-09-07 14:32:49 +00001719
Guido van Rossumaee08791992-09-08 09:05:33 +00001720/* s.setsockopt() method.
1721 With an integer third argument, sets an integer option.
1722 With a string third argument, sets an option from a buffer;
1723 use optional built-in module 'struct' to encode the string. */
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001724
Guido van Rossum73624e91994-10-10 17:59:00 +00001725static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001726sock_setsockopt(PySocketSockObject *s, PyObject *args)
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001727{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001728 int level;
1729 int optname;
1730 int res;
1731 char *buf;
1732 int buflen;
1733 int flag;
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001734
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001735 if (PyArg_ParseTuple(args, "iii:setsockopt",
1736 &level, &optname, &flag)) {
1737 buf = (char *) &flag;
1738 buflen = sizeof flag;
1739 }
1740 else {
1741 PyErr_Clear();
1742 if (!PyArg_ParseTuple(args, "iiy#:setsockopt",
1743 &level, &optname, &buf, &buflen))
1744 return NULL;
1745 }
1746 res = setsockopt(s->sock_fd, level, optname, (void *)buf, buflen);
1747 if (res < 0)
1748 return s->errorhandler();
1749 Py_INCREF(Py_None);
1750 return Py_None;
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001751}
1752
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001753PyDoc_STRVAR(setsockopt_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001754"setsockopt(level, option, value)\n\
1755\n\
1756Set a socket option. See the Unix manual for level and option.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001757The value argument can either be an integer or a string.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001758
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001759
Guido van Rossumaee08791992-09-08 09:05:33 +00001760/* s.getsockopt() method.
1761 With two arguments, retrieves an integer option.
1762 With a third integer argument, retrieves a string buffer of that size;
1763 use optional built-in module 'struct' to decode the string. */
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001764
Guido van Rossum73624e91994-10-10 17:59:00 +00001765static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001766sock_getsockopt(PySocketSockObject *s, PyObject *args)
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001767{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001768 int level;
1769 int optname;
1770 int res;
1771 PyObject *buf;
1772 socklen_t buflen = 0;
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001773
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001774 if (!PyArg_ParseTuple(args, "ii|i:getsockopt",
1775 &level, &optname, &buflen))
1776 return NULL;
Guido van Rossum48a680c2001-03-02 06:34:14 +00001777
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001778 if (buflen == 0) {
1779 int flag = 0;
1780 socklen_t flagsize = sizeof flag;
1781 res = getsockopt(s->sock_fd, level, optname,
1782 (void *)&flag, &flagsize);
1783 if (res < 0)
1784 return s->errorhandler();
1785 return PyLong_FromLong(flag);
1786 }
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001787#ifdef __VMS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001788 /* socklen_t is unsigned so no negative test is needed,
1789 test buflen == 0 is previously done */
1790 if (buflen > 1024) {
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001791#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001792 if (buflen <= 0 || buflen > 1024) {
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001793#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001794 PyErr_SetString(socket_error,
1795 "getsockopt buflen out of range");
1796 return NULL;
1797 }
1798 buf = PyBytes_FromStringAndSize((char *)NULL, buflen);
1799 if (buf == NULL)
1800 return NULL;
1801 res = getsockopt(s->sock_fd, level, optname,
1802 (void *)PyBytes_AS_STRING(buf), &buflen);
1803 if (res < 0) {
1804 Py_DECREF(buf);
1805 return s->errorhandler();
1806 }
1807 _PyBytes_Resize(&buf, buflen);
1808 return buf;
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001809}
1810
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001811PyDoc_STRVAR(getsockopt_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001812"getsockopt(level, option[, buffersize]) -> value\n\
1813\n\
1814Get a socket option. See the Unix manual for level and option.\n\
1815If a nonzero buffersize argument is given, the return value is a\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001816string of that length; otherwise it is an integer.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001817
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001818
Fred Drake728819a2000-07-01 03:40:12 +00001819/* s.bind(sockaddr) method */
Guido van Rossum30a685f1991-06-27 15:51:29 +00001820
Guido van Rossum73624e91994-10-10 17:59:00 +00001821static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001822sock_bind(PySocketSockObject *s, PyObject *addro)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001823{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001824 sock_addr_t addrbuf;
1825 int addrlen;
1826 int res;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00001827
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001828 if (!getsockaddrarg(s, addro, SAS2SA(&addrbuf), &addrlen))
1829 return NULL;
1830 Py_BEGIN_ALLOW_THREADS
1831 res = bind(s->sock_fd, SAS2SA(&addrbuf), addrlen);
1832 Py_END_ALLOW_THREADS
1833 if (res < 0)
1834 return s->errorhandler();
1835 Py_INCREF(Py_None);
1836 return Py_None;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001837}
1838
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001839PyDoc_STRVAR(bind_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001840"bind(address)\n\
1841\n\
1842Bind the socket to a local address. For IP sockets, the address is a\n\
Jeremy Hylton22308652001-02-02 03:23:09 +00001843pair (host, port); the host must refer to the local host. For raw packet\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001844sockets the address is a tuple (ifname, proto [,pkttype [,hatype]])");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001845
Guido van Rossum30a685f1991-06-27 15:51:29 +00001846
1847/* s.close() method.
1848 Set the file descriptor to -1 so operations tried subsequently
1849 will surely fail. */
1850
Guido van Rossum73624e91994-10-10 17:59:00 +00001851static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001852sock_close(PySocketSockObject *s)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001853{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001854 SOCKET_T fd;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00001855
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001856 if ((fd = s->sock_fd) != -1) {
1857 s->sock_fd = -1;
1858 Py_BEGIN_ALLOW_THREADS
1859 (void) SOCKETCLOSE(fd);
1860 Py_END_ALLOW_THREADS
1861 }
1862 Py_INCREF(Py_None);
1863 return Py_None;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001864}
1865
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001866PyDoc_STRVAR(close_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001867"close()\n\
1868\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001869Close the socket. It cannot be used after this call.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001870
Antoine Pitroue43f9d02010-08-08 23:24:50 +00001871static PyObject *
Antoine Pitrou6e451df2010-08-09 20:39:54 +00001872sock_detach(PySocketSockObject *s)
Antoine Pitroue43f9d02010-08-08 23:24:50 +00001873{
Antoine Pitrou6e451df2010-08-09 20:39:54 +00001874 SOCKET_T fd = s->sock_fd;
Antoine Pitroue43f9d02010-08-08 23:24:50 +00001875 s->sock_fd = -1;
Antoine Pitrou6e451df2010-08-09 20:39:54 +00001876 return PyLong_FromSocket_t(fd);
Antoine Pitroue43f9d02010-08-08 23:24:50 +00001877}
1878
Antoine Pitrou6e451df2010-08-09 20:39:54 +00001879PyDoc_STRVAR(detach_doc,
1880"detach()\n\
Antoine Pitroue43f9d02010-08-08 23:24:50 +00001881\n\
1882Close the socket object without closing the underlying file descriptor.\
1883The object cannot be used after this call, but the file descriptor\
Antoine Pitrou6e451df2010-08-09 20:39:54 +00001884can be reused for other purposes. The file descriptor is returned.");
Antoine Pitroue43f9d02010-08-08 23:24:50 +00001885
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001886static int
Raymond Hettingeref7343c2003-06-29 03:08:05 +00001887internal_connect(PySocketSockObject *s, struct sockaddr *addr, int addrlen,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001888 int *timeoutp)
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001889{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001890 int res, timeout;
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001891
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001892 timeout = 0;
1893 res = connect(s->sock_fd, addr, addrlen);
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001894
1895#ifdef MS_WINDOWS
1896
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001897 if (s->sock_timeout > 0.0) {
1898 if (res < 0 && WSAGetLastError() == WSAEWOULDBLOCK &&
1899 IS_SELECTABLE(s)) {
1900 /* This is a mess. Best solution: trust select */
1901 fd_set fds;
1902 fd_set fds_exc;
1903 struct timeval tv;
1904 tv.tv_sec = (int)s->sock_timeout;
1905 tv.tv_usec = (int)((s->sock_timeout - tv.tv_sec) * 1e6);
1906 FD_ZERO(&fds);
1907 FD_SET(s->sock_fd, &fds);
1908 FD_ZERO(&fds_exc);
1909 FD_SET(s->sock_fd, &fds_exc);
Antoine Pitrou19467d22010-08-17 19:33:30 +00001910 res = select(Py_SAFE_DOWNCAST(s->sock_fd+1, SOCKET_T, int),
1911 NULL, &fds, &fds_exc, &tv);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001912 if (res == 0) {
1913 res = WSAEWOULDBLOCK;
1914 timeout = 1;
1915 } else if (res > 0) {
1916 if (FD_ISSET(s->sock_fd, &fds))
1917 /* The socket is in the writable set - this
1918 means connected */
1919 res = 0;
1920 else {
1921 /* As per MS docs, we need to call getsockopt()
1922 to get the underlying error */
1923 int res_size = sizeof res;
1924 /* It must be in the exception set */
1925 assert(FD_ISSET(s->sock_fd, &fds_exc));
1926 if (0 == getsockopt(s->sock_fd, SOL_SOCKET, SO_ERROR,
1927 (char *)&res, &res_size))
1928 /* getsockopt also clears WSAGetLastError,
1929 so reset it back. */
1930 WSASetLastError(res);
1931 else
1932 res = WSAGetLastError();
1933 }
1934 }
1935 /* else if (res < 0) an error occurred */
1936 }
1937 }
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001938
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001939 if (res < 0)
1940 res = WSAGetLastError();
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001941
1942#else
1943
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001944 if (s->sock_timeout > 0.0) {
1945 if (res < 0 && errno == EINPROGRESS && IS_SELECTABLE(s)) {
1946 timeout = internal_select(s, 1);
1947 if (timeout == 0) {
1948 /* Bug #1019808: in case of an EINPROGRESS,
1949 use getsockopt(SO_ERROR) to get the real
1950 error. */
1951 socklen_t res_size = sizeof res;
1952 (void)getsockopt(s->sock_fd, SOL_SOCKET,
1953 SO_ERROR, &res, &res_size);
1954 if (res == EISCONN)
1955 res = 0;
1956 errno = res;
1957 }
1958 else if (timeout == -1) {
1959 res = errno; /* had error */
1960 }
1961 else
1962 res = EWOULDBLOCK; /* timed out */
1963 }
1964 }
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001965
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001966 if (res < 0)
1967 res = errno;
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001968
1969#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001970 *timeoutp = timeout;
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001971
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001972 return res;
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001973}
Guido van Rossum30a685f1991-06-27 15:51:29 +00001974
Fred Drake728819a2000-07-01 03:40:12 +00001975/* s.connect(sockaddr) method */
Guido van Rossum30a685f1991-06-27 15:51:29 +00001976
Guido van Rossum73624e91994-10-10 17:59:00 +00001977static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001978sock_connect(PySocketSockObject *s, PyObject *addro)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001979{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001980 sock_addr_t addrbuf;
1981 int addrlen;
1982 int res;
1983 int timeout;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00001984
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001985 if (!getsockaddrarg(s, addro, SAS2SA(&addrbuf), &addrlen))
1986 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001987
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001988 Py_BEGIN_ALLOW_THREADS
1989 res = internal_connect(s, SAS2SA(&addrbuf), addrlen, &timeout);
1990 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00001991
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001992 if (timeout == 1) {
1993 PyErr_SetString(socket_timeout, "timed out");
1994 return NULL;
1995 }
1996 if (res != 0)
1997 return s->errorhandler();
1998 Py_INCREF(Py_None);
1999 return Py_None;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002000}
2001
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002002PyDoc_STRVAR(connect_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002003"connect(address)\n\
2004\n\
2005Connect the socket to a remote address. For IP sockets, the address\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002006is a pair (host, port).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002007
Guido van Rossum30a685f1991-06-27 15:51:29 +00002008
Fred Drake728819a2000-07-01 03:40:12 +00002009/* s.connect_ex(sockaddr) method */
Guido van Rossumfc4255d1997-11-19 18:57:13 +00002010
2011static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002012sock_connect_ex(PySocketSockObject *s, PyObject *addro)
Guido van Rossumfc4255d1997-11-19 18:57:13 +00002013{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002014 sock_addr_t addrbuf;
2015 int addrlen;
2016 int res;
2017 int timeout;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00002018
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002019 if (!getsockaddrarg(s, addro, SAS2SA(&addrbuf), &addrlen))
2020 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002021
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002022 Py_BEGIN_ALLOW_THREADS
2023 res = internal_connect(s, SAS2SA(&addrbuf), addrlen, &timeout);
2024 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00002025
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002026 /* Signals are not errors (though they may raise exceptions). Adapted
2027 from PyErr_SetFromErrnoWithFilenameObject(). */
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002028#ifdef EINTR
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002029 if (res == EINTR && PyErr_CheckSignals())
2030 return NULL;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002031#endif
2032
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002033 return PyLong_FromLong((long) res);
Guido van Rossumfc4255d1997-11-19 18:57:13 +00002034}
2035
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002036PyDoc_STRVAR(connect_ex_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00002037"connect_ex(address) -> errno\n\
Guido van Rossum82a5c661998-07-07 20:45:43 +00002038\n\
2039This is like connect(address), but returns an error code (the errno value)\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002040instead of raising an exception when an error occurs.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002041
Guido van Rossumfc4255d1997-11-19 18:57:13 +00002042
Guido van Rossumed233a51992-06-23 09:07:03 +00002043/* s.fileno() method */
2044
Guido van Rossum73624e91994-10-10 17:59:00 +00002045static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002046sock_fileno(PySocketSockObject *s)
Guido van Rossumed233a51992-06-23 09:07:03 +00002047{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002048 return PyLong_FromSocket_t(s->sock_fd);
Guido van Rossumed233a51992-06-23 09:07:03 +00002049}
2050
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002051PyDoc_STRVAR(fileno_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002052"fileno() -> integer\n\
2053\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002054Return the integer file descriptor of the socket.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002055
Guido van Rossumed233a51992-06-23 09:07:03 +00002056
Guido van Rossumc89705d1992-11-26 08:54:07 +00002057/* s.getsockname() method */
2058
Guido van Rossum73624e91994-10-10 17:59:00 +00002059static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002060sock_getsockname(PySocketSockObject *s)
Guido van Rossumc89705d1992-11-26 08:54:07 +00002061{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002062 sock_addr_t addrbuf;
2063 int res;
2064 socklen_t addrlen;
Guido van Rossumff3ab422000-04-24 15:16:03 +00002065
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002066 if (!getsockaddrlen(s, &addrlen))
2067 return NULL;
2068 memset(&addrbuf, 0, addrlen);
2069 Py_BEGIN_ALLOW_THREADS
2070 res = getsockname(s->sock_fd, SAS2SA(&addrbuf), &addrlen);
2071 Py_END_ALLOW_THREADS
2072 if (res < 0)
2073 return s->errorhandler();
2074 return makesockaddr(s->sock_fd, SAS2SA(&addrbuf), addrlen,
2075 s->sock_proto);
Guido van Rossumc89705d1992-11-26 08:54:07 +00002076}
2077
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002078PyDoc_STRVAR(getsockname_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002079"getsockname() -> address info\n\
2080\n\
2081Return the address of the local endpoint. For IP sockets, the address\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002082info is a pair (hostaddr, port).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002083
Guido van Rossumc89705d1992-11-26 08:54:07 +00002084
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002085#ifdef HAVE_GETPEERNAME /* Cray APP doesn't have this :-( */
Guido van Rossumc89705d1992-11-26 08:54:07 +00002086/* s.getpeername() method */
2087
Guido van Rossum73624e91994-10-10 17:59:00 +00002088static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002089sock_getpeername(PySocketSockObject *s)
Guido van Rossumc89705d1992-11-26 08:54:07 +00002090{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002091 sock_addr_t addrbuf;
2092 int res;
2093 socklen_t addrlen;
Guido van Rossumff3ab422000-04-24 15:16:03 +00002094
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002095 if (!getsockaddrlen(s, &addrlen))
2096 return NULL;
2097 memset(&addrbuf, 0, addrlen);
2098 Py_BEGIN_ALLOW_THREADS
2099 res = getpeername(s->sock_fd, SAS2SA(&addrbuf), &addrlen);
2100 Py_END_ALLOW_THREADS
2101 if (res < 0)
2102 return s->errorhandler();
2103 return makesockaddr(s->sock_fd, SAS2SA(&addrbuf), addrlen,
2104 s->sock_proto);
Guido van Rossumc89705d1992-11-26 08:54:07 +00002105}
Guido van Rossum82a5c661998-07-07 20:45:43 +00002106
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002107PyDoc_STRVAR(getpeername_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002108"getpeername() -> address info\n\
2109\n\
2110Return the address of the remote endpoint. For IP sockets, the address\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002111info is a pair (hostaddr, port).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002112
Guido van Rossumb6775db1994-08-01 11:34:53 +00002113#endif /* HAVE_GETPEERNAME */
Guido van Rossumc89705d1992-11-26 08:54:07 +00002114
2115
Guido van Rossum30a685f1991-06-27 15:51:29 +00002116/* s.listen(n) method */
2117
Guido van Rossum73624e91994-10-10 17:59:00 +00002118static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002119sock_listen(PySocketSockObject *s, PyObject *arg)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002120{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002121 int backlog;
2122 int res;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00002123
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002124 backlog = PyLong_AsLong(arg);
2125 if (backlog == -1 && PyErr_Occurred())
2126 return NULL;
2127 Py_BEGIN_ALLOW_THREADS
2128 if (backlog < 1)
2129 backlog = 1;
2130 res = listen(s->sock_fd, backlog);
2131 Py_END_ALLOW_THREADS
2132 if (res < 0)
2133 return s->errorhandler();
2134 Py_INCREF(Py_None);
2135 return Py_None;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002136}
2137
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002138PyDoc_STRVAR(listen_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002139"listen(backlog)\n\
2140\n\
2141Enable a server to accept connections. The backlog argument must be at\n\
2142least 1; it specifies the number of unaccepted connection that the system\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002143will allow before refusing new connections.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002144
2145
Thomas Wouters477c8d52006-05-27 19:21:47 +00002146/*
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002147 * This is the guts of the recv() and recv_into() methods, which reads into a
Thomas Wouters902d6eb2007-01-09 23:18:33 +00002148 * char buffer. If you have any inc/dec ref to do to the objects that contain
Thomas Wouters477c8d52006-05-27 19:21:47 +00002149 * the buffer, do it in the caller. This function returns the number of bytes
2150 * succesfully read. If there was an error, it returns -1. Note that it is
2151 * also possible that we return a number of bytes smaller than the request
2152 * bytes.
2153 */
Antoine Pitrou19467d22010-08-17 19:33:30 +00002154static Py_ssize_t
2155sock_recv_guts(PySocketSockObject *s, char* cbuf, Py_ssize_t len, int flags)
Thomas Wouters477c8d52006-05-27 19:21:47 +00002156{
Antoine Pitrou19467d22010-08-17 19:33:30 +00002157 Py_ssize_t outlen = -1;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002158 int timeout;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002159#ifdef __VMS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002160 int remaining;
2161 char *read_buf;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002162#endif
2163
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002164 if (!IS_SELECTABLE(s)) {
2165 select_error();
2166 return -1;
2167 }
2168 if (len == 0) {
2169 /* If 0 bytes were requested, do nothing. */
2170 return 0;
2171 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002172
2173#ifndef __VMS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002174 Py_BEGIN_ALLOW_THREADS
2175 timeout = internal_select(s, 0);
2176 if (!timeout)
2177 outlen = recv(s->sock_fd, cbuf, len, flags);
2178 Py_END_ALLOW_THREADS
Thomas Wouters477c8d52006-05-27 19:21:47 +00002179
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002180 if (timeout == 1) {
2181 PyErr_SetString(socket_timeout, "timed out");
2182 return -1;
2183 }
2184 if (outlen < 0) {
2185 /* Note: the call to errorhandler() ALWAYS indirectly returned
2186 NULL, so ignore its return value */
2187 s->errorhandler();
2188 return -1;
2189 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002190#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002191 read_buf = cbuf;
2192 remaining = len;
2193 while (remaining != 0) {
2194 unsigned int segment;
2195 int nread = -1;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002196
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002197 segment = remaining /SEGMENT_SIZE;
2198 if (segment != 0) {
2199 segment = SEGMENT_SIZE;
2200 }
2201 else {
2202 segment = remaining;
2203 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002204
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002205 Py_BEGIN_ALLOW_THREADS
2206 timeout = internal_select(s, 0);
2207 if (!timeout)
2208 nread = recv(s->sock_fd, read_buf, segment, flags);
2209 Py_END_ALLOW_THREADS
Thomas Wouters477c8d52006-05-27 19:21:47 +00002210
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002211 if (timeout == 1) {
2212 PyErr_SetString(socket_timeout, "timed out");
2213 return -1;
2214 }
2215 if (nread < 0) {
2216 s->errorhandler();
2217 return -1;
2218 }
2219 if (nread != remaining) {
2220 read_buf += nread;
2221 break;
2222 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002223
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002224 remaining -= segment;
2225 read_buf += segment;
2226 }
2227 outlen = read_buf - cbuf;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002228#endif /* !__VMS */
2229
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002230 return outlen;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002231}
2232
Guido van Rossum48a680c2001-03-02 06:34:14 +00002233
Guido van Rossumeb6b33a1993-05-25 09:38:27 +00002234/* s.recv(nbytes [,flags]) method */
Guido van Rossum30a685f1991-06-27 15:51:29 +00002235
Guido van Rossum73624e91994-10-10 17:59:00 +00002236static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002237sock_recv(PySocketSockObject *s, PyObject *args)
Guido van Rossum30a685f1991-06-27 15:51:29 +00002238{
Antoine Pitrou19467d22010-08-17 19:33:30 +00002239 Py_ssize_t recvlen, outlen;
2240 int flags = 0;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002241 PyObject *buf;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002242
Antoine Pitrou19467d22010-08-17 19:33:30 +00002243 if (!PyArg_ParseTuple(args, "n|i:recv", &recvlen, &flags))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002244 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002245
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002246 if (recvlen < 0) {
2247 PyErr_SetString(PyExc_ValueError,
2248 "negative buffersize in recv");
2249 return NULL;
2250 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00002251
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002252 /* Allocate a new string. */
2253 buf = PyBytes_FromStringAndSize((char *) 0, recvlen);
2254 if (buf == NULL)
2255 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002256
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002257 /* Call the guts */
2258 outlen = sock_recv_guts(s, PyBytes_AS_STRING(buf), recvlen, flags);
2259 if (outlen < 0) {
2260 /* An error occurred, release the string and return an
2261 error. */
2262 Py_DECREF(buf);
2263 return NULL;
2264 }
2265 if (outlen != recvlen) {
2266 /* We did not read as many bytes as we anticipated, resize the
2267 string if possible and be successful. */
2268 _PyBytes_Resize(&buf, outlen);
2269 }
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00002270
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002271 return buf;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002272}
2273
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002274PyDoc_STRVAR(recv_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002275"recv(buffersize[, flags]) -> data\n\
2276\n\
2277Receive up to buffersize bytes from the socket. For the optional flags\n\
2278argument, see the Unix manual. When no data is available, block until\n\
2279at least one byte is available or until the remote end is closed. When\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002280the remote end is closed and all data is read, return the empty string.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002281
Guido van Rossum30a685f1991-06-27 15:51:29 +00002282
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002283/* s.recv_into(buffer, [nbytes [,flags]]) method */
Guido van Rossum30a685f1991-06-27 15:51:29 +00002284
Thomas Wouters477c8d52006-05-27 19:21:47 +00002285static PyObject*
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002286sock_recv_into(PySocketSockObject *s, PyObject *args, PyObject *kwds)
Thomas Wouters477c8d52006-05-27 19:21:47 +00002287{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002288 static char *kwlist[] = {"buffer", "nbytes", "flags", 0};
Thomas Wouters477c8d52006-05-27 19:21:47 +00002289
Antoine Pitrou19467d22010-08-17 19:33:30 +00002290 int flags = 0;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002291 Py_buffer pbuf;
2292 char *buf;
Antoine Pitrou19467d22010-08-17 19:33:30 +00002293 Py_ssize_t buflen, readlen, recvlen = 0;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002294
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002295 /* Get the buffer's memory */
Antoine Pitrou19467d22010-08-17 19:33:30 +00002296 if (!PyArg_ParseTupleAndKeywords(args, kwds, "w*|ni:recv_into", kwlist,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002297 &pbuf, &recvlen, &flags))
2298 return NULL;
2299 buf = pbuf.buf;
2300 buflen = pbuf.len;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002301
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002302 if (recvlen < 0) {
2303 PyBuffer_Release(&pbuf);
2304 PyErr_SetString(PyExc_ValueError,
2305 "negative buffersize in recv_into");
2306 return NULL;
2307 }
2308 if (recvlen == 0) {
2309 /* If nbytes was not specified, use the buffer's length */
2310 recvlen = buflen;
2311 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002312
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002313 /* Check if the buffer is large enough */
2314 if (buflen < recvlen) {
2315 PyBuffer_Release(&pbuf);
2316 PyErr_SetString(PyExc_ValueError,
2317 "buffer too small for requested bytes");
2318 return NULL;
2319 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002320
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002321 /* Call the guts */
2322 readlen = sock_recv_guts(s, buf, recvlen, flags);
2323 if (readlen < 0) {
2324 /* Return an error. */
2325 PyBuffer_Release(&pbuf);
2326 return NULL;
2327 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002328
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002329 PyBuffer_Release(&pbuf);
2330 /* Return the number of bytes read. Note that we do not do anything
2331 special here in the case that readlen < recvlen. */
2332 return PyLong_FromSsize_t(readlen);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002333}
2334
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002335PyDoc_STRVAR(recv_into_doc,
2336"recv_into(buffer, [nbytes[, flags]]) -> nbytes_read\n\
Thomas Wouters477c8d52006-05-27 19:21:47 +00002337\n\
2338A version of recv() that stores its data into a buffer rather than creating \n\
2339a new string. Receive up to buffersize bytes from the socket. If buffersize \n\
2340is not specified (or 0), receive up to the size available in the given buffer.\n\
2341\n\
2342See recv() for documentation about the flags.");
2343
2344
2345/*
Christian Heimes99170a52007-12-19 02:07:34 +00002346 * This is the guts of the recvfrom() and recvfrom_into() methods, which reads
2347 * into a char buffer. If you have any inc/def ref to do to the objects that
2348 * contain the buffer, do it in the caller. This function returns the number
2349 * of bytes succesfully read. If there was an error, it returns -1. Note
2350 * that it is also possible that we return a number of bytes smaller than the
2351 * request bytes.
Thomas Wouters477c8d52006-05-27 19:21:47 +00002352 *
2353 * 'addr' is a return value for the address object. Note that you must decref
2354 * it yourself.
2355 */
Antoine Pitrou19467d22010-08-17 19:33:30 +00002356static Py_ssize_t
2357sock_recvfrom_guts(PySocketSockObject *s, char* cbuf, Py_ssize_t len, int flags,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002358 PyObject** addr)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002359{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002360 sock_addr_t addrbuf;
2361 int timeout;
Antoine Pitrou19467d22010-08-17 19:33:30 +00002362 Py_ssize_t n = -1;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002363 socklen_t addrlen;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002364
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002365 *addr = NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002366
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002367 if (!getsockaddrlen(s, &addrlen))
2368 return -1;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002369
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002370 if (!IS_SELECTABLE(s)) {
2371 select_error();
2372 return -1;
2373 }
Neal Norwitz082b2df2006-02-07 07:04:46 +00002374
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002375 Py_BEGIN_ALLOW_THREADS
2376 memset(&addrbuf, 0, addrlen);
2377 timeout = internal_select(s, 0);
2378 if (!timeout) {
Guido van Rossum8d665e61996-06-26 18:22:49 +00002379#ifndef MS_WINDOWS
Andrew MacIntyreba43e872002-03-03 03:03:52 +00002380#if defined(PYOS_OS2) && !defined(PYCC_GCC)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002381 n = recvfrom(s->sock_fd, cbuf, len, flags,
2382 SAS2SA(&addrbuf), &addrlen);
Guido van Rossum32c575d1997-12-02 20:37:32 +00002383#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002384 n = recvfrom(s->sock_fd, cbuf, len, flags,
2385 (void *) &addrbuf, &addrlen);
Guido van Rossum32c575d1997-12-02 20:37:32 +00002386#endif
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00002387#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002388 n = recvfrom(s->sock_fd, cbuf, len, flags,
2389 SAS2SA(&addrbuf), &addrlen);
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00002390#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002391 }
2392 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00002393
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002394 if (timeout == 1) {
2395 PyErr_SetString(socket_timeout, "timed out");
2396 return -1;
2397 }
2398 if (n < 0) {
2399 s->errorhandler();
2400 return -1;
2401 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00002402
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002403 if (!(*addr = makesockaddr(s->sock_fd, SAS2SA(&addrbuf),
2404 addrlen, s->sock_proto)))
2405 return -1;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002406
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002407 return n;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002408}
2409
2410/* s.recvfrom(nbytes [,flags]) method */
2411
2412static PyObject *
2413sock_recvfrom(PySocketSockObject *s, PyObject *args)
2414{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002415 PyObject *buf = NULL;
2416 PyObject *addr = NULL;
2417 PyObject *ret = NULL;
Antoine Pitrou19467d22010-08-17 19:33:30 +00002418 int flags = 0;
2419 Py_ssize_t recvlen, outlen;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002420
Antoine Pitrou19467d22010-08-17 19:33:30 +00002421 if (!PyArg_ParseTuple(args, "n|i:recvfrom", &recvlen, &flags))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002422 return NULL;
Guido van Rossum48a680c2001-03-02 06:34:14 +00002423
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002424 if (recvlen < 0) {
2425 PyErr_SetString(PyExc_ValueError,
2426 "negative buffersize in recvfrom");
2427 return NULL;
2428 }
Guido van Rossumd8faa362007-04-27 19:54:29 +00002429
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002430 buf = PyBytes_FromStringAndSize((char *) 0, recvlen);
2431 if (buf == NULL)
2432 return NULL;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002433
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002434 outlen = sock_recvfrom_guts(s, PyBytes_AS_STRING(buf),
2435 recvlen, flags, &addr);
2436 if (outlen < 0) {
2437 goto finally;
2438 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002439
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002440 if (outlen != recvlen) {
2441 /* We did not read as many bytes as we anticipated, resize the
2442 string if possible and be succesful. */
2443 if (_PyBytes_Resize(&buf, outlen) < 0)
2444 /* Oopsy, not so succesful after all. */
2445 goto finally;
2446 }
Barry Warsaw752300b1997-01-03 17:18:10 +00002447
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002448 ret = PyTuple_Pack(2, buf, addr);
Guido van Rossum67f7a382002-06-06 21:08:16 +00002449
2450finally:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002451 Py_XDECREF(buf);
2452 Py_XDECREF(addr);
2453 return ret;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002454}
2455
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002456PyDoc_STRVAR(recvfrom_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002457"recvfrom(buffersize[, flags]) -> (data, address info)\n\
2458\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002459Like recv(buffersize, flags) but also return the sender's address info.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002460
Thomas Wouters477c8d52006-05-27 19:21:47 +00002461
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002462/* s.recvfrom_into(buffer[, nbytes [,flags]]) method */
Thomas Wouters477c8d52006-05-27 19:21:47 +00002463
2464static PyObject *
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002465sock_recvfrom_into(PySocketSockObject *s, PyObject *args, PyObject* kwds)
Thomas Wouters477c8d52006-05-27 19:21:47 +00002466{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002467 static char *kwlist[] = {"buffer", "nbytes", "flags", 0};
Thomas Wouters477c8d52006-05-27 19:21:47 +00002468
Antoine Pitrou19467d22010-08-17 19:33:30 +00002469 int flags = 0;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002470 Py_buffer pbuf;
2471 char *buf;
Antoine Pitrou19467d22010-08-17 19:33:30 +00002472 Py_ssize_t readlen, buflen, recvlen = 0;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002473
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002474 PyObject *addr = NULL;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002475
Antoine Pitrou19467d22010-08-17 19:33:30 +00002476 if (!PyArg_ParseTupleAndKeywords(args, kwds, "w*|ni:recvfrom_into",
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002477 kwlist, &pbuf,
2478 &recvlen, &flags))
2479 return NULL;
2480 buf = pbuf.buf;
2481 buflen = pbuf.len;
2482 assert(buf != 0 && buflen > 0);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002483
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002484 if (recvlen < 0) {
2485 PyBuffer_Release(&pbuf);
2486 PyErr_SetString(PyExc_ValueError,
2487 "negative buffersize in recvfrom_into");
2488 return NULL;
2489 }
2490 if (recvlen == 0) {
2491 /* If nbytes was not specified, use the buffer's length */
2492 recvlen = buflen;
2493 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002494
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002495 readlen = sock_recvfrom_guts(s, buf, recvlen, flags, &addr);
2496 if (readlen < 0) {
2497 PyBuffer_Release(&pbuf);
2498 /* Return an error */
2499 Py_XDECREF(addr);
2500 return NULL;
2501 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002502
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002503 PyBuffer_Release(&pbuf);
2504 /* Return the number of bytes read and the address. Note that we do
2505 not do anything special here in the case that readlen < recvlen. */
Antoine Pitrou19467d22010-08-17 19:33:30 +00002506 return Py_BuildValue("nN", readlen, addr);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002507}
2508
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002509PyDoc_STRVAR(recvfrom_into_doc,
2510"recvfrom_into(buffer[, nbytes[, flags]]) -> (nbytes, address info)\n\
Thomas Wouters477c8d52006-05-27 19:21:47 +00002511\n\
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002512Like recv_into(buffer[, nbytes[, flags]]) but also return the sender's address info.");
Thomas Wouters477c8d52006-05-27 19:21:47 +00002513
2514
Guido van Rossumeb6b33a1993-05-25 09:38:27 +00002515/* s.send(data [,flags]) method */
Guido van Rossum30a685f1991-06-27 15:51:29 +00002516
Guido van Rossum73624e91994-10-10 17:59:00 +00002517static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002518sock_send(PySocketSockObject *s, PyObject *args)
Guido van Rossum30a685f1991-06-27 15:51:29 +00002519{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002520 char *buf;
Antoine Pitrou19467d22010-08-17 19:33:30 +00002521 Py_ssize_t len, n = -1;
2522 int flags = 0, timeout;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002523 Py_buffer pbuf;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002524
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002525 if (!PyArg_ParseTuple(args, "y*|i:send", &pbuf, &flags))
2526 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002527
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002528 if (!IS_SELECTABLE(s)) {
2529 PyBuffer_Release(&pbuf);
2530 return select_error();
2531 }
2532 buf = pbuf.buf;
2533 len = pbuf.len;
Neal Norwitz082b2df2006-02-07 07:04:46 +00002534
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002535 Py_BEGIN_ALLOW_THREADS
2536 timeout = internal_select(s, 1);
2537 if (!timeout)
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002538#ifdef __VMS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002539 n = sendsegmented(s->sock_fd, buf, len, flags);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002540#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002541 n = send(s->sock_fd, buf, len, flags);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002542#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002543 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00002544
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002545 PyBuffer_Release(&pbuf);
Martin v. Löwis423be952008-08-13 15:53:07 +00002546
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002547 if (timeout == 1) {
2548 PyErr_SetString(socket_timeout, "timed out");
2549 return NULL;
2550 }
2551 if (n < 0)
2552 return s->errorhandler();
Antoine Pitrou19467d22010-08-17 19:33:30 +00002553 return PyLong_FromSsize_t(n);
Guido van Rossum30a685f1991-06-27 15:51:29 +00002554}
2555
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002556PyDoc_STRVAR(send_doc,
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002557"send(data[, flags]) -> count\n\
Guido van Rossum82a5c661998-07-07 20:45:43 +00002558\n\
2559Send a data string to the socket. For the optional flags\n\
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002560argument, see the Unix manual. Return the number of bytes\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002561sent; this may be less than len(data) if the network is busy.");
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002562
2563
2564/* s.sendall(data [,flags]) method */
2565
2566static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002567sock_sendall(PySocketSockObject *s, PyObject *args)
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002568{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002569 char *buf;
Antoine Pitrou19467d22010-08-17 19:33:30 +00002570 Py_ssize_t len, n = -1;
Antoine Pitrou6d7df632010-09-27 17:52:25 +00002571 int flags = 0, timeout, saved_errno;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002572 Py_buffer pbuf;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002573
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002574 if (!PyArg_ParseTuple(args, "y*|i:sendall", &pbuf, &flags))
2575 return NULL;
2576 buf = pbuf.buf;
2577 len = pbuf.len;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002578
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002579 if (!IS_SELECTABLE(s)) {
2580 PyBuffer_Release(&pbuf);
2581 return select_error();
2582 }
Neal Norwitz082b2df2006-02-07 07:04:46 +00002583
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002584 do {
Antoine Pitrou6d7df632010-09-27 17:52:25 +00002585 Py_BEGIN_ALLOW_THREADS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002586 timeout = internal_select(s, 1);
2587 n = -1;
Antoine Pitrou6d7df632010-09-27 17:52:25 +00002588 if (!timeout) {
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002589#ifdef __VMS
Antoine Pitrou6d7df632010-09-27 17:52:25 +00002590 n = sendsegmented(s->sock_fd, buf, len, flags);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002591#else
Antoine Pitrou6d7df632010-09-27 17:52:25 +00002592 n = send(s->sock_fd, buf, len, flags);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002593#endif
Antoine Pitrou6d7df632010-09-27 17:52:25 +00002594 }
2595 Py_END_ALLOW_THREADS
2596 if (timeout == 1) {
2597 PyBuffer_Release(&pbuf);
2598 PyErr_SetString(socket_timeout, "timed out");
2599 return NULL;
2600 }
2601 /* PyErr_CheckSignals() might change errno */
2602 saved_errno = errno;
2603 /* We must run our signal handlers before looping again.
2604 send() can return a successful partial write when it is
2605 interrupted, so we can't restrict ourselves to EINTR. */
2606 if (PyErr_CheckSignals()) {
2607 PyBuffer_Release(&pbuf);
2608 return NULL;
2609 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002610 if (n < 0) {
Antoine Pitrou6d7df632010-09-27 17:52:25 +00002611 /* If interrupted, try again */
2612 if (saved_errno == EINTR)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002613 continue;
Antoine Pitrou6d7df632010-09-27 17:52:25 +00002614 else
2615 break;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002616 }
2617 buf += n;
2618 len -= n;
2619 } while (len > 0);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002620 PyBuffer_Release(&pbuf);
Guido van Rossum67f7a382002-06-06 21:08:16 +00002621
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002622 if (n < 0)
2623 return s->errorhandler();
Guido van Rossum67f7a382002-06-06 21:08:16 +00002624
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002625 Py_INCREF(Py_None);
2626 return Py_None;
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002627}
2628
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002629PyDoc_STRVAR(sendall_doc,
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002630"sendall(data[, flags])\n\
2631\n\
2632Send a data string to the socket. For the optional flags\n\
2633argument, see the Unix manual. This calls send() repeatedly\n\
2634until all data is sent. If an error occurs, it's impossible\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002635to tell how much data has been sent.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002636
Guido van Rossum30a685f1991-06-27 15:51:29 +00002637
Guido van Rossumeb6b33a1993-05-25 09:38:27 +00002638/* s.sendto(data, [flags,] sockaddr) method */
Guido van Rossum30a685f1991-06-27 15:51:29 +00002639
Guido van Rossum73624e91994-10-10 17:59:00 +00002640static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002641sock_sendto(PySocketSockObject *s, PyObject *args)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002642{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002643 Py_buffer pbuf;
2644 PyObject *addro;
2645 char *buf;
2646 Py_ssize_t len;
2647 sock_addr_t addrbuf;
2648 int addrlen, n = -1, flags, timeout;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002649
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002650 flags = 0;
2651 if (!PyArg_ParseTuple(args, "y*O:sendto", &pbuf, &addro)) {
2652 PyErr_Clear();
2653 if (!PyArg_ParseTuple(args, "y*iO:sendto",
2654 &pbuf, &flags, &addro))
2655 return NULL;
2656 }
2657 buf = pbuf.buf;
2658 len = pbuf.len;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002659
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002660 if (!IS_SELECTABLE(s)) {
2661 PyBuffer_Release(&pbuf);
2662 return select_error();
2663 }
Neal Norwitz082b2df2006-02-07 07:04:46 +00002664
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002665 if (!getsockaddrarg(s, addro, SAS2SA(&addrbuf), &addrlen)) {
2666 PyBuffer_Release(&pbuf);
2667 return NULL;
2668 }
Thomas Wouters89f507f2006-12-13 04:49:30 +00002669
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002670 Py_BEGIN_ALLOW_THREADS
2671 timeout = internal_select(s, 1);
2672 if (!timeout)
2673 n = sendto(s->sock_fd, buf, len, flags, SAS2SA(&addrbuf), addrlen);
2674 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00002675
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002676 PyBuffer_Release(&pbuf);
2677 if (timeout == 1) {
2678 PyErr_SetString(socket_timeout, "timed out");
2679 return NULL;
2680 }
2681 if (n < 0)
2682 return s->errorhandler();
Antoine Pitrou19467d22010-08-17 19:33:30 +00002683 return PyLong_FromSsize_t(n);
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002684}
2685
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002686PyDoc_STRVAR(sendto_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00002687"sendto(data[, flags], address) -> count\n\
Guido van Rossum82a5c661998-07-07 20:45:43 +00002688\n\
2689Like send(data, flags) but allows specifying the destination address.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002690For IP sockets, the address is a pair (hostaddr, port).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002691
Guido van Rossum30a685f1991-06-27 15:51:29 +00002692
2693/* s.shutdown(how) method */
2694
Guido van Rossum73624e91994-10-10 17:59:00 +00002695static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002696sock_shutdown(PySocketSockObject *s, PyObject *arg)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002697{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002698 int how;
2699 int res;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00002700
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002701 how = PyLong_AsLong(arg);
2702 if (how == -1 && PyErr_Occurred())
2703 return NULL;
2704 Py_BEGIN_ALLOW_THREADS
2705 res = shutdown(s->sock_fd, how);
2706 Py_END_ALLOW_THREADS
2707 if (res < 0)
2708 return s->errorhandler();
2709 Py_INCREF(Py_None);
2710 return Py_None;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002711}
2712
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002713PyDoc_STRVAR(shutdown_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002714"shutdown(flag)\n\
2715\n\
Martin v. Löwis94681fc2003-11-27 19:40:22 +00002716Shut down the reading side of the socket (flag == SHUT_RD), the writing side\n\
2717of the socket (flag == SHUT_WR), or both ends (flag == SHUT_RDWR).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002718
Amaury Forgeot d'Arc3d17a5c2008-06-13 01:09:34 +00002719#if defined(MS_WINDOWS) && defined(SIO_RCVALL)
Christian Heimesfaf2f632008-01-06 16:59:19 +00002720static PyObject*
2721sock_ioctl(PySocketSockObject *s, PyObject *arg)
2722{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002723 unsigned long cmd = SIO_RCVALL;
2724 PyObject *argO;
2725 DWORD recv;
Christian Heimesfaf2f632008-01-06 16:59:19 +00002726
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002727 if (!PyArg_ParseTuple(arg, "kO:ioctl", &cmd, &argO))
2728 return NULL;
Christian Heimesfaf2f632008-01-06 16:59:19 +00002729
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002730 switch (cmd) {
2731 case SIO_RCVALL: {
2732 unsigned int option = RCVALL_ON;
2733 if (!PyArg_ParseTuple(arg, "kI:ioctl", &cmd, &option))
2734 return NULL;
2735 if (WSAIoctl(s->sock_fd, cmd, &option, sizeof(option),
2736 NULL, 0, &recv, NULL, NULL) == SOCKET_ERROR) {
2737 return set_error();
2738 }
2739 return PyLong_FromUnsignedLong(recv); }
2740 case SIO_KEEPALIVE_VALS: {
2741 struct tcp_keepalive ka;
2742 if (!PyArg_ParseTuple(arg, "k(kkk):ioctl", &cmd,
2743 &ka.onoff, &ka.keepalivetime, &ka.keepaliveinterval))
2744 return NULL;
2745 if (WSAIoctl(s->sock_fd, cmd, &ka, sizeof(ka),
2746 NULL, 0, &recv, NULL, NULL) == SOCKET_ERROR) {
2747 return set_error();
2748 }
2749 return PyLong_FromUnsignedLong(recv); }
2750 default:
2751 PyErr_Format(PyExc_ValueError, "invalid ioctl command %d", cmd);
2752 return NULL;
2753 }
Christian Heimesfaf2f632008-01-06 16:59:19 +00002754}
2755PyDoc_STRVAR(sock_ioctl_doc,
2756"ioctl(cmd, option) -> long\n\
2757\n\
Kristján Valur Jónsson847ec752009-09-27 21:10:38 +00002758Control the socket with WSAIoctl syscall. Currently supported 'cmd' values are\n\
2759SIO_RCVALL: 'option' must be one of the socket.RCVALL_* constants.\n\
2760SIO_KEEPALIVE_VALS: 'option' is a tuple of (onoff, timeout, interval).");
Christian Heimesfaf2f632008-01-06 16:59:19 +00002761
2762#endif
Guido van Rossum30a685f1991-06-27 15:51:29 +00002763
2764/* List of methods for socket objects */
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002765
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002766static PyMethodDef sock_methods[] = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002767 {"_accept", (PyCFunction)sock_accept, METH_NOARGS,
2768 accept_doc},
2769 {"bind", (PyCFunction)sock_bind, METH_O,
2770 bind_doc},
2771 {"close", (PyCFunction)sock_close, METH_NOARGS,
2772 close_doc},
2773 {"connect", (PyCFunction)sock_connect, METH_O,
2774 connect_doc},
2775 {"connect_ex", (PyCFunction)sock_connect_ex, METH_O,
2776 connect_ex_doc},
Antoine Pitrou6e451df2010-08-09 20:39:54 +00002777 {"detach", (PyCFunction)sock_detach, METH_NOARGS,
2778 detach_doc},
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002779 {"fileno", (PyCFunction)sock_fileno, METH_NOARGS,
2780 fileno_doc},
Guido van Rossumb6775db1994-08-01 11:34:53 +00002781#ifdef HAVE_GETPEERNAME
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002782 {"getpeername", (PyCFunction)sock_getpeername,
2783 METH_NOARGS, getpeername_doc},
Guido van Rossum9575a441993-04-07 14:06:14 +00002784#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002785 {"getsockname", (PyCFunction)sock_getsockname,
2786 METH_NOARGS, getsockname_doc},
2787 {"getsockopt", (PyCFunction)sock_getsockopt, METH_VARARGS,
2788 getsockopt_doc},
Amaury Forgeot d'Arc3d17a5c2008-06-13 01:09:34 +00002789#if defined(MS_WINDOWS) && defined(SIO_RCVALL)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002790 {"ioctl", (PyCFunction)sock_ioctl, METH_VARARGS,
2791 sock_ioctl_doc},
Christian Heimesfaf2f632008-01-06 16:59:19 +00002792#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002793 {"listen", (PyCFunction)sock_listen, METH_O,
2794 listen_doc},
2795 {"recv", (PyCFunction)sock_recv, METH_VARARGS,
2796 recv_doc},
2797 {"recv_into", (PyCFunction)sock_recv_into, METH_VARARGS | METH_KEYWORDS,
2798 recv_into_doc},
2799 {"recvfrom", (PyCFunction)sock_recvfrom, METH_VARARGS,
2800 recvfrom_doc},
2801 {"recvfrom_into", (PyCFunction)sock_recvfrom_into, METH_VARARGS | METH_KEYWORDS,
2802 recvfrom_into_doc},
2803 {"send", (PyCFunction)sock_send, METH_VARARGS,
2804 send_doc},
2805 {"sendall", (PyCFunction)sock_sendall, METH_VARARGS,
2806 sendall_doc},
2807 {"sendto", (PyCFunction)sock_sendto, METH_VARARGS,
2808 sendto_doc},
2809 {"setblocking", (PyCFunction)sock_setblocking, METH_O,
2810 setblocking_doc},
2811 {"settimeout", (PyCFunction)sock_settimeout, METH_O,
2812 settimeout_doc},
2813 {"gettimeout", (PyCFunction)sock_gettimeout, METH_NOARGS,
2814 gettimeout_doc},
2815 {"setsockopt", (PyCFunction)sock_setsockopt, METH_VARARGS,
2816 setsockopt_doc},
2817 {"shutdown", (PyCFunction)sock_shutdown, METH_O,
2818 shutdown_doc},
2819 {NULL, NULL} /* sentinel */
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002820};
2821
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002822/* SockObject members */
2823static PyMemberDef sock_memberlist[] = {
2824 {"family", T_INT, offsetof(PySocketSockObject, sock_family), READONLY, "the socket family"},
2825 {"type", T_INT, offsetof(PySocketSockObject, sock_type), READONLY, "the socket type"},
2826 {"proto", T_INT, offsetof(PySocketSockObject, sock_proto), READONLY, "the socket protocol"},
2827 {"timeout", T_DOUBLE, offsetof(PySocketSockObject, sock_timeout), READONLY, "the socket timeout"},
2828 {0},
2829};
Guido van Rossum30a685f1991-06-27 15:51:29 +00002830
Guido van Rossum73624e91994-10-10 17:59:00 +00002831/* Deallocate a socket object in response to the last Py_DECREF().
Guido van Rossum30a685f1991-06-27 15:51:29 +00002832 First close the file description. */
2833
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002834static void
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002835sock_dealloc(PySocketSockObject *s)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002836{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002837 if (s->sock_fd != -1)
2838 (void) SOCKETCLOSE(s->sock_fd);
2839 Py_TYPE(s)->tp_free((PyObject *)s);
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002840}
2841
Guido van Rossum30a685f1991-06-27 15:51:29 +00002842
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00002843static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002844sock_repr(PySocketSockObject *s)
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00002845{
Fred Drakea04eaad2000-06-30 02:46:07 +00002846#if SIZEOF_SOCKET_T > SIZEOF_LONG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002847 if (s->sock_fd > LONG_MAX) {
2848 /* this can occur on Win64, and actually there is a special
2849 ugly printf formatter for decimal pointer length integer
2850 printing, only bother if necessary*/
2851 PyErr_SetString(PyExc_OverflowError,
2852 "no printf formatter to display "
2853 "the socket descriptor in decimal");
2854 return NULL;
2855 }
Fred Drakea04eaad2000-06-30 02:46:07 +00002856#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002857 return PyUnicode_FromFormat(
2858 "<socket object, fd=%ld, family=%d, type=%d, proto=%d>",
2859 (long)s->sock_fd, s->sock_family,
2860 s->sock_type,
2861 s->sock_proto);
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00002862}
2863
2864
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002865/* Create a new, uninitialized socket object. */
2866
2867static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002868sock_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002869{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002870 PyObject *new;
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002871
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002872 new = type->tp_alloc(type, 0);
2873 if (new != NULL) {
2874 ((PySocketSockObject *)new)->sock_fd = -1;
2875 ((PySocketSockObject *)new)->sock_timeout = -1.0;
2876 ((PySocketSockObject *)new)->errorhandler = &set_error;
2877 }
2878 return new;
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002879}
2880
2881
2882/* Initialize a new socket object. */
2883
2884/*ARGSUSED*/
2885static int
Andrew MacIntyre7aec4a22002-06-13 11:53:52 +00002886sock_initobj(PyObject *self, PyObject *args, PyObject *kwds)
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002887{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002888 PySocketSockObject *s = (PySocketSockObject *)self;
2889 PyObject *fdobj = NULL;
2890 SOCKET_T fd = INVALID_SOCKET;
2891 int family = AF_INET, type = SOCK_STREAM, proto = 0;
2892 static char *keywords[] = {"family", "type", "proto", "fileno", 0};
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002893
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002894 if (!PyArg_ParseTupleAndKeywords(args, kwds,
2895 "|iiiO:socket", keywords,
2896 &family, &type, &proto, &fdobj))
2897 return -1;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002898
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002899 if (fdobj != NULL && fdobj != Py_None) {
2900 fd = PyLong_AsSocket_t(fdobj);
2901 if (fd == (SOCKET_T)(-1) && PyErr_Occurred())
2902 return -1;
2903 if (fd == INVALID_SOCKET) {
2904 PyErr_SetString(PyExc_ValueError,
2905 "can't use invalid socket value");
2906 return -1;
2907 }
2908 }
2909 else {
2910 Py_BEGIN_ALLOW_THREADS
2911 fd = socket(family, type, proto);
2912 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00002913
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002914 if (fd == INVALID_SOCKET) {
2915 set_error();
2916 return -1;
2917 }
2918 }
2919 init_sockobject(s, fd, family, type, proto);
Guido van Rossum67f7a382002-06-06 21:08:16 +00002920
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002921 return 0;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002922
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002923}
2924
2925
Guido van Rossumb6775db1994-08-01 11:34:53 +00002926/* Type object for socket objects. */
Guido van Rossum30a685f1991-06-27 15:51:29 +00002927
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002928static PyTypeObject sock_type = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002929 PyVarObject_HEAD_INIT(0, 0) /* Must fill in type value later */
2930 "_socket.socket", /* tp_name */
2931 sizeof(PySocketSockObject), /* tp_basicsize */
2932 0, /* tp_itemsize */
2933 (destructor)sock_dealloc, /* tp_dealloc */
2934 0, /* tp_print */
2935 0, /* tp_getattr */
2936 0, /* tp_setattr */
2937 0, /* tp_reserved */
2938 (reprfunc)sock_repr, /* tp_repr */
2939 0, /* tp_as_number */
2940 0, /* tp_as_sequence */
2941 0, /* tp_as_mapping */
2942 0, /* tp_hash */
2943 0, /* tp_call */
2944 0, /* tp_str */
2945 PyObject_GenericGetAttr, /* tp_getattro */
2946 0, /* tp_setattro */
2947 0, /* tp_as_buffer */
2948 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE, /* tp_flags */
2949 sock_doc, /* tp_doc */
2950 0, /* tp_traverse */
2951 0, /* tp_clear */
2952 0, /* tp_richcompare */
2953 0, /* tp_weaklistoffset */
2954 0, /* tp_iter */
2955 0, /* tp_iternext */
2956 sock_methods, /* tp_methods */
2957 sock_memberlist, /* tp_members */
2958 0, /* tp_getset */
2959 0, /* tp_base */
2960 0, /* tp_dict */
2961 0, /* tp_descr_get */
2962 0, /* tp_descr_set */
2963 0, /* tp_dictoffset */
2964 sock_initobj, /* tp_init */
2965 PyType_GenericAlloc, /* tp_alloc */
2966 sock_new, /* tp_new */
2967 PyObject_Del, /* tp_free */
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002968};
2969
Guido van Rossum30a685f1991-06-27 15:51:29 +00002970
Guido van Rossum81194471991-07-27 21:42:02 +00002971/* Python interface to gethostname(). */
2972
2973/*ARGSUSED*/
Guido van Rossum73624e91994-10-10 17:59:00 +00002974static PyObject *
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00002975socket_gethostname(PyObject *self, PyObject *unused)
Guido van Rossum81194471991-07-27 21:42:02 +00002976{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002977 char buf[1024];
2978 int res;
2979 Py_BEGIN_ALLOW_THREADS
2980 res = gethostname(buf, (int) sizeof buf - 1);
2981 Py_END_ALLOW_THREADS
2982 if (res < 0)
2983 return set_error();
2984 buf[sizeof buf - 1] = '\0';
2985 return PyUnicode_FromString(buf);
Guido van Rossum81194471991-07-27 21:42:02 +00002986}
Guido van Rossumff4949e1992-08-05 19:58:53 +00002987
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002988PyDoc_STRVAR(gethostname_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002989"gethostname() -> string\n\
2990\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002991Return the current host name.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002992
Guido van Rossumff4949e1992-08-05 19:58:53 +00002993
Guido van Rossum30a685f1991-06-27 15:51:29 +00002994/* Python interface to gethostbyname(name). */
2995
2996/*ARGSUSED*/
Guido van Rossum73624e91994-10-10 17:59:00 +00002997static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002998socket_gethostbyname(PyObject *self, PyObject *args)
Guido van Rossum30a685f1991-06-27 15:51:29 +00002999{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003000 char *name;
3001 sock_addr_t addrbuf;
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003002 PyObject *ret = NULL;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003003
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003004 if (!PyArg_ParseTuple(args, "et:gethostbyname", "idna", &name))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003005 return NULL;
3006 if (setipaddr(name, SAS2SA(&addrbuf), sizeof(addrbuf), AF_INET) < 0)
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003007 goto finally;
3008 ret = makeipaddr(SAS2SA(&addrbuf), sizeof(struct sockaddr_in));
3009finally:
3010 PyMem_Free(name);
3011 return ret;
Guido van Rossum30a685f1991-06-27 15:51:29 +00003012}
3013
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003014PyDoc_STRVAR(gethostbyname_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003015"gethostbyname(host) -> address\n\
3016\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003017Return the IP address (a string of the form '255.255.255.255') for a host.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003018
3019
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003020/* Convenience function common to gethostbyname_ex and gethostbyaddr */
3021
3022static PyObject *
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003023gethost_common(struct hostent *h, struct sockaddr *addr, int alen, int af)
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003024{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003025 char **pch;
3026 PyObject *rtn_tuple = (PyObject *)NULL;
3027 PyObject *name_list = (PyObject *)NULL;
3028 PyObject *addr_list = (PyObject *)NULL;
3029 PyObject *tmp;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003030
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003031 if (h == NULL) {
3032 /* Let's get real error message to return */
3033 set_herror(h_errno);
3034 return NULL;
3035 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00003036
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003037 if (h->h_addrtype != af) {
3038 /* Let's get real error message to return */
3039 PyErr_SetString(socket_error,
3040 (char *)strerror(EAFNOSUPPORT));
Christian Heimesada8c3b2008-03-18 18:26:33 +00003041
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003042 return NULL;
3043 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00003044
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003045 switch (af) {
Guido van Rossum67f7a382002-06-06 21:08:16 +00003046
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003047 case AF_INET:
3048 if (alen < sizeof(struct sockaddr_in))
3049 return NULL;
3050 break;
Guido van Rossum67f7a382002-06-06 21:08:16 +00003051
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00003052#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003053 case AF_INET6:
3054 if (alen < sizeof(struct sockaddr_in6))
3055 return NULL;
3056 break;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003057#endif
Guido van Rossum67f7a382002-06-06 21:08:16 +00003058
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003059 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00003060
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003061 if ((name_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 if ((addr_list = PyList_New(0)) == NULL)
3065 goto err;
Guido van Rossum67f7a382002-06-06 21:08:16 +00003066
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003067 /* SF #1511317: h_aliases can be NULL */
3068 if (h->h_aliases) {
3069 for (pch = h->h_aliases; *pch != NULL; pch++) {
3070 int status;
3071 tmp = PyUnicode_FromString(*pch);
3072 if (tmp == NULL)
3073 goto err;
Guido van Rossum67f7a382002-06-06 21:08:16 +00003074
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003075 status = PyList_Append(name_list, tmp);
3076 Py_DECREF(tmp);
Guido van Rossum67f7a382002-06-06 21:08:16 +00003077
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003078 if (status)
3079 goto err;
3080 }
3081 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00003082
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003083 for (pch = h->h_addr_list; *pch != NULL; pch++) {
3084 int status;
Guido van Rossum67f7a382002-06-06 21:08:16 +00003085
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003086 switch (af) {
Guido van Rossum67f7a382002-06-06 21:08:16 +00003087
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003088 case AF_INET:
3089 {
3090 struct sockaddr_in sin;
3091 memset(&sin, 0, sizeof(sin));
3092 sin.sin_family = af;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003093#ifdef HAVE_SOCKADDR_SA_LEN
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003094 sin.sin_len = sizeof(sin);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003095#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003096 memcpy(&sin.sin_addr, *pch, sizeof(sin.sin_addr));
3097 tmp = makeipaddr((struct sockaddr *)&sin, sizeof(sin));
Guido van Rossum67f7a382002-06-06 21:08:16 +00003098
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003099 if (pch == h->h_addr_list && alen >= sizeof(sin))
3100 memcpy((char *) addr, &sin, sizeof(sin));
3101 break;
3102 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00003103
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00003104#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003105 case AF_INET6:
3106 {
3107 struct sockaddr_in6 sin6;
3108 memset(&sin6, 0, sizeof(sin6));
3109 sin6.sin6_family = af;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003110#ifdef HAVE_SOCKADDR_SA_LEN
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003111 sin6.sin6_len = sizeof(sin6);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003112#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003113 memcpy(&sin6.sin6_addr, *pch, sizeof(sin6.sin6_addr));
3114 tmp = makeipaddr((struct sockaddr *)&sin6,
3115 sizeof(sin6));
Guido van Rossum67f7a382002-06-06 21:08:16 +00003116
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003117 if (pch == h->h_addr_list && alen >= sizeof(sin6))
3118 memcpy((char *) addr, &sin6, sizeof(sin6));
3119 break;
3120 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003121#endif
Guido van Rossum67f7a382002-06-06 21:08:16 +00003122
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003123 default: /* can't happen */
3124 PyErr_SetString(socket_error,
3125 "unsupported address family");
3126 return NULL;
3127 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00003128
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003129 if (tmp == NULL)
3130 goto err;
Guido van Rossum67f7a382002-06-06 21:08:16 +00003131
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003132 status = PyList_Append(addr_list, tmp);
3133 Py_DECREF(tmp);
Guido van Rossum67f7a382002-06-06 21:08:16 +00003134
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003135 if (status)
3136 goto err;
3137 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00003138
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003139 rtn_tuple = Py_BuildValue("sOO", h->h_name, name_list, addr_list);
Guido van Rossum67f7a382002-06-06 21:08:16 +00003140
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003141 err:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003142 Py_XDECREF(name_list);
3143 Py_XDECREF(addr_list);
3144 return rtn_tuple;
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003145}
3146
3147
3148/* Python interface to gethostbyname_ex(name). */
3149
3150/*ARGSUSED*/
3151static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003152socket_gethostbyname_ex(PyObject *self, PyObject *args)
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003153{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003154 char *name;
3155 struct hostent *h;
Martin v. Löwisc16f3bd2003-05-03 09:14:54 +00003156#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003157 struct sockaddr_storage addr;
Martin v. Löwisc16f3bd2003-05-03 09:14:54 +00003158#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003159 struct sockaddr_in addr;
Martin v. Löwisc16f3bd2003-05-03 09:14:54 +00003160#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003161 struct sockaddr *sa;
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003162 PyObject *ret = NULL;
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003163#ifdef HAVE_GETHOSTBYNAME_R
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003164 struct hostent hp_allocated;
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003165#ifdef HAVE_GETHOSTBYNAME_R_3_ARG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003166 struct hostent_data data;
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003167#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003168 char buf[16384];
3169 int buf_len = (sizeof buf) - 1;
3170 int errnop;
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003171#endif
3172#if defined(HAVE_GETHOSTBYNAME_R_3_ARG) || defined(HAVE_GETHOSTBYNAME_R_6_ARG)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003173 int result;
Guido van Rossume9cd07b1999-03-15 21:40:14 +00003174#endif
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003175#endif /* HAVE_GETHOSTBYNAME_R */
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003176
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003177 if (!PyArg_ParseTuple(args, "et:gethostbyname_ex", "idna", &name))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003178 return NULL;
3179 if (setipaddr(name, (struct sockaddr *)&addr, sizeof(addr), AF_INET) < 0)
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003180 goto finally;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003181 Py_BEGIN_ALLOW_THREADS
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003182#ifdef HAVE_GETHOSTBYNAME_R
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003183#if defined(HAVE_GETHOSTBYNAME_R_6_ARG)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003184 result = gethostbyname_r(name, &hp_allocated, buf, buf_len,
3185 &h, &errnop);
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003186#elif defined(HAVE_GETHOSTBYNAME_R_5_ARG)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003187 h = gethostbyname_r(name, &hp_allocated, buf, buf_len, &errnop);
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003188#else /* HAVE_GETHOSTBYNAME_R_3_ARG */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003189 memset((void *) &data, '\0', sizeof(data));
3190 result = gethostbyname_r(name, &hp_allocated, &data);
3191 h = (result != 0) ? NULL : &hp_allocated;
Guido van Rossume9cd07b1999-03-15 21:40:14 +00003192#endif
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003193#else /* not HAVE_GETHOSTBYNAME_R */
Guido van Rossum3baaa131999-03-22 21:44:51 +00003194#ifdef USE_GETHOSTBYNAME_LOCK
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003195 PyThread_acquire_lock(netdb_lock, 1);
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003196#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003197 h = gethostbyname(name);
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003198#endif /* HAVE_GETHOSTBYNAME_R */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003199 Py_END_ALLOW_THREADS
3200 /* Some C libraries would require addr.__ss_family instead of
3201 addr.ss_family.
3202 Therefore, we cast the sockaddr_storage into sockaddr to
3203 access sa_family. */
3204 sa = (struct sockaddr*)&addr;
3205 ret = gethost_common(h, (struct sockaddr *)&addr, sizeof(addr),
3206 sa->sa_family);
Guido van Rossum3baaa131999-03-22 21:44:51 +00003207#ifdef USE_GETHOSTBYNAME_LOCK
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003208 PyThread_release_lock(netdb_lock);
Guido van Rossum955becc1999-03-22 20:14:53 +00003209#endif
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003210finally:
3211 PyMem_Free(name);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003212 return ret;
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003213}
3214
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003215PyDoc_STRVAR(ghbn_ex_doc,
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003216"gethostbyname_ex(host) -> (name, aliaslist, addresslist)\n\
3217\n\
3218Return the true host name, a list of aliases, and a list of IP addresses,\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003219for a host. The host argument is a string giving a host name or IP number.");
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003220
3221
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00003222/* Python interface to gethostbyaddr(IP). */
3223
3224/*ARGSUSED*/
3225static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003226socket_gethostbyaddr(PyObject *self, PyObject *args)
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00003227{
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00003228#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003229 struct sockaddr_storage addr;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003230#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003231 struct sockaddr_in addr;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003232#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003233 struct sockaddr *sa = (struct sockaddr *)&addr;
3234 char *ip_num;
3235 struct hostent *h;
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003236 PyObject *ret = NULL;
Guido van Rossum4f199ea1998-04-09 20:56:35 +00003237#ifdef HAVE_GETHOSTBYNAME_R
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003238 struct hostent hp_allocated;
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003239#ifdef HAVE_GETHOSTBYNAME_R_3_ARG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003240 struct hostent_data data;
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003241#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003242 /* glibcs up to 2.10 assume that the buf argument to
3243 gethostbyaddr_r is 8-byte aligned, which at least llvm-gcc
3244 does not ensure. The attribute below instructs the compiler
3245 to maintain this alignment. */
3246 char buf[16384] Py_ALIGNED(8);
3247 int buf_len = (sizeof buf) - 1;
3248 int errnop;
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003249#endif
3250#if defined(HAVE_GETHOSTBYNAME_R_3_ARG) || defined(HAVE_GETHOSTBYNAME_R_6_ARG)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003251 int result;
Guido van Rossume9cd07b1999-03-15 21:40:14 +00003252#endif
Guido van Rossum4f199ea1998-04-09 20:56:35 +00003253#endif /* HAVE_GETHOSTBYNAME_R */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003254 char *ap;
3255 int al;
3256 int af;
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00003257
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003258 if (!PyArg_ParseTuple(args, "et:gethostbyaddr", "idna", &ip_num))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003259 return NULL;
3260 af = AF_UNSPEC;
3261 if (setipaddr(ip_num, sa, sizeof(addr), af) < 0)
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003262 goto finally;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003263 af = sa->sa_family;
3264 ap = NULL;
3265 al = 0;
3266 switch (af) {
3267 case AF_INET:
3268 ap = (char *)&((struct sockaddr_in *)sa)->sin_addr;
3269 al = sizeof(((struct sockaddr_in *)sa)->sin_addr);
3270 break;
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00003271#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003272 case AF_INET6:
3273 ap = (char *)&((struct sockaddr_in6 *)sa)->sin6_addr;
3274 al = sizeof(((struct sockaddr_in6 *)sa)->sin6_addr);
3275 break;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003276#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003277 default:
3278 PyErr_SetString(socket_error, "unsupported address family");
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003279 goto finally;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003280 }
3281 Py_BEGIN_ALLOW_THREADS
Guido van Rossum4f199ea1998-04-09 20:56:35 +00003282#ifdef HAVE_GETHOSTBYNAME_R
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003283#if defined(HAVE_GETHOSTBYNAME_R_6_ARG)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003284 result = gethostbyaddr_r(ap, al, af,
3285 &hp_allocated, buf, buf_len,
3286 &h, &errnop);
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003287#elif defined(HAVE_GETHOSTBYNAME_R_5_ARG)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003288 h = gethostbyaddr_r(ap, al, af,
3289 &hp_allocated, buf, buf_len, &errnop);
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003290#else /* HAVE_GETHOSTBYNAME_R_3_ARG */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003291 memset((void *) &data, '\0', sizeof(data));
3292 result = gethostbyaddr_r(ap, al, af, &hp_allocated, &data);
3293 h = (result != 0) ? NULL : &hp_allocated;
Guido van Rossume9cd07b1999-03-15 21:40:14 +00003294#endif
Guido van Rossum4f199ea1998-04-09 20:56:35 +00003295#else /* not HAVE_GETHOSTBYNAME_R */
Guido van Rossum3baaa131999-03-22 21:44:51 +00003296#ifdef USE_GETHOSTBYNAME_LOCK
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003297 PyThread_acquire_lock(netdb_lock, 1);
Guido van Rossum4f199ea1998-04-09 20:56:35 +00003298#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003299 h = gethostbyaddr(ap, al, af);
Guido van Rossum4f199ea1998-04-09 20:56:35 +00003300#endif /* HAVE_GETHOSTBYNAME_R */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003301 Py_END_ALLOW_THREADS
3302 ret = gethost_common(h, (struct sockaddr *)&addr, sizeof(addr), af);
Guido van Rossum3baaa131999-03-22 21:44:51 +00003303#ifdef USE_GETHOSTBYNAME_LOCK
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003304 PyThread_release_lock(netdb_lock);
Guido van Rossum3baaa131999-03-22 21:44:51 +00003305#endif
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003306finally:
3307 PyMem_Free(ip_num);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003308 return ret;
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00003309}
3310
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003311PyDoc_STRVAR(gethostbyaddr_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003312"gethostbyaddr(host) -> (name, aliaslist, addresslist)\n\
3313\n\
3314Return the true host name, a list of aliases, and a list of IP addresses,\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003315for a host. The host argument is a string giving a host name or IP number.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003316
Guido van Rossum30a685f1991-06-27 15:51:29 +00003317
3318/* Python interface to getservbyname(name).
3319 This only returns the port number, since the other info is already
3320 known or not useful (like the list of aliases). */
3321
3322/*ARGSUSED*/
Guido van Rossum73624e91994-10-10 17:59:00 +00003323static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003324socket_getservbyname(PyObject *self, PyObject *args)
Guido van Rossum30a685f1991-06-27 15:51:29 +00003325{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003326 char *name, *proto=NULL;
3327 struct servent *sp;
3328 if (!PyArg_ParseTuple(args, "s|s:getservbyname", &name, &proto))
3329 return NULL;
3330 Py_BEGIN_ALLOW_THREADS
3331 sp = getservbyname(name, proto);
3332 Py_END_ALLOW_THREADS
3333 if (sp == NULL) {
3334 PyErr_SetString(socket_error, "service/proto not found");
3335 return NULL;
3336 }
3337 return PyLong_FromLong((long) ntohs(sp->s_port));
Guido van Rossum30a685f1991-06-27 15:51:29 +00003338}
3339
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003340PyDoc_STRVAR(getservbyname_doc,
Barry Warsaw11b91a02004-06-28 00:50:43 +00003341"getservbyname(servicename[, protocolname]) -> integer\n\
Guido van Rossum82a5c661998-07-07 20:45:43 +00003342\n\
3343Return a port number from a service name and protocol name.\n\
Barry Warsaw11b91a02004-06-28 00:50:43 +00003344The optional protocol name, if given, should be 'tcp' or 'udp',\n\
3345otherwise any protocol will match.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003346
Guido van Rossum30a685f1991-06-27 15:51:29 +00003347
Barry Warsaw11b91a02004-06-28 00:50:43 +00003348/* Python interface to getservbyport(port).
3349 This only returns the service name, since the other info is already
3350 known or not useful (like the list of aliases). */
3351
3352/*ARGSUSED*/
3353static PyObject *
3354socket_getservbyport(PyObject *self, PyObject *args)
3355{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003356 int port;
3357 char *proto=NULL;
3358 struct servent *sp;
3359 if (!PyArg_ParseTuple(args, "i|s:getservbyport", &port, &proto))
3360 return NULL;
3361 if (port < 0 || port > 0xffff) {
3362 PyErr_SetString(
3363 PyExc_OverflowError,
3364 "getservbyport: port must be 0-65535.");
3365 return NULL;
3366 }
3367 Py_BEGIN_ALLOW_THREADS
3368 sp = getservbyport(htons((short)port), proto);
3369 Py_END_ALLOW_THREADS
3370 if (sp == NULL) {
3371 PyErr_SetString(socket_error, "port/proto not found");
3372 return NULL;
3373 }
3374 return PyUnicode_FromString(sp->s_name);
Barry Warsaw11b91a02004-06-28 00:50:43 +00003375}
3376
3377PyDoc_STRVAR(getservbyport_doc,
3378"getservbyport(port[, protocolname]) -> string\n\
3379\n\
3380Return the service name from a port number and protocol name.\n\
3381The optional protocol name, if given, should be 'tcp' or 'udp',\n\
3382otherwise any protocol will match.");
3383
Guido van Rossum3901d851996-12-19 16:35:04 +00003384/* Python interface to getprotobyname(name).
3385 This only returns the protocol number, since the other info is
3386 already known or not useful (like the list of aliases). */
3387
3388/*ARGSUSED*/
3389static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003390socket_getprotobyname(PyObject *self, PyObject *args)
Guido van Rossum3901d851996-12-19 16:35:04 +00003391{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003392 char *name;
3393 struct protoent *sp;
3394 if (!PyArg_ParseTuple(args, "s:getprotobyname", &name))
3395 return NULL;
3396 Py_BEGIN_ALLOW_THREADS
3397 sp = getprotobyname(name);
3398 Py_END_ALLOW_THREADS
3399 if (sp == NULL) {
3400 PyErr_SetString(socket_error, "protocol not found");
3401 return NULL;
3402 }
3403 return PyLong_FromLong((long) sp->p_proto);
Guido van Rossum3901d851996-12-19 16:35:04 +00003404}
3405
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003406PyDoc_STRVAR(getprotobyname_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003407"getprotobyname(name) -> integer\n\
3408\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003409Return the protocol number for the named protocol. (Rarely used.)");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003410
Guido van Rossum3901d851996-12-19 16:35:04 +00003411
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00003412#ifndef NO_DUP
3413/* dup() function for socket fds */
3414
3415static PyObject *
3416socket_dup(PyObject *self, PyObject *fdobj)
3417{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003418 SOCKET_T fd, newfd;
3419 PyObject *newfdobj;
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00003420
3421
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003422 fd = PyLong_AsSocket_t(fdobj);
3423 if (fd == (SOCKET_T)(-1) && PyErr_Occurred())
3424 return NULL;
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00003425
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003426 newfd = dup_socket(fd);
3427 if (newfd == INVALID_SOCKET)
3428 return set_error();
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00003429
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003430 newfdobj = PyLong_FromSocket_t(newfd);
3431 if (newfdobj == NULL)
3432 SOCKETCLOSE(newfd);
3433 return newfdobj;
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00003434}
3435
3436PyDoc_STRVAR(dup_doc,
3437"dup(integer) -> integer\n\
3438\n\
3439Duplicate an integer socket file descriptor. This is like os.dup(), but for\n\
3440sockets; on some platforms os.dup() won't work for socket file descriptors.");
3441#endif
3442
3443
Dave Cole331708b2004-08-09 04:51:41 +00003444#ifdef HAVE_SOCKETPAIR
3445/* Create a pair of sockets using the socketpair() function.
Dave Cole07fda7e2004-08-23 05:16:23 +00003446 Arguments as for socket() except the default family is AF_UNIX if
Dave Colee8bbfe42004-08-26 00:51:16 +00003447 defined on the platform; otherwise, the default is AF_INET. */
Dave Cole331708b2004-08-09 04:51:41 +00003448
3449/*ARGSUSED*/
3450static PyObject *
3451socket_socketpair(PyObject *self, PyObject *args)
3452{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003453 PySocketSockObject *s0 = NULL, *s1 = NULL;
3454 SOCKET_T sv[2];
3455 int family, type = SOCK_STREAM, proto = 0;
3456 PyObject *res = NULL;
Dave Cole331708b2004-08-09 04:51:41 +00003457
3458#if defined(AF_UNIX)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003459 family = AF_UNIX;
Dave Cole331708b2004-08-09 04:51:41 +00003460#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003461 family = AF_INET;
Dave Cole331708b2004-08-09 04:51:41 +00003462#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003463 if (!PyArg_ParseTuple(args, "|iii:socketpair",
3464 &family, &type, &proto))
3465 return NULL;
3466 /* Create a pair of socket fds */
3467 if (socketpair(family, type, proto, sv) < 0)
3468 return set_error();
3469 s0 = new_sockobject(sv[0], family, type, proto);
3470 if (s0 == NULL)
3471 goto finally;
3472 s1 = new_sockobject(sv[1], family, type, proto);
3473 if (s1 == NULL)
3474 goto finally;
3475 res = PyTuple_Pack(2, s0, s1);
Dave Cole331708b2004-08-09 04:51:41 +00003476
3477finally:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003478 if (res == NULL) {
3479 if (s0 == NULL)
3480 SOCKETCLOSE(sv[0]);
3481 if (s1 == NULL)
3482 SOCKETCLOSE(sv[1]);
3483 }
3484 Py_XDECREF(s0);
3485 Py_XDECREF(s1);
3486 return res;
Dave Cole331708b2004-08-09 04:51:41 +00003487}
3488
3489PyDoc_STRVAR(socketpair_doc,
3490"socketpair([family[, type[, proto]]]) -> (socket object, socket object)\n\
3491\n\
3492Create a pair of socket objects from the sockets returned by the platform\n\
3493socketpair() function.\n\
Dave Cole07fda7e2004-08-23 05:16:23 +00003494The arguments are the same as for socket() except the default family is\n\
Dave Colee8bbfe42004-08-26 00:51:16 +00003495AF_UNIX if defined on the platform; otherwise, the default is AF_INET.");
Dave Cole331708b2004-08-09 04:51:41 +00003496
3497#endif /* HAVE_SOCKETPAIR */
3498
3499
Guido van Rossum006bf911996-06-12 04:04:55 +00003500static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003501socket_ntohs(PyObject *self, PyObject *args)
Guido van Rossum006bf911996-06-12 04:04:55 +00003502{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003503 int x1, x2;
Guido van Rossum006bf911996-06-12 04:04:55 +00003504
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003505 if (!PyArg_ParseTuple(args, "i:ntohs", &x1)) {
3506 return NULL;
3507 }
3508 if (x1 < 0) {
3509 PyErr_SetString(PyExc_OverflowError,
3510 "can't convert negative number to unsigned long");
3511 return NULL;
3512 }
3513 x2 = (unsigned int)ntohs((unsigned short)x1);
3514 return PyLong_FromLong(x2);
Guido van Rossum006bf911996-06-12 04:04:55 +00003515}
3516
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003517PyDoc_STRVAR(ntohs_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003518"ntohs(integer) -> integer\n\
3519\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003520Convert a 16-bit integer from network to host byte order.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003521
3522
Guido van Rossum006bf911996-06-12 04:04:55 +00003523static PyObject *
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003524socket_ntohl(PyObject *self, PyObject *arg)
Guido van Rossum006bf911996-06-12 04:04:55 +00003525{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003526 unsigned long x;
Guido van Rossum006bf911996-06-12 04:04:55 +00003527
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003528 if (PyLong_Check(arg)) {
3529 x = PyLong_AsUnsignedLong(arg);
3530 if (x == (unsigned long) -1 && PyErr_Occurred())
3531 return NULL;
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003532#if SIZEOF_LONG > 4
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003533 {
3534 unsigned long y;
3535 /* only want the trailing 32 bits */
3536 y = x & 0xFFFFFFFFUL;
3537 if (y ^ x)
3538 return PyErr_Format(PyExc_OverflowError,
3539 "long int larger than 32 bits");
3540 x = y;
3541 }
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003542#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003543 }
3544 else
3545 return PyErr_Format(PyExc_TypeError,
3546 "expected int/long, %s found",
3547 Py_TYPE(arg)->tp_name);
3548 if (x == (unsigned long) -1 && PyErr_Occurred())
3549 return NULL;
3550 return PyLong_FromUnsignedLong(ntohl(x));
Guido van Rossum006bf911996-06-12 04:04:55 +00003551}
3552
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003553PyDoc_STRVAR(ntohl_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003554"ntohl(integer) -> integer\n\
3555\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003556Convert a 32-bit integer from network to host byte order.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003557
3558
Guido van Rossum006bf911996-06-12 04:04:55 +00003559static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003560socket_htons(PyObject *self, PyObject *args)
Guido van Rossum006bf911996-06-12 04:04:55 +00003561{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003562 int x1, x2;
Guido van Rossum006bf911996-06-12 04:04:55 +00003563
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003564 if (!PyArg_ParseTuple(args, "i:htons", &x1)) {
3565 return NULL;
3566 }
3567 if (x1 < 0) {
3568 PyErr_SetString(PyExc_OverflowError,
3569 "can't convert negative number to unsigned long");
3570 return NULL;
3571 }
3572 x2 = (unsigned int)htons((unsigned short)x1);
3573 return PyLong_FromLong(x2);
Guido van Rossum006bf911996-06-12 04:04:55 +00003574}
3575
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003576PyDoc_STRVAR(htons_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003577"htons(integer) -> integer\n\
3578\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003579Convert a 16-bit integer from host to network byte order.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003580
3581
Guido van Rossum006bf911996-06-12 04:04:55 +00003582static PyObject *
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003583socket_htonl(PyObject *self, PyObject *arg)
Guido van Rossum006bf911996-06-12 04:04:55 +00003584{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003585 unsigned long x;
Guido van Rossum006bf911996-06-12 04:04:55 +00003586
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003587 if (PyLong_Check(arg)) {
3588 x = PyLong_AsUnsignedLong(arg);
3589 if (x == (unsigned long) -1 && PyErr_Occurred())
3590 return NULL;
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003591#if SIZEOF_LONG > 4
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003592 {
3593 unsigned long y;
3594 /* only want the trailing 32 bits */
3595 y = x & 0xFFFFFFFFUL;
3596 if (y ^ x)
3597 return PyErr_Format(PyExc_OverflowError,
3598 "long int larger than 32 bits");
3599 x = y;
3600 }
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003601#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003602 }
3603 else
3604 return PyErr_Format(PyExc_TypeError,
3605 "expected int/long, %s found",
3606 Py_TYPE(arg)->tp_name);
3607 return PyLong_FromUnsignedLong(htonl((unsigned long)x));
Guido van Rossum006bf911996-06-12 04:04:55 +00003608}
3609
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003610PyDoc_STRVAR(htonl_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003611"htonl(integer) -> integer\n\
3612\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003613Convert a 32-bit integer from host to network byte order.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003614
Guido van Rossum3eede5a2002-06-07 02:08:35 +00003615/* socket.inet_aton() and socket.inet_ntoa() functions. */
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003616
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003617PyDoc_STRVAR(inet_aton_doc,
Guido van Rossum7d0a8262007-05-21 23:13:11 +00003618"inet_aton(string) -> bytes giving packed 32-bit IP representation\n\
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003619\n\
Guido van Rossumc6a164b1999-08-20 19:11:27 +00003620Convert 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 +00003621binary format used in low-level network functions.");
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003622
3623static PyObject*
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003624socket_inet_aton(PyObject *self, PyObject *args)
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003625{
Guido van Rossuma2e48551999-09-09 15:42:59 +00003626#ifndef INADDR_NONE
3627#define INADDR_NONE (-1)
3628#endif
Neal Norwitz88f115b2003-02-13 02:15:42 +00003629#ifdef HAVE_INET_ATON
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003630 struct in_addr buf;
Thomas Wouters477c8d52006-05-27 19:21:47 +00003631#endif
3632
3633#if !defined(HAVE_INET_ATON) || defined(USE_INET_ATON_WEAKLINK)
Benjamin Petersonf91df042009-02-13 02:50:59 +00003634#if (SIZEOF_INT != 4)
3635#error "Not sure if in_addr_t exists and int is not 32-bits."
3636#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003637 /* Have to use inet_addr() instead */
3638 unsigned int packed_addr;
Tim Peters1df9fdd2003-02-13 03:13:40 +00003639#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003640 char *ip_addr;
Guido van Rossumc6a164b1999-08-20 19:11:27 +00003641
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003642 if (!PyArg_ParseTuple(args, "s:inet_aton", &ip_addr))
3643 return NULL;
Guido van Rossumad05cdf2003-02-12 23:08:22 +00003644
Tim Peters1df9fdd2003-02-13 03:13:40 +00003645
3646#ifdef HAVE_INET_ATON
Thomas Wouters477c8d52006-05-27 19:21:47 +00003647
3648#ifdef USE_INET_ATON_WEAKLINK
3649 if (inet_aton != NULL) {
3650#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003651 if (inet_aton(ip_addr, &buf))
3652 return PyBytes_FromStringAndSize((char *)(&buf),
3653 sizeof(buf));
Guido van Rossumad05cdf2003-02-12 23:08:22 +00003654
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003655 PyErr_SetString(socket_error,
3656 "illegal IP address string passed to inet_aton");
3657 return NULL;
Guido van Rossumad05cdf2003-02-12 23:08:22 +00003658
Thomas Wouters477c8d52006-05-27 19:21:47 +00003659#ifdef USE_INET_ATON_WEAKLINK
3660 } else {
3661#endif
3662
3663#endif
3664
3665#if !defined(HAVE_INET_ATON) || defined(USE_INET_ATON_WEAKLINK)
3666
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003667 /* special-case this address as inet_addr might return INADDR_NONE
3668 * for this */
3669 if (strcmp(ip_addr, "255.255.255.255") == 0) {
3670 packed_addr = 0xFFFFFFFF;
3671 } else {
Thomas Wouters477c8d52006-05-27 19:21:47 +00003672
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003673 packed_addr = inet_addr(ip_addr);
Guido van Rossumc6a164b1999-08-20 19:11:27 +00003674
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003675 if (packed_addr == INADDR_NONE) { /* invalid address */
3676 PyErr_SetString(socket_error,
3677 "illegal IP address string passed to inet_aton");
3678 return NULL;
3679 }
3680 }
3681 return PyBytes_FromStringAndSize((char *) &packed_addr,
3682 sizeof(packed_addr));
Thomas Wouters477c8d52006-05-27 19:21:47 +00003683
3684#ifdef USE_INET_ATON_WEAKLINK
3685 }
3686#endif
3687
Guido van Rossumad05cdf2003-02-12 23:08:22 +00003688#endif
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003689}
3690
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003691PyDoc_STRVAR(inet_ntoa_doc,
Fred Drakee0661342000-03-07 14:05:16 +00003692"inet_ntoa(packed_ip) -> ip_address_string\n\
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003693\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003694Convert an IP address from 32-bit packed binary format to string format");
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003695
3696static PyObject*
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003697socket_inet_ntoa(PyObject *self, PyObject *args)
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003698{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003699 char *packed_str;
3700 int addr_len;
3701 struct in_addr packed_addr;
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003702
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003703 if (!PyArg_ParseTuple(args, "y#:inet_ntoa", &packed_str, &addr_len)) {
3704 return NULL;
3705 }
Guido van Rossum48a680c2001-03-02 06:34:14 +00003706
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003707 if (addr_len != sizeof(packed_addr)) {
3708 PyErr_SetString(socket_error,
3709 "packed IP wrong length for inet_ntoa");
3710 return NULL;
3711 }
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003712
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003713 memcpy(&packed_addr, packed_str, addr_len);
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003714
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003715 return PyUnicode_FromString(inet_ntoa(packed_addr));
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003716}
Guido van Rossum82a5c661998-07-07 20:45:43 +00003717
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003718#ifdef HAVE_INET_PTON
3719
3720PyDoc_STRVAR(inet_pton_doc,
3721"inet_pton(af, ip) -> packed IP address string\n\
3722\n\
3723Convert an IP address from string format to a packed string suitable\n\
3724for use with low-level network functions.");
3725
3726static PyObject *
3727socket_inet_pton(PyObject *self, PyObject *args)
3728{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003729 int af;
3730 char* ip;
3731 int retval;
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003732#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003733 char packed[MAX(sizeof(struct in_addr), sizeof(struct in6_addr))];
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003734#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003735 char packed[sizeof(struct in_addr)];
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003736#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003737 if (!PyArg_ParseTuple(args, "is:inet_pton", &af, &ip)) {
3738 return NULL;
3739 }
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003740
Martin v. Löwis04697e82004-06-02 12:35:29 +00003741#if !defined(ENABLE_IPV6) && defined(AF_INET6)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003742 if(af == AF_INET6) {
3743 PyErr_SetString(socket_error,
3744 "can't use AF_INET6, IPv6 is disabled");
3745 return NULL;
3746 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00003747#endif
Martin v. Löwis10649092003-08-05 06:25:06 +00003748
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003749 retval = inet_pton(af, ip, packed);
3750 if (retval < 0) {
3751 PyErr_SetFromErrno(socket_error);
3752 return NULL;
3753 } else if (retval == 0) {
3754 PyErr_SetString(socket_error,
3755 "illegal IP address string passed to inet_pton");
3756 return NULL;
3757 } else if (af == AF_INET) {
3758 return PyBytes_FromStringAndSize(packed,
3759 sizeof(struct in_addr));
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003760#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003761 } else if (af == AF_INET6) {
3762 return PyBytes_FromStringAndSize(packed,
3763 sizeof(struct in6_addr));
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003764#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003765 } else {
3766 PyErr_SetString(socket_error, "unknown address family");
3767 return NULL;
3768 }
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003769}
Thomas Wouters477c8d52006-05-27 19:21:47 +00003770
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003771PyDoc_STRVAR(inet_ntop_doc,
3772"inet_ntop(af, packed_ip) -> string formatted IP address\n\
3773\n\
3774Convert a packed IP address of the given family to string format.");
3775
3776static PyObject *
3777socket_inet_ntop(PyObject *self, PyObject *args)
3778{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003779 int af;
3780 char* packed;
3781 int len;
3782 const char* retval;
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003783#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003784 char ip[MAX(INET_ADDRSTRLEN, INET6_ADDRSTRLEN) + 1];
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003785#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003786 char ip[INET_ADDRSTRLEN + 1];
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003787#endif
Thomas Wouters477c8d52006-05-27 19:21:47 +00003788
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003789 /* Guarantee NUL-termination for PyUnicode_FromString() below */
3790 memset((void *) &ip[0], '\0', sizeof(ip));
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003791
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003792 if (!PyArg_ParseTuple(args, "iy#:inet_ntop", &af, &packed, &len)) {
3793 return NULL;
3794 }
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003795
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003796 if (af == AF_INET) {
3797 if (len != sizeof(struct in_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#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003803 } else if (af == AF_INET6) {
3804 if (len != sizeof(struct in6_addr)) {
3805 PyErr_SetString(PyExc_ValueError,
3806 "invalid length of packed IP address string");
3807 return NULL;
3808 }
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003809#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003810 } else {
3811 PyErr_Format(PyExc_ValueError,
3812 "unknown address family %d", af);
3813 return NULL;
3814 }
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003815
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003816 retval = inet_ntop(af, packed, ip, sizeof(ip));
3817 if (!retval) {
3818 PyErr_SetFromErrno(socket_error);
3819 return NULL;
3820 } else {
3821 return PyUnicode_FromString(retval);
3822 }
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003823
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003824 /* NOTREACHED */
3825 PyErr_SetString(PyExc_RuntimeError, "invalid handling of inet_ntop");
3826 return NULL;
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003827}
3828
3829#endif /* HAVE_INET_PTON */
3830
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003831/* Python interface to getaddrinfo(host, port). */
3832
3833/*ARGSUSED*/
3834static PyObject *
Giampaolo Rodolàccfb91c2010-08-17 15:30:23 +00003835socket_getaddrinfo(PyObject *self, PyObject *args, PyObject* kwargs)
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003836{
Giampaolo Rodolàccfb91c2010-08-17 15:30:23 +00003837 static char* kwnames[] = {"host", "port", "family", "type", "proto",
3838 "flags", 0};
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003839 struct addrinfo hints, *res;
3840 struct addrinfo *res0 = NULL;
3841 PyObject *hobj = NULL;
3842 PyObject *pobj = (PyObject *)NULL;
3843 char pbuf[30];
3844 char *hptr, *pptr;
3845 int family, socktype, protocol, flags;
3846 int error;
3847 PyObject *all = (PyObject *)NULL;
3848 PyObject *idna = NULL;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003849
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003850 family = socktype = protocol = flags = 0;
3851 family = AF_UNSPEC;
Giampaolo Rodolàccfb91c2010-08-17 15:30:23 +00003852 if (!PyArg_ParseTupleAndKeywords(args, kwargs, "OO|iiii:getaddrinfo",
3853 kwnames, &hobj, &pobj, &family, &socktype,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003854 &protocol, &flags)) {
3855 return NULL;
3856 }
3857 if (hobj == Py_None) {
3858 hptr = NULL;
3859 } else if (PyUnicode_Check(hobj)) {
3860 idna = PyObject_CallMethod(hobj, "encode", "s", "idna");
3861 if (!idna)
3862 return NULL;
3863 assert(PyBytes_Check(idna));
3864 hptr = PyBytes_AS_STRING(idna);
3865 } else if (PyBytes_Check(hobj)) {
3866 hptr = PyBytes_AsString(hobj);
3867 } else {
3868 PyErr_SetString(PyExc_TypeError,
3869 "getaddrinfo() argument 1 must be string or None");
3870 return NULL;
3871 }
3872 if (PyLong_CheckExact(pobj)) {
3873 long value = PyLong_AsLong(pobj);
3874 if (value == -1 && PyErr_Occurred())
3875 goto err;
3876 PyOS_snprintf(pbuf, sizeof(pbuf), "%ld", value);
3877 pptr = pbuf;
3878 } else if (PyUnicode_Check(pobj)) {
3879 pptr = _PyUnicode_AsString(pobj);
3880 } else if (PyBytes_Check(pobj)) {
3881 pptr = PyBytes_AsString(pobj);
3882 } else if (pobj == Py_None) {
3883 pptr = (char *)NULL;
3884 } else {
3885 PyErr_SetString(socket_error, "Int or String expected");
3886 goto err;
3887 }
3888 memset(&hints, 0, sizeof(hints));
3889 hints.ai_family = family;
3890 hints.ai_socktype = socktype;
3891 hints.ai_protocol = protocol;
3892 hints.ai_flags = flags;
3893 Py_BEGIN_ALLOW_THREADS
3894 ACQUIRE_GETADDRINFO_LOCK
3895 error = getaddrinfo(hptr, pptr, &hints, &res0);
3896 Py_END_ALLOW_THREADS
3897 RELEASE_GETADDRINFO_LOCK /* see comment in setipaddr() */
3898 if (error) {
3899 set_gaierror(error);
3900 goto err;
3901 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003902
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003903 if ((all = PyList_New(0)) == NULL)
3904 goto err;
3905 for (res = res0; res; res = res->ai_next) {
3906 PyObject *single;
3907 PyObject *addr =
3908 makesockaddr(-1, res->ai_addr, res->ai_addrlen, protocol);
3909 if (addr == NULL)
3910 goto err;
3911 single = Py_BuildValue("iiisO", res->ai_family,
3912 res->ai_socktype, res->ai_protocol,
3913 res->ai_canonname ? res->ai_canonname : "",
3914 addr);
3915 Py_DECREF(addr);
3916 if (single == NULL)
3917 goto err;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003918
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003919 if (PyList_Append(all, single))
3920 goto err;
3921 Py_XDECREF(single);
3922 }
3923 Py_XDECREF(idna);
3924 if (res0)
3925 freeaddrinfo(res0);
3926 return all;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003927 err:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003928 Py_XDECREF(all);
3929 Py_XDECREF(idna);
3930 if (res0)
3931 freeaddrinfo(res0);
3932 return (PyObject *)NULL;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003933}
3934
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003935PyDoc_STRVAR(getaddrinfo_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00003936"getaddrinfo(host, port [, family, socktype, proto, flags])\n\
3937 -> list of (family, socktype, proto, canonname, sockaddr)\n\
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003938\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003939Resolve host and port into addrinfo struct.");
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003940
3941/* Python interface to getnameinfo(sa, flags). */
3942
3943/*ARGSUSED*/
3944static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003945socket_getnameinfo(PyObject *self, PyObject *args)
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003946{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003947 PyObject *sa = (PyObject *)NULL;
3948 int flags;
3949 char *hostp;
3950 int port, flowinfo, scope_id;
3951 char hbuf[NI_MAXHOST], pbuf[NI_MAXSERV];
3952 struct addrinfo hints, *res = NULL;
3953 int error;
3954 PyObject *ret = (PyObject *)NULL;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003955
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003956 flags = flowinfo = scope_id = 0;
3957 if (!PyArg_ParseTuple(args, "Oi:getnameinfo", &sa, &flags))
3958 return NULL;
3959 if (!PyTuple_Check(sa)) {
3960 PyErr_SetString(PyExc_TypeError,
3961 "getnameinfo() argument 1 must be a tuple");
3962 return NULL;
3963 }
3964 if (!PyArg_ParseTuple(sa, "si|ii",
3965 &hostp, &port, &flowinfo, &scope_id))
3966 return NULL;
3967 PyOS_snprintf(pbuf, sizeof(pbuf), "%d", port);
3968 memset(&hints, 0, sizeof(hints));
3969 hints.ai_family = AF_UNSPEC;
3970 hints.ai_socktype = SOCK_DGRAM; /* make numeric port happy */
Martin v. Löwis112c0f32010-08-25 07:38:15 +00003971 hints.ai_flags = AI_NUMERICHOST; /* don't do any name resolution */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003972 Py_BEGIN_ALLOW_THREADS
3973 ACQUIRE_GETADDRINFO_LOCK
3974 error = getaddrinfo(hostp, pbuf, &hints, &res);
3975 Py_END_ALLOW_THREADS
3976 RELEASE_GETADDRINFO_LOCK /* see comment in setipaddr() */
3977 if (error) {
3978 set_gaierror(error);
3979 goto fail;
3980 }
3981 if (res->ai_next) {
3982 PyErr_SetString(socket_error,
3983 "sockaddr resolved to multiple addresses");
3984 goto fail;
3985 }
3986 switch (res->ai_family) {
3987 case AF_INET:
3988 {
3989 if (PyTuple_GET_SIZE(sa) != 2) {
3990 PyErr_SetString(socket_error,
3991 "IPv4 sockaddr must be 2 tuple");
3992 goto fail;
3993 }
3994 break;
3995 }
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00003996#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003997 case AF_INET6:
3998 {
3999 struct sockaddr_in6 *sin6;
4000 sin6 = (struct sockaddr_in6 *)res->ai_addr;
4001 sin6->sin6_flowinfo = flowinfo;
4002 sin6->sin6_scope_id = scope_id;
4003 break;
4004 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004005#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004006 }
Antoine Pitrou19467d22010-08-17 19:33:30 +00004007 error = getnameinfo(res->ai_addr, (socklen_t) res->ai_addrlen,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004008 hbuf, sizeof(hbuf), pbuf, sizeof(pbuf), flags);
4009 if (error) {
4010 set_gaierror(error);
4011 goto fail;
4012 }
4013 ret = Py_BuildValue("ss", hbuf, pbuf);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004014
4015fail:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004016 if (res)
4017 freeaddrinfo(res);
4018 return ret;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004019}
4020
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004021PyDoc_STRVAR(getnameinfo_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00004022"getnameinfo(sockaddr, flags) --> (host, port)\n\
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004023\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004024Get host and port for a sockaddr.");
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004025
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004026
4027/* Python API to getting and setting the default timeout value. */
4028
4029static PyObject *
4030socket_getdefaulttimeout(PyObject *self)
4031{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004032 if (defaulttimeout < 0.0) {
4033 Py_INCREF(Py_None);
4034 return Py_None;
4035 }
4036 else
4037 return PyFloat_FromDouble(defaulttimeout);
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004038}
4039
4040PyDoc_STRVAR(getdefaulttimeout_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00004041"getdefaulttimeout() -> timeout\n\
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004042\n\
4043Returns the default timeout in floating seconds for new socket objects.\n\
4044A value of None indicates that new socket objects have no timeout.\n\
4045When the socket module is first imported, the default is None.");
4046
4047static PyObject *
4048socket_setdefaulttimeout(PyObject *self, PyObject *arg)
4049{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004050 double timeout;
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004051
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004052 if (arg == Py_None)
4053 timeout = -1.0;
4054 else {
4055 timeout = PyFloat_AsDouble(arg);
4056 if (timeout < 0.0) {
4057 if (!PyErr_Occurred())
4058 PyErr_SetString(PyExc_ValueError,
4059 "Timeout value out of range");
4060 return NULL;
4061 }
4062 }
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004063
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004064 defaulttimeout = timeout;
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004065
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004066 Py_INCREF(Py_None);
4067 return Py_None;
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004068}
4069
4070PyDoc_STRVAR(setdefaulttimeout_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00004071"setdefaulttimeout(timeout)\n\
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004072\n\
4073Set the default timeout in floating seconds for new socket objects.\n\
4074A value of None indicates that new socket objects have no timeout.\n\
4075When the socket module is first imported, the default is None.");
4076
4077
Guido van Rossum30a685f1991-06-27 15:51:29 +00004078/* List of functions exported by this module. */
4079
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004080static PyMethodDef socket_methods[] = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004081 {"gethostbyname", socket_gethostbyname,
4082 METH_VARARGS, gethostbyname_doc},
4083 {"gethostbyname_ex", socket_gethostbyname_ex,
4084 METH_VARARGS, ghbn_ex_doc},
4085 {"gethostbyaddr", socket_gethostbyaddr,
4086 METH_VARARGS, gethostbyaddr_doc},
4087 {"gethostname", socket_gethostname,
4088 METH_NOARGS, gethostname_doc},
4089 {"getservbyname", socket_getservbyname,
4090 METH_VARARGS, getservbyname_doc},
4091 {"getservbyport", socket_getservbyport,
4092 METH_VARARGS, getservbyport_doc},
4093 {"getprotobyname", socket_getprotobyname,
4094 METH_VARARGS, getprotobyname_doc},
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00004095#ifndef NO_DUP
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004096 {"dup", socket_dup,
4097 METH_O, dup_doc},
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00004098#endif
Dave Cole331708b2004-08-09 04:51:41 +00004099#ifdef HAVE_SOCKETPAIR
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004100 {"socketpair", socket_socketpair,
4101 METH_VARARGS, socketpair_doc},
Dave Cole331708b2004-08-09 04:51:41 +00004102#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004103 {"ntohs", socket_ntohs,
4104 METH_VARARGS, ntohs_doc},
4105 {"ntohl", socket_ntohl,
4106 METH_O, ntohl_doc},
4107 {"htons", socket_htons,
4108 METH_VARARGS, htons_doc},
4109 {"htonl", socket_htonl,
4110 METH_O, htonl_doc},
4111 {"inet_aton", socket_inet_aton,
4112 METH_VARARGS, inet_aton_doc},
4113 {"inet_ntoa", socket_inet_ntoa,
4114 METH_VARARGS, inet_ntoa_doc},
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00004115#ifdef HAVE_INET_PTON
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004116 {"inet_pton", socket_inet_pton,
4117 METH_VARARGS, inet_pton_doc},
4118 {"inet_ntop", socket_inet_ntop,
4119 METH_VARARGS, inet_ntop_doc},
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00004120#endif
Giampaolo Rodolàccfb91c2010-08-17 15:30:23 +00004121 {"getaddrinfo", (PyCFunction)socket_getaddrinfo,
4122 METH_VARARGS | METH_KEYWORDS, getaddrinfo_doc},
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004123 {"getnameinfo", socket_getnameinfo,
4124 METH_VARARGS, getnameinfo_doc},
4125 {"getdefaulttimeout", (PyCFunction)socket_getdefaulttimeout,
4126 METH_NOARGS, getdefaulttimeout_doc},
4127 {"setdefaulttimeout", socket_setdefaulttimeout,
4128 METH_O, setdefaulttimeout_doc},
4129 {NULL, NULL} /* Sentinel */
Guido van Rossum6574b3e1991-06-25 21:36:08 +00004130};
4131
Guido van Rossum30a685f1991-06-27 15:51:29 +00004132
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004133#ifdef MS_WINDOWS
4134#define OS_INIT_DEFINED
4135
4136/* Additional initialization and cleanup for Windows */
Guido van Rossumbe32c891996-06-20 16:25:29 +00004137
4138static void
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004139os_cleanup(void)
Guido van Rossumbe32c891996-06-20 16:25:29 +00004140{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004141 WSACleanup();
Guido van Rossumbe32c891996-06-20 16:25:29 +00004142}
4143
4144static int
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004145os_init(void)
Guido van Rossumbe32c891996-06-20 16:25:29 +00004146{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004147 WSADATA WSAData;
4148 int ret;
4149 ret = WSAStartup(0x0101, &WSAData);
4150 switch (ret) {
4151 case 0: /* No error */
4152 Py_AtExit(os_cleanup);
4153 return 1; /* Success */
4154 case WSASYSNOTREADY:
4155 PyErr_SetString(PyExc_ImportError,
4156 "WSAStartup failed: network not ready");
4157 break;
4158 case WSAVERNOTSUPPORTED:
4159 case WSAEINVAL:
4160 PyErr_SetString(
4161 PyExc_ImportError,
4162 "WSAStartup failed: requested version not supported");
4163 break;
4164 default:
4165 PyErr_Format(PyExc_ImportError, "WSAStartup failed: error code %d", ret);
4166 break;
4167 }
4168 return 0; /* Failure */
Guido van Rossumbe32c891996-06-20 16:25:29 +00004169}
4170
Guido van Rossum8d665e61996-06-26 18:22:49 +00004171#endif /* MS_WINDOWS */
Guido van Rossumbe32c891996-06-20 16:25:29 +00004172
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004173
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004174#ifdef PYOS_OS2
4175#define OS_INIT_DEFINED
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004176
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004177/* Additional initialization for OS/2 */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004178
4179static int
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004180os_init(void)
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004181{
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004182#ifndef PYCC_GCC
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004183 int rc = sock_init();
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004184
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004185 if (rc == 0) {
4186 return 1; /* Success */
4187 }
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004188
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004189 PyErr_Format(PyExc_ImportError, "OS/2 TCP/IP Error# %d", sock_errno());
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004190
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004191 return 0; /* Failure */
Andrew MacIntyreba43e872002-03-03 03:03:52 +00004192#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004193 /* No need to initialise sockets with GCC/EMX */
4194 return 1; /* Success */
Andrew MacIntyreba43e872002-03-03 03:03:52 +00004195#endif
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004196}
4197
4198#endif /* PYOS_OS2 */
4199
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004200
4201#ifndef OS_INIT_DEFINED
4202static int
4203os_init(void)
4204{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004205 return 1; /* Success */
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004206}
4207#endif
4208
4209
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00004210/* C API table - always add new things to the end for binary
4211 compatibility. */
4212static
4213PySocketModule_APIObject PySocketModuleAPI =
4214{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004215 &sock_type,
4216 NULL
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00004217};
4218
Guido van Rossum3eede5a2002-06-07 02:08:35 +00004219
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004220/* Initialize the _socket module.
Guido van Rossum3eede5a2002-06-07 02:08:35 +00004221
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004222 This module is actually called "_socket", and there's a wrapper
Guido van Rossum7d0a8262007-05-21 23:13:11 +00004223 "socket.py" which implements some additional functionality.
4224 The import of "_socket" may fail with an ImportError exception if
4225 os-specific initialization fails. On Windows, this does WINSOCK
4226 initialization. When WINSOCK is initialized succesfully, a call to
4227 WSACleanup() is scheduled to be made at exit time.
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004228*/
Guido van Rossum3eede5a2002-06-07 02:08:35 +00004229
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004230PyDoc_STRVAR(socket_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00004231"Implementation module for socket operations.\n\
4232\n\
4233See the socket module for documentation.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00004234
Martin v. Löwis1a214512008-06-11 05:26:20 +00004235static struct PyModuleDef socketmodule = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004236 PyModuleDef_HEAD_INIT,
4237 PySocket_MODULE_NAME,
4238 socket_doc,
4239 -1,
4240 socket_methods,
4241 NULL,
4242 NULL,
4243 NULL,
4244 NULL
Martin v. Löwis1a214512008-06-11 05:26:20 +00004245};
4246
Mark Hammondfe51c6d2002-08-02 02:27:13 +00004247PyMODINIT_FUNC
Martin v. Löwis1a214512008-06-11 05:26:20 +00004248PyInit__socket(void)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00004249{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004250 PyObject *m, *has_ipv6;
Fred Drake4baedc12002-04-01 14:53:37 +00004251
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004252 if (!os_init())
4253 return NULL;
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004254
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004255 Py_TYPE(&sock_type) = &PyType_Type;
4256 m = PyModule_Create(&socketmodule);
4257 if (m == NULL)
4258 return NULL;
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004259
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004260 socket_error = PyErr_NewException("socket.error",
4261 PyExc_IOError, NULL);
4262 if (socket_error == NULL)
4263 return NULL;
4264 PySocketModuleAPI.error = socket_error;
4265 Py_INCREF(socket_error);
4266 PyModule_AddObject(m, "error", socket_error);
4267 socket_herror = PyErr_NewException("socket.herror",
4268 socket_error, NULL);
4269 if (socket_herror == NULL)
4270 return NULL;
4271 Py_INCREF(socket_herror);
4272 PyModule_AddObject(m, "herror", socket_herror);
4273 socket_gaierror = PyErr_NewException("socket.gaierror", socket_error,
4274 NULL);
4275 if (socket_gaierror == NULL)
4276 return NULL;
4277 Py_INCREF(socket_gaierror);
4278 PyModule_AddObject(m, "gaierror", socket_gaierror);
4279 socket_timeout = PyErr_NewException("socket.timeout",
4280 socket_error, NULL);
4281 if (socket_timeout == NULL)
4282 return NULL;
4283 Py_INCREF(socket_timeout);
4284 PyModule_AddObject(m, "timeout", socket_timeout);
4285 Py_INCREF((PyObject *)&sock_type);
4286 if (PyModule_AddObject(m, "SocketType",
4287 (PyObject *)&sock_type) != 0)
4288 return NULL;
4289 Py_INCREF((PyObject *)&sock_type);
4290 if (PyModule_AddObject(m, "socket",
4291 (PyObject *)&sock_type) != 0)
4292 return NULL;
Guido van Rossum09be4091999-08-09 14:40:40 +00004293
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00004294#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004295 has_ipv6 = Py_True;
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00004296#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004297 has_ipv6 = Py_False;
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00004298#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004299 Py_INCREF(has_ipv6);
4300 PyModule_AddObject(m, "has_ipv6", has_ipv6);
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00004301
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004302 /* Export C API */
4303 if (PyModule_AddObject(m, PySocket_CAPI_NAME,
4304 PyCapsule_New(&PySocketModuleAPI, PySocket_CAPSULE_NAME, NULL)
4305 ) != 0)
4306 return NULL;
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00004307
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004308 /* Address families (we only support AF_INET and AF_UNIX) */
Guido van Rossum09be4091999-08-09 14:40:40 +00004309#ifdef AF_UNSPEC
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004310 PyModule_AddIntConstant(m, "AF_UNSPEC", AF_UNSPEC);
Guido van Rossum09be4091999-08-09 14:40:40 +00004311#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004312 PyModule_AddIntConstant(m, "AF_INET", AF_INET);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004313#ifdef AF_INET6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004314 PyModule_AddIntConstant(m, "AF_INET6", AF_INET6);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004315#endif /* AF_INET6 */
Andrew MacIntyred12dfbb2004-04-04 07:13:49 +00004316#if defined(AF_UNIX)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004317 PyModule_AddIntConstant(m, "AF_UNIX", AF_UNIX);
Guido van Rossumb6775db1994-08-01 11:34:53 +00004318#endif /* AF_UNIX */
Guido van Rossum09be4091999-08-09 14:40:40 +00004319#ifdef AF_AX25
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004320 /* Amateur Radio AX.25 */
4321 PyModule_AddIntConstant(m, "AF_AX25", AF_AX25);
Guido van Rossum09be4091999-08-09 14:40:40 +00004322#endif
4323#ifdef AF_IPX
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004324 PyModule_AddIntConstant(m, "AF_IPX", AF_IPX); /* Novell IPX */
Guido van Rossum09be4091999-08-09 14:40:40 +00004325#endif
4326#ifdef AF_APPLETALK
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004327 /* Appletalk DDP */
4328 PyModule_AddIntConstant(m, "AF_APPLETALK", AF_APPLETALK);
Guido van Rossum09be4091999-08-09 14:40:40 +00004329#endif
4330#ifdef AF_NETROM
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004331 /* Amateur radio NetROM */
4332 PyModule_AddIntConstant(m, "AF_NETROM", AF_NETROM);
Guido van Rossum09be4091999-08-09 14:40:40 +00004333#endif
4334#ifdef AF_BRIDGE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004335 /* Multiprotocol bridge */
4336 PyModule_AddIntConstant(m, "AF_BRIDGE", AF_BRIDGE);
Guido van Rossum09be4091999-08-09 14:40:40 +00004337#endif
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004338#ifdef AF_ATMPVC
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004339 /* ATM PVCs */
4340 PyModule_AddIntConstant(m, "AF_ATMPVC", AF_ATMPVC);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004341#endif
Guido van Rossum09be4091999-08-09 14:40:40 +00004342#ifdef AF_AAL5
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004343 /* Reserved for Werner's ATM */
4344 PyModule_AddIntConstant(m, "AF_AAL5", AF_AAL5);
Guido van Rossum09be4091999-08-09 14:40:40 +00004345#endif
4346#ifdef AF_X25
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004347 /* Reserved for X.25 project */
4348 PyModule_AddIntConstant(m, "AF_X25", AF_X25);
Guido van Rossum09be4091999-08-09 14:40:40 +00004349#endif
4350#ifdef AF_INET6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004351 PyModule_AddIntConstant(m, "AF_INET6", AF_INET6); /* IP version 6 */
Guido van Rossum09be4091999-08-09 14:40:40 +00004352#endif
4353#ifdef AF_ROSE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004354 /* Amateur Radio X.25 PLP */
4355 PyModule_AddIntConstant(m, "AF_ROSE", AF_ROSE);
Guido van Rossum09be4091999-08-09 14:40:40 +00004356#endif
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004357#ifdef AF_DECnet
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004358 /* Reserved for DECnet project */
4359 PyModule_AddIntConstant(m, "AF_DECnet", AF_DECnet);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004360#endif
4361#ifdef AF_NETBEUI
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004362 /* Reserved for 802.2LLC project */
4363 PyModule_AddIntConstant(m, "AF_NETBEUI", AF_NETBEUI);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004364#endif
4365#ifdef AF_SECURITY
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004366 /* Security callback pseudo AF */
4367 PyModule_AddIntConstant(m, "AF_SECURITY", AF_SECURITY);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004368#endif
4369#ifdef AF_KEY
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004370 /* PF_KEY key management API */
4371 PyModule_AddIntConstant(m, "AF_KEY", AF_KEY);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004372#endif
4373#ifdef AF_NETLINK
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004374 /* */
4375 PyModule_AddIntConstant(m, "AF_NETLINK", AF_NETLINK);
4376 PyModule_AddIntConstant(m, "NETLINK_ROUTE", NETLINK_ROUTE);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004377#ifdef NETLINK_SKIP
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004378 PyModule_AddIntConstant(m, "NETLINK_SKIP", NETLINK_SKIP);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004379#endif
4380#ifdef NETLINK_W1
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004381 PyModule_AddIntConstant(m, "NETLINK_W1", NETLINK_W1);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004382#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004383 PyModule_AddIntConstant(m, "NETLINK_USERSOCK", NETLINK_USERSOCK);
4384 PyModule_AddIntConstant(m, "NETLINK_FIREWALL", NETLINK_FIREWALL);
Guido van Rossum668a94a2006-02-21 01:07:27 +00004385#ifdef NETLINK_TCPDIAG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004386 PyModule_AddIntConstant(m, "NETLINK_TCPDIAG", NETLINK_TCPDIAG);
Guido van Rossum668a94a2006-02-21 01:07:27 +00004387#endif
4388#ifdef NETLINK_NFLOG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004389 PyModule_AddIntConstant(m, "NETLINK_NFLOG", NETLINK_NFLOG);
Guido van Rossum668a94a2006-02-21 01:07:27 +00004390#endif
Neal Norwitz65851662006-01-16 04:31:40 +00004391#ifdef NETLINK_XFRM
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004392 PyModule_AddIntConstant(m, "NETLINK_XFRM", NETLINK_XFRM);
Neal Norwitz65851662006-01-16 04:31:40 +00004393#endif
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004394#ifdef NETLINK_ARPD
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004395 PyModule_AddIntConstant(m, "NETLINK_ARPD", NETLINK_ARPD);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004396#endif
4397#ifdef NETLINK_ROUTE6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004398 PyModule_AddIntConstant(m, "NETLINK_ROUTE6", NETLINK_ROUTE6);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004399#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004400 PyModule_AddIntConstant(m, "NETLINK_IP6_FW", NETLINK_IP6_FW);
Thomas Wouterscf297e42007-02-23 15:07:44 +00004401#ifdef NETLINK_DNRTMSG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004402 PyModule_AddIntConstant(m, "NETLINK_DNRTMSG", NETLINK_DNRTMSG);
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00004403#endif
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004404#ifdef NETLINK_TAPBASE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004405 PyModule_AddIntConstant(m, "NETLINK_TAPBASE", NETLINK_TAPBASE);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004406#endif
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004407#endif /* AF_NETLINK */
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004408#ifdef AF_ROUTE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004409 /* Alias to emulate 4.4BSD */
4410 PyModule_AddIntConstant(m, "AF_ROUTE", AF_ROUTE);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004411#endif
4412#ifdef AF_ASH
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004413 /* Ash */
4414 PyModule_AddIntConstant(m, "AF_ASH", AF_ASH);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004415#endif
4416#ifdef AF_ECONET
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004417 /* Acorn Econet */
4418 PyModule_AddIntConstant(m, "AF_ECONET", AF_ECONET);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004419#endif
4420#ifdef AF_ATMSVC
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004421 /* ATM SVCs */
4422 PyModule_AddIntConstant(m, "AF_ATMSVC", AF_ATMSVC);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004423#endif
4424#ifdef AF_SNA
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004425 /* Linux SNA Project (nutters!) */
4426 PyModule_AddIntConstant(m, "AF_SNA", AF_SNA);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004427#endif
4428#ifdef AF_IRDA
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004429 /* IRDA sockets */
4430 PyModule_AddIntConstant(m, "AF_IRDA", AF_IRDA);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004431#endif
4432#ifdef AF_PPPOX
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004433 /* PPPoX sockets */
4434 PyModule_AddIntConstant(m, "AF_PPPOX", AF_PPPOX);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004435#endif
4436#ifdef AF_WANPIPE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004437 /* Wanpipe API Sockets */
4438 PyModule_AddIntConstant(m, "AF_WANPIPE", AF_WANPIPE);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004439#endif
4440#ifdef AF_LLC
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004441 /* Linux LLC */
4442 PyModule_AddIntConstant(m, "AF_LLC", AF_LLC);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004443#endif
Martin v. Löwis12af0482004-01-31 12:34:17 +00004444
Hye-Shik Chang81268602004-02-02 06:05:24 +00004445#ifdef USE_BLUETOOTH
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004446 PyModule_AddIntConstant(m, "AF_BLUETOOTH", AF_BLUETOOTH);
4447 PyModule_AddIntConstant(m, "BTPROTO_L2CAP", BTPROTO_L2CAP);
4448 PyModule_AddIntConstant(m, "BTPROTO_HCI", BTPROTO_HCI);
4449 PyModule_AddIntConstant(m, "SOL_HCI", SOL_HCI);
4450 PyModule_AddIntConstant(m, "HCI_FILTER", HCI_FILTER);
Hye-Shik Chang81268602004-02-02 06:05:24 +00004451#if !defined(__FreeBSD__)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004452 PyModule_AddIntConstant(m, "HCI_TIME_STAMP", HCI_TIME_STAMP);
4453 PyModule_AddIntConstant(m, "HCI_DATA_DIR", HCI_DATA_DIR);
4454 PyModule_AddIntConstant(m, "BTPROTO_SCO", BTPROTO_SCO);
Hye-Shik Chang81268602004-02-02 06:05:24 +00004455#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004456 PyModule_AddIntConstant(m, "BTPROTO_RFCOMM", BTPROTO_RFCOMM);
4457 PyModule_AddStringConstant(m, "BDADDR_ANY", "00:00:00:00:00:00");
4458 PyModule_AddStringConstant(m, "BDADDR_LOCAL", "00:00:00:FF:FF:FF");
Martin v. Löwis12af0482004-01-31 12:34:17 +00004459#endif
4460
Martin v. Löwis1ba3fd52001-08-10 20:29:40 +00004461#ifdef HAVE_NETPACKET_PACKET_H
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004462 PyModule_AddIntConstant(m, "AF_PACKET", AF_PACKET);
4463 PyModule_AddIntConstant(m, "PF_PACKET", PF_PACKET);
4464 PyModule_AddIntConstant(m, "PACKET_HOST", PACKET_HOST);
4465 PyModule_AddIntConstant(m, "PACKET_BROADCAST", PACKET_BROADCAST);
4466 PyModule_AddIntConstant(m, "PACKET_MULTICAST", PACKET_MULTICAST);
4467 PyModule_AddIntConstant(m, "PACKET_OTHERHOST", PACKET_OTHERHOST);
4468 PyModule_AddIntConstant(m, "PACKET_OUTGOING", PACKET_OUTGOING);
4469 PyModule_AddIntConstant(m, "PACKET_LOOPBACK", PACKET_LOOPBACK);
4470 PyModule_AddIntConstant(m, "PACKET_FASTROUTE", PACKET_FASTROUTE);
Guido van Rossum48a680c2001-03-02 06:34:14 +00004471#endif
Guido van Rossum09be4091999-08-09 14:40:40 +00004472
Christian Heimes043d6f62008-01-07 17:19:16 +00004473#ifdef HAVE_LINUX_TIPC_H
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004474 PyModule_AddIntConstant(m, "AF_TIPC", AF_TIPC);
Christian Heimes043d6f62008-01-07 17:19:16 +00004475
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004476 /* for addresses */
4477 PyModule_AddIntConstant(m, "TIPC_ADDR_NAMESEQ", TIPC_ADDR_NAMESEQ);
4478 PyModule_AddIntConstant(m, "TIPC_ADDR_NAME", TIPC_ADDR_NAME);
4479 PyModule_AddIntConstant(m, "TIPC_ADDR_ID", TIPC_ADDR_ID);
Christian Heimes043d6f62008-01-07 17:19:16 +00004480
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004481 PyModule_AddIntConstant(m, "TIPC_ZONE_SCOPE", TIPC_ZONE_SCOPE);
4482 PyModule_AddIntConstant(m, "TIPC_CLUSTER_SCOPE", TIPC_CLUSTER_SCOPE);
4483 PyModule_AddIntConstant(m, "TIPC_NODE_SCOPE", TIPC_NODE_SCOPE);
Christian Heimes043d6f62008-01-07 17:19:16 +00004484
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004485 /* for setsockopt() */
4486 PyModule_AddIntConstant(m, "SOL_TIPC", SOL_TIPC);
4487 PyModule_AddIntConstant(m, "TIPC_IMPORTANCE", TIPC_IMPORTANCE);
4488 PyModule_AddIntConstant(m, "TIPC_SRC_DROPPABLE", TIPC_SRC_DROPPABLE);
4489 PyModule_AddIntConstant(m, "TIPC_DEST_DROPPABLE",
4490 TIPC_DEST_DROPPABLE);
4491 PyModule_AddIntConstant(m, "TIPC_CONN_TIMEOUT", TIPC_CONN_TIMEOUT);
Christian Heimes043d6f62008-01-07 17:19:16 +00004492
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004493 PyModule_AddIntConstant(m, "TIPC_LOW_IMPORTANCE",
4494 TIPC_LOW_IMPORTANCE);
4495 PyModule_AddIntConstant(m, "TIPC_MEDIUM_IMPORTANCE",
4496 TIPC_MEDIUM_IMPORTANCE);
4497 PyModule_AddIntConstant(m, "TIPC_HIGH_IMPORTANCE",
4498 TIPC_HIGH_IMPORTANCE);
4499 PyModule_AddIntConstant(m, "TIPC_CRITICAL_IMPORTANCE",
4500 TIPC_CRITICAL_IMPORTANCE);
Christian Heimes043d6f62008-01-07 17:19:16 +00004501
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004502 /* for subscriptions */
4503 PyModule_AddIntConstant(m, "TIPC_SUB_PORTS", TIPC_SUB_PORTS);
4504 PyModule_AddIntConstant(m, "TIPC_SUB_SERVICE", TIPC_SUB_SERVICE);
Christian Heimes25bb7832008-01-11 16:17:00 +00004505#ifdef TIPC_SUB_CANCEL
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004506 /* doesn't seem to be available everywhere */
4507 PyModule_AddIntConstant(m, "TIPC_SUB_CANCEL", TIPC_SUB_CANCEL);
Christian Heimes25bb7832008-01-11 16:17:00 +00004508#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004509 PyModule_AddIntConstant(m, "TIPC_WAIT_FOREVER", TIPC_WAIT_FOREVER);
4510 PyModule_AddIntConstant(m, "TIPC_PUBLISHED", TIPC_PUBLISHED);
4511 PyModule_AddIntConstant(m, "TIPC_WITHDRAWN", TIPC_WITHDRAWN);
4512 PyModule_AddIntConstant(m, "TIPC_SUBSCR_TIMEOUT", TIPC_SUBSCR_TIMEOUT);
4513 PyModule_AddIntConstant(m, "TIPC_CFG_SRV", TIPC_CFG_SRV);
4514 PyModule_AddIntConstant(m, "TIPC_TOP_SRV", TIPC_TOP_SRV);
Christian Heimes043d6f62008-01-07 17:19:16 +00004515#endif
4516
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004517 /* Socket types */
4518 PyModule_AddIntConstant(m, "SOCK_STREAM", SOCK_STREAM);
4519 PyModule_AddIntConstant(m, "SOCK_DGRAM", SOCK_DGRAM);
Guido van Rossumbcc20741998-08-04 22:53:56 +00004520/* We have incomplete socket support. */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004521 PyModule_AddIntConstant(m, "SOCK_RAW", SOCK_RAW);
4522 PyModule_AddIntConstant(m, "SOCK_SEQPACKET", SOCK_SEQPACKET);
Martin v. Löwiscf8f47e2002-12-11 13:10:57 +00004523#if defined(SOCK_RDM)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004524 PyModule_AddIntConstant(m, "SOCK_RDM", SOCK_RDM);
Guido van Rossumbcc20741998-08-04 22:53:56 +00004525#endif
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004526
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004527#ifdef SO_DEBUG
4528 PyModule_AddIntConstant(m, "SO_DEBUG", SO_DEBUG);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004529#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004530#ifdef SO_ACCEPTCONN
4531 PyModule_AddIntConstant(m, "SO_ACCEPTCONN", SO_ACCEPTCONN);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004532#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004533#ifdef SO_REUSEADDR
4534 PyModule_AddIntConstant(m, "SO_REUSEADDR", SO_REUSEADDR);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004535#endif
Andrew M. Kuchling42851ab2004-07-10 14:19:21 +00004536#ifdef SO_EXCLUSIVEADDRUSE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004537 PyModule_AddIntConstant(m, "SO_EXCLUSIVEADDRUSE", SO_EXCLUSIVEADDRUSE);
Andrew M. Kuchling42851ab2004-07-10 14:19:21 +00004538#endif
4539
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004540#ifdef SO_KEEPALIVE
4541 PyModule_AddIntConstant(m, "SO_KEEPALIVE", SO_KEEPALIVE);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004542#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004543#ifdef SO_DONTROUTE
4544 PyModule_AddIntConstant(m, "SO_DONTROUTE", SO_DONTROUTE);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004545#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004546#ifdef SO_BROADCAST
4547 PyModule_AddIntConstant(m, "SO_BROADCAST", SO_BROADCAST);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004548#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004549#ifdef SO_USELOOPBACK
4550 PyModule_AddIntConstant(m, "SO_USELOOPBACK", SO_USELOOPBACK);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004551#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004552#ifdef SO_LINGER
4553 PyModule_AddIntConstant(m, "SO_LINGER", SO_LINGER);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004554#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004555#ifdef SO_OOBINLINE
4556 PyModule_AddIntConstant(m, "SO_OOBINLINE", SO_OOBINLINE);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004557#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004558#ifdef SO_REUSEPORT
4559 PyModule_AddIntConstant(m, "SO_REUSEPORT", SO_REUSEPORT);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004560#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004561#ifdef SO_SNDBUF
4562 PyModule_AddIntConstant(m, "SO_SNDBUF", SO_SNDBUF);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004563#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004564#ifdef SO_RCVBUF
4565 PyModule_AddIntConstant(m, "SO_RCVBUF", SO_RCVBUF);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004566#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004567#ifdef SO_SNDLOWAT
4568 PyModule_AddIntConstant(m, "SO_SNDLOWAT", SO_SNDLOWAT);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004569#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004570#ifdef SO_RCVLOWAT
4571 PyModule_AddIntConstant(m, "SO_RCVLOWAT", SO_RCVLOWAT);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004572#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004573#ifdef SO_SNDTIMEO
4574 PyModule_AddIntConstant(m, "SO_SNDTIMEO", SO_SNDTIMEO);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004575#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004576#ifdef SO_RCVTIMEO
4577 PyModule_AddIntConstant(m, "SO_RCVTIMEO", SO_RCVTIMEO);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004578#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004579#ifdef SO_ERROR
4580 PyModule_AddIntConstant(m, "SO_ERROR", SO_ERROR);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004581#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004582#ifdef SO_TYPE
4583 PyModule_AddIntConstant(m, "SO_TYPE", SO_TYPE);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004584#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004585#ifdef SO_SETFIB
4586 PyModule_AddIntConstant(m, "SO_SETFIB", SO_SETFIB);
Larry Hastingsf0f37952010-04-02 11:47:10 +00004587#endif
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004588
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004589 /* Maximum number of connections for "listen" */
4590#ifdef SOMAXCONN
4591 PyModule_AddIntConstant(m, "SOMAXCONN", SOMAXCONN);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004592#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004593 PyModule_AddIntConstant(m, "SOMAXCONN", 5); /* Common value */
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004594#endif
4595
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004596 /* Flags for send, recv */
4597#ifdef MSG_OOB
4598 PyModule_AddIntConstant(m, "MSG_OOB", MSG_OOB);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004599#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004600#ifdef MSG_PEEK
4601 PyModule_AddIntConstant(m, "MSG_PEEK", MSG_PEEK);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004602#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004603#ifdef MSG_DONTROUTE
4604 PyModule_AddIntConstant(m, "MSG_DONTROUTE", MSG_DONTROUTE);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004605#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004606#ifdef MSG_DONTWAIT
4607 PyModule_AddIntConstant(m, "MSG_DONTWAIT", MSG_DONTWAIT);
Guido van Rossum2c8bcb82000-04-25 21:34:53 +00004608#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004609#ifdef MSG_EOR
4610 PyModule_AddIntConstant(m, "MSG_EOR", MSG_EOR);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004611#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004612#ifdef MSG_TRUNC
4613 PyModule_AddIntConstant(m, "MSG_TRUNC", MSG_TRUNC);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004614#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004615#ifdef MSG_CTRUNC
4616 PyModule_AddIntConstant(m, "MSG_CTRUNC", MSG_CTRUNC);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004617#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004618#ifdef MSG_WAITALL
4619 PyModule_AddIntConstant(m, "MSG_WAITALL", MSG_WAITALL);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004620#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004621#ifdef MSG_BTAG
4622 PyModule_AddIntConstant(m, "MSG_BTAG", MSG_BTAG);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004623#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004624#ifdef MSG_ETAG
4625 PyModule_AddIntConstant(m, "MSG_ETAG", MSG_ETAG);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004626#endif
4627
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004628 /* Protocol level and numbers, usable for [gs]etsockopt */
4629#ifdef SOL_SOCKET
4630 PyModule_AddIntConstant(m, "SOL_SOCKET", SOL_SOCKET);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004631#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004632#ifdef SOL_IP
4633 PyModule_AddIntConstant(m, "SOL_IP", SOL_IP);
Guido van Rossum09be4091999-08-09 14:40:40 +00004634#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004635 PyModule_AddIntConstant(m, "SOL_IP", 0);
Guido van Rossum09be4091999-08-09 14:40:40 +00004636#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004637#ifdef SOL_IPX
4638 PyModule_AddIntConstant(m, "SOL_IPX", SOL_IPX);
Guido van Rossum09be4091999-08-09 14:40:40 +00004639#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004640#ifdef SOL_AX25
4641 PyModule_AddIntConstant(m, "SOL_AX25", SOL_AX25);
Guido van Rossum09be4091999-08-09 14:40:40 +00004642#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004643#ifdef SOL_ATALK
4644 PyModule_AddIntConstant(m, "SOL_ATALK", SOL_ATALK);
Guido van Rossum09be4091999-08-09 14:40:40 +00004645#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004646#ifdef SOL_NETROM
4647 PyModule_AddIntConstant(m, "SOL_NETROM", SOL_NETROM);
Guido van Rossum09be4091999-08-09 14:40:40 +00004648#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004649#ifdef SOL_ROSE
4650 PyModule_AddIntConstant(m, "SOL_ROSE", SOL_ROSE);
Guido van Rossum09be4091999-08-09 14:40:40 +00004651#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004652#ifdef SOL_TCP
4653 PyModule_AddIntConstant(m, "SOL_TCP", SOL_TCP);
Guido van Rossum09be4091999-08-09 14:40:40 +00004654#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004655 PyModule_AddIntConstant(m, "SOL_TCP", 6);
Guido van Rossum09be4091999-08-09 14:40:40 +00004656#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004657#ifdef SOL_UDP
4658 PyModule_AddIntConstant(m, "SOL_UDP", SOL_UDP);
Guido van Rossum09be4091999-08-09 14:40:40 +00004659#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004660 PyModule_AddIntConstant(m, "SOL_UDP", 17);
Guido van Rossum09be4091999-08-09 14:40:40 +00004661#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004662#ifdef IPPROTO_IP
4663 PyModule_AddIntConstant(m, "IPPROTO_IP", IPPROTO_IP);
Guido van Rossum578de301998-05-28 20:18:18 +00004664#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004665 PyModule_AddIntConstant(m, "IPPROTO_IP", 0);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004666#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004667#ifdef IPPROTO_HOPOPTS
4668 PyModule_AddIntConstant(m, "IPPROTO_HOPOPTS", IPPROTO_HOPOPTS);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004669#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004670#ifdef IPPROTO_ICMP
4671 PyModule_AddIntConstant(m, "IPPROTO_ICMP", IPPROTO_ICMP);
Guido van Rossum578de301998-05-28 20:18:18 +00004672#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004673 PyModule_AddIntConstant(m, "IPPROTO_ICMP", 1);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004674#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004675#ifdef IPPROTO_IGMP
4676 PyModule_AddIntConstant(m, "IPPROTO_IGMP", IPPROTO_IGMP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004677#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004678#ifdef IPPROTO_GGP
4679 PyModule_AddIntConstant(m, "IPPROTO_GGP", IPPROTO_GGP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004680#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004681#ifdef IPPROTO_IPV4
4682 PyModule_AddIntConstant(m, "IPPROTO_IPV4", IPPROTO_IPV4);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004683#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004684#ifdef IPPROTO_IPV6
4685 PyModule_AddIntConstant(m, "IPPROTO_IPV6", IPPROTO_IPV6);
Martin v. Löwisa0f17342003-10-03 13:56:20 +00004686#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004687#ifdef IPPROTO_IPIP
4688 PyModule_AddIntConstant(m, "IPPROTO_IPIP", IPPROTO_IPIP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004689#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004690#ifdef IPPROTO_TCP
4691 PyModule_AddIntConstant(m, "IPPROTO_TCP", IPPROTO_TCP);
Guido van Rossum578de301998-05-28 20:18:18 +00004692#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004693 PyModule_AddIntConstant(m, "IPPROTO_TCP", 6);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004694#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004695#ifdef IPPROTO_EGP
4696 PyModule_AddIntConstant(m, "IPPROTO_EGP", IPPROTO_EGP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004697#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004698#ifdef IPPROTO_PUP
4699 PyModule_AddIntConstant(m, "IPPROTO_PUP", IPPROTO_PUP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004700#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004701#ifdef IPPROTO_UDP
4702 PyModule_AddIntConstant(m, "IPPROTO_UDP", IPPROTO_UDP);
Guido van Rossum578de301998-05-28 20:18:18 +00004703#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004704 PyModule_AddIntConstant(m, "IPPROTO_UDP", 17);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004705#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004706#ifdef IPPROTO_IDP
4707 PyModule_AddIntConstant(m, "IPPROTO_IDP", IPPROTO_IDP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004708#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004709#ifdef IPPROTO_HELLO
4710 PyModule_AddIntConstant(m, "IPPROTO_HELLO", IPPROTO_HELLO);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004711#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004712#ifdef IPPROTO_ND
4713 PyModule_AddIntConstant(m, "IPPROTO_ND", IPPROTO_ND);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004714#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004715#ifdef IPPROTO_TP
4716 PyModule_AddIntConstant(m, "IPPROTO_TP", IPPROTO_TP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004717#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004718#ifdef IPPROTO_IPV6
4719 PyModule_AddIntConstant(m, "IPPROTO_IPV6", IPPROTO_IPV6);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004720#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004721#ifdef IPPROTO_ROUTING
4722 PyModule_AddIntConstant(m, "IPPROTO_ROUTING", IPPROTO_ROUTING);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004723#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004724#ifdef IPPROTO_FRAGMENT
4725 PyModule_AddIntConstant(m, "IPPROTO_FRAGMENT", IPPROTO_FRAGMENT);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004726#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004727#ifdef IPPROTO_RSVP
4728 PyModule_AddIntConstant(m, "IPPROTO_RSVP", IPPROTO_RSVP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004729#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004730#ifdef IPPROTO_GRE
4731 PyModule_AddIntConstant(m, "IPPROTO_GRE", IPPROTO_GRE);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004732#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004733#ifdef IPPROTO_ESP
4734 PyModule_AddIntConstant(m, "IPPROTO_ESP", IPPROTO_ESP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004735#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004736#ifdef IPPROTO_AH
4737 PyModule_AddIntConstant(m, "IPPROTO_AH", IPPROTO_AH);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004738#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004739#ifdef IPPROTO_MOBILE
4740 PyModule_AddIntConstant(m, "IPPROTO_MOBILE", IPPROTO_MOBILE);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004741#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004742#ifdef IPPROTO_ICMPV6
4743 PyModule_AddIntConstant(m, "IPPROTO_ICMPV6", IPPROTO_ICMPV6);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004744#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004745#ifdef IPPROTO_NONE
4746 PyModule_AddIntConstant(m, "IPPROTO_NONE", IPPROTO_NONE);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004747#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004748#ifdef IPPROTO_DSTOPTS
4749 PyModule_AddIntConstant(m, "IPPROTO_DSTOPTS", IPPROTO_DSTOPTS);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004750#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004751#ifdef IPPROTO_XTP
4752 PyModule_AddIntConstant(m, "IPPROTO_XTP", IPPROTO_XTP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004753#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004754#ifdef IPPROTO_EON
4755 PyModule_AddIntConstant(m, "IPPROTO_EON", IPPROTO_EON);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004756#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004757#ifdef IPPROTO_PIM
4758 PyModule_AddIntConstant(m, "IPPROTO_PIM", IPPROTO_PIM);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004759#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004760#ifdef IPPROTO_IPCOMP
4761 PyModule_AddIntConstant(m, "IPPROTO_IPCOMP", IPPROTO_IPCOMP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004762#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004763#ifdef IPPROTO_VRRP
4764 PyModule_AddIntConstant(m, "IPPROTO_VRRP", IPPROTO_VRRP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004765#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004766#ifdef IPPROTO_BIP
4767 PyModule_AddIntConstant(m, "IPPROTO_BIP", IPPROTO_BIP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004768#endif
4769/**/
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004770#ifdef IPPROTO_RAW
4771 PyModule_AddIntConstant(m, "IPPROTO_RAW", IPPROTO_RAW);
Guido van Rossum578de301998-05-28 20:18:18 +00004772#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004773 PyModule_AddIntConstant(m, "IPPROTO_RAW", 255);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004774#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004775#ifdef IPPROTO_MAX
4776 PyModule_AddIntConstant(m, "IPPROTO_MAX", IPPROTO_MAX);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004777#endif
4778
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004779 /* Some port configuration */
4780#ifdef IPPORT_RESERVED
4781 PyModule_AddIntConstant(m, "IPPORT_RESERVED", IPPORT_RESERVED);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004782#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004783 PyModule_AddIntConstant(m, "IPPORT_RESERVED", 1024);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004784#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004785#ifdef IPPORT_USERRESERVED
4786 PyModule_AddIntConstant(m, "IPPORT_USERRESERVED", IPPORT_USERRESERVED);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004787#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004788 PyModule_AddIntConstant(m, "IPPORT_USERRESERVED", 5000);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004789#endif
4790
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004791 /* Some reserved IP v.4 addresses */
4792#ifdef INADDR_ANY
4793 PyModule_AddIntConstant(m, "INADDR_ANY", INADDR_ANY);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004794#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004795 PyModule_AddIntConstant(m, "INADDR_ANY", 0x00000000);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004796#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004797#ifdef INADDR_BROADCAST
4798 PyModule_AddIntConstant(m, "INADDR_BROADCAST", INADDR_BROADCAST);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004799#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004800 PyModule_AddIntConstant(m, "INADDR_BROADCAST", 0xffffffff);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004801#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004802#ifdef INADDR_LOOPBACK
4803 PyModule_AddIntConstant(m, "INADDR_LOOPBACK", INADDR_LOOPBACK);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004804#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004805 PyModule_AddIntConstant(m, "INADDR_LOOPBACK", 0x7F000001);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004806#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004807#ifdef INADDR_UNSPEC_GROUP
4808 PyModule_AddIntConstant(m, "INADDR_UNSPEC_GROUP", INADDR_UNSPEC_GROUP);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004809#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004810 PyModule_AddIntConstant(m, "INADDR_UNSPEC_GROUP", 0xe0000000);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004811#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004812#ifdef INADDR_ALLHOSTS_GROUP
4813 PyModule_AddIntConstant(m, "INADDR_ALLHOSTS_GROUP",
4814 INADDR_ALLHOSTS_GROUP);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004815#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004816 PyModule_AddIntConstant(m, "INADDR_ALLHOSTS_GROUP", 0xe0000001);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004817#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004818#ifdef INADDR_MAX_LOCAL_GROUP
4819 PyModule_AddIntConstant(m, "INADDR_MAX_LOCAL_GROUP",
4820 INADDR_MAX_LOCAL_GROUP);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004821#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004822 PyModule_AddIntConstant(m, "INADDR_MAX_LOCAL_GROUP", 0xe00000ff);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004823#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004824#ifdef INADDR_NONE
4825 PyModule_AddIntConstant(m, "INADDR_NONE", INADDR_NONE);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004826#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004827 PyModule_AddIntConstant(m, "INADDR_NONE", 0xffffffff);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004828#endif
4829
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004830 /* IPv4 [gs]etsockopt options */
4831#ifdef IP_OPTIONS
4832 PyModule_AddIntConstant(m, "IP_OPTIONS", IP_OPTIONS);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004833#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004834#ifdef IP_HDRINCL
4835 PyModule_AddIntConstant(m, "IP_HDRINCL", IP_HDRINCL);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004836#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004837#ifdef IP_TOS
4838 PyModule_AddIntConstant(m, "IP_TOS", IP_TOS);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004839#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004840#ifdef IP_TTL
4841 PyModule_AddIntConstant(m, "IP_TTL", IP_TTL);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004842#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004843#ifdef IP_RECVOPTS
4844 PyModule_AddIntConstant(m, "IP_RECVOPTS", IP_RECVOPTS);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004845#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004846#ifdef IP_RECVRETOPTS
4847 PyModule_AddIntConstant(m, "IP_RECVRETOPTS", IP_RECVRETOPTS);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004848#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004849#ifdef IP_RECVDSTADDR
4850 PyModule_AddIntConstant(m, "IP_RECVDSTADDR", IP_RECVDSTADDR);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004851#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004852#ifdef IP_RETOPTS
4853 PyModule_AddIntConstant(m, "IP_RETOPTS", IP_RETOPTS);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004854#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004855#ifdef IP_MULTICAST_IF
4856 PyModule_AddIntConstant(m, "IP_MULTICAST_IF", IP_MULTICAST_IF);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004857#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004858#ifdef IP_MULTICAST_TTL
4859 PyModule_AddIntConstant(m, "IP_MULTICAST_TTL", IP_MULTICAST_TTL);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004860#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004861#ifdef IP_MULTICAST_LOOP
4862 PyModule_AddIntConstant(m, "IP_MULTICAST_LOOP", IP_MULTICAST_LOOP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004863#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004864#ifdef IP_ADD_MEMBERSHIP
4865 PyModule_AddIntConstant(m, "IP_ADD_MEMBERSHIP", IP_ADD_MEMBERSHIP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004866#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004867#ifdef IP_DROP_MEMBERSHIP
4868 PyModule_AddIntConstant(m, "IP_DROP_MEMBERSHIP", IP_DROP_MEMBERSHIP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004869#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004870#ifdef IP_DEFAULT_MULTICAST_TTL
4871 PyModule_AddIntConstant(m, "IP_DEFAULT_MULTICAST_TTL",
4872 IP_DEFAULT_MULTICAST_TTL);
Guido van Rossum09be4091999-08-09 14:40:40 +00004873#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004874#ifdef IP_DEFAULT_MULTICAST_LOOP
4875 PyModule_AddIntConstant(m, "IP_DEFAULT_MULTICAST_LOOP",
4876 IP_DEFAULT_MULTICAST_LOOP);
Guido van Rossum09be4091999-08-09 14:40:40 +00004877#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004878#ifdef IP_MAX_MEMBERSHIPS
4879 PyModule_AddIntConstant(m, "IP_MAX_MEMBERSHIPS", IP_MAX_MEMBERSHIPS);
Guido van Rossum09be4091999-08-09 14:40:40 +00004880#endif
4881
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004882 /* IPv6 [gs]etsockopt options, defined in RFC2553 */
4883#ifdef IPV6_JOIN_GROUP
4884 PyModule_AddIntConstant(m, "IPV6_JOIN_GROUP", IPV6_JOIN_GROUP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004885#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004886#ifdef IPV6_LEAVE_GROUP
4887 PyModule_AddIntConstant(m, "IPV6_LEAVE_GROUP", IPV6_LEAVE_GROUP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004888#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004889#ifdef IPV6_MULTICAST_HOPS
4890 PyModule_AddIntConstant(m, "IPV6_MULTICAST_HOPS", IPV6_MULTICAST_HOPS);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004891#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004892#ifdef IPV6_MULTICAST_IF
4893 PyModule_AddIntConstant(m, "IPV6_MULTICAST_IF", IPV6_MULTICAST_IF);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004894#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004895#ifdef IPV6_MULTICAST_LOOP
4896 PyModule_AddIntConstant(m, "IPV6_MULTICAST_LOOP", IPV6_MULTICAST_LOOP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004897#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004898#ifdef IPV6_UNICAST_HOPS
4899 PyModule_AddIntConstant(m, "IPV6_UNICAST_HOPS", IPV6_UNICAST_HOPS);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004900#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004901 /* Additional IPV6 socket options, defined in RFC 3493 */
Martin v. Löwisda91d022003-12-30 11:14:01 +00004902#ifdef IPV6_V6ONLY
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004903 PyModule_AddIntConstant(m, "IPV6_V6ONLY", IPV6_V6ONLY);
Martin v. Löwisda91d022003-12-30 11:14:01 +00004904#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004905 /* Advanced IPV6 socket options, from RFC 3542 */
Martin v. Löwisda91d022003-12-30 11:14:01 +00004906#ifdef IPV6_CHECKSUM
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004907 PyModule_AddIntConstant(m, "IPV6_CHECKSUM", IPV6_CHECKSUM);
Martin v. Löwisda91d022003-12-30 11:14:01 +00004908#endif
4909#ifdef IPV6_DONTFRAG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004910 PyModule_AddIntConstant(m, "IPV6_DONTFRAG", IPV6_DONTFRAG);
Martin v. Löwisda91d022003-12-30 11:14:01 +00004911#endif
4912#ifdef IPV6_DSTOPTS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004913 PyModule_AddIntConstant(m, "IPV6_DSTOPTS", IPV6_DSTOPTS);
Martin v. Löwisda91d022003-12-30 11:14:01 +00004914#endif
4915#ifdef IPV6_HOPLIMIT
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004916 PyModule_AddIntConstant(m, "IPV6_HOPLIMIT", IPV6_HOPLIMIT);
Martin v. Löwisda91d022003-12-30 11:14:01 +00004917#endif
4918#ifdef IPV6_HOPOPTS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004919 PyModule_AddIntConstant(m, "IPV6_HOPOPTS", IPV6_HOPOPTS);
Martin v. Löwisda91d022003-12-30 11:14:01 +00004920#endif
4921#ifdef IPV6_NEXTHOP
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004922 PyModule_AddIntConstant(m, "IPV6_NEXTHOP", IPV6_NEXTHOP);
Martin v. Löwisda91d022003-12-30 11:14:01 +00004923#endif
4924#ifdef IPV6_PATHMTU
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004925 PyModule_AddIntConstant(m, "IPV6_PATHMTU", IPV6_PATHMTU);
Martin v. Löwisda91d022003-12-30 11:14:01 +00004926#endif
4927#ifdef IPV6_PKTINFO
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004928 PyModule_AddIntConstant(m, "IPV6_PKTINFO", IPV6_PKTINFO);
Martin v. Löwisda91d022003-12-30 11:14:01 +00004929#endif
4930#ifdef IPV6_RECVDSTOPTS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004931 PyModule_AddIntConstant(m, "IPV6_RECVDSTOPTS", IPV6_RECVDSTOPTS);
Martin v. Löwisda91d022003-12-30 11:14:01 +00004932#endif
4933#ifdef IPV6_RECVHOPLIMIT
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004934 PyModule_AddIntConstant(m, "IPV6_RECVHOPLIMIT", IPV6_RECVHOPLIMIT);
Martin v. Löwisda91d022003-12-30 11:14:01 +00004935#endif
4936#ifdef IPV6_RECVHOPOPTS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004937 PyModule_AddIntConstant(m, "IPV6_RECVHOPOPTS", IPV6_RECVHOPOPTS);
Martin v. Löwisda91d022003-12-30 11:14:01 +00004938#endif
4939#ifdef IPV6_RECVPKTINFO
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004940 PyModule_AddIntConstant(m, "IPV6_RECVPKTINFO", IPV6_RECVPKTINFO);
Martin v. Löwisda91d022003-12-30 11:14:01 +00004941#endif
4942#ifdef IPV6_RECVRTHDR
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004943 PyModule_AddIntConstant(m, "IPV6_RECVRTHDR", IPV6_RECVRTHDR);
Martin v. Löwisda91d022003-12-30 11:14:01 +00004944#endif
4945#ifdef IPV6_RECVTCLASS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004946 PyModule_AddIntConstant(m, "IPV6_RECVTCLASS", IPV6_RECVTCLASS);
Martin v. Löwisda91d022003-12-30 11:14:01 +00004947#endif
4948#ifdef IPV6_RTHDR
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004949 PyModule_AddIntConstant(m, "IPV6_RTHDR", IPV6_RTHDR);
Martin v. Löwisda91d022003-12-30 11:14:01 +00004950#endif
4951#ifdef IPV6_RTHDRDSTOPTS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004952 PyModule_AddIntConstant(m, "IPV6_RTHDRDSTOPTS", IPV6_RTHDRDSTOPTS);
Martin v. Löwisda91d022003-12-30 11:14:01 +00004953#endif
4954#ifdef IPV6_RTHDR_TYPE_0
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004955 PyModule_AddIntConstant(m, "IPV6_RTHDR_TYPE_0", IPV6_RTHDR_TYPE_0);
Martin v. Löwisda91d022003-12-30 11:14:01 +00004956#endif
4957#ifdef IPV6_RECVPATHMTU
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004958 PyModule_AddIntConstant(m, "IPV6_RECVPATHMTU", IPV6_RECVPATHMTU);
Martin v. Löwisda91d022003-12-30 11:14:01 +00004959#endif
4960#ifdef IPV6_TCLASS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004961 PyModule_AddIntConstant(m, "IPV6_TCLASS", IPV6_TCLASS);
Martin v. Löwisda91d022003-12-30 11:14:01 +00004962#endif
4963#ifdef IPV6_USE_MIN_MTU
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004964 PyModule_AddIntConstant(m, "IPV6_USE_MIN_MTU", IPV6_USE_MIN_MTU);
Martin v. Löwisda91d022003-12-30 11:14:01 +00004965#endif
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004966
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004967 /* TCP options */
4968#ifdef TCP_NODELAY
4969 PyModule_AddIntConstant(m, "TCP_NODELAY", TCP_NODELAY);
Guido van Rossum09be4091999-08-09 14:40:40 +00004970#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004971#ifdef TCP_MAXSEG
4972 PyModule_AddIntConstant(m, "TCP_MAXSEG", TCP_MAXSEG);
Guido van Rossum09be4091999-08-09 14:40:40 +00004973#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004974#ifdef TCP_CORK
4975 PyModule_AddIntConstant(m, "TCP_CORK", TCP_CORK);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00004976#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004977#ifdef TCP_KEEPIDLE
4978 PyModule_AddIntConstant(m, "TCP_KEEPIDLE", TCP_KEEPIDLE);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00004979#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004980#ifdef TCP_KEEPINTVL
4981 PyModule_AddIntConstant(m, "TCP_KEEPINTVL", TCP_KEEPINTVL);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00004982#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004983#ifdef TCP_KEEPCNT
4984 PyModule_AddIntConstant(m, "TCP_KEEPCNT", TCP_KEEPCNT);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00004985#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004986#ifdef TCP_SYNCNT
4987 PyModule_AddIntConstant(m, "TCP_SYNCNT", TCP_SYNCNT);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00004988#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004989#ifdef TCP_LINGER2
4990 PyModule_AddIntConstant(m, "TCP_LINGER2", TCP_LINGER2);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00004991#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004992#ifdef TCP_DEFER_ACCEPT
4993 PyModule_AddIntConstant(m, "TCP_DEFER_ACCEPT", TCP_DEFER_ACCEPT);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00004994#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004995#ifdef TCP_WINDOW_CLAMP
4996 PyModule_AddIntConstant(m, "TCP_WINDOW_CLAMP", TCP_WINDOW_CLAMP);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00004997#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004998#ifdef TCP_INFO
4999 PyModule_AddIntConstant(m, "TCP_INFO", TCP_INFO);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00005000#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005001#ifdef TCP_QUICKACK
5002 PyModule_AddIntConstant(m, "TCP_QUICKACK", TCP_QUICKACK);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00005003#endif
5004
Guido van Rossum09be4091999-08-09 14:40:40 +00005005
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005006 /* IPX options */
5007#ifdef IPX_TYPE
5008 PyModule_AddIntConstant(m, "IPX_TYPE", IPX_TYPE);
Guido van Rossum09be4091999-08-09 14:40:40 +00005009#endif
Guido van Rossum4f199ea1998-04-09 20:56:35 +00005010
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005011 /* get{addr,name}info parameters */
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005012#ifdef EAI_ADDRFAMILY
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005013 PyModule_AddIntConstant(m, "EAI_ADDRFAMILY", EAI_ADDRFAMILY);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005014#endif
5015#ifdef EAI_AGAIN
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005016 PyModule_AddIntConstant(m, "EAI_AGAIN", EAI_AGAIN);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005017#endif
5018#ifdef EAI_BADFLAGS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005019 PyModule_AddIntConstant(m, "EAI_BADFLAGS", EAI_BADFLAGS);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005020#endif
5021#ifdef EAI_FAIL
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005022 PyModule_AddIntConstant(m, "EAI_FAIL", EAI_FAIL);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005023#endif
5024#ifdef EAI_FAMILY
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005025 PyModule_AddIntConstant(m, "EAI_FAMILY", EAI_FAMILY);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005026#endif
5027#ifdef EAI_MEMORY
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005028 PyModule_AddIntConstant(m, "EAI_MEMORY", EAI_MEMORY);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005029#endif
5030#ifdef EAI_NODATA
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005031 PyModule_AddIntConstant(m, "EAI_NODATA", EAI_NODATA);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005032#endif
5033#ifdef EAI_NONAME
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005034 PyModule_AddIntConstant(m, "EAI_NONAME", EAI_NONAME);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005035#endif
Martin v. Löwisda91d022003-12-30 11:14:01 +00005036#ifdef EAI_OVERFLOW
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005037 PyModule_AddIntConstant(m, "EAI_OVERFLOW", EAI_OVERFLOW);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005038#endif
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005039#ifdef EAI_SERVICE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005040 PyModule_AddIntConstant(m, "EAI_SERVICE", EAI_SERVICE);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005041#endif
5042#ifdef EAI_SOCKTYPE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005043 PyModule_AddIntConstant(m, "EAI_SOCKTYPE", EAI_SOCKTYPE);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005044#endif
5045#ifdef EAI_SYSTEM
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005046 PyModule_AddIntConstant(m, "EAI_SYSTEM", EAI_SYSTEM);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005047#endif
5048#ifdef EAI_BADHINTS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005049 PyModule_AddIntConstant(m, "EAI_BADHINTS", EAI_BADHINTS);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005050#endif
5051#ifdef EAI_PROTOCOL
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005052 PyModule_AddIntConstant(m, "EAI_PROTOCOL", EAI_PROTOCOL);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005053#endif
5054#ifdef EAI_MAX
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005055 PyModule_AddIntConstant(m, "EAI_MAX", EAI_MAX);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005056#endif
5057#ifdef AI_PASSIVE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005058 PyModule_AddIntConstant(m, "AI_PASSIVE", AI_PASSIVE);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005059#endif
5060#ifdef AI_CANONNAME
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005061 PyModule_AddIntConstant(m, "AI_CANONNAME", AI_CANONNAME);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005062#endif
5063#ifdef AI_NUMERICHOST
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005064 PyModule_AddIntConstant(m, "AI_NUMERICHOST", AI_NUMERICHOST);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005065#endif
Martin v. Löwisda91d022003-12-30 11:14:01 +00005066#ifdef AI_NUMERICSERV
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005067 PyModule_AddIntConstant(m, "AI_NUMERICSERV", AI_NUMERICSERV);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005068#endif
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005069#ifdef AI_MASK
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005070 PyModule_AddIntConstant(m, "AI_MASK", AI_MASK);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005071#endif
5072#ifdef AI_ALL
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005073 PyModule_AddIntConstant(m, "AI_ALL", AI_ALL);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005074#endif
5075#ifdef AI_V4MAPPED_CFG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005076 PyModule_AddIntConstant(m, "AI_V4MAPPED_CFG", AI_V4MAPPED_CFG);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005077#endif
5078#ifdef AI_ADDRCONFIG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005079 PyModule_AddIntConstant(m, "AI_ADDRCONFIG", AI_ADDRCONFIG);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005080#endif
5081#ifdef AI_V4MAPPED
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005082 PyModule_AddIntConstant(m, "AI_V4MAPPED", AI_V4MAPPED);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005083#endif
5084#ifdef AI_DEFAULT
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005085 PyModule_AddIntConstant(m, "AI_DEFAULT", AI_DEFAULT);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005086#endif
5087#ifdef NI_MAXHOST
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005088 PyModule_AddIntConstant(m, "NI_MAXHOST", NI_MAXHOST);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005089#endif
5090#ifdef NI_MAXSERV
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005091 PyModule_AddIntConstant(m, "NI_MAXSERV", NI_MAXSERV);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005092#endif
5093#ifdef NI_NOFQDN
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005094 PyModule_AddIntConstant(m, "NI_NOFQDN", NI_NOFQDN);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005095#endif
5096#ifdef NI_NUMERICHOST
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005097 PyModule_AddIntConstant(m, "NI_NUMERICHOST", NI_NUMERICHOST);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005098#endif
5099#ifdef NI_NAMEREQD
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005100 PyModule_AddIntConstant(m, "NI_NAMEREQD", NI_NAMEREQD);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005101#endif
5102#ifdef NI_NUMERICSERV
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005103 PyModule_AddIntConstant(m, "NI_NUMERICSERV", NI_NUMERICSERV);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005104#endif
5105#ifdef NI_DGRAM
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005106 PyModule_AddIntConstant(m, "NI_DGRAM", NI_DGRAM);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005107#endif
5108
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005109 /* shutdown() parameters */
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005110#ifdef SHUT_RD
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005111 PyModule_AddIntConstant(m, "SHUT_RD", SHUT_RD);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005112#elif defined(SD_RECEIVE)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005113 PyModule_AddIntConstant(m, "SHUT_RD", SD_RECEIVE);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005114#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005115 PyModule_AddIntConstant(m, "SHUT_RD", 0);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005116#endif
5117#ifdef SHUT_WR
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005118 PyModule_AddIntConstant(m, "SHUT_WR", SHUT_WR);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005119#elif defined(SD_SEND)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005120 PyModule_AddIntConstant(m, "SHUT_WR", SD_SEND);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005121#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005122 PyModule_AddIntConstant(m, "SHUT_WR", 1);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005123#endif
5124#ifdef SHUT_RDWR
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005125 PyModule_AddIntConstant(m, "SHUT_RDWR", SHUT_RDWR);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005126#elif defined(SD_BOTH)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005127 PyModule_AddIntConstant(m, "SHUT_RDWR", SD_BOTH);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005128#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005129 PyModule_AddIntConstant(m, "SHUT_RDWR", 2);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005130#endif
5131
Christian Heimesfaf2f632008-01-06 16:59:19 +00005132#ifdef SIO_RCVALL
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005133 {
5134 DWORD codes[] = {SIO_RCVALL, SIO_KEEPALIVE_VALS};
5135 const char *names[] = {"SIO_RCVALL", "SIO_KEEPALIVE_VALS"};
5136 int i;
5137 for(i = 0; i<sizeof(codes)/sizeof(*codes); ++i) {
5138 PyObject *tmp;
5139 tmp = PyLong_FromUnsignedLong(codes[i]);
5140 if (tmp == NULL)
5141 return NULL;
5142 PyModule_AddObject(m, names[i], tmp);
5143 }
5144 }
5145 PyModule_AddIntConstant(m, "RCVALL_OFF", RCVALL_OFF);
5146 PyModule_AddIntConstant(m, "RCVALL_ON", RCVALL_ON);
5147 PyModule_AddIntConstant(m, "RCVALL_SOCKETLEVELONLY", RCVALL_SOCKETLEVELONLY);
Amaury Forgeot d'Arc762681b2008-06-12 23:03:41 +00005148#ifdef RCVALL_IPLEVEL
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005149 PyModule_AddIntConstant(m, "RCVALL_IPLEVEL", RCVALL_IPLEVEL);
Amaury Forgeot d'Arc762681b2008-06-12 23:03:41 +00005150#endif
5151#ifdef RCVALL_MAX
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005152 PyModule_AddIntConstant(m, "RCVALL_MAX", RCVALL_MAX);
Amaury Forgeot d'Arc762681b2008-06-12 23:03:41 +00005153#endif
Christian Heimesfaf2f632008-01-06 16:59:19 +00005154#endif /* _MSTCPIP_ */
5155
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005156 /* Initialize gethostbyname lock */
Just van Rossum1040d2c2003-05-09 07:53:18 +00005157#if defined(USE_GETHOSTBYNAME_LOCK) || defined(USE_GETADDRINFO_LOCK)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005158 netdb_lock = PyThread_allocate_lock();
Guido van Rossum4f199ea1998-04-09 20:56:35 +00005159#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005160 return m;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00005161}
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005162
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00005163
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005164#ifndef HAVE_INET_PTON
Christian Heimes96e7b3d2007-11-20 06:51:17 +00005165#if !defined(NTDDI_VERSION) || (NTDDI_VERSION < NTDDI_LONGHORN)
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00005166
5167/* Simplistic emulation code for inet_pton that only works for IPv4 */
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00005168/* These are not exposed because they do not set errno properly */
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00005169
Guido van Rossum3eede5a2002-06-07 02:08:35 +00005170int
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00005171inet_pton(int af, const char *src, void *dst)
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005172{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005173 if (af == AF_INET) {
Benjamin Petersonf91df042009-02-13 02:50:59 +00005174#if (SIZEOF_INT != 4)
5175#error "Not sure if in_addr_t exists and int is not 32-bits."
5176#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005177 unsigned int packed_addr;
5178 packed_addr = inet_addr(src);
5179 if (packed_addr == INADDR_NONE)
5180 return 0;
5181 memcpy(dst, &packed_addr, 4);
5182 return 1;
5183 }
5184 /* Should set errno to EAFNOSUPPORT */
5185 return -1;
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005186}
5187
Martin v. Löwisc925b1532001-07-21 09:42:15 +00005188const char *
5189inet_ntop(int af, const void *src, char *dst, socklen_t size)
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005190{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005191 if (af == AF_INET) {
5192 struct in_addr packed_addr;
5193 if (size < 16)
5194 /* Should set errno to ENOSPC. */
5195 return NULL;
5196 memcpy(&packed_addr, src, sizeof(packed_addr));
5197 return strncpy(dst, inet_ntoa(packed_addr), size);
5198 }
5199 /* Should set errno to EAFNOSUPPORT */
5200 return NULL;
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005201}
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00005202
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005203#endif
Christian Heimesb6150692007-11-15 23:37:07 +00005204#endif