blob: d0e1ed6657b124d0e79bd5397c4fcd3f35f4b931 [file] [log] [blame]
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001/* Socket module */
2
3/*
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004
Guido van Rossum6574b3e1991-06-25 21:36:08 +00005This module provides an interface to Berkeley socket IPC.
6
7Limitations:
8
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00009- Only AF_INET, AF_INET6 and AF_UNIX address families are supported in a
Christian Heimes043d6f62008-01-07 17:19:16 +000010 portable manner, though AF_PACKET, AF_NETLINK and AF_TIPC are supported
11 under Linux.
Guido van Rossumc4fcfa32002-06-07 03:19:37 +000012- No read/write operations (use sendall/recv or makefile instead).
13- Additional restrictions apply on some non-Unix platforms (compensated
14 for by socket.py).
Guido van Rossum6574b3e1991-06-25 21:36:08 +000015
Guido van Rossum27e177d1995-03-16 15:43:47 +000016Module interface:
Guido van Rossum6574b3e1991-06-25 21:36:08 +000017
Guido van Rossum27e177d1995-03-16 15:43:47 +000018- socket.error: exception raised for socket specific errors
Martin v. Löwis2d8d4272001-07-21 18:05:31 +000019- socket.gaierror: exception raised for getaddrinfo/getnameinfo errors,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000020 a subclass of socket.error
Martin v. Löwis2d8d4272001-07-21 18:05:31 +000021- socket.herror: exception raised for gethostby* errors,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000022 a subclass of socket.error
Guido van Rossum30a685f1991-06-27 15:51:29 +000023- socket.gethostbyname(hostname) --> host IP address (string: 'dd.dd.dd.dd')
Guido van Rossum3bbc62e1995-01-02 19:30:30 +000024- socket.gethostbyaddr(IP address) --> (hostname, [alias, ...], [IP addr, ...])
Guido van Rossum27e177d1995-03-16 15:43:47 +000025- socket.gethostname() --> host name (string: 'spam' or 'spam.domain.com')
Guido van Rossum25405c71996-12-19 16:42:52 +000026- socket.getprotobyname(protocolname) --> protocol number
Barry Warsaw11b91a02004-06-28 00:50:43 +000027- socket.getservbyname(servicename[, protocolname]) --> port number
28- socket.getservbyport(portnumber[, protocolname]) --> service name
Guido van Rossum7d0a8262007-05-21 23:13:11 +000029- socket.socket([family[, type [, proto, fileno]]]) --> new socket object
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000030 (fileno specifies a pre-existing socket file descriptor)
Dave Cole331708b2004-08-09 04:51:41 +000031- socket.socketpair([family[, type [, proto]]]) --> (socket, socket)
Guido van Rossum006bf911996-06-12 04:04:55 +000032- socket.ntohs(16 bit value) --> new int object
33- socket.ntohl(32 bit value) --> new int object
34- socket.htons(16 bit value) --> new int object
35- socket.htonl(32 bit value) --> new int object
Martin v. Löwis2d8d4272001-07-21 18:05:31 +000036- socket.getaddrinfo(host, port [, family, socktype, proto, flags])
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000037 --> List of (family, socktype, proto, canonname, sockaddr)
Martin v. Löwis2d8d4272001-07-21 18:05:31 +000038- socket.getnameinfo(sockaddr, flags) --> (host, port)
Guido van Rossum27e177d1995-03-16 15:43:47 +000039- socket.AF_INET, socket.SOCK_STREAM, etc.: constants from <socket.h>
Guido van Rossum47dfa4a2003-04-25 05:48:32 +000040- socket.has_ipv6: boolean value indicating if IPv6 is supported
Guido van Rossum5c9eb211999-08-20 18:21:51 +000041- socket.inet_aton(IP address) -> 32-bit packed IP representation
42- socket.inet_ntoa(packed IP) -> IP address string
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +000043- socket.getdefaulttimeout() -> None | float
44- socket.setdefaulttimeout(None | float)
Guido van Rossum6574b3e1991-06-25 21:36:08 +000045- an Internet socket address is a pair (hostname, port)
46 where hostname can be anything recognized by gethostbyname()
47 (including the dd.dd.dd.dd notation) and port is in host byte order
48- where a hostname is returned, the dd.dd.dd.dd notation is used
Guido van Rossum27e177d1995-03-16 15:43:47 +000049- a UNIX domain socket address is a string specifying the pathname
Jeremy Hylton22308652001-02-02 03:23:09 +000050- an AF_PACKET socket address is a tuple containing a string
51 specifying the ethernet interface and an integer specifying
52 the Ethernet protocol number to be received. For example:
Jeremy Hyltondbfb6622001-02-02 19:55:17 +000053 ("eth0",0x1234). Optional 3rd,4th,5th elements in the tuple
Andrew M. Kuchlingb8e17172004-07-10 23:39:35 +000054 specify packet-type and ha-type/addr.
Christian Heimes043d6f62008-01-07 17:19:16 +000055- an AF_TIPC socket address is expressed as
56 (addr_type, v1, v2, v3 [, scope]); where addr_type can be one of:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000057 TIPC_ADDR_NAMESEQ, TIPC_ADDR_NAME, and TIPC_ADDR_ID;
Christian Heimes043d6f62008-01-07 17:19:16 +000058 and scope can be one of:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000059 TIPC_ZONE_SCOPE, TIPC_CLUSTER_SCOPE, and TIPC_NODE_SCOPE.
Christian Heimes043d6f62008-01-07 17:19:16 +000060 The meaning of v1, v2 and v3 depends on the value of addr_type:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000061 if addr_type is TIPC_ADDR_NAME:
62 v1 is the server type
63 v2 is the port identifier
64 v3 is ignored
65 if addr_type is TIPC_ADDR_NAMESEQ:
66 v1 is the server type
67 v2 is the lower port number
68 v3 is the upper port number
69 if addr_type is TIPC_ADDR_ID:
70 v1 is the node
71 v2 is the ref
72 v3 is ignored
Christian Heimes043d6f62008-01-07 17:19:16 +000073
Guido van Rossum6574b3e1991-06-25 21:36:08 +000074
Guido van Rossumc4fcfa32002-06-07 03:19:37 +000075Local naming conventions:
Guido van Rossum6574b3e1991-06-25 21:36:08 +000076
Guido van Rossumc4fcfa32002-06-07 03:19:37 +000077- names starting with sock_ are socket object methods
78- names starting with socket_ are module-level functions
79- names starting with PySocket are exported through socketmodule.h
Guido van Rossum30a685f1991-06-27 15:51:29 +000080
Guido van Rossum6574b3e1991-06-25 21:36:08 +000081*/
82
Thomas Wouters477c8d52006-05-27 19:21:47 +000083#ifdef __APPLE__
84 /*
85 * inet_aton is not available on OSX 10.3, yet we want to use a binary
86 * that was build on 10.4 or later to work on that release, weak linking
87 * comes to the rescue.
88 */
89# pragma weak inet_aton
90#endif
91
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000092#include "Python.h"
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000093#include "structmember.h"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000094
Guido van Rossum47dfa4a2003-04-25 05:48:32 +000095#undef MAX
96#define MAX(x, y) ((x) < (y) ? (y) : (x))
97
Guido van Rossumc4fcfa32002-06-07 03:19:37 +000098/* Socket object documentation */
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000099PyDoc_STRVAR(sock_doc,
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000100"socket([family[, type[, proto]]]) -> socket object\n\
101\n\
102Open a socket of the given type. The family argument specifies the\n\
103address family; it defaults to AF_INET. The type argument specifies\n\
104whether this is a stream (SOCK_STREAM, this is the default)\n\
105or datagram (SOCK_DGRAM) socket. The protocol argument defaults to 0,\n\
106specifying the default protocol. Keyword arguments are accepted.\n\
107\n\
108A socket object represents one endpoint of a network connection.\n\
109\n\
110Methods of socket objects (keyword arguments not allowed):\n\
111\n\
Guido van Rossum39eb8fa2007-11-16 01:24:05 +0000112_accept() -- accept connection, returning new socket fd and client address\n\
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000113bind(addr) -- bind the socket to a local address\n\
114close() -- close the socket\n\
115connect(addr) -- connect the socket to a remote address\n\
116connect_ex(addr) -- connect, return an error code instead of an exception\n\
Guido van Rossum39eb8fa2007-11-16 01:24:05 +0000117_dup() -- return a new socket fd duplicated from fileno()\n\
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000118fileno() -- return underlying file descriptor\n\
119getpeername() -- return remote address [*]\n\
120getsockname() -- return local address\n\
121getsockopt(level, optname[, buflen]) -- get socket options\n\
122gettimeout() -- return timeout or None\n\
123listen(n) -- start listening for incoming connections\n\
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000124recv(buflen[, flags]) -- receive data\n\
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000125recv_into(buffer[, nbytes[, flags]]) -- receive data (into a buffer)\n\
Thomas Wouters477c8d52006-05-27 19:21:47 +0000126recvfrom(buflen[, flags]) -- receive data and sender\'s address\n\
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000127recvfrom_into(buffer[, nbytes, [, flags])\n\
Thomas Wouters477c8d52006-05-27 19:21:47 +0000128 -- receive data and sender\'s address (into a buffer)\n\
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000129sendall(data[, flags]) -- send all data\n\
130send(data[, flags]) -- send data, may not send all of it\n\
131sendto(data[, flags], addr) -- send data to a given address\n\
132setblocking(0 | 1) -- set or clear the blocking I/O flag\n\
133setsockopt(level, optname, value) -- set socket options\n\
134settimeout(None | float) -- set or clear the timeout\n\
135shutdown(how) -- shut down traffic in one or both directions\n\
136\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000137 [*] not available on all platforms!");
Guido van Rossum3baaa131999-03-22 21:44:51 +0000138
Walter Dörwaldf0dfc7a2003-10-20 14:01:56 +0000139/* XXX This is a terrible mess of platform-dependent preprocessor hacks.
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000140 I hope some day someone can clean this up please... */
141
Guido van Rossum9376b741999-09-15 22:01:40 +0000142/* Hacks for gethostbyname_r(). On some non-Linux platforms, the configure
143 script doesn't get this right, so we hardcode some platform checks below.
144 On the other hand, not all Linux versions agree, so there the settings
145 computed by the configure script are needed! */
146
147#ifndef linux
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000148# undef HAVE_GETHOSTBYNAME_R_3_ARG
149# undef HAVE_GETHOSTBYNAME_R_5_ARG
150# undef HAVE_GETHOSTBYNAME_R_6_ARG
Guido van Rossum9376b741999-09-15 22:01:40 +0000151#endif
Guido van Rossume7de2061999-03-24 17:24:33 +0000152
Guido van Rossum7a122991999-04-13 04:07:32 +0000153#ifndef WITH_THREAD
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000154# undef HAVE_GETHOSTBYNAME_R
Guido van Rossum7a122991999-04-13 04:07:32 +0000155#endif
156
Guido van Rossume7de2061999-03-24 17:24:33 +0000157#ifdef HAVE_GETHOSTBYNAME_R
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000158# if defined(_AIX) || defined(__osf__)
159# define HAVE_GETHOSTBYNAME_R_3_ARG
160# elif defined(__sun) || defined(__sgi)
161# define HAVE_GETHOSTBYNAME_R_5_ARG
162# elif defined(linux)
Guido van Rossum9376b741999-09-15 22:01:40 +0000163/* Rely on the configure script */
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000164# else
165# undef HAVE_GETHOSTBYNAME_R
166# endif
Guido van Rossume7de2061999-03-24 17:24:33 +0000167#endif
168
Guido van Rossum3eede5a2002-06-07 02:08:35 +0000169#if !defined(HAVE_GETHOSTBYNAME_R) && defined(WITH_THREAD) && \
170 !defined(MS_WINDOWS)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000171# define USE_GETHOSTBYNAME_LOCK
Guido van Rossum3baaa131999-03-22 21:44:51 +0000172#endif
173
Hye-Shik Chang9ceebd52005-09-24 14:58:47 +0000174/* To use __FreeBSD_version */
175#ifdef HAVE_SYS_PARAM_H
176#include <sys/param.h>
177#endif
Just van Rossum1040d2c2003-05-09 07:53:18 +0000178/* On systems on which getaddrinfo() is believed to not be thread-safe,
Just van Rossum09aecd72003-05-09 08:03:44 +0000179 (this includes the getaddrinfo emulation) protect access with a lock. */
Hye-Shik Chang9ceebd52005-09-24 14:58:47 +0000180#if defined(WITH_THREAD) && (defined(__APPLE__) || \
181 (defined(__FreeBSD__) && __FreeBSD_version+0 < 503000) || \
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000182 defined(__OpenBSD__) || defined(__NetBSD__) || \
183 defined(__VMS) || !defined(HAVE_GETADDRINFO))
Just van Rossum1040d2c2003-05-09 07:53:18 +0000184#define USE_GETADDRINFO_LOCK
185#endif
186
187#ifdef USE_GETADDRINFO_LOCK
188#define ACQUIRE_GETADDRINFO_LOCK PyThread_acquire_lock(netdb_lock, 1);
189#define RELEASE_GETADDRINFO_LOCK PyThread_release_lock(netdb_lock);
190#else
191#define ACQUIRE_GETADDRINFO_LOCK
192#define RELEASE_GETADDRINFO_LOCK
193#endif
194
195#if defined(USE_GETHOSTBYNAME_LOCK) || defined(USE_GETADDRINFO_LOCK)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000196# include "pythread.h"
Guido van Rossum4f199ea1998-04-09 20:56:35 +0000197#endif
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000198
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000199#if defined(PYCC_VACPP)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000200# include <types.h>
201# include <io.h>
202# include <sys/ioctl.h>
203# include <utils.h>
204# include <ctype.h>
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000205#endif
206
Martin v. Löwis9e437302002-12-06 12:57:26 +0000207#if defined(__VMS)
Martin v. Löwis79acb9e2002-12-06 12:48:53 +0000208# include <ioctl.h>
209#endif
210
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000211#if defined(PYOS_OS2)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000212# define INCL_DOS
213# define INCL_DOSERRORS
214# define INCL_NOPMAPI
215# include <os2.h>
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000216#endif
217
Jeremy Hyltonfb509a32003-07-17 16:58:48 +0000218#if defined(__sgi) && _COMPILER_VERSION>700 && !_SGIAPI
Martin v. Löwis8eb92a02002-09-19 08:03:21 +0000219/* make sure that the reentrant (gethostbyaddr_r etc)
220 functions are declared correctly if compiling with
221 MIPSPro 7.x in ANSI C mode (default) */
Jeremy Hyltonfb509a32003-07-17 16:58:48 +0000222
Thomas Wouters477c8d52006-05-27 19:21:47 +0000223/* XXX Using _SGIAPI is the wrong thing,
Jeremy Hyltonfb509a32003-07-17 16:58:48 +0000224 but I don't know what the right thing is. */
Trent Mick8ea5bdf2004-09-13 17:48:41 +0000225#undef _SGIAPI /* to avoid warning */
Martin v. Löwis8eb92a02002-09-19 08:03:21 +0000226#define _SGIAPI 1
Jeremy Hyltonfb509a32003-07-17 16:58:48 +0000227
Trent Mick8ea5bdf2004-09-13 17:48:41 +0000228#undef _XOPEN_SOURCE
229#include <sys/socket.h>
230#include <sys/types.h>
231#include <netinet/in.h>
232#ifdef _SS_ALIGNSIZE
233#define HAVE_GETADDRINFO 1
234#define HAVE_GETNAMEINFO 1
235#endif
236
Jeremy Hyltonfb509a32003-07-17 16:58:48 +0000237#define HAVE_INET_PTON
238#include <netdb.h>
Martin v. Löwis8eb92a02002-09-19 08:03:21 +0000239#endif
240
Thomas Wouters477c8d52006-05-27 19:21:47 +0000241/* Irix 6.5 fails to define this variable at all. This is needed
242 for both GCC and SGI's compiler. I'd say that the SGI headers
Georg Brandldbd83392006-02-20 09:42:33 +0000243 are just busted. Same thing for Solaris. */
244#if (defined(__sgi) || defined(sun)) && !defined(INET_ADDRSTRLEN)
Anthony Baxterbab23cf2003-10-04 08:00:49 +0000245#define INET_ADDRSTRLEN 16
246#endif
247
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000248/* Generic includes */
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000249#ifdef HAVE_SYS_TYPES_H
Guido van Rossumb6775db1994-08-01 11:34:53 +0000250#include <sys/types.h>
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000251#endif
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000252
Marc-André Lemburg976ade62002-02-16 18:47:07 +0000253/* Generic socket object definitions and includes */
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000254#define PySocket_BUILDING_SOCKET
Marc-André Lemburgbb8b78b2002-02-16 18:44:52 +0000255#include "socketmodule.h"
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000256
257/* Addressing includes */
258
Guido van Rossum6f489d91996-06-28 20:15:15 +0000259#ifndef MS_WINDOWS
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000260
261/* Non-MS WINDOWS includes */
262# include <netdb.h>
Guido van Rossum5c9eb211999-08-20 18:21:51 +0000263
Guido van Rossum9376b741999-09-15 22:01:40 +0000264/* Headers needed for inet_ntoa() and inet_addr() */
Skip Montanaroeb33e5a2007-08-17 12:57:41 +0000265# if defined(PYOS_OS2) && defined(PYCC_VACPP)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000266# include <netdb.h>
Tim Peters603c6832001-11-05 02:45:59 +0000267typedef size_t socklen_t;
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000268# else
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000269# include <arpa/inet.h>
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000270# endif
Guido van Rossum5c9eb211999-08-20 18:21:51 +0000271
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000272# include <fcntl.h>
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000273
Guido van Rossume2ae77b2001-10-24 20:42:55 +0000274#else
Guido van Rossum48a680c2001-03-02 06:34:14 +0000275
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000276/* MS_WINDOWS includes */
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000277# ifdef HAVE_FCNTL_H
278# include <fcntl.h>
279# endif
Guido van Rossum48a680c2001-03-02 06:34:14 +0000280
Jeremy Hylton22308652001-02-02 03:23:09 +0000281#endif
282
Skip Montanaro7befb992004-02-10 16:50:21 +0000283#include <stddef.h>
Martin v. Löwisa45ecae2001-06-24 21:28:42 +0000284
285#ifndef offsetof
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000286# define offsetof(type, member) ((size_t)(&((type *)0)->member))
Martin v. Löwisa45ecae2001-06-24 21:28:42 +0000287#endif
288
Neal Norwitz39d22e52002-11-02 19:55:21 +0000289#ifndef O_NONBLOCK
290# define O_NONBLOCK O_NDELAY
Guido van Rossum3bbc62e1995-01-02 19:30:30 +0000291#endif
292
Trent Micka708d6e2004-09-07 17:48:26 +0000293/* include Python's addrinfo.h unless it causes trouble */
294#if defined(__sgi) && _COMPILER_VERSION>700 && defined(_SS_ALIGNSIZE)
295 /* Do not include addinfo.h on some newer IRIX versions.
296 * _SS_ALIGNSIZE is defined in sys/socket.h by 6.5.21,
297 * for example, but not by 6.5.10.
298 */
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000299#elif defined(_MSC_VER) && _MSC_VER>1201
Trent Micka708d6e2004-09-07 17:48:26 +0000300 /* Do not include addrinfo.h for MSVC7 or greater. 'addrinfo' and
301 * EAI_* constants are defined in (the already included) ws2tcpip.h.
302 */
303#else
304# include "addrinfo.h"
305#endif
Jason Tishlerc246cb72004-08-09 13:25:59 +0000306
Martin v. Löwisb9ab1592001-06-24 21:18:26 +0000307#ifndef HAVE_INET_PTON
Christian Heimes96e7b3d2007-11-20 06:51:17 +0000308#if !defined(NTDDI_VERSION) || (NTDDI_VERSION < NTDDI_LONGHORN)
Guido van Rossum3eede5a2002-06-07 02:08:35 +0000309int inet_pton(int af, const char *src, void *dst);
Martin v. Löwisc925b1532001-07-21 09:42:15 +0000310const char *inet_ntop(int af, const void *src, char *dst, socklen_t size);
Martin v. Löwisb9ab1592001-06-24 21:18:26 +0000311#endif
Christian Heimesb6150692007-11-15 23:37:07 +0000312#endif
Martin v. Löwisb9ab1592001-06-24 21:18:26 +0000313
Martin v. Löwisae26dc22001-08-03 10:02:29 +0000314#ifdef __APPLE__
315/* On OS X, getaddrinfo returns no error indication of lookup
316 failure, so we must use the emulation instead of the libinfo
317 implementation. Unfortunately, performing an autoconf test
318 for this bug would require DNS access for the machine performing
319 the configuration, which is not acceptable. Therefore, we
320 determine the bug just by checking for __APPLE__. If this bug
321 gets ever fixed, perhaps checking for sys/version.h would be
322 appropriate, which is 10/0 on the system with the bug. */
Jack Jansen84262fb2002-07-02 14:40:42 +0000323#ifndef HAVE_GETNAMEINFO
324/* This bug seems to be fixed in Jaguar. Ths easiest way I could
325 Find to check for Jaguar is that it has getnameinfo(), which
326 older releases don't have */
Martin v. Löwisae26dc22001-08-03 10:02:29 +0000327#undef HAVE_GETADDRINFO
Martin v. Löwisae26dc22001-08-03 10:02:29 +0000328#endif
Thomas Wouters477c8d52006-05-27 19:21:47 +0000329
330#ifdef HAVE_INET_ATON
331#define USE_INET_ATON_WEAKLINK
332#endif
333
Jack Jansen84262fb2002-07-02 14:40:42 +0000334#endif
Martin v. Löwisae26dc22001-08-03 10:02:29 +0000335
Martin v. Löwis01dfdb32001-06-23 16:30:13 +0000336/* I know this is a bad practice, but it is the easiest... */
Martin v. Löwisae26dc22001-08-03 10:02:29 +0000337#if !defined(HAVE_GETADDRINFO)
Martin v. Löwisfccac2e2003-05-01 05:20:46 +0000338/* avoid clashes with the C library definition of the symbol. */
339#define getaddrinfo fake_getaddrinfo
340#define gai_strerror fake_gai_strerror
341#define freeaddrinfo fake_freeaddrinfo
Martin v. Löwis01dfdb32001-06-23 16:30:13 +0000342#include "getaddrinfo.c"
343#endif
Martin v. Löwisae26dc22001-08-03 10:02:29 +0000344#if !defined(HAVE_GETNAMEINFO)
Martin v. Löwisfccac2e2003-05-01 05:20:46 +0000345#define getnameinfo fake_getnameinfo
Martin v. Löwis01dfdb32001-06-23 16:30:13 +0000346#include "getnameinfo.c"
347#endif
348
Guido van Rossum39eb8fa2007-11-16 01:24:05 +0000349#ifdef MS_WINDOWS
350/* On Windows a socket is really a handle not an fd */
351static SOCKET
352dup_socket(SOCKET handle)
353{
Daniel Stutzbach6c765282010-09-03 12:38:33 +0000354 WSAPROTOCOL_INFO info;
Guido van Rossum39eb8fa2007-11-16 01:24:05 +0000355
Daniel Stutzbach6c765282010-09-03 12:38:33 +0000356 if (WSADuplicateSocket(handle, GetCurrentProcessId(), &info))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000357 return INVALID_SOCKET;
Daniel Stutzbach6c765282010-09-03 12:38:33 +0000358
359 return WSASocket(FROM_PROTOCOL_INFO, FROM_PROTOCOL_INFO,
360 FROM_PROTOCOL_INFO, &info, 0, 0);
Guido van Rossum39eb8fa2007-11-16 01:24:05 +0000361}
Guido van Rossum2dd8ddd2000-04-21 20:33:00 +0000362#define SOCKETCLOSE closesocket
Guido van Rossum39eb8fa2007-11-16 01:24:05 +0000363#else
364/* On Unix we can use dup to duplicate the file descriptor of a socket*/
365#define dup_socket(fd) dup(fd)
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +0000366#endif
367
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000368#ifdef MS_WIN32
Guido van Rossum3eede5a2002-06-07 02:08:35 +0000369#define EAFNOSUPPORT WSAEAFNOSUPPORT
370#define snprintf _snprintf
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000371#endif
Fred Drakea04eaad2000-06-30 02:46:07 +0000372
Andrew MacIntyreba43e872002-03-03 03:03:52 +0000373#if defined(PYOS_OS2) && !defined(PYCC_GCC)
Guido van Rossum2dd8ddd2000-04-21 20:33:00 +0000374#define SOCKETCLOSE soclose
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000375#define NO_DUP /* Sockets are Not Actual File Handles under OS/2 */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000376#endif
377
Guido van Rossum2dd8ddd2000-04-21 20:33:00 +0000378#ifndef SOCKETCLOSE
379#define SOCKETCLOSE close
380#endif
381
Gregory P. Smith3e85dfd2010-10-17 00:43:10 +0000382#if (defined(HAVE_BLUETOOTH_H) || defined(HAVE_BLUETOOTH_BLUETOOTH_H)) && !defined(__NetBSD__) && !defined(__DragonFly__)
Hye-Shik Chang81268602004-02-02 06:05:24 +0000383#define USE_BLUETOOTH 1
384#if defined(__FreeBSD__)
385#define BTPROTO_L2CAP BLUETOOTH_PROTO_L2CAP
386#define BTPROTO_RFCOMM BLUETOOTH_PROTO_RFCOMM
Thomas Wouterscf297e42007-02-23 15:07:44 +0000387#define BTPROTO_HCI BLUETOOTH_PROTO_HCI
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000388#define SOL_HCI SOL_HCI_RAW
389#define HCI_FILTER SO_HCI_RAW_FILTER
Hye-Shik Chang81268602004-02-02 06:05:24 +0000390#define sockaddr_l2 sockaddr_l2cap
391#define sockaddr_rc sockaddr_rfcomm
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000392#define hci_dev hci_node
Hye-Shik Chang81268602004-02-02 06:05:24 +0000393#define _BT_L2_MEMB(sa, memb) ((sa)->l2cap_##memb)
394#define _BT_RC_MEMB(sa, memb) ((sa)->rfcomm_##memb)
Thomas Wouterscf297e42007-02-23 15:07:44 +0000395#define _BT_HCI_MEMB(sa, memb) ((sa)->hci_##memb)
Gregory P. Smith3e85dfd2010-10-17 00:43:10 +0000396#elif defined(__NetBSD__) || defined(__DragonFly__)
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000397#define sockaddr_l2 sockaddr_bt
398#define sockaddr_rc sockaddr_bt
Thomas Wouterscf297e42007-02-23 15:07:44 +0000399#define sockaddr_hci sockaddr_bt
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000400#define sockaddr_sco sockaddr_bt
Gregory P. Smith3e85dfd2010-10-17 00:43:10 +0000401#define SOL_HCI BTPROTO_HCI
402#define HCI_DATA_DIR SO_HCI_DIRECTION
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000403#define _BT_L2_MEMB(sa, memb) ((sa)->bt_##memb)
404#define _BT_RC_MEMB(sa, memb) ((sa)->bt_##memb)
Thomas Wouterscf297e42007-02-23 15:07:44 +0000405#define _BT_HCI_MEMB(sa, memb) ((sa)->bt_##memb)
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000406#define _BT_SCO_MEMB(sa, memb) ((sa)->bt_##memb)
Hye-Shik Chang81268602004-02-02 06:05:24 +0000407#else
Hye-Shik Chang81268602004-02-02 06:05:24 +0000408#define _BT_L2_MEMB(sa, memb) ((sa)->l2_##memb)
409#define _BT_RC_MEMB(sa, memb) ((sa)->rc_##memb)
Thomas Wouterscf297e42007-02-23 15:07:44 +0000410#define _BT_HCI_MEMB(sa, memb) ((sa)->hci_##memb)
Hye-Shik Chang81268602004-02-02 06:05:24 +0000411#define _BT_SCO_MEMB(sa, memb) ((sa)->sco_##memb)
412#endif
413#endif
414
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000415#ifdef __VMS
416/* TCP/IP Services for VMS uses a maximum send/recv buffer length */
417#define SEGMENT_SIZE (32 * 1024 -1)
418#endif
419
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000420#define SAS2SA(x) ((struct sockaddr *)(x))
Thomas Wouters89f507f2006-12-13 04:49:30 +0000421
Martin v. Löwise9416172003-05-03 10:12:45 +0000422/*
423 * Constants for getnameinfo()
424 */
425#if !defined(NI_MAXHOST)
426#define NI_MAXHOST 1025
427#endif
428#if !defined(NI_MAXSERV)
429#define NI_MAXSERV 32
430#endif
431
Guido van Rossum7d0a8262007-05-21 23:13:11 +0000432#ifndef INVALID_SOCKET /* MS defines this */
433#define INVALID_SOCKET (-1)
434#endif
435
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000436/* XXX There's a problem here: *static* functions are not supposed to have
437 a Py prefix (or use CapitalizedWords). Later... */
438
Guido van Rossum30a685f1991-06-27 15:51:29 +0000439/* Global variable holding the exception type for errors detected
440 by this module (but not argument type or memory errors, etc.). */
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000441static PyObject *socket_error;
442static PyObject *socket_herror;
443static PyObject *socket_gaierror;
Raymond Hettingeref7343c2003-06-29 03:08:05 +0000444static PyObject *socket_timeout;
Guido van Rossum30a685f1991-06-27 15:51:29 +0000445
Tim Peters643a7fc2002-02-17 04:13:21 +0000446/* A forward reference to the socket type object.
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000447 The sock_type variable contains pointers to various functions,
448 some of which call new_sockobject(), which uses sock_type, so
Tim Peters643a7fc2002-02-17 04:13:21 +0000449 there has to be a circular reference. */
Jeremy Hylton938ace62002-07-17 16:30:39 +0000450static PyTypeObject sock_type;
Guido van Rossum48a680c2001-03-02 06:34:14 +0000451
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000452#if defined(HAVE_POLL_H)
453#include <poll.h>
454#elif defined(HAVE_SYS_POLL_H)
455#include <sys/poll.h>
456#endif
457
Martin v. Löwisf84d1b92006-02-11 09:27:05 +0000458#ifdef Py_SOCKET_FD_CAN_BE_GE_FD_SETSIZE
459/* Platform can select file descriptors beyond FD_SETSIZE */
460#define IS_SELECTABLE(s) 1
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000461#elif defined(HAVE_POLL)
462/* Instead of select(), we'll use poll() since poll() works on any fd. */
463#define IS_SELECTABLE(s) 1
464/* Can we call select() with this socket without a buffer overrun? */
Martin v. Löwisf84d1b92006-02-11 09:27:05 +0000465#else
466/* POSIX says selecting file descriptors beyond FD_SETSIZE
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000467 has undefined behaviour. If there's no timeout left, we don't have to
468 call select, so it's a safe, little white lie. */
469#define IS_SELECTABLE(s) ((s)->sock_fd < FD_SETSIZE || s->sock_timeout <= 0.0)
Martin v. Löwisf84d1b92006-02-11 09:27:05 +0000470#endif
Neal Norwitz082b2df2006-02-07 07:04:46 +0000471
472static PyObject*
473select_error(void)
474{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000475 PyErr_SetString(socket_error, "unable to select on socket");
476 return NULL;
Neal Norwitz082b2df2006-02-07 07:04:46 +0000477}
478
Antoine Pitrou3e1fd272010-09-28 21:23:11 +0000479#ifdef MS_WINDOWS
Antoine Pitrouc8f01962010-09-28 22:03:27 +0000480#ifndef WSAEAGAIN
481#define WSAEAGAIN WSAEWOULDBLOCK
482#endif
Antoine Pitrou3e1fd272010-09-28 21:23:11 +0000483#define CHECK_ERRNO(expected) \
484 (WSAGetLastError() == WSA ## expected)
485#else
486#define CHECK_ERRNO(expected) \
487 (errno == expected)
488#endif
489
Guido van Rossum30a685f1991-06-27 15:51:29 +0000490/* Convenience function to raise an error according to errno
491 and return a NULL pointer from a function. */
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000492
Guido van Rossum73624e91994-10-10 17:59:00 +0000493static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000494set_error(void)
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000495{
Guido van Rossum8d665e61996-06-26 18:22:49 +0000496#ifdef MS_WINDOWS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000497 int err_no = WSAGetLastError();
498 /* PyErr_SetExcFromWindowsErr() invokes FormatMessage() which
499 recognizes the error codes used by both GetLastError() and
500 WSAGetLastError */
501 if (err_no)
502 return PyErr_SetExcFromWindowsErr(socket_error, err_no);
Guido van Rossum14ed0b21994-09-29 09:50:09 +0000503#endif
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000504
Andrew MacIntyreba43e872002-03-03 03:03:52 +0000505#if defined(PYOS_OS2) && !defined(PYCC_GCC)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000506 if (sock_errno() != NO_ERROR) {
507 APIRET rc;
508 ULONG msglen;
509 char outbuf[100];
510 int myerrorcode = sock_errno();
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000511
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000512 /* Retrieve socket-related error message from MPTN.MSG file */
513 rc = DosGetMessage(NULL, 0, outbuf, sizeof(outbuf),
514 myerrorcode - SOCBASEERR + 26,
515 "mptn.msg",
516 &msglen);
517 if (rc == NO_ERROR) {
518 PyObject *v;
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000519
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000520 /* OS/2 doesn't guarantee a terminator */
521 outbuf[msglen] = '\0';
522 if (strlen(outbuf) > 0) {
523 /* If non-empty msg, trim CRLF */
524 char *lastc = &outbuf[ strlen(outbuf)-1 ];
525 while (lastc > outbuf &&
526 isspace(Py_CHARMASK(*lastc))) {
527 /* Trim trailing whitespace (CRLF) */
528 *lastc-- = '\0';
529 }
530 }
531 v = Py_BuildValue("(is)", myerrorcode, outbuf);
532 if (v != NULL) {
533 PyErr_SetObject(socket_error, v);
534 Py_DECREF(v);
535 }
536 return NULL;
537 }
538 }
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000539#endif
540
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000541 return PyErr_SetFromErrno(socket_error);
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000542}
543
Guido van Rossum30a685f1991-06-27 15:51:29 +0000544
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000545static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000546set_herror(int h_error)
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000547{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000548 PyObject *v;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000549
550#ifdef HAVE_HSTRERROR
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000551 v = Py_BuildValue("(is)", h_error, (char *)hstrerror(h_error));
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000552#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000553 v = Py_BuildValue("(is)", h_error, "host not found");
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000554#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000555 if (v != NULL) {
556 PyErr_SetObject(socket_herror, v);
557 Py_DECREF(v);
558 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000559
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000560 return NULL;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000561}
562
563
564static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000565set_gaierror(int error)
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000566{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000567 PyObject *v;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000568
Martin v. Löwis272cb402002-03-01 08:31:07 +0000569#ifdef EAI_SYSTEM
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000570 /* EAI_SYSTEM is not available on Windows XP. */
571 if (error == EAI_SYSTEM)
572 return set_error();
Martin v. Löwis272cb402002-03-01 08:31:07 +0000573#endif
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000574
Martin v. Löwisf95dd0a2001-08-15 17:14:33 +0000575#ifdef HAVE_GAI_STRERROR
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000576 v = Py_BuildValue("(is)", error, gai_strerror(error));
Martin v. Löwisf95dd0a2001-08-15 17:14:33 +0000577#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000578 v = Py_BuildValue("(is)", error, "getaddrinfo failed");
Martin v. Löwisf95dd0a2001-08-15 17:14:33 +0000579#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000580 if (v != NULL) {
581 PyErr_SetObject(socket_gaierror, v);
582 Py_DECREF(v);
583 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000584
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000585 return NULL;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000586}
587
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000588#ifdef __VMS
589/* Function to send in segments */
590static int
591sendsegmented(int sock_fd, char *buf, int len, int flags)
592{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000593 int n = 0;
594 int remaining = len;
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000595
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000596 while (remaining > 0) {
597 unsigned int segment;
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000598
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000599 segment = (remaining >= SEGMENT_SIZE ? SEGMENT_SIZE : remaining);
600 n = send(sock_fd, buf, segment, flags);
601 if (n < 0) {
602 return n;
603 }
604 remaining -= segment;
605 buf += segment;
606 } /* end while */
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000607
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000608 return len;
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000609}
610#endif
611
Guido van Rossum3eede5a2002-06-07 02:08:35 +0000612/* Function to perform the setting of socket blocking mode
613 internally. block = (1 | 0). */
Guido van Rossum67f7a382002-06-06 21:08:16 +0000614static int
615internal_setblocking(PySocketSockObject *s, int block)
616{
Guido van Rossum67f7a382002-06-06 21:08:16 +0000617#ifndef MS_WINDOWS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000618 int delay_flag;
Guido van Rossum67f7a382002-06-06 21:08:16 +0000619#endif
Antoine Pitroub1c54962010-10-14 15:05:38 +0000620#ifdef SOCK_NONBLOCK
621 if (block)
622 s->sock_type &= (~SOCK_NONBLOCK);
623 else
624 s->sock_type |= SOCK_NONBLOCK;
625#endif
Guido van Rossum67f7a382002-06-06 21:08:16 +0000626
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000627 Py_BEGIN_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +0000628#ifndef MS_WINDOWS
629#if defined(PYOS_OS2) && !defined(PYCC_GCC)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000630 block = !block;
631 ioctl(s->sock_fd, FIONBIO, (caddr_t)&block, sizeof(block));
Martin v. Löwis79acb9e2002-12-06 12:48:53 +0000632#elif defined(__VMS)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000633 block = !block;
634 ioctl(s->sock_fd, FIONBIO, (unsigned int *)&block);
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000635#else /* !PYOS_OS2 && !__VMS */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000636 delay_flag = fcntl(s->sock_fd, F_GETFL, 0);
637 if (block)
638 delay_flag &= (~O_NONBLOCK);
639 else
640 delay_flag |= O_NONBLOCK;
641 fcntl(s->sock_fd, F_SETFL, delay_flag);
Guido van Rossum67f7a382002-06-06 21:08:16 +0000642#endif /* !PYOS_OS2 */
643#else /* MS_WINDOWS */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000644 block = !block;
645 ioctlsocket(s->sock_fd, FIONBIO, (u_long*)&block);
Guido van Rossum67f7a382002-06-06 21:08:16 +0000646#endif /* MS_WINDOWS */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000647 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +0000648
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000649 /* Since these don't return anything */
650 return 1;
Guido van Rossum67f7a382002-06-06 21:08:16 +0000651}
652
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000653/* Do a select()/poll() on the socket, if necessary (sock_timeout > 0).
Guido van Rossum11ba0942002-06-13 15:07:44 +0000654 The argument writing indicates the direction.
Raymond Hettingeref7343c2003-06-29 03:08:05 +0000655 This does not raise an exception; we'll let our caller do that
656 after they've reacquired the interpreter lock.
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000657 Returns 1 on timeout, -1 on error, 0 otherwise. */
Raymond Hettingeref7343c2003-06-29 03:08:05 +0000658static int
Antoine Pitrou3e1fd272010-09-28 21:23:11 +0000659internal_select_ex(PySocketSockObject *s, int writing, double interval)
Guido van Rossum67f7a382002-06-06 21:08:16 +0000660{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000661 int n;
Guido van Rossum11ba0942002-06-13 15:07:44 +0000662
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000663 /* Nothing to do unless we're in timeout mode (not non-blocking) */
664 if (s->sock_timeout <= 0.0)
665 return 0;
Guido van Rossum67f7a382002-06-06 21:08:16 +0000666
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000667 /* Guard against closed socket */
668 if (s->sock_fd < 0)
669 return 0;
Guido van Rossumad654902002-07-19 12:44:59 +0000670
Antoine Pitrou3e1fd272010-09-28 21:23:11 +0000671 /* Handling this condition here simplifies the select loops */
672 if (interval < 0.0)
673 return 1;
674
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000675 /* Prefer poll, if available, since you can poll() any fd
676 * which can't be done with select(). */
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000677#ifdef HAVE_POLL
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000678 {
679 struct pollfd pollfd;
680 int timeout;
Guido van Rossum67f7a382002-06-06 21:08:16 +0000681
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000682 pollfd.fd = s->sock_fd;
683 pollfd.events = writing ? POLLOUT : POLLIN;
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000684
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000685 /* s->sock_timeout is in seconds, timeout in ms */
Antoine Pitrou3e1fd272010-09-28 21:23:11 +0000686 timeout = (int)(interval * 1000 + 0.5);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000687 n = poll(&pollfd, 1, timeout);
688 }
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000689#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000690 {
691 /* Construct the arguments to select */
692 fd_set fds;
693 struct timeval tv;
Antoine Pitrou3e1fd272010-09-28 21:23:11 +0000694 tv.tv_sec = (int)interval;
695 tv.tv_usec = (int)((interval - tv.tv_sec) * 1e6);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000696 FD_ZERO(&fds);
697 FD_SET(s->sock_fd, &fds);
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000698
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000699 /* See if the socket is ready */
700 if (writing)
Antoine Pitrou19467d22010-08-17 19:33:30 +0000701 n = select(Py_SAFE_DOWNCAST(s->sock_fd+1, SOCKET_T, int),
702 NULL, &fds, NULL, &tv);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000703 else
Antoine Pitrou19467d22010-08-17 19:33:30 +0000704 n = select(Py_SAFE_DOWNCAST(s->sock_fd+1, SOCKET_T, int),
705 &fds, NULL, NULL, &tv);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000706 }
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000707#endif
Guido van Rossum39eb8fa2007-11-16 01:24:05 +0000708
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000709 if (n < 0)
710 return -1;
711 if (n == 0)
712 return 1;
713 return 0;
Guido van Rossum67f7a382002-06-06 21:08:16 +0000714}
715
Antoine Pitrou3e1fd272010-09-28 21:23:11 +0000716static int
717internal_select(PySocketSockObject *s, int writing)
718{
719 return internal_select_ex(s, writing, s->sock_timeout);
720}
721
722/*
723 Two macros for automatic retry of select() in case of false positives
724 (for example, select() could indicate a socket is ready for reading
725 but the data then discarded by the OS because of a wrong checksum).
726 Here is an example of use:
727
728 BEGIN_SELECT_LOOP(s)
729 Py_BEGIN_ALLOW_THREADS
730 timeout = internal_select_ex(s, 0, interval);
731 if (!timeout)
732 outlen = recv(s->sock_fd, cbuf, len, flags);
733 Py_END_ALLOW_THREADS
734 if (timeout == 1) {
735 PyErr_SetString(socket_timeout, "timed out");
736 return -1;
737 }
738 END_SELECT_LOOP(s)
739*/
740
741#define BEGIN_SELECT_LOOP(s) \
742 { \
743 _PyTime_timeval now, deadline = {0, 0}; \
744 double interval = s->sock_timeout; \
745 int has_timeout = s->sock_timeout > 0.0; \
746 if (has_timeout) { \
747 _PyTime_gettimeofday(&now); \
748 deadline = now; \
749 _PyTime_ADD_SECONDS(deadline, s->sock_timeout); \
750 } \
751 while (1) { \
752 errno = 0; \
753
754#define END_SELECT_LOOP(s) \
755 if (!has_timeout || \
756 (!CHECK_ERRNO(EWOULDBLOCK) && !CHECK_ERRNO(EAGAIN))) \
757 break; \
758 _PyTime_gettimeofday(&now); \
759 interval = _PyTime_INTERVAL(now, deadline); \
760 } \
761 } \
762
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000763/* Initialize a new socket object. */
764
Tim Petersa12b4cf2002-07-18 22:38:44 +0000765static double defaulttimeout = -1.0; /* Default timeout for new sockets */
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +0000766
Martin v. Löwis1a214512008-06-11 05:26:20 +0000767static void
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000768init_sockobject(PySocketSockObject *s,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000769 SOCKET_T fd, int family, int type, int proto)
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000770{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000771 s->sock_fd = fd;
772 s->sock_family = family;
773 s->sock_type = type;
774 s->sock_proto = proto;
Guido van Rossum67f7a382002-06-06 21:08:16 +0000775
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000776 s->errorhandler = &set_error;
Antoine Pitroub1c54962010-10-14 15:05:38 +0000777#ifdef SOCK_NONBLOCK
778 if (type & SOCK_NONBLOCK)
779 s->sock_timeout = 0.0;
780 else
781#endif
782 {
783 s->sock_timeout = defaulttimeout;
784 if (defaulttimeout >= 0.0)
785 internal_setblocking(s, 0);
786 }
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +0000787
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000788}
789
790
Guido van Rossum30a685f1991-06-27 15:51:29 +0000791/* Create a new socket object.
792 This just creates the object and initializes it.
793 If the creation fails, return NULL and set an exception (implicit
794 in NEWOBJ()). */
795
Guido van Rossum73624e91994-10-10 17:59:00 +0000796static PySocketSockObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000797new_sockobject(SOCKET_T fd, int family, int type, int proto)
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000798{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000799 PySocketSockObject *s;
800 s = (PySocketSockObject *)
801 PyType_GenericNew(&sock_type, NULL, NULL);
802 if (s != NULL)
803 init_sockobject(s, fd, family, type, proto);
804 return s;
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000805}
806
Guido van Rossum30a685f1991-06-27 15:51:29 +0000807
Guido van Rossum48a680c2001-03-02 06:34:14 +0000808/* Lock to allow python interpreter to continue, but only allow one
Just van Rossum1040d2c2003-05-09 07:53:18 +0000809 thread to be in gethostbyname or getaddrinfo */
810#if defined(USE_GETHOSTBYNAME_LOCK) || defined(USE_GETADDRINFO_LOCK)
811PyThread_type_lock netdb_lock;
Guido van Rossum4f199ea1998-04-09 20:56:35 +0000812#endif
813
814
Guido van Rossum30a685f1991-06-27 15:51:29 +0000815/* Convert a string specifying a host name or one of a few symbolic
816 names to a numeric IP address. This usually calls gethostbyname()
817 to do the work; the names "" and "<broadcast>" are special.
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000818 Return the length (IPv4 should be 4 bytes), or negative if
Guido van Rossum30a685f1991-06-27 15:51:29 +0000819 an error occurred; then an exception is raised. */
820
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000821static int
Martin v. Löwisddc6f472002-07-28 16:10:31 +0000822setipaddr(char *name, struct sockaddr *addr_ret, size_t addr_ret_size, int af)
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000823{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000824 struct addrinfo hints, *res;
825 int error;
826 int d1, d2, d3, d4;
827 char ch;
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000828
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000829 memset((void *) addr_ret, '\0', sizeof(*addr_ret));
830 if (name[0] == '\0') {
831 int siz;
832 memset(&hints, 0, sizeof(hints));
833 hints.ai_family = af;
834 hints.ai_socktype = SOCK_DGRAM; /*dummy*/
835 hints.ai_flags = AI_PASSIVE;
836 Py_BEGIN_ALLOW_THREADS
837 ACQUIRE_GETADDRINFO_LOCK
838 error = getaddrinfo(NULL, "0", &hints, &res);
839 Py_END_ALLOW_THREADS
840 /* We assume that those thread-unsafe getaddrinfo() versions
841 *are* safe regarding their return value, ie. that a
842 subsequent call to getaddrinfo() does not destroy the
843 outcome of the first call. */
844 RELEASE_GETADDRINFO_LOCK
845 if (error) {
846 set_gaierror(error);
847 return -1;
848 }
849 switch (res->ai_family) {
850 case AF_INET:
851 siz = 4;
852 break;
Martin v. Löwis44ddbde2001-12-02 10:15:37 +0000853#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000854 case AF_INET6:
855 siz = 16;
856 break;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000857#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000858 default:
859 freeaddrinfo(res);
860 PyErr_SetString(socket_error,
861 "unsupported address family");
862 return -1;
863 }
864 if (res->ai_next) {
865 freeaddrinfo(res);
866 PyErr_SetString(socket_error,
867 "wildcard resolved to multiple address");
868 return -1;
869 }
870 if (res->ai_addrlen < addr_ret_size)
871 addr_ret_size = res->ai_addrlen;
872 memcpy(addr_ret, res->ai_addr, addr_ret_size);
873 freeaddrinfo(res);
874 return siz;
875 }
876 if (name[0] == '<' && strcmp(name, "<broadcast>") == 0) {
877 struct sockaddr_in *sin;
878 if (af != AF_INET && af != AF_UNSPEC) {
879 PyErr_SetString(socket_error,
880 "address family mismatched");
881 return -1;
882 }
883 sin = (struct sockaddr_in *)addr_ret;
884 memset((void *) sin, '\0', sizeof(*sin));
885 sin->sin_family = AF_INET;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000886#ifdef HAVE_SOCKADDR_SA_LEN
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000887 sin->sin_len = sizeof(*sin);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000888#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000889 sin->sin_addr.s_addr = INADDR_BROADCAST;
890 return sizeof(sin->sin_addr);
891 }
892 if (sscanf(name, "%d.%d.%d.%d%c", &d1, &d2, &d3, &d4, &ch) == 4 &&
893 0 <= d1 && d1 <= 255 && 0 <= d2 && d2 <= 255 &&
894 0 <= d3 && d3 <= 255 && 0 <= d4 && d4 <= 255) {
895 struct sockaddr_in *sin;
896 sin = (struct sockaddr_in *)addr_ret;
897 sin->sin_addr.s_addr = htonl(
898 ((long) d1 << 24) | ((long) d2 << 16) |
899 ((long) d3 << 8) | ((long) d4 << 0));
900 sin->sin_family = AF_INET;
Anthony Baxter0e85f9d2003-05-02 15:40:46 +0000901#ifdef HAVE_SOCKADDR_SA_LEN
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000902 sin->sin_len = sizeof(*sin);
Anthony Baxter0e85f9d2003-05-02 15:40:46 +0000903#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000904 return 4;
905 }
906 memset(&hints, 0, sizeof(hints));
907 hints.ai_family = af;
908 Py_BEGIN_ALLOW_THREADS
909 ACQUIRE_GETADDRINFO_LOCK
910 error = getaddrinfo(name, NULL, &hints, &res);
Martin v. Löwis7c4b5fa2001-10-25 09:04:03 +0000911#if defined(__digital__) && defined(__unix__)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000912 if (error == EAI_NONAME && af == AF_UNSPEC) {
913 /* On Tru64 V5.1, numeric-to-addr conversion fails
914 if no address family is given. Assume IPv4 for now.*/
915 hints.ai_family = AF_INET;
916 error = getaddrinfo(name, NULL, &hints, &res);
917 }
Martin v. Löwis7c4b5fa2001-10-25 09:04:03 +0000918#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000919 Py_END_ALLOW_THREADS
920 RELEASE_GETADDRINFO_LOCK /* see comment in setipaddr() */
921 if (error) {
922 set_gaierror(error);
923 return -1;
924 }
925 if (res->ai_addrlen < addr_ret_size)
926 addr_ret_size = res->ai_addrlen;
927 memcpy((char *) addr_ret, res->ai_addr, addr_ret_size);
928 freeaddrinfo(res);
929 switch (addr_ret->sa_family) {
930 case AF_INET:
931 return 4;
Martin v. Löwis44ddbde2001-12-02 10:15:37 +0000932#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000933 case AF_INET6:
934 return 16;
Guido van Rossum955becc1999-03-22 20:14:53 +0000935#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000936 default:
937 PyErr_SetString(socket_error, "unknown address family");
938 return -1;
939 }
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000940}
941
Guido van Rossum30a685f1991-06-27 15:51:29 +0000942
Guido van Rossum30a685f1991-06-27 15:51:29 +0000943/* Create a string object representing an IP address.
944 This is always a string of the form 'dd.dd.dd.dd' (with variable
945 size numbers). */
946
Guido van Rossum73624e91994-10-10 17:59:00 +0000947static PyObject *
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000948makeipaddr(struct sockaddr *addr, int addrlen)
Guido van Rossum30a685f1991-06-27 15:51:29 +0000949{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000950 char buf[NI_MAXHOST];
951 int error;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000952
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000953 error = getnameinfo(addr, addrlen, buf, sizeof(buf), NULL, 0,
954 NI_NUMERICHOST);
955 if (error) {
956 set_gaierror(error);
957 return NULL;
958 }
959 return PyUnicode_FromString(buf);
Guido van Rossum30a685f1991-06-27 15:51:29 +0000960}
961
962
Martin v. Löwis558d9bf2004-06-03 09:24:42 +0000963#ifdef USE_BLUETOOTH
964/* Convert a string representation of a Bluetooth address into a numeric
965 address. Returns the length (6), or raises an exception and returns -1 if
966 an error occurred. */
967
968static int
969setbdaddr(char *name, bdaddr_t *bdaddr)
970{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000971 unsigned int b0, b1, b2, b3, b4, b5;
972 char ch;
973 int n;
Martin v. Löwis558d9bf2004-06-03 09:24:42 +0000974
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000975 n = sscanf(name, "%X:%X:%X:%X:%X:%X%c",
976 &b5, &b4, &b3, &b2, &b1, &b0, &ch);
977 if (n == 6 && (b0 | b1 | b2 | b3 | b4 | b5) < 256) {
978 bdaddr->b[0] = b0;
979 bdaddr->b[1] = b1;
980 bdaddr->b[2] = b2;
981 bdaddr->b[3] = b3;
982 bdaddr->b[4] = b4;
983 bdaddr->b[5] = b5;
984 return 6;
985 } else {
986 PyErr_SetString(socket_error, "bad bluetooth address");
987 return -1;
988 }
Martin v. Löwis558d9bf2004-06-03 09:24:42 +0000989}
990
991/* Create a string representation of the Bluetooth address. This is always a
992 string of the form 'XX:XX:XX:XX:XX:XX' where XX is a two digit hexadecimal
993 value (zero padded if necessary). */
994
995static PyObject *
996makebdaddr(bdaddr_t *bdaddr)
997{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000998 char buf[(6 * 2) + 5 + 1];
Martin v. Löwis558d9bf2004-06-03 09:24:42 +0000999
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001000 sprintf(buf, "%02X:%02X:%02X:%02X:%02X:%02X",
1001 bdaddr->b[5], bdaddr->b[4], bdaddr->b[3],
1002 bdaddr->b[2], bdaddr->b[1], bdaddr->b[0]);
1003 return PyUnicode_FromString(buf);
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001004}
1005#endif
1006
1007
Guido van Rossum30a685f1991-06-27 15:51:29 +00001008/* Create an object representing the given socket address,
1009 suitable for passing it back to bind(), connect() etc.
1010 The family field of the sockaddr structure is inspected
1011 to determine what kind of address it really is. */
1012
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001013/*ARGSUSED*/
Guido van Rossum73624e91994-10-10 17:59:00 +00001014static PyObject *
Antoine Pitrou19467d22010-08-17 19:33:30 +00001015makesockaddr(SOCKET_T sockfd, struct sockaddr *addr, size_t addrlen, int proto)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001016{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001017 if (addrlen == 0) {
1018 /* No address -- may be recvfrom() from known socket */
1019 Py_INCREF(Py_None);
1020 return Py_None;
1021 }
Guido van Rossum25bec8c1992-08-05 19:00:45 +00001022
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001023 switch (addr->sa_family) {
Guido van Rossum30a685f1991-06-27 15:51:29 +00001024
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001025 case AF_INET:
1026 {
1027 struct sockaddr_in *a;
1028 PyObject *addrobj = makeipaddr(addr, sizeof(*a));
1029 PyObject *ret = NULL;
1030 if (addrobj) {
1031 a = (struct sockaddr_in *)addr;
1032 ret = Py_BuildValue("Oi", addrobj, ntohs(a->sin_port));
1033 Py_DECREF(addrobj);
1034 }
1035 return ret;
1036 }
Guido van Rossum30a685f1991-06-27 15:51:29 +00001037
Andrew MacIntyred12dfbb2004-04-04 07:13:49 +00001038#if defined(AF_UNIX)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001039 case AF_UNIX:
1040 {
1041 struct sockaddr_un *a = (struct sockaddr_un *) addr;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001042#ifdef linux
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001043 if (a->sun_path[0] == 0) { /* Linux abstract namespace */
1044 addrlen -= offsetof(struct sockaddr_un, sun_path);
1045 return PyBytes_FromStringAndSize(a->sun_path, addrlen);
1046 }
1047 else
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001048#endif /* linux */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001049 {
1050 /* regular NULL-terminated string */
1051 return PyUnicode_FromString(a->sun_path);
1052 }
1053 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00001054#endif /* AF_UNIX */
1055
Martin v. Löwis11017b12006-01-14 18:12:57 +00001056#if defined(AF_NETLINK)
1057 case AF_NETLINK:
1058 {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001059 struct sockaddr_nl *a = (struct sockaddr_nl *) addr;
1060 return Py_BuildValue("II", a->nl_pid, a->nl_groups);
Martin v. Löwis11017b12006-01-14 18:12:57 +00001061 }
1062#endif /* AF_NETLINK */
1063
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00001064#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001065 case AF_INET6:
1066 {
1067 struct sockaddr_in6 *a;
1068 PyObject *addrobj = makeipaddr(addr, sizeof(*a));
1069 PyObject *ret = NULL;
1070 if (addrobj) {
1071 a = (struct sockaddr_in6 *)addr;
1072 ret = Py_BuildValue("Oiii",
1073 addrobj,
1074 ntohs(a->sin6_port),
1075 a->sin6_flowinfo,
1076 a->sin6_scope_id);
1077 Py_DECREF(addrobj);
1078 }
1079 return ret;
1080 }
Jeremy Hylton22308652001-02-02 03:23:09 +00001081#endif
Guido van Rossum30a685f1991-06-27 15:51:29 +00001082
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001083#ifdef USE_BLUETOOTH
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001084 case AF_BLUETOOTH:
1085 switch (proto) {
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001086
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001087 case BTPROTO_L2CAP:
1088 {
1089 struct sockaddr_l2 *a = (struct sockaddr_l2 *) addr;
1090 PyObject *addrobj = makebdaddr(&_BT_L2_MEMB(a, bdaddr));
1091 PyObject *ret = NULL;
1092 if (addrobj) {
1093 ret = Py_BuildValue("Oi",
1094 addrobj,
1095 _BT_L2_MEMB(a, psm));
1096 Py_DECREF(addrobj);
1097 }
1098 return ret;
1099 }
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001100
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001101 case BTPROTO_RFCOMM:
1102 {
1103 struct sockaddr_rc *a = (struct sockaddr_rc *) addr;
1104 PyObject *addrobj = makebdaddr(&_BT_RC_MEMB(a, bdaddr));
1105 PyObject *ret = NULL;
1106 if (addrobj) {
1107 ret = Py_BuildValue("Oi",
1108 addrobj,
1109 _BT_RC_MEMB(a, channel));
1110 Py_DECREF(addrobj);
1111 }
1112 return ret;
1113 }
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001114
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001115 case BTPROTO_HCI:
1116 {
1117 struct sockaddr_hci *a = (struct sockaddr_hci *) addr;
Gregory P. Smith3e85dfd2010-10-17 00:43:10 +00001118#if defined(__NetBSD__) || defined(__DragonFly__)
1119 return makebdaddr(&_BT_HCI_MEMB(a, bdaddr));
1120#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001121 PyObject *ret = NULL;
1122 ret = Py_BuildValue("i", _BT_HCI_MEMB(a, dev));
1123 return ret;
Gregory P. Smith3e85dfd2010-10-17 00:43:10 +00001124#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001125 }
Thomas Wouterscf297e42007-02-23 15:07:44 +00001126
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001127#if !defined(__FreeBSD__)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001128 case BTPROTO_SCO:
1129 {
1130 struct sockaddr_sco *a = (struct sockaddr_sco *) addr;
1131 return makebdaddr(&_BT_SCO_MEMB(a, bdaddr));
1132 }
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001133#endif
1134
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001135 default:
1136 PyErr_SetString(PyExc_ValueError,
1137 "Unknown Bluetooth protocol");
1138 return NULL;
1139 }
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001140#endif
1141
Martin v. Löwis1ba3fd52001-08-10 20:29:40 +00001142#ifdef HAVE_NETPACKET_PACKET_H
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001143 case AF_PACKET:
1144 {
1145 struct sockaddr_ll *a = (struct sockaddr_ll *)addr;
1146 char *ifname = "";
1147 struct ifreq ifr;
1148 /* need to look up interface name give index */
1149 if (a->sll_ifindex) {
1150 ifr.ifr_ifindex = a->sll_ifindex;
1151 if (ioctl(sockfd, SIOCGIFNAME, &ifr) == 0)
1152 ifname = ifr.ifr_name;
1153 }
1154 return Py_BuildValue("shbhy#",
1155 ifname,
1156 ntohs(a->sll_protocol),
1157 a->sll_pkttype,
1158 a->sll_hatype,
1159 a->sll_addr,
1160 a->sll_halen);
1161 }
Jeremy Hylton22308652001-02-02 03:23:09 +00001162#endif
Guido van Rossum48a680c2001-03-02 06:34:14 +00001163
Christian Heimes043d6f62008-01-07 17:19:16 +00001164#ifdef HAVE_LINUX_TIPC_H
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001165 case AF_TIPC:
1166 {
1167 struct sockaddr_tipc *a = (struct sockaddr_tipc *) addr;
1168 if (a->addrtype == TIPC_ADDR_NAMESEQ) {
1169 return Py_BuildValue("IIIII",
1170 a->addrtype,
1171 a->addr.nameseq.type,
1172 a->addr.nameseq.lower,
1173 a->addr.nameseq.upper,
1174 a->scope);
1175 } else if (a->addrtype == TIPC_ADDR_NAME) {
1176 return Py_BuildValue("IIIII",
1177 a->addrtype,
1178 a->addr.name.name.type,
1179 a->addr.name.name.instance,
1180 a->addr.name.name.instance,
1181 a->scope);
1182 } else if (a->addrtype == TIPC_ADDR_ID) {
1183 return Py_BuildValue("IIIII",
1184 a->addrtype,
1185 a->addr.id.node,
1186 a->addr.id.ref,
1187 0,
1188 a->scope);
1189 } else {
1190 PyErr_SetString(PyExc_ValueError,
1191 "Invalid address type");
1192 return NULL;
1193 }
1194 }
Christian Heimes043d6f62008-01-07 17:19:16 +00001195#endif
1196
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001197 /* More cases here... */
Guido van Rossum30a685f1991-06-27 15:51:29 +00001198
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001199 default:
1200 /* If we don't know the address family, don't raise an
1201 exception -- return it as an (int, bytes) tuple. */
1202 return Py_BuildValue("iy#",
1203 addr->sa_family,
1204 addr->sa_data,
1205 sizeof(addr->sa_data));
Guido van Rossum25bec8c1992-08-05 19:00:45 +00001206
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001207 }
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001208}
1209
Guido van Rossum30a685f1991-06-27 15:51:29 +00001210
1211/* Parse a socket address argument according to the socket object's
1212 address family. Return 1 if the address was in the proper format,
1213 0 of not. The address is returned through addr_ret, its length
1214 through len_ret. */
1215
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001216static int
Guido van Rossum48a680c2001-03-02 06:34:14 +00001217getsockaddrarg(PySocketSockObject *s, PyObject *args,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001218 struct sockaddr *addr_ret, int *len_ret)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001219{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001220 switch (s->sock_family) {
Guido van Rossum30a685f1991-06-27 15:51:29 +00001221
Andrew MacIntyred12dfbb2004-04-04 07:13:49 +00001222#if defined(AF_UNIX)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001223 case AF_UNIX:
1224 {
1225 struct sockaddr_un* addr;
1226 char *path;
1227 int len;
1228 if (!PyArg_Parse(args, "s#", &path, &len))
1229 return 0;
Thomas Wouters89f507f2006-12-13 04:49:30 +00001230
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001231 addr = (struct sockaddr_un*)addr_ret;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001232#ifdef linux
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001233 if (len > 0 && path[0] == 0) {
1234 /* Linux abstract namespace extension */
1235 if (len > sizeof addr->sun_path) {
1236 PyErr_SetString(socket_error,
1237 "AF_UNIX path too long");
1238 return 0;
1239 }
1240 }
1241 else
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001242#endif /* linux */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001243 {
1244 /* regular NULL-terminated string */
1245 if (len >= sizeof addr->sun_path) {
1246 PyErr_SetString(socket_error,
1247 "AF_UNIX path too long");
1248 return 0;
1249 }
1250 addr->sun_path[len] = 0;
1251 }
1252 addr->sun_family = s->sock_family;
1253 memcpy(addr->sun_path, path, len);
Andrew MacIntyredaedf212004-04-11 12:03:57 +00001254#if defined(PYOS_OS2)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001255 *len_ret = sizeof(*addr);
Andrew MacIntyredaedf212004-04-11 12:03:57 +00001256#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001257 *len_ret = len + offsetof(struct sockaddr_un, sun_path);
Andrew MacIntyredaedf212004-04-11 12:03:57 +00001258#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001259 return 1;
1260 }
Guido van Rossumb6775db1994-08-01 11:34:53 +00001261#endif /* AF_UNIX */
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001262
Martin v. Löwis11017b12006-01-14 18:12:57 +00001263#if defined(AF_NETLINK)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001264 case AF_NETLINK:
1265 {
1266 struct sockaddr_nl* addr;
1267 int pid, groups;
1268 addr = (struct sockaddr_nl *)addr_ret;
1269 if (!PyTuple_Check(args)) {
1270 PyErr_Format(
1271 PyExc_TypeError,
1272 "getsockaddrarg: "
1273 "AF_NETLINK address must be tuple, not %.500s",
1274 Py_TYPE(args)->tp_name);
1275 return 0;
1276 }
1277 if (!PyArg_ParseTuple(args, "II:getsockaddrarg", &pid, &groups))
1278 return 0;
1279 addr->nl_family = AF_NETLINK;
1280 addr->nl_pid = pid;
1281 addr->nl_groups = groups;
1282 *len_ret = sizeof(*addr);
1283 return 1;
1284 }
Martin v. Löwis11017b12006-01-14 18:12:57 +00001285#endif
1286
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001287 case AF_INET:
1288 {
1289 struct sockaddr_in* addr;
1290 char *host;
1291 int port, result;
1292 if (!PyTuple_Check(args)) {
1293 PyErr_Format(
1294 PyExc_TypeError,
1295 "getsockaddrarg: "
1296 "AF_INET address must be tuple, not %.500s",
1297 Py_TYPE(args)->tp_name);
1298 return 0;
1299 }
1300 if (!PyArg_ParseTuple(args, "eti:getsockaddrarg",
1301 "idna", &host, &port))
1302 return 0;
1303 addr=(struct sockaddr_in*)addr_ret;
1304 result = setipaddr(host, (struct sockaddr *)addr,
1305 sizeof(*addr), AF_INET);
1306 PyMem_Free(host);
1307 if (result < 0)
1308 return 0;
1309 if (port < 0 || port > 0xffff) {
1310 PyErr_SetString(
1311 PyExc_OverflowError,
1312 "getsockaddrarg: port must be 0-65535.");
1313 return 0;
1314 }
1315 addr->sin_family = AF_INET;
1316 addr->sin_port = htons((short)port);
1317 *len_ret = sizeof *addr;
1318 return 1;
1319 }
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001320
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00001321#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001322 case AF_INET6:
1323 {
1324 struct sockaddr_in6* addr;
1325 char *host;
1326 int port, flowinfo, scope_id, result;
1327 flowinfo = scope_id = 0;
1328 if (!PyTuple_Check(args)) {
1329 PyErr_Format(
1330 PyExc_TypeError,
1331 "getsockaddrarg: "
1332 "AF_INET6 address must be tuple, not %.500s",
1333 Py_TYPE(args)->tp_name);
1334 return 0;
1335 }
1336 if (!PyArg_ParseTuple(args, "eti|ii",
1337 "idna", &host, &port, &flowinfo,
1338 &scope_id)) {
1339 return 0;
1340 }
1341 addr = (struct sockaddr_in6*)addr_ret;
1342 result = setipaddr(host, (struct sockaddr *)addr,
1343 sizeof(*addr), AF_INET6);
1344 PyMem_Free(host);
1345 if (result < 0)
1346 return 0;
1347 if (port < 0 || port > 0xffff) {
1348 PyErr_SetString(
1349 PyExc_OverflowError,
1350 "getsockaddrarg: port must be 0-65535.");
1351 return 0;
1352 }
1353 addr->sin6_family = s->sock_family;
1354 addr->sin6_port = htons((short)port);
1355 addr->sin6_flowinfo = flowinfo;
1356 addr->sin6_scope_id = scope_id;
1357 *len_ret = sizeof *addr;
1358 return 1;
1359 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00001360#endif
1361
Hye-Shik Chang81268602004-02-02 06:05:24 +00001362#ifdef USE_BLUETOOTH
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001363 case AF_BLUETOOTH:
1364 {
1365 switch (s->sock_proto) {
1366 case BTPROTO_L2CAP:
1367 {
1368 struct sockaddr_l2 *addr;
1369 char *straddr;
Martin v. Löwis12af0482004-01-31 12:34:17 +00001370
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001371 addr = (struct sockaddr_l2 *)addr_ret;
1372 memset(addr, 0, sizeof(struct sockaddr_l2));
1373 _BT_L2_MEMB(addr, family) = AF_BLUETOOTH;
1374 if (!PyArg_ParseTuple(args, "si", &straddr,
1375 &_BT_L2_MEMB(addr, psm))) {
1376 PyErr_SetString(socket_error, "getsockaddrarg: "
1377 "wrong format");
1378 return 0;
1379 }
1380 if (setbdaddr(straddr, &_BT_L2_MEMB(addr, bdaddr)) < 0)
1381 return 0;
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001382
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001383 *len_ret = sizeof *addr;
1384 return 1;
1385 }
1386 case BTPROTO_RFCOMM:
1387 {
1388 struct sockaddr_rc *addr;
1389 char *straddr;
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001390
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001391 addr = (struct sockaddr_rc *)addr_ret;
1392 _BT_RC_MEMB(addr, family) = AF_BLUETOOTH;
1393 if (!PyArg_ParseTuple(args, "si", &straddr,
1394 &_BT_RC_MEMB(addr, channel))) {
1395 PyErr_SetString(socket_error, "getsockaddrarg: "
1396 "wrong format");
1397 return 0;
1398 }
1399 if (setbdaddr(straddr, &_BT_RC_MEMB(addr, bdaddr)) < 0)
1400 return 0;
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001401
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001402 *len_ret = sizeof *addr;
1403 return 1;
1404 }
1405 case BTPROTO_HCI:
1406 {
1407 struct sockaddr_hci *addr = (struct sockaddr_hci *)addr_ret;
Gregory P. Smith3e85dfd2010-10-17 00:43:10 +00001408#if defined(__NetBSD__) || defined(__DragonFly__)
1409 char *straddr = PyBytes_AS_STRING(args);
1410
1411 _BT_HCI_MEMB(addr, family) = AF_BLUETOOTH;
1412 if (straddr == NULL) {
1413 PyErr_SetString(socket_error, "getsockaddrarg: "
1414 "wrong format");
1415 return 0;
1416 }
1417 if (setbdaddr(straddr, &_BT_HCI_MEMB(addr, bdaddr)) < 0)
1418 return 0;
1419#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001420 _BT_HCI_MEMB(addr, family) = AF_BLUETOOTH;
1421 if (!PyArg_ParseTuple(args, "i", &_BT_HCI_MEMB(addr, dev))) {
1422 PyErr_SetString(socket_error, "getsockaddrarg: "
1423 "wrong format");
1424 return 0;
1425 }
Gregory P. Smith3e85dfd2010-10-17 00:43:10 +00001426#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001427 *len_ret = sizeof *addr;
1428 return 1;
1429 }
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001430#if !defined(__FreeBSD__)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001431 case BTPROTO_SCO:
1432 {
1433 struct sockaddr_sco *addr;
1434 char *straddr;
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001435
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001436 addr = (struct sockaddr_sco *)addr_ret;
1437 _BT_SCO_MEMB(addr, family) = AF_BLUETOOTH;
1438 if (!PyBytes_Check(args)) {
1439 PyErr_SetString(socket_error, "getsockaddrarg: "
1440 "wrong format");
1441 return 0;
1442 }
1443 straddr = PyBytes_AS_STRING(args);
1444 if (setbdaddr(straddr, &_BT_SCO_MEMB(addr, bdaddr)) < 0)
1445 return 0;
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001446
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001447 *len_ret = sizeof *addr;
1448 return 1;
1449 }
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001450#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001451 default:
1452 PyErr_SetString(socket_error, "getsockaddrarg: unknown Bluetooth protocol");
1453 return 0;
1454 }
1455 }
Martin v. Löwis12af0482004-01-31 12:34:17 +00001456#endif
1457
Martin v. Löwis1ba3fd52001-08-10 20:29:40 +00001458#ifdef HAVE_NETPACKET_PACKET_H
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001459 case AF_PACKET:
1460 {
1461 struct sockaddr_ll* addr;
1462 struct ifreq ifr;
1463 char *interfaceName;
1464 int protoNumber;
1465 int hatype = 0;
1466 int pkttype = 0;
1467 char *haddr = NULL;
1468 unsigned int halen = 0;
Guido van Rossum48a680c2001-03-02 06:34:14 +00001469
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001470 if (!PyTuple_Check(args)) {
1471 PyErr_Format(
1472 PyExc_TypeError,
1473 "getsockaddrarg: "
1474 "AF_PACKET address must be tuple, not %.500s",
1475 Py_TYPE(args)->tp_name);
1476 return 0;
1477 }
1478 if (!PyArg_ParseTuple(args, "si|iiy#", &interfaceName,
1479 &protoNumber, &pkttype, &hatype,
1480 &haddr, &halen))
1481 return 0;
1482 strncpy(ifr.ifr_name, interfaceName, sizeof(ifr.ifr_name));
1483 ifr.ifr_name[(sizeof(ifr.ifr_name))-1] = '\0';
1484 if (ioctl(s->sock_fd, SIOCGIFINDEX, &ifr) < 0) {
1485 s->errorhandler();
1486 return 0;
1487 }
1488 if (halen > 8) {
1489 PyErr_SetString(PyExc_ValueError,
1490 "Hardware address must be 8 bytes or less");
1491 return 0;
1492 }
1493 if (protoNumber < 0 || protoNumber > 0xffff) {
1494 PyErr_SetString(
1495 PyExc_OverflowError,
1496 "getsockaddrarg: protoNumber must be 0-65535.");
1497 return 0;
1498 }
1499 addr = (struct sockaddr_ll*)addr_ret;
1500 addr->sll_family = AF_PACKET;
1501 addr->sll_protocol = htons((short)protoNumber);
1502 addr->sll_ifindex = ifr.ifr_ifindex;
1503 addr->sll_pkttype = pkttype;
1504 addr->sll_hatype = hatype;
1505 if (halen != 0) {
1506 memcpy(&addr->sll_addr, haddr, halen);
1507 }
1508 addr->sll_halen = halen;
1509 *len_ret = sizeof *addr;
1510 return 1;
1511 }
Guido van Rossum48a680c2001-03-02 06:34:14 +00001512#endif
1513
Christian Heimes043d6f62008-01-07 17:19:16 +00001514#ifdef HAVE_LINUX_TIPC_H
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001515 case AF_TIPC:
1516 {
1517 unsigned int atype, v1, v2, v3;
1518 unsigned int scope = TIPC_CLUSTER_SCOPE;
1519 struct sockaddr_tipc *addr;
Christian Heimes043d6f62008-01-07 17:19:16 +00001520
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001521 if (!PyTuple_Check(args)) {
1522 PyErr_Format(
1523 PyExc_TypeError,
1524 "getsockaddrarg: "
1525 "AF_TIPC address must be tuple, not %.500s",
1526 Py_TYPE(args)->tp_name);
1527 return 0;
1528 }
Christian Heimes043d6f62008-01-07 17:19:16 +00001529
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001530 if (!PyArg_ParseTuple(args,
1531 "IIII|I;Invalid TIPC address format",
1532 &atype, &v1, &v2, &v3, &scope))
1533 return 0;
Christian Heimes043d6f62008-01-07 17:19:16 +00001534
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001535 addr = (struct sockaddr_tipc *) addr_ret;
1536 memset(addr, 0, sizeof(struct sockaddr_tipc));
Christian Heimes043d6f62008-01-07 17:19:16 +00001537
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001538 addr->family = AF_TIPC;
1539 addr->scope = scope;
1540 addr->addrtype = atype;
Christian Heimes043d6f62008-01-07 17:19:16 +00001541
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001542 if (atype == TIPC_ADDR_NAMESEQ) {
1543 addr->addr.nameseq.type = v1;
1544 addr->addr.nameseq.lower = v2;
1545 addr->addr.nameseq.upper = v3;
1546 } else if (atype == TIPC_ADDR_NAME) {
1547 addr->addr.name.name.type = v1;
1548 addr->addr.name.name.instance = v2;
1549 } else if (atype == TIPC_ADDR_ID) {
1550 addr->addr.id.node = v1;
1551 addr->addr.id.ref = v2;
1552 } else {
1553 /* Shouldn't happen */
1554 PyErr_SetString(PyExc_TypeError, "Invalid address type");
1555 return 0;
1556 }
Christian Heimes043d6f62008-01-07 17:19:16 +00001557
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001558 *len_ret = sizeof(*addr);
Christian Heimes043d6f62008-01-07 17:19:16 +00001559
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001560 return 1;
1561 }
Christian Heimes043d6f62008-01-07 17:19:16 +00001562#endif
1563
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001564 /* More cases here... */
Guido van Rossum30a685f1991-06-27 15:51:29 +00001565
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001566 default:
1567 PyErr_SetString(socket_error, "getsockaddrarg: bad family");
1568 return 0;
Guido van Rossum30a685f1991-06-27 15:51:29 +00001569
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001570 }
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001571}
1572
Guido van Rossum30a685f1991-06-27 15:51:29 +00001573
Guido van Rossum48a680c2001-03-02 06:34:14 +00001574/* Get the address length according to the socket object's address family.
Guido van Rossum710e1df1992-06-12 10:39:36 +00001575 Return 1 if the family is known, 0 otherwise. The length is returned
1576 through len_ret. */
1577
1578static int
Peter Schneider-Kamp618e25d2000-07-11 23:00:12 +00001579getsockaddrlen(PySocketSockObject *s, socklen_t *len_ret)
Guido van Rossum710e1df1992-06-12 10:39:36 +00001580{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001581 switch (s->sock_family) {
Guido van Rossum710e1df1992-06-12 10:39:36 +00001582
Andrew MacIntyred12dfbb2004-04-04 07:13:49 +00001583#if defined(AF_UNIX)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001584 case AF_UNIX:
1585 {
1586 *len_ret = sizeof (struct sockaddr_un);
1587 return 1;
1588 }
Guido van Rossumb6775db1994-08-01 11:34:53 +00001589#endif /* AF_UNIX */
Martin v. Löwis11017b12006-01-14 18:12:57 +00001590#if defined(AF_NETLINK)
1591 case AF_NETLINK:
1592 {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001593 *len_ret = sizeof (struct sockaddr_nl);
1594 return 1;
Martin v. Löwis11017b12006-01-14 18:12:57 +00001595 }
1596#endif
Guido van Rossum710e1df1992-06-12 10:39:36 +00001597
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001598 case AF_INET:
1599 {
1600 *len_ret = sizeof (struct sockaddr_in);
1601 return 1;
1602 }
Guido van Rossum710e1df1992-06-12 10:39:36 +00001603
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00001604#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001605 case AF_INET6:
1606 {
1607 *len_ret = sizeof (struct sockaddr_in6);
1608 return 1;
1609 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00001610#endif
1611
Hye-Shik Chang81268602004-02-02 06:05:24 +00001612#ifdef USE_BLUETOOTH
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001613 case AF_BLUETOOTH:
1614 {
1615 switch(s->sock_proto)
1616 {
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001617
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001618 case BTPROTO_L2CAP:
1619 *len_ret = sizeof (struct sockaddr_l2);
1620 return 1;
1621 case BTPROTO_RFCOMM:
1622 *len_ret = sizeof (struct sockaddr_rc);
1623 return 1;
1624 case BTPROTO_HCI:
1625 *len_ret = sizeof (struct sockaddr_hci);
1626 return 1;
Hye-Shik Chang81268602004-02-02 06:05:24 +00001627#if !defined(__FreeBSD__)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001628 case BTPROTO_SCO:
1629 *len_ret = sizeof (struct sockaddr_sco);
1630 return 1;
Hye-Shik Chang81268602004-02-02 06:05:24 +00001631#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001632 default:
1633 PyErr_SetString(socket_error, "getsockaddrlen: "
1634 "unknown BT protocol");
1635 return 0;
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001636
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001637 }
1638 }
Martin v. Löwis12af0482004-01-31 12:34:17 +00001639#endif
1640
Martin v. Löwis1ba3fd52001-08-10 20:29:40 +00001641#ifdef HAVE_NETPACKET_PACKET_H
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001642 case AF_PACKET:
1643 {
1644 *len_ret = sizeof (struct sockaddr_ll);
1645 return 1;
1646 }
Jeremy Hylton22308652001-02-02 03:23:09 +00001647#endif
Guido van Rossum48a680c2001-03-02 06:34:14 +00001648
Christian Heimes043d6f62008-01-07 17:19:16 +00001649#ifdef HAVE_LINUX_TIPC_H
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001650 case AF_TIPC:
1651 {
1652 *len_ret = sizeof (struct sockaddr_tipc);
1653 return 1;
1654 }
Christian Heimes043d6f62008-01-07 17:19:16 +00001655#endif
1656
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001657 /* More cases here... */
Guido van Rossum710e1df1992-06-12 10:39:36 +00001658
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001659 default:
1660 PyErr_SetString(socket_error, "getsockaddrlen: bad family");
1661 return 0;
Guido van Rossum710e1df1992-06-12 10:39:36 +00001662
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001663 }
Guido van Rossum710e1df1992-06-12 10:39:36 +00001664}
1665
1666
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00001667/* s._accept() -> (fd, address) */
Guido van Rossum30a685f1991-06-27 15:51:29 +00001668
Guido van Rossum73624e91994-10-10 17:59:00 +00001669static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001670sock_accept(PySocketSockObject *s)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001671{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001672 sock_addr_t addrbuf;
1673 SOCKET_T newfd = INVALID_SOCKET;
1674 socklen_t addrlen;
1675 PyObject *sock = NULL;
1676 PyObject *addr = NULL;
1677 PyObject *res = NULL;
1678 int timeout;
Antoine Pitroub1c54962010-10-14 15:05:38 +00001679#ifdef HAVE_ACCEPT4
1680 int flags = 0;
1681#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001682 if (!getsockaddrlen(s, &addrlen))
1683 return NULL;
1684 memset(&addrbuf, 0, addrlen);
Guido van Rossum67f7a382002-06-06 21:08:16 +00001685
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001686 if (!IS_SELECTABLE(s))
1687 return select_error();
Neal Norwitz082b2df2006-02-07 07:04:46 +00001688
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00001689 BEGIN_SELECT_LOOP(s)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001690 Py_BEGIN_ALLOW_THREADS
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00001691 timeout = internal_select_ex(s, 0, interval);
Antoine Pitroub1c54962010-10-14 15:05:38 +00001692 if (!timeout) {
1693#ifdef HAVE_ACCEPT4
1694 /* inherit socket flags and use accept4 call */
1695 flags = s->sock_type & (SOCK_CLOEXEC | SOCK_NONBLOCK);
1696 newfd = accept4(s->sock_fd, SAS2SA(&addrbuf), &addrlen, flags);
1697#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001698 newfd = accept(s->sock_fd, SAS2SA(&addrbuf), &addrlen);
Antoine Pitroub1c54962010-10-14 15:05:38 +00001699#endif /* HAVE_ACCEPT4 */
1700 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001701 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00001702
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001703 if (timeout == 1) {
1704 PyErr_SetString(socket_timeout, "timed out");
1705 return NULL;
1706 }
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00001707 END_SELECT_LOOP(s)
Raymond Hettingeref7343c2003-06-29 03:08:05 +00001708
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001709 if (newfd == INVALID_SOCKET)
1710 return s->errorhandler();
Barry Warsaw752300b1997-01-03 17:18:10 +00001711
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001712 sock = PyLong_FromSocket_t(newfd);
1713 if (sock == NULL) {
1714 SOCKETCLOSE(newfd);
1715 goto finally;
1716 }
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00001717
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001718 addr = makesockaddr(s->sock_fd, SAS2SA(&addrbuf),
1719 addrlen, s->sock_proto);
1720 if (addr == NULL)
1721 goto finally;
Barry Warsaw752300b1997-01-03 17:18:10 +00001722
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001723 res = PyTuple_Pack(2, sock, addr);
Barry Warsaw752300b1997-01-03 17:18:10 +00001724
Guido van Rossum67f7a382002-06-06 21:08:16 +00001725finally:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001726 Py_XDECREF(sock);
1727 Py_XDECREF(addr);
1728 return res;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001729}
1730
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001731PyDoc_STRVAR(accept_doc,
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00001732"_accept() -> (integer, address info)\n\
Guido van Rossum82a5c661998-07-07 20:45:43 +00001733\n\
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00001734Wait for an incoming connection. Return a new socket file descriptor\n\
1735representing the connection, and the address of the client.\n\
1736For IP sockets, the address info is a pair (hostaddr, port).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001737
Guido van Rossum11ba0942002-06-13 15:07:44 +00001738/* s.setblocking(flag) method. Argument:
1739 False -- non-blocking mode; same as settimeout(0)
1740 True -- blocking mode; same as settimeout(None)
1741*/
Guido van Rossume4485b01994-09-07 14:32:49 +00001742
Guido van Rossum73624e91994-10-10 17:59:00 +00001743static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001744sock_setblocking(PySocketSockObject *s, PyObject *arg)
Guido van Rossume4485b01994-09-07 14:32:49 +00001745{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001746 int block;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001747
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001748 block = PyLong_AsLong(arg);
1749 if (block == -1 && PyErr_Occurred())
1750 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001751
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001752 s->sock_timeout = block ? -1.0 : 0.0;
1753 internal_setblocking(s, block);
Guido van Rossume4485b01994-09-07 14:32:49 +00001754
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001755 Py_INCREF(Py_None);
1756 return Py_None;
Guido van Rossume4485b01994-09-07 14:32:49 +00001757}
Guido van Rossume4485b01994-09-07 14:32:49 +00001758
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001759PyDoc_STRVAR(setblocking_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001760"setblocking(flag)\n\
1761\n\
1762Set the socket to blocking (flag is true) or non-blocking (false).\n\
Guido van Rossum11ba0942002-06-13 15:07:44 +00001763setblocking(True) is equivalent to settimeout(None);\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001764setblocking(False) is equivalent to settimeout(0.0).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001765
Guido van Rossum11ba0942002-06-13 15:07:44 +00001766/* s.settimeout(timeout) method. Argument:
1767 None -- no timeout, blocking mode; same as setblocking(True)
1768 0.0 -- non-blocking mode; same as setblocking(False)
1769 > 0 -- timeout mode; operations time out after timeout seconds
1770 < 0 -- illegal; raises an exception
1771*/
Guido van Rossum67f7a382002-06-06 21:08:16 +00001772static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001773sock_settimeout(PySocketSockObject *s, PyObject *arg)
Guido van Rossum67f7a382002-06-06 21:08:16 +00001774{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001775 double timeout;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001776
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001777 if (arg == Py_None)
1778 timeout = -1.0;
1779 else {
1780 timeout = PyFloat_AsDouble(arg);
1781 if (timeout < 0.0) {
1782 if (!PyErr_Occurred())
1783 PyErr_SetString(PyExc_ValueError,
1784 "Timeout value out of range");
1785 return NULL;
1786 }
1787 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00001788
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001789 s->sock_timeout = timeout;
1790 internal_setblocking(s, timeout < 0.0);
Guido van Rossum67f7a382002-06-06 21:08:16 +00001791
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001792 Py_INCREF(Py_None);
1793 return Py_None;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001794}
1795
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001796PyDoc_STRVAR(settimeout_doc,
Guido van Rossum3eede5a2002-06-07 02:08:35 +00001797"settimeout(timeout)\n\
Guido van Rossum67f7a382002-06-06 21:08:16 +00001798\n\
Guido van Rossum11ba0942002-06-13 15:07:44 +00001799Set a timeout on socket operations. 'timeout' can be a float,\n\
1800giving in seconds, or None. Setting a timeout of None disables\n\
1801the timeout feature and is equivalent to setblocking(1).\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001802Setting a timeout of zero is the same as setblocking(0).");
Guido van Rossum67f7a382002-06-06 21:08:16 +00001803
Guido van Rossum3eede5a2002-06-07 02:08:35 +00001804/* s.gettimeout() method.
1805 Returns the timeout associated with a socket. */
Guido van Rossum67f7a382002-06-06 21:08:16 +00001806static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001807sock_gettimeout(PySocketSockObject *s)
Guido van Rossum67f7a382002-06-06 21:08:16 +00001808{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001809 if (s->sock_timeout < 0.0) {
1810 Py_INCREF(Py_None);
1811 return Py_None;
1812 }
1813 else
1814 return PyFloat_FromDouble(s->sock_timeout);
Guido van Rossum67f7a382002-06-06 21:08:16 +00001815}
1816
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001817PyDoc_STRVAR(gettimeout_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00001818"gettimeout() -> timeout\n\
Guido van Rossum67f7a382002-06-06 21:08:16 +00001819\n\
1820Returns the timeout in floating seconds associated with socket \n\
1821operations. A timeout of None indicates that timeouts on socket \n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001822operations are disabled.");
Guido van Rossume4485b01994-09-07 14:32:49 +00001823
Guido van Rossumaee08791992-09-08 09:05:33 +00001824/* s.setsockopt() method.
1825 With an integer third argument, sets an integer option.
1826 With a string third argument, sets an option from a buffer;
1827 use optional built-in module 'struct' to encode the string. */
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001828
Guido van Rossum73624e91994-10-10 17:59:00 +00001829static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001830sock_setsockopt(PySocketSockObject *s, PyObject *args)
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001831{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001832 int level;
1833 int optname;
1834 int res;
1835 char *buf;
1836 int buflen;
1837 int flag;
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001838
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001839 if (PyArg_ParseTuple(args, "iii:setsockopt",
1840 &level, &optname, &flag)) {
1841 buf = (char *) &flag;
1842 buflen = sizeof flag;
1843 }
1844 else {
1845 PyErr_Clear();
1846 if (!PyArg_ParseTuple(args, "iiy#:setsockopt",
1847 &level, &optname, &buf, &buflen))
1848 return NULL;
1849 }
1850 res = setsockopt(s->sock_fd, level, optname, (void *)buf, buflen);
1851 if (res < 0)
1852 return s->errorhandler();
1853 Py_INCREF(Py_None);
1854 return Py_None;
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001855}
1856
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001857PyDoc_STRVAR(setsockopt_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001858"setsockopt(level, option, value)\n\
1859\n\
1860Set a socket option. See the Unix manual for level and option.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001861The value argument can either be an integer or a string.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001862
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001863
Guido van Rossumaee08791992-09-08 09:05:33 +00001864/* s.getsockopt() method.
1865 With two arguments, retrieves an integer option.
1866 With a third integer argument, retrieves a string buffer of that size;
1867 use optional built-in module 'struct' to decode the string. */
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001868
Guido van Rossum73624e91994-10-10 17:59:00 +00001869static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001870sock_getsockopt(PySocketSockObject *s, PyObject *args)
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001871{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001872 int level;
1873 int optname;
1874 int res;
1875 PyObject *buf;
1876 socklen_t buflen = 0;
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001877
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001878 if (!PyArg_ParseTuple(args, "ii|i:getsockopt",
1879 &level, &optname, &buflen))
1880 return NULL;
Guido van Rossum48a680c2001-03-02 06:34:14 +00001881
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001882 if (buflen == 0) {
1883 int flag = 0;
1884 socklen_t flagsize = sizeof flag;
1885 res = getsockopt(s->sock_fd, level, optname,
1886 (void *)&flag, &flagsize);
1887 if (res < 0)
1888 return s->errorhandler();
1889 return PyLong_FromLong(flag);
1890 }
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001891#ifdef __VMS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001892 /* socklen_t is unsigned so no negative test is needed,
1893 test buflen == 0 is previously done */
1894 if (buflen > 1024) {
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001895#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001896 if (buflen <= 0 || buflen > 1024) {
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001897#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001898 PyErr_SetString(socket_error,
1899 "getsockopt buflen out of range");
1900 return NULL;
1901 }
1902 buf = PyBytes_FromStringAndSize((char *)NULL, buflen);
1903 if (buf == NULL)
1904 return NULL;
1905 res = getsockopt(s->sock_fd, level, optname,
1906 (void *)PyBytes_AS_STRING(buf), &buflen);
1907 if (res < 0) {
1908 Py_DECREF(buf);
1909 return s->errorhandler();
1910 }
1911 _PyBytes_Resize(&buf, buflen);
1912 return buf;
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001913}
1914
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001915PyDoc_STRVAR(getsockopt_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001916"getsockopt(level, option[, buffersize]) -> value\n\
1917\n\
1918Get a socket option. See the Unix manual for level and option.\n\
1919If a nonzero buffersize argument is given, the return value is a\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001920string of that length; otherwise it is an integer.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001921
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001922
Fred Drake728819a2000-07-01 03:40:12 +00001923/* s.bind(sockaddr) method */
Guido van Rossum30a685f1991-06-27 15:51:29 +00001924
Guido van Rossum73624e91994-10-10 17:59:00 +00001925static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001926sock_bind(PySocketSockObject *s, PyObject *addro)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001927{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001928 sock_addr_t addrbuf;
1929 int addrlen;
1930 int res;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00001931
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001932 if (!getsockaddrarg(s, addro, SAS2SA(&addrbuf), &addrlen))
1933 return NULL;
1934 Py_BEGIN_ALLOW_THREADS
1935 res = bind(s->sock_fd, SAS2SA(&addrbuf), addrlen);
1936 Py_END_ALLOW_THREADS
1937 if (res < 0)
1938 return s->errorhandler();
1939 Py_INCREF(Py_None);
1940 return Py_None;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001941}
1942
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001943PyDoc_STRVAR(bind_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001944"bind(address)\n\
1945\n\
1946Bind the socket to a local address. For IP sockets, the address is a\n\
Jeremy Hylton22308652001-02-02 03:23:09 +00001947pair (host, port); the host must refer to the local host. For raw packet\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001948sockets the address is a tuple (ifname, proto [,pkttype [,hatype]])");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001949
Guido van Rossum30a685f1991-06-27 15:51:29 +00001950
1951/* s.close() method.
1952 Set the file descriptor to -1 so operations tried subsequently
1953 will surely fail. */
1954
Guido van Rossum73624e91994-10-10 17:59:00 +00001955static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001956sock_close(PySocketSockObject *s)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001957{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001958 SOCKET_T fd;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00001959
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001960 if ((fd = s->sock_fd) != -1) {
1961 s->sock_fd = -1;
1962 Py_BEGIN_ALLOW_THREADS
1963 (void) SOCKETCLOSE(fd);
1964 Py_END_ALLOW_THREADS
1965 }
1966 Py_INCREF(Py_None);
1967 return Py_None;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001968}
1969
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001970PyDoc_STRVAR(close_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001971"close()\n\
1972\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001973Close the socket. It cannot be used after this call.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001974
Antoine Pitroue43f9d02010-08-08 23:24:50 +00001975static PyObject *
Antoine Pitrou6e451df2010-08-09 20:39:54 +00001976sock_detach(PySocketSockObject *s)
Antoine Pitroue43f9d02010-08-08 23:24:50 +00001977{
Antoine Pitrou6e451df2010-08-09 20:39:54 +00001978 SOCKET_T fd = s->sock_fd;
Antoine Pitroue43f9d02010-08-08 23:24:50 +00001979 s->sock_fd = -1;
Antoine Pitrou6e451df2010-08-09 20:39:54 +00001980 return PyLong_FromSocket_t(fd);
Antoine Pitroue43f9d02010-08-08 23:24:50 +00001981}
1982
Antoine Pitrou6e451df2010-08-09 20:39:54 +00001983PyDoc_STRVAR(detach_doc,
1984"detach()\n\
Antoine Pitroue43f9d02010-08-08 23:24:50 +00001985\n\
1986Close the socket object without closing the underlying file descriptor.\
1987The object cannot be used after this call, but the file descriptor\
Antoine Pitrou6e451df2010-08-09 20:39:54 +00001988can be reused for other purposes. The file descriptor is returned.");
Antoine Pitroue43f9d02010-08-08 23:24:50 +00001989
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001990static int
Raymond Hettingeref7343c2003-06-29 03:08:05 +00001991internal_connect(PySocketSockObject *s, struct sockaddr *addr, int addrlen,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001992 int *timeoutp)
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001993{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001994 int res, timeout;
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001995
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001996 timeout = 0;
1997 res = connect(s->sock_fd, addr, addrlen);
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001998
1999#ifdef MS_WINDOWS
2000
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002001 if (s->sock_timeout > 0.0) {
2002 if (res < 0 && WSAGetLastError() == WSAEWOULDBLOCK &&
2003 IS_SELECTABLE(s)) {
2004 /* This is a mess. Best solution: trust select */
2005 fd_set fds;
2006 fd_set fds_exc;
2007 struct timeval tv;
2008 tv.tv_sec = (int)s->sock_timeout;
2009 tv.tv_usec = (int)((s->sock_timeout - tv.tv_sec) * 1e6);
2010 FD_ZERO(&fds);
2011 FD_SET(s->sock_fd, &fds);
2012 FD_ZERO(&fds_exc);
2013 FD_SET(s->sock_fd, &fds_exc);
Antoine Pitrou19467d22010-08-17 19:33:30 +00002014 res = select(Py_SAFE_DOWNCAST(s->sock_fd+1, SOCKET_T, int),
2015 NULL, &fds, &fds_exc, &tv);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002016 if (res == 0) {
2017 res = WSAEWOULDBLOCK;
2018 timeout = 1;
2019 } else if (res > 0) {
2020 if (FD_ISSET(s->sock_fd, &fds))
2021 /* The socket is in the writable set - this
2022 means connected */
2023 res = 0;
2024 else {
2025 /* As per MS docs, we need to call getsockopt()
2026 to get the underlying error */
2027 int res_size = sizeof res;
2028 /* It must be in the exception set */
2029 assert(FD_ISSET(s->sock_fd, &fds_exc));
2030 if (0 == getsockopt(s->sock_fd, SOL_SOCKET, SO_ERROR,
2031 (char *)&res, &res_size))
2032 /* getsockopt also clears WSAGetLastError,
2033 so reset it back. */
2034 WSASetLastError(res);
2035 else
2036 res = WSAGetLastError();
2037 }
2038 }
2039 /* else if (res < 0) an error occurred */
2040 }
2041 }
Guido van Rossum7b8bac12002-06-13 16:07:04 +00002042
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002043 if (res < 0)
2044 res = WSAGetLastError();
Guido van Rossum7b8bac12002-06-13 16:07:04 +00002045
2046#else
2047
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002048 if (s->sock_timeout > 0.0) {
2049 if (res < 0 && errno == EINPROGRESS && IS_SELECTABLE(s)) {
2050 timeout = internal_select(s, 1);
2051 if (timeout == 0) {
2052 /* Bug #1019808: in case of an EINPROGRESS,
2053 use getsockopt(SO_ERROR) to get the real
2054 error. */
2055 socklen_t res_size = sizeof res;
2056 (void)getsockopt(s->sock_fd, SOL_SOCKET,
2057 SO_ERROR, &res, &res_size);
2058 if (res == EISCONN)
2059 res = 0;
2060 errno = res;
2061 }
2062 else if (timeout == -1) {
2063 res = errno; /* had error */
2064 }
2065 else
2066 res = EWOULDBLOCK; /* timed out */
2067 }
2068 }
Guido van Rossum7b8bac12002-06-13 16:07:04 +00002069
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002070 if (res < 0)
2071 res = errno;
Guido van Rossum7b8bac12002-06-13 16:07:04 +00002072
2073#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002074 *timeoutp = timeout;
Guido van Rossum7b8bac12002-06-13 16:07:04 +00002075
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002076 return res;
Guido van Rossum7b8bac12002-06-13 16:07:04 +00002077}
Guido van Rossum30a685f1991-06-27 15:51:29 +00002078
Fred Drake728819a2000-07-01 03:40:12 +00002079/* s.connect(sockaddr) method */
Guido van Rossum30a685f1991-06-27 15:51:29 +00002080
Guido van Rossum73624e91994-10-10 17:59:00 +00002081static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002082sock_connect(PySocketSockObject *s, PyObject *addro)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002083{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002084 sock_addr_t addrbuf;
2085 int addrlen;
2086 int res;
2087 int timeout;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00002088
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002089 if (!getsockaddrarg(s, addro, SAS2SA(&addrbuf), &addrlen))
2090 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002091
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002092 Py_BEGIN_ALLOW_THREADS
2093 res = internal_connect(s, SAS2SA(&addrbuf), addrlen, &timeout);
2094 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00002095
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002096 if (timeout == 1) {
2097 PyErr_SetString(socket_timeout, "timed out");
2098 return NULL;
2099 }
2100 if (res != 0)
2101 return s->errorhandler();
2102 Py_INCREF(Py_None);
2103 return Py_None;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002104}
2105
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002106PyDoc_STRVAR(connect_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002107"connect(address)\n\
2108\n\
2109Connect the socket to a remote address. For IP sockets, the address\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002110is a pair (host, port).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002111
Guido van Rossum30a685f1991-06-27 15:51:29 +00002112
Fred Drake728819a2000-07-01 03:40:12 +00002113/* s.connect_ex(sockaddr) method */
Guido van Rossumfc4255d1997-11-19 18:57:13 +00002114
2115static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002116sock_connect_ex(PySocketSockObject *s, PyObject *addro)
Guido van Rossumfc4255d1997-11-19 18:57:13 +00002117{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002118 sock_addr_t addrbuf;
2119 int addrlen;
2120 int res;
2121 int timeout;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00002122
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002123 if (!getsockaddrarg(s, addro, SAS2SA(&addrbuf), &addrlen))
2124 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002125
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002126 Py_BEGIN_ALLOW_THREADS
2127 res = internal_connect(s, SAS2SA(&addrbuf), addrlen, &timeout);
2128 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00002129
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002130 /* Signals are not errors (though they may raise exceptions). Adapted
2131 from PyErr_SetFromErrnoWithFilenameObject(). */
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002132#ifdef EINTR
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002133 if (res == EINTR && PyErr_CheckSignals())
2134 return NULL;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002135#endif
2136
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002137 return PyLong_FromLong((long) res);
Guido van Rossumfc4255d1997-11-19 18:57:13 +00002138}
2139
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002140PyDoc_STRVAR(connect_ex_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00002141"connect_ex(address) -> errno\n\
Guido van Rossum82a5c661998-07-07 20:45:43 +00002142\n\
2143This is like connect(address), but returns an error code (the errno value)\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002144instead of raising an exception when an error occurs.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002145
Guido van Rossumfc4255d1997-11-19 18:57:13 +00002146
Guido van Rossumed233a51992-06-23 09:07:03 +00002147/* s.fileno() method */
2148
Guido van Rossum73624e91994-10-10 17:59:00 +00002149static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002150sock_fileno(PySocketSockObject *s)
Guido van Rossumed233a51992-06-23 09:07:03 +00002151{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002152 return PyLong_FromSocket_t(s->sock_fd);
Guido van Rossumed233a51992-06-23 09:07:03 +00002153}
2154
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002155PyDoc_STRVAR(fileno_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002156"fileno() -> integer\n\
2157\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002158Return the integer file descriptor of the socket.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002159
Guido van Rossumed233a51992-06-23 09:07:03 +00002160
Guido van Rossumc89705d1992-11-26 08:54:07 +00002161/* s.getsockname() method */
2162
Guido van Rossum73624e91994-10-10 17:59:00 +00002163static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002164sock_getsockname(PySocketSockObject *s)
Guido van Rossumc89705d1992-11-26 08:54:07 +00002165{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002166 sock_addr_t addrbuf;
2167 int res;
2168 socklen_t addrlen;
Guido van Rossumff3ab422000-04-24 15:16:03 +00002169
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002170 if (!getsockaddrlen(s, &addrlen))
2171 return NULL;
2172 memset(&addrbuf, 0, addrlen);
2173 Py_BEGIN_ALLOW_THREADS
2174 res = getsockname(s->sock_fd, SAS2SA(&addrbuf), &addrlen);
2175 Py_END_ALLOW_THREADS
2176 if (res < 0)
2177 return s->errorhandler();
2178 return makesockaddr(s->sock_fd, SAS2SA(&addrbuf), addrlen,
2179 s->sock_proto);
Guido van Rossumc89705d1992-11-26 08:54:07 +00002180}
2181
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002182PyDoc_STRVAR(getsockname_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002183"getsockname() -> address info\n\
2184\n\
2185Return the address of the local endpoint. For IP sockets, the address\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002186info is a pair (hostaddr, port).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002187
Guido van Rossumc89705d1992-11-26 08:54:07 +00002188
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002189#ifdef HAVE_GETPEERNAME /* Cray APP doesn't have this :-( */
Guido van Rossumc89705d1992-11-26 08:54:07 +00002190/* s.getpeername() method */
2191
Guido van Rossum73624e91994-10-10 17:59:00 +00002192static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002193sock_getpeername(PySocketSockObject *s)
Guido van Rossumc89705d1992-11-26 08:54:07 +00002194{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002195 sock_addr_t addrbuf;
2196 int res;
2197 socklen_t addrlen;
Guido van Rossumff3ab422000-04-24 15:16:03 +00002198
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002199 if (!getsockaddrlen(s, &addrlen))
2200 return NULL;
2201 memset(&addrbuf, 0, addrlen);
2202 Py_BEGIN_ALLOW_THREADS
2203 res = getpeername(s->sock_fd, SAS2SA(&addrbuf), &addrlen);
2204 Py_END_ALLOW_THREADS
2205 if (res < 0)
2206 return s->errorhandler();
2207 return makesockaddr(s->sock_fd, SAS2SA(&addrbuf), addrlen,
2208 s->sock_proto);
Guido van Rossumc89705d1992-11-26 08:54:07 +00002209}
Guido van Rossum82a5c661998-07-07 20:45:43 +00002210
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002211PyDoc_STRVAR(getpeername_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002212"getpeername() -> address info\n\
2213\n\
2214Return the address of the remote endpoint. For IP sockets, the address\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002215info is a pair (hostaddr, port).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002216
Guido van Rossumb6775db1994-08-01 11:34:53 +00002217#endif /* HAVE_GETPEERNAME */
Guido van Rossumc89705d1992-11-26 08:54:07 +00002218
2219
Guido van Rossum30a685f1991-06-27 15:51:29 +00002220/* s.listen(n) method */
2221
Guido van Rossum73624e91994-10-10 17:59:00 +00002222static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002223sock_listen(PySocketSockObject *s, PyObject *arg)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002224{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002225 int backlog;
2226 int res;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00002227
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002228 backlog = PyLong_AsLong(arg);
2229 if (backlog == -1 && PyErr_Occurred())
2230 return NULL;
2231 Py_BEGIN_ALLOW_THREADS
2232 if (backlog < 1)
2233 backlog = 1;
2234 res = listen(s->sock_fd, backlog);
2235 Py_END_ALLOW_THREADS
2236 if (res < 0)
2237 return s->errorhandler();
2238 Py_INCREF(Py_None);
2239 return Py_None;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002240}
2241
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002242PyDoc_STRVAR(listen_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002243"listen(backlog)\n\
2244\n\
2245Enable a server to accept connections. The backlog argument must be at\n\
2246least 1; it specifies the number of unaccepted connection that the system\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002247will allow before refusing new connections.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002248
2249
Thomas Wouters477c8d52006-05-27 19:21:47 +00002250/*
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002251 * This is the guts of the recv() and recv_into() methods, which reads into a
Thomas Wouters902d6eb2007-01-09 23:18:33 +00002252 * char buffer. If you have any inc/dec ref to do to the objects that contain
Thomas Wouters477c8d52006-05-27 19:21:47 +00002253 * the buffer, do it in the caller. This function returns the number of bytes
2254 * succesfully read. If there was an error, it returns -1. Note that it is
2255 * also possible that we return a number of bytes smaller than the request
2256 * bytes.
2257 */
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002258
Antoine Pitrou19467d22010-08-17 19:33:30 +00002259static Py_ssize_t
2260sock_recv_guts(PySocketSockObject *s, char* cbuf, Py_ssize_t len, int flags)
Thomas Wouters477c8d52006-05-27 19:21:47 +00002261{
Antoine Pitrou19467d22010-08-17 19:33:30 +00002262 Py_ssize_t outlen = -1;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002263 int timeout;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002264#ifdef __VMS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002265 int remaining;
2266 char *read_buf;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002267#endif
2268
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002269 if (!IS_SELECTABLE(s)) {
2270 select_error();
2271 return -1;
2272 }
2273 if (len == 0) {
2274 /* If 0 bytes were requested, do nothing. */
2275 return 0;
2276 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002277
2278#ifndef __VMS
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002279 BEGIN_SELECT_LOOP(s)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002280 Py_BEGIN_ALLOW_THREADS
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002281 timeout = internal_select_ex(s, 0, interval);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002282 if (!timeout)
2283 outlen = recv(s->sock_fd, cbuf, len, flags);
2284 Py_END_ALLOW_THREADS
Thomas Wouters477c8d52006-05-27 19:21:47 +00002285
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002286 if (timeout == 1) {
2287 PyErr_SetString(socket_timeout, "timed out");
2288 return -1;
2289 }
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002290 END_SELECT_LOOP(s)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002291 if (outlen < 0) {
2292 /* Note: the call to errorhandler() ALWAYS indirectly returned
2293 NULL, so ignore its return value */
2294 s->errorhandler();
2295 return -1;
2296 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002297#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002298 read_buf = cbuf;
2299 remaining = len;
2300 while (remaining != 0) {
2301 unsigned int segment;
2302 int nread = -1;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002303
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002304 segment = remaining /SEGMENT_SIZE;
2305 if (segment != 0) {
2306 segment = SEGMENT_SIZE;
2307 }
2308 else {
2309 segment = remaining;
2310 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002311
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002312 BEGIN_SELECT_LOOP(s)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002313 Py_BEGIN_ALLOW_THREADS
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002314 timeout = internal_select_ex(s, 0, interval);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002315 if (!timeout)
2316 nread = recv(s->sock_fd, read_buf, segment, flags);
2317 Py_END_ALLOW_THREADS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002318 if (timeout == 1) {
2319 PyErr_SetString(socket_timeout, "timed out");
2320 return -1;
2321 }
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002322 END_SELECT_LOOP(s)
2323
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002324 if (nread < 0) {
2325 s->errorhandler();
2326 return -1;
2327 }
2328 if (nread != remaining) {
2329 read_buf += nread;
2330 break;
2331 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002332
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002333 remaining -= segment;
2334 read_buf += segment;
2335 }
2336 outlen = read_buf - cbuf;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002337#endif /* !__VMS */
2338
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002339 return outlen;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002340}
2341
Guido van Rossum48a680c2001-03-02 06:34:14 +00002342
Guido van Rossumeb6b33a1993-05-25 09:38:27 +00002343/* s.recv(nbytes [,flags]) method */
Guido van Rossum30a685f1991-06-27 15:51:29 +00002344
Guido van Rossum73624e91994-10-10 17:59:00 +00002345static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002346sock_recv(PySocketSockObject *s, PyObject *args)
Guido van Rossum30a685f1991-06-27 15:51:29 +00002347{
Antoine Pitrou19467d22010-08-17 19:33:30 +00002348 Py_ssize_t recvlen, outlen;
2349 int flags = 0;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002350 PyObject *buf;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002351
Antoine Pitrou19467d22010-08-17 19:33:30 +00002352 if (!PyArg_ParseTuple(args, "n|i:recv", &recvlen, &flags))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002353 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002354
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002355 if (recvlen < 0) {
2356 PyErr_SetString(PyExc_ValueError,
2357 "negative buffersize in recv");
2358 return NULL;
2359 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00002360
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002361 /* Allocate a new string. */
2362 buf = PyBytes_FromStringAndSize((char *) 0, recvlen);
2363 if (buf == NULL)
2364 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002365
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002366 /* Call the guts */
2367 outlen = sock_recv_guts(s, PyBytes_AS_STRING(buf), recvlen, flags);
2368 if (outlen < 0) {
2369 /* An error occurred, release the string and return an
2370 error. */
2371 Py_DECREF(buf);
2372 return NULL;
2373 }
2374 if (outlen != recvlen) {
2375 /* We did not read as many bytes as we anticipated, resize the
2376 string if possible and be successful. */
2377 _PyBytes_Resize(&buf, outlen);
2378 }
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00002379
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002380 return buf;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002381}
2382
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002383PyDoc_STRVAR(recv_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002384"recv(buffersize[, flags]) -> data\n\
2385\n\
2386Receive up to buffersize bytes from the socket. For the optional flags\n\
2387argument, see the Unix manual. When no data is available, block until\n\
2388at least one byte is available or until the remote end is closed. When\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002389the remote end is closed and all data is read, return the empty string.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002390
Guido van Rossum30a685f1991-06-27 15:51:29 +00002391
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002392/* s.recv_into(buffer, [nbytes [,flags]]) method */
Guido van Rossum30a685f1991-06-27 15:51:29 +00002393
Thomas Wouters477c8d52006-05-27 19:21:47 +00002394static PyObject*
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002395sock_recv_into(PySocketSockObject *s, PyObject *args, PyObject *kwds)
Thomas Wouters477c8d52006-05-27 19:21:47 +00002396{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002397 static char *kwlist[] = {"buffer", "nbytes", "flags", 0};
Thomas Wouters477c8d52006-05-27 19:21:47 +00002398
Antoine Pitrou19467d22010-08-17 19:33:30 +00002399 int flags = 0;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002400 Py_buffer pbuf;
2401 char *buf;
Antoine Pitrou19467d22010-08-17 19:33:30 +00002402 Py_ssize_t buflen, readlen, recvlen = 0;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002403
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002404 /* Get the buffer's memory */
Antoine Pitrou19467d22010-08-17 19:33:30 +00002405 if (!PyArg_ParseTupleAndKeywords(args, kwds, "w*|ni:recv_into", kwlist,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002406 &pbuf, &recvlen, &flags))
2407 return NULL;
2408 buf = pbuf.buf;
2409 buflen = pbuf.len;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002410
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002411 if (recvlen < 0) {
2412 PyBuffer_Release(&pbuf);
2413 PyErr_SetString(PyExc_ValueError,
2414 "negative buffersize in recv_into");
2415 return NULL;
2416 }
2417 if (recvlen == 0) {
2418 /* If nbytes was not specified, use the buffer's length */
2419 recvlen = buflen;
2420 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002421
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002422 /* Check if the buffer is large enough */
2423 if (buflen < recvlen) {
2424 PyBuffer_Release(&pbuf);
2425 PyErr_SetString(PyExc_ValueError,
2426 "buffer too small for requested bytes");
2427 return NULL;
2428 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002429
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002430 /* Call the guts */
2431 readlen = sock_recv_guts(s, buf, recvlen, flags);
2432 if (readlen < 0) {
2433 /* Return an error. */
2434 PyBuffer_Release(&pbuf);
2435 return NULL;
2436 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002437
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002438 PyBuffer_Release(&pbuf);
2439 /* Return the number of bytes read. Note that we do not do anything
2440 special here in the case that readlen < recvlen. */
2441 return PyLong_FromSsize_t(readlen);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002442}
2443
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002444PyDoc_STRVAR(recv_into_doc,
2445"recv_into(buffer, [nbytes[, flags]]) -> nbytes_read\n\
Thomas Wouters477c8d52006-05-27 19:21:47 +00002446\n\
2447A version of recv() that stores its data into a buffer rather than creating \n\
2448a new string. Receive up to buffersize bytes from the socket. If buffersize \n\
2449is not specified (or 0), receive up to the size available in the given buffer.\n\
2450\n\
2451See recv() for documentation about the flags.");
2452
2453
2454/*
Christian Heimes99170a52007-12-19 02:07:34 +00002455 * This is the guts of the recvfrom() and recvfrom_into() methods, which reads
2456 * into a char buffer. If you have any inc/def ref to do to the objects that
2457 * contain the buffer, do it in the caller. This function returns the number
2458 * of bytes succesfully read. If there was an error, it returns -1. Note
2459 * that it is also possible that we return a number of bytes smaller than the
2460 * request bytes.
Thomas Wouters477c8d52006-05-27 19:21:47 +00002461 *
2462 * 'addr' is a return value for the address object. Note that you must decref
2463 * it yourself.
2464 */
Antoine Pitrou19467d22010-08-17 19:33:30 +00002465static Py_ssize_t
2466sock_recvfrom_guts(PySocketSockObject *s, char* cbuf, Py_ssize_t len, int flags,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002467 PyObject** addr)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002468{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002469 sock_addr_t addrbuf;
2470 int timeout;
Antoine Pitrou19467d22010-08-17 19:33:30 +00002471 Py_ssize_t n = -1;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002472 socklen_t addrlen;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002473
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002474 *addr = NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002475
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002476 if (!getsockaddrlen(s, &addrlen))
2477 return -1;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002478
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002479 if (!IS_SELECTABLE(s)) {
2480 select_error();
2481 return -1;
2482 }
Neal Norwitz082b2df2006-02-07 07:04:46 +00002483
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002484 BEGIN_SELECT_LOOP(s)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002485 Py_BEGIN_ALLOW_THREADS
2486 memset(&addrbuf, 0, addrlen);
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002487 timeout = internal_select_ex(s, 0, interval);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002488 if (!timeout) {
Guido van Rossum8d665e61996-06-26 18:22:49 +00002489#ifndef MS_WINDOWS
Andrew MacIntyreba43e872002-03-03 03:03:52 +00002490#if defined(PYOS_OS2) && !defined(PYCC_GCC)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002491 n = recvfrom(s->sock_fd, cbuf, len, flags,
2492 SAS2SA(&addrbuf), &addrlen);
Guido van Rossum32c575d1997-12-02 20:37:32 +00002493#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002494 n = recvfrom(s->sock_fd, cbuf, len, flags,
2495 (void *) &addrbuf, &addrlen);
Guido van Rossum32c575d1997-12-02 20:37:32 +00002496#endif
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00002497#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002498 n = recvfrom(s->sock_fd, cbuf, len, flags,
2499 SAS2SA(&addrbuf), &addrlen);
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00002500#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002501 }
2502 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00002503
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002504 if (timeout == 1) {
2505 PyErr_SetString(socket_timeout, "timed out");
2506 return -1;
2507 }
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002508 END_SELECT_LOOP(s)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002509 if (n < 0) {
2510 s->errorhandler();
2511 return -1;
2512 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00002513
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002514 if (!(*addr = makesockaddr(s->sock_fd, SAS2SA(&addrbuf),
2515 addrlen, s->sock_proto)))
2516 return -1;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002517
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002518 return n;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002519}
2520
2521/* s.recvfrom(nbytes [,flags]) method */
2522
2523static PyObject *
2524sock_recvfrom(PySocketSockObject *s, PyObject *args)
2525{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002526 PyObject *buf = NULL;
2527 PyObject *addr = NULL;
2528 PyObject *ret = NULL;
Antoine Pitrou19467d22010-08-17 19:33:30 +00002529 int flags = 0;
2530 Py_ssize_t recvlen, outlen;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002531
Antoine Pitrou19467d22010-08-17 19:33:30 +00002532 if (!PyArg_ParseTuple(args, "n|i:recvfrom", &recvlen, &flags))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002533 return NULL;
Guido van Rossum48a680c2001-03-02 06:34:14 +00002534
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002535 if (recvlen < 0) {
2536 PyErr_SetString(PyExc_ValueError,
2537 "negative buffersize in recvfrom");
2538 return NULL;
2539 }
Guido van Rossumd8faa362007-04-27 19:54:29 +00002540
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002541 buf = PyBytes_FromStringAndSize((char *) 0, recvlen);
2542 if (buf == NULL)
2543 return NULL;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002544
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002545 outlen = sock_recvfrom_guts(s, PyBytes_AS_STRING(buf),
2546 recvlen, flags, &addr);
2547 if (outlen < 0) {
2548 goto finally;
2549 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002550
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002551 if (outlen != recvlen) {
2552 /* We did not read as many bytes as we anticipated, resize the
2553 string if possible and be succesful. */
2554 if (_PyBytes_Resize(&buf, outlen) < 0)
2555 /* Oopsy, not so succesful after all. */
2556 goto finally;
2557 }
Barry Warsaw752300b1997-01-03 17:18:10 +00002558
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002559 ret = PyTuple_Pack(2, buf, addr);
Guido van Rossum67f7a382002-06-06 21:08:16 +00002560
2561finally:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002562 Py_XDECREF(buf);
2563 Py_XDECREF(addr);
2564 return ret;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002565}
2566
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002567PyDoc_STRVAR(recvfrom_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002568"recvfrom(buffersize[, flags]) -> (data, address info)\n\
2569\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002570Like recv(buffersize, flags) but also return the sender's address info.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002571
Thomas Wouters477c8d52006-05-27 19:21:47 +00002572
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002573/* s.recvfrom_into(buffer[, nbytes [,flags]]) method */
Thomas Wouters477c8d52006-05-27 19:21:47 +00002574
2575static PyObject *
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002576sock_recvfrom_into(PySocketSockObject *s, PyObject *args, PyObject* kwds)
Thomas Wouters477c8d52006-05-27 19:21:47 +00002577{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002578 static char *kwlist[] = {"buffer", "nbytes", "flags", 0};
Thomas Wouters477c8d52006-05-27 19:21:47 +00002579
Antoine Pitrou19467d22010-08-17 19:33:30 +00002580 int flags = 0;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002581 Py_buffer pbuf;
2582 char *buf;
Antoine Pitrou19467d22010-08-17 19:33:30 +00002583 Py_ssize_t readlen, buflen, recvlen = 0;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002584
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002585 PyObject *addr = NULL;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002586
Antoine Pitrou19467d22010-08-17 19:33:30 +00002587 if (!PyArg_ParseTupleAndKeywords(args, kwds, "w*|ni:recvfrom_into",
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002588 kwlist, &pbuf,
2589 &recvlen, &flags))
2590 return NULL;
2591 buf = pbuf.buf;
2592 buflen = pbuf.len;
2593 assert(buf != 0 && buflen > 0);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002594
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002595 if (recvlen < 0) {
2596 PyBuffer_Release(&pbuf);
2597 PyErr_SetString(PyExc_ValueError,
2598 "negative buffersize in recvfrom_into");
2599 return NULL;
2600 }
2601 if (recvlen == 0) {
2602 /* If nbytes was not specified, use the buffer's length */
2603 recvlen = buflen;
2604 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002605
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002606 readlen = sock_recvfrom_guts(s, buf, recvlen, flags, &addr);
2607 if (readlen < 0) {
2608 PyBuffer_Release(&pbuf);
2609 /* Return an error */
2610 Py_XDECREF(addr);
2611 return NULL;
2612 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002613
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002614 PyBuffer_Release(&pbuf);
2615 /* Return the number of bytes read and the address. Note that we do
2616 not do anything special here in the case that readlen < recvlen. */
Antoine Pitrou19467d22010-08-17 19:33:30 +00002617 return Py_BuildValue("nN", readlen, addr);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002618}
2619
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002620PyDoc_STRVAR(recvfrom_into_doc,
2621"recvfrom_into(buffer[, nbytes[, flags]]) -> (nbytes, address info)\n\
Thomas Wouters477c8d52006-05-27 19:21:47 +00002622\n\
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002623Like recv_into(buffer[, nbytes[, flags]]) but also return the sender's address info.");
Thomas Wouters477c8d52006-05-27 19:21:47 +00002624
2625
Guido van Rossumeb6b33a1993-05-25 09:38:27 +00002626/* s.send(data [,flags]) method */
Guido van Rossum30a685f1991-06-27 15:51:29 +00002627
Guido van Rossum73624e91994-10-10 17:59:00 +00002628static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002629sock_send(PySocketSockObject *s, PyObject *args)
Guido van Rossum30a685f1991-06-27 15:51:29 +00002630{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002631 char *buf;
Antoine Pitrou19467d22010-08-17 19:33:30 +00002632 Py_ssize_t len, n = -1;
2633 int flags = 0, timeout;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002634 Py_buffer pbuf;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002635
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002636 if (!PyArg_ParseTuple(args, "y*|i:send", &pbuf, &flags))
2637 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002638
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002639 if (!IS_SELECTABLE(s)) {
2640 PyBuffer_Release(&pbuf);
2641 return select_error();
2642 }
2643 buf = pbuf.buf;
2644 len = pbuf.len;
Neal Norwitz082b2df2006-02-07 07:04:46 +00002645
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002646 BEGIN_SELECT_LOOP(s)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002647 Py_BEGIN_ALLOW_THREADS
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002648 timeout = internal_select_ex(s, 1, interval);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002649 if (!timeout)
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002650#ifdef __VMS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002651 n = sendsegmented(s->sock_fd, buf, len, flags);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002652#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002653 n = send(s->sock_fd, buf, len, flags);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002654#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002655 Py_END_ALLOW_THREADS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002656 if (timeout == 1) {
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002657 PyBuffer_Release(&pbuf);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002658 PyErr_SetString(socket_timeout, "timed out");
2659 return NULL;
2660 }
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002661 END_SELECT_LOOP(s)
2662
2663 PyBuffer_Release(&pbuf);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002664 if (n < 0)
2665 return s->errorhandler();
Antoine Pitrou19467d22010-08-17 19:33:30 +00002666 return PyLong_FromSsize_t(n);
Guido van Rossum30a685f1991-06-27 15:51:29 +00002667}
2668
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002669PyDoc_STRVAR(send_doc,
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002670"send(data[, flags]) -> count\n\
Guido van Rossum82a5c661998-07-07 20:45:43 +00002671\n\
2672Send a data string to the socket. For the optional flags\n\
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002673argument, see the Unix manual. Return the number of bytes\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002674sent; this may be less than len(data) if the network is busy.");
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002675
2676
2677/* s.sendall(data [,flags]) method */
2678
2679static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002680sock_sendall(PySocketSockObject *s, PyObject *args)
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002681{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002682 char *buf;
Antoine Pitrou19467d22010-08-17 19:33:30 +00002683 Py_ssize_t len, n = -1;
Antoine Pitrou6d7df632010-09-27 17:52:25 +00002684 int flags = 0, timeout, saved_errno;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002685 Py_buffer pbuf;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002686
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002687 if (!PyArg_ParseTuple(args, "y*|i:sendall", &pbuf, &flags))
2688 return NULL;
2689 buf = pbuf.buf;
2690 len = pbuf.len;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002691
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002692 if (!IS_SELECTABLE(s)) {
2693 PyBuffer_Release(&pbuf);
2694 return select_error();
2695 }
Neal Norwitz082b2df2006-02-07 07:04:46 +00002696
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002697 do {
Antoine Pitrou6d7df632010-09-27 17:52:25 +00002698 Py_BEGIN_ALLOW_THREADS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002699 timeout = internal_select(s, 1);
2700 n = -1;
Antoine Pitrou6d7df632010-09-27 17:52:25 +00002701 if (!timeout) {
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002702#ifdef __VMS
Antoine Pitrou6d7df632010-09-27 17:52:25 +00002703 n = sendsegmented(s->sock_fd, buf, len, flags);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002704#else
Antoine Pitrou6d7df632010-09-27 17:52:25 +00002705 n = send(s->sock_fd, buf, len, flags);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002706#endif
Antoine Pitrou6d7df632010-09-27 17:52:25 +00002707 }
2708 Py_END_ALLOW_THREADS
2709 if (timeout == 1) {
2710 PyBuffer_Release(&pbuf);
2711 PyErr_SetString(socket_timeout, "timed out");
2712 return NULL;
2713 }
2714 /* PyErr_CheckSignals() might change errno */
2715 saved_errno = errno;
2716 /* We must run our signal handlers before looping again.
2717 send() can return a successful partial write when it is
2718 interrupted, so we can't restrict ourselves to EINTR. */
2719 if (PyErr_CheckSignals()) {
2720 PyBuffer_Release(&pbuf);
2721 return NULL;
2722 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002723 if (n < 0) {
Antoine Pitrou6d7df632010-09-27 17:52:25 +00002724 /* If interrupted, try again */
2725 if (saved_errno == EINTR)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002726 continue;
Antoine Pitrou6d7df632010-09-27 17:52:25 +00002727 else
2728 break;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002729 }
2730 buf += n;
2731 len -= n;
2732 } while (len > 0);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002733 PyBuffer_Release(&pbuf);
Guido van Rossum67f7a382002-06-06 21:08:16 +00002734
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002735 if (n < 0)
2736 return s->errorhandler();
Guido van Rossum67f7a382002-06-06 21:08:16 +00002737
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002738 Py_INCREF(Py_None);
2739 return Py_None;
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002740}
2741
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002742PyDoc_STRVAR(sendall_doc,
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002743"sendall(data[, flags])\n\
2744\n\
2745Send a data string to the socket. For the optional flags\n\
2746argument, see the Unix manual. This calls send() repeatedly\n\
2747until all data is sent. If an error occurs, it's impossible\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002748to tell how much data has been sent.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002749
Guido van Rossum30a685f1991-06-27 15:51:29 +00002750
Guido van Rossumeb6b33a1993-05-25 09:38:27 +00002751/* s.sendto(data, [flags,] sockaddr) method */
Guido van Rossum30a685f1991-06-27 15:51:29 +00002752
Guido van Rossum73624e91994-10-10 17:59:00 +00002753static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002754sock_sendto(PySocketSockObject *s, PyObject *args)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002755{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002756 Py_buffer pbuf;
2757 PyObject *addro;
2758 char *buf;
2759 Py_ssize_t len;
2760 sock_addr_t addrbuf;
2761 int addrlen, n = -1, flags, timeout;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002762
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002763 flags = 0;
2764 if (!PyArg_ParseTuple(args, "y*O:sendto", &pbuf, &addro)) {
2765 PyErr_Clear();
2766 if (!PyArg_ParseTuple(args, "y*iO:sendto",
2767 &pbuf, &flags, &addro))
2768 return NULL;
2769 }
2770 buf = pbuf.buf;
2771 len = pbuf.len;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002772
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002773 if (!IS_SELECTABLE(s)) {
2774 PyBuffer_Release(&pbuf);
2775 return select_error();
2776 }
Neal Norwitz082b2df2006-02-07 07:04:46 +00002777
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002778 if (!getsockaddrarg(s, addro, SAS2SA(&addrbuf), &addrlen)) {
2779 PyBuffer_Release(&pbuf);
2780 return NULL;
2781 }
Thomas Wouters89f507f2006-12-13 04:49:30 +00002782
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002783 BEGIN_SELECT_LOOP(s)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002784 Py_BEGIN_ALLOW_THREADS
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002785 timeout = internal_select_ex(s, 1, interval);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002786 if (!timeout)
2787 n = sendto(s->sock_fd, buf, len, flags, SAS2SA(&addrbuf), addrlen);
2788 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00002789
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002790 if (timeout == 1) {
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002791 PyBuffer_Release(&pbuf);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002792 PyErr_SetString(socket_timeout, "timed out");
2793 return NULL;
2794 }
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002795 END_SELECT_LOOP(s)
2796 PyBuffer_Release(&pbuf);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002797 if (n < 0)
2798 return s->errorhandler();
Antoine Pitrou19467d22010-08-17 19:33:30 +00002799 return PyLong_FromSsize_t(n);
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002800}
2801
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002802PyDoc_STRVAR(sendto_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00002803"sendto(data[, flags], address) -> count\n\
Guido van Rossum82a5c661998-07-07 20:45:43 +00002804\n\
2805Like send(data, flags) but allows specifying the destination address.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002806For IP sockets, the address is a pair (hostaddr, port).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002807
Guido van Rossum30a685f1991-06-27 15:51:29 +00002808
2809/* s.shutdown(how) method */
2810
Guido van Rossum73624e91994-10-10 17:59:00 +00002811static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002812sock_shutdown(PySocketSockObject *s, PyObject *arg)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002813{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002814 int how;
2815 int res;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00002816
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002817 how = PyLong_AsLong(arg);
2818 if (how == -1 && PyErr_Occurred())
2819 return NULL;
2820 Py_BEGIN_ALLOW_THREADS
2821 res = shutdown(s->sock_fd, how);
2822 Py_END_ALLOW_THREADS
2823 if (res < 0)
2824 return s->errorhandler();
2825 Py_INCREF(Py_None);
2826 return Py_None;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002827}
2828
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002829PyDoc_STRVAR(shutdown_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002830"shutdown(flag)\n\
2831\n\
Martin v. Löwis94681fc2003-11-27 19:40:22 +00002832Shut down the reading side of the socket (flag == SHUT_RD), the writing side\n\
2833of the socket (flag == SHUT_WR), or both ends (flag == SHUT_RDWR).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002834
Amaury Forgeot d'Arc3d17a5c2008-06-13 01:09:34 +00002835#if defined(MS_WINDOWS) && defined(SIO_RCVALL)
Christian Heimesfaf2f632008-01-06 16:59:19 +00002836static PyObject*
2837sock_ioctl(PySocketSockObject *s, PyObject *arg)
2838{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002839 unsigned long cmd = SIO_RCVALL;
2840 PyObject *argO;
2841 DWORD recv;
Christian Heimesfaf2f632008-01-06 16:59:19 +00002842
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002843 if (!PyArg_ParseTuple(arg, "kO:ioctl", &cmd, &argO))
2844 return NULL;
Christian Heimesfaf2f632008-01-06 16:59:19 +00002845
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002846 switch (cmd) {
2847 case SIO_RCVALL: {
2848 unsigned int option = RCVALL_ON;
2849 if (!PyArg_ParseTuple(arg, "kI:ioctl", &cmd, &option))
2850 return NULL;
2851 if (WSAIoctl(s->sock_fd, cmd, &option, sizeof(option),
2852 NULL, 0, &recv, NULL, NULL) == SOCKET_ERROR) {
2853 return set_error();
2854 }
2855 return PyLong_FromUnsignedLong(recv); }
2856 case SIO_KEEPALIVE_VALS: {
2857 struct tcp_keepalive ka;
2858 if (!PyArg_ParseTuple(arg, "k(kkk):ioctl", &cmd,
2859 &ka.onoff, &ka.keepalivetime, &ka.keepaliveinterval))
2860 return NULL;
2861 if (WSAIoctl(s->sock_fd, cmd, &ka, sizeof(ka),
2862 NULL, 0, &recv, NULL, NULL) == SOCKET_ERROR) {
2863 return set_error();
2864 }
2865 return PyLong_FromUnsignedLong(recv); }
2866 default:
2867 PyErr_Format(PyExc_ValueError, "invalid ioctl command %d", cmd);
2868 return NULL;
2869 }
Christian Heimesfaf2f632008-01-06 16:59:19 +00002870}
2871PyDoc_STRVAR(sock_ioctl_doc,
2872"ioctl(cmd, option) -> long\n\
2873\n\
Kristján Valur Jónsson847ec752009-09-27 21:10:38 +00002874Control the socket with WSAIoctl syscall. Currently supported 'cmd' values are\n\
2875SIO_RCVALL: 'option' must be one of the socket.RCVALL_* constants.\n\
2876SIO_KEEPALIVE_VALS: 'option' is a tuple of (onoff, timeout, interval).");
Christian Heimesfaf2f632008-01-06 16:59:19 +00002877
2878#endif
Guido van Rossum30a685f1991-06-27 15:51:29 +00002879
2880/* List of methods for socket objects */
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002881
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002882static PyMethodDef sock_methods[] = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002883 {"_accept", (PyCFunction)sock_accept, METH_NOARGS,
2884 accept_doc},
2885 {"bind", (PyCFunction)sock_bind, METH_O,
2886 bind_doc},
2887 {"close", (PyCFunction)sock_close, METH_NOARGS,
2888 close_doc},
2889 {"connect", (PyCFunction)sock_connect, METH_O,
2890 connect_doc},
2891 {"connect_ex", (PyCFunction)sock_connect_ex, METH_O,
2892 connect_ex_doc},
Antoine Pitrou6e451df2010-08-09 20:39:54 +00002893 {"detach", (PyCFunction)sock_detach, METH_NOARGS,
2894 detach_doc},
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002895 {"fileno", (PyCFunction)sock_fileno, METH_NOARGS,
2896 fileno_doc},
Guido van Rossumb6775db1994-08-01 11:34:53 +00002897#ifdef HAVE_GETPEERNAME
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002898 {"getpeername", (PyCFunction)sock_getpeername,
2899 METH_NOARGS, getpeername_doc},
Guido van Rossum9575a441993-04-07 14:06:14 +00002900#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002901 {"getsockname", (PyCFunction)sock_getsockname,
2902 METH_NOARGS, getsockname_doc},
2903 {"getsockopt", (PyCFunction)sock_getsockopt, METH_VARARGS,
2904 getsockopt_doc},
Amaury Forgeot d'Arc3d17a5c2008-06-13 01:09:34 +00002905#if defined(MS_WINDOWS) && defined(SIO_RCVALL)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002906 {"ioctl", (PyCFunction)sock_ioctl, METH_VARARGS,
2907 sock_ioctl_doc},
Christian Heimesfaf2f632008-01-06 16:59:19 +00002908#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002909 {"listen", (PyCFunction)sock_listen, METH_O,
2910 listen_doc},
2911 {"recv", (PyCFunction)sock_recv, METH_VARARGS,
2912 recv_doc},
2913 {"recv_into", (PyCFunction)sock_recv_into, METH_VARARGS | METH_KEYWORDS,
2914 recv_into_doc},
2915 {"recvfrom", (PyCFunction)sock_recvfrom, METH_VARARGS,
2916 recvfrom_doc},
2917 {"recvfrom_into", (PyCFunction)sock_recvfrom_into, METH_VARARGS | METH_KEYWORDS,
2918 recvfrom_into_doc},
2919 {"send", (PyCFunction)sock_send, METH_VARARGS,
2920 send_doc},
2921 {"sendall", (PyCFunction)sock_sendall, METH_VARARGS,
2922 sendall_doc},
2923 {"sendto", (PyCFunction)sock_sendto, METH_VARARGS,
2924 sendto_doc},
2925 {"setblocking", (PyCFunction)sock_setblocking, METH_O,
2926 setblocking_doc},
2927 {"settimeout", (PyCFunction)sock_settimeout, METH_O,
2928 settimeout_doc},
2929 {"gettimeout", (PyCFunction)sock_gettimeout, METH_NOARGS,
2930 gettimeout_doc},
2931 {"setsockopt", (PyCFunction)sock_setsockopt, METH_VARARGS,
2932 setsockopt_doc},
2933 {"shutdown", (PyCFunction)sock_shutdown, METH_O,
2934 shutdown_doc},
2935 {NULL, NULL} /* sentinel */
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002936};
2937
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002938/* SockObject members */
2939static PyMemberDef sock_memberlist[] = {
2940 {"family", T_INT, offsetof(PySocketSockObject, sock_family), READONLY, "the socket family"},
2941 {"type", T_INT, offsetof(PySocketSockObject, sock_type), READONLY, "the socket type"},
2942 {"proto", T_INT, offsetof(PySocketSockObject, sock_proto), READONLY, "the socket protocol"},
2943 {"timeout", T_DOUBLE, offsetof(PySocketSockObject, sock_timeout), READONLY, "the socket timeout"},
2944 {0},
2945};
Guido van Rossum30a685f1991-06-27 15:51:29 +00002946
Guido van Rossum73624e91994-10-10 17:59:00 +00002947/* Deallocate a socket object in response to the last Py_DECREF().
Guido van Rossum30a685f1991-06-27 15:51:29 +00002948 First close the file description. */
2949
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002950static void
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002951sock_dealloc(PySocketSockObject *s)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002952{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002953 if (s->sock_fd != -1)
2954 (void) SOCKETCLOSE(s->sock_fd);
2955 Py_TYPE(s)->tp_free((PyObject *)s);
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002956}
2957
Guido van Rossum30a685f1991-06-27 15:51:29 +00002958
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00002959static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002960sock_repr(PySocketSockObject *s)
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00002961{
Fred Drakea04eaad2000-06-30 02:46:07 +00002962#if SIZEOF_SOCKET_T > SIZEOF_LONG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002963 if (s->sock_fd > LONG_MAX) {
2964 /* this can occur on Win64, and actually there is a special
2965 ugly printf formatter for decimal pointer length integer
2966 printing, only bother if necessary*/
2967 PyErr_SetString(PyExc_OverflowError,
2968 "no printf formatter to display "
2969 "the socket descriptor in decimal");
2970 return NULL;
2971 }
Fred Drakea04eaad2000-06-30 02:46:07 +00002972#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002973 return PyUnicode_FromFormat(
2974 "<socket object, fd=%ld, family=%d, type=%d, proto=%d>",
2975 (long)s->sock_fd, s->sock_family,
2976 s->sock_type,
2977 s->sock_proto);
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00002978}
2979
2980
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002981/* Create a new, uninitialized socket object. */
2982
2983static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002984sock_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002985{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002986 PyObject *new;
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002987
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002988 new = type->tp_alloc(type, 0);
2989 if (new != NULL) {
2990 ((PySocketSockObject *)new)->sock_fd = -1;
2991 ((PySocketSockObject *)new)->sock_timeout = -1.0;
2992 ((PySocketSockObject *)new)->errorhandler = &set_error;
2993 }
2994 return new;
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002995}
2996
2997
2998/* Initialize a new socket object. */
2999
3000/*ARGSUSED*/
3001static int
Andrew MacIntyre7aec4a22002-06-13 11:53:52 +00003002sock_initobj(PyObject *self, PyObject *args, PyObject *kwds)
Guido van Rossum384ca9c2001-10-27 22:20:47 +00003003{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003004 PySocketSockObject *s = (PySocketSockObject *)self;
3005 PyObject *fdobj = NULL;
3006 SOCKET_T fd = INVALID_SOCKET;
3007 int family = AF_INET, type = SOCK_STREAM, proto = 0;
3008 static char *keywords[] = {"family", "type", "proto", "fileno", 0};
Guido van Rossum384ca9c2001-10-27 22:20:47 +00003009
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003010 if (!PyArg_ParseTupleAndKeywords(args, kwds,
3011 "|iiiO:socket", keywords,
3012 &family, &type, &proto, &fdobj))
3013 return -1;
Guido van Rossum67f7a382002-06-06 21:08:16 +00003014
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003015 if (fdobj != NULL && fdobj != Py_None) {
3016 fd = PyLong_AsSocket_t(fdobj);
3017 if (fd == (SOCKET_T)(-1) && PyErr_Occurred())
3018 return -1;
3019 if (fd == INVALID_SOCKET) {
3020 PyErr_SetString(PyExc_ValueError,
3021 "can't use invalid socket value");
3022 return -1;
3023 }
3024 }
3025 else {
3026 Py_BEGIN_ALLOW_THREADS
3027 fd = socket(family, type, proto);
3028 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00003029
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003030 if (fd == INVALID_SOCKET) {
3031 set_error();
3032 return -1;
3033 }
3034 }
3035 init_sockobject(s, fd, family, type, proto);
Guido van Rossum67f7a382002-06-06 21:08:16 +00003036
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003037 return 0;
Guido van Rossum67f7a382002-06-06 21:08:16 +00003038
Guido van Rossum384ca9c2001-10-27 22:20:47 +00003039}
3040
3041
Guido van Rossumb6775db1994-08-01 11:34:53 +00003042/* Type object for socket objects. */
Guido van Rossum30a685f1991-06-27 15:51:29 +00003043
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003044static PyTypeObject sock_type = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003045 PyVarObject_HEAD_INIT(0, 0) /* Must fill in type value later */
3046 "_socket.socket", /* tp_name */
3047 sizeof(PySocketSockObject), /* tp_basicsize */
3048 0, /* tp_itemsize */
3049 (destructor)sock_dealloc, /* tp_dealloc */
3050 0, /* tp_print */
3051 0, /* tp_getattr */
3052 0, /* tp_setattr */
3053 0, /* tp_reserved */
3054 (reprfunc)sock_repr, /* tp_repr */
3055 0, /* tp_as_number */
3056 0, /* tp_as_sequence */
3057 0, /* tp_as_mapping */
3058 0, /* tp_hash */
3059 0, /* tp_call */
3060 0, /* tp_str */
3061 PyObject_GenericGetAttr, /* tp_getattro */
3062 0, /* tp_setattro */
3063 0, /* tp_as_buffer */
3064 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE, /* tp_flags */
3065 sock_doc, /* tp_doc */
3066 0, /* tp_traverse */
3067 0, /* tp_clear */
3068 0, /* tp_richcompare */
3069 0, /* tp_weaklistoffset */
3070 0, /* tp_iter */
3071 0, /* tp_iternext */
3072 sock_methods, /* tp_methods */
3073 sock_memberlist, /* tp_members */
3074 0, /* tp_getset */
3075 0, /* tp_base */
3076 0, /* tp_dict */
3077 0, /* tp_descr_get */
3078 0, /* tp_descr_set */
3079 0, /* tp_dictoffset */
3080 sock_initobj, /* tp_init */
3081 PyType_GenericAlloc, /* tp_alloc */
3082 sock_new, /* tp_new */
3083 PyObject_Del, /* tp_free */
Guido van Rossum6574b3e1991-06-25 21:36:08 +00003084};
3085
Guido van Rossum30a685f1991-06-27 15:51:29 +00003086
Guido van Rossum81194471991-07-27 21:42:02 +00003087/* Python interface to gethostname(). */
3088
3089/*ARGSUSED*/
Guido van Rossum73624e91994-10-10 17:59:00 +00003090static PyObject *
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00003091socket_gethostname(PyObject *self, PyObject *unused)
Guido van Rossum81194471991-07-27 21:42:02 +00003092{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003093 char buf[1024];
3094 int res;
3095 Py_BEGIN_ALLOW_THREADS
3096 res = gethostname(buf, (int) sizeof buf - 1);
3097 Py_END_ALLOW_THREADS
3098 if (res < 0)
3099 return set_error();
3100 buf[sizeof buf - 1] = '\0';
3101 return PyUnicode_FromString(buf);
Guido van Rossum81194471991-07-27 21:42:02 +00003102}
Guido van Rossumff4949e1992-08-05 19:58:53 +00003103
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003104PyDoc_STRVAR(gethostname_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003105"gethostname() -> string\n\
3106\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003107Return the current host name.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003108
Guido van Rossumff4949e1992-08-05 19:58:53 +00003109
Guido van Rossum30a685f1991-06-27 15:51:29 +00003110/* Python interface to gethostbyname(name). */
3111
3112/*ARGSUSED*/
Guido van Rossum73624e91994-10-10 17:59:00 +00003113static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003114socket_gethostbyname(PyObject *self, PyObject *args)
Guido van Rossum30a685f1991-06-27 15:51:29 +00003115{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003116 char *name;
3117 sock_addr_t addrbuf;
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003118 PyObject *ret = NULL;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003119
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003120 if (!PyArg_ParseTuple(args, "et:gethostbyname", "idna", &name))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003121 return NULL;
3122 if (setipaddr(name, SAS2SA(&addrbuf), sizeof(addrbuf), AF_INET) < 0)
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003123 goto finally;
3124 ret = makeipaddr(SAS2SA(&addrbuf), sizeof(struct sockaddr_in));
3125finally:
3126 PyMem_Free(name);
3127 return ret;
Guido van Rossum30a685f1991-06-27 15:51:29 +00003128}
3129
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003130PyDoc_STRVAR(gethostbyname_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003131"gethostbyname(host) -> address\n\
3132\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003133Return the IP address (a string of the form '255.255.255.255') for a host.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003134
3135
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003136/* Convenience function common to gethostbyname_ex and gethostbyaddr */
3137
3138static PyObject *
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003139gethost_common(struct hostent *h, struct sockaddr *addr, int alen, int af)
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003140{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003141 char **pch;
3142 PyObject *rtn_tuple = (PyObject *)NULL;
3143 PyObject *name_list = (PyObject *)NULL;
3144 PyObject *addr_list = (PyObject *)NULL;
3145 PyObject *tmp;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003146
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003147 if (h == NULL) {
3148 /* Let's get real error message to return */
3149 set_herror(h_errno);
3150 return NULL;
3151 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00003152
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003153 if (h->h_addrtype != af) {
3154 /* Let's get real error message to return */
3155 PyErr_SetString(socket_error,
3156 (char *)strerror(EAFNOSUPPORT));
Christian Heimesada8c3b2008-03-18 18:26:33 +00003157
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003158 return NULL;
3159 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00003160
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003161 switch (af) {
Guido van Rossum67f7a382002-06-06 21:08:16 +00003162
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003163 case AF_INET:
3164 if (alen < sizeof(struct sockaddr_in))
3165 return NULL;
3166 break;
Guido van Rossum67f7a382002-06-06 21:08:16 +00003167
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00003168#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003169 case AF_INET6:
3170 if (alen < sizeof(struct sockaddr_in6))
3171 return NULL;
3172 break;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003173#endif
Guido van Rossum67f7a382002-06-06 21:08:16 +00003174
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003175 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00003176
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003177 if ((name_list = PyList_New(0)) == NULL)
3178 goto err;
Guido van Rossum67f7a382002-06-06 21:08:16 +00003179
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003180 if ((addr_list = PyList_New(0)) == NULL)
3181 goto err;
Guido van Rossum67f7a382002-06-06 21:08:16 +00003182
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003183 /* SF #1511317: h_aliases can be NULL */
3184 if (h->h_aliases) {
3185 for (pch = h->h_aliases; *pch != NULL; pch++) {
3186 int status;
3187 tmp = PyUnicode_FromString(*pch);
3188 if (tmp == NULL)
3189 goto err;
Guido van Rossum67f7a382002-06-06 21:08:16 +00003190
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003191 status = PyList_Append(name_list, tmp);
3192 Py_DECREF(tmp);
Guido van Rossum67f7a382002-06-06 21:08:16 +00003193
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003194 if (status)
3195 goto err;
3196 }
3197 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00003198
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003199 for (pch = h->h_addr_list; *pch != NULL; pch++) {
3200 int status;
Guido van Rossum67f7a382002-06-06 21:08:16 +00003201
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003202 switch (af) {
Guido van Rossum67f7a382002-06-06 21:08:16 +00003203
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003204 case AF_INET:
3205 {
3206 struct sockaddr_in sin;
3207 memset(&sin, 0, sizeof(sin));
3208 sin.sin_family = af;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003209#ifdef HAVE_SOCKADDR_SA_LEN
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003210 sin.sin_len = sizeof(sin);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003211#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003212 memcpy(&sin.sin_addr, *pch, sizeof(sin.sin_addr));
3213 tmp = makeipaddr((struct sockaddr *)&sin, sizeof(sin));
Guido van Rossum67f7a382002-06-06 21:08:16 +00003214
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003215 if (pch == h->h_addr_list && alen >= sizeof(sin))
3216 memcpy((char *) addr, &sin, sizeof(sin));
3217 break;
3218 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00003219
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00003220#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003221 case AF_INET6:
3222 {
3223 struct sockaddr_in6 sin6;
3224 memset(&sin6, 0, sizeof(sin6));
3225 sin6.sin6_family = af;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003226#ifdef HAVE_SOCKADDR_SA_LEN
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003227 sin6.sin6_len = sizeof(sin6);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003228#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003229 memcpy(&sin6.sin6_addr, *pch, sizeof(sin6.sin6_addr));
3230 tmp = makeipaddr((struct sockaddr *)&sin6,
3231 sizeof(sin6));
Guido van Rossum67f7a382002-06-06 21:08:16 +00003232
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003233 if (pch == h->h_addr_list && alen >= sizeof(sin6))
3234 memcpy((char *) addr, &sin6, sizeof(sin6));
3235 break;
3236 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003237#endif
Guido van Rossum67f7a382002-06-06 21:08:16 +00003238
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003239 default: /* can't happen */
3240 PyErr_SetString(socket_error,
3241 "unsupported address family");
3242 return NULL;
3243 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00003244
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003245 if (tmp == NULL)
3246 goto err;
Guido van Rossum67f7a382002-06-06 21:08:16 +00003247
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003248 status = PyList_Append(addr_list, tmp);
3249 Py_DECREF(tmp);
Guido van Rossum67f7a382002-06-06 21:08:16 +00003250
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003251 if (status)
3252 goto err;
3253 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00003254
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003255 rtn_tuple = Py_BuildValue("sOO", h->h_name, name_list, addr_list);
Guido van Rossum67f7a382002-06-06 21:08:16 +00003256
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003257 err:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003258 Py_XDECREF(name_list);
3259 Py_XDECREF(addr_list);
3260 return rtn_tuple;
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003261}
3262
3263
3264/* Python interface to gethostbyname_ex(name). */
3265
3266/*ARGSUSED*/
3267static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003268socket_gethostbyname_ex(PyObject *self, PyObject *args)
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003269{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003270 char *name;
3271 struct hostent *h;
Martin v. Löwisc16f3bd2003-05-03 09:14:54 +00003272#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003273 struct sockaddr_storage addr;
Martin v. Löwisc16f3bd2003-05-03 09:14:54 +00003274#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003275 struct sockaddr_in addr;
Martin v. Löwisc16f3bd2003-05-03 09:14:54 +00003276#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003277 struct sockaddr *sa;
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003278 PyObject *ret = NULL;
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003279#ifdef HAVE_GETHOSTBYNAME_R
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003280 struct hostent hp_allocated;
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003281#ifdef HAVE_GETHOSTBYNAME_R_3_ARG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003282 struct hostent_data data;
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003283#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003284 char buf[16384];
3285 int buf_len = (sizeof buf) - 1;
3286 int errnop;
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003287#endif
3288#if defined(HAVE_GETHOSTBYNAME_R_3_ARG) || defined(HAVE_GETHOSTBYNAME_R_6_ARG)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003289 int result;
Guido van Rossume9cd07b1999-03-15 21:40:14 +00003290#endif
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003291#endif /* HAVE_GETHOSTBYNAME_R */
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003292
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003293 if (!PyArg_ParseTuple(args, "et:gethostbyname_ex", "idna", &name))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003294 return NULL;
3295 if (setipaddr(name, (struct sockaddr *)&addr, sizeof(addr), AF_INET) < 0)
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003296 goto finally;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003297 Py_BEGIN_ALLOW_THREADS
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003298#ifdef HAVE_GETHOSTBYNAME_R
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003299#if defined(HAVE_GETHOSTBYNAME_R_6_ARG)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003300 result = gethostbyname_r(name, &hp_allocated, buf, buf_len,
3301 &h, &errnop);
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003302#elif defined(HAVE_GETHOSTBYNAME_R_5_ARG)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003303 h = gethostbyname_r(name, &hp_allocated, buf, buf_len, &errnop);
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003304#else /* HAVE_GETHOSTBYNAME_R_3_ARG */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003305 memset((void *) &data, '\0', sizeof(data));
3306 result = gethostbyname_r(name, &hp_allocated, &data);
3307 h = (result != 0) ? NULL : &hp_allocated;
Guido van Rossume9cd07b1999-03-15 21:40:14 +00003308#endif
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003309#else /* not HAVE_GETHOSTBYNAME_R */
Guido van Rossum3baaa131999-03-22 21:44:51 +00003310#ifdef USE_GETHOSTBYNAME_LOCK
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003311 PyThread_acquire_lock(netdb_lock, 1);
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003312#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003313 h = gethostbyname(name);
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003314#endif /* HAVE_GETHOSTBYNAME_R */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003315 Py_END_ALLOW_THREADS
3316 /* Some C libraries would require addr.__ss_family instead of
3317 addr.ss_family.
3318 Therefore, we cast the sockaddr_storage into sockaddr to
3319 access sa_family. */
3320 sa = (struct sockaddr*)&addr;
3321 ret = gethost_common(h, (struct sockaddr *)&addr, sizeof(addr),
3322 sa->sa_family);
Guido van Rossum3baaa131999-03-22 21:44:51 +00003323#ifdef USE_GETHOSTBYNAME_LOCK
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003324 PyThread_release_lock(netdb_lock);
Guido van Rossum955becc1999-03-22 20:14:53 +00003325#endif
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003326finally:
3327 PyMem_Free(name);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003328 return ret;
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003329}
3330
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003331PyDoc_STRVAR(ghbn_ex_doc,
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003332"gethostbyname_ex(host) -> (name, aliaslist, addresslist)\n\
3333\n\
3334Return the true host name, a list of aliases, and a list of IP addresses,\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003335for a host. The host argument is a string giving a host name or IP number.");
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003336
3337
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00003338/* Python interface to gethostbyaddr(IP). */
3339
3340/*ARGSUSED*/
3341static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003342socket_gethostbyaddr(PyObject *self, PyObject *args)
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00003343{
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00003344#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003345 struct sockaddr_storage addr;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003346#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003347 struct sockaddr_in addr;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003348#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003349 struct sockaddr *sa = (struct sockaddr *)&addr;
3350 char *ip_num;
3351 struct hostent *h;
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003352 PyObject *ret = NULL;
Guido van Rossum4f199ea1998-04-09 20:56:35 +00003353#ifdef HAVE_GETHOSTBYNAME_R
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003354 struct hostent hp_allocated;
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003355#ifdef HAVE_GETHOSTBYNAME_R_3_ARG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003356 struct hostent_data data;
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003357#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003358 /* glibcs up to 2.10 assume that the buf argument to
3359 gethostbyaddr_r is 8-byte aligned, which at least llvm-gcc
3360 does not ensure. The attribute below instructs the compiler
3361 to maintain this alignment. */
3362 char buf[16384] Py_ALIGNED(8);
3363 int buf_len = (sizeof buf) - 1;
3364 int errnop;
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003365#endif
3366#if defined(HAVE_GETHOSTBYNAME_R_3_ARG) || defined(HAVE_GETHOSTBYNAME_R_6_ARG)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003367 int result;
Guido van Rossume9cd07b1999-03-15 21:40:14 +00003368#endif
Guido van Rossum4f199ea1998-04-09 20:56:35 +00003369#endif /* HAVE_GETHOSTBYNAME_R */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003370 char *ap;
3371 int al;
3372 int af;
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00003373
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003374 if (!PyArg_ParseTuple(args, "et:gethostbyaddr", "idna", &ip_num))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003375 return NULL;
3376 af = AF_UNSPEC;
3377 if (setipaddr(ip_num, sa, sizeof(addr), af) < 0)
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003378 goto finally;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003379 af = sa->sa_family;
3380 ap = NULL;
3381 al = 0;
3382 switch (af) {
3383 case AF_INET:
3384 ap = (char *)&((struct sockaddr_in *)sa)->sin_addr;
3385 al = sizeof(((struct sockaddr_in *)sa)->sin_addr);
3386 break;
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00003387#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003388 case AF_INET6:
3389 ap = (char *)&((struct sockaddr_in6 *)sa)->sin6_addr;
3390 al = sizeof(((struct sockaddr_in6 *)sa)->sin6_addr);
3391 break;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003392#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003393 default:
3394 PyErr_SetString(socket_error, "unsupported address family");
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003395 goto finally;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003396 }
3397 Py_BEGIN_ALLOW_THREADS
Guido van Rossum4f199ea1998-04-09 20:56:35 +00003398#ifdef HAVE_GETHOSTBYNAME_R
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003399#if defined(HAVE_GETHOSTBYNAME_R_6_ARG)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003400 result = gethostbyaddr_r(ap, al, af,
3401 &hp_allocated, buf, buf_len,
3402 &h, &errnop);
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003403#elif defined(HAVE_GETHOSTBYNAME_R_5_ARG)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003404 h = gethostbyaddr_r(ap, al, af,
3405 &hp_allocated, buf, buf_len, &errnop);
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003406#else /* HAVE_GETHOSTBYNAME_R_3_ARG */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003407 memset((void *) &data, '\0', sizeof(data));
3408 result = gethostbyaddr_r(ap, al, af, &hp_allocated, &data);
3409 h = (result != 0) ? NULL : &hp_allocated;
Guido van Rossume9cd07b1999-03-15 21:40:14 +00003410#endif
Guido van Rossum4f199ea1998-04-09 20:56:35 +00003411#else /* not HAVE_GETHOSTBYNAME_R */
Guido van Rossum3baaa131999-03-22 21:44:51 +00003412#ifdef USE_GETHOSTBYNAME_LOCK
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003413 PyThread_acquire_lock(netdb_lock, 1);
Guido van Rossum4f199ea1998-04-09 20:56:35 +00003414#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003415 h = gethostbyaddr(ap, al, af);
Guido van Rossum4f199ea1998-04-09 20:56:35 +00003416#endif /* HAVE_GETHOSTBYNAME_R */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003417 Py_END_ALLOW_THREADS
3418 ret = gethost_common(h, (struct sockaddr *)&addr, sizeof(addr), af);
Guido van Rossum3baaa131999-03-22 21:44:51 +00003419#ifdef USE_GETHOSTBYNAME_LOCK
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003420 PyThread_release_lock(netdb_lock);
Guido van Rossum3baaa131999-03-22 21:44:51 +00003421#endif
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003422finally:
3423 PyMem_Free(ip_num);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003424 return ret;
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00003425}
3426
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003427PyDoc_STRVAR(gethostbyaddr_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003428"gethostbyaddr(host) -> (name, aliaslist, addresslist)\n\
3429\n\
3430Return the true host name, a list of aliases, and a list of IP addresses,\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003431for a host. The host argument is a string giving a host name or IP number.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003432
Guido van Rossum30a685f1991-06-27 15:51:29 +00003433
3434/* Python interface to getservbyname(name).
3435 This only returns the port number, since the other info is already
3436 known or not useful (like the list of aliases). */
3437
3438/*ARGSUSED*/
Guido van Rossum73624e91994-10-10 17:59:00 +00003439static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003440socket_getservbyname(PyObject *self, PyObject *args)
Guido van Rossum30a685f1991-06-27 15:51:29 +00003441{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003442 char *name, *proto=NULL;
3443 struct servent *sp;
3444 if (!PyArg_ParseTuple(args, "s|s:getservbyname", &name, &proto))
3445 return NULL;
3446 Py_BEGIN_ALLOW_THREADS
3447 sp = getservbyname(name, proto);
3448 Py_END_ALLOW_THREADS
3449 if (sp == NULL) {
3450 PyErr_SetString(socket_error, "service/proto not found");
3451 return NULL;
3452 }
3453 return PyLong_FromLong((long) ntohs(sp->s_port));
Guido van Rossum30a685f1991-06-27 15:51:29 +00003454}
3455
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003456PyDoc_STRVAR(getservbyname_doc,
Barry Warsaw11b91a02004-06-28 00:50:43 +00003457"getservbyname(servicename[, protocolname]) -> integer\n\
Guido van Rossum82a5c661998-07-07 20:45:43 +00003458\n\
3459Return a port number from a service name and protocol name.\n\
Barry Warsaw11b91a02004-06-28 00:50:43 +00003460The optional protocol name, if given, should be 'tcp' or 'udp',\n\
3461otherwise any protocol will match.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003462
Guido van Rossum30a685f1991-06-27 15:51:29 +00003463
Barry Warsaw11b91a02004-06-28 00:50:43 +00003464/* Python interface to getservbyport(port).
3465 This only returns the service name, since the other info is already
3466 known or not useful (like the list of aliases). */
3467
3468/*ARGSUSED*/
3469static PyObject *
3470socket_getservbyport(PyObject *self, PyObject *args)
3471{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003472 int port;
3473 char *proto=NULL;
3474 struct servent *sp;
3475 if (!PyArg_ParseTuple(args, "i|s:getservbyport", &port, &proto))
3476 return NULL;
3477 if (port < 0 || port > 0xffff) {
3478 PyErr_SetString(
3479 PyExc_OverflowError,
3480 "getservbyport: port must be 0-65535.");
3481 return NULL;
3482 }
3483 Py_BEGIN_ALLOW_THREADS
3484 sp = getservbyport(htons((short)port), proto);
3485 Py_END_ALLOW_THREADS
3486 if (sp == NULL) {
3487 PyErr_SetString(socket_error, "port/proto not found");
3488 return NULL;
3489 }
3490 return PyUnicode_FromString(sp->s_name);
Barry Warsaw11b91a02004-06-28 00:50:43 +00003491}
3492
3493PyDoc_STRVAR(getservbyport_doc,
3494"getservbyport(port[, protocolname]) -> string\n\
3495\n\
3496Return the service name from a port number and protocol name.\n\
3497The optional protocol name, if given, should be 'tcp' or 'udp',\n\
3498otherwise any protocol will match.");
3499
Guido van Rossum3901d851996-12-19 16:35:04 +00003500/* Python interface to getprotobyname(name).
3501 This only returns the protocol number, since the other info is
3502 already known or not useful (like the list of aliases). */
3503
3504/*ARGSUSED*/
3505static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003506socket_getprotobyname(PyObject *self, PyObject *args)
Guido van Rossum3901d851996-12-19 16:35:04 +00003507{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003508 char *name;
3509 struct protoent *sp;
3510 if (!PyArg_ParseTuple(args, "s:getprotobyname", &name))
3511 return NULL;
3512 Py_BEGIN_ALLOW_THREADS
3513 sp = getprotobyname(name);
3514 Py_END_ALLOW_THREADS
3515 if (sp == NULL) {
3516 PyErr_SetString(socket_error, "protocol not found");
3517 return NULL;
3518 }
3519 return PyLong_FromLong((long) sp->p_proto);
Guido van Rossum3901d851996-12-19 16:35:04 +00003520}
3521
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003522PyDoc_STRVAR(getprotobyname_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003523"getprotobyname(name) -> integer\n\
3524\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003525Return the protocol number for the named protocol. (Rarely used.)");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003526
Guido van Rossum3901d851996-12-19 16:35:04 +00003527
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00003528#ifndef NO_DUP
3529/* dup() function for socket fds */
3530
3531static PyObject *
3532socket_dup(PyObject *self, PyObject *fdobj)
3533{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003534 SOCKET_T fd, newfd;
3535 PyObject *newfdobj;
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00003536
3537
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003538 fd = PyLong_AsSocket_t(fdobj);
3539 if (fd == (SOCKET_T)(-1) && PyErr_Occurred())
3540 return NULL;
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00003541
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003542 newfd = dup_socket(fd);
3543 if (newfd == INVALID_SOCKET)
3544 return set_error();
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00003545
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003546 newfdobj = PyLong_FromSocket_t(newfd);
3547 if (newfdobj == NULL)
3548 SOCKETCLOSE(newfd);
3549 return newfdobj;
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00003550}
3551
3552PyDoc_STRVAR(dup_doc,
3553"dup(integer) -> integer\n\
3554\n\
3555Duplicate an integer socket file descriptor. This is like os.dup(), but for\n\
3556sockets; on some platforms os.dup() won't work for socket file descriptors.");
3557#endif
3558
3559
Dave Cole331708b2004-08-09 04:51:41 +00003560#ifdef HAVE_SOCKETPAIR
3561/* Create a pair of sockets using the socketpair() function.
Dave Cole07fda7e2004-08-23 05:16:23 +00003562 Arguments as for socket() except the default family is AF_UNIX if
Dave Colee8bbfe42004-08-26 00:51:16 +00003563 defined on the platform; otherwise, the default is AF_INET. */
Dave Cole331708b2004-08-09 04:51:41 +00003564
3565/*ARGSUSED*/
3566static PyObject *
3567socket_socketpair(PyObject *self, PyObject *args)
3568{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003569 PySocketSockObject *s0 = NULL, *s1 = NULL;
3570 SOCKET_T sv[2];
3571 int family, type = SOCK_STREAM, proto = 0;
3572 PyObject *res = NULL;
Dave Cole331708b2004-08-09 04:51:41 +00003573
3574#if defined(AF_UNIX)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003575 family = AF_UNIX;
Dave Cole331708b2004-08-09 04:51:41 +00003576#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003577 family = AF_INET;
Dave Cole331708b2004-08-09 04:51:41 +00003578#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003579 if (!PyArg_ParseTuple(args, "|iii:socketpair",
3580 &family, &type, &proto))
3581 return NULL;
3582 /* Create a pair of socket fds */
3583 if (socketpair(family, type, proto, sv) < 0)
3584 return set_error();
3585 s0 = new_sockobject(sv[0], family, type, proto);
3586 if (s0 == NULL)
3587 goto finally;
3588 s1 = new_sockobject(sv[1], family, type, proto);
3589 if (s1 == NULL)
3590 goto finally;
3591 res = PyTuple_Pack(2, s0, s1);
Dave Cole331708b2004-08-09 04:51:41 +00003592
3593finally:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003594 if (res == NULL) {
3595 if (s0 == NULL)
3596 SOCKETCLOSE(sv[0]);
3597 if (s1 == NULL)
3598 SOCKETCLOSE(sv[1]);
3599 }
3600 Py_XDECREF(s0);
3601 Py_XDECREF(s1);
3602 return res;
Dave Cole331708b2004-08-09 04:51:41 +00003603}
3604
3605PyDoc_STRVAR(socketpair_doc,
3606"socketpair([family[, type[, proto]]]) -> (socket object, socket object)\n\
3607\n\
3608Create a pair of socket objects from the sockets returned by the platform\n\
3609socketpair() function.\n\
Dave Cole07fda7e2004-08-23 05:16:23 +00003610The arguments are the same as for socket() except the default family is\n\
Dave Colee8bbfe42004-08-26 00:51:16 +00003611AF_UNIX if defined on the platform; otherwise, the default is AF_INET.");
Dave Cole331708b2004-08-09 04:51:41 +00003612
3613#endif /* HAVE_SOCKETPAIR */
3614
3615
Guido van Rossum006bf911996-06-12 04:04:55 +00003616static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003617socket_ntohs(PyObject *self, PyObject *args)
Guido van Rossum006bf911996-06-12 04:04:55 +00003618{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003619 int x1, x2;
Guido van Rossum006bf911996-06-12 04:04:55 +00003620
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003621 if (!PyArg_ParseTuple(args, "i:ntohs", &x1)) {
3622 return NULL;
3623 }
3624 if (x1 < 0) {
3625 PyErr_SetString(PyExc_OverflowError,
3626 "can't convert negative number to unsigned long");
3627 return NULL;
3628 }
3629 x2 = (unsigned int)ntohs((unsigned short)x1);
3630 return PyLong_FromLong(x2);
Guido van Rossum006bf911996-06-12 04:04:55 +00003631}
3632
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003633PyDoc_STRVAR(ntohs_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003634"ntohs(integer) -> integer\n\
3635\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003636Convert a 16-bit integer from network to host byte order.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003637
3638
Guido van Rossum006bf911996-06-12 04:04:55 +00003639static PyObject *
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003640socket_ntohl(PyObject *self, PyObject *arg)
Guido van Rossum006bf911996-06-12 04:04:55 +00003641{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003642 unsigned long x;
Guido van Rossum006bf911996-06-12 04:04:55 +00003643
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003644 if (PyLong_Check(arg)) {
3645 x = PyLong_AsUnsignedLong(arg);
3646 if (x == (unsigned long) -1 && PyErr_Occurred())
3647 return NULL;
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003648#if SIZEOF_LONG > 4
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003649 {
3650 unsigned long y;
3651 /* only want the trailing 32 bits */
3652 y = x & 0xFFFFFFFFUL;
3653 if (y ^ x)
3654 return PyErr_Format(PyExc_OverflowError,
3655 "long int larger than 32 bits");
3656 x = y;
3657 }
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003658#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003659 }
3660 else
3661 return PyErr_Format(PyExc_TypeError,
3662 "expected int/long, %s found",
3663 Py_TYPE(arg)->tp_name);
3664 if (x == (unsigned long) -1 && PyErr_Occurred())
3665 return NULL;
3666 return PyLong_FromUnsignedLong(ntohl(x));
Guido van Rossum006bf911996-06-12 04:04:55 +00003667}
3668
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003669PyDoc_STRVAR(ntohl_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003670"ntohl(integer) -> integer\n\
3671\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003672Convert a 32-bit integer from network to host byte order.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003673
3674
Guido van Rossum006bf911996-06-12 04:04:55 +00003675static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003676socket_htons(PyObject *self, PyObject *args)
Guido van Rossum006bf911996-06-12 04:04:55 +00003677{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003678 int x1, x2;
Guido van Rossum006bf911996-06-12 04:04:55 +00003679
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003680 if (!PyArg_ParseTuple(args, "i:htons", &x1)) {
3681 return NULL;
3682 }
3683 if (x1 < 0) {
3684 PyErr_SetString(PyExc_OverflowError,
3685 "can't convert negative number to unsigned long");
3686 return NULL;
3687 }
3688 x2 = (unsigned int)htons((unsigned short)x1);
3689 return PyLong_FromLong(x2);
Guido van Rossum006bf911996-06-12 04:04:55 +00003690}
3691
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003692PyDoc_STRVAR(htons_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003693"htons(integer) -> integer\n\
3694\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003695Convert a 16-bit integer from host to network byte order.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003696
3697
Guido van Rossum006bf911996-06-12 04:04:55 +00003698static PyObject *
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003699socket_htonl(PyObject *self, PyObject *arg)
Guido van Rossum006bf911996-06-12 04:04:55 +00003700{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003701 unsigned long x;
Guido van Rossum006bf911996-06-12 04:04:55 +00003702
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003703 if (PyLong_Check(arg)) {
3704 x = PyLong_AsUnsignedLong(arg);
3705 if (x == (unsigned long) -1 && PyErr_Occurred())
3706 return NULL;
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003707#if SIZEOF_LONG > 4
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003708 {
3709 unsigned long y;
3710 /* only want the trailing 32 bits */
3711 y = x & 0xFFFFFFFFUL;
3712 if (y ^ x)
3713 return PyErr_Format(PyExc_OverflowError,
3714 "long int larger than 32 bits");
3715 x = y;
3716 }
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003717#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003718 }
3719 else
3720 return PyErr_Format(PyExc_TypeError,
3721 "expected int/long, %s found",
3722 Py_TYPE(arg)->tp_name);
3723 return PyLong_FromUnsignedLong(htonl((unsigned long)x));
Guido van Rossum006bf911996-06-12 04:04:55 +00003724}
3725
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003726PyDoc_STRVAR(htonl_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003727"htonl(integer) -> integer\n\
3728\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003729Convert a 32-bit integer from host to network byte order.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003730
Guido van Rossum3eede5a2002-06-07 02:08:35 +00003731/* socket.inet_aton() and socket.inet_ntoa() functions. */
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003732
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003733PyDoc_STRVAR(inet_aton_doc,
Guido van Rossum7d0a8262007-05-21 23:13:11 +00003734"inet_aton(string) -> bytes giving packed 32-bit IP representation\n\
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003735\n\
Guido van Rossumc6a164b1999-08-20 19:11:27 +00003736Convert 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 +00003737binary format used in low-level network functions.");
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003738
3739static PyObject*
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003740socket_inet_aton(PyObject *self, PyObject *args)
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003741{
Guido van Rossuma2e48551999-09-09 15:42:59 +00003742#ifndef INADDR_NONE
3743#define INADDR_NONE (-1)
3744#endif
Neal Norwitz88f115b2003-02-13 02:15:42 +00003745#ifdef HAVE_INET_ATON
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003746 struct in_addr buf;
Thomas Wouters477c8d52006-05-27 19:21:47 +00003747#endif
3748
3749#if !defined(HAVE_INET_ATON) || defined(USE_INET_ATON_WEAKLINK)
Benjamin Petersonf91df042009-02-13 02:50:59 +00003750#if (SIZEOF_INT != 4)
3751#error "Not sure if in_addr_t exists and int is not 32-bits."
3752#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003753 /* Have to use inet_addr() instead */
3754 unsigned int packed_addr;
Tim Peters1df9fdd2003-02-13 03:13:40 +00003755#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003756 char *ip_addr;
Guido van Rossumc6a164b1999-08-20 19:11:27 +00003757
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003758 if (!PyArg_ParseTuple(args, "s:inet_aton", &ip_addr))
3759 return NULL;
Guido van Rossumad05cdf2003-02-12 23:08:22 +00003760
Tim Peters1df9fdd2003-02-13 03:13:40 +00003761
3762#ifdef HAVE_INET_ATON
Thomas Wouters477c8d52006-05-27 19:21:47 +00003763
3764#ifdef USE_INET_ATON_WEAKLINK
3765 if (inet_aton != NULL) {
3766#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003767 if (inet_aton(ip_addr, &buf))
3768 return PyBytes_FromStringAndSize((char *)(&buf),
3769 sizeof(buf));
Guido van Rossumad05cdf2003-02-12 23:08:22 +00003770
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003771 PyErr_SetString(socket_error,
3772 "illegal IP address string passed to inet_aton");
3773 return NULL;
Guido van Rossumad05cdf2003-02-12 23:08:22 +00003774
Thomas Wouters477c8d52006-05-27 19:21:47 +00003775#ifdef USE_INET_ATON_WEAKLINK
3776 } else {
3777#endif
3778
3779#endif
3780
3781#if !defined(HAVE_INET_ATON) || defined(USE_INET_ATON_WEAKLINK)
3782
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003783 /* special-case this address as inet_addr might return INADDR_NONE
3784 * for this */
3785 if (strcmp(ip_addr, "255.255.255.255") == 0) {
3786 packed_addr = 0xFFFFFFFF;
3787 } else {
Thomas Wouters477c8d52006-05-27 19:21:47 +00003788
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003789 packed_addr = inet_addr(ip_addr);
Guido van Rossumc6a164b1999-08-20 19:11:27 +00003790
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003791 if (packed_addr == INADDR_NONE) { /* invalid address */
3792 PyErr_SetString(socket_error,
3793 "illegal IP address string passed to inet_aton");
3794 return NULL;
3795 }
3796 }
3797 return PyBytes_FromStringAndSize((char *) &packed_addr,
3798 sizeof(packed_addr));
Thomas Wouters477c8d52006-05-27 19:21:47 +00003799
3800#ifdef USE_INET_ATON_WEAKLINK
3801 }
3802#endif
3803
Guido van Rossumad05cdf2003-02-12 23:08:22 +00003804#endif
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003805}
3806
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003807PyDoc_STRVAR(inet_ntoa_doc,
Fred Drakee0661342000-03-07 14:05:16 +00003808"inet_ntoa(packed_ip) -> ip_address_string\n\
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003809\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003810Convert an IP address from 32-bit packed binary format to string format");
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003811
3812static PyObject*
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003813socket_inet_ntoa(PyObject *self, PyObject *args)
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003814{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003815 char *packed_str;
3816 int addr_len;
3817 struct in_addr packed_addr;
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003818
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003819 if (!PyArg_ParseTuple(args, "y#:inet_ntoa", &packed_str, &addr_len)) {
3820 return NULL;
3821 }
Guido van Rossum48a680c2001-03-02 06:34:14 +00003822
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003823 if (addr_len != sizeof(packed_addr)) {
3824 PyErr_SetString(socket_error,
3825 "packed IP wrong length for inet_ntoa");
3826 return NULL;
3827 }
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003828
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003829 memcpy(&packed_addr, packed_str, addr_len);
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003830
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003831 return PyUnicode_FromString(inet_ntoa(packed_addr));
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003832}
Guido van Rossum82a5c661998-07-07 20:45:43 +00003833
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003834#ifdef HAVE_INET_PTON
3835
3836PyDoc_STRVAR(inet_pton_doc,
3837"inet_pton(af, ip) -> packed IP address string\n\
3838\n\
3839Convert an IP address from string format to a packed string suitable\n\
3840for use with low-level network functions.");
3841
3842static PyObject *
3843socket_inet_pton(PyObject *self, PyObject *args)
3844{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003845 int af;
3846 char* ip;
3847 int retval;
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003848#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003849 char packed[MAX(sizeof(struct in_addr), sizeof(struct in6_addr))];
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003850#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003851 char packed[sizeof(struct in_addr)];
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003852#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003853 if (!PyArg_ParseTuple(args, "is:inet_pton", &af, &ip)) {
3854 return NULL;
3855 }
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003856
Martin v. Löwis04697e82004-06-02 12:35:29 +00003857#if !defined(ENABLE_IPV6) && defined(AF_INET6)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003858 if(af == AF_INET6) {
3859 PyErr_SetString(socket_error,
3860 "can't use AF_INET6, IPv6 is disabled");
3861 return NULL;
3862 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00003863#endif
Martin v. Löwis10649092003-08-05 06:25:06 +00003864
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003865 retval = inet_pton(af, ip, packed);
3866 if (retval < 0) {
3867 PyErr_SetFromErrno(socket_error);
3868 return NULL;
3869 } else if (retval == 0) {
3870 PyErr_SetString(socket_error,
3871 "illegal IP address string passed to inet_pton");
3872 return NULL;
3873 } else if (af == AF_INET) {
3874 return PyBytes_FromStringAndSize(packed,
3875 sizeof(struct in_addr));
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003876#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003877 } else if (af == AF_INET6) {
3878 return PyBytes_FromStringAndSize(packed,
3879 sizeof(struct in6_addr));
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003880#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003881 } else {
3882 PyErr_SetString(socket_error, "unknown address family");
3883 return NULL;
3884 }
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003885}
Thomas Wouters477c8d52006-05-27 19:21:47 +00003886
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003887PyDoc_STRVAR(inet_ntop_doc,
3888"inet_ntop(af, packed_ip) -> string formatted IP address\n\
3889\n\
3890Convert a packed IP address of the given family to string format.");
3891
3892static PyObject *
3893socket_inet_ntop(PyObject *self, PyObject *args)
3894{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003895 int af;
3896 char* packed;
3897 int len;
3898 const char* retval;
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003899#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003900 char ip[MAX(INET_ADDRSTRLEN, INET6_ADDRSTRLEN) + 1];
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003901#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003902 char ip[INET_ADDRSTRLEN + 1];
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003903#endif
Thomas Wouters477c8d52006-05-27 19:21:47 +00003904
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003905 /* Guarantee NUL-termination for PyUnicode_FromString() below */
3906 memset((void *) &ip[0], '\0', sizeof(ip));
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003907
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003908 if (!PyArg_ParseTuple(args, "iy#:inet_ntop", &af, &packed, &len)) {
3909 return NULL;
3910 }
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003911
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003912 if (af == AF_INET) {
3913 if (len != sizeof(struct in_addr)) {
3914 PyErr_SetString(PyExc_ValueError,
3915 "invalid length of packed IP address string");
3916 return NULL;
3917 }
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003918#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003919 } else if (af == AF_INET6) {
3920 if (len != sizeof(struct in6_addr)) {
3921 PyErr_SetString(PyExc_ValueError,
3922 "invalid length of packed IP address string");
3923 return NULL;
3924 }
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003925#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003926 } else {
3927 PyErr_Format(PyExc_ValueError,
3928 "unknown address family %d", af);
3929 return NULL;
3930 }
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003931
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003932 retval = inet_ntop(af, packed, ip, sizeof(ip));
3933 if (!retval) {
3934 PyErr_SetFromErrno(socket_error);
3935 return NULL;
3936 } else {
3937 return PyUnicode_FromString(retval);
3938 }
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003939
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003940 /* NOTREACHED */
3941 PyErr_SetString(PyExc_RuntimeError, "invalid handling of inet_ntop");
3942 return NULL;
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003943}
3944
3945#endif /* HAVE_INET_PTON */
3946
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003947/* Python interface to getaddrinfo(host, port). */
3948
3949/*ARGSUSED*/
3950static PyObject *
Giampaolo Rodolàccfb91c2010-08-17 15:30:23 +00003951socket_getaddrinfo(PyObject *self, PyObject *args, PyObject* kwargs)
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003952{
Giampaolo Rodolàccfb91c2010-08-17 15:30:23 +00003953 static char* kwnames[] = {"host", "port", "family", "type", "proto",
3954 "flags", 0};
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003955 struct addrinfo hints, *res;
3956 struct addrinfo *res0 = NULL;
3957 PyObject *hobj = NULL;
3958 PyObject *pobj = (PyObject *)NULL;
3959 char pbuf[30];
3960 char *hptr, *pptr;
3961 int family, socktype, protocol, flags;
3962 int error;
3963 PyObject *all = (PyObject *)NULL;
3964 PyObject *idna = NULL;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003965
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003966 family = socktype = protocol = flags = 0;
3967 family = AF_UNSPEC;
Giampaolo Rodolàccfb91c2010-08-17 15:30:23 +00003968 if (!PyArg_ParseTupleAndKeywords(args, kwargs, "OO|iiii:getaddrinfo",
3969 kwnames, &hobj, &pobj, &family, &socktype,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003970 &protocol, &flags)) {
3971 return NULL;
3972 }
3973 if (hobj == Py_None) {
3974 hptr = NULL;
3975 } else if (PyUnicode_Check(hobj)) {
3976 idna = PyObject_CallMethod(hobj, "encode", "s", "idna");
3977 if (!idna)
3978 return NULL;
3979 assert(PyBytes_Check(idna));
3980 hptr = PyBytes_AS_STRING(idna);
3981 } else if (PyBytes_Check(hobj)) {
3982 hptr = PyBytes_AsString(hobj);
3983 } else {
3984 PyErr_SetString(PyExc_TypeError,
3985 "getaddrinfo() argument 1 must be string or None");
3986 return NULL;
3987 }
3988 if (PyLong_CheckExact(pobj)) {
3989 long value = PyLong_AsLong(pobj);
3990 if (value == -1 && PyErr_Occurred())
3991 goto err;
3992 PyOS_snprintf(pbuf, sizeof(pbuf), "%ld", value);
3993 pptr = pbuf;
3994 } else if (PyUnicode_Check(pobj)) {
3995 pptr = _PyUnicode_AsString(pobj);
3996 } else if (PyBytes_Check(pobj)) {
3997 pptr = PyBytes_AsString(pobj);
3998 } else if (pobj == Py_None) {
3999 pptr = (char *)NULL;
4000 } else {
4001 PyErr_SetString(socket_error, "Int or String expected");
4002 goto err;
4003 }
4004 memset(&hints, 0, sizeof(hints));
4005 hints.ai_family = family;
4006 hints.ai_socktype = socktype;
4007 hints.ai_protocol = protocol;
4008 hints.ai_flags = flags;
4009 Py_BEGIN_ALLOW_THREADS
4010 ACQUIRE_GETADDRINFO_LOCK
4011 error = getaddrinfo(hptr, pptr, &hints, &res0);
4012 Py_END_ALLOW_THREADS
4013 RELEASE_GETADDRINFO_LOCK /* see comment in setipaddr() */
4014 if (error) {
4015 set_gaierror(error);
4016 goto err;
4017 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004018
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004019 if ((all = PyList_New(0)) == NULL)
4020 goto err;
4021 for (res = res0; res; res = res->ai_next) {
4022 PyObject *single;
4023 PyObject *addr =
4024 makesockaddr(-1, res->ai_addr, res->ai_addrlen, protocol);
4025 if (addr == NULL)
4026 goto err;
4027 single = Py_BuildValue("iiisO", res->ai_family,
4028 res->ai_socktype, res->ai_protocol,
4029 res->ai_canonname ? res->ai_canonname : "",
4030 addr);
4031 Py_DECREF(addr);
4032 if (single == NULL)
4033 goto err;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004034
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004035 if (PyList_Append(all, single))
4036 goto err;
4037 Py_XDECREF(single);
4038 }
4039 Py_XDECREF(idna);
4040 if (res0)
4041 freeaddrinfo(res0);
4042 return all;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004043 err:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004044 Py_XDECREF(all);
4045 Py_XDECREF(idna);
4046 if (res0)
4047 freeaddrinfo(res0);
4048 return (PyObject *)NULL;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004049}
4050
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004051PyDoc_STRVAR(getaddrinfo_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00004052"getaddrinfo(host, port [, family, socktype, proto, flags])\n\
4053 -> list of (family, socktype, proto, canonname, sockaddr)\n\
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004054\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004055Resolve host and port into addrinfo struct.");
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004056
4057/* Python interface to getnameinfo(sa, flags). */
4058
4059/*ARGSUSED*/
4060static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004061socket_getnameinfo(PyObject *self, PyObject *args)
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004062{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004063 PyObject *sa = (PyObject *)NULL;
4064 int flags;
4065 char *hostp;
4066 int port, flowinfo, scope_id;
4067 char hbuf[NI_MAXHOST], pbuf[NI_MAXSERV];
4068 struct addrinfo hints, *res = NULL;
4069 int error;
4070 PyObject *ret = (PyObject *)NULL;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004071
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004072 flags = flowinfo = scope_id = 0;
4073 if (!PyArg_ParseTuple(args, "Oi:getnameinfo", &sa, &flags))
4074 return NULL;
4075 if (!PyTuple_Check(sa)) {
4076 PyErr_SetString(PyExc_TypeError,
4077 "getnameinfo() argument 1 must be a tuple");
4078 return NULL;
4079 }
4080 if (!PyArg_ParseTuple(sa, "si|ii",
4081 &hostp, &port, &flowinfo, &scope_id))
4082 return NULL;
4083 PyOS_snprintf(pbuf, sizeof(pbuf), "%d", port);
4084 memset(&hints, 0, sizeof(hints));
4085 hints.ai_family = AF_UNSPEC;
4086 hints.ai_socktype = SOCK_DGRAM; /* make numeric port happy */
Martin v. Löwis112c0f32010-08-25 07:38:15 +00004087 hints.ai_flags = AI_NUMERICHOST; /* don't do any name resolution */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004088 Py_BEGIN_ALLOW_THREADS
4089 ACQUIRE_GETADDRINFO_LOCK
4090 error = getaddrinfo(hostp, pbuf, &hints, &res);
4091 Py_END_ALLOW_THREADS
4092 RELEASE_GETADDRINFO_LOCK /* see comment in setipaddr() */
4093 if (error) {
4094 set_gaierror(error);
4095 goto fail;
4096 }
4097 if (res->ai_next) {
4098 PyErr_SetString(socket_error,
4099 "sockaddr resolved to multiple addresses");
4100 goto fail;
4101 }
4102 switch (res->ai_family) {
4103 case AF_INET:
4104 {
4105 if (PyTuple_GET_SIZE(sa) != 2) {
4106 PyErr_SetString(socket_error,
4107 "IPv4 sockaddr must be 2 tuple");
4108 goto fail;
4109 }
4110 break;
4111 }
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00004112#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004113 case AF_INET6:
4114 {
4115 struct sockaddr_in6 *sin6;
4116 sin6 = (struct sockaddr_in6 *)res->ai_addr;
4117 sin6->sin6_flowinfo = flowinfo;
4118 sin6->sin6_scope_id = scope_id;
4119 break;
4120 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004121#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004122 }
Antoine Pitrou19467d22010-08-17 19:33:30 +00004123 error = getnameinfo(res->ai_addr, (socklen_t) res->ai_addrlen,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004124 hbuf, sizeof(hbuf), pbuf, sizeof(pbuf), flags);
4125 if (error) {
4126 set_gaierror(error);
4127 goto fail;
4128 }
4129 ret = Py_BuildValue("ss", hbuf, pbuf);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004130
4131fail:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004132 if (res)
4133 freeaddrinfo(res);
4134 return ret;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004135}
4136
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004137PyDoc_STRVAR(getnameinfo_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00004138"getnameinfo(sockaddr, flags) --> (host, port)\n\
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004139\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004140Get host and port for a sockaddr.");
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004141
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004142
4143/* Python API to getting and setting the default timeout value. */
4144
4145static PyObject *
4146socket_getdefaulttimeout(PyObject *self)
4147{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004148 if (defaulttimeout < 0.0) {
4149 Py_INCREF(Py_None);
4150 return Py_None;
4151 }
4152 else
4153 return PyFloat_FromDouble(defaulttimeout);
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004154}
4155
4156PyDoc_STRVAR(getdefaulttimeout_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00004157"getdefaulttimeout() -> timeout\n\
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004158\n\
4159Returns the default timeout in floating seconds for new socket objects.\n\
4160A value of None indicates that new socket objects have no timeout.\n\
4161When the socket module is first imported, the default is None.");
4162
4163static PyObject *
4164socket_setdefaulttimeout(PyObject *self, PyObject *arg)
4165{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004166 double timeout;
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004167
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004168 if (arg == Py_None)
4169 timeout = -1.0;
4170 else {
4171 timeout = PyFloat_AsDouble(arg);
4172 if (timeout < 0.0) {
4173 if (!PyErr_Occurred())
4174 PyErr_SetString(PyExc_ValueError,
4175 "Timeout value out of range");
4176 return NULL;
4177 }
4178 }
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004179
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004180 defaulttimeout = timeout;
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004181
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004182 Py_INCREF(Py_None);
4183 return Py_None;
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004184}
4185
4186PyDoc_STRVAR(setdefaulttimeout_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00004187"setdefaulttimeout(timeout)\n\
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004188\n\
4189Set the default timeout in floating seconds for new socket objects.\n\
4190A value of None indicates that new socket objects have no timeout.\n\
4191When the socket module is first imported, the default is None.");
4192
4193
Guido van Rossum30a685f1991-06-27 15:51:29 +00004194/* List of functions exported by this module. */
4195
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004196static PyMethodDef socket_methods[] = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004197 {"gethostbyname", socket_gethostbyname,
4198 METH_VARARGS, gethostbyname_doc},
4199 {"gethostbyname_ex", socket_gethostbyname_ex,
4200 METH_VARARGS, ghbn_ex_doc},
4201 {"gethostbyaddr", socket_gethostbyaddr,
4202 METH_VARARGS, gethostbyaddr_doc},
4203 {"gethostname", socket_gethostname,
4204 METH_NOARGS, gethostname_doc},
4205 {"getservbyname", socket_getservbyname,
4206 METH_VARARGS, getservbyname_doc},
4207 {"getservbyport", socket_getservbyport,
4208 METH_VARARGS, getservbyport_doc},
4209 {"getprotobyname", socket_getprotobyname,
4210 METH_VARARGS, getprotobyname_doc},
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00004211#ifndef NO_DUP
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004212 {"dup", socket_dup,
4213 METH_O, dup_doc},
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00004214#endif
Dave Cole331708b2004-08-09 04:51:41 +00004215#ifdef HAVE_SOCKETPAIR
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004216 {"socketpair", socket_socketpair,
4217 METH_VARARGS, socketpair_doc},
Dave Cole331708b2004-08-09 04:51:41 +00004218#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004219 {"ntohs", socket_ntohs,
4220 METH_VARARGS, ntohs_doc},
4221 {"ntohl", socket_ntohl,
4222 METH_O, ntohl_doc},
4223 {"htons", socket_htons,
4224 METH_VARARGS, htons_doc},
4225 {"htonl", socket_htonl,
4226 METH_O, htonl_doc},
4227 {"inet_aton", socket_inet_aton,
4228 METH_VARARGS, inet_aton_doc},
4229 {"inet_ntoa", socket_inet_ntoa,
4230 METH_VARARGS, inet_ntoa_doc},
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00004231#ifdef HAVE_INET_PTON
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004232 {"inet_pton", socket_inet_pton,
4233 METH_VARARGS, inet_pton_doc},
4234 {"inet_ntop", socket_inet_ntop,
4235 METH_VARARGS, inet_ntop_doc},
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00004236#endif
Giampaolo Rodolàccfb91c2010-08-17 15:30:23 +00004237 {"getaddrinfo", (PyCFunction)socket_getaddrinfo,
4238 METH_VARARGS | METH_KEYWORDS, getaddrinfo_doc},
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004239 {"getnameinfo", socket_getnameinfo,
4240 METH_VARARGS, getnameinfo_doc},
4241 {"getdefaulttimeout", (PyCFunction)socket_getdefaulttimeout,
4242 METH_NOARGS, getdefaulttimeout_doc},
4243 {"setdefaulttimeout", socket_setdefaulttimeout,
4244 METH_O, setdefaulttimeout_doc},
4245 {NULL, NULL} /* Sentinel */
Guido van Rossum6574b3e1991-06-25 21:36:08 +00004246};
4247
Guido van Rossum30a685f1991-06-27 15:51:29 +00004248
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004249#ifdef MS_WINDOWS
4250#define OS_INIT_DEFINED
4251
4252/* Additional initialization and cleanup for Windows */
Guido van Rossumbe32c891996-06-20 16:25:29 +00004253
4254static void
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004255os_cleanup(void)
Guido van Rossumbe32c891996-06-20 16:25:29 +00004256{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004257 WSACleanup();
Guido van Rossumbe32c891996-06-20 16:25:29 +00004258}
4259
4260static int
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004261os_init(void)
Guido van Rossumbe32c891996-06-20 16:25:29 +00004262{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004263 WSADATA WSAData;
4264 int ret;
4265 ret = WSAStartup(0x0101, &WSAData);
4266 switch (ret) {
4267 case 0: /* No error */
4268 Py_AtExit(os_cleanup);
4269 return 1; /* Success */
4270 case WSASYSNOTREADY:
4271 PyErr_SetString(PyExc_ImportError,
4272 "WSAStartup failed: network not ready");
4273 break;
4274 case WSAVERNOTSUPPORTED:
4275 case WSAEINVAL:
4276 PyErr_SetString(
4277 PyExc_ImportError,
4278 "WSAStartup failed: requested version not supported");
4279 break;
4280 default:
4281 PyErr_Format(PyExc_ImportError, "WSAStartup failed: error code %d", ret);
4282 break;
4283 }
4284 return 0; /* Failure */
Guido van Rossumbe32c891996-06-20 16:25:29 +00004285}
4286
Guido van Rossum8d665e61996-06-26 18:22:49 +00004287#endif /* MS_WINDOWS */
Guido van Rossumbe32c891996-06-20 16:25:29 +00004288
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004289
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004290#ifdef PYOS_OS2
4291#define OS_INIT_DEFINED
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004292
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004293/* Additional initialization for OS/2 */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004294
4295static int
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004296os_init(void)
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004297{
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004298#ifndef PYCC_GCC
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004299 int rc = sock_init();
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004300
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004301 if (rc == 0) {
4302 return 1; /* Success */
4303 }
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004304
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004305 PyErr_Format(PyExc_ImportError, "OS/2 TCP/IP Error# %d", sock_errno());
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004306
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004307 return 0; /* Failure */
Andrew MacIntyreba43e872002-03-03 03:03:52 +00004308#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004309 /* No need to initialise sockets with GCC/EMX */
4310 return 1; /* Success */
Andrew MacIntyreba43e872002-03-03 03:03:52 +00004311#endif
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004312}
4313
4314#endif /* PYOS_OS2 */
4315
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004316
4317#ifndef OS_INIT_DEFINED
4318static int
4319os_init(void)
4320{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004321 return 1; /* Success */
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004322}
4323#endif
4324
4325
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00004326/* C API table - always add new things to the end for binary
4327 compatibility. */
4328static
4329PySocketModule_APIObject PySocketModuleAPI =
4330{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004331 &sock_type,
4332 NULL
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00004333};
4334
Guido van Rossum3eede5a2002-06-07 02:08:35 +00004335
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004336/* Initialize the _socket module.
Guido van Rossum3eede5a2002-06-07 02:08:35 +00004337
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004338 This module is actually called "_socket", and there's a wrapper
Guido van Rossum7d0a8262007-05-21 23:13:11 +00004339 "socket.py" which implements some additional functionality.
4340 The import of "_socket" may fail with an ImportError exception if
4341 os-specific initialization fails. On Windows, this does WINSOCK
4342 initialization. When WINSOCK is initialized succesfully, a call to
4343 WSACleanup() is scheduled to be made at exit time.
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004344*/
Guido van Rossum3eede5a2002-06-07 02:08:35 +00004345
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004346PyDoc_STRVAR(socket_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00004347"Implementation module for socket operations.\n\
4348\n\
4349See the socket module for documentation.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00004350
Martin v. Löwis1a214512008-06-11 05:26:20 +00004351static struct PyModuleDef socketmodule = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004352 PyModuleDef_HEAD_INIT,
4353 PySocket_MODULE_NAME,
4354 socket_doc,
4355 -1,
4356 socket_methods,
4357 NULL,
4358 NULL,
4359 NULL,
4360 NULL
Martin v. Löwis1a214512008-06-11 05:26:20 +00004361};
4362
Mark Hammondfe51c6d2002-08-02 02:27:13 +00004363PyMODINIT_FUNC
Martin v. Löwis1a214512008-06-11 05:26:20 +00004364PyInit__socket(void)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00004365{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004366 PyObject *m, *has_ipv6;
Fred Drake4baedc12002-04-01 14:53:37 +00004367
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004368 if (!os_init())
4369 return NULL;
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004370
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004371 Py_TYPE(&sock_type) = &PyType_Type;
4372 m = PyModule_Create(&socketmodule);
4373 if (m == NULL)
4374 return NULL;
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004375
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004376 socket_error = PyErr_NewException("socket.error",
4377 PyExc_IOError, NULL);
4378 if (socket_error == NULL)
4379 return NULL;
4380 PySocketModuleAPI.error = socket_error;
4381 Py_INCREF(socket_error);
4382 PyModule_AddObject(m, "error", socket_error);
4383 socket_herror = PyErr_NewException("socket.herror",
4384 socket_error, NULL);
4385 if (socket_herror == NULL)
4386 return NULL;
4387 Py_INCREF(socket_herror);
4388 PyModule_AddObject(m, "herror", socket_herror);
4389 socket_gaierror = PyErr_NewException("socket.gaierror", socket_error,
4390 NULL);
4391 if (socket_gaierror == NULL)
4392 return NULL;
4393 Py_INCREF(socket_gaierror);
4394 PyModule_AddObject(m, "gaierror", socket_gaierror);
4395 socket_timeout = PyErr_NewException("socket.timeout",
4396 socket_error, NULL);
4397 if (socket_timeout == NULL)
4398 return NULL;
4399 Py_INCREF(socket_timeout);
4400 PyModule_AddObject(m, "timeout", socket_timeout);
4401 Py_INCREF((PyObject *)&sock_type);
4402 if (PyModule_AddObject(m, "SocketType",
4403 (PyObject *)&sock_type) != 0)
4404 return NULL;
4405 Py_INCREF((PyObject *)&sock_type);
4406 if (PyModule_AddObject(m, "socket",
4407 (PyObject *)&sock_type) != 0)
4408 return NULL;
Guido van Rossum09be4091999-08-09 14:40:40 +00004409
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00004410#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004411 has_ipv6 = Py_True;
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00004412#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004413 has_ipv6 = Py_False;
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00004414#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004415 Py_INCREF(has_ipv6);
4416 PyModule_AddObject(m, "has_ipv6", has_ipv6);
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00004417
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004418 /* Export C API */
4419 if (PyModule_AddObject(m, PySocket_CAPI_NAME,
4420 PyCapsule_New(&PySocketModuleAPI, PySocket_CAPSULE_NAME, NULL)
4421 ) != 0)
4422 return NULL;
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00004423
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004424 /* Address families (we only support AF_INET and AF_UNIX) */
Guido van Rossum09be4091999-08-09 14:40:40 +00004425#ifdef AF_UNSPEC
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004426 PyModule_AddIntConstant(m, "AF_UNSPEC", AF_UNSPEC);
Guido van Rossum09be4091999-08-09 14:40:40 +00004427#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004428 PyModule_AddIntConstant(m, "AF_INET", AF_INET);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004429#ifdef AF_INET6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004430 PyModule_AddIntConstant(m, "AF_INET6", AF_INET6);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004431#endif /* AF_INET6 */
Andrew MacIntyred12dfbb2004-04-04 07:13:49 +00004432#if defined(AF_UNIX)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004433 PyModule_AddIntConstant(m, "AF_UNIX", AF_UNIX);
Guido van Rossumb6775db1994-08-01 11:34:53 +00004434#endif /* AF_UNIX */
Guido van Rossum09be4091999-08-09 14:40:40 +00004435#ifdef AF_AX25
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004436 /* Amateur Radio AX.25 */
4437 PyModule_AddIntConstant(m, "AF_AX25", AF_AX25);
Guido van Rossum09be4091999-08-09 14:40:40 +00004438#endif
4439#ifdef AF_IPX
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004440 PyModule_AddIntConstant(m, "AF_IPX", AF_IPX); /* Novell IPX */
Guido van Rossum09be4091999-08-09 14:40:40 +00004441#endif
4442#ifdef AF_APPLETALK
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004443 /* Appletalk DDP */
4444 PyModule_AddIntConstant(m, "AF_APPLETALK", AF_APPLETALK);
Guido van Rossum09be4091999-08-09 14:40:40 +00004445#endif
4446#ifdef AF_NETROM
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004447 /* Amateur radio NetROM */
4448 PyModule_AddIntConstant(m, "AF_NETROM", AF_NETROM);
Guido van Rossum09be4091999-08-09 14:40:40 +00004449#endif
4450#ifdef AF_BRIDGE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004451 /* Multiprotocol bridge */
4452 PyModule_AddIntConstant(m, "AF_BRIDGE", AF_BRIDGE);
Guido van Rossum09be4091999-08-09 14:40:40 +00004453#endif
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004454#ifdef AF_ATMPVC
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004455 /* ATM PVCs */
4456 PyModule_AddIntConstant(m, "AF_ATMPVC", AF_ATMPVC);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004457#endif
Guido van Rossum09be4091999-08-09 14:40:40 +00004458#ifdef AF_AAL5
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004459 /* Reserved for Werner's ATM */
4460 PyModule_AddIntConstant(m, "AF_AAL5", AF_AAL5);
Guido van Rossum09be4091999-08-09 14:40:40 +00004461#endif
4462#ifdef AF_X25
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004463 /* Reserved for X.25 project */
4464 PyModule_AddIntConstant(m, "AF_X25", AF_X25);
Guido van Rossum09be4091999-08-09 14:40:40 +00004465#endif
4466#ifdef AF_INET6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004467 PyModule_AddIntConstant(m, "AF_INET6", AF_INET6); /* IP version 6 */
Guido van Rossum09be4091999-08-09 14:40:40 +00004468#endif
4469#ifdef AF_ROSE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004470 /* Amateur Radio X.25 PLP */
4471 PyModule_AddIntConstant(m, "AF_ROSE", AF_ROSE);
Guido van Rossum09be4091999-08-09 14:40:40 +00004472#endif
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004473#ifdef AF_DECnet
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004474 /* Reserved for DECnet project */
4475 PyModule_AddIntConstant(m, "AF_DECnet", AF_DECnet);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004476#endif
4477#ifdef AF_NETBEUI
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004478 /* Reserved for 802.2LLC project */
4479 PyModule_AddIntConstant(m, "AF_NETBEUI", AF_NETBEUI);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004480#endif
4481#ifdef AF_SECURITY
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004482 /* Security callback pseudo AF */
4483 PyModule_AddIntConstant(m, "AF_SECURITY", AF_SECURITY);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004484#endif
4485#ifdef AF_KEY
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004486 /* PF_KEY key management API */
4487 PyModule_AddIntConstant(m, "AF_KEY", AF_KEY);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004488#endif
4489#ifdef AF_NETLINK
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004490 /* */
4491 PyModule_AddIntConstant(m, "AF_NETLINK", AF_NETLINK);
4492 PyModule_AddIntConstant(m, "NETLINK_ROUTE", NETLINK_ROUTE);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004493#ifdef NETLINK_SKIP
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004494 PyModule_AddIntConstant(m, "NETLINK_SKIP", NETLINK_SKIP);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004495#endif
4496#ifdef NETLINK_W1
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004497 PyModule_AddIntConstant(m, "NETLINK_W1", NETLINK_W1);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004498#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004499 PyModule_AddIntConstant(m, "NETLINK_USERSOCK", NETLINK_USERSOCK);
4500 PyModule_AddIntConstant(m, "NETLINK_FIREWALL", NETLINK_FIREWALL);
Guido van Rossum668a94a2006-02-21 01:07:27 +00004501#ifdef NETLINK_TCPDIAG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004502 PyModule_AddIntConstant(m, "NETLINK_TCPDIAG", NETLINK_TCPDIAG);
Guido van Rossum668a94a2006-02-21 01:07:27 +00004503#endif
4504#ifdef NETLINK_NFLOG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004505 PyModule_AddIntConstant(m, "NETLINK_NFLOG", NETLINK_NFLOG);
Guido van Rossum668a94a2006-02-21 01:07:27 +00004506#endif
Neal Norwitz65851662006-01-16 04:31:40 +00004507#ifdef NETLINK_XFRM
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004508 PyModule_AddIntConstant(m, "NETLINK_XFRM", NETLINK_XFRM);
Neal Norwitz65851662006-01-16 04:31:40 +00004509#endif
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004510#ifdef NETLINK_ARPD
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004511 PyModule_AddIntConstant(m, "NETLINK_ARPD", NETLINK_ARPD);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004512#endif
4513#ifdef NETLINK_ROUTE6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004514 PyModule_AddIntConstant(m, "NETLINK_ROUTE6", NETLINK_ROUTE6);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004515#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004516 PyModule_AddIntConstant(m, "NETLINK_IP6_FW", NETLINK_IP6_FW);
Thomas Wouterscf297e42007-02-23 15:07:44 +00004517#ifdef NETLINK_DNRTMSG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004518 PyModule_AddIntConstant(m, "NETLINK_DNRTMSG", NETLINK_DNRTMSG);
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00004519#endif
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004520#ifdef NETLINK_TAPBASE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004521 PyModule_AddIntConstant(m, "NETLINK_TAPBASE", NETLINK_TAPBASE);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004522#endif
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004523#endif /* AF_NETLINK */
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004524#ifdef AF_ROUTE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004525 /* Alias to emulate 4.4BSD */
4526 PyModule_AddIntConstant(m, "AF_ROUTE", AF_ROUTE);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004527#endif
4528#ifdef AF_ASH
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004529 /* Ash */
4530 PyModule_AddIntConstant(m, "AF_ASH", AF_ASH);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004531#endif
4532#ifdef AF_ECONET
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004533 /* Acorn Econet */
4534 PyModule_AddIntConstant(m, "AF_ECONET", AF_ECONET);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004535#endif
4536#ifdef AF_ATMSVC
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004537 /* ATM SVCs */
4538 PyModule_AddIntConstant(m, "AF_ATMSVC", AF_ATMSVC);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004539#endif
4540#ifdef AF_SNA
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004541 /* Linux SNA Project (nutters!) */
4542 PyModule_AddIntConstant(m, "AF_SNA", AF_SNA);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004543#endif
4544#ifdef AF_IRDA
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004545 /* IRDA sockets */
4546 PyModule_AddIntConstant(m, "AF_IRDA", AF_IRDA);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004547#endif
4548#ifdef AF_PPPOX
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004549 /* PPPoX sockets */
4550 PyModule_AddIntConstant(m, "AF_PPPOX", AF_PPPOX);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004551#endif
4552#ifdef AF_WANPIPE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004553 /* Wanpipe API Sockets */
4554 PyModule_AddIntConstant(m, "AF_WANPIPE", AF_WANPIPE);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004555#endif
4556#ifdef AF_LLC
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004557 /* Linux LLC */
4558 PyModule_AddIntConstant(m, "AF_LLC", AF_LLC);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004559#endif
Martin v. Löwis12af0482004-01-31 12:34:17 +00004560
Hye-Shik Chang81268602004-02-02 06:05:24 +00004561#ifdef USE_BLUETOOTH
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004562 PyModule_AddIntConstant(m, "AF_BLUETOOTH", AF_BLUETOOTH);
4563 PyModule_AddIntConstant(m, "BTPROTO_L2CAP", BTPROTO_L2CAP);
4564 PyModule_AddIntConstant(m, "BTPROTO_HCI", BTPROTO_HCI);
4565 PyModule_AddIntConstant(m, "SOL_HCI", SOL_HCI);
Gregory P. Smith3e85dfd2010-10-17 00:43:10 +00004566#if !defined(__NetBSD__) && !defined(__DragonFly__)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004567 PyModule_AddIntConstant(m, "HCI_FILTER", HCI_FILTER);
Gregory P. Smith3e85dfd2010-10-17 00:43:10 +00004568#endif
Hye-Shik Chang81268602004-02-02 06:05:24 +00004569#if !defined(__FreeBSD__)
Gregory P. Smith3e85dfd2010-10-17 00:43:10 +00004570#if !defined(__NetBSD__) && !defined(__DragonFly__)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004571 PyModule_AddIntConstant(m, "HCI_TIME_STAMP", HCI_TIME_STAMP);
Gregory P. Smith3e85dfd2010-10-17 00:43:10 +00004572#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004573 PyModule_AddIntConstant(m, "HCI_DATA_DIR", HCI_DATA_DIR);
4574 PyModule_AddIntConstant(m, "BTPROTO_SCO", BTPROTO_SCO);
Hye-Shik Chang81268602004-02-02 06:05:24 +00004575#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004576 PyModule_AddIntConstant(m, "BTPROTO_RFCOMM", BTPROTO_RFCOMM);
4577 PyModule_AddStringConstant(m, "BDADDR_ANY", "00:00:00:00:00:00");
4578 PyModule_AddStringConstant(m, "BDADDR_LOCAL", "00:00:00:FF:FF:FF");
Martin v. Löwis12af0482004-01-31 12:34:17 +00004579#endif
4580
Martin v. Löwis1ba3fd52001-08-10 20:29:40 +00004581#ifdef HAVE_NETPACKET_PACKET_H
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004582 PyModule_AddIntConstant(m, "AF_PACKET", AF_PACKET);
4583 PyModule_AddIntConstant(m, "PF_PACKET", PF_PACKET);
4584 PyModule_AddIntConstant(m, "PACKET_HOST", PACKET_HOST);
4585 PyModule_AddIntConstant(m, "PACKET_BROADCAST", PACKET_BROADCAST);
4586 PyModule_AddIntConstant(m, "PACKET_MULTICAST", PACKET_MULTICAST);
4587 PyModule_AddIntConstant(m, "PACKET_OTHERHOST", PACKET_OTHERHOST);
4588 PyModule_AddIntConstant(m, "PACKET_OUTGOING", PACKET_OUTGOING);
4589 PyModule_AddIntConstant(m, "PACKET_LOOPBACK", PACKET_LOOPBACK);
4590 PyModule_AddIntConstant(m, "PACKET_FASTROUTE", PACKET_FASTROUTE);
Guido van Rossum48a680c2001-03-02 06:34:14 +00004591#endif
Guido van Rossum09be4091999-08-09 14:40:40 +00004592
Christian Heimes043d6f62008-01-07 17:19:16 +00004593#ifdef HAVE_LINUX_TIPC_H
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004594 PyModule_AddIntConstant(m, "AF_TIPC", AF_TIPC);
Christian Heimes043d6f62008-01-07 17:19:16 +00004595
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004596 /* for addresses */
4597 PyModule_AddIntConstant(m, "TIPC_ADDR_NAMESEQ", TIPC_ADDR_NAMESEQ);
4598 PyModule_AddIntConstant(m, "TIPC_ADDR_NAME", TIPC_ADDR_NAME);
4599 PyModule_AddIntConstant(m, "TIPC_ADDR_ID", TIPC_ADDR_ID);
Christian Heimes043d6f62008-01-07 17:19:16 +00004600
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004601 PyModule_AddIntConstant(m, "TIPC_ZONE_SCOPE", TIPC_ZONE_SCOPE);
4602 PyModule_AddIntConstant(m, "TIPC_CLUSTER_SCOPE", TIPC_CLUSTER_SCOPE);
4603 PyModule_AddIntConstant(m, "TIPC_NODE_SCOPE", TIPC_NODE_SCOPE);
Christian Heimes043d6f62008-01-07 17:19:16 +00004604
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004605 /* for setsockopt() */
4606 PyModule_AddIntConstant(m, "SOL_TIPC", SOL_TIPC);
4607 PyModule_AddIntConstant(m, "TIPC_IMPORTANCE", TIPC_IMPORTANCE);
4608 PyModule_AddIntConstant(m, "TIPC_SRC_DROPPABLE", TIPC_SRC_DROPPABLE);
4609 PyModule_AddIntConstant(m, "TIPC_DEST_DROPPABLE",
4610 TIPC_DEST_DROPPABLE);
4611 PyModule_AddIntConstant(m, "TIPC_CONN_TIMEOUT", TIPC_CONN_TIMEOUT);
Christian Heimes043d6f62008-01-07 17:19:16 +00004612
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004613 PyModule_AddIntConstant(m, "TIPC_LOW_IMPORTANCE",
4614 TIPC_LOW_IMPORTANCE);
4615 PyModule_AddIntConstant(m, "TIPC_MEDIUM_IMPORTANCE",
4616 TIPC_MEDIUM_IMPORTANCE);
4617 PyModule_AddIntConstant(m, "TIPC_HIGH_IMPORTANCE",
4618 TIPC_HIGH_IMPORTANCE);
4619 PyModule_AddIntConstant(m, "TIPC_CRITICAL_IMPORTANCE",
4620 TIPC_CRITICAL_IMPORTANCE);
Christian Heimes043d6f62008-01-07 17:19:16 +00004621
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004622 /* for subscriptions */
4623 PyModule_AddIntConstant(m, "TIPC_SUB_PORTS", TIPC_SUB_PORTS);
4624 PyModule_AddIntConstant(m, "TIPC_SUB_SERVICE", TIPC_SUB_SERVICE);
Christian Heimes25bb7832008-01-11 16:17:00 +00004625#ifdef TIPC_SUB_CANCEL
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004626 /* doesn't seem to be available everywhere */
4627 PyModule_AddIntConstant(m, "TIPC_SUB_CANCEL", TIPC_SUB_CANCEL);
Christian Heimes25bb7832008-01-11 16:17:00 +00004628#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004629 PyModule_AddIntConstant(m, "TIPC_WAIT_FOREVER", TIPC_WAIT_FOREVER);
4630 PyModule_AddIntConstant(m, "TIPC_PUBLISHED", TIPC_PUBLISHED);
4631 PyModule_AddIntConstant(m, "TIPC_WITHDRAWN", TIPC_WITHDRAWN);
4632 PyModule_AddIntConstant(m, "TIPC_SUBSCR_TIMEOUT", TIPC_SUBSCR_TIMEOUT);
4633 PyModule_AddIntConstant(m, "TIPC_CFG_SRV", TIPC_CFG_SRV);
4634 PyModule_AddIntConstant(m, "TIPC_TOP_SRV", TIPC_TOP_SRV);
Christian Heimes043d6f62008-01-07 17:19:16 +00004635#endif
4636
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004637 /* Socket types */
4638 PyModule_AddIntConstant(m, "SOCK_STREAM", SOCK_STREAM);
4639 PyModule_AddIntConstant(m, "SOCK_DGRAM", SOCK_DGRAM);
Guido van Rossumbcc20741998-08-04 22:53:56 +00004640/* We have incomplete socket support. */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004641 PyModule_AddIntConstant(m, "SOCK_RAW", SOCK_RAW);
4642 PyModule_AddIntConstant(m, "SOCK_SEQPACKET", SOCK_SEQPACKET);
Martin v. Löwiscf8f47e2002-12-11 13:10:57 +00004643#if defined(SOCK_RDM)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004644 PyModule_AddIntConstant(m, "SOCK_RDM", SOCK_RDM);
Guido van Rossumbcc20741998-08-04 22:53:56 +00004645#endif
Antoine Pitroub1c54962010-10-14 15:05:38 +00004646#ifdef SOCK_CLOEXEC
4647 PyModule_AddIntConstant(m, "SOCK_CLOEXEC", SOCK_CLOEXEC);
4648#endif
4649#ifdef SOCK_NONBLOCK
4650 PyModule_AddIntConstant(m, "SOCK_NONBLOCK", SOCK_NONBLOCK);
4651#endif
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004652
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004653#ifdef SO_DEBUG
4654 PyModule_AddIntConstant(m, "SO_DEBUG", SO_DEBUG);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004655#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004656#ifdef SO_ACCEPTCONN
4657 PyModule_AddIntConstant(m, "SO_ACCEPTCONN", SO_ACCEPTCONN);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004658#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004659#ifdef SO_REUSEADDR
4660 PyModule_AddIntConstant(m, "SO_REUSEADDR", SO_REUSEADDR);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004661#endif
Andrew M. Kuchling42851ab2004-07-10 14:19:21 +00004662#ifdef SO_EXCLUSIVEADDRUSE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004663 PyModule_AddIntConstant(m, "SO_EXCLUSIVEADDRUSE", SO_EXCLUSIVEADDRUSE);
Andrew M. Kuchling42851ab2004-07-10 14:19:21 +00004664#endif
4665
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004666#ifdef SO_KEEPALIVE
4667 PyModule_AddIntConstant(m, "SO_KEEPALIVE", SO_KEEPALIVE);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004668#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004669#ifdef SO_DONTROUTE
4670 PyModule_AddIntConstant(m, "SO_DONTROUTE", SO_DONTROUTE);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004671#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004672#ifdef SO_BROADCAST
4673 PyModule_AddIntConstant(m, "SO_BROADCAST", SO_BROADCAST);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004674#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004675#ifdef SO_USELOOPBACK
4676 PyModule_AddIntConstant(m, "SO_USELOOPBACK", SO_USELOOPBACK);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004677#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004678#ifdef SO_LINGER
4679 PyModule_AddIntConstant(m, "SO_LINGER", SO_LINGER);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004680#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004681#ifdef SO_OOBINLINE
4682 PyModule_AddIntConstant(m, "SO_OOBINLINE", SO_OOBINLINE);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004683#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004684#ifdef SO_REUSEPORT
4685 PyModule_AddIntConstant(m, "SO_REUSEPORT", SO_REUSEPORT);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004686#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004687#ifdef SO_SNDBUF
4688 PyModule_AddIntConstant(m, "SO_SNDBUF", SO_SNDBUF);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004689#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004690#ifdef SO_RCVBUF
4691 PyModule_AddIntConstant(m, "SO_RCVBUF", SO_RCVBUF);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004692#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004693#ifdef SO_SNDLOWAT
4694 PyModule_AddIntConstant(m, "SO_SNDLOWAT", SO_SNDLOWAT);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004695#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004696#ifdef SO_RCVLOWAT
4697 PyModule_AddIntConstant(m, "SO_RCVLOWAT", SO_RCVLOWAT);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004698#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004699#ifdef SO_SNDTIMEO
4700 PyModule_AddIntConstant(m, "SO_SNDTIMEO", SO_SNDTIMEO);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004701#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004702#ifdef SO_RCVTIMEO
4703 PyModule_AddIntConstant(m, "SO_RCVTIMEO", SO_RCVTIMEO);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004704#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004705#ifdef SO_ERROR
4706 PyModule_AddIntConstant(m, "SO_ERROR", SO_ERROR);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004707#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004708#ifdef SO_TYPE
4709 PyModule_AddIntConstant(m, "SO_TYPE", SO_TYPE);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004710#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004711#ifdef SO_SETFIB
4712 PyModule_AddIntConstant(m, "SO_SETFIB", SO_SETFIB);
Larry Hastingsf0f37952010-04-02 11:47:10 +00004713#endif
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004714
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004715 /* Maximum number of connections for "listen" */
4716#ifdef SOMAXCONN
4717 PyModule_AddIntConstant(m, "SOMAXCONN", SOMAXCONN);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004718#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004719 PyModule_AddIntConstant(m, "SOMAXCONN", 5); /* Common value */
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004720#endif
4721
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004722 /* Flags for send, recv */
4723#ifdef MSG_OOB
4724 PyModule_AddIntConstant(m, "MSG_OOB", MSG_OOB);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004725#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004726#ifdef MSG_PEEK
4727 PyModule_AddIntConstant(m, "MSG_PEEK", MSG_PEEK);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004728#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004729#ifdef MSG_DONTROUTE
4730 PyModule_AddIntConstant(m, "MSG_DONTROUTE", MSG_DONTROUTE);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004731#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004732#ifdef MSG_DONTWAIT
4733 PyModule_AddIntConstant(m, "MSG_DONTWAIT", MSG_DONTWAIT);
Guido van Rossum2c8bcb82000-04-25 21:34:53 +00004734#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004735#ifdef MSG_EOR
4736 PyModule_AddIntConstant(m, "MSG_EOR", MSG_EOR);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004737#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004738#ifdef MSG_TRUNC
4739 PyModule_AddIntConstant(m, "MSG_TRUNC", MSG_TRUNC);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004740#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004741#ifdef MSG_CTRUNC
4742 PyModule_AddIntConstant(m, "MSG_CTRUNC", MSG_CTRUNC);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004743#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004744#ifdef MSG_WAITALL
4745 PyModule_AddIntConstant(m, "MSG_WAITALL", MSG_WAITALL);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004746#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004747#ifdef MSG_BTAG
4748 PyModule_AddIntConstant(m, "MSG_BTAG", MSG_BTAG);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004749#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004750#ifdef MSG_ETAG
4751 PyModule_AddIntConstant(m, "MSG_ETAG", MSG_ETAG);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004752#endif
4753
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004754 /* Protocol level and numbers, usable for [gs]etsockopt */
4755#ifdef SOL_SOCKET
4756 PyModule_AddIntConstant(m, "SOL_SOCKET", SOL_SOCKET);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004757#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004758#ifdef SOL_IP
4759 PyModule_AddIntConstant(m, "SOL_IP", SOL_IP);
Guido van Rossum09be4091999-08-09 14:40:40 +00004760#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004761 PyModule_AddIntConstant(m, "SOL_IP", 0);
Guido van Rossum09be4091999-08-09 14:40:40 +00004762#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004763#ifdef SOL_IPX
4764 PyModule_AddIntConstant(m, "SOL_IPX", SOL_IPX);
Guido van Rossum09be4091999-08-09 14:40:40 +00004765#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004766#ifdef SOL_AX25
4767 PyModule_AddIntConstant(m, "SOL_AX25", SOL_AX25);
Guido van Rossum09be4091999-08-09 14:40:40 +00004768#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004769#ifdef SOL_ATALK
4770 PyModule_AddIntConstant(m, "SOL_ATALK", SOL_ATALK);
Guido van Rossum09be4091999-08-09 14:40:40 +00004771#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004772#ifdef SOL_NETROM
4773 PyModule_AddIntConstant(m, "SOL_NETROM", SOL_NETROM);
Guido van Rossum09be4091999-08-09 14:40:40 +00004774#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004775#ifdef SOL_ROSE
4776 PyModule_AddIntConstant(m, "SOL_ROSE", SOL_ROSE);
Guido van Rossum09be4091999-08-09 14:40:40 +00004777#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004778#ifdef SOL_TCP
4779 PyModule_AddIntConstant(m, "SOL_TCP", SOL_TCP);
Guido van Rossum09be4091999-08-09 14:40:40 +00004780#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004781 PyModule_AddIntConstant(m, "SOL_TCP", 6);
Guido van Rossum09be4091999-08-09 14:40:40 +00004782#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004783#ifdef SOL_UDP
4784 PyModule_AddIntConstant(m, "SOL_UDP", SOL_UDP);
Guido van Rossum09be4091999-08-09 14:40:40 +00004785#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004786 PyModule_AddIntConstant(m, "SOL_UDP", 17);
Guido van Rossum09be4091999-08-09 14:40:40 +00004787#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004788#ifdef IPPROTO_IP
4789 PyModule_AddIntConstant(m, "IPPROTO_IP", IPPROTO_IP);
Guido van Rossum578de301998-05-28 20:18:18 +00004790#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004791 PyModule_AddIntConstant(m, "IPPROTO_IP", 0);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004792#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004793#ifdef IPPROTO_HOPOPTS
4794 PyModule_AddIntConstant(m, "IPPROTO_HOPOPTS", IPPROTO_HOPOPTS);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004795#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004796#ifdef IPPROTO_ICMP
4797 PyModule_AddIntConstant(m, "IPPROTO_ICMP", IPPROTO_ICMP);
Guido van Rossum578de301998-05-28 20:18:18 +00004798#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004799 PyModule_AddIntConstant(m, "IPPROTO_ICMP", 1);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004800#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004801#ifdef IPPROTO_IGMP
4802 PyModule_AddIntConstant(m, "IPPROTO_IGMP", IPPROTO_IGMP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004803#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004804#ifdef IPPROTO_GGP
4805 PyModule_AddIntConstant(m, "IPPROTO_GGP", IPPROTO_GGP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004806#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004807#ifdef IPPROTO_IPV4
4808 PyModule_AddIntConstant(m, "IPPROTO_IPV4", IPPROTO_IPV4);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004809#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004810#ifdef IPPROTO_IPV6
4811 PyModule_AddIntConstant(m, "IPPROTO_IPV6", IPPROTO_IPV6);
Martin v. Löwisa0f17342003-10-03 13:56:20 +00004812#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004813#ifdef IPPROTO_IPIP
4814 PyModule_AddIntConstant(m, "IPPROTO_IPIP", IPPROTO_IPIP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004815#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004816#ifdef IPPROTO_TCP
4817 PyModule_AddIntConstant(m, "IPPROTO_TCP", IPPROTO_TCP);
Guido van Rossum578de301998-05-28 20:18:18 +00004818#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004819 PyModule_AddIntConstant(m, "IPPROTO_TCP", 6);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004820#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004821#ifdef IPPROTO_EGP
4822 PyModule_AddIntConstant(m, "IPPROTO_EGP", IPPROTO_EGP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004823#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004824#ifdef IPPROTO_PUP
4825 PyModule_AddIntConstant(m, "IPPROTO_PUP", IPPROTO_PUP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004826#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004827#ifdef IPPROTO_UDP
4828 PyModule_AddIntConstant(m, "IPPROTO_UDP", IPPROTO_UDP);
Guido van Rossum578de301998-05-28 20:18:18 +00004829#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004830 PyModule_AddIntConstant(m, "IPPROTO_UDP", 17);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004831#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004832#ifdef IPPROTO_IDP
4833 PyModule_AddIntConstant(m, "IPPROTO_IDP", IPPROTO_IDP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004834#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004835#ifdef IPPROTO_HELLO
4836 PyModule_AddIntConstant(m, "IPPROTO_HELLO", IPPROTO_HELLO);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004837#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004838#ifdef IPPROTO_ND
4839 PyModule_AddIntConstant(m, "IPPROTO_ND", IPPROTO_ND);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004840#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004841#ifdef IPPROTO_TP
4842 PyModule_AddIntConstant(m, "IPPROTO_TP", IPPROTO_TP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004843#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004844#ifdef IPPROTO_IPV6
4845 PyModule_AddIntConstant(m, "IPPROTO_IPV6", IPPROTO_IPV6);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004846#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004847#ifdef IPPROTO_ROUTING
4848 PyModule_AddIntConstant(m, "IPPROTO_ROUTING", IPPROTO_ROUTING);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004849#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004850#ifdef IPPROTO_FRAGMENT
4851 PyModule_AddIntConstant(m, "IPPROTO_FRAGMENT", IPPROTO_FRAGMENT);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004852#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004853#ifdef IPPROTO_RSVP
4854 PyModule_AddIntConstant(m, "IPPROTO_RSVP", IPPROTO_RSVP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004855#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004856#ifdef IPPROTO_GRE
4857 PyModule_AddIntConstant(m, "IPPROTO_GRE", IPPROTO_GRE);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004858#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004859#ifdef IPPROTO_ESP
4860 PyModule_AddIntConstant(m, "IPPROTO_ESP", IPPROTO_ESP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004861#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004862#ifdef IPPROTO_AH
4863 PyModule_AddIntConstant(m, "IPPROTO_AH", IPPROTO_AH);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004864#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004865#ifdef IPPROTO_MOBILE
4866 PyModule_AddIntConstant(m, "IPPROTO_MOBILE", IPPROTO_MOBILE);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004867#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004868#ifdef IPPROTO_ICMPV6
4869 PyModule_AddIntConstant(m, "IPPROTO_ICMPV6", IPPROTO_ICMPV6);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004870#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004871#ifdef IPPROTO_NONE
4872 PyModule_AddIntConstant(m, "IPPROTO_NONE", IPPROTO_NONE);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004873#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004874#ifdef IPPROTO_DSTOPTS
4875 PyModule_AddIntConstant(m, "IPPROTO_DSTOPTS", IPPROTO_DSTOPTS);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004876#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004877#ifdef IPPROTO_XTP
4878 PyModule_AddIntConstant(m, "IPPROTO_XTP", IPPROTO_XTP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004879#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004880#ifdef IPPROTO_EON
4881 PyModule_AddIntConstant(m, "IPPROTO_EON", IPPROTO_EON);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004882#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004883#ifdef IPPROTO_PIM
4884 PyModule_AddIntConstant(m, "IPPROTO_PIM", IPPROTO_PIM);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004885#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004886#ifdef IPPROTO_IPCOMP
4887 PyModule_AddIntConstant(m, "IPPROTO_IPCOMP", IPPROTO_IPCOMP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004888#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004889#ifdef IPPROTO_VRRP
4890 PyModule_AddIntConstant(m, "IPPROTO_VRRP", IPPROTO_VRRP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004891#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004892#ifdef IPPROTO_BIP
4893 PyModule_AddIntConstant(m, "IPPROTO_BIP", IPPROTO_BIP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004894#endif
4895/**/
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004896#ifdef IPPROTO_RAW
4897 PyModule_AddIntConstant(m, "IPPROTO_RAW", IPPROTO_RAW);
Guido van Rossum578de301998-05-28 20:18:18 +00004898#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004899 PyModule_AddIntConstant(m, "IPPROTO_RAW", 255);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004900#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004901#ifdef IPPROTO_MAX
4902 PyModule_AddIntConstant(m, "IPPROTO_MAX", IPPROTO_MAX);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004903#endif
4904
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004905 /* Some port configuration */
4906#ifdef IPPORT_RESERVED
4907 PyModule_AddIntConstant(m, "IPPORT_RESERVED", IPPORT_RESERVED);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004908#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004909 PyModule_AddIntConstant(m, "IPPORT_RESERVED", 1024);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004910#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004911#ifdef IPPORT_USERRESERVED
4912 PyModule_AddIntConstant(m, "IPPORT_USERRESERVED", IPPORT_USERRESERVED);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004913#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004914 PyModule_AddIntConstant(m, "IPPORT_USERRESERVED", 5000);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004915#endif
4916
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004917 /* Some reserved IP v.4 addresses */
4918#ifdef INADDR_ANY
4919 PyModule_AddIntConstant(m, "INADDR_ANY", INADDR_ANY);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004920#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004921 PyModule_AddIntConstant(m, "INADDR_ANY", 0x00000000);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004922#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004923#ifdef INADDR_BROADCAST
4924 PyModule_AddIntConstant(m, "INADDR_BROADCAST", INADDR_BROADCAST);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004925#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004926 PyModule_AddIntConstant(m, "INADDR_BROADCAST", 0xffffffff);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004927#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004928#ifdef INADDR_LOOPBACK
4929 PyModule_AddIntConstant(m, "INADDR_LOOPBACK", INADDR_LOOPBACK);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004930#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004931 PyModule_AddIntConstant(m, "INADDR_LOOPBACK", 0x7F000001);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004932#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004933#ifdef INADDR_UNSPEC_GROUP
4934 PyModule_AddIntConstant(m, "INADDR_UNSPEC_GROUP", INADDR_UNSPEC_GROUP);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004935#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004936 PyModule_AddIntConstant(m, "INADDR_UNSPEC_GROUP", 0xe0000000);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004937#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004938#ifdef INADDR_ALLHOSTS_GROUP
4939 PyModule_AddIntConstant(m, "INADDR_ALLHOSTS_GROUP",
4940 INADDR_ALLHOSTS_GROUP);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004941#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004942 PyModule_AddIntConstant(m, "INADDR_ALLHOSTS_GROUP", 0xe0000001);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004943#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004944#ifdef INADDR_MAX_LOCAL_GROUP
4945 PyModule_AddIntConstant(m, "INADDR_MAX_LOCAL_GROUP",
4946 INADDR_MAX_LOCAL_GROUP);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004947#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004948 PyModule_AddIntConstant(m, "INADDR_MAX_LOCAL_GROUP", 0xe00000ff);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004949#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004950#ifdef INADDR_NONE
4951 PyModule_AddIntConstant(m, "INADDR_NONE", INADDR_NONE);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004952#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004953 PyModule_AddIntConstant(m, "INADDR_NONE", 0xffffffff);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004954#endif
4955
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004956 /* IPv4 [gs]etsockopt options */
4957#ifdef IP_OPTIONS
4958 PyModule_AddIntConstant(m, "IP_OPTIONS", IP_OPTIONS);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004959#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004960#ifdef IP_HDRINCL
4961 PyModule_AddIntConstant(m, "IP_HDRINCL", IP_HDRINCL);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004962#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004963#ifdef IP_TOS
4964 PyModule_AddIntConstant(m, "IP_TOS", IP_TOS);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004965#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004966#ifdef IP_TTL
4967 PyModule_AddIntConstant(m, "IP_TTL", IP_TTL);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004968#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004969#ifdef IP_RECVOPTS
4970 PyModule_AddIntConstant(m, "IP_RECVOPTS", IP_RECVOPTS);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004971#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004972#ifdef IP_RECVRETOPTS
4973 PyModule_AddIntConstant(m, "IP_RECVRETOPTS", IP_RECVRETOPTS);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004974#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004975#ifdef IP_RECVDSTADDR
4976 PyModule_AddIntConstant(m, "IP_RECVDSTADDR", IP_RECVDSTADDR);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004977#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004978#ifdef IP_RETOPTS
4979 PyModule_AddIntConstant(m, "IP_RETOPTS", IP_RETOPTS);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004980#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004981#ifdef IP_MULTICAST_IF
4982 PyModule_AddIntConstant(m, "IP_MULTICAST_IF", IP_MULTICAST_IF);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004983#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004984#ifdef IP_MULTICAST_TTL
4985 PyModule_AddIntConstant(m, "IP_MULTICAST_TTL", IP_MULTICAST_TTL);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004986#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004987#ifdef IP_MULTICAST_LOOP
4988 PyModule_AddIntConstant(m, "IP_MULTICAST_LOOP", IP_MULTICAST_LOOP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004989#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004990#ifdef IP_ADD_MEMBERSHIP
4991 PyModule_AddIntConstant(m, "IP_ADD_MEMBERSHIP", IP_ADD_MEMBERSHIP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004992#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004993#ifdef IP_DROP_MEMBERSHIP
4994 PyModule_AddIntConstant(m, "IP_DROP_MEMBERSHIP", IP_DROP_MEMBERSHIP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004995#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004996#ifdef IP_DEFAULT_MULTICAST_TTL
4997 PyModule_AddIntConstant(m, "IP_DEFAULT_MULTICAST_TTL",
4998 IP_DEFAULT_MULTICAST_TTL);
Guido van Rossum09be4091999-08-09 14:40:40 +00004999#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005000#ifdef IP_DEFAULT_MULTICAST_LOOP
5001 PyModule_AddIntConstant(m, "IP_DEFAULT_MULTICAST_LOOP",
5002 IP_DEFAULT_MULTICAST_LOOP);
Guido van Rossum09be4091999-08-09 14:40:40 +00005003#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005004#ifdef IP_MAX_MEMBERSHIPS
5005 PyModule_AddIntConstant(m, "IP_MAX_MEMBERSHIPS", IP_MAX_MEMBERSHIPS);
Guido van Rossum09be4091999-08-09 14:40:40 +00005006#endif
5007
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005008 /* IPv6 [gs]etsockopt options, defined in RFC2553 */
5009#ifdef IPV6_JOIN_GROUP
5010 PyModule_AddIntConstant(m, "IPV6_JOIN_GROUP", IPV6_JOIN_GROUP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00005011#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005012#ifdef IPV6_LEAVE_GROUP
5013 PyModule_AddIntConstant(m, "IPV6_LEAVE_GROUP", IPV6_LEAVE_GROUP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00005014#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005015#ifdef IPV6_MULTICAST_HOPS
5016 PyModule_AddIntConstant(m, "IPV6_MULTICAST_HOPS", IPV6_MULTICAST_HOPS);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00005017#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005018#ifdef IPV6_MULTICAST_IF
5019 PyModule_AddIntConstant(m, "IPV6_MULTICAST_IF", IPV6_MULTICAST_IF);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00005020#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005021#ifdef IPV6_MULTICAST_LOOP
5022 PyModule_AddIntConstant(m, "IPV6_MULTICAST_LOOP", IPV6_MULTICAST_LOOP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00005023#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005024#ifdef IPV6_UNICAST_HOPS
5025 PyModule_AddIntConstant(m, "IPV6_UNICAST_HOPS", IPV6_UNICAST_HOPS);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00005026#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005027 /* Additional IPV6 socket options, defined in RFC 3493 */
Martin v. Löwisda91d022003-12-30 11:14:01 +00005028#ifdef IPV6_V6ONLY
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005029 PyModule_AddIntConstant(m, "IPV6_V6ONLY", IPV6_V6ONLY);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005030#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005031 /* Advanced IPV6 socket options, from RFC 3542 */
Martin v. Löwisda91d022003-12-30 11:14:01 +00005032#ifdef IPV6_CHECKSUM
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005033 PyModule_AddIntConstant(m, "IPV6_CHECKSUM", IPV6_CHECKSUM);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005034#endif
5035#ifdef IPV6_DONTFRAG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005036 PyModule_AddIntConstant(m, "IPV6_DONTFRAG", IPV6_DONTFRAG);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005037#endif
5038#ifdef IPV6_DSTOPTS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005039 PyModule_AddIntConstant(m, "IPV6_DSTOPTS", IPV6_DSTOPTS);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005040#endif
5041#ifdef IPV6_HOPLIMIT
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005042 PyModule_AddIntConstant(m, "IPV6_HOPLIMIT", IPV6_HOPLIMIT);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005043#endif
5044#ifdef IPV6_HOPOPTS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005045 PyModule_AddIntConstant(m, "IPV6_HOPOPTS", IPV6_HOPOPTS);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005046#endif
5047#ifdef IPV6_NEXTHOP
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005048 PyModule_AddIntConstant(m, "IPV6_NEXTHOP", IPV6_NEXTHOP);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005049#endif
5050#ifdef IPV6_PATHMTU
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005051 PyModule_AddIntConstant(m, "IPV6_PATHMTU", IPV6_PATHMTU);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005052#endif
5053#ifdef IPV6_PKTINFO
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005054 PyModule_AddIntConstant(m, "IPV6_PKTINFO", IPV6_PKTINFO);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005055#endif
5056#ifdef IPV6_RECVDSTOPTS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005057 PyModule_AddIntConstant(m, "IPV6_RECVDSTOPTS", IPV6_RECVDSTOPTS);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005058#endif
5059#ifdef IPV6_RECVHOPLIMIT
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005060 PyModule_AddIntConstant(m, "IPV6_RECVHOPLIMIT", IPV6_RECVHOPLIMIT);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005061#endif
5062#ifdef IPV6_RECVHOPOPTS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005063 PyModule_AddIntConstant(m, "IPV6_RECVHOPOPTS", IPV6_RECVHOPOPTS);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005064#endif
5065#ifdef IPV6_RECVPKTINFO
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005066 PyModule_AddIntConstant(m, "IPV6_RECVPKTINFO", IPV6_RECVPKTINFO);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005067#endif
5068#ifdef IPV6_RECVRTHDR
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005069 PyModule_AddIntConstant(m, "IPV6_RECVRTHDR", IPV6_RECVRTHDR);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005070#endif
5071#ifdef IPV6_RECVTCLASS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005072 PyModule_AddIntConstant(m, "IPV6_RECVTCLASS", IPV6_RECVTCLASS);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005073#endif
5074#ifdef IPV6_RTHDR
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005075 PyModule_AddIntConstant(m, "IPV6_RTHDR", IPV6_RTHDR);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005076#endif
5077#ifdef IPV6_RTHDRDSTOPTS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005078 PyModule_AddIntConstant(m, "IPV6_RTHDRDSTOPTS", IPV6_RTHDRDSTOPTS);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005079#endif
5080#ifdef IPV6_RTHDR_TYPE_0
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005081 PyModule_AddIntConstant(m, "IPV6_RTHDR_TYPE_0", IPV6_RTHDR_TYPE_0);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005082#endif
5083#ifdef IPV6_RECVPATHMTU
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005084 PyModule_AddIntConstant(m, "IPV6_RECVPATHMTU", IPV6_RECVPATHMTU);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005085#endif
5086#ifdef IPV6_TCLASS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005087 PyModule_AddIntConstant(m, "IPV6_TCLASS", IPV6_TCLASS);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005088#endif
5089#ifdef IPV6_USE_MIN_MTU
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005090 PyModule_AddIntConstant(m, "IPV6_USE_MIN_MTU", IPV6_USE_MIN_MTU);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005091#endif
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00005092
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005093 /* TCP options */
5094#ifdef TCP_NODELAY
5095 PyModule_AddIntConstant(m, "TCP_NODELAY", TCP_NODELAY);
Guido van Rossum09be4091999-08-09 14:40:40 +00005096#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005097#ifdef TCP_MAXSEG
5098 PyModule_AddIntConstant(m, "TCP_MAXSEG", TCP_MAXSEG);
Guido van Rossum09be4091999-08-09 14:40:40 +00005099#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005100#ifdef TCP_CORK
5101 PyModule_AddIntConstant(m, "TCP_CORK", TCP_CORK);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00005102#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005103#ifdef TCP_KEEPIDLE
5104 PyModule_AddIntConstant(m, "TCP_KEEPIDLE", TCP_KEEPIDLE);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00005105#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005106#ifdef TCP_KEEPINTVL
5107 PyModule_AddIntConstant(m, "TCP_KEEPINTVL", TCP_KEEPINTVL);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00005108#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005109#ifdef TCP_KEEPCNT
5110 PyModule_AddIntConstant(m, "TCP_KEEPCNT", TCP_KEEPCNT);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00005111#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005112#ifdef TCP_SYNCNT
5113 PyModule_AddIntConstant(m, "TCP_SYNCNT", TCP_SYNCNT);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00005114#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005115#ifdef TCP_LINGER2
5116 PyModule_AddIntConstant(m, "TCP_LINGER2", TCP_LINGER2);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00005117#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005118#ifdef TCP_DEFER_ACCEPT
5119 PyModule_AddIntConstant(m, "TCP_DEFER_ACCEPT", TCP_DEFER_ACCEPT);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00005120#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005121#ifdef TCP_WINDOW_CLAMP
5122 PyModule_AddIntConstant(m, "TCP_WINDOW_CLAMP", TCP_WINDOW_CLAMP);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00005123#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005124#ifdef TCP_INFO
5125 PyModule_AddIntConstant(m, "TCP_INFO", TCP_INFO);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00005126#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005127#ifdef TCP_QUICKACK
5128 PyModule_AddIntConstant(m, "TCP_QUICKACK", TCP_QUICKACK);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00005129#endif
5130
Guido van Rossum09be4091999-08-09 14:40:40 +00005131
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005132 /* IPX options */
5133#ifdef IPX_TYPE
5134 PyModule_AddIntConstant(m, "IPX_TYPE", IPX_TYPE);
Guido van Rossum09be4091999-08-09 14:40:40 +00005135#endif
Guido van Rossum4f199ea1998-04-09 20:56:35 +00005136
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005137 /* get{addr,name}info parameters */
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005138#ifdef EAI_ADDRFAMILY
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005139 PyModule_AddIntConstant(m, "EAI_ADDRFAMILY", EAI_ADDRFAMILY);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005140#endif
5141#ifdef EAI_AGAIN
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005142 PyModule_AddIntConstant(m, "EAI_AGAIN", EAI_AGAIN);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005143#endif
5144#ifdef EAI_BADFLAGS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005145 PyModule_AddIntConstant(m, "EAI_BADFLAGS", EAI_BADFLAGS);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005146#endif
5147#ifdef EAI_FAIL
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005148 PyModule_AddIntConstant(m, "EAI_FAIL", EAI_FAIL);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005149#endif
5150#ifdef EAI_FAMILY
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005151 PyModule_AddIntConstant(m, "EAI_FAMILY", EAI_FAMILY);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005152#endif
5153#ifdef EAI_MEMORY
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005154 PyModule_AddIntConstant(m, "EAI_MEMORY", EAI_MEMORY);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005155#endif
5156#ifdef EAI_NODATA
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005157 PyModule_AddIntConstant(m, "EAI_NODATA", EAI_NODATA);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005158#endif
5159#ifdef EAI_NONAME
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005160 PyModule_AddIntConstant(m, "EAI_NONAME", EAI_NONAME);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005161#endif
Martin v. Löwisda91d022003-12-30 11:14:01 +00005162#ifdef EAI_OVERFLOW
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005163 PyModule_AddIntConstant(m, "EAI_OVERFLOW", EAI_OVERFLOW);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005164#endif
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005165#ifdef EAI_SERVICE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005166 PyModule_AddIntConstant(m, "EAI_SERVICE", EAI_SERVICE);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005167#endif
5168#ifdef EAI_SOCKTYPE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005169 PyModule_AddIntConstant(m, "EAI_SOCKTYPE", EAI_SOCKTYPE);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005170#endif
5171#ifdef EAI_SYSTEM
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005172 PyModule_AddIntConstant(m, "EAI_SYSTEM", EAI_SYSTEM);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005173#endif
5174#ifdef EAI_BADHINTS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005175 PyModule_AddIntConstant(m, "EAI_BADHINTS", EAI_BADHINTS);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005176#endif
5177#ifdef EAI_PROTOCOL
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005178 PyModule_AddIntConstant(m, "EAI_PROTOCOL", EAI_PROTOCOL);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005179#endif
5180#ifdef EAI_MAX
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005181 PyModule_AddIntConstant(m, "EAI_MAX", EAI_MAX);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005182#endif
5183#ifdef AI_PASSIVE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005184 PyModule_AddIntConstant(m, "AI_PASSIVE", AI_PASSIVE);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005185#endif
5186#ifdef AI_CANONNAME
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005187 PyModule_AddIntConstant(m, "AI_CANONNAME", AI_CANONNAME);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005188#endif
5189#ifdef AI_NUMERICHOST
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005190 PyModule_AddIntConstant(m, "AI_NUMERICHOST", AI_NUMERICHOST);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005191#endif
Martin v. Löwisda91d022003-12-30 11:14:01 +00005192#ifdef AI_NUMERICSERV
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005193 PyModule_AddIntConstant(m, "AI_NUMERICSERV", AI_NUMERICSERV);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005194#endif
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005195#ifdef AI_MASK
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005196 PyModule_AddIntConstant(m, "AI_MASK", AI_MASK);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005197#endif
5198#ifdef AI_ALL
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005199 PyModule_AddIntConstant(m, "AI_ALL", AI_ALL);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005200#endif
5201#ifdef AI_V4MAPPED_CFG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005202 PyModule_AddIntConstant(m, "AI_V4MAPPED_CFG", AI_V4MAPPED_CFG);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005203#endif
5204#ifdef AI_ADDRCONFIG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005205 PyModule_AddIntConstant(m, "AI_ADDRCONFIG", AI_ADDRCONFIG);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005206#endif
5207#ifdef AI_V4MAPPED
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005208 PyModule_AddIntConstant(m, "AI_V4MAPPED", AI_V4MAPPED);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005209#endif
5210#ifdef AI_DEFAULT
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005211 PyModule_AddIntConstant(m, "AI_DEFAULT", AI_DEFAULT);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005212#endif
5213#ifdef NI_MAXHOST
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005214 PyModule_AddIntConstant(m, "NI_MAXHOST", NI_MAXHOST);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005215#endif
5216#ifdef NI_MAXSERV
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005217 PyModule_AddIntConstant(m, "NI_MAXSERV", NI_MAXSERV);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005218#endif
5219#ifdef NI_NOFQDN
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005220 PyModule_AddIntConstant(m, "NI_NOFQDN", NI_NOFQDN);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005221#endif
5222#ifdef NI_NUMERICHOST
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005223 PyModule_AddIntConstant(m, "NI_NUMERICHOST", NI_NUMERICHOST);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005224#endif
5225#ifdef NI_NAMEREQD
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005226 PyModule_AddIntConstant(m, "NI_NAMEREQD", NI_NAMEREQD);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005227#endif
5228#ifdef NI_NUMERICSERV
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005229 PyModule_AddIntConstant(m, "NI_NUMERICSERV", NI_NUMERICSERV);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005230#endif
5231#ifdef NI_DGRAM
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005232 PyModule_AddIntConstant(m, "NI_DGRAM", NI_DGRAM);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005233#endif
5234
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005235 /* shutdown() parameters */
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005236#ifdef SHUT_RD
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005237 PyModule_AddIntConstant(m, "SHUT_RD", SHUT_RD);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005238#elif defined(SD_RECEIVE)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005239 PyModule_AddIntConstant(m, "SHUT_RD", SD_RECEIVE);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005240#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005241 PyModule_AddIntConstant(m, "SHUT_RD", 0);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005242#endif
5243#ifdef SHUT_WR
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005244 PyModule_AddIntConstant(m, "SHUT_WR", SHUT_WR);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005245#elif defined(SD_SEND)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005246 PyModule_AddIntConstant(m, "SHUT_WR", SD_SEND);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005247#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005248 PyModule_AddIntConstant(m, "SHUT_WR", 1);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005249#endif
5250#ifdef SHUT_RDWR
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005251 PyModule_AddIntConstant(m, "SHUT_RDWR", SHUT_RDWR);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005252#elif defined(SD_BOTH)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005253 PyModule_AddIntConstant(m, "SHUT_RDWR", SD_BOTH);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005254#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005255 PyModule_AddIntConstant(m, "SHUT_RDWR", 2);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005256#endif
5257
Christian Heimesfaf2f632008-01-06 16:59:19 +00005258#ifdef SIO_RCVALL
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005259 {
5260 DWORD codes[] = {SIO_RCVALL, SIO_KEEPALIVE_VALS};
5261 const char *names[] = {"SIO_RCVALL", "SIO_KEEPALIVE_VALS"};
5262 int i;
5263 for(i = 0; i<sizeof(codes)/sizeof(*codes); ++i) {
5264 PyObject *tmp;
5265 tmp = PyLong_FromUnsignedLong(codes[i]);
5266 if (tmp == NULL)
5267 return NULL;
5268 PyModule_AddObject(m, names[i], tmp);
5269 }
5270 }
5271 PyModule_AddIntConstant(m, "RCVALL_OFF", RCVALL_OFF);
5272 PyModule_AddIntConstant(m, "RCVALL_ON", RCVALL_ON);
5273 PyModule_AddIntConstant(m, "RCVALL_SOCKETLEVELONLY", RCVALL_SOCKETLEVELONLY);
Amaury Forgeot d'Arc762681b2008-06-12 23:03:41 +00005274#ifdef RCVALL_IPLEVEL
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005275 PyModule_AddIntConstant(m, "RCVALL_IPLEVEL", RCVALL_IPLEVEL);
Amaury Forgeot d'Arc762681b2008-06-12 23:03:41 +00005276#endif
5277#ifdef RCVALL_MAX
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005278 PyModule_AddIntConstant(m, "RCVALL_MAX", RCVALL_MAX);
Amaury Forgeot d'Arc762681b2008-06-12 23:03:41 +00005279#endif
Christian Heimesfaf2f632008-01-06 16:59:19 +00005280#endif /* _MSTCPIP_ */
5281
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005282 /* Initialize gethostbyname lock */
Just van Rossum1040d2c2003-05-09 07:53:18 +00005283#if defined(USE_GETHOSTBYNAME_LOCK) || defined(USE_GETADDRINFO_LOCK)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005284 netdb_lock = PyThread_allocate_lock();
Guido van Rossum4f199ea1998-04-09 20:56:35 +00005285#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005286 return m;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00005287}
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005288
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00005289
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005290#ifndef HAVE_INET_PTON
Christian Heimes96e7b3d2007-11-20 06:51:17 +00005291#if !defined(NTDDI_VERSION) || (NTDDI_VERSION < NTDDI_LONGHORN)
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00005292
5293/* Simplistic emulation code for inet_pton that only works for IPv4 */
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00005294/* These are not exposed because they do not set errno properly */
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00005295
Guido van Rossum3eede5a2002-06-07 02:08:35 +00005296int
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00005297inet_pton(int af, const char *src, void *dst)
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005298{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005299 if (af == AF_INET) {
Benjamin Petersonf91df042009-02-13 02:50:59 +00005300#if (SIZEOF_INT != 4)
5301#error "Not sure if in_addr_t exists and int is not 32-bits."
5302#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005303 unsigned int packed_addr;
5304 packed_addr = inet_addr(src);
5305 if (packed_addr == INADDR_NONE)
5306 return 0;
5307 memcpy(dst, &packed_addr, 4);
5308 return 1;
5309 }
5310 /* Should set errno to EAFNOSUPPORT */
5311 return -1;
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005312}
5313
Martin v. Löwisc925b1532001-07-21 09:42:15 +00005314const char *
5315inet_ntop(int af, const void *src, char *dst, socklen_t size)
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005316{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005317 if (af == AF_INET) {
5318 struct in_addr packed_addr;
5319 if (size < 16)
5320 /* Should set errno to ENOSPC. */
5321 return NULL;
5322 memcpy(&packed_addr, src, sizeof(packed_addr));
5323 return strncpy(dst, inet_ntoa(packed_addr), size);
5324 }
5325 /* Should set errno to EAFNOSUPPORT */
5326 return NULL;
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005327}
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00005328
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005329#endif
Christian Heimesb6150692007-11-15 23:37:07 +00005330#endif