blob: fc671e06648a1bf6042ebcdea02a22d451f5342b [file] [log] [blame]
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001/* Socket module */
2
3/*
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004
Guido van Rossum6574b3e1991-06-25 21:36:08 +00005This module provides an interface to Berkeley socket IPC.
6
7Limitations:
8
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00009- Only AF_INET, AF_INET6 and AF_UNIX address families are supported in a
Christian Heimes043d6f62008-01-07 17:19:16 +000010 portable manner, though AF_PACKET, AF_NETLINK and AF_TIPC are supported
11 under Linux.
Guido van Rossumc4fcfa32002-06-07 03:19:37 +000012- No read/write operations (use sendall/recv or makefile instead).
13- Additional restrictions apply on some non-Unix platforms (compensated
14 for by socket.py).
Guido van Rossum6574b3e1991-06-25 21:36:08 +000015
Guido van Rossum27e177d1995-03-16 15:43:47 +000016Module interface:
Guido van Rossum6574b3e1991-06-25 21:36:08 +000017
Guido van Rossum27e177d1995-03-16 15:43:47 +000018- socket.error: exception raised for socket specific errors
Martin v. Löwis2d8d4272001-07-21 18:05:31 +000019- socket.gaierror: exception raised for getaddrinfo/getnameinfo errors,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000020 a subclass of socket.error
Martin v. Löwis2d8d4272001-07-21 18:05:31 +000021- socket.herror: exception raised for gethostby* errors,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000022 a subclass of socket.error
Guido van Rossum30a685f1991-06-27 15:51:29 +000023- socket.gethostbyname(hostname) --> host IP address (string: 'dd.dd.dd.dd')
Guido van Rossum3bbc62e1995-01-02 19:30:30 +000024- socket.gethostbyaddr(IP address) --> (hostname, [alias, ...], [IP addr, ...])
Guido van Rossum27e177d1995-03-16 15:43:47 +000025- socket.gethostname() --> host name (string: 'spam' or 'spam.domain.com')
Guido van Rossum25405c71996-12-19 16:42:52 +000026- socket.getprotobyname(protocolname) --> protocol number
Barry Warsaw11b91a02004-06-28 00:50:43 +000027- socket.getservbyname(servicename[, protocolname]) --> port number
28- socket.getservbyport(portnumber[, protocolname]) --> service name
Guido van Rossum7d0a8262007-05-21 23:13:11 +000029- socket.socket([family[, type [, proto, fileno]]]) --> new socket object
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000030 (fileno specifies a pre-existing socket file descriptor)
Dave Cole331708b2004-08-09 04:51:41 +000031- socket.socketpair([family[, type [, proto]]]) --> (socket, socket)
Guido van Rossum006bf911996-06-12 04:04:55 +000032- socket.ntohs(16 bit value) --> new int object
33- socket.ntohl(32 bit value) --> new int object
34- socket.htons(16 bit value) --> new int object
35- socket.htonl(32 bit value) --> new int object
Martin v. Löwis2d8d4272001-07-21 18:05:31 +000036- socket.getaddrinfo(host, port [, family, socktype, proto, flags])
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000037 --> List of (family, socktype, proto, canonname, sockaddr)
Martin v. Löwis2d8d4272001-07-21 18:05:31 +000038- socket.getnameinfo(sockaddr, flags) --> (host, port)
Guido van Rossum27e177d1995-03-16 15:43:47 +000039- socket.AF_INET, socket.SOCK_STREAM, etc.: constants from <socket.h>
Guido van Rossum47dfa4a2003-04-25 05:48:32 +000040- socket.has_ipv6: boolean value indicating if IPv6 is supported
Guido van Rossum5c9eb211999-08-20 18:21:51 +000041- socket.inet_aton(IP address) -> 32-bit packed IP representation
42- socket.inet_ntoa(packed IP) -> IP address string
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +000043- socket.getdefaulttimeout() -> None | float
44- socket.setdefaulttimeout(None | float)
Guido van Rossum6574b3e1991-06-25 21:36:08 +000045- an Internet socket address is a pair (hostname, port)
46 where hostname can be anything recognized by gethostbyname()
47 (including the dd.dd.dd.dd notation) and port is in host byte order
48- where a hostname is returned, the dd.dd.dd.dd notation is used
Guido van Rossum27e177d1995-03-16 15:43:47 +000049- a UNIX domain socket address is a string specifying the pathname
Jeremy Hylton22308652001-02-02 03:23:09 +000050- an AF_PACKET socket address is a tuple containing a string
51 specifying the ethernet interface and an integer specifying
52 the Ethernet protocol number to be received. For example:
Jeremy Hyltondbfb6622001-02-02 19:55:17 +000053 ("eth0",0x1234). Optional 3rd,4th,5th elements in the tuple
Andrew M. Kuchlingb8e17172004-07-10 23:39:35 +000054 specify packet-type and ha-type/addr.
Christian Heimes043d6f62008-01-07 17:19:16 +000055- an AF_TIPC socket address is expressed as
56 (addr_type, v1, v2, v3 [, scope]); where addr_type can be one of:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000057 TIPC_ADDR_NAMESEQ, TIPC_ADDR_NAME, and TIPC_ADDR_ID;
Christian Heimes043d6f62008-01-07 17:19:16 +000058 and scope can be one of:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000059 TIPC_ZONE_SCOPE, TIPC_CLUSTER_SCOPE, and TIPC_NODE_SCOPE.
Christian Heimes043d6f62008-01-07 17:19:16 +000060 The meaning of v1, v2 and v3 depends on the value of addr_type:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000061 if addr_type is TIPC_ADDR_NAME:
62 v1 is the server type
63 v2 is the port identifier
64 v3 is ignored
65 if addr_type is TIPC_ADDR_NAMESEQ:
66 v1 is the server type
67 v2 is the lower port number
68 v3 is the upper port number
69 if addr_type is TIPC_ADDR_ID:
70 v1 is the node
71 v2 is the ref
72 v3 is ignored
Christian Heimes043d6f62008-01-07 17:19:16 +000073
Guido van Rossum6574b3e1991-06-25 21:36:08 +000074
Guido van Rossumc4fcfa32002-06-07 03:19:37 +000075Local naming conventions:
Guido van Rossum6574b3e1991-06-25 21:36:08 +000076
Guido van Rossumc4fcfa32002-06-07 03:19:37 +000077- names starting with sock_ are socket object methods
78- names starting with socket_ are module-level functions
79- names starting with PySocket are exported through socketmodule.h
Guido van Rossum30a685f1991-06-27 15:51:29 +000080
Guido van Rossum6574b3e1991-06-25 21:36:08 +000081*/
82
Thomas Wouters477c8d52006-05-27 19:21:47 +000083#ifdef __APPLE__
84 /*
85 * inet_aton is not available on OSX 10.3, yet we want to use a binary
86 * that was build on 10.4 or later to work on that release, weak linking
87 * comes to the rescue.
88 */
89# pragma weak inet_aton
90#endif
91
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000092#include "Python.h"
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000093#include "structmember.h"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000094
Guido van Rossum47dfa4a2003-04-25 05:48:32 +000095#undef MAX
96#define MAX(x, y) ((x) < (y) ? (y) : (x))
97
Guido van Rossumc4fcfa32002-06-07 03:19:37 +000098/* Socket object documentation */
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000099PyDoc_STRVAR(sock_doc,
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000100"socket([family[, type[, proto]]]) -> socket object\n\
101\n\
102Open a socket of the given type. The family argument specifies the\n\
103address family; it defaults to AF_INET. The type argument specifies\n\
104whether this is a stream (SOCK_STREAM, this is the default)\n\
105or datagram (SOCK_DGRAM) socket. The protocol argument defaults to 0,\n\
106specifying the default protocol. Keyword arguments are accepted.\n\
107\n\
108A socket object represents one endpoint of a network connection.\n\
109\n\
110Methods of socket objects (keyword arguments not allowed):\n\
111\n\
Guido van Rossum39eb8fa2007-11-16 01:24:05 +0000112_accept() -- accept connection, returning new socket fd and client address\n\
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000113bind(addr) -- bind the socket to a local address\n\
114close() -- close the socket\n\
115connect(addr) -- connect the socket to a remote address\n\
116connect_ex(addr) -- connect, return an error code instead of an exception\n\
Guido van Rossum39eb8fa2007-11-16 01:24:05 +0000117_dup() -- return a new socket fd duplicated from fileno()\n\
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000118fileno() -- return underlying file descriptor\n\
119getpeername() -- return remote address [*]\n\
120getsockname() -- return local address\n\
121getsockopt(level, optname[, buflen]) -- get socket options\n\
122gettimeout() -- return timeout or None\n\
123listen(n) -- start listening for incoming connections\n\
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000124recv(buflen[, flags]) -- receive data\n\
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000125recv_into(buffer[, nbytes[, flags]]) -- receive data (into a buffer)\n\
Thomas Wouters477c8d52006-05-27 19:21:47 +0000126recvfrom(buflen[, flags]) -- receive data and sender\'s address\n\
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000127recvfrom_into(buffer[, nbytes, [, flags])\n\
Thomas Wouters477c8d52006-05-27 19:21:47 +0000128 -- receive data and sender\'s address (into a buffer)\n\
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000129sendall(data[, flags]) -- send all data\n\
130send(data[, flags]) -- send data, may not send all of it\n\
131sendto(data[, flags], addr) -- send data to a given address\n\
132setblocking(0 | 1) -- set or clear the blocking I/O flag\n\
133setsockopt(level, optname, value) -- set socket options\n\
134settimeout(None | float) -- set or clear the timeout\n\
135shutdown(how) -- shut down traffic in one or both directions\n\
136\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000137 [*] not available on all platforms!");
Guido van Rossum3baaa131999-03-22 21:44:51 +0000138
Walter Dörwaldf0dfc7a2003-10-20 14:01:56 +0000139/* XXX This is a terrible mess of platform-dependent preprocessor hacks.
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000140 I hope some day someone can clean this up please... */
141
Guido van Rossum9376b741999-09-15 22:01:40 +0000142/* Hacks for gethostbyname_r(). On some non-Linux platforms, the configure
143 script doesn't get this right, so we hardcode some platform checks below.
144 On the other hand, not all Linux versions agree, so there the settings
145 computed by the configure script are needed! */
146
147#ifndef linux
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000148# undef HAVE_GETHOSTBYNAME_R_3_ARG
149# undef HAVE_GETHOSTBYNAME_R_5_ARG
150# undef HAVE_GETHOSTBYNAME_R_6_ARG
Guido van Rossum9376b741999-09-15 22:01:40 +0000151#endif
Guido van Rossume7de2061999-03-24 17:24:33 +0000152
Guido van Rossum7a122991999-04-13 04:07:32 +0000153#ifndef WITH_THREAD
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000154# undef HAVE_GETHOSTBYNAME_R
Guido van Rossum7a122991999-04-13 04:07:32 +0000155#endif
156
Guido van Rossume7de2061999-03-24 17:24:33 +0000157#ifdef HAVE_GETHOSTBYNAME_R
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000158# if defined(_AIX) || defined(__osf__)
159# define HAVE_GETHOSTBYNAME_R_3_ARG
160# elif defined(__sun) || defined(__sgi)
161# define HAVE_GETHOSTBYNAME_R_5_ARG
162# elif defined(linux)
Guido van Rossum9376b741999-09-15 22:01:40 +0000163/* Rely on the configure script */
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000164# else
165# undef HAVE_GETHOSTBYNAME_R
166# endif
Guido van Rossume7de2061999-03-24 17:24:33 +0000167#endif
168
Guido van Rossum3eede5a2002-06-07 02:08:35 +0000169#if !defined(HAVE_GETHOSTBYNAME_R) && defined(WITH_THREAD) && \
170 !defined(MS_WINDOWS)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000171# define USE_GETHOSTBYNAME_LOCK
Guido van Rossum3baaa131999-03-22 21:44:51 +0000172#endif
173
Hye-Shik Chang9ceebd52005-09-24 14:58:47 +0000174/* To use __FreeBSD_version */
175#ifdef HAVE_SYS_PARAM_H
176#include <sys/param.h>
177#endif
Just van Rossum1040d2c2003-05-09 07:53:18 +0000178/* On systems on which getaddrinfo() is believed to not be thread-safe,
Just van Rossum09aecd72003-05-09 08:03:44 +0000179 (this includes the getaddrinfo emulation) protect access with a lock. */
Hye-Shik Chang9ceebd52005-09-24 14:58:47 +0000180#if defined(WITH_THREAD) && (defined(__APPLE__) || \
181 (defined(__FreeBSD__) && __FreeBSD_version+0 < 503000) || \
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000182 defined(__OpenBSD__) || defined(__NetBSD__) || \
183 defined(__VMS) || !defined(HAVE_GETADDRINFO))
Just van Rossum1040d2c2003-05-09 07:53:18 +0000184#define USE_GETADDRINFO_LOCK
185#endif
186
187#ifdef USE_GETADDRINFO_LOCK
188#define ACQUIRE_GETADDRINFO_LOCK PyThread_acquire_lock(netdb_lock, 1);
189#define RELEASE_GETADDRINFO_LOCK PyThread_release_lock(netdb_lock);
190#else
191#define ACQUIRE_GETADDRINFO_LOCK
192#define RELEASE_GETADDRINFO_LOCK
193#endif
194
195#if defined(USE_GETHOSTBYNAME_LOCK) || defined(USE_GETADDRINFO_LOCK)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000196# include "pythread.h"
Guido van Rossum4f199ea1998-04-09 20:56:35 +0000197#endif
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000198
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000199#if defined(PYCC_VACPP)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000200# include <types.h>
201# include <io.h>
202# include <sys/ioctl.h>
203# include <utils.h>
204# include <ctype.h>
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000205#endif
206
Martin v. Löwis9e437302002-12-06 12:57:26 +0000207#if defined(__VMS)
Martin v. Löwis79acb9e2002-12-06 12:48:53 +0000208# include <ioctl.h>
209#endif
210
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000211#if defined(PYOS_OS2)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000212# define INCL_DOS
213# define INCL_DOSERRORS
214# define INCL_NOPMAPI
215# include <os2.h>
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000216#endif
217
Jeremy Hyltonfb509a32003-07-17 16:58:48 +0000218#if defined(__sgi) && _COMPILER_VERSION>700 && !_SGIAPI
Martin v. Löwis8eb92a02002-09-19 08:03:21 +0000219/* make sure that the reentrant (gethostbyaddr_r etc)
220 functions are declared correctly if compiling with
221 MIPSPro 7.x in ANSI C mode (default) */
Jeremy Hyltonfb509a32003-07-17 16:58:48 +0000222
Thomas Wouters477c8d52006-05-27 19:21:47 +0000223/* XXX Using _SGIAPI is the wrong thing,
Jeremy Hyltonfb509a32003-07-17 16:58:48 +0000224 but I don't know what the right thing is. */
Trent Mick8ea5bdf2004-09-13 17:48:41 +0000225#undef _SGIAPI /* to avoid warning */
Martin v. Löwis8eb92a02002-09-19 08:03:21 +0000226#define _SGIAPI 1
Jeremy Hyltonfb509a32003-07-17 16:58:48 +0000227
Trent Mick8ea5bdf2004-09-13 17:48:41 +0000228#undef _XOPEN_SOURCE
229#include <sys/socket.h>
230#include <sys/types.h>
231#include <netinet/in.h>
232#ifdef _SS_ALIGNSIZE
233#define HAVE_GETADDRINFO 1
234#define HAVE_GETNAMEINFO 1
235#endif
236
Jeremy Hyltonfb509a32003-07-17 16:58:48 +0000237#define HAVE_INET_PTON
238#include <netdb.h>
Martin v. Löwis8eb92a02002-09-19 08:03:21 +0000239#endif
240
Thomas Wouters477c8d52006-05-27 19:21:47 +0000241/* Irix 6.5 fails to define this variable at all. This is needed
242 for both GCC and SGI's compiler. I'd say that the SGI headers
Georg Brandldbd83392006-02-20 09:42:33 +0000243 are just busted. Same thing for Solaris. */
244#if (defined(__sgi) || defined(sun)) && !defined(INET_ADDRSTRLEN)
Anthony Baxterbab23cf2003-10-04 08:00:49 +0000245#define INET_ADDRSTRLEN 16
246#endif
247
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000248/* Generic includes */
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000249#ifdef HAVE_SYS_TYPES_H
Guido van Rossumb6775db1994-08-01 11:34:53 +0000250#include <sys/types.h>
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000251#endif
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000252
Marc-André Lemburg976ade62002-02-16 18:47:07 +0000253/* Generic socket object definitions and includes */
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000254#define PySocket_BUILDING_SOCKET
Marc-André Lemburgbb8b78b2002-02-16 18:44:52 +0000255#include "socketmodule.h"
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000256
257/* Addressing includes */
258
Guido van Rossum6f489d91996-06-28 20:15:15 +0000259#ifndef MS_WINDOWS
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000260
261/* Non-MS WINDOWS includes */
262# include <netdb.h>
Guido van Rossum5c9eb211999-08-20 18:21:51 +0000263
Guido van Rossum9376b741999-09-15 22:01:40 +0000264/* Headers needed for inet_ntoa() and inet_addr() */
Skip Montanaroeb33e5a2007-08-17 12:57:41 +0000265# if defined(PYOS_OS2) && defined(PYCC_VACPP)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000266# include <netdb.h>
Tim Peters603c6832001-11-05 02:45:59 +0000267typedef size_t socklen_t;
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000268# else
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000269# include <arpa/inet.h>
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000270# endif
Guido van Rossum5c9eb211999-08-20 18:21:51 +0000271
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000272# include <fcntl.h>
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000273
Guido van Rossume2ae77b2001-10-24 20:42:55 +0000274#else
Guido van Rossum48a680c2001-03-02 06:34:14 +0000275
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000276/* MS_WINDOWS includes */
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000277# ifdef HAVE_FCNTL_H
278# include <fcntl.h>
279# endif
Guido van Rossum48a680c2001-03-02 06:34:14 +0000280
Jeremy Hylton22308652001-02-02 03:23:09 +0000281#endif
282
Skip Montanaro7befb992004-02-10 16:50:21 +0000283#include <stddef.h>
Martin v. Löwisa45ecae2001-06-24 21:28:42 +0000284
285#ifndef offsetof
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000286# define offsetof(type, member) ((size_t)(&((type *)0)->member))
Martin v. Löwisa45ecae2001-06-24 21:28:42 +0000287#endif
288
Neal Norwitz39d22e52002-11-02 19:55:21 +0000289#ifndef O_NONBLOCK
290# define O_NONBLOCK O_NDELAY
Guido van Rossum3bbc62e1995-01-02 19:30:30 +0000291#endif
292
Trent Micka708d6e2004-09-07 17:48:26 +0000293/* include Python's addrinfo.h unless it causes trouble */
294#if defined(__sgi) && _COMPILER_VERSION>700 && defined(_SS_ALIGNSIZE)
295 /* Do not include addinfo.h on some newer IRIX versions.
296 * _SS_ALIGNSIZE is defined in sys/socket.h by 6.5.21,
297 * for example, but not by 6.5.10.
298 */
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000299#elif defined(_MSC_VER) && _MSC_VER>1201
Trent Micka708d6e2004-09-07 17:48:26 +0000300 /* Do not include addrinfo.h for MSVC7 or greater. 'addrinfo' and
301 * EAI_* constants are defined in (the already included) ws2tcpip.h.
302 */
303#else
304# include "addrinfo.h"
305#endif
Jason Tishlerc246cb72004-08-09 13:25:59 +0000306
Martin v. Löwisb9ab1592001-06-24 21:18:26 +0000307#ifndef HAVE_INET_PTON
Christian Heimes96e7b3d2007-11-20 06:51:17 +0000308#if !defined(NTDDI_VERSION) || (NTDDI_VERSION < NTDDI_LONGHORN)
Guido van Rossum3eede5a2002-06-07 02:08:35 +0000309int inet_pton(int af, const char *src, void *dst);
Martin v. Löwisc925b1532001-07-21 09:42:15 +0000310const char *inet_ntop(int af, const void *src, char *dst, socklen_t size);
Martin v. Löwisb9ab1592001-06-24 21:18:26 +0000311#endif
Christian Heimesb6150692007-11-15 23:37:07 +0000312#endif
Martin v. Löwisb9ab1592001-06-24 21:18:26 +0000313
Martin v. Löwisae26dc22001-08-03 10:02:29 +0000314#ifdef __APPLE__
315/* On OS X, getaddrinfo returns no error indication of lookup
316 failure, so we must use the emulation instead of the libinfo
317 implementation. Unfortunately, performing an autoconf test
318 for this bug would require DNS access for the machine performing
319 the configuration, which is not acceptable. Therefore, we
320 determine the bug just by checking for __APPLE__. If this bug
321 gets ever fixed, perhaps checking for sys/version.h would be
322 appropriate, which is 10/0 on the system with the bug. */
Jack Jansen84262fb2002-07-02 14:40:42 +0000323#ifndef HAVE_GETNAMEINFO
324/* This bug seems to be fixed in Jaguar. Ths easiest way I could
325 Find to check for Jaguar is that it has getnameinfo(), which
326 older releases don't have */
Martin v. Löwisae26dc22001-08-03 10:02:29 +0000327#undef HAVE_GETADDRINFO
Martin v. Löwisae26dc22001-08-03 10:02:29 +0000328#endif
Thomas Wouters477c8d52006-05-27 19:21:47 +0000329
330#ifdef HAVE_INET_ATON
331#define USE_INET_ATON_WEAKLINK
332#endif
333
Jack Jansen84262fb2002-07-02 14:40:42 +0000334#endif
Martin v. Löwisae26dc22001-08-03 10:02:29 +0000335
Martin v. Löwis01dfdb32001-06-23 16:30:13 +0000336/* I know this is a bad practice, but it is the easiest... */
Martin v. Löwisae26dc22001-08-03 10:02:29 +0000337#if !defined(HAVE_GETADDRINFO)
Martin v. Löwisfccac2e2003-05-01 05:20:46 +0000338/* avoid clashes with the C library definition of the symbol. */
339#define getaddrinfo fake_getaddrinfo
340#define gai_strerror fake_gai_strerror
341#define freeaddrinfo fake_freeaddrinfo
Martin v. Löwis01dfdb32001-06-23 16:30:13 +0000342#include "getaddrinfo.c"
343#endif
Martin v. Löwisae26dc22001-08-03 10:02:29 +0000344#if !defined(HAVE_GETNAMEINFO)
Martin v. Löwisfccac2e2003-05-01 05:20:46 +0000345#define getnameinfo fake_getnameinfo
Martin v. Löwis01dfdb32001-06-23 16:30:13 +0000346#include "getnameinfo.c"
347#endif
348
Guido van Rossum39eb8fa2007-11-16 01:24:05 +0000349#ifdef MS_WINDOWS
350/* On Windows a socket is really a handle not an fd */
351static SOCKET
352dup_socket(SOCKET handle)
353{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000354 HANDLE newhandle;
Guido van Rossum39eb8fa2007-11-16 01:24:05 +0000355
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000356 if (!DuplicateHandle(GetCurrentProcess(), (HANDLE)handle,
357 GetCurrentProcess(), &newhandle,
358 0, FALSE, DUPLICATE_SAME_ACCESS))
359 {
360 WSASetLastError(GetLastError());
361 return INVALID_SOCKET;
362 }
363 return (SOCKET)newhandle;
Guido van Rossum39eb8fa2007-11-16 01:24:05 +0000364}
Guido van Rossum2dd8ddd2000-04-21 20:33:00 +0000365#define SOCKETCLOSE closesocket
Guido van Rossum39eb8fa2007-11-16 01:24:05 +0000366#else
367/* On Unix we can use dup to duplicate the file descriptor of a socket*/
368#define dup_socket(fd) dup(fd)
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +0000369#endif
370
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000371#ifdef MS_WIN32
Guido van Rossum3eede5a2002-06-07 02:08:35 +0000372#define EAFNOSUPPORT WSAEAFNOSUPPORT
373#define snprintf _snprintf
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000374#endif
Fred Drakea04eaad2000-06-30 02:46:07 +0000375
Andrew MacIntyreba43e872002-03-03 03:03:52 +0000376#if defined(PYOS_OS2) && !defined(PYCC_GCC)
Guido van Rossum2dd8ddd2000-04-21 20:33:00 +0000377#define SOCKETCLOSE soclose
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000378#define NO_DUP /* Sockets are Not Actual File Handles under OS/2 */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000379#endif
380
Guido van Rossum2dd8ddd2000-04-21 20:33:00 +0000381#ifndef SOCKETCLOSE
382#define SOCKETCLOSE close
383#endif
384
Jesse Noller32d68c22009-03-31 18:48:42 +0000385#if defined(HAVE_BLUETOOTH_H) || defined(HAVE_BLUETOOTH_BLUETOOTH_H) && !defined(__NetBSD__)
Hye-Shik Chang81268602004-02-02 06:05:24 +0000386#define USE_BLUETOOTH 1
387#if defined(__FreeBSD__)
388#define BTPROTO_L2CAP BLUETOOTH_PROTO_L2CAP
389#define BTPROTO_RFCOMM BLUETOOTH_PROTO_RFCOMM
Thomas Wouterscf297e42007-02-23 15:07:44 +0000390#define BTPROTO_HCI BLUETOOTH_PROTO_HCI
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000391#define SOL_HCI SOL_HCI_RAW
392#define HCI_FILTER SO_HCI_RAW_FILTER
Hye-Shik Chang81268602004-02-02 06:05:24 +0000393#define sockaddr_l2 sockaddr_l2cap
394#define sockaddr_rc sockaddr_rfcomm
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000395#define hci_dev hci_node
Hye-Shik Chang81268602004-02-02 06:05:24 +0000396#define _BT_L2_MEMB(sa, memb) ((sa)->l2cap_##memb)
397#define _BT_RC_MEMB(sa, memb) ((sa)->rfcomm_##memb)
Thomas Wouterscf297e42007-02-23 15:07:44 +0000398#define _BT_HCI_MEMB(sa, memb) ((sa)->hci_##memb)
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000399#elif defined(__NetBSD__)
400#define sockaddr_l2 sockaddr_bt
401#define sockaddr_rc sockaddr_bt
Thomas Wouterscf297e42007-02-23 15:07:44 +0000402#define sockaddr_hci sockaddr_bt
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000403#define sockaddr_sco sockaddr_bt
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000404#define _BT_L2_MEMB(sa, memb) ((sa)->bt_##memb)
405#define _BT_RC_MEMB(sa, memb) ((sa)->bt_##memb)
Thomas Wouterscf297e42007-02-23 15:07:44 +0000406#define _BT_HCI_MEMB(sa, memb) ((sa)->bt_##memb)
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000407#define _BT_SCO_MEMB(sa, memb) ((sa)->bt_##memb)
Hye-Shik Chang81268602004-02-02 06:05:24 +0000408#else
Hye-Shik Chang81268602004-02-02 06:05:24 +0000409#define _BT_L2_MEMB(sa, memb) ((sa)->l2_##memb)
410#define _BT_RC_MEMB(sa, memb) ((sa)->rc_##memb)
Thomas Wouterscf297e42007-02-23 15:07:44 +0000411#define _BT_HCI_MEMB(sa, memb) ((sa)->hci_##memb)
Hye-Shik Chang81268602004-02-02 06:05:24 +0000412#define _BT_SCO_MEMB(sa, memb) ((sa)->sco_##memb)
413#endif
414#endif
415
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000416#ifdef __VMS
417/* TCP/IP Services for VMS uses a maximum send/recv buffer length */
418#define SEGMENT_SIZE (32 * 1024 -1)
419#endif
420
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000421#define SAS2SA(x) ((struct sockaddr *)(x))
Thomas Wouters89f507f2006-12-13 04:49:30 +0000422
Martin v. Löwise9416172003-05-03 10:12:45 +0000423/*
424 * Constants for getnameinfo()
425 */
426#if !defined(NI_MAXHOST)
427#define NI_MAXHOST 1025
428#endif
429#if !defined(NI_MAXSERV)
430#define NI_MAXSERV 32
431#endif
432
Guido van Rossum7d0a8262007-05-21 23:13:11 +0000433#ifndef INVALID_SOCKET /* MS defines this */
434#define INVALID_SOCKET (-1)
435#endif
436
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000437/* XXX There's a problem here: *static* functions are not supposed to have
438 a Py prefix (or use CapitalizedWords). Later... */
439
Guido van Rossum30a685f1991-06-27 15:51:29 +0000440/* Global variable holding the exception type for errors detected
441 by this module (but not argument type or memory errors, etc.). */
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000442static PyObject *socket_error;
443static PyObject *socket_herror;
444static PyObject *socket_gaierror;
Raymond Hettingeref7343c2003-06-29 03:08:05 +0000445static PyObject *socket_timeout;
Guido van Rossum30a685f1991-06-27 15:51:29 +0000446
Tim Peters643a7fc2002-02-17 04:13:21 +0000447/* A forward reference to the socket type object.
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000448 The sock_type variable contains pointers to various functions,
449 some of which call new_sockobject(), which uses sock_type, so
Tim Peters643a7fc2002-02-17 04:13:21 +0000450 there has to be a circular reference. */
Jeremy Hylton938ace62002-07-17 16:30:39 +0000451static PyTypeObject sock_type;
Guido van Rossum48a680c2001-03-02 06:34:14 +0000452
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000453#if defined(HAVE_POLL_H)
454#include <poll.h>
455#elif defined(HAVE_SYS_POLL_H)
456#include <sys/poll.h>
457#endif
458
Martin v. Löwisf84d1b92006-02-11 09:27:05 +0000459#ifdef Py_SOCKET_FD_CAN_BE_GE_FD_SETSIZE
460/* Platform can select file descriptors beyond FD_SETSIZE */
461#define IS_SELECTABLE(s) 1
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000462#elif defined(HAVE_POLL)
463/* Instead of select(), we'll use poll() since poll() works on any fd. */
464#define IS_SELECTABLE(s) 1
465/* Can we call select() with this socket without a buffer overrun? */
Martin v. Löwisf84d1b92006-02-11 09:27:05 +0000466#else
467/* POSIX says selecting file descriptors beyond FD_SETSIZE
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000468 has undefined behaviour. If there's no timeout left, we don't have to
469 call select, so it's a safe, little white lie. */
470#define IS_SELECTABLE(s) ((s)->sock_fd < FD_SETSIZE || s->sock_timeout <= 0.0)
Martin v. Löwisf84d1b92006-02-11 09:27:05 +0000471#endif
Neal Norwitz082b2df2006-02-07 07:04:46 +0000472
473static PyObject*
474select_error(void)
475{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000476 PyErr_SetString(socket_error, "unable to select on socket");
477 return NULL;
Neal Norwitz082b2df2006-02-07 07:04:46 +0000478}
479
Guido van Rossum30a685f1991-06-27 15:51:29 +0000480/* Convenience function to raise an error according to errno
481 and return a NULL pointer from a function. */
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000482
Guido van Rossum73624e91994-10-10 17:59:00 +0000483static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000484set_error(void)
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000485{
Guido van Rossum8d665e61996-06-26 18:22:49 +0000486#ifdef MS_WINDOWS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000487 int err_no = WSAGetLastError();
488 /* PyErr_SetExcFromWindowsErr() invokes FormatMessage() which
489 recognizes the error codes used by both GetLastError() and
490 WSAGetLastError */
491 if (err_no)
492 return PyErr_SetExcFromWindowsErr(socket_error, err_no);
Guido van Rossum14ed0b21994-09-29 09:50:09 +0000493#endif
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000494
Andrew MacIntyreba43e872002-03-03 03:03:52 +0000495#if defined(PYOS_OS2) && !defined(PYCC_GCC)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000496 if (sock_errno() != NO_ERROR) {
497 APIRET rc;
498 ULONG msglen;
499 char outbuf[100];
500 int myerrorcode = sock_errno();
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000501
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000502 /* Retrieve socket-related error message from MPTN.MSG file */
503 rc = DosGetMessage(NULL, 0, outbuf, sizeof(outbuf),
504 myerrorcode - SOCBASEERR + 26,
505 "mptn.msg",
506 &msglen);
507 if (rc == NO_ERROR) {
508 PyObject *v;
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000509
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000510 /* OS/2 doesn't guarantee a terminator */
511 outbuf[msglen] = '\0';
512 if (strlen(outbuf) > 0) {
513 /* If non-empty msg, trim CRLF */
514 char *lastc = &outbuf[ strlen(outbuf)-1 ];
515 while (lastc > outbuf &&
516 isspace(Py_CHARMASK(*lastc))) {
517 /* Trim trailing whitespace (CRLF) */
518 *lastc-- = '\0';
519 }
520 }
521 v = Py_BuildValue("(is)", myerrorcode, outbuf);
522 if (v != NULL) {
523 PyErr_SetObject(socket_error, v);
524 Py_DECREF(v);
525 }
526 return NULL;
527 }
528 }
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000529#endif
530
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000531 return PyErr_SetFromErrno(socket_error);
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000532}
533
Guido van Rossum30a685f1991-06-27 15:51:29 +0000534
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000535static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000536set_herror(int h_error)
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000537{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000538 PyObject *v;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000539
540#ifdef HAVE_HSTRERROR
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000541 v = Py_BuildValue("(is)", h_error, (char *)hstrerror(h_error));
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000542#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000543 v = Py_BuildValue("(is)", h_error, "host not found");
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000544#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000545 if (v != NULL) {
546 PyErr_SetObject(socket_herror, v);
547 Py_DECREF(v);
548 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000549
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000550 return NULL;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000551}
552
553
554static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000555set_gaierror(int error)
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000556{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000557 PyObject *v;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000558
Martin v. Löwis272cb402002-03-01 08:31:07 +0000559#ifdef EAI_SYSTEM
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000560 /* EAI_SYSTEM is not available on Windows XP. */
561 if (error == EAI_SYSTEM)
562 return set_error();
Martin v. Löwis272cb402002-03-01 08:31:07 +0000563#endif
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000564
Martin v. Löwisf95dd0a2001-08-15 17:14:33 +0000565#ifdef HAVE_GAI_STRERROR
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000566 v = Py_BuildValue("(is)", error, gai_strerror(error));
Martin v. Löwisf95dd0a2001-08-15 17:14:33 +0000567#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000568 v = Py_BuildValue("(is)", error, "getaddrinfo failed");
Martin v. Löwisf95dd0a2001-08-15 17:14:33 +0000569#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000570 if (v != NULL) {
571 PyErr_SetObject(socket_gaierror, v);
572 Py_DECREF(v);
573 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000574
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000575 return NULL;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000576}
577
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000578#ifdef __VMS
579/* Function to send in segments */
580static int
581sendsegmented(int sock_fd, char *buf, int len, int flags)
582{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000583 int n = 0;
584 int remaining = len;
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000585
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000586 while (remaining > 0) {
587 unsigned int segment;
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000588
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000589 segment = (remaining >= SEGMENT_SIZE ? SEGMENT_SIZE : remaining);
590 n = send(sock_fd, buf, segment, flags);
591 if (n < 0) {
592 return n;
593 }
594 remaining -= segment;
595 buf += segment;
596 } /* end while */
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000597
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000598 return len;
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000599}
600#endif
601
Guido van Rossum3eede5a2002-06-07 02:08:35 +0000602/* Function to perform the setting of socket blocking mode
603 internally. block = (1 | 0). */
Guido van Rossum67f7a382002-06-06 21:08:16 +0000604static int
605internal_setblocking(PySocketSockObject *s, int block)
606{
Guido van Rossum67f7a382002-06-06 21:08:16 +0000607#ifndef MS_WINDOWS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000608 int delay_flag;
Guido van Rossum67f7a382002-06-06 21:08:16 +0000609#endif
Guido van Rossum67f7a382002-06-06 21:08:16 +0000610
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000611 Py_BEGIN_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +0000612#ifndef MS_WINDOWS
613#if defined(PYOS_OS2) && !defined(PYCC_GCC)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000614 block = !block;
615 ioctl(s->sock_fd, FIONBIO, (caddr_t)&block, sizeof(block));
Martin v. Löwis79acb9e2002-12-06 12:48:53 +0000616#elif defined(__VMS)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000617 block = !block;
618 ioctl(s->sock_fd, FIONBIO, (unsigned int *)&block);
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000619#else /* !PYOS_OS2 && !__VMS */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000620 delay_flag = fcntl(s->sock_fd, F_GETFL, 0);
621 if (block)
622 delay_flag &= (~O_NONBLOCK);
623 else
624 delay_flag |= O_NONBLOCK;
625 fcntl(s->sock_fd, F_SETFL, delay_flag);
Guido van Rossum67f7a382002-06-06 21:08:16 +0000626#endif /* !PYOS_OS2 */
627#else /* MS_WINDOWS */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000628 block = !block;
629 ioctlsocket(s->sock_fd, FIONBIO, (u_long*)&block);
Guido van Rossum67f7a382002-06-06 21:08:16 +0000630#endif /* MS_WINDOWS */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000631 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +0000632
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000633 /* Since these don't return anything */
634 return 1;
Guido van Rossum67f7a382002-06-06 21:08:16 +0000635}
636
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000637/* Do a select()/poll() on the socket, if necessary (sock_timeout > 0).
Guido van Rossum11ba0942002-06-13 15:07:44 +0000638 The argument writing indicates the direction.
Raymond Hettingeref7343c2003-06-29 03:08:05 +0000639 This does not raise an exception; we'll let our caller do that
640 after they've reacquired the interpreter lock.
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000641 Returns 1 on timeout, -1 on error, 0 otherwise. */
Raymond Hettingeref7343c2003-06-29 03:08:05 +0000642static int
Guido van Rossumb9e916a2002-06-07 01:42:47 +0000643internal_select(PySocketSockObject *s, int writing)
Guido van Rossum67f7a382002-06-06 21:08:16 +0000644{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000645 int n;
Guido van Rossum11ba0942002-06-13 15:07:44 +0000646
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000647 /* Nothing to do unless we're in timeout mode (not non-blocking) */
648 if (s->sock_timeout <= 0.0)
649 return 0;
Guido van Rossum67f7a382002-06-06 21:08:16 +0000650
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000651 /* Guard against closed socket */
652 if (s->sock_fd < 0)
653 return 0;
Guido van Rossumad654902002-07-19 12:44:59 +0000654
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000655 /* Prefer poll, if available, since you can poll() any fd
656 * which can't be done with select(). */
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000657#ifdef HAVE_POLL
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000658 {
659 struct pollfd pollfd;
660 int timeout;
Guido van Rossum67f7a382002-06-06 21:08:16 +0000661
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000662 pollfd.fd = s->sock_fd;
663 pollfd.events = writing ? POLLOUT : POLLIN;
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000664
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000665 /* s->sock_timeout is in seconds, timeout in ms */
666 timeout = (int)(s->sock_timeout * 1000 + 0.5);
667 n = poll(&pollfd, 1, timeout);
668 }
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000669#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000670 {
671 /* Construct the arguments to select */
672 fd_set fds;
673 struct timeval tv;
674 tv.tv_sec = (int)s->sock_timeout;
675 tv.tv_usec = (int)((s->sock_timeout - tv.tv_sec) * 1e6);
676 FD_ZERO(&fds);
677 FD_SET(s->sock_fd, &fds);
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000678
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000679 /* See if the socket is ready */
680 if (writing)
681 n = select(s->sock_fd+1, NULL, &fds, NULL, &tv);
682 else
683 n = select(s->sock_fd+1, &fds, NULL, NULL, &tv);
684 }
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000685#endif
Guido van Rossum39eb8fa2007-11-16 01:24:05 +0000686
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000687 if (n < 0)
688 return -1;
689 if (n == 0)
690 return 1;
691 return 0;
Guido van Rossum67f7a382002-06-06 21:08:16 +0000692}
693
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000694/* Initialize a new socket object. */
695
Tim Petersa12b4cf2002-07-18 22:38:44 +0000696static double defaulttimeout = -1.0; /* Default timeout for new sockets */
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +0000697
Martin v. Löwis1a214512008-06-11 05:26:20 +0000698static void
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000699init_sockobject(PySocketSockObject *s,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000700 SOCKET_T fd, int family, int type, int proto)
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000701{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000702 s->sock_fd = fd;
703 s->sock_family = family;
704 s->sock_type = type;
705 s->sock_proto = proto;
706 s->sock_timeout = defaulttimeout;
Guido van Rossum67f7a382002-06-06 21:08:16 +0000707
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000708 s->errorhandler = &set_error;
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +0000709
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000710 if (defaulttimeout >= 0.0)
711 internal_setblocking(s, 0);
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +0000712
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000713}
714
715
Guido van Rossum30a685f1991-06-27 15:51:29 +0000716/* Create a new socket object.
717 This just creates the object and initializes it.
718 If the creation fails, return NULL and set an exception (implicit
719 in NEWOBJ()). */
720
Guido van Rossum73624e91994-10-10 17:59:00 +0000721static PySocketSockObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000722new_sockobject(SOCKET_T fd, int family, int type, int proto)
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000723{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000724 PySocketSockObject *s;
725 s = (PySocketSockObject *)
726 PyType_GenericNew(&sock_type, NULL, NULL);
727 if (s != NULL)
728 init_sockobject(s, fd, family, type, proto);
729 return s;
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000730}
731
Guido van Rossum30a685f1991-06-27 15:51:29 +0000732
Guido van Rossum48a680c2001-03-02 06:34:14 +0000733/* Lock to allow python interpreter to continue, but only allow one
Just van Rossum1040d2c2003-05-09 07:53:18 +0000734 thread to be in gethostbyname or getaddrinfo */
735#if defined(USE_GETHOSTBYNAME_LOCK) || defined(USE_GETADDRINFO_LOCK)
736PyThread_type_lock netdb_lock;
Guido van Rossum4f199ea1998-04-09 20:56:35 +0000737#endif
738
739
Guido van Rossum30a685f1991-06-27 15:51:29 +0000740/* Convert a string specifying a host name or one of a few symbolic
741 names to a numeric IP address. This usually calls gethostbyname()
742 to do the work; the names "" and "<broadcast>" are special.
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000743 Return the length (IPv4 should be 4 bytes), or negative if
Guido van Rossum30a685f1991-06-27 15:51:29 +0000744 an error occurred; then an exception is raised. */
745
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000746static int
Martin v. Löwisddc6f472002-07-28 16:10:31 +0000747setipaddr(char *name, struct sockaddr *addr_ret, size_t addr_ret_size, int af)
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000748{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000749 struct addrinfo hints, *res;
750 int error;
751 int d1, d2, d3, d4;
752 char ch;
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000753
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000754 memset((void *) addr_ret, '\0', sizeof(*addr_ret));
755 if (name[0] == '\0') {
756 int siz;
757 memset(&hints, 0, sizeof(hints));
758 hints.ai_family = af;
759 hints.ai_socktype = SOCK_DGRAM; /*dummy*/
760 hints.ai_flags = AI_PASSIVE;
761 Py_BEGIN_ALLOW_THREADS
762 ACQUIRE_GETADDRINFO_LOCK
763 error = getaddrinfo(NULL, "0", &hints, &res);
764 Py_END_ALLOW_THREADS
765 /* We assume that those thread-unsafe getaddrinfo() versions
766 *are* safe regarding their return value, ie. that a
767 subsequent call to getaddrinfo() does not destroy the
768 outcome of the first call. */
769 RELEASE_GETADDRINFO_LOCK
770 if (error) {
771 set_gaierror(error);
772 return -1;
773 }
774 switch (res->ai_family) {
775 case AF_INET:
776 siz = 4;
777 break;
Martin v. Löwis44ddbde2001-12-02 10:15:37 +0000778#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000779 case AF_INET6:
780 siz = 16;
781 break;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000782#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000783 default:
784 freeaddrinfo(res);
785 PyErr_SetString(socket_error,
786 "unsupported address family");
787 return -1;
788 }
789 if (res->ai_next) {
790 freeaddrinfo(res);
791 PyErr_SetString(socket_error,
792 "wildcard resolved to multiple address");
793 return -1;
794 }
795 if (res->ai_addrlen < addr_ret_size)
796 addr_ret_size = res->ai_addrlen;
797 memcpy(addr_ret, res->ai_addr, addr_ret_size);
798 freeaddrinfo(res);
799 return siz;
800 }
801 if (name[0] == '<' && strcmp(name, "<broadcast>") == 0) {
802 struct sockaddr_in *sin;
803 if (af != AF_INET && af != AF_UNSPEC) {
804 PyErr_SetString(socket_error,
805 "address family mismatched");
806 return -1;
807 }
808 sin = (struct sockaddr_in *)addr_ret;
809 memset((void *) sin, '\0', sizeof(*sin));
810 sin->sin_family = AF_INET;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000811#ifdef HAVE_SOCKADDR_SA_LEN
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000812 sin->sin_len = sizeof(*sin);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000813#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000814 sin->sin_addr.s_addr = INADDR_BROADCAST;
815 return sizeof(sin->sin_addr);
816 }
817 if (sscanf(name, "%d.%d.%d.%d%c", &d1, &d2, &d3, &d4, &ch) == 4 &&
818 0 <= d1 && d1 <= 255 && 0 <= d2 && d2 <= 255 &&
819 0 <= d3 && d3 <= 255 && 0 <= d4 && d4 <= 255) {
820 struct sockaddr_in *sin;
821 sin = (struct sockaddr_in *)addr_ret;
822 sin->sin_addr.s_addr = htonl(
823 ((long) d1 << 24) | ((long) d2 << 16) |
824 ((long) d3 << 8) | ((long) d4 << 0));
825 sin->sin_family = AF_INET;
Anthony Baxter0e85f9d2003-05-02 15:40:46 +0000826#ifdef HAVE_SOCKADDR_SA_LEN
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000827 sin->sin_len = sizeof(*sin);
Anthony Baxter0e85f9d2003-05-02 15:40:46 +0000828#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000829 return 4;
830 }
831 memset(&hints, 0, sizeof(hints));
832 hints.ai_family = af;
833 Py_BEGIN_ALLOW_THREADS
834 ACQUIRE_GETADDRINFO_LOCK
835 error = getaddrinfo(name, NULL, &hints, &res);
Martin v. Löwis7c4b5fa2001-10-25 09:04:03 +0000836#if defined(__digital__) && defined(__unix__)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000837 if (error == EAI_NONAME && af == AF_UNSPEC) {
838 /* On Tru64 V5.1, numeric-to-addr conversion fails
839 if no address family is given. Assume IPv4 for now.*/
840 hints.ai_family = AF_INET;
841 error = getaddrinfo(name, NULL, &hints, &res);
842 }
Martin v. Löwis7c4b5fa2001-10-25 09:04:03 +0000843#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000844 Py_END_ALLOW_THREADS
845 RELEASE_GETADDRINFO_LOCK /* see comment in setipaddr() */
846 if (error) {
847 set_gaierror(error);
848 return -1;
849 }
850 if (res->ai_addrlen < addr_ret_size)
851 addr_ret_size = res->ai_addrlen;
852 memcpy((char *) addr_ret, res->ai_addr, addr_ret_size);
853 freeaddrinfo(res);
854 switch (addr_ret->sa_family) {
855 case AF_INET:
856 return 4;
Martin v. Löwis44ddbde2001-12-02 10:15:37 +0000857#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000858 case AF_INET6:
859 return 16;
Guido van Rossum955becc1999-03-22 20:14:53 +0000860#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000861 default:
862 PyErr_SetString(socket_error, "unknown address family");
863 return -1;
864 }
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000865}
866
Guido van Rossum30a685f1991-06-27 15:51:29 +0000867
Guido van Rossum30a685f1991-06-27 15:51:29 +0000868/* Create a string object representing an IP address.
869 This is always a string of the form 'dd.dd.dd.dd' (with variable
870 size numbers). */
871
Guido van Rossum73624e91994-10-10 17:59:00 +0000872static PyObject *
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000873makeipaddr(struct sockaddr *addr, int addrlen)
Guido van Rossum30a685f1991-06-27 15:51:29 +0000874{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000875 char buf[NI_MAXHOST];
876 int error;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000877
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000878 error = getnameinfo(addr, addrlen, buf, sizeof(buf), NULL, 0,
879 NI_NUMERICHOST);
880 if (error) {
881 set_gaierror(error);
882 return NULL;
883 }
884 return PyUnicode_FromString(buf);
Guido van Rossum30a685f1991-06-27 15:51:29 +0000885}
886
887
Martin v. Löwis558d9bf2004-06-03 09:24:42 +0000888#ifdef USE_BLUETOOTH
889/* Convert a string representation of a Bluetooth address into a numeric
890 address. Returns the length (6), or raises an exception and returns -1 if
891 an error occurred. */
892
893static int
894setbdaddr(char *name, bdaddr_t *bdaddr)
895{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000896 unsigned int b0, b1, b2, b3, b4, b5;
897 char ch;
898 int n;
Martin v. Löwis558d9bf2004-06-03 09:24:42 +0000899
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000900 n = sscanf(name, "%X:%X:%X:%X:%X:%X%c",
901 &b5, &b4, &b3, &b2, &b1, &b0, &ch);
902 if (n == 6 && (b0 | b1 | b2 | b3 | b4 | b5) < 256) {
903 bdaddr->b[0] = b0;
904 bdaddr->b[1] = b1;
905 bdaddr->b[2] = b2;
906 bdaddr->b[3] = b3;
907 bdaddr->b[4] = b4;
908 bdaddr->b[5] = b5;
909 return 6;
910 } else {
911 PyErr_SetString(socket_error, "bad bluetooth address");
912 return -1;
913 }
Martin v. Löwis558d9bf2004-06-03 09:24:42 +0000914}
915
916/* Create a string representation of the Bluetooth address. This is always a
917 string of the form 'XX:XX:XX:XX:XX:XX' where XX is a two digit hexadecimal
918 value (zero padded if necessary). */
919
920static PyObject *
921makebdaddr(bdaddr_t *bdaddr)
922{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000923 char buf[(6 * 2) + 5 + 1];
Martin v. Löwis558d9bf2004-06-03 09:24:42 +0000924
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000925 sprintf(buf, "%02X:%02X:%02X:%02X:%02X:%02X",
926 bdaddr->b[5], bdaddr->b[4], bdaddr->b[3],
927 bdaddr->b[2], bdaddr->b[1], bdaddr->b[0]);
928 return PyUnicode_FromString(buf);
Martin v. Löwis558d9bf2004-06-03 09:24:42 +0000929}
930#endif
931
932
Guido van Rossum30a685f1991-06-27 15:51:29 +0000933/* Create an object representing the given socket address,
934 suitable for passing it back to bind(), connect() etc.
935 The family field of the sockaddr structure is inspected
936 to determine what kind of address it really is. */
937
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000938/*ARGSUSED*/
Guido van Rossum73624e91994-10-10 17:59:00 +0000939static PyObject *
Martin v. Löwis558d9bf2004-06-03 09:24:42 +0000940makesockaddr(int sockfd, struct sockaddr *addr, int addrlen, int proto)
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000941{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000942 if (addrlen == 0) {
943 /* No address -- may be recvfrom() from known socket */
944 Py_INCREF(Py_None);
945 return Py_None;
946 }
Guido van Rossum25bec8c1992-08-05 19:00:45 +0000947
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000948 switch (addr->sa_family) {
Guido van Rossum30a685f1991-06-27 15:51:29 +0000949
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000950 case AF_INET:
951 {
952 struct sockaddr_in *a;
953 PyObject *addrobj = makeipaddr(addr, sizeof(*a));
954 PyObject *ret = NULL;
955 if (addrobj) {
956 a = (struct sockaddr_in *)addr;
957 ret = Py_BuildValue("Oi", addrobj, ntohs(a->sin_port));
958 Py_DECREF(addrobj);
959 }
960 return ret;
961 }
Guido van Rossum30a685f1991-06-27 15:51:29 +0000962
Andrew MacIntyred12dfbb2004-04-04 07:13:49 +0000963#if defined(AF_UNIX)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000964 case AF_UNIX:
965 {
966 struct sockaddr_un *a = (struct sockaddr_un *) addr;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000967#ifdef linux
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000968 if (a->sun_path[0] == 0) { /* Linux abstract namespace */
969 addrlen -= offsetof(struct sockaddr_un, sun_path);
970 return PyBytes_FromStringAndSize(a->sun_path, addrlen);
971 }
972 else
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000973#endif /* linux */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000974 {
975 /* regular NULL-terminated string */
976 return PyUnicode_FromString(a->sun_path);
977 }
978 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000979#endif /* AF_UNIX */
980
Martin v. Löwis11017b12006-01-14 18:12:57 +0000981#if defined(AF_NETLINK)
982 case AF_NETLINK:
983 {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000984 struct sockaddr_nl *a = (struct sockaddr_nl *) addr;
985 return Py_BuildValue("II", a->nl_pid, a->nl_groups);
Martin v. Löwis11017b12006-01-14 18:12:57 +0000986 }
987#endif /* AF_NETLINK */
988
Martin v. Löwis44ddbde2001-12-02 10:15:37 +0000989#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000990 case AF_INET6:
991 {
992 struct sockaddr_in6 *a;
993 PyObject *addrobj = makeipaddr(addr, sizeof(*a));
994 PyObject *ret = NULL;
995 if (addrobj) {
996 a = (struct sockaddr_in6 *)addr;
997 ret = Py_BuildValue("Oiii",
998 addrobj,
999 ntohs(a->sin6_port),
1000 a->sin6_flowinfo,
1001 a->sin6_scope_id);
1002 Py_DECREF(addrobj);
1003 }
1004 return ret;
1005 }
Jeremy Hylton22308652001-02-02 03:23:09 +00001006#endif
Guido van Rossum30a685f1991-06-27 15:51:29 +00001007
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001008#ifdef USE_BLUETOOTH
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001009 case AF_BLUETOOTH:
1010 switch (proto) {
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001011
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001012 case BTPROTO_L2CAP:
1013 {
1014 struct sockaddr_l2 *a = (struct sockaddr_l2 *) addr;
1015 PyObject *addrobj = makebdaddr(&_BT_L2_MEMB(a, bdaddr));
1016 PyObject *ret = NULL;
1017 if (addrobj) {
1018 ret = Py_BuildValue("Oi",
1019 addrobj,
1020 _BT_L2_MEMB(a, psm));
1021 Py_DECREF(addrobj);
1022 }
1023 return ret;
1024 }
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001025
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001026 case BTPROTO_RFCOMM:
1027 {
1028 struct sockaddr_rc *a = (struct sockaddr_rc *) addr;
1029 PyObject *addrobj = makebdaddr(&_BT_RC_MEMB(a, bdaddr));
1030 PyObject *ret = NULL;
1031 if (addrobj) {
1032 ret = Py_BuildValue("Oi",
1033 addrobj,
1034 _BT_RC_MEMB(a, channel));
1035 Py_DECREF(addrobj);
1036 }
1037 return ret;
1038 }
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001039
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001040 case BTPROTO_HCI:
1041 {
1042 struct sockaddr_hci *a = (struct sockaddr_hci *) addr;
1043 PyObject *ret = NULL;
1044 ret = Py_BuildValue("i", _BT_HCI_MEMB(a, dev));
1045 return ret;
1046 }
Thomas Wouterscf297e42007-02-23 15:07:44 +00001047
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001048#if !defined(__FreeBSD__)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001049 case BTPROTO_SCO:
1050 {
1051 struct sockaddr_sco *a = (struct sockaddr_sco *) addr;
1052 return makebdaddr(&_BT_SCO_MEMB(a, bdaddr));
1053 }
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001054#endif
1055
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001056 default:
1057 PyErr_SetString(PyExc_ValueError,
1058 "Unknown Bluetooth protocol");
1059 return NULL;
1060 }
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001061#endif
1062
Martin v. Löwis1ba3fd52001-08-10 20:29:40 +00001063#ifdef HAVE_NETPACKET_PACKET_H
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001064 case AF_PACKET:
1065 {
1066 struct sockaddr_ll *a = (struct sockaddr_ll *)addr;
1067 char *ifname = "";
1068 struct ifreq ifr;
1069 /* need to look up interface name give index */
1070 if (a->sll_ifindex) {
1071 ifr.ifr_ifindex = a->sll_ifindex;
1072 if (ioctl(sockfd, SIOCGIFNAME, &ifr) == 0)
1073 ifname = ifr.ifr_name;
1074 }
1075 return Py_BuildValue("shbhy#",
1076 ifname,
1077 ntohs(a->sll_protocol),
1078 a->sll_pkttype,
1079 a->sll_hatype,
1080 a->sll_addr,
1081 a->sll_halen);
1082 }
Jeremy Hylton22308652001-02-02 03:23:09 +00001083#endif
Guido van Rossum48a680c2001-03-02 06:34:14 +00001084
Christian Heimes043d6f62008-01-07 17:19:16 +00001085#ifdef HAVE_LINUX_TIPC_H
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001086 case AF_TIPC:
1087 {
1088 struct sockaddr_tipc *a = (struct sockaddr_tipc *) addr;
1089 if (a->addrtype == TIPC_ADDR_NAMESEQ) {
1090 return Py_BuildValue("IIIII",
1091 a->addrtype,
1092 a->addr.nameseq.type,
1093 a->addr.nameseq.lower,
1094 a->addr.nameseq.upper,
1095 a->scope);
1096 } else if (a->addrtype == TIPC_ADDR_NAME) {
1097 return Py_BuildValue("IIIII",
1098 a->addrtype,
1099 a->addr.name.name.type,
1100 a->addr.name.name.instance,
1101 a->addr.name.name.instance,
1102 a->scope);
1103 } else if (a->addrtype == TIPC_ADDR_ID) {
1104 return Py_BuildValue("IIIII",
1105 a->addrtype,
1106 a->addr.id.node,
1107 a->addr.id.ref,
1108 0,
1109 a->scope);
1110 } else {
1111 PyErr_SetString(PyExc_ValueError,
1112 "Invalid address type");
1113 return NULL;
1114 }
1115 }
Christian Heimes043d6f62008-01-07 17:19:16 +00001116#endif
1117
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001118 /* More cases here... */
Guido van Rossum30a685f1991-06-27 15:51:29 +00001119
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001120 default:
1121 /* If we don't know the address family, don't raise an
1122 exception -- return it as an (int, bytes) tuple. */
1123 return Py_BuildValue("iy#",
1124 addr->sa_family,
1125 addr->sa_data,
1126 sizeof(addr->sa_data));
Guido van Rossum25bec8c1992-08-05 19:00:45 +00001127
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001128 }
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001129}
1130
Guido van Rossum30a685f1991-06-27 15:51:29 +00001131
1132/* Parse a socket address argument according to the socket object's
1133 address family. Return 1 if the address was in the proper format,
1134 0 of not. The address is returned through addr_ret, its length
1135 through len_ret. */
1136
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001137static int
Guido van Rossum48a680c2001-03-02 06:34:14 +00001138getsockaddrarg(PySocketSockObject *s, PyObject *args,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001139 struct sockaddr *addr_ret, int *len_ret)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001140{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001141 switch (s->sock_family) {
Guido van Rossum30a685f1991-06-27 15:51:29 +00001142
Andrew MacIntyred12dfbb2004-04-04 07:13:49 +00001143#if defined(AF_UNIX)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001144 case AF_UNIX:
1145 {
1146 struct sockaddr_un* addr;
1147 char *path;
1148 int len;
1149 if (!PyArg_Parse(args, "s#", &path, &len))
1150 return 0;
Thomas Wouters89f507f2006-12-13 04:49:30 +00001151
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001152 addr = (struct sockaddr_un*)addr_ret;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001153#ifdef linux
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001154 if (len > 0 && path[0] == 0) {
1155 /* Linux abstract namespace extension */
1156 if (len > sizeof addr->sun_path) {
1157 PyErr_SetString(socket_error,
1158 "AF_UNIX path too long");
1159 return 0;
1160 }
1161 }
1162 else
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001163#endif /* linux */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001164 {
1165 /* regular NULL-terminated string */
1166 if (len >= sizeof addr->sun_path) {
1167 PyErr_SetString(socket_error,
1168 "AF_UNIX path too long");
1169 return 0;
1170 }
1171 addr->sun_path[len] = 0;
1172 }
1173 addr->sun_family = s->sock_family;
1174 memcpy(addr->sun_path, path, len);
Andrew MacIntyredaedf212004-04-11 12:03:57 +00001175#if defined(PYOS_OS2)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001176 *len_ret = sizeof(*addr);
Andrew MacIntyredaedf212004-04-11 12:03:57 +00001177#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001178 *len_ret = len + offsetof(struct sockaddr_un, sun_path);
Andrew MacIntyredaedf212004-04-11 12:03:57 +00001179#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001180 return 1;
1181 }
Guido van Rossumb6775db1994-08-01 11:34:53 +00001182#endif /* AF_UNIX */
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001183
Martin v. Löwis11017b12006-01-14 18:12:57 +00001184#if defined(AF_NETLINK)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001185 case AF_NETLINK:
1186 {
1187 struct sockaddr_nl* addr;
1188 int pid, groups;
1189 addr = (struct sockaddr_nl *)addr_ret;
1190 if (!PyTuple_Check(args)) {
1191 PyErr_Format(
1192 PyExc_TypeError,
1193 "getsockaddrarg: "
1194 "AF_NETLINK address must be tuple, not %.500s",
1195 Py_TYPE(args)->tp_name);
1196 return 0;
1197 }
1198 if (!PyArg_ParseTuple(args, "II:getsockaddrarg", &pid, &groups))
1199 return 0;
1200 addr->nl_family = AF_NETLINK;
1201 addr->nl_pid = pid;
1202 addr->nl_groups = groups;
1203 *len_ret = sizeof(*addr);
1204 return 1;
1205 }
Martin v. Löwis11017b12006-01-14 18:12:57 +00001206#endif
1207
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001208 case AF_INET:
1209 {
1210 struct sockaddr_in* addr;
1211 char *host;
1212 int port, result;
1213 if (!PyTuple_Check(args)) {
1214 PyErr_Format(
1215 PyExc_TypeError,
1216 "getsockaddrarg: "
1217 "AF_INET address must be tuple, not %.500s",
1218 Py_TYPE(args)->tp_name);
1219 return 0;
1220 }
1221 if (!PyArg_ParseTuple(args, "eti:getsockaddrarg",
1222 "idna", &host, &port))
1223 return 0;
1224 addr=(struct sockaddr_in*)addr_ret;
1225 result = setipaddr(host, (struct sockaddr *)addr,
1226 sizeof(*addr), AF_INET);
1227 PyMem_Free(host);
1228 if (result < 0)
1229 return 0;
1230 if (port < 0 || port > 0xffff) {
1231 PyErr_SetString(
1232 PyExc_OverflowError,
1233 "getsockaddrarg: port must be 0-65535.");
1234 return 0;
1235 }
1236 addr->sin_family = AF_INET;
1237 addr->sin_port = htons((short)port);
1238 *len_ret = sizeof *addr;
1239 return 1;
1240 }
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001241
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00001242#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001243 case AF_INET6:
1244 {
1245 struct sockaddr_in6* addr;
1246 char *host;
1247 int port, flowinfo, scope_id, result;
1248 flowinfo = scope_id = 0;
1249 if (!PyTuple_Check(args)) {
1250 PyErr_Format(
1251 PyExc_TypeError,
1252 "getsockaddrarg: "
1253 "AF_INET6 address must be tuple, not %.500s",
1254 Py_TYPE(args)->tp_name);
1255 return 0;
1256 }
1257 if (!PyArg_ParseTuple(args, "eti|ii",
1258 "idna", &host, &port, &flowinfo,
1259 &scope_id)) {
1260 return 0;
1261 }
1262 addr = (struct sockaddr_in6*)addr_ret;
1263 result = setipaddr(host, (struct sockaddr *)addr,
1264 sizeof(*addr), AF_INET6);
1265 PyMem_Free(host);
1266 if (result < 0)
1267 return 0;
1268 if (port < 0 || port > 0xffff) {
1269 PyErr_SetString(
1270 PyExc_OverflowError,
1271 "getsockaddrarg: port must be 0-65535.");
1272 return 0;
1273 }
1274 addr->sin6_family = s->sock_family;
1275 addr->sin6_port = htons((short)port);
1276 addr->sin6_flowinfo = flowinfo;
1277 addr->sin6_scope_id = scope_id;
1278 *len_ret = sizeof *addr;
1279 return 1;
1280 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00001281#endif
1282
Hye-Shik Chang81268602004-02-02 06:05:24 +00001283#ifdef USE_BLUETOOTH
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001284 case AF_BLUETOOTH:
1285 {
1286 switch (s->sock_proto) {
1287 case BTPROTO_L2CAP:
1288 {
1289 struct sockaddr_l2 *addr;
1290 char *straddr;
Martin v. Löwis12af0482004-01-31 12:34:17 +00001291
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001292 addr = (struct sockaddr_l2 *)addr_ret;
1293 memset(addr, 0, sizeof(struct sockaddr_l2));
1294 _BT_L2_MEMB(addr, family) = AF_BLUETOOTH;
1295 if (!PyArg_ParseTuple(args, "si", &straddr,
1296 &_BT_L2_MEMB(addr, psm))) {
1297 PyErr_SetString(socket_error, "getsockaddrarg: "
1298 "wrong format");
1299 return 0;
1300 }
1301 if (setbdaddr(straddr, &_BT_L2_MEMB(addr, bdaddr)) < 0)
1302 return 0;
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001303
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001304 *len_ret = sizeof *addr;
1305 return 1;
1306 }
1307 case BTPROTO_RFCOMM:
1308 {
1309 struct sockaddr_rc *addr;
1310 char *straddr;
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001311
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001312 addr = (struct sockaddr_rc *)addr_ret;
1313 _BT_RC_MEMB(addr, family) = AF_BLUETOOTH;
1314 if (!PyArg_ParseTuple(args, "si", &straddr,
1315 &_BT_RC_MEMB(addr, channel))) {
1316 PyErr_SetString(socket_error, "getsockaddrarg: "
1317 "wrong format");
1318 return 0;
1319 }
1320 if (setbdaddr(straddr, &_BT_RC_MEMB(addr, bdaddr)) < 0)
1321 return 0;
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001322
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001323 *len_ret = sizeof *addr;
1324 return 1;
1325 }
1326 case BTPROTO_HCI:
1327 {
1328 struct sockaddr_hci *addr = (struct sockaddr_hci *)addr_ret;
1329 _BT_HCI_MEMB(addr, family) = AF_BLUETOOTH;
1330 if (!PyArg_ParseTuple(args, "i", &_BT_HCI_MEMB(addr, dev))) {
1331 PyErr_SetString(socket_error, "getsockaddrarg: "
1332 "wrong format");
1333 return 0;
1334 }
1335 *len_ret = sizeof *addr;
1336 return 1;
1337 }
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001338#if !defined(__FreeBSD__)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001339 case BTPROTO_SCO:
1340 {
1341 struct sockaddr_sco *addr;
1342 char *straddr;
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001343
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001344 addr = (struct sockaddr_sco *)addr_ret;
1345 _BT_SCO_MEMB(addr, family) = AF_BLUETOOTH;
1346 if (!PyBytes_Check(args)) {
1347 PyErr_SetString(socket_error, "getsockaddrarg: "
1348 "wrong format");
1349 return 0;
1350 }
1351 straddr = PyBytes_AS_STRING(args);
1352 if (setbdaddr(straddr, &_BT_SCO_MEMB(addr, bdaddr)) < 0)
1353 return 0;
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001354
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001355 *len_ret = sizeof *addr;
1356 return 1;
1357 }
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001358#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001359 default:
1360 PyErr_SetString(socket_error, "getsockaddrarg: unknown Bluetooth protocol");
1361 return 0;
1362 }
1363 }
Martin v. Löwis12af0482004-01-31 12:34:17 +00001364#endif
1365
Martin v. Löwis1ba3fd52001-08-10 20:29:40 +00001366#ifdef HAVE_NETPACKET_PACKET_H
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001367 case AF_PACKET:
1368 {
1369 struct sockaddr_ll* addr;
1370 struct ifreq ifr;
1371 char *interfaceName;
1372 int protoNumber;
1373 int hatype = 0;
1374 int pkttype = 0;
1375 char *haddr = NULL;
1376 unsigned int halen = 0;
Guido van Rossum48a680c2001-03-02 06:34:14 +00001377
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001378 if (!PyTuple_Check(args)) {
1379 PyErr_Format(
1380 PyExc_TypeError,
1381 "getsockaddrarg: "
1382 "AF_PACKET address must be tuple, not %.500s",
1383 Py_TYPE(args)->tp_name);
1384 return 0;
1385 }
1386 if (!PyArg_ParseTuple(args, "si|iiy#", &interfaceName,
1387 &protoNumber, &pkttype, &hatype,
1388 &haddr, &halen))
1389 return 0;
1390 strncpy(ifr.ifr_name, interfaceName, sizeof(ifr.ifr_name));
1391 ifr.ifr_name[(sizeof(ifr.ifr_name))-1] = '\0';
1392 if (ioctl(s->sock_fd, SIOCGIFINDEX, &ifr) < 0) {
1393 s->errorhandler();
1394 return 0;
1395 }
1396 if (halen > 8) {
1397 PyErr_SetString(PyExc_ValueError,
1398 "Hardware address must be 8 bytes or less");
1399 return 0;
1400 }
1401 if (protoNumber < 0 || protoNumber > 0xffff) {
1402 PyErr_SetString(
1403 PyExc_OverflowError,
1404 "getsockaddrarg: protoNumber must be 0-65535.");
1405 return 0;
1406 }
1407 addr = (struct sockaddr_ll*)addr_ret;
1408 addr->sll_family = AF_PACKET;
1409 addr->sll_protocol = htons((short)protoNumber);
1410 addr->sll_ifindex = ifr.ifr_ifindex;
1411 addr->sll_pkttype = pkttype;
1412 addr->sll_hatype = hatype;
1413 if (halen != 0) {
1414 memcpy(&addr->sll_addr, haddr, halen);
1415 }
1416 addr->sll_halen = halen;
1417 *len_ret = sizeof *addr;
1418 return 1;
1419 }
Guido van Rossum48a680c2001-03-02 06:34:14 +00001420#endif
1421
Christian Heimes043d6f62008-01-07 17:19:16 +00001422#ifdef HAVE_LINUX_TIPC_H
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001423 case AF_TIPC:
1424 {
1425 unsigned int atype, v1, v2, v3;
1426 unsigned int scope = TIPC_CLUSTER_SCOPE;
1427 struct sockaddr_tipc *addr;
Christian Heimes043d6f62008-01-07 17:19:16 +00001428
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001429 if (!PyTuple_Check(args)) {
1430 PyErr_Format(
1431 PyExc_TypeError,
1432 "getsockaddrarg: "
1433 "AF_TIPC address must be tuple, not %.500s",
1434 Py_TYPE(args)->tp_name);
1435 return 0;
1436 }
Christian Heimes043d6f62008-01-07 17:19:16 +00001437
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001438 if (!PyArg_ParseTuple(args,
1439 "IIII|I;Invalid TIPC address format",
1440 &atype, &v1, &v2, &v3, &scope))
1441 return 0;
Christian Heimes043d6f62008-01-07 17:19:16 +00001442
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001443 addr = (struct sockaddr_tipc *) addr_ret;
1444 memset(addr, 0, sizeof(struct sockaddr_tipc));
Christian Heimes043d6f62008-01-07 17:19:16 +00001445
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001446 addr->family = AF_TIPC;
1447 addr->scope = scope;
1448 addr->addrtype = atype;
Christian Heimes043d6f62008-01-07 17:19:16 +00001449
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001450 if (atype == TIPC_ADDR_NAMESEQ) {
1451 addr->addr.nameseq.type = v1;
1452 addr->addr.nameseq.lower = v2;
1453 addr->addr.nameseq.upper = v3;
1454 } else if (atype == TIPC_ADDR_NAME) {
1455 addr->addr.name.name.type = v1;
1456 addr->addr.name.name.instance = v2;
1457 } else if (atype == TIPC_ADDR_ID) {
1458 addr->addr.id.node = v1;
1459 addr->addr.id.ref = v2;
1460 } else {
1461 /* Shouldn't happen */
1462 PyErr_SetString(PyExc_TypeError, "Invalid address type");
1463 return 0;
1464 }
Christian Heimes043d6f62008-01-07 17:19:16 +00001465
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001466 *len_ret = sizeof(*addr);
Christian Heimes043d6f62008-01-07 17:19:16 +00001467
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001468 return 1;
1469 }
Christian Heimes043d6f62008-01-07 17:19:16 +00001470#endif
1471
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001472 /* More cases here... */
Guido van Rossum30a685f1991-06-27 15:51:29 +00001473
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001474 default:
1475 PyErr_SetString(socket_error, "getsockaddrarg: bad family");
1476 return 0;
Guido van Rossum30a685f1991-06-27 15:51:29 +00001477
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001478 }
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001479}
1480
Guido van Rossum30a685f1991-06-27 15:51:29 +00001481
Guido van Rossum48a680c2001-03-02 06:34:14 +00001482/* Get the address length according to the socket object's address family.
Guido van Rossum710e1df1992-06-12 10:39:36 +00001483 Return 1 if the family is known, 0 otherwise. The length is returned
1484 through len_ret. */
1485
1486static int
Peter Schneider-Kamp618e25d2000-07-11 23:00:12 +00001487getsockaddrlen(PySocketSockObject *s, socklen_t *len_ret)
Guido van Rossum710e1df1992-06-12 10:39:36 +00001488{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001489 switch (s->sock_family) {
Guido van Rossum710e1df1992-06-12 10:39:36 +00001490
Andrew MacIntyred12dfbb2004-04-04 07:13:49 +00001491#if defined(AF_UNIX)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001492 case AF_UNIX:
1493 {
1494 *len_ret = sizeof (struct sockaddr_un);
1495 return 1;
1496 }
Guido van Rossumb6775db1994-08-01 11:34:53 +00001497#endif /* AF_UNIX */
Martin v. Löwis11017b12006-01-14 18:12:57 +00001498#if defined(AF_NETLINK)
1499 case AF_NETLINK:
1500 {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001501 *len_ret = sizeof (struct sockaddr_nl);
1502 return 1;
Martin v. Löwis11017b12006-01-14 18:12:57 +00001503 }
1504#endif
Guido van Rossum710e1df1992-06-12 10:39:36 +00001505
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001506 case AF_INET:
1507 {
1508 *len_ret = sizeof (struct sockaddr_in);
1509 return 1;
1510 }
Guido van Rossum710e1df1992-06-12 10:39:36 +00001511
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00001512#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001513 case AF_INET6:
1514 {
1515 *len_ret = sizeof (struct sockaddr_in6);
1516 return 1;
1517 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00001518#endif
1519
Hye-Shik Chang81268602004-02-02 06:05:24 +00001520#ifdef USE_BLUETOOTH
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001521 case AF_BLUETOOTH:
1522 {
1523 switch(s->sock_proto)
1524 {
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001525
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001526 case BTPROTO_L2CAP:
1527 *len_ret = sizeof (struct sockaddr_l2);
1528 return 1;
1529 case BTPROTO_RFCOMM:
1530 *len_ret = sizeof (struct sockaddr_rc);
1531 return 1;
1532 case BTPROTO_HCI:
1533 *len_ret = sizeof (struct sockaddr_hci);
1534 return 1;
Hye-Shik Chang81268602004-02-02 06:05:24 +00001535#if !defined(__FreeBSD__)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001536 case BTPROTO_SCO:
1537 *len_ret = sizeof (struct sockaddr_sco);
1538 return 1;
Hye-Shik Chang81268602004-02-02 06:05:24 +00001539#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001540 default:
1541 PyErr_SetString(socket_error, "getsockaddrlen: "
1542 "unknown BT protocol");
1543 return 0;
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001544
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001545 }
1546 }
Martin v. Löwis12af0482004-01-31 12:34:17 +00001547#endif
1548
Martin v. Löwis1ba3fd52001-08-10 20:29:40 +00001549#ifdef HAVE_NETPACKET_PACKET_H
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001550 case AF_PACKET:
1551 {
1552 *len_ret = sizeof (struct sockaddr_ll);
1553 return 1;
1554 }
Jeremy Hylton22308652001-02-02 03:23:09 +00001555#endif
Guido van Rossum48a680c2001-03-02 06:34:14 +00001556
Christian Heimes043d6f62008-01-07 17:19:16 +00001557#ifdef HAVE_LINUX_TIPC_H
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001558 case AF_TIPC:
1559 {
1560 *len_ret = sizeof (struct sockaddr_tipc);
1561 return 1;
1562 }
Christian Heimes043d6f62008-01-07 17:19:16 +00001563#endif
1564
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001565 /* More cases here... */
Guido van Rossum710e1df1992-06-12 10:39:36 +00001566
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001567 default:
1568 PyErr_SetString(socket_error, "getsockaddrlen: bad family");
1569 return 0;
Guido van Rossum710e1df1992-06-12 10:39:36 +00001570
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001571 }
Guido van Rossum710e1df1992-06-12 10:39:36 +00001572}
1573
1574
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00001575/* s._accept() -> (fd, address) */
Guido van Rossum30a685f1991-06-27 15:51:29 +00001576
Guido van Rossum73624e91994-10-10 17:59:00 +00001577static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001578sock_accept(PySocketSockObject *s)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001579{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001580 sock_addr_t addrbuf;
1581 SOCKET_T newfd = INVALID_SOCKET;
1582 socklen_t addrlen;
1583 PyObject *sock = NULL;
1584 PyObject *addr = NULL;
1585 PyObject *res = NULL;
1586 int timeout;
Barry Warsaw752300b1997-01-03 17:18:10 +00001587
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001588 if (!getsockaddrlen(s, &addrlen))
1589 return NULL;
1590 memset(&addrbuf, 0, addrlen);
Guido van Rossum67f7a382002-06-06 21:08:16 +00001591
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001592 if (!IS_SELECTABLE(s))
1593 return select_error();
Neal Norwitz082b2df2006-02-07 07:04:46 +00001594
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001595 Py_BEGIN_ALLOW_THREADS
1596 timeout = internal_select(s, 0);
1597 if (!timeout)
1598 newfd = accept(s->sock_fd, SAS2SA(&addrbuf), &addrlen);
1599 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00001600
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001601 if (timeout == 1) {
1602 PyErr_SetString(socket_timeout, "timed out");
1603 return NULL;
1604 }
Raymond Hettingeref7343c2003-06-29 03:08:05 +00001605
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001606 if (newfd == INVALID_SOCKET)
1607 return s->errorhandler();
Barry Warsaw752300b1997-01-03 17:18:10 +00001608
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001609 sock = PyLong_FromSocket_t(newfd);
1610 if (sock == NULL) {
1611 SOCKETCLOSE(newfd);
1612 goto finally;
1613 }
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00001614
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001615 addr = makesockaddr(s->sock_fd, SAS2SA(&addrbuf),
1616 addrlen, s->sock_proto);
1617 if (addr == NULL)
1618 goto finally;
Barry Warsaw752300b1997-01-03 17:18:10 +00001619
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001620 res = PyTuple_Pack(2, sock, addr);
Barry Warsaw752300b1997-01-03 17:18:10 +00001621
Guido van Rossum67f7a382002-06-06 21:08:16 +00001622finally:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001623 Py_XDECREF(sock);
1624 Py_XDECREF(addr);
1625 return res;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001626}
1627
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001628PyDoc_STRVAR(accept_doc,
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00001629"_accept() -> (integer, address info)\n\
Guido van Rossum82a5c661998-07-07 20:45:43 +00001630\n\
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00001631Wait for an incoming connection. Return a new socket file descriptor\n\
1632representing the connection, and the address of the client.\n\
1633For IP sockets, the address info is a pair (hostaddr, port).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001634
Guido van Rossum11ba0942002-06-13 15:07:44 +00001635/* s.setblocking(flag) method. Argument:
1636 False -- non-blocking mode; same as settimeout(0)
1637 True -- blocking mode; same as settimeout(None)
1638*/
Guido van Rossume4485b01994-09-07 14:32:49 +00001639
Guido van Rossum73624e91994-10-10 17:59:00 +00001640static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001641sock_setblocking(PySocketSockObject *s, PyObject *arg)
Guido van Rossume4485b01994-09-07 14:32:49 +00001642{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001643 int block;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001644
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001645 block = PyLong_AsLong(arg);
1646 if (block == -1 && PyErr_Occurred())
1647 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001648
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001649 s->sock_timeout = block ? -1.0 : 0.0;
1650 internal_setblocking(s, block);
Guido van Rossume4485b01994-09-07 14:32:49 +00001651
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001652 Py_INCREF(Py_None);
1653 return Py_None;
Guido van Rossume4485b01994-09-07 14:32:49 +00001654}
Guido van Rossume4485b01994-09-07 14:32:49 +00001655
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001656PyDoc_STRVAR(setblocking_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001657"setblocking(flag)\n\
1658\n\
1659Set the socket to blocking (flag is true) or non-blocking (false).\n\
Guido van Rossum11ba0942002-06-13 15:07:44 +00001660setblocking(True) is equivalent to settimeout(None);\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001661setblocking(False) is equivalent to settimeout(0.0).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001662
Guido van Rossum11ba0942002-06-13 15:07:44 +00001663/* s.settimeout(timeout) method. Argument:
1664 None -- no timeout, blocking mode; same as setblocking(True)
1665 0.0 -- non-blocking mode; same as setblocking(False)
1666 > 0 -- timeout mode; operations time out after timeout seconds
1667 < 0 -- illegal; raises an exception
1668*/
Guido van Rossum67f7a382002-06-06 21:08:16 +00001669static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001670sock_settimeout(PySocketSockObject *s, PyObject *arg)
Guido van Rossum67f7a382002-06-06 21:08:16 +00001671{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001672 double timeout;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001673
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001674 if (arg == Py_None)
1675 timeout = -1.0;
1676 else {
1677 timeout = PyFloat_AsDouble(arg);
1678 if (timeout < 0.0) {
1679 if (!PyErr_Occurred())
1680 PyErr_SetString(PyExc_ValueError,
1681 "Timeout value out of range");
1682 return NULL;
1683 }
1684 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00001685
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001686 s->sock_timeout = timeout;
1687 internal_setblocking(s, timeout < 0.0);
Guido van Rossum67f7a382002-06-06 21:08:16 +00001688
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001689 Py_INCREF(Py_None);
1690 return Py_None;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001691}
1692
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001693PyDoc_STRVAR(settimeout_doc,
Guido van Rossum3eede5a2002-06-07 02:08:35 +00001694"settimeout(timeout)\n\
Guido van Rossum67f7a382002-06-06 21:08:16 +00001695\n\
Guido van Rossum11ba0942002-06-13 15:07:44 +00001696Set a timeout on socket operations. 'timeout' can be a float,\n\
1697giving in seconds, or None. Setting a timeout of None disables\n\
1698the timeout feature and is equivalent to setblocking(1).\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001699Setting a timeout of zero is the same as setblocking(0).");
Guido van Rossum67f7a382002-06-06 21:08:16 +00001700
Guido van Rossum3eede5a2002-06-07 02:08:35 +00001701/* s.gettimeout() method.
1702 Returns the timeout associated with a socket. */
Guido van Rossum67f7a382002-06-06 21:08:16 +00001703static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001704sock_gettimeout(PySocketSockObject *s)
Guido van Rossum67f7a382002-06-06 21:08:16 +00001705{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001706 if (s->sock_timeout < 0.0) {
1707 Py_INCREF(Py_None);
1708 return Py_None;
1709 }
1710 else
1711 return PyFloat_FromDouble(s->sock_timeout);
Guido van Rossum67f7a382002-06-06 21:08:16 +00001712}
1713
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001714PyDoc_STRVAR(gettimeout_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00001715"gettimeout() -> timeout\n\
Guido van Rossum67f7a382002-06-06 21:08:16 +00001716\n\
1717Returns the timeout in floating seconds associated with socket \n\
1718operations. A timeout of None indicates that timeouts on socket \n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001719operations are disabled.");
Guido van Rossume4485b01994-09-07 14:32:49 +00001720
Guido van Rossumaee08791992-09-08 09:05:33 +00001721/* s.setsockopt() method.
1722 With an integer third argument, sets an integer option.
1723 With a string third argument, sets an option from a buffer;
1724 use optional built-in module 'struct' to encode the string. */
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001725
Guido van Rossum73624e91994-10-10 17:59:00 +00001726static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001727sock_setsockopt(PySocketSockObject *s, PyObject *args)
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001728{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001729 int level;
1730 int optname;
1731 int res;
1732 char *buf;
1733 int buflen;
1734 int flag;
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001735
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001736 if (PyArg_ParseTuple(args, "iii:setsockopt",
1737 &level, &optname, &flag)) {
1738 buf = (char *) &flag;
1739 buflen = sizeof flag;
1740 }
1741 else {
1742 PyErr_Clear();
1743 if (!PyArg_ParseTuple(args, "iiy#:setsockopt",
1744 &level, &optname, &buf, &buflen))
1745 return NULL;
1746 }
1747 res = setsockopt(s->sock_fd, level, optname, (void *)buf, buflen);
1748 if (res < 0)
1749 return s->errorhandler();
1750 Py_INCREF(Py_None);
1751 return Py_None;
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001752}
1753
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001754PyDoc_STRVAR(setsockopt_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001755"setsockopt(level, option, value)\n\
1756\n\
1757Set a socket option. See the Unix manual for level and option.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001758The value argument can either be an integer or a string.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001759
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001760
Guido van Rossumaee08791992-09-08 09:05:33 +00001761/* s.getsockopt() method.
1762 With two arguments, retrieves an integer option.
1763 With a third integer argument, retrieves a string buffer of that size;
1764 use optional built-in module 'struct' to decode the string. */
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001765
Guido van Rossum73624e91994-10-10 17:59:00 +00001766static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001767sock_getsockopt(PySocketSockObject *s, PyObject *args)
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001768{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001769 int level;
1770 int optname;
1771 int res;
1772 PyObject *buf;
1773 socklen_t buflen = 0;
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001774
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001775 if (!PyArg_ParseTuple(args, "ii|i:getsockopt",
1776 &level, &optname, &buflen))
1777 return NULL;
Guido van Rossum48a680c2001-03-02 06:34:14 +00001778
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001779 if (buflen == 0) {
1780 int flag = 0;
1781 socklen_t flagsize = sizeof flag;
1782 res = getsockopt(s->sock_fd, level, optname,
1783 (void *)&flag, &flagsize);
1784 if (res < 0)
1785 return s->errorhandler();
1786 return PyLong_FromLong(flag);
1787 }
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001788#ifdef __VMS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001789 /* socklen_t is unsigned so no negative test is needed,
1790 test buflen == 0 is previously done */
1791 if (buflen > 1024) {
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001792#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001793 if (buflen <= 0 || buflen > 1024) {
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001794#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001795 PyErr_SetString(socket_error,
1796 "getsockopt buflen out of range");
1797 return NULL;
1798 }
1799 buf = PyBytes_FromStringAndSize((char *)NULL, buflen);
1800 if (buf == NULL)
1801 return NULL;
1802 res = getsockopt(s->sock_fd, level, optname,
1803 (void *)PyBytes_AS_STRING(buf), &buflen);
1804 if (res < 0) {
1805 Py_DECREF(buf);
1806 return s->errorhandler();
1807 }
1808 _PyBytes_Resize(&buf, buflen);
1809 return buf;
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001810}
1811
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001812PyDoc_STRVAR(getsockopt_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001813"getsockopt(level, option[, buffersize]) -> value\n\
1814\n\
1815Get a socket option. See the Unix manual for level and option.\n\
1816If a nonzero buffersize argument is given, the return value is a\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001817string of that length; otherwise it is an integer.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001818
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001819
Fred Drake728819a2000-07-01 03:40:12 +00001820/* s.bind(sockaddr) method */
Guido van Rossum30a685f1991-06-27 15:51:29 +00001821
Guido van Rossum73624e91994-10-10 17:59:00 +00001822static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001823sock_bind(PySocketSockObject *s, PyObject *addro)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001824{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001825 sock_addr_t addrbuf;
1826 int addrlen;
1827 int res;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00001828
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001829 if (!getsockaddrarg(s, addro, SAS2SA(&addrbuf), &addrlen))
1830 return NULL;
1831 Py_BEGIN_ALLOW_THREADS
1832 res = bind(s->sock_fd, SAS2SA(&addrbuf), addrlen);
1833 Py_END_ALLOW_THREADS
1834 if (res < 0)
1835 return s->errorhandler();
1836 Py_INCREF(Py_None);
1837 return Py_None;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001838}
1839
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001840PyDoc_STRVAR(bind_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001841"bind(address)\n\
1842\n\
1843Bind the socket to a local address. For IP sockets, the address is a\n\
Jeremy Hylton22308652001-02-02 03:23:09 +00001844pair (host, port); the host must refer to the local host. For raw packet\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001845sockets the address is a tuple (ifname, proto [,pkttype [,hatype]])");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001846
Guido van Rossum30a685f1991-06-27 15:51:29 +00001847
1848/* s.close() method.
1849 Set the file descriptor to -1 so operations tried subsequently
1850 will surely fail. */
1851
Guido van Rossum73624e91994-10-10 17:59:00 +00001852static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001853sock_close(PySocketSockObject *s)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001854{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001855 SOCKET_T fd;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00001856
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001857 if ((fd = s->sock_fd) != -1) {
1858 s->sock_fd = -1;
1859 Py_BEGIN_ALLOW_THREADS
1860 (void) SOCKETCLOSE(fd);
1861 Py_END_ALLOW_THREADS
1862 }
1863 Py_INCREF(Py_None);
1864 return Py_None;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001865}
1866
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001867PyDoc_STRVAR(close_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001868"close()\n\
1869\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001870Close the socket. It cannot be used after this call.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001871
Antoine Pitroue43f9d02010-08-08 23:24:50 +00001872static PyObject *
1873sock_forget(PySocketSockObject *s)
1874{
1875 s->sock_fd = -1;
1876 Py_INCREF(Py_None);
1877 return Py_None;
1878}
1879
1880PyDoc_STRVAR(forget_doc,
1881"forget()\n\
1882\n\
1883Close the socket object without closing the underlying file descriptor.\
1884The object cannot be used after this call, but the file descriptor\
1885can be reused for other purposes.");
1886
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001887static int
Raymond Hettingeref7343c2003-06-29 03:08:05 +00001888internal_connect(PySocketSockObject *s, struct sockaddr *addr, int addrlen,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001889 int *timeoutp)
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001890{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001891 int res, timeout;
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001892
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001893 timeout = 0;
1894 res = connect(s->sock_fd, addr, addrlen);
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001895
1896#ifdef MS_WINDOWS
1897
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001898 if (s->sock_timeout > 0.0) {
1899 if (res < 0 && WSAGetLastError() == WSAEWOULDBLOCK &&
1900 IS_SELECTABLE(s)) {
1901 /* This is a mess. Best solution: trust select */
1902 fd_set fds;
1903 fd_set fds_exc;
1904 struct timeval tv;
1905 tv.tv_sec = (int)s->sock_timeout;
1906 tv.tv_usec = (int)((s->sock_timeout - tv.tv_sec) * 1e6);
1907 FD_ZERO(&fds);
1908 FD_SET(s->sock_fd, &fds);
1909 FD_ZERO(&fds_exc);
1910 FD_SET(s->sock_fd, &fds_exc);
1911 res = select(s->sock_fd+1, NULL, &fds, &fds_exc, &tv);
1912 if (res == 0) {
1913 res = WSAEWOULDBLOCK;
1914 timeout = 1;
1915 } else if (res > 0) {
1916 if (FD_ISSET(s->sock_fd, &fds))
1917 /* The socket is in the writable set - this
1918 means connected */
1919 res = 0;
1920 else {
1921 /* As per MS docs, we need to call getsockopt()
1922 to get the underlying error */
1923 int res_size = sizeof res;
1924 /* It must be in the exception set */
1925 assert(FD_ISSET(s->sock_fd, &fds_exc));
1926 if (0 == getsockopt(s->sock_fd, SOL_SOCKET, SO_ERROR,
1927 (char *)&res, &res_size))
1928 /* getsockopt also clears WSAGetLastError,
1929 so reset it back. */
1930 WSASetLastError(res);
1931 else
1932 res = WSAGetLastError();
1933 }
1934 }
1935 /* else if (res < 0) an error occurred */
1936 }
1937 }
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001938
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001939 if (res < 0)
1940 res = WSAGetLastError();
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001941
1942#else
1943
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001944 if (s->sock_timeout > 0.0) {
1945 if (res < 0 && errno == EINPROGRESS && IS_SELECTABLE(s)) {
1946 timeout = internal_select(s, 1);
1947 if (timeout == 0) {
1948 /* Bug #1019808: in case of an EINPROGRESS,
1949 use getsockopt(SO_ERROR) to get the real
1950 error. */
1951 socklen_t res_size = sizeof res;
1952 (void)getsockopt(s->sock_fd, SOL_SOCKET,
1953 SO_ERROR, &res, &res_size);
1954 if (res == EISCONN)
1955 res = 0;
1956 errno = res;
1957 }
1958 else if (timeout == -1) {
1959 res = errno; /* had error */
1960 }
1961 else
1962 res = EWOULDBLOCK; /* timed out */
1963 }
1964 }
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001965
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001966 if (res < 0)
1967 res = errno;
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001968
1969#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001970 *timeoutp = timeout;
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001971
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001972 return res;
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001973}
Guido van Rossum30a685f1991-06-27 15:51:29 +00001974
Fred Drake728819a2000-07-01 03:40:12 +00001975/* s.connect(sockaddr) method */
Guido van Rossum30a685f1991-06-27 15:51:29 +00001976
Guido van Rossum73624e91994-10-10 17:59:00 +00001977static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001978sock_connect(PySocketSockObject *s, PyObject *addro)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001979{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001980 sock_addr_t addrbuf;
1981 int addrlen;
1982 int res;
1983 int timeout;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00001984
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001985 if (!getsockaddrarg(s, addro, SAS2SA(&addrbuf), &addrlen))
1986 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001987
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001988 Py_BEGIN_ALLOW_THREADS
1989 res = internal_connect(s, SAS2SA(&addrbuf), addrlen, &timeout);
1990 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00001991
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001992 if (timeout == 1) {
1993 PyErr_SetString(socket_timeout, "timed out");
1994 return NULL;
1995 }
1996 if (res != 0)
1997 return s->errorhandler();
1998 Py_INCREF(Py_None);
1999 return Py_None;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002000}
2001
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002002PyDoc_STRVAR(connect_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002003"connect(address)\n\
2004\n\
2005Connect the socket to a remote address. For IP sockets, the address\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002006is a pair (host, port).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002007
Guido van Rossum30a685f1991-06-27 15:51:29 +00002008
Fred Drake728819a2000-07-01 03:40:12 +00002009/* s.connect_ex(sockaddr) method */
Guido van Rossumfc4255d1997-11-19 18:57:13 +00002010
2011static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002012sock_connect_ex(PySocketSockObject *s, PyObject *addro)
Guido van Rossumfc4255d1997-11-19 18:57:13 +00002013{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002014 sock_addr_t addrbuf;
2015 int addrlen;
2016 int res;
2017 int timeout;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00002018
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002019 if (!getsockaddrarg(s, addro, SAS2SA(&addrbuf), &addrlen))
2020 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002021
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002022 Py_BEGIN_ALLOW_THREADS
2023 res = internal_connect(s, SAS2SA(&addrbuf), addrlen, &timeout);
2024 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00002025
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002026 /* Signals are not errors (though they may raise exceptions). Adapted
2027 from PyErr_SetFromErrnoWithFilenameObject(). */
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002028#ifdef EINTR
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002029 if (res == EINTR && PyErr_CheckSignals())
2030 return NULL;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002031#endif
2032
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002033 return PyLong_FromLong((long) res);
Guido van Rossumfc4255d1997-11-19 18:57:13 +00002034}
2035
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002036PyDoc_STRVAR(connect_ex_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00002037"connect_ex(address) -> errno\n\
Guido van Rossum82a5c661998-07-07 20:45:43 +00002038\n\
2039This is like connect(address), but returns an error code (the errno value)\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002040instead of raising an exception when an error occurs.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002041
Guido van Rossumfc4255d1997-11-19 18:57:13 +00002042
Guido van Rossumed233a51992-06-23 09:07:03 +00002043/* s.fileno() method */
2044
Guido van Rossum73624e91994-10-10 17:59:00 +00002045static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002046sock_fileno(PySocketSockObject *s)
Guido van Rossumed233a51992-06-23 09:07:03 +00002047{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002048 return PyLong_FromSocket_t(s->sock_fd);
Guido van Rossumed233a51992-06-23 09:07:03 +00002049}
2050
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002051PyDoc_STRVAR(fileno_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002052"fileno() -> integer\n\
2053\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002054Return the integer file descriptor of the socket.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002055
Guido van Rossumed233a51992-06-23 09:07:03 +00002056
Guido van Rossumc89705d1992-11-26 08:54:07 +00002057/* s.getsockname() method */
2058
Guido van Rossum73624e91994-10-10 17:59:00 +00002059static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002060sock_getsockname(PySocketSockObject *s)
Guido van Rossumc89705d1992-11-26 08:54:07 +00002061{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002062 sock_addr_t addrbuf;
2063 int res;
2064 socklen_t addrlen;
Guido van Rossumff3ab422000-04-24 15:16:03 +00002065
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002066 if (!getsockaddrlen(s, &addrlen))
2067 return NULL;
2068 memset(&addrbuf, 0, addrlen);
2069 Py_BEGIN_ALLOW_THREADS
2070 res = getsockname(s->sock_fd, SAS2SA(&addrbuf), &addrlen);
2071 Py_END_ALLOW_THREADS
2072 if (res < 0)
2073 return s->errorhandler();
2074 return makesockaddr(s->sock_fd, SAS2SA(&addrbuf), addrlen,
2075 s->sock_proto);
Guido van Rossumc89705d1992-11-26 08:54:07 +00002076}
2077
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002078PyDoc_STRVAR(getsockname_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002079"getsockname() -> address info\n\
2080\n\
2081Return the address of the local endpoint. For IP sockets, the address\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002082info is a pair (hostaddr, port).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002083
Guido van Rossumc89705d1992-11-26 08:54:07 +00002084
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002085#ifdef HAVE_GETPEERNAME /* Cray APP doesn't have this :-( */
Guido van Rossumc89705d1992-11-26 08:54:07 +00002086/* s.getpeername() method */
2087
Guido van Rossum73624e91994-10-10 17:59:00 +00002088static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002089sock_getpeername(PySocketSockObject *s)
Guido van Rossumc89705d1992-11-26 08:54:07 +00002090{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002091 sock_addr_t addrbuf;
2092 int res;
2093 socklen_t addrlen;
Guido van Rossumff3ab422000-04-24 15:16:03 +00002094
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002095 if (!getsockaddrlen(s, &addrlen))
2096 return NULL;
2097 memset(&addrbuf, 0, addrlen);
2098 Py_BEGIN_ALLOW_THREADS
2099 res = getpeername(s->sock_fd, SAS2SA(&addrbuf), &addrlen);
2100 Py_END_ALLOW_THREADS
2101 if (res < 0)
2102 return s->errorhandler();
2103 return makesockaddr(s->sock_fd, SAS2SA(&addrbuf), addrlen,
2104 s->sock_proto);
Guido van Rossumc89705d1992-11-26 08:54:07 +00002105}
Guido van Rossum82a5c661998-07-07 20:45:43 +00002106
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002107PyDoc_STRVAR(getpeername_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002108"getpeername() -> address info\n\
2109\n\
2110Return the address of the remote endpoint. For IP sockets, the address\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002111info is a pair (hostaddr, port).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002112
Guido van Rossumb6775db1994-08-01 11:34:53 +00002113#endif /* HAVE_GETPEERNAME */
Guido van Rossumc89705d1992-11-26 08:54:07 +00002114
2115
Guido van Rossum30a685f1991-06-27 15:51:29 +00002116/* s.listen(n) method */
2117
Guido van Rossum73624e91994-10-10 17:59:00 +00002118static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002119sock_listen(PySocketSockObject *s, PyObject *arg)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002120{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002121 int backlog;
2122 int res;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00002123
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002124 backlog = PyLong_AsLong(arg);
2125 if (backlog == -1 && PyErr_Occurred())
2126 return NULL;
2127 Py_BEGIN_ALLOW_THREADS
2128 if (backlog < 1)
2129 backlog = 1;
2130 res = listen(s->sock_fd, backlog);
2131 Py_END_ALLOW_THREADS
2132 if (res < 0)
2133 return s->errorhandler();
2134 Py_INCREF(Py_None);
2135 return Py_None;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002136}
2137
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002138PyDoc_STRVAR(listen_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002139"listen(backlog)\n\
2140\n\
2141Enable a server to accept connections. The backlog argument must be at\n\
2142least 1; it specifies the number of unaccepted connection that the system\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002143will allow before refusing new connections.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002144
2145
Thomas Wouters477c8d52006-05-27 19:21:47 +00002146/*
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002147 * This is the guts of the recv() and recv_into() methods, which reads into a
Thomas Wouters902d6eb2007-01-09 23:18:33 +00002148 * char buffer. If you have any inc/dec ref to do to the objects that contain
Thomas Wouters477c8d52006-05-27 19:21:47 +00002149 * the buffer, do it in the caller. This function returns the number of bytes
2150 * succesfully read. If there was an error, it returns -1. Note that it is
2151 * also possible that we return a number of bytes smaller than the request
2152 * bytes.
2153 */
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002154static ssize_t
Thomas Wouters477c8d52006-05-27 19:21:47 +00002155sock_recv_guts(PySocketSockObject *s, char* cbuf, int len, int flags)
2156{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002157 ssize_t outlen = -1;
2158 int timeout;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002159#ifdef __VMS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002160 int remaining;
2161 char *read_buf;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002162#endif
2163
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002164 if (!IS_SELECTABLE(s)) {
2165 select_error();
2166 return -1;
2167 }
2168 if (len == 0) {
2169 /* If 0 bytes were requested, do nothing. */
2170 return 0;
2171 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002172
2173#ifndef __VMS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002174 Py_BEGIN_ALLOW_THREADS
2175 timeout = internal_select(s, 0);
2176 if (!timeout)
2177 outlen = recv(s->sock_fd, cbuf, len, flags);
2178 Py_END_ALLOW_THREADS
Thomas Wouters477c8d52006-05-27 19:21:47 +00002179
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002180 if (timeout == 1) {
2181 PyErr_SetString(socket_timeout, "timed out");
2182 return -1;
2183 }
2184 if (outlen < 0) {
2185 /* Note: the call to errorhandler() ALWAYS indirectly returned
2186 NULL, so ignore its return value */
2187 s->errorhandler();
2188 return -1;
2189 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002190#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002191 read_buf = cbuf;
2192 remaining = len;
2193 while (remaining != 0) {
2194 unsigned int segment;
2195 int nread = -1;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002196
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002197 segment = remaining /SEGMENT_SIZE;
2198 if (segment != 0) {
2199 segment = SEGMENT_SIZE;
2200 }
2201 else {
2202 segment = remaining;
2203 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002204
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002205 Py_BEGIN_ALLOW_THREADS
2206 timeout = internal_select(s, 0);
2207 if (!timeout)
2208 nread = recv(s->sock_fd, read_buf, segment, flags);
2209 Py_END_ALLOW_THREADS
Thomas Wouters477c8d52006-05-27 19:21:47 +00002210
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002211 if (timeout == 1) {
2212 PyErr_SetString(socket_timeout, "timed out");
2213 return -1;
2214 }
2215 if (nread < 0) {
2216 s->errorhandler();
2217 return -1;
2218 }
2219 if (nread != remaining) {
2220 read_buf += nread;
2221 break;
2222 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002223
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002224 remaining -= segment;
2225 read_buf += segment;
2226 }
2227 outlen = read_buf - cbuf;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002228#endif /* !__VMS */
2229
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002230 return outlen;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002231}
2232
Guido van Rossum48a680c2001-03-02 06:34:14 +00002233
Guido van Rossumeb6b33a1993-05-25 09:38:27 +00002234/* s.recv(nbytes [,flags]) method */
Guido van Rossum30a685f1991-06-27 15:51:29 +00002235
Guido van Rossum73624e91994-10-10 17:59:00 +00002236static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002237sock_recv(PySocketSockObject *s, PyObject *args)
Guido van Rossum30a685f1991-06-27 15:51:29 +00002238{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002239 int recvlen, flags = 0;
2240 ssize_t outlen;
2241 PyObject *buf;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002242
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002243 if (!PyArg_ParseTuple(args, "i|i:recv", &recvlen, &flags))
2244 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002245
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002246 if (recvlen < 0) {
2247 PyErr_SetString(PyExc_ValueError,
2248 "negative buffersize in recv");
2249 return NULL;
2250 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00002251
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002252 /* Allocate a new string. */
2253 buf = PyBytes_FromStringAndSize((char *) 0, recvlen);
2254 if (buf == NULL)
2255 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002256
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002257 /* Call the guts */
2258 outlen = sock_recv_guts(s, PyBytes_AS_STRING(buf), recvlen, flags);
2259 if (outlen < 0) {
2260 /* An error occurred, release the string and return an
2261 error. */
2262 Py_DECREF(buf);
2263 return NULL;
2264 }
2265 if (outlen != recvlen) {
2266 /* We did not read as many bytes as we anticipated, resize the
2267 string if possible and be successful. */
2268 _PyBytes_Resize(&buf, outlen);
2269 }
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00002270
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002271 return buf;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002272}
2273
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002274PyDoc_STRVAR(recv_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002275"recv(buffersize[, flags]) -> data\n\
2276\n\
2277Receive up to buffersize bytes from the socket. For the optional flags\n\
2278argument, see the Unix manual. When no data is available, block until\n\
2279at least one byte is available or until the remote end is closed. When\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002280the remote end is closed and all data is read, return the empty string.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002281
Guido van Rossum30a685f1991-06-27 15:51:29 +00002282
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002283/* s.recv_into(buffer, [nbytes [,flags]]) method */
Guido van Rossum30a685f1991-06-27 15:51:29 +00002284
Thomas Wouters477c8d52006-05-27 19:21:47 +00002285static PyObject*
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002286sock_recv_into(PySocketSockObject *s, PyObject *args, PyObject *kwds)
Thomas Wouters477c8d52006-05-27 19:21:47 +00002287{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002288 static char *kwlist[] = {"buffer", "nbytes", "flags", 0};
Thomas Wouters477c8d52006-05-27 19:21:47 +00002289
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002290 int recvlen = 0, flags = 0;
2291 ssize_t readlen;
2292 Py_buffer pbuf;
2293 char *buf;
2294 int buflen;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002295
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002296 /* Get the buffer's memory */
2297 if (!PyArg_ParseTupleAndKeywords(args, kwds, "w*|ii:recv_into", kwlist,
2298 &pbuf, &recvlen, &flags))
2299 return NULL;
2300 buf = pbuf.buf;
2301 buflen = pbuf.len;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002302
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002303 if (recvlen < 0) {
2304 PyBuffer_Release(&pbuf);
2305 PyErr_SetString(PyExc_ValueError,
2306 "negative buffersize in recv_into");
2307 return NULL;
2308 }
2309 if (recvlen == 0) {
2310 /* If nbytes was not specified, use the buffer's length */
2311 recvlen = buflen;
2312 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002313
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002314 /* Check if the buffer is large enough */
2315 if (buflen < recvlen) {
2316 PyBuffer_Release(&pbuf);
2317 PyErr_SetString(PyExc_ValueError,
2318 "buffer too small for requested bytes");
2319 return NULL;
2320 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002321
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002322 /* Call the guts */
2323 readlen = sock_recv_guts(s, buf, recvlen, flags);
2324 if (readlen < 0) {
2325 /* Return an error. */
2326 PyBuffer_Release(&pbuf);
2327 return NULL;
2328 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002329
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002330 PyBuffer_Release(&pbuf);
2331 /* Return the number of bytes read. Note that we do not do anything
2332 special here in the case that readlen < recvlen. */
2333 return PyLong_FromSsize_t(readlen);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002334}
2335
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002336PyDoc_STRVAR(recv_into_doc,
2337"recv_into(buffer, [nbytes[, flags]]) -> nbytes_read\n\
Thomas Wouters477c8d52006-05-27 19:21:47 +00002338\n\
2339A version of recv() that stores its data into a buffer rather than creating \n\
2340a new string. Receive up to buffersize bytes from the socket. If buffersize \n\
2341is not specified (or 0), receive up to the size available in the given buffer.\n\
2342\n\
2343See recv() for documentation about the flags.");
2344
2345
2346/*
Christian Heimes99170a52007-12-19 02:07:34 +00002347 * This is the guts of the recvfrom() and recvfrom_into() methods, which reads
2348 * into a char buffer. If you have any inc/def ref to do to the objects that
2349 * contain the buffer, do it in the caller. This function returns the number
2350 * of bytes succesfully read. If there was an error, it returns -1. Note
2351 * that it is also possible that we return a number of bytes smaller than the
2352 * request bytes.
Thomas Wouters477c8d52006-05-27 19:21:47 +00002353 *
2354 * 'addr' is a return value for the address object. Note that you must decref
2355 * it yourself.
2356 */
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002357static ssize_t
Thomas Wouters477c8d52006-05-27 19:21:47 +00002358sock_recvfrom_guts(PySocketSockObject *s, char* cbuf, int len, int flags,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002359 PyObject** addr)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002360{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002361 sock_addr_t addrbuf;
2362 int timeout;
2363 ssize_t n = -1;
2364 socklen_t addrlen;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002365
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002366 *addr = NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002367
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002368 if (!getsockaddrlen(s, &addrlen))
2369 return -1;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002370
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002371 if (!IS_SELECTABLE(s)) {
2372 select_error();
2373 return -1;
2374 }
Neal Norwitz082b2df2006-02-07 07:04:46 +00002375
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002376 Py_BEGIN_ALLOW_THREADS
2377 memset(&addrbuf, 0, addrlen);
2378 timeout = internal_select(s, 0);
2379 if (!timeout) {
Guido van Rossum8d665e61996-06-26 18:22:49 +00002380#ifndef MS_WINDOWS
Andrew MacIntyreba43e872002-03-03 03:03:52 +00002381#if defined(PYOS_OS2) && !defined(PYCC_GCC)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002382 n = recvfrom(s->sock_fd, cbuf, len, flags,
2383 SAS2SA(&addrbuf), &addrlen);
Guido van Rossum32c575d1997-12-02 20:37:32 +00002384#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002385 n = recvfrom(s->sock_fd, cbuf, len, flags,
2386 (void *) &addrbuf, &addrlen);
Guido van Rossum32c575d1997-12-02 20:37:32 +00002387#endif
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00002388#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002389 n = recvfrom(s->sock_fd, cbuf, len, flags,
2390 SAS2SA(&addrbuf), &addrlen);
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00002391#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002392 }
2393 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00002394
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002395 if (timeout == 1) {
2396 PyErr_SetString(socket_timeout, "timed out");
2397 return -1;
2398 }
2399 if (n < 0) {
2400 s->errorhandler();
2401 return -1;
2402 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00002403
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002404 if (!(*addr = makesockaddr(s->sock_fd, SAS2SA(&addrbuf),
2405 addrlen, s->sock_proto)))
2406 return -1;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002407
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002408 return n;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002409}
2410
2411/* s.recvfrom(nbytes [,flags]) method */
2412
2413static PyObject *
2414sock_recvfrom(PySocketSockObject *s, PyObject *args)
2415{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002416 PyObject *buf = NULL;
2417 PyObject *addr = NULL;
2418 PyObject *ret = NULL;
2419 int recvlen, flags = 0;
2420 ssize_t outlen;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002421
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002422 if (!PyArg_ParseTuple(args, "i|i:recvfrom", &recvlen, &flags))
2423 return NULL;
Guido van Rossum48a680c2001-03-02 06:34:14 +00002424
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002425 if (recvlen < 0) {
2426 PyErr_SetString(PyExc_ValueError,
2427 "negative buffersize in recvfrom");
2428 return NULL;
2429 }
Guido van Rossumd8faa362007-04-27 19:54:29 +00002430
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002431 buf = PyBytes_FromStringAndSize((char *) 0, recvlen);
2432 if (buf == NULL)
2433 return NULL;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002434
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002435 outlen = sock_recvfrom_guts(s, PyBytes_AS_STRING(buf),
2436 recvlen, flags, &addr);
2437 if (outlen < 0) {
2438 goto finally;
2439 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002440
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002441 if (outlen != recvlen) {
2442 /* We did not read as many bytes as we anticipated, resize the
2443 string if possible and be succesful. */
2444 if (_PyBytes_Resize(&buf, outlen) < 0)
2445 /* Oopsy, not so succesful after all. */
2446 goto finally;
2447 }
Barry Warsaw752300b1997-01-03 17:18:10 +00002448
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002449 ret = PyTuple_Pack(2, buf, addr);
Guido van Rossum67f7a382002-06-06 21:08:16 +00002450
2451finally:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002452 Py_XDECREF(buf);
2453 Py_XDECREF(addr);
2454 return ret;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002455}
2456
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002457PyDoc_STRVAR(recvfrom_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002458"recvfrom(buffersize[, flags]) -> (data, address info)\n\
2459\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002460Like recv(buffersize, flags) but also return the sender's address info.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002461
Thomas Wouters477c8d52006-05-27 19:21:47 +00002462
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002463/* s.recvfrom_into(buffer[, nbytes [,flags]]) method */
Thomas Wouters477c8d52006-05-27 19:21:47 +00002464
2465static PyObject *
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002466sock_recvfrom_into(PySocketSockObject *s, PyObject *args, PyObject* kwds)
Thomas Wouters477c8d52006-05-27 19:21:47 +00002467{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002468 static char *kwlist[] = {"buffer", "nbytes", "flags", 0};
Thomas Wouters477c8d52006-05-27 19:21:47 +00002469
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002470 int recvlen = 0, flags = 0;
2471 ssize_t readlen;
2472 Py_buffer pbuf;
2473 char *buf;
2474 int buflen;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002475
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002476 PyObject *addr = NULL;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002477
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002478 if (!PyArg_ParseTupleAndKeywords(args, kwds, "w*|ii:recvfrom_into",
2479 kwlist, &pbuf,
2480 &recvlen, &flags))
2481 return NULL;
2482 buf = pbuf.buf;
2483 buflen = pbuf.len;
2484 assert(buf != 0 && buflen > 0);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002485
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002486 if (recvlen < 0) {
2487 PyBuffer_Release(&pbuf);
2488 PyErr_SetString(PyExc_ValueError,
2489 "negative buffersize in recvfrom_into");
2490 return NULL;
2491 }
2492 if (recvlen == 0) {
2493 /* If nbytes was not specified, use the buffer's length */
2494 recvlen = buflen;
2495 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002496
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002497 readlen = sock_recvfrom_guts(s, buf, recvlen, flags, &addr);
2498 if (readlen < 0) {
2499 PyBuffer_Release(&pbuf);
2500 /* Return an error */
2501 Py_XDECREF(addr);
2502 return NULL;
2503 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002504
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002505 PyBuffer_Release(&pbuf);
2506 /* Return the number of bytes read and the address. Note that we do
2507 not do anything special here in the case that readlen < recvlen. */
2508 return Py_BuildValue("lN", readlen, addr);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002509}
2510
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002511PyDoc_STRVAR(recvfrom_into_doc,
2512"recvfrom_into(buffer[, nbytes[, flags]]) -> (nbytes, address info)\n\
Thomas Wouters477c8d52006-05-27 19:21:47 +00002513\n\
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002514Like recv_into(buffer[, nbytes[, flags]]) but also return the sender's address info.");
Thomas Wouters477c8d52006-05-27 19:21:47 +00002515
2516
Guido van Rossumeb6b33a1993-05-25 09:38:27 +00002517/* s.send(data [,flags]) method */
Guido van Rossum30a685f1991-06-27 15:51:29 +00002518
Guido van Rossum73624e91994-10-10 17:59:00 +00002519static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002520sock_send(PySocketSockObject *s, PyObject *args)
Guido van Rossum30a685f1991-06-27 15:51:29 +00002521{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002522 char *buf;
2523 int len, n = -1, flags = 0, timeout;
2524 Py_buffer pbuf;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002525
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002526 if (!PyArg_ParseTuple(args, "y*|i:send", &pbuf, &flags))
2527 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002528
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002529 if (!IS_SELECTABLE(s)) {
2530 PyBuffer_Release(&pbuf);
2531 return select_error();
2532 }
2533 buf = pbuf.buf;
2534 len = pbuf.len;
Neal Norwitz082b2df2006-02-07 07:04:46 +00002535
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002536 Py_BEGIN_ALLOW_THREADS
2537 timeout = internal_select(s, 1);
2538 if (!timeout)
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002539#ifdef __VMS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002540 n = sendsegmented(s->sock_fd, buf, len, flags);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002541#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002542 n = send(s->sock_fd, buf, len, flags);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002543#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002544 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00002545
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002546 PyBuffer_Release(&pbuf);
Martin v. Löwis423be952008-08-13 15:53:07 +00002547
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002548 if (timeout == 1) {
2549 PyErr_SetString(socket_timeout, "timed out");
2550 return NULL;
2551 }
2552 if (n < 0)
2553 return s->errorhandler();
2554 return PyLong_FromLong((long)n);
Guido van Rossum30a685f1991-06-27 15:51:29 +00002555}
2556
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002557PyDoc_STRVAR(send_doc,
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002558"send(data[, flags]) -> count\n\
Guido van Rossum82a5c661998-07-07 20:45:43 +00002559\n\
2560Send a data string to the socket. For the optional flags\n\
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002561argument, see the Unix manual. Return the number of bytes\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002562sent; this may be less than len(data) if the network is busy.");
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002563
2564
2565/* s.sendall(data [,flags]) method */
2566
2567static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002568sock_sendall(PySocketSockObject *s, PyObject *args)
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002569{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002570 char *buf;
2571 int len, n = -1, flags = 0, timeout;
2572 Py_buffer pbuf;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002573
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002574 if (!PyArg_ParseTuple(args, "y*|i:sendall", &pbuf, &flags))
2575 return NULL;
2576 buf = pbuf.buf;
2577 len = pbuf.len;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002578
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002579 if (!IS_SELECTABLE(s)) {
2580 PyBuffer_Release(&pbuf);
2581 return select_error();
2582 }
Neal Norwitz082b2df2006-02-07 07:04:46 +00002583
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002584 Py_BEGIN_ALLOW_THREADS
2585 do {
2586 timeout = internal_select(s, 1);
2587 n = -1;
2588 if (timeout)
2589 break;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002590#ifdef __VMS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002591 n = sendsegmented(s->sock_fd, buf, len, flags);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002592#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002593 n = send(s->sock_fd, buf, len, flags);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002594#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002595 if (n < 0) {
Gregory P. Smithb585a0f2010-01-04 03:29:50 +00002596#ifdef EINTR
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002597 /* We must handle EINTR here as there is no way for
2598 * the caller to know how much was sent otherwise. */
2599 if (errno == EINTR) {
2600 /* Run signal handlers. If an exception was
2601 * raised, abort and leave this socket in
2602 * an unknown state. */
2603 if (PyErr_CheckSignals())
2604 return NULL;
2605 continue;
2606 }
Gregory P. Smithb585a0f2010-01-04 03:29:50 +00002607#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002608 break;
2609 }
2610 buf += n;
2611 len -= n;
2612 } while (len > 0);
2613 Py_END_ALLOW_THREADS
2614 PyBuffer_Release(&pbuf);
Guido van Rossum67f7a382002-06-06 21:08:16 +00002615
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002616 if (timeout == 1) {
2617 PyErr_SetString(socket_timeout, "timed out");
2618 return NULL;
2619 }
2620 if (n < 0)
2621 return s->errorhandler();
Guido van Rossum67f7a382002-06-06 21:08:16 +00002622
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002623 Py_INCREF(Py_None);
2624 return Py_None;
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002625}
2626
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002627PyDoc_STRVAR(sendall_doc,
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002628"sendall(data[, flags])\n\
2629\n\
2630Send a data string to the socket. For the optional flags\n\
2631argument, see the Unix manual. This calls send() repeatedly\n\
2632until all data is sent. If an error occurs, it's impossible\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002633to tell how much data has been sent.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002634
Guido van Rossum30a685f1991-06-27 15:51:29 +00002635
Guido van Rossumeb6b33a1993-05-25 09:38:27 +00002636/* s.sendto(data, [flags,] sockaddr) method */
Guido van Rossum30a685f1991-06-27 15:51:29 +00002637
Guido van Rossum73624e91994-10-10 17:59:00 +00002638static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002639sock_sendto(PySocketSockObject *s, PyObject *args)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002640{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002641 Py_buffer pbuf;
2642 PyObject *addro;
2643 char *buf;
2644 Py_ssize_t len;
2645 sock_addr_t addrbuf;
2646 int addrlen, n = -1, flags, timeout;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002647
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002648 flags = 0;
2649 if (!PyArg_ParseTuple(args, "y*O:sendto", &pbuf, &addro)) {
2650 PyErr_Clear();
2651 if (!PyArg_ParseTuple(args, "y*iO:sendto",
2652 &pbuf, &flags, &addro))
2653 return NULL;
2654 }
2655 buf = pbuf.buf;
2656 len = pbuf.len;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002657
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002658 if (!IS_SELECTABLE(s)) {
2659 PyBuffer_Release(&pbuf);
2660 return select_error();
2661 }
Neal Norwitz082b2df2006-02-07 07:04:46 +00002662
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002663 if (!getsockaddrarg(s, addro, SAS2SA(&addrbuf), &addrlen)) {
2664 PyBuffer_Release(&pbuf);
2665 return NULL;
2666 }
Thomas Wouters89f507f2006-12-13 04:49:30 +00002667
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002668 Py_BEGIN_ALLOW_THREADS
2669 timeout = internal_select(s, 1);
2670 if (!timeout)
2671 n = sendto(s->sock_fd, buf, len, flags, SAS2SA(&addrbuf), addrlen);
2672 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00002673
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002674 PyBuffer_Release(&pbuf);
2675 if (timeout == 1) {
2676 PyErr_SetString(socket_timeout, "timed out");
2677 return NULL;
2678 }
2679 if (n < 0)
2680 return s->errorhandler();
2681 return PyLong_FromLong((long)n);
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002682}
2683
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002684PyDoc_STRVAR(sendto_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00002685"sendto(data[, flags], address) -> count\n\
Guido van Rossum82a5c661998-07-07 20:45:43 +00002686\n\
2687Like send(data, flags) but allows specifying the destination address.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002688For IP sockets, the address is a pair (hostaddr, port).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002689
Guido van Rossum30a685f1991-06-27 15:51:29 +00002690
2691/* s.shutdown(how) method */
2692
Guido van Rossum73624e91994-10-10 17:59:00 +00002693static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002694sock_shutdown(PySocketSockObject *s, PyObject *arg)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002695{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002696 int how;
2697 int res;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00002698
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002699 how = PyLong_AsLong(arg);
2700 if (how == -1 && PyErr_Occurred())
2701 return NULL;
2702 Py_BEGIN_ALLOW_THREADS
2703 res = shutdown(s->sock_fd, how);
2704 Py_END_ALLOW_THREADS
2705 if (res < 0)
2706 return s->errorhandler();
2707 Py_INCREF(Py_None);
2708 return Py_None;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002709}
2710
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002711PyDoc_STRVAR(shutdown_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002712"shutdown(flag)\n\
2713\n\
Martin v. Löwis94681fc2003-11-27 19:40:22 +00002714Shut down the reading side of the socket (flag == SHUT_RD), the writing side\n\
2715of the socket (flag == SHUT_WR), or both ends (flag == SHUT_RDWR).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002716
Amaury Forgeot d'Arc3d17a5c2008-06-13 01:09:34 +00002717#if defined(MS_WINDOWS) && defined(SIO_RCVALL)
Christian Heimesfaf2f632008-01-06 16:59:19 +00002718static PyObject*
2719sock_ioctl(PySocketSockObject *s, PyObject *arg)
2720{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002721 unsigned long cmd = SIO_RCVALL;
2722 PyObject *argO;
2723 DWORD recv;
Christian Heimesfaf2f632008-01-06 16:59:19 +00002724
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002725 if (!PyArg_ParseTuple(arg, "kO:ioctl", &cmd, &argO))
2726 return NULL;
Christian Heimesfaf2f632008-01-06 16:59:19 +00002727
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002728 switch (cmd) {
2729 case SIO_RCVALL: {
2730 unsigned int option = RCVALL_ON;
2731 if (!PyArg_ParseTuple(arg, "kI:ioctl", &cmd, &option))
2732 return NULL;
2733 if (WSAIoctl(s->sock_fd, cmd, &option, sizeof(option),
2734 NULL, 0, &recv, NULL, NULL) == SOCKET_ERROR) {
2735 return set_error();
2736 }
2737 return PyLong_FromUnsignedLong(recv); }
2738 case SIO_KEEPALIVE_VALS: {
2739 struct tcp_keepalive ka;
2740 if (!PyArg_ParseTuple(arg, "k(kkk):ioctl", &cmd,
2741 &ka.onoff, &ka.keepalivetime, &ka.keepaliveinterval))
2742 return NULL;
2743 if (WSAIoctl(s->sock_fd, cmd, &ka, sizeof(ka),
2744 NULL, 0, &recv, NULL, NULL) == SOCKET_ERROR) {
2745 return set_error();
2746 }
2747 return PyLong_FromUnsignedLong(recv); }
2748 default:
2749 PyErr_Format(PyExc_ValueError, "invalid ioctl command %d", cmd);
2750 return NULL;
2751 }
Christian Heimesfaf2f632008-01-06 16:59:19 +00002752}
2753PyDoc_STRVAR(sock_ioctl_doc,
2754"ioctl(cmd, option) -> long\n\
2755\n\
Kristján Valur Jónsson847ec752009-09-27 21:10:38 +00002756Control the socket with WSAIoctl syscall. Currently supported 'cmd' values are\n\
2757SIO_RCVALL: 'option' must be one of the socket.RCVALL_* constants.\n\
2758SIO_KEEPALIVE_VALS: 'option' is a tuple of (onoff, timeout, interval).");
Christian Heimesfaf2f632008-01-06 16:59:19 +00002759
2760#endif
Guido van Rossum30a685f1991-06-27 15:51:29 +00002761
2762/* List of methods for socket objects */
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002763
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002764static PyMethodDef sock_methods[] = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002765 {"_accept", (PyCFunction)sock_accept, METH_NOARGS,
2766 accept_doc},
2767 {"bind", (PyCFunction)sock_bind, METH_O,
2768 bind_doc},
2769 {"close", (PyCFunction)sock_close, METH_NOARGS,
2770 close_doc},
2771 {"connect", (PyCFunction)sock_connect, METH_O,
2772 connect_doc},
2773 {"connect_ex", (PyCFunction)sock_connect_ex, METH_O,
2774 connect_ex_doc},
2775 {"fileno", (PyCFunction)sock_fileno, METH_NOARGS,
2776 fileno_doc},
Antoine Pitroue43f9d02010-08-08 23:24:50 +00002777 {"forget", (PyCFunction)sock_forget, METH_NOARGS,
2778 forget_doc},
Guido van Rossumb6775db1994-08-01 11:34:53 +00002779#ifdef HAVE_GETPEERNAME
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002780 {"getpeername", (PyCFunction)sock_getpeername,
2781 METH_NOARGS, getpeername_doc},
Guido van Rossum9575a441993-04-07 14:06:14 +00002782#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002783 {"getsockname", (PyCFunction)sock_getsockname,
2784 METH_NOARGS, getsockname_doc},
2785 {"getsockopt", (PyCFunction)sock_getsockopt, METH_VARARGS,
2786 getsockopt_doc},
Amaury Forgeot d'Arc3d17a5c2008-06-13 01:09:34 +00002787#if defined(MS_WINDOWS) && defined(SIO_RCVALL)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002788 {"ioctl", (PyCFunction)sock_ioctl, METH_VARARGS,
2789 sock_ioctl_doc},
Christian Heimesfaf2f632008-01-06 16:59:19 +00002790#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002791 {"listen", (PyCFunction)sock_listen, METH_O,
2792 listen_doc},
2793 {"recv", (PyCFunction)sock_recv, METH_VARARGS,
2794 recv_doc},
2795 {"recv_into", (PyCFunction)sock_recv_into, METH_VARARGS | METH_KEYWORDS,
2796 recv_into_doc},
2797 {"recvfrom", (PyCFunction)sock_recvfrom, METH_VARARGS,
2798 recvfrom_doc},
2799 {"recvfrom_into", (PyCFunction)sock_recvfrom_into, METH_VARARGS | METH_KEYWORDS,
2800 recvfrom_into_doc},
2801 {"send", (PyCFunction)sock_send, METH_VARARGS,
2802 send_doc},
2803 {"sendall", (PyCFunction)sock_sendall, METH_VARARGS,
2804 sendall_doc},
2805 {"sendto", (PyCFunction)sock_sendto, METH_VARARGS,
2806 sendto_doc},
2807 {"setblocking", (PyCFunction)sock_setblocking, METH_O,
2808 setblocking_doc},
2809 {"settimeout", (PyCFunction)sock_settimeout, METH_O,
2810 settimeout_doc},
2811 {"gettimeout", (PyCFunction)sock_gettimeout, METH_NOARGS,
2812 gettimeout_doc},
2813 {"setsockopt", (PyCFunction)sock_setsockopt, METH_VARARGS,
2814 setsockopt_doc},
2815 {"shutdown", (PyCFunction)sock_shutdown, METH_O,
2816 shutdown_doc},
2817 {NULL, NULL} /* sentinel */
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002818};
2819
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002820/* SockObject members */
2821static PyMemberDef sock_memberlist[] = {
2822 {"family", T_INT, offsetof(PySocketSockObject, sock_family), READONLY, "the socket family"},
2823 {"type", T_INT, offsetof(PySocketSockObject, sock_type), READONLY, "the socket type"},
2824 {"proto", T_INT, offsetof(PySocketSockObject, sock_proto), READONLY, "the socket protocol"},
2825 {"timeout", T_DOUBLE, offsetof(PySocketSockObject, sock_timeout), READONLY, "the socket timeout"},
2826 {0},
2827};
Guido van Rossum30a685f1991-06-27 15:51:29 +00002828
Guido van Rossum73624e91994-10-10 17:59:00 +00002829/* Deallocate a socket object in response to the last Py_DECREF().
Guido van Rossum30a685f1991-06-27 15:51:29 +00002830 First close the file description. */
2831
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002832static void
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002833sock_dealloc(PySocketSockObject *s)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002834{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002835 if (s->sock_fd != -1)
2836 (void) SOCKETCLOSE(s->sock_fd);
2837 Py_TYPE(s)->tp_free((PyObject *)s);
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002838}
2839
Guido van Rossum30a685f1991-06-27 15:51:29 +00002840
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00002841static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002842sock_repr(PySocketSockObject *s)
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00002843{
Fred Drakea04eaad2000-06-30 02:46:07 +00002844#if SIZEOF_SOCKET_T > SIZEOF_LONG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002845 if (s->sock_fd > LONG_MAX) {
2846 /* this can occur on Win64, and actually there is a special
2847 ugly printf formatter for decimal pointer length integer
2848 printing, only bother if necessary*/
2849 PyErr_SetString(PyExc_OverflowError,
2850 "no printf formatter to display "
2851 "the socket descriptor in decimal");
2852 return NULL;
2853 }
Fred Drakea04eaad2000-06-30 02:46:07 +00002854#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002855 return PyUnicode_FromFormat(
2856 "<socket object, fd=%ld, family=%d, type=%d, proto=%d>",
2857 (long)s->sock_fd, s->sock_family,
2858 s->sock_type,
2859 s->sock_proto);
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00002860}
2861
2862
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002863/* Create a new, uninitialized socket object. */
2864
2865static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002866sock_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002867{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002868 PyObject *new;
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002869
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002870 new = type->tp_alloc(type, 0);
2871 if (new != NULL) {
2872 ((PySocketSockObject *)new)->sock_fd = -1;
2873 ((PySocketSockObject *)new)->sock_timeout = -1.0;
2874 ((PySocketSockObject *)new)->errorhandler = &set_error;
2875 }
2876 return new;
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002877}
2878
2879
2880/* Initialize a new socket object. */
2881
2882/*ARGSUSED*/
2883static int
Andrew MacIntyre7aec4a22002-06-13 11:53:52 +00002884sock_initobj(PyObject *self, PyObject *args, PyObject *kwds)
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002885{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002886 PySocketSockObject *s = (PySocketSockObject *)self;
2887 PyObject *fdobj = NULL;
2888 SOCKET_T fd = INVALID_SOCKET;
2889 int family = AF_INET, type = SOCK_STREAM, proto = 0;
2890 static char *keywords[] = {"family", "type", "proto", "fileno", 0};
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002891
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002892 if (!PyArg_ParseTupleAndKeywords(args, kwds,
2893 "|iiiO:socket", keywords,
2894 &family, &type, &proto, &fdobj))
2895 return -1;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002896
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002897 if (fdobj != NULL && fdobj != Py_None) {
2898 fd = PyLong_AsSocket_t(fdobj);
2899 if (fd == (SOCKET_T)(-1) && PyErr_Occurred())
2900 return -1;
2901 if (fd == INVALID_SOCKET) {
2902 PyErr_SetString(PyExc_ValueError,
2903 "can't use invalid socket value");
2904 return -1;
2905 }
2906 }
2907 else {
2908 Py_BEGIN_ALLOW_THREADS
2909 fd = socket(family, type, proto);
2910 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00002911
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002912 if (fd == INVALID_SOCKET) {
2913 set_error();
2914 return -1;
2915 }
2916 }
2917 init_sockobject(s, fd, family, type, proto);
Guido van Rossum67f7a382002-06-06 21:08:16 +00002918
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002919 return 0;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002920
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002921}
2922
2923
Guido van Rossumb6775db1994-08-01 11:34:53 +00002924/* Type object for socket objects. */
Guido van Rossum30a685f1991-06-27 15:51:29 +00002925
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002926static PyTypeObject sock_type = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002927 PyVarObject_HEAD_INIT(0, 0) /* Must fill in type value later */
2928 "_socket.socket", /* tp_name */
2929 sizeof(PySocketSockObject), /* tp_basicsize */
2930 0, /* tp_itemsize */
2931 (destructor)sock_dealloc, /* tp_dealloc */
2932 0, /* tp_print */
2933 0, /* tp_getattr */
2934 0, /* tp_setattr */
2935 0, /* tp_reserved */
2936 (reprfunc)sock_repr, /* tp_repr */
2937 0, /* tp_as_number */
2938 0, /* tp_as_sequence */
2939 0, /* tp_as_mapping */
2940 0, /* tp_hash */
2941 0, /* tp_call */
2942 0, /* tp_str */
2943 PyObject_GenericGetAttr, /* tp_getattro */
2944 0, /* tp_setattro */
2945 0, /* tp_as_buffer */
2946 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE, /* tp_flags */
2947 sock_doc, /* tp_doc */
2948 0, /* tp_traverse */
2949 0, /* tp_clear */
2950 0, /* tp_richcompare */
2951 0, /* tp_weaklistoffset */
2952 0, /* tp_iter */
2953 0, /* tp_iternext */
2954 sock_methods, /* tp_methods */
2955 sock_memberlist, /* tp_members */
2956 0, /* tp_getset */
2957 0, /* tp_base */
2958 0, /* tp_dict */
2959 0, /* tp_descr_get */
2960 0, /* tp_descr_set */
2961 0, /* tp_dictoffset */
2962 sock_initobj, /* tp_init */
2963 PyType_GenericAlloc, /* tp_alloc */
2964 sock_new, /* tp_new */
2965 PyObject_Del, /* tp_free */
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002966};
2967
Guido van Rossum30a685f1991-06-27 15:51:29 +00002968
Guido van Rossum81194471991-07-27 21:42:02 +00002969/* Python interface to gethostname(). */
2970
2971/*ARGSUSED*/
Guido van Rossum73624e91994-10-10 17:59:00 +00002972static PyObject *
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00002973socket_gethostname(PyObject *self, PyObject *unused)
Guido van Rossum81194471991-07-27 21:42:02 +00002974{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002975 char buf[1024];
2976 int res;
2977 Py_BEGIN_ALLOW_THREADS
2978 res = gethostname(buf, (int) sizeof buf - 1);
2979 Py_END_ALLOW_THREADS
2980 if (res < 0)
2981 return set_error();
2982 buf[sizeof buf - 1] = '\0';
2983 return PyUnicode_FromString(buf);
Guido van Rossum81194471991-07-27 21:42:02 +00002984}
Guido van Rossumff4949e1992-08-05 19:58:53 +00002985
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002986PyDoc_STRVAR(gethostname_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002987"gethostname() -> string\n\
2988\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002989Return the current host name.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002990
Guido van Rossumff4949e1992-08-05 19:58:53 +00002991
Guido van Rossum30a685f1991-06-27 15:51:29 +00002992/* Python interface to gethostbyname(name). */
2993
2994/*ARGSUSED*/
Guido van Rossum73624e91994-10-10 17:59:00 +00002995static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002996socket_gethostbyname(PyObject *self, PyObject *args)
Guido van Rossum30a685f1991-06-27 15:51:29 +00002997{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002998 char *name;
2999 sock_addr_t addrbuf;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003000
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003001 if (!PyArg_ParseTuple(args, "s:gethostbyname", &name))
3002 return NULL;
3003 if (setipaddr(name, SAS2SA(&addrbuf), sizeof(addrbuf), AF_INET) < 0)
3004 return NULL;
3005 return makeipaddr(SAS2SA(&addrbuf), sizeof(struct sockaddr_in));
Guido van Rossum30a685f1991-06-27 15:51:29 +00003006}
3007
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003008PyDoc_STRVAR(gethostbyname_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003009"gethostbyname(host) -> address\n\
3010\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003011Return the IP address (a string of the form '255.255.255.255') for a host.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003012
3013
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003014/* Convenience function common to gethostbyname_ex and gethostbyaddr */
3015
3016static PyObject *
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003017gethost_common(struct hostent *h, struct sockaddr *addr, int alen, int af)
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003018{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003019 char **pch;
3020 PyObject *rtn_tuple = (PyObject *)NULL;
3021 PyObject *name_list = (PyObject *)NULL;
3022 PyObject *addr_list = (PyObject *)NULL;
3023 PyObject *tmp;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003024
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003025 if (h == NULL) {
3026 /* Let's get real error message to return */
3027 set_herror(h_errno);
3028 return NULL;
3029 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00003030
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003031 if (h->h_addrtype != af) {
3032 /* Let's get real error message to return */
3033 PyErr_SetString(socket_error,
3034 (char *)strerror(EAFNOSUPPORT));
Christian Heimesada8c3b2008-03-18 18:26:33 +00003035
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003036 return NULL;
3037 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00003038
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003039 switch (af) {
Guido van Rossum67f7a382002-06-06 21:08:16 +00003040
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003041 case AF_INET:
3042 if (alen < sizeof(struct sockaddr_in))
3043 return NULL;
3044 break;
Guido van Rossum67f7a382002-06-06 21:08:16 +00003045
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00003046#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003047 case AF_INET6:
3048 if (alen < sizeof(struct sockaddr_in6))
3049 return NULL;
3050 break;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003051#endif
Guido van Rossum67f7a382002-06-06 21:08:16 +00003052
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003053 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00003054
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003055 if ((name_list = PyList_New(0)) == NULL)
3056 goto err;
Guido van Rossum67f7a382002-06-06 21:08:16 +00003057
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003058 if ((addr_list = PyList_New(0)) == NULL)
3059 goto err;
Guido van Rossum67f7a382002-06-06 21:08:16 +00003060
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003061 /* SF #1511317: h_aliases can be NULL */
3062 if (h->h_aliases) {
3063 for (pch = h->h_aliases; *pch != NULL; pch++) {
3064 int status;
3065 tmp = PyUnicode_FromString(*pch);
3066 if (tmp == NULL)
3067 goto err;
Guido van Rossum67f7a382002-06-06 21:08:16 +00003068
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003069 status = PyList_Append(name_list, tmp);
3070 Py_DECREF(tmp);
Guido van Rossum67f7a382002-06-06 21:08:16 +00003071
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003072 if (status)
3073 goto err;
3074 }
3075 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00003076
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003077 for (pch = h->h_addr_list; *pch != NULL; pch++) {
3078 int status;
Guido van Rossum67f7a382002-06-06 21:08:16 +00003079
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003080 switch (af) {
Guido van Rossum67f7a382002-06-06 21:08:16 +00003081
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003082 case AF_INET:
3083 {
3084 struct sockaddr_in sin;
3085 memset(&sin, 0, sizeof(sin));
3086 sin.sin_family = af;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003087#ifdef HAVE_SOCKADDR_SA_LEN
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003088 sin.sin_len = sizeof(sin);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003089#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003090 memcpy(&sin.sin_addr, *pch, sizeof(sin.sin_addr));
3091 tmp = makeipaddr((struct sockaddr *)&sin, sizeof(sin));
Guido van Rossum67f7a382002-06-06 21:08:16 +00003092
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003093 if (pch == h->h_addr_list && alen >= sizeof(sin))
3094 memcpy((char *) addr, &sin, sizeof(sin));
3095 break;
3096 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00003097
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00003098#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003099 case AF_INET6:
3100 {
3101 struct sockaddr_in6 sin6;
3102 memset(&sin6, 0, sizeof(sin6));
3103 sin6.sin6_family = af;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003104#ifdef HAVE_SOCKADDR_SA_LEN
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003105 sin6.sin6_len = sizeof(sin6);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003106#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003107 memcpy(&sin6.sin6_addr, *pch, sizeof(sin6.sin6_addr));
3108 tmp = makeipaddr((struct sockaddr *)&sin6,
3109 sizeof(sin6));
Guido van Rossum67f7a382002-06-06 21:08:16 +00003110
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003111 if (pch == h->h_addr_list && alen >= sizeof(sin6))
3112 memcpy((char *) addr, &sin6, sizeof(sin6));
3113 break;
3114 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003115#endif
Guido van Rossum67f7a382002-06-06 21:08:16 +00003116
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003117 default: /* can't happen */
3118 PyErr_SetString(socket_error,
3119 "unsupported address family");
3120 return NULL;
3121 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00003122
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003123 if (tmp == NULL)
3124 goto err;
Guido van Rossum67f7a382002-06-06 21:08:16 +00003125
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003126 status = PyList_Append(addr_list, tmp);
3127 Py_DECREF(tmp);
Guido van Rossum67f7a382002-06-06 21:08:16 +00003128
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003129 if (status)
3130 goto err;
3131 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00003132
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003133 rtn_tuple = Py_BuildValue("sOO", h->h_name, name_list, addr_list);
Guido van Rossum67f7a382002-06-06 21:08:16 +00003134
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003135 err:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003136 Py_XDECREF(name_list);
3137 Py_XDECREF(addr_list);
3138 return rtn_tuple;
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003139}
3140
3141
3142/* Python interface to gethostbyname_ex(name). */
3143
3144/*ARGSUSED*/
3145static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003146socket_gethostbyname_ex(PyObject *self, PyObject *args)
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003147{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003148 char *name;
3149 struct hostent *h;
Martin v. Löwisc16f3bd2003-05-03 09:14:54 +00003150#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003151 struct sockaddr_storage addr;
Martin v. Löwisc16f3bd2003-05-03 09:14:54 +00003152#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003153 struct sockaddr_in addr;
Martin v. Löwisc16f3bd2003-05-03 09:14:54 +00003154#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003155 struct sockaddr *sa;
3156 PyObject *ret;
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003157#ifdef HAVE_GETHOSTBYNAME_R
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003158 struct hostent hp_allocated;
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003159#ifdef HAVE_GETHOSTBYNAME_R_3_ARG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003160 struct hostent_data data;
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003161#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003162 char buf[16384];
3163 int buf_len = (sizeof buf) - 1;
3164 int errnop;
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003165#endif
3166#if defined(HAVE_GETHOSTBYNAME_R_3_ARG) || defined(HAVE_GETHOSTBYNAME_R_6_ARG)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003167 int result;
Guido van Rossume9cd07b1999-03-15 21:40:14 +00003168#endif
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003169#endif /* HAVE_GETHOSTBYNAME_R */
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003170
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003171 if (!PyArg_ParseTuple(args, "s:gethostbyname_ex", &name))
3172 return NULL;
3173 if (setipaddr(name, (struct sockaddr *)&addr, sizeof(addr), AF_INET) < 0)
3174 return NULL;
3175 Py_BEGIN_ALLOW_THREADS
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003176#ifdef HAVE_GETHOSTBYNAME_R
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003177#if defined(HAVE_GETHOSTBYNAME_R_6_ARG)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003178 result = gethostbyname_r(name, &hp_allocated, buf, buf_len,
3179 &h, &errnop);
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003180#elif defined(HAVE_GETHOSTBYNAME_R_5_ARG)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003181 h = gethostbyname_r(name, &hp_allocated, buf, buf_len, &errnop);
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003182#else /* HAVE_GETHOSTBYNAME_R_3_ARG */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003183 memset((void *) &data, '\0', sizeof(data));
3184 result = gethostbyname_r(name, &hp_allocated, &data);
3185 h = (result != 0) ? NULL : &hp_allocated;
Guido van Rossume9cd07b1999-03-15 21:40:14 +00003186#endif
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003187#else /* not HAVE_GETHOSTBYNAME_R */
Guido van Rossum3baaa131999-03-22 21:44:51 +00003188#ifdef USE_GETHOSTBYNAME_LOCK
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003189 PyThread_acquire_lock(netdb_lock, 1);
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003190#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003191 h = gethostbyname(name);
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003192#endif /* HAVE_GETHOSTBYNAME_R */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003193 Py_END_ALLOW_THREADS
3194 /* Some C libraries would require addr.__ss_family instead of
3195 addr.ss_family.
3196 Therefore, we cast the sockaddr_storage into sockaddr to
3197 access sa_family. */
3198 sa = (struct sockaddr*)&addr;
3199 ret = gethost_common(h, (struct sockaddr *)&addr, sizeof(addr),
3200 sa->sa_family);
Guido van Rossum3baaa131999-03-22 21:44:51 +00003201#ifdef USE_GETHOSTBYNAME_LOCK
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003202 PyThread_release_lock(netdb_lock);
Guido van Rossum955becc1999-03-22 20:14:53 +00003203#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003204 return ret;
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003205}
3206
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003207PyDoc_STRVAR(ghbn_ex_doc,
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003208"gethostbyname_ex(host) -> (name, aliaslist, addresslist)\n\
3209\n\
3210Return the true host name, a list of aliases, and a list of IP addresses,\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003211for a host. The host argument is a string giving a host name or IP number.");
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003212
3213
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00003214/* Python interface to gethostbyaddr(IP). */
3215
3216/*ARGSUSED*/
3217static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003218socket_gethostbyaddr(PyObject *self, PyObject *args)
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00003219{
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00003220#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003221 struct sockaddr_storage addr;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003222#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003223 struct sockaddr_in addr;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003224#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003225 struct sockaddr *sa = (struct sockaddr *)&addr;
3226 char *ip_num;
3227 struct hostent *h;
3228 PyObject *ret;
Guido van Rossum4f199ea1998-04-09 20:56:35 +00003229#ifdef HAVE_GETHOSTBYNAME_R
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003230 struct hostent hp_allocated;
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003231#ifdef HAVE_GETHOSTBYNAME_R_3_ARG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003232 struct hostent_data data;
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003233#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003234 /* glibcs up to 2.10 assume that the buf argument to
3235 gethostbyaddr_r is 8-byte aligned, which at least llvm-gcc
3236 does not ensure. The attribute below instructs the compiler
3237 to maintain this alignment. */
3238 char buf[16384] Py_ALIGNED(8);
3239 int buf_len = (sizeof buf) - 1;
3240 int errnop;
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003241#endif
3242#if defined(HAVE_GETHOSTBYNAME_R_3_ARG) || defined(HAVE_GETHOSTBYNAME_R_6_ARG)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003243 int result;
Guido van Rossume9cd07b1999-03-15 21:40:14 +00003244#endif
Guido van Rossum4f199ea1998-04-09 20:56:35 +00003245#endif /* HAVE_GETHOSTBYNAME_R */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003246 char *ap;
3247 int al;
3248 int af;
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00003249
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003250 if (!PyArg_ParseTuple(args, "s:gethostbyaddr", &ip_num))
3251 return NULL;
3252 af = AF_UNSPEC;
3253 if (setipaddr(ip_num, sa, sizeof(addr), af) < 0)
3254 return NULL;
3255 af = sa->sa_family;
3256 ap = NULL;
3257 al = 0;
3258 switch (af) {
3259 case AF_INET:
3260 ap = (char *)&((struct sockaddr_in *)sa)->sin_addr;
3261 al = sizeof(((struct sockaddr_in *)sa)->sin_addr);
3262 break;
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00003263#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003264 case AF_INET6:
3265 ap = (char *)&((struct sockaddr_in6 *)sa)->sin6_addr;
3266 al = sizeof(((struct sockaddr_in6 *)sa)->sin6_addr);
3267 break;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003268#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003269 default:
3270 PyErr_SetString(socket_error, "unsupported address family");
3271 return NULL;
3272 }
3273 Py_BEGIN_ALLOW_THREADS
Guido van Rossum4f199ea1998-04-09 20:56:35 +00003274#ifdef HAVE_GETHOSTBYNAME_R
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003275#if defined(HAVE_GETHOSTBYNAME_R_6_ARG)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003276 result = gethostbyaddr_r(ap, al, af,
3277 &hp_allocated, buf, buf_len,
3278 &h, &errnop);
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003279#elif defined(HAVE_GETHOSTBYNAME_R_5_ARG)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003280 h = gethostbyaddr_r(ap, al, af,
3281 &hp_allocated, buf, buf_len, &errnop);
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003282#else /* HAVE_GETHOSTBYNAME_R_3_ARG */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003283 memset((void *) &data, '\0', sizeof(data));
3284 result = gethostbyaddr_r(ap, al, af, &hp_allocated, &data);
3285 h = (result != 0) ? NULL : &hp_allocated;
Guido van Rossume9cd07b1999-03-15 21:40:14 +00003286#endif
Guido van Rossum4f199ea1998-04-09 20:56:35 +00003287#else /* not HAVE_GETHOSTBYNAME_R */
Guido van Rossum3baaa131999-03-22 21:44:51 +00003288#ifdef USE_GETHOSTBYNAME_LOCK
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003289 PyThread_acquire_lock(netdb_lock, 1);
Guido van Rossum4f199ea1998-04-09 20:56:35 +00003290#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003291 h = gethostbyaddr(ap, al, af);
Guido van Rossum4f199ea1998-04-09 20:56:35 +00003292#endif /* HAVE_GETHOSTBYNAME_R */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003293 Py_END_ALLOW_THREADS
3294 ret = gethost_common(h, (struct sockaddr *)&addr, sizeof(addr), af);
Guido van Rossum3baaa131999-03-22 21:44:51 +00003295#ifdef USE_GETHOSTBYNAME_LOCK
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003296 PyThread_release_lock(netdb_lock);
Guido van Rossum3baaa131999-03-22 21:44:51 +00003297#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003298 return ret;
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00003299}
3300
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003301PyDoc_STRVAR(gethostbyaddr_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003302"gethostbyaddr(host) -> (name, aliaslist, addresslist)\n\
3303\n\
3304Return the true host name, a list of aliases, and a list of IP addresses,\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003305for a host. The host argument is a string giving a host name or IP number.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003306
Guido van Rossum30a685f1991-06-27 15:51:29 +00003307
3308/* Python interface to getservbyname(name).
3309 This only returns the port number, since the other info is already
3310 known or not useful (like the list of aliases). */
3311
3312/*ARGSUSED*/
Guido van Rossum73624e91994-10-10 17:59:00 +00003313static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003314socket_getservbyname(PyObject *self, PyObject *args)
Guido van Rossum30a685f1991-06-27 15:51:29 +00003315{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003316 char *name, *proto=NULL;
3317 struct servent *sp;
3318 if (!PyArg_ParseTuple(args, "s|s:getservbyname", &name, &proto))
3319 return NULL;
3320 Py_BEGIN_ALLOW_THREADS
3321 sp = getservbyname(name, proto);
3322 Py_END_ALLOW_THREADS
3323 if (sp == NULL) {
3324 PyErr_SetString(socket_error, "service/proto not found");
3325 return NULL;
3326 }
3327 return PyLong_FromLong((long) ntohs(sp->s_port));
Guido van Rossum30a685f1991-06-27 15:51:29 +00003328}
3329
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003330PyDoc_STRVAR(getservbyname_doc,
Barry Warsaw11b91a02004-06-28 00:50:43 +00003331"getservbyname(servicename[, protocolname]) -> integer\n\
Guido van Rossum82a5c661998-07-07 20:45:43 +00003332\n\
3333Return a port number from a service name and protocol name.\n\
Barry Warsaw11b91a02004-06-28 00:50:43 +00003334The optional protocol name, if given, should be 'tcp' or 'udp',\n\
3335otherwise any protocol will match.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003336
Guido van Rossum30a685f1991-06-27 15:51:29 +00003337
Barry Warsaw11b91a02004-06-28 00:50:43 +00003338/* Python interface to getservbyport(port).
3339 This only returns the service name, since the other info is already
3340 known or not useful (like the list of aliases). */
3341
3342/*ARGSUSED*/
3343static PyObject *
3344socket_getservbyport(PyObject *self, PyObject *args)
3345{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003346 int port;
3347 char *proto=NULL;
3348 struct servent *sp;
3349 if (!PyArg_ParseTuple(args, "i|s:getservbyport", &port, &proto))
3350 return NULL;
3351 if (port < 0 || port > 0xffff) {
3352 PyErr_SetString(
3353 PyExc_OverflowError,
3354 "getservbyport: port must be 0-65535.");
3355 return NULL;
3356 }
3357 Py_BEGIN_ALLOW_THREADS
3358 sp = getservbyport(htons((short)port), proto);
3359 Py_END_ALLOW_THREADS
3360 if (sp == NULL) {
3361 PyErr_SetString(socket_error, "port/proto not found");
3362 return NULL;
3363 }
3364 return PyUnicode_FromString(sp->s_name);
Barry Warsaw11b91a02004-06-28 00:50:43 +00003365}
3366
3367PyDoc_STRVAR(getservbyport_doc,
3368"getservbyport(port[, protocolname]) -> string\n\
3369\n\
3370Return the service name from a port number and protocol name.\n\
3371The optional protocol name, if given, should be 'tcp' or 'udp',\n\
3372otherwise any protocol will match.");
3373
Guido van Rossum3901d851996-12-19 16:35:04 +00003374/* Python interface to getprotobyname(name).
3375 This only returns the protocol number, since the other info is
3376 already known or not useful (like the list of aliases). */
3377
3378/*ARGSUSED*/
3379static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003380socket_getprotobyname(PyObject *self, PyObject *args)
Guido van Rossum3901d851996-12-19 16:35:04 +00003381{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003382 char *name;
3383 struct protoent *sp;
3384 if (!PyArg_ParseTuple(args, "s:getprotobyname", &name))
3385 return NULL;
3386 Py_BEGIN_ALLOW_THREADS
3387 sp = getprotobyname(name);
3388 Py_END_ALLOW_THREADS
3389 if (sp == NULL) {
3390 PyErr_SetString(socket_error, "protocol not found");
3391 return NULL;
3392 }
3393 return PyLong_FromLong((long) sp->p_proto);
Guido van Rossum3901d851996-12-19 16:35:04 +00003394}
3395
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003396PyDoc_STRVAR(getprotobyname_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003397"getprotobyname(name) -> integer\n\
3398\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003399Return the protocol number for the named protocol. (Rarely used.)");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003400
Guido van Rossum3901d851996-12-19 16:35:04 +00003401
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00003402#ifndef NO_DUP
3403/* dup() function for socket fds */
3404
3405static PyObject *
3406socket_dup(PyObject *self, PyObject *fdobj)
3407{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003408 SOCKET_T fd, newfd;
3409 PyObject *newfdobj;
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00003410
3411
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003412 fd = PyLong_AsSocket_t(fdobj);
3413 if (fd == (SOCKET_T)(-1) && PyErr_Occurred())
3414 return NULL;
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00003415
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003416 newfd = dup_socket(fd);
3417 if (newfd == INVALID_SOCKET)
3418 return set_error();
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00003419
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003420 newfdobj = PyLong_FromSocket_t(newfd);
3421 if (newfdobj == NULL)
3422 SOCKETCLOSE(newfd);
3423 return newfdobj;
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00003424}
3425
3426PyDoc_STRVAR(dup_doc,
3427"dup(integer) -> integer\n\
3428\n\
3429Duplicate an integer socket file descriptor. This is like os.dup(), but for\n\
3430sockets; on some platforms os.dup() won't work for socket file descriptors.");
3431#endif
3432
3433
Dave Cole331708b2004-08-09 04:51:41 +00003434#ifdef HAVE_SOCKETPAIR
3435/* Create a pair of sockets using the socketpair() function.
Dave Cole07fda7e2004-08-23 05:16:23 +00003436 Arguments as for socket() except the default family is AF_UNIX if
Dave Colee8bbfe42004-08-26 00:51:16 +00003437 defined on the platform; otherwise, the default is AF_INET. */
Dave Cole331708b2004-08-09 04:51:41 +00003438
3439/*ARGSUSED*/
3440static PyObject *
3441socket_socketpair(PyObject *self, PyObject *args)
3442{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003443 PySocketSockObject *s0 = NULL, *s1 = NULL;
3444 SOCKET_T sv[2];
3445 int family, type = SOCK_STREAM, proto = 0;
3446 PyObject *res = NULL;
Dave Cole331708b2004-08-09 04:51:41 +00003447
3448#if defined(AF_UNIX)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003449 family = AF_UNIX;
Dave Cole331708b2004-08-09 04:51:41 +00003450#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003451 family = AF_INET;
Dave Cole331708b2004-08-09 04:51:41 +00003452#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003453 if (!PyArg_ParseTuple(args, "|iii:socketpair",
3454 &family, &type, &proto))
3455 return NULL;
3456 /* Create a pair of socket fds */
3457 if (socketpair(family, type, proto, sv) < 0)
3458 return set_error();
3459 s0 = new_sockobject(sv[0], family, type, proto);
3460 if (s0 == NULL)
3461 goto finally;
3462 s1 = new_sockobject(sv[1], family, type, proto);
3463 if (s1 == NULL)
3464 goto finally;
3465 res = PyTuple_Pack(2, s0, s1);
Dave Cole331708b2004-08-09 04:51:41 +00003466
3467finally:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003468 if (res == NULL) {
3469 if (s0 == NULL)
3470 SOCKETCLOSE(sv[0]);
3471 if (s1 == NULL)
3472 SOCKETCLOSE(sv[1]);
3473 }
3474 Py_XDECREF(s0);
3475 Py_XDECREF(s1);
3476 return res;
Dave Cole331708b2004-08-09 04:51:41 +00003477}
3478
3479PyDoc_STRVAR(socketpair_doc,
3480"socketpair([family[, type[, proto]]]) -> (socket object, socket object)\n\
3481\n\
3482Create a pair of socket objects from the sockets returned by the platform\n\
3483socketpair() function.\n\
Dave Cole07fda7e2004-08-23 05:16:23 +00003484The arguments are the same as for socket() except the default family is\n\
Dave Colee8bbfe42004-08-26 00:51:16 +00003485AF_UNIX if defined on the platform; otherwise, the default is AF_INET.");
Dave Cole331708b2004-08-09 04:51:41 +00003486
3487#endif /* HAVE_SOCKETPAIR */
3488
3489
Guido van Rossum006bf911996-06-12 04:04:55 +00003490static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003491socket_ntohs(PyObject *self, PyObject *args)
Guido van Rossum006bf911996-06-12 04:04:55 +00003492{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003493 int x1, x2;
Guido van Rossum006bf911996-06-12 04:04:55 +00003494
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003495 if (!PyArg_ParseTuple(args, "i:ntohs", &x1)) {
3496 return NULL;
3497 }
3498 if (x1 < 0) {
3499 PyErr_SetString(PyExc_OverflowError,
3500 "can't convert negative number to unsigned long");
3501 return NULL;
3502 }
3503 x2 = (unsigned int)ntohs((unsigned short)x1);
3504 return PyLong_FromLong(x2);
Guido van Rossum006bf911996-06-12 04:04:55 +00003505}
3506
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003507PyDoc_STRVAR(ntohs_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003508"ntohs(integer) -> integer\n\
3509\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003510Convert a 16-bit integer from network to host byte order.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003511
3512
Guido van Rossum006bf911996-06-12 04:04:55 +00003513static PyObject *
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003514socket_ntohl(PyObject *self, PyObject *arg)
Guido van Rossum006bf911996-06-12 04:04:55 +00003515{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003516 unsigned long x;
Guido van Rossum006bf911996-06-12 04:04:55 +00003517
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003518 if (PyLong_Check(arg)) {
3519 x = PyLong_AsUnsignedLong(arg);
3520 if (x == (unsigned long) -1 && PyErr_Occurred())
3521 return NULL;
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003522#if SIZEOF_LONG > 4
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003523 {
3524 unsigned long y;
3525 /* only want the trailing 32 bits */
3526 y = x & 0xFFFFFFFFUL;
3527 if (y ^ x)
3528 return PyErr_Format(PyExc_OverflowError,
3529 "long int larger than 32 bits");
3530 x = y;
3531 }
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003532#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003533 }
3534 else
3535 return PyErr_Format(PyExc_TypeError,
3536 "expected int/long, %s found",
3537 Py_TYPE(arg)->tp_name);
3538 if (x == (unsigned long) -1 && PyErr_Occurred())
3539 return NULL;
3540 return PyLong_FromUnsignedLong(ntohl(x));
Guido van Rossum006bf911996-06-12 04:04:55 +00003541}
3542
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003543PyDoc_STRVAR(ntohl_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003544"ntohl(integer) -> integer\n\
3545\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003546Convert a 32-bit integer from network to host byte order.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003547
3548
Guido van Rossum006bf911996-06-12 04:04:55 +00003549static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003550socket_htons(PyObject *self, PyObject *args)
Guido van Rossum006bf911996-06-12 04:04:55 +00003551{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003552 int x1, x2;
Guido van Rossum006bf911996-06-12 04:04:55 +00003553
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003554 if (!PyArg_ParseTuple(args, "i:htons", &x1)) {
3555 return NULL;
3556 }
3557 if (x1 < 0) {
3558 PyErr_SetString(PyExc_OverflowError,
3559 "can't convert negative number to unsigned long");
3560 return NULL;
3561 }
3562 x2 = (unsigned int)htons((unsigned short)x1);
3563 return PyLong_FromLong(x2);
Guido van Rossum006bf911996-06-12 04:04:55 +00003564}
3565
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003566PyDoc_STRVAR(htons_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003567"htons(integer) -> integer\n\
3568\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003569Convert a 16-bit integer from host to network byte order.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003570
3571
Guido van Rossum006bf911996-06-12 04:04:55 +00003572static PyObject *
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003573socket_htonl(PyObject *self, PyObject *arg)
Guido van Rossum006bf911996-06-12 04:04:55 +00003574{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003575 unsigned long x;
Guido van Rossum006bf911996-06-12 04:04:55 +00003576
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003577 if (PyLong_Check(arg)) {
3578 x = PyLong_AsUnsignedLong(arg);
3579 if (x == (unsigned long) -1 && PyErr_Occurred())
3580 return NULL;
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003581#if SIZEOF_LONG > 4
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003582 {
3583 unsigned long y;
3584 /* only want the trailing 32 bits */
3585 y = x & 0xFFFFFFFFUL;
3586 if (y ^ x)
3587 return PyErr_Format(PyExc_OverflowError,
3588 "long int larger than 32 bits");
3589 x = y;
3590 }
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003591#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003592 }
3593 else
3594 return PyErr_Format(PyExc_TypeError,
3595 "expected int/long, %s found",
3596 Py_TYPE(arg)->tp_name);
3597 return PyLong_FromUnsignedLong(htonl((unsigned long)x));
Guido van Rossum006bf911996-06-12 04:04:55 +00003598}
3599
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003600PyDoc_STRVAR(htonl_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003601"htonl(integer) -> integer\n\
3602\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003603Convert a 32-bit integer from host to network byte order.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003604
Guido van Rossum3eede5a2002-06-07 02:08:35 +00003605/* socket.inet_aton() and socket.inet_ntoa() functions. */
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003606
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003607PyDoc_STRVAR(inet_aton_doc,
Guido van Rossum7d0a8262007-05-21 23:13:11 +00003608"inet_aton(string) -> bytes giving packed 32-bit IP representation\n\
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003609\n\
Guido van Rossumc6a164b1999-08-20 19:11:27 +00003610Convert 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 +00003611binary format used in low-level network functions.");
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003612
3613static PyObject*
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003614socket_inet_aton(PyObject *self, PyObject *args)
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003615{
Guido van Rossuma2e48551999-09-09 15:42:59 +00003616#ifndef INADDR_NONE
3617#define INADDR_NONE (-1)
3618#endif
Neal Norwitz88f115b2003-02-13 02:15:42 +00003619#ifdef HAVE_INET_ATON
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003620 struct in_addr buf;
Thomas Wouters477c8d52006-05-27 19:21:47 +00003621#endif
3622
3623#if !defined(HAVE_INET_ATON) || defined(USE_INET_ATON_WEAKLINK)
Benjamin Petersonf91df042009-02-13 02:50:59 +00003624#if (SIZEOF_INT != 4)
3625#error "Not sure if in_addr_t exists and int is not 32-bits."
3626#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003627 /* Have to use inet_addr() instead */
3628 unsigned int packed_addr;
Tim Peters1df9fdd2003-02-13 03:13:40 +00003629#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003630 char *ip_addr;
Guido van Rossumc6a164b1999-08-20 19:11:27 +00003631
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003632 if (!PyArg_ParseTuple(args, "s:inet_aton", &ip_addr))
3633 return NULL;
Guido van Rossumad05cdf2003-02-12 23:08:22 +00003634
Tim Peters1df9fdd2003-02-13 03:13:40 +00003635
3636#ifdef HAVE_INET_ATON
Thomas Wouters477c8d52006-05-27 19:21:47 +00003637
3638#ifdef USE_INET_ATON_WEAKLINK
3639 if (inet_aton != NULL) {
3640#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003641 if (inet_aton(ip_addr, &buf))
3642 return PyBytes_FromStringAndSize((char *)(&buf),
3643 sizeof(buf));
Guido van Rossumad05cdf2003-02-12 23:08:22 +00003644
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003645 PyErr_SetString(socket_error,
3646 "illegal IP address string passed to inet_aton");
3647 return NULL;
Guido van Rossumad05cdf2003-02-12 23:08:22 +00003648
Thomas Wouters477c8d52006-05-27 19:21:47 +00003649#ifdef USE_INET_ATON_WEAKLINK
3650 } else {
3651#endif
3652
3653#endif
3654
3655#if !defined(HAVE_INET_ATON) || defined(USE_INET_ATON_WEAKLINK)
3656
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003657 /* special-case this address as inet_addr might return INADDR_NONE
3658 * for this */
3659 if (strcmp(ip_addr, "255.255.255.255") == 0) {
3660 packed_addr = 0xFFFFFFFF;
3661 } else {
Thomas Wouters477c8d52006-05-27 19:21:47 +00003662
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003663 packed_addr = inet_addr(ip_addr);
Guido van Rossumc6a164b1999-08-20 19:11:27 +00003664
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003665 if (packed_addr == INADDR_NONE) { /* invalid address */
3666 PyErr_SetString(socket_error,
3667 "illegal IP address string passed to inet_aton");
3668 return NULL;
3669 }
3670 }
3671 return PyBytes_FromStringAndSize((char *) &packed_addr,
3672 sizeof(packed_addr));
Thomas Wouters477c8d52006-05-27 19:21:47 +00003673
3674#ifdef USE_INET_ATON_WEAKLINK
3675 }
3676#endif
3677
Guido van Rossumad05cdf2003-02-12 23:08:22 +00003678#endif
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003679}
3680
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003681PyDoc_STRVAR(inet_ntoa_doc,
Fred Drakee0661342000-03-07 14:05:16 +00003682"inet_ntoa(packed_ip) -> ip_address_string\n\
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003683\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003684Convert an IP address from 32-bit packed binary format to string format");
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003685
3686static PyObject*
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003687socket_inet_ntoa(PyObject *self, PyObject *args)
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003688{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003689 char *packed_str;
3690 int addr_len;
3691 struct in_addr packed_addr;
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003692
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003693 if (!PyArg_ParseTuple(args, "y#:inet_ntoa", &packed_str, &addr_len)) {
3694 return NULL;
3695 }
Guido van Rossum48a680c2001-03-02 06:34:14 +00003696
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003697 if (addr_len != sizeof(packed_addr)) {
3698 PyErr_SetString(socket_error,
3699 "packed IP wrong length for inet_ntoa");
3700 return NULL;
3701 }
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003702
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003703 memcpy(&packed_addr, packed_str, addr_len);
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003704
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003705 return PyUnicode_FromString(inet_ntoa(packed_addr));
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003706}
Guido van Rossum82a5c661998-07-07 20:45:43 +00003707
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003708#ifdef HAVE_INET_PTON
3709
3710PyDoc_STRVAR(inet_pton_doc,
3711"inet_pton(af, ip) -> packed IP address string\n\
3712\n\
3713Convert an IP address from string format to a packed string suitable\n\
3714for use with low-level network functions.");
3715
3716static PyObject *
3717socket_inet_pton(PyObject *self, PyObject *args)
3718{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003719 int af;
3720 char* ip;
3721 int retval;
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003722#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003723 char packed[MAX(sizeof(struct in_addr), sizeof(struct in6_addr))];
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003724#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003725 char packed[sizeof(struct in_addr)];
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003726#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003727 if (!PyArg_ParseTuple(args, "is:inet_pton", &af, &ip)) {
3728 return NULL;
3729 }
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003730
Martin v. Löwis04697e82004-06-02 12:35:29 +00003731#if !defined(ENABLE_IPV6) && defined(AF_INET6)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003732 if(af == AF_INET6) {
3733 PyErr_SetString(socket_error,
3734 "can't use AF_INET6, IPv6 is disabled");
3735 return NULL;
3736 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00003737#endif
Martin v. Löwis10649092003-08-05 06:25:06 +00003738
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003739 retval = inet_pton(af, ip, packed);
3740 if (retval < 0) {
3741 PyErr_SetFromErrno(socket_error);
3742 return NULL;
3743 } else if (retval == 0) {
3744 PyErr_SetString(socket_error,
3745 "illegal IP address string passed to inet_pton");
3746 return NULL;
3747 } else if (af == AF_INET) {
3748 return PyBytes_FromStringAndSize(packed,
3749 sizeof(struct in_addr));
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003750#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003751 } else if (af == AF_INET6) {
3752 return PyBytes_FromStringAndSize(packed,
3753 sizeof(struct in6_addr));
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003754#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003755 } else {
3756 PyErr_SetString(socket_error, "unknown address family");
3757 return NULL;
3758 }
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003759}
Thomas Wouters477c8d52006-05-27 19:21:47 +00003760
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003761PyDoc_STRVAR(inet_ntop_doc,
3762"inet_ntop(af, packed_ip) -> string formatted IP address\n\
3763\n\
3764Convert a packed IP address of the given family to string format.");
3765
3766static PyObject *
3767socket_inet_ntop(PyObject *self, PyObject *args)
3768{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003769 int af;
3770 char* packed;
3771 int len;
3772 const char* retval;
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003773#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003774 char ip[MAX(INET_ADDRSTRLEN, INET6_ADDRSTRLEN) + 1];
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003775#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003776 char ip[INET_ADDRSTRLEN + 1];
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003777#endif
Thomas Wouters477c8d52006-05-27 19:21:47 +00003778
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003779 /* Guarantee NUL-termination for PyUnicode_FromString() below */
3780 memset((void *) &ip[0], '\0', sizeof(ip));
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003781
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003782 if (!PyArg_ParseTuple(args, "iy#:inet_ntop", &af, &packed, &len)) {
3783 return NULL;
3784 }
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003785
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003786 if (af == AF_INET) {
3787 if (len != sizeof(struct in_addr)) {
3788 PyErr_SetString(PyExc_ValueError,
3789 "invalid length of packed IP address string");
3790 return NULL;
3791 }
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003792#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003793 } else if (af == AF_INET6) {
3794 if (len != sizeof(struct in6_addr)) {
3795 PyErr_SetString(PyExc_ValueError,
3796 "invalid length of packed IP address string");
3797 return NULL;
3798 }
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003799#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003800 } else {
3801 PyErr_Format(PyExc_ValueError,
3802 "unknown address family %d", af);
3803 return NULL;
3804 }
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003805
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003806 retval = inet_ntop(af, packed, ip, sizeof(ip));
3807 if (!retval) {
3808 PyErr_SetFromErrno(socket_error);
3809 return NULL;
3810 } else {
3811 return PyUnicode_FromString(retval);
3812 }
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003813
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003814 /* NOTREACHED */
3815 PyErr_SetString(PyExc_RuntimeError, "invalid handling of inet_ntop");
3816 return NULL;
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003817}
3818
3819#endif /* HAVE_INET_PTON */
3820
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003821/* Python interface to getaddrinfo(host, port). */
3822
3823/*ARGSUSED*/
3824static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003825socket_getaddrinfo(PyObject *self, PyObject *args)
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003826{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003827 struct addrinfo hints, *res;
3828 struct addrinfo *res0 = NULL;
3829 PyObject *hobj = NULL;
3830 PyObject *pobj = (PyObject *)NULL;
3831 char pbuf[30];
3832 char *hptr, *pptr;
3833 int family, socktype, protocol, flags;
3834 int error;
3835 PyObject *all = (PyObject *)NULL;
3836 PyObject *idna = NULL;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003837
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003838 family = socktype = protocol = flags = 0;
3839 family = AF_UNSPEC;
3840 if (!PyArg_ParseTuple(args, "OO|iiii:getaddrinfo",
3841 &hobj, &pobj, &family, &socktype,
3842 &protocol, &flags)) {
3843 return NULL;
3844 }
3845 if (hobj == Py_None) {
3846 hptr = NULL;
3847 } else if (PyUnicode_Check(hobj)) {
3848 idna = PyObject_CallMethod(hobj, "encode", "s", "idna");
3849 if (!idna)
3850 return NULL;
3851 assert(PyBytes_Check(idna));
3852 hptr = PyBytes_AS_STRING(idna);
3853 } else if (PyBytes_Check(hobj)) {
3854 hptr = PyBytes_AsString(hobj);
3855 } else {
3856 PyErr_SetString(PyExc_TypeError,
3857 "getaddrinfo() argument 1 must be string or None");
3858 return NULL;
3859 }
3860 if (PyLong_CheckExact(pobj)) {
3861 long value = PyLong_AsLong(pobj);
3862 if (value == -1 && PyErr_Occurred())
3863 goto err;
3864 PyOS_snprintf(pbuf, sizeof(pbuf), "%ld", value);
3865 pptr = pbuf;
3866 } else if (PyUnicode_Check(pobj)) {
3867 pptr = _PyUnicode_AsString(pobj);
3868 } else if (PyBytes_Check(pobj)) {
3869 pptr = PyBytes_AsString(pobj);
3870 } else if (pobj == Py_None) {
3871 pptr = (char *)NULL;
3872 } else {
3873 PyErr_SetString(socket_error, "Int or String expected");
3874 goto err;
3875 }
3876 memset(&hints, 0, sizeof(hints));
3877 hints.ai_family = family;
3878 hints.ai_socktype = socktype;
3879 hints.ai_protocol = protocol;
3880 hints.ai_flags = flags;
3881 Py_BEGIN_ALLOW_THREADS
3882 ACQUIRE_GETADDRINFO_LOCK
3883 error = getaddrinfo(hptr, pptr, &hints, &res0);
3884 Py_END_ALLOW_THREADS
3885 RELEASE_GETADDRINFO_LOCK /* see comment in setipaddr() */
3886 if (error) {
3887 set_gaierror(error);
3888 goto err;
3889 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003890
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003891 if ((all = PyList_New(0)) == NULL)
3892 goto err;
3893 for (res = res0; res; res = res->ai_next) {
3894 PyObject *single;
3895 PyObject *addr =
3896 makesockaddr(-1, res->ai_addr, res->ai_addrlen, protocol);
3897 if (addr == NULL)
3898 goto err;
3899 single = Py_BuildValue("iiisO", res->ai_family,
3900 res->ai_socktype, res->ai_protocol,
3901 res->ai_canonname ? res->ai_canonname : "",
3902 addr);
3903 Py_DECREF(addr);
3904 if (single == NULL)
3905 goto err;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003906
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003907 if (PyList_Append(all, single))
3908 goto err;
3909 Py_XDECREF(single);
3910 }
3911 Py_XDECREF(idna);
3912 if (res0)
3913 freeaddrinfo(res0);
3914 return all;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003915 err:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003916 Py_XDECREF(all);
3917 Py_XDECREF(idna);
3918 if (res0)
3919 freeaddrinfo(res0);
3920 return (PyObject *)NULL;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003921}
3922
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003923PyDoc_STRVAR(getaddrinfo_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00003924"getaddrinfo(host, port [, family, socktype, proto, flags])\n\
3925 -> list of (family, socktype, proto, canonname, sockaddr)\n\
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003926\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003927Resolve host and port into addrinfo struct.");
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003928
3929/* Python interface to getnameinfo(sa, flags). */
3930
3931/*ARGSUSED*/
3932static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003933socket_getnameinfo(PyObject *self, PyObject *args)
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003934{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003935 PyObject *sa = (PyObject *)NULL;
3936 int flags;
3937 char *hostp;
3938 int port, flowinfo, scope_id;
3939 char hbuf[NI_MAXHOST], pbuf[NI_MAXSERV];
3940 struct addrinfo hints, *res = NULL;
3941 int error;
3942 PyObject *ret = (PyObject *)NULL;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003943
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003944 flags = flowinfo = scope_id = 0;
3945 if (!PyArg_ParseTuple(args, "Oi:getnameinfo", &sa, &flags))
3946 return NULL;
3947 if (!PyTuple_Check(sa)) {
3948 PyErr_SetString(PyExc_TypeError,
3949 "getnameinfo() argument 1 must be a tuple");
3950 return NULL;
3951 }
3952 if (!PyArg_ParseTuple(sa, "si|ii",
3953 &hostp, &port, &flowinfo, &scope_id))
3954 return NULL;
3955 PyOS_snprintf(pbuf, sizeof(pbuf), "%d", port);
3956 memset(&hints, 0, sizeof(hints));
3957 hints.ai_family = AF_UNSPEC;
3958 hints.ai_socktype = SOCK_DGRAM; /* make numeric port happy */
3959 Py_BEGIN_ALLOW_THREADS
3960 ACQUIRE_GETADDRINFO_LOCK
3961 error = getaddrinfo(hostp, pbuf, &hints, &res);
3962 Py_END_ALLOW_THREADS
3963 RELEASE_GETADDRINFO_LOCK /* see comment in setipaddr() */
3964 if (error) {
3965 set_gaierror(error);
3966 goto fail;
3967 }
3968 if (res->ai_next) {
3969 PyErr_SetString(socket_error,
3970 "sockaddr resolved to multiple addresses");
3971 goto fail;
3972 }
3973 switch (res->ai_family) {
3974 case AF_INET:
3975 {
3976 if (PyTuple_GET_SIZE(sa) != 2) {
3977 PyErr_SetString(socket_error,
3978 "IPv4 sockaddr must be 2 tuple");
3979 goto fail;
3980 }
3981 break;
3982 }
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00003983#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003984 case AF_INET6:
3985 {
3986 struct sockaddr_in6 *sin6;
3987 sin6 = (struct sockaddr_in6 *)res->ai_addr;
3988 sin6->sin6_flowinfo = flowinfo;
3989 sin6->sin6_scope_id = scope_id;
3990 break;
3991 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003992#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003993 }
3994 error = getnameinfo(res->ai_addr, res->ai_addrlen,
3995 hbuf, sizeof(hbuf), pbuf, sizeof(pbuf), flags);
3996 if (error) {
3997 set_gaierror(error);
3998 goto fail;
3999 }
4000 ret = Py_BuildValue("ss", hbuf, pbuf);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004001
4002fail:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004003 if (res)
4004 freeaddrinfo(res);
4005 return ret;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004006}
4007
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004008PyDoc_STRVAR(getnameinfo_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00004009"getnameinfo(sockaddr, flags) --> (host, port)\n\
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004010\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004011Get host and port for a sockaddr.");
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004012
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004013
4014/* Python API to getting and setting the default timeout value. */
4015
4016static PyObject *
4017socket_getdefaulttimeout(PyObject *self)
4018{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004019 if (defaulttimeout < 0.0) {
4020 Py_INCREF(Py_None);
4021 return Py_None;
4022 }
4023 else
4024 return PyFloat_FromDouble(defaulttimeout);
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004025}
4026
4027PyDoc_STRVAR(getdefaulttimeout_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00004028"getdefaulttimeout() -> timeout\n\
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004029\n\
4030Returns the default timeout in floating seconds for new socket objects.\n\
4031A value of None indicates that new socket objects have no timeout.\n\
4032When the socket module is first imported, the default is None.");
4033
4034static PyObject *
4035socket_setdefaulttimeout(PyObject *self, PyObject *arg)
4036{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004037 double timeout;
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004038
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004039 if (arg == Py_None)
4040 timeout = -1.0;
4041 else {
4042 timeout = PyFloat_AsDouble(arg);
4043 if (timeout < 0.0) {
4044 if (!PyErr_Occurred())
4045 PyErr_SetString(PyExc_ValueError,
4046 "Timeout value out of range");
4047 return NULL;
4048 }
4049 }
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004050
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004051 defaulttimeout = timeout;
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004052
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004053 Py_INCREF(Py_None);
4054 return Py_None;
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004055}
4056
4057PyDoc_STRVAR(setdefaulttimeout_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00004058"setdefaulttimeout(timeout)\n\
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004059\n\
4060Set the default timeout in floating seconds for new socket objects.\n\
4061A value of None indicates that new socket objects have no timeout.\n\
4062When the socket module is first imported, the default is None.");
4063
4064
Guido van Rossum30a685f1991-06-27 15:51:29 +00004065/* List of functions exported by this module. */
4066
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004067static PyMethodDef socket_methods[] = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004068 {"gethostbyname", socket_gethostbyname,
4069 METH_VARARGS, gethostbyname_doc},
4070 {"gethostbyname_ex", socket_gethostbyname_ex,
4071 METH_VARARGS, ghbn_ex_doc},
4072 {"gethostbyaddr", socket_gethostbyaddr,
4073 METH_VARARGS, gethostbyaddr_doc},
4074 {"gethostname", socket_gethostname,
4075 METH_NOARGS, gethostname_doc},
4076 {"getservbyname", socket_getservbyname,
4077 METH_VARARGS, getservbyname_doc},
4078 {"getservbyport", socket_getservbyport,
4079 METH_VARARGS, getservbyport_doc},
4080 {"getprotobyname", socket_getprotobyname,
4081 METH_VARARGS, getprotobyname_doc},
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00004082#ifndef NO_DUP
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004083 {"dup", socket_dup,
4084 METH_O, dup_doc},
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00004085#endif
Dave Cole331708b2004-08-09 04:51:41 +00004086#ifdef HAVE_SOCKETPAIR
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004087 {"socketpair", socket_socketpair,
4088 METH_VARARGS, socketpair_doc},
Dave Cole331708b2004-08-09 04:51:41 +00004089#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004090 {"ntohs", socket_ntohs,
4091 METH_VARARGS, ntohs_doc},
4092 {"ntohl", socket_ntohl,
4093 METH_O, ntohl_doc},
4094 {"htons", socket_htons,
4095 METH_VARARGS, htons_doc},
4096 {"htonl", socket_htonl,
4097 METH_O, htonl_doc},
4098 {"inet_aton", socket_inet_aton,
4099 METH_VARARGS, inet_aton_doc},
4100 {"inet_ntoa", socket_inet_ntoa,
4101 METH_VARARGS, inet_ntoa_doc},
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00004102#ifdef HAVE_INET_PTON
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004103 {"inet_pton", socket_inet_pton,
4104 METH_VARARGS, inet_pton_doc},
4105 {"inet_ntop", socket_inet_ntop,
4106 METH_VARARGS, inet_ntop_doc},
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00004107#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004108 {"getaddrinfo", socket_getaddrinfo,
4109 METH_VARARGS, getaddrinfo_doc},
4110 {"getnameinfo", socket_getnameinfo,
4111 METH_VARARGS, getnameinfo_doc},
4112 {"getdefaulttimeout", (PyCFunction)socket_getdefaulttimeout,
4113 METH_NOARGS, getdefaulttimeout_doc},
4114 {"setdefaulttimeout", socket_setdefaulttimeout,
4115 METH_O, setdefaulttimeout_doc},
4116 {NULL, NULL} /* Sentinel */
Guido van Rossum6574b3e1991-06-25 21:36:08 +00004117};
4118
Guido van Rossum30a685f1991-06-27 15:51:29 +00004119
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004120#ifdef MS_WINDOWS
4121#define OS_INIT_DEFINED
4122
4123/* Additional initialization and cleanup for Windows */
Guido van Rossumbe32c891996-06-20 16:25:29 +00004124
4125static void
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004126os_cleanup(void)
Guido van Rossumbe32c891996-06-20 16:25:29 +00004127{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004128 WSACleanup();
Guido van Rossumbe32c891996-06-20 16:25:29 +00004129}
4130
4131static int
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004132os_init(void)
Guido van Rossumbe32c891996-06-20 16:25:29 +00004133{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004134 WSADATA WSAData;
4135 int ret;
4136 ret = WSAStartup(0x0101, &WSAData);
4137 switch (ret) {
4138 case 0: /* No error */
4139 Py_AtExit(os_cleanup);
4140 return 1; /* Success */
4141 case WSASYSNOTREADY:
4142 PyErr_SetString(PyExc_ImportError,
4143 "WSAStartup failed: network not ready");
4144 break;
4145 case WSAVERNOTSUPPORTED:
4146 case WSAEINVAL:
4147 PyErr_SetString(
4148 PyExc_ImportError,
4149 "WSAStartup failed: requested version not supported");
4150 break;
4151 default:
4152 PyErr_Format(PyExc_ImportError, "WSAStartup failed: error code %d", ret);
4153 break;
4154 }
4155 return 0; /* Failure */
Guido van Rossumbe32c891996-06-20 16:25:29 +00004156}
4157
Guido van Rossum8d665e61996-06-26 18:22:49 +00004158#endif /* MS_WINDOWS */
Guido van Rossumbe32c891996-06-20 16:25:29 +00004159
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004160
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004161#ifdef PYOS_OS2
4162#define OS_INIT_DEFINED
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004163
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004164/* Additional initialization for OS/2 */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004165
4166static int
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004167os_init(void)
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004168{
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004169#ifndef PYCC_GCC
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004170 int rc = sock_init();
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004171
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004172 if (rc == 0) {
4173 return 1; /* Success */
4174 }
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004175
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004176 PyErr_Format(PyExc_ImportError, "OS/2 TCP/IP Error# %d", sock_errno());
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004177
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004178 return 0; /* Failure */
Andrew MacIntyreba43e872002-03-03 03:03:52 +00004179#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004180 /* No need to initialise sockets with GCC/EMX */
4181 return 1; /* Success */
Andrew MacIntyreba43e872002-03-03 03:03:52 +00004182#endif
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004183}
4184
4185#endif /* PYOS_OS2 */
4186
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004187
4188#ifndef OS_INIT_DEFINED
4189static int
4190os_init(void)
4191{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004192 return 1; /* Success */
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004193}
4194#endif
4195
4196
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00004197/* C API table - always add new things to the end for binary
4198 compatibility. */
4199static
4200PySocketModule_APIObject PySocketModuleAPI =
4201{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004202 &sock_type,
4203 NULL
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00004204};
4205
Guido van Rossum3eede5a2002-06-07 02:08:35 +00004206
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004207/* Initialize the _socket module.
Guido van Rossum3eede5a2002-06-07 02:08:35 +00004208
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004209 This module is actually called "_socket", and there's a wrapper
Guido van Rossum7d0a8262007-05-21 23:13:11 +00004210 "socket.py" which implements some additional functionality.
4211 The import of "_socket" may fail with an ImportError exception if
4212 os-specific initialization fails. On Windows, this does WINSOCK
4213 initialization. When WINSOCK is initialized succesfully, a call to
4214 WSACleanup() is scheduled to be made at exit time.
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004215*/
Guido van Rossum3eede5a2002-06-07 02:08:35 +00004216
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004217PyDoc_STRVAR(socket_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00004218"Implementation module for socket operations.\n\
4219\n\
4220See the socket module for documentation.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00004221
Martin v. Löwis1a214512008-06-11 05:26:20 +00004222static struct PyModuleDef socketmodule = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004223 PyModuleDef_HEAD_INIT,
4224 PySocket_MODULE_NAME,
4225 socket_doc,
4226 -1,
4227 socket_methods,
4228 NULL,
4229 NULL,
4230 NULL,
4231 NULL
Martin v. Löwis1a214512008-06-11 05:26:20 +00004232};
4233
Mark Hammondfe51c6d2002-08-02 02:27:13 +00004234PyMODINIT_FUNC
Martin v. Löwis1a214512008-06-11 05:26:20 +00004235PyInit__socket(void)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00004236{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004237 PyObject *m, *has_ipv6;
Fred Drake4baedc12002-04-01 14:53:37 +00004238
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004239 if (!os_init())
4240 return NULL;
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004241
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004242 Py_TYPE(&sock_type) = &PyType_Type;
4243 m = PyModule_Create(&socketmodule);
4244 if (m == NULL)
4245 return NULL;
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004246
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004247 socket_error = PyErr_NewException("socket.error",
4248 PyExc_IOError, NULL);
4249 if (socket_error == NULL)
4250 return NULL;
4251 PySocketModuleAPI.error = socket_error;
4252 Py_INCREF(socket_error);
4253 PyModule_AddObject(m, "error", socket_error);
4254 socket_herror = PyErr_NewException("socket.herror",
4255 socket_error, NULL);
4256 if (socket_herror == NULL)
4257 return NULL;
4258 Py_INCREF(socket_herror);
4259 PyModule_AddObject(m, "herror", socket_herror);
4260 socket_gaierror = PyErr_NewException("socket.gaierror", socket_error,
4261 NULL);
4262 if (socket_gaierror == NULL)
4263 return NULL;
4264 Py_INCREF(socket_gaierror);
4265 PyModule_AddObject(m, "gaierror", socket_gaierror);
4266 socket_timeout = PyErr_NewException("socket.timeout",
4267 socket_error, NULL);
4268 if (socket_timeout == NULL)
4269 return NULL;
4270 Py_INCREF(socket_timeout);
4271 PyModule_AddObject(m, "timeout", socket_timeout);
4272 Py_INCREF((PyObject *)&sock_type);
4273 if (PyModule_AddObject(m, "SocketType",
4274 (PyObject *)&sock_type) != 0)
4275 return NULL;
4276 Py_INCREF((PyObject *)&sock_type);
4277 if (PyModule_AddObject(m, "socket",
4278 (PyObject *)&sock_type) != 0)
4279 return NULL;
Guido van Rossum09be4091999-08-09 14:40:40 +00004280
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00004281#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004282 has_ipv6 = Py_True;
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00004283#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004284 has_ipv6 = Py_False;
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00004285#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004286 Py_INCREF(has_ipv6);
4287 PyModule_AddObject(m, "has_ipv6", has_ipv6);
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00004288
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004289 /* Export C API */
4290 if (PyModule_AddObject(m, PySocket_CAPI_NAME,
4291 PyCapsule_New(&PySocketModuleAPI, PySocket_CAPSULE_NAME, NULL)
4292 ) != 0)
4293 return NULL;
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00004294
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004295 /* Address families (we only support AF_INET and AF_UNIX) */
Guido van Rossum09be4091999-08-09 14:40:40 +00004296#ifdef AF_UNSPEC
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004297 PyModule_AddIntConstant(m, "AF_UNSPEC", AF_UNSPEC);
Guido van Rossum09be4091999-08-09 14:40:40 +00004298#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004299 PyModule_AddIntConstant(m, "AF_INET", AF_INET);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004300#ifdef AF_INET6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004301 PyModule_AddIntConstant(m, "AF_INET6", AF_INET6);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004302#endif /* AF_INET6 */
Andrew MacIntyred12dfbb2004-04-04 07:13:49 +00004303#if defined(AF_UNIX)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004304 PyModule_AddIntConstant(m, "AF_UNIX", AF_UNIX);
Guido van Rossumb6775db1994-08-01 11:34:53 +00004305#endif /* AF_UNIX */
Guido van Rossum09be4091999-08-09 14:40:40 +00004306#ifdef AF_AX25
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004307 /* Amateur Radio AX.25 */
4308 PyModule_AddIntConstant(m, "AF_AX25", AF_AX25);
Guido van Rossum09be4091999-08-09 14:40:40 +00004309#endif
4310#ifdef AF_IPX
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004311 PyModule_AddIntConstant(m, "AF_IPX", AF_IPX); /* Novell IPX */
Guido van Rossum09be4091999-08-09 14:40:40 +00004312#endif
4313#ifdef AF_APPLETALK
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004314 /* Appletalk DDP */
4315 PyModule_AddIntConstant(m, "AF_APPLETALK", AF_APPLETALK);
Guido van Rossum09be4091999-08-09 14:40:40 +00004316#endif
4317#ifdef AF_NETROM
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004318 /* Amateur radio NetROM */
4319 PyModule_AddIntConstant(m, "AF_NETROM", AF_NETROM);
Guido van Rossum09be4091999-08-09 14:40:40 +00004320#endif
4321#ifdef AF_BRIDGE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004322 /* Multiprotocol bridge */
4323 PyModule_AddIntConstant(m, "AF_BRIDGE", AF_BRIDGE);
Guido van Rossum09be4091999-08-09 14:40:40 +00004324#endif
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004325#ifdef AF_ATMPVC
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004326 /* ATM PVCs */
4327 PyModule_AddIntConstant(m, "AF_ATMPVC", AF_ATMPVC);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004328#endif
Guido van Rossum09be4091999-08-09 14:40:40 +00004329#ifdef AF_AAL5
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004330 /* Reserved for Werner's ATM */
4331 PyModule_AddIntConstant(m, "AF_AAL5", AF_AAL5);
Guido van Rossum09be4091999-08-09 14:40:40 +00004332#endif
4333#ifdef AF_X25
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004334 /* Reserved for X.25 project */
4335 PyModule_AddIntConstant(m, "AF_X25", AF_X25);
Guido van Rossum09be4091999-08-09 14:40:40 +00004336#endif
4337#ifdef AF_INET6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004338 PyModule_AddIntConstant(m, "AF_INET6", AF_INET6); /* IP version 6 */
Guido van Rossum09be4091999-08-09 14:40:40 +00004339#endif
4340#ifdef AF_ROSE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004341 /* Amateur Radio X.25 PLP */
4342 PyModule_AddIntConstant(m, "AF_ROSE", AF_ROSE);
Guido van Rossum09be4091999-08-09 14:40:40 +00004343#endif
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004344#ifdef AF_DECnet
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004345 /* Reserved for DECnet project */
4346 PyModule_AddIntConstant(m, "AF_DECnet", AF_DECnet);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004347#endif
4348#ifdef AF_NETBEUI
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004349 /* Reserved for 802.2LLC project */
4350 PyModule_AddIntConstant(m, "AF_NETBEUI", AF_NETBEUI);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004351#endif
4352#ifdef AF_SECURITY
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004353 /* Security callback pseudo AF */
4354 PyModule_AddIntConstant(m, "AF_SECURITY", AF_SECURITY);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004355#endif
4356#ifdef AF_KEY
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004357 /* PF_KEY key management API */
4358 PyModule_AddIntConstant(m, "AF_KEY", AF_KEY);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004359#endif
4360#ifdef AF_NETLINK
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004361 /* */
4362 PyModule_AddIntConstant(m, "AF_NETLINK", AF_NETLINK);
4363 PyModule_AddIntConstant(m, "NETLINK_ROUTE", NETLINK_ROUTE);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004364#ifdef NETLINK_SKIP
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004365 PyModule_AddIntConstant(m, "NETLINK_SKIP", NETLINK_SKIP);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004366#endif
4367#ifdef NETLINK_W1
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004368 PyModule_AddIntConstant(m, "NETLINK_W1", NETLINK_W1);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004369#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004370 PyModule_AddIntConstant(m, "NETLINK_USERSOCK", NETLINK_USERSOCK);
4371 PyModule_AddIntConstant(m, "NETLINK_FIREWALL", NETLINK_FIREWALL);
Guido van Rossum668a94a2006-02-21 01:07:27 +00004372#ifdef NETLINK_TCPDIAG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004373 PyModule_AddIntConstant(m, "NETLINK_TCPDIAG", NETLINK_TCPDIAG);
Guido van Rossum668a94a2006-02-21 01:07:27 +00004374#endif
4375#ifdef NETLINK_NFLOG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004376 PyModule_AddIntConstant(m, "NETLINK_NFLOG", NETLINK_NFLOG);
Guido van Rossum668a94a2006-02-21 01:07:27 +00004377#endif
Neal Norwitz65851662006-01-16 04:31:40 +00004378#ifdef NETLINK_XFRM
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004379 PyModule_AddIntConstant(m, "NETLINK_XFRM", NETLINK_XFRM);
Neal Norwitz65851662006-01-16 04:31:40 +00004380#endif
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004381#ifdef NETLINK_ARPD
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004382 PyModule_AddIntConstant(m, "NETLINK_ARPD", NETLINK_ARPD);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004383#endif
4384#ifdef NETLINK_ROUTE6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004385 PyModule_AddIntConstant(m, "NETLINK_ROUTE6", NETLINK_ROUTE6);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004386#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004387 PyModule_AddIntConstant(m, "NETLINK_IP6_FW", NETLINK_IP6_FW);
Thomas Wouterscf297e42007-02-23 15:07:44 +00004388#ifdef NETLINK_DNRTMSG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004389 PyModule_AddIntConstant(m, "NETLINK_DNRTMSG", NETLINK_DNRTMSG);
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00004390#endif
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004391#ifdef NETLINK_TAPBASE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004392 PyModule_AddIntConstant(m, "NETLINK_TAPBASE", NETLINK_TAPBASE);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004393#endif
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004394#endif /* AF_NETLINK */
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004395#ifdef AF_ROUTE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004396 /* Alias to emulate 4.4BSD */
4397 PyModule_AddIntConstant(m, "AF_ROUTE", AF_ROUTE);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004398#endif
4399#ifdef AF_ASH
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004400 /* Ash */
4401 PyModule_AddIntConstant(m, "AF_ASH", AF_ASH);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004402#endif
4403#ifdef AF_ECONET
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004404 /* Acorn Econet */
4405 PyModule_AddIntConstant(m, "AF_ECONET", AF_ECONET);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004406#endif
4407#ifdef AF_ATMSVC
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004408 /* ATM SVCs */
4409 PyModule_AddIntConstant(m, "AF_ATMSVC", AF_ATMSVC);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004410#endif
4411#ifdef AF_SNA
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004412 /* Linux SNA Project (nutters!) */
4413 PyModule_AddIntConstant(m, "AF_SNA", AF_SNA);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004414#endif
4415#ifdef AF_IRDA
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004416 /* IRDA sockets */
4417 PyModule_AddIntConstant(m, "AF_IRDA", AF_IRDA);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004418#endif
4419#ifdef AF_PPPOX
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004420 /* PPPoX sockets */
4421 PyModule_AddIntConstant(m, "AF_PPPOX", AF_PPPOX);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004422#endif
4423#ifdef AF_WANPIPE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004424 /* Wanpipe API Sockets */
4425 PyModule_AddIntConstant(m, "AF_WANPIPE", AF_WANPIPE);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004426#endif
4427#ifdef AF_LLC
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004428 /* Linux LLC */
4429 PyModule_AddIntConstant(m, "AF_LLC", AF_LLC);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004430#endif
Martin v. Löwis12af0482004-01-31 12:34:17 +00004431
Hye-Shik Chang81268602004-02-02 06:05:24 +00004432#ifdef USE_BLUETOOTH
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004433 PyModule_AddIntConstant(m, "AF_BLUETOOTH", AF_BLUETOOTH);
4434 PyModule_AddIntConstant(m, "BTPROTO_L2CAP", BTPROTO_L2CAP);
4435 PyModule_AddIntConstant(m, "BTPROTO_HCI", BTPROTO_HCI);
4436 PyModule_AddIntConstant(m, "SOL_HCI", SOL_HCI);
4437 PyModule_AddIntConstant(m, "HCI_FILTER", HCI_FILTER);
Hye-Shik Chang81268602004-02-02 06:05:24 +00004438#if !defined(__FreeBSD__)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004439 PyModule_AddIntConstant(m, "HCI_TIME_STAMP", HCI_TIME_STAMP);
4440 PyModule_AddIntConstant(m, "HCI_DATA_DIR", HCI_DATA_DIR);
4441 PyModule_AddIntConstant(m, "BTPROTO_SCO", BTPROTO_SCO);
Hye-Shik Chang81268602004-02-02 06:05:24 +00004442#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004443 PyModule_AddIntConstant(m, "BTPROTO_RFCOMM", BTPROTO_RFCOMM);
4444 PyModule_AddStringConstant(m, "BDADDR_ANY", "00:00:00:00:00:00");
4445 PyModule_AddStringConstant(m, "BDADDR_LOCAL", "00:00:00:FF:FF:FF");
Martin v. Löwis12af0482004-01-31 12:34:17 +00004446#endif
4447
Martin v. Löwis1ba3fd52001-08-10 20:29:40 +00004448#ifdef HAVE_NETPACKET_PACKET_H
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004449 PyModule_AddIntConstant(m, "AF_PACKET", AF_PACKET);
4450 PyModule_AddIntConstant(m, "PF_PACKET", PF_PACKET);
4451 PyModule_AddIntConstant(m, "PACKET_HOST", PACKET_HOST);
4452 PyModule_AddIntConstant(m, "PACKET_BROADCAST", PACKET_BROADCAST);
4453 PyModule_AddIntConstant(m, "PACKET_MULTICAST", PACKET_MULTICAST);
4454 PyModule_AddIntConstant(m, "PACKET_OTHERHOST", PACKET_OTHERHOST);
4455 PyModule_AddIntConstant(m, "PACKET_OUTGOING", PACKET_OUTGOING);
4456 PyModule_AddIntConstant(m, "PACKET_LOOPBACK", PACKET_LOOPBACK);
4457 PyModule_AddIntConstant(m, "PACKET_FASTROUTE", PACKET_FASTROUTE);
Guido van Rossum48a680c2001-03-02 06:34:14 +00004458#endif
Guido van Rossum09be4091999-08-09 14:40:40 +00004459
Christian Heimes043d6f62008-01-07 17:19:16 +00004460#ifdef HAVE_LINUX_TIPC_H
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004461 PyModule_AddIntConstant(m, "AF_TIPC", AF_TIPC);
Christian Heimes043d6f62008-01-07 17:19:16 +00004462
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004463 /* for addresses */
4464 PyModule_AddIntConstant(m, "TIPC_ADDR_NAMESEQ", TIPC_ADDR_NAMESEQ);
4465 PyModule_AddIntConstant(m, "TIPC_ADDR_NAME", TIPC_ADDR_NAME);
4466 PyModule_AddIntConstant(m, "TIPC_ADDR_ID", TIPC_ADDR_ID);
Christian Heimes043d6f62008-01-07 17:19:16 +00004467
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004468 PyModule_AddIntConstant(m, "TIPC_ZONE_SCOPE", TIPC_ZONE_SCOPE);
4469 PyModule_AddIntConstant(m, "TIPC_CLUSTER_SCOPE", TIPC_CLUSTER_SCOPE);
4470 PyModule_AddIntConstant(m, "TIPC_NODE_SCOPE", TIPC_NODE_SCOPE);
Christian Heimes043d6f62008-01-07 17:19:16 +00004471
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004472 /* for setsockopt() */
4473 PyModule_AddIntConstant(m, "SOL_TIPC", SOL_TIPC);
4474 PyModule_AddIntConstant(m, "TIPC_IMPORTANCE", TIPC_IMPORTANCE);
4475 PyModule_AddIntConstant(m, "TIPC_SRC_DROPPABLE", TIPC_SRC_DROPPABLE);
4476 PyModule_AddIntConstant(m, "TIPC_DEST_DROPPABLE",
4477 TIPC_DEST_DROPPABLE);
4478 PyModule_AddIntConstant(m, "TIPC_CONN_TIMEOUT", TIPC_CONN_TIMEOUT);
Christian Heimes043d6f62008-01-07 17:19:16 +00004479
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004480 PyModule_AddIntConstant(m, "TIPC_LOW_IMPORTANCE",
4481 TIPC_LOW_IMPORTANCE);
4482 PyModule_AddIntConstant(m, "TIPC_MEDIUM_IMPORTANCE",
4483 TIPC_MEDIUM_IMPORTANCE);
4484 PyModule_AddIntConstant(m, "TIPC_HIGH_IMPORTANCE",
4485 TIPC_HIGH_IMPORTANCE);
4486 PyModule_AddIntConstant(m, "TIPC_CRITICAL_IMPORTANCE",
4487 TIPC_CRITICAL_IMPORTANCE);
Christian Heimes043d6f62008-01-07 17:19:16 +00004488
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004489 /* for subscriptions */
4490 PyModule_AddIntConstant(m, "TIPC_SUB_PORTS", TIPC_SUB_PORTS);
4491 PyModule_AddIntConstant(m, "TIPC_SUB_SERVICE", TIPC_SUB_SERVICE);
Christian Heimes25bb7832008-01-11 16:17:00 +00004492#ifdef TIPC_SUB_CANCEL
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004493 /* doesn't seem to be available everywhere */
4494 PyModule_AddIntConstant(m, "TIPC_SUB_CANCEL", TIPC_SUB_CANCEL);
Christian Heimes25bb7832008-01-11 16:17:00 +00004495#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004496 PyModule_AddIntConstant(m, "TIPC_WAIT_FOREVER", TIPC_WAIT_FOREVER);
4497 PyModule_AddIntConstant(m, "TIPC_PUBLISHED", TIPC_PUBLISHED);
4498 PyModule_AddIntConstant(m, "TIPC_WITHDRAWN", TIPC_WITHDRAWN);
4499 PyModule_AddIntConstant(m, "TIPC_SUBSCR_TIMEOUT", TIPC_SUBSCR_TIMEOUT);
4500 PyModule_AddIntConstant(m, "TIPC_CFG_SRV", TIPC_CFG_SRV);
4501 PyModule_AddIntConstant(m, "TIPC_TOP_SRV", TIPC_TOP_SRV);
Christian Heimes043d6f62008-01-07 17:19:16 +00004502#endif
4503
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004504 /* Socket types */
4505 PyModule_AddIntConstant(m, "SOCK_STREAM", SOCK_STREAM);
4506 PyModule_AddIntConstant(m, "SOCK_DGRAM", SOCK_DGRAM);
Guido van Rossumbcc20741998-08-04 22:53:56 +00004507/* We have incomplete socket support. */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004508 PyModule_AddIntConstant(m, "SOCK_RAW", SOCK_RAW);
4509 PyModule_AddIntConstant(m, "SOCK_SEQPACKET", SOCK_SEQPACKET);
Martin v. Löwiscf8f47e2002-12-11 13:10:57 +00004510#if defined(SOCK_RDM)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004511 PyModule_AddIntConstant(m, "SOCK_RDM", SOCK_RDM);
Guido van Rossumbcc20741998-08-04 22:53:56 +00004512#endif
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004513
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004514#ifdef SO_DEBUG
4515 PyModule_AddIntConstant(m, "SO_DEBUG", SO_DEBUG);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004516#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004517#ifdef SO_ACCEPTCONN
4518 PyModule_AddIntConstant(m, "SO_ACCEPTCONN", SO_ACCEPTCONN);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004519#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004520#ifdef SO_REUSEADDR
4521 PyModule_AddIntConstant(m, "SO_REUSEADDR", SO_REUSEADDR);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004522#endif
Andrew M. Kuchling42851ab2004-07-10 14:19:21 +00004523#ifdef SO_EXCLUSIVEADDRUSE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004524 PyModule_AddIntConstant(m, "SO_EXCLUSIVEADDRUSE", SO_EXCLUSIVEADDRUSE);
Andrew M. Kuchling42851ab2004-07-10 14:19:21 +00004525#endif
4526
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004527#ifdef SO_KEEPALIVE
4528 PyModule_AddIntConstant(m, "SO_KEEPALIVE", SO_KEEPALIVE);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004529#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004530#ifdef SO_DONTROUTE
4531 PyModule_AddIntConstant(m, "SO_DONTROUTE", SO_DONTROUTE);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004532#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004533#ifdef SO_BROADCAST
4534 PyModule_AddIntConstant(m, "SO_BROADCAST", SO_BROADCAST);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004535#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004536#ifdef SO_USELOOPBACK
4537 PyModule_AddIntConstant(m, "SO_USELOOPBACK", SO_USELOOPBACK);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004538#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004539#ifdef SO_LINGER
4540 PyModule_AddIntConstant(m, "SO_LINGER", SO_LINGER);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004541#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004542#ifdef SO_OOBINLINE
4543 PyModule_AddIntConstant(m, "SO_OOBINLINE", SO_OOBINLINE);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004544#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004545#ifdef SO_REUSEPORT
4546 PyModule_AddIntConstant(m, "SO_REUSEPORT", SO_REUSEPORT);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004547#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004548#ifdef SO_SNDBUF
4549 PyModule_AddIntConstant(m, "SO_SNDBUF", SO_SNDBUF);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004550#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004551#ifdef SO_RCVBUF
4552 PyModule_AddIntConstant(m, "SO_RCVBUF", SO_RCVBUF);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004553#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004554#ifdef SO_SNDLOWAT
4555 PyModule_AddIntConstant(m, "SO_SNDLOWAT", SO_SNDLOWAT);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004556#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004557#ifdef SO_RCVLOWAT
4558 PyModule_AddIntConstant(m, "SO_RCVLOWAT", SO_RCVLOWAT);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004559#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004560#ifdef SO_SNDTIMEO
4561 PyModule_AddIntConstant(m, "SO_SNDTIMEO", SO_SNDTIMEO);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004562#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004563#ifdef SO_RCVTIMEO
4564 PyModule_AddIntConstant(m, "SO_RCVTIMEO", SO_RCVTIMEO);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004565#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004566#ifdef SO_ERROR
4567 PyModule_AddIntConstant(m, "SO_ERROR", SO_ERROR);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004568#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004569#ifdef SO_TYPE
4570 PyModule_AddIntConstant(m, "SO_TYPE", SO_TYPE);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004571#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004572#ifdef SO_SETFIB
4573 PyModule_AddIntConstant(m, "SO_SETFIB", SO_SETFIB);
Larry Hastingsf0f37952010-04-02 11:47:10 +00004574#endif
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004575
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004576 /* Maximum number of connections for "listen" */
4577#ifdef SOMAXCONN
4578 PyModule_AddIntConstant(m, "SOMAXCONN", SOMAXCONN);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004579#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004580 PyModule_AddIntConstant(m, "SOMAXCONN", 5); /* Common value */
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004581#endif
4582
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004583 /* Flags for send, recv */
4584#ifdef MSG_OOB
4585 PyModule_AddIntConstant(m, "MSG_OOB", MSG_OOB);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004586#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004587#ifdef MSG_PEEK
4588 PyModule_AddIntConstant(m, "MSG_PEEK", MSG_PEEK);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004589#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004590#ifdef MSG_DONTROUTE
4591 PyModule_AddIntConstant(m, "MSG_DONTROUTE", MSG_DONTROUTE);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004592#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004593#ifdef MSG_DONTWAIT
4594 PyModule_AddIntConstant(m, "MSG_DONTWAIT", MSG_DONTWAIT);
Guido van Rossum2c8bcb82000-04-25 21:34:53 +00004595#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004596#ifdef MSG_EOR
4597 PyModule_AddIntConstant(m, "MSG_EOR", MSG_EOR);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004598#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004599#ifdef MSG_TRUNC
4600 PyModule_AddIntConstant(m, "MSG_TRUNC", MSG_TRUNC);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004601#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004602#ifdef MSG_CTRUNC
4603 PyModule_AddIntConstant(m, "MSG_CTRUNC", MSG_CTRUNC);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004604#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004605#ifdef MSG_WAITALL
4606 PyModule_AddIntConstant(m, "MSG_WAITALL", MSG_WAITALL);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004607#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004608#ifdef MSG_BTAG
4609 PyModule_AddIntConstant(m, "MSG_BTAG", MSG_BTAG);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004610#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004611#ifdef MSG_ETAG
4612 PyModule_AddIntConstant(m, "MSG_ETAG", MSG_ETAG);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004613#endif
4614
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004615 /* Protocol level and numbers, usable for [gs]etsockopt */
4616#ifdef SOL_SOCKET
4617 PyModule_AddIntConstant(m, "SOL_SOCKET", SOL_SOCKET);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004618#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004619#ifdef SOL_IP
4620 PyModule_AddIntConstant(m, "SOL_IP", SOL_IP);
Guido van Rossum09be4091999-08-09 14:40:40 +00004621#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004622 PyModule_AddIntConstant(m, "SOL_IP", 0);
Guido van Rossum09be4091999-08-09 14:40:40 +00004623#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004624#ifdef SOL_IPX
4625 PyModule_AddIntConstant(m, "SOL_IPX", SOL_IPX);
Guido van Rossum09be4091999-08-09 14:40:40 +00004626#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004627#ifdef SOL_AX25
4628 PyModule_AddIntConstant(m, "SOL_AX25", SOL_AX25);
Guido van Rossum09be4091999-08-09 14:40:40 +00004629#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004630#ifdef SOL_ATALK
4631 PyModule_AddIntConstant(m, "SOL_ATALK", SOL_ATALK);
Guido van Rossum09be4091999-08-09 14:40:40 +00004632#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004633#ifdef SOL_NETROM
4634 PyModule_AddIntConstant(m, "SOL_NETROM", SOL_NETROM);
Guido van Rossum09be4091999-08-09 14:40:40 +00004635#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004636#ifdef SOL_ROSE
4637 PyModule_AddIntConstant(m, "SOL_ROSE", SOL_ROSE);
Guido van Rossum09be4091999-08-09 14:40:40 +00004638#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004639#ifdef SOL_TCP
4640 PyModule_AddIntConstant(m, "SOL_TCP", SOL_TCP);
Guido van Rossum09be4091999-08-09 14:40:40 +00004641#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004642 PyModule_AddIntConstant(m, "SOL_TCP", 6);
Guido van Rossum09be4091999-08-09 14:40:40 +00004643#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004644#ifdef SOL_UDP
4645 PyModule_AddIntConstant(m, "SOL_UDP", SOL_UDP);
Guido van Rossum09be4091999-08-09 14:40:40 +00004646#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004647 PyModule_AddIntConstant(m, "SOL_UDP", 17);
Guido van Rossum09be4091999-08-09 14:40:40 +00004648#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004649#ifdef IPPROTO_IP
4650 PyModule_AddIntConstant(m, "IPPROTO_IP", IPPROTO_IP);
Guido van Rossum578de301998-05-28 20:18:18 +00004651#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004652 PyModule_AddIntConstant(m, "IPPROTO_IP", 0);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004653#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004654#ifdef IPPROTO_HOPOPTS
4655 PyModule_AddIntConstant(m, "IPPROTO_HOPOPTS", IPPROTO_HOPOPTS);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004656#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004657#ifdef IPPROTO_ICMP
4658 PyModule_AddIntConstant(m, "IPPROTO_ICMP", IPPROTO_ICMP);
Guido van Rossum578de301998-05-28 20:18:18 +00004659#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004660 PyModule_AddIntConstant(m, "IPPROTO_ICMP", 1);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004661#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004662#ifdef IPPROTO_IGMP
4663 PyModule_AddIntConstant(m, "IPPROTO_IGMP", IPPROTO_IGMP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004664#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004665#ifdef IPPROTO_GGP
4666 PyModule_AddIntConstant(m, "IPPROTO_GGP", IPPROTO_GGP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004667#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004668#ifdef IPPROTO_IPV4
4669 PyModule_AddIntConstant(m, "IPPROTO_IPV4", IPPROTO_IPV4);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004670#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004671#ifdef IPPROTO_IPV6
4672 PyModule_AddIntConstant(m, "IPPROTO_IPV6", IPPROTO_IPV6);
Martin v. Löwisa0f17342003-10-03 13:56:20 +00004673#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004674#ifdef IPPROTO_IPIP
4675 PyModule_AddIntConstant(m, "IPPROTO_IPIP", IPPROTO_IPIP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004676#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004677#ifdef IPPROTO_TCP
4678 PyModule_AddIntConstant(m, "IPPROTO_TCP", IPPROTO_TCP);
Guido van Rossum578de301998-05-28 20:18:18 +00004679#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004680 PyModule_AddIntConstant(m, "IPPROTO_TCP", 6);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004681#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004682#ifdef IPPROTO_EGP
4683 PyModule_AddIntConstant(m, "IPPROTO_EGP", IPPROTO_EGP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004684#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004685#ifdef IPPROTO_PUP
4686 PyModule_AddIntConstant(m, "IPPROTO_PUP", IPPROTO_PUP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004687#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004688#ifdef IPPROTO_UDP
4689 PyModule_AddIntConstant(m, "IPPROTO_UDP", IPPROTO_UDP);
Guido van Rossum578de301998-05-28 20:18:18 +00004690#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004691 PyModule_AddIntConstant(m, "IPPROTO_UDP", 17);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004692#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004693#ifdef IPPROTO_IDP
4694 PyModule_AddIntConstant(m, "IPPROTO_IDP", IPPROTO_IDP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004695#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004696#ifdef IPPROTO_HELLO
4697 PyModule_AddIntConstant(m, "IPPROTO_HELLO", IPPROTO_HELLO);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004698#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004699#ifdef IPPROTO_ND
4700 PyModule_AddIntConstant(m, "IPPROTO_ND", IPPROTO_ND);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004701#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004702#ifdef IPPROTO_TP
4703 PyModule_AddIntConstant(m, "IPPROTO_TP", IPPROTO_TP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004704#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004705#ifdef IPPROTO_IPV6
4706 PyModule_AddIntConstant(m, "IPPROTO_IPV6", IPPROTO_IPV6);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004707#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004708#ifdef IPPROTO_ROUTING
4709 PyModule_AddIntConstant(m, "IPPROTO_ROUTING", IPPROTO_ROUTING);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004710#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004711#ifdef IPPROTO_FRAGMENT
4712 PyModule_AddIntConstant(m, "IPPROTO_FRAGMENT", IPPROTO_FRAGMENT);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004713#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004714#ifdef IPPROTO_RSVP
4715 PyModule_AddIntConstant(m, "IPPROTO_RSVP", IPPROTO_RSVP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004716#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004717#ifdef IPPROTO_GRE
4718 PyModule_AddIntConstant(m, "IPPROTO_GRE", IPPROTO_GRE);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004719#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004720#ifdef IPPROTO_ESP
4721 PyModule_AddIntConstant(m, "IPPROTO_ESP", IPPROTO_ESP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004722#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004723#ifdef IPPROTO_AH
4724 PyModule_AddIntConstant(m, "IPPROTO_AH", IPPROTO_AH);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004725#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004726#ifdef IPPROTO_MOBILE
4727 PyModule_AddIntConstant(m, "IPPROTO_MOBILE", IPPROTO_MOBILE);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004728#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004729#ifdef IPPROTO_ICMPV6
4730 PyModule_AddIntConstant(m, "IPPROTO_ICMPV6", IPPROTO_ICMPV6);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004731#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004732#ifdef IPPROTO_NONE
4733 PyModule_AddIntConstant(m, "IPPROTO_NONE", IPPROTO_NONE);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004734#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004735#ifdef IPPROTO_DSTOPTS
4736 PyModule_AddIntConstant(m, "IPPROTO_DSTOPTS", IPPROTO_DSTOPTS);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004737#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004738#ifdef IPPROTO_XTP
4739 PyModule_AddIntConstant(m, "IPPROTO_XTP", IPPROTO_XTP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004740#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004741#ifdef IPPROTO_EON
4742 PyModule_AddIntConstant(m, "IPPROTO_EON", IPPROTO_EON);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004743#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004744#ifdef IPPROTO_PIM
4745 PyModule_AddIntConstant(m, "IPPROTO_PIM", IPPROTO_PIM);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004746#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004747#ifdef IPPROTO_IPCOMP
4748 PyModule_AddIntConstant(m, "IPPROTO_IPCOMP", IPPROTO_IPCOMP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004749#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004750#ifdef IPPROTO_VRRP
4751 PyModule_AddIntConstant(m, "IPPROTO_VRRP", IPPROTO_VRRP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004752#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004753#ifdef IPPROTO_BIP
4754 PyModule_AddIntConstant(m, "IPPROTO_BIP", IPPROTO_BIP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004755#endif
4756/**/
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004757#ifdef IPPROTO_RAW
4758 PyModule_AddIntConstant(m, "IPPROTO_RAW", IPPROTO_RAW);
Guido van Rossum578de301998-05-28 20:18:18 +00004759#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004760 PyModule_AddIntConstant(m, "IPPROTO_RAW", 255);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004761#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004762#ifdef IPPROTO_MAX
4763 PyModule_AddIntConstant(m, "IPPROTO_MAX", IPPROTO_MAX);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004764#endif
4765
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004766 /* Some port configuration */
4767#ifdef IPPORT_RESERVED
4768 PyModule_AddIntConstant(m, "IPPORT_RESERVED", IPPORT_RESERVED);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004769#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004770 PyModule_AddIntConstant(m, "IPPORT_RESERVED", 1024);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004771#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004772#ifdef IPPORT_USERRESERVED
4773 PyModule_AddIntConstant(m, "IPPORT_USERRESERVED", IPPORT_USERRESERVED);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004774#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004775 PyModule_AddIntConstant(m, "IPPORT_USERRESERVED", 5000);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004776#endif
4777
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004778 /* Some reserved IP v.4 addresses */
4779#ifdef INADDR_ANY
4780 PyModule_AddIntConstant(m, "INADDR_ANY", INADDR_ANY);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004781#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004782 PyModule_AddIntConstant(m, "INADDR_ANY", 0x00000000);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004783#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004784#ifdef INADDR_BROADCAST
4785 PyModule_AddIntConstant(m, "INADDR_BROADCAST", INADDR_BROADCAST);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004786#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004787 PyModule_AddIntConstant(m, "INADDR_BROADCAST", 0xffffffff);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004788#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004789#ifdef INADDR_LOOPBACK
4790 PyModule_AddIntConstant(m, "INADDR_LOOPBACK", INADDR_LOOPBACK);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004791#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004792 PyModule_AddIntConstant(m, "INADDR_LOOPBACK", 0x7F000001);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004793#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004794#ifdef INADDR_UNSPEC_GROUP
4795 PyModule_AddIntConstant(m, "INADDR_UNSPEC_GROUP", INADDR_UNSPEC_GROUP);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004796#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004797 PyModule_AddIntConstant(m, "INADDR_UNSPEC_GROUP", 0xe0000000);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004798#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004799#ifdef INADDR_ALLHOSTS_GROUP
4800 PyModule_AddIntConstant(m, "INADDR_ALLHOSTS_GROUP",
4801 INADDR_ALLHOSTS_GROUP);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004802#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004803 PyModule_AddIntConstant(m, "INADDR_ALLHOSTS_GROUP", 0xe0000001);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004804#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004805#ifdef INADDR_MAX_LOCAL_GROUP
4806 PyModule_AddIntConstant(m, "INADDR_MAX_LOCAL_GROUP",
4807 INADDR_MAX_LOCAL_GROUP);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004808#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004809 PyModule_AddIntConstant(m, "INADDR_MAX_LOCAL_GROUP", 0xe00000ff);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004810#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004811#ifdef INADDR_NONE
4812 PyModule_AddIntConstant(m, "INADDR_NONE", INADDR_NONE);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004813#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004814 PyModule_AddIntConstant(m, "INADDR_NONE", 0xffffffff);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004815#endif
4816
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004817 /* IPv4 [gs]etsockopt options */
4818#ifdef IP_OPTIONS
4819 PyModule_AddIntConstant(m, "IP_OPTIONS", IP_OPTIONS);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004820#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004821#ifdef IP_HDRINCL
4822 PyModule_AddIntConstant(m, "IP_HDRINCL", IP_HDRINCL);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004823#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004824#ifdef IP_TOS
4825 PyModule_AddIntConstant(m, "IP_TOS", IP_TOS);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004826#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004827#ifdef IP_TTL
4828 PyModule_AddIntConstant(m, "IP_TTL", IP_TTL);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004829#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004830#ifdef IP_RECVOPTS
4831 PyModule_AddIntConstant(m, "IP_RECVOPTS", IP_RECVOPTS);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004832#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004833#ifdef IP_RECVRETOPTS
4834 PyModule_AddIntConstant(m, "IP_RECVRETOPTS", IP_RECVRETOPTS);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004835#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004836#ifdef IP_RECVDSTADDR
4837 PyModule_AddIntConstant(m, "IP_RECVDSTADDR", IP_RECVDSTADDR);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004838#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004839#ifdef IP_RETOPTS
4840 PyModule_AddIntConstant(m, "IP_RETOPTS", IP_RETOPTS);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004841#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004842#ifdef IP_MULTICAST_IF
4843 PyModule_AddIntConstant(m, "IP_MULTICAST_IF", IP_MULTICAST_IF);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004844#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004845#ifdef IP_MULTICAST_TTL
4846 PyModule_AddIntConstant(m, "IP_MULTICAST_TTL", IP_MULTICAST_TTL);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004847#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004848#ifdef IP_MULTICAST_LOOP
4849 PyModule_AddIntConstant(m, "IP_MULTICAST_LOOP", IP_MULTICAST_LOOP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004850#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004851#ifdef IP_ADD_MEMBERSHIP
4852 PyModule_AddIntConstant(m, "IP_ADD_MEMBERSHIP", IP_ADD_MEMBERSHIP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004853#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004854#ifdef IP_DROP_MEMBERSHIP
4855 PyModule_AddIntConstant(m, "IP_DROP_MEMBERSHIP", IP_DROP_MEMBERSHIP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004856#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004857#ifdef IP_DEFAULT_MULTICAST_TTL
4858 PyModule_AddIntConstant(m, "IP_DEFAULT_MULTICAST_TTL",
4859 IP_DEFAULT_MULTICAST_TTL);
Guido van Rossum09be4091999-08-09 14:40:40 +00004860#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004861#ifdef IP_DEFAULT_MULTICAST_LOOP
4862 PyModule_AddIntConstant(m, "IP_DEFAULT_MULTICAST_LOOP",
4863 IP_DEFAULT_MULTICAST_LOOP);
Guido van Rossum09be4091999-08-09 14:40:40 +00004864#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004865#ifdef IP_MAX_MEMBERSHIPS
4866 PyModule_AddIntConstant(m, "IP_MAX_MEMBERSHIPS", IP_MAX_MEMBERSHIPS);
Guido van Rossum09be4091999-08-09 14:40:40 +00004867#endif
4868
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004869 /* IPv6 [gs]etsockopt options, defined in RFC2553 */
4870#ifdef IPV6_JOIN_GROUP
4871 PyModule_AddIntConstant(m, "IPV6_JOIN_GROUP", IPV6_JOIN_GROUP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004872#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004873#ifdef IPV6_LEAVE_GROUP
4874 PyModule_AddIntConstant(m, "IPV6_LEAVE_GROUP", IPV6_LEAVE_GROUP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004875#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004876#ifdef IPV6_MULTICAST_HOPS
4877 PyModule_AddIntConstant(m, "IPV6_MULTICAST_HOPS", IPV6_MULTICAST_HOPS);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004878#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004879#ifdef IPV6_MULTICAST_IF
4880 PyModule_AddIntConstant(m, "IPV6_MULTICAST_IF", IPV6_MULTICAST_IF);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004881#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004882#ifdef IPV6_MULTICAST_LOOP
4883 PyModule_AddIntConstant(m, "IPV6_MULTICAST_LOOP", IPV6_MULTICAST_LOOP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004884#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004885#ifdef IPV6_UNICAST_HOPS
4886 PyModule_AddIntConstant(m, "IPV6_UNICAST_HOPS", IPV6_UNICAST_HOPS);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004887#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004888 /* Additional IPV6 socket options, defined in RFC 3493 */
Martin v. Löwisda91d022003-12-30 11:14:01 +00004889#ifdef IPV6_V6ONLY
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004890 PyModule_AddIntConstant(m, "IPV6_V6ONLY", IPV6_V6ONLY);
Martin v. Löwisda91d022003-12-30 11:14:01 +00004891#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004892 /* Advanced IPV6 socket options, from RFC 3542 */
Martin v. Löwisda91d022003-12-30 11:14:01 +00004893#ifdef IPV6_CHECKSUM
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004894 PyModule_AddIntConstant(m, "IPV6_CHECKSUM", IPV6_CHECKSUM);
Martin v. Löwisda91d022003-12-30 11:14:01 +00004895#endif
4896#ifdef IPV6_DONTFRAG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004897 PyModule_AddIntConstant(m, "IPV6_DONTFRAG", IPV6_DONTFRAG);
Martin v. Löwisda91d022003-12-30 11:14:01 +00004898#endif
4899#ifdef IPV6_DSTOPTS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004900 PyModule_AddIntConstant(m, "IPV6_DSTOPTS", IPV6_DSTOPTS);
Martin v. Löwisda91d022003-12-30 11:14:01 +00004901#endif
4902#ifdef IPV6_HOPLIMIT
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004903 PyModule_AddIntConstant(m, "IPV6_HOPLIMIT", IPV6_HOPLIMIT);
Martin v. Löwisda91d022003-12-30 11:14:01 +00004904#endif
4905#ifdef IPV6_HOPOPTS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004906 PyModule_AddIntConstant(m, "IPV6_HOPOPTS", IPV6_HOPOPTS);
Martin v. Löwisda91d022003-12-30 11:14:01 +00004907#endif
4908#ifdef IPV6_NEXTHOP
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004909 PyModule_AddIntConstant(m, "IPV6_NEXTHOP", IPV6_NEXTHOP);
Martin v. Löwisda91d022003-12-30 11:14:01 +00004910#endif
4911#ifdef IPV6_PATHMTU
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004912 PyModule_AddIntConstant(m, "IPV6_PATHMTU", IPV6_PATHMTU);
Martin v. Löwisda91d022003-12-30 11:14:01 +00004913#endif
4914#ifdef IPV6_PKTINFO
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004915 PyModule_AddIntConstant(m, "IPV6_PKTINFO", IPV6_PKTINFO);
Martin v. Löwisda91d022003-12-30 11:14:01 +00004916#endif
4917#ifdef IPV6_RECVDSTOPTS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004918 PyModule_AddIntConstant(m, "IPV6_RECVDSTOPTS", IPV6_RECVDSTOPTS);
Martin v. Löwisda91d022003-12-30 11:14:01 +00004919#endif
4920#ifdef IPV6_RECVHOPLIMIT
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004921 PyModule_AddIntConstant(m, "IPV6_RECVHOPLIMIT", IPV6_RECVHOPLIMIT);
Martin v. Löwisda91d022003-12-30 11:14:01 +00004922#endif
4923#ifdef IPV6_RECVHOPOPTS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004924 PyModule_AddIntConstant(m, "IPV6_RECVHOPOPTS", IPV6_RECVHOPOPTS);
Martin v. Löwisda91d022003-12-30 11:14:01 +00004925#endif
4926#ifdef IPV6_RECVPKTINFO
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004927 PyModule_AddIntConstant(m, "IPV6_RECVPKTINFO", IPV6_RECVPKTINFO);
Martin v. Löwisda91d022003-12-30 11:14:01 +00004928#endif
4929#ifdef IPV6_RECVRTHDR
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004930 PyModule_AddIntConstant(m, "IPV6_RECVRTHDR", IPV6_RECVRTHDR);
Martin v. Löwisda91d022003-12-30 11:14:01 +00004931#endif
4932#ifdef IPV6_RECVTCLASS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004933 PyModule_AddIntConstant(m, "IPV6_RECVTCLASS", IPV6_RECVTCLASS);
Martin v. Löwisda91d022003-12-30 11:14:01 +00004934#endif
4935#ifdef IPV6_RTHDR
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004936 PyModule_AddIntConstant(m, "IPV6_RTHDR", IPV6_RTHDR);
Martin v. Löwisda91d022003-12-30 11:14:01 +00004937#endif
4938#ifdef IPV6_RTHDRDSTOPTS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004939 PyModule_AddIntConstant(m, "IPV6_RTHDRDSTOPTS", IPV6_RTHDRDSTOPTS);
Martin v. Löwisda91d022003-12-30 11:14:01 +00004940#endif
4941#ifdef IPV6_RTHDR_TYPE_0
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004942 PyModule_AddIntConstant(m, "IPV6_RTHDR_TYPE_0", IPV6_RTHDR_TYPE_0);
Martin v. Löwisda91d022003-12-30 11:14:01 +00004943#endif
4944#ifdef IPV6_RECVPATHMTU
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004945 PyModule_AddIntConstant(m, "IPV6_RECVPATHMTU", IPV6_RECVPATHMTU);
Martin v. Löwisda91d022003-12-30 11:14:01 +00004946#endif
4947#ifdef IPV6_TCLASS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004948 PyModule_AddIntConstant(m, "IPV6_TCLASS", IPV6_TCLASS);
Martin v. Löwisda91d022003-12-30 11:14:01 +00004949#endif
4950#ifdef IPV6_USE_MIN_MTU
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004951 PyModule_AddIntConstant(m, "IPV6_USE_MIN_MTU", IPV6_USE_MIN_MTU);
Martin v. Löwisda91d022003-12-30 11:14:01 +00004952#endif
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004953
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004954 /* TCP options */
4955#ifdef TCP_NODELAY
4956 PyModule_AddIntConstant(m, "TCP_NODELAY", TCP_NODELAY);
Guido van Rossum09be4091999-08-09 14:40:40 +00004957#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004958#ifdef TCP_MAXSEG
4959 PyModule_AddIntConstant(m, "TCP_MAXSEG", TCP_MAXSEG);
Guido van Rossum09be4091999-08-09 14:40:40 +00004960#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004961#ifdef TCP_CORK
4962 PyModule_AddIntConstant(m, "TCP_CORK", TCP_CORK);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00004963#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004964#ifdef TCP_KEEPIDLE
4965 PyModule_AddIntConstant(m, "TCP_KEEPIDLE", TCP_KEEPIDLE);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00004966#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004967#ifdef TCP_KEEPINTVL
4968 PyModule_AddIntConstant(m, "TCP_KEEPINTVL", TCP_KEEPINTVL);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00004969#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004970#ifdef TCP_KEEPCNT
4971 PyModule_AddIntConstant(m, "TCP_KEEPCNT", TCP_KEEPCNT);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00004972#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004973#ifdef TCP_SYNCNT
4974 PyModule_AddIntConstant(m, "TCP_SYNCNT", TCP_SYNCNT);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00004975#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004976#ifdef TCP_LINGER2
4977 PyModule_AddIntConstant(m, "TCP_LINGER2", TCP_LINGER2);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00004978#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004979#ifdef TCP_DEFER_ACCEPT
4980 PyModule_AddIntConstant(m, "TCP_DEFER_ACCEPT", TCP_DEFER_ACCEPT);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00004981#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004982#ifdef TCP_WINDOW_CLAMP
4983 PyModule_AddIntConstant(m, "TCP_WINDOW_CLAMP", TCP_WINDOW_CLAMP);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00004984#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004985#ifdef TCP_INFO
4986 PyModule_AddIntConstant(m, "TCP_INFO", TCP_INFO);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00004987#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004988#ifdef TCP_QUICKACK
4989 PyModule_AddIntConstant(m, "TCP_QUICKACK", TCP_QUICKACK);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00004990#endif
4991
Guido van Rossum09be4091999-08-09 14:40:40 +00004992
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004993 /* IPX options */
4994#ifdef IPX_TYPE
4995 PyModule_AddIntConstant(m, "IPX_TYPE", IPX_TYPE);
Guido van Rossum09be4091999-08-09 14:40:40 +00004996#endif
Guido van Rossum4f199ea1998-04-09 20:56:35 +00004997
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004998 /* get{addr,name}info parameters */
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004999#ifdef EAI_ADDRFAMILY
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005000 PyModule_AddIntConstant(m, "EAI_ADDRFAMILY", EAI_ADDRFAMILY);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005001#endif
5002#ifdef EAI_AGAIN
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005003 PyModule_AddIntConstant(m, "EAI_AGAIN", EAI_AGAIN);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005004#endif
5005#ifdef EAI_BADFLAGS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005006 PyModule_AddIntConstant(m, "EAI_BADFLAGS", EAI_BADFLAGS);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005007#endif
5008#ifdef EAI_FAIL
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005009 PyModule_AddIntConstant(m, "EAI_FAIL", EAI_FAIL);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005010#endif
5011#ifdef EAI_FAMILY
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005012 PyModule_AddIntConstant(m, "EAI_FAMILY", EAI_FAMILY);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005013#endif
5014#ifdef EAI_MEMORY
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005015 PyModule_AddIntConstant(m, "EAI_MEMORY", EAI_MEMORY);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005016#endif
5017#ifdef EAI_NODATA
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005018 PyModule_AddIntConstant(m, "EAI_NODATA", EAI_NODATA);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005019#endif
5020#ifdef EAI_NONAME
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005021 PyModule_AddIntConstant(m, "EAI_NONAME", EAI_NONAME);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005022#endif
Martin v. Löwisda91d022003-12-30 11:14:01 +00005023#ifdef EAI_OVERFLOW
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005024 PyModule_AddIntConstant(m, "EAI_OVERFLOW", EAI_OVERFLOW);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005025#endif
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005026#ifdef EAI_SERVICE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005027 PyModule_AddIntConstant(m, "EAI_SERVICE", EAI_SERVICE);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005028#endif
5029#ifdef EAI_SOCKTYPE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005030 PyModule_AddIntConstant(m, "EAI_SOCKTYPE", EAI_SOCKTYPE);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005031#endif
5032#ifdef EAI_SYSTEM
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005033 PyModule_AddIntConstant(m, "EAI_SYSTEM", EAI_SYSTEM);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005034#endif
5035#ifdef EAI_BADHINTS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005036 PyModule_AddIntConstant(m, "EAI_BADHINTS", EAI_BADHINTS);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005037#endif
5038#ifdef EAI_PROTOCOL
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005039 PyModule_AddIntConstant(m, "EAI_PROTOCOL", EAI_PROTOCOL);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005040#endif
5041#ifdef EAI_MAX
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005042 PyModule_AddIntConstant(m, "EAI_MAX", EAI_MAX);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005043#endif
5044#ifdef AI_PASSIVE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005045 PyModule_AddIntConstant(m, "AI_PASSIVE", AI_PASSIVE);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005046#endif
5047#ifdef AI_CANONNAME
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005048 PyModule_AddIntConstant(m, "AI_CANONNAME", AI_CANONNAME);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005049#endif
5050#ifdef AI_NUMERICHOST
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005051 PyModule_AddIntConstant(m, "AI_NUMERICHOST", AI_NUMERICHOST);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005052#endif
Martin v. Löwisda91d022003-12-30 11:14:01 +00005053#ifdef AI_NUMERICSERV
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005054 PyModule_AddIntConstant(m, "AI_NUMERICSERV", AI_NUMERICSERV);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005055#endif
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005056#ifdef AI_MASK
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005057 PyModule_AddIntConstant(m, "AI_MASK", AI_MASK);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005058#endif
5059#ifdef AI_ALL
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005060 PyModule_AddIntConstant(m, "AI_ALL", AI_ALL);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005061#endif
5062#ifdef AI_V4MAPPED_CFG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005063 PyModule_AddIntConstant(m, "AI_V4MAPPED_CFG", AI_V4MAPPED_CFG);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005064#endif
5065#ifdef AI_ADDRCONFIG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005066 PyModule_AddIntConstant(m, "AI_ADDRCONFIG", AI_ADDRCONFIG);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005067#endif
5068#ifdef AI_V4MAPPED
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005069 PyModule_AddIntConstant(m, "AI_V4MAPPED", AI_V4MAPPED);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005070#endif
5071#ifdef AI_DEFAULT
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005072 PyModule_AddIntConstant(m, "AI_DEFAULT", AI_DEFAULT);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005073#endif
5074#ifdef NI_MAXHOST
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005075 PyModule_AddIntConstant(m, "NI_MAXHOST", NI_MAXHOST);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005076#endif
5077#ifdef NI_MAXSERV
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005078 PyModule_AddIntConstant(m, "NI_MAXSERV", NI_MAXSERV);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005079#endif
5080#ifdef NI_NOFQDN
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005081 PyModule_AddIntConstant(m, "NI_NOFQDN", NI_NOFQDN);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005082#endif
5083#ifdef NI_NUMERICHOST
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005084 PyModule_AddIntConstant(m, "NI_NUMERICHOST", NI_NUMERICHOST);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005085#endif
5086#ifdef NI_NAMEREQD
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005087 PyModule_AddIntConstant(m, "NI_NAMEREQD", NI_NAMEREQD);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005088#endif
5089#ifdef NI_NUMERICSERV
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005090 PyModule_AddIntConstant(m, "NI_NUMERICSERV", NI_NUMERICSERV);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005091#endif
5092#ifdef NI_DGRAM
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005093 PyModule_AddIntConstant(m, "NI_DGRAM", NI_DGRAM);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005094#endif
5095
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005096 /* shutdown() parameters */
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005097#ifdef SHUT_RD
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005098 PyModule_AddIntConstant(m, "SHUT_RD", SHUT_RD);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005099#elif defined(SD_RECEIVE)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005100 PyModule_AddIntConstant(m, "SHUT_RD", SD_RECEIVE);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005101#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005102 PyModule_AddIntConstant(m, "SHUT_RD", 0);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005103#endif
5104#ifdef SHUT_WR
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005105 PyModule_AddIntConstant(m, "SHUT_WR", SHUT_WR);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005106#elif defined(SD_SEND)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005107 PyModule_AddIntConstant(m, "SHUT_WR", SD_SEND);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005108#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005109 PyModule_AddIntConstant(m, "SHUT_WR", 1);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005110#endif
5111#ifdef SHUT_RDWR
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005112 PyModule_AddIntConstant(m, "SHUT_RDWR", SHUT_RDWR);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005113#elif defined(SD_BOTH)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005114 PyModule_AddIntConstant(m, "SHUT_RDWR", SD_BOTH);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005115#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005116 PyModule_AddIntConstant(m, "SHUT_RDWR", 2);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005117#endif
5118
Christian Heimesfaf2f632008-01-06 16:59:19 +00005119#ifdef SIO_RCVALL
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005120 {
5121 DWORD codes[] = {SIO_RCVALL, SIO_KEEPALIVE_VALS};
5122 const char *names[] = {"SIO_RCVALL", "SIO_KEEPALIVE_VALS"};
5123 int i;
5124 for(i = 0; i<sizeof(codes)/sizeof(*codes); ++i) {
5125 PyObject *tmp;
5126 tmp = PyLong_FromUnsignedLong(codes[i]);
5127 if (tmp == NULL)
5128 return NULL;
5129 PyModule_AddObject(m, names[i], tmp);
5130 }
5131 }
5132 PyModule_AddIntConstant(m, "RCVALL_OFF", RCVALL_OFF);
5133 PyModule_AddIntConstant(m, "RCVALL_ON", RCVALL_ON);
5134 PyModule_AddIntConstant(m, "RCVALL_SOCKETLEVELONLY", RCVALL_SOCKETLEVELONLY);
Amaury Forgeot d'Arc762681b2008-06-12 23:03:41 +00005135#ifdef RCVALL_IPLEVEL
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005136 PyModule_AddIntConstant(m, "RCVALL_IPLEVEL", RCVALL_IPLEVEL);
Amaury Forgeot d'Arc762681b2008-06-12 23:03:41 +00005137#endif
5138#ifdef RCVALL_MAX
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005139 PyModule_AddIntConstant(m, "RCVALL_MAX", RCVALL_MAX);
Amaury Forgeot d'Arc762681b2008-06-12 23:03:41 +00005140#endif
Christian Heimesfaf2f632008-01-06 16:59:19 +00005141#endif /* _MSTCPIP_ */
5142
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005143 /* Initialize gethostbyname lock */
Just van Rossum1040d2c2003-05-09 07:53:18 +00005144#if defined(USE_GETHOSTBYNAME_LOCK) || defined(USE_GETADDRINFO_LOCK)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005145 netdb_lock = PyThread_allocate_lock();
Guido van Rossum4f199ea1998-04-09 20:56:35 +00005146#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005147 return m;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00005148}
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005149
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00005150
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005151#ifndef HAVE_INET_PTON
Christian Heimes96e7b3d2007-11-20 06:51:17 +00005152#if !defined(NTDDI_VERSION) || (NTDDI_VERSION < NTDDI_LONGHORN)
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00005153
5154/* Simplistic emulation code for inet_pton that only works for IPv4 */
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00005155/* These are not exposed because they do not set errno properly */
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00005156
Guido van Rossum3eede5a2002-06-07 02:08:35 +00005157int
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00005158inet_pton(int af, const char *src, void *dst)
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005159{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005160 if (af == AF_INET) {
Benjamin Petersonf91df042009-02-13 02:50:59 +00005161#if (SIZEOF_INT != 4)
5162#error "Not sure if in_addr_t exists and int is not 32-bits."
5163#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005164 unsigned int packed_addr;
5165 packed_addr = inet_addr(src);
5166 if (packed_addr == INADDR_NONE)
5167 return 0;
5168 memcpy(dst, &packed_addr, 4);
5169 return 1;
5170 }
5171 /* Should set errno to EAFNOSUPPORT */
5172 return -1;
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005173}
5174
Martin v. Löwisc925b1532001-07-21 09:42:15 +00005175const char *
5176inet_ntop(int af, const void *src, char *dst, socklen_t size)
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005177{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005178 if (af == AF_INET) {
5179 struct in_addr packed_addr;
5180 if (size < 16)
5181 /* Should set errno to ENOSPC. */
5182 return NULL;
5183 memcpy(&packed_addr, src, sizeof(packed_addr));
5184 return strncpy(dst, inet_ntoa(packed_addr), size);
5185 }
5186 /* Should set errno to EAFNOSUPPORT */
5187 return NULL;
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005188}
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00005189
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005190#endif
Christian Heimesb6150692007-11-15 23:37:07 +00005191#endif