blob: d3e5c753155f503117858859575477817d43b08c [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
Gregory P. Smith3e85dfd2010-10-17 00:43:10 +0000382#if (defined(HAVE_BLUETOOTH_H) || defined(HAVE_BLUETOOTH_BLUETOOTH_H)) && !defined(__NetBSD__) && !defined(__DragonFly__)
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)
Gregory P. Smith3e85dfd2010-10-17 00:43:10 +0000396#elif defined(__NetBSD__) || defined(__DragonFly__)
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000397#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
Gregory P. Smith3e85dfd2010-10-17 00:43:10 +0000401#define SOL_HCI BTPROTO_HCI
402#define HCI_DATA_DIR SO_HCI_DIRECTION
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000403#define _BT_L2_MEMB(sa, memb) ((sa)->bt_##memb)
404#define _BT_RC_MEMB(sa, memb) ((sa)->bt_##memb)
Thomas Wouterscf297e42007-02-23 15:07:44 +0000405#define _BT_HCI_MEMB(sa, memb) ((sa)->bt_##memb)
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000406#define _BT_SCO_MEMB(sa, memb) ((sa)->bt_##memb)
Hye-Shik Chang81268602004-02-02 06:05:24 +0000407#else
Hye-Shik Chang81268602004-02-02 06:05:24 +0000408#define _BT_L2_MEMB(sa, memb) ((sa)->l2_##memb)
409#define _BT_RC_MEMB(sa, memb) ((sa)->rc_##memb)
Thomas Wouterscf297e42007-02-23 15:07:44 +0000410#define _BT_HCI_MEMB(sa, memb) ((sa)->hci_##memb)
Hye-Shik Chang81268602004-02-02 06:05:24 +0000411#define _BT_SCO_MEMB(sa, memb) ((sa)->sco_##memb)
412#endif
413#endif
414
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000415#ifdef __VMS
416/* TCP/IP Services for VMS uses a maximum send/recv buffer length */
417#define SEGMENT_SIZE (32 * 1024 -1)
418#endif
419
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000420#define SAS2SA(x) ((struct sockaddr *)(x))
Thomas Wouters89f507f2006-12-13 04:49:30 +0000421
Martin v. Löwise9416172003-05-03 10:12:45 +0000422/*
423 * Constants for getnameinfo()
424 */
425#if !defined(NI_MAXHOST)
426#define NI_MAXHOST 1025
427#endif
428#if !defined(NI_MAXSERV)
429#define NI_MAXSERV 32
430#endif
431
Guido van Rossum7d0a8262007-05-21 23:13:11 +0000432#ifndef INVALID_SOCKET /* MS defines this */
433#define INVALID_SOCKET (-1)
434#endif
435
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000436/* XXX There's a problem here: *static* functions are not supposed to have
437 a Py prefix (or use CapitalizedWords). Later... */
438
Guido van Rossum30a685f1991-06-27 15:51:29 +0000439/* Global variable holding the exception type for errors detected
440 by this module (but not argument type or memory errors, etc.). */
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000441static PyObject *socket_error;
442static PyObject *socket_herror;
443static PyObject *socket_gaierror;
Raymond Hettingeref7343c2003-06-29 03:08:05 +0000444static PyObject *socket_timeout;
Guido van Rossum30a685f1991-06-27 15:51:29 +0000445
Tim Peters643a7fc2002-02-17 04:13:21 +0000446/* A forward reference to the socket type object.
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000447 The sock_type variable contains pointers to various functions,
448 some of which call new_sockobject(), which uses sock_type, so
Tim Peters643a7fc2002-02-17 04:13:21 +0000449 there has to be a circular reference. */
Jeremy Hylton938ace62002-07-17 16:30:39 +0000450static PyTypeObject sock_type;
Guido van Rossum48a680c2001-03-02 06:34:14 +0000451
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000452#if defined(HAVE_POLL_H)
453#include <poll.h>
454#elif defined(HAVE_SYS_POLL_H)
455#include <sys/poll.h>
456#endif
457
Charles-François Nataliaa26b272011-08-28 17:51:43 +0200458#ifdef HAVE_POLL
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000459/* Instead of select(), we'll use poll() since poll() works on any fd. */
460#define IS_SELECTABLE(s) 1
461/* Can we call select() with this socket without a buffer overrun? */
Martin v. Löwisf84d1b92006-02-11 09:27:05 +0000462#else
Charles-François Nataliaa26b272011-08-28 17:51:43 +0200463/* If there's no timeout left, we don't have to call select, so it's a safe,
464 * little white lie. */
465#define IS_SELECTABLE(s) (_PyIsSelectable_fd((s)->sock_fd) || (s)->sock_timeout <= 0.0)
Martin v. Löwisf84d1b92006-02-11 09:27:05 +0000466#endif
Neal Norwitz082b2df2006-02-07 07:04:46 +0000467
468static PyObject*
469select_error(void)
470{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000471 PyErr_SetString(socket_error, "unable to select on socket");
472 return NULL;
Neal Norwitz082b2df2006-02-07 07:04:46 +0000473}
474
Antoine Pitrou3e1fd272010-09-28 21:23:11 +0000475#ifdef MS_WINDOWS
Antoine Pitrouc8f01962010-09-28 22:03:27 +0000476#ifndef WSAEAGAIN
477#define WSAEAGAIN WSAEWOULDBLOCK
478#endif
Antoine Pitrou3e1fd272010-09-28 21:23:11 +0000479#define CHECK_ERRNO(expected) \
480 (WSAGetLastError() == WSA ## expected)
481#else
482#define CHECK_ERRNO(expected) \
483 (errno == expected)
484#endif
485
Guido van Rossum30a685f1991-06-27 15:51:29 +0000486/* Convenience function to raise an error according to errno
487 and return a NULL pointer from a function. */
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000488
Guido van Rossum73624e91994-10-10 17:59:00 +0000489static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000490set_error(void)
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000491{
Guido van Rossum8d665e61996-06-26 18:22:49 +0000492#ifdef MS_WINDOWS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000493 int err_no = WSAGetLastError();
494 /* PyErr_SetExcFromWindowsErr() invokes FormatMessage() which
495 recognizes the error codes used by both GetLastError() and
496 WSAGetLastError */
497 if (err_no)
498 return PyErr_SetExcFromWindowsErr(socket_error, err_no);
Guido van Rossum14ed0b21994-09-29 09:50:09 +0000499#endif
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000500
Andrew MacIntyreba43e872002-03-03 03:03:52 +0000501#if defined(PYOS_OS2) && !defined(PYCC_GCC)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000502 if (sock_errno() != NO_ERROR) {
503 APIRET rc;
504 ULONG msglen;
505 char outbuf[100];
506 int myerrorcode = sock_errno();
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000507
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000508 /* Retrieve socket-related error message from MPTN.MSG file */
509 rc = DosGetMessage(NULL, 0, outbuf, sizeof(outbuf),
510 myerrorcode - SOCBASEERR + 26,
511 "mptn.msg",
512 &msglen);
513 if (rc == NO_ERROR) {
514 PyObject *v;
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000515
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000516 /* OS/2 doesn't guarantee a terminator */
517 outbuf[msglen] = '\0';
518 if (strlen(outbuf) > 0) {
519 /* If non-empty msg, trim CRLF */
520 char *lastc = &outbuf[ strlen(outbuf)-1 ];
521 while (lastc > outbuf &&
522 isspace(Py_CHARMASK(*lastc))) {
523 /* Trim trailing whitespace (CRLF) */
524 *lastc-- = '\0';
525 }
526 }
527 v = Py_BuildValue("(is)", myerrorcode, outbuf);
528 if (v != NULL) {
529 PyErr_SetObject(socket_error, v);
530 Py_DECREF(v);
531 }
532 return NULL;
533 }
534 }
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000535#endif
536
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000537 return PyErr_SetFromErrno(socket_error);
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000538}
539
Guido van Rossum30a685f1991-06-27 15:51:29 +0000540
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000541static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000542set_herror(int h_error)
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000543{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000544 PyObject *v;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000545
546#ifdef HAVE_HSTRERROR
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000547 v = Py_BuildValue("(is)", h_error, (char *)hstrerror(h_error));
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000548#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000549 v = Py_BuildValue("(is)", h_error, "host not found");
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000550#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000551 if (v != NULL) {
552 PyErr_SetObject(socket_herror, v);
553 Py_DECREF(v);
554 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000555
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000556 return NULL;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000557}
558
559
560static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000561set_gaierror(int error)
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000562{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000563 PyObject *v;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000564
Martin v. Löwis272cb402002-03-01 08:31:07 +0000565#ifdef EAI_SYSTEM
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000566 /* EAI_SYSTEM is not available on Windows XP. */
567 if (error == EAI_SYSTEM)
568 return set_error();
Martin v. Löwis272cb402002-03-01 08:31:07 +0000569#endif
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000570
Martin v. Löwisf95dd0a2001-08-15 17:14:33 +0000571#ifdef HAVE_GAI_STRERROR
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000572 v = Py_BuildValue("(is)", error, gai_strerror(error));
Martin v. Löwisf95dd0a2001-08-15 17:14:33 +0000573#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000574 v = Py_BuildValue("(is)", error, "getaddrinfo failed");
Martin v. Löwisf95dd0a2001-08-15 17:14:33 +0000575#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000576 if (v != NULL) {
577 PyErr_SetObject(socket_gaierror, v);
578 Py_DECREF(v);
579 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000580
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000581 return NULL;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000582}
583
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000584#ifdef __VMS
585/* Function to send in segments */
586static int
587sendsegmented(int sock_fd, char *buf, int len, int flags)
588{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000589 int n = 0;
590 int remaining = len;
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000591
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000592 while (remaining > 0) {
593 unsigned int segment;
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000594
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000595 segment = (remaining >= SEGMENT_SIZE ? SEGMENT_SIZE : remaining);
596 n = send(sock_fd, buf, segment, flags);
597 if (n < 0) {
598 return n;
599 }
600 remaining -= segment;
601 buf += segment;
602 } /* end while */
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000603
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000604 return len;
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000605}
606#endif
607
Guido van Rossum3eede5a2002-06-07 02:08:35 +0000608/* Function to perform the setting of socket blocking mode
609 internally. block = (1 | 0). */
Guido van Rossum67f7a382002-06-06 21:08:16 +0000610static int
611internal_setblocking(PySocketSockObject *s, int block)
612{
Guido van Rossum67f7a382002-06-06 21:08:16 +0000613#ifndef MS_WINDOWS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000614 int delay_flag;
Guido van Rossum67f7a382002-06-06 21:08:16 +0000615#endif
Antoine Pitroub1c54962010-10-14 15:05:38 +0000616#ifdef SOCK_NONBLOCK
617 if (block)
618 s->sock_type &= (~SOCK_NONBLOCK);
619 else
620 s->sock_type |= SOCK_NONBLOCK;
621#endif
Guido van Rossum67f7a382002-06-06 21:08:16 +0000622
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000623 Py_BEGIN_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +0000624#ifndef MS_WINDOWS
625#if defined(PYOS_OS2) && !defined(PYCC_GCC)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000626 block = !block;
627 ioctl(s->sock_fd, FIONBIO, (caddr_t)&block, sizeof(block));
Martin v. Löwis79acb9e2002-12-06 12:48:53 +0000628#elif defined(__VMS)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000629 block = !block;
630 ioctl(s->sock_fd, FIONBIO, (unsigned int *)&block);
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000631#else /* !PYOS_OS2 && !__VMS */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000632 delay_flag = fcntl(s->sock_fd, F_GETFL, 0);
633 if (block)
634 delay_flag &= (~O_NONBLOCK);
635 else
636 delay_flag |= O_NONBLOCK;
637 fcntl(s->sock_fd, F_SETFL, delay_flag);
Guido van Rossum67f7a382002-06-06 21:08:16 +0000638#endif /* !PYOS_OS2 */
639#else /* MS_WINDOWS */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000640 block = !block;
641 ioctlsocket(s->sock_fd, FIONBIO, (u_long*)&block);
Guido van Rossum67f7a382002-06-06 21:08:16 +0000642#endif /* MS_WINDOWS */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000643 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +0000644
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000645 /* Since these don't return anything */
646 return 1;
Guido van Rossum67f7a382002-06-06 21:08:16 +0000647}
648
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000649/* Do a select()/poll() on the socket, if necessary (sock_timeout > 0).
Guido van Rossum11ba0942002-06-13 15:07:44 +0000650 The argument writing indicates the direction.
Raymond Hettingeref7343c2003-06-29 03:08:05 +0000651 This does not raise an exception; we'll let our caller do that
652 after they've reacquired the interpreter lock.
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000653 Returns 1 on timeout, -1 on error, 0 otherwise. */
Raymond Hettingeref7343c2003-06-29 03:08:05 +0000654static int
Antoine Pitrou3e1fd272010-09-28 21:23:11 +0000655internal_select_ex(PySocketSockObject *s, int writing, double interval)
Guido van Rossum67f7a382002-06-06 21:08:16 +0000656{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000657 int n;
Guido van Rossum11ba0942002-06-13 15:07:44 +0000658
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000659 /* Nothing to do unless we're in timeout mode (not non-blocking) */
660 if (s->sock_timeout <= 0.0)
661 return 0;
Guido van Rossum67f7a382002-06-06 21:08:16 +0000662
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000663 /* Guard against closed socket */
664 if (s->sock_fd < 0)
665 return 0;
Guido van Rossumad654902002-07-19 12:44:59 +0000666
Antoine Pitrou3e1fd272010-09-28 21:23:11 +0000667 /* Handling this condition here simplifies the select loops */
668 if (interval < 0.0)
669 return 1;
670
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000671 /* Prefer poll, if available, since you can poll() any fd
672 * which can't be done with select(). */
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000673#ifdef HAVE_POLL
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000674 {
675 struct pollfd pollfd;
676 int timeout;
Guido van Rossum67f7a382002-06-06 21:08:16 +0000677
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000678 pollfd.fd = s->sock_fd;
679 pollfd.events = writing ? POLLOUT : POLLIN;
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000680
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000681 /* s->sock_timeout is in seconds, timeout in ms */
Antoine Pitrou3e1fd272010-09-28 21:23:11 +0000682 timeout = (int)(interval * 1000 + 0.5);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000683 n = poll(&pollfd, 1, timeout);
684 }
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000685#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000686 {
687 /* Construct the arguments to select */
688 fd_set fds;
689 struct timeval tv;
Antoine Pitrou3e1fd272010-09-28 21:23:11 +0000690 tv.tv_sec = (int)interval;
691 tv.tv_usec = (int)((interval - tv.tv_sec) * 1e6);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000692 FD_ZERO(&fds);
693 FD_SET(s->sock_fd, &fds);
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000694
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000695 /* See if the socket is ready */
696 if (writing)
Antoine Pitrou19467d22010-08-17 19:33:30 +0000697 n = select(Py_SAFE_DOWNCAST(s->sock_fd+1, SOCKET_T, int),
698 NULL, &fds, NULL, &tv);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000699 else
Antoine Pitrou19467d22010-08-17 19:33:30 +0000700 n = select(Py_SAFE_DOWNCAST(s->sock_fd+1, SOCKET_T, int),
701 &fds, NULL, NULL, &tv);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000702 }
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000703#endif
Guido van Rossum39eb8fa2007-11-16 01:24:05 +0000704
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000705 if (n < 0)
706 return -1;
707 if (n == 0)
708 return 1;
709 return 0;
Guido van Rossum67f7a382002-06-06 21:08:16 +0000710}
711
Antoine Pitrou3e1fd272010-09-28 21:23:11 +0000712static int
713internal_select(PySocketSockObject *s, int writing)
714{
715 return internal_select_ex(s, writing, s->sock_timeout);
716}
717
718/*
719 Two macros for automatic retry of select() in case of false positives
720 (for example, select() could indicate a socket is ready for reading
721 but the data then discarded by the OS because of a wrong checksum).
722 Here is an example of use:
723
724 BEGIN_SELECT_LOOP(s)
725 Py_BEGIN_ALLOW_THREADS
726 timeout = internal_select_ex(s, 0, interval);
727 if (!timeout)
728 outlen = recv(s->sock_fd, cbuf, len, flags);
729 Py_END_ALLOW_THREADS
730 if (timeout == 1) {
731 PyErr_SetString(socket_timeout, "timed out");
732 return -1;
733 }
734 END_SELECT_LOOP(s)
735*/
736
737#define BEGIN_SELECT_LOOP(s) \
738 { \
739 _PyTime_timeval now, deadline = {0, 0}; \
740 double interval = s->sock_timeout; \
741 int has_timeout = s->sock_timeout > 0.0; \
742 if (has_timeout) { \
743 _PyTime_gettimeofday(&now); \
744 deadline = now; \
745 _PyTime_ADD_SECONDS(deadline, s->sock_timeout); \
746 } \
747 while (1) { \
748 errno = 0; \
749
750#define END_SELECT_LOOP(s) \
751 if (!has_timeout || \
752 (!CHECK_ERRNO(EWOULDBLOCK) && !CHECK_ERRNO(EAGAIN))) \
753 break; \
754 _PyTime_gettimeofday(&now); \
755 interval = _PyTime_INTERVAL(now, deadline); \
756 } \
757 } \
758
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000759/* Initialize a new socket object. */
760
Tim Petersa12b4cf2002-07-18 22:38:44 +0000761static double defaulttimeout = -1.0; /* Default timeout for new sockets */
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +0000762
Martin v. Löwis1a214512008-06-11 05:26:20 +0000763static void
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000764init_sockobject(PySocketSockObject *s,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000765 SOCKET_T fd, int family, int type, int proto)
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000766{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000767 s->sock_fd = fd;
768 s->sock_family = family;
769 s->sock_type = type;
770 s->sock_proto = proto;
Guido van Rossum67f7a382002-06-06 21:08:16 +0000771
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000772 s->errorhandler = &set_error;
Antoine Pitroub1c54962010-10-14 15:05:38 +0000773#ifdef SOCK_NONBLOCK
774 if (type & SOCK_NONBLOCK)
775 s->sock_timeout = 0.0;
776 else
777#endif
778 {
779 s->sock_timeout = defaulttimeout;
780 if (defaulttimeout >= 0.0)
781 internal_setblocking(s, 0);
782 }
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +0000783
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000784}
785
786
Guido van Rossum30a685f1991-06-27 15:51:29 +0000787/* Create a new socket object.
788 This just creates the object and initializes it.
789 If the creation fails, return NULL and set an exception (implicit
790 in NEWOBJ()). */
791
Guido van Rossum73624e91994-10-10 17:59:00 +0000792static PySocketSockObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000793new_sockobject(SOCKET_T fd, int family, int type, int proto)
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000794{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000795 PySocketSockObject *s;
796 s = (PySocketSockObject *)
797 PyType_GenericNew(&sock_type, NULL, NULL);
798 if (s != NULL)
799 init_sockobject(s, fd, family, type, proto);
800 return s;
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000801}
802
Guido van Rossum30a685f1991-06-27 15:51:29 +0000803
Guido van Rossum48a680c2001-03-02 06:34:14 +0000804/* Lock to allow python interpreter to continue, but only allow one
Just van Rossum1040d2c2003-05-09 07:53:18 +0000805 thread to be in gethostbyname or getaddrinfo */
806#if defined(USE_GETHOSTBYNAME_LOCK) || defined(USE_GETADDRINFO_LOCK)
807PyThread_type_lock netdb_lock;
Guido van Rossum4f199ea1998-04-09 20:56:35 +0000808#endif
809
810
Guido van Rossum30a685f1991-06-27 15:51:29 +0000811/* Convert a string specifying a host name or one of a few symbolic
812 names to a numeric IP address. This usually calls gethostbyname()
813 to do the work; the names "" and "<broadcast>" are special.
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000814 Return the length (IPv4 should be 4 bytes), or negative if
Guido van Rossum30a685f1991-06-27 15:51:29 +0000815 an error occurred; then an exception is raised. */
816
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000817static int
Martin v. Löwisddc6f472002-07-28 16:10:31 +0000818setipaddr(char *name, struct sockaddr *addr_ret, size_t addr_ret_size, int af)
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000819{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000820 struct addrinfo hints, *res;
821 int error;
822 int d1, d2, d3, d4;
823 char ch;
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000824
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000825 memset((void *) addr_ret, '\0', sizeof(*addr_ret));
826 if (name[0] == '\0') {
827 int siz;
828 memset(&hints, 0, sizeof(hints));
829 hints.ai_family = af;
830 hints.ai_socktype = SOCK_DGRAM; /*dummy*/
831 hints.ai_flags = AI_PASSIVE;
832 Py_BEGIN_ALLOW_THREADS
833 ACQUIRE_GETADDRINFO_LOCK
834 error = getaddrinfo(NULL, "0", &hints, &res);
835 Py_END_ALLOW_THREADS
836 /* We assume that those thread-unsafe getaddrinfo() versions
837 *are* safe regarding their return value, ie. that a
838 subsequent call to getaddrinfo() does not destroy the
839 outcome of the first call. */
840 RELEASE_GETADDRINFO_LOCK
841 if (error) {
842 set_gaierror(error);
843 return -1;
844 }
845 switch (res->ai_family) {
846 case AF_INET:
847 siz = 4;
848 break;
Martin v. Löwis44ddbde2001-12-02 10:15:37 +0000849#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000850 case AF_INET6:
851 siz = 16;
852 break;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000853#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000854 default:
855 freeaddrinfo(res);
856 PyErr_SetString(socket_error,
857 "unsupported address family");
858 return -1;
859 }
860 if (res->ai_next) {
861 freeaddrinfo(res);
862 PyErr_SetString(socket_error,
863 "wildcard resolved to multiple address");
864 return -1;
865 }
866 if (res->ai_addrlen < addr_ret_size)
867 addr_ret_size = res->ai_addrlen;
868 memcpy(addr_ret, res->ai_addr, addr_ret_size);
869 freeaddrinfo(res);
870 return siz;
871 }
872 if (name[0] == '<' && strcmp(name, "<broadcast>") == 0) {
873 struct sockaddr_in *sin;
874 if (af != AF_INET && af != AF_UNSPEC) {
875 PyErr_SetString(socket_error,
876 "address family mismatched");
877 return -1;
878 }
879 sin = (struct sockaddr_in *)addr_ret;
880 memset((void *) sin, '\0', sizeof(*sin));
881 sin->sin_family = AF_INET;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000882#ifdef HAVE_SOCKADDR_SA_LEN
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000883 sin->sin_len = sizeof(*sin);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000884#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000885 sin->sin_addr.s_addr = INADDR_BROADCAST;
886 return sizeof(sin->sin_addr);
887 }
888 if (sscanf(name, "%d.%d.%d.%d%c", &d1, &d2, &d3, &d4, &ch) == 4 &&
889 0 <= d1 && d1 <= 255 && 0 <= d2 && d2 <= 255 &&
890 0 <= d3 && d3 <= 255 && 0 <= d4 && d4 <= 255) {
891 struct sockaddr_in *sin;
892 sin = (struct sockaddr_in *)addr_ret;
893 sin->sin_addr.s_addr = htonl(
894 ((long) d1 << 24) | ((long) d2 << 16) |
895 ((long) d3 << 8) | ((long) d4 << 0));
896 sin->sin_family = AF_INET;
Anthony Baxter0e85f9d2003-05-02 15:40:46 +0000897#ifdef HAVE_SOCKADDR_SA_LEN
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000898 sin->sin_len = sizeof(*sin);
Anthony Baxter0e85f9d2003-05-02 15:40:46 +0000899#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000900 return 4;
901 }
902 memset(&hints, 0, sizeof(hints));
903 hints.ai_family = af;
904 Py_BEGIN_ALLOW_THREADS
905 ACQUIRE_GETADDRINFO_LOCK
906 error = getaddrinfo(name, NULL, &hints, &res);
Martin v. Löwis7c4b5fa2001-10-25 09:04:03 +0000907#if defined(__digital__) && defined(__unix__)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000908 if (error == EAI_NONAME && af == AF_UNSPEC) {
909 /* On Tru64 V5.1, numeric-to-addr conversion fails
910 if no address family is given. Assume IPv4 for now.*/
911 hints.ai_family = AF_INET;
912 error = getaddrinfo(name, NULL, &hints, &res);
913 }
Martin v. Löwis7c4b5fa2001-10-25 09:04:03 +0000914#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000915 Py_END_ALLOW_THREADS
916 RELEASE_GETADDRINFO_LOCK /* see comment in setipaddr() */
917 if (error) {
918 set_gaierror(error);
919 return -1;
920 }
921 if (res->ai_addrlen < addr_ret_size)
922 addr_ret_size = res->ai_addrlen;
923 memcpy((char *) addr_ret, res->ai_addr, addr_ret_size);
924 freeaddrinfo(res);
925 switch (addr_ret->sa_family) {
926 case AF_INET:
927 return 4;
Martin v. Löwis44ddbde2001-12-02 10:15:37 +0000928#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000929 case AF_INET6:
930 return 16;
Guido van Rossum955becc1999-03-22 20:14:53 +0000931#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000932 default:
933 PyErr_SetString(socket_error, "unknown address family");
934 return -1;
935 }
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000936}
937
Guido van Rossum30a685f1991-06-27 15:51:29 +0000938
Guido van Rossum30a685f1991-06-27 15:51:29 +0000939/* Create a string object representing an IP address.
940 This is always a string of the form 'dd.dd.dd.dd' (with variable
941 size numbers). */
942
Guido van Rossum73624e91994-10-10 17:59:00 +0000943static PyObject *
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000944makeipaddr(struct sockaddr *addr, int addrlen)
Guido van Rossum30a685f1991-06-27 15:51:29 +0000945{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000946 char buf[NI_MAXHOST];
947 int error;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000948
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000949 error = getnameinfo(addr, addrlen, buf, sizeof(buf), NULL, 0,
950 NI_NUMERICHOST);
951 if (error) {
952 set_gaierror(error);
953 return NULL;
954 }
955 return PyUnicode_FromString(buf);
Guido van Rossum30a685f1991-06-27 15:51:29 +0000956}
957
958
Martin v. Löwis558d9bf2004-06-03 09:24:42 +0000959#ifdef USE_BLUETOOTH
960/* Convert a string representation of a Bluetooth address into a numeric
961 address. Returns the length (6), or raises an exception and returns -1 if
962 an error occurred. */
963
964static int
965setbdaddr(char *name, bdaddr_t *bdaddr)
966{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000967 unsigned int b0, b1, b2, b3, b4, b5;
968 char ch;
969 int n;
Martin v. Löwis558d9bf2004-06-03 09:24:42 +0000970
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000971 n = sscanf(name, "%X:%X:%X:%X:%X:%X%c",
972 &b5, &b4, &b3, &b2, &b1, &b0, &ch);
973 if (n == 6 && (b0 | b1 | b2 | b3 | b4 | b5) < 256) {
974 bdaddr->b[0] = b0;
975 bdaddr->b[1] = b1;
976 bdaddr->b[2] = b2;
977 bdaddr->b[3] = b3;
978 bdaddr->b[4] = b4;
979 bdaddr->b[5] = b5;
980 return 6;
981 } else {
982 PyErr_SetString(socket_error, "bad bluetooth address");
983 return -1;
984 }
Martin v. Löwis558d9bf2004-06-03 09:24:42 +0000985}
986
987/* Create a string representation of the Bluetooth address. This is always a
988 string of the form 'XX:XX:XX:XX:XX:XX' where XX is a two digit hexadecimal
989 value (zero padded if necessary). */
990
991static PyObject *
992makebdaddr(bdaddr_t *bdaddr)
993{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000994 char buf[(6 * 2) + 5 + 1];
Martin v. Löwis558d9bf2004-06-03 09:24:42 +0000995
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000996 sprintf(buf, "%02X:%02X:%02X:%02X:%02X:%02X",
997 bdaddr->b[5], bdaddr->b[4], bdaddr->b[3],
998 bdaddr->b[2], bdaddr->b[1], bdaddr->b[0]);
999 return PyUnicode_FromString(buf);
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001000}
1001#endif
1002
1003
Guido van Rossum30a685f1991-06-27 15:51:29 +00001004/* Create an object representing the given socket address,
1005 suitable for passing it back to bind(), connect() etc.
1006 The family field of the sockaddr structure is inspected
1007 to determine what kind of address it really is. */
1008
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001009/*ARGSUSED*/
Guido van Rossum73624e91994-10-10 17:59:00 +00001010static PyObject *
Antoine Pitrou19467d22010-08-17 19:33:30 +00001011makesockaddr(SOCKET_T sockfd, struct sockaddr *addr, size_t addrlen, int proto)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001012{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001013 if (addrlen == 0) {
1014 /* No address -- may be recvfrom() from known socket */
1015 Py_INCREF(Py_None);
1016 return Py_None;
1017 }
Guido van Rossum25bec8c1992-08-05 19:00:45 +00001018
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001019 switch (addr->sa_family) {
Guido van Rossum30a685f1991-06-27 15:51:29 +00001020
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001021 case AF_INET:
1022 {
1023 struct sockaddr_in *a;
1024 PyObject *addrobj = makeipaddr(addr, sizeof(*a));
1025 PyObject *ret = NULL;
1026 if (addrobj) {
1027 a = (struct sockaddr_in *)addr;
1028 ret = Py_BuildValue("Oi", addrobj, ntohs(a->sin_port));
1029 Py_DECREF(addrobj);
1030 }
1031 return ret;
1032 }
Guido van Rossum30a685f1991-06-27 15:51:29 +00001033
Andrew MacIntyred12dfbb2004-04-04 07:13:49 +00001034#if defined(AF_UNIX)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001035 case AF_UNIX:
1036 {
1037 struct sockaddr_un *a = (struct sockaddr_un *) addr;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001038#ifdef linux
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001039 if (a->sun_path[0] == 0) { /* Linux abstract namespace */
1040 addrlen -= offsetof(struct sockaddr_un, sun_path);
1041 return PyBytes_FromStringAndSize(a->sun_path, addrlen);
1042 }
1043 else
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001044#endif /* linux */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001045 {
1046 /* regular NULL-terminated string */
1047 return PyUnicode_FromString(a->sun_path);
1048 }
1049 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00001050#endif /* AF_UNIX */
1051
Martin v. Löwis11017b12006-01-14 18:12:57 +00001052#if defined(AF_NETLINK)
1053 case AF_NETLINK:
1054 {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001055 struct sockaddr_nl *a = (struct sockaddr_nl *) addr;
1056 return Py_BuildValue("II", a->nl_pid, a->nl_groups);
Martin v. Löwis11017b12006-01-14 18:12:57 +00001057 }
1058#endif /* AF_NETLINK */
1059
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00001060#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001061 case AF_INET6:
1062 {
1063 struct sockaddr_in6 *a;
1064 PyObject *addrobj = makeipaddr(addr, sizeof(*a));
1065 PyObject *ret = NULL;
1066 if (addrobj) {
1067 a = (struct sockaddr_in6 *)addr;
1068 ret = Py_BuildValue("Oiii",
1069 addrobj,
1070 ntohs(a->sin6_port),
1071 a->sin6_flowinfo,
1072 a->sin6_scope_id);
1073 Py_DECREF(addrobj);
1074 }
1075 return ret;
1076 }
Jeremy Hylton22308652001-02-02 03:23:09 +00001077#endif
Guido van Rossum30a685f1991-06-27 15:51:29 +00001078
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001079#ifdef USE_BLUETOOTH
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001080 case AF_BLUETOOTH:
1081 switch (proto) {
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001082
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001083 case BTPROTO_L2CAP:
1084 {
1085 struct sockaddr_l2 *a = (struct sockaddr_l2 *) addr;
1086 PyObject *addrobj = makebdaddr(&_BT_L2_MEMB(a, bdaddr));
1087 PyObject *ret = NULL;
1088 if (addrobj) {
1089 ret = Py_BuildValue("Oi",
1090 addrobj,
1091 _BT_L2_MEMB(a, psm));
1092 Py_DECREF(addrobj);
1093 }
1094 return ret;
1095 }
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001096
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001097 case BTPROTO_RFCOMM:
1098 {
1099 struct sockaddr_rc *a = (struct sockaddr_rc *) addr;
1100 PyObject *addrobj = makebdaddr(&_BT_RC_MEMB(a, bdaddr));
1101 PyObject *ret = NULL;
1102 if (addrobj) {
1103 ret = Py_BuildValue("Oi",
1104 addrobj,
1105 _BT_RC_MEMB(a, channel));
1106 Py_DECREF(addrobj);
1107 }
1108 return ret;
1109 }
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001110
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001111 case BTPROTO_HCI:
1112 {
1113 struct sockaddr_hci *a = (struct sockaddr_hci *) addr;
Gregory P. Smith3e85dfd2010-10-17 00:43:10 +00001114#if defined(__NetBSD__) || defined(__DragonFly__)
1115 return makebdaddr(&_BT_HCI_MEMB(a, bdaddr));
1116#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001117 PyObject *ret = NULL;
1118 ret = Py_BuildValue("i", _BT_HCI_MEMB(a, dev));
1119 return ret;
Gregory P. Smith3e85dfd2010-10-17 00:43:10 +00001120#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001121 }
Thomas Wouterscf297e42007-02-23 15:07:44 +00001122
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001123#if !defined(__FreeBSD__)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001124 case BTPROTO_SCO:
1125 {
1126 struct sockaddr_sco *a = (struct sockaddr_sco *) addr;
1127 return makebdaddr(&_BT_SCO_MEMB(a, bdaddr));
1128 }
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001129#endif
1130
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001131 default:
1132 PyErr_SetString(PyExc_ValueError,
1133 "Unknown Bluetooth protocol");
1134 return NULL;
1135 }
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001136#endif
1137
Antoine Pitroub156a462010-10-27 20:13:57 +00001138#if defined(HAVE_NETPACKET_PACKET_H) && defined(SIOCGIFNAME)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001139 case AF_PACKET:
1140 {
1141 struct sockaddr_ll *a = (struct sockaddr_ll *)addr;
1142 char *ifname = "";
1143 struct ifreq ifr;
1144 /* need to look up interface name give index */
1145 if (a->sll_ifindex) {
1146 ifr.ifr_ifindex = a->sll_ifindex;
1147 if (ioctl(sockfd, SIOCGIFNAME, &ifr) == 0)
1148 ifname = ifr.ifr_name;
1149 }
1150 return Py_BuildValue("shbhy#",
1151 ifname,
1152 ntohs(a->sll_protocol),
1153 a->sll_pkttype,
1154 a->sll_hatype,
1155 a->sll_addr,
1156 a->sll_halen);
1157 }
Jeremy Hylton22308652001-02-02 03:23:09 +00001158#endif
Guido van Rossum48a680c2001-03-02 06:34:14 +00001159
Christian Heimes043d6f62008-01-07 17:19:16 +00001160#ifdef HAVE_LINUX_TIPC_H
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001161 case AF_TIPC:
1162 {
1163 struct sockaddr_tipc *a = (struct sockaddr_tipc *) addr;
1164 if (a->addrtype == TIPC_ADDR_NAMESEQ) {
1165 return Py_BuildValue("IIIII",
1166 a->addrtype,
1167 a->addr.nameseq.type,
1168 a->addr.nameseq.lower,
1169 a->addr.nameseq.upper,
1170 a->scope);
1171 } else if (a->addrtype == TIPC_ADDR_NAME) {
1172 return Py_BuildValue("IIIII",
1173 a->addrtype,
1174 a->addr.name.name.type,
1175 a->addr.name.name.instance,
1176 a->addr.name.name.instance,
1177 a->scope);
1178 } else if (a->addrtype == TIPC_ADDR_ID) {
1179 return Py_BuildValue("IIIII",
1180 a->addrtype,
1181 a->addr.id.node,
1182 a->addr.id.ref,
1183 0,
1184 a->scope);
1185 } else {
1186 PyErr_SetString(PyExc_ValueError,
1187 "Invalid address type");
1188 return NULL;
1189 }
1190 }
Christian Heimes043d6f62008-01-07 17:19:16 +00001191#endif
1192
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001193 /* More cases here... */
Guido van Rossum30a685f1991-06-27 15:51:29 +00001194
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001195 default:
1196 /* If we don't know the address family, don't raise an
1197 exception -- return it as an (int, bytes) tuple. */
1198 return Py_BuildValue("iy#",
1199 addr->sa_family,
1200 addr->sa_data,
1201 sizeof(addr->sa_data));
Guido van Rossum25bec8c1992-08-05 19:00:45 +00001202
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001203 }
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001204}
1205
Guido van Rossum30a685f1991-06-27 15:51:29 +00001206
1207/* Parse a socket address argument according to the socket object's
1208 address family. Return 1 if the address was in the proper format,
1209 0 of not. The address is returned through addr_ret, its length
1210 through len_ret. */
1211
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001212static int
Guido van Rossum48a680c2001-03-02 06:34:14 +00001213getsockaddrarg(PySocketSockObject *s, PyObject *args,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001214 struct sockaddr *addr_ret, int *len_ret)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001215{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001216 switch (s->sock_family) {
Guido van Rossum30a685f1991-06-27 15:51:29 +00001217
Andrew MacIntyred12dfbb2004-04-04 07:13:49 +00001218#if defined(AF_UNIX)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001219 case AF_UNIX:
1220 {
1221 struct sockaddr_un* addr;
1222 char *path;
1223 int len;
1224 if (!PyArg_Parse(args, "s#", &path, &len))
1225 return 0;
Thomas Wouters89f507f2006-12-13 04:49:30 +00001226
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001227 addr = (struct sockaddr_un*)addr_ret;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001228#ifdef linux
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001229 if (len > 0 && path[0] == 0) {
1230 /* Linux abstract namespace extension */
1231 if (len > sizeof addr->sun_path) {
1232 PyErr_SetString(socket_error,
1233 "AF_UNIX path too long");
1234 return 0;
1235 }
1236 }
1237 else
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001238#endif /* linux */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001239 {
1240 /* regular NULL-terminated string */
1241 if (len >= sizeof addr->sun_path) {
1242 PyErr_SetString(socket_error,
1243 "AF_UNIX path too long");
1244 return 0;
1245 }
1246 addr->sun_path[len] = 0;
1247 }
1248 addr->sun_family = s->sock_family;
1249 memcpy(addr->sun_path, path, len);
Andrew MacIntyredaedf212004-04-11 12:03:57 +00001250#if defined(PYOS_OS2)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001251 *len_ret = sizeof(*addr);
Andrew MacIntyredaedf212004-04-11 12:03:57 +00001252#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001253 *len_ret = len + offsetof(struct sockaddr_un, sun_path);
Andrew MacIntyredaedf212004-04-11 12:03:57 +00001254#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001255 return 1;
1256 }
Guido van Rossumb6775db1994-08-01 11:34:53 +00001257#endif /* AF_UNIX */
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001258
Martin v. Löwis11017b12006-01-14 18:12:57 +00001259#if defined(AF_NETLINK)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001260 case AF_NETLINK:
1261 {
1262 struct sockaddr_nl* addr;
1263 int pid, groups;
1264 addr = (struct sockaddr_nl *)addr_ret;
1265 if (!PyTuple_Check(args)) {
1266 PyErr_Format(
1267 PyExc_TypeError,
1268 "getsockaddrarg: "
1269 "AF_NETLINK address must be tuple, not %.500s",
1270 Py_TYPE(args)->tp_name);
1271 return 0;
1272 }
1273 if (!PyArg_ParseTuple(args, "II:getsockaddrarg", &pid, &groups))
1274 return 0;
1275 addr->nl_family = AF_NETLINK;
1276 addr->nl_pid = pid;
1277 addr->nl_groups = groups;
1278 *len_ret = sizeof(*addr);
1279 return 1;
1280 }
Martin v. Löwis11017b12006-01-14 18:12:57 +00001281#endif
1282
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001283 case AF_INET:
1284 {
1285 struct sockaddr_in* addr;
1286 char *host;
1287 int port, result;
1288 if (!PyTuple_Check(args)) {
1289 PyErr_Format(
1290 PyExc_TypeError,
1291 "getsockaddrarg: "
1292 "AF_INET address must be tuple, not %.500s",
1293 Py_TYPE(args)->tp_name);
1294 return 0;
1295 }
1296 if (!PyArg_ParseTuple(args, "eti:getsockaddrarg",
1297 "idna", &host, &port))
1298 return 0;
1299 addr=(struct sockaddr_in*)addr_ret;
1300 result = setipaddr(host, (struct sockaddr *)addr,
1301 sizeof(*addr), AF_INET);
1302 PyMem_Free(host);
1303 if (result < 0)
1304 return 0;
1305 if (port < 0 || port > 0xffff) {
1306 PyErr_SetString(
1307 PyExc_OverflowError,
1308 "getsockaddrarg: port must be 0-65535.");
1309 return 0;
1310 }
1311 addr->sin_family = AF_INET;
1312 addr->sin_port = htons((short)port);
1313 *len_ret = sizeof *addr;
1314 return 1;
1315 }
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001316
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00001317#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001318 case AF_INET6:
1319 {
1320 struct sockaddr_in6* addr;
1321 char *host;
1322 int port, flowinfo, scope_id, result;
1323 flowinfo = scope_id = 0;
1324 if (!PyTuple_Check(args)) {
1325 PyErr_Format(
1326 PyExc_TypeError,
1327 "getsockaddrarg: "
1328 "AF_INET6 address must be tuple, not %.500s",
1329 Py_TYPE(args)->tp_name);
1330 return 0;
1331 }
1332 if (!PyArg_ParseTuple(args, "eti|ii",
1333 "idna", &host, &port, &flowinfo,
1334 &scope_id)) {
1335 return 0;
1336 }
1337 addr = (struct sockaddr_in6*)addr_ret;
1338 result = setipaddr(host, (struct sockaddr *)addr,
1339 sizeof(*addr), AF_INET6);
1340 PyMem_Free(host);
1341 if (result < 0)
1342 return 0;
1343 if (port < 0 || port > 0xffff) {
1344 PyErr_SetString(
1345 PyExc_OverflowError,
1346 "getsockaddrarg: port must be 0-65535.");
1347 return 0;
1348 }
1349 addr->sin6_family = s->sock_family;
1350 addr->sin6_port = htons((short)port);
1351 addr->sin6_flowinfo = flowinfo;
1352 addr->sin6_scope_id = scope_id;
1353 *len_ret = sizeof *addr;
1354 return 1;
1355 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00001356#endif
1357
Hye-Shik Chang81268602004-02-02 06:05:24 +00001358#ifdef USE_BLUETOOTH
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001359 case AF_BLUETOOTH:
1360 {
1361 switch (s->sock_proto) {
1362 case BTPROTO_L2CAP:
1363 {
1364 struct sockaddr_l2 *addr;
1365 char *straddr;
Martin v. Löwis12af0482004-01-31 12:34:17 +00001366
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001367 addr = (struct sockaddr_l2 *)addr_ret;
1368 memset(addr, 0, sizeof(struct sockaddr_l2));
1369 _BT_L2_MEMB(addr, family) = AF_BLUETOOTH;
1370 if (!PyArg_ParseTuple(args, "si", &straddr,
1371 &_BT_L2_MEMB(addr, psm))) {
1372 PyErr_SetString(socket_error, "getsockaddrarg: "
1373 "wrong format");
1374 return 0;
1375 }
1376 if (setbdaddr(straddr, &_BT_L2_MEMB(addr, bdaddr)) < 0)
1377 return 0;
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001378
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001379 *len_ret = sizeof *addr;
1380 return 1;
1381 }
1382 case BTPROTO_RFCOMM:
1383 {
1384 struct sockaddr_rc *addr;
1385 char *straddr;
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001386
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001387 addr = (struct sockaddr_rc *)addr_ret;
1388 _BT_RC_MEMB(addr, family) = AF_BLUETOOTH;
1389 if (!PyArg_ParseTuple(args, "si", &straddr,
1390 &_BT_RC_MEMB(addr, channel))) {
1391 PyErr_SetString(socket_error, "getsockaddrarg: "
1392 "wrong format");
1393 return 0;
1394 }
1395 if (setbdaddr(straddr, &_BT_RC_MEMB(addr, bdaddr)) < 0)
1396 return 0;
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001397
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001398 *len_ret = sizeof *addr;
1399 return 1;
1400 }
1401 case BTPROTO_HCI:
1402 {
1403 struct sockaddr_hci *addr = (struct sockaddr_hci *)addr_ret;
Gregory P. Smith3e85dfd2010-10-17 00:43:10 +00001404#if defined(__NetBSD__) || defined(__DragonFly__)
Alexander Belopolskye239d232010-12-08 23:31:48 +00001405 char *straddr = PyBytes_AS_STRING(args);
Gregory P. Smith3e85dfd2010-10-17 00:43:10 +00001406
Alexander Belopolskye239d232010-12-08 23:31:48 +00001407 _BT_HCI_MEMB(addr, family) = AF_BLUETOOTH;
Gregory P. Smith3e85dfd2010-10-17 00:43:10 +00001408 if (straddr == NULL) {
1409 PyErr_SetString(socket_error, "getsockaddrarg: "
1410 "wrong format");
1411 return 0;
1412 }
1413 if (setbdaddr(straddr, &_BT_HCI_MEMB(addr, bdaddr)) < 0)
1414 return 0;
1415#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001416 _BT_HCI_MEMB(addr, family) = AF_BLUETOOTH;
1417 if (!PyArg_ParseTuple(args, "i", &_BT_HCI_MEMB(addr, dev))) {
1418 PyErr_SetString(socket_error, "getsockaddrarg: "
1419 "wrong format");
1420 return 0;
1421 }
Gregory P. Smith3e85dfd2010-10-17 00:43:10 +00001422#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001423 *len_ret = sizeof *addr;
1424 return 1;
1425 }
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001426#if !defined(__FreeBSD__)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001427 case BTPROTO_SCO:
1428 {
1429 struct sockaddr_sco *addr;
1430 char *straddr;
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001431
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001432 addr = (struct sockaddr_sco *)addr_ret;
1433 _BT_SCO_MEMB(addr, family) = AF_BLUETOOTH;
1434 if (!PyBytes_Check(args)) {
1435 PyErr_SetString(socket_error, "getsockaddrarg: "
1436 "wrong format");
1437 return 0;
1438 }
1439 straddr = PyBytes_AS_STRING(args);
1440 if (setbdaddr(straddr, &_BT_SCO_MEMB(addr, bdaddr)) < 0)
1441 return 0;
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001442
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001443 *len_ret = sizeof *addr;
1444 return 1;
1445 }
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001446#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001447 default:
1448 PyErr_SetString(socket_error, "getsockaddrarg: unknown Bluetooth protocol");
1449 return 0;
1450 }
1451 }
Martin v. Löwis12af0482004-01-31 12:34:17 +00001452#endif
1453
Antoine Pitroub156a462010-10-27 20:13:57 +00001454#if defined(HAVE_NETPACKET_PACKET_H) && defined(SIOCGIFINDEX)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001455 case AF_PACKET:
1456 {
1457 struct sockaddr_ll* addr;
1458 struct ifreq ifr;
1459 char *interfaceName;
1460 int protoNumber;
1461 int hatype = 0;
1462 int pkttype = 0;
1463 char *haddr = NULL;
1464 unsigned int halen = 0;
Guido van Rossum48a680c2001-03-02 06:34:14 +00001465
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001466 if (!PyTuple_Check(args)) {
1467 PyErr_Format(
1468 PyExc_TypeError,
1469 "getsockaddrarg: "
1470 "AF_PACKET address must be tuple, not %.500s",
1471 Py_TYPE(args)->tp_name);
1472 return 0;
1473 }
1474 if (!PyArg_ParseTuple(args, "si|iiy#", &interfaceName,
1475 &protoNumber, &pkttype, &hatype,
1476 &haddr, &halen))
1477 return 0;
1478 strncpy(ifr.ifr_name, interfaceName, sizeof(ifr.ifr_name));
1479 ifr.ifr_name[(sizeof(ifr.ifr_name))-1] = '\0';
1480 if (ioctl(s->sock_fd, SIOCGIFINDEX, &ifr) < 0) {
1481 s->errorhandler();
1482 return 0;
1483 }
1484 if (halen > 8) {
1485 PyErr_SetString(PyExc_ValueError,
1486 "Hardware address must be 8 bytes or less");
1487 return 0;
1488 }
1489 if (protoNumber < 0 || protoNumber > 0xffff) {
1490 PyErr_SetString(
1491 PyExc_OverflowError,
1492 "getsockaddrarg: protoNumber must be 0-65535.");
1493 return 0;
1494 }
1495 addr = (struct sockaddr_ll*)addr_ret;
1496 addr->sll_family = AF_PACKET;
1497 addr->sll_protocol = htons((short)protoNumber);
1498 addr->sll_ifindex = ifr.ifr_ifindex;
1499 addr->sll_pkttype = pkttype;
1500 addr->sll_hatype = hatype;
1501 if (halen != 0) {
1502 memcpy(&addr->sll_addr, haddr, halen);
1503 }
1504 addr->sll_halen = halen;
1505 *len_ret = sizeof *addr;
1506 return 1;
1507 }
Guido van Rossum48a680c2001-03-02 06:34:14 +00001508#endif
1509
Christian Heimes043d6f62008-01-07 17:19:16 +00001510#ifdef HAVE_LINUX_TIPC_H
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001511 case AF_TIPC:
1512 {
1513 unsigned int atype, v1, v2, v3;
1514 unsigned int scope = TIPC_CLUSTER_SCOPE;
1515 struct sockaddr_tipc *addr;
Christian Heimes043d6f62008-01-07 17:19:16 +00001516
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001517 if (!PyTuple_Check(args)) {
1518 PyErr_Format(
1519 PyExc_TypeError,
1520 "getsockaddrarg: "
1521 "AF_TIPC address must be tuple, not %.500s",
1522 Py_TYPE(args)->tp_name);
1523 return 0;
1524 }
Christian Heimes043d6f62008-01-07 17:19:16 +00001525
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001526 if (!PyArg_ParseTuple(args,
1527 "IIII|I;Invalid TIPC address format",
1528 &atype, &v1, &v2, &v3, &scope))
1529 return 0;
Christian Heimes043d6f62008-01-07 17:19:16 +00001530
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001531 addr = (struct sockaddr_tipc *) addr_ret;
1532 memset(addr, 0, sizeof(struct sockaddr_tipc));
Christian Heimes043d6f62008-01-07 17:19:16 +00001533
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001534 addr->family = AF_TIPC;
1535 addr->scope = scope;
1536 addr->addrtype = atype;
Christian Heimes043d6f62008-01-07 17:19:16 +00001537
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001538 if (atype == TIPC_ADDR_NAMESEQ) {
1539 addr->addr.nameseq.type = v1;
1540 addr->addr.nameseq.lower = v2;
1541 addr->addr.nameseq.upper = v3;
1542 } else if (atype == TIPC_ADDR_NAME) {
1543 addr->addr.name.name.type = v1;
1544 addr->addr.name.name.instance = v2;
1545 } else if (atype == TIPC_ADDR_ID) {
1546 addr->addr.id.node = v1;
1547 addr->addr.id.ref = v2;
1548 } else {
1549 /* Shouldn't happen */
1550 PyErr_SetString(PyExc_TypeError, "Invalid address type");
1551 return 0;
1552 }
Christian Heimes043d6f62008-01-07 17:19:16 +00001553
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001554 *len_ret = sizeof(*addr);
Christian Heimes043d6f62008-01-07 17:19:16 +00001555
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001556 return 1;
1557 }
Christian Heimes043d6f62008-01-07 17:19:16 +00001558#endif
1559
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001560 /* More cases here... */
Guido van Rossum30a685f1991-06-27 15:51:29 +00001561
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001562 default:
1563 PyErr_SetString(socket_error, "getsockaddrarg: bad family");
1564 return 0;
Guido van Rossum30a685f1991-06-27 15:51:29 +00001565
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001566 }
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001567}
1568
Guido van Rossum30a685f1991-06-27 15:51:29 +00001569
Guido van Rossum48a680c2001-03-02 06:34:14 +00001570/* Get the address length according to the socket object's address family.
Guido van Rossum710e1df1992-06-12 10:39:36 +00001571 Return 1 if the family is known, 0 otherwise. The length is returned
1572 through len_ret. */
1573
1574static int
Peter Schneider-Kamp618e25d2000-07-11 23:00:12 +00001575getsockaddrlen(PySocketSockObject *s, socklen_t *len_ret)
Guido van Rossum710e1df1992-06-12 10:39:36 +00001576{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001577 switch (s->sock_family) {
Guido van Rossum710e1df1992-06-12 10:39:36 +00001578
Andrew MacIntyred12dfbb2004-04-04 07:13:49 +00001579#if defined(AF_UNIX)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001580 case AF_UNIX:
1581 {
1582 *len_ret = sizeof (struct sockaddr_un);
1583 return 1;
1584 }
Guido van Rossumb6775db1994-08-01 11:34:53 +00001585#endif /* AF_UNIX */
Martin v. Löwis11017b12006-01-14 18:12:57 +00001586#if defined(AF_NETLINK)
1587 case AF_NETLINK:
1588 {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001589 *len_ret = sizeof (struct sockaddr_nl);
1590 return 1;
Martin v. Löwis11017b12006-01-14 18:12:57 +00001591 }
1592#endif
Guido van Rossum710e1df1992-06-12 10:39:36 +00001593
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001594 case AF_INET:
1595 {
1596 *len_ret = sizeof (struct sockaddr_in);
1597 return 1;
1598 }
Guido van Rossum710e1df1992-06-12 10:39:36 +00001599
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00001600#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001601 case AF_INET6:
1602 {
1603 *len_ret = sizeof (struct sockaddr_in6);
1604 return 1;
1605 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00001606#endif
1607
Hye-Shik Chang81268602004-02-02 06:05:24 +00001608#ifdef USE_BLUETOOTH
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001609 case AF_BLUETOOTH:
1610 {
1611 switch(s->sock_proto)
1612 {
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001613
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001614 case BTPROTO_L2CAP:
1615 *len_ret = sizeof (struct sockaddr_l2);
1616 return 1;
1617 case BTPROTO_RFCOMM:
1618 *len_ret = sizeof (struct sockaddr_rc);
1619 return 1;
1620 case BTPROTO_HCI:
1621 *len_ret = sizeof (struct sockaddr_hci);
1622 return 1;
Hye-Shik Chang81268602004-02-02 06:05:24 +00001623#if !defined(__FreeBSD__)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001624 case BTPROTO_SCO:
1625 *len_ret = sizeof (struct sockaddr_sco);
1626 return 1;
Hye-Shik Chang81268602004-02-02 06:05:24 +00001627#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001628 default:
1629 PyErr_SetString(socket_error, "getsockaddrlen: "
1630 "unknown BT protocol");
1631 return 0;
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001632
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001633 }
1634 }
Martin v. Löwis12af0482004-01-31 12:34:17 +00001635#endif
1636
Martin v. Löwis1ba3fd52001-08-10 20:29:40 +00001637#ifdef HAVE_NETPACKET_PACKET_H
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001638 case AF_PACKET:
1639 {
1640 *len_ret = sizeof (struct sockaddr_ll);
1641 return 1;
1642 }
Jeremy Hylton22308652001-02-02 03:23:09 +00001643#endif
Guido van Rossum48a680c2001-03-02 06:34:14 +00001644
Christian Heimes043d6f62008-01-07 17:19:16 +00001645#ifdef HAVE_LINUX_TIPC_H
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001646 case AF_TIPC:
1647 {
1648 *len_ret = sizeof (struct sockaddr_tipc);
1649 return 1;
1650 }
Christian Heimes043d6f62008-01-07 17:19:16 +00001651#endif
1652
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001653 /* More cases here... */
Guido van Rossum710e1df1992-06-12 10:39:36 +00001654
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001655 default:
1656 PyErr_SetString(socket_error, "getsockaddrlen: bad family");
1657 return 0;
Guido van Rossum710e1df1992-06-12 10:39:36 +00001658
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001659 }
Guido van Rossum710e1df1992-06-12 10:39:36 +00001660}
1661
1662
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00001663/* s._accept() -> (fd, address) */
Guido van Rossum30a685f1991-06-27 15:51:29 +00001664
Guido van Rossum73624e91994-10-10 17:59:00 +00001665static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001666sock_accept(PySocketSockObject *s)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001667{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001668 sock_addr_t addrbuf;
1669 SOCKET_T newfd = INVALID_SOCKET;
1670 socklen_t addrlen;
1671 PyObject *sock = NULL;
1672 PyObject *addr = NULL;
1673 PyObject *res = NULL;
1674 int timeout;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001675 if (!getsockaddrlen(s, &addrlen))
1676 return NULL;
1677 memset(&addrbuf, 0, addrlen);
Guido van Rossum67f7a382002-06-06 21:08:16 +00001678
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001679 if (!IS_SELECTABLE(s))
1680 return select_error();
Neal Norwitz082b2df2006-02-07 07:04:46 +00001681
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00001682 BEGIN_SELECT_LOOP(s)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001683 Py_BEGIN_ALLOW_THREADS
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00001684 timeout = internal_select_ex(s, 0, interval);
Antoine Pitroub1c54962010-10-14 15:05:38 +00001685 if (!timeout) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001686 newfd = accept(s->sock_fd, SAS2SA(&addrbuf), &addrlen);
Antoine Pitroub1c54962010-10-14 15:05:38 +00001687 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001688 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00001689
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001690 if (timeout == 1) {
1691 PyErr_SetString(socket_timeout, "timed out");
1692 return NULL;
1693 }
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00001694 END_SELECT_LOOP(s)
Raymond Hettingeref7343c2003-06-29 03:08:05 +00001695
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001696 if (newfd == INVALID_SOCKET)
1697 return s->errorhandler();
Barry Warsaw752300b1997-01-03 17:18:10 +00001698
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001699 sock = PyLong_FromSocket_t(newfd);
1700 if (sock == NULL) {
1701 SOCKETCLOSE(newfd);
1702 goto finally;
1703 }
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00001704
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001705 addr = makesockaddr(s->sock_fd, SAS2SA(&addrbuf),
1706 addrlen, s->sock_proto);
1707 if (addr == NULL)
1708 goto finally;
Barry Warsaw752300b1997-01-03 17:18:10 +00001709
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001710 res = PyTuple_Pack(2, sock, addr);
Barry Warsaw752300b1997-01-03 17:18:10 +00001711
Guido van Rossum67f7a382002-06-06 21:08:16 +00001712finally:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001713 Py_XDECREF(sock);
1714 Py_XDECREF(addr);
1715 return res;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001716}
1717
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001718PyDoc_STRVAR(accept_doc,
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00001719"_accept() -> (integer, address info)\n\
Guido van Rossum82a5c661998-07-07 20:45:43 +00001720\n\
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00001721Wait for an incoming connection. Return a new socket file descriptor\n\
1722representing the connection, and the address of the client.\n\
1723For IP sockets, the address info is a pair (hostaddr, port).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001724
Guido van Rossum11ba0942002-06-13 15:07:44 +00001725/* s.setblocking(flag) method. Argument:
1726 False -- non-blocking mode; same as settimeout(0)
1727 True -- blocking mode; same as settimeout(None)
1728*/
Guido van Rossume4485b01994-09-07 14:32:49 +00001729
Guido van Rossum73624e91994-10-10 17:59:00 +00001730static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001731sock_setblocking(PySocketSockObject *s, PyObject *arg)
Guido van Rossume4485b01994-09-07 14:32:49 +00001732{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001733 int block;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001734
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001735 block = PyLong_AsLong(arg);
1736 if (block == -1 && PyErr_Occurred())
1737 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001738
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001739 s->sock_timeout = block ? -1.0 : 0.0;
1740 internal_setblocking(s, block);
Guido van Rossume4485b01994-09-07 14:32:49 +00001741
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001742 Py_INCREF(Py_None);
1743 return Py_None;
Guido van Rossume4485b01994-09-07 14:32:49 +00001744}
Guido van Rossume4485b01994-09-07 14:32:49 +00001745
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001746PyDoc_STRVAR(setblocking_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001747"setblocking(flag)\n\
1748\n\
1749Set the socket to blocking (flag is true) or non-blocking (false).\n\
Guido van Rossum11ba0942002-06-13 15:07:44 +00001750setblocking(True) is equivalent to settimeout(None);\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001751setblocking(False) is equivalent to settimeout(0.0).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001752
Guido van Rossum11ba0942002-06-13 15:07:44 +00001753/* s.settimeout(timeout) method. Argument:
1754 None -- no timeout, blocking mode; same as setblocking(True)
1755 0.0 -- non-blocking mode; same as setblocking(False)
1756 > 0 -- timeout mode; operations time out after timeout seconds
1757 < 0 -- illegal; raises an exception
1758*/
Guido van Rossum67f7a382002-06-06 21:08:16 +00001759static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001760sock_settimeout(PySocketSockObject *s, PyObject *arg)
Guido van Rossum67f7a382002-06-06 21:08:16 +00001761{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001762 double timeout;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001763
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001764 if (arg == Py_None)
1765 timeout = -1.0;
1766 else {
1767 timeout = PyFloat_AsDouble(arg);
1768 if (timeout < 0.0) {
1769 if (!PyErr_Occurred())
1770 PyErr_SetString(PyExc_ValueError,
1771 "Timeout value out of range");
1772 return NULL;
1773 }
1774 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00001775
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001776 s->sock_timeout = timeout;
1777 internal_setblocking(s, timeout < 0.0);
Guido van Rossum67f7a382002-06-06 21:08:16 +00001778
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001779 Py_INCREF(Py_None);
1780 return Py_None;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001781}
1782
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001783PyDoc_STRVAR(settimeout_doc,
Guido van Rossum3eede5a2002-06-07 02:08:35 +00001784"settimeout(timeout)\n\
Guido van Rossum67f7a382002-06-06 21:08:16 +00001785\n\
Guido van Rossum11ba0942002-06-13 15:07:44 +00001786Set a timeout on socket operations. 'timeout' can be a float,\n\
1787giving in seconds, or None. Setting a timeout of None disables\n\
1788the timeout feature and is equivalent to setblocking(1).\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001789Setting a timeout of zero is the same as setblocking(0).");
Guido van Rossum67f7a382002-06-06 21:08:16 +00001790
Guido van Rossum3eede5a2002-06-07 02:08:35 +00001791/* s.gettimeout() method.
1792 Returns the timeout associated with a socket. */
Guido van Rossum67f7a382002-06-06 21:08:16 +00001793static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001794sock_gettimeout(PySocketSockObject *s)
Guido van Rossum67f7a382002-06-06 21:08:16 +00001795{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001796 if (s->sock_timeout < 0.0) {
1797 Py_INCREF(Py_None);
1798 return Py_None;
1799 }
1800 else
1801 return PyFloat_FromDouble(s->sock_timeout);
Guido van Rossum67f7a382002-06-06 21:08:16 +00001802}
1803
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001804PyDoc_STRVAR(gettimeout_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00001805"gettimeout() -> timeout\n\
Guido van Rossum67f7a382002-06-06 21:08:16 +00001806\n\
Ezio Melotti388c9452011-08-14 08:28:57 +03001807Returns the timeout in seconds (float) associated with socket \n\
Guido van Rossum67f7a382002-06-06 21:08:16 +00001808operations. A timeout of None indicates that timeouts on socket \n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001809operations are disabled.");
Guido van Rossume4485b01994-09-07 14:32:49 +00001810
Guido van Rossumaee08791992-09-08 09:05:33 +00001811/* s.setsockopt() method.
1812 With an integer third argument, sets an integer option.
1813 With a string third argument, sets an option from a buffer;
1814 use optional built-in module 'struct' to encode the string. */
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001815
Guido van Rossum73624e91994-10-10 17:59:00 +00001816static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001817sock_setsockopt(PySocketSockObject *s, PyObject *args)
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001818{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001819 int level;
1820 int optname;
1821 int res;
1822 char *buf;
1823 int buflen;
1824 int flag;
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001825
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001826 if (PyArg_ParseTuple(args, "iii:setsockopt",
1827 &level, &optname, &flag)) {
1828 buf = (char *) &flag;
1829 buflen = sizeof flag;
1830 }
1831 else {
1832 PyErr_Clear();
1833 if (!PyArg_ParseTuple(args, "iiy#:setsockopt",
1834 &level, &optname, &buf, &buflen))
1835 return NULL;
1836 }
1837 res = setsockopt(s->sock_fd, level, optname, (void *)buf, buflen);
1838 if (res < 0)
1839 return s->errorhandler();
1840 Py_INCREF(Py_None);
1841 return Py_None;
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001842}
1843
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001844PyDoc_STRVAR(setsockopt_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001845"setsockopt(level, option, value)\n\
1846\n\
1847Set a socket option. See the Unix manual for level and option.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001848The value argument can either be an integer or a string.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001849
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001850
Guido van Rossumaee08791992-09-08 09:05:33 +00001851/* s.getsockopt() method.
1852 With two arguments, retrieves an integer option.
1853 With a third integer argument, retrieves a string buffer of that size;
1854 use optional built-in module 'struct' to decode the string. */
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001855
Guido van Rossum73624e91994-10-10 17:59:00 +00001856static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001857sock_getsockopt(PySocketSockObject *s, PyObject *args)
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001858{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001859 int level;
1860 int optname;
1861 int res;
1862 PyObject *buf;
1863 socklen_t buflen = 0;
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001864
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001865 if (!PyArg_ParseTuple(args, "ii|i:getsockopt",
1866 &level, &optname, &buflen))
1867 return NULL;
Guido van Rossum48a680c2001-03-02 06:34:14 +00001868
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001869 if (buflen == 0) {
1870 int flag = 0;
1871 socklen_t flagsize = sizeof flag;
1872 res = getsockopt(s->sock_fd, level, optname,
1873 (void *)&flag, &flagsize);
1874 if (res < 0)
1875 return s->errorhandler();
1876 return PyLong_FromLong(flag);
1877 }
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001878#ifdef __VMS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001879 /* socklen_t is unsigned so no negative test is needed,
1880 test buflen == 0 is previously done */
1881 if (buflen > 1024) {
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001882#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001883 if (buflen <= 0 || buflen > 1024) {
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001884#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001885 PyErr_SetString(socket_error,
1886 "getsockopt buflen out of range");
1887 return NULL;
1888 }
1889 buf = PyBytes_FromStringAndSize((char *)NULL, buflen);
1890 if (buf == NULL)
1891 return NULL;
1892 res = getsockopt(s->sock_fd, level, optname,
1893 (void *)PyBytes_AS_STRING(buf), &buflen);
1894 if (res < 0) {
1895 Py_DECREF(buf);
1896 return s->errorhandler();
1897 }
1898 _PyBytes_Resize(&buf, buflen);
1899 return buf;
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001900}
1901
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001902PyDoc_STRVAR(getsockopt_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001903"getsockopt(level, option[, buffersize]) -> value\n\
1904\n\
1905Get a socket option. See the Unix manual for level and option.\n\
1906If a nonzero buffersize argument is given, the return value is a\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001907string of that length; otherwise it is an integer.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001908
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001909
Fred Drake728819a2000-07-01 03:40:12 +00001910/* s.bind(sockaddr) method */
Guido van Rossum30a685f1991-06-27 15:51:29 +00001911
Guido van Rossum73624e91994-10-10 17:59:00 +00001912static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001913sock_bind(PySocketSockObject *s, PyObject *addro)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001914{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001915 sock_addr_t addrbuf;
1916 int addrlen;
1917 int res;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00001918
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001919 if (!getsockaddrarg(s, addro, SAS2SA(&addrbuf), &addrlen))
1920 return NULL;
1921 Py_BEGIN_ALLOW_THREADS
1922 res = bind(s->sock_fd, SAS2SA(&addrbuf), addrlen);
1923 Py_END_ALLOW_THREADS
1924 if (res < 0)
1925 return s->errorhandler();
1926 Py_INCREF(Py_None);
1927 return Py_None;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001928}
1929
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001930PyDoc_STRVAR(bind_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001931"bind(address)\n\
1932\n\
1933Bind the socket to a local address. For IP sockets, the address is a\n\
Jeremy Hylton22308652001-02-02 03:23:09 +00001934pair (host, port); the host must refer to the local host. For raw packet\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001935sockets the address is a tuple (ifname, proto [,pkttype [,hatype]])");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001936
Guido van Rossum30a685f1991-06-27 15:51:29 +00001937
1938/* s.close() method.
1939 Set the file descriptor to -1 so operations tried subsequently
1940 will surely fail. */
1941
Guido van Rossum73624e91994-10-10 17:59:00 +00001942static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001943sock_close(PySocketSockObject *s)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001944{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001945 SOCKET_T fd;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00001946
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001947 if ((fd = s->sock_fd) != -1) {
1948 s->sock_fd = -1;
1949 Py_BEGIN_ALLOW_THREADS
1950 (void) SOCKETCLOSE(fd);
1951 Py_END_ALLOW_THREADS
1952 }
1953 Py_INCREF(Py_None);
1954 return Py_None;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001955}
1956
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001957PyDoc_STRVAR(close_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001958"close()\n\
1959\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001960Close the socket. It cannot be used after this call.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001961
Antoine Pitroue43f9d02010-08-08 23:24:50 +00001962static PyObject *
Antoine Pitrou6e451df2010-08-09 20:39:54 +00001963sock_detach(PySocketSockObject *s)
Antoine Pitroue43f9d02010-08-08 23:24:50 +00001964{
Antoine Pitrou6e451df2010-08-09 20:39:54 +00001965 SOCKET_T fd = s->sock_fd;
Antoine Pitroue43f9d02010-08-08 23:24:50 +00001966 s->sock_fd = -1;
Antoine Pitrou6e451df2010-08-09 20:39:54 +00001967 return PyLong_FromSocket_t(fd);
Antoine Pitroue43f9d02010-08-08 23:24:50 +00001968}
1969
Antoine Pitrou6e451df2010-08-09 20:39:54 +00001970PyDoc_STRVAR(detach_doc,
1971"detach()\n\
Antoine Pitroue43f9d02010-08-08 23:24:50 +00001972\n\
1973Close the socket object without closing the underlying file descriptor.\
1974The object cannot be used after this call, but the file descriptor\
Antoine Pitrou6e451df2010-08-09 20:39:54 +00001975can be reused for other purposes. The file descriptor is returned.");
Antoine Pitroue43f9d02010-08-08 23:24:50 +00001976
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001977static int
Raymond Hettingeref7343c2003-06-29 03:08:05 +00001978internal_connect(PySocketSockObject *s, struct sockaddr *addr, int addrlen,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001979 int *timeoutp)
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001980{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001981 int res, timeout;
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001982
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001983 timeout = 0;
1984 res = connect(s->sock_fd, addr, addrlen);
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001985
1986#ifdef MS_WINDOWS
1987
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001988 if (s->sock_timeout > 0.0) {
1989 if (res < 0 && WSAGetLastError() == WSAEWOULDBLOCK &&
1990 IS_SELECTABLE(s)) {
1991 /* This is a mess. Best solution: trust select */
1992 fd_set fds;
1993 fd_set fds_exc;
1994 struct timeval tv;
1995 tv.tv_sec = (int)s->sock_timeout;
1996 tv.tv_usec = (int)((s->sock_timeout - tv.tv_sec) * 1e6);
1997 FD_ZERO(&fds);
1998 FD_SET(s->sock_fd, &fds);
1999 FD_ZERO(&fds_exc);
2000 FD_SET(s->sock_fd, &fds_exc);
Antoine Pitrou19467d22010-08-17 19:33:30 +00002001 res = select(Py_SAFE_DOWNCAST(s->sock_fd+1, SOCKET_T, int),
2002 NULL, &fds, &fds_exc, &tv);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002003 if (res == 0) {
2004 res = WSAEWOULDBLOCK;
2005 timeout = 1;
2006 } else if (res > 0) {
2007 if (FD_ISSET(s->sock_fd, &fds))
2008 /* The socket is in the writable set - this
2009 means connected */
2010 res = 0;
2011 else {
2012 /* As per MS docs, we need to call getsockopt()
2013 to get the underlying error */
2014 int res_size = sizeof res;
2015 /* It must be in the exception set */
2016 assert(FD_ISSET(s->sock_fd, &fds_exc));
2017 if (0 == getsockopt(s->sock_fd, SOL_SOCKET, SO_ERROR,
2018 (char *)&res, &res_size))
2019 /* getsockopt also clears WSAGetLastError,
2020 so reset it back. */
2021 WSASetLastError(res);
2022 else
2023 res = WSAGetLastError();
2024 }
2025 }
2026 /* else if (res < 0) an error occurred */
2027 }
2028 }
Guido van Rossum7b8bac12002-06-13 16:07:04 +00002029
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002030 if (res < 0)
2031 res = WSAGetLastError();
Guido van Rossum7b8bac12002-06-13 16:07:04 +00002032
2033#else
2034
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002035 if (s->sock_timeout > 0.0) {
2036 if (res < 0 && errno == EINPROGRESS && IS_SELECTABLE(s)) {
2037 timeout = internal_select(s, 1);
2038 if (timeout == 0) {
2039 /* Bug #1019808: in case of an EINPROGRESS,
2040 use getsockopt(SO_ERROR) to get the real
2041 error. */
2042 socklen_t res_size = sizeof res;
2043 (void)getsockopt(s->sock_fd, SOL_SOCKET,
2044 SO_ERROR, &res, &res_size);
2045 if (res == EISCONN)
2046 res = 0;
2047 errno = res;
2048 }
2049 else if (timeout == -1) {
2050 res = errno; /* had error */
2051 }
2052 else
2053 res = EWOULDBLOCK; /* timed out */
2054 }
2055 }
Guido van Rossum7b8bac12002-06-13 16:07:04 +00002056
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002057 if (res < 0)
2058 res = errno;
Guido van Rossum7b8bac12002-06-13 16:07:04 +00002059
2060#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002061 *timeoutp = timeout;
Guido van Rossum7b8bac12002-06-13 16:07:04 +00002062
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002063 return res;
Guido van Rossum7b8bac12002-06-13 16:07:04 +00002064}
Guido van Rossum30a685f1991-06-27 15:51:29 +00002065
Fred Drake728819a2000-07-01 03:40:12 +00002066/* s.connect(sockaddr) method */
Guido van Rossum30a685f1991-06-27 15:51:29 +00002067
Guido van Rossum73624e91994-10-10 17:59:00 +00002068static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002069sock_connect(PySocketSockObject *s, PyObject *addro)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002070{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002071 sock_addr_t addrbuf;
2072 int addrlen;
2073 int res;
2074 int timeout;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00002075
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002076 if (!getsockaddrarg(s, addro, SAS2SA(&addrbuf), &addrlen))
2077 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002078
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002079 Py_BEGIN_ALLOW_THREADS
2080 res = internal_connect(s, SAS2SA(&addrbuf), addrlen, &timeout);
2081 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00002082
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002083 if (timeout == 1) {
2084 PyErr_SetString(socket_timeout, "timed out");
2085 return NULL;
2086 }
2087 if (res != 0)
2088 return s->errorhandler();
2089 Py_INCREF(Py_None);
2090 return Py_None;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002091}
2092
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002093PyDoc_STRVAR(connect_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002094"connect(address)\n\
2095\n\
2096Connect the socket to a remote address. For IP sockets, the address\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002097is a pair (host, port).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002098
Guido van Rossum30a685f1991-06-27 15:51:29 +00002099
Fred Drake728819a2000-07-01 03:40:12 +00002100/* s.connect_ex(sockaddr) method */
Guido van Rossumfc4255d1997-11-19 18:57:13 +00002101
2102static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002103sock_connect_ex(PySocketSockObject *s, PyObject *addro)
Guido van Rossumfc4255d1997-11-19 18:57:13 +00002104{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002105 sock_addr_t addrbuf;
2106 int addrlen;
2107 int res;
2108 int timeout;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00002109
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002110 if (!getsockaddrarg(s, addro, SAS2SA(&addrbuf), &addrlen))
2111 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002112
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002113 Py_BEGIN_ALLOW_THREADS
2114 res = internal_connect(s, SAS2SA(&addrbuf), addrlen, &timeout);
2115 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00002116
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002117 /* Signals are not errors (though they may raise exceptions). Adapted
2118 from PyErr_SetFromErrnoWithFilenameObject(). */
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002119#ifdef EINTR
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002120 if (res == EINTR && PyErr_CheckSignals())
2121 return NULL;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002122#endif
2123
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002124 return PyLong_FromLong((long) res);
Guido van Rossumfc4255d1997-11-19 18:57:13 +00002125}
2126
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002127PyDoc_STRVAR(connect_ex_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00002128"connect_ex(address) -> errno\n\
Guido van Rossum82a5c661998-07-07 20:45:43 +00002129\n\
2130This is like connect(address), but returns an error code (the errno value)\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002131instead of raising an exception when an error occurs.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002132
Guido van Rossumfc4255d1997-11-19 18:57:13 +00002133
Guido van Rossumed233a51992-06-23 09:07:03 +00002134/* s.fileno() method */
2135
Guido van Rossum73624e91994-10-10 17:59:00 +00002136static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002137sock_fileno(PySocketSockObject *s)
Guido van Rossumed233a51992-06-23 09:07:03 +00002138{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002139 return PyLong_FromSocket_t(s->sock_fd);
Guido van Rossumed233a51992-06-23 09:07:03 +00002140}
2141
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002142PyDoc_STRVAR(fileno_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002143"fileno() -> integer\n\
2144\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002145Return the integer file descriptor of the socket.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002146
Guido van Rossumed233a51992-06-23 09:07:03 +00002147
Guido van Rossumc89705d1992-11-26 08:54:07 +00002148/* s.getsockname() method */
2149
Guido van Rossum73624e91994-10-10 17:59:00 +00002150static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002151sock_getsockname(PySocketSockObject *s)
Guido van Rossumc89705d1992-11-26 08:54:07 +00002152{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002153 sock_addr_t addrbuf;
2154 int res;
2155 socklen_t addrlen;
Guido van Rossumff3ab422000-04-24 15:16:03 +00002156
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002157 if (!getsockaddrlen(s, &addrlen))
2158 return NULL;
2159 memset(&addrbuf, 0, addrlen);
2160 Py_BEGIN_ALLOW_THREADS
2161 res = getsockname(s->sock_fd, SAS2SA(&addrbuf), &addrlen);
2162 Py_END_ALLOW_THREADS
2163 if (res < 0)
2164 return s->errorhandler();
2165 return makesockaddr(s->sock_fd, SAS2SA(&addrbuf), addrlen,
2166 s->sock_proto);
Guido van Rossumc89705d1992-11-26 08:54:07 +00002167}
2168
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002169PyDoc_STRVAR(getsockname_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002170"getsockname() -> address info\n\
2171\n\
2172Return the address of the local endpoint. For IP sockets, the address\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002173info is a pair (hostaddr, port).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002174
Guido van Rossumc89705d1992-11-26 08:54:07 +00002175
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002176#ifdef HAVE_GETPEERNAME /* Cray APP doesn't have this :-( */
Guido van Rossumc89705d1992-11-26 08:54:07 +00002177/* s.getpeername() method */
2178
Guido van Rossum73624e91994-10-10 17:59:00 +00002179static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002180sock_getpeername(PySocketSockObject *s)
Guido van Rossumc89705d1992-11-26 08:54:07 +00002181{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002182 sock_addr_t addrbuf;
2183 int res;
2184 socklen_t addrlen;
Guido van Rossumff3ab422000-04-24 15:16:03 +00002185
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002186 if (!getsockaddrlen(s, &addrlen))
2187 return NULL;
2188 memset(&addrbuf, 0, addrlen);
2189 Py_BEGIN_ALLOW_THREADS
2190 res = getpeername(s->sock_fd, SAS2SA(&addrbuf), &addrlen);
2191 Py_END_ALLOW_THREADS
2192 if (res < 0)
2193 return s->errorhandler();
2194 return makesockaddr(s->sock_fd, SAS2SA(&addrbuf), addrlen,
2195 s->sock_proto);
Guido van Rossumc89705d1992-11-26 08:54:07 +00002196}
Guido van Rossum82a5c661998-07-07 20:45:43 +00002197
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002198PyDoc_STRVAR(getpeername_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002199"getpeername() -> address info\n\
2200\n\
2201Return the address of the remote endpoint. For IP sockets, the address\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002202info is a pair (hostaddr, port).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002203
Guido van Rossumb6775db1994-08-01 11:34:53 +00002204#endif /* HAVE_GETPEERNAME */
Guido van Rossumc89705d1992-11-26 08:54:07 +00002205
2206
Guido van Rossum30a685f1991-06-27 15:51:29 +00002207/* s.listen(n) method */
2208
Guido van Rossum73624e91994-10-10 17:59:00 +00002209static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002210sock_listen(PySocketSockObject *s, PyObject *arg)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002211{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002212 int backlog;
2213 int res;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00002214
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002215 backlog = PyLong_AsLong(arg);
2216 if (backlog == -1 && PyErr_Occurred())
2217 return NULL;
2218 Py_BEGIN_ALLOW_THREADS
Antoine Pitrou1be815a2011-05-10 19:16:29 +02002219 /* To avoid problems on systems that don't allow a negative backlog
2220 * (which doesn't make sense anyway) we force a minimum value of 0. */
2221 if (backlog < 0)
2222 backlog = 0;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002223 res = listen(s->sock_fd, backlog);
2224 Py_END_ALLOW_THREADS
2225 if (res < 0)
2226 return s->errorhandler();
2227 Py_INCREF(Py_None);
2228 return Py_None;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002229}
2230
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002231PyDoc_STRVAR(listen_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002232"listen(backlog)\n\
2233\n\
2234Enable a server to accept connections. The backlog argument must be at\n\
Antoine Pitrou1be815a2011-05-10 19:16:29 +02002235least 0 (if it is lower, it is set to 0); it specifies the number of\n\
2236unaccepted connections that the system will allow before refusing new\n\
2237connections.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002238
2239
Thomas Wouters477c8d52006-05-27 19:21:47 +00002240/*
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002241 * This is the guts of the recv() and recv_into() methods, which reads into a
Thomas Wouters902d6eb2007-01-09 23:18:33 +00002242 * char buffer. If you have any inc/dec ref to do to the objects that contain
Thomas Wouters477c8d52006-05-27 19:21:47 +00002243 * the buffer, do it in the caller. This function returns the number of bytes
Ezio Melotti13925002011-03-16 11:05:33 +02002244 * successfully read. If there was an error, it returns -1. Note that it is
Thomas Wouters477c8d52006-05-27 19:21:47 +00002245 * also possible that we return a number of bytes smaller than the request
2246 * bytes.
2247 */
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002248
Antoine Pitrou19467d22010-08-17 19:33:30 +00002249static Py_ssize_t
2250sock_recv_guts(PySocketSockObject *s, char* cbuf, Py_ssize_t len, int flags)
Thomas Wouters477c8d52006-05-27 19:21:47 +00002251{
Antoine Pitrou19467d22010-08-17 19:33:30 +00002252 Py_ssize_t outlen = -1;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002253 int timeout;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002254#ifdef __VMS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002255 int remaining;
2256 char *read_buf;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002257#endif
2258
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002259 if (!IS_SELECTABLE(s)) {
2260 select_error();
2261 return -1;
2262 }
2263 if (len == 0) {
2264 /* If 0 bytes were requested, do nothing. */
2265 return 0;
2266 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002267
2268#ifndef __VMS
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002269 BEGIN_SELECT_LOOP(s)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002270 Py_BEGIN_ALLOW_THREADS
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002271 timeout = internal_select_ex(s, 0, interval);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002272 if (!timeout)
2273 outlen = recv(s->sock_fd, cbuf, len, flags);
2274 Py_END_ALLOW_THREADS
Thomas Wouters477c8d52006-05-27 19:21:47 +00002275
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002276 if (timeout == 1) {
2277 PyErr_SetString(socket_timeout, "timed out");
2278 return -1;
2279 }
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002280 END_SELECT_LOOP(s)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002281 if (outlen < 0) {
2282 /* Note: the call to errorhandler() ALWAYS indirectly returned
2283 NULL, so ignore its return value */
2284 s->errorhandler();
2285 return -1;
2286 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002287#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002288 read_buf = cbuf;
2289 remaining = len;
2290 while (remaining != 0) {
2291 unsigned int segment;
2292 int nread = -1;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002293
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002294 segment = remaining /SEGMENT_SIZE;
2295 if (segment != 0) {
2296 segment = SEGMENT_SIZE;
2297 }
2298 else {
2299 segment = remaining;
2300 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002301
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002302 BEGIN_SELECT_LOOP(s)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002303 Py_BEGIN_ALLOW_THREADS
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002304 timeout = internal_select_ex(s, 0, interval);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002305 if (!timeout)
2306 nread = recv(s->sock_fd, read_buf, segment, flags);
2307 Py_END_ALLOW_THREADS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002308 if (timeout == 1) {
2309 PyErr_SetString(socket_timeout, "timed out");
2310 return -1;
2311 }
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002312 END_SELECT_LOOP(s)
2313
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002314 if (nread < 0) {
2315 s->errorhandler();
2316 return -1;
2317 }
2318 if (nread != remaining) {
2319 read_buf += nread;
2320 break;
2321 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002322
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002323 remaining -= segment;
2324 read_buf += segment;
2325 }
2326 outlen = read_buf - cbuf;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002327#endif /* !__VMS */
2328
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002329 return outlen;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002330}
2331
Guido van Rossum48a680c2001-03-02 06:34:14 +00002332
Guido van Rossumeb6b33a1993-05-25 09:38:27 +00002333/* s.recv(nbytes [,flags]) method */
Guido van Rossum30a685f1991-06-27 15:51:29 +00002334
Guido van Rossum73624e91994-10-10 17:59:00 +00002335static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002336sock_recv(PySocketSockObject *s, PyObject *args)
Guido van Rossum30a685f1991-06-27 15:51:29 +00002337{
Antoine Pitrou19467d22010-08-17 19:33:30 +00002338 Py_ssize_t recvlen, outlen;
2339 int flags = 0;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002340 PyObject *buf;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002341
Antoine Pitrou19467d22010-08-17 19:33:30 +00002342 if (!PyArg_ParseTuple(args, "n|i:recv", &recvlen, &flags))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002343 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002344
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002345 if (recvlen < 0) {
2346 PyErr_SetString(PyExc_ValueError,
2347 "negative buffersize in recv");
2348 return NULL;
2349 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00002350
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002351 /* Allocate a new string. */
2352 buf = PyBytes_FromStringAndSize((char *) 0, recvlen);
2353 if (buf == NULL)
2354 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002355
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002356 /* Call the guts */
2357 outlen = sock_recv_guts(s, PyBytes_AS_STRING(buf), recvlen, flags);
2358 if (outlen < 0) {
2359 /* An error occurred, release the string and return an
2360 error. */
2361 Py_DECREF(buf);
2362 return NULL;
2363 }
2364 if (outlen != recvlen) {
2365 /* We did not read as many bytes as we anticipated, resize the
2366 string if possible and be successful. */
2367 _PyBytes_Resize(&buf, outlen);
2368 }
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00002369
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002370 return buf;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002371}
2372
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002373PyDoc_STRVAR(recv_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002374"recv(buffersize[, flags]) -> data\n\
2375\n\
2376Receive up to buffersize bytes from the socket. For the optional flags\n\
2377argument, see the Unix manual. When no data is available, block until\n\
2378at least one byte is available or until the remote end is closed. When\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002379the remote end is closed and all data is read, return the empty string.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002380
Guido van Rossum30a685f1991-06-27 15:51:29 +00002381
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002382/* s.recv_into(buffer, [nbytes [,flags]]) method */
Guido van Rossum30a685f1991-06-27 15:51:29 +00002383
Thomas Wouters477c8d52006-05-27 19:21:47 +00002384static PyObject*
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002385sock_recv_into(PySocketSockObject *s, PyObject *args, PyObject *kwds)
Thomas Wouters477c8d52006-05-27 19:21:47 +00002386{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002387 static char *kwlist[] = {"buffer", "nbytes", "flags", 0};
Thomas Wouters477c8d52006-05-27 19:21:47 +00002388
Antoine Pitrou19467d22010-08-17 19:33:30 +00002389 int flags = 0;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002390 Py_buffer pbuf;
2391 char *buf;
Antoine Pitrou19467d22010-08-17 19:33:30 +00002392 Py_ssize_t buflen, readlen, recvlen = 0;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002393
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002394 /* Get the buffer's memory */
Antoine Pitrou19467d22010-08-17 19:33:30 +00002395 if (!PyArg_ParseTupleAndKeywords(args, kwds, "w*|ni:recv_into", kwlist,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002396 &pbuf, &recvlen, &flags))
2397 return NULL;
2398 buf = pbuf.buf;
2399 buflen = pbuf.len;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002400
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002401 if (recvlen < 0) {
2402 PyBuffer_Release(&pbuf);
2403 PyErr_SetString(PyExc_ValueError,
2404 "negative buffersize in recv_into");
2405 return NULL;
2406 }
2407 if (recvlen == 0) {
2408 /* If nbytes was not specified, use the buffer's length */
2409 recvlen = buflen;
2410 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002411
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002412 /* Check if the buffer is large enough */
2413 if (buflen < recvlen) {
2414 PyBuffer_Release(&pbuf);
2415 PyErr_SetString(PyExc_ValueError,
2416 "buffer too small for requested bytes");
2417 return NULL;
2418 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002419
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002420 /* Call the guts */
2421 readlen = sock_recv_guts(s, buf, recvlen, flags);
2422 if (readlen < 0) {
2423 /* Return an error. */
2424 PyBuffer_Release(&pbuf);
2425 return NULL;
2426 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002427
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002428 PyBuffer_Release(&pbuf);
2429 /* Return the number of bytes read. Note that we do not do anything
2430 special here in the case that readlen < recvlen. */
2431 return PyLong_FromSsize_t(readlen);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002432}
2433
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002434PyDoc_STRVAR(recv_into_doc,
2435"recv_into(buffer, [nbytes[, flags]]) -> nbytes_read\n\
Thomas Wouters477c8d52006-05-27 19:21:47 +00002436\n\
2437A version of recv() that stores its data into a buffer rather than creating \n\
2438a new string. Receive up to buffersize bytes from the socket. If buffersize \n\
2439is not specified (or 0), receive up to the size available in the given buffer.\n\
2440\n\
2441See recv() for documentation about the flags.");
2442
2443
2444/*
Christian Heimes99170a52007-12-19 02:07:34 +00002445 * This is the guts of the recvfrom() and recvfrom_into() methods, which reads
2446 * into a char buffer. If you have any inc/def ref to do to the objects that
2447 * contain the buffer, do it in the caller. This function returns the number
Ezio Melotti13925002011-03-16 11:05:33 +02002448 * of bytes successfully read. If there was an error, it returns -1. Note
Christian Heimes99170a52007-12-19 02:07:34 +00002449 * that it is also possible that we return a number of bytes smaller than the
2450 * request bytes.
Thomas Wouters477c8d52006-05-27 19:21:47 +00002451 *
2452 * 'addr' is a return value for the address object. Note that you must decref
2453 * it yourself.
2454 */
Antoine Pitrou19467d22010-08-17 19:33:30 +00002455static Py_ssize_t
2456sock_recvfrom_guts(PySocketSockObject *s, char* cbuf, Py_ssize_t len, int flags,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002457 PyObject** addr)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002458{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002459 sock_addr_t addrbuf;
2460 int timeout;
Antoine Pitrou19467d22010-08-17 19:33:30 +00002461 Py_ssize_t n = -1;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002462 socklen_t addrlen;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002463
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002464 *addr = NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002465
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002466 if (!getsockaddrlen(s, &addrlen))
2467 return -1;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002468
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002469 if (!IS_SELECTABLE(s)) {
2470 select_error();
2471 return -1;
2472 }
Neal Norwitz082b2df2006-02-07 07:04:46 +00002473
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002474 BEGIN_SELECT_LOOP(s)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002475 Py_BEGIN_ALLOW_THREADS
2476 memset(&addrbuf, 0, addrlen);
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002477 timeout = internal_select_ex(s, 0, interval);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002478 if (!timeout) {
Guido van Rossum8d665e61996-06-26 18:22:49 +00002479#ifndef MS_WINDOWS
Andrew MacIntyreba43e872002-03-03 03:03:52 +00002480#if defined(PYOS_OS2) && !defined(PYCC_GCC)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002481 n = recvfrom(s->sock_fd, cbuf, len, flags,
2482 SAS2SA(&addrbuf), &addrlen);
Guido van Rossum32c575d1997-12-02 20:37:32 +00002483#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002484 n = recvfrom(s->sock_fd, cbuf, len, flags,
2485 (void *) &addrbuf, &addrlen);
Guido van Rossum32c575d1997-12-02 20:37:32 +00002486#endif
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00002487#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002488 n = recvfrom(s->sock_fd, cbuf, len, flags,
2489 SAS2SA(&addrbuf), &addrlen);
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00002490#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002491 }
2492 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00002493
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002494 if (timeout == 1) {
2495 PyErr_SetString(socket_timeout, "timed out");
2496 return -1;
2497 }
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002498 END_SELECT_LOOP(s)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002499 if (n < 0) {
2500 s->errorhandler();
2501 return -1;
2502 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00002503
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002504 if (!(*addr = makesockaddr(s->sock_fd, SAS2SA(&addrbuf),
2505 addrlen, s->sock_proto)))
2506 return -1;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002507
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002508 return n;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002509}
2510
2511/* s.recvfrom(nbytes [,flags]) method */
2512
2513static PyObject *
2514sock_recvfrom(PySocketSockObject *s, PyObject *args)
2515{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002516 PyObject *buf = NULL;
2517 PyObject *addr = NULL;
2518 PyObject *ret = NULL;
Antoine Pitrou19467d22010-08-17 19:33:30 +00002519 int flags = 0;
2520 Py_ssize_t recvlen, outlen;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002521
Antoine Pitrou19467d22010-08-17 19:33:30 +00002522 if (!PyArg_ParseTuple(args, "n|i:recvfrom", &recvlen, &flags))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002523 return NULL;
Guido van Rossum48a680c2001-03-02 06:34:14 +00002524
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002525 if (recvlen < 0) {
2526 PyErr_SetString(PyExc_ValueError,
2527 "negative buffersize in recvfrom");
2528 return NULL;
2529 }
Guido van Rossumd8faa362007-04-27 19:54:29 +00002530
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002531 buf = PyBytes_FromStringAndSize((char *) 0, recvlen);
2532 if (buf == NULL)
2533 return NULL;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002534
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002535 outlen = sock_recvfrom_guts(s, PyBytes_AS_STRING(buf),
2536 recvlen, flags, &addr);
2537 if (outlen < 0) {
2538 goto finally;
2539 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002540
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002541 if (outlen != recvlen) {
2542 /* We did not read as many bytes as we anticipated, resize the
Ezio Melotti13925002011-03-16 11:05:33 +02002543 string if possible and be successful. */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002544 if (_PyBytes_Resize(&buf, outlen) < 0)
Ezio Melotti13925002011-03-16 11:05:33 +02002545 /* Oopsy, not so successful after all. */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002546 goto finally;
2547 }
Barry Warsaw752300b1997-01-03 17:18:10 +00002548
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002549 ret = PyTuple_Pack(2, buf, addr);
Guido van Rossum67f7a382002-06-06 21:08:16 +00002550
2551finally:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002552 Py_XDECREF(buf);
2553 Py_XDECREF(addr);
2554 return ret;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002555}
2556
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002557PyDoc_STRVAR(recvfrom_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002558"recvfrom(buffersize[, flags]) -> (data, address info)\n\
2559\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002560Like recv(buffersize, flags) but also return the sender's address info.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002561
Thomas Wouters477c8d52006-05-27 19:21:47 +00002562
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002563/* s.recvfrom_into(buffer[, nbytes [,flags]]) method */
Thomas Wouters477c8d52006-05-27 19:21:47 +00002564
2565static PyObject *
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002566sock_recvfrom_into(PySocketSockObject *s, PyObject *args, PyObject* kwds)
Thomas Wouters477c8d52006-05-27 19:21:47 +00002567{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002568 static char *kwlist[] = {"buffer", "nbytes", "flags", 0};
Thomas Wouters477c8d52006-05-27 19:21:47 +00002569
Antoine Pitrou19467d22010-08-17 19:33:30 +00002570 int flags = 0;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002571 Py_buffer pbuf;
2572 char *buf;
Antoine Pitrou19467d22010-08-17 19:33:30 +00002573 Py_ssize_t readlen, buflen, recvlen = 0;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002574
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002575 PyObject *addr = NULL;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002576
Antoine Pitrou19467d22010-08-17 19:33:30 +00002577 if (!PyArg_ParseTupleAndKeywords(args, kwds, "w*|ni:recvfrom_into",
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002578 kwlist, &pbuf,
2579 &recvlen, &flags))
2580 return NULL;
2581 buf = pbuf.buf;
2582 buflen = pbuf.len;
2583 assert(buf != 0 && buflen > 0);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002584
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002585 if (recvlen < 0) {
2586 PyBuffer_Release(&pbuf);
2587 PyErr_SetString(PyExc_ValueError,
2588 "negative buffersize in recvfrom_into");
2589 return NULL;
2590 }
2591 if (recvlen == 0) {
2592 /* If nbytes was not specified, use the buffer's length */
2593 recvlen = buflen;
2594 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002595
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002596 readlen = sock_recvfrom_guts(s, buf, recvlen, flags, &addr);
2597 if (readlen < 0) {
2598 PyBuffer_Release(&pbuf);
2599 /* Return an error */
2600 Py_XDECREF(addr);
2601 return NULL;
2602 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002603
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002604 PyBuffer_Release(&pbuf);
2605 /* Return the number of bytes read and the address. Note that we do
2606 not do anything special here in the case that readlen < recvlen. */
Antoine Pitrou19467d22010-08-17 19:33:30 +00002607 return Py_BuildValue("nN", readlen, addr);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002608}
2609
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002610PyDoc_STRVAR(recvfrom_into_doc,
2611"recvfrom_into(buffer[, nbytes[, flags]]) -> (nbytes, address info)\n\
Thomas Wouters477c8d52006-05-27 19:21:47 +00002612\n\
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002613Like recv_into(buffer[, nbytes[, flags]]) but also return the sender's address info.");
Thomas Wouters477c8d52006-05-27 19:21:47 +00002614
2615
Guido van Rossumeb6b33a1993-05-25 09:38:27 +00002616/* s.send(data [,flags]) method */
Guido van Rossum30a685f1991-06-27 15:51:29 +00002617
Guido van Rossum73624e91994-10-10 17:59:00 +00002618static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002619sock_send(PySocketSockObject *s, PyObject *args)
Guido van Rossum30a685f1991-06-27 15:51:29 +00002620{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002621 char *buf;
Antoine Pitrou19467d22010-08-17 19:33:30 +00002622 Py_ssize_t len, n = -1;
2623 int flags = 0, timeout;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002624 Py_buffer pbuf;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002625
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002626 if (!PyArg_ParseTuple(args, "y*|i:send", &pbuf, &flags))
2627 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002628
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002629 if (!IS_SELECTABLE(s)) {
2630 PyBuffer_Release(&pbuf);
2631 return select_error();
2632 }
2633 buf = pbuf.buf;
2634 len = pbuf.len;
Neal Norwitz082b2df2006-02-07 07:04:46 +00002635
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002636 BEGIN_SELECT_LOOP(s)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002637 Py_BEGIN_ALLOW_THREADS
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002638 timeout = internal_select_ex(s, 1, interval);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002639 if (!timeout)
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002640#ifdef __VMS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002641 n = sendsegmented(s->sock_fd, buf, len, flags);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002642#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002643 n = send(s->sock_fd, buf, len, flags);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002644#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002645 Py_END_ALLOW_THREADS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002646 if (timeout == 1) {
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002647 PyBuffer_Release(&pbuf);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002648 PyErr_SetString(socket_timeout, "timed out");
2649 return NULL;
2650 }
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002651 END_SELECT_LOOP(s)
2652
2653 PyBuffer_Release(&pbuf);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002654 if (n < 0)
2655 return s->errorhandler();
Antoine Pitrou19467d22010-08-17 19:33:30 +00002656 return PyLong_FromSsize_t(n);
Guido van Rossum30a685f1991-06-27 15:51:29 +00002657}
2658
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002659PyDoc_STRVAR(send_doc,
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002660"send(data[, flags]) -> count\n\
Guido van Rossum82a5c661998-07-07 20:45:43 +00002661\n\
2662Send a data string to the socket. For the optional flags\n\
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002663argument, see the Unix manual. Return the number of bytes\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002664sent; this may be less than len(data) if the network is busy.");
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002665
2666
2667/* s.sendall(data [,flags]) method */
2668
2669static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002670sock_sendall(PySocketSockObject *s, PyObject *args)
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002671{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002672 char *buf;
Antoine Pitrou19467d22010-08-17 19:33:30 +00002673 Py_ssize_t len, n = -1;
Antoine Pitrou6d7df632010-09-27 17:52:25 +00002674 int flags = 0, timeout, saved_errno;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002675 Py_buffer pbuf;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002676
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002677 if (!PyArg_ParseTuple(args, "y*|i:sendall", &pbuf, &flags))
2678 return NULL;
2679 buf = pbuf.buf;
2680 len = pbuf.len;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002681
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002682 if (!IS_SELECTABLE(s)) {
2683 PyBuffer_Release(&pbuf);
2684 return select_error();
2685 }
Neal Norwitz082b2df2006-02-07 07:04:46 +00002686
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002687 do {
Antoine Pitrou6d7df632010-09-27 17:52:25 +00002688 Py_BEGIN_ALLOW_THREADS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002689 timeout = internal_select(s, 1);
2690 n = -1;
Antoine Pitrou6d7df632010-09-27 17:52:25 +00002691 if (!timeout) {
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002692#ifdef __VMS
Antoine Pitrou6d7df632010-09-27 17:52:25 +00002693 n = sendsegmented(s->sock_fd, buf, len, flags);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002694#else
Antoine Pitrou6d7df632010-09-27 17:52:25 +00002695 n = send(s->sock_fd, buf, len, flags);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002696#endif
Antoine Pitrou6d7df632010-09-27 17:52:25 +00002697 }
2698 Py_END_ALLOW_THREADS
2699 if (timeout == 1) {
2700 PyBuffer_Release(&pbuf);
2701 PyErr_SetString(socket_timeout, "timed out");
2702 return NULL;
2703 }
2704 /* PyErr_CheckSignals() might change errno */
2705 saved_errno = errno;
2706 /* We must run our signal handlers before looping again.
2707 send() can return a successful partial write when it is
2708 interrupted, so we can't restrict ourselves to EINTR. */
2709 if (PyErr_CheckSignals()) {
2710 PyBuffer_Release(&pbuf);
2711 return NULL;
2712 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002713 if (n < 0) {
Antoine Pitrou6d7df632010-09-27 17:52:25 +00002714 /* If interrupted, try again */
2715 if (saved_errno == EINTR)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002716 continue;
Antoine Pitrou6d7df632010-09-27 17:52:25 +00002717 else
2718 break;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002719 }
2720 buf += n;
2721 len -= n;
2722 } while (len > 0);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002723 PyBuffer_Release(&pbuf);
Guido van Rossum67f7a382002-06-06 21:08:16 +00002724
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002725 if (n < 0)
2726 return s->errorhandler();
Guido van Rossum67f7a382002-06-06 21:08:16 +00002727
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002728 Py_INCREF(Py_None);
2729 return Py_None;
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002730}
2731
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002732PyDoc_STRVAR(sendall_doc,
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002733"sendall(data[, flags])\n\
2734\n\
2735Send a data string to the socket. For the optional flags\n\
2736argument, see the Unix manual. This calls send() repeatedly\n\
2737until all data is sent. If an error occurs, it's impossible\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002738to tell how much data has been sent.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002739
Guido van Rossum30a685f1991-06-27 15:51:29 +00002740
Guido van Rossumeb6b33a1993-05-25 09:38:27 +00002741/* s.sendto(data, [flags,] sockaddr) method */
Guido van Rossum30a685f1991-06-27 15:51:29 +00002742
Guido van Rossum73624e91994-10-10 17:59:00 +00002743static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002744sock_sendto(PySocketSockObject *s, PyObject *args)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002745{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002746 Py_buffer pbuf;
2747 PyObject *addro;
2748 char *buf;
Antoine Pitrou5e981412011-03-17 22:38:37 +01002749 Py_ssize_t len, arglen;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002750 sock_addr_t addrbuf;
2751 int addrlen, n = -1, flags, timeout;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002752
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002753 flags = 0;
Antoine Pitrou5e981412011-03-17 22:38:37 +01002754 arglen = PyTuple_Size(args);
2755 switch (arglen) {
2756 case 2:
2757 PyArg_ParseTuple(args, "y*O:sendto", &pbuf, &addro);
2758 break;
2759 case 3:
2760 PyArg_ParseTuple(args, "y*iO:sendto",
2761 &pbuf, &flags, &addro);
2762 break;
2763 default:
2764 PyErr_Format(PyExc_TypeError,
2765 "sendto() takes 2 or 3 arguments (%d given)",
2766 arglen);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002767 }
Antoine Pitrou5e981412011-03-17 22:38:37 +01002768 if (PyErr_Occurred())
2769 return NULL;
2770
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002771 buf = pbuf.buf;
2772 len = pbuf.len;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002773
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002774 if (!IS_SELECTABLE(s)) {
2775 PyBuffer_Release(&pbuf);
2776 return select_error();
2777 }
Neal Norwitz082b2df2006-02-07 07:04:46 +00002778
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002779 if (!getsockaddrarg(s, addro, SAS2SA(&addrbuf), &addrlen)) {
2780 PyBuffer_Release(&pbuf);
2781 return NULL;
2782 }
Thomas Wouters89f507f2006-12-13 04:49:30 +00002783
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002784 BEGIN_SELECT_LOOP(s)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002785 Py_BEGIN_ALLOW_THREADS
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002786 timeout = internal_select_ex(s, 1, interval);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002787 if (!timeout)
2788 n = sendto(s->sock_fd, buf, len, flags, SAS2SA(&addrbuf), addrlen);
2789 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00002790
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002791 if (timeout == 1) {
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002792 PyBuffer_Release(&pbuf);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002793 PyErr_SetString(socket_timeout, "timed out");
2794 return NULL;
2795 }
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002796 END_SELECT_LOOP(s)
2797 PyBuffer_Release(&pbuf);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002798 if (n < 0)
2799 return s->errorhandler();
Antoine Pitrou19467d22010-08-17 19:33:30 +00002800 return PyLong_FromSsize_t(n);
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002801}
2802
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002803PyDoc_STRVAR(sendto_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00002804"sendto(data[, flags], address) -> count\n\
Guido van Rossum82a5c661998-07-07 20:45:43 +00002805\n\
2806Like send(data, flags) but allows specifying the destination address.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002807For IP sockets, the address is a pair (hostaddr, port).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002808
Guido van Rossum30a685f1991-06-27 15:51:29 +00002809
2810/* s.shutdown(how) method */
2811
Guido van Rossum73624e91994-10-10 17:59:00 +00002812static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002813sock_shutdown(PySocketSockObject *s, PyObject *arg)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002814{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002815 int how;
2816 int res;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00002817
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002818 how = PyLong_AsLong(arg);
2819 if (how == -1 && PyErr_Occurred())
2820 return NULL;
2821 Py_BEGIN_ALLOW_THREADS
2822 res = shutdown(s->sock_fd, how);
2823 Py_END_ALLOW_THREADS
2824 if (res < 0)
2825 return s->errorhandler();
2826 Py_INCREF(Py_None);
2827 return Py_None;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002828}
2829
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002830PyDoc_STRVAR(shutdown_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002831"shutdown(flag)\n\
2832\n\
Martin v. Löwis94681fc2003-11-27 19:40:22 +00002833Shut down the reading side of the socket (flag == SHUT_RD), the writing side\n\
2834of the socket (flag == SHUT_WR), or both ends (flag == SHUT_RDWR).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002835
Amaury Forgeot d'Arc3d17a5c2008-06-13 01:09:34 +00002836#if defined(MS_WINDOWS) && defined(SIO_RCVALL)
Christian Heimesfaf2f632008-01-06 16:59:19 +00002837static PyObject*
2838sock_ioctl(PySocketSockObject *s, PyObject *arg)
2839{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002840 unsigned long cmd = SIO_RCVALL;
2841 PyObject *argO;
2842 DWORD recv;
Christian Heimesfaf2f632008-01-06 16:59:19 +00002843
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002844 if (!PyArg_ParseTuple(arg, "kO:ioctl", &cmd, &argO))
2845 return NULL;
Christian Heimesfaf2f632008-01-06 16:59:19 +00002846
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002847 switch (cmd) {
2848 case SIO_RCVALL: {
2849 unsigned int option = RCVALL_ON;
2850 if (!PyArg_ParseTuple(arg, "kI:ioctl", &cmd, &option))
2851 return NULL;
2852 if (WSAIoctl(s->sock_fd, cmd, &option, sizeof(option),
2853 NULL, 0, &recv, NULL, NULL) == SOCKET_ERROR) {
2854 return set_error();
2855 }
2856 return PyLong_FromUnsignedLong(recv); }
2857 case SIO_KEEPALIVE_VALS: {
2858 struct tcp_keepalive ka;
2859 if (!PyArg_ParseTuple(arg, "k(kkk):ioctl", &cmd,
2860 &ka.onoff, &ka.keepalivetime, &ka.keepaliveinterval))
2861 return NULL;
2862 if (WSAIoctl(s->sock_fd, cmd, &ka, sizeof(ka),
2863 NULL, 0, &recv, NULL, NULL) == SOCKET_ERROR) {
2864 return set_error();
2865 }
2866 return PyLong_FromUnsignedLong(recv); }
2867 default:
2868 PyErr_Format(PyExc_ValueError, "invalid ioctl command %d", cmd);
2869 return NULL;
2870 }
Christian Heimesfaf2f632008-01-06 16:59:19 +00002871}
2872PyDoc_STRVAR(sock_ioctl_doc,
2873"ioctl(cmd, option) -> long\n\
2874\n\
Kristján Valur Jónsson847ec752009-09-27 21:10:38 +00002875Control the socket with WSAIoctl syscall. Currently supported 'cmd' values are\n\
2876SIO_RCVALL: 'option' must be one of the socket.RCVALL_* constants.\n\
2877SIO_KEEPALIVE_VALS: 'option' is a tuple of (onoff, timeout, interval).");
Christian Heimesfaf2f632008-01-06 16:59:19 +00002878
2879#endif
Guido van Rossum30a685f1991-06-27 15:51:29 +00002880
2881/* List of methods for socket objects */
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002882
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002883static PyMethodDef sock_methods[] = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002884 {"_accept", (PyCFunction)sock_accept, METH_NOARGS,
2885 accept_doc},
2886 {"bind", (PyCFunction)sock_bind, METH_O,
2887 bind_doc},
2888 {"close", (PyCFunction)sock_close, METH_NOARGS,
2889 close_doc},
2890 {"connect", (PyCFunction)sock_connect, METH_O,
2891 connect_doc},
2892 {"connect_ex", (PyCFunction)sock_connect_ex, METH_O,
2893 connect_ex_doc},
Antoine Pitrou6e451df2010-08-09 20:39:54 +00002894 {"detach", (PyCFunction)sock_detach, METH_NOARGS,
2895 detach_doc},
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002896 {"fileno", (PyCFunction)sock_fileno, METH_NOARGS,
2897 fileno_doc},
Guido van Rossumb6775db1994-08-01 11:34:53 +00002898#ifdef HAVE_GETPEERNAME
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002899 {"getpeername", (PyCFunction)sock_getpeername,
2900 METH_NOARGS, getpeername_doc},
Guido van Rossum9575a441993-04-07 14:06:14 +00002901#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002902 {"getsockname", (PyCFunction)sock_getsockname,
2903 METH_NOARGS, getsockname_doc},
2904 {"getsockopt", (PyCFunction)sock_getsockopt, METH_VARARGS,
2905 getsockopt_doc},
Amaury Forgeot d'Arc3d17a5c2008-06-13 01:09:34 +00002906#if defined(MS_WINDOWS) && defined(SIO_RCVALL)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002907 {"ioctl", (PyCFunction)sock_ioctl, METH_VARARGS,
2908 sock_ioctl_doc},
Christian Heimesfaf2f632008-01-06 16:59:19 +00002909#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002910 {"listen", (PyCFunction)sock_listen, METH_O,
2911 listen_doc},
2912 {"recv", (PyCFunction)sock_recv, METH_VARARGS,
2913 recv_doc},
2914 {"recv_into", (PyCFunction)sock_recv_into, METH_VARARGS | METH_KEYWORDS,
2915 recv_into_doc},
2916 {"recvfrom", (PyCFunction)sock_recvfrom, METH_VARARGS,
2917 recvfrom_doc},
2918 {"recvfrom_into", (PyCFunction)sock_recvfrom_into, METH_VARARGS | METH_KEYWORDS,
2919 recvfrom_into_doc},
2920 {"send", (PyCFunction)sock_send, METH_VARARGS,
2921 send_doc},
2922 {"sendall", (PyCFunction)sock_sendall, METH_VARARGS,
2923 sendall_doc},
2924 {"sendto", (PyCFunction)sock_sendto, METH_VARARGS,
2925 sendto_doc},
2926 {"setblocking", (PyCFunction)sock_setblocking, METH_O,
2927 setblocking_doc},
2928 {"settimeout", (PyCFunction)sock_settimeout, METH_O,
2929 settimeout_doc},
2930 {"gettimeout", (PyCFunction)sock_gettimeout, METH_NOARGS,
2931 gettimeout_doc},
2932 {"setsockopt", (PyCFunction)sock_setsockopt, METH_VARARGS,
2933 setsockopt_doc},
2934 {"shutdown", (PyCFunction)sock_shutdown, METH_O,
2935 shutdown_doc},
2936 {NULL, NULL} /* sentinel */
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002937};
2938
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002939/* SockObject members */
2940static PyMemberDef sock_memberlist[] = {
2941 {"family", T_INT, offsetof(PySocketSockObject, sock_family), READONLY, "the socket family"},
2942 {"type", T_INT, offsetof(PySocketSockObject, sock_type), READONLY, "the socket type"},
2943 {"proto", T_INT, offsetof(PySocketSockObject, sock_proto), READONLY, "the socket protocol"},
2944 {"timeout", T_DOUBLE, offsetof(PySocketSockObject, sock_timeout), READONLY, "the socket timeout"},
2945 {0},
2946};
Guido van Rossum30a685f1991-06-27 15:51:29 +00002947
Guido van Rossum73624e91994-10-10 17:59:00 +00002948/* Deallocate a socket object in response to the last Py_DECREF().
Guido van Rossum30a685f1991-06-27 15:51:29 +00002949 First close the file description. */
2950
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002951static void
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002952sock_dealloc(PySocketSockObject *s)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002953{
Antoine Pitroue033e062010-10-29 10:38:18 +00002954 if (s->sock_fd != -1) {
2955 PyObject *exc, *val, *tb;
2956 Py_ssize_t old_refcount = Py_REFCNT(s);
2957 ++Py_REFCNT(s);
2958 PyErr_Fetch(&exc, &val, &tb);
2959 if (PyErr_WarnFormat(PyExc_ResourceWarning, 1,
2960 "unclosed %R", s))
2961 /* Spurious errors can appear at shutdown */
2962 if (PyErr_ExceptionMatches(PyExc_Warning))
2963 PyErr_WriteUnraisable((PyObject *) s);
2964 PyErr_Restore(exc, val, tb);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002965 (void) SOCKETCLOSE(s->sock_fd);
Antoine Pitroue033e062010-10-29 10:38:18 +00002966 Py_REFCNT(s) = old_refcount;
2967 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002968 Py_TYPE(s)->tp_free((PyObject *)s);
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002969}
2970
Guido van Rossum30a685f1991-06-27 15:51:29 +00002971
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00002972static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002973sock_repr(PySocketSockObject *s)
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00002974{
Fred Drakea04eaad2000-06-30 02:46:07 +00002975#if SIZEOF_SOCKET_T > SIZEOF_LONG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002976 if (s->sock_fd > LONG_MAX) {
2977 /* this can occur on Win64, and actually there is a special
2978 ugly printf formatter for decimal pointer length integer
2979 printing, only bother if necessary*/
2980 PyErr_SetString(PyExc_OverflowError,
2981 "no printf formatter to display "
2982 "the socket descriptor in decimal");
2983 return NULL;
2984 }
Fred Drakea04eaad2000-06-30 02:46:07 +00002985#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002986 return PyUnicode_FromFormat(
2987 "<socket object, fd=%ld, family=%d, type=%d, proto=%d>",
2988 (long)s->sock_fd, s->sock_family,
2989 s->sock_type,
2990 s->sock_proto);
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00002991}
2992
2993
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002994/* Create a new, uninitialized socket object. */
2995
2996static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002997sock_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002998{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002999 PyObject *new;
Guido van Rossum384ca9c2001-10-27 22:20:47 +00003000
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003001 new = type->tp_alloc(type, 0);
3002 if (new != NULL) {
3003 ((PySocketSockObject *)new)->sock_fd = -1;
3004 ((PySocketSockObject *)new)->sock_timeout = -1.0;
3005 ((PySocketSockObject *)new)->errorhandler = &set_error;
3006 }
3007 return new;
Guido van Rossum384ca9c2001-10-27 22:20:47 +00003008}
3009
3010
3011/* Initialize a new socket object. */
3012
3013/*ARGSUSED*/
3014static int
Andrew MacIntyre7aec4a22002-06-13 11:53:52 +00003015sock_initobj(PyObject *self, PyObject *args, PyObject *kwds)
Guido van Rossum384ca9c2001-10-27 22:20:47 +00003016{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003017 PySocketSockObject *s = (PySocketSockObject *)self;
3018 PyObject *fdobj = NULL;
3019 SOCKET_T fd = INVALID_SOCKET;
3020 int family = AF_INET, type = SOCK_STREAM, proto = 0;
3021 static char *keywords[] = {"family", "type", "proto", "fileno", 0};
Guido van Rossum384ca9c2001-10-27 22:20:47 +00003022
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003023 if (!PyArg_ParseTupleAndKeywords(args, kwds,
3024 "|iiiO:socket", keywords,
3025 &family, &type, &proto, &fdobj))
3026 return -1;
Guido van Rossum67f7a382002-06-06 21:08:16 +00003027
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003028 if (fdobj != NULL && fdobj != Py_None) {
3029 fd = PyLong_AsSocket_t(fdobj);
3030 if (fd == (SOCKET_T)(-1) && PyErr_Occurred())
3031 return -1;
3032 if (fd == INVALID_SOCKET) {
3033 PyErr_SetString(PyExc_ValueError,
3034 "can't use invalid socket value");
3035 return -1;
3036 }
3037 }
3038 else {
3039 Py_BEGIN_ALLOW_THREADS
3040 fd = socket(family, type, proto);
3041 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00003042
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003043 if (fd == INVALID_SOCKET) {
3044 set_error();
3045 return -1;
3046 }
3047 }
3048 init_sockobject(s, fd, family, type, proto);
Guido van Rossum67f7a382002-06-06 21:08:16 +00003049
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003050 return 0;
Guido van Rossum67f7a382002-06-06 21:08:16 +00003051
Guido van Rossum384ca9c2001-10-27 22:20:47 +00003052}
3053
3054
Guido van Rossumb6775db1994-08-01 11:34:53 +00003055/* Type object for socket objects. */
Guido van Rossum30a685f1991-06-27 15:51:29 +00003056
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003057static PyTypeObject sock_type = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003058 PyVarObject_HEAD_INIT(0, 0) /* Must fill in type value later */
3059 "_socket.socket", /* tp_name */
3060 sizeof(PySocketSockObject), /* tp_basicsize */
3061 0, /* tp_itemsize */
3062 (destructor)sock_dealloc, /* tp_dealloc */
3063 0, /* tp_print */
3064 0, /* tp_getattr */
3065 0, /* tp_setattr */
3066 0, /* tp_reserved */
3067 (reprfunc)sock_repr, /* tp_repr */
3068 0, /* tp_as_number */
3069 0, /* tp_as_sequence */
3070 0, /* tp_as_mapping */
3071 0, /* tp_hash */
3072 0, /* tp_call */
3073 0, /* tp_str */
3074 PyObject_GenericGetAttr, /* tp_getattro */
3075 0, /* tp_setattro */
3076 0, /* tp_as_buffer */
3077 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE, /* tp_flags */
3078 sock_doc, /* tp_doc */
3079 0, /* tp_traverse */
3080 0, /* tp_clear */
3081 0, /* tp_richcompare */
3082 0, /* tp_weaklistoffset */
3083 0, /* tp_iter */
3084 0, /* tp_iternext */
3085 sock_methods, /* tp_methods */
3086 sock_memberlist, /* tp_members */
3087 0, /* tp_getset */
3088 0, /* tp_base */
3089 0, /* tp_dict */
3090 0, /* tp_descr_get */
3091 0, /* tp_descr_set */
3092 0, /* tp_dictoffset */
3093 sock_initobj, /* tp_init */
3094 PyType_GenericAlloc, /* tp_alloc */
3095 sock_new, /* tp_new */
3096 PyObject_Del, /* tp_free */
Guido van Rossum6574b3e1991-06-25 21:36:08 +00003097};
3098
Guido van Rossum30a685f1991-06-27 15:51:29 +00003099
Guido van Rossum81194471991-07-27 21:42:02 +00003100/* Python interface to gethostname(). */
3101
3102/*ARGSUSED*/
Guido van Rossum73624e91994-10-10 17:59:00 +00003103static PyObject *
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00003104socket_gethostname(PyObject *self, PyObject *unused)
Guido van Rossum81194471991-07-27 21:42:02 +00003105{
Martin v. Löwis72f48422010-10-29 18:20:08 +00003106#ifdef MS_WINDOWS
3107 /* Don't use winsock's gethostname, as this returns the ANSI
3108 version of the hostname, whereas we need a Unicode string.
3109 Otherwise, gethostname apparently also returns the DNS name. */
Hirokazu Yamamoto09fff7a2010-11-05 17:24:13 +00003110 wchar_t buf[MAX_COMPUTERNAME_LENGTH + 1];
3111 DWORD size = sizeof(buf) / sizeof(wchar_t);
3112 PyObject *result;
Martin v. Löwis72f48422010-10-29 18:20:08 +00003113 if (!GetComputerNameExW(ComputerNamePhysicalDnsHostname, buf, &size)) {
3114 if (GetLastError() == ERROR_MORE_DATA) {
3115 /* MSDN says this may occur "because DNS allows longer names */
Hirokazu Yamamoto09fff7a2010-11-05 17:24:13 +00003116 if (size == 0) /* XXX: I'm not sure how to handle this */
3117 return PyUnicode_FromUnicode(NULL, 0);
3118 result = PyUnicode_FromUnicode(NULL, size - 1);
Martin v. Löwis72f48422010-10-29 18:20:08 +00003119 if (!result)
3120 return NULL;
Hirokazu Yamamoto09fff7a2010-11-05 17:24:13 +00003121 if (GetComputerNameExW(ComputerNamePhysicalDnsHostname,
3122 PyUnicode_AS_UNICODE(result),
3123 &size))
Martin v. Löwis72f48422010-10-29 18:20:08 +00003124 return result;
Benjamin Petersone857b292010-10-29 21:37:26 +00003125 Py_DECREF(result);
Martin v. Löwis72f48422010-10-29 18:20:08 +00003126 }
3127 return PyErr_SetExcFromWindowsErr(PyExc_WindowsError, GetLastError());
3128 }
3129 return PyUnicode_FromUnicode(buf, size);
3130#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003131 char buf[1024];
3132 int res;
3133 Py_BEGIN_ALLOW_THREADS
3134 res = gethostname(buf, (int) sizeof buf - 1);
3135 Py_END_ALLOW_THREADS
3136 if (res < 0)
3137 return set_error();
3138 buf[sizeof buf - 1] = '\0';
3139 return PyUnicode_FromString(buf);
Martin v. Löwis72f48422010-10-29 18:20:08 +00003140#endif
Guido van Rossum81194471991-07-27 21:42:02 +00003141}
Guido van Rossumff4949e1992-08-05 19:58:53 +00003142
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003143PyDoc_STRVAR(gethostname_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003144"gethostname() -> string\n\
3145\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003146Return the current host name.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003147
Guido van Rossumff4949e1992-08-05 19:58:53 +00003148
Guido van Rossum30a685f1991-06-27 15:51:29 +00003149/* Python interface to gethostbyname(name). */
3150
3151/*ARGSUSED*/
Guido van Rossum73624e91994-10-10 17:59:00 +00003152static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003153socket_gethostbyname(PyObject *self, PyObject *args)
Guido van Rossum30a685f1991-06-27 15:51:29 +00003154{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003155 char *name;
3156 sock_addr_t addrbuf;
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003157 PyObject *ret = NULL;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003158
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003159 if (!PyArg_ParseTuple(args, "et:gethostbyname", "idna", &name))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003160 return NULL;
3161 if (setipaddr(name, SAS2SA(&addrbuf), sizeof(addrbuf), AF_INET) < 0)
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003162 goto finally;
3163 ret = makeipaddr(SAS2SA(&addrbuf), sizeof(struct sockaddr_in));
3164finally:
3165 PyMem_Free(name);
3166 return ret;
Guido van Rossum30a685f1991-06-27 15:51:29 +00003167}
3168
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003169PyDoc_STRVAR(gethostbyname_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003170"gethostbyname(host) -> address\n\
3171\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003172Return the IP address (a string of the form '255.255.255.255') for a host.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003173
3174
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003175/* Convenience function common to gethostbyname_ex and gethostbyaddr */
3176
3177static PyObject *
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003178gethost_common(struct hostent *h, struct sockaddr *addr, int alen, int af)
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003179{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003180 char **pch;
3181 PyObject *rtn_tuple = (PyObject *)NULL;
3182 PyObject *name_list = (PyObject *)NULL;
3183 PyObject *addr_list = (PyObject *)NULL;
3184 PyObject *tmp;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003185
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003186 if (h == NULL) {
3187 /* Let's get real error message to return */
3188 set_herror(h_errno);
3189 return NULL;
3190 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00003191
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003192 if (h->h_addrtype != af) {
3193 /* Let's get real error message to return */
3194 PyErr_SetString(socket_error,
3195 (char *)strerror(EAFNOSUPPORT));
Christian Heimesada8c3b2008-03-18 18:26:33 +00003196
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003197 return NULL;
3198 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00003199
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003200 switch (af) {
Guido van Rossum67f7a382002-06-06 21:08:16 +00003201
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003202 case AF_INET:
3203 if (alen < sizeof(struct sockaddr_in))
3204 return NULL;
3205 break;
Guido van Rossum67f7a382002-06-06 21:08:16 +00003206
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00003207#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003208 case AF_INET6:
3209 if (alen < sizeof(struct sockaddr_in6))
3210 return NULL;
3211 break;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003212#endif
Guido van Rossum67f7a382002-06-06 21:08:16 +00003213
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003214 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00003215
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003216 if ((name_list = PyList_New(0)) == NULL)
3217 goto err;
Guido van Rossum67f7a382002-06-06 21:08:16 +00003218
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003219 if ((addr_list = PyList_New(0)) == NULL)
3220 goto err;
Guido van Rossum67f7a382002-06-06 21:08:16 +00003221
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003222 /* SF #1511317: h_aliases can be NULL */
3223 if (h->h_aliases) {
3224 for (pch = h->h_aliases; *pch != NULL; pch++) {
3225 int status;
3226 tmp = PyUnicode_FromString(*pch);
3227 if (tmp == NULL)
3228 goto err;
Guido van Rossum67f7a382002-06-06 21:08:16 +00003229
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003230 status = PyList_Append(name_list, tmp);
3231 Py_DECREF(tmp);
Guido van Rossum67f7a382002-06-06 21:08:16 +00003232
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003233 if (status)
3234 goto err;
3235 }
3236 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00003237
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003238 for (pch = h->h_addr_list; *pch != NULL; pch++) {
3239 int status;
Guido van Rossum67f7a382002-06-06 21:08:16 +00003240
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003241 switch (af) {
Guido van Rossum67f7a382002-06-06 21:08:16 +00003242
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003243 case AF_INET:
3244 {
3245 struct sockaddr_in sin;
3246 memset(&sin, 0, sizeof(sin));
3247 sin.sin_family = af;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003248#ifdef HAVE_SOCKADDR_SA_LEN
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003249 sin.sin_len = sizeof(sin);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003250#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003251 memcpy(&sin.sin_addr, *pch, sizeof(sin.sin_addr));
3252 tmp = makeipaddr((struct sockaddr *)&sin, sizeof(sin));
Guido van Rossum67f7a382002-06-06 21:08:16 +00003253
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003254 if (pch == h->h_addr_list && alen >= sizeof(sin))
3255 memcpy((char *) addr, &sin, sizeof(sin));
3256 break;
3257 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00003258
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00003259#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003260 case AF_INET6:
3261 {
3262 struct sockaddr_in6 sin6;
3263 memset(&sin6, 0, sizeof(sin6));
3264 sin6.sin6_family = af;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003265#ifdef HAVE_SOCKADDR_SA_LEN
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003266 sin6.sin6_len = sizeof(sin6);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003267#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003268 memcpy(&sin6.sin6_addr, *pch, sizeof(sin6.sin6_addr));
3269 tmp = makeipaddr((struct sockaddr *)&sin6,
3270 sizeof(sin6));
Guido van Rossum67f7a382002-06-06 21:08:16 +00003271
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003272 if (pch == h->h_addr_list && alen >= sizeof(sin6))
3273 memcpy((char *) addr, &sin6, sizeof(sin6));
3274 break;
3275 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003276#endif
Guido van Rossum67f7a382002-06-06 21:08:16 +00003277
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003278 default: /* can't happen */
3279 PyErr_SetString(socket_error,
3280 "unsupported address family");
3281 return NULL;
3282 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00003283
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003284 if (tmp == NULL)
3285 goto err;
Guido van Rossum67f7a382002-06-06 21:08:16 +00003286
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003287 status = PyList_Append(addr_list, tmp);
3288 Py_DECREF(tmp);
Guido van Rossum67f7a382002-06-06 21:08:16 +00003289
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003290 if (status)
3291 goto err;
3292 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00003293
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003294 rtn_tuple = Py_BuildValue("sOO", h->h_name, name_list, addr_list);
Guido van Rossum67f7a382002-06-06 21:08:16 +00003295
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003296 err:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003297 Py_XDECREF(name_list);
3298 Py_XDECREF(addr_list);
3299 return rtn_tuple;
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003300}
3301
3302
3303/* Python interface to gethostbyname_ex(name). */
3304
3305/*ARGSUSED*/
3306static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003307socket_gethostbyname_ex(PyObject *self, PyObject *args)
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003308{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003309 char *name;
3310 struct hostent *h;
Martin v. Löwisc16f3bd2003-05-03 09:14:54 +00003311#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003312 struct sockaddr_storage addr;
Martin v. Löwisc16f3bd2003-05-03 09:14:54 +00003313#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003314 struct sockaddr_in addr;
Martin v. Löwisc16f3bd2003-05-03 09:14:54 +00003315#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003316 struct sockaddr *sa;
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003317 PyObject *ret = NULL;
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003318#ifdef HAVE_GETHOSTBYNAME_R
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003319 struct hostent hp_allocated;
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003320#ifdef HAVE_GETHOSTBYNAME_R_3_ARG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003321 struct hostent_data data;
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003322#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003323 char buf[16384];
3324 int buf_len = (sizeof buf) - 1;
3325 int errnop;
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003326#endif
3327#if defined(HAVE_GETHOSTBYNAME_R_3_ARG) || defined(HAVE_GETHOSTBYNAME_R_6_ARG)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003328 int result;
Guido van Rossume9cd07b1999-03-15 21:40:14 +00003329#endif
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003330#endif /* HAVE_GETHOSTBYNAME_R */
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003331
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003332 if (!PyArg_ParseTuple(args, "et:gethostbyname_ex", "idna", &name))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003333 return NULL;
3334 if (setipaddr(name, (struct sockaddr *)&addr, sizeof(addr), AF_INET) < 0)
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003335 goto finally;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003336 Py_BEGIN_ALLOW_THREADS
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003337#ifdef HAVE_GETHOSTBYNAME_R
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003338#if defined(HAVE_GETHOSTBYNAME_R_6_ARG)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003339 result = gethostbyname_r(name, &hp_allocated, buf, buf_len,
3340 &h, &errnop);
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003341#elif defined(HAVE_GETHOSTBYNAME_R_5_ARG)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003342 h = gethostbyname_r(name, &hp_allocated, buf, buf_len, &errnop);
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003343#else /* HAVE_GETHOSTBYNAME_R_3_ARG */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003344 memset((void *) &data, '\0', sizeof(data));
3345 result = gethostbyname_r(name, &hp_allocated, &data);
3346 h = (result != 0) ? NULL : &hp_allocated;
Guido van Rossume9cd07b1999-03-15 21:40:14 +00003347#endif
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003348#else /* not HAVE_GETHOSTBYNAME_R */
Guido van Rossum3baaa131999-03-22 21:44:51 +00003349#ifdef USE_GETHOSTBYNAME_LOCK
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003350 PyThread_acquire_lock(netdb_lock, 1);
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003351#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003352 h = gethostbyname(name);
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003353#endif /* HAVE_GETHOSTBYNAME_R */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003354 Py_END_ALLOW_THREADS
3355 /* Some C libraries would require addr.__ss_family instead of
3356 addr.ss_family.
3357 Therefore, we cast the sockaddr_storage into sockaddr to
3358 access sa_family. */
3359 sa = (struct sockaddr*)&addr;
3360 ret = gethost_common(h, (struct sockaddr *)&addr, sizeof(addr),
3361 sa->sa_family);
Guido van Rossum3baaa131999-03-22 21:44:51 +00003362#ifdef USE_GETHOSTBYNAME_LOCK
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003363 PyThread_release_lock(netdb_lock);
Guido van Rossum955becc1999-03-22 20:14:53 +00003364#endif
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003365finally:
3366 PyMem_Free(name);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003367 return ret;
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003368}
3369
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003370PyDoc_STRVAR(ghbn_ex_doc,
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003371"gethostbyname_ex(host) -> (name, aliaslist, addresslist)\n\
3372\n\
3373Return the true host name, a list of aliases, and a list of IP addresses,\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003374for a host. The host argument is a string giving a host name or IP number.");
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003375
3376
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00003377/* Python interface to gethostbyaddr(IP). */
3378
3379/*ARGSUSED*/
3380static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003381socket_gethostbyaddr(PyObject *self, PyObject *args)
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00003382{
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00003383#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003384 struct sockaddr_storage addr;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003385#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003386 struct sockaddr_in addr;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003387#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003388 struct sockaddr *sa = (struct sockaddr *)&addr;
3389 char *ip_num;
3390 struct hostent *h;
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003391 PyObject *ret = NULL;
Guido van Rossum4f199ea1998-04-09 20:56:35 +00003392#ifdef HAVE_GETHOSTBYNAME_R
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003393 struct hostent hp_allocated;
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003394#ifdef HAVE_GETHOSTBYNAME_R_3_ARG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003395 struct hostent_data data;
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003396#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003397 /* glibcs up to 2.10 assume that the buf argument to
3398 gethostbyaddr_r is 8-byte aligned, which at least llvm-gcc
3399 does not ensure. The attribute below instructs the compiler
3400 to maintain this alignment. */
3401 char buf[16384] Py_ALIGNED(8);
3402 int buf_len = (sizeof buf) - 1;
3403 int errnop;
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003404#endif
3405#if defined(HAVE_GETHOSTBYNAME_R_3_ARG) || defined(HAVE_GETHOSTBYNAME_R_6_ARG)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003406 int result;
Guido van Rossume9cd07b1999-03-15 21:40:14 +00003407#endif
Guido van Rossum4f199ea1998-04-09 20:56:35 +00003408#endif /* HAVE_GETHOSTBYNAME_R */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003409 char *ap;
3410 int al;
3411 int af;
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00003412
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003413 if (!PyArg_ParseTuple(args, "et:gethostbyaddr", "idna", &ip_num))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003414 return NULL;
3415 af = AF_UNSPEC;
3416 if (setipaddr(ip_num, sa, sizeof(addr), af) < 0)
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003417 goto finally;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003418 af = sa->sa_family;
3419 ap = NULL;
3420 al = 0;
3421 switch (af) {
3422 case AF_INET:
3423 ap = (char *)&((struct sockaddr_in *)sa)->sin_addr;
3424 al = sizeof(((struct sockaddr_in *)sa)->sin_addr);
3425 break;
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00003426#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003427 case AF_INET6:
3428 ap = (char *)&((struct sockaddr_in6 *)sa)->sin6_addr;
3429 al = sizeof(((struct sockaddr_in6 *)sa)->sin6_addr);
3430 break;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003431#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003432 default:
3433 PyErr_SetString(socket_error, "unsupported address family");
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003434 goto finally;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003435 }
3436 Py_BEGIN_ALLOW_THREADS
Guido van Rossum4f199ea1998-04-09 20:56:35 +00003437#ifdef HAVE_GETHOSTBYNAME_R
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003438#if defined(HAVE_GETHOSTBYNAME_R_6_ARG)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003439 result = gethostbyaddr_r(ap, al, af,
3440 &hp_allocated, buf, buf_len,
3441 &h, &errnop);
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003442#elif defined(HAVE_GETHOSTBYNAME_R_5_ARG)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003443 h = gethostbyaddr_r(ap, al, af,
3444 &hp_allocated, buf, buf_len, &errnop);
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003445#else /* HAVE_GETHOSTBYNAME_R_3_ARG */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003446 memset((void *) &data, '\0', sizeof(data));
3447 result = gethostbyaddr_r(ap, al, af, &hp_allocated, &data);
3448 h = (result != 0) ? NULL : &hp_allocated;
Guido van Rossume9cd07b1999-03-15 21:40:14 +00003449#endif
Guido van Rossum4f199ea1998-04-09 20:56:35 +00003450#else /* not HAVE_GETHOSTBYNAME_R */
Guido van Rossum3baaa131999-03-22 21:44:51 +00003451#ifdef USE_GETHOSTBYNAME_LOCK
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003452 PyThread_acquire_lock(netdb_lock, 1);
Guido van Rossum4f199ea1998-04-09 20:56:35 +00003453#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003454 h = gethostbyaddr(ap, al, af);
Guido van Rossum4f199ea1998-04-09 20:56:35 +00003455#endif /* HAVE_GETHOSTBYNAME_R */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003456 Py_END_ALLOW_THREADS
3457 ret = gethost_common(h, (struct sockaddr *)&addr, sizeof(addr), af);
Guido van Rossum3baaa131999-03-22 21:44:51 +00003458#ifdef USE_GETHOSTBYNAME_LOCK
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003459 PyThread_release_lock(netdb_lock);
Guido van Rossum3baaa131999-03-22 21:44:51 +00003460#endif
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003461finally:
3462 PyMem_Free(ip_num);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003463 return ret;
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00003464}
3465
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003466PyDoc_STRVAR(gethostbyaddr_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003467"gethostbyaddr(host) -> (name, aliaslist, addresslist)\n\
3468\n\
3469Return the true host name, a list of aliases, and a list of IP addresses,\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003470for a host. The host argument is a string giving a host name or IP number.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003471
Guido van Rossum30a685f1991-06-27 15:51:29 +00003472
3473/* Python interface to getservbyname(name).
3474 This only returns the port number, since the other info is already
3475 known or not useful (like the list of aliases). */
3476
3477/*ARGSUSED*/
Guido van Rossum73624e91994-10-10 17:59:00 +00003478static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003479socket_getservbyname(PyObject *self, PyObject *args)
Guido van Rossum30a685f1991-06-27 15:51:29 +00003480{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003481 char *name, *proto=NULL;
3482 struct servent *sp;
3483 if (!PyArg_ParseTuple(args, "s|s:getservbyname", &name, &proto))
3484 return NULL;
3485 Py_BEGIN_ALLOW_THREADS
3486 sp = getservbyname(name, proto);
3487 Py_END_ALLOW_THREADS
3488 if (sp == NULL) {
3489 PyErr_SetString(socket_error, "service/proto not found");
3490 return NULL;
3491 }
3492 return PyLong_FromLong((long) ntohs(sp->s_port));
Guido van Rossum30a685f1991-06-27 15:51:29 +00003493}
3494
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003495PyDoc_STRVAR(getservbyname_doc,
Barry Warsaw11b91a02004-06-28 00:50:43 +00003496"getservbyname(servicename[, protocolname]) -> integer\n\
Guido van Rossum82a5c661998-07-07 20:45:43 +00003497\n\
3498Return a port number from a service name and protocol name.\n\
Barry Warsaw11b91a02004-06-28 00:50:43 +00003499The optional protocol name, if given, should be 'tcp' or 'udp',\n\
3500otherwise any protocol will match.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003501
Guido van Rossum30a685f1991-06-27 15:51:29 +00003502
Barry Warsaw11b91a02004-06-28 00:50:43 +00003503/* Python interface to getservbyport(port).
3504 This only returns the service name, since the other info is already
3505 known or not useful (like the list of aliases). */
3506
3507/*ARGSUSED*/
3508static PyObject *
3509socket_getservbyport(PyObject *self, PyObject *args)
3510{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003511 int port;
3512 char *proto=NULL;
3513 struct servent *sp;
3514 if (!PyArg_ParseTuple(args, "i|s:getservbyport", &port, &proto))
3515 return NULL;
3516 if (port < 0 || port > 0xffff) {
3517 PyErr_SetString(
3518 PyExc_OverflowError,
3519 "getservbyport: port must be 0-65535.");
3520 return NULL;
3521 }
3522 Py_BEGIN_ALLOW_THREADS
3523 sp = getservbyport(htons((short)port), proto);
3524 Py_END_ALLOW_THREADS
3525 if (sp == NULL) {
3526 PyErr_SetString(socket_error, "port/proto not found");
3527 return NULL;
3528 }
3529 return PyUnicode_FromString(sp->s_name);
Barry Warsaw11b91a02004-06-28 00:50:43 +00003530}
3531
3532PyDoc_STRVAR(getservbyport_doc,
3533"getservbyport(port[, protocolname]) -> string\n\
3534\n\
3535Return the service name from a port number and protocol name.\n\
3536The optional protocol name, if given, should be 'tcp' or 'udp',\n\
3537otherwise any protocol will match.");
3538
Guido van Rossum3901d851996-12-19 16:35:04 +00003539/* Python interface to getprotobyname(name).
3540 This only returns the protocol number, since the other info is
3541 already known or not useful (like the list of aliases). */
3542
3543/*ARGSUSED*/
3544static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003545socket_getprotobyname(PyObject *self, PyObject *args)
Guido van Rossum3901d851996-12-19 16:35:04 +00003546{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003547 char *name;
3548 struct protoent *sp;
3549 if (!PyArg_ParseTuple(args, "s:getprotobyname", &name))
3550 return NULL;
3551 Py_BEGIN_ALLOW_THREADS
3552 sp = getprotobyname(name);
3553 Py_END_ALLOW_THREADS
3554 if (sp == NULL) {
3555 PyErr_SetString(socket_error, "protocol not found");
3556 return NULL;
3557 }
3558 return PyLong_FromLong((long) sp->p_proto);
Guido van Rossum3901d851996-12-19 16:35:04 +00003559}
3560
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003561PyDoc_STRVAR(getprotobyname_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003562"getprotobyname(name) -> integer\n\
3563\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003564Return the protocol number for the named protocol. (Rarely used.)");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003565
Guido van Rossum3901d851996-12-19 16:35:04 +00003566
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00003567#ifndef NO_DUP
3568/* dup() function for socket fds */
3569
3570static PyObject *
3571socket_dup(PyObject *self, PyObject *fdobj)
3572{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003573 SOCKET_T fd, newfd;
3574 PyObject *newfdobj;
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00003575
3576
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003577 fd = PyLong_AsSocket_t(fdobj);
3578 if (fd == (SOCKET_T)(-1) && PyErr_Occurred())
3579 return NULL;
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00003580
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003581 newfd = dup_socket(fd);
3582 if (newfd == INVALID_SOCKET)
3583 return set_error();
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00003584
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003585 newfdobj = PyLong_FromSocket_t(newfd);
3586 if (newfdobj == NULL)
3587 SOCKETCLOSE(newfd);
3588 return newfdobj;
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00003589}
3590
3591PyDoc_STRVAR(dup_doc,
3592"dup(integer) -> integer\n\
3593\n\
3594Duplicate an integer socket file descriptor. This is like os.dup(), but for\n\
3595sockets; on some platforms os.dup() won't work for socket file descriptors.");
3596#endif
3597
3598
Dave Cole331708b2004-08-09 04:51:41 +00003599#ifdef HAVE_SOCKETPAIR
3600/* Create a pair of sockets using the socketpair() function.
Dave Cole07fda7e2004-08-23 05:16:23 +00003601 Arguments as for socket() except the default family is AF_UNIX if
Dave Colee8bbfe42004-08-26 00:51:16 +00003602 defined on the platform; otherwise, the default is AF_INET. */
Dave Cole331708b2004-08-09 04:51:41 +00003603
3604/*ARGSUSED*/
3605static PyObject *
3606socket_socketpair(PyObject *self, PyObject *args)
3607{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003608 PySocketSockObject *s0 = NULL, *s1 = NULL;
3609 SOCKET_T sv[2];
3610 int family, type = SOCK_STREAM, proto = 0;
3611 PyObject *res = NULL;
Dave Cole331708b2004-08-09 04:51:41 +00003612
3613#if defined(AF_UNIX)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003614 family = AF_UNIX;
Dave Cole331708b2004-08-09 04:51:41 +00003615#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003616 family = AF_INET;
Dave Cole331708b2004-08-09 04:51:41 +00003617#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003618 if (!PyArg_ParseTuple(args, "|iii:socketpair",
3619 &family, &type, &proto))
3620 return NULL;
3621 /* Create a pair of socket fds */
3622 if (socketpair(family, type, proto, sv) < 0)
3623 return set_error();
3624 s0 = new_sockobject(sv[0], family, type, proto);
3625 if (s0 == NULL)
3626 goto finally;
3627 s1 = new_sockobject(sv[1], family, type, proto);
3628 if (s1 == NULL)
3629 goto finally;
3630 res = PyTuple_Pack(2, s0, s1);
Dave Cole331708b2004-08-09 04:51:41 +00003631
3632finally:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003633 if (res == NULL) {
3634 if (s0 == NULL)
3635 SOCKETCLOSE(sv[0]);
3636 if (s1 == NULL)
3637 SOCKETCLOSE(sv[1]);
3638 }
3639 Py_XDECREF(s0);
3640 Py_XDECREF(s1);
3641 return res;
Dave Cole331708b2004-08-09 04:51:41 +00003642}
3643
3644PyDoc_STRVAR(socketpair_doc,
3645"socketpair([family[, type[, proto]]]) -> (socket object, socket object)\n\
3646\n\
3647Create a pair of socket objects from the sockets returned by the platform\n\
3648socketpair() function.\n\
Dave Cole07fda7e2004-08-23 05:16:23 +00003649The arguments are the same as for socket() except the default family is\n\
Dave Colee8bbfe42004-08-26 00:51:16 +00003650AF_UNIX if defined on the platform; otherwise, the default is AF_INET.");
Dave Cole331708b2004-08-09 04:51:41 +00003651
3652#endif /* HAVE_SOCKETPAIR */
3653
3654
Guido van Rossum006bf911996-06-12 04:04:55 +00003655static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003656socket_ntohs(PyObject *self, PyObject *args)
Guido van Rossum006bf911996-06-12 04:04:55 +00003657{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003658 int x1, x2;
Guido van Rossum006bf911996-06-12 04:04:55 +00003659
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003660 if (!PyArg_ParseTuple(args, "i:ntohs", &x1)) {
3661 return NULL;
3662 }
3663 if (x1 < 0) {
3664 PyErr_SetString(PyExc_OverflowError,
3665 "can't convert negative number to unsigned long");
3666 return NULL;
3667 }
3668 x2 = (unsigned int)ntohs((unsigned short)x1);
3669 return PyLong_FromLong(x2);
Guido van Rossum006bf911996-06-12 04:04:55 +00003670}
3671
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003672PyDoc_STRVAR(ntohs_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003673"ntohs(integer) -> integer\n\
3674\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003675Convert a 16-bit integer from network to host byte order.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003676
3677
Guido van Rossum006bf911996-06-12 04:04:55 +00003678static PyObject *
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003679socket_ntohl(PyObject *self, PyObject *arg)
Guido van Rossum006bf911996-06-12 04:04:55 +00003680{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003681 unsigned long x;
Guido van Rossum006bf911996-06-12 04:04:55 +00003682
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003683 if (PyLong_Check(arg)) {
3684 x = PyLong_AsUnsignedLong(arg);
3685 if (x == (unsigned long) -1 && PyErr_Occurred())
3686 return NULL;
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003687#if SIZEOF_LONG > 4
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003688 {
3689 unsigned long y;
3690 /* only want the trailing 32 bits */
3691 y = x & 0xFFFFFFFFUL;
3692 if (y ^ x)
3693 return PyErr_Format(PyExc_OverflowError,
3694 "long int larger than 32 bits");
3695 x = y;
3696 }
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003697#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003698 }
3699 else
3700 return PyErr_Format(PyExc_TypeError,
3701 "expected int/long, %s found",
3702 Py_TYPE(arg)->tp_name);
3703 if (x == (unsigned long) -1 && PyErr_Occurred())
3704 return NULL;
3705 return PyLong_FromUnsignedLong(ntohl(x));
Guido van Rossum006bf911996-06-12 04:04:55 +00003706}
3707
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003708PyDoc_STRVAR(ntohl_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003709"ntohl(integer) -> integer\n\
3710\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003711Convert a 32-bit integer from network to host byte order.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003712
3713
Guido van Rossum006bf911996-06-12 04:04:55 +00003714static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003715socket_htons(PyObject *self, PyObject *args)
Guido van Rossum006bf911996-06-12 04:04:55 +00003716{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003717 int x1, x2;
Guido van Rossum006bf911996-06-12 04:04:55 +00003718
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003719 if (!PyArg_ParseTuple(args, "i:htons", &x1)) {
3720 return NULL;
3721 }
3722 if (x1 < 0) {
3723 PyErr_SetString(PyExc_OverflowError,
3724 "can't convert negative number to unsigned long");
3725 return NULL;
3726 }
3727 x2 = (unsigned int)htons((unsigned short)x1);
3728 return PyLong_FromLong(x2);
Guido van Rossum006bf911996-06-12 04:04:55 +00003729}
3730
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003731PyDoc_STRVAR(htons_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003732"htons(integer) -> integer\n\
3733\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003734Convert a 16-bit integer from host to network byte order.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003735
3736
Guido van Rossum006bf911996-06-12 04:04:55 +00003737static PyObject *
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003738socket_htonl(PyObject *self, PyObject *arg)
Guido van Rossum006bf911996-06-12 04:04:55 +00003739{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003740 unsigned long x;
Guido van Rossum006bf911996-06-12 04:04:55 +00003741
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003742 if (PyLong_Check(arg)) {
3743 x = PyLong_AsUnsignedLong(arg);
3744 if (x == (unsigned long) -1 && PyErr_Occurred())
3745 return NULL;
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003746#if SIZEOF_LONG > 4
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003747 {
3748 unsigned long y;
3749 /* only want the trailing 32 bits */
3750 y = x & 0xFFFFFFFFUL;
3751 if (y ^ x)
3752 return PyErr_Format(PyExc_OverflowError,
3753 "long int larger than 32 bits");
3754 x = y;
3755 }
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003756#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003757 }
3758 else
3759 return PyErr_Format(PyExc_TypeError,
3760 "expected int/long, %s found",
3761 Py_TYPE(arg)->tp_name);
3762 return PyLong_FromUnsignedLong(htonl((unsigned long)x));
Guido van Rossum006bf911996-06-12 04:04:55 +00003763}
3764
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003765PyDoc_STRVAR(htonl_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003766"htonl(integer) -> integer\n\
3767\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003768Convert a 32-bit integer from host to network byte order.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003769
Guido van Rossum3eede5a2002-06-07 02:08:35 +00003770/* socket.inet_aton() and socket.inet_ntoa() functions. */
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003771
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003772PyDoc_STRVAR(inet_aton_doc,
Guido van Rossum7d0a8262007-05-21 23:13:11 +00003773"inet_aton(string) -> bytes giving packed 32-bit IP representation\n\
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003774\n\
Guido van Rossumc6a164b1999-08-20 19:11:27 +00003775Convert 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 +00003776binary format used in low-level network functions.");
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003777
3778static PyObject*
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003779socket_inet_aton(PyObject *self, PyObject *args)
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003780{
Guido van Rossuma2e48551999-09-09 15:42:59 +00003781#ifndef INADDR_NONE
3782#define INADDR_NONE (-1)
3783#endif
Neal Norwitz88f115b2003-02-13 02:15:42 +00003784#ifdef HAVE_INET_ATON
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003785 struct in_addr buf;
Thomas Wouters477c8d52006-05-27 19:21:47 +00003786#endif
3787
3788#if !defined(HAVE_INET_ATON) || defined(USE_INET_ATON_WEAKLINK)
Benjamin Petersonf91df042009-02-13 02:50:59 +00003789#if (SIZEOF_INT != 4)
3790#error "Not sure if in_addr_t exists and int is not 32-bits."
3791#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003792 /* Have to use inet_addr() instead */
3793 unsigned int packed_addr;
Tim Peters1df9fdd2003-02-13 03:13:40 +00003794#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003795 char *ip_addr;
Guido van Rossumc6a164b1999-08-20 19:11:27 +00003796
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003797 if (!PyArg_ParseTuple(args, "s:inet_aton", &ip_addr))
3798 return NULL;
Guido van Rossumad05cdf2003-02-12 23:08:22 +00003799
Tim Peters1df9fdd2003-02-13 03:13:40 +00003800
3801#ifdef HAVE_INET_ATON
Thomas Wouters477c8d52006-05-27 19:21:47 +00003802
3803#ifdef USE_INET_ATON_WEAKLINK
3804 if (inet_aton != NULL) {
3805#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003806 if (inet_aton(ip_addr, &buf))
3807 return PyBytes_FromStringAndSize((char *)(&buf),
3808 sizeof(buf));
Guido van Rossumad05cdf2003-02-12 23:08:22 +00003809
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003810 PyErr_SetString(socket_error,
3811 "illegal IP address string passed to inet_aton");
3812 return NULL;
Guido van Rossumad05cdf2003-02-12 23:08:22 +00003813
Thomas Wouters477c8d52006-05-27 19:21:47 +00003814#ifdef USE_INET_ATON_WEAKLINK
3815 } else {
3816#endif
3817
3818#endif
3819
3820#if !defined(HAVE_INET_ATON) || defined(USE_INET_ATON_WEAKLINK)
3821
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003822 /* special-case this address as inet_addr might return INADDR_NONE
3823 * for this */
3824 if (strcmp(ip_addr, "255.255.255.255") == 0) {
3825 packed_addr = 0xFFFFFFFF;
3826 } else {
Thomas Wouters477c8d52006-05-27 19:21:47 +00003827
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003828 packed_addr = inet_addr(ip_addr);
Guido van Rossumc6a164b1999-08-20 19:11:27 +00003829
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003830 if (packed_addr == INADDR_NONE) { /* invalid address */
3831 PyErr_SetString(socket_error,
3832 "illegal IP address string passed to inet_aton");
3833 return NULL;
3834 }
3835 }
3836 return PyBytes_FromStringAndSize((char *) &packed_addr,
3837 sizeof(packed_addr));
Thomas Wouters477c8d52006-05-27 19:21:47 +00003838
3839#ifdef USE_INET_ATON_WEAKLINK
3840 }
3841#endif
3842
Guido van Rossumad05cdf2003-02-12 23:08:22 +00003843#endif
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003844}
3845
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003846PyDoc_STRVAR(inet_ntoa_doc,
Fred Drakee0661342000-03-07 14:05:16 +00003847"inet_ntoa(packed_ip) -> ip_address_string\n\
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003848\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003849Convert an IP address from 32-bit packed binary format to string format");
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003850
3851static PyObject*
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003852socket_inet_ntoa(PyObject *self, PyObject *args)
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003853{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003854 char *packed_str;
3855 int addr_len;
3856 struct in_addr packed_addr;
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003857
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003858 if (!PyArg_ParseTuple(args, "y#:inet_ntoa", &packed_str, &addr_len)) {
3859 return NULL;
3860 }
Guido van Rossum48a680c2001-03-02 06:34:14 +00003861
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003862 if (addr_len != sizeof(packed_addr)) {
3863 PyErr_SetString(socket_error,
3864 "packed IP wrong length for inet_ntoa");
3865 return NULL;
3866 }
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003867
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003868 memcpy(&packed_addr, packed_str, addr_len);
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003869
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003870 return PyUnicode_FromString(inet_ntoa(packed_addr));
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003871}
Guido van Rossum82a5c661998-07-07 20:45:43 +00003872
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003873#ifdef HAVE_INET_PTON
3874
3875PyDoc_STRVAR(inet_pton_doc,
3876"inet_pton(af, ip) -> packed IP address string\n\
3877\n\
3878Convert an IP address from string format to a packed string suitable\n\
3879for use with low-level network functions.");
3880
3881static PyObject *
3882socket_inet_pton(PyObject *self, PyObject *args)
3883{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003884 int af;
3885 char* ip;
3886 int retval;
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003887#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003888 char packed[MAX(sizeof(struct in_addr), sizeof(struct in6_addr))];
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003889#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003890 char packed[sizeof(struct in_addr)];
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003891#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003892 if (!PyArg_ParseTuple(args, "is:inet_pton", &af, &ip)) {
3893 return NULL;
3894 }
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003895
Martin v. Löwis04697e82004-06-02 12:35:29 +00003896#if !defined(ENABLE_IPV6) && defined(AF_INET6)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003897 if(af == AF_INET6) {
3898 PyErr_SetString(socket_error,
3899 "can't use AF_INET6, IPv6 is disabled");
3900 return NULL;
3901 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00003902#endif
Martin v. Löwis10649092003-08-05 06:25:06 +00003903
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003904 retval = inet_pton(af, ip, packed);
3905 if (retval < 0) {
3906 PyErr_SetFromErrno(socket_error);
3907 return NULL;
3908 } else if (retval == 0) {
3909 PyErr_SetString(socket_error,
3910 "illegal IP address string passed to inet_pton");
3911 return NULL;
3912 } else if (af == AF_INET) {
3913 return PyBytes_FromStringAndSize(packed,
3914 sizeof(struct in_addr));
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003915#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003916 } else if (af == AF_INET6) {
3917 return PyBytes_FromStringAndSize(packed,
3918 sizeof(struct in6_addr));
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003919#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003920 } else {
3921 PyErr_SetString(socket_error, "unknown address family");
3922 return NULL;
3923 }
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003924}
Thomas Wouters477c8d52006-05-27 19:21:47 +00003925
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003926PyDoc_STRVAR(inet_ntop_doc,
3927"inet_ntop(af, packed_ip) -> string formatted IP address\n\
3928\n\
3929Convert a packed IP address of the given family to string format.");
3930
3931static PyObject *
3932socket_inet_ntop(PyObject *self, PyObject *args)
3933{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003934 int af;
3935 char* packed;
3936 int len;
3937 const char* retval;
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003938#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003939 char ip[MAX(INET_ADDRSTRLEN, INET6_ADDRSTRLEN) + 1];
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003940#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003941 char ip[INET_ADDRSTRLEN + 1];
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003942#endif
Thomas Wouters477c8d52006-05-27 19:21:47 +00003943
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003944 /* Guarantee NUL-termination for PyUnicode_FromString() below */
3945 memset((void *) &ip[0], '\0', sizeof(ip));
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003946
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003947 if (!PyArg_ParseTuple(args, "iy#:inet_ntop", &af, &packed, &len)) {
3948 return NULL;
3949 }
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003950
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003951 if (af == AF_INET) {
3952 if (len != sizeof(struct in_addr)) {
3953 PyErr_SetString(PyExc_ValueError,
3954 "invalid length of packed IP address string");
3955 return NULL;
3956 }
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003957#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003958 } else if (af == AF_INET6) {
3959 if (len != sizeof(struct in6_addr)) {
3960 PyErr_SetString(PyExc_ValueError,
3961 "invalid length of packed IP address string");
3962 return NULL;
3963 }
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003964#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003965 } else {
3966 PyErr_Format(PyExc_ValueError,
3967 "unknown address family %d", af);
3968 return NULL;
3969 }
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003970
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003971 retval = inet_ntop(af, packed, ip, sizeof(ip));
3972 if (!retval) {
3973 PyErr_SetFromErrno(socket_error);
3974 return NULL;
3975 } else {
3976 return PyUnicode_FromString(retval);
3977 }
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003978
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003979 /* NOTREACHED */
3980 PyErr_SetString(PyExc_RuntimeError, "invalid handling of inet_ntop");
3981 return NULL;
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003982}
3983
3984#endif /* HAVE_INET_PTON */
3985
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003986/* Python interface to getaddrinfo(host, port). */
3987
3988/*ARGSUSED*/
3989static PyObject *
Giampaolo Rodolàccfb91c2010-08-17 15:30:23 +00003990socket_getaddrinfo(PyObject *self, PyObject *args, PyObject* kwargs)
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003991{
Giampaolo Rodolàccfb91c2010-08-17 15:30:23 +00003992 static char* kwnames[] = {"host", "port", "family", "type", "proto",
3993 "flags", 0};
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003994 struct addrinfo hints, *res;
3995 struct addrinfo *res0 = NULL;
3996 PyObject *hobj = NULL;
3997 PyObject *pobj = (PyObject *)NULL;
3998 char pbuf[30];
3999 char *hptr, *pptr;
4000 int family, socktype, protocol, flags;
4001 int error;
4002 PyObject *all = (PyObject *)NULL;
4003 PyObject *idna = NULL;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004004
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004005 family = socktype = protocol = flags = 0;
4006 family = AF_UNSPEC;
Giampaolo Rodolàccfb91c2010-08-17 15:30:23 +00004007 if (!PyArg_ParseTupleAndKeywords(args, kwargs, "OO|iiii:getaddrinfo",
4008 kwnames, &hobj, &pobj, &family, &socktype,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004009 &protocol, &flags)) {
4010 return NULL;
4011 }
4012 if (hobj == Py_None) {
4013 hptr = NULL;
4014 } else if (PyUnicode_Check(hobj)) {
4015 idna = PyObject_CallMethod(hobj, "encode", "s", "idna");
4016 if (!idna)
4017 return NULL;
4018 assert(PyBytes_Check(idna));
4019 hptr = PyBytes_AS_STRING(idna);
4020 } else if (PyBytes_Check(hobj)) {
4021 hptr = PyBytes_AsString(hobj);
4022 } else {
4023 PyErr_SetString(PyExc_TypeError,
4024 "getaddrinfo() argument 1 must be string or None");
4025 return NULL;
4026 }
4027 if (PyLong_CheckExact(pobj)) {
4028 long value = PyLong_AsLong(pobj);
4029 if (value == -1 && PyErr_Occurred())
4030 goto err;
4031 PyOS_snprintf(pbuf, sizeof(pbuf), "%ld", value);
4032 pptr = pbuf;
4033 } else if (PyUnicode_Check(pobj)) {
4034 pptr = _PyUnicode_AsString(pobj);
Alexander Belopolskye239d232010-12-08 23:31:48 +00004035 if (pptr == NULL)
4036 goto err;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004037 } else if (PyBytes_Check(pobj)) {
Alexander Belopolskye239d232010-12-08 23:31:48 +00004038 pptr = PyBytes_AS_STRING(pobj);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004039 } else if (pobj == Py_None) {
4040 pptr = (char *)NULL;
4041 } else {
4042 PyErr_SetString(socket_error, "Int or String expected");
4043 goto err;
4044 }
4045 memset(&hints, 0, sizeof(hints));
4046 hints.ai_family = family;
4047 hints.ai_socktype = socktype;
4048 hints.ai_protocol = protocol;
4049 hints.ai_flags = flags;
4050 Py_BEGIN_ALLOW_THREADS
4051 ACQUIRE_GETADDRINFO_LOCK
4052 error = getaddrinfo(hptr, pptr, &hints, &res0);
4053 Py_END_ALLOW_THREADS
4054 RELEASE_GETADDRINFO_LOCK /* see comment in setipaddr() */
4055 if (error) {
4056 set_gaierror(error);
4057 goto err;
4058 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004059
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004060 if ((all = PyList_New(0)) == NULL)
4061 goto err;
4062 for (res = res0; res; res = res->ai_next) {
4063 PyObject *single;
4064 PyObject *addr =
4065 makesockaddr(-1, res->ai_addr, res->ai_addrlen, protocol);
4066 if (addr == NULL)
4067 goto err;
4068 single = Py_BuildValue("iiisO", res->ai_family,
4069 res->ai_socktype, res->ai_protocol,
4070 res->ai_canonname ? res->ai_canonname : "",
4071 addr);
4072 Py_DECREF(addr);
4073 if (single == NULL)
4074 goto err;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004075
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004076 if (PyList_Append(all, single))
4077 goto err;
4078 Py_XDECREF(single);
4079 }
4080 Py_XDECREF(idna);
4081 if (res0)
4082 freeaddrinfo(res0);
4083 return all;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004084 err:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004085 Py_XDECREF(all);
4086 Py_XDECREF(idna);
4087 if (res0)
4088 freeaddrinfo(res0);
4089 return (PyObject *)NULL;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004090}
4091
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004092PyDoc_STRVAR(getaddrinfo_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00004093"getaddrinfo(host, port [, family, socktype, proto, flags])\n\
4094 -> list of (family, socktype, proto, canonname, sockaddr)\n\
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004095\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004096Resolve host and port into addrinfo struct.");
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004097
4098/* Python interface to getnameinfo(sa, flags). */
4099
4100/*ARGSUSED*/
4101static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004102socket_getnameinfo(PyObject *self, PyObject *args)
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004103{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004104 PyObject *sa = (PyObject *)NULL;
4105 int flags;
4106 char *hostp;
4107 int port, flowinfo, scope_id;
4108 char hbuf[NI_MAXHOST], pbuf[NI_MAXSERV];
4109 struct addrinfo hints, *res = NULL;
4110 int error;
4111 PyObject *ret = (PyObject *)NULL;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004112
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004113 flags = flowinfo = scope_id = 0;
4114 if (!PyArg_ParseTuple(args, "Oi:getnameinfo", &sa, &flags))
4115 return NULL;
4116 if (!PyTuple_Check(sa)) {
4117 PyErr_SetString(PyExc_TypeError,
4118 "getnameinfo() argument 1 must be a tuple");
4119 return NULL;
4120 }
4121 if (!PyArg_ParseTuple(sa, "si|ii",
4122 &hostp, &port, &flowinfo, &scope_id))
4123 return NULL;
4124 PyOS_snprintf(pbuf, sizeof(pbuf), "%d", port);
4125 memset(&hints, 0, sizeof(hints));
4126 hints.ai_family = AF_UNSPEC;
4127 hints.ai_socktype = SOCK_DGRAM; /* make numeric port happy */
Martin v. Löwis112c0f32010-08-25 07:38:15 +00004128 hints.ai_flags = AI_NUMERICHOST; /* don't do any name resolution */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004129 Py_BEGIN_ALLOW_THREADS
4130 ACQUIRE_GETADDRINFO_LOCK
4131 error = getaddrinfo(hostp, pbuf, &hints, &res);
4132 Py_END_ALLOW_THREADS
4133 RELEASE_GETADDRINFO_LOCK /* see comment in setipaddr() */
4134 if (error) {
4135 set_gaierror(error);
4136 goto fail;
4137 }
4138 if (res->ai_next) {
4139 PyErr_SetString(socket_error,
4140 "sockaddr resolved to multiple addresses");
4141 goto fail;
4142 }
4143 switch (res->ai_family) {
4144 case AF_INET:
4145 {
4146 if (PyTuple_GET_SIZE(sa) != 2) {
4147 PyErr_SetString(socket_error,
4148 "IPv4 sockaddr must be 2 tuple");
4149 goto fail;
4150 }
4151 break;
4152 }
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00004153#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004154 case AF_INET6:
4155 {
4156 struct sockaddr_in6 *sin6;
4157 sin6 = (struct sockaddr_in6 *)res->ai_addr;
4158 sin6->sin6_flowinfo = flowinfo;
4159 sin6->sin6_scope_id = scope_id;
4160 break;
4161 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004162#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004163 }
Antoine Pitrou19467d22010-08-17 19:33:30 +00004164 error = getnameinfo(res->ai_addr, (socklen_t) res->ai_addrlen,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004165 hbuf, sizeof(hbuf), pbuf, sizeof(pbuf), flags);
4166 if (error) {
4167 set_gaierror(error);
4168 goto fail;
4169 }
4170 ret = Py_BuildValue("ss", hbuf, pbuf);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004171
4172fail:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004173 if (res)
4174 freeaddrinfo(res);
4175 return ret;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004176}
4177
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004178PyDoc_STRVAR(getnameinfo_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00004179"getnameinfo(sockaddr, flags) --> (host, port)\n\
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004180\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004181Get host and port for a sockaddr.");
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004182
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004183
4184/* Python API to getting and setting the default timeout value. */
4185
4186static PyObject *
4187socket_getdefaulttimeout(PyObject *self)
4188{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004189 if (defaulttimeout < 0.0) {
4190 Py_INCREF(Py_None);
4191 return Py_None;
4192 }
4193 else
4194 return PyFloat_FromDouble(defaulttimeout);
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004195}
4196
4197PyDoc_STRVAR(getdefaulttimeout_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00004198"getdefaulttimeout() -> timeout\n\
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004199\n\
Ezio Melotti388c9452011-08-14 08:28:57 +03004200Returns the default timeout in seconds (float) for new socket objects.\n\
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004201A value of None indicates that new socket objects have no timeout.\n\
4202When the socket module is first imported, the default is None.");
4203
4204static PyObject *
4205socket_setdefaulttimeout(PyObject *self, PyObject *arg)
4206{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004207 double timeout;
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004208
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004209 if (arg == Py_None)
4210 timeout = -1.0;
4211 else {
4212 timeout = PyFloat_AsDouble(arg);
4213 if (timeout < 0.0) {
4214 if (!PyErr_Occurred())
4215 PyErr_SetString(PyExc_ValueError,
4216 "Timeout value out of range");
4217 return NULL;
4218 }
4219 }
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004220
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004221 defaulttimeout = timeout;
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004222
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004223 Py_INCREF(Py_None);
4224 return Py_None;
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004225}
4226
4227PyDoc_STRVAR(setdefaulttimeout_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00004228"setdefaulttimeout(timeout)\n\
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004229\n\
Ezio Melotti388c9452011-08-14 08:28:57 +03004230Set the default timeout in seconds (float) for new socket objects.\n\
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004231A value of None indicates that new socket objects have no timeout.\n\
4232When the socket module is first imported, the default is None.");
4233
4234
Guido van Rossum30a685f1991-06-27 15:51:29 +00004235/* List of functions exported by this module. */
4236
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004237static PyMethodDef socket_methods[] = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004238 {"gethostbyname", socket_gethostbyname,
4239 METH_VARARGS, gethostbyname_doc},
4240 {"gethostbyname_ex", socket_gethostbyname_ex,
4241 METH_VARARGS, ghbn_ex_doc},
4242 {"gethostbyaddr", socket_gethostbyaddr,
4243 METH_VARARGS, gethostbyaddr_doc},
4244 {"gethostname", socket_gethostname,
4245 METH_NOARGS, gethostname_doc},
4246 {"getservbyname", socket_getservbyname,
4247 METH_VARARGS, getservbyname_doc},
4248 {"getservbyport", socket_getservbyport,
4249 METH_VARARGS, getservbyport_doc},
4250 {"getprotobyname", socket_getprotobyname,
4251 METH_VARARGS, getprotobyname_doc},
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00004252#ifndef NO_DUP
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004253 {"dup", socket_dup,
4254 METH_O, dup_doc},
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00004255#endif
Dave Cole331708b2004-08-09 04:51:41 +00004256#ifdef HAVE_SOCKETPAIR
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004257 {"socketpair", socket_socketpair,
4258 METH_VARARGS, socketpair_doc},
Dave Cole331708b2004-08-09 04:51:41 +00004259#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004260 {"ntohs", socket_ntohs,
4261 METH_VARARGS, ntohs_doc},
4262 {"ntohl", socket_ntohl,
4263 METH_O, ntohl_doc},
4264 {"htons", socket_htons,
4265 METH_VARARGS, htons_doc},
4266 {"htonl", socket_htonl,
4267 METH_O, htonl_doc},
4268 {"inet_aton", socket_inet_aton,
4269 METH_VARARGS, inet_aton_doc},
4270 {"inet_ntoa", socket_inet_ntoa,
4271 METH_VARARGS, inet_ntoa_doc},
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00004272#ifdef HAVE_INET_PTON
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004273 {"inet_pton", socket_inet_pton,
4274 METH_VARARGS, inet_pton_doc},
4275 {"inet_ntop", socket_inet_ntop,
4276 METH_VARARGS, inet_ntop_doc},
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00004277#endif
Giampaolo Rodolàccfb91c2010-08-17 15:30:23 +00004278 {"getaddrinfo", (PyCFunction)socket_getaddrinfo,
4279 METH_VARARGS | METH_KEYWORDS, getaddrinfo_doc},
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004280 {"getnameinfo", socket_getnameinfo,
4281 METH_VARARGS, getnameinfo_doc},
4282 {"getdefaulttimeout", (PyCFunction)socket_getdefaulttimeout,
4283 METH_NOARGS, getdefaulttimeout_doc},
4284 {"setdefaulttimeout", socket_setdefaulttimeout,
4285 METH_O, setdefaulttimeout_doc},
4286 {NULL, NULL} /* Sentinel */
Guido van Rossum6574b3e1991-06-25 21:36:08 +00004287};
4288
Guido van Rossum30a685f1991-06-27 15:51:29 +00004289
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004290#ifdef MS_WINDOWS
4291#define OS_INIT_DEFINED
4292
4293/* Additional initialization and cleanup for Windows */
Guido van Rossumbe32c891996-06-20 16:25:29 +00004294
4295static void
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004296os_cleanup(void)
Guido van Rossumbe32c891996-06-20 16:25:29 +00004297{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004298 WSACleanup();
Guido van Rossumbe32c891996-06-20 16:25:29 +00004299}
4300
4301static int
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004302os_init(void)
Guido van Rossumbe32c891996-06-20 16:25:29 +00004303{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004304 WSADATA WSAData;
4305 int ret;
4306 ret = WSAStartup(0x0101, &WSAData);
4307 switch (ret) {
4308 case 0: /* No error */
4309 Py_AtExit(os_cleanup);
4310 return 1; /* Success */
4311 case WSASYSNOTREADY:
4312 PyErr_SetString(PyExc_ImportError,
4313 "WSAStartup failed: network not ready");
4314 break;
4315 case WSAVERNOTSUPPORTED:
4316 case WSAEINVAL:
4317 PyErr_SetString(
4318 PyExc_ImportError,
4319 "WSAStartup failed: requested version not supported");
4320 break;
4321 default:
4322 PyErr_Format(PyExc_ImportError, "WSAStartup failed: error code %d", ret);
4323 break;
4324 }
4325 return 0; /* Failure */
Guido van Rossumbe32c891996-06-20 16:25:29 +00004326}
4327
Guido van Rossum8d665e61996-06-26 18:22:49 +00004328#endif /* MS_WINDOWS */
Guido van Rossumbe32c891996-06-20 16:25:29 +00004329
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004330
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004331#ifdef PYOS_OS2
4332#define OS_INIT_DEFINED
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004333
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004334/* Additional initialization for OS/2 */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004335
4336static int
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004337os_init(void)
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004338{
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004339#ifndef PYCC_GCC
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004340 int rc = sock_init();
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004341
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004342 if (rc == 0) {
4343 return 1; /* Success */
4344 }
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004345
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004346 PyErr_Format(PyExc_ImportError, "OS/2 TCP/IP Error# %d", sock_errno());
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004347
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004348 return 0; /* Failure */
Andrew MacIntyreba43e872002-03-03 03:03:52 +00004349#else
Ezio Melotti13925002011-03-16 11:05:33 +02004350 /* No need to initialize sockets with GCC/EMX */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004351 return 1; /* Success */
Andrew MacIntyreba43e872002-03-03 03:03:52 +00004352#endif
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004353}
4354
4355#endif /* PYOS_OS2 */
4356
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004357
4358#ifndef OS_INIT_DEFINED
4359static int
4360os_init(void)
4361{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004362 return 1; /* Success */
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004363}
4364#endif
4365
4366
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00004367/* C API table - always add new things to the end for binary
4368 compatibility. */
4369static
4370PySocketModule_APIObject PySocketModuleAPI =
4371{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004372 &sock_type,
Antoine Pitrouc4df7842010-12-03 19:59:41 +00004373 NULL,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004374 NULL
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00004375};
4376
Guido van Rossum3eede5a2002-06-07 02:08:35 +00004377
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004378/* Initialize the _socket module.
Guido van Rossum3eede5a2002-06-07 02:08:35 +00004379
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004380 This module is actually called "_socket", and there's a wrapper
Guido van Rossum7d0a8262007-05-21 23:13:11 +00004381 "socket.py" which implements some additional functionality.
4382 The import of "_socket" may fail with an ImportError exception if
4383 os-specific initialization fails. On Windows, this does WINSOCK
Ezio Melotti13925002011-03-16 11:05:33 +02004384 initialization. When WINSOCK is initialized successfully, a call to
Guido van Rossum7d0a8262007-05-21 23:13:11 +00004385 WSACleanup() is scheduled to be made at exit time.
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004386*/
Guido van Rossum3eede5a2002-06-07 02:08:35 +00004387
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004388PyDoc_STRVAR(socket_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00004389"Implementation module for socket operations.\n\
4390\n\
4391See the socket module for documentation.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00004392
Martin v. Löwis1a214512008-06-11 05:26:20 +00004393static struct PyModuleDef socketmodule = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004394 PyModuleDef_HEAD_INIT,
4395 PySocket_MODULE_NAME,
4396 socket_doc,
4397 -1,
4398 socket_methods,
4399 NULL,
4400 NULL,
4401 NULL,
4402 NULL
Martin v. Löwis1a214512008-06-11 05:26:20 +00004403};
4404
Mark Hammondfe51c6d2002-08-02 02:27:13 +00004405PyMODINIT_FUNC
Martin v. Löwis1a214512008-06-11 05:26:20 +00004406PyInit__socket(void)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00004407{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004408 PyObject *m, *has_ipv6;
Fred Drake4baedc12002-04-01 14:53:37 +00004409
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004410 if (!os_init())
4411 return NULL;
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004412
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004413 Py_TYPE(&sock_type) = &PyType_Type;
4414 m = PyModule_Create(&socketmodule);
4415 if (m == NULL)
4416 return NULL;
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004417
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004418 socket_error = PyErr_NewException("socket.error",
4419 PyExc_IOError, NULL);
4420 if (socket_error == NULL)
4421 return NULL;
4422 PySocketModuleAPI.error = socket_error;
4423 Py_INCREF(socket_error);
4424 PyModule_AddObject(m, "error", socket_error);
4425 socket_herror = PyErr_NewException("socket.herror",
4426 socket_error, NULL);
4427 if (socket_herror == NULL)
4428 return NULL;
4429 Py_INCREF(socket_herror);
4430 PyModule_AddObject(m, "herror", socket_herror);
4431 socket_gaierror = PyErr_NewException("socket.gaierror", socket_error,
4432 NULL);
4433 if (socket_gaierror == NULL)
4434 return NULL;
4435 Py_INCREF(socket_gaierror);
4436 PyModule_AddObject(m, "gaierror", socket_gaierror);
4437 socket_timeout = PyErr_NewException("socket.timeout",
4438 socket_error, NULL);
4439 if (socket_timeout == NULL)
4440 return NULL;
Antoine Pitrouc4df7842010-12-03 19:59:41 +00004441 PySocketModuleAPI.timeout_error = socket_timeout;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004442 Py_INCREF(socket_timeout);
4443 PyModule_AddObject(m, "timeout", socket_timeout);
4444 Py_INCREF((PyObject *)&sock_type);
4445 if (PyModule_AddObject(m, "SocketType",
4446 (PyObject *)&sock_type) != 0)
4447 return NULL;
4448 Py_INCREF((PyObject *)&sock_type);
4449 if (PyModule_AddObject(m, "socket",
4450 (PyObject *)&sock_type) != 0)
4451 return NULL;
Guido van Rossum09be4091999-08-09 14:40:40 +00004452
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00004453#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004454 has_ipv6 = Py_True;
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00004455#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004456 has_ipv6 = Py_False;
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00004457#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004458 Py_INCREF(has_ipv6);
4459 PyModule_AddObject(m, "has_ipv6", has_ipv6);
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00004460
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004461 /* Export C API */
4462 if (PyModule_AddObject(m, PySocket_CAPI_NAME,
4463 PyCapsule_New(&PySocketModuleAPI, PySocket_CAPSULE_NAME, NULL)
4464 ) != 0)
4465 return NULL;
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00004466
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004467 /* Address families (we only support AF_INET and AF_UNIX) */
Guido van Rossum09be4091999-08-09 14:40:40 +00004468#ifdef AF_UNSPEC
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004469 PyModule_AddIntConstant(m, "AF_UNSPEC", AF_UNSPEC);
Guido van Rossum09be4091999-08-09 14:40:40 +00004470#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004471 PyModule_AddIntConstant(m, "AF_INET", AF_INET);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004472#ifdef AF_INET6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004473 PyModule_AddIntConstant(m, "AF_INET6", AF_INET6);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004474#endif /* AF_INET6 */
Andrew MacIntyred12dfbb2004-04-04 07:13:49 +00004475#if defined(AF_UNIX)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004476 PyModule_AddIntConstant(m, "AF_UNIX", AF_UNIX);
Guido van Rossumb6775db1994-08-01 11:34:53 +00004477#endif /* AF_UNIX */
Guido van Rossum09be4091999-08-09 14:40:40 +00004478#ifdef AF_AX25
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004479 /* Amateur Radio AX.25 */
4480 PyModule_AddIntConstant(m, "AF_AX25", AF_AX25);
Guido van Rossum09be4091999-08-09 14:40:40 +00004481#endif
4482#ifdef AF_IPX
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004483 PyModule_AddIntConstant(m, "AF_IPX", AF_IPX); /* Novell IPX */
Guido van Rossum09be4091999-08-09 14:40:40 +00004484#endif
4485#ifdef AF_APPLETALK
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004486 /* Appletalk DDP */
4487 PyModule_AddIntConstant(m, "AF_APPLETALK", AF_APPLETALK);
Guido van Rossum09be4091999-08-09 14:40:40 +00004488#endif
4489#ifdef AF_NETROM
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004490 /* Amateur radio NetROM */
4491 PyModule_AddIntConstant(m, "AF_NETROM", AF_NETROM);
Guido van Rossum09be4091999-08-09 14:40:40 +00004492#endif
4493#ifdef AF_BRIDGE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004494 /* Multiprotocol bridge */
4495 PyModule_AddIntConstant(m, "AF_BRIDGE", AF_BRIDGE);
Guido van Rossum09be4091999-08-09 14:40:40 +00004496#endif
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004497#ifdef AF_ATMPVC
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004498 /* ATM PVCs */
4499 PyModule_AddIntConstant(m, "AF_ATMPVC", AF_ATMPVC);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004500#endif
Guido van Rossum09be4091999-08-09 14:40:40 +00004501#ifdef AF_AAL5
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004502 /* Reserved for Werner's ATM */
4503 PyModule_AddIntConstant(m, "AF_AAL5", AF_AAL5);
Guido van Rossum09be4091999-08-09 14:40:40 +00004504#endif
4505#ifdef AF_X25
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004506 /* Reserved for X.25 project */
4507 PyModule_AddIntConstant(m, "AF_X25", AF_X25);
Guido van Rossum09be4091999-08-09 14:40:40 +00004508#endif
4509#ifdef AF_INET6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004510 PyModule_AddIntConstant(m, "AF_INET6", AF_INET6); /* IP version 6 */
Guido van Rossum09be4091999-08-09 14:40:40 +00004511#endif
4512#ifdef AF_ROSE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004513 /* Amateur Radio X.25 PLP */
4514 PyModule_AddIntConstant(m, "AF_ROSE", AF_ROSE);
Guido van Rossum09be4091999-08-09 14:40:40 +00004515#endif
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004516#ifdef AF_DECnet
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004517 /* Reserved for DECnet project */
4518 PyModule_AddIntConstant(m, "AF_DECnet", AF_DECnet);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004519#endif
4520#ifdef AF_NETBEUI
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004521 /* Reserved for 802.2LLC project */
4522 PyModule_AddIntConstant(m, "AF_NETBEUI", AF_NETBEUI);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004523#endif
4524#ifdef AF_SECURITY
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004525 /* Security callback pseudo AF */
4526 PyModule_AddIntConstant(m, "AF_SECURITY", AF_SECURITY);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004527#endif
4528#ifdef AF_KEY
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004529 /* PF_KEY key management API */
4530 PyModule_AddIntConstant(m, "AF_KEY", AF_KEY);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004531#endif
4532#ifdef AF_NETLINK
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004533 /* */
4534 PyModule_AddIntConstant(m, "AF_NETLINK", AF_NETLINK);
4535 PyModule_AddIntConstant(m, "NETLINK_ROUTE", NETLINK_ROUTE);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004536#ifdef NETLINK_SKIP
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004537 PyModule_AddIntConstant(m, "NETLINK_SKIP", NETLINK_SKIP);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004538#endif
4539#ifdef NETLINK_W1
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004540 PyModule_AddIntConstant(m, "NETLINK_W1", NETLINK_W1);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004541#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004542 PyModule_AddIntConstant(m, "NETLINK_USERSOCK", NETLINK_USERSOCK);
4543 PyModule_AddIntConstant(m, "NETLINK_FIREWALL", NETLINK_FIREWALL);
Guido van Rossum668a94a2006-02-21 01:07:27 +00004544#ifdef NETLINK_TCPDIAG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004545 PyModule_AddIntConstant(m, "NETLINK_TCPDIAG", NETLINK_TCPDIAG);
Guido van Rossum668a94a2006-02-21 01:07:27 +00004546#endif
4547#ifdef NETLINK_NFLOG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004548 PyModule_AddIntConstant(m, "NETLINK_NFLOG", NETLINK_NFLOG);
Guido van Rossum668a94a2006-02-21 01:07:27 +00004549#endif
Neal Norwitz65851662006-01-16 04:31:40 +00004550#ifdef NETLINK_XFRM
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004551 PyModule_AddIntConstant(m, "NETLINK_XFRM", NETLINK_XFRM);
Neal Norwitz65851662006-01-16 04:31:40 +00004552#endif
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004553#ifdef NETLINK_ARPD
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004554 PyModule_AddIntConstant(m, "NETLINK_ARPD", NETLINK_ARPD);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004555#endif
4556#ifdef NETLINK_ROUTE6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004557 PyModule_AddIntConstant(m, "NETLINK_ROUTE6", NETLINK_ROUTE6);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004558#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004559 PyModule_AddIntConstant(m, "NETLINK_IP6_FW", NETLINK_IP6_FW);
Thomas Wouterscf297e42007-02-23 15:07:44 +00004560#ifdef NETLINK_DNRTMSG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004561 PyModule_AddIntConstant(m, "NETLINK_DNRTMSG", NETLINK_DNRTMSG);
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00004562#endif
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004563#ifdef NETLINK_TAPBASE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004564 PyModule_AddIntConstant(m, "NETLINK_TAPBASE", NETLINK_TAPBASE);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004565#endif
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004566#endif /* AF_NETLINK */
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004567#ifdef AF_ROUTE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004568 /* Alias to emulate 4.4BSD */
4569 PyModule_AddIntConstant(m, "AF_ROUTE", AF_ROUTE);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004570#endif
4571#ifdef AF_ASH
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004572 /* Ash */
4573 PyModule_AddIntConstant(m, "AF_ASH", AF_ASH);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004574#endif
4575#ifdef AF_ECONET
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004576 /* Acorn Econet */
4577 PyModule_AddIntConstant(m, "AF_ECONET", AF_ECONET);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004578#endif
4579#ifdef AF_ATMSVC
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004580 /* ATM SVCs */
4581 PyModule_AddIntConstant(m, "AF_ATMSVC", AF_ATMSVC);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004582#endif
4583#ifdef AF_SNA
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004584 /* Linux SNA Project (nutters!) */
4585 PyModule_AddIntConstant(m, "AF_SNA", AF_SNA);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004586#endif
4587#ifdef AF_IRDA
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004588 /* IRDA sockets */
4589 PyModule_AddIntConstant(m, "AF_IRDA", AF_IRDA);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004590#endif
4591#ifdef AF_PPPOX
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004592 /* PPPoX sockets */
4593 PyModule_AddIntConstant(m, "AF_PPPOX", AF_PPPOX);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004594#endif
4595#ifdef AF_WANPIPE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004596 /* Wanpipe API Sockets */
4597 PyModule_AddIntConstant(m, "AF_WANPIPE", AF_WANPIPE);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004598#endif
4599#ifdef AF_LLC
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004600 /* Linux LLC */
4601 PyModule_AddIntConstant(m, "AF_LLC", AF_LLC);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004602#endif
Martin v. Löwis12af0482004-01-31 12:34:17 +00004603
Hye-Shik Chang81268602004-02-02 06:05:24 +00004604#ifdef USE_BLUETOOTH
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004605 PyModule_AddIntConstant(m, "AF_BLUETOOTH", AF_BLUETOOTH);
4606 PyModule_AddIntConstant(m, "BTPROTO_L2CAP", BTPROTO_L2CAP);
4607 PyModule_AddIntConstant(m, "BTPROTO_HCI", BTPROTO_HCI);
4608 PyModule_AddIntConstant(m, "SOL_HCI", SOL_HCI);
Gregory P. Smith3e85dfd2010-10-17 00:43:10 +00004609#if !defined(__NetBSD__) && !defined(__DragonFly__)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004610 PyModule_AddIntConstant(m, "HCI_FILTER", HCI_FILTER);
Gregory P. Smith3e85dfd2010-10-17 00:43:10 +00004611#endif
Hye-Shik Chang81268602004-02-02 06:05:24 +00004612#if !defined(__FreeBSD__)
Gregory P. Smith3e85dfd2010-10-17 00:43:10 +00004613#if !defined(__NetBSD__) && !defined(__DragonFly__)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004614 PyModule_AddIntConstant(m, "HCI_TIME_STAMP", HCI_TIME_STAMP);
Gregory P. Smith3e85dfd2010-10-17 00:43:10 +00004615#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004616 PyModule_AddIntConstant(m, "HCI_DATA_DIR", HCI_DATA_DIR);
4617 PyModule_AddIntConstant(m, "BTPROTO_SCO", BTPROTO_SCO);
Hye-Shik Chang81268602004-02-02 06:05:24 +00004618#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004619 PyModule_AddIntConstant(m, "BTPROTO_RFCOMM", BTPROTO_RFCOMM);
4620 PyModule_AddStringConstant(m, "BDADDR_ANY", "00:00:00:00:00:00");
4621 PyModule_AddStringConstant(m, "BDADDR_LOCAL", "00:00:00:FF:FF:FF");
Martin v. Löwis12af0482004-01-31 12:34:17 +00004622#endif
4623
Antoine Pitroub156a462010-10-27 20:13:57 +00004624#ifdef AF_PACKET
4625 PyModule_AddIntMacro(m, AF_PACKET);
4626#endif
4627#ifdef PF_PACKET
4628 PyModule_AddIntMacro(m, PF_PACKET);
4629#endif
4630#ifdef PACKET_HOST
4631 PyModule_AddIntMacro(m, PACKET_HOST);
4632#endif
4633#ifdef PACKET_BROADCAST
4634 PyModule_AddIntMacro(m, PACKET_BROADCAST);
4635#endif
4636#ifdef PACKET_MULTICAST
4637 PyModule_AddIntMacro(m, PACKET_MULTICAST);
4638#endif
4639#ifdef PACKET_OTHERHOST
4640 PyModule_AddIntMacro(m, PACKET_OTHERHOST);
4641#endif
4642#ifdef PACKET_OUTGOING
4643 PyModule_AddIntMacro(m, PACKET_OUTGOING);
4644#endif
4645#ifdef PACKET_LOOPBACK
4646 PyModule_AddIntMacro(m, PACKET_LOOPBACK);
4647#endif
4648#ifdef PACKET_FASTROUTE
4649 PyModule_AddIntMacro(m, PACKET_FASTROUTE);
Guido van Rossum48a680c2001-03-02 06:34:14 +00004650#endif
Guido van Rossum09be4091999-08-09 14:40:40 +00004651
Christian Heimes043d6f62008-01-07 17:19:16 +00004652#ifdef HAVE_LINUX_TIPC_H
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004653 PyModule_AddIntConstant(m, "AF_TIPC", AF_TIPC);
Christian Heimes043d6f62008-01-07 17:19:16 +00004654
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004655 /* for addresses */
4656 PyModule_AddIntConstant(m, "TIPC_ADDR_NAMESEQ", TIPC_ADDR_NAMESEQ);
4657 PyModule_AddIntConstant(m, "TIPC_ADDR_NAME", TIPC_ADDR_NAME);
4658 PyModule_AddIntConstant(m, "TIPC_ADDR_ID", TIPC_ADDR_ID);
Christian Heimes043d6f62008-01-07 17:19:16 +00004659
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004660 PyModule_AddIntConstant(m, "TIPC_ZONE_SCOPE", TIPC_ZONE_SCOPE);
4661 PyModule_AddIntConstant(m, "TIPC_CLUSTER_SCOPE", TIPC_CLUSTER_SCOPE);
4662 PyModule_AddIntConstant(m, "TIPC_NODE_SCOPE", TIPC_NODE_SCOPE);
Christian Heimes043d6f62008-01-07 17:19:16 +00004663
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004664 /* for setsockopt() */
4665 PyModule_AddIntConstant(m, "SOL_TIPC", SOL_TIPC);
4666 PyModule_AddIntConstant(m, "TIPC_IMPORTANCE", TIPC_IMPORTANCE);
4667 PyModule_AddIntConstant(m, "TIPC_SRC_DROPPABLE", TIPC_SRC_DROPPABLE);
4668 PyModule_AddIntConstant(m, "TIPC_DEST_DROPPABLE",
4669 TIPC_DEST_DROPPABLE);
4670 PyModule_AddIntConstant(m, "TIPC_CONN_TIMEOUT", TIPC_CONN_TIMEOUT);
Christian Heimes043d6f62008-01-07 17:19:16 +00004671
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004672 PyModule_AddIntConstant(m, "TIPC_LOW_IMPORTANCE",
4673 TIPC_LOW_IMPORTANCE);
4674 PyModule_AddIntConstant(m, "TIPC_MEDIUM_IMPORTANCE",
4675 TIPC_MEDIUM_IMPORTANCE);
4676 PyModule_AddIntConstant(m, "TIPC_HIGH_IMPORTANCE",
4677 TIPC_HIGH_IMPORTANCE);
4678 PyModule_AddIntConstant(m, "TIPC_CRITICAL_IMPORTANCE",
4679 TIPC_CRITICAL_IMPORTANCE);
Christian Heimes043d6f62008-01-07 17:19:16 +00004680
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004681 /* for subscriptions */
4682 PyModule_AddIntConstant(m, "TIPC_SUB_PORTS", TIPC_SUB_PORTS);
4683 PyModule_AddIntConstant(m, "TIPC_SUB_SERVICE", TIPC_SUB_SERVICE);
Christian Heimes25bb7832008-01-11 16:17:00 +00004684#ifdef TIPC_SUB_CANCEL
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004685 /* doesn't seem to be available everywhere */
4686 PyModule_AddIntConstant(m, "TIPC_SUB_CANCEL", TIPC_SUB_CANCEL);
Christian Heimes25bb7832008-01-11 16:17:00 +00004687#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004688 PyModule_AddIntConstant(m, "TIPC_WAIT_FOREVER", TIPC_WAIT_FOREVER);
4689 PyModule_AddIntConstant(m, "TIPC_PUBLISHED", TIPC_PUBLISHED);
4690 PyModule_AddIntConstant(m, "TIPC_WITHDRAWN", TIPC_WITHDRAWN);
4691 PyModule_AddIntConstant(m, "TIPC_SUBSCR_TIMEOUT", TIPC_SUBSCR_TIMEOUT);
4692 PyModule_AddIntConstant(m, "TIPC_CFG_SRV", TIPC_CFG_SRV);
4693 PyModule_AddIntConstant(m, "TIPC_TOP_SRV", TIPC_TOP_SRV);
Christian Heimes043d6f62008-01-07 17:19:16 +00004694#endif
4695
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004696 /* Socket types */
4697 PyModule_AddIntConstant(m, "SOCK_STREAM", SOCK_STREAM);
4698 PyModule_AddIntConstant(m, "SOCK_DGRAM", SOCK_DGRAM);
Guido van Rossumbcc20741998-08-04 22:53:56 +00004699/* We have incomplete socket support. */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004700 PyModule_AddIntConstant(m, "SOCK_RAW", SOCK_RAW);
4701 PyModule_AddIntConstant(m, "SOCK_SEQPACKET", SOCK_SEQPACKET);
Martin v. Löwiscf8f47e2002-12-11 13:10:57 +00004702#if defined(SOCK_RDM)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004703 PyModule_AddIntConstant(m, "SOCK_RDM", SOCK_RDM);
Guido van Rossumbcc20741998-08-04 22:53:56 +00004704#endif
Antoine Pitroub1c54962010-10-14 15:05:38 +00004705#ifdef SOCK_CLOEXEC
4706 PyModule_AddIntConstant(m, "SOCK_CLOEXEC", SOCK_CLOEXEC);
4707#endif
4708#ifdef SOCK_NONBLOCK
4709 PyModule_AddIntConstant(m, "SOCK_NONBLOCK", SOCK_NONBLOCK);
4710#endif
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004711
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004712#ifdef SO_DEBUG
4713 PyModule_AddIntConstant(m, "SO_DEBUG", SO_DEBUG);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004714#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004715#ifdef SO_ACCEPTCONN
4716 PyModule_AddIntConstant(m, "SO_ACCEPTCONN", SO_ACCEPTCONN);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004717#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004718#ifdef SO_REUSEADDR
4719 PyModule_AddIntConstant(m, "SO_REUSEADDR", SO_REUSEADDR);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004720#endif
Andrew M. Kuchling42851ab2004-07-10 14:19:21 +00004721#ifdef SO_EXCLUSIVEADDRUSE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004722 PyModule_AddIntConstant(m, "SO_EXCLUSIVEADDRUSE", SO_EXCLUSIVEADDRUSE);
Andrew M. Kuchling42851ab2004-07-10 14:19:21 +00004723#endif
4724
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004725#ifdef SO_KEEPALIVE
4726 PyModule_AddIntConstant(m, "SO_KEEPALIVE", SO_KEEPALIVE);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004727#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004728#ifdef SO_DONTROUTE
4729 PyModule_AddIntConstant(m, "SO_DONTROUTE", SO_DONTROUTE);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004730#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004731#ifdef SO_BROADCAST
4732 PyModule_AddIntConstant(m, "SO_BROADCAST", SO_BROADCAST);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004733#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004734#ifdef SO_USELOOPBACK
4735 PyModule_AddIntConstant(m, "SO_USELOOPBACK", SO_USELOOPBACK);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004736#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004737#ifdef SO_LINGER
4738 PyModule_AddIntConstant(m, "SO_LINGER", SO_LINGER);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004739#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004740#ifdef SO_OOBINLINE
4741 PyModule_AddIntConstant(m, "SO_OOBINLINE", SO_OOBINLINE);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004742#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004743#ifdef SO_REUSEPORT
4744 PyModule_AddIntConstant(m, "SO_REUSEPORT", SO_REUSEPORT);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004745#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004746#ifdef SO_SNDBUF
4747 PyModule_AddIntConstant(m, "SO_SNDBUF", SO_SNDBUF);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004748#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004749#ifdef SO_RCVBUF
4750 PyModule_AddIntConstant(m, "SO_RCVBUF", SO_RCVBUF);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004751#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004752#ifdef SO_SNDLOWAT
4753 PyModule_AddIntConstant(m, "SO_SNDLOWAT", SO_SNDLOWAT);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004754#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004755#ifdef SO_RCVLOWAT
4756 PyModule_AddIntConstant(m, "SO_RCVLOWAT", SO_RCVLOWAT);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004757#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004758#ifdef SO_SNDTIMEO
4759 PyModule_AddIntConstant(m, "SO_SNDTIMEO", SO_SNDTIMEO);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004760#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004761#ifdef SO_RCVTIMEO
4762 PyModule_AddIntConstant(m, "SO_RCVTIMEO", SO_RCVTIMEO);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004763#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004764#ifdef SO_ERROR
4765 PyModule_AddIntConstant(m, "SO_ERROR", SO_ERROR);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004766#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004767#ifdef SO_TYPE
4768 PyModule_AddIntConstant(m, "SO_TYPE", SO_TYPE);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004769#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004770#ifdef SO_SETFIB
4771 PyModule_AddIntConstant(m, "SO_SETFIB", SO_SETFIB);
Larry Hastingsf0f37952010-04-02 11:47:10 +00004772#endif
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004773
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004774 /* Maximum number of connections for "listen" */
4775#ifdef SOMAXCONN
4776 PyModule_AddIntConstant(m, "SOMAXCONN", SOMAXCONN);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004777#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004778 PyModule_AddIntConstant(m, "SOMAXCONN", 5); /* Common value */
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004779#endif
4780
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004781 /* Flags for send, recv */
4782#ifdef MSG_OOB
4783 PyModule_AddIntConstant(m, "MSG_OOB", MSG_OOB);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004784#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004785#ifdef MSG_PEEK
4786 PyModule_AddIntConstant(m, "MSG_PEEK", MSG_PEEK);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004787#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004788#ifdef MSG_DONTROUTE
4789 PyModule_AddIntConstant(m, "MSG_DONTROUTE", MSG_DONTROUTE);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004790#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004791#ifdef MSG_DONTWAIT
4792 PyModule_AddIntConstant(m, "MSG_DONTWAIT", MSG_DONTWAIT);
Guido van Rossum2c8bcb82000-04-25 21:34:53 +00004793#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004794#ifdef MSG_EOR
4795 PyModule_AddIntConstant(m, "MSG_EOR", MSG_EOR);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004796#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004797#ifdef MSG_TRUNC
4798 PyModule_AddIntConstant(m, "MSG_TRUNC", MSG_TRUNC);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004799#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004800#ifdef MSG_CTRUNC
4801 PyModule_AddIntConstant(m, "MSG_CTRUNC", MSG_CTRUNC);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004802#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004803#ifdef MSG_WAITALL
4804 PyModule_AddIntConstant(m, "MSG_WAITALL", MSG_WAITALL);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004805#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004806#ifdef MSG_BTAG
4807 PyModule_AddIntConstant(m, "MSG_BTAG", MSG_BTAG);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004808#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004809#ifdef MSG_ETAG
4810 PyModule_AddIntConstant(m, "MSG_ETAG", MSG_ETAG);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004811#endif
4812
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004813 /* Protocol level and numbers, usable for [gs]etsockopt */
4814#ifdef SOL_SOCKET
4815 PyModule_AddIntConstant(m, "SOL_SOCKET", SOL_SOCKET);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004816#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004817#ifdef SOL_IP
4818 PyModule_AddIntConstant(m, "SOL_IP", SOL_IP);
Guido van Rossum09be4091999-08-09 14:40:40 +00004819#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004820 PyModule_AddIntConstant(m, "SOL_IP", 0);
Guido van Rossum09be4091999-08-09 14:40:40 +00004821#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004822#ifdef SOL_IPX
4823 PyModule_AddIntConstant(m, "SOL_IPX", SOL_IPX);
Guido van Rossum09be4091999-08-09 14:40:40 +00004824#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004825#ifdef SOL_AX25
4826 PyModule_AddIntConstant(m, "SOL_AX25", SOL_AX25);
Guido van Rossum09be4091999-08-09 14:40:40 +00004827#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004828#ifdef SOL_ATALK
4829 PyModule_AddIntConstant(m, "SOL_ATALK", SOL_ATALK);
Guido van Rossum09be4091999-08-09 14:40:40 +00004830#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004831#ifdef SOL_NETROM
4832 PyModule_AddIntConstant(m, "SOL_NETROM", SOL_NETROM);
Guido van Rossum09be4091999-08-09 14:40:40 +00004833#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004834#ifdef SOL_ROSE
4835 PyModule_AddIntConstant(m, "SOL_ROSE", SOL_ROSE);
Guido van Rossum09be4091999-08-09 14:40:40 +00004836#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004837#ifdef SOL_TCP
4838 PyModule_AddIntConstant(m, "SOL_TCP", SOL_TCP);
Guido van Rossum09be4091999-08-09 14:40:40 +00004839#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004840 PyModule_AddIntConstant(m, "SOL_TCP", 6);
Guido van Rossum09be4091999-08-09 14:40:40 +00004841#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004842#ifdef SOL_UDP
4843 PyModule_AddIntConstant(m, "SOL_UDP", SOL_UDP);
Guido van Rossum09be4091999-08-09 14:40:40 +00004844#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004845 PyModule_AddIntConstant(m, "SOL_UDP", 17);
Guido van Rossum09be4091999-08-09 14:40:40 +00004846#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004847#ifdef IPPROTO_IP
4848 PyModule_AddIntConstant(m, "IPPROTO_IP", IPPROTO_IP);
Guido van Rossum578de301998-05-28 20:18:18 +00004849#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004850 PyModule_AddIntConstant(m, "IPPROTO_IP", 0);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004851#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004852#ifdef IPPROTO_HOPOPTS
4853 PyModule_AddIntConstant(m, "IPPROTO_HOPOPTS", IPPROTO_HOPOPTS);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004854#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004855#ifdef IPPROTO_ICMP
4856 PyModule_AddIntConstant(m, "IPPROTO_ICMP", IPPROTO_ICMP);
Guido van Rossum578de301998-05-28 20:18:18 +00004857#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004858 PyModule_AddIntConstant(m, "IPPROTO_ICMP", 1);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004859#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004860#ifdef IPPROTO_IGMP
4861 PyModule_AddIntConstant(m, "IPPROTO_IGMP", IPPROTO_IGMP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004862#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004863#ifdef IPPROTO_GGP
4864 PyModule_AddIntConstant(m, "IPPROTO_GGP", IPPROTO_GGP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004865#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004866#ifdef IPPROTO_IPV4
4867 PyModule_AddIntConstant(m, "IPPROTO_IPV4", IPPROTO_IPV4);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004868#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004869#ifdef IPPROTO_IPV6
4870 PyModule_AddIntConstant(m, "IPPROTO_IPV6", IPPROTO_IPV6);
Martin v. Löwisa0f17342003-10-03 13:56:20 +00004871#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004872#ifdef IPPROTO_IPIP
4873 PyModule_AddIntConstant(m, "IPPROTO_IPIP", IPPROTO_IPIP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004874#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004875#ifdef IPPROTO_TCP
4876 PyModule_AddIntConstant(m, "IPPROTO_TCP", IPPROTO_TCP);
Guido van Rossum578de301998-05-28 20:18:18 +00004877#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004878 PyModule_AddIntConstant(m, "IPPROTO_TCP", 6);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004879#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004880#ifdef IPPROTO_EGP
4881 PyModule_AddIntConstant(m, "IPPROTO_EGP", IPPROTO_EGP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004882#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004883#ifdef IPPROTO_PUP
4884 PyModule_AddIntConstant(m, "IPPROTO_PUP", IPPROTO_PUP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004885#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004886#ifdef IPPROTO_UDP
4887 PyModule_AddIntConstant(m, "IPPROTO_UDP", IPPROTO_UDP);
Guido van Rossum578de301998-05-28 20:18:18 +00004888#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004889 PyModule_AddIntConstant(m, "IPPROTO_UDP", 17);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004890#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004891#ifdef IPPROTO_IDP
4892 PyModule_AddIntConstant(m, "IPPROTO_IDP", IPPROTO_IDP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004893#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004894#ifdef IPPROTO_HELLO
4895 PyModule_AddIntConstant(m, "IPPROTO_HELLO", IPPROTO_HELLO);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004896#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004897#ifdef IPPROTO_ND
4898 PyModule_AddIntConstant(m, "IPPROTO_ND", IPPROTO_ND);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004899#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004900#ifdef IPPROTO_TP
4901 PyModule_AddIntConstant(m, "IPPROTO_TP", IPPROTO_TP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004902#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004903#ifdef IPPROTO_IPV6
4904 PyModule_AddIntConstant(m, "IPPROTO_IPV6", IPPROTO_IPV6);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004905#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004906#ifdef IPPROTO_ROUTING
4907 PyModule_AddIntConstant(m, "IPPROTO_ROUTING", IPPROTO_ROUTING);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004908#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004909#ifdef IPPROTO_FRAGMENT
4910 PyModule_AddIntConstant(m, "IPPROTO_FRAGMENT", IPPROTO_FRAGMENT);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004911#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004912#ifdef IPPROTO_RSVP
4913 PyModule_AddIntConstant(m, "IPPROTO_RSVP", IPPROTO_RSVP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004914#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004915#ifdef IPPROTO_GRE
4916 PyModule_AddIntConstant(m, "IPPROTO_GRE", IPPROTO_GRE);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004917#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004918#ifdef IPPROTO_ESP
4919 PyModule_AddIntConstant(m, "IPPROTO_ESP", IPPROTO_ESP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004920#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004921#ifdef IPPROTO_AH
4922 PyModule_AddIntConstant(m, "IPPROTO_AH", IPPROTO_AH);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004923#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004924#ifdef IPPROTO_MOBILE
4925 PyModule_AddIntConstant(m, "IPPROTO_MOBILE", IPPROTO_MOBILE);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004926#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004927#ifdef IPPROTO_ICMPV6
4928 PyModule_AddIntConstant(m, "IPPROTO_ICMPV6", IPPROTO_ICMPV6);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004929#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004930#ifdef IPPROTO_NONE
4931 PyModule_AddIntConstant(m, "IPPROTO_NONE", IPPROTO_NONE);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004932#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004933#ifdef IPPROTO_DSTOPTS
4934 PyModule_AddIntConstant(m, "IPPROTO_DSTOPTS", IPPROTO_DSTOPTS);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004935#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004936#ifdef IPPROTO_XTP
4937 PyModule_AddIntConstant(m, "IPPROTO_XTP", IPPROTO_XTP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004938#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004939#ifdef IPPROTO_EON
4940 PyModule_AddIntConstant(m, "IPPROTO_EON", IPPROTO_EON);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004941#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004942#ifdef IPPROTO_PIM
4943 PyModule_AddIntConstant(m, "IPPROTO_PIM", IPPROTO_PIM);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004944#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004945#ifdef IPPROTO_IPCOMP
4946 PyModule_AddIntConstant(m, "IPPROTO_IPCOMP", IPPROTO_IPCOMP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004947#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004948#ifdef IPPROTO_VRRP
4949 PyModule_AddIntConstant(m, "IPPROTO_VRRP", IPPROTO_VRRP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004950#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004951#ifdef IPPROTO_BIP
4952 PyModule_AddIntConstant(m, "IPPROTO_BIP", IPPROTO_BIP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004953#endif
4954/**/
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004955#ifdef IPPROTO_RAW
4956 PyModule_AddIntConstant(m, "IPPROTO_RAW", IPPROTO_RAW);
Guido van Rossum578de301998-05-28 20:18:18 +00004957#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004958 PyModule_AddIntConstant(m, "IPPROTO_RAW", 255);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004959#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004960#ifdef IPPROTO_MAX
4961 PyModule_AddIntConstant(m, "IPPROTO_MAX", IPPROTO_MAX);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004962#endif
4963
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004964 /* Some port configuration */
4965#ifdef IPPORT_RESERVED
4966 PyModule_AddIntConstant(m, "IPPORT_RESERVED", IPPORT_RESERVED);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004967#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004968 PyModule_AddIntConstant(m, "IPPORT_RESERVED", 1024);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004969#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004970#ifdef IPPORT_USERRESERVED
4971 PyModule_AddIntConstant(m, "IPPORT_USERRESERVED", IPPORT_USERRESERVED);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004972#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004973 PyModule_AddIntConstant(m, "IPPORT_USERRESERVED", 5000);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004974#endif
4975
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004976 /* Some reserved IP v.4 addresses */
4977#ifdef INADDR_ANY
4978 PyModule_AddIntConstant(m, "INADDR_ANY", INADDR_ANY);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004979#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004980 PyModule_AddIntConstant(m, "INADDR_ANY", 0x00000000);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004981#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004982#ifdef INADDR_BROADCAST
4983 PyModule_AddIntConstant(m, "INADDR_BROADCAST", INADDR_BROADCAST);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004984#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004985 PyModule_AddIntConstant(m, "INADDR_BROADCAST", 0xffffffff);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004986#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004987#ifdef INADDR_LOOPBACK
4988 PyModule_AddIntConstant(m, "INADDR_LOOPBACK", INADDR_LOOPBACK);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004989#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004990 PyModule_AddIntConstant(m, "INADDR_LOOPBACK", 0x7F000001);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004991#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004992#ifdef INADDR_UNSPEC_GROUP
4993 PyModule_AddIntConstant(m, "INADDR_UNSPEC_GROUP", INADDR_UNSPEC_GROUP);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004994#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004995 PyModule_AddIntConstant(m, "INADDR_UNSPEC_GROUP", 0xe0000000);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004996#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004997#ifdef INADDR_ALLHOSTS_GROUP
4998 PyModule_AddIntConstant(m, "INADDR_ALLHOSTS_GROUP",
4999 INADDR_ALLHOSTS_GROUP);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00005000#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005001 PyModule_AddIntConstant(m, "INADDR_ALLHOSTS_GROUP", 0xe0000001);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00005002#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005003#ifdef INADDR_MAX_LOCAL_GROUP
5004 PyModule_AddIntConstant(m, "INADDR_MAX_LOCAL_GROUP",
5005 INADDR_MAX_LOCAL_GROUP);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00005006#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005007 PyModule_AddIntConstant(m, "INADDR_MAX_LOCAL_GROUP", 0xe00000ff);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00005008#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005009#ifdef INADDR_NONE
5010 PyModule_AddIntConstant(m, "INADDR_NONE", INADDR_NONE);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00005011#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005012 PyModule_AddIntConstant(m, "INADDR_NONE", 0xffffffff);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00005013#endif
5014
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005015 /* IPv4 [gs]etsockopt options */
5016#ifdef IP_OPTIONS
5017 PyModule_AddIntConstant(m, "IP_OPTIONS", IP_OPTIONS);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00005018#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005019#ifdef IP_HDRINCL
5020 PyModule_AddIntConstant(m, "IP_HDRINCL", IP_HDRINCL);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00005021#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005022#ifdef IP_TOS
5023 PyModule_AddIntConstant(m, "IP_TOS", IP_TOS);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00005024#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005025#ifdef IP_TTL
5026 PyModule_AddIntConstant(m, "IP_TTL", IP_TTL);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00005027#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005028#ifdef IP_RECVOPTS
5029 PyModule_AddIntConstant(m, "IP_RECVOPTS", IP_RECVOPTS);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00005030#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005031#ifdef IP_RECVRETOPTS
5032 PyModule_AddIntConstant(m, "IP_RECVRETOPTS", IP_RECVRETOPTS);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00005033#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005034#ifdef IP_RECVDSTADDR
5035 PyModule_AddIntConstant(m, "IP_RECVDSTADDR", IP_RECVDSTADDR);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00005036#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005037#ifdef IP_RETOPTS
5038 PyModule_AddIntConstant(m, "IP_RETOPTS", IP_RETOPTS);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00005039#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005040#ifdef IP_MULTICAST_IF
5041 PyModule_AddIntConstant(m, "IP_MULTICAST_IF", IP_MULTICAST_IF);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00005042#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005043#ifdef IP_MULTICAST_TTL
5044 PyModule_AddIntConstant(m, "IP_MULTICAST_TTL", IP_MULTICAST_TTL);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00005045#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005046#ifdef IP_MULTICAST_LOOP
5047 PyModule_AddIntConstant(m, "IP_MULTICAST_LOOP", IP_MULTICAST_LOOP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00005048#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005049#ifdef IP_ADD_MEMBERSHIP
5050 PyModule_AddIntConstant(m, "IP_ADD_MEMBERSHIP", IP_ADD_MEMBERSHIP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00005051#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005052#ifdef IP_DROP_MEMBERSHIP
5053 PyModule_AddIntConstant(m, "IP_DROP_MEMBERSHIP", IP_DROP_MEMBERSHIP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00005054#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005055#ifdef IP_DEFAULT_MULTICAST_TTL
5056 PyModule_AddIntConstant(m, "IP_DEFAULT_MULTICAST_TTL",
5057 IP_DEFAULT_MULTICAST_TTL);
Guido van Rossum09be4091999-08-09 14:40:40 +00005058#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005059#ifdef IP_DEFAULT_MULTICAST_LOOP
5060 PyModule_AddIntConstant(m, "IP_DEFAULT_MULTICAST_LOOP",
5061 IP_DEFAULT_MULTICAST_LOOP);
Guido van Rossum09be4091999-08-09 14:40:40 +00005062#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005063#ifdef IP_MAX_MEMBERSHIPS
5064 PyModule_AddIntConstant(m, "IP_MAX_MEMBERSHIPS", IP_MAX_MEMBERSHIPS);
Guido van Rossum09be4091999-08-09 14:40:40 +00005065#endif
5066
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005067 /* IPv6 [gs]etsockopt options, defined in RFC2553 */
5068#ifdef IPV6_JOIN_GROUP
5069 PyModule_AddIntConstant(m, "IPV6_JOIN_GROUP", IPV6_JOIN_GROUP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00005070#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005071#ifdef IPV6_LEAVE_GROUP
5072 PyModule_AddIntConstant(m, "IPV6_LEAVE_GROUP", IPV6_LEAVE_GROUP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00005073#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005074#ifdef IPV6_MULTICAST_HOPS
5075 PyModule_AddIntConstant(m, "IPV6_MULTICAST_HOPS", IPV6_MULTICAST_HOPS);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00005076#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005077#ifdef IPV6_MULTICAST_IF
5078 PyModule_AddIntConstant(m, "IPV6_MULTICAST_IF", IPV6_MULTICAST_IF);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00005079#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005080#ifdef IPV6_MULTICAST_LOOP
5081 PyModule_AddIntConstant(m, "IPV6_MULTICAST_LOOP", IPV6_MULTICAST_LOOP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00005082#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005083#ifdef IPV6_UNICAST_HOPS
5084 PyModule_AddIntConstant(m, "IPV6_UNICAST_HOPS", IPV6_UNICAST_HOPS);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00005085#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005086 /* Additional IPV6 socket options, defined in RFC 3493 */
Martin v. Löwisda91d022003-12-30 11:14:01 +00005087#ifdef IPV6_V6ONLY
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005088 PyModule_AddIntConstant(m, "IPV6_V6ONLY", IPV6_V6ONLY);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005089#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005090 /* Advanced IPV6 socket options, from RFC 3542 */
Martin v. Löwisda91d022003-12-30 11:14:01 +00005091#ifdef IPV6_CHECKSUM
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005092 PyModule_AddIntConstant(m, "IPV6_CHECKSUM", IPV6_CHECKSUM);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005093#endif
5094#ifdef IPV6_DONTFRAG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005095 PyModule_AddIntConstant(m, "IPV6_DONTFRAG", IPV6_DONTFRAG);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005096#endif
5097#ifdef IPV6_DSTOPTS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005098 PyModule_AddIntConstant(m, "IPV6_DSTOPTS", IPV6_DSTOPTS);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005099#endif
5100#ifdef IPV6_HOPLIMIT
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005101 PyModule_AddIntConstant(m, "IPV6_HOPLIMIT", IPV6_HOPLIMIT);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005102#endif
5103#ifdef IPV6_HOPOPTS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005104 PyModule_AddIntConstant(m, "IPV6_HOPOPTS", IPV6_HOPOPTS);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005105#endif
5106#ifdef IPV6_NEXTHOP
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005107 PyModule_AddIntConstant(m, "IPV6_NEXTHOP", IPV6_NEXTHOP);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005108#endif
5109#ifdef IPV6_PATHMTU
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005110 PyModule_AddIntConstant(m, "IPV6_PATHMTU", IPV6_PATHMTU);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005111#endif
5112#ifdef IPV6_PKTINFO
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005113 PyModule_AddIntConstant(m, "IPV6_PKTINFO", IPV6_PKTINFO);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005114#endif
5115#ifdef IPV6_RECVDSTOPTS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005116 PyModule_AddIntConstant(m, "IPV6_RECVDSTOPTS", IPV6_RECVDSTOPTS);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005117#endif
5118#ifdef IPV6_RECVHOPLIMIT
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005119 PyModule_AddIntConstant(m, "IPV6_RECVHOPLIMIT", IPV6_RECVHOPLIMIT);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005120#endif
5121#ifdef IPV6_RECVHOPOPTS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005122 PyModule_AddIntConstant(m, "IPV6_RECVHOPOPTS", IPV6_RECVHOPOPTS);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005123#endif
5124#ifdef IPV6_RECVPKTINFO
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005125 PyModule_AddIntConstant(m, "IPV6_RECVPKTINFO", IPV6_RECVPKTINFO);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005126#endif
5127#ifdef IPV6_RECVRTHDR
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005128 PyModule_AddIntConstant(m, "IPV6_RECVRTHDR", IPV6_RECVRTHDR);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005129#endif
5130#ifdef IPV6_RECVTCLASS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005131 PyModule_AddIntConstant(m, "IPV6_RECVTCLASS", IPV6_RECVTCLASS);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005132#endif
5133#ifdef IPV6_RTHDR
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005134 PyModule_AddIntConstant(m, "IPV6_RTHDR", IPV6_RTHDR);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005135#endif
5136#ifdef IPV6_RTHDRDSTOPTS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005137 PyModule_AddIntConstant(m, "IPV6_RTHDRDSTOPTS", IPV6_RTHDRDSTOPTS);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005138#endif
5139#ifdef IPV6_RTHDR_TYPE_0
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005140 PyModule_AddIntConstant(m, "IPV6_RTHDR_TYPE_0", IPV6_RTHDR_TYPE_0);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005141#endif
5142#ifdef IPV6_RECVPATHMTU
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005143 PyModule_AddIntConstant(m, "IPV6_RECVPATHMTU", IPV6_RECVPATHMTU);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005144#endif
5145#ifdef IPV6_TCLASS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005146 PyModule_AddIntConstant(m, "IPV6_TCLASS", IPV6_TCLASS);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005147#endif
5148#ifdef IPV6_USE_MIN_MTU
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005149 PyModule_AddIntConstant(m, "IPV6_USE_MIN_MTU", IPV6_USE_MIN_MTU);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005150#endif
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00005151
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005152 /* TCP options */
5153#ifdef TCP_NODELAY
5154 PyModule_AddIntConstant(m, "TCP_NODELAY", TCP_NODELAY);
Guido van Rossum09be4091999-08-09 14:40:40 +00005155#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005156#ifdef TCP_MAXSEG
5157 PyModule_AddIntConstant(m, "TCP_MAXSEG", TCP_MAXSEG);
Guido van Rossum09be4091999-08-09 14:40:40 +00005158#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005159#ifdef TCP_CORK
5160 PyModule_AddIntConstant(m, "TCP_CORK", TCP_CORK);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00005161#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005162#ifdef TCP_KEEPIDLE
5163 PyModule_AddIntConstant(m, "TCP_KEEPIDLE", TCP_KEEPIDLE);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00005164#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005165#ifdef TCP_KEEPINTVL
5166 PyModule_AddIntConstant(m, "TCP_KEEPINTVL", TCP_KEEPINTVL);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00005167#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005168#ifdef TCP_KEEPCNT
5169 PyModule_AddIntConstant(m, "TCP_KEEPCNT", TCP_KEEPCNT);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00005170#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005171#ifdef TCP_SYNCNT
5172 PyModule_AddIntConstant(m, "TCP_SYNCNT", TCP_SYNCNT);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00005173#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005174#ifdef TCP_LINGER2
5175 PyModule_AddIntConstant(m, "TCP_LINGER2", TCP_LINGER2);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00005176#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005177#ifdef TCP_DEFER_ACCEPT
5178 PyModule_AddIntConstant(m, "TCP_DEFER_ACCEPT", TCP_DEFER_ACCEPT);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00005179#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005180#ifdef TCP_WINDOW_CLAMP
5181 PyModule_AddIntConstant(m, "TCP_WINDOW_CLAMP", TCP_WINDOW_CLAMP);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00005182#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005183#ifdef TCP_INFO
5184 PyModule_AddIntConstant(m, "TCP_INFO", TCP_INFO);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00005185#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005186#ifdef TCP_QUICKACK
5187 PyModule_AddIntConstant(m, "TCP_QUICKACK", TCP_QUICKACK);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00005188#endif
5189
Guido van Rossum09be4091999-08-09 14:40:40 +00005190
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005191 /* IPX options */
5192#ifdef IPX_TYPE
5193 PyModule_AddIntConstant(m, "IPX_TYPE", IPX_TYPE);
Guido van Rossum09be4091999-08-09 14:40:40 +00005194#endif
Guido van Rossum4f199ea1998-04-09 20:56:35 +00005195
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005196 /* get{addr,name}info parameters */
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005197#ifdef EAI_ADDRFAMILY
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005198 PyModule_AddIntConstant(m, "EAI_ADDRFAMILY", EAI_ADDRFAMILY);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005199#endif
5200#ifdef EAI_AGAIN
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005201 PyModule_AddIntConstant(m, "EAI_AGAIN", EAI_AGAIN);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005202#endif
5203#ifdef EAI_BADFLAGS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005204 PyModule_AddIntConstant(m, "EAI_BADFLAGS", EAI_BADFLAGS);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005205#endif
5206#ifdef EAI_FAIL
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005207 PyModule_AddIntConstant(m, "EAI_FAIL", EAI_FAIL);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005208#endif
5209#ifdef EAI_FAMILY
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005210 PyModule_AddIntConstant(m, "EAI_FAMILY", EAI_FAMILY);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005211#endif
5212#ifdef EAI_MEMORY
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005213 PyModule_AddIntConstant(m, "EAI_MEMORY", EAI_MEMORY);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005214#endif
5215#ifdef EAI_NODATA
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005216 PyModule_AddIntConstant(m, "EAI_NODATA", EAI_NODATA);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005217#endif
5218#ifdef EAI_NONAME
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005219 PyModule_AddIntConstant(m, "EAI_NONAME", EAI_NONAME);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005220#endif
Martin v. Löwisda91d022003-12-30 11:14:01 +00005221#ifdef EAI_OVERFLOW
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005222 PyModule_AddIntConstant(m, "EAI_OVERFLOW", EAI_OVERFLOW);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005223#endif
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005224#ifdef EAI_SERVICE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005225 PyModule_AddIntConstant(m, "EAI_SERVICE", EAI_SERVICE);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005226#endif
5227#ifdef EAI_SOCKTYPE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005228 PyModule_AddIntConstant(m, "EAI_SOCKTYPE", EAI_SOCKTYPE);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005229#endif
5230#ifdef EAI_SYSTEM
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005231 PyModule_AddIntConstant(m, "EAI_SYSTEM", EAI_SYSTEM);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005232#endif
5233#ifdef EAI_BADHINTS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005234 PyModule_AddIntConstant(m, "EAI_BADHINTS", EAI_BADHINTS);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005235#endif
5236#ifdef EAI_PROTOCOL
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005237 PyModule_AddIntConstant(m, "EAI_PROTOCOL", EAI_PROTOCOL);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005238#endif
5239#ifdef EAI_MAX
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005240 PyModule_AddIntConstant(m, "EAI_MAX", EAI_MAX);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005241#endif
5242#ifdef AI_PASSIVE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005243 PyModule_AddIntConstant(m, "AI_PASSIVE", AI_PASSIVE);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005244#endif
5245#ifdef AI_CANONNAME
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005246 PyModule_AddIntConstant(m, "AI_CANONNAME", AI_CANONNAME);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005247#endif
5248#ifdef AI_NUMERICHOST
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005249 PyModule_AddIntConstant(m, "AI_NUMERICHOST", AI_NUMERICHOST);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005250#endif
Martin v. Löwisda91d022003-12-30 11:14:01 +00005251#ifdef AI_NUMERICSERV
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005252 PyModule_AddIntConstant(m, "AI_NUMERICSERV", AI_NUMERICSERV);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005253#endif
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005254#ifdef AI_MASK
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005255 PyModule_AddIntConstant(m, "AI_MASK", AI_MASK);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005256#endif
5257#ifdef AI_ALL
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005258 PyModule_AddIntConstant(m, "AI_ALL", AI_ALL);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005259#endif
5260#ifdef AI_V4MAPPED_CFG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005261 PyModule_AddIntConstant(m, "AI_V4MAPPED_CFG", AI_V4MAPPED_CFG);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005262#endif
5263#ifdef AI_ADDRCONFIG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005264 PyModule_AddIntConstant(m, "AI_ADDRCONFIG", AI_ADDRCONFIG);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005265#endif
5266#ifdef AI_V4MAPPED
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005267 PyModule_AddIntConstant(m, "AI_V4MAPPED", AI_V4MAPPED);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005268#endif
5269#ifdef AI_DEFAULT
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005270 PyModule_AddIntConstant(m, "AI_DEFAULT", AI_DEFAULT);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005271#endif
5272#ifdef NI_MAXHOST
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005273 PyModule_AddIntConstant(m, "NI_MAXHOST", NI_MAXHOST);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005274#endif
5275#ifdef NI_MAXSERV
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005276 PyModule_AddIntConstant(m, "NI_MAXSERV", NI_MAXSERV);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005277#endif
5278#ifdef NI_NOFQDN
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005279 PyModule_AddIntConstant(m, "NI_NOFQDN", NI_NOFQDN);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005280#endif
5281#ifdef NI_NUMERICHOST
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005282 PyModule_AddIntConstant(m, "NI_NUMERICHOST", NI_NUMERICHOST);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005283#endif
5284#ifdef NI_NAMEREQD
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005285 PyModule_AddIntConstant(m, "NI_NAMEREQD", NI_NAMEREQD);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005286#endif
5287#ifdef NI_NUMERICSERV
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005288 PyModule_AddIntConstant(m, "NI_NUMERICSERV", NI_NUMERICSERV);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005289#endif
5290#ifdef NI_DGRAM
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005291 PyModule_AddIntConstant(m, "NI_DGRAM", NI_DGRAM);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005292#endif
5293
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005294 /* shutdown() parameters */
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005295#ifdef SHUT_RD
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005296 PyModule_AddIntConstant(m, "SHUT_RD", SHUT_RD);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005297#elif defined(SD_RECEIVE)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005298 PyModule_AddIntConstant(m, "SHUT_RD", SD_RECEIVE);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005299#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005300 PyModule_AddIntConstant(m, "SHUT_RD", 0);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005301#endif
5302#ifdef SHUT_WR
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005303 PyModule_AddIntConstant(m, "SHUT_WR", SHUT_WR);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005304#elif defined(SD_SEND)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005305 PyModule_AddIntConstant(m, "SHUT_WR", SD_SEND);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005306#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005307 PyModule_AddIntConstant(m, "SHUT_WR", 1);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005308#endif
5309#ifdef SHUT_RDWR
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005310 PyModule_AddIntConstant(m, "SHUT_RDWR", SHUT_RDWR);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005311#elif defined(SD_BOTH)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005312 PyModule_AddIntConstant(m, "SHUT_RDWR", SD_BOTH);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005313#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005314 PyModule_AddIntConstant(m, "SHUT_RDWR", 2);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005315#endif
5316
Christian Heimesfaf2f632008-01-06 16:59:19 +00005317#ifdef SIO_RCVALL
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005318 {
5319 DWORD codes[] = {SIO_RCVALL, SIO_KEEPALIVE_VALS};
5320 const char *names[] = {"SIO_RCVALL", "SIO_KEEPALIVE_VALS"};
5321 int i;
5322 for(i = 0; i<sizeof(codes)/sizeof(*codes); ++i) {
5323 PyObject *tmp;
5324 tmp = PyLong_FromUnsignedLong(codes[i]);
5325 if (tmp == NULL)
5326 return NULL;
5327 PyModule_AddObject(m, names[i], tmp);
5328 }
5329 }
5330 PyModule_AddIntConstant(m, "RCVALL_OFF", RCVALL_OFF);
5331 PyModule_AddIntConstant(m, "RCVALL_ON", RCVALL_ON);
5332 PyModule_AddIntConstant(m, "RCVALL_SOCKETLEVELONLY", RCVALL_SOCKETLEVELONLY);
Amaury Forgeot d'Arc762681b2008-06-12 23:03:41 +00005333#ifdef RCVALL_IPLEVEL
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005334 PyModule_AddIntConstant(m, "RCVALL_IPLEVEL", RCVALL_IPLEVEL);
Amaury Forgeot d'Arc762681b2008-06-12 23:03:41 +00005335#endif
5336#ifdef RCVALL_MAX
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005337 PyModule_AddIntConstant(m, "RCVALL_MAX", RCVALL_MAX);
Amaury Forgeot d'Arc762681b2008-06-12 23:03:41 +00005338#endif
Christian Heimesfaf2f632008-01-06 16:59:19 +00005339#endif /* _MSTCPIP_ */
5340
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005341 /* Initialize gethostbyname lock */
Just van Rossum1040d2c2003-05-09 07:53:18 +00005342#if defined(USE_GETHOSTBYNAME_LOCK) || defined(USE_GETADDRINFO_LOCK)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005343 netdb_lock = PyThread_allocate_lock();
Guido van Rossum4f199ea1998-04-09 20:56:35 +00005344#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005345 return m;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00005346}
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005347
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00005348
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005349#ifndef HAVE_INET_PTON
Christian Heimes96e7b3d2007-11-20 06:51:17 +00005350#if !defined(NTDDI_VERSION) || (NTDDI_VERSION < NTDDI_LONGHORN)
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00005351
5352/* Simplistic emulation code for inet_pton that only works for IPv4 */
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00005353/* These are not exposed because they do not set errno properly */
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00005354
Guido van Rossum3eede5a2002-06-07 02:08:35 +00005355int
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00005356inet_pton(int af, const char *src, void *dst)
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005357{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005358 if (af == AF_INET) {
Benjamin Petersonf91df042009-02-13 02:50:59 +00005359#if (SIZEOF_INT != 4)
5360#error "Not sure if in_addr_t exists and int is not 32-bits."
5361#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005362 unsigned int packed_addr;
5363 packed_addr = inet_addr(src);
5364 if (packed_addr == INADDR_NONE)
5365 return 0;
5366 memcpy(dst, &packed_addr, 4);
5367 return 1;
5368 }
5369 /* Should set errno to EAFNOSUPPORT */
5370 return -1;
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005371}
5372
Martin v. Löwisc925b1532001-07-21 09:42:15 +00005373const char *
5374inet_ntop(int af, const void *src, char *dst, socklen_t size)
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005375{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005376 if (af == AF_INET) {
5377 struct in_addr packed_addr;
5378 if (size < 16)
5379 /* Should set errno to ENOSPC. */
5380 return NULL;
5381 memcpy(&packed_addr, src, sizeof(packed_addr));
5382 return strncpy(dst, inet_ntoa(packed_addr), size);
5383 }
5384 /* Should set errno to EAFNOSUPPORT */
5385 return NULL;
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005386}
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00005387
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005388#endif
Christian Heimesb6150692007-11-15 23:37:07 +00005389#endif