blob: 903e10c98cc81c251879ed6561b473ce4ffab123 [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 &&
Antoine Pitrou4de74572013-02-09 23:11:27 +0100522 Py_ISSPACE(Py_CHARMASK(*lastc))) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000523 /* 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)
Matthias Klosed182a6c2012-08-14 17:29:04 +0200807static PyThread_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;
Charles-François Natali366999a2012-01-02 15:47:29 +01001068 ret = Py_BuildValue("OiII",
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001069 addrobj,
1070 ntohs(a->sin6_port),
Charles-François Natali366999a2012-01-02 15:47:29 +01001071 ntohl(a->sin6_flowinfo),
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001072 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;
Charles-François Natali366999a2012-01-02 15:47:29 +01001322 int port, result;
1323 unsigned int flowinfo, scope_id;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001324 flowinfo = scope_id = 0;
1325 if (!PyTuple_Check(args)) {
1326 PyErr_Format(
1327 PyExc_TypeError,
1328 "getsockaddrarg: "
1329 "AF_INET6 address must be tuple, not %.500s",
1330 Py_TYPE(args)->tp_name);
1331 return 0;
1332 }
Charles-François Natali366999a2012-01-02 15:47:29 +01001333 if (!PyArg_ParseTuple(args, "eti|II",
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001334 "idna", &host, &port, &flowinfo,
1335 &scope_id)) {
1336 return 0;
1337 }
1338 addr = (struct sockaddr_in6*)addr_ret;
1339 result = setipaddr(host, (struct sockaddr *)addr,
1340 sizeof(*addr), AF_INET6);
1341 PyMem_Free(host);
1342 if (result < 0)
1343 return 0;
1344 if (port < 0 || port > 0xffff) {
1345 PyErr_SetString(
1346 PyExc_OverflowError,
1347 "getsockaddrarg: port must be 0-65535.");
1348 return 0;
1349 }
Charles-François Natali9b0c0062012-06-23 10:17:05 +02001350 if (flowinfo > 0xfffff) {
Charles-François Natali366999a2012-01-02 15:47:29 +01001351 PyErr_SetString(
1352 PyExc_OverflowError,
1353 "getsockaddrarg: flowinfo must be 0-1048575.");
1354 return 0;
1355 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001356 addr->sin6_family = s->sock_family;
1357 addr->sin6_port = htons((short)port);
Charles-François Natali366999a2012-01-02 15:47:29 +01001358 addr->sin6_flowinfo = htonl(flowinfo);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001359 addr->sin6_scope_id = scope_id;
1360 *len_ret = sizeof *addr;
1361 return 1;
1362 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00001363#endif
1364
Hye-Shik Chang81268602004-02-02 06:05:24 +00001365#ifdef USE_BLUETOOTH
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001366 case AF_BLUETOOTH:
1367 {
1368 switch (s->sock_proto) {
1369 case BTPROTO_L2CAP:
1370 {
1371 struct sockaddr_l2 *addr;
1372 char *straddr;
Martin v. Löwis12af0482004-01-31 12:34:17 +00001373
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001374 addr = (struct sockaddr_l2 *)addr_ret;
1375 memset(addr, 0, sizeof(struct sockaddr_l2));
1376 _BT_L2_MEMB(addr, family) = AF_BLUETOOTH;
1377 if (!PyArg_ParseTuple(args, "si", &straddr,
1378 &_BT_L2_MEMB(addr, psm))) {
1379 PyErr_SetString(socket_error, "getsockaddrarg: "
1380 "wrong format");
1381 return 0;
1382 }
1383 if (setbdaddr(straddr, &_BT_L2_MEMB(addr, bdaddr)) < 0)
1384 return 0;
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001385
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001386 *len_ret = sizeof *addr;
1387 return 1;
1388 }
1389 case BTPROTO_RFCOMM:
1390 {
1391 struct sockaddr_rc *addr;
1392 char *straddr;
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001393
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001394 addr = (struct sockaddr_rc *)addr_ret;
1395 _BT_RC_MEMB(addr, family) = AF_BLUETOOTH;
1396 if (!PyArg_ParseTuple(args, "si", &straddr,
1397 &_BT_RC_MEMB(addr, channel))) {
1398 PyErr_SetString(socket_error, "getsockaddrarg: "
1399 "wrong format");
1400 return 0;
1401 }
1402 if (setbdaddr(straddr, &_BT_RC_MEMB(addr, bdaddr)) < 0)
1403 return 0;
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001404
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001405 *len_ret = sizeof *addr;
1406 return 1;
1407 }
1408 case BTPROTO_HCI:
1409 {
1410 struct sockaddr_hci *addr = (struct sockaddr_hci *)addr_ret;
Gregory P. Smith3e85dfd2010-10-17 00:43:10 +00001411#if defined(__NetBSD__) || defined(__DragonFly__)
Alexander Belopolskye239d232010-12-08 23:31:48 +00001412 char *straddr = PyBytes_AS_STRING(args);
Gregory P. Smith3e85dfd2010-10-17 00:43:10 +00001413
Alexander Belopolskye239d232010-12-08 23:31:48 +00001414 _BT_HCI_MEMB(addr, family) = AF_BLUETOOTH;
Gregory P. Smith3e85dfd2010-10-17 00:43:10 +00001415 if (straddr == NULL) {
1416 PyErr_SetString(socket_error, "getsockaddrarg: "
1417 "wrong format");
1418 return 0;
1419 }
1420 if (setbdaddr(straddr, &_BT_HCI_MEMB(addr, bdaddr)) < 0)
1421 return 0;
1422#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001423 _BT_HCI_MEMB(addr, family) = AF_BLUETOOTH;
1424 if (!PyArg_ParseTuple(args, "i", &_BT_HCI_MEMB(addr, dev))) {
1425 PyErr_SetString(socket_error, "getsockaddrarg: "
1426 "wrong format");
1427 return 0;
1428 }
Gregory P. Smith3e85dfd2010-10-17 00:43:10 +00001429#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001430 *len_ret = sizeof *addr;
1431 return 1;
1432 }
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001433#if !defined(__FreeBSD__)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001434 case BTPROTO_SCO:
1435 {
1436 struct sockaddr_sco *addr;
1437 char *straddr;
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001438
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001439 addr = (struct sockaddr_sco *)addr_ret;
1440 _BT_SCO_MEMB(addr, family) = AF_BLUETOOTH;
1441 if (!PyBytes_Check(args)) {
1442 PyErr_SetString(socket_error, "getsockaddrarg: "
1443 "wrong format");
1444 return 0;
1445 }
1446 straddr = PyBytes_AS_STRING(args);
1447 if (setbdaddr(straddr, &_BT_SCO_MEMB(addr, bdaddr)) < 0)
1448 return 0;
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001449
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001450 *len_ret = sizeof *addr;
1451 return 1;
1452 }
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001453#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001454 default:
1455 PyErr_SetString(socket_error, "getsockaddrarg: unknown Bluetooth protocol");
1456 return 0;
1457 }
1458 }
Martin v. Löwis12af0482004-01-31 12:34:17 +00001459#endif
1460
Antoine Pitroub156a462010-10-27 20:13:57 +00001461#if defined(HAVE_NETPACKET_PACKET_H) && defined(SIOCGIFINDEX)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001462 case AF_PACKET:
1463 {
1464 struct sockaddr_ll* addr;
1465 struct ifreq ifr;
1466 char *interfaceName;
1467 int protoNumber;
1468 int hatype = 0;
1469 int pkttype = 0;
1470 char *haddr = NULL;
1471 unsigned int halen = 0;
Guido van Rossum48a680c2001-03-02 06:34:14 +00001472
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001473 if (!PyTuple_Check(args)) {
1474 PyErr_Format(
1475 PyExc_TypeError,
1476 "getsockaddrarg: "
1477 "AF_PACKET address must be tuple, not %.500s",
1478 Py_TYPE(args)->tp_name);
1479 return 0;
1480 }
1481 if (!PyArg_ParseTuple(args, "si|iiy#", &interfaceName,
1482 &protoNumber, &pkttype, &hatype,
1483 &haddr, &halen))
1484 return 0;
1485 strncpy(ifr.ifr_name, interfaceName, sizeof(ifr.ifr_name));
1486 ifr.ifr_name[(sizeof(ifr.ifr_name))-1] = '\0';
1487 if (ioctl(s->sock_fd, SIOCGIFINDEX, &ifr) < 0) {
1488 s->errorhandler();
1489 return 0;
1490 }
1491 if (halen > 8) {
1492 PyErr_SetString(PyExc_ValueError,
1493 "Hardware address must be 8 bytes or less");
1494 return 0;
1495 }
1496 if (protoNumber < 0 || protoNumber > 0xffff) {
1497 PyErr_SetString(
1498 PyExc_OverflowError,
1499 "getsockaddrarg: protoNumber must be 0-65535.");
1500 return 0;
1501 }
1502 addr = (struct sockaddr_ll*)addr_ret;
1503 addr->sll_family = AF_PACKET;
1504 addr->sll_protocol = htons((short)protoNumber);
1505 addr->sll_ifindex = ifr.ifr_ifindex;
1506 addr->sll_pkttype = pkttype;
1507 addr->sll_hatype = hatype;
1508 if (halen != 0) {
1509 memcpy(&addr->sll_addr, haddr, halen);
1510 }
1511 addr->sll_halen = halen;
1512 *len_ret = sizeof *addr;
1513 return 1;
1514 }
Guido van Rossum48a680c2001-03-02 06:34:14 +00001515#endif
1516
Christian Heimes043d6f62008-01-07 17:19:16 +00001517#ifdef HAVE_LINUX_TIPC_H
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001518 case AF_TIPC:
1519 {
1520 unsigned int atype, v1, v2, v3;
1521 unsigned int scope = TIPC_CLUSTER_SCOPE;
1522 struct sockaddr_tipc *addr;
Christian Heimes043d6f62008-01-07 17:19:16 +00001523
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001524 if (!PyTuple_Check(args)) {
1525 PyErr_Format(
1526 PyExc_TypeError,
1527 "getsockaddrarg: "
1528 "AF_TIPC address must be tuple, not %.500s",
1529 Py_TYPE(args)->tp_name);
1530 return 0;
1531 }
Christian Heimes043d6f62008-01-07 17:19:16 +00001532
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001533 if (!PyArg_ParseTuple(args,
1534 "IIII|I;Invalid TIPC address format",
1535 &atype, &v1, &v2, &v3, &scope))
1536 return 0;
Christian Heimes043d6f62008-01-07 17:19:16 +00001537
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001538 addr = (struct sockaddr_tipc *) addr_ret;
1539 memset(addr, 0, sizeof(struct sockaddr_tipc));
Christian Heimes043d6f62008-01-07 17:19:16 +00001540
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001541 addr->family = AF_TIPC;
1542 addr->scope = scope;
1543 addr->addrtype = atype;
Christian Heimes043d6f62008-01-07 17:19:16 +00001544
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001545 if (atype == TIPC_ADDR_NAMESEQ) {
1546 addr->addr.nameseq.type = v1;
1547 addr->addr.nameseq.lower = v2;
1548 addr->addr.nameseq.upper = v3;
1549 } else if (atype == TIPC_ADDR_NAME) {
1550 addr->addr.name.name.type = v1;
1551 addr->addr.name.name.instance = v2;
1552 } else if (atype == TIPC_ADDR_ID) {
1553 addr->addr.id.node = v1;
1554 addr->addr.id.ref = v2;
1555 } else {
1556 /* Shouldn't happen */
1557 PyErr_SetString(PyExc_TypeError, "Invalid address type");
1558 return 0;
1559 }
Christian Heimes043d6f62008-01-07 17:19:16 +00001560
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001561 *len_ret = sizeof(*addr);
Christian Heimes043d6f62008-01-07 17:19:16 +00001562
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001563 return 1;
1564 }
Christian Heimes043d6f62008-01-07 17:19:16 +00001565#endif
1566
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001567 /* More cases here... */
Guido van Rossum30a685f1991-06-27 15:51:29 +00001568
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001569 default:
1570 PyErr_SetString(socket_error, "getsockaddrarg: bad family");
1571 return 0;
Guido van Rossum30a685f1991-06-27 15:51:29 +00001572
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001573 }
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001574}
1575
Guido van Rossum30a685f1991-06-27 15:51:29 +00001576
Guido van Rossum48a680c2001-03-02 06:34:14 +00001577/* Get the address length according to the socket object's address family.
Guido van Rossum710e1df1992-06-12 10:39:36 +00001578 Return 1 if the family is known, 0 otherwise. The length is returned
1579 through len_ret. */
1580
1581static int
Peter Schneider-Kamp618e25d2000-07-11 23:00:12 +00001582getsockaddrlen(PySocketSockObject *s, socklen_t *len_ret)
Guido van Rossum710e1df1992-06-12 10:39:36 +00001583{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001584 switch (s->sock_family) {
Guido van Rossum710e1df1992-06-12 10:39:36 +00001585
Andrew MacIntyred12dfbb2004-04-04 07:13:49 +00001586#if defined(AF_UNIX)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001587 case AF_UNIX:
1588 {
1589 *len_ret = sizeof (struct sockaddr_un);
1590 return 1;
1591 }
Guido van Rossumb6775db1994-08-01 11:34:53 +00001592#endif /* AF_UNIX */
Martin v. Löwis11017b12006-01-14 18:12:57 +00001593#if defined(AF_NETLINK)
1594 case AF_NETLINK:
1595 {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001596 *len_ret = sizeof (struct sockaddr_nl);
1597 return 1;
Martin v. Löwis11017b12006-01-14 18:12:57 +00001598 }
1599#endif
Guido van Rossum710e1df1992-06-12 10:39:36 +00001600
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001601 case AF_INET:
1602 {
1603 *len_ret = sizeof (struct sockaddr_in);
1604 return 1;
1605 }
Guido van Rossum710e1df1992-06-12 10:39:36 +00001606
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00001607#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001608 case AF_INET6:
1609 {
1610 *len_ret = sizeof (struct sockaddr_in6);
1611 return 1;
1612 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00001613#endif
1614
Hye-Shik Chang81268602004-02-02 06:05:24 +00001615#ifdef USE_BLUETOOTH
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001616 case AF_BLUETOOTH:
1617 {
1618 switch(s->sock_proto)
1619 {
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001620
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001621 case BTPROTO_L2CAP:
1622 *len_ret = sizeof (struct sockaddr_l2);
1623 return 1;
1624 case BTPROTO_RFCOMM:
1625 *len_ret = sizeof (struct sockaddr_rc);
1626 return 1;
1627 case BTPROTO_HCI:
1628 *len_ret = sizeof (struct sockaddr_hci);
1629 return 1;
Hye-Shik Chang81268602004-02-02 06:05:24 +00001630#if !defined(__FreeBSD__)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001631 case BTPROTO_SCO:
1632 *len_ret = sizeof (struct sockaddr_sco);
1633 return 1;
Hye-Shik Chang81268602004-02-02 06:05:24 +00001634#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001635 default:
1636 PyErr_SetString(socket_error, "getsockaddrlen: "
1637 "unknown BT protocol");
1638 return 0;
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001639
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001640 }
1641 }
Martin v. Löwis12af0482004-01-31 12:34:17 +00001642#endif
1643
Martin v. Löwis1ba3fd52001-08-10 20:29:40 +00001644#ifdef HAVE_NETPACKET_PACKET_H
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001645 case AF_PACKET:
1646 {
1647 *len_ret = sizeof (struct sockaddr_ll);
1648 return 1;
1649 }
Jeremy Hylton22308652001-02-02 03:23:09 +00001650#endif
Guido van Rossum48a680c2001-03-02 06:34:14 +00001651
Christian Heimes043d6f62008-01-07 17:19:16 +00001652#ifdef HAVE_LINUX_TIPC_H
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001653 case AF_TIPC:
1654 {
1655 *len_ret = sizeof (struct sockaddr_tipc);
1656 return 1;
1657 }
Christian Heimes043d6f62008-01-07 17:19:16 +00001658#endif
1659
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001660 /* More cases here... */
Guido van Rossum710e1df1992-06-12 10:39:36 +00001661
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001662 default:
1663 PyErr_SetString(socket_error, "getsockaddrlen: bad family");
1664 return 0;
Guido van Rossum710e1df1992-06-12 10:39:36 +00001665
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001666 }
Guido van Rossum710e1df1992-06-12 10:39:36 +00001667}
1668
1669
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00001670/* s._accept() -> (fd, address) */
Guido van Rossum30a685f1991-06-27 15:51:29 +00001671
Guido van Rossum73624e91994-10-10 17:59:00 +00001672static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001673sock_accept(PySocketSockObject *s)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001674{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001675 sock_addr_t addrbuf;
1676 SOCKET_T newfd = INVALID_SOCKET;
1677 socklen_t addrlen;
1678 PyObject *sock = NULL;
1679 PyObject *addr = NULL;
1680 PyObject *res = NULL;
1681 int timeout;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001682 if (!getsockaddrlen(s, &addrlen))
1683 return NULL;
1684 memset(&addrbuf, 0, addrlen);
Guido van Rossum67f7a382002-06-06 21:08:16 +00001685
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001686 if (!IS_SELECTABLE(s))
1687 return select_error();
Neal Norwitz082b2df2006-02-07 07:04:46 +00001688
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00001689 BEGIN_SELECT_LOOP(s)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001690 Py_BEGIN_ALLOW_THREADS
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00001691 timeout = internal_select_ex(s, 0, interval);
Antoine Pitroub1c54962010-10-14 15:05:38 +00001692 if (!timeout) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001693 newfd = accept(s->sock_fd, SAS2SA(&addrbuf), &addrlen);
Antoine Pitroub1c54962010-10-14 15:05:38 +00001694 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001695 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00001696
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001697 if (timeout == 1) {
1698 PyErr_SetString(socket_timeout, "timed out");
1699 return NULL;
1700 }
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00001701 END_SELECT_LOOP(s)
Raymond Hettingeref7343c2003-06-29 03:08:05 +00001702
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001703 if (newfd == INVALID_SOCKET)
1704 return s->errorhandler();
Barry Warsaw752300b1997-01-03 17:18:10 +00001705
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001706 sock = PyLong_FromSocket_t(newfd);
1707 if (sock == NULL) {
1708 SOCKETCLOSE(newfd);
1709 goto finally;
1710 }
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00001711
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001712 addr = makesockaddr(s->sock_fd, SAS2SA(&addrbuf),
1713 addrlen, s->sock_proto);
1714 if (addr == NULL)
1715 goto finally;
Barry Warsaw752300b1997-01-03 17:18:10 +00001716
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001717 res = PyTuple_Pack(2, sock, addr);
Barry Warsaw752300b1997-01-03 17:18:10 +00001718
Guido van Rossum67f7a382002-06-06 21:08:16 +00001719finally:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001720 Py_XDECREF(sock);
1721 Py_XDECREF(addr);
1722 return res;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001723}
1724
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001725PyDoc_STRVAR(accept_doc,
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00001726"_accept() -> (integer, address info)\n\
Guido van Rossum82a5c661998-07-07 20:45:43 +00001727\n\
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00001728Wait for an incoming connection. Return a new socket file descriptor\n\
1729representing the connection, and the address of the client.\n\
1730For IP sockets, the address info is a pair (hostaddr, port).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001731
Guido van Rossum11ba0942002-06-13 15:07:44 +00001732/* s.setblocking(flag) method. Argument:
1733 False -- non-blocking mode; same as settimeout(0)
1734 True -- blocking mode; same as settimeout(None)
1735*/
Guido van Rossume4485b01994-09-07 14:32:49 +00001736
Guido van Rossum73624e91994-10-10 17:59:00 +00001737static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001738sock_setblocking(PySocketSockObject *s, PyObject *arg)
Guido van Rossume4485b01994-09-07 14:32:49 +00001739{
Serhiy Storchaka441d30f2013-01-19 12:26:26 +02001740 long block;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001741
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001742 block = PyLong_AsLong(arg);
1743 if (block == -1 && PyErr_Occurred())
1744 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001745
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001746 s->sock_timeout = block ? -1.0 : 0.0;
1747 internal_setblocking(s, block);
Guido van Rossume4485b01994-09-07 14:32:49 +00001748
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001749 Py_INCREF(Py_None);
1750 return Py_None;
Guido van Rossume4485b01994-09-07 14:32:49 +00001751}
Guido van Rossume4485b01994-09-07 14:32:49 +00001752
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001753PyDoc_STRVAR(setblocking_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001754"setblocking(flag)\n\
1755\n\
1756Set the socket to blocking (flag is true) or non-blocking (false).\n\
Guido van Rossum11ba0942002-06-13 15:07:44 +00001757setblocking(True) is equivalent to settimeout(None);\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001758setblocking(False) is equivalent to settimeout(0.0).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001759
Guido van Rossum11ba0942002-06-13 15:07:44 +00001760/* s.settimeout(timeout) method. Argument:
1761 None -- no timeout, blocking mode; same as setblocking(True)
1762 0.0 -- non-blocking mode; same as setblocking(False)
1763 > 0 -- timeout mode; operations time out after timeout seconds
1764 < 0 -- illegal; raises an exception
1765*/
Guido van Rossum67f7a382002-06-06 21:08:16 +00001766static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001767sock_settimeout(PySocketSockObject *s, PyObject *arg)
Guido van Rossum67f7a382002-06-06 21:08:16 +00001768{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001769 double timeout;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001770
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001771 if (arg == Py_None)
1772 timeout = -1.0;
1773 else {
1774 timeout = PyFloat_AsDouble(arg);
1775 if (timeout < 0.0) {
1776 if (!PyErr_Occurred())
1777 PyErr_SetString(PyExc_ValueError,
1778 "Timeout value out of range");
1779 return NULL;
1780 }
1781 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00001782
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001783 s->sock_timeout = timeout;
1784 internal_setblocking(s, timeout < 0.0);
Guido van Rossum67f7a382002-06-06 21:08:16 +00001785
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001786 Py_INCREF(Py_None);
1787 return Py_None;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001788}
1789
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001790PyDoc_STRVAR(settimeout_doc,
Guido van Rossum3eede5a2002-06-07 02:08:35 +00001791"settimeout(timeout)\n\
Guido van Rossum67f7a382002-06-06 21:08:16 +00001792\n\
Guido van Rossum11ba0942002-06-13 15:07:44 +00001793Set a timeout on socket operations. 'timeout' can be a float,\n\
1794giving in seconds, or None. Setting a timeout of None disables\n\
1795the timeout feature and is equivalent to setblocking(1).\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001796Setting a timeout of zero is the same as setblocking(0).");
Guido van Rossum67f7a382002-06-06 21:08:16 +00001797
Guido van Rossum3eede5a2002-06-07 02:08:35 +00001798/* s.gettimeout() method.
1799 Returns the timeout associated with a socket. */
Guido van Rossum67f7a382002-06-06 21:08:16 +00001800static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001801sock_gettimeout(PySocketSockObject *s)
Guido van Rossum67f7a382002-06-06 21:08:16 +00001802{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001803 if (s->sock_timeout < 0.0) {
1804 Py_INCREF(Py_None);
1805 return Py_None;
1806 }
1807 else
1808 return PyFloat_FromDouble(s->sock_timeout);
Guido van Rossum67f7a382002-06-06 21:08:16 +00001809}
1810
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001811PyDoc_STRVAR(gettimeout_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00001812"gettimeout() -> timeout\n\
Guido van Rossum67f7a382002-06-06 21:08:16 +00001813\n\
Ezio Melotti388c9452011-08-14 08:28:57 +03001814Returns the timeout in seconds (float) associated with socket \n\
Guido van Rossum67f7a382002-06-06 21:08:16 +00001815operations. A timeout of None indicates that timeouts on socket \n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001816operations are disabled.");
Guido van Rossume4485b01994-09-07 14:32:49 +00001817
Guido van Rossumaee08791992-09-08 09:05:33 +00001818/* s.setsockopt() method.
1819 With an integer third argument, sets an integer option.
1820 With a string third argument, sets an option from a buffer;
1821 use optional built-in module 'struct' to encode the string. */
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001822
Guido van Rossum73624e91994-10-10 17:59:00 +00001823static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001824sock_setsockopt(PySocketSockObject *s, PyObject *args)
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001825{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001826 int level;
1827 int optname;
1828 int res;
1829 char *buf;
1830 int buflen;
1831 int flag;
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001832
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001833 if (PyArg_ParseTuple(args, "iii:setsockopt",
1834 &level, &optname, &flag)) {
1835 buf = (char *) &flag;
1836 buflen = sizeof flag;
1837 }
1838 else {
1839 PyErr_Clear();
1840 if (!PyArg_ParseTuple(args, "iiy#:setsockopt",
1841 &level, &optname, &buf, &buflen))
1842 return NULL;
1843 }
1844 res = setsockopt(s->sock_fd, level, optname, (void *)buf, buflen);
1845 if (res < 0)
1846 return s->errorhandler();
1847 Py_INCREF(Py_None);
1848 return Py_None;
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001849}
1850
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001851PyDoc_STRVAR(setsockopt_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001852"setsockopt(level, option, value)\n\
1853\n\
1854Set a socket option. See the Unix manual for level and option.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001855The value argument can either be an integer or a string.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001856
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001857
Guido van Rossumaee08791992-09-08 09:05:33 +00001858/* s.getsockopt() method.
1859 With two arguments, retrieves an integer option.
1860 With a third integer argument, retrieves a string buffer of that size;
1861 use optional built-in module 'struct' to decode the string. */
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001862
Guido van Rossum73624e91994-10-10 17:59:00 +00001863static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001864sock_getsockopt(PySocketSockObject *s, PyObject *args)
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001865{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001866 int level;
1867 int optname;
1868 int res;
1869 PyObject *buf;
1870 socklen_t buflen = 0;
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001871
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001872 if (!PyArg_ParseTuple(args, "ii|i:getsockopt",
1873 &level, &optname, &buflen))
1874 return NULL;
Guido van Rossum48a680c2001-03-02 06:34:14 +00001875
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001876 if (buflen == 0) {
1877 int flag = 0;
1878 socklen_t flagsize = sizeof flag;
1879 res = getsockopt(s->sock_fd, level, optname,
1880 (void *)&flag, &flagsize);
1881 if (res < 0)
1882 return s->errorhandler();
1883 return PyLong_FromLong(flag);
1884 }
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001885#ifdef __VMS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001886 /* socklen_t is unsigned so no negative test is needed,
1887 test buflen == 0 is previously done */
1888 if (buflen > 1024) {
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001889#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001890 if (buflen <= 0 || buflen > 1024) {
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001891#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001892 PyErr_SetString(socket_error,
1893 "getsockopt buflen out of range");
1894 return NULL;
1895 }
1896 buf = PyBytes_FromStringAndSize((char *)NULL, buflen);
1897 if (buf == NULL)
1898 return NULL;
1899 res = getsockopt(s->sock_fd, level, optname,
1900 (void *)PyBytes_AS_STRING(buf), &buflen);
1901 if (res < 0) {
1902 Py_DECREF(buf);
1903 return s->errorhandler();
1904 }
1905 _PyBytes_Resize(&buf, buflen);
1906 return buf;
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001907}
1908
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001909PyDoc_STRVAR(getsockopt_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001910"getsockopt(level, option[, buffersize]) -> value\n\
1911\n\
1912Get a socket option. See the Unix manual for level and option.\n\
1913If a nonzero buffersize argument is given, the return value is a\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001914string of that length; otherwise it is an integer.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001915
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001916
Fred Drake728819a2000-07-01 03:40:12 +00001917/* s.bind(sockaddr) method */
Guido van Rossum30a685f1991-06-27 15:51:29 +00001918
Guido van Rossum73624e91994-10-10 17:59:00 +00001919static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001920sock_bind(PySocketSockObject *s, PyObject *addro)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001921{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001922 sock_addr_t addrbuf;
1923 int addrlen;
1924 int res;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00001925
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001926 if (!getsockaddrarg(s, addro, SAS2SA(&addrbuf), &addrlen))
1927 return NULL;
1928 Py_BEGIN_ALLOW_THREADS
1929 res = bind(s->sock_fd, SAS2SA(&addrbuf), addrlen);
1930 Py_END_ALLOW_THREADS
1931 if (res < 0)
1932 return s->errorhandler();
1933 Py_INCREF(Py_None);
1934 return Py_None;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001935}
1936
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001937PyDoc_STRVAR(bind_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001938"bind(address)\n\
1939\n\
1940Bind the socket to a local address. For IP sockets, the address is a\n\
Jeremy Hylton22308652001-02-02 03:23:09 +00001941pair (host, port); the host must refer to the local host. For raw packet\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001942sockets the address is a tuple (ifname, proto [,pkttype [,hatype]])");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001943
Guido van Rossum30a685f1991-06-27 15:51:29 +00001944
1945/* s.close() method.
1946 Set the file descriptor to -1 so operations tried subsequently
1947 will surely fail. */
1948
Guido van Rossum73624e91994-10-10 17:59:00 +00001949static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001950sock_close(PySocketSockObject *s)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001951{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001952 SOCKET_T fd;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00001953
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001954 if ((fd = s->sock_fd) != -1) {
1955 s->sock_fd = -1;
1956 Py_BEGIN_ALLOW_THREADS
1957 (void) SOCKETCLOSE(fd);
1958 Py_END_ALLOW_THREADS
1959 }
1960 Py_INCREF(Py_None);
1961 return Py_None;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001962}
1963
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001964PyDoc_STRVAR(close_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001965"close()\n\
1966\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001967Close the socket. It cannot be used after this call.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001968
Antoine Pitroue43f9d02010-08-08 23:24:50 +00001969static PyObject *
Antoine Pitrou6e451df2010-08-09 20:39:54 +00001970sock_detach(PySocketSockObject *s)
Antoine Pitroue43f9d02010-08-08 23:24:50 +00001971{
Antoine Pitrou6e451df2010-08-09 20:39:54 +00001972 SOCKET_T fd = s->sock_fd;
Antoine Pitroue43f9d02010-08-08 23:24:50 +00001973 s->sock_fd = -1;
Antoine Pitrou6e451df2010-08-09 20:39:54 +00001974 return PyLong_FromSocket_t(fd);
Antoine Pitroue43f9d02010-08-08 23:24:50 +00001975}
1976
Antoine Pitrou6e451df2010-08-09 20:39:54 +00001977PyDoc_STRVAR(detach_doc,
1978"detach()\n\
Antoine Pitroue43f9d02010-08-08 23:24:50 +00001979\n\
1980Close the socket object without closing the underlying file descriptor.\
1981The object cannot be used after this call, but the file descriptor\
Antoine Pitrou6e451df2010-08-09 20:39:54 +00001982can be reused for other purposes. The file descriptor is returned.");
Antoine Pitroue43f9d02010-08-08 23:24:50 +00001983
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001984static int
Raymond Hettingeref7343c2003-06-29 03:08:05 +00001985internal_connect(PySocketSockObject *s, struct sockaddr *addr, int addrlen,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001986 int *timeoutp)
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001987{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001988 int res, timeout;
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001989
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001990 timeout = 0;
1991 res = connect(s->sock_fd, addr, addrlen);
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001992
1993#ifdef MS_WINDOWS
1994
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001995 if (s->sock_timeout > 0.0) {
1996 if (res < 0 && WSAGetLastError() == WSAEWOULDBLOCK &&
1997 IS_SELECTABLE(s)) {
1998 /* This is a mess. Best solution: trust select */
1999 fd_set fds;
2000 fd_set fds_exc;
2001 struct timeval tv;
2002 tv.tv_sec = (int)s->sock_timeout;
2003 tv.tv_usec = (int)((s->sock_timeout - tv.tv_sec) * 1e6);
2004 FD_ZERO(&fds);
2005 FD_SET(s->sock_fd, &fds);
2006 FD_ZERO(&fds_exc);
2007 FD_SET(s->sock_fd, &fds_exc);
Antoine Pitrou19467d22010-08-17 19:33:30 +00002008 res = select(Py_SAFE_DOWNCAST(s->sock_fd+1, SOCKET_T, int),
2009 NULL, &fds, &fds_exc, &tv);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002010 if (res == 0) {
2011 res = WSAEWOULDBLOCK;
2012 timeout = 1;
2013 } else if (res > 0) {
2014 if (FD_ISSET(s->sock_fd, &fds))
2015 /* The socket is in the writable set - this
2016 means connected */
2017 res = 0;
2018 else {
2019 /* As per MS docs, we need to call getsockopt()
2020 to get the underlying error */
2021 int res_size = sizeof res;
2022 /* It must be in the exception set */
2023 assert(FD_ISSET(s->sock_fd, &fds_exc));
2024 if (0 == getsockopt(s->sock_fd, SOL_SOCKET, SO_ERROR,
2025 (char *)&res, &res_size))
2026 /* getsockopt also clears WSAGetLastError,
2027 so reset it back. */
2028 WSASetLastError(res);
2029 else
2030 res = WSAGetLastError();
2031 }
2032 }
2033 /* else if (res < 0) an error occurred */
2034 }
2035 }
Guido van Rossum7b8bac12002-06-13 16:07:04 +00002036
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002037 if (res < 0)
2038 res = WSAGetLastError();
Guido van Rossum7b8bac12002-06-13 16:07:04 +00002039
2040#else
2041
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002042 if (s->sock_timeout > 0.0) {
2043 if (res < 0 && errno == EINPROGRESS && IS_SELECTABLE(s)) {
2044 timeout = internal_select(s, 1);
2045 if (timeout == 0) {
2046 /* Bug #1019808: in case of an EINPROGRESS,
2047 use getsockopt(SO_ERROR) to get the real
2048 error. */
2049 socklen_t res_size = sizeof res;
2050 (void)getsockopt(s->sock_fd, SOL_SOCKET,
2051 SO_ERROR, &res, &res_size);
2052 if (res == EISCONN)
2053 res = 0;
2054 errno = res;
2055 }
2056 else if (timeout == -1) {
2057 res = errno; /* had error */
2058 }
2059 else
2060 res = EWOULDBLOCK; /* timed out */
2061 }
2062 }
Guido van Rossum7b8bac12002-06-13 16:07:04 +00002063
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002064 if (res < 0)
2065 res = errno;
Guido van Rossum7b8bac12002-06-13 16:07:04 +00002066
2067#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002068 *timeoutp = timeout;
Guido van Rossum7b8bac12002-06-13 16:07:04 +00002069
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002070 return res;
Guido van Rossum7b8bac12002-06-13 16:07:04 +00002071}
Guido van Rossum30a685f1991-06-27 15:51:29 +00002072
Fred Drake728819a2000-07-01 03:40:12 +00002073/* s.connect(sockaddr) method */
Guido van Rossum30a685f1991-06-27 15:51:29 +00002074
Guido van Rossum73624e91994-10-10 17:59:00 +00002075static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002076sock_connect(PySocketSockObject *s, PyObject *addro)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002077{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002078 sock_addr_t addrbuf;
2079 int addrlen;
2080 int res;
2081 int timeout;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00002082
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002083 if (!getsockaddrarg(s, addro, SAS2SA(&addrbuf), &addrlen))
2084 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002085
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002086 Py_BEGIN_ALLOW_THREADS
2087 res = internal_connect(s, SAS2SA(&addrbuf), addrlen, &timeout);
2088 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00002089
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002090 if (timeout == 1) {
2091 PyErr_SetString(socket_timeout, "timed out");
2092 return NULL;
2093 }
2094 if (res != 0)
2095 return s->errorhandler();
2096 Py_INCREF(Py_None);
2097 return Py_None;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002098}
2099
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002100PyDoc_STRVAR(connect_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002101"connect(address)\n\
2102\n\
2103Connect the socket to a remote address. For IP sockets, the address\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002104is a pair (host, port).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002105
Guido van Rossum30a685f1991-06-27 15:51:29 +00002106
Fred Drake728819a2000-07-01 03:40:12 +00002107/* s.connect_ex(sockaddr) method */
Guido van Rossumfc4255d1997-11-19 18:57:13 +00002108
2109static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002110sock_connect_ex(PySocketSockObject *s, PyObject *addro)
Guido van Rossumfc4255d1997-11-19 18:57:13 +00002111{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002112 sock_addr_t addrbuf;
2113 int addrlen;
2114 int res;
2115 int timeout;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00002116
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002117 if (!getsockaddrarg(s, addro, SAS2SA(&addrbuf), &addrlen))
2118 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002119
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002120 Py_BEGIN_ALLOW_THREADS
2121 res = internal_connect(s, SAS2SA(&addrbuf), addrlen, &timeout);
2122 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00002123
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002124 /* Signals are not errors (though they may raise exceptions). Adapted
2125 from PyErr_SetFromErrnoWithFilenameObject(). */
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002126#ifdef EINTR
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002127 if (res == EINTR && PyErr_CheckSignals())
2128 return NULL;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002129#endif
2130
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002131 return PyLong_FromLong((long) res);
Guido van Rossumfc4255d1997-11-19 18:57:13 +00002132}
2133
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002134PyDoc_STRVAR(connect_ex_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00002135"connect_ex(address) -> errno\n\
Guido van Rossum82a5c661998-07-07 20:45:43 +00002136\n\
2137This is like connect(address), but returns an error code (the errno value)\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002138instead of raising an exception when an error occurs.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002139
Guido van Rossumfc4255d1997-11-19 18:57:13 +00002140
Guido van Rossumed233a51992-06-23 09:07:03 +00002141/* s.fileno() method */
2142
Guido van Rossum73624e91994-10-10 17:59:00 +00002143static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002144sock_fileno(PySocketSockObject *s)
Guido van Rossumed233a51992-06-23 09:07:03 +00002145{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002146 return PyLong_FromSocket_t(s->sock_fd);
Guido van Rossumed233a51992-06-23 09:07:03 +00002147}
2148
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002149PyDoc_STRVAR(fileno_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002150"fileno() -> integer\n\
2151\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002152Return the integer file descriptor of the socket.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002153
Guido van Rossumed233a51992-06-23 09:07:03 +00002154
Guido van Rossumc89705d1992-11-26 08:54:07 +00002155/* s.getsockname() method */
2156
Guido van Rossum73624e91994-10-10 17:59:00 +00002157static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002158sock_getsockname(PySocketSockObject *s)
Guido van Rossumc89705d1992-11-26 08:54:07 +00002159{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002160 sock_addr_t addrbuf;
2161 int res;
2162 socklen_t addrlen;
Guido van Rossumff3ab422000-04-24 15:16:03 +00002163
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002164 if (!getsockaddrlen(s, &addrlen))
2165 return NULL;
2166 memset(&addrbuf, 0, addrlen);
2167 Py_BEGIN_ALLOW_THREADS
2168 res = getsockname(s->sock_fd, SAS2SA(&addrbuf), &addrlen);
2169 Py_END_ALLOW_THREADS
2170 if (res < 0)
2171 return s->errorhandler();
2172 return makesockaddr(s->sock_fd, SAS2SA(&addrbuf), addrlen,
2173 s->sock_proto);
Guido van Rossumc89705d1992-11-26 08:54:07 +00002174}
2175
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002176PyDoc_STRVAR(getsockname_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002177"getsockname() -> address info\n\
2178\n\
2179Return the address of the local endpoint. For IP sockets, the address\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002180info is a pair (hostaddr, port).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002181
Guido van Rossumc89705d1992-11-26 08:54:07 +00002182
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002183#ifdef HAVE_GETPEERNAME /* Cray APP doesn't have this :-( */
Guido van Rossumc89705d1992-11-26 08:54:07 +00002184/* s.getpeername() method */
2185
Guido van Rossum73624e91994-10-10 17:59:00 +00002186static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002187sock_getpeername(PySocketSockObject *s)
Guido van Rossumc89705d1992-11-26 08:54:07 +00002188{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002189 sock_addr_t addrbuf;
2190 int res;
2191 socklen_t addrlen;
Guido van Rossumff3ab422000-04-24 15:16:03 +00002192
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002193 if (!getsockaddrlen(s, &addrlen))
2194 return NULL;
2195 memset(&addrbuf, 0, addrlen);
2196 Py_BEGIN_ALLOW_THREADS
2197 res = getpeername(s->sock_fd, SAS2SA(&addrbuf), &addrlen);
2198 Py_END_ALLOW_THREADS
2199 if (res < 0)
2200 return s->errorhandler();
2201 return makesockaddr(s->sock_fd, SAS2SA(&addrbuf), addrlen,
2202 s->sock_proto);
Guido van Rossumc89705d1992-11-26 08:54:07 +00002203}
Guido van Rossum82a5c661998-07-07 20:45:43 +00002204
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002205PyDoc_STRVAR(getpeername_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002206"getpeername() -> address info\n\
2207\n\
2208Return the address of the remote endpoint. For IP sockets, the address\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002209info is a pair (hostaddr, port).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002210
Guido van Rossumb6775db1994-08-01 11:34:53 +00002211#endif /* HAVE_GETPEERNAME */
Guido van Rossumc89705d1992-11-26 08:54:07 +00002212
2213
Guido van Rossum30a685f1991-06-27 15:51:29 +00002214/* s.listen(n) method */
2215
Guido van Rossum73624e91994-10-10 17:59:00 +00002216static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002217sock_listen(PySocketSockObject *s, PyObject *arg)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002218{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002219 int backlog;
2220 int res;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00002221
Serhiy Storchaka441d30f2013-01-19 12:26:26 +02002222 backlog = _PyLong_AsInt(arg);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002223 if (backlog == -1 && PyErr_Occurred())
2224 return NULL;
2225 Py_BEGIN_ALLOW_THREADS
Antoine Pitrou1be815a2011-05-10 19:16:29 +02002226 /* To avoid problems on systems that don't allow a negative backlog
2227 * (which doesn't make sense anyway) we force a minimum value of 0. */
2228 if (backlog < 0)
2229 backlog = 0;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002230 res = listen(s->sock_fd, backlog);
2231 Py_END_ALLOW_THREADS
2232 if (res < 0)
2233 return s->errorhandler();
2234 Py_INCREF(Py_None);
2235 return Py_None;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002236}
2237
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002238PyDoc_STRVAR(listen_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002239"listen(backlog)\n\
2240\n\
2241Enable a server to accept connections. The backlog argument must be at\n\
Antoine Pitrou1be815a2011-05-10 19:16:29 +02002242least 0 (if it is lower, it is set to 0); it specifies the number of\n\
2243unaccepted connections that the system will allow before refusing new\n\
2244connections.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002245
2246
Thomas Wouters477c8d52006-05-27 19:21:47 +00002247/*
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002248 * This is the guts of the recv() and recv_into() methods, which reads into a
Thomas Wouters902d6eb2007-01-09 23:18:33 +00002249 * char buffer. If you have any inc/dec ref to do to the objects that contain
Thomas Wouters477c8d52006-05-27 19:21:47 +00002250 * the buffer, do it in the caller. This function returns the number of bytes
Ezio Melotti13925002011-03-16 11:05:33 +02002251 * successfully read. If there was an error, it returns -1. Note that it is
Thomas Wouters477c8d52006-05-27 19:21:47 +00002252 * also possible that we return a number of bytes smaller than the request
2253 * bytes.
2254 */
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002255
Antoine Pitrou19467d22010-08-17 19:33:30 +00002256static Py_ssize_t
2257sock_recv_guts(PySocketSockObject *s, char* cbuf, Py_ssize_t len, int flags)
Thomas Wouters477c8d52006-05-27 19:21:47 +00002258{
Antoine Pitrou19467d22010-08-17 19:33:30 +00002259 Py_ssize_t outlen = -1;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002260 int timeout;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002261#ifdef __VMS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002262 int remaining;
2263 char *read_buf;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002264#endif
2265
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002266 if (!IS_SELECTABLE(s)) {
2267 select_error();
2268 return -1;
2269 }
2270 if (len == 0) {
2271 /* If 0 bytes were requested, do nothing. */
2272 return 0;
2273 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002274
2275#ifndef __VMS
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002276 BEGIN_SELECT_LOOP(s)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002277 Py_BEGIN_ALLOW_THREADS
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002278 timeout = internal_select_ex(s, 0, interval);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002279 if (!timeout)
2280 outlen = recv(s->sock_fd, cbuf, len, flags);
2281 Py_END_ALLOW_THREADS
Thomas Wouters477c8d52006-05-27 19:21:47 +00002282
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002283 if (timeout == 1) {
2284 PyErr_SetString(socket_timeout, "timed out");
2285 return -1;
2286 }
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002287 END_SELECT_LOOP(s)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002288 if (outlen < 0) {
2289 /* Note: the call to errorhandler() ALWAYS indirectly returned
2290 NULL, so ignore its return value */
2291 s->errorhandler();
2292 return -1;
2293 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002294#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002295 read_buf = cbuf;
2296 remaining = len;
2297 while (remaining != 0) {
2298 unsigned int segment;
2299 int nread = -1;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002300
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002301 segment = remaining /SEGMENT_SIZE;
2302 if (segment != 0) {
2303 segment = SEGMENT_SIZE;
2304 }
2305 else {
2306 segment = remaining;
2307 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002308
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002309 BEGIN_SELECT_LOOP(s)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002310 Py_BEGIN_ALLOW_THREADS
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002311 timeout = internal_select_ex(s, 0, interval);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002312 if (!timeout)
2313 nread = recv(s->sock_fd, read_buf, segment, flags);
2314 Py_END_ALLOW_THREADS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002315 if (timeout == 1) {
2316 PyErr_SetString(socket_timeout, "timed out");
2317 return -1;
2318 }
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002319 END_SELECT_LOOP(s)
2320
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002321 if (nread < 0) {
2322 s->errorhandler();
2323 return -1;
2324 }
2325 if (nread != remaining) {
2326 read_buf += nread;
2327 break;
2328 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002329
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002330 remaining -= segment;
2331 read_buf += segment;
2332 }
2333 outlen = read_buf - cbuf;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002334#endif /* !__VMS */
2335
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002336 return outlen;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002337}
2338
Guido van Rossum48a680c2001-03-02 06:34:14 +00002339
Guido van Rossumeb6b33a1993-05-25 09:38:27 +00002340/* s.recv(nbytes [,flags]) method */
Guido van Rossum30a685f1991-06-27 15:51:29 +00002341
Guido van Rossum73624e91994-10-10 17:59:00 +00002342static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002343sock_recv(PySocketSockObject *s, PyObject *args)
Guido van Rossum30a685f1991-06-27 15:51:29 +00002344{
Antoine Pitrou19467d22010-08-17 19:33:30 +00002345 Py_ssize_t recvlen, outlen;
2346 int flags = 0;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002347 PyObject *buf;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002348
Antoine Pitrou19467d22010-08-17 19:33:30 +00002349 if (!PyArg_ParseTuple(args, "n|i:recv", &recvlen, &flags))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002350 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002351
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002352 if (recvlen < 0) {
2353 PyErr_SetString(PyExc_ValueError,
2354 "negative buffersize in recv");
2355 return NULL;
2356 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00002357
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002358 /* Allocate a new string. */
2359 buf = PyBytes_FromStringAndSize((char *) 0, recvlen);
2360 if (buf == NULL)
2361 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002362
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002363 /* Call the guts */
2364 outlen = sock_recv_guts(s, PyBytes_AS_STRING(buf), recvlen, flags);
2365 if (outlen < 0) {
2366 /* An error occurred, release the string and return an
2367 error. */
2368 Py_DECREF(buf);
2369 return NULL;
2370 }
2371 if (outlen != recvlen) {
2372 /* We did not read as many bytes as we anticipated, resize the
2373 string if possible and be successful. */
2374 _PyBytes_Resize(&buf, outlen);
2375 }
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00002376
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002377 return buf;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002378}
2379
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002380PyDoc_STRVAR(recv_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002381"recv(buffersize[, flags]) -> data\n\
2382\n\
2383Receive up to buffersize bytes from the socket. For the optional flags\n\
2384argument, see the Unix manual. When no data is available, block until\n\
2385at least one byte is available or until the remote end is closed. When\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002386the remote end is closed and all data is read, return the empty string.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002387
Guido van Rossum30a685f1991-06-27 15:51:29 +00002388
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002389/* s.recv_into(buffer, [nbytes [,flags]]) method */
Guido van Rossum30a685f1991-06-27 15:51:29 +00002390
Thomas Wouters477c8d52006-05-27 19:21:47 +00002391static PyObject*
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002392sock_recv_into(PySocketSockObject *s, PyObject *args, PyObject *kwds)
Thomas Wouters477c8d52006-05-27 19:21:47 +00002393{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002394 static char *kwlist[] = {"buffer", "nbytes", "flags", 0};
Thomas Wouters477c8d52006-05-27 19:21:47 +00002395
Antoine Pitrou19467d22010-08-17 19:33:30 +00002396 int flags = 0;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002397 Py_buffer pbuf;
2398 char *buf;
Antoine Pitrou19467d22010-08-17 19:33:30 +00002399 Py_ssize_t buflen, readlen, recvlen = 0;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002400
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002401 /* Get the buffer's memory */
Antoine Pitrou19467d22010-08-17 19:33:30 +00002402 if (!PyArg_ParseTupleAndKeywords(args, kwds, "w*|ni:recv_into", kwlist,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002403 &pbuf, &recvlen, &flags))
2404 return NULL;
2405 buf = pbuf.buf;
2406 buflen = pbuf.len;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002407
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002408 if (recvlen < 0) {
2409 PyBuffer_Release(&pbuf);
2410 PyErr_SetString(PyExc_ValueError,
2411 "negative buffersize in recv_into");
2412 return NULL;
2413 }
2414 if (recvlen == 0) {
2415 /* If nbytes was not specified, use the buffer's length */
2416 recvlen = buflen;
2417 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002418
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002419 /* Check if the buffer is large enough */
2420 if (buflen < recvlen) {
2421 PyBuffer_Release(&pbuf);
2422 PyErr_SetString(PyExc_ValueError,
2423 "buffer too small for requested bytes");
2424 return NULL;
2425 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002426
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002427 /* Call the guts */
2428 readlen = sock_recv_guts(s, buf, recvlen, flags);
2429 if (readlen < 0) {
2430 /* Return an error. */
2431 PyBuffer_Release(&pbuf);
2432 return NULL;
2433 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002434
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002435 PyBuffer_Release(&pbuf);
2436 /* Return the number of bytes read. Note that we do not do anything
2437 special here in the case that readlen < recvlen. */
2438 return PyLong_FromSsize_t(readlen);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002439}
2440
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002441PyDoc_STRVAR(recv_into_doc,
2442"recv_into(buffer, [nbytes[, flags]]) -> nbytes_read\n\
Thomas Wouters477c8d52006-05-27 19:21:47 +00002443\n\
2444A version of recv() that stores its data into a buffer rather than creating \n\
2445a new string. Receive up to buffersize bytes from the socket. If buffersize \n\
2446is not specified (or 0), receive up to the size available in the given buffer.\n\
2447\n\
2448See recv() for documentation about the flags.");
2449
2450
2451/*
Christian Heimes99170a52007-12-19 02:07:34 +00002452 * This is the guts of the recvfrom() and recvfrom_into() methods, which reads
2453 * into a char buffer. If you have any inc/def ref to do to the objects that
2454 * contain the buffer, do it in the caller. This function returns the number
Ezio Melotti13925002011-03-16 11:05:33 +02002455 * of bytes successfully read. If there was an error, it returns -1. Note
Christian Heimes99170a52007-12-19 02:07:34 +00002456 * that it is also possible that we return a number of bytes smaller than the
2457 * request bytes.
Thomas Wouters477c8d52006-05-27 19:21:47 +00002458 *
2459 * 'addr' is a return value for the address object. Note that you must decref
2460 * it yourself.
2461 */
Antoine Pitrou19467d22010-08-17 19:33:30 +00002462static Py_ssize_t
2463sock_recvfrom_guts(PySocketSockObject *s, char* cbuf, Py_ssize_t len, int flags,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002464 PyObject** addr)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002465{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002466 sock_addr_t addrbuf;
2467 int timeout;
Antoine Pitrou19467d22010-08-17 19:33:30 +00002468 Py_ssize_t n = -1;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002469 socklen_t addrlen;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002470
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002471 *addr = NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002472
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002473 if (!getsockaddrlen(s, &addrlen))
2474 return -1;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002475
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002476 if (!IS_SELECTABLE(s)) {
2477 select_error();
2478 return -1;
2479 }
Neal Norwitz082b2df2006-02-07 07:04:46 +00002480
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002481 BEGIN_SELECT_LOOP(s)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002482 Py_BEGIN_ALLOW_THREADS
2483 memset(&addrbuf, 0, addrlen);
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002484 timeout = internal_select_ex(s, 0, interval);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002485 if (!timeout) {
Guido van Rossum8d665e61996-06-26 18:22:49 +00002486#ifndef MS_WINDOWS
Andrew MacIntyreba43e872002-03-03 03:03:52 +00002487#if defined(PYOS_OS2) && !defined(PYCC_GCC)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002488 n = recvfrom(s->sock_fd, cbuf, len, flags,
2489 SAS2SA(&addrbuf), &addrlen);
Guido van Rossum32c575d1997-12-02 20:37:32 +00002490#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002491 n = recvfrom(s->sock_fd, cbuf, len, flags,
2492 (void *) &addrbuf, &addrlen);
Guido van Rossum32c575d1997-12-02 20:37:32 +00002493#endif
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00002494#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002495 n = recvfrom(s->sock_fd, cbuf, len, flags,
2496 SAS2SA(&addrbuf), &addrlen);
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00002497#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002498 }
2499 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00002500
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002501 if (timeout == 1) {
2502 PyErr_SetString(socket_timeout, "timed out");
2503 return -1;
2504 }
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002505 END_SELECT_LOOP(s)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002506 if (n < 0) {
2507 s->errorhandler();
2508 return -1;
2509 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00002510
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002511 if (!(*addr = makesockaddr(s->sock_fd, SAS2SA(&addrbuf),
2512 addrlen, s->sock_proto)))
2513 return -1;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002514
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002515 return n;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002516}
2517
2518/* s.recvfrom(nbytes [,flags]) method */
2519
2520static PyObject *
2521sock_recvfrom(PySocketSockObject *s, PyObject *args)
2522{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002523 PyObject *buf = NULL;
2524 PyObject *addr = NULL;
2525 PyObject *ret = NULL;
Antoine Pitrou19467d22010-08-17 19:33:30 +00002526 int flags = 0;
2527 Py_ssize_t recvlen, outlen;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002528
Antoine Pitrou19467d22010-08-17 19:33:30 +00002529 if (!PyArg_ParseTuple(args, "n|i:recvfrom", &recvlen, &flags))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002530 return NULL;
Guido van Rossum48a680c2001-03-02 06:34:14 +00002531
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002532 if (recvlen < 0) {
2533 PyErr_SetString(PyExc_ValueError,
2534 "negative buffersize in recvfrom");
2535 return NULL;
2536 }
Guido van Rossumd8faa362007-04-27 19:54:29 +00002537
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002538 buf = PyBytes_FromStringAndSize((char *) 0, recvlen);
2539 if (buf == NULL)
2540 return NULL;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002541
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002542 outlen = sock_recvfrom_guts(s, PyBytes_AS_STRING(buf),
2543 recvlen, flags, &addr);
2544 if (outlen < 0) {
2545 goto finally;
2546 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002547
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002548 if (outlen != recvlen) {
2549 /* We did not read as many bytes as we anticipated, resize the
Ezio Melotti13925002011-03-16 11:05:33 +02002550 string if possible and be successful. */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002551 if (_PyBytes_Resize(&buf, outlen) < 0)
Ezio Melotti13925002011-03-16 11:05:33 +02002552 /* Oopsy, not so successful after all. */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002553 goto finally;
2554 }
Barry Warsaw752300b1997-01-03 17:18:10 +00002555
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002556 ret = PyTuple_Pack(2, buf, addr);
Guido van Rossum67f7a382002-06-06 21:08:16 +00002557
2558finally:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002559 Py_XDECREF(buf);
2560 Py_XDECREF(addr);
2561 return ret;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002562}
2563
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002564PyDoc_STRVAR(recvfrom_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002565"recvfrom(buffersize[, flags]) -> (data, address info)\n\
2566\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002567Like recv(buffersize, flags) but also return the sender's address info.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002568
Thomas Wouters477c8d52006-05-27 19:21:47 +00002569
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002570/* s.recvfrom_into(buffer[, nbytes [,flags]]) method */
Thomas Wouters477c8d52006-05-27 19:21:47 +00002571
2572static PyObject *
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002573sock_recvfrom_into(PySocketSockObject *s, PyObject *args, PyObject* kwds)
Thomas Wouters477c8d52006-05-27 19:21:47 +00002574{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002575 static char *kwlist[] = {"buffer", "nbytes", "flags", 0};
Thomas Wouters477c8d52006-05-27 19:21:47 +00002576
Antoine Pitrou19467d22010-08-17 19:33:30 +00002577 int flags = 0;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002578 Py_buffer pbuf;
2579 char *buf;
Antoine Pitrou19467d22010-08-17 19:33:30 +00002580 Py_ssize_t readlen, buflen, recvlen = 0;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002581
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002582 PyObject *addr = NULL;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002583
Antoine Pitrou19467d22010-08-17 19:33:30 +00002584 if (!PyArg_ParseTupleAndKeywords(args, kwds, "w*|ni:recvfrom_into",
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002585 kwlist, &pbuf,
2586 &recvlen, &flags))
2587 return NULL;
2588 buf = pbuf.buf;
2589 buflen = pbuf.len;
2590 assert(buf != 0 && buflen > 0);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002591
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002592 if (recvlen < 0) {
2593 PyBuffer_Release(&pbuf);
2594 PyErr_SetString(PyExc_ValueError,
2595 "negative buffersize in recvfrom_into");
2596 return NULL;
2597 }
2598 if (recvlen == 0) {
2599 /* If nbytes was not specified, use the buffer's length */
2600 recvlen = buflen;
Benjamin Petersonfbf648e2014-01-13 22:59:38 -05002601 } else if (recvlen > buflen) {
2602 PyBuffer_Release(&pbuf);
2603 PyErr_SetString(PyExc_ValueError,
2604 "nbytes is greater than the length of the buffer");
2605 return NULL;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002606 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002607
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002608 readlen = sock_recvfrom_guts(s, buf, recvlen, flags, &addr);
2609 if (readlen < 0) {
2610 PyBuffer_Release(&pbuf);
2611 /* Return an error */
2612 Py_XDECREF(addr);
2613 return NULL;
2614 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002615
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002616 PyBuffer_Release(&pbuf);
2617 /* Return the number of bytes read and the address. Note that we do
2618 not do anything special here in the case that readlen < recvlen. */
Antoine Pitrou19467d22010-08-17 19:33:30 +00002619 return Py_BuildValue("nN", readlen, addr);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002620}
2621
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002622PyDoc_STRVAR(recvfrom_into_doc,
2623"recvfrom_into(buffer[, nbytes[, flags]]) -> (nbytes, address info)\n\
Thomas Wouters477c8d52006-05-27 19:21:47 +00002624\n\
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002625Like recv_into(buffer[, nbytes[, flags]]) but also return the sender's address info.");
Thomas Wouters477c8d52006-05-27 19:21:47 +00002626
2627
Guido van Rossumeb6b33a1993-05-25 09:38:27 +00002628/* s.send(data [,flags]) method */
Guido van Rossum30a685f1991-06-27 15:51:29 +00002629
Guido van Rossum73624e91994-10-10 17:59:00 +00002630static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002631sock_send(PySocketSockObject *s, PyObject *args)
Guido van Rossum30a685f1991-06-27 15:51:29 +00002632{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002633 char *buf;
Antoine Pitrou19467d22010-08-17 19:33:30 +00002634 Py_ssize_t len, n = -1;
2635 int flags = 0, timeout;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002636 Py_buffer pbuf;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002637
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002638 if (!PyArg_ParseTuple(args, "y*|i:send", &pbuf, &flags))
2639 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002640
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002641 if (!IS_SELECTABLE(s)) {
2642 PyBuffer_Release(&pbuf);
2643 return select_error();
2644 }
2645 buf = pbuf.buf;
2646 len = pbuf.len;
Neal Norwitz082b2df2006-02-07 07:04:46 +00002647
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002648 BEGIN_SELECT_LOOP(s)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002649 Py_BEGIN_ALLOW_THREADS
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002650 timeout = internal_select_ex(s, 1, interval);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002651 if (!timeout)
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002652#ifdef __VMS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002653 n = sendsegmented(s->sock_fd, buf, len, flags);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002654#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002655 n = send(s->sock_fd, buf, len, flags);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002656#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002657 Py_END_ALLOW_THREADS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002658 if (timeout == 1) {
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002659 PyBuffer_Release(&pbuf);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002660 PyErr_SetString(socket_timeout, "timed out");
2661 return NULL;
2662 }
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002663 END_SELECT_LOOP(s)
2664
2665 PyBuffer_Release(&pbuf);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002666 if (n < 0)
2667 return s->errorhandler();
Antoine Pitrou19467d22010-08-17 19:33:30 +00002668 return PyLong_FromSsize_t(n);
Guido van Rossum30a685f1991-06-27 15:51:29 +00002669}
2670
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002671PyDoc_STRVAR(send_doc,
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002672"send(data[, flags]) -> count\n\
Guido van Rossum82a5c661998-07-07 20:45:43 +00002673\n\
2674Send a data string to the socket. For the optional flags\n\
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002675argument, see the Unix manual. Return the number of bytes\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002676sent; this may be less than len(data) if the network is busy.");
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002677
2678
2679/* s.sendall(data [,flags]) method */
2680
2681static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002682sock_sendall(PySocketSockObject *s, PyObject *args)
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002683{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002684 char *buf;
Antoine Pitrou19467d22010-08-17 19:33:30 +00002685 Py_ssize_t len, n = -1;
Antoine Pitrou6d7df632010-09-27 17:52:25 +00002686 int flags = 0, timeout, saved_errno;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002687 Py_buffer pbuf;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002688
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002689 if (!PyArg_ParseTuple(args, "y*|i:sendall", &pbuf, &flags))
2690 return NULL;
2691 buf = pbuf.buf;
2692 len = pbuf.len;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002693
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002694 if (!IS_SELECTABLE(s)) {
2695 PyBuffer_Release(&pbuf);
2696 return select_error();
2697 }
Neal Norwitz082b2df2006-02-07 07:04:46 +00002698
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002699 do {
Antoine Pitrou6d7df632010-09-27 17:52:25 +00002700 Py_BEGIN_ALLOW_THREADS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002701 timeout = internal_select(s, 1);
2702 n = -1;
Antoine Pitrou6d7df632010-09-27 17:52:25 +00002703 if (!timeout) {
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002704#ifdef __VMS
Antoine Pitrou6d7df632010-09-27 17:52:25 +00002705 n = sendsegmented(s->sock_fd, buf, len, flags);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002706#else
Antoine Pitrou6d7df632010-09-27 17:52:25 +00002707 n = send(s->sock_fd, buf, len, flags);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002708#endif
Antoine Pitrou6d7df632010-09-27 17:52:25 +00002709 }
2710 Py_END_ALLOW_THREADS
2711 if (timeout == 1) {
2712 PyBuffer_Release(&pbuf);
2713 PyErr_SetString(socket_timeout, "timed out");
2714 return NULL;
2715 }
2716 /* PyErr_CheckSignals() might change errno */
2717 saved_errno = errno;
2718 /* We must run our signal handlers before looping again.
2719 send() can return a successful partial write when it is
2720 interrupted, so we can't restrict ourselves to EINTR. */
2721 if (PyErr_CheckSignals()) {
2722 PyBuffer_Release(&pbuf);
2723 return NULL;
2724 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002725 if (n < 0) {
Antoine Pitrou6d7df632010-09-27 17:52:25 +00002726 /* If interrupted, try again */
2727 if (saved_errno == EINTR)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002728 continue;
Antoine Pitrou6d7df632010-09-27 17:52:25 +00002729 else
2730 break;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002731 }
2732 buf += n;
2733 len -= n;
2734 } while (len > 0);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002735 PyBuffer_Release(&pbuf);
Guido van Rossum67f7a382002-06-06 21:08:16 +00002736
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002737 if (n < 0)
2738 return s->errorhandler();
Guido van Rossum67f7a382002-06-06 21:08:16 +00002739
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002740 Py_INCREF(Py_None);
2741 return Py_None;
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002742}
2743
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002744PyDoc_STRVAR(sendall_doc,
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002745"sendall(data[, flags])\n\
2746\n\
2747Send a data string to the socket. For the optional flags\n\
2748argument, see the Unix manual. This calls send() repeatedly\n\
2749until all data is sent. If an error occurs, it's impossible\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002750to tell how much data has been sent.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002751
Guido van Rossum30a685f1991-06-27 15:51:29 +00002752
Guido van Rossumeb6b33a1993-05-25 09:38:27 +00002753/* s.sendto(data, [flags,] sockaddr) method */
Guido van Rossum30a685f1991-06-27 15:51:29 +00002754
Guido van Rossum73624e91994-10-10 17:59:00 +00002755static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002756sock_sendto(PySocketSockObject *s, PyObject *args)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002757{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002758 Py_buffer pbuf;
2759 PyObject *addro;
2760 char *buf;
Antoine Pitrou5e981412011-03-17 22:38:37 +01002761 Py_ssize_t len, arglen;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002762 sock_addr_t addrbuf;
2763 int addrlen, n = -1, flags, timeout;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002764
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002765 flags = 0;
Antoine Pitrou5e981412011-03-17 22:38:37 +01002766 arglen = PyTuple_Size(args);
2767 switch (arglen) {
2768 case 2:
2769 PyArg_ParseTuple(args, "y*O:sendto", &pbuf, &addro);
2770 break;
2771 case 3:
2772 PyArg_ParseTuple(args, "y*iO:sendto",
2773 &pbuf, &flags, &addro);
2774 break;
2775 default:
2776 PyErr_Format(PyExc_TypeError,
2777 "sendto() takes 2 or 3 arguments (%d given)",
2778 arglen);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002779 }
Antoine Pitrou5e981412011-03-17 22:38:37 +01002780 if (PyErr_Occurred())
2781 return NULL;
2782
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002783 buf = pbuf.buf;
2784 len = pbuf.len;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002785
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002786 if (!IS_SELECTABLE(s)) {
2787 PyBuffer_Release(&pbuf);
2788 return select_error();
2789 }
Neal Norwitz082b2df2006-02-07 07:04:46 +00002790
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002791 if (!getsockaddrarg(s, addro, SAS2SA(&addrbuf), &addrlen)) {
2792 PyBuffer_Release(&pbuf);
2793 return NULL;
2794 }
Thomas Wouters89f507f2006-12-13 04:49:30 +00002795
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002796 BEGIN_SELECT_LOOP(s)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002797 Py_BEGIN_ALLOW_THREADS
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002798 timeout = internal_select_ex(s, 1, interval);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002799 if (!timeout)
2800 n = sendto(s->sock_fd, buf, len, flags, SAS2SA(&addrbuf), addrlen);
2801 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00002802
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002803 if (timeout == 1) {
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002804 PyBuffer_Release(&pbuf);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002805 PyErr_SetString(socket_timeout, "timed out");
2806 return NULL;
2807 }
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002808 END_SELECT_LOOP(s)
2809 PyBuffer_Release(&pbuf);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002810 if (n < 0)
2811 return s->errorhandler();
Antoine Pitrou19467d22010-08-17 19:33:30 +00002812 return PyLong_FromSsize_t(n);
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002813}
2814
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002815PyDoc_STRVAR(sendto_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00002816"sendto(data[, flags], address) -> count\n\
Guido van Rossum82a5c661998-07-07 20:45:43 +00002817\n\
2818Like send(data, flags) but allows specifying the destination address.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002819For IP sockets, the address is a pair (hostaddr, port).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002820
Guido van Rossum30a685f1991-06-27 15:51:29 +00002821
2822/* s.shutdown(how) method */
2823
Guido van Rossum73624e91994-10-10 17:59:00 +00002824static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002825sock_shutdown(PySocketSockObject *s, PyObject *arg)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002826{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002827 int how;
2828 int res;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00002829
Serhiy Storchaka441d30f2013-01-19 12:26:26 +02002830 how = _PyLong_AsInt(arg);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002831 if (how == -1 && PyErr_Occurred())
2832 return NULL;
2833 Py_BEGIN_ALLOW_THREADS
2834 res = shutdown(s->sock_fd, how);
2835 Py_END_ALLOW_THREADS
2836 if (res < 0)
2837 return s->errorhandler();
2838 Py_INCREF(Py_None);
2839 return Py_None;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002840}
2841
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002842PyDoc_STRVAR(shutdown_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002843"shutdown(flag)\n\
2844\n\
Martin v. Löwis94681fc2003-11-27 19:40:22 +00002845Shut down the reading side of the socket (flag == SHUT_RD), the writing side\n\
2846of the socket (flag == SHUT_WR), or both ends (flag == SHUT_RDWR).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002847
Amaury Forgeot d'Arc3d17a5c2008-06-13 01:09:34 +00002848#if defined(MS_WINDOWS) && defined(SIO_RCVALL)
Christian Heimesfaf2f632008-01-06 16:59:19 +00002849static PyObject*
2850sock_ioctl(PySocketSockObject *s, PyObject *arg)
2851{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002852 unsigned long cmd = SIO_RCVALL;
2853 PyObject *argO;
2854 DWORD recv;
Christian Heimesfaf2f632008-01-06 16:59:19 +00002855
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002856 if (!PyArg_ParseTuple(arg, "kO:ioctl", &cmd, &argO))
2857 return NULL;
Christian Heimesfaf2f632008-01-06 16:59:19 +00002858
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002859 switch (cmd) {
2860 case SIO_RCVALL: {
2861 unsigned int option = RCVALL_ON;
2862 if (!PyArg_ParseTuple(arg, "kI:ioctl", &cmd, &option))
2863 return NULL;
2864 if (WSAIoctl(s->sock_fd, cmd, &option, sizeof(option),
2865 NULL, 0, &recv, NULL, NULL) == SOCKET_ERROR) {
2866 return set_error();
2867 }
2868 return PyLong_FromUnsignedLong(recv); }
2869 case SIO_KEEPALIVE_VALS: {
2870 struct tcp_keepalive ka;
2871 if (!PyArg_ParseTuple(arg, "k(kkk):ioctl", &cmd,
2872 &ka.onoff, &ka.keepalivetime, &ka.keepaliveinterval))
2873 return NULL;
2874 if (WSAIoctl(s->sock_fd, cmd, &ka, sizeof(ka),
2875 NULL, 0, &recv, NULL, NULL) == SOCKET_ERROR) {
2876 return set_error();
2877 }
2878 return PyLong_FromUnsignedLong(recv); }
2879 default:
2880 PyErr_Format(PyExc_ValueError, "invalid ioctl command %d", cmd);
2881 return NULL;
2882 }
Christian Heimesfaf2f632008-01-06 16:59:19 +00002883}
2884PyDoc_STRVAR(sock_ioctl_doc,
2885"ioctl(cmd, option) -> long\n\
2886\n\
Kristján Valur Jónsson847ec752009-09-27 21:10:38 +00002887Control the socket with WSAIoctl syscall. Currently supported 'cmd' values are\n\
2888SIO_RCVALL: 'option' must be one of the socket.RCVALL_* constants.\n\
2889SIO_KEEPALIVE_VALS: 'option' is a tuple of (onoff, timeout, interval).");
Christian Heimesfaf2f632008-01-06 16:59:19 +00002890
2891#endif
Guido van Rossum30a685f1991-06-27 15:51:29 +00002892
2893/* List of methods for socket objects */
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002894
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002895static PyMethodDef sock_methods[] = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002896 {"_accept", (PyCFunction)sock_accept, METH_NOARGS,
2897 accept_doc},
2898 {"bind", (PyCFunction)sock_bind, METH_O,
2899 bind_doc},
2900 {"close", (PyCFunction)sock_close, METH_NOARGS,
2901 close_doc},
2902 {"connect", (PyCFunction)sock_connect, METH_O,
2903 connect_doc},
2904 {"connect_ex", (PyCFunction)sock_connect_ex, METH_O,
2905 connect_ex_doc},
Antoine Pitrou6e451df2010-08-09 20:39:54 +00002906 {"detach", (PyCFunction)sock_detach, METH_NOARGS,
2907 detach_doc},
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002908 {"fileno", (PyCFunction)sock_fileno, METH_NOARGS,
2909 fileno_doc},
Guido van Rossumb6775db1994-08-01 11:34:53 +00002910#ifdef HAVE_GETPEERNAME
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002911 {"getpeername", (PyCFunction)sock_getpeername,
2912 METH_NOARGS, getpeername_doc},
Guido van Rossum9575a441993-04-07 14:06:14 +00002913#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002914 {"getsockname", (PyCFunction)sock_getsockname,
2915 METH_NOARGS, getsockname_doc},
2916 {"getsockopt", (PyCFunction)sock_getsockopt, METH_VARARGS,
2917 getsockopt_doc},
Amaury Forgeot d'Arc3d17a5c2008-06-13 01:09:34 +00002918#if defined(MS_WINDOWS) && defined(SIO_RCVALL)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002919 {"ioctl", (PyCFunction)sock_ioctl, METH_VARARGS,
2920 sock_ioctl_doc},
Christian Heimesfaf2f632008-01-06 16:59:19 +00002921#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002922 {"listen", (PyCFunction)sock_listen, METH_O,
2923 listen_doc},
2924 {"recv", (PyCFunction)sock_recv, METH_VARARGS,
2925 recv_doc},
2926 {"recv_into", (PyCFunction)sock_recv_into, METH_VARARGS | METH_KEYWORDS,
2927 recv_into_doc},
2928 {"recvfrom", (PyCFunction)sock_recvfrom, METH_VARARGS,
2929 recvfrom_doc},
2930 {"recvfrom_into", (PyCFunction)sock_recvfrom_into, METH_VARARGS | METH_KEYWORDS,
2931 recvfrom_into_doc},
2932 {"send", (PyCFunction)sock_send, METH_VARARGS,
2933 send_doc},
2934 {"sendall", (PyCFunction)sock_sendall, METH_VARARGS,
2935 sendall_doc},
2936 {"sendto", (PyCFunction)sock_sendto, METH_VARARGS,
2937 sendto_doc},
2938 {"setblocking", (PyCFunction)sock_setblocking, METH_O,
2939 setblocking_doc},
2940 {"settimeout", (PyCFunction)sock_settimeout, METH_O,
2941 settimeout_doc},
2942 {"gettimeout", (PyCFunction)sock_gettimeout, METH_NOARGS,
2943 gettimeout_doc},
2944 {"setsockopt", (PyCFunction)sock_setsockopt, METH_VARARGS,
2945 setsockopt_doc},
2946 {"shutdown", (PyCFunction)sock_shutdown, METH_O,
2947 shutdown_doc},
2948 {NULL, NULL} /* sentinel */
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002949};
2950
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002951/* SockObject members */
2952static PyMemberDef sock_memberlist[] = {
2953 {"family", T_INT, offsetof(PySocketSockObject, sock_family), READONLY, "the socket family"},
2954 {"type", T_INT, offsetof(PySocketSockObject, sock_type), READONLY, "the socket type"},
2955 {"proto", T_INT, offsetof(PySocketSockObject, sock_proto), READONLY, "the socket protocol"},
2956 {"timeout", T_DOUBLE, offsetof(PySocketSockObject, sock_timeout), READONLY, "the socket timeout"},
2957 {0},
2958};
Guido van Rossum30a685f1991-06-27 15:51:29 +00002959
Guido van Rossum73624e91994-10-10 17:59:00 +00002960/* Deallocate a socket object in response to the last Py_DECREF().
Guido van Rossum30a685f1991-06-27 15:51:29 +00002961 First close the file description. */
2962
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002963static void
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002964sock_dealloc(PySocketSockObject *s)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002965{
Antoine Pitroue033e062010-10-29 10:38:18 +00002966 if (s->sock_fd != -1) {
2967 PyObject *exc, *val, *tb;
2968 Py_ssize_t old_refcount = Py_REFCNT(s);
2969 ++Py_REFCNT(s);
2970 PyErr_Fetch(&exc, &val, &tb);
2971 if (PyErr_WarnFormat(PyExc_ResourceWarning, 1,
2972 "unclosed %R", s))
2973 /* Spurious errors can appear at shutdown */
2974 if (PyErr_ExceptionMatches(PyExc_Warning))
2975 PyErr_WriteUnraisable((PyObject *) s);
2976 PyErr_Restore(exc, val, tb);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002977 (void) SOCKETCLOSE(s->sock_fd);
Antoine Pitroue033e062010-10-29 10:38:18 +00002978 Py_REFCNT(s) = old_refcount;
2979 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002980 Py_TYPE(s)->tp_free((PyObject *)s);
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002981}
2982
Guido van Rossum30a685f1991-06-27 15:51:29 +00002983
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00002984static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002985sock_repr(PySocketSockObject *s)
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00002986{
Fred Drakea04eaad2000-06-30 02:46:07 +00002987#if SIZEOF_SOCKET_T > SIZEOF_LONG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002988 if (s->sock_fd > LONG_MAX) {
2989 /* this can occur on Win64, and actually there is a special
2990 ugly printf formatter for decimal pointer length integer
2991 printing, only bother if necessary*/
2992 PyErr_SetString(PyExc_OverflowError,
2993 "no printf formatter to display "
2994 "the socket descriptor in decimal");
2995 return NULL;
2996 }
Fred Drakea04eaad2000-06-30 02:46:07 +00002997#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002998 return PyUnicode_FromFormat(
2999 "<socket object, fd=%ld, family=%d, type=%d, proto=%d>",
3000 (long)s->sock_fd, s->sock_family,
3001 s->sock_type,
3002 s->sock_proto);
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00003003}
3004
3005
Guido van Rossum384ca9c2001-10-27 22:20:47 +00003006/* Create a new, uninitialized socket object. */
3007
3008static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003009sock_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
Guido van Rossum384ca9c2001-10-27 22:20:47 +00003010{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003011 PyObject *new;
Guido van Rossum384ca9c2001-10-27 22:20:47 +00003012
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003013 new = type->tp_alloc(type, 0);
3014 if (new != NULL) {
3015 ((PySocketSockObject *)new)->sock_fd = -1;
3016 ((PySocketSockObject *)new)->sock_timeout = -1.0;
3017 ((PySocketSockObject *)new)->errorhandler = &set_error;
3018 }
3019 return new;
Guido van Rossum384ca9c2001-10-27 22:20:47 +00003020}
3021
3022
3023/* Initialize a new socket object. */
3024
3025/*ARGSUSED*/
3026static int
Andrew MacIntyre7aec4a22002-06-13 11:53:52 +00003027sock_initobj(PyObject *self, PyObject *args, PyObject *kwds)
Guido van Rossum384ca9c2001-10-27 22:20:47 +00003028{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003029 PySocketSockObject *s = (PySocketSockObject *)self;
3030 PyObject *fdobj = NULL;
3031 SOCKET_T fd = INVALID_SOCKET;
3032 int family = AF_INET, type = SOCK_STREAM, proto = 0;
3033 static char *keywords[] = {"family", "type", "proto", "fileno", 0};
Guido van Rossum384ca9c2001-10-27 22:20:47 +00003034
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003035 if (!PyArg_ParseTupleAndKeywords(args, kwds,
3036 "|iiiO:socket", keywords,
3037 &family, &type, &proto, &fdobj))
3038 return -1;
Guido van Rossum67f7a382002-06-06 21:08:16 +00003039
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003040 if (fdobj != NULL && fdobj != Py_None) {
3041 fd = PyLong_AsSocket_t(fdobj);
3042 if (fd == (SOCKET_T)(-1) && PyErr_Occurred())
3043 return -1;
3044 if (fd == INVALID_SOCKET) {
3045 PyErr_SetString(PyExc_ValueError,
3046 "can't use invalid socket value");
3047 return -1;
3048 }
3049 }
3050 else {
3051 Py_BEGIN_ALLOW_THREADS
3052 fd = socket(family, type, proto);
3053 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00003054
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003055 if (fd == INVALID_SOCKET) {
3056 set_error();
3057 return -1;
3058 }
3059 }
3060 init_sockobject(s, fd, family, type, proto);
Guido van Rossum67f7a382002-06-06 21:08:16 +00003061
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003062 return 0;
Guido van Rossum67f7a382002-06-06 21:08:16 +00003063
Guido van Rossum384ca9c2001-10-27 22:20:47 +00003064}
3065
3066
Guido van Rossumb6775db1994-08-01 11:34:53 +00003067/* Type object for socket objects. */
Guido van Rossum30a685f1991-06-27 15:51:29 +00003068
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003069static PyTypeObject sock_type = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003070 PyVarObject_HEAD_INIT(0, 0) /* Must fill in type value later */
3071 "_socket.socket", /* tp_name */
3072 sizeof(PySocketSockObject), /* tp_basicsize */
3073 0, /* tp_itemsize */
3074 (destructor)sock_dealloc, /* tp_dealloc */
3075 0, /* tp_print */
3076 0, /* tp_getattr */
3077 0, /* tp_setattr */
3078 0, /* tp_reserved */
3079 (reprfunc)sock_repr, /* tp_repr */
3080 0, /* tp_as_number */
3081 0, /* tp_as_sequence */
3082 0, /* tp_as_mapping */
3083 0, /* tp_hash */
3084 0, /* tp_call */
3085 0, /* tp_str */
3086 PyObject_GenericGetAttr, /* tp_getattro */
3087 0, /* tp_setattro */
3088 0, /* tp_as_buffer */
3089 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE, /* tp_flags */
3090 sock_doc, /* tp_doc */
3091 0, /* tp_traverse */
3092 0, /* tp_clear */
3093 0, /* tp_richcompare */
3094 0, /* tp_weaklistoffset */
3095 0, /* tp_iter */
3096 0, /* tp_iternext */
3097 sock_methods, /* tp_methods */
3098 sock_memberlist, /* tp_members */
3099 0, /* tp_getset */
3100 0, /* tp_base */
3101 0, /* tp_dict */
3102 0, /* tp_descr_get */
3103 0, /* tp_descr_set */
3104 0, /* tp_dictoffset */
3105 sock_initobj, /* tp_init */
3106 PyType_GenericAlloc, /* tp_alloc */
3107 sock_new, /* tp_new */
3108 PyObject_Del, /* tp_free */
Guido van Rossum6574b3e1991-06-25 21:36:08 +00003109};
3110
Guido van Rossum30a685f1991-06-27 15:51:29 +00003111
Guido van Rossum81194471991-07-27 21:42:02 +00003112/* Python interface to gethostname(). */
3113
3114/*ARGSUSED*/
Guido van Rossum73624e91994-10-10 17:59:00 +00003115static PyObject *
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00003116socket_gethostname(PyObject *self, PyObject *unused)
Guido van Rossum81194471991-07-27 21:42:02 +00003117{
Martin v. Löwis72f48422010-10-29 18:20:08 +00003118#ifdef MS_WINDOWS
3119 /* Don't use winsock's gethostname, as this returns the ANSI
3120 version of the hostname, whereas we need a Unicode string.
3121 Otherwise, gethostname apparently also returns the DNS name. */
Hirokazu Yamamoto09fff7a2010-11-05 17:24:13 +00003122 wchar_t buf[MAX_COMPUTERNAME_LENGTH + 1];
3123 DWORD size = sizeof(buf) / sizeof(wchar_t);
3124 PyObject *result;
Martin v. Löwis72f48422010-10-29 18:20:08 +00003125 if (!GetComputerNameExW(ComputerNamePhysicalDnsHostname, buf, &size)) {
3126 if (GetLastError() == ERROR_MORE_DATA) {
3127 /* MSDN says this may occur "because DNS allows longer names */
Hirokazu Yamamoto09fff7a2010-11-05 17:24:13 +00003128 if (size == 0) /* XXX: I'm not sure how to handle this */
3129 return PyUnicode_FromUnicode(NULL, 0);
3130 result = PyUnicode_FromUnicode(NULL, size - 1);
Martin v. Löwis72f48422010-10-29 18:20:08 +00003131 if (!result)
3132 return NULL;
Hirokazu Yamamoto09fff7a2010-11-05 17:24:13 +00003133 if (GetComputerNameExW(ComputerNamePhysicalDnsHostname,
3134 PyUnicode_AS_UNICODE(result),
3135 &size))
Martin v. Löwis72f48422010-10-29 18:20:08 +00003136 return result;
Benjamin Petersone857b292010-10-29 21:37:26 +00003137 Py_DECREF(result);
Martin v. Löwis72f48422010-10-29 18:20:08 +00003138 }
3139 return PyErr_SetExcFromWindowsErr(PyExc_WindowsError, GetLastError());
3140 }
3141 return PyUnicode_FromUnicode(buf, size);
3142#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003143 char buf[1024];
3144 int res;
3145 Py_BEGIN_ALLOW_THREADS
3146 res = gethostname(buf, (int) sizeof buf - 1);
3147 Py_END_ALLOW_THREADS
3148 if (res < 0)
3149 return set_error();
3150 buf[sizeof buf - 1] = '\0';
3151 return PyUnicode_FromString(buf);
Martin v. Löwis72f48422010-10-29 18:20:08 +00003152#endif
Guido van Rossum81194471991-07-27 21:42:02 +00003153}
Guido van Rossumff4949e1992-08-05 19:58:53 +00003154
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003155PyDoc_STRVAR(gethostname_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003156"gethostname() -> string\n\
3157\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003158Return the current host name.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003159
Guido van Rossumff4949e1992-08-05 19:58:53 +00003160
Guido van Rossum30a685f1991-06-27 15:51:29 +00003161/* Python interface to gethostbyname(name). */
3162
3163/*ARGSUSED*/
Guido van Rossum73624e91994-10-10 17:59:00 +00003164static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003165socket_gethostbyname(PyObject *self, PyObject *args)
Guido van Rossum30a685f1991-06-27 15:51:29 +00003166{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003167 char *name;
3168 sock_addr_t addrbuf;
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003169 PyObject *ret = NULL;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003170
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003171 if (!PyArg_ParseTuple(args, "et:gethostbyname", "idna", &name))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003172 return NULL;
3173 if (setipaddr(name, SAS2SA(&addrbuf), sizeof(addrbuf), AF_INET) < 0)
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003174 goto finally;
3175 ret = makeipaddr(SAS2SA(&addrbuf), sizeof(struct sockaddr_in));
3176finally:
3177 PyMem_Free(name);
3178 return ret;
Guido van Rossum30a685f1991-06-27 15:51:29 +00003179}
3180
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003181PyDoc_STRVAR(gethostbyname_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003182"gethostbyname(host) -> address\n\
3183\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003184Return the IP address (a string of the form '255.255.255.255') for a host.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003185
3186
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003187/* Convenience function common to gethostbyname_ex and gethostbyaddr */
3188
3189static PyObject *
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003190gethost_common(struct hostent *h, struct sockaddr *addr, int alen, int af)
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003191{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003192 char **pch;
3193 PyObject *rtn_tuple = (PyObject *)NULL;
3194 PyObject *name_list = (PyObject *)NULL;
3195 PyObject *addr_list = (PyObject *)NULL;
3196 PyObject *tmp;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003197
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003198 if (h == NULL) {
3199 /* Let's get real error message to return */
3200 set_herror(h_errno);
3201 return NULL;
3202 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00003203
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003204 if (h->h_addrtype != af) {
3205 /* Let's get real error message to return */
3206 PyErr_SetString(socket_error,
3207 (char *)strerror(EAFNOSUPPORT));
Christian Heimesada8c3b2008-03-18 18:26:33 +00003208
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003209 return NULL;
3210 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00003211
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003212 switch (af) {
Guido van Rossum67f7a382002-06-06 21:08:16 +00003213
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003214 case AF_INET:
3215 if (alen < sizeof(struct sockaddr_in))
3216 return NULL;
3217 break;
Guido van Rossum67f7a382002-06-06 21:08:16 +00003218
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00003219#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003220 case AF_INET6:
3221 if (alen < sizeof(struct sockaddr_in6))
3222 return NULL;
3223 break;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003224#endif
Guido van Rossum67f7a382002-06-06 21:08:16 +00003225
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003226 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00003227
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003228 if ((name_list = PyList_New(0)) == NULL)
3229 goto err;
Guido van Rossum67f7a382002-06-06 21:08:16 +00003230
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003231 if ((addr_list = PyList_New(0)) == NULL)
3232 goto err;
Guido van Rossum67f7a382002-06-06 21:08:16 +00003233
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003234 /* SF #1511317: h_aliases can be NULL */
3235 if (h->h_aliases) {
3236 for (pch = h->h_aliases; *pch != NULL; pch++) {
3237 int status;
3238 tmp = PyUnicode_FromString(*pch);
3239 if (tmp == NULL)
3240 goto err;
Guido van Rossum67f7a382002-06-06 21:08:16 +00003241
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003242 status = PyList_Append(name_list, tmp);
3243 Py_DECREF(tmp);
Guido van Rossum67f7a382002-06-06 21:08:16 +00003244
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003245 if (status)
3246 goto err;
3247 }
3248 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00003249
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003250 for (pch = h->h_addr_list; *pch != NULL; pch++) {
3251 int status;
Guido van Rossum67f7a382002-06-06 21:08:16 +00003252
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003253 switch (af) {
Guido van Rossum67f7a382002-06-06 21:08:16 +00003254
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003255 case AF_INET:
3256 {
3257 struct sockaddr_in sin;
3258 memset(&sin, 0, sizeof(sin));
3259 sin.sin_family = af;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003260#ifdef HAVE_SOCKADDR_SA_LEN
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003261 sin.sin_len = sizeof(sin);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003262#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003263 memcpy(&sin.sin_addr, *pch, sizeof(sin.sin_addr));
3264 tmp = makeipaddr((struct sockaddr *)&sin, sizeof(sin));
Guido van Rossum67f7a382002-06-06 21:08:16 +00003265
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003266 if (pch == h->h_addr_list && alen >= sizeof(sin))
3267 memcpy((char *) addr, &sin, sizeof(sin));
3268 break;
3269 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00003270
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00003271#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003272 case AF_INET6:
3273 {
3274 struct sockaddr_in6 sin6;
3275 memset(&sin6, 0, sizeof(sin6));
3276 sin6.sin6_family = af;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003277#ifdef HAVE_SOCKADDR_SA_LEN
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003278 sin6.sin6_len = sizeof(sin6);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003279#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003280 memcpy(&sin6.sin6_addr, *pch, sizeof(sin6.sin6_addr));
3281 tmp = makeipaddr((struct sockaddr *)&sin6,
3282 sizeof(sin6));
Guido van Rossum67f7a382002-06-06 21:08:16 +00003283
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003284 if (pch == h->h_addr_list && alen >= sizeof(sin6))
3285 memcpy((char *) addr, &sin6, sizeof(sin6));
3286 break;
3287 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003288#endif
Guido van Rossum67f7a382002-06-06 21:08:16 +00003289
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003290 default: /* can't happen */
3291 PyErr_SetString(socket_error,
3292 "unsupported address family");
3293 return NULL;
3294 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00003295
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003296 if (tmp == NULL)
3297 goto err;
Guido van Rossum67f7a382002-06-06 21:08:16 +00003298
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003299 status = PyList_Append(addr_list, tmp);
3300 Py_DECREF(tmp);
Guido van Rossum67f7a382002-06-06 21:08:16 +00003301
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003302 if (status)
3303 goto err;
3304 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00003305
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003306 rtn_tuple = Py_BuildValue("sOO", h->h_name, name_list, addr_list);
Guido van Rossum67f7a382002-06-06 21:08:16 +00003307
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003308 err:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003309 Py_XDECREF(name_list);
3310 Py_XDECREF(addr_list);
3311 return rtn_tuple;
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003312}
3313
3314
3315/* Python interface to gethostbyname_ex(name). */
3316
3317/*ARGSUSED*/
3318static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003319socket_gethostbyname_ex(PyObject *self, PyObject *args)
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003320{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003321 char *name;
3322 struct hostent *h;
Martin v. Löwisc16f3bd2003-05-03 09:14:54 +00003323#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003324 struct sockaddr_storage addr;
Martin v. Löwisc16f3bd2003-05-03 09:14:54 +00003325#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003326 struct sockaddr_in addr;
Martin v. Löwisc16f3bd2003-05-03 09:14:54 +00003327#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003328 struct sockaddr *sa;
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003329 PyObject *ret = NULL;
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003330#ifdef HAVE_GETHOSTBYNAME_R
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003331 struct hostent hp_allocated;
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003332#ifdef HAVE_GETHOSTBYNAME_R_3_ARG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003333 struct hostent_data data;
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003334#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003335 char buf[16384];
3336 int buf_len = (sizeof buf) - 1;
3337 int errnop;
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003338#endif
3339#if defined(HAVE_GETHOSTBYNAME_R_3_ARG) || defined(HAVE_GETHOSTBYNAME_R_6_ARG)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003340 int result;
Guido van Rossume9cd07b1999-03-15 21:40:14 +00003341#endif
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003342#endif /* HAVE_GETHOSTBYNAME_R */
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003343
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003344 if (!PyArg_ParseTuple(args, "et:gethostbyname_ex", "idna", &name))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003345 return NULL;
3346 if (setipaddr(name, (struct sockaddr *)&addr, sizeof(addr), AF_INET) < 0)
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003347 goto finally;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003348 Py_BEGIN_ALLOW_THREADS
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003349#ifdef HAVE_GETHOSTBYNAME_R
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003350#if defined(HAVE_GETHOSTBYNAME_R_6_ARG)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003351 result = gethostbyname_r(name, &hp_allocated, buf, buf_len,
3352 &h, &errnop);
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003353#elif defined(HAVE_GETHOSTBYNAME_R_5_ARG)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003354 h = gethostbyname_r(name, &hp_allocated, buf, buf_len, &errnop);
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003355#else /* HAVE_GETHOSTBYNAME_R_3_ARG */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003356 memset((void *) &data, '\0', sizeof(data));
3357 result = gethostbyname_r(name, &hp_allocated, &data);
3358 h = (result != 0) ? NULL : &hp_allocated;
Guido van Rossume9cd07b1999-03-15 21:40:14 +00003359#endif
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003360#else /* not HAVE_GETHOSTBYNAME_R */
Guido van Rossum3baaa131999-03-22 21:44:51 +00003361#ifdef USE_GETHOSTBYNAME_LOCK
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003362 PyThread_acquire_lock(netdb_lock, 1);
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003363#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003364 h = gethostbyname(name);
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003365#endif /* HAVE_GETHOSTBYNAME_R */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003366 Py_END_ALLOW_THREADS
3367 /* Some C libraries would require addr.__ss_family instead of
3368 addr.ss_family.
3369 Therefore, we cast the sockaddr_storage into sockaddr to
3370 access sa_family. */
3371 sa = (struct sockaddr*)&addr;
3372 ret = gethost_common(h, (struct sockaddr *)&addr, sizeof(addr),
3373 sa->sa_family);
Guido van Rossum3baaa131999-03-22 21:44:51 +00003374#ifdef USE_GETHOSTBYNAME_LOCK
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003375 PyThread_release_lock(netdb_lock);
Guido van Rossum955becc1999-03-22 20:14:53 +00003376#endif
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003377finally:
3378 PyMem_Free(name);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003379 return ret;
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003380}
3381
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003382PyDoc_STRVAR(ghbn_ex_doc,
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003383"gethostbyname_ex(host) -> (name, aliaslist, addresslist)\n\
3384\n\
3385Return the true host name, a list of aliases, and a list of IP addresses,\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003386for a host. The host argument is a string giving a host name or IP number.");
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003387
3388
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00003389/* Python interface to gethostbyaddr(IP). */
3390
3391/*ARGSUSED*/
3392static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003393socket_gethostbyaddr(PyObject *self, PyObject *args)
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00003394{
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00003395#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003396 struct sockaddr_storage addr;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003397#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003398 struct sockaddr_in addr;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003399#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003400 struct sockaddr *sa = (struct sockaddr *)&addr;
3401 char *ip_num;
3402 struct hostent *h;
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003403 PyObject *ret = NULL;
Guido van Rossum4f199ea1998-04-09 20:56:35 +00003404#ifdef HAVE_GETHOSTBYNAME_R
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003405 struct hostent hp_allocated;
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003406#ifdef HAVE_GETHOSTBYNAME_R_3_ARG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003407 struct hostent_data data;
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003408#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003409 /* glibcs up to 2.10 assume that the buf argument to
3410 gethostbyaddr_r is 8-byte aligned, which at least llvm-gcc
3411 does not ensure. The attribute below instructs the compiler
3412 to maintain this alignment. */
3413 char buf[16384] Py_ALIGNED(8);
3414 int buf_len = (sizeof buf) - 1;
3415 int errnop;
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003416#endif
3417#if defined(HAVE_GETHOSTBYNAME_R_3_ARG) || defined(HAVE_GETHOSTBYNAME_R_6_ARG)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003418 int result;
Guido van Rossume9cd07b1999-03-15 21:40:14 +00003419#endif
Guido van Rossum4f199ea1998-04-09 20:56:35 +00003420#endif /* HAVE_GETHOSTBYNAME_R */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003421 char *ap;
3422 int al;
3423 int af;
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00003424
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003425 if (!PyArg_ParseTuple(args, "et:gethostbyaddr", "idna", &ip_num))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003426 return NULL;
3427 af = AF_UNSPEC;
3428 if (setipaddr(ip_num, sa, sizeof(addr), af) < 0)
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003429 goto finally;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003430 af = sa->sa_family;
3431 ap = NULL;
3432 al = 0;
3433 switch (af) {
3434 case AF_INET:
3435 ap = (char *)&((struct sockaddr_in *)sa)->sin_addr;
3436 al = sizeof(((struct sockaddr_in *)sa)->sin_addr);
3437 break;
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00003438#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003439 case AF_INET6:
3440 ap = (char *)&((struct sockaddr_in6 *)sa)->sin6_addr;
3441 al = sizeof(((struct sockaddr_in6 *)sa)->sin6_addr);
3442 break;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003443#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003444 default:
3445 PyErr_SetString(socket_error, "unsupported address family");
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003446 goto finally;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003447 }
3448 Py_BEGIN_ALLOW_THREADS
Guido van Rossum4f199ea1998-04-09 20:56:35 +00003449#ifdef HAVE_GETHOSTBYNAME_R
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003450#if defined(HAVE_GETHOSTBYNAME_R_6_ARG)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003451 result = gethostbyaddr_r(ap, al, af,
3452 &hp_allocated, buf, buf_len,
3453 &h, &errnop);
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003454#elif defined(HAVE_GETHOSTBYNAME_R_5_ARG)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003455 h = gethostbyaddr_r(ap, al, af,
3456 &hp_allocated, buf, buf_len, &errnop);
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003457#else /* HAVE_GETHOSTBYNAME_R_3_ARG */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003458 memset((void *) &data, '\0', sizeof(data));
3459 result = gethostbyaddr_r(ap, al, af, &hp_allocated, &data);
3460 h = (result != 0) ? NULL : &hp_allocated;
Guido van Rossume9cd07b1999-03-15 21:40:14 +00003461#endif
Guido van Rossum4f199ea1998-04-09 20:56:35 +00003462#else /* not HAVE_GETHOSTBYNAME_R */
Guido van Rossum3baaa131999-03-22 21:44:51 +00003463#ifdef USE_GETHOSTBYNAME_LOCK
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003464 PyThread_acquire_lock(netdb_lock, 1);
Guido van Rossum4f199ea1998-04-09 20:56:35 +00003465#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003466 h = gethostbyaddr(ap, al, af);
Guido van Rossum4f199ea1998-04-09 20:56:35 +00003467#endif /* HAVE_GETHOSTBYNAME_R */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003468 Py_END_ALLOW_THREADS
3469 ret = gethost_common(h, (struct sockaddr *)&addr, sizeof(addr), af);
Guido van Rossum3baaa131999-03-22 21:44:51 +00003470#ifdef USE_GETHOSTBYNAME_LOCK
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003471 PyThread_release_lock(netdb_lock);
Guido van Rossum3baaa131999-03-22 21:44:51 +00003472#endif
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003473finally:
3474 PyMem_Free(ip_num);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003475 return ret;
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00003476}
3477
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003478PyDoc_STRVAR(gethostbyaddr_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003479"gethostbyaddr(host) -> (name, aliaslist, addresslist)\n\
3480\n\
3481Return the true host name, a list of aliases, and a list of IP addresses,\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003482for a host. The host argument is a string giving a host name or IP number.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003483
Guido van Rossum30a685f1991-06-27 15:51:29 +00003484
3485/* Python interface to getservbyname(name).
3486 This only returns the port number, since the other info is already
3487 known or not useful (like the list of aliases). */
3488
3489/*ARGSUSED*/
Guido van Rossum73624e91994-10-10 17:59:00 +00003490static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003491socket_getservbyname(PyObject *self, PyObject *args)
Guido van Rossum30a685f1991-06-27 15:51:29 +00003492{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003493 char *name, *proto=NULL;
3494 struct servent *sp;
3495 if (!PyArg_ParseTuple(args, "s|s:getservbyname", &name, &proto))
3496 return NULL;
3497 Py_BEGIN_ALLOW_THREADS
3498 sp = getservbyname(name, proto);
3499 Py_END_ALLOW_THREADS
3500 if (sp == NULL) {
3501 PyErr_SetString(socket_error, "service/proto not found");
3502 return NULL;
3503 }
3504 return PyLong_FromLong((long) ntohs(sp->s_port));
Guido van Rossum30a685f1991-06-27 15:51:29 +00003505}
3506
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003507PyDoc_STRVAR(getservbyname_doc,
Barry Warsaw11b91a02004-06-28 00:50:43 +00003508"getservbyname(servicename[, protocolname]) -> integer\n\
Guido van Rossum82a5c661998-07-07 20:45:43 +00003509\n\
3510Return a port number from a service name and protocol name.\n\
Barry Warsaw11b91a02004-06-28 00:50:43 +00003511The optional protocol name, if given, should be 'tcp' or 'udp',\n\
3512otherwise any protocol will match.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003513
Guido van Rossum30a685f1991-06-27 15:51:29 +00003514
Barry Warsaw11b91a02004-06-28 00:50:43 +00003515/* Python interface to getservbyport(port).
3516 This only returns the service name, since the other info is already
3517 known or not useful (like the list of aliases). */
3518
3519/*ARGSUSED*/
3520static PyObject *
3521socket_getservbyport(PyObject *self, PyObject *args)
3522{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003523 int port;
3524 char *proto=NULL;
3525 struct servent *sp;
3526 if (!PyArg_ParseTuple(args, "i|s:getservbyport", &port, &proto))
3527 return NULL;
3528 if (port < 0 || port > 0xffff) {
3529 PyErr_SetString(
3530 PyExc_OverflowError,
3531 "getservbyport: port must be 0-65535.");
3532 return NULL;
3533 }
3534 Py_BEGIN_ALLOW_THREADS
3535 sp = getservbyport(htons((short)port), proto);
3536 Py_END_ALLOW_THREADS
3537 if (sp == NULL) {
3538 PyErr_SetString(socket_error, "port/proto not found");
3539 return NULL;
3540 }
3541 return PyUnicode_FromString(sp->s_name);
Barry Warsaw11b91a02004-06-28 00:50:43 +00003542}
3543
3544PyDoc_STRVAR(getservbyport_doc,
3545"getservbyport(port[, protocolname]) -> string\n\
3546\n\
3547Return the service name from a port number and protocol name.\n\
3548The optional protocol name, if given, should be 'tcp' or 'udp',\n\
3549otherwise any protocol will match.");
3550
Guido van Rossum3901d851996-12-19 16:35:04 +00003551/* Python interface to getprotobyname(name).
3552 This only returns the protocol number, since the other info is
3553 already known or not useful (like the list of aliases). */
3554
3555/*ARGSUSED*/
3556static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003557socket_getprotobyname(PyObject *self, PyObject *args)
Guido van Rossum3901d851996-12-19 16:35:04 +00003558{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003559 char *name;
3560 struct protoent *sp;
3561 if (!PyArg_ParseTuple(args, "s:getprotobyname", &name))
3562 return NULL;
3563 Py_BEGIN_ALLOW_THREADS
3564 sp = getprotobyname(name);
3565 Py_END_ALLOW_THREADS
3566 if (sp == NULL) {
3567 PyErr_SetString(socket_error, "protocol not found");
3568 return NULL;
3569 }
3570 return PyLong_FromLong((long) sp->p_proto);
Guido van Rossum3901d851996-12-19 16:35:04 +00003571}
3572
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003573PyDoc_STRVAR(getprotobyname_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003574"getprotobyname(name) -> integer\n\
3575\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003576Return the protocol number for the named protocol. (Rarely used.)");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003577
Guido van Rossum3901d851996-12-19 16:35:04 +00003578
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00003579#ifndef NO_DUP
3580/* dup() function for socket fds */
3581
3582static PyObject *
3583socket_dup(PyObject *self, PyObject *fdobj)
3584{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003585 SOCKET_T fd, newfd;
3586 PyObject *newfdobj;
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00003587
3588
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003589 fd = PyLong_AsSocket_t(fdobj);
3590 if (fd == (SOCKET_T)(-1) && PyErr_Occurred())
3591 return NULL;
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00003592
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003593 newfd = dup_socket(fd);
3594 if (newfd == INVALID_SOCKET)
3595 return set_error();
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00003596
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003597 newfdobj = PyLong_FromSocket_t(newfd);
3598 if (newfdobj == NULL)
3599 SOCKETCLOSE(newfd);
3600 return newfdobj;
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00003601}
3602
3603PyDoc_STRVAR(dup_doc,
3604"dup(integer) -> integer\n\
3605\n\
3606Duplicate an integer socket file descriptor. This is like os.dup(), but for\n\
3607sockets; on some platforms os.dup() won't work for socket file descriptors.");
3608#endif
3609
3610
Dave Cole331708b2004-08-09 04:51:41 +00003611#ifdef HAVE_SOCKETPAIR
3612/* Create a pair of sockets using the socketpair() function.
Dave Cole07fda7e2004-08-23 05:16:23 +00003613 Arguments as for socket() except the default family is AF_UNIX if
Dave Colee8bbfe42004-08-26 00:51:16 +00003614 defined on the platform; otherwise, the default is AF_INET. */
Dave Cole331708b2004-08-09 04:51:41 +00003615
3616/*ARGSUSED*/
3617static PyObject *
3618socket_socketpair(PyObject *self, PyObject *args)
3619{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003620 PySocketSockObject *s0 = NULL, *s1 = NULL;
3621 SOCKET_T sv[2];
3622 int family, type = SOCK_STREAM, proto = 0;
3623 PyObject *res = NULL;
Dave Cole331708b2004-08-09 04:51:41 +00003624
3625#if defined(AF_UNIX)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003626 family = AF_UNIX;
Dave Cole331708b2004-08-09 04:51:41 +00003627#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003628 family = AF_INET;
Dave Cole331708b2004-08-09 04:51:41 +00003629#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003630 if (!PyArg_ParseTuple(args, "|iii:socketpair",
3631 &family, &type, &proto))
3632 return NULL;
3633 /* Create a pair of socket fds */
3634 if (socketpair(family, type, proto, sv) < 0)
3635 return set_error();
3636 s0 = new_sockobject(sv[0], family, type, proto);
3637 if (s0 == NULL)
3638 goto finally;
3639 s1 = new_sockobject(sv[1], family, type, proto);
3640 if (s1 == NULL)
3641 goto finally;
3642 res = PyTuple_Pack(2, s0, s1);
Dave Cole331708b2004-08-09 04:51:41 +00003643
3644finally:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003645 if (res == NULL) {
3646 if (s0 == NULL)
3647 SOCKETCLOSE(sv[0]);
3648 if (s1 == NULL)
3649 SOCKETCLOSE(sv[1]);
3650 }
3651 Py_XDECREF(s0);
3652 Py_XDECREF(s1);
3653 return res;
Dave Cole331708b2004-08-09 04:51:41 +00003654}
3655
3656PyDoc_STRVAR(socketpair_doc,
3657"socketpair([family[, type[, proto]]]) -> (socket object, socket object)\n\
3658\n\
3659Create a pair of socket objects from the sockets returned by the platform\n\
3660socketpair() function.\n\
Dave Cole07fda7e2004-08-23 05:16:23 +00003661The arguments are the same as for socket() except the default family is\n\
Dave Colee8bbfe42004-08-26 00:51:16 +00003662AF_UNIX if defined on the platform; otherwise, the default is AF_INET.");
Dave Cole331708b2004-08-09 04:51:41 +00003663
3664#endif /* HAVE_SOCKETPAIR */
3665
3666
Guido van Rossum006bf911996-06-12 04:04:55 +00003667static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003668socket_ntohs(PyObject *self, PyObject *args)
Guido van Rossum006bf911996-06-12 04:04:55 +00003669{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003670 int x1, x2;
Guido van Rossum006bf911996-06-12 04:04:55 +00003671
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003672 if (!PyArg_ParseTuple(args, "i:ntohs", &x1)) {
3673 return NULL;
3674 }
3675 if (x1 < 0) {
3676 PyErr_SetString(PyExc_OverflowError,
3677 "can't convert negative number to unsigned long");
3678 return NULL;
3679 }
3680 x2 = (unsigned int)ntohs((unsigned short)x1);
3681 return PyLong_FromLong(x2);
Guido van Rossum006bf911996-06-12 04:04:55 +00003682}
3683
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003684PyDoc_STRVAR(ntohs_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003685"ntohs(integer) -> integer\n\
3686\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003687Convert a 16-bit integer from network to host byte order.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003688
3689
Guido van Rossum006bf911996-06-12 04:04:55 +00003690static PyObject *
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003691socket_ntohl(PyObject *self, PyObject *arg)
Guido van Rossum006bf911996-06-12 04:04:55 +00003692{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003693 unsigned long x;
Guido van Rossum006bf911996-06-12 04:04:55 +00003694
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003695 if (PyLong_Check(arg)) {
3696 x = PyLong_AsUnsignedLong(arg);
3697 if (x == (unsigned long) -1 && PyErr_Occurred())
3698 return NULL;
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003699#if SIZEOF_LONG > 4
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003700 {
3701 unsigned long y;
3702 /* only want the trailing 32 bits */
3703 y = x & 0xFFFFFFFFUL;
3704 if (y ^ x)
3705 return PyErr_Format(PyExc_OverflowError,
3706 "long int larger than 32 bits");
3707 x = y;
3708 }
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003709#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003710 }
3711 else
3712 return PyErr_Format(PyExc_TypeError,
3713 "expected int/long, %s found",
3714 Py_TYPE(arg)->tp_name);
3715 if (x == (unsigned long) -1 && PyErr_Occurred())
3716 return NULL;
3717 return PyLong_FromUnsignedLong(ntohl(x));
Guido van Rossum006bf911996-06-12 04:04:55 +00003718}
3719
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003720PyDoc_STRVAR(ntohl_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003721"ntohl(integer) -> integer\n\
3722\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003723Convert a 32-bit integer from network to host byte order.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003724
3725
Guido van Rossum006bf911996-06-12 04:04:55 +00003726static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003727socket_htons(PyObject *self, PyObject *args)
Guido van Rossum006bf911996-06-12 04:04:55 +00003728{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003729 int x1, x2;
Guido van Rossum006bf911996-06-12 04:04:55 +00003730
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003731 if (!PyArg_ParseTuple(args, "i:htons", &x1)) {
3732 return NULL;
3733 }
3734 if (x1 < 0) {
3735 PyErr_SetString(PyExc_OverflowError,
3736 "can't convert negative number to unsigned long");
3737 return NULL;
3738 }
3739 x2 = (unsigned int)htons((unsigned short)x1);
3740 return PyLong_FromLong(x2);
Guido van Rossum006bf911996-06-12 04:04:55 +00003741}
3742
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003743PyDoc_STRVAR(htons_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003744"htons(integer) -> integer\n\
3745\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003746Convert a 16-bit integer from host to network byte order.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003747
3748
Guido van Rossum006bf911996-06-12 04:04:55 +00003749static PyObject *
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003750socket_htonl(PyObject *self, PyObject *arg)
Guido van Rossum006bf911996-06-12 04:04:55 +00003751{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003752 unsigned long x;
Guido van Rossum006bf911996-06-12 04:04:55 +00003753
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003754 if (PyLong_Check(arg)) {
3755 x = PyLong_AsUnsignedLong(arg);
3756 if (x == (unsigned long) -1 && PyErr_Occurred())
3757 return NULL;
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003758#if SIZEOF_LONG > 4
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003759 {
3760 unsigned long y;
3761 /* only want the trailing 32 bits */
3762 y = x & 0xFFFFFFFFUL;
3763 if (y ^ x)
3764 return PyErr_Format(PyExc_OverflowError,
3765 "long int larger than 32 bits");
3766 x = y;
3767 }
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003768#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003769 }
3770 else
3771 return PyErr_Format(PyExc_TypeError,
3772 "expected int/long, %s found",
3773 Py_TYPE(arg)->tp_name);
3774 return PyLong_FromUnsignedLong(htonl((unsigned long)x));
Guido van Rossum006bf911996-06-12 04:04:55 +00003775}
3776
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003777PyDoc_STRVAR(htonl_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003778"htonl(integer) -> integer\n\
3779\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003780Convert a 32-bit integer from host to network byte order.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003781
Guido van Rossum3eede5a2002-06-07 02:08:35 +00003782/* socket.inet_aton() and socket.inet_ntoa() functions. */
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003783
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003784PyDoc_STRVAR(inet_aton_doc,
Guido van Rossum7d0a8262007-05-21 23:13:11 +00003785"inet_aton(string) -> bytes giving packed 32-bit IP representation\n\
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003786\n\
Guido van Rossumc6a164b1999-08-20 19:11:27 +00003787Convert 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 +00003788binary format used in low-level network functions.");
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003789
3790static PyObject*
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003791socket_inet_aton(PyObject *self, PyObject *args)
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003792{
Guido van Rossuma2e48551999-09-09 15:42:59 +00003793#ifndef INADDR_NONE
3794#define INADDR_NONE (-1)
3795#endif
Neal Norwitz88f115b2003-02-13 02:15:42 +00003796#ifdef HAVE_INET_ATON
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003797 struct in_addr buf;
Thomas Wouters477c8d52006-05-27 19:21:47 +00003798#endif
3799
3800#if !defined(HAVE_INET_ATON) || defined(USE_INET_ATON_WEAKLINK)
Benjamin Petersonf91df042009-02-13 02:50:59 +00003801#if (SIZEOF_INT != 4)
3802#error "Not sure if in_addr_t exists and int is not 32-bits."
3803#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003804 /* Have to use inet_addr() instead */
3805 unsigned int packed_addr;
Tim Peters1df9fdd2003-02-13 03:13:40 +00003806#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003807 char *ip_addr;
Guido van Rossumc6a164b1999-08-20 19:11:27 +00003808
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003809 if (!PyArg_ParseTuple(args, "s:inet_aton", &ip_addr))
3810 return NULL;
Guido van Rossumad05cdf2003-02-12 23:08:22 +00003811
Tim Peters1df9fdd2003-02-13 03:13:40 +00003812
3813#ifdef HAVE_INET_ATON
Thomas Wouters477c8d52006-05-27 19:21:47 +00003814
3815#ifdef USE_INET_ATON_WEAKLINK
3816 if (inet_aton != NULL) {
3817#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003818 if (inet_aton(ip_addr, &buf))
3819 return PyBytes_FromStringAndSize((char *)(&buf),
3820 sizeof(buf));
Guido van Rossumad05cdf2003-02-12 23:08:22 +00003821
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003822 PyErr_SetString(socket_error,
3823 "illegal IP address string passed to inet_aton");
3824 return NULL;
Guido van Rossumad05cdf2003-02-12 23:08:22 +00003825
Thomas Wouters477c8d52006-05-27 19:21:47 +00003826#ifdef USE_INET_ATON_WEAKLINK
3827 } else {
3828#endif
3829
3830#endif
3831
3832#if !defined(HAVE_INET_ATON) || defined(USE_INET_ATON_WEAKLINK)
3833
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003834 /* special-case this address as inet_addr might return INADDR_NONE
3835 * for this */
3836 if (strcmp(ip_addr, "255.255.255.255") == 0) {
3837 packed_addr = 0xFFFFFFFF;
3838 } else {
Thomas Wouters477c8d52006-05-27 19:21:47 +00003839
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003840 packed_addr = inet_addr(ip_addr);
Guido van Rossumc6a164b1999-08-20 19:11:27 +00003841
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003842 if (packed_addr == INADDR_NONE) { /* invalid address */
3843 PyErr_SetString(socket_error,
3844 "illegal IP address string passed to inet_aton");
3845 return NULL;
3846 }
3847 }
3848 return PyBytes_FromStringAndSize((char *) &packed_addr,
3849 sizeof(packed_addr));
Thomas Wouters477c8d52006-05-27 19:21:47 +00003850
3851#ifdef USE_INET_ATON_WEAKLINK
3852 }
3853#endif
3854
Guido van Rossumad05cdf2003-02-12 23:08:22 +00003855#endif
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003856}
3857
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003858PyDoc_STRVAR(inet_ntoa_doc,
Fred Drakee0661342000-03-07 14:05:16 +00003859"inet_ntoa(packed_ip) -> ip_address_string\n\
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003860\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003861Convert an IP address from 32-bit packed binary format to string format");
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003862
3863static PyObject*
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003864socket_inet_ntoa(PyObject *self, PyObject *args)
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003865{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003866 char *packed_str;
3867 int addr_len;
3868 struct in_addr packed_addr;
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003869
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003870 if (!PyArg_ParseTuple(args, "y#:inet_ntoa", &packed_str, &addr_len)) {
3871 return NULL;
3872 }
Guido van Rossum48a680c2001-03-02 06:34:14 +00003873
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003874 if (addr_len != sizeof(packed_addr)) {
3875 PyErr_SetString(socket_error,
3876 "packed IP wrong length for inet_ntoa");
3877 return NULL;
3878 }
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003879
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003880 memcpy(&packed_addr, packed_str, addr_len);
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003881
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003882 return PyUnicode_FromString(inet_ntoa(packed_addr));
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003883}
Guido van Rossum82a5c661998-07-07 20:45:43 +00003884
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003885#ifdef HAVE_INET_PTON
3886
3887PyDoc_STRVAR(inet_pton_doc,
3888"inet_pton(af, ip) -> packed IP address string\n\
3889\n\
3890Convert an IP address from string format to a packed string suitable\n\
3891for use with low-level network functions.");
3892
3893static PyObject *
3894socket_inet_pton(PyObject *self, PyObject *args)
3895{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003896 int af;
3897 char* ip;
3898 int retval;
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003899#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003900 char packed[MAX(sizeof(struct in_addr), sizeof(struct in6_addr))];
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003901#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003902 char packed[sizeof(struct in_addr)];
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003903#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003904 if (!PyArg_ParseTuple(args, "is:inet_pton", &af, &ip)) {
3905 return NULL;
3906 }
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003907
Martin v. Löwis04697e82004-06-02 12:35:29 +00003908#if !defined(ENABLE_IPV6) && defined(AF_INET6)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003909 if(af == AF_INET6) {
3910 PyErr_SetString(socket_error,
3911 "can't use AF_INET6, IPv6 is disabled");
3912 return NULL;
3913 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00003914#endif
Martin v. Löwis10649092003-08-05 06:25:06 +00003915
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003916 retval = inet_pton(af, ip, packed);
3917 if (retval < 0) {
3918 PyErr_SetFromErrno(socket_error);
3919 return NULL;
3920 } else if (retval == 0) {
3921 PyErr_SetString(socket_error,
3922 "illegal IP address string passed to inet_pton");
3923 return NULL;
3924 } else if (af == AF_INET) {
3925 return PyBytes_FromStringAndSize(packed,
3926 sizeof(struct in_addr));
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003927#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003928 } else if (af == AF_INET6) {
3929 return PyBytes_FromStringAndSize(packed,
3930 sizeof(struct in6_addr));
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003931#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003932 } else {
3933 PyErr_SetString(socket_error, "unknown address family");
3934 return NULL;
3935 }
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003936}
Thomas Wouters477c8d52006-05-27 19:21:47 +00003937
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003938PyDoc_STRVAR(inet_ntop_doc,
3939"inet_ntop(af, packed_ip) -> string formatted IP address\n\
3940\n\
3941Convert a packed IP address of the given family to string format.");
3942
3943static PyObject *
3944socket_inet_ntop(PyObject *self, PyObject *args)
3945{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003946 int af;
3947 char* packed;
3948 int len;
3949 const char* retval;
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003950#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003951 char ip[MAX(INET_ADDRSTRLEN, INET6_ADDRSTRLEN) + 1];
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003952#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003953 char ip[INET_ADDRSTRLEN + 1];
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003954#endif
Thomas Wouters477c8d52006-05-27 19:21:47 +00003955
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003956 /* Guarantee NUL-termination for PyUnicode_FromString() below */
3957 memset((void *) &ip[0], '\0', sizeof(ip));
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003958
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003959 if (!PyArg_ParseTuple(args, "iy#:inet_ntop", &af, &packed, &len)) {
3960 return NULL;
3961 }
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003962
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003963 if (af == AF_INET) {
3964 if (len != sizeof(struct in_addr)) {
3965 PyErr_SetString(PyExc_ValueError,
3966 "invalid length of packed IP address string");
3967 return NULL;
3968 }
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003969#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003970 } else if (af == AF_INET6) {
3971 if (len != sizeof(struct in6_addr)) {
3972 PyErr_SetString(PyExc_ValueError,
3973 "invalid length of packed IP address string");
3974 return NULL;
3975 }
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003976#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003977 } else {
3978 PyErr_Format(PyExc_ValueError,
3979 "unknown address family %d", af);
3980 return NULL;
3981 }
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003982
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003983 retval = inet_ntop(af, packed, ip, sizeof(ip));
3984 if (!retval) {
3985 PyErr_SetFromErrno(socket_error);
3986 return NULL;
3987 } else {
3988 return PyUnicode_FromString(retval);
3989 }
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003990
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003991 /* NOTREACHED */
3992 PyErr_SetString(PyExc_RuntimeError, "invalid handling of inet_ntop");
3993 return NULL;
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003994}
3995
3996#endif /* HAVE_INET_PTON */
3997
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003998/* Python interface to getaddrinfo(host, port). */
3999
4000/*ARGSUSED*/
4001static PyObject *
Giampaolo Rodolàccfb91c2010-08-17 15:30:23 +00004002socket_getaddrinfo(PyObject *self, PyObject *args, PyObject* kwargs)
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004003{
Giampaolo Rodolàccfb91c2010-08-17 15:30:23 +00004004 static char* kwnames[] = {"host", "port", "family", "type", "proto",
4005 "flags", 0};
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004006 struct addrinfo hints, *res;
4007 struct addrinfo *res0 = NULL;
4008 PyObject *hobj = NULL;
4009 PyObject *pobj = (PyObject *)NULL;
4010 char pbuf[30];
4011 char *hptr, *pptr;
4012 int family, socktype, protocol, flags;
4013 int error;
4014 PyObject *all = (PyObject *)NULL;
4015 PyObject *idna = NULL;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004016
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004017 family = socktype = protocol = flags = 0;
4018 family = AF_UNSPEC;
Giampaolo Rodolàccfb91c2010-08-17 15:30:23 +00004019 if (!PyArg_ParseTupleAndKeywords(args, kwargs, "OO|iiii:getaddrinfo",
4020 kwnames, &hobj, &pobj, &family, &socktype,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004021 &protocol, &flags)) {
4022 return NULL;
4023 }
4024 if (hobj == Py_None) {
4025 hptr = NULL;
4026 } else if (PyUnicode_Check(hobj)) {
4027 idna = PyObject_CallMethod(hobj, "encode", "s", "idna");
4028 if (!idna)
4029 return NULL;
4030 assert(PyBytes_Check(idna));
4031 hptr = PyBytes_AS_STRING(idna);
4032 } else if (PyBytes_Check(hobj)) {
4033 hptr = PyBytes_AsString(hobj);
4034 } else {
4035 PyErr_SetString(PyExc_TypeError,
4036 "getaddrinfo() argument 1 must be string or None");
4037 return NULL;
4038 }
4039 if (PyLong_CheckExact(pobj)) {
4040 long value = PyLong_AsLong(pobj);
4041 if (value == -1 && PyErr_Occurred())
4042 goto err;
4043 PyOS_snprintf(pbuf, sizeof(pbuf), "%ld", value);
4044 pptr = pbuf;
4045 } else if (PyUnicode_Check(pobj)) {
4046 pptr = _PyUnicode_AsString(pobj);
Alexander Belopolskye239d232010-12-08 23:31:48 +00004047 if (pptr == NULL)
4048 goto err;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004049 } else if (PyBytes_Check(pobj)) {
Alexander Belopolskye239d232010-12-08 23:31:48 +00004050 pptr = PyBytes_AS_STRING(pobj);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004051 } else if (pobj == Py_None) {
4052 pptr = (char *)NULL;
4053 } else {
4054 PyErr_SetString(socket_error, "Int or String expected");
4055 goto err;
4056 }
4057 memset(&hints, 0, sizeof(hints));
4058 hints.ai_family = family;
4059 hints.ai_socktype = socktype;
4060 hints.ai_protocol = protocol;
4061 hints.ai_flags = flags;
4062 Py_BEGIN_ALLOW_THREADS
4063 ACQUIRE_GETADDRINFO_LOCK
4064 error = getaddrinfo(hptr, pptr, &hints, &res0);
4065 Py_END_ALLOW_THREADS
4066 RELEASE_GETADDRINFO_LOCK /* see comment in setipaddr() */
4067 if (error) {
4068 set_gaierror(error);
4069 goto err;
4070 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004071
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004072 if ((all = PyList_New(0)) == NULL)
4073 goto err;
4074 for (res = res0; res; res = res->ai_next) {
4075 PyObject *single;
4076 PyObject *addr =
4077 makesockaddr(-1, res->ai_addr, res->ai_addrlen, protocol);
4078 if (addr == NULL)
4079 goto err;
4080 single = Py_BuildValue("iiisO", res->ai_family,
4081 res->ai_socktype, res->ai_protocol,
4082 res->ai_canonname ? res->ai_canonname : "",
4083 addr);
4084 Py_DECREF(addr);
4085 if (single == NULL)
4086 goto err;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004087
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004088 if (PyList_Append(all, single))
4089 goto err;
4090 Py_XDECREF(single);
4091 }
4092 Py_XDECREF(idna);
4093 if (res0)
4094 freeaddrinfo(res0);
4095 return all;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004096 err:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004097 Py_XDECREF(all);
4098 Py_XDECREF(idna);
4099 if (res0)
4100 freeaddrinfo(res0);
4101 return (PyObject *)NULL;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004102}
4103
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004104PyDoc_STRVAR(getaddrinfo_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00004105"getaddrinfo(host, port [, family, socktype, proto, flags])\n\
4106 -> list of (family, socktype, proto, canonname, sockaddr)\n\
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004107\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004108Resolve host and port into addrinfo struct.");
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004109
4110/* Python interface to getnameinfo(sa, flags). */
4111
4112/*ARGSUSED*/
4113static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004114socket_getnameinfo(PyObject *self, PyObject *args)
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004115{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004116 PyObject *sa = (PyObject *)NULL;
4117 int flags;
4118 char *hostp;
Charles-François Natali366999a2012-01-02 15:47:29 +01004119 int port;
4120 unsigned int flowinfo, scope_id;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004121 char hbuf[NI_MAXHOST], pbuf[NI_MAXSERV];
4122 struct addrinfo hints, *res = NULL;
4123 int error;
4124 PyObject *ret = (PyObject *)NULL;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004125
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004126 flags = flowinfo = scope_id = 0;
4127 if (!PyArg_ParseTuple(args, "Oi:getnameinfo", &sa, &flags))
4128 return NULL;
4129 if (!PyTuple_Check(sa)) {
4130 PyErr_SetString(PyExc_TypeError,
4131 "getnameinfo() argument 1 must be a tuple");
4132 return NULL;
4133 }
Charles-François Natali366999a2012-01-02 15:47:29 +01004134 if (!PyArg_ParseTuple(sa, "si|II",
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004135 &hostp, &port, &flowinfo, &scope_id))
4136 return NULL;
Charles-François Natali9b0c0062012-06-23 10:17:05 +02004137 if (flowinfo > 0xfffff) {
Charles-François Natali366999a2012-01-02 15:47:29 +01004138 PyErr_SetString(PyExc_OverflowError,
4139 "getsockaddrarg: flowinfo must be 0-1048575.");
4140 return NULL;
4141 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004142 PyOS_snprintf(pbuf, sizeof(pbuf), "%d", port);
4143 memset(&hints, 0, sizeof(hints));
4144 hints.ai_family = AF_UNSPEC;
4145 hints.ai_socktype = SOCK_DGRAM; /* make numeric port happy */
Martin v. Löwis112c0f32010-08-25 07:38:15 +00004146 hints.ai_flags = AI_NUMERICHOST; /* don't do any name resolution */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004147 Py_BEGIN_ALLOW_THREADS
4148 ACQUIRE_GETADDRINFO_LOCK
4149 error = getaddrinfo(hostp, pbuf, &hints, &res);
4150 Py_END_ALLOW_THREADS
4151 RELEASE_GETADDRINFO_LOCK /* see comment in setipaddr() */
4152 if (error) {
4153 set_gaierror(error);
4154 goto fail;
4155 }
4156 if (res->ai_next) {
4157 PyErr_SetString(socket_error,
4158 "sockaddr resolved to multiple addresses");
4159 goto fail;
4160 }
4161 switch (res->ai_family) {
4162 case AF_INET:
4163 {
4164 if (PyTuple_GET_SIZE(sa) != 2) {
4165 PyErr_SetString(socket_error,
4166 "IPv4 sockaddr must be 2 tuple");
4167 goto fail;
4168 }
4169 break;
4170 }
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00004171#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004172 case AF_INET6:
4173 {
4174 struct sockaddr_in6 *sin6;
4175 sin6 = (struct sockaddr_in6 *)res->ai_addr;
Charles-François Natali366999a2012-01-02 15:47:29 +01004176 sin6->sin6_flowinfo = htonl(flowinfo);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004177 sin6->sin6_scope_id = scope_id;
4178 break;
4179 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004180#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004181 }
Antoine Pitrou19467d22010-08-17 19:33:30 +00004182 error = getnameinfo(res->ai_addr, (socklen_t) res->ai_addrlen,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004183 hbuf, sizeof(hbuf), pbuf, sizeof(pbuf), flags);
4184 if (error) {
4185 set_gaierror(error);
4186 goto fail;
4187 }
4188 ret = Py_BuildValue("ss", hbuf, pbuf);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004189
4190fail:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004191 if (res)
4192 freeaddrinfo(res);
4193 return ret;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004194}
4195
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004196PyDoc_STRVAR(getnameinfo_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00004197"getnameinfo(sockaddr, flags) --> (host, port)\n\
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004198\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004199Get host and port for a sockaddr.");
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004200
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004201
4202/* Python API to getting and setting the default timeout value. */
4203
4204static PyObject *
4205socket_getdefaulttimeout(PyObject *self)
4206{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004207 if (defaulttimeout < 0.0) {
4208 Py_INCREF(Py_None);
4209 return Py_None;
4210 }
4211 else
4212 return PyFloat_FromDouble(defaulttimeout);
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004213}
4214
4215PyDoc_STRVAR(getdefaulttimeout_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00004216"getdefaulttimeout() -> timeout\n\
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004217\n\
Ezio Melotti388c9452011-08-14 08:28:57 +03004218Returns the default timeout in seconds (float) for new socket objects.\n\
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004219A value of None indicates that new socket objects have no timeout.\n\
4220When the socket module is first imported, the default is None.");
4221
4222static PyObject *
4223socket_setdefaulttimeout(PyObject *self, PyObject *arg)
4224{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004225 double timeout;
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004226
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004227 if (arg == Py_None)
4228 timeout = -1.0;
4229 else {
4230 timeout = PyFloat_AsDouble(arg);
4231 if (timeout < 0.0) {
4232 if (!PyErr_Occurred())
4233 PyErr_SetString(PyExc_ValueError,
4234 "Timeout value out of range");
4235 return NULL;
4236 }
4237 }
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004238
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004239 defaulttimeout = timeout;
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004240
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004241 Py_INCREF(Py_None);
4242 return Py_None;
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004243}
4244
4245PyDoc_STRVAR(setdefaulttimeout_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00004246"setdefaulttimeout(timeout)\n\
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004247\n\
Ezio Melotti388c9452011-08-14 08:28:57 +03004248Set the default timeout in seconds (float) for new socket objects.\n\
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004249A value of None indicates that new socket objects have no timeout.\n\
4250When the socket module is first imported, the default is None.");
4251
4252
Guido van Rossum30a685f1991-06-27 15:51:29 +00004253/* List of functions exported by this module. */
4254
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004255static PyMethodDef socket_methods[] = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004256 {"gethostbyname", socket_gethostbyname,
4257 METH_VARARGS, gethostbyname_doc},
4258 {"gethostbyname_ex", socket_gethostbyname_ex,
4259 METH_VARARGS, ghbn_ex_doc},
4260 {"gethostbyaddr", socket_gethostbyaddr,
4261 METH_VARARGS, gethostbyaddr_doc},
4262 {"gethostname", socket_gethostname,
4263 METH_NOARGS, gethostname_doc},
4264 {"getservbyname", socket_getservbyname,
4265 METH_VARARGS, getservbyname_doc},
4266 {"getservbyport", socket_getservbyport,
4267 METH_VARARGS, getservbyport_doc},
4268 {"getprotobyname", socket_getprotobyname,
4269 METH_VARARGS, getprotobyname_doc},
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00004270#ifndef NO_DUP
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004271 {"dup", socket_dup,
4272 METH_O, dup_doc},
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00004273#endif
Dave Cole331708b2004-08-09 04:51:41 +00004274#ifdef HAVE_SOCKETPAIR
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004275 {"socketpair", socket_socketpair,
4276 METH_VARARGS, socketpair_doc},
Dave Cole331708b2004-08-09 04:51:41 +00004277#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004278 {"ntohs", socket_ntohs,
4279 METH_VARARGS, ntohs_doc},
4280 {"ntohl", socket_ntohl,
4281 METH_O, ntohl_doc},
4282 {"htons", socket_htons,
4283 METH_VARARGS, htons_doc},
4284 {"htonl", socket_htonl,
4285 METH_O, htonl_doc},
4286 {"inet_aton", socket_inet_aton,
4287 METH_VARARGS, inet_aton_doc},
4288 {"inet_ntoa", socket_inet_ntoa,
4289 METH_VARARGS, inet_ntoa_doc},
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00004290#ifdef HAVE_INET_PTON
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004291 {"inet_pton", socket_inet_pton,
4292 METH_VARARGS, inet_pton_doc},
4293 {"inet_ntop", socket_inet_ntop,
4294 METH_VARARGS, inet_ntop_doc},
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00004295#endif
Giampaolo Rodolàccfb91c2010-08-17 15:30:23 +00004296 {"getaddrinfo", (PyCFunction)socket_getaddrinfo,
4297 METH_VARARGS | METH_KEYWORDS, getaddrinfo_doc},
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004298 {"getnameinfo", socket_getnameinfo,
4299 METH_VARARGS, getnameinfo_doc},
4300 {"getdefaulttimeout", (PyCFunction)socket_getdefaulttimeout,
4301 METH_NOARGS, getdefaulttimeout_doc},
4302 {"setdefaulttimeout", socket_setdefaulttimeout,
4303 METH_O, setdefaulttimeout_doc},
4304 {NULL, NULL} /* Sentinel */
Guido van Rossum6574b3e1991-06-25 21:36:08 +00004305};
4306
Guido van Rossum30a685f1991-06-27 15:51:29 +00004307
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004308#ifdef MS_WINDOWS
4309#define OS_INIT_DEFINED
4310
4311/* Additional initialization and cleanup for Windows */
Guido van Rossumbe32c891996-06-20 16:25:29 +00004312
4313static void
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004314os_cleanup(void)
Guido van Rossumbe32c891996-06-20 16:25:29 +00004315{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004316 WSACleanup();
Guido van Rossumbe32c891996-06-20 16:25:29 +00004317}
4318
4319static int
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004320os_init(void)
Guido van Rossumbe32c891996-06-20 16:25:29 +00004321{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004322 WSADATA WSAData;
4323 int ret;
4324 ret = WSAStartup(0x0101, &WSAData);
4325 switch (ret) {
4326 case 0: /* No error */
4327 Py_AtExit(os_cleanup);
4328 return 1; /* Success */
4329 case WSASYSNOTREADY:
4330 PyErr_SetString(PyExc_ImportError,
4331 "WSAStartup failed: network not ready");
4332 break;
4333 case WSAVERNOTSUPPORTED:
4334 case WSAEINVAL:
4335 PyErr_SetString(
4336 PyExc_ImportError,
4337 "WSAStartup failed: requested version not supported");
4338 break;
4339 default:
4340 PyErr_Format(PyExc_ImportError, "WSAStartup failed: error code %d", ret);
4341 break;
4342 }
4343 return 0; /* Failure */
Guido van Rossumbe32c891996-06-20 16:25:29 +00004344}
4345
Guido van Rossum8d665e61996-06-26 18:22:49 +00004346#endif /* MS_WINDOWS */
Guido van Rossumbe32c891996-06-20 16:25:29 +00004347
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004348
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004349#ifdef PYOS_OS2
4350#define OS_INIT_DEFINED
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004351
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004352/* Additional initialization for OS/2 */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004353
4354static int
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004355os_init(void)
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004356{
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004357#ifndef PYCC_GCC
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004358 int rc = sock_init();
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004359
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004360 if (rc == 0) {
4361 return 1; /* Success */
4362 }
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004363
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004364 PyErr_Format(PyExc_ImportError, "OS/2 TCP/IP Error# %d", sock_errno());
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004365
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004366 return 0; /* Failure */
Andrew MacIntyreba43e872002-03-03 03:03:52 +00004367#else
Ezio Melotti13925002011-03-16 11:05:33 +02004368 /* No need to initialize sockets with GCC/EMX */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004369 return 1; /* Success */
Andrew MacIntyreba43e872002-03-03 03:03:52 +00004370#endif
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004371}
4372
4373#endif /* PYOS_OS2 */
4374
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004375
4376#ifndef OS_INIT_DEFINED
4377static int
4378os_init(void)
4379{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004380 return 1; /* Success */
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004381}
4382#endif
4383
4384
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00004385/* C API table - always add new things to the end for binary
4386 compatibility. */
4387static
4388PySocketModule_APIObject PySocketModuleAPI =
4389{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004390 &sock_type,
Antoine Pitrouc4df7842010-12-03 19:59:41 +00004391 NULL,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004392 NULL
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00004393};
4394
Guido van Rossum3eede5a2002-06-07 02:08:35 +00004395
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004396/* Initialize the _socket module.
Guido van Rossum3eede5a2002-06-07 02:08:35 +00004397
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004398 This module is actually called "_socket", and there's a wrapper
Guido van Rossum7d0a8262007-05-21 23:13:11 +00004399 "socket.py" which implements some additional functionality.
4400 The import of "_socket" may fail with an ImportError exception if
4401 os-specific initialization fails. On Windows, this does WINSOCK
Ezio Melotti13925002011-03-16 11:05:33 +02004402 initialization. When WINSOCK is initialized successfully, a call to
Guido van Rossum7d0a8262007-05-21 23:13:11 +00004403 WSACleanup() is scheduled to be made at exit time.
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004404*/
Guido van Rossum3eede5a2002-06-07 02:08:35 +00004405
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004406PyDoc_STRVAR(socket_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00004407"Implementation module for socket operations.\n\
4408\n\
4409See the socket module for documentation.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00004410
Martin v. Löwis1a214512008-06-11 05:26:20 +00004411static struct PyModuleDef socketmodule = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004412 PyModuleDef_HEAD_INIT,
4413 PySocket_MODULE_NAME,
4414 socket_doc,
4415 -1,
4416 socket_methods,
4417 NULL,
4418 NULL,
4419 NULL,
4420 NULL
Martin v. Löwis1a214512008-06-11 05:26:20 +00004421};
4422
Mark Hammondfe51c6d2002-08-02 02:27:13 +00004423PyMODINIT_FUNC
Martin v. Löwis1a214512008-06-11 05:26:20 +00004424PyInit__socket(void)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00004425{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004426 PyObject *m, *has_ipv6;
Fred Drake4baedc12002-04-01 14:53:37 +00004427
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004428 if (!os_init())
4429 return NULL;
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004430
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004431 Py_TYPE(&sock_type) = &PyType_Type;
4432 m = PyModule_Create(&socketmodule);
4433 if (m == NULL)
4434 return NULL;
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004435
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004436 socket_error = PyErr_NewException("socket.error",
4437 PyExc_IOError, NULL);
4438 if (socket_error == NULL)
4439 return NULL;
4440 PySocketModuleAPI.error = socket_error;
4441 Py_INCREF(socket_error);
4442 PyModule_AddObject(m, "error", socket_error);
4443 socket_herror = PyErr_NewException("socket.herror",
4444 socket_error, NULL);
4445 if (socket_herror == NULL)
4446 return NULL;
4447 Py_INCREF(socket_herror);
4448 PyModule_AddObject(m, "herror", socket_herror);
4449 socket_gaierror = PyErr_NewException("socket.gaierror", socket_error,
4450 NULL);
4451 if (socket_gaierror == NULL)
4452 return NULL;
4453 Py_INCREF(socket_gaierror);
4454 PyModule_AddObject(m, "gaierror", socket_gaierror);
4455 socket_timeout = PyErr_NewException("socket.timeout",
4456 socket_error, NULL);
4457 if (socket_timeout == NULL)
4458 return NULL;
Antoine Pitrouc4df7842010-12-03 19:59:41 +00004459 PySocketModuleAPI.timeout_error = socket_timeout;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004460 Py_INCREF(socket_timeout);
4461 PyModule_AddObject(m, "timeout", socket_timeout);
4462 Py_INCREF((PyObject *)&sock_type);
4463 if (PyModule_AddObject(m, "SocketType",
4464 (PyObject *)&sock_type) != 0)
4465 return NULL;
4466 Py_INCREF((PyObject *)&sock_type);
4467 if (PyModule_AddObject(m, "socket",
4468 (PyObject *)&sock_type) != 0)
4469 return NULL;
Guido van Rossum09be4091999-08-09 14:40:40 +00004470
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00004471#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004472 has_ipv6 = Py_True;
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00004473#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004474 has_ipv6 = Py_False;
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00004475#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004476 Py_INCREF(has_ipv6);
4477 PyModule_AddObject(m, "has_ipv6", has_ipv6);
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00004478
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004479 /* Export C API */
4480 if (PyModule_AddObject(m, PySocket_CAPI_NAME,
4481 PyCapsule_New(&PySocketModuleAPI, PySocket_CAPSULE_NAME, NULL)
4482 ) != 0)
4483 return NULL;
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00004484
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004485 /* Address families (we only support AF_INET and AF_UNIX) */
Guido van Rossum09be4091999-08-09 14:40:40 +00004486#ifdef AF_UNSPEC
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004487 PyModule_AddIntConstant(m, "AF_UNSPEC", AF_UNSPEC);
Guido van Rossum09be4091999-08-09 14:40:40 +00004488#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004489 PyModule_AddIntConstant(m, "AF_INET", AF_INET);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004490#ifdef AF_INET6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004491 PyModule_AddIntConstant(m, "AF_INET6", AF_INET6);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004492#endif /* AF_INET6 */
Andrew MacIntyred12dfbb2004-04-04 07:13:49 +00004493#if defined(AF_UNIX)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004494 PyModule_AddIntConstant(m, "AF_UNIX", AF_UNIX);
Guido van Rossumb6775db1994-08-01 11:34:53 +00004495#endif /* AF_UNIX */
Guido van Rossum09be4091999-08-09 14:40:40 +00004496#ifdef AF_AX25
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004497 /* Amateur Radio AX.25 */
4498 PyModule_AddIntConstant(m, "AF_AX25", AF_AX25);
Guido van Rossum09be4091999-08-09 14:40:40 +00004499#endif
4500#ifdef AF_IPX
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004501 PyModule_AddIntConstant(m, "AF_IPX", AF_IPX); /* Novell IPX */
Guido van Rossum09be4091999-08-09 14:40:40 +00004502#endif
4503#ifdef AF_APPLETALK
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004504 /* Appletalk DDP */
4505 PyModule_AddIntConstant(m, "AF_APPLETALK", AF_APPLETALK);
Guido van Rossum09be4091999-08-09 14:40:40 +00004506#endif
4507#ifdef AF_NETROM
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004508 /* Amateur radio NetROM */
4509 PyModule_AddIntConstant(m, "AF_NETROM", AF_NETROM);
Guido van Rossum09be4091999-08-09 14:40:40 +00004510#endif
4511#ifdef AF_BRIDGE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004512 /* Multiprotocol bridge */
4513 PyModule_AddIntConstant(m, "AF_BRIDGE", AF_BRIDGE);
Guido van Rossum09be4091999-08-09 14:40:40 +00004514#endif
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004515#ifdef AF_ATMPVC
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004516 /* ATM PVCs */
4517 PyModule_AddIntConstant(m, "AF_ATMPVC", AF_ATMPVC);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004518#endif
Guido van Rossum09be4091999-08-09 14:40:40 +00004519#ifdef AF_AAL5
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004520 /* Reserved for Werner's ATM */
4521 PyModule_AddIntConstant(m, "AF_AAL5", AF_AAL5);
Guido van Rossum09be4091999-08-09 14:40:40 +00004522#endif
4523#ifdef AF_X25
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004524 /* Reserved for X.25 project */
4525 PyModule_AddIntConstant(m, "AF_X25", AF_X25);
Guido van Rossum09be4091999-08-09 14:40:40 +00004526#endif
4527#ifdef AF_INET6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004528 PyModule_AddIntConstant(m, "AF_INET6", AF_INET6); /* IP version 6 */
Guido van Rossum09be4091999-08-09 14:40:40 +00004529#endif
4530#ifdef AF_ROSE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004531 /* Amateur Radio X.25 PLP */
4532 PyModule_AddIntConstant(m, "AF_ROSE", AF_ROSE);
Guido van Rossum09be4091999-08-09 14:40:40 +00004533#endif
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004534#ifdef AF_DECnet
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004535 /* Reserved for DECnet project */
4536 PyModule_AddIntConstant(m, "AF_DECnet", AF_DECnet);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004537#endif
4538#ifdef AF_NETBEUI
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004539 /* Reserved for 802.2LLC project */
4540 PyModule_AddIntConstant(m, "AF_NETBEUI", AF_NETBEUI);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004541#endif
4542#ifdef AF_SECURITY
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004543 /* Security callback pseudo AF */
4544 PyModule_AddIntConstant(m, "AF_SECURITY", AF_SECURITY);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004545#endif
4546#ifdef AF_KEY
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004547 /* PF_KEY key management API */
4548 PyModule_AddIntConstant(m, "AF_KEY", AF_KEY);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004549#endif
4550#ifdef AF_NETLINK
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004551 /* */
4552 PyModule_AddIntConstant(m, "AF_NETLINK", AF_NETLINK);
4553 PyModule_AddIntConstant(m, "NETLINK_ROUTE", NETLINK_ROUTE);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004554#ifdef NETLINK_SKIP
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004555 PyModule_AddIntConstant(m, "NETLINK_SKIP", NETLINK_SKIP);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004556#endif
4557#ifdef NETLINK_W1
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004558 PyModule_AddIntConstant(m, "NETLINK_W1", NETLINK_W1);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004559#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004560 PyModule_AddIntConstant(m, "NETLINK_USERSOCK", NETLINK_USERSOCK);
4561 PyModule_AddIntConstant(m, "NETLINK_FIREWALL", NETLINK_FIREWALL);
Guido van Rossum668a94a2006-02-21 01:07:27 +00004562#ifdef NETLINK_TCPDIAG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004563 PyModule_AddIntConstant(m, "NETLINK_TCPDIAG", NETLINK_TCPDIAG);
Guido van Rossum668a94a2006-02-21 01:07:27 +00004564#endif
4565#ifdef NETLINK_NFLOG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004566 PyModule_AddIntConstant(m, "NETLINK_NFLOG", NETLINK_NFLOG);
Guido van Rossum668a94a2006-02-21 01:07:27 +00004567#endif
Neal Norwitz65851662006-01-16 04:31:40 +00004568#ifdef NETLINK_XFRM
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004569 PyModule_AddIntConstant(m, "NETLINK_XFRM", NETLINK_XFRM);
Neal Norwitz65851662006-01-16 04:31:40 +00004570#endif
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004571#ifdef NETLINK_ARPD
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004572 PyModule_AddIntConstant(m, "NETLINK_ARPD", NETLINK_ARPD);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004573#endif
4574#ifdef NETLINK_ROUTE6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004575 PyModule_AddIntConstant(m, "NETLINK_ROUTE6", NETLINK_ROUTE6);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004576#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004577 PyModule_AddIntConstant(m, "NETLINK_IP6_FW", NETLINK_IP6_FW);
Thomas Wouterscf297e42007-02-23 15:07:44 +00004578#ifdef NETLINK_DNRTMSG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004579 PyModule_AddIntConstant(m, "NETLINK_DNRTMSG", NETLINK_DNRTMSG);
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00004580#endif
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004581#ifdef NETLINK_TAPBASE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004582 PyModule_AddIntConstant(m, "NETLINK_TAPBASE", NETLINK_TAPBASE);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004583#endif
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004584#endif /* AF_NETLINK */
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004585#ifdef AF_ROUTE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004586 /* Alias to emulate 4.4BSD */
4587 PyModule_AddIntConstant(m, "AF_ROUTE", AF_ROUTE);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004588#endif
4589#ifdef AF_ASH
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004590 /* Ash */
4591 PyModule_AddIntConstant(m, "AF_ASH", AF_ASH);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004592#endif
4593#ifdef AF_ECONET
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004594 /* Acorn Econet */
4595 PyModule_AddIntConstant(m, "AF_ECONET", AF_ECONET);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004596#endif
4597#ifdef AF_ATMSVC
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004598 /* ATM SVCs */
4599 PyModule_AddIntConstant(m, "AF_ATMSVC", AF_ATMSVC);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004600#endif
4601#ifdef AF_SNA
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004602 /* Linux SNA Project (nutters!) */
4603 PyModule_AddIntConstant(m, "AF_SNA", AF_SNA);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004604#endif
4605#ifdef AF_IRDA
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004606 /* IRDA sockets */
4607 PyModule_AddIntConstant(m, "AF_IRDA", AF_IRDA);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004608#endif
4609#ifdef AF_PPPOX
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004610 /* PPPoX sockets */
4611 PyModule_AddIntConstant(m, "AF_PPPOX", AF_PPPOX);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004612#endif
4613#ifdef AF_WANPIPE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004614 /* Wanpipe API Sockets */
4615 PyModule_AddIntConstant(m, "AF_WANPIPE", AF_WANPIPE);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004616#endif
4617#ifdef AF_LLC
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004618 /* Linux LLC */
4619 PyModule_AddIntConstant(m, "AF_LLC", AF_LLC);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004620#endif
Martin v. Löwis12af0482004-01-31 12:34:17 +00004621
Hye-Shik Chang81268602004-02-02 06:05:24 +00004622#ifdef USE_BLUETOOTH
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004623 PyModule_AddIntConstant(m, "AF_BLUETOOTH", AF_BLUETOOTH);
4624 PyModule_AddIntConstant(m, "BTPROTO_L2CAP", BTPROTO_L2CAP);
4625 PyModule_AddIntConstant(m, "BTPROTO_HCI", BTPROTO_HCI);
4626 PyModule_AddIntConstant(m, "SOL_HCI", SOL_HCI);
Gregory P. Smith3e85dfd2010-10-17 00:43:10 +00004627#if !defined(__NetBSD__) && !defined(__DragonFly__)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004628 PyModule_AddIntConstant(m, "HCI_FILTER", HCI_FILTER);
Gregory P. Smith3e85dfd2010-10-17 00:43:10 +00004629#endif
Hye-Shik Chang81268602004-02-02 06:05:24 +00004630#if !defined(__FreeBSD__)
Gregory P. Smith3e85dfd2010-10-17 00:43:10 +00004631#if !defined(__NetBSD__) && !defined(__DragonFly__)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004632 PyModule_AddIntConstant(m, "HCI_TIME_STAMP", HCI_TIME_STAMP);
Gregory P. Smith3e85dfd2010-10-17 00:43:10 +00004633#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004634 PyModule_AddIntConstant(m, "HCI_DATA_DIR", HCI_DATA_DIR);
4635 PyModule_AddIntConstant(m, "BTPROTO_SCO", BTPROTO_SCO);
Hye-Shik Chang81268602004-02-02 06:05:24 +00004636#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004637 PyModule_AddIntConstant(m, "BTPROTO_RFCOMM", BTPROTO_RFCOMM);
4638 PyModule_AddStringConstant(m, "BDADDR_ANY", "00:00:00:00:00:00");
4639 PyModule_AddStringConstant(m, "BDADDR_LOCAL", "00:00:00:FF:FF:FF");
Martin v. Löwis12af0482004-01-31 12:34:17 +00004640#endif
4641
Antoine Pitroub156a462010-10-27 20:13:57 +00004642#ifdef AF_PACKET
4643 PyModule_AddIntMacro(m, AF_PACKET);
4644#endif
4645#ifdef PF_PACKET
4646 PyModule_AddIntMacro(m, PF_PACKET);
4647#endif
4648#ifdef PACKET_HOST
4649 PyModule_AddIntMacro(m, PACKET_HOST);
4650#endif
4651#ifdef PACKET_BROADCAST
4652 PyModule_AddIntMacro(m, PACKET_BROADCAST);
4653#endif
4654#ifdef PACKET_MULTICAST
4655 PyModule_AddIntMacro(m, PACKET_MULTICAST);
4656#endif
4657#ifdef PACKET_OTHERHOST
4658 PyModule_AddIntMacro(m, PACKET_OTHERHOST);
4659#endif
4660#ifdef PACKET_OUTGOING
4661 PyModule_AddIntMacro(m, PACKET_OUTGOING);
4662#endif
4663#ifdef PACKET_LOOPBACK
4664 PyModule_AddIntMacro(m, PACKET_LOOPBACK);
4665#endif
4666#ifdef PACKET_FASTROUTE
4667 PyModule_AddIntMacro(m, PACKET_FASTROUTE);
Guido van Rossum48a680c2001-03-02 06:34:14 +00004668#endif
Guido van Rossum09be4091999-08-09 14:40:40 +00004669
Christian Heimes043d6f62008-01-07 17:19:16 +00004670#ifdef HAVE_LINUX_TIPC_H
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004671 PyModule_AddIntConstant(m, "AF_TIPC", AF_TIPC);
Christian Heimes043d6f62008-01-07 17:19:16 +00004672
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004673 /* for addresses */
4674 PyModule_AddIntConstant(m, "TIPC_ADDR_NAMESEQ", TIPC_ADDR_NAMESEQ);
4675 PyModule_AddIntConstant(m, "TIPC_ADDR_NAME", TIPC_ADDR_NAME);
4676 PyModule_AddIntConstant(m, "TIPC_ADDR_ID", TIPC_ADDR_ID);
Christian Heimes043d6f62008-01-07 17:19:16 +00004677
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004678 PyModule_AddIntConstant(m, "TIPC_ZONE_SCOPE", TIPC_ZONE_SCOPE);
4679 PyModule_AddIntConstant(m, "TIPC_CLUSTER_SCOPE", TIPC_CLUSTER_SCOPE);
4680 PyModule_AddIntConstant(m, "TIPC_NODE_SCOPE", TIPC_NODE_SCOPE);
Christian Heimes043d6f62008-01-07 17:19:16 +00004681
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004682 /* for setsockopt() */
4683 PyModule_AddIntConstant(m, "SOL_TIPC", SOL_TIPC);
4684 PyModule_AddIntConstant(m, "TIPC_IMPORTANCE", TIPC_IMPORTANCE);
4685 PyModule_AddIntConstant(m, "TIPC_SRC_DROPPABLE", TIPC_SRC_DROPPABLE);
4686 PyModule_AddIntConstant(m, "TIPC_DEST_DROPPABLE",
4687 TIPC_DEST_DROPPABLE);
4688 PyModule_AddIntConstant(m, "TIPC_CONN_TIMEOUT", TIPC_CONN_TIMEOUT);
Christian Heimes043d6f62008-01-07 17:19:16 +00004689
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004690 PyModule_AddIntConstant(m, "TIPC_LOW_IMPORTANCE",
4691 TIPC_LOW_IMPORTANCE);
4692 PyModule_AddIntConstant(m, "TIPC_MEDIUM_IMPORTANCE",
4693 TIPC_MEDIUM_IMPORTANCE);
4694 PyModule_AddIntConstant(m, "TIPC_HIGH_IMPORTANCE",
4695 TIPC_HIGH_IMPORTANCE);
4696 PyModule_AddIntConstant(m, "TIPC_CRITICAL_IMPORTANCE",
4697 TIPC_CRITICAL_IMPORTANCE);
Christian Heimes043d6f62008-01-07 17:19:16 +00004698
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004699 /* for subscriptions */
4700 PyModule_AddIntConstant(m, "TIPC_SUB_PORTS", TIPC_SUB_PORTS);
4701 PyModule_AddIntConstant(m, "TIPC_SUB_SERVICE", TIPC_SUB_SERVICE);
Christian Heimes25bb7832008-01-11 16:17:00 +00004702#ifdef TIPC_SUB_CANCEL
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004703 /* doesn't seem to be available everywhere */
4704 PyModule_AddIntConstant(m, "TIPC_SUB_CANCEL", TIPC_SUB_CANCEL);
Christian Heimes25bb7832008-01-11 16:17:00 +00004705#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004706 PyModule_AddIntConstant(m, "TIPC_WAIT_FOREVER", TIPC_WAIT_FOREVER);
4707 PyModule_AddIntConstant(m, "TIPC_PUBLISHED", TIPC_PUBLISHED);
4708 PyModule_AddIntConstant(m, "TIPC_WITHDRAWN", TIPC_WITHDRAWN);
4709 PyModule_AddIntConstant(m, "TIPC_SUBSCR_TIMEOUT", TIPC_SUBSCR_TIMEOUT);
4710 PyModule_AddIntConstant(m, "TIPC_CFG_SRV", TIPC_CFG_SRV);
4711 PyModule_AddIntConstant(m, "TIPC_TOP_SRV", TIPC_TOP_SRV);
Christian Heimes043d6f62008-01-07 17:19:16 +00004712#endif
4713
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004714 /* Socket types */
4715 PyModule_AddIntConstant(m, "SOCK_STREAM", SOCK_STREAM);
4716 PyModule_AddIntConstant(m, "SOCK_DGRAM", SOCK_DGRAM);
Guido van Rossumbcc20741998-08-04 22:53:56 +00004717/* We have incomplete socket support. */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004718 PyModule_AddIntConstant(m, "SOCK_RAW", SOCK_RAW);
4719 PyModule_AddIntConstant(m, "SOCK_SEQPACKET", SOCK_SEQPACKET);
Martin v. Löwiscf8f47e2002-12-11 13:10:57 +00004720#if defined(SOCK_RDM)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004721 PyModule_AddIntConstant(m, "SOCK_RDM", SOCK_RDM);
Guido van Rossumbcc20741998-08-04 22:53:56 +00004722#endif
Antoine Pitroub1c54962010-10-14 15:05:38 +00004723#ifdef SOCK_CLOEXEC
4724 PyModule_AddIntConstant(m, "SOCK_CLOEXEC", SOCK_CLOEXEC);
4725#endif
4726#ifdef SOCK_NONBLOCK
4727 PyModule_AddIntConstant(m, "SOCK_NONBLOCK", SOCK_NONBLOCK);
4728#endif
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004729
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004730#ifdef SO_DEBUG
4731 PyModule_AddIntConstant(m, "SO_DEBUG", SO_DEBUG);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004732#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004733#ifdef SO_ACCEPTCONN
4734 PyModule_AddIntConstant(m, "SO_ACCEPTCONN", SO_ACCEPTCONN);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004735#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004736#ifdef SO_REUSEADDR
4737 PyModule_AddIntConstant(m, "SO_REUSEADDR", SO_REUSEADDR);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004738#endif
Andrew M. Kuchling42851ab2004-07-10 14:19:21 +00004739#ifdef SO_EXCLUSIVEADDRUSE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004740 PyModule_AddIntConstant(m, "SO_EXCLUSIVEADDRUSE", SO_EXCLUSIVEADDRUSE);
Andrew M. Kuchling42851ab2004-07-10 14:19:21 +00004741#endif
4742
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004743#ifdef SO_KEEPALIVE
4744 PyModule_AddIntConstant(m, "SO_KEEPALIVE", SO_KEEPALIVE);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004745#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004746#ifdef SO_DONTROUTE
4747 PyModule_AddIntConstant(m, "SO_DONTROUTE", SO_DONTROUTE);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004748#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004749#ifdef SO_BROADCAST
4750 PyModule_AddIntConstant(m, "SO_BROADCAST", SO_BROADCAST);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004751#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004752#ifdef SO_USELOOPBACK
4753 PyModule_AddIntConstant(m, "SO_USELOOPBACK", SO_USELOOPBACK);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004754#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004755#ifdef SO_LINGER
4756 PyModule_AddIntConstant(m, "SO_LINGER", SO_LINGER);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004757#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004758#ifdef SO_OOBINLINE
4759 PyModule_AddIntConstant(m, "SO_OOBINLINE", SO_OOBINLINE);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004760#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004761#ifdef SO_REUSEPORT
4762 PyModule_AddIntConstant(m, "SO_REUSEPORT", SO_REUSEPORT);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004763#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004764#ifdef SO_SNDBUF
4765 PyModule_AddIntConstant(m, "SO_SNDBUF", SO_SNDBUF);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004766#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004767#ifdef SO_RCVBUF
4768 PyModule_AddIntConstant(m, "SO_RCVBUF", SO_RCVBUF);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004769#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004770#ifdef SO_SNDLOWAT
4771 PyModule_AddIntConstant(m, "SO_SNDLOWAT", SO_SNDLOWAT);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004772#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004773#ifdef SO_RCVLOWAT
4774 PyModule_AddIntConstant(m, "SO_RCVLOWAT", SO_RCVLOWAT);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004775#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004776#ifdef SO_SNDTIMEO
4777 PyModule_AddIntConstant(m, "SO_SNDTIMEO", SO_SNDTIMEO);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004778#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004779#ifdef SO_RCVTIMEO
4780 PyModule_AddIntConstant(m, "SO_RCVTIMEO", SO_RCVTIMEO);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004781#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004782#ifdef SO_ERROR
4783 PyModule_AddIntConstant(m, "SO_ERROR", SO_ERROR);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004784#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004785#ifdef SO_TYPE
4786 PyModule_AddIntConstant(m, "SO_TYPE", SO_TYPE);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004787#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004788#ifdef SO_SETFIB
4789 PyModule_AddIntConstant(m, "SO_SETFIB", SO_SETFIB);
Larry Hastingsf0f37952010-04-02 11:47:10 +00004790#endif
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004791
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004792 /* Maximum number of connections for "listen" */
4793#ifdef SOMAXCONN
4794 PyModule_AddIntConstant(m, "SOMAXCONN", SOMAXCONN);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004795#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004796 PyModule_AddIntConstant(m, "SOMAXCONN", 5); /* Common value */
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004797#endif
4798
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004799 /* Flags for send, recv */
4800#ifdef MSG_OOB
4801 PyModule_AddIntConstant(m, "MSG_OOB", MSG_OOB);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004802#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004803#ifdef MSG_PEEK
4804 PyModule_AddIntConstant(m, "MSG_PEEK", MSG_PEEK);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004805#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004806#ifdef MSG_DONTROUTE
4807 PyModule_AddIntConstant(m, "MSG_DONTROUTE", MSG_DONTROUTE);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004808#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004809#ifdef MSG_DONTWAIT
4810 PyModule_AddIntConstant(m, "MSG_DONTWAIT", MSG_DONTWAIT);
Guido van Rossum2c8bcb82000-04-25 21:34:53 +00004811#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004812#ifdef MSG_EOR
4813 PyModule_AddIntConstant(m, "MSG_EOR", MSG_EOR);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004814#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004815#ifdef MSG_TRUNC
4816 PyModule_AddIntConstant(m, "MSG_TRUNC", MSG_TRUNC);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004817#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004818#ifdef MSG_CTRUNC
4819 PyModule_AddIntConstant(m, "MSG_CTRUNC", MSG_CTRUNC);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004820#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004821#ifdef MSG_WAITALL
4822 PyModule_AddIntConstant(m, "MSG_WAITALL", MSG_WAITALL);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004823#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004824#ifdef MSG_BTAG
4825 PyModule_AddIntConstant(m, "MSG_BTAG", MSG_BTAG);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004826#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004827#ifdef MSG_ETAG
4828 PyModule_AddIntConstant(m, "MSG_ETAG", MSG_ETAG);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004829#endif
4830
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004831 /* Protocol level and numbers, usable for [gs]etsockopt */
4832#ifdef SOL_SOCKET
4833 PyModule_AddIntConstant(m, "SOL_SOCKET", SOL_SOCKET);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004834#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004835#ifdef SOL_IP
4836 PyModule_AddIntConstant(m, "SOL_IP", SOL_IP);
Guido van Rossum09be4091999-08-09 14:40:40 +00004837#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004838 PyModule_AddIntConstant(m, "SOL_IP", 0);
Guido van Rossum09be4091999-08-09 14:40:40 +00004839#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004840#ifdef SOL_IPX
4841 PyModule_AddIntConstant(m, "SOL_IPX", SOL_IPX);
Guido van Rossum09be4091999-08-09 14:40:40 +00004842#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004843#ifdef SOL_AX25
4844 PyModule_AddIntConstant(m, "SOL_AX25", SOL_AX25);
Guido van Rossum09be4091999-08-09 14:40:40 +00004845#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004846#ifdef SOL_ATALK
4847 PyModule_AddIntConstant(m, "SOL_ATALK", SOL_ATALK);
Guido van Rossum09be4091999-08-09 14:40:40 +00004848#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004849#ifdef SOL_NETROM
4850 PyModule_AddIntConstant(m, "SOL_NETROM", SOL_NETROM);
Guido van Rossum09be4091999-08-09 14:40:40 +00004851#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004852#ifdef SOL_ROSE
4853 PyModule_AddIntConstant(m, "SOL_ROSE", SOL_ROSE);
Guido van Rossum09be4091999-08-09 14:40:40 +00004854#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004855#ifdef SOL_TCP
4856 PyModule_AddIntConstant(m, "SOL_TCP", SOL_TCP);
Guido van Rossum09be4091999-08-09 14:40:40 +00004857#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004858 PyModule_AddIntConstant(m, "SOL_TCP", 6);
Guido van Rossum09be4091999-08-09 14:40:40 +00004859#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004860#ifdef SOL_UDP
4861 PyModule_AddIntConstant(m, "SOL_UDP", SOL_UDP);
Guido van Rossum09be4091999-08-09 14:40:40 +00004862#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004863 PyModule_AddIntConstant(m, "SOL_UDP", 17);
Guido van Rossum09be4091999-08-09 14:40:40 +00004864#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004865#ifdef IPPROTO_IP
4866 PyModule_AddIntConstant(m, "IPPROTO_IP", IPPROTO_IP);
Guido van Rossum578de301998-05-28 20:18:18 +00004867#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004868 PyModule_AddIntConstant(m, "IPPROTO_IP", 0);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004869#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004870#ifdef IPPROTO_HOPOPTS
4871 PyModule_AddIntConstant(m, "IPPROTO_HOPOPTS", IPPROTO_HOPOPTS);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004872#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004873#ifdef IPPROTO_ICMP
4874 PyModule_AddIntConstant(m, "IPPROTO_ICMP", IPPROTO_ICMP);
Guido van Rossum578de301998-05-28 20:18:18 +00004875#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004876 PyModule_AddIntConstant(m, "IPPROTO_ICMP", 1);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004877#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004878#ifdef IPPROTO_IGMP
4879 PyModule_AddIntConstant(m, "IPPROTO_IGMP", IPPROTO_IGMP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004880#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004881#ifdef IPPROTO_GGP
4882 PyModule_AddIntConstant(m, "IPPROTO_GGP", IPPROTO_GGP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004883#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004884#ifdef IPPROTO_IPV4
4885 PyModule_AddIntConstant(m, "IPPROTO_IPV4", IPPROTO_IPV4);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004886#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004887#ifdef IPPROTO_IPV6
4888 PyModule_AddIntConstant(m, "IPPROTO_IPV6", IPPROTO_IPV6);
Martin v. Löwisa0f17342003-10-03 13:56:20 +00004889#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004890#ifdef IPPROTO_IPIP
4891 PyModule_AddIntConstant(m, "IPPROTO_IPIP", IPPROTO_IPIP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004892#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004893#ifdef IPPROTO_TCP
4894 PyModule_AddIntConstant(m, "IPPROTO_TCP", IPPROTO_TCP);
Guido van Rossum578de301998-05-28 20:18:18 +00004895#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004896 PyModule_AddIntConstant(m, "IPPROTO_TCP", 6);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004897#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004898#ifdef IPPROTO_EGP
4899 PyModule_AddIntConstant(m, "IPPROTO_EGP", IPPROTO_EGP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004900#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004901#ifdef IPPROTO_PUP
4902 PyModule_AddIntConstant(m, "IPPROTO_PUP", IPPROTO_PUP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004903#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004904#ifdef IPPROTO_UDP
4905 PyModule_AddIntConstant(m, "IPPROTO_UDP", IPPROTO_UDP);
Guido van Rossum578de301998-05-28 20:18:18 +00004906#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004907 PyModule_AddIntConstant(m, "IPPROTO_UDP", 17);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004908#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004909#ifdef IPPROTO_IDP
4910 PyModule_AddIntConstant(m, "IPPROTO_IDP", IPPROTO_IDP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004911#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004912#ifdef IPPROTO_HELLO
4913 PyModule_AddIntConstant(m, "IPPROTO_HELLO", IPPROTO_HELLO);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004914#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004915#ifdef IPPROTO_ND
4916 PyModule_AddIntConstant(m, "IPPROTO_ND", IPPROTO_ND);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004917#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004918#ifdef IPPROTO_TP
4919 PyModule_AddIntConstant(m, "IPPROTO_TP", IPPROTO_TP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004920#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004921#ifdef IPPROTO_IPV6
4922 PyModule_AddIntConstant(m, "IPPROTO_IPV6", IPPROTO_IPV6);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004923#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004924#ifdef IPPROTO_ROUTING
4925 PyModule_AddIntConstant(m, "IPPROTO_ROUTING", IPPROTO_ROUTING);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004926#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004927#ifdef IPPROTO_FRAGMENT
4928 PyModule_AddIntConstant(m, "IPPROTO_FRAGMENT", IPPROTO_FRAGMENT);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004929#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004930#ifdef IPPROTO_RSVP
4931 PyModule_AddIntConstant(m, "IPPROTO_RSVP", IPPROTO_RSVP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004932#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004933#ifdef IPPROTO_GRE
4934 PyModule_AddIntConstant(m, "IPPROTO_GRE", IPPROTO_GRE);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004935#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004936#ifdef IPPROTO_ESP
4937 PyModule_AddIntConstant(m, "IPPROTO_ESP", IPPROTO_ESP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004938#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004939#ifdef IPPROTO_AH
4940 PyModule_AddIntConstant(m, "IPPROTO_AH", IPPROTO_AH);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004941#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004942#ifdef IPPROTO_MOBILE
4943 PyModule_AddIntConstant(m, "IPPROTO_MOBILE", IPPROTO_MOBILE);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004944#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004945#ifdef IPPROTO_ICMPV6
4946 PyModule_AddIntConstant(m, "IPPROTO_ICMPV6", IPPROTO_ICMPV6);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004947#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004948#ifdef IPPROTO_NONE
4949 PyModule_AddIntConstant(m, "IPPROTO_NONE", IPPROTO_NONE);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004950#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004951#ifdef IPPROTO_DSTOPTS
4952 PyModule_AddIntConstant(m, "IPPROTO_DSTOPTS", IPPROTO_DSTOPTS);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004953#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004954#ifdef IPPROTO_XTP
4955 PyModule_AddIntConstant(m, "IPPROTO_XTP", IPPROTO_XTP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004956#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004957#ifdef IPPROTO_EON
4958 PyModule_AddIntConstant(m, "IPPROTO_EON", IPPROTO_EON);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004959#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004960#ifdef IPPROTO_PIM
4961 PyModule_AddIntConstant(m, "IPPROTO_PIM", IPPROTO_PIM);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004962#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004963#ifdef IPPROTO_IPCOMP
4964 PyModule_AddIntConstant(m, "IPPROTO_IPCOMP", IPPROTO_IPCOMP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004965#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004966#ifdef IPPROTO_VRRP
4967 PyModule_AddIntConstant(m, "IPPROTO_VRRP", IPPROTO_VRRP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004968#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004969#ifdef IPPROTO_BIP
4970 PyModule_AddIntConstant(m, "IPPROTO_BIP", IPPROTO_BIP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004971#endif
4972/**/
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004973#ifdef IPPROTO_RAW
4974 PyModule_AddIntConstant(m, "IPPROTO_RAW", IPPROTO_RAW);
Guido van Rossum578de301998-05-28 20:18:18 +00004975#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004976 PyModule_AddIntConstant(m, "IPPROTO_RAW", 255);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004977#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004978#ifdef IPPROTO_MAX
4979 PyModule_AddIntConstant(m, "IPPROTO_MAX", IPPROTO_MAX);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004980#endif
4981
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004982 /* Some port configuration */
4983#ifdef IPPORT_RESERVED
4984 PyModule_AddIntConstant(m, "IPPORT_RESERVED", IPPORT_RESERVED);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004985#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004986 PyModule_AddIntConstant(m, "IPPORT_RESERVED", 1024);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004987#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004988#ifdef IPPORT_USERRESERVED
4989 PyModule_AddIntConstant(m, "IPPORT_USERRESERVED", IPPORT_USERRESERVED);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004990#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004991 PyModule_AddIntConstant(m, "IPPORT_USERRESERVED", 5000);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004992#endif
4993
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004994 /* Some reserved IP v.4 addresses */
4995#ifdef INADDR_ANY
4996 PyModule_AddIntConstant(m, "INADDR_ANY", INADDR_ANY);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004997#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004998 PyModule_AddIntConstant(m, "INADDR_ANY", 0x00000000);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004999#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005000#ifdef INADDR_BROADCAST
5001 PyModule_AddIntConstant(m, "INADDR_BROADCAST", INADDR_BROADCAST);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00005002#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005003 PyModule_AddIntConstant(m, "INADDR_BROADCAST", 0xffffffff);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00005004#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005005#ifdef INADDR_LOOPBACK
5006 PyModule_AddIntConstant(m, "INADDR_LOOPBACK", INADDR_LOOPBACK);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00005007#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005008 PyModule_AddIntConstant(m, "INADDR_LOOPBACK", 0x7F000001);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00005009#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005010#ifdef INADDR_UNSPEC_GROUP
5011 PyModule_AddIntConstant(m, "INADDR_UNSPEC_GROUP", INADDR_UNSPEC_GROUP);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00005012#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005013 PyModule_AddIntConstant(m, "INADDR_UNSPEC_GROUP", 0xe0000000);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00005014#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005015#ifdef INADDR_ALLHOSTS_GROUP
5016 PyModule_AddIntConstant(m, "INADDR_ALLHOSTS_GROUP",
5017 INADDR_ALLHOSTS_GROUP);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00005018#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005019 PyModule_AddIntConstant(m, "INADDR_ALLHOSTS_GROUP", 0xe0000001);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00005020#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005021#ifdef INADDR_MAX_LOCAL_GROUP
5022 PyModule_AddIntConstant(m, "INADDR_MAX_LOCAL_GROUP",
5023 INADDR_MAX_LOCAL_GROUP);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00005024#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005025 PyModule_AddIntConstant(m, "INADDR_MAX_LOCAL_GROUP", 0xe00000ff);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00005026#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005027#ifdef INADDR_NONE
5028 PyModule_AddIntConstant(m, "INADDR_NONE", INADDR_NONE);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00005029#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005030 PyModule_AddIntConstant(m, "INADDR_NONE", 0xffffffff);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00005031#endif
5032
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005033 /* IPv4 [gs]etsockopt options */
5034#ifdef IP_OPTIONS
5035 PyModule_AddIntConstant(m, "IP_OPTIONS", IP_OPTIONS);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00005036#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005037#ifdef IP_HDRINCL
5038 PyModule_AddIntConstant(m, "IP_HDRINCL", IP_HDRINCL);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00005039#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005040#ifdef IP_TOS
5041 PyModule_AddIntConstant(m, "IP_TOS", IP_TOS);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00005042#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005043#ifdef IP_TTL
5044 PyModule_AddIntConstant(m, "IP_TTL", IP_TTL);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00005045#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005046#ifdef IP_RECVOPTS
5047 PyModule_AddIntConstant(m, "IP_RECVOPTS", IP_RECVOPTS);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00005048#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005049#ifdef IP_RECVRETOPTS
5050 PyModule_AddIntConstant(m, "IP_RECVRETOPTS", IP_RECVRETOPTS);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00005051#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005052#ifdef IP_RECVDSTADDR
5053 PyModule_AddIntConstant(m, "IP_RECVDSTADDR", IP_RECVDSTADDR);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00005054#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005055#ifdef IP_RETOPTS
5056 PyModule_AddIntConstant(m, "IP_RETOPTS", IP_RETOPTS);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00005057#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005058#ifdef IP_MULTICAST_IF
5059 PyModule_AddIntConstant(m, "IP_MULTICAST_IF", IP_MULTICAST_IF);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00005060#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005061#ifdef IP_MULTICAST_TTL
5062 PyModule_AddIntConstant(m, "IP_MULTICAST_TTL", IP_MULTICAST_TTL);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00005063#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005064#ifdef IP_MULTICAST_LOOP
5065 PyModule_AddIntConstant(m, "IP_MULTICAST_LOOP", IP_MULTICAST_LOOP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00005066#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005067#ifdef IP_ADD_MEMBERSHIP
5068 PyModule_AddIntConstant(m, "IP_ADD_MEMBERSHIP", IP_ADD_MEMBERSHIP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00005069#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005070#ifdef IP_DROP_MEMBERSHIP
5071 PyModule_AddIntConstant(m, "IP_DROP_MEMBERSHIP", IP_DROP_MEMBERSHIP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00005072#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005073#ifdef IP_DEFAULT_MULTICAST_TTL
5074 PyModule_AddIntConstant(m, "IP_DEFAULT_MULTICAST_TTL",
5075 IP_DEFAULT_MULTICAST_TTL);
Guido van Rossum09be4091999-08-09 14:40:40 +00005076#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005077#ifdef IP_DEFAULT_MULTICAST_LOOP
5078 PyModule_AddIntConstant(m, "IP_DEFAULT_MULTICAST_LOOP",
5079 IP_DEFAULT_MULTICAST_LOOP);
Guido van Rossum09be4091999-08-09 14:40:40 +00005080#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005081#ifdef IP_MAX_MEMBERSHIPS
5082 PyModule_AddIntConstant(m, "IP_MAX_MEMBERSHIPS", IP_MAX_MEMBERSHIPS);
Guido van Rossum09be4091999-08-09 14:40:40 +00005083#endif
5084
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005085 /* IPv6 [gs]etsockopt options, defined in RFC2553 */
5086#ifdef IPV6_JOIN_GROUP
5087 PyModule_AddIntConstant(m, "IPV6_JOIN_GROUP", IPV6_JOIN_GROUP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00005088#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005089#ifdef IPV6_LEAVE_GROUP
5090 PyModule_AddIntConstant(m, "IPV6_LEAVE_GROUP", IPV6_LEAVE_GROUP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00005091#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005092#ifdef IPV6_MULTICAST_HOPS
5093 PyModule_AddIntConstant(m, "IPV6_MULTICAST_HOPS", IPV6_MULTICAST_HOPS);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00005094#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005095#ifdef IPV6_MULTICAST_IF
5096 PyModule_AddIntConstant(m, "IPV6_MULTICAST_IF", IPV6_MULTICAST_IF);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00005097#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005098#ifdef IPV6_MULTICAST_LOOP
5099 PyModule_AddIntConstant(m, "IPV6_MULTICAST_LOOP", IPV6_MULTICAST_LOOP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00005100#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005101#ifdef IPV6_UNICAST_HOPS
5102 PyModule_AddIntConstant(m, "IPV6_UNICAST_HOPS", IPV6_UNICAST_HOPS);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00005103#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005104 /* Additional IPV6 socket options, defined in RFC 3493 */
Martin v. Löwisda91d022003-12-30 11:14:01 +00005105#ifdef IPV6_V6ONLY
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005106 PyModule_AddIntConstant(m, "IPV6_V6ONLY", IPV6_V6ONLY);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005107#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005108 /* Advanced IPV6 socket options, from RFC 3542 */
Martin v. Löwisda91d022003-12-30 11:14:01 +00005109#ifdef IPV6_CHECKSUM
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005110 PyModule_AddIntConstant(m, "IPV6_CHECKSUM", IPV6_CHECKSUM);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005111#endif
5112#ifdef IPV6_DONTFRAG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005113 PyModule_AddIntConstant(m, "IPV6_DONTFRAG", IPV6_DONTFRAG);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005114#endif
5115#ifdef IPV6_DSTOPTS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005116 PyModule_AddIntConstant(m, "IPV6_DSTOPTS", IPV6_DSTOPTS);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005117#endif
5118#ifdef IPV6_HOPLIMIT
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005119 PyModule_AddIntConstant(m, "IPV6_HOPLIMIT", IPV6_HOPLIMIT);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005120#endif
5121#ifdef IPV6_HOPOPTS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005122 PyModule_AddIntConstant(m, "IPV6_HOPOPTS", IPV6_HOPOPTS);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005123#endif
5124#ifdef IPV6_NEXTHOP
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005125 PyModule_AddIntConstant(m, "IPV6_NEXTHOP", IPV6_NEXTHOP);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005126#endif
5127#ifdef IPV6_PATHMTU
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005128 PyModule_AddIntConstant(m, "IPV6_PATHMTU", IPV6_PATHMTU);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005129#endif
5130#ifdef IPV6_PKTINFO
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005131 PyModule_AddIntConstant(m, "IPV6_PKTINFO", IPV6_PKTINFO);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005132#endif
5133#ifdef IPV6_RECVDSTOPTS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005134 PyModule_AddIntConstant(m, "IPV6_RECVDSTOPTS", IPV6_RECVDSTOPTS);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005135#endif
5136#ifdef IPV6_RECVHOPLIMIT
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005137 PyModule_AddIntConstant(m, "IPV6_RECVHOPLIMIT", IPV6_RECVHOPLIMIT);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005138#endif
5139#ifdef IPV6_RECVHOPOPTS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005140 PyModule_AddIntConstant(m, "IPV6_RECVHOPOPTS", IPV6_RECVHOPOPTS);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005141#endif
5142#ifdef IPV6_RECVPKTINFO
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005143 PyModule_AddIntConstant(m, "IPV6_RECVPKTINFO", IPV6_RECVPKTINFO);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005144#endif
5145#ifdef IPV6_RECVRTHDR
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005146 PyModule_AddIntConstant(m, "IPV6_RECVRTHDR", IPV6_RECVRTHDR);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005147#endif
5148#ifdef IPV6_RECVTCLASS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005149 PyModule_AddIntConstant(m, "IPV6_RECVTCLASS", IPV6_RECVTCLASS);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005150#endif
5151#ifdef IPV6_RTHDR
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005152 PyModule_AddIntConstant(m, "IPV6_RTHDR", IPV6_RTHDR);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005153#endif
5154#ifdef IPV6_RTHDRDSTOPTS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005155 PyModule_AddIntConstant(m, "IPV6_RTHDRDSTOPTS", IPV6_RTHDRDSTOPTS);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005156#endif
5157#ifdef IPV6_RTHDR_TYPE_0
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005158 PyModule_AddIntConstant(m, "IPV6_RTHDR_TYPE_0", IPV6_RTHDR_TYPE_0);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005159#endif
5160#ifdef IPV6_RECVPATHMTU
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005161 PyModule_AddIntConstant(m, "IPV6_RECVPATHMTU", IPV6_RECVPATHMTU);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005162#endif
5163#ifdef IPV6_TCLASS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005164 PyModule_AddIntConstant(m, "IPV6_TCLASS", IPV6_TCLASS);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005165#endif
5166#ifdef IPV6_USE_MIN_MTU
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005167 PyModule_AddIntConstant(m, "IPV6_USE_MIN_MTU", IPV6_USE_MIN_MTU);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005168#endif
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00005169
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005170 /* TCP options */
5171#ifdef TCP_NODELAY
5172 PyModule_AddIntConstant(m, "TCP_NODELAY", TCP_NODELAY);
Guido van Rossum09be4091999-08-09 14:40:40 +00005173#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005174#ifdef TCP_MAXSEG
5175 PyModule_AddIntConstant(m, "TCP_MAXSEG", TCP_MAXSEG);
Guido van Rossum09be4091999-08-09 14:40:40 +00005176#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005177#ifdef TCP_CORK
5178 PyModule_AddIntConstant(m, "TCP_CORK", TCP_CORK);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00005179#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005180#ifdef TCP_KEEPIDLE
5181 PyModule_AddIntConstant(m, "TCP_KEEPIDLE", TCP_KEEPIDLE);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00005182#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005183#ifdef TCP_KEEPINTVL
5184 PyModule_AddIntConstant(m, "TCP_KEEPINTVL", TCP_KEEPINTVL);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00005185#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005186#ifdef TCP_KEEPCNT
5187 PyModule_AddIntConstant(m, "TCP_KEEPCNT", TCP_KEEPCNT);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00005188#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005189#ifdef TCP_SYNCNT
5190 PyModule_AddIntConstant(m, "TCP_SYNCNT", TCP_SYNCNT);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00005191#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005192#ifdef TCP_LINGER2
5193 PyModule_AddIntConstant(m, "TCP_LINGER2", TCP_LINGER2);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00005194#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005195#ifdef TCP_DEFER_ACCEPT
5196 PyModule_AddIntConstant(m, "TCP_DEFER_ACCEPT", TCP_DEFER_ACCEPT);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00005197#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005198#ifdef TCP_WINDOW_CLAMP
5199 PyModule_AddIntConstant(m, "TCP_WINDOW_CLAMP", TCP_WINDOW_CLAMP);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00005200#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005201#ifdef TCP_INFO
5202 PyModule_AddIntConstant(m, "TCP_INFO", TCP_INFO);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00005203#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005204#ifdef TCP_QUICKACK
5205 PyModule_AddIntConstant(m, "TCP_QUICKACK", TCP_QUICKACK);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00005206#endif
5207
Guido van Rossum09be4091999-08-09 14:40:40 +00005208
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005209 /* IPX options */
5210#ifdef IPX_TYPE
5211 PyModule_AddIntConstant(m, "IPX_TYPE", IPX_TYPE);
Guido van Rossum09be4091999-08-09 14:40:40 +00005212#endif
Guido van Rossum4f199ea1998-04-09 20:56:35 +00005213
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005214 /* get{addr,name}info parameters */
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005215#ifdef EAI_ADDRFAMILY
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005216 PyModule_AddIntConstant(m, "EAI_ADDRFAMILY", EAI_ADDRFAMILY);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005217#endif
5218#ifdef EAI_AGAIN
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005219 PyModule_AddIntConstant(m, "EAI_AGAIN", EAI_AGAIN);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005220#endif
5221#ifdef EAI_BADFLAGS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005222 PyModule_AddIntConstant(m, "EAI_BADFLAGS", EAI_BADFLAGS);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005223#endif
5224#ifdef EAI_FAIL
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005225 PyModule_AddIntConstant(m, "EAI_FAIL", EAI_FAIL);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005226#endif
5227#ifdef EAI_FAMILY
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005228 PyModule_AddIntConstant(m, "EAI_FAMILY", EAI_FAMILY);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005229#endif
5230#ifdef EAI_MEMORY
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005231 PyModule_AddIntConstant(m, "EAI_MEMORY", EAI_MEMORY);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005232#endif
5233#ifdef EAI_NODATA
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005234 PyModule_AddIntConstant(m, "EAI_NODATA", EAI_NODATA);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005235#endif
5236#ifdef EAI_NONAME
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005237 PyModule_AddIntConstant(m, "EAI_NONAME", EAI_NONAME);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005238#endif
Martin v. Löwisda91d022003-12-30 11:14:01 +00005239#ifdef EAI_OVERFLOW
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005240 PyModule_AddIntConstant(m, "EAI_OVERFLOW", EAI_OVERFLOW);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005241#endif
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005242#ifdef EAI_SERVICE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005243 PyModule_AddIntConstant(m, "EAI_SERVICE", EAI_SERVICE);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005244#endif
5245#ifdef EAI_SOCKTYPE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005246 PyModule_AddIntConstant(m, "EAI_SOCKTYPE", EAI_SOCKTYPE);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005247#endif
5248#ifdef EAI_SYSTEM
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005249 PyModule_AddIntConstant(m, "EAI_SYSTEM", EAI_SYSTEM);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005250#endif
5251#ifdef EAI_BADHINTS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005252 PyModule_AddIntConstant(m, "EAI_BADHINTS", EAI_BADHINTS);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005253#endif
5254#ifdef EAI_PROTOCOL
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005255 PyModule_AddIntConstant(m, "EAI_PROTOCOL", EAI_PROTOCOL);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005256#endif
5257#ifdef EAI_MAX
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005258 PyModule_AddIntConstant(m, "EAI_MAX", EAI_MAX);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005259#endif
5260#ifdef AI_PASSIVE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005261 PyModule_AddIntConstant(m, "AI_PASSIVE", AI_PASSIVE);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005262#endif
5263#ifdef AI_CANONNAME
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005264 PyModule_AddIntConstant(m, "AI_CANONNAME", AI_CANONNAME);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005265#endif
5266#ifdef AI_NUMERICHOST
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005267 PyModule_AddIntConstant(m, "AI_NUMERICHOST", AI_NUMERICHOST);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005268#endif
Martin v. Löwisda91d022003-12-30 11:14:01 +00005269#ifdef AI_NUMERICSERV
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005270 PyModule_AddIntConstant(m, "AI_NUMERICSERV", AI_NUMERICSERV);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005271#endif
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005272#ifdef AI_MASK
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005273 PyModule_AddIntConstant(m, "AI_MASK", AI_MASK);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005274#endif
5275#ifdef AI_ALL
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005276 PyModule_AddIntConstant(m, "AI_ALL", AI_ALL);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005277#endif
5278#ifdef AI_V4MAPPED_CFG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005279 PyModule_AddIntConstant(m, "AI_V4MAPPED_CFG", AI_V4MAPPED_CFG);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005280#endif
5281#ifdef AI_ADDRCONFIG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005282 PyModule_AddIntConstant(m, "AI_ADDRCONFIG", AI_ADDRCONFIG);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005283#endif
5284#ifdef AI_V4MAPPED
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005285 PyModule_AddIntConstant(m, "AI_V4MAPPED", AI_V4MAPPED);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005286#endif
5287#ifdef AI_DEFAULT
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005288 PyModule_AddIntConstant(m, "AI_DEFAULT", AI_DEFAULT);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005289#endif
5290#ifdef NI_MAXHOST
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005291 PyModule_AddIntConstant(m, "NI_MAXHOST", NI_MAXHOST);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005292#endif
5293#ifdef NI_MAXSERV
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005294 PyModule_AddIntConstant(m, "NI_MAXSERV", NI_MAXSERV);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005295#endif
5296#ifdef NI_NOFQDN
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005297 PyModule_AddIntConstant(m, "NI_NOFQDN", NI_NOFQDN);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005298#endif
5299#ifdef NI_NUMERICHOST
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005300 PyModule_AddIntConstant(m, "NI_NUMERICHOST", NI_NUMERICHOST);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005301#endif
5302#ifdef NI_NAMEREQD
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005303 PyModule_AddIntConstant(m, "NI_NAMEREQD", NI_NAMEREQD);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005304#endif
5305#ifdef NI_NUMERICSERV
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005306 PyModule_AddIntConstant(m, "NI_NUMERICSERV", NI_NUMERICSERV);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005307#endif
5308#ifdef NI_DGRAM
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005309 PyModule_AddIntConstant(m, "NI_DGRAM", NI_DGRAM);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005310#endif
5311
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005312 /* shutdown() parameters */
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005313#ifdef SHUT_RD
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005314 PyModule_AddIntConstant(m, "SHUT_RD", SHUT_RD);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005315#elif defined(SD_RECEIVE)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005316 PyModule_AddIntConstant(m, "SHUT_RD", SD_RECEIVE);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005317#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005318 PyModule_AddIntConstant(m, "SHUT_RD", 0);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005319#endif
5320#ifdef SHUT_WR
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005321 PyModule_AddIntConstant(m, "SHUT_WR", SHUT_WR);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005322#elif defined(SD_SEND)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005323 PyModule_AddIntConstant(m, "SHUT_WR", SD_SEND);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005324#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005325 PyModule_AddIntConstant(m, "SHUT_WR", 1);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005326#endif
5327#ifdef SHUT_RDWR
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005328 PyModule_AddIntConstant(m, "SHUT_RDWR", SHUT_RDWR);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005329#elif defined(SD_BOTH)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005330 PyModule_AddIntConstant(m, "SHUT_RDWR", SD_BOTH);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005331#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005332 PyModule_AddIntConstant(m, "SHUT_RDWR", 2);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005333#endif
5334
Christian Heimesfaf2f632008-01-06 16:59:19 +00005335#ifdef SIO_RCVALL
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005336 {
5337 DWORD codes[] = {SIO_RCVALL, SIO_KEEPALIVE_VALS};
5338 const char *names[] = {"SIO_RCVALL", "SIO_KEEPALIVE_VALS"};
5339 int i;
5340 for(i = 0; i<sizeof(codes)/sizeof(*codes); ++i) {
5341 PyObject *tmp;
5342 tmp = PyLong_FromUnsignedLong(codes[i]);
5343 if (tmp == NULL)
5344 return NULL;
5345 PyModule_AddObject(m, names[i], tmp);
5346 }
5347 }
5348 PyModule_AddIntConstant(m, "RCVALL_OFF", RCVALL_OFF);
5349 PyModule_AddIntConstant(m, "RCVALL_ON", RCVALL_ON);
5350 PyModule_AddIntConstant(m, "RCVALL_SOCKETLEVELONLY", RCVALL_SOCKETLEVELONLY);
Amaury Forgeot d'Arc762681b2008-06-12 23:03:41 +00005351#ifdef RCVALL_IPLEVEL
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005352 PyModule_AddIntConstant(m, "RCVALL_IPLEVEL", RCVALL_IPLEVEL);
Amaury Forgeot d'Arc762681b2008-06-12 23:03:41 +00005353#endif
5354#ifdef RCVALL_MAX
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005355 PyModule_AddIntConstant(m, "RCVALL_MAX", RCVALL_MAX);
Amaury Forgeot d'Arc762681b2008-06-12 23:03:41 +00005356#endif
Christian Heimesfaf2f632008-01-06 16:59:19 +00005357#endif /* _MSTCPIP_ */
5358
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005359 /* Initialize gethostbyname lock */
Just van Rossum1040d2c2003-05-09 07:53:18 +00005360#if defined(USE_GETHOSTBYNAME_LOCK) || defined(USE_GETADDRINFO_LOCK)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005361 netdb_lock = PyThread_allocate_lock();
Guido van Rossum4f199ea1998-04-09 20:56:35 +00005362#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005363 return m;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00005364}
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005365
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00005366
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005367#ifndef HAVE_INET_PTON
Christian Heimes96e7b3d2007-11-20 06:51:17 +00005368#if !defined(NTDDI_VERSION) || (NTDDI_VERSION < NTDDI_LONGHORN)
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00005369
5370/* Simplistic emulation code for inet_pton that only works for IPv4 */
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00005371/* These are not exposed because they do not set errno properly */
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00005372
Guido van Rossum3eede5a2002-06-07 02:08:35 +00005373int
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00005374inet_pton(int af, const char *src, void *dst)
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005375{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005376 if (af == AF_INET) {
Benjamin Petersonf91df042009-02-13 02:50:59 +00005377#if (SIZEOF_INT != 4)
5378#error "Not sure if in_addr_t exists and int is not 32-bits."
5379#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005380 unsigned int packed_addr;
5381 packed_addr = inet_addr(src);
5382 if (packed_addr == INADDR_NONE)
5383 return 0;
5384 memcpy(dst, &packed_addr, 4);
5385 return 1;
5386 }
5387 /* Should set errno to EAFNOSUPPORT */
5388 return -1;
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005389}
5390
Martin v. Löwisc925b1532001-07-21 09:42:15 +00005391const char *
5392inet_ntop(int af, const void *src, char *dst, socklen_t size)
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005393{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005394 if (af == AF_INET) {
5395 struct in_addr packed_addr;
5396 if (size < 16)
5397 /* Should set errno to ENOSPC. */
5398 return NULL;
5399 memcpy(&packed_addr, src, sizeof(packed_addr));
5400 return strncpy(dst, inet_ntoa(packed_addr), size);
5401 }
5402 /* Should set errno to EAFNOSUPPORT */
5403 return NULL;
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005404}
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00005405
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005406#endif
Christian Heimesb6150692007-11-15 23:37:07 +00005407#endif