blob: 701af2f0bf0b70e00e5fa6ac8ccb17e169c3ae92 [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)
Gregory P. Smith5ed2e772011-05-15 00:26:45 -070045- socket.if_nameindex() -> list of tuples (if_index, if_name)
46- socket.if_nametoindex(name) -> corresponding interface index
47- socket.if_indextoname(index) -> corresponding interface name
Guido van Rossum6574b3e1991-06-25 21:36:08 +000048- an Internet socket address is a pair (hostname, port)
49 where hostname can be anything recognized by gethostbyname()
50 (including the dd.dd.dd.dd notation) and port is in host byte order
51- where a hostname is returned, the dd.dd.dd.dd notation is used
Guido van Rossum27e177d1995-03-16 15:43:47 +000052- a UNIX domain socket address is a string specifying the pathname
Jeremy Hylton22308652001-02-02 03:23:09 +000053- an AF_PACKET socket address is a tuple containing a string
54 specifying the ethernet interface and an integer specifying
55 the Ethernet protocol number to be received. For example:
Jeremy Hyltondbfb6622001-02-02 19:55:17 +000056 ("eth0",0x1234). Optional 3rd,4th,5th elements in the tuple
Andrew M. Kuchlingb8e17172004-07-10 23:39:35 +000057 specify packet-type and ha-type/addr.
Christian Heimes043d6f62008-01-07 17:19:16 +000058- an AF_TIPC socket address is expressed as
59 (addr_type, v1, v2, v3 [, scope]); where addr_type can be one of:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000060 TIPC_ADDR_NAMESEQ, TIPC_ADDR_NAME, and TIPC_ADDR_ID;
Christian Heimes043d6f62008-01-07 17:19:16 +000061 and scope can be one of:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000062 TIPC_ZONE_SCOPE, TIPC_CLUSTER_SCOPE, and TIPC_NODE_SCOPE.
Christian Heimes043d6f62008-01-07 17:19:16 +000063 The meaning of v1, v2 and v3 depends on the value of addr_type:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000064 if addr_type is TIPC_ADDR_NAME:
65 v1 is the server type
66 v2 is the port identifier
67 v3 is ignored
68 if addr_type is TIPC_ADDR_NAMESEQ:
69 v1 is the server type
70 v2 is the lower port number
71 v3 is the upper port number
72 if addr_type is TIPC_ADDR_ID:
73 v1 is the node
74 v2 is the ref
75 v3 is ignored
Christian Heimes043d6f62008-01-07 17:19:16 +000076
Guido van Rossum6574b3e1991-06-25 21:36:08 +000077
Guido van Rossumc4fcfa32002-06-07 03:19:37 +000078Local naming conventions:
Guido van Rossum6574b3e1991-06-25 21:36:08 +000079
Guido van Rossumc4fcfa32002-06-07 03:19:37 +000080- names starting with sock_ are socket object methods
81- names starting with socket_ are module-level functions
82- names starting with PySocket are exported through socketmodule.h
Guido van Rossum30a685f1991-06-27 15:51:29 +000083
Guido van Rossum6574b3e1991-06-25 21:36:08 +000084*/
85
Thomas Wouters477c8d52006-05-27 19:21:47 +000086#ifdef __APPLE__
87 /*
88 * inet_aton is not available on OSX 10.3, yet we want to use a binary
89 * that was build on 10.4 or later to work on that release, weak linking
90 * comes to the rescue.
91 */
92# pragma weak inet_aton
93#endif
94
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000095#include "Python.h"
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000096#include "structmember.h"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000097
Guido van Rossum47dfa4a2003-04-25 05:48:32 +000098#undef MAX
99#define MAX(x, y) ((x) < (y) ? (y) : (x))
100
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000101/* Socket object documentation */
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000102PyDoc_STRVAR(sock_doc,
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000103"socket([family[, type[, proto]]]) -> socket object\n\
104\n\
105Open a socket of the given type. The family argument specifies the\n\
106address family; it defaults to AF_INET. The type argument specifies\n\
107whether this is a stream (SOCK_STREAM, this is the default)\n\
108or datagram (SOCK_DGRAM) socket. The protocol argument defaults to 0,\n\
109specifying the default protocol. Keyword arguments are accepted.\n\
110\n\
111A socket object represents one endpoint of a network connection.\n\
112\n\
113Methods of socket objects (keyword arguments not allowed):\n\
114\n\
Guido van Rossum39eb8fa2007-11-16 01:24:05 +0000115_accept() -- accept connection, returning new socket fd and client address\n\
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000116bind(addr) -- bind the socket to a local address\n\
117close() -- close the socket\n\
118connect(addr) -- connect the socket to a remote address\n\
119connect_ex(addr) -- connect, return an error code instead of an exception\n\
Guido van Rossum39eb8fa2007-11-16 01:24:05 +0000120_dup() -- return a new socket fd duplicated from fileno()\n\
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000121fileno() -- return underlying file descriptor\n\
122getpeername() -- return remote address [*]\n\
123getsockname() -- return local address\n\
124getsockopt(level, optname[, buflen]) -- get socket options\n\
125gettimeout() -- return timeout or None\n\
126listen(n) -- start listening for incoming connections\n\
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000127recv(buflen[, flags]) -- receive data\n\
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000128recv_into(buffer[, nbytes[, flags]]) -- receive data (into a buffer)\n\
Thomas Wouters477c8d52006-05-27 19:21:47 +0000129recvfrom(buflen[, flags]) -- receive data and sender\'s address\n\
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000130recvfrom_into(buffer[, nbytes, [, flags])\n\
Thomas Wouters477c8d52006-05-27 19:21:47 +0000131 -- receive data and sender\'s address (into a buffer)\n\
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000132sendall(data[, flags]) -- send all data\n\
133send(data[, flags]) -- send data, may not send all of it\n\
134sendto(data[, flags], addr) -- send data to a given address\n\
135setblocking(0 | 1) -- set or clear the blocking I/O flag\n\
136setsockopt(level, optname, value) -- set socket options\n\
137settimeout(None | float) -- set or clear the timeout\n\
138shutdown(how) -- shut down traffic in one or both directions\n\
Gregory P. Smith5ed2e772011-05-15 00:26:45 -0700139if_nameindex() -- return all network interface indices and names\n\
140if_nametoindex(name) -- returns the corresponding interface index\n\
141if_indextoname(index) -- returns the corresponding interface name\n\
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000142\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000143 [*] not available on all platforms!");
Guido van Rossum3baaa131999-03-22 21:44:51 +0000144
Walter Dörwaldf0dfc7a2003-10-20 14:01:56 +0000145/* XXX This is a terrible mess of platform-dependent preprocessor hacks.
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000146 I hope some day someone can clean this up please... */
147
Guido van Rossum9376b741999-09-15 22:01:40 +0000148/* Hacks for gethostbyname_r(). On some non-Linux platforms, the configure
149 script doesn't get this right, so we hardcode some platform checks below.
150 On the other hand, not all Linux versions agree, so there the settings
151 computed by the configure script are needed! */
152
153#ifndef linux
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000154# undef HAVE_GETHOSTBYNAME_R_3_ARG
155# undef HAVE_GETHOSTBYNAME_R_5_ARG
156# undef HAVE_GETHOSTBYNAME_R_6_ARG
Guido van Rossum9376b741999-09-15 22:01:40 +0000157#endif
Guido van Rossume7de2061999-03-24 17:24:33 +0000158
Guido van Rossum7a122991999-04-13 04:07:32 +0000159#ifndef WITH_THREAD
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000160# undef HAVE_GETHOSTBYNAME_R
Guido van Rossum7a122991999-04-13 04:07:32 +0000161#endif
162
Guido van Rossume7de2061999-03-24 17:24:33 +0000163#ifdef HAVE_GETHOSTBYNAME_R
Jesus Cea736e7fc2011-03-14 17:36:54 +0100164# if defined(_AIX)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000165# define HAVE_GETHOSTBYNAME_R_3_ARG
166# elif defined(__sun) || defined(__sgi)
167# define HAVE_GETHOSTBYNAME_R_5_ARG
168# elif defined(linux)
Guido van Rossum9376b741999-09-15 22:01:40 +0000169/* Rely on the configure script */
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000170# else
171# undef HAVE_GETHOSTBYNAME_R
172# endif
Guido van Rossume7de2061999-03-24 17:24:33 +0000173#endif
174
Guido van Rossum3eede5a2002-06-07 02:08:35 +0000175#if !defined(HAVE_GETHOSTBYNAME_R) && defined(WITH_THREAD) && \
176 !defined(MS_WINDOWS)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000177# define USE_GETHOSTBYNAME_LOCK
Guido van Rossum3baaa131999-03-22 21:44:51 +0000178#endif
179
Hye-Shik Chang9ceebd52005-09-24 14:58:47 +0000180/* To use __FreeBSD_version */
181#ifdef HAVE_SYS_PARAM_H
182#include <sys/param.h>
183#endif
Just van Rossum1040d2c2003-05-09 07:53:18 +0000184/* On systems on which getaddrinfo() is believed to not be thread-safe,
Just van Rossum09aecd72003-05-09 08:03:44 +0000185 (this includes the getaddrinfo emulation) protect access with a lock. */
Hye-Shik Chang9ceebd52005-09-24 14:58:47 +0000186#if defined(WITH_THREAD) && (defined(__APPLE__) || \
187 (defined(__FreeBSD__) && __FreeBSD_version+0 < 503000) || \
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000188 defined(__OpenBSD__) || defined(__NetBSD__) || \
189 defined(__VMS) || !defined(HAVE_GETADDRINFO))
Just van Rossum1040d2c2003-05-09 07:53:18 +0000190#define USE_GETADDRINFO_LOCK
191#endif
192
193#ifdef USE_GETADDRINFO_LOCK
194#define ACQUIRE_GETADDRINFO_LOCK PyThread_acquire_lock(netdb_lock, 1);
195#define RELEASE_GETADDRINFO_LOCK PyThread_release_lock(netdb_lock);
196#else
197#define ACQUIRE_GETADDRINFO_LOCK
198#define RELEASE_GETADDRINFO_LOCK
199#endif
200
201#if defined(USE_GETHOSTBYNAME_LOCK) || defined(USE_GETADDRINFO_LOCK)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000202# include "pythread.h"
Guido van Rossum4f199ea1998-04-09 20:56:35 +0000203#endif
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000204
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000205#if defined(PYCC_VACPP)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000206# include <types.h>
207# include <io.h>
208# include <sys/ioctl.h>
209# include <utils.h>
210# include <ctype.h>
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000211#endif
212
Martin v. Löwis9e437302002-12-06 12:57:26 +0000213#if defined(__VMS)
Martin v. Löwis79acb9e2002-12-06 12:48:53 +0000214# include <ioctl.h>
215#endif
216
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000217#if defined(PYOS_OS2)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000218# define INCL_DOS
219# define INCL_DOSERRORS
220# define INCL_NOPMAPI
221# include <os2.h>
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000222#endif
223
Jeremy Hyltonfb509a32003-07-17 16:58:48 +0000224#if defined(__sgi) && _COMPILER_VERSION>700 && !_SGIAPI
Martin v. Löwis8eb92a02002-09-19 08:03:21 +0000225/* make sure that the reentrant (gethostbyaddr_r etc)
226 functions are declared correctly if compiling with
227 MIPSPro 7.x in ANSI C mode (default) */
Jeremy Hyltonfb509a32003-07-17 16:58:48 +0000228
Thomas Wouters477c8d52006-05-27 19:21:47 +0000229/* XXX Using _SGIAPI is the wrong thing,
Jeremy Hyltonfb509a32003-07-17 16:58:48 +0000230 but I don't know what the right thing is. */
Trent Mick8ea5bdf2004-09-13 17:48:41 +0000231#undef _SGIAPI /* to avoid warning */
Martin v. Löwis8eb92a02002-09-19 08:03:21 +0000232#define _SGIAPI 1
Jeremy Hyltonfb509a32003-07-17 16:58:48 +0000233
Trent Mick8ea5bdf2004-09-13 17:48:41 +0000234#undef _XOPEN_SOURCE
235#include <sys/socket.h>
236#include <sys/types.h>
237#include <netinet/in.h>
238#ifdef _SS_ALIGNSIZE
239#define HAVE_GETADDRINFO 1
240#define HAVE_GETNAMEINFO 1
241#endif
242
Jeremy Hyltonfb509a32003-07-17 16:58:48 +0000243#define HAVE_INET_PTON
244#include <netdb.h>
Martin v. Löwis8eb92a02002-09-19 08:03:21 +0000245#endif
246
Thomas Wouters477c8d52006-05-27 19:21:47 +0000247/* Irix 6.5 fails to define this variable at all. This is needed
248 for both GCC and SGI's compiler. I'd say that the SGI headers
Georg Brandldbd83392006-02-20 09:42:33 +0000249 are just busted. Same thing for Solaris. */
250#if (defined(__sgi) || defined(sun)) && !defined(INET_ADDRSTRLEN)
Anthony Baxterbab23cf2003-10-04 08:00:49 +0000251#define INET_ADDRSTRLEN 16
252#endif
253
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000254/* Generic includes */
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000255#ifdef HAVE_SYS_TYPES_H
Guido van Rossumb6775db1994-08-01 11:34:53 +0000256#include <sys/types.h>
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000257#endif
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000258
Marc-André Lemburg976ade62002-02-16 18:47:07 +0000259/* Generic socket object definitions and includes */
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000260#define PySocket_BUILDING_SOCKET
Marc-André Lemburgbb8b78b2002-02-16 18:44:52 +0000261#include "socketmodule.h"
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000262
263/* Addressing includes */
264
Guido van Rossum6f489d91996-06-28 20:15:15 +0000265#ifndef MS_WINDOWS
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000266
267/* Non-MS WINDOWS includes */
268# include <netdb.h>
Guido van Rossum5c9eb211999-08-20 18:21:51 +0000269
Guido van Rossum9376b741999-09-15 22:01:40 +0000270/* Headers needed for inet_ntoa() and inet_addr() */
Skip Montanaroeb33e5a2007-08-17 12:57:41 +0000271# if defined(PYOS_OS2) && defined(PYCC_VACPP)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000272# include <netdb.h>
Tim Peters603c6832001-11-05 02:45:59 +0000273typedef size_t socklen_t;
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000274# else
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000275# include <arpa/inet.h>
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000276# endif
Guido van Rossum5c9eb211999-08-20 18:21:51 +0000277
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000278# include <fcntl.h>
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000279
Guido van Rossume2ae77b2001-10-24 20:42:55 +0000280#else
Guido van Rossum48a680c2001-03-02 06:34:14 +0000281
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000282/* MS_WINDOWS includes */
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000283# ifdef HAVE_FCNTL_H
284# include <fcntl.h>
285# endif
Guido van Rossum48a680c2001-03-02 06:34:14 +0000286
Jeremy Hylton22308652001-02-02 03:23:09 +0000287#endif
288
Skip Montanaro7befb992004-02-10 16:50:21 +0000289#include <stddef.h>
Martin v. Löwisa45ecae2001-06-24 21:28:42 +0000290
291#ifndef offsetof
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000292# define offsetof(type, member) ((size_t)(&((type *)0)->member))
Martin v. Löwisa45ecae2001-06-24 21:28:42 +0000293#endif
294
Neal Norwitz39d22e52002-11-02 19:55:21 +0000295#ifndef O_NONBLOCK
296# define O_NONBLOCK O_NDELAY
Guido van Rossum3bbc62e1995-01-02 19:30:30 +0000297#endif
298
Trent Micka708d6e2004-09-07 17:48:26 +0000299/* include Python's addrinfo.h unless it causes trouble */
300#if defined(__sgi) && _COMPILER_VERSION>700 && defined(_SS_ALIGNSIZE)
301 /* Do not include addinfo.h on some newer IRIX versions.
302 * _SS_ALIGNSIZE is defined in sys/socket.h by 6.5.21,
303 * for example, but not by 6.5.10.
304 */
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000305#elif defined(_MSC_VER) && _MSC_VER>1201
Trent Micka708d6e2004-09-07 17:48:26 +0000306 /* Do not include addrinfo.h for MSVC7 or greater. 'addrinfo' and
307 * EAI_* constants are defined in (the already included) ws2tcpip.h.
308 */
309#else
310# include "addrinfo.h"
311#endif
Jason Tishlerc246cb72004-08-09 13:25:59 +0000312
Martin v. Löwisb9ab1592001-06-24 21:18:26 +0000313#ifndef HAVE_INET_PTON
Christian Heimes96e7b3d2007-11-20 06:51:17 +0000314#if !defined(NTDDI_VERSION) || (NTDDI_VERSION < NTDDI_LONGHORN)
Guido van Rossum3eede5a2002-06-07 02:08:35 +0000315int inet_pton(int af, const char *src, void *dst);
Martin v. Löwisc925b1532001-07-21 09:42:15 +0000316const char *inet_ntop(int af, const void *src, char *dst, socklen_t size);
Martin v. Löwisb9ab1592001-06-24 21:18:26 +0000317#endif
Christian Heimesb6150692007-11-15 23:37:07 +0000318#endif
Martin v. Löwisb9ab1592001-06-24 21:18:26 +0000319
Martin v. Löwisae26dc22001-08-03 10:02:29 +0000320#ifdef __APPLE__
321/* On OS X, getaddrinfo returns no error indication of lookup
322 failure, so we must use the emulation instead of the libinfo
323 implementation. Unfortunately, performing an autoconf test
324 for this bug would require DNS access for the machine performing
325 the configuration, which is not acceptable. Therefore, we
326 determine the bug just by checking for __APPLE__. If this bug
327 gets ever fixed, perhaps checking for sys/version.h would be
328 appropriate, which is 10/0 on the system with the bug. */
Jack Jansen84262fb2002-07-02 14:40:42 +0000329#ifndef HAVE_GETNAMEINFO
330/* This bug seems to be fixed in Jaguar. Ths easiest way I could
331 Find to check for Jaguar is that it has getnameinfo(), which
332 older releases don't have */
Martin v. Löwisae26dc22001-08-03 10:02:29 +0000333#undef HAVE_GETADDRINFO
Martin v. Löwisae26dc22001-08-03 10:02:29 +0000334#endif
Thomas Wouters477c8d52006-05-27 19:21:47 +0000335
336#ifdef HAVE_INET_ATON
337#define USE_INET_ATON_WEAKLINK
338#endif
339
Jack Jansen84262fb2002-07-02 14:40:42 +0000340#endif
Martin v. Löwisae26dc22001-08-03 10:02:29 +0000341
Martin v. Löwis01dfdb32001-06-23 16:30:13 +0000342/* I know this is a bad practice, but it is the easiest... */
Martin v. Löwisae26dc22001-08-03 10:02:29 +0000343#if !defined(HAVE_GETADDRINFO)
Martin v. Löwisfccac2e2003-05-01 05:20:46 +0000344/* avoid clashes with the C library definition of the symbol. */
345#define getaddrinfo fake_getaddrinfo
346#define gai_strerror fake_gai_strerror
347#define freeaddrinfo fake_freeaddrinfo
Martin v. Löwis01dfdb32001-06-23 16:30:13 +0000348#include "getaddrinfo.c"
349#endif
Martin v. Löwisae26dc22001-08-03 10:02:29 +0000350#if !defined(HAVE_GETNAMEINFO)
Martin v. Löwisfccac2e2003-05-01 05:20:46 +0000351#define getnameinfo fake_getnameinfo
Martin v. Löwis01dfdb32001-06-23 16:30:13 +0000352#include "getnameinfo.c"
353#endif
354
Guido van Rossum39eb8fa2007-11-16 01:24:05 +0000355#ifdef MS_WINDOWS
356/* On Windows a socket is really a handle not an fd */
357static SOCKET
358dup_socket(SOCKET handle)
359{
Daniel Stutzbach6c765282010-09-03 12:38:33 +0000360 WSAPROTOCOL_INFO info;
Guido van Rossum39eb8fa2007-11-16 01:24:05 +0000361
Daniel Stutzbach6c765282010-09-03 12:38:33 +0000362 if (WSADuplicateSocket(handle, GetCurrentProcessId(), &info))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000363 return INVALID_SOCKET;
Daniel Stutzbach6c765282010-09-03 12:38:33 +0000364
365 return WSASocket(FROM_PROTOCOL_INFO, FROM_PROTOCOL_INFO,
366 FROM_PROTOCOL_INFO, &info, 0, 0);
Guido van Rossum39eb8fa2007-11-16 01:24:05 +0000367}
Guido van Rossum2dd8ddd2000-04-21 20:33:00 +0000368#define SOCKETCLOSE closesocket
Guido van Rossum39eb8fa2007-11-16 01:24:05 +0000369#else
370/* On Unix we can use dup to duplicate the file descriptor of a socket*/
371#define dup_socket(fd) dup(fd)
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +0000372#endif
373
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000374#ifdef MS_WIN32
Guido van Rossum3eede5a2002-06-07 02:08:35 +0000375#define EAFNOSUPPORT WSAEAFNOSUPPORT
376#define snprintf _snprintf
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000377#endif
Fred Drakea04eaad2000-06-30 02:46:07 +0000378
Andrew MacIntyreba43e872002-03-03 03:03:52 +0000379#if defined(PYOS_OS2) && !defined(PYCC_GCC)
Guido van Rossum2dd8ddd2000-04-21 20:33:00 +0000380#define SOCKETCLOSE soclose
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000381#define NO_DUP /* Sockets are Not Actual File Handles under OS/2 */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000382#endif
383
Guido van Rossum2dd8ddd2000-04-21 20:33:00 +0000384#ifndef SOCKETCLOSE
385#define SOCKETCLOSE close
386#endif
387
Gregory P. Smith3e85dfd2010-10-17 00:43:10 +0000388#if (defined(HAVE_BLUETOOTH_H) || defined(HAVE_BLUETOOTH_BLUETOOTH_H)) && !defined(__NetBSD__) && !defined(__DragonFly__)
Hye-Shik Chang81268602004-02-02 06:05:24 +0000389#define USE_BLUETOOTH 1
390#if defined(__FreeBSD__)
391#define BTPROTO_L2CAP BLUETOOTH_PROTO_L2CAP
392#define BTPROTO_RFCOMM BLUETOOTH_PROTO_RFCOMM
Thomas Wouterscf297e42007-02-23 15:07:44 +0000393#define BTPROTO_HCI BLUETOOTH_PROTO_HCI
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000394#define SOL_HCI SOL_HCI_RAW
395#define HCI_FILTER SO_HCI_RAW_FILTER
Hye-Shik Chang81268602004-02-02 06:05:24 +0000396#define sockaddr_l2 sockaddr_l2cap
397#define sockaddr_rc sockaddr_rfcomm
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000398#define hci_dev hci_node
Hye-Shik Chang81268602004-02-02 06:05:24 +0000399#define _BT_L2_MEMB(sa, memb) ((sa)->l2cap_##memb)
400#define _BT_RC_MEMB(sa, memb) ((sa)->rfcomm_##memb)
Thomas Wouterscf297e42007-02-23 15:07:44 +0000401#define _BT_HCI_MEMB(sa, memb) ((sa)->hci_##memb)
Gregory P. Smith3e85dfd2010-10-17 00:43:10 +0000402#elif defined(__NetBSD__) || defined(__DragonFly__)
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000403#define sockaddr_l2 sockaddr_bt
404#define sockaddr_rc sockaddr_bt
Thomas Wouterscf297e42007-02-23 15:07:44 +0000405#define sockaddr_hci sockaddr_bt
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000406#define sockaddr_sco sockaddr_bt
Gregory P. Smith3e85dfd2010-10-17 00:43:10 +0000407#define SOL_HCI BTPROTO_HCI
408#define HCI_DATA_DIR SO_HCI_DIRECTION
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000409#define _BT_L2_MEMB(sa, memb) ((sa)->bt_##memb)
410#define _BT_RC_MEMB(sa, memb) ((sa)->bt_##memb)
Thomas Wouterscf297e42007-02-23 15:07:44 +0000411#define _BT_HCI_MEMB(sa, memb) ((sa)->bt_##memb)
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000412#define _BT_SCO_MEMB(sa, memb) ((sa)->bt_##memb)
Hye-Shik Chang81268602004-02-02 06:05:24 +0000413#else
Hye-Shik Chang81268602004-02-02 06:05:24 +0000414#define _BT_L2_MEMB(sa, memb) ((sa)->l2_##memb)
415#define _BT_RC_MEMB(sa, memb) ((sa)->rc_##memb)
Thomas Wouterscf297e42007-02-23 15:07:44 +0000416#define _BT_HCI_MEMB(sa, memb) ((sa)->hci_##memb)
Hye-Shik Chang81268602004-02-02 06:05:24 +0000417#define _BT_SCO_MEMB(sa, memb) ((sa)->sco_##memb)
418#endif
419#endif
420
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000421#ifdef __VMS
422/* TCP/IP Services for VMS uses a maximum send/recv buffer length */
423#define SEGMENT_SIZE (32 * 1024 -1)
424#endif
425
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000426#define SAS2SA(x) ((struct sockaddr *)(x))
Thomas Wouters89f507f2006-12-13 04:49:30 +0000427
Martin v. Löwise9416172003-05-03 10:12:45 +0000428/*
429 * Constants for getnameinfo()
430 */
431#if !defined(NI_MAXHOST)
432#define NI_MAXHOST 1025
433#endif
434#if !defined(NI_MAXSERV)
435#define NI_MAXSERV 32
436#endif
437
Guido van Rossum7d0a8262007-05-21 23:13:11 +0000438#ifndef INVALID_SOCKET /* MS defines this */
439#define INVALID_SOCKET (-1)
440#endif
441
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000442/* XXX There's a problem here: *static* functions are not supposed to have
443 a Py prefix (or use CapitalizedWords). Later... */
444
Guido van Rossum30a685f1991-06-27 15:51:29 +0000445/* Global variable holding the exception type for errors detected
446 by this module (but not argument type or memory errors, etc.). */
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000447static PyObject *socket_error;
448static PyObject *socket_herror;
449static PyObject *socket_gaierror;
Raymond Hettingeref7343c2003-06-29 03:08:05 +0000450static PyObject *socket_timeout;
Guido van Rossum30a685f1991-06-27 15:51:29 +0000451
Tim Peters643a7fc2002-02-17 04:13:21 +0000452/* A forward reference to the socket type object.
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000453 The sock_type variable contains pointers to various functions,
454 some of which call new_sockobject(), which uses sock_type, so
Tim Peters643a7fc2002-02-17 04:13:21 +0000455 there has to be a circular reference. */
Jeremy Hylton938ace62002-07-17 16:30:39 +0000456static PyTypeObject sock_type;
Guido van Rossum48a680c2001-03-02 06:34:14 +0000457
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000458#if defined(HAVE_POLL_H)
459#include <poll.h>
460#elif defined(HAVE_SYS_POLL_H)
461#include <sys/poll.h>
462#endif
463
Martin v. Löwisf84d1b92006-02-11 09:27:05 +0000464#ifdef Py_SOCKET_FD_CAN_BE_GE_FD_SETSIZE
465/* Platform can select file descriptors beyond FD_SETSIZE */
466#define IS_SELECTABLE(s) 1
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000467#elif defined(HAVE_POLL)
468/* Instead of select(), we'll use poll() since poll() works on any fd. */
469#define IS_SELECTABLE(s) 1
470/* Can we call select() with this socket without a buffer overrun? */
Martin v. Löwisf84d1b92006-02-11 09:27:05 +0000471#else
472/* POSIX says selecting file descriptors beyond FD_SETSIZE
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000473 has undefined behaviour. If there's no timeout left, we don't have to
474 call select, so it's a safe, little white lie. */
475#define IS_SELECTABLE(s) ((s)->sock_fd < FD_SETSIZE || s->sock_timeout <= 0.0)
Martin v. Löwisf84d1b92006-02-11 09:27:05 +0000476#endif
Neal Norwitz082b2df2006-02-07 07:04:46 +0000477
478static PyObject*
479select_error(void)
480{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000481 PyErr_SetString(socket_error, "unable to select on socket");
482 return NULL;
Neal Norwitz082b2df2006-02-07 07:04:46 +0000483}
484
Antoine Pitrou3e1fd272010-09-28 21:23:11 +0000485#ifdef MS_WINDOWS
Antoine Pitrouc8f01962010-09-28 22:03:27 +0000486#ifndef WSAEAGAIN
487#define WSAEAGAIN WSAEWOULDBLOCK
488#endif
Antoine Pitrou3e1fd272010-09-28 21:23:11 +0000489#define CHECK_ERRNO(expected) \
490 (WSAGetLastError() == WSA ## expected)
491#else
492#define CHECK_ERRNO(expected) \
493 (errno == expected)
494#endif
495
Guido van Rossum30a685f1991-06-27 15:51:29 +0000496/* Convenience function to raise an error according to errno
497 and return a NULL pointer from a function. */
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000498
Guido van Rossum73624e91994-10-10 17:59:00 +0000499static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000500set_error(void)
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000501{
Guido van Rossum8d665e61996-06-26 18:22:49 +0000502#ifdef MS_WINDOWS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000503 int err_no = WSAGetLastError();
504 /* PyErr_SetExcFromWindowsErr() invokes FormatMessage() which
505 recognizes the error codes used by both GetLastError() and
506 WSAGetLastError */
507 if (err_no)
508 return PyErr_SetExcFromWindowsErr(socket_error, err_no);
Guido van Rossum14ed0b21994-09-29 09:50:09 +0000509#endif
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000510
Andrew MacIntyreba43e872002-03-03 03:03:52 +0000511#if defined(PYOS_OS2) && !defined(PYCC_GCC)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000512 if (sock_errno() != NO_ERROR) {
513 APIRET rc;
514 ULONG msglen;
515 char outbuf[100];
516 int myerrorcode = sock_errno();
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000517
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000518 /* Retrieve socket-related error message from MPTN.MSG file */
519 rc = DosGetMessage(NULL, 0, outbuf, sizeof(outbuf),
520 myerrorcode - SOCBASEERR + 26,
521 "mptn.msg",
522 &msglen);
523 if (rc == NO_ERROR) {
524 PyObject *v;
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000525
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000526 /* OS/2 doesn't guarantee a terminator */
527 outbuf[msglen] = '\0';
528 if (strlen(outbuf) > 0) {
529 /* If non-empty msg, trim CRLF */
530 char *lastc = &outbuf[ strlen(outbuf)-1 ];
531 while (lastc > outbuf &&
532 isspace(Py_CHARMASK(*lastc))) {
533 /* Trim trailing whitespace (CRLF) */
534 *lastc-- = '\0';
535 }
536 }
537 v = Py_BuildValue("(is)", myerrorcode, outbuf);
538 if (v != NULL) {
539 PyErr_SetObject(socket_error, v);
540 Py_DECREF(v);
541 }
542 return NULL;
543 }
544 }
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000545#endif
546
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000547 return PyErr_SetFromErrno(socket_error);
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000548}
549
Guido van Rossum30a685f1991-06-27 15:51:29 +0000550
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000551static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000552set_herror(int h_error)
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000553{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000554 PyObject *v;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000555
556#ifdef HAVE_HSTRERROR
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000557 v = Py_BuildValue("(is)", h_error, (char *)hstrerror(h_error));
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000558#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000559 v = Py_BuildValue("(is)", h_error, "host not found");
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000560#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000561 if (v != NULL) {
562 PyErr_SetObject(socket_herror, v);
563 Py_DECREF(v);
564 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000565
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000566 return NULL;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000567}
568
569
570static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000571set_gaierror(int error)
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000572{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000573 PyObject *v;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000574
Martin v. Löwis272cb402002-03-01 08:31:07 +0000575#ifdef EAI_SYSTEM
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000576 /* EAI_SYSTEM is not available on Windows XP. */
577 if (error == EAI_SYSTEM)
578 return set_error();
Martin v. Löwis272cb402002-03-01 08:31:07 +0000579#endif
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000580
Martin v. Löwisf95dd0a2001-08-15 17:14:33 +0000581#ifdef HAVE_GAI_STRERROR
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000582 v = Py_BuildValue("(is)", error, gai_strerror(error));
Martin v. Löwisf95dd0a2001-08-15 17:14:33 +0000583#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000584 v = Py_BuildValue("(is)", error, "getaddrinfo failed");
Martin v. Löwisf95dd0a2001-08-15 17:14:33 +0000585#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000586 if (v != NULL) {
587 PyErr_SetObject(socket_gaierror, v);
588 Py_DECREF(v);
589 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000590
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000591 return NULL;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000592}
593
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000594#ifdef __VMS
595/* Function to send in segments */
596static int
597sendsegmented(int sock_fd, char *buf, int len, int flags)
598{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000599 int n = 0;
600 int remaining = len;
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000601
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000602 while (remaining > 0) {
603 unsigned int segment;
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000604
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000605 segment = (remaining >= SEGMENT_SIZE ? SEGMENT_SIZE : remaining);
606 n = send(sock_fd, buf, segment, flags);
607 if (n < 0) {
608 return n;
609 }
610 remaining -= segment;
611 buf += segment;
612 } /* end while */
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000613
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000614 return len;
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000615}
616#endif
617
Guido van Rossum3eede5a2002-06-07 02:08:35 +0000618/* Function to perform the setting of socket blocking mode
619 internally. block = (1 | 0). */
Guido van Rossum67f7a382002-06-06 21:08:16 +0000620static int
621internal_setblocking(PySocketSockObject *s, int block)
622{
Guido van Rossum67f7a382002-06-06 21:08:16 +0000623#ifndef MS_WINDOWS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000624 int delay_flag;
Guido van Rossum67f7a382002-06-06 21:08:16 +0000625#endif
Antoine Pitroub1c54962010-10-14 15:05:38 +0000626#ifdef SOCK_NONBLOCK
627 if (block)
628 s->sock_type &= (~SOCK_NONBLOCK);
629 else
630 s->sock_type |= SOCK_NONBLOCK;
631#endif
Guido van Rossum67f7a382002-06-06 21:08:16 +0000632
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000633 Py_BEGIN_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +0000634#ifndef MS_WINDOWS
635#if defined(PYOS_OS2) && !defined(PYCC_GCC)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000636 block = !block;
637 ioctl(s->sock_fd, FIONBIO, (caddr_t)&block, sizeof(block));
Martin v. Löwis79acb9e2002-12-06 12:48:53 +0000638#elif defined(__VMS)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000639 block = !block;
640 ioctl(s->sock_fd, FIONBIO, (unsigned int *)&block);
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000641#else /* !PYOS_OS2 && !__VMS */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000642 delay_flag = fcntl(s->sock_fd, F_GETFL, 0);
643 if (block)
644 delay_flag &= (~O_NONBLOCK);
645 else
646 delay_flag |= O_NONBLOCK;
647 fcntl(s->sock_fd, F_SETFL, delay_flag);
Guido van Rossum67f7a382002-06-06 21:08:16 +0000648#endif /* !PYOS_OS2 */
649#else /* MS_WINDOWS */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000650 block = !block;
651 ioctlsocket(s->sock_fd, FIONBIO, (u_long*)&block);
Guido van Rossum67f7a382002-06-06 21:08:16 +0000652#endif /* MS_WINDOWS */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000653 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +0000654
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000655 /* Since these don't return anything */
656 return 1;
Guido van Rossum67f7a382002-06-06 21:08:16 +0000657}
658
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000659/* Do a select()/poll() on the socket, if necessary (sock_timeout > 0).
Guido van Rossum11ba0942002-06-13 15:07:44 +0000660 The argument writing indicates the direction.
Raymond Hettingeref7343c2003-06-29 03:08:05 +0000661 This does not raise an exception; we'll let our caller do that
662 after they've reacquired the interpreter lock.
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000663 Returns 1 on timeout, -1 on error, 0 otherwise. */
Raymond Hettingeref7343c2003-06-29 03:08:05 +0000664static int
Antoine Pitrou3e1fd272010-09-28 21:23:11 +0000665internal_select_ex(PySocketSockObject *s, int writing, double interval)
Guido van Rossum67f7a382002-06-06 21:08:16 +0000666{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000667 int n;
Guido van Rossum11ba0942002-06-13 15:07:44 +0000668
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000669 /* Nothing to do unless we're in timeout mode (not non-blocking) */
670 if (s->sock_timeout <= 0.0)
671 return 0;
Guido van Rossum67f7a382002-06-06 21:08:16 +0000672
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000673 /* Guard against closed socket */
674 if (s->sock_fd < 0)
675 return 0;
Guido van Rossumad654902002-07-19 12:44:59 +0000676
Antoine Pitrou3e1fd272010-09-28 21:23:11 +0000677 /* Handling this condition here simplifies the select loops */
678 if (interval < 0.0)
679 return 1;
680
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000681 /* Prefer poll, if available, since you can poll() any fd
682 * which can't be done with select(). */
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000683#ifdef HAVE_POLL
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000684 {
685 struct pollfd pollfd;
686 int timeout;
Guido van Rossum67f7a382002-06-06 21:08:16 +0000687
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000688 pollfd.fd = s->sock_fd;
689 pollfd.events = writing ? POLLOUT : POLLIN;
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000690
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000691 /* s->sock_timeout is in seconds, timeout in ms */
Antoine Pitrou3e1fd272010-09-28 21:23:11 +0000692 timeout = (int)(interval * 1000 + 0.5);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000693 n = poll(&pollfd, 1, timeout);
694 }
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000695#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000696 {
697 /* Construct the arguments to select */
698 fd_set fds;
699 struct timeval tv;
Antoine Pitrou3e1fd272010-09-28 21:23:11 +0000700 tv.tv_sec = (int)interval;
701 tv.tv_usec = (int)((interval - tv.tv_sec) * 1e6);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000702 FD_ZERO(&fds);
703 FD_SET(s->sock_fd, &fds);
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000704
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000705 /* See if the socket is ready */
706 if (writing)
Antoine Pitrou19467d22010-08-17 19:33:30 +0000707 n = select(Py_SAFE_DOWNCAST(s->sock_fd+1, SOCKET_T, int),
708 NULL, &fds, NULL, &tv);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000709 else
Antoine Pitrou19467d22010-08-17 19:33:30 +0000710 n = select(Py_SAFE_DOWNCAST(s->sock_fd+1, SOCKET_T, int),
711 &fds, NULL, NULL, &tv);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000712 }
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000713#endif
Guido van Rossum39eb8fa2007-11-16 01:24:05 +0000714
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000715 if (n < 0)
716 return -1;
717 if (n == 0)
718 return 1;
719 return 0;
Guido van Rossum67f7a382002-06-06 21:08:16 +0000720}
721
Antoine Pitrou3e1fd272010-09-28 21:23:11 +0000722static int
723internal_select(PySocketSockObject *s, int writing)
724{
725 return internal_select_ex(s, writing, s->sock_timeout);
726}
727
728/*
729 Two macros for automatic retry of select() in case of false positives
730 (for example, select() could indicate a socket is ready for reading
731 but the data then discarded by the OS because of a wrong checksum).
732 Here is an example of use:
733
734 BEGIN_SELECT_LOOP(s)
735 Py_BEGIN_ALLOW_THREADS
736 timeout = internal_select_ex(s, 0, interval);
737 if (!timeout)
738 outlen = recv(s->sock_fd, cbuf, len, flags);
739 Py_END_ALLOW_THREADS
740 if (timeout == 1) {
741 PyErr_SetString(socket_timeout, "timed out");
742 return -1;
743 }
744 END_SELECT_LOOP(s)
745*/
746
747#define BEGIN_SELECT_LOOP(s) \
748 { \
749 _PyTime_timeval now, deadline = {0, 0}; \
750 double interval = s->sock_timeout; \
751 int has_timeout = s->sock_timeout > 0.0; \
752 if (has_timeout) { \
753 _PyTime_gettimeofday(&now); \
754 deadline = now; \
755 _PyTime_ADD_SECONDS(deadline, s->sock_timeout); \
756 } \
757 while (1) { \
758 errno = 0; \
759
760#define END_SELECT_LOOP(s) \
761 if (!has_timeout || \
762 (!CHECK_ERRNO(EWOULDBLOCK) && !CHECK_ERRNO(EAGAIN))) \
763 break; \
764 _PyTime_gettimeofday(&now); \
765 interval = _PyTime_INTERVAL(now, deadline); \
766 } \
767 } \
768
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000769/* Initialize a new socket object. */
770
Tim Petersa12b4cf2002-07-18 22:38:44 +0000771static double defaulttimeout = -1.0; /* Default timeout for new sockets */
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +0000772
Martin v. Löwis1a214512008-06-11 05:26:20 +0000773static void
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000774init_sockobject(PySocketSockObject *s,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000775 SOCKET_T fd, int family, int type, int proto)
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000776{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000777 s->sock_fd = fd;
778 s->sock_family = family;
779 s->sock_type = type;
780 s->sock_proto = proto;
Guido van Rossum67f7a382002-06-06 21:08:16 +0000781
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000782 s->errorhandler = &set_error;
Antoine Pitroub1c54962010-10-14 15:05:38 +0000783#ifdef SOCK_NONBLOCK
784 if (type & SOCK_NONBLOCK)
785 s->sock_timeout = 0.0;
786 else
787#endif
788 {
789 s->sock_timeout = defaulttimeout;
790 if (defaulttimeout >= 0.0)
791 internal_setblocking(s, 0);
792 }
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +0000793
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000794}
795
796
Guido van Rossum30a685f1991-06-27 15:51:29 +0000797/* Create a new socket object.
798 This just creates the object and initializes it.
799 If the creation fails, return NULL and set an exception (implicit
800 in NEWOBJ()). */
801
Guido van Rossum73624e91994-10-10 17:59:00 +0000802static PySocketSockObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000803new_sockobject(SOCKET_T fd, int family, int type, int proto)
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000804{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000805 PySocketSockObject *s;
806 s = (PySocketSockObject *)
807 PyType_GenericNew(&sock_type, NULL, NULL);
808 if (s != NULL)
809 init_sockobject(s, fd, family, type, proto);
810 return s;
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000811}
812
Guido van Rossum30a685f1991-06-27 15:51:29 +0000813
Guido van Rossum48a680c2001-03-02 06:34:14 +0000814/* Lock to allow python interpreter to continue, but only allow one
Just van Rossum1040d2c2003-05-09 07:53:18 +0000815 thread to be in gethostbyname or getaddrinfo */
816#if defined(USE_GETHOSTBYNAME_LOCK) || defined(USE_GETADDRINFO_LOCK)
817PyThread_type_lock netdb_lock;
Guido van Rossum4f199ea1998-04-09 20:56:35 +0000818#endif
819
820
Guido van Rossum30a685f1991-06-27 15:51:29 +0000821/* Convert a string specifying a host name or one of a few symbolic
822 names to a numeric IP address. This usually calls gethostbyname()
823 to do the work; the names "" and "<broadcast>" are special.
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000824 Return the length (IPv4 should be 4 bytes), or negative if
Guido van Rossum30a685f1991-06-27 15:51:29 +0000825 an error occurred; then an exception is raised. */
826
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000827static int
Martin v. Löwisddc6f472002-07-28 16:10:31 +0000828setipaddr(char *name, struct sockaddr *addr_ret, size_t addr_ret_size, int af)
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000829{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000830 struct addrinfo hints, *res;
831 int error;
832 int d1, d2, d3, d4;
833 char ch;
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000834
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000835 memset((void *) addr_ret, '\0', sizeof(*addr_ret));
836 if (name[0] == '\0') {
837 int siz;
838 memset(&hints, 0, sizeof(hints));
839 hints.ai_family = af;
840 hints.ai_socktype = SOCK_DGRAM; /*dummy*/
841 hints.ai_flags = AI_PASSIVE;
842 Py_BEGIN_ALLOW_THREADS
843 ACQUIRE_GETADDRINFO_LOCK
844 error = getaddrinfo(NULL, "0", &hints, &res);
845 Py_END_ALLOW_THREADS
846 /* We assume that those thread-unsafe getaddrinfo() versions
847 *are* safe regarding their return value, ie. that a
848 subsequent call to getaddrinfo() does not destroy the
849 outcome of the first call. */
850 RELEASE_GETADDRINFO_LOCK
851 if (error) {
852 set_gaierror(error);
853 return -1;
854 }
855 switch (res->ai_family) {
856 case AF_INET:
857 siz = 4;
858 break;
Martin v. Löwis44ddbde2001-12-02 10:15:37 +0000859#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000860 case AF_INET6:
861 siz = 16;
862 break;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000863#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000864 default:
865 freeaddrinfo(res);
866 PyErr_SetString(socket_error,
867 "unsupported address family");
868 return -1;
869 }
870 if (res->ai_next) {
871 freeaddrinfo(res);
872 PyErr_SetString(socket_error,
873 "wildcard resolved to multiple address");
874 return -1;
875 }
876 if (res->ai_addrlen < addr_ret_size)
877 addr_ret_size = res->ai_addrlen;
878 memcpy(addr_ret, res->ai_addr, addr_ret_size);
879 freeaddrinfo(res);
880 return siz;
881 }
882 if (name[0] == '<' && strcmp(name, "<broadcast>") == 0) {
883 struct sockaddr_in *sin;
884 if (af != AF_INET && af != AF_UNSPEC) {
885 PyErr_SetString(socket_error,
886 "address family mismatched");
887 return -1;
888 }
889 sin = (struct sockaddr_in *)addr_ret;
890 memset((void *) sin, '\0', sizeof(*sin));
891 sin->sin_family = AF_INET;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000892#ifdef HAVE_SOCKADDR_SA_LEN
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000893 sin->sin_len = sizeof(*sin);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000894#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000895 sin->sin_addr.s_addr = INADDR_BROADCAST;
896 return sizeof(sin->sin_addr);
897 }
898 if (sscanf(name, "%d.%d.%d.%d%c", &d1, &d2, &d3, &d4, &ch) == 4 &&
899 0 <= d1 && d1 <= 255 && 0 <= d2 && d2 <= 255 &&
900 0 <= d3 && d3 <= 255 && 0 <= d4 && d4 <= 255) {
901 struct sockaddr_in *sin;
902 sin = (struct sockaddr_in *)addr_ret;
903 sin->sin_addr.s_addr = htonl(
904 ((long) d1 << 24) | ((long) d2 << 16) |
905 ((long) d3 << 8) | ((long) d4 << 0));
906 sin->sin_family = AF_INET;
Anthony Baxter0e85f9d2003-05-02 15:40:46 +0000907#ifdef HAVE_SOCKADDR_SA_LEN
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000908 sin->sin_len = sizeof(*sin);
Anthony Baxter0e85f9d2003-05-02 15:40:46 +0000909#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000910 return 4;
911 }
912 memset(&hints, 0, sizeof(hints));
913 hints.ai_family = af;
914 Py_BEGIN_ALLOW_THREADS
915 ACQUIRE_GETADDRINFO_LOCK
916 error = getaddrinfo(name, NULL, &hints, &res);
Martin v. Löwis7c4b5fa2001-10-25 09:04:03 +0000917#if defined(__digital__) && defined(__unix__)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000918 if (error == EAI_NONAME && af == AF_UNSPEC) {
919 /* On Tru64 V5.1, numeric-to-addr conversion fails
920 if no address family is given. Assume IPv4 for now.*/
921 hints.ai_family = AF_INET;
922 error = getaddrinfo(name, NULL, &hints, &res);
923 }
Martin v. Löwis7c4b5fa2001-10-25 09:04:03 +0000924#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000925 Py_END_ALLOW_THREADS
926 RELEASE_GETADDRINFO_LOCK /* see comment in setipaddr() */
927 if (error) {
928 set_gaierror(error);
929 return -1;
930 }
931 if (res->ai_addrlen < addr_ret_size)
932 addr_ret_size = res->ai_addrlen;
933 memcpy((char *) addr_ret, res->ai_addr, addr_ret_size);
934 freeaddrinfo(res);
935 switch (addr_ret->sa_family) {
936 case AF_INET:
937 return 4;
Martin v. Löwis44ddbde2001-12-02 10:15:37 +0000938#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000939 case AF_INET6:
940 return 16;
Guido van Rossum955becc1999-03-22 20:14:53 +0000941#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000942 default:
943 PyErr_SetString(socket_error, "unknown address family");
944 return -1;
945 }
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000946}
947
Guido van Rossum30a685f1991-06-27 15:51:29 +0000948
Guido van Rossum30a685f1991-06-27 15:51:29 +0000949/* Create a string object representing an IP address.
950 This is always a string of the form 'dd.dd.dd.dd' (with variable
951 size numbers). */
952
Guido van Rossum73624e91994-10-10 17:59:00 +0000953static PyObject *
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000954makeipaddr(struct sockaddr *addr, int addrlen)
Guido van Rossum30a685f1991-06-27 15:51:29 +0000955{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000956 char buf[NI_MAXHOST];
957 int error;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000958
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000959 error = getnameinfo(addr, addrlen, buf, sizeof(buf), NULL, 0,
960 NI_NUMERICHOST);
961 if (error) {
962 set_gaierror(error);
963 return NULL;
964 }
965 return PyUnicode_FromString(buf);
Guido van Rossum30a685f1991-06-27 15:51:29 +0000966}
967
968
Martin v. Löwis558d9bf2004-06-03 09:24:42 +0000969#ifdef USE_BLUETOOTH
970/* Convert a string representation of a Bluetooth address into a numeric
971 address. Returns the length (6), or raises an exception and returns -1 if
972 an error occurred. */
973
974static int
975setbdaddr(char *name, bdaddr_t *bdaddr)
976{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000977 unsigned int b0, b1, b2, b3, b4, b5;
978 char ch;
979 int n;
Martin v. Löwis558d9bf2004-06-03 09:24:42 +0000980
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000981 n = sscanf(name, "%X:%X:%X:%X:%X:%X%c",
982 &b5, &b4, &b3, &b2, &b1, &b0, &ch);
983 if (n == 6 && (b0 | b1 | b2 | b3 | b4 | b5) < 256) {
984 bdaddr->b[0] = b0;
985 bdaddr->b[1] = b1;
986 bdaddr->b[2] = b2;
987 bdaddr->b[3] = b3;
988 bdaddr->b[4] = b4;
989 bdaddr->b[5] = b5;
990 return 6;
991 } else {
992 PyErr_SetString(socket_error, "bad bluetooth address");
993 return -1;
994 }
Martin v. Löwis558d9bf2004-06-03 09:24:42 +0000995}
996
997/* Create a string representation of the Bluetooth address. This is always a
998 string of the form 'XX:XX:XX:XX:XX:XX' where XX is a two digit hexadecimal
999 value (zero padded if necessary). */
1000
1001static PyObject *
1002makebdaddr(bdaddr_t *bdaddr)
1003{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001004 char buf[(6 * 2) + 5 + 1];
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001005
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001006 sprintf(buf, "%02X:%02X:%02X:%02X:%02X:%02X",
1007 bdaddr->b[5], bdaddr->b[4], bdaddr->b[3],
1008 bdaddr->b[2], bdaddr->b[1], bdaddr->b[0]);
1009 return PyUnicode_FromString(buf);
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001010}
1011#endif
1012
1013
Guido van Rossum30a685f1991-06-27 15:51:29 +00001014/* Create an object representing the given socket address,
1015 suitable for passing it back to bind(), connect() etc.
1016 The family field of the sockaddr structure is inspected
1017 to determine what kind of address it really is. */
1018
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001019/*ARGSUSED*/
Guido van Rossum73624e91994-10-10 17:59:00 +00001020static PyObject *
Antoine Pitrou19467d22010-08-17 19:33:30 +00001021makesockaddr(SOCKET_T sockfd, struct sockaddr *addr, size_t addrlen, int proto)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001022{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001023 if (addrlen == 0) {
1024 /* No address -- may be recvfrom() from known socket */
1025 Py_INCREF(Py_None);
1026 return Py_None;
1027 }
Guido van Rossum25bec8c1992-08-05 19:00:45 +00001028
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001029 switch (addr->sa_family) {
Guido van Rossum30a685f1991-06-27 15:51:29 +00001030
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001031 case AF_INET:
1032 {
1033 struct sockaddr_in *a;
1034 PyObject *addrobj = makeipaddr(addr, sizeof(*a));
1035 PyObject *ret = NULL;
1036 if (addrobj) {
1037 a = (struct sockaddr_in *)addr;
1038 ret = Py_BuildValue("Oi", addrobj, ntohs(a->sin_port));
1039 Py_DECREF(addrobj);
1040 }
1041 return ret;
1042 }
Guido van Rossum30a685f1991-06-27 15:51:29 +00001043
Andrew MacIntyred12dfbb2004-04-04 07:13:49 +00001044#if defined(AF_UNIX)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001045 case AF_UNIX:
1046 {
1047 struct sockaddr_un *a = (struct sockaddr_un *) addr;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001048#ifdef linux
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001049 if (a->sun_path[0] == 0) { /* Linux abstract namespace */
1050 addrlen -= offsetof(struct sockaddr_un, sun_path);
1051 return PyBytes_FromStringAndSize(a->sun_path, addrlen);
1052 }
1053 else
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001054#endif /* linux */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001055 {
1056 /* regular NULL-terminated string */
1057 return PyUnicode_FromString(a->sun_path);
1058 }
1059 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00001060#endif /* AF_UNIX */
1061
Martin v. Löwis11017b12006-01-14 18:12:57 +00001062#if defined(AF_NETLINK)
1063 case AF_NETLINK:
1064 {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001065 struct sockaddr_nl *a = (struct sockaddr_nl *) addr;
1066 return Py_BuildValue("II", a->nl_pid, a->nl_groups);
Martin v. Löwis11017b12006-01-14 18:12:57 +00001067 }
1068#endif /* AF_NETLINK */
1069
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00001070#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001071 case AF_INET6:
1072 {
1073 struct sockaddr_in6 *a;
1074 PyObject *addrobj = makeipaddr(addr, sizeof(*a));
1075 PyObject *ret = NULL;
1076 if (addrobj) {
1077 a = (struct sockaddr_in6 *)addr;
1078 ret = Py_BuildValue("Oiii",
1079 addrobj,
1080 ntohs(a->sin6_port),
1081 a->sin6_flowinfo,
1082 a->sin6_scope_id);
1083 Py_DECREF(addrobj);
1084 }
1085 return ret;
1086 }
Jeremy Hylton22308652001-02-02 03:23:09 +00001087#endif
Guido van Rossum30a685f1991-06-27 15:51:29 +00001088
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001089#ifdef USE_BLUETOOTH
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001090 case AF_BLUETOOTH:
1091 switch (proto) {
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001092
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001093 case BTPROTO_L2CAP:
1094 {
1095 struct sockaddr_l2 *a = (struct sockaddr_l2 *) addr;
1096 PyObject *addrobj = makebdaddr(&_BT_L2_MEMB(a, bdaddr));
1097 PyObject *ret = NULL;
1098 if (addrobj) {
1099 ret = Py_BuildValue("Oi",
1100 addrobj,
1101 _BT_L2_MEMB(a, psm));
1102 Py_DECREF(addrobj);
1103 }
1104 return ret;
1105 }
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001106
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001107 case BTPROTO_RFCOMM:
1108 {
1109 struct sockaddr_rc *a = (struct sockaddr_rc *) addr;
1110 PyObject *addrobj = makebdaddr(&_BT_RC_MEMB(a, bdaddr));
1111 PyObject *ret = NULL;
1112 if (addrobj) {
1113 ret = Py_BuildValue("Oi",
1114 addrobj,
1115 _BT_RC_MEMB(a, channel));
1116 Py_DECREF(addrobj);
1117 }
1118 return ret;
1119 }
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001120
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001121 case BTPROTO_HCI:
1122 {
1123 struct sockaddr_hci *a = (struct sockaddr_hci *) addr;
Gregory P. Smith3e85dfd2010-10-17 00:43:10 +00001124#if defined(__NetBSD__) || defined(__DragonFly__)
1125 return makebdaddr(&_BT_HCI_MEMB(a, bdaddr));
1126#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001127 PyObject *ret = NULL;
1128 ret = Py_BuildValue("i", _BT_HCI_MEMB(a, dev));
1129 return ret;
Gregory P. Smith3e85dfd2010-10-17 00:43:10 +00001130#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001131 }
Thomas Wouterscf297e42007-02-23 15:07:44 +00001132
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001133#if !defined(__FreeBSD__)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001134 case BTPROTO_SCO:
1135 {
1136 struct sockaddr_sco *a = (struct sockaddr_sco *) addr;
1137 return makebdaddr(&_BT_SCO_MEMB(a, bdaddr));
1138 }
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001139#endif
1140
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001141 default:
1142 PyErr_SetString(PyExc_ValueError,
1143 "Unknown Bluetooth protocol");
1144 return NULL;
1145 }
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001146#endif
1147
Antoine Pitroub156a462010-10-27 20:13:57 +00001148#if defined(HAVE_NETPACKET_PACKET_H) && defined(SIOCGIFNAME)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001149 case AF_PACKET:
1150 {
1151 struct sockaddr_ll *a = (struct sockaddr_ll *)addr;
1152 char *ifname = "";
1153 struct ifreq ifr;
1154 /* need to look up interface name give index */
1155 if (a->sll_ifindex) {
1156 ifr.ifr_ifindex = a->sll_ifindex;
1157 if (ioctl(sockfd, SIOCGIFNAME, &ifr) == 0)
1158 ifname = ifr.ifr_name;
1159 }
1160 return Py_BuildValue("shbhy#",
1161 ifname,
1162 ntohs(a->sll_protocol),
1163 a->sll_pkttype,
1164 a->sll_hatype,
1165 a->sll_addr,
1166 a->sll_halen);
1167 }
Jeremy Hylton22308652001-02-02 03:23:09 +00001168#endif
Guido van Rossum48a680c2001-03-02 06:34:14 +00001169
Christian Heimes043d6f62008-01-07 17:19:16 +00001170#ifdef HAVE_LINUX_TIPC_H
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001171 case AF_TIPC:
1172 {
1173 struct sockaddr_tipc *a = (struct sockaddr_tipc *) addr;
1174 if (a->addrtype == TIPC_ADDR_NAMESEQ) {
1175 return Py_BuildValue("IIIII",
1176 a->addrtype,
1177 a->addr.nameseq.type,
1178 a->addr.nameseq.lower,
1179 a->addr.nameseq.upper,
1180 a->scope);
1181 } else if (a->addrtype == TIPC_ADDR_NAME) {
1182 return Py_BuildValue("IIIII",
1183 a->addrtype,
1184 a->addr.name.name.type,
1185 a->addr.name.name.instance,
1186 a->addr.name.name.instance,
1187 a->scope);
1188 } else if (a->addrtype == TIPC_ADDR_ID) {
1189 return Py_BuildValue("IIIII",
1190 a->addrtype,
1191 a->addr.id.node,
1192 a->addr.id.ref,
1193 0,
1194 a->scope);
1195 } else {
1196 PyErr_SetString(PyExc_ValueError,
1197 "Invalid address type");
1198 return NULL;
1199 }
1200 }
Christian Heimes043d6f62008-01-07 17:19:16 +00001201#endif
1202
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001203 /* More cases here... */
Guido van Rossum30a685f1991-06-27 15:51:29 +00001204
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001205 default:
1206 /* If we don't know the address family, don't raise an
1207 exception -- return it as an (int, bytes) tuple. */
1208 return Py_BuildValue("iy#",
1209 addr->sa_family,
1210 addr->sa_data,
1211 sizeof(addr->sa_data));
Guido van Rossum25bec8c1992-08-05 19:00:45 +00001212
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001213 }
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001214}
1215
Guido van Rossum30a685f1991-06-27 15:51:29 +00001216
1217/* Parse a socket address argument according to the socket object's
1218 address family. Return 1 if the address was in the proper format,
1219 0 of not. The address is returned through addr_ret, its length
1220 through len_ret. */
1221
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001222static int
Guido van Rossum48a680c2001-03-02 06:34:14 +00001223getsockaddrarg(PySocketSockObject *s, PyObject *args,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001224 struct sockaddr *addr_ret, int *len_ret)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001225{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001226 switch (s->sock_family) {
Guido van Rossum30a685f1991-06-27 15:51:29 +00001227
Andrew MacIntyred12dfbb2004-04-04 07:13:49 +00001228#if defined(AF_UNIX)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001229 case AF_UNIX:
1230 {
1231 struct sockaddr_un* addr;
1232 char *path;
1233 int len;
1234 if (!PyArg_Parse(args, "s#", &path, &len))
1235 return 0;
Thomas Wouters89f507f2006-12-13 04:49:30 +00001236
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001237 addr = (struct sockaddr_un*)addr_ret;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001238#ifdef linux
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001239 if (len > 0 && path[0] == 0) {
1240 /* Linux abstract namespace extension */
1241 if (len > sizeof addr->sun_path) {
1242 PyErr_SetString(socket_error,
1243 "AF_UNIX path too long");
1244 return 0;
1245 }
1246 }
1247 else
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001248#endif /* linux */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001249 {
1250 /* regular NULL-terminated string */
1251 if (len >= sizeof addr->sun_path) {
1252 PyErr_SetString(socket_error,
1253 "AF_UNIX path too long");
1254 return 0;
1255 }
1256 addr->sun_path[len] = 0;
1257 }
1258 addr->sun_family = s->sock_family;
1259 memcpy(addr->sun_path, path, len);
Andrew MacIntyredaedf212004-04-11 12:03:57 +00001260#if defined(PYOS_OS2)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001261 *len_ret = sizeof(*addr);
Andrew MacIntyredaedf212004-04-11 12:03:57 +00001262#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001263 *len_ret = len + offsetof(struct sockaddr_un, sun_path);
Andrew MacIntyredaedf212004-04-11 12:03:57 +00001264#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001265 return 1;
1266 }
Guido van Rossumb6775db1994-08-01 11:34:53 +00001267#endif /* AF_UNIX */
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001268
Martin v. Löwis11017b12006-01-14 18:12:57 +00001269#if defined(AF_NETLINK)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001270 case AF_NETLINK:
1271 {
1272 struct sockaddr_nl* addr;
1273 int pid, groups;
1274 addr = (struct sockaddr_nl *)addr_ret;
1275 if (!PyTuple_Check(args)) {
1276 PyErr_Format(
1277 PyExc_TypeError,
1278 "getsockaddrarg: "
1279 "AF_NETLINK address must be tuple, not %.500s",
1280 Py_TYPE(args)->tp_name);
1281 return 0;
1282 }
1283 if (!PyArg_ParseTuple(args, "II:getsockaddrarg", &pid, &groups))
1284 return 0;
1285 addr->nl_family = AF_NETLINK;
1286 addr->nl_pid = pid;
1287 addr->nl_groups = groups;
1288 *len_ret = sizeof(*addr);
1289 return 1;
1290 }
Martin v. Löwis11017b12006-01-14 18:12:57 +00001291#endif
1292
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001293 case AF_INET:
1294 {
1295 struct sockaddr_in* addr;
1296 char *host;
1297 int port, result;
1298 if (!PyTuple_Check(args)) {
1299 PyErr_Format(
1300 PyExc_TypeError,
1301 "getsockaddrarg: "
1302 "AF_INET address must be tuple, not %.500s",
1303 Py_TYPE(args)->tp_name);
1304 return 0;
1305 }
1306 if (!PyArg_ParseTuple(args, "eti:getsockaddrarg",
1307 "idna", &host, &port))
1308 return 0;
1309 addr=(struct sockaddr_in*)addr_ret;
1310 result = setipaddr(host, (struct sockaddr *)addr,
1311 sizeof(*addr), AF_INET);
1312 PyMem_Free(host);
1313 if (result < 0)
1314 return 0;
1315 if (port < 0 || port > 0xffff) {
1316 PyErr_SetString(
1317 PyExc_OverflowError,
1318 "getsockaddrarg: port must be 0-65535.");
1319 return 0;
1320 }
1321 addr->sin_family = AF_INET;
1322 addr->sin_port = htons((short)port);
1323 *len_ret = sizeof *addr;
1324 return 1;
1325 }
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001326
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00001327#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001328 case AF_INET6:
1329 {
1330 struct sockaddr_in6* addr;
1331 char *host;
1332 int port, flowinfo, scope_id, result;
1333 flowinfo = scope_id = 0;
1334 if (!PyTuple_Check(args)) {
1335 PyErr_Format(
1336 PyExc_TypeError,
1337 "getsockaddrarg: "
1338 "AF_INET6 address must be tuple, not %.500s",
1339 Py_TYPE(args)->tp_name);
1340 return 0;
1341 }
1342 if (!PyArg_ParseTuple(args, "eti|ii",
1343 "idna", &host, &port, &flowinfo,
1344 &scope_id)) {
1345 return 0;
1346 }
1347 addr = (struct sockaddr_in6*)addr_ret;
1348 result = setipaddr(host, (struct sockaddr *)addr,
1349 sizeof(*addr), AF_INET6);
1350 PyMem_Free(host);
1351 if (result < 0)
1352 return 0;
1353 if (port < 0 || port > 0xffff) {
1354 PyErr_SetString(
1355 PyExc_OverflowError,
1356 "getsockaddrarg: port must be 0-65535.");
1357 return 0;
1358 }
1359 addr->sin6_family = s->sock_family;
1360 addr->sin6_port = htons((short)port);
1361 addr->sin6_flowinfo = flowinfo;
1362 addr->sin6_scope_id = scope_id;
1363 *len_ret = sizeof *addr;
1364 return 1;
1365 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00001366#endif
1367
Hye-Shik Chang81268602004-02-02 06:05:24 +00001368#ifdef USE_BLUETOOTH
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001369 case AF_BLUETOOTH:
1370 {
1371 switch (s->sock_proto) {
1372 case BTPROTO_L2CAP:
1373 {
1374 struct sockaddr_l2 *addr;
1375 char *straddr;
Martin v. Löwis12af0482004-01-31 12:34:17 +00001376
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001377 addr = (struct sockaddr_l2 *)addr_ret;
1378 memset(addr, 0, sizeof(struct sockaddr_l2));
1379 _BT_L2_MEMB(addr, family) = AF_BLUETOOTH;
1380 if (!PyArg_ParseTuple(args, "si", &straddr,
1381 &_BT_L2_MEMB(addr, psm))) {
1382 PyErr_SetString(socket_error, "getsockaddrarg: "
1383 "wrong format");
1384 return 0;
1385 }
1386 if (setbdaddr(straddr, &_BT_L2_MEMB(addr, bdaddr)) < 0)
1387 return 0;
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001388
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001389 *len_ret = sizeof *addr;
1390 return 1;
1391 }
1392 case BTPROTO_RFCOMM:
1393 {
1394 struct sockaddr_rc *addr;
1395 char *straddr;
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001396
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001397 addr = (struct sockaddr_rc *)addr_ret;
1398 _BT_RC_MEMB(addr, family) = AF_BLUETOOTH;
1399 if (!PyArg_ParseTuple(args, "si", &straddr,
1400 &_BT_RC_MEMB(addr, channel))) {
1401 PyErr_SetString(socket_error, "getsockaddrarg: "
1402 "wrong format");
1403 return 0;
1404 }
1405 if (setbdaddr(straddr, &_BT_RC_MEMB(addr, bdaddr)) < 0)
1406 return 0;
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001407
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001408 *len_ret = sizeof *addr;
1409 return 1;
1410 }
1411 case BTPROTO_HCI:
1412 {
1413 struct sockaddr_hci *addr = (struct sockaddr_hci *)addr_ret;
Gregory P. Smith3e85dfd2010-10-17 00:43:10 +00001414#if defined(__NetBSD__) || defined(__DragonFly__)
Alexander Belopolskye239d232010-12-08 23:31:48 +00001415 char *straddr = PyBytes_AS_STRING(args);
Gregory P. Smith3e85dfd2010-10-17 00:43:10 +00001416
Alexander Belopolskye239d232010-12-08 23:31:48 +00001417 _BT_HCI_MEMB(addr, family) = AF_BLUETOOTH;
Gregory P. Smith3e85dfd2010-10-17 00:43:10 +00001418 if (straddr == NULL) {
1419 PyErr_SetString(socket_error, "getsockaddrarg: "
1420 "wrong format");
1421 return 0;
1422 }
1423 if (setbdaddr(straddr, &_BT_HCI_MEMB(addr, bdaddr)) < 0)
1424 return 0;
1425#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001426 _BT_HCI_MEMB(addr, family) = AF_BLUETOOTH;
1427 if (!PyArg_ParseTuple(args, "i", &_BT_HCI_MEMB(addr, dev))) {
1428 PyErr_SetString(socket_error, "getsockaddrarg: "
1429 "wrong format");
1430 return 0;
1431 }
Gregory P. Smith3e85dfd2010-10-17 00:43:10 +00001432#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001433 *len_ret = sizeof *addr;
1434 return 1;
1435 }
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001436#if !defined(__FreeBSD__)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001437 case BTPROTO_SCO:
1438 {
1439 struct sockaddr_sco *addr;
1440 char *straddr;
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001441
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001442 addr = (struct sockaddr_sco *)addr_ret;
1443 _BT_SCO_MEMB(addr, family) = AF_BLUETOOTH;
1444 if (!PyBytes_Check(args)) {
1445 PyErr_SetString(socket_error, "getsockaddrarg: "
1446 "wrong format");
1447 return 0;
1448 }
1449 straddr = PyBytes_AS_STRING(args);
1450 if (setbdaddr(straddr, &_BT_SCO_MEMB(addr, bdaddr)) < 0)
1451 return 0;
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001452
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001453 *len_ret = sizeof *addr;
1454 return 1;
1455 }
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001456#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001457 default:
1458 PyErr_SetString(socket_error, "getsockaddrarg: unknown Bluetooth protocol");
1459 return 0;
1460 }
1461 }
Martin v. Löwis12af0482004-01-31 12:34:17 +00001462#endif
1463
Antoine Pitroub156a462010-10-27 20:13:57 +00001464#if defined(HAVE_NETPACKET_PACKET_H) && defined(SIOCGIFINDEX)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001465 case AF_PACKET:
1466 {
1467 struct sockaddr_ll* addr;
1468 struct ifreq ifr;
1469 char *interfaceName;
1470 int protoNumber;
1471 int hatype = 0;
1472 int pkttype = 0;
1473 char *haddr = NULL;
1474 unsigned int halen = 0;
Guido van Rossum48a680c2001-03-02 06:34:14 +00001475
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001476 if (!PyTuple_Check(args)) {
1477 PyErr_Format(
1478 PyExc_TypeError,
1479 "getsockaddrarg: "
1480 "AF_PACKET address must be tuple, not %.500s",
1481 Py_TYPE(args)->tp_name);
1482 return 0;
1483 }
1484 if (!PyArg_ParseTuple(args, "si|iiy#", &interfaceName,
1485 &protoNumber, &pkttype, &hatype,
1486 &haddr, &halen))
1487 return 0;
1488 strncpy(ifr.ifr_name, interfaceName, sizeof(ifr.ifr_name));
1489 ifr.ifr_name[(sizeof(ifr.ifr_name))-1] = '\0';
1490 if (ioctl(s->sock_fd, SIOCGIFINDEX, &ifr) < 0) {
1491 s->errorhandler();
1492 return 0;
1493 }
1494 if (halen > 8) {
1495 PyErr_SetString(PyExc_ValueError,
1496 "Hardware address must be 8 bytes or less");
1497 return 0;
1498 }
1499 if (protoNumber < 0 || protoNumber > 0xffff) {
1500 PyErr_SetString(
1501 PyExc_OverflowError,
1502 "getsockaddrarg: protoNumber must be 0-65535.");
1503 return 0;
1504 }
1505 addr = (struct sockaddr_ll*)addr_ret;
1506 addr->sll_family = AF_PACKET;
1507 addr->sll_protocol = htons((short)protoNumber);
1508 addr->sll_ifindex = ifr.ifr_ifindex;
1509 addr->sll_pkttype = pkttype;
1510 addr->sll_hatype = hatype;
1511 if (halen != 0) {
1512 memcpy(&addr->sll_addr, haddr, halen);
1513 }
1514 addr->sll_halen = halen;
1515 *len_ret = sizeof *addr;
1516 return 1;
1517 }
Guido van Rossum48a680c2001-03-02 06:34:14 +00001518#endif
1519
Christian Heimes043d6f62008-01-07 17:19:16 +00001520#ifdef HAVE_LINUX_TIPC_H
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001521 case AF_TIPC:
1522 {
1523 unsigned int atype, v1, v2, v3;
1524 unsigned int scope = TIPC_CLUSTER_SCOPE;
1525 struct sockaddr_tipc *addr;
Christian Heimes043d6f62008-01-07 17:19:16 +00001526
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001527 if (!PyTuple_Check(args)) {
1528 PyErr_Format(
1529 PyExc_TypeError,
1530 "getsockaddrarg: "
1531 "AF_TIPC address must be tuple, not %.500s",
1532 Py_TYPE(args)->tp_name);
1533 return 0;
1534 }
Christian Heimes043d6f62008-01-07 17:19:16 +00001535
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001536 if (!PyArg_ParseTuple(args,
1537 "IIII|I;Invalid TIPC address format",
1538 &atype, &v1, &v2, &v3, &scope))
1539 return 0;
Christian Heimes043d6f62008-01-07 17:19:16 +00001540
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001541 addr = (struct sockaddr_tipc *) addr_ret;
1542 memset(addr, 0, sizeof(struct sockaddr_tipc));
Christian Heimes043d6f62008-01-07 17:19:16 +00001543
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001544 addr->family = AF_TIPC;
1545 addr->scope = scope;
1546 addr->addrtype = atype;
Christian Heimes043d6f62008-01-07 17:19:16 +00001547
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001548 if (atype == TIPC_ADDR_NAMESEQ) {
1549 addr->addr.nameseq.type = v1;
1550 addr->addr.nameseq.lower = v2;
1551 addr->addr.nameseq.upper = v3;
1552 } else if (atype == TIPC_ADDR_NAME) {
1553 addr->addr.name.name.type = v1;
1554 addr->addr.name.name.instance = v2;
1555 } else if (atype == TIPC_ADDR_ID) {
1556 addr->addr.id.node = v1;
1557 addr->addr.id.ref = v2;
1558 } else {
1559 /* Shouldn't happen */
1560 PyErr_SetString(PyExc_TypeError, "Invalid address type");
1561 return 0;
1562 }
Christian Heimes043d6f62008-01-07 17:19:16 +00001563
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001564 *len_ret = sizeof(*addr);
Christian Heimes043d6f62008-01-07 17:19:16 +00001565
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001566 return 1;
1567 }
Christian Heimes043d6f62008-01-07 17:19:16 +00001568#endif
1569
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001570 /* More cases here... */
Guido van Rossum30a685f1991-06-27 15:51:29 +00001571
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001572 default:
1573 PyErr_SetString(socket_error, "getsockaddrarg: bad family");
1574 return 0;
Guido van Rossum30a685f1991-06-27 15:51:29 +00001575
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001576 }
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001577}
1578
Guido van Rossum30a685f1991-06-27 15:51:29 +00001579
Guido van Rossum48a680c2001-03-02 06:34:14 +00001580/* Get the address length according to the socket object's address family.
Guido van Rossum710e1df1992-06-12 10:39:36 +00001581 Return 1 if the family is known, 0 otherwise. The length is returned
1582 through len_ret. */
1583
1584static int
Peter Schneider-Kamp618e25d2000-07-11 23:00:12 +00001585getsockaddrlen(PySocketSockObject *s, socklen_t *len_ret)
Guido van Rossum710e1df1992-06-12 10:39:36 +00001586{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001587 switch (s->sock_family) {
Guido van Rossum710e1df1992-06-12 10:39:36 +00001588
Andrew MacIntyred12dfbb2004-04-04 07:13:49 +00001589#if defined(AF_UNIX)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001590 case AF_UNIX:
1591 {
1592 *len_ret = sizeof (struct sockaddr_un);
1593 return 1;
1594 }
Guido van Rossumb6775db1994-08-01 11:34:53 +00001595#endif /* AF_UNIX */
Martin v. Löwis11017b12006-01-14 18:12:57 +00001596#if defined(AF_NETLINK)
1597 case AF_NETLINK:
1598 {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001599 *len_ret = sizeof (struct sockaddr_nl);
1600 return 1;
Martin v. Löwis11017b12006-01-14 18:12:57 +00001601 }
1602#endif
Guido van Rossum710e1df1992-06-12 10:39:36 +00001603
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001604 case AF_INET:
1605 {
1606 *len_ret = sizeof (struct sockaddr_in);
1607 return 1;
1608 }
Guido van Rossum710e1df1992-06-12 10:39:36 +00001609
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00001610#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001611 case AF_INET6:
1612 {
1613 *len_ret = sizeof (struct sockaddr_in6);
1614 return 1;
1615 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00001616#endif
1617
Hye-Shik Chang81268602004-02-02 06:05:24 +00001618#ifdef USE_BLUETOOTH
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001619 case AF_BLUETOOTH:
1620 {
1621 switch(s->sock_proto)
1622 {
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001623
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001624 case BTPROTO_L2CAP:
1625 *len_ret = sizeof (struct sockaddr_l2);
1626 return 1;
1627 case BTPROTO_RFCOMM:
1628 *len_ret = sizeof (struct sockaddr_rc);
1629 return 1;
1630 case BTPROTO_HCI:
1631 *len_ret = sizeof (struct sockaddr_hci);
1632 return 1;
Hye-Shik Chang81268602004-02-02 06:05:24 +00001633#if !defined(__FreeBSD__)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001634 case BTPROTO_SCO:
1635 *len_ret = sizeof (struct sockaddr_sco);
1636 return 1;
Hye-Shik Chang81268602004-02-02 06:05:24 +00001637#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001638 default:
1639 PyErr_SetString(socket_error, "getsockaddrlen: "
1640 "unknown BT protocol");
1641 return 0;
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001642
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001643 }
1644 }
Martin v. Löwis12af0482004-01-31 12:34:17 +00001645#endif
1646
Martin v. Löwis1ba3fd52001-08-10 20:29:40 +00001647#ifdef HAVE_NETPACKET_PACKET_H
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001648 case AF_PACKET:
1649 {
1650 *len_ret = sizeof (struct sockaddr_ll);
1651 return 1;
1652 }
Jeremy Hylton22308652001-02-02 03:23:09 +00001653#endif
Guido van Rossum48a680c2001-03-02 06:34:14 +00001654
Christian Heimes043d6f62008-01-07 17:19:16 +00001655#ifdef HAVE_LINUX_TIPC_H
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001656 case AF_TIPC:
1657 {
1658 *len_ret = sizeof (struct sockaddr_tipc);
1659 return 1;
1660 }
Christian Heimes043d6f62008-01-07 17:19:16 +00001661#endif
1662
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001663 /* More cases here... */
Guido van Rossum710e1df1992-06-12 10:39:36 +00001664
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001665 default:
1666 PyErr_SetString(socket_error, "getsockaddrlen: bad family");
1667 return 0;
Guido van Rossum710e1df1992-06-12 10:39:36 +00001668
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001669 }
Guido van Rossum710e1df1992-06-12 10:39:36 +00001670}
1671
1672
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00001673/* s._accept() -> (fd, address) */
Guido van Rossum30a685f1991-06-27 15:51:29 +00001674
Guido van Rossum73624e91994-10-10 17:59:00 +00001675static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001676sock_accept(PySocketSockObject *s)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001677{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001678 sock_addr_t addrbuf;
1679 SOCKET_T newfd = INVALID_SOCKET;
1680 socklen_t addrlen;
1681 PyObject *sock = NULL;
1682 PyObject *addr = NULL;
1683 PyObject *res = NULL;
1684 int timeout;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001685 if (!getsockaddrlen(s, &addrlen))
1686 return NULL;
1687 memset(&addrbuf, 0, addrlen);
Guido van Rossum67f7a382002-06-06 21:08:16 +00001688
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001689 if (!IS_SELECTABLE(s))
1690 return select_error();
Neal Norwitz082b2df2006-02-07 07:04:46 +00001691
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00001692 BEGIN_SELECT_LOOP(s)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001693 Py_BEGIN_ALLOW_THREADS
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00001694 timeout = internal_select_ex(s, 0, interval);
Antoine Pitroub1c54962010-10-14 15:05:38 +00001695 if (!timeout) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001696 newfd = accept(s->sock_fd, SAS2SA(&addrbuf), &addrlen);
Antoine Pitroub1c54962010-10-14 15:05:38 +00001697 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001698 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00001699
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001700 if (timeout == 1) {
1701 PyErr_SetString(socket_timeout, "timed out");
1702 return NULL;
1703 }
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00001704 END_SELECT_LOOP(s)
Raymond Hettingeref7343c2003-06-29 03:08:05 +00001705
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001706 if (newfd == INVALID_SOCKET)
1707 return s->errorhandler();
Barry Warsaw752300b1997-01-03 17:18:10 +00001708
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001709 sock = PyLong_FromSocket_t(newfd);
1710 if (sock == NULL) {
1711 SOCKETCLOSE(newfd);
1712 goto finally;
1713 }
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00001714
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001715 addr = makesockaddr(s->sock_fd, SAS2SA(&addrbuf),
1716 addrlen, s->sock_proto);
1717 if (addr == NULL)
1718 goto finally;
Barry Warsaw752300b1997-01-03 17:18:10 +00001719
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001720 res = PyTuple_Pack(2, sock, addr);
Barry Warsaw752300b1997-01-03 17:18:10 +00001721
Guido van Rossum67f7a382002-06-06 21:08:16 +00001722finally:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001723 Py_XDECREF(sock);
1724 Py_XDECREF(addr);
1725 return res;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001726}
1727
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001728PyDoc_STRVAR(accept_doc,
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00001729"_accept() -> (integer, address info)\n\
Guido van Rossum82a5c661998-07-07 20:45:43 +00001730\n\
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00001731Wait for an incoming connection. Return a new socket file descriptor\n\
1732representing the connection, and the address of the client.\n\
1733For IP sockets, the address info is a pair (hostaddr, port).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001734
Guido van Rossum11ba0942002-06-13 15:07:44 +00001735/* s.setblocking(flag) method. Argument:
1736 False -- non-blocking mode; same as settimeout(0)
1737 True -- blocking mode; same as settimeout(None)
1738*/
Guido van Rossume4485b01994-09-07 14:32:49 +00001739
Guido van Rossum73624e91994-10-10 17:59:00 +00001740static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001741sock_setblocking(PySocketSockObject *s, PyObject *arg)
Guido van Rossume4485b01994-09-07 14:32:49 +00001742{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001743 int block;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001744
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001745 block = PyLong_AsLong(arg);
1746 if (block == -1 && PyErr_Occurred())
1747 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001748
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001749 s->sock_timeout = block ? -1.0 : 0.0;
1750 internal_setblocking(s, block);
Guido van Rossume4485b01994-09-07 14:32:49 +00001751
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001752 Py_INCREF(Py_None);
1753 return Py_None;
Guido van Rossume4485b01994-09-07 14:32:49 +00001754}
Guido van Rossume4485b01994-09-07 14:32:49 +00001755
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001756PyDoc_STRVAR(setblocking_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001757"setblocking(flag)\n\
1758\n\
1759Set the socket to blocking (flag is true) or non-blocking (false).\n\
Guido van Rossum11ba0942002-06-13 15:07:44 +00001760setblocking(True) is equivalent to settimeout(None);\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001761setblocking(False) is equivalent to settimeout(0.0).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001762
Guido van Rossum11ba0942002-06-13 15:07:44 +00001763/* s.settimeout(timeout) method. Argument:
1764 None -- no timeout, blocking mode; same as setblocking(True)
1765 0.0 -- non-blocking mode; same as setblocking(False)
1766 > 0 -- timeout mode; operations time out after timeout seconds
1767 < 0 -- illegal; raises an exception
1768*/
Guido van Rossum67f7a382002-06-06 21:08:16 +00001769static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001770sock_settimeout(PySocketSockObject *s, PyObject *arg)
Guido van Rossum67f7a382002-06-06 21:08:16 +00001771{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001772 double timeout;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001773
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001774 if (arg == Py_None)
1775 timeout = -1.0;
1776 else {
1777 timeout = PyFloat_AsDouble(arg);
1778 if (timeout < 0.0) {
1779 if (!PyErr_Occurred())
1780 PyErr_SetString(PyExc_ValueError,
1781 "Timeout value out of range");
1782 return NULL;
1783 }
1784 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00001785
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001786 s->sock_timeout = timeout;
1787 internal_setblocking(s, timeout < 0.0);
Guido van Rossum67f7a382002-06-06 21:08:16 +00001788
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001789 Py_INCREF(Py_None);
1790 return Py_None;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001791}
1792
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001793PyDoc_STRVAR(settimeout_doc,
Guido van Rossum3eede5a2002-06-07 02:08:35 +00001794"settimeout(timeout)\n\
Guido van Rossum67f7a382002-06-06 21:08:16 +00001795\n\
Guido van Rossum11ba0942002-06-13 15:07:44 +00001796Set a timeout on socket operations. 'timeout' can be a float,\n\
1797giving in seconds, or None. Setting a timeout of None disables\n\
1798the timeout feature and is equivalent to setblocking(1).\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001799Setting a timeout of zero is the same as setblocking(0).");
Guido van Rossum67f7a382002-06-06 21:08:16 +00001800
Guido van Rossum3eede5a2002-06-07 02:08:35 +00001801/* s.gettimeout() method.
1802 Returns the timeout associated with a socket. */
Guido van Rossum67f7a382002-06-06 21:08:16 +00001803static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001804sock_gettimeout(PySocketSockObject *s)
Guido van Rossum67f7a382002-06-06 21:08:16 +00001805{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001806 if (s->sock_timeout < 0.0) {
1807 Py_INCREF(Py_None);
1808 return Py_None;
1809 }
1810 else
1811 return PyFloat_FromDouble(s->sock_timeout);
Guido van Rossum67f7a382002-06-06 21:08:16 +00001812}
1813
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001814PyDoc_STRVAR(gettimeout_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00001815"gettimeout() -> timeout\n\
Guido van Rossum67f7a382002-06-06 21:08:16 +00001816\n\
1817Returns the timeout in floating seconds associated with socket \n\
1818operations. A timeout of None indicates that timeouts on socket \n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001819operations are disabled.");
Guido van Rossume4485b01994-09-07 14:32:49 +00001820
Guido van Rossumaee08791992-09-08 09:05:33 +00001821/* s.setsockopt() method.
1822 With an integer third argument, sets an integer option.
1823 With a string third argument, sets an option from a buffer;
1824 use optional built-in module 'struct' to encode the string. */
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001825
Guido van Rossum73624e91994-10-10 17:59:00 +00001826static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001827sock_setsockopt(PySocketSockObject *s, PyObject *args)
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001828{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001829 int level;
1830 int optname;
1831 int res;
1832 char *buf;
1833 int buflen;
1834 int flag;
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001835
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001836 if (PyArg_ParseTuple(args, "iii:setsockopt",
1837 &level, &optname, &flag)) {
1838 buf = (char *) &flag;
1839 buflen = sizeof flag;
1840 }
1841 else {
1842 PyErr_Clear();
1843 if (!PyArg_ParseTuple(args, "iiy#:setsockopt",
1844 &level, &optname, &buf, &buflen))
1845 return NULL;
1846 }
1847 res = setsockopt(s->sock_fd, level, optname, (void *)buf, buflen);
1848 if (res < 0)
1849 return s->errorhandler();
1850 Py_INCREF(Py_None);
1851 return Py_None;
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001852}
1853
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001854PyDoc_STRVAR(setsockopt_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001855"setsockopt(level, option, value)\n\
1856\n\
1857Set a socket option. See the Unix manual for level and option.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001858The value argument can either be an integer or a string.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001859
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001860
Guido van Rossumaee08791992-09-08 09:05:33 +00001861/* s.getsockopt() method.
1862 With two arguments, retrieves an integer option.
1863 With a third integer argument, retrieves a string buffer of that size;
1864 use optional built-in module 'struct' to decode the string. */
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001865
Guido van Rossum73624e91994-10-10 17:59:00 +00001866static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001867sock_getsockopt(PySocketSockObject *s, PyObject *args)
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001868{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001869 int level;
1870 int optname;
1871 int res;
1872 PyObject *buf;
1873 socklen_t buflen = 0;
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001874
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001875 if (!PyArg_ParseTuple(args, "ii|i:getsockopt",
1876 &level, &optname, &buflen))
1877 return NULL;
Guido van Rossum48a680c2001-03-02 06:34:14 +00001878
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001879 if (buflen == 0) {
1880 int flag = 0;
1881 socklen_t flagsize = sizeof flag;
1882 res = getsockopt(s->sock_fd, level, optname,
1883 (void *)&flag, &flagsize);
1884 if (res < 0)
1885 return s->errorhandler();
1886 return PyLong_FromLong(flag);
1887 }
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001888#ifdef __VMS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001889 /* socklen_t is unsigned so no negative test is needed,
1890 test buflen == 0 is previously done */
1891 if (buflen > 1024) {
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001892#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001893 if (buflen <= 0 || buflen > 1024) {
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001894#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001895 PyErr_SetString(socket_error,
1896 "getsockopt buflen out of range");
1897 return NULL;
1898 }
1899 buf = PyBytes_FromStringAndSize((char *)NULL, buflen);
1900 if (buf == NULL)
1901 return NULL;
1902 res = getsockopt(s->sock_fd, level, optname,
1903 (void *)PyBytes_AS_STRING(buf), &buflen);
1904 if (res < 0) {
1905 Py_DECREF(buf);
1906 return s->errorhandler();
1907 }
1908 _PyBytes_Resize(&buf, buflen);
1909 return buf;
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001910}
1911
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001912PyDoc_STRVAR(getsockopt_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001913"getsockopt(level, option[, buffersize]) -> value\n\
1914\n\
1915Get a socket option. See the Unix manual for level and option.\n\
1916If a nonzero buffersize argument is given, the return value is a\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001917string of that length; otherwise it is an integer.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001918
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001919
Fred Drake728819a2000-07-01 03:40:12 +00001920/* s.bind(sockaddr) method */
Guido van Rossum30a685f1991-06-27 15:51:29 +00001921
Guido van Rossum73624e91994-10-10 17:59:00 +00001922static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001923sock_bind(PySocketSockObject *s, PyObject *addro)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001924{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001925 sock_addr_t addrbuf;
1926 int addrlen;
1927 int res;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00001928
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001929 if (!getsockaddrarg(s, addro, SAS2SA(&addrbuf), &addrlen))
1930 return NULL;
1931 Py_BEGIN_ALLOW_THREADS
1932 res = bind(s->sock_fd, SAS2SA(&addrbuf), addrlen);
1933 Py_END_ALLOW_THREADS
1934 if (res < 0)
1935 return s->errorhandler();
1936 Py_INCREF(Py_None);
1937 return Py_None;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001938}
1939
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001940PyDoc_STRVAR(bind_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001941"bind(address)\n\
1942\n\
1943Bind the socket to a local address. For IP sockets, the address is a\n\
Jeremy Hylton22308652001-02-02 03:23:09 +00001944pair (host, port); the host must refer to the local host. For raw packet\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001945sockets the address is a tuple (ifname, proto [,pkttype [,hatype]])");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001946
Guido van Rossum30a685f1991-06-27 15:51:29 +00001947
1948/* s.close() method.
1949 Set the file descriptor to -1 so operations tried subsequently
1950 will surely fail. */
1951
Guido van Rossum73624e91994-10-10 17:59:00 +00001952static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001953sock_close(PySocketSockObject *s)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001954{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001955 SOCKET_T fd;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00001956
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001957 if ((fd = s->sock_fd) != -1) {
1958 s->sock_fd = -1;
1959 Py_BEGIN_ALLOW_THREADS
1960 (void) SOCKETCLOSE(fd);
1961 Py_END_ALLOW_THREADS
1962 }
1963 Py_INCREF(Py_None);
1964 return Py_None;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001965}
1966
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001967PyDoc_STRVAR(close_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001968"close()\n\
1969\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001970Close the socket. It cannot be used after this call.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001971
Antoine Pitroue43f9d02010-08-08 23:24:50 +00001972static PyObject *
Antoine Pitrou6e451df2010-08-09 20:39:54 +00001973sock_detach(PySocketSockObject *s)
Antoine Pitroue43f9d02010-08-08 23:24:50 +00001974{
Antoine Pitrou6e451df2010-08-09 20:39:54 +00001975 SOCKET_T fd = s->sock_fd;
Antoine Pitroue43f9d02010-08-08 23:24:50 +00001976 s->sock_fd = -1;
Antoine Pitrou6e451df2010-08-09 20:39:54 +00001977 return PyLong_FromSocket_t(fd);
Antoine Pitroue43f9d02010-08-08 23:24:50 +00001978}
1979
Antoine Pitrou6e451df2010-08-09 20:39:54 +00001980PyDoc_STRVAR(detach_doc,
1981"detach()\n\
Antoine Pitroue43f9d02010-08-08 23:24:50 +00001982\n\
1983Close the socket object without closing the underlying file descriptor.\
1984The object cannot be used after this call, but the file descriptor\
Antoine Pitrou6e451df2010-08-09 20:39:54 +00001985can be reused for other purposes. The file descriptor is returned.");
Antoine Pitroue43f9d02010-08-08 23:24:50 +00001986
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001987static int
Raymond Hettingeref7343c2003-06-29 03:08:05 +00001988internal_connect(PySocketSockObject *s, struct sockaddr *addr, int addrlen,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001989 int *timeoutp)
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001990{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001991 int res, timeout;
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001992
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001993 timeout = 0;
1994 res = connect(s->sock_fd, addr, addrlen);
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001995
1996#ifdef MS_WINDOWS
1997
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001998 if (s->sock_timeout > 0.0) {
1999 if (res < 0 && WSAGetLastError() == WSAEWOULDBLOCK &&
2000 IS_SELECTABLE(s)) {
2001 /* This is a mess. Best solution: trust select */
2002 fd_set fds;
2003 fd_set fds_exc;
2004 struct timeval tv;
2005 tv.tv_sec = (int)s->sock_timeout;
2006 tv.tv_usec = (int)((s->sock_timeout - tv.tv_sec) * 1e6);
2007 FD_ZERO(&fds);
2008 FD_SET(s->sock_fd, &fds);
2009 FD_ZERO(&fds_exc);
2010 FD_SET(s->sock_fd, &fds_exc);
Antoine Pitrou19467d22010-08-17 19:33:30 +00002011 res = select(Py_SAFE_DOWNCAST(s->sock_fd+1, SOCKET_T, int),
2012 NULL, &fds, &fds_exc, &tv);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002013 if (res == 0) {
2014 res = WSAEWOULDBLOCK;
2015 timeout = 1;
2016 } else if (res > 0) {
2017 if (FD_ISSET(s->sock_fd, &fds))
2018 /* The socket is in the writable set - this
2019 means connected */
2020 res = 0;
2021 else {
2022 /* As per MS docs, we need to call getsockopt()
2023 to get the underlying error */
2024 int res_size = sizeof res;
2025 /* It must be in the exception set */
2026 assert(FD_ISSET(s->sock_fd, &fds_exc));
2027 if (0 == getsockopt(s->sock_fd, SOL_SOCKET, SO_ERROR,
2028 (char *)&res, &res_size))
2029 /* getsockopt also clears WSAGetLastError,
2030 so reset it back. */
2031 WSASetLastError(res);
2032 else
2033 res = WSAGetLastError();
2034 }
2035 }
2036 /* else if (res < 0) an error occurred */
2037 }
2038 }
Guido van Rossum7b8bac12002-06-13 16:07:04 +00002039
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002040 if (res < 0)
2041 res = WSAGetLastError();
Guido van Rossum7b8bac12002-06-13 16:07:04 +00002042
2043#else
2044
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002045 if (s->sock_timeout > 0.0) {
2046 if (res < 0 && errno == EINPROGRESS && IS_SELECTABLE(s)) {
2047 timeout = internal_select(s, 1);
2048 if (timeout == 0) {
2049 /* Bug #1019808: in case of an EINPROGRESS,
2050 use getsockopt(SO_ERROR) to get the real
2051 error. */
2052 socklen_t res_size = sizeof res;
2053 (void)getsockopt(s->sock_fd, SOL_SOCKET,
2054 SO_ERROR, &res, &res_size);
2055 if (res == EISCONN)
2056 res = 0;
2057 errno = res;
2058 }
2059 else if (timeout == -1) {
2060 res = errno; /* had error */
2061 }
2062 else
2063 res = EWOULDBLOCK; /* timed out */
2064 }
2065 }
Guido van Rossum7b8bac12002-06-13 16:07:04 +00002066
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002067 if (res < 0)
2068 res = errno;
Guido van Rossum7b8bac12002-06-13 16:07:04 +00002069
2070#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002071 *timeoutp = timeout;
Guido van Rossum7b8bac12002-06-13 16:07:04 +00002072
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002073 return res;
Guido van Rossum7b8bac12002-06-13 16:07:04 +00002074}
Guido van Rossum30a685f1991-06-27 15:51:29 +00002075
Fred Drake728819a2000-07-01 03:40:12 +00002076/* s.connect(sockaddr) method */
Guido van Rossum30a685f1991-06-27 15:51:29 +00002077
Guido van Rossum73624e91994-10-10 17:59:00 +00002078static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002079sock_connect(PySocketSockObject *s, PyObject *addro)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002080{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002081 sock_addr_t addrbuf;
2082 int addrlen;
2083 int res;
2084 int timeout;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00002085
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002086 if (!getsockaddrarg(s, addro, SAS2SA(&addrbuf), &addrlen))
2087 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002088
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002089 Py_BEGIN_ALLOW_THREADS
2090 res = internal_connect(s, SAS2SA(&addrbuf), addrlen, &timeout);
2091 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00002092
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002093 if (timeout == 1) {
2094 PyErr_SetString(socket_timeout, "timed out");
2095 return NULL;
2096 }
2097 if (res != 0)
2098 return s->errorhandler();
2099 Py_INCREF(Py_None);
2100 return Py_None;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002101}
2102
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002103PyDoc_STRVAR(connect_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002104"connect(address)\n\
2105\n\
2106Connect the socket to a remote address. For IP sockets, the address\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002107is a pair (host, port).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002108
Guido van Rossum30a685f1991-06-27 15:51:29 +00002109
Fred Drake728819a2000-07-01 03:40:12 +00002110/* s.connect_ex(sockaddr) method */
Guido van Rossumfc4255d1997-11-19 18:57:13 +00002111
2112static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002113sock_connect_ex(PySocketSockObject *s, PyObject *addro)
Guido van Rossumfc4255d1997-11-19 18:57:13 +00002114{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002115 sock_addr_t addrbuf;
2116 int addrlen;
2117 int res;
2118 int timeout;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00002119
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002120 if (!getsockaddrarg(s, addro, SAS2SA(&addrbuf), &addrlen))
2121 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002122
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002123 Py_BEGIN_ALLOW_THREADS
2124 res = internal_connect(s, SAS2SA(&addrbuf), addrlen, &timeout);
2125 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00002126
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002127 /* Signals are not errors (though they may raise exceptions). Adapted
2128 from PyErr_SetFromErrnoWithFilenameObject(). */
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002129#ifdef EINTR
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002130 if (res == EINTR && PyErr_CheckSignals())
2131 return NULL;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002132#endif
2133
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002134 return PyLong_FromLong((long) res);
Guido van Rossumfc4255d1997-11-19 18:57:13 +00002135}
2136
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002137PyDoc_STRVAR(connect_ex_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00002138"connect_ex(address) -> errno\n\
Guido van Rossum82a5c661998-07-07 20:45:43 +00002139\n\
2140This is like connect(address), but returns an error code (the errno value)\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002141instead of raising an exception when an error occurs.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002142
Guido van Rossumfc4255d1997-11-19 18:57:13 +00002143
Guido van Rossumed233a51992-06-23 09:07:03 +00002144/* s.fileno() method */
2145
Guido van Rossum73624e91994-10-10 17:59:00 +00002146static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002147sock_fileno(PySocketSockObject *s)
Guido van Rossumed233a51992-06-23 09:07:03 +00002148{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002149 return PyLong_FromSocket_t(s->sock_fd);
Guido van Rossumed233a51992-06-23 09:07:03 +00002150}
2151
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002152PyDoc_STRVAR(fileno_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002153"fileno() -> integer\n\
2154\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002155Return the integer file descriptor of the socket.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002156
Guido van Rossumed233a51992-06-23 09:07:03 +00002157
Guido van Rossumc89705d1992-11-26 08:54:07 +00002158/* s.getsockname() method */
2159
Guido van Rossum73624e91994-10-10 17:59:00 +00002160static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002161sock_getsockname(PySocketSockObject *s)
Guido van Rossumc89705d1992-11-26 08:54:07 +00002162{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002163 sock_addr_t addrbuf;
2164 int res;
2165 socklen_t addrlen;
Guido van Rossumff3ab422000-04-24 15:16:03 +00002166
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002167 if (!getsockaddrlen(s, &addrlen))
2168 return NULL;
2169 memset(&addrbuf, 0, addrlen);
2170 Py_BEGIN_ALLOW_THREADS
2171 res = getsockname(s->sock_fd, SAS2SA(&addrbuf), &addrlen);
2172 Py_END_ALLOW_THREADS
2173 if (res < 0)
2174 return s->errorhandler();
2175 return makesockaddr(s->sock_fd, SAS2SA(&addrbuf), addrlen,
2176 s->sock_proto);
Guido van Rossumc89705d1992-11-26 08:54:07 +00002177}
2178
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002179PyDoc_STRVAR(getsockname_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002180"getsockname() -> address info\n\
2181\n\
2182Return the address of the local endpoint. For IP sockets, the address\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002183info is a pair (hostaddr, port).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002184
Guido van Rossumc89705d1992-11-26 08:54:07 +00002185
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002186#ifdef HAVE_GETPEERNAME /* Cray APP doesn't have this :-( */
Guido van Rossumc89705d1992-11-26 08:54:07 +00002187/* s.getpeername() method */
2188
Guido van Rossum73624e91994-10-10 17:59:00 +00002189static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002190sock_getpeername(PySocketSockObject *s)
Guido van Rossumc89705d1992-11-26 08:54:07 +00002191{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002192 sock_addr_t addrbuf;
2193 int res;
2194 socklen_t addrlen;
Guido van Rossumff3ab422000-04-24 15:16:03 +00002195
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002196 if (!getsockaddrlen(s, &addrlen))
2197 return NULL;
2198 memset(&addrbuf, 0, addrlen);
2199 Py_BEGIN_ALLOW_THREADS
2200 res = getpeername(s->sock_fd, SAS2SA(&addrbuf), &addrlen);
2201 Py_END_ALLOW_THREADS
2202 if (res < 0)
2203 return s->errorhandler();
2204 return makesockaddr(s->sock_fd, SAS2SA(&addrbuf), addrlen,
2205 s->sock_proto);
Guido van Rossumc89705d1992-11-26 08:54:07 +00002206}
Guido van Rossum82a5c661998-07-07 20:45:43 +00002207
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002208PyDoc_STRVAR(getpeername_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002209"getpeername() -> address info\n\
2210\n\
2211Return the address of the remote endpoint. For IP sockets, the address\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002212info is a pair (hostaddr, port).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002213
Guido van Rossumb6775db1994-08-01 11:34:53 +00002214#endif /* HAVE_GETPEERNAME */
Guido van Rossumc89705d1992-11-26 08:54:07 +00002215
2216
Guido van Rossum30a685f1991-06-27 15:51:29 +00002217/* s.listen(n) method */
2218
Guido van Rossum73624e91994-10-10 17:59:00 +00002219static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002220sock_listen(PySocketSockObject *s, PyObject *arg)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002221{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002222 int backlog;
2223 int res;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00002224
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002225 backlog = PyLong_AsLong(arg);
2226 if (backlog == -1 && PyErr_Occurred())
2227 return NULL;
2228 Py_BEGIN_ALLOW_THREADS
Antoine Pitrou1be815a2011-05-10 19:16:29 +02002229 /* To avoid problems on systems that don't allow a negative backlog
2230 * (which doesn't make sense anyway) we force a minimum value of 0. */
2231 if (backlog < 0)
2232 backlog = 0;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002233 res = listen(s->sock_fd, backlog);
2234 Py_END_ALLOW_THREADS
2235 if (res < 0)
2236 return s->errorhandler();
2237 Py_INCREF(Py_None);
2238 return Py_None;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002239}
2240
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002241PyDoc_STRVAR(listen_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002242"listen(backlog)\n\
2243\n\
2244Enable a server to accept connections. The backlog argument must be at\n\
Antoine Pitrou1be815a2011-05-10 19:16:29 +02002245least 0 (if it is lower, it is set to 0); it specifies the number of\n\
2246unaccepted connections that the system will allow before refusing new\n\
2247connections.");
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
Ezio Melotti13925002011-03-16 11:05:33 +02002254 * successfully read. If there was an error, it returns -1. Note that it is
Thomas Wouters477c8d52006-05-27 19:21:47 +00002255 * 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
Ezio Melotti13925002011-03-16 11:05:33 +02002458 * of bytes successfully read. If there was an error, it returns -1. Note
Christian Heimes99170a52007-12-19 02:07:34 +00002459 * 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
Ezio Melotti13925002011-03-16 11:05:33 +02002553 string if possible and be successful. */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002554 if (_PyBytes_Resize(&buf, outlen) < 0)
Ezio Melotti13925002011-03-16 11:05:33 +02002555 /* Oopsy, not so successful after all. */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002556 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;
Antoine Pitrou5e981412011-03-17 22:38:37 +01002759 Py_ssize_t len, arglen;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002760 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;
Antoine Pitrou5e981412011-03-17 22:38:37 +01002764 arglen = PyTuple_Size(args);
2765 switch (arglen) {
2766 case 2:
2767 PyArg_ParseTuple(args, "y*O:sendto", &pbuf, &addro);
2768 break;
2769 case 3:
2770 PyArg_ParseTuple(args, "y*iO:sendto",
2771 &pbuf, &flags, &addro);
2772 break;
2773 default:
2774 PyErr_Format(PyExc_TypeError,
2775 "sendto() takes 2 or 3 arguments (%d given)",
2776 arglen);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002777 }
Antoine Pitrou5e981412011-03-17 22:38:37 +01002778 if (PyErr_Occurred())
2779 return NULL;
2780
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002781 buf = pbuf.buf;
2782 len = pbuf.len;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002783
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002784 if (!IS_SELECTABLE(s)) {
2785 PyBuffer_Release(&pbuf);
2786 return select_error();
2787 }
Neal Norwitz082b2df2006-02-07 07:04:46 +00002788
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002789 if (!getsockaddrarg(s, addro, SAS2SA(&addrbuf), &addrlen)) {
2790 PyBuffer_Release(&pbuf);
2791 return NULL;
2792 }
Thomas Wouters89f507f2006-12-13 04:49:30 +00002793
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002794 BEGIN_SELECT_LOOP(s)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002795 Py_BEGIN_ALLOW_THREADS
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002796 timeout = internal_select_ex(s, 1, interval);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002797 if (!timeout)
2798 n = sendto(s->sock_fd, buf, len, flags, SAS2SA(&addrbuf), addrlen);
2799 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00002800
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002801 if (timeout == 1) {
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002802 PyBuffer_Release(&pbuf);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002803 PyErr_SetString(socket_timeout, "timed out");
2804 return NULL;
2805 }
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002806 END_SELECT_LOOP(s)
2807 PyBuffer_Release(&pbuf);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002808 if (n < 0)
2809 return s->errorhandler();
Antoine Pitrou19467d22010-08-17 19:33:30 +00002810 return PyLong_FromSsize_t(n);
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002811}
2812
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002813PyDoc_STRVAR(sendto_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00002814"sendto(data[, flags], address) -> count\n\
Guido van Rossum82a5c661998-07-07 20:45:43 +00002815\n\
2816Like send(data, flags) but allows specifying the destination address.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002817For IP sockets, the address is a pair (hostaddr, port).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002818
Guido van Rossum30a685f1991-06-27 15:51:29 +00002819
2820/* s.shutdown(how) method */
2821
Guido van Rossum73624e91994-10-10 17:59:00 +00002822static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002823sock_shutdown(PySocketSockObject *s, PyObject *arg)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002824{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002825 int how;
2826 int res;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00002827
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002828 how = PyLong_AsLong(arg);
2829 if (how == -1 && PyErr_Occurred())
2830 return NULL;
2831 Py_BEGIN_ALLOW_THREADS
2832 res = shutdown(s->sock_fd, how);
2833 Py_END_ALLOW_THREADS
2834 if (res < 0)
2835 return s->errorhandler();
2836 Py_INCREF(Py_None);
2837 return Py_None;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002838}
2839
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002840PyDoc_STRVAR(shutdown_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002841"shutdown(flag)\n\
2842\n\
Martin v. Löwis94681fc2003-11-27 19:40:22 +00002843Shut down the reading side of the socket (flag == SHUT_RD), the writing side\n\
2844of the socket (flag == SHUT_WR), or both ends (flag == SHUT_RDWR).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002845
Amaury Forgeot d'Arc3d17a5c2008-06-13 01:09:34 +00002846#if defined(MS_WINDOWS) && defined(SIO_RCVALL)
Christian Heimesfaf2f632008-01-06 16:59:19 +00002847static PyObject*
2848sock_ioctl(PySocketSockObject *s, PyObject *arg)
2849{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002850 unsigned long cmd = SIO_RCVALL;
2851 PyObject *argO;
2852 DWORD recv;
Christian Heimesfaf2f632008-01-06 16:59:19 +00002853
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002854 if (!PyArg_ParseTuple(arg, "kO:ioctl", &cmd, &argO))
2855 return NULL;
Christian Heimesfaf2f632008-01-06 16:59:19 +00002856
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002857 switch (cmd) {
2858 case SIO_RCVALL: {
2859 unsigned int option = RCVALL_ON;
2860 if (!PyArg_ParseTuple(arg, "kI:ioctl", &cmd, &option))
2861 return NULL;
2862 if (WSAIoctl(s->sock_fd, cmd, &option, sizeof(option),
2863 NULL, 0, &recv, NULL, NULL) == SOCKET_ERROR) {
2864 return set_error();
2865 }
2866 return PyLong_FromUnsignedLong(recv); }
2867 case SIO_KEEPALIVE_VALS: {
2868 struct tcp_keepalive ka;
2869 if (!PyArg_ParseTuple(arg, "k(kkk):ioctl", &cmd,
2870 &ka.onoff, &ka.keepalivetime, &ka.keepaliveinterval))
2871 return NULL;
2872 if (WSAIoctl(s->sock_fd, cmd, &ka, sizeof(ka),
2873 NULL, 0, &recv, NULL, NULL) == SOCKET_ERROR) {
2874 return set_error();
2875 }
2876 return PyLong_FromUnsignedLong(recv); }
2877 default:
2878 PyErr_Format(PyExc_ValueError, "invalid ioctl command %d", cmd);
2879 return NULL;
2880 }
Christian Heimesfaf2f632008-01-06 16:59:19 +00002881}
2882PyDoc_STRVAR(sock_ioctl_doc,
2883"ioctl(cmd, option) -> long\n\
2884\n\
Kristján Valur Jónsson847ec752009-09-27 21:10:38 +00002885Control the socket with WSAIoctl syscall. Currently supported 'cmd' values are\n\
2886SIO_RCVALL: 'option' must be one of the socket.RCVALL_* constants.\n\
2887SIO_KEEPALIVE_VALS: 'option' is a tuple of (onoff, timeout, interval).");
Christian Heimesfaf2f632008-01-06 16:59:19 +00002888
2889#endif
Guido van Rossum30a685f1991-06-27 15:51:29 +00002890
2891/* List of methods for socket objects */
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002892
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002893static PyMethodDef sock_methods[] = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002894 {"_accept", (PyCFunction)sock_accept, METH_NOARGS,
2895 accept_doc},
2896 {"bind", (PyCFunction)sock_bind, METH_O,
2897 bind_doc},
2898 {"close", (PyCFunction)sock_close, METH_NOARGS,
2899 close_doc},
2900 {"connect", (PyCFunction)sock_connect, METH_O,
2901 connect_doc},
2902 {"connect_ex", (PyCFunction)sock_connect_ex, METH_O,
2903 connect_ex_doc},
Antoine Pitrou6e451df2010-08-09 20:39:54 +00002904 {"detach", (PyCFunction)sock_detach, METH_NOARGS,
2905 detach_doc},
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002906 {"fileno", (PyCFunction)sock_fileno, METH_NOARGS,
2907 fileno_doc},
Guido van Rossumb6775db1994-08-01 11:34:53 +00002908#ifdef HAVE_GETPEERNAME
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002909 {"getpeername", (PyCFunction)sock_getpeername,
2910 METH_NOARGS, getpeername_doc},
Guido van Rossum9575a441993-04-07 14:06:14 +00002911#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002912 {"getsockname", (PyCFunction)sock_getsockname,
2913 METH_NOARGS, getsockname_doc},
2914 {"getsockopt", (PyCFunction)sock_getsockopt, METH_VARARGS,
2915 getsockopt_doc},
Amaury Forgeot d'Arc3d17a5c2008-06-13 01:09:34 +00002916#if defined(MS_WINDOWS) && defined(SIO_RCVALL)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002917 {"ioctl", (PyCFunction)sock_ioctl, METH_VARARGS,
2918 sock_ioctl_doc},
Christian Heimesfaf2f632008-01-06 16:59:19 +00002919#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002920 {"listen", (PyCFunction)sock_listen, METH_O,
2921 listen_doc},
2922 {"recv", (PyCFunction)sock_recv, METH_VARARGS,
2923 recv_doc},
2924 {"recv_into", (PyCFunction)sock_recv_into, METH_VARARGS | METH_KEYWORDS,
2925 recv_into_doc},
2926 {"recvfrom", (PyCFunction)sock_recvfrom, METH_VARARGS,
2927 recvfrom_doc},
2928 {"recvfrom_into", (PyCFunction)sock_recvfrom_into, METH_VARARGS | METH_KEYWORDS,
2929 recvfrom_into_doc},
2930 {"send", (PyCFunction)sock_send, METH_VARARGS,
2931 send_doc},
2932 {"sendall", (PyCFunction)sock_sendall, METH_VARARGS,
2933 sendall_doc},
2934 {"sendto", (PyCFunction)sock_sendto, METH_VARARGS,
2935 sendto_doc},
2936 {"setblocking", (PyCFunction)sock_setblocking, METH_O,
2937 setblocking_doc},
2938 {"settimeout", (PyCFunction)sock_settimeout, METH_O,
2939 settimeout_doc},
2940 {"gettimeout", (PyCFunction)sock_gettimeout, METH_NOARGS,
2941 gettimeout_doc},
2942 {"setsockopt", (PyCFunction)sock_setsockopt, METH_VARARGS,
2943 setsockopt_doc},
2944 {"shutdown", (PyCFunction)sock_shutdown, METH_O,
2945 shutdown_doc},
2946 {NULL, NULL} /* sentinel */
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002947};
2948
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002949/* SockObject members */
2950static PyMemberDef sock_memberlist[] = {
2951 {"family", T_INT, offsetof(PySocketSockObject, sock_family), READONLY, "the socket family"},
2952 {"type", T_INT, offsetof(PySocketSockObject, sock_type), READONLY, "the socket type"},
2953 {"proto", T_INT, offsetof(PySocketSockObject, sock_proto), READONLY, "the socket protocol"},
2954 {"timeout", T_DOUBLE, offsetof(PySocketSockObject, sock_timeout), READONLY, "the socket timeout"},
2955 {0},
2956};
Guido van Rossum30a685f1991-06-27 15:51:29 +00002957
Guido van Rossum73624e91994-10-10 17:59:00 +00002958/* Deallocate a socket object in response to the last Py_DECREF().
Guido van Rossum30a685f1991-06-27 15:51:29 +00002959 First close the file description. */
2960
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002961static void
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002962sock_dealloc(PySocketSockObject *s)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002963{
Antoine Pitroue033e062010-10-29 10:38:18 +00002964 if (s->sock_fd != -1) {
2965 PyObject *exc, *val, *tb;
2966 Py_ssize_t old_refcount = Py_REFCNT(s);
2967 ++Py_REFCNT(s);
2968 PyErr_Fetch(&exc, &val, &tb);
2969 if (PyErr_WarnFormat(PyExc_ResourceWarning, 1,
2970 "unclosed %R", s))
2971 /* Spurious errors can appear at shutdown */
2972 if (PyErr_ExceptionMatches(PyExc_Warning))
2973 PyErr_WriteUnraisable((PyObject *) s);
2974 PyErr_Restore(exc, val, tb);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002975 (void) SOCKETCLOSE(s->sock_fd);
Antoine Pitroue033e062010-10-29 10:38:18 +00002976 Py_REFCNT(s) = old_refcount;
2977 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002978 Py_TYPE(s)->tp_free((PyObject *)s);
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002979}
2980
Guido van Rossum30a685f1991-06-27 15:51:29 +00002981
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00002982static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002983sock_repr(PySocketSockObject *s)
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00002984{
Fred Drakea04eaad2000-06-30 02:46:07 +00002985#if SIZEOF_SOCKET_T > SIZEOF_LONG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002986 if (s->sock_fd > LONG_MAX) {
2987 /* this can occur on Win64, and actually there is a special
2988 ugly printf formatter for decimal pointer length integer
2989 printing, only bother if necessary*/
2990 PyErr_SetString(PyExc_OverflowError,
2991 "no printf formatter to display "
2992 "the socket descriptor in decimal");
2993 return NULL;
2994 }
Fred Drakea04eaad2000-06-30 02:46:07 +00002995#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002996 return PyUnicode_FromFormat(
2997 "<socket object, fd=%ld, family=%d, type=%d, proto=%d>",
2998 (long)s->sock_fd, s->sock_family,
2999 s->sock_type,
3000 s->sock_proto);
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00003001}
3002
3003
Guido van Rossum384ca9c2001-10-27 22:20:47 +00003004/* Create a new, uninitialized socket object. */
3005
3006static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003007sock_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
Guido van Rossum384ca9c2001-10-27 22:20:47 +00003008{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003009 PyObject *new;
Guido van Rossum384ca9c2001-10-27 22:20:47 +00003010
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003011 new = type->tp_alloc(type, 0);
3012 if (new != NULL) {
3013 ((PySocketSockObject *)new)->sock_fd = -1;
3014 ((PySocketSockObject *)new)->sock_timeout = -1.0;
3015 ((PySocketSockObject *)new)->errorhandler = &set_error;
3016 }
3017 return new;
Guido van Rossum384ca9c2001-10-27 22:20:47 +00003018}
3019
3020
3021/* Initialize a new socket object. */
3022
3023/*ARGSUSED*/
3024static int
Andrew MacIntyre7aec4a22002-06-13 11:53:52 +00003025sock_initobj(PyObject *self, PyObject *args, PyObject *kwds)
Guido van Rossum384ca9c2001-10-27 22:20:47 +00003026{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003027 PySocketSockObject *s = (PySocketSockObject *)self;
3028 PyObject *fdobj = NULL;
3029 SOCKET_T fd = INVALID_SOCKET;
3030 int family = AF_INET, type = SOCK_STREAM, proto = 0;
3031 static char *keywords[] = {"family", "type", "proto", "fileno", 0};
Guido van Rossum384ca9c2001-10-27 22:20:47 +00003032
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003033 if (!PyArg_ParseTupleAndKeywords(args, kwds,
3034 "|iiiO:socket", keywords,
3035 &family, &type, &proto, &fdobj))
3036 return -1;
Guido van Rossum67f7a382002-06-06 21:08:16 +00003037
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003038 if (fdobj != NULL && fdobj != Py_None) {
3039 fd = PyLong_AsSocket_t(fdobj);
3040 if (fd == (SOCKET_T)(-1) && PyErr_Occurred())
3041 return -1;
3042 if (fd == INVALID_SOCKET) {
3043 PyErr_SetString(PyExc_ValueError,
3044 "can't use invalid socket value");
3045 return -1;
3046 }
3047 }
3048 else {
3049 Py_BEGIN_ALLOW_THREADS
3050 fd = socket(family, type, proto);
3051 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00003052
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003053 if (fd == INVALID_SOCKET) {
3054 set_error();
3055 return -1;
3056 }
3057 }
3058 init_sockobject(s, fd, family, type, proto);
Guido van Rossum67f7a382002-06-06 21:08:16 +00003059
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003060 return 0;
Guido van Rossum67f7a382002-06-06 21:08:16 +00003061
Guido van Rossum384ca9c2001-10-27 22:20:47 +00003062}
3063
3064
Guido van Rossumb6775db1994-08-01 11:34:53 +00003065/* Type object for socket objects. */
Guido van Rossum30a685f1991-06-27 15:51:29 +00003066
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003067static PyTypeObject sock_type = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003068 PyVarObject_HEAD_INIT(0, 0) /* Must fill in type value later */
3069 "_socket.socket", /* tp_name */
3070 sizeof(PySocketSockObject), /* tp_basicsize */
3071 0, /* tp_itemsize */
3072 (destructor)sock_dealloc, /* tp_dealloc */
3073 0, /* tp_print */
3074 0, /* tp_getattr */
3075 0, /* tp_setattr */
3076 0, /* tp_reserved */
3077 (reprfunc)sock_repr, /* tp_repr */
3078 0, /* tp_as_number */
3079 0, /* tp_as_sequence */
3080 0, /* tp_as_mapping */
3081 0, /* tp_hash */
3082 0, /* tp_call */
3083 0, /* tp_str */
3084 PyObject_GenericGetAttr, /* tp_getattro */
3085 0, /* tp_setattro */
3086 0, /* tp_as_buffer */
3087 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE, /* tp_flags */
3088 sock_doc, /* tp_doc */
3089 0, /* tp_traverse */
3090 0, /* tp_clear */
3091 0, /* tp_richcompare */
3092 0, /* tp_weaklistoffset */
3093 0, /* tp_iter */
3094 0, /* tp_iternext */
3095 sock_methods, /* tp_methods */
3096 sock_memberlist, /* tp_members */
3097 0, /* tp_getset */
3098 0, /* tp_base */
3099 0, /* tp_dict */
3100 0, /* tp_descr_get */
3101 0, /* tp_descr_set */
3102 0, /* tp_dictoffset */
3103 sock_initobj, /* tp_init */
3104 PyType_GenericAlloc, /* tp_alloc */
3105 sock_new, /* tp_new */
3106 PyObject_Del, /* tp_free */
Guido van Rossum6574b3e1991-06-25 21:36:08 +00003107};
3108
Guido van Rossum30a685f1991-06-27 15:51:29 +00003109
Guido van Rossum81194471991-07-27 21:42:02 +00003110/* Python interface to gethostname(). */
3111
3112/*ARGSUSED*/
Guido van Rossum73624e91994-10-10 17:59:00 +00003113static PyObject *
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00003114socket_gethostname(PyObject *self, PyObject *unused)
Guido van Rossum81194471991-07-27 21:42:02 +00003115{
Martin v. Löwis72f48422010-10-29 18:20:08 +00003116#ifdef MS_WINDOWS
3117 /* Don't use winsock's gethostname, as this returns the ANSI
3118 version of the hostname, whereas we need a Unicode string.
3119 Otherwise, gethostname apparently also returns the DNS name. */
Hirokazu Yamamoto09fff7a2010-11-05 17:24:13 +00003120 wchar_t buf[MAX_COMPUTERNAME_LENGTH + 1];
3121 DWORD size = sizeof(buf) / sizeof(wchar_t);
3122 PyObject *result;
Martin v. Löwis72f48422010-10-29 18:20:08 +00003123 if (!GetComputerNameExW(ComputerNamePhysicalDnsHostname, buf, &size)) {
3124 if (GetLastError() == ERROR_MORE_DATA) {
3125 /* MSDN says this may occur "because DNS allows longer names */
Hirokazu Yamamoto09fff7a2010-11-05 17:24:13 +00003126 if (size == 0) /* XXX: I'm not sure how to handle this */
3127 return PyUnicode_FromUnicode(NULL, 0);
3128 result = PyUnicode_FromUnicode(NULL, size - 1);
Martin v. Löwis72f48422010-10-29 18:20:08 +00003129 if (!result)
3130 return NULL;
Hirokazu Yamamoto09fff7a2010-11-05 17:24:13 +00003131 if (GetComputerNameExW(ComputerNamePhysicalDnsHostname,
3132 PyUnicode_AS_UNICODE(result),
3133 &size))
Martin v. Löwis72f48422010-10-29 18:20:08 +00003134 return result;
Benjamin Petersone857b292010-10-29 21:37:26 +00003135 Py_DECREF(result);
Martin v. Löwis72f48422010-10-29 18:20:08 +00003136 }
3137 return PyErr_SetExcFromWindowsErr(PyExc_WindowsError, GetLastError());
3138 }
3139 return PyUnicode_FromUnicode(buf, size);
3140#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003141 char buf[1024];
3142 int res;
3143 Py_BEGIN_ALLOW_THREADS
3144 res = gethostname(buf, (int) sizeof buf - 1);
3145 Py_END_ALLOW_THREADS
3146 if (res < 0)
3147 return set_error();
3148 buf[sizeof buf - 1] = '\0';
3149 return PyUnicode_FromString(buf);
Martin v. Löwis72f48422010-10-29 18:20:08 +00003150#endif
Guido van Rossum81194471991-07-27 21:42:02 +00003151}
Guido van Rossumff4949e1992-08-05 19:58:53 +00003152
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003153PyDoc_STRVAR(gethostname_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003154"gethostname() -> string\n\
3155\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003156Return the current host name.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003157
Antoine Pitrou061cfb52011-02-28 22:25:22 +00003158#ifdef HAVE_SETHOSTNAME
3159PyDoc_STRVAR(sethostname_doc,
3160"sethostname(name)\n\n\
3161Sets the hostname to name.");
3162
3163static PyObject *
3164socket_sethostname(PyObject *self, PyObject *args)
3165{
3166 PyObject *hnobj;
3167 Py_buffer buf;
3168 int res, flag = 0;
3169
3170 if (!PyArg_ParseTuple(args, "S:sethostname", &hnobj)) {
3171 PyErr_Clear();
3172 if (!PyArg_ParseTuple(args, "O&:sethostname",
3173 PyUnicode_FSConverter, &hnobj))
3174 return NULL;
3175 flag = 1;
3176 }
3177 res = PyObject_GetBuffer(hnobj, &buf, PyBUF_SIMPLE);
3178 if (!res) {
3179 res = sethostname(buf.buf, buf.len);
3180 PyBuffer_Release(&buf);
3181 }
3182 if (flag)
3183 Py_DECREF(hnobj);
3184 if (res)
3185 return set_error();
3186 Py_RETURN_NONE;
3187}
3188#endif
Guido van Rossumff4949e1992-08-05 19:58:53 +00003189
Guido van Rossum30a685f1991-06-27 15:51:29 +00003190/* Python interface to gethostbyname(name). */
3191
3192/*ARGSUSED*/
Guido van Rossum73624e91994-10-10 17:59:00 +00003193static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003194socket_gethostbyname(PyObject *self, PyObject *args)
Guido van Rossum30a685f1991-06-27 15:51:29 +00003195{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003196 char *name;
3197 sock_addr_t addrbuf;
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003198 PyObject *ret = NULL;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003199
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003200 if (!PyArg_ParseTuple(args, "et:gethostbyname", "idna", &name))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003201 return NULL;
3202 if (setipaddr(name, SAS2SA(&addrbuf), sizeof(addrbuf), AF_INET) < 0)
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003203 goto finally;
3204 ret = makeipaddr(SAS2SA(&addrbuf), sizeof(struct sockaddr_in));
3205finally:
3206 PyMem_Free(name);
3207 return ret;
Guido van Rossum30a685f1991-06-27 15:51:29 +00003208}
3209
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003210PyDoc_STRVAR(gethostbyname_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003211"gethostbyname(host) -> address\n\
3212\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003213Return the IP address (a string of the form '255.255.255.255') for a host.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003214
3215
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003216/* Convenience function common to gethostbyname_ex and gethostbyaddr */
3217
3218static PyObject *
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003219gethost_common(struct hostent *h, struct sockaddr *addr, int alen, int af)
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003220{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003221 char **pch;
3222 PyObject *rtn_tuple = (PyObject *)NULL;
3223 PyObject *name_list = (PyObject *)NULL;
3224 PyObject *addr_list = (PyObject *)NULL;
3225 PyObject *tmp;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003226
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003227 if (h == NULL) {
3228 /* Let's get real error message to return */
3229 set_herror(h_errno);
3230 return NULL;
3231 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00003232
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003233 if (h->h_addrtype != af) {
3234 /* Let's get real error message to return */
3235 PyErr_SetString(socket_error,
3236 (char *)strerror(EAFNOSUPPORT));
Christian Heimesada8c3b2008-03-18 18:26:33 +00003237
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003238 return NULL;
3239 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00003240
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003241 switch (af) {
Guido van Rossum67f7a382002-06-06 21:08:16 +00003242
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003243 case AF_INET:
3244 if (alen < sizeof(struct sockaddr_in))
3245 return NULL;
3246 break;
Guido van Rossum67f7a382002-06-06 21:08:16 +00003247
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00003248#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003249 case AF_INET6:
3250 if (alen < sizeof(struct sockaddr_in6))
3251 return NULL;
3252 break;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003253#endif
Guido van Rossum67f7a382002-06-06 21:08:16 +00003254
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003255 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00003256
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003257 if ((name_list = PyList_New(0)) == NULL)
3258 goto err;
Guido van Rossum67f7a382002-06-06 21:08:16 +00003259
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003260 if ((addr_list = PyList_New(0)) == NULL)
3261 goto err;
Guido van Rossum67f7a382002-06-06 21:08:16 +00003262
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003263 /* SF #1511317: h_aliases can be NULL */
3264 if (h->h_aliases) {
3265 for (pch = h->h_aliases; *pch != NULL; pch++) {
3266 int status;
3267 tmp = PyUnicode_FromString(*pch);
3268 if (tmp == NULL)
3269 goto err;
Guido van Rossum67f7a382002-06-06 21:08:16 +00003270
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003271 status = PyList_Append(name_list, tmp);
3272 Py_DECREF(tmp);
Guido van Rossum67f7a382002-06-06 21:08:16 +00003273
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003274 if (status)
3275 goto err;
3276 }
3277 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00003278
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003279 for (pch = h->h_addr_list; *pch != NULL; pch++) {
3280 int status;
Guido van Rossum67f7a382002-06-06 21:08:16 +00003281
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003282 switch (af) {
Guido van Rossum67f7a382002-06-06 21:08:16 +00003283
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003284 case AF_INET:
3285 {
3286 struct sockaddr_in sin;
3287 memset(&sin, 0, sizeof(sin));
3288 sin.sin_family = af;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003289#ifdef HAVE_SOCKADDR_SA_LEN
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003290 sin.sin_len = sizeof(sin);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003291#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003292 memcpy(&sin.sin_addr, *pch, sizeof(sin.sin_addr));
3293 tmp = makeipaddr((struct sockaddr *)&sin, sizeof(sin));
Guido van Rossum67f7a382002-06-06 21:08:16 +00003294
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003295 if (pch == h->h_addr_list && alen >= sizeof(sin))
3296 memcpy((char *) addr, &sin, sizeof(sin));
3297 break;
3298 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00003299
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00003300#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003301 case AF_INET6:
3302 {
3303 struct sockaddr_in6 sin6;
3304 memset(&sin6, 0, sizeof(sin6));
3305 sin6.sin6_family = af;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003306#ifdef HAVE_SOCKADDR_SA_LEN
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003307 sin6.sin6_len = sizeof(sin6);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003308#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003309 memcpy(&sin6.sin6_addr, *pch, sizeof(sin6.sin6_addr));
3310 tmp = makeipaddr((struct sockaddr *)&sin6,
3311 sizeof(sin6));
Guido van Rossum67f7a382002-06-06 21:08:16 +00003312
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003313 if (pch == h->h_addr_list && alen >= sizeof(sin6))
3314 memcpy((char *) addr, &sin6, sizeof(sin6));
3315 break;
3316 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003317#endif
Guido van Rossum67f7a382002-06-06 21:08:16 +00003318
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003319 default: /* can't happen */
3320 PyErr_SetString(socket_error,
3321 "unsupported address family");
3322 return NULL;
3323 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00003324
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003325 if (tmp == NULL)
3326 goto err;
Guido van Rossum67f7a382002-06-06 21:08:16 +00003327
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003328 status = PyList_Append(addr_list, tmp);
3329 Py_DECREF(tmp);
Guido van Rossum67f7a382002-06-06 21:08:16 +00003330
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003331 if (status)
3332 goto err;
3333 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00003334
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003335 rtn_tuple = Py_BuildValue("sOO", h->h_name, name_list, addr_list);
Guido van Rossum67f7a382002-06-06 21:08:16 +00003336
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003337 err:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003338 Py_XDECREF(name_list);
3339 Py_XDECREF(addr_list);
3340 return rtn_tuple;
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003341}
3342
3343
3344/* Python interface to gethostbyname_ex(name). */
3345
3346/*ARGSUSED*/
3347static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003348socket_gethostbyname_ex(PyObject *self, PyObject *args)
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003349{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003350 char *name;
3351 struct hostent *h;
Martin v. Löwisc16f3bd2003-05-03 09:14:54 +00003352#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003353 struct sockaddr_storage addr;
Martin v. Löwisc16f3bd2003-05-03 09:14:54 +00003354#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003355 struct sockaddr_in addr;
Martin v. Löwisc16f3bd2003-05-03 09:14:54 +00003356#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003357 struct sockaddr *sa;
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003358 PyObject *ret = NULL;
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003359#ifdef HAVE_GETHOSTBYNAME_R
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003360 struct hostent hp_allocated;
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003361#ifdef HAVE_GETHOSTBYNAME_R_3_ARG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003362 struct hostent_data data;
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003363#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003364 char buf[16384];
3365 int buf_len = (sizeof buf) - 1;
3366 int errnop;
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003367#endif
3368#if defined(HAVE_GETHOSTBYNAME_R_3_ARG) || defined(HAVE_GETHOSTBYNAME_R_6_ARG)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003369 int result;
Guido van Rossume9cd07b1999-03-15 21:40:14 +00003370#endif
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003371#endif /* HAVE_GETHOSTBYNAME_R */
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003372
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003373 if (!PyArg_ParseTuple(args, "et:gethostbyname_ex", "idna", &name))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003374 return NULL;
3375 if (setipaddr(name, (struct sockaddr *)&addr, sizeof(addr), AF_INET) < 0)
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003376 goto finally;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003377 Py_BEGIN_ALLOW_THREADS
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003378#ifdef HAVE_GETHOSTBYNAME_R
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003379#if defined(HAVE_GETHOSTBYNAME_R_6_ARG)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003380 result = gethostbyname_r(name, &hp_allocated, buf, buf_len,
3381 &h, &errnop);
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003382#elif defined(HAVE_GETHOSTBYNAME_R_5_ARG)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003383 h = gethostbyname_r(name, &hp_allocated, buf, buf_len, &errnop);
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003384#else /* HAVE_GETHOSTBYNAME_R_3_ARG */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003385 memset((void *) &data, '\0', sizeof(data));
3386 result = gethostbyname_r(name, &hp_allocated, &data);
3387 h = (result != 0) ? NULL : &hp_allocated;
Guido van Rossume9cd07b1999-03-15 21:40:14 +00003388#endif
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003389#else /* not HAVE_GETHOSTBYNAME_R */
Guido van Rossum3baaa131999-03-22 21:44:51 +00003390#ifdef USE_GETHOSTBYNAME_LOCK
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003391 PyThread_acquire_lock(netdb_lock, 1);
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003392#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003393 h = gethostbyname(name);
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003394#endif /* HAVE_GETHOSTBYNAME_R */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003395 Py_END_ALLOW_THREADS
3396 /* Some C libraries would require addr.__ss_family instead of
3397 addr.ss_family.
3398 Therefore, we cast the sockaddr_storage into sockaddr to
3399 access sa_family. */
3400 sa = (struct sockaddr*)&addr;
3401 ret = gethost_common(h, (struct sockaddr *)&addr, sizeof(addr),
3402 sa->sa_family);
Guido van Rossum3baaa131999-03-22 21:44:51 +00003403#ifdef USE_GETHOSTBYNAME_LOCK
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003404 PyThread_release_lock(netdb_lock);
Guido van Rossum955becc1999-03-22 20:14:53 +00003405#endif
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003406finally:
3407 PyMem_Free(name);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003408 return ret;
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003409}
3410
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003411PyDoc_STRVAR(ghbn_ex_doc,
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003412"gethostbyname_ex(host) -> (name, aliaslist, addresslist)\n\
3413\n\
3414Return the true host name, a list of aliases, and a list of IP addresses,\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003415for a host. The host argument is a string giving a host name or IP number.");
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003416
3417
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00003418/* Python interface to gethostbyaddr(IP). */
3419
3420/*ARGSUSED*/
3421static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003422socket_gethostbyaddr(PyObject *self, PyObject *args)
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00003423{
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00003424#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003425 struct sockaddr_storage addr;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003426#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003427 struct sockaddr_in addr;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003428#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003429 struct sockaddr *sa = (struct sockaddr *)&addr;
3430 char *ip_num;
3431 struct hostent *h;
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003432 PyObject *ret = NULL;
Guido van Rossum4f199ea1998-04-09 20:56:35 +00003433#ifdef HAVE_GETHOSTBYNAME_R
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003434 struct hostent hp_allocated;
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003435#ifdef HAVE_GETHOSTBYNAME_R_3_ARG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003436 struct hostent_data data;
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003437#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003438 /* glibcs up to 2.10 assume that the buf argument to
3439 gethostbyaddr_r is 8-byte aligned, which at least llvm-gcc
3440 does not ensure. The attribute below instructs the compiler
3441 to maintain this alignment. */
3442 char buf[16384] Py_ALIGNED(8);
3443 int buf_len = (sizeof buf) - 1;
3444 int errnop;
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003445#endif
3446#if defined(HAVE_GETHOSTBYNAME_R_3_ARG) || defined(HAVE_GETHOSTBYNAME_R_6_ARG)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003447 int result;
Guido van Rossume9cd07b1999-03-15 21:40:14 +00003448#endif
Guido van Rossum4f199ea1998-04-09 20:56:35 +00003449#endif /* HAVE_GETHOSTBYNAME_R */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003450 char *ap;
3451 int al;
3452 int af;
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00003453
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003454 if (!PyArg_ParseTuple(args, "et:gethostbyaddr", "idna", &ip_num))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003455 return NULL;
3456 af = AF_UNSPEC;
3457 if (setipaddr(ip_num, sa, sizeof(addr), af) < 0)
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003458 goto finally;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003459 af = sa->sa_family;
3460 ap = NULL;
Brett Cannonb94767f2011-02-22 20:15:44 +00003461 /* al = 0; */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003462 switch (af) {
3463 case AF_INET:
3464 ap = (char *)&((struct sockaddr_in *)sa)->sin_addr;
3465 al = sizeof(((struct sockaddr_in *)sa)->sin_addr);
3466 break;
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00003467#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003468 case AF_INET6:
3469 ap = (char *)&((struct sockaddr_in6 *)sa)->sin6_addr;
3470 al = sizeof(((struct sockaddr_in6 *)sa)->sin6_addr);
3471 break;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003472#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003473 default:
3474 PyErr_SetString(socket_error, "unsupported address family");
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003475 goto finally;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003476 }
3477 Py_BEGIN_ALLOW_THREADS
Guido van Rossum4f199ea1998-04-09 20:56:35 +00003478#ifdef HAVE_GETHOSTBYNAME_R
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003479#if defined(HAVE_GETHOSTBYNAME_R_6_ARG)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003480 result = gethostbyaddr_r(ap, al, af,
3481 &hp_allocated, buf, buf_len,
3482 &h, &errnop);
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003483#elif defined(HAVE_GETHOSTBYNAME_R_5_ARG)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003484 h = gethostbyaddr_r(ap, al, af,
3485 &hp_allocated, buf, buf_len, &errnop);
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003486#else /* HAVE_GETHOSTBYNAME_R_3_ARG */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003487 memset((void *) &data, '\0', sizeof(data));
3488 result = gethostbyaddr_r(ap, al, af, &hp_allocated, &data);
3489 h = (result != 0) ? NULL : &hp_allocated;
Guido van Rossume9cd07b1999-03-15 21:40:14 +00003490#endif
Guido van Rossum4f199ea1998-04-09 20:56:35 +00003491#else /* not HAVE_GETHOSTBYNAME_R */
Guido van Rossum3baaa131999-03-22 21:44:51 +00003492#ifdef USE_GETHOSTBYNAME_LOCK
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003493 PyThread_acquire_lock(netdb_lock, 1);
Guido van Rossum4f199ea1998-04-09 20:56:35 +00003494#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003495 h = gethostbyaddr(ap, al, af);
Guido van Rossum4f199ea1998-04-09 20:56:35 +00003496#endif /* HAVE_GETHOSTBYNAME_R */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003497 Py_END_ALLOW_THREADS
3498 ret = gethost_common(h, (struct sockaddr *)&addr, sizeof(addr), af);
Guido van Rossum3baaa131999-03-22 21:44:51 +00003499#ifdef USE_GETHOSTBYNAME_LOCK
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003500 PyThread_release_lock(netdb_lock);
Guido van Rossum3baaa131999-03-22 21:44:51 +00003501#endif
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003502finally:
3503 PyMem_Free(ip_num);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003504 return ret;
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00003505}
3506
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003507PyDoc_STRVAR(gethostbyaddr_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003508"gethostbyaddr(host) -> (name, aliaslist, addresslist)\n\
3509\n\
3510Return the true host name, a list of aliases, and a list of IP addresses,\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003511for a host. The host argument is a string giving a host name or IP number.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003512
Guido van Rossum30a685f1991-06-27 15:51:29 +00003513
3514/* Python interface to getservbyname(name).
3515 This only returns the port number, since the other info is already
3516 known or not useful (like the list of aliases). */
3517
3518/*ARGSUSED*/
Guido van Rossum73624e91994-10-10 17:59:00 +00003519static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003520socket_getservbyname(PyObject *self, PyObject *args)
Guido van Rossum30a685f1991-06-27 15:51:29 +00003521{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003522 char *name, *proto=NULL;
3523 struct servent *sp;
3524 if (!PyArg_ParseTuple(args, "s|s:getservbyname", &name, &proto))
3525 return NULL;
3526 Py_BEGIN_ALLOW_THREADS
3527 sp = getservbyname(name, proto);
3528 Py_END_ALLOW_THREADS
3529 if (sp == NULL) {
3530 PyErr_SetString(socket_error, "service/proto not found");
3531 return NULL;
3532 }
3533 return PyLong_FromLong((long) ntohs(sp->s_port));
Guido van Rossum30a685f1991-06-27 15:51:29 +00003534}
3535
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003536PyDoc_STRVAR(getservbyname_doc,
Barry Warsaw11b91a02004-06-28 00:50:43 +00003537"getservbyname(servicename[, protocolname]) -> integer\n\
Guido van Rossum82a5c661998-07-07 20:45:43 +00003538\n\
3539Return a port number from a service name and protocol name.\n\
Barry Warsaw11b91a02004-06-28 00:50:43 +00003540The optional protocol name, if given, should be 'tcp' or 'udp',\n\
3541otherwise any protocol will match.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003542
Guido van Rossum30a685f1991-06-27 15:51:29 +00003543
Barry Warsaw11b91a02004-06-28 00:50:43 +00003544/* Python interface to getservbyport(port).
3545 This only returns the service name, since the other info is already
3546 known or not useful (like the list of aliases). */
3547
3548/*ARGSUSED*/
3549static PyObject *
3550socket_getservbyport(PyObject *self, PyObject *args)
3551{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003552 int port;
3553 char *proto=NULL;
3554 struct servent *sp;
3555 if (!PyArg_ParseTuple(args, "i|s:getservbyport", &port, &proto))
3556 return NULL;
3557 if (port < 0 || port > 0xffff) {
3558 PyErr_SetString(
3559 PyExc_OverflowError,
3560 "getservbyport: port must be 0-65535.");
3561 return NULL;
3562 }
3563 Py_BEGIN_ALLOW_THREADS
3564 sp = getservbyport(htons((short)port), proto);
3565 Py_END_ALLOW_THREADS
3566 if (sp == NULL) {
3567 PyErr_SetString(socket_error, "port/proto not found");
3568 return NULL;
3569 }
3570 return PyUnicode_FromString(sp->s_name);
Barry Warsaw11b91a02004-06-28 00:50:43 +00003571}
3572
3573PyDoc_STRVAR(getservbyport_doc,
3574"getservbyport(port[, protocolname]) -> string\n\
3575\n\
3576Return the service name from a port number and protocol name.\n\
3577The optional protocol name, if given, should be 'tcp' or 'udp',\n\
3578otherwise any protocol will match.");
3579
Guido van Rossum3901d851996-12-19 16:35:04 +00003580/* Python interface to getprotobyname(name).
3581 This only returns the protocol number, since the other info is
3582 already known or not useful (like the list of aliases). */
3583
3584/*ARGSUSED*/
3585static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003586socket_getprotobyname(PyObject *self, PyObject *args)
Guido van Rossum3901d851996-12-19 16:35:04 +00003587{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003588 char *name;
3589 struct protoent *sp;
3590 if (!PyArg_ParseTuple(args, "s:getprotobyname", &name))
3591 return NULL;
3592 Py_BEGIN_ALLOW_THREADS
3593 sp = getprotobyname(name);
3594 Py_END_ALLOW_THREADS
3595 if (sp == NULL) {
3596 PyErr_SetString(socket_error, "protocol not found");
3597 return NULL;
3598 }
3599 return PyLong_FromLong((long) sp->p_proto);
Guido van Rossum3901d851996-12-19 16:35:04 +00003600}
3601
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003602PyDoc_STRVAR(getprotobyname_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003603"getprotobyname(name) -> integer\n\
3604\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003605Return the protocol number for the named protocol. (Rarely used.)");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003606
Guido van Rossum3901d851996-12-19 16:35:04 +00003607
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00003608#ifndef NO_DUP
3609/* dup() function for socket fds */
3610
3611static PyObject *
3612socket_dup(PyObject *self, PyObject *fdobj)
3613{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003614 SOCKET_T fd, newfd;
3615 PyObject *newfdobj;
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00003616
3617
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003618 fd = PyLong_AsSocket_t(fdobj);
3619 if (fd == (SOCKET_T)(-1) && PyErr_Occurred())
3620 return NULL;
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00003621
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003622 newfd = dup_socket(fd);
3623 if (newfd == INVALID_SOCKET)
3624 return set_error();
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00003625
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003626 newfdobj = PyLong_FromSocket_t(newfd);
3627 if (newfdobj == NULL)
3628 SOCKETCLOSE(newfd);
3629 return newfdobj;
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00003630}
3631
3632PyDoc_STRVAR(dup_doc,
3633"dup(integer) -> integer\n\
3634\n\
3635Duplicate an integer socket file descriptor. This is like os.dup(), but for\n\
3636sockets; on some platforms os.dup() won't work for socket file descriptors.");
3637#endif
3638
3639
Dave Cole331708b2004-08-09 04:51:41 +00003640#ifdef HAVE_SOCKETPAIR
3641/* Create a pair of sockets using the socketpair() function.
Dave Cole07fda7e2004-08-23 05:16:23 +00003642 Arguments as for socket() except the default family is AF_UNIX if
Dave Colee8bbfe42004-08-26 00:51:16 +00003643 defined on the platform; otherwise, the default is AF_INET. */
Dave Cole331708b2004-08-09 04:51:41 +00003644
3645/*ARGSUSED*/
3646static PyObject *
3647socket_socketpair(PyObject *self, PyObject *args)
3648{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003649 PySocketSockObject *s0 = NULL, *s1 = NULL;
3650 SOCKET_T sv[2];
3651 int family, type = SOCK_STREAM, proto = 0;
3652 PyObject *res = NULL;
Dave Cole331708b2004-08-09 04:51:41 +00003653
3654#if defined(AF_UNIX)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003655 family = AF_UNIX;
Dave Cole331708b2004-08-09 04:51:41 +00003656#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003657 family = AF_INET;
Dave Cole331708b2004-08-09 04:51:41 +00003658#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003659 if (!PyArg_ParseTuple(args, "|iii:socketpair",
3660 &family, &type, &proto))
3661 return NULL;
3662 /* Create a pair of socket fds */
3663 if (socketpair(family, type, proto, sv) < 0)
3664 return set_error();
3665 s0 = new_sockobject(sv[0], family, type, proto);
3666 if (s0 == NULL)
3667 goto finally;
3668 s1 = new_sockobject(sv[1], family, type, proto);
3669 if (s1 == NULL)
3670 goto finally;
3671 res = PyTuple_Pack(2, s0, s1);
Dave Cole331708b2004-08-09 04:51:41 +00003672
3673finally:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003674 if (res == NULL) {
3675 if (s0 == NULL)
3676 SOCKETCLOSE(sv[0]);
3677 if (s1 == NULL)
3678 SOCKETCLOSE(sv[1]);
3679 }
3680 Py_XDECREF(s0);
3681 Py_XDECREF(s1);
3682 return res;
Dave Cole331708b2004-08-09 04:51:41 +00003683}
3684
3685PyDoc_STRVAR(socketpair_doc,
3686"socketpair([family[, type[, proto]]]) -> (socket object, socket object)\n\
3687\n\
3688Create a pair of socket objects from the sockets returned by the platform\n\
3689socketpair() function.\n\
Dave Cole07fda7e2004-08-23 05:16:23 +00003690The arguments are the same as for socket() except the default family is\n\
Dave Colee8bbfe42004-08-26 00:51:16 +00003691AF_UNIX if defined on the platform; otherwise, the default is AF_INET.");
Dave Cole331708b2004-08-09 04:51:41 +00003692
3693#endif /* HAVE_SOCKETPAIR */
3694
3695
Guido van Rossum006bf911996-06-12 04:04:55 +00003696static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003697socket_ntohs(PyObject *self, PyObject *args)
Guido van Rossum006bf911996-06-12 04:04:55 +00003698{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003699 int x1, x2;
Guido van Rossum006bf911996-06-12 04:04:55 +00003700
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003701 if (!PyArg_ParseTuple(args, "i:ntohs", &x1)) {
3702 return NULL;
3703 }
3704 if (x1 < 0) {
3705 PyErr_SetString(PyExc_OverflowError,
3706 "can't convert negative number to unsigned long");
3707 return NULL;
3708 }
3709 x2 = (unsigned int)ntohs((unsigned short)x1);
3710 return PyLong_FromLong(x2);
Guido van Rossum006bf911996-06-12 04:04:55 +00003711}
3712
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003713PyDoc_STRVAR(ntohs_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003714"ntohs(integer) -> integer\n\
3715\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003716Convert a 16-bit integer from network to host byte order.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003717
3718
Guido van Rossum006bf911996-06-12 04:04:55 +00003719static PyObject *
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003720socket_ntohl(PyObject *self, PyObject *arg)
Guido van Rossum006bf911996-06-12 04:04:55 +00003721{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003722 unsigned long x;
Guido van Rossum006bf911996-06-12 04:04:55 +00003723
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003724 if (PyLong_Check(arg)) {
3725 x = PyLong_AsUnsignedLong(arg);
3726 if (x == (unsigned long) -1 && PyErr_Occurred())
3727 return NULL;
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003728#if SIZEOF_LONG > 4
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003729 {
3730 unsigned long y;
3731 /* only want the trailing 32 bits */
3732 y = x & 0xFFFFFFFFUL;
3733 if (y ^ x)
3734 return PyErr_Format(PyExc_OverflowError,
3735 "long int larger than 32 bits");
3736 x = y;
3737 }
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003738#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003739 }
3740 else
3741 return PyErr_Format(PyExc_TypeError,
3742 "expected int/long, %s found",
3743 Py_TYPE(arg)->tp_name);
3744 if (x == (unsigned long) -1 && PyErr_Occurred())
3745 return NULL;
3746 return PyLong_FromUnsignedLong(ntohl(x));
Guido van Rossum006bf911996-06-12 04:04:55 +00003747}
3748
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003749PyDoc_STRVAR(ntohl_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003750"ntohl(integer) -> integer\n\
3751\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003752Convert a 32-bit integer from network to host byte order.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003753
3754
Guido van Rossum006bf911996-06-12 04:04:55 +00003755static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003756socket_htons(PyObject *self, PyObject *args)
Guido van Rossum006bf911996-06-12 04:04:55 +00003757{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003758 int x1, x2;
Guido van Rossum006bf911996-06-12 04:04:55 +00003759
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003760 if (!PyArg_ParseTuple(args, "i:htons", &x1)) {
3761 return NULL;
3762 }
3763 if (x1 < 0) {
3764 PyErr_SetString(PyExc_OverflowError,
3765 "can't convert negative number to unsigned long");
3766 return NULL;
3767 }
3768 x2 = (unsigned int)htons((unsigned short)x1);
3769 return PyLong_FromLong(x2);
Guido van Rossum006bf911996-06-12 04:04:55 +00003770}
3771
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003772PyDoc_STRVAR(htons_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003773"htons(integer) -> integer\n\
3774\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003775Convert a 16-bit integer from host to network byte order.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003776
3777
Guido van Rossum006bf911996-06-12 04:04:55 +00003778static PyObject *
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003779socket_htonl(PyObject *self, PyObject *arg)
Guido van Rossum006bf911996-06-12 04:04:55 +00003780{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003781 unsigned long x;
Guido van Rossum006bf911996-06-12 04:04:55 +00003782
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003783 if (PyLong_Check(arg)) {
3784 x = PyLong_AsUnsignedLong(arg);
3785 if (x == (unsigned long) -1 && PyErr_Occurred())
3786 return NULL;
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003787#if SIZEOF_LONG > 4
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003788 {
3789 unsigned long y;
3790 /* only want the trailing 32 bits */
3791 y = x & 0xFFFFFFFFUL;
3792 if (y ^ x)
3793 return PyErr_Format(PyExc_OverflowError,
3794 "long int larger than 32 bits");
3795 x = y;
3796 }
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003797#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003798 }
3799 else
3800 return PyErr_Format(PyExc_TypeError,
3801 "expected int/long, %s found",
3802 Py_TYPE(arg)->tp_name);
3803 return PyLong_FromUnsignedLong(htonl((unsigned long)x));
Guido van Rossum006bf911996-06-12 04:04:55 +00003804}
3805
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003806PyDoc_STRVAR(htonl_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003807"htonl(integer) -> integer\n\
3808\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003809Convert a 32-bit integer from host to network byte order.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003810
Guido van Rossum3eede5a2002-06-07 02:08:35 +00003811/* socket.inet_aton() and socket.inet_ntoa() functions. */
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003812
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003813PyDoc_STRVAR(inet_aton_doc,
Guido van Rossum7d0a8262007-05-21 23:13:11 +00003814"inet_aton(string) -> bytes giving packed 32-bit IP representation\n\
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003815\n\
Guido van Rossumc6a164b1999-08-20 19:11:27 +00003816Convert 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 +00003817binary format used in low-level network functions.");
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003818
3819static PyObject*
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003820socket_inet_aton(PyObject *self, PyObject *args)
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003821{
Guido van Rossuma2e48551999-09-09 15:42:59 +00003822#ifndef INADDR_NONE
3823#define INADDR_NONE (-1)
3824#endif
Neal Norwitz88f115b2003-02-13 02:15:42 +00003825#ifdef HAVE_INET_ATON
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003826 struct in_addr buf;
Thomas Wouters477c8d52006-05-27 19:21:47 +00003827#endif
3828
3829#if !defined(HAVE_INET_ATON) || defined(USE_INET_ATON_WEAKLINK)
Benjamin Petersonf91df042009-02-13 02:50:59 +00003830#if (SIZEOF_INT != 4)
3831#error "Not sure if in_addr_t exists and int is not 32-bits."
3832#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003833 /* Have to use inet_addr() instead */
3834 unsigned int packed_addr;
Tim Peters1df9fdd2003-02-13 03:13:40 +00003835#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003836 char *ip_addr;
Guido van Rossumc6a164b1999-08-20 19:11:27 +00003837
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003838 if (!PyArg_ParseTuple(args, "s:inet_aton", &ip_addr))
3839 return NULL;
Guido van Rossumad05cdf2003-02-12 23:08:22 +00003840
Tim Peters1df9fdd2003-02-13 03:13:40 +00003841
3842#ifdef HAVE_INET_ATON
Thomas Wouters477c8d52006-05-27 19:21:47 +00003843
3844#ifdef USE_INET_ATON_WEAKLINK
3845 if (inet_aton != NULL) {
3846#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003847 if (inet_aton(ip_addr, &buf))
3848 return PyBytes_FromStringAndSize((char *)(&buf),
3849 sizeof(buf));
Guido van Rossumad05cdf2003-02-12 23:08:22 +00003850
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003851 PyErr_SetString(socket_error,
3852 "illegal IP address string passed to inet_aton");
3853 return NULL;
Guido van Rossumad05cdf2003-02-12 23:08:22 +00003854
Thomas Wouters477c8d52006-05-27 19:21:47 +00003855#ifdef USE_INET_ATON_WEAKLINK
3856 } else {
3857#endif
3858
3859#endif
3860
3861#if !defined(HAVE_INET_ATON) || defined(USE_INET_ATON_WEAKLINK)
3862
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003863 /* special-case this address as inet_addr might return INADDR_NONE
3864 * for this */
3865 if (strcmp(ip_addr, "255.255.255.255") == 0) {
3866 packed_addr = 0xFFFFFFFF;
3867 } else {
Thomas Wouters477c8d52006-05-27 19:21:47 +00003868
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003869 packed_addr = inet_addr(ip_addr);
Guido van Rossumc6a164b1999-08-20 19:11:27 +00003870
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003871 if (packed_addr == INADDR_NONE) { /* invalid address */
3872 PyErr_SetString(socket_error,
3873 "illegal IP address string passed to inet_aton");
3874 return NULL;
3875 }
3876 }
3877 return PyBytes_FromStringAndSize((char *) &packed_addr,
3878 sizeof(packed_addr));
Thomas Wouters477c8d52006-05-27 19:21:47 +00003879
3880#ifdef USE_INET_ATON_WEAKLINK
3881 }
3882#endif
3883
Guido van Rossumad05cdf2003-02-12 23:08:22 +00003884#endif
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003885}
3886
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003887PyDoc_STRVAR(inet_ntoa_doc,
Fred Drakee0661342000-03-07 14:05:16 +00003888"inet_ntoa(packed_ip) -> ip_address_string\n\
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003889\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003890Convert an IP address from 32-bit packed binary format to string format");
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003891
3892static PyObject*
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003893socket_inet_ntoa(PyObject *self, PyObject *args)
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003894{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003895 char *packed_str;
3896 int addr_len;
3897 struct in_addr packed_addr;
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003898
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003899 if (!PyArg_ParseTuple(args, "y#:inet_ntoa", &packed_str, &addr_len)) {
3900 return NULL;
3901 }
Guido van Rossum48a680c2001-03-02 06:34:14 +00003902
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003903 if (addr_len != sizeof(packed_addr)) {
3904 PyErr_SetString(socket_error,
3905 "packed IP wrong length for inet_ntoa");
3906 return NULL;
3907 }
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003908
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003909 memcpy(&packed_addr, packed_str, addr_len);
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003910
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003911 return PyUnicode_FromString(inet_ntoa(packed_addr));
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003912}
Guido van Rossum82a5c661998-07-07 20:45:43 +00003913
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003914#ifdef HAVE_INET_PTON
3915
3916PyDoc_STRVAR(inet_pton_doc,
3917"inet_pton(af, ip) -> packed IP address string\n\
3918\n\
3919Convert an IP address from string format to a packed string suitable\n\
3920for use with low-level network functions.");
3921
3922static PyObject *
3923socket_inet_pton(PyObject *self, PyObject *args)
3924{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003925 int af;
3926 char* ip;
3927 int retval;
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003928#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003929 char packed[MAX(sizeof(struct in_addr), sizeof(struct in6_addr))];
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003930#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003931 char packed[sizeof(struct in_addr)];
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003932#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003933 if (!PyArg_ParseTuple(args, "is:inet_pton", &af, &ip)) {
3934 return NULL;
3935 }
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003936
Martin v. Löwis04697e82004-06-02 12:35:29 +00003937#if !defined(ENABLE_IPV6) && defined(AF_INET6)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003938 if(af == AF_INET6) {
3939 PyErr_SetString(socket_error,
3940 "can't use AF_INET6, IPv6 is disabled");
3941 return NULL;
3942 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00003943#endif
Martin v. Löwis10649092003-08-05 06:25:06 +00003944
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003945 retval = inet_pton(af, ip, packed);
3946 if (retval < 0) {
3947 PyErr_SetFromErrno(socket_error);
3948 return NULL;
3949 } else if (retval == 0) {
3950 PyErr_SetString(socket_error,
3951 "illegal IP address string passed to inet_pton");
3952 return NULL;
3953 } else if (af == AF_INET) {
3954 return PyBytes_FromStringAndSize(packed,
3955 sizeof(struct in_addr));
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003956#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003957 } else if (af == AF_INET6) {
3958 return PyBytes_FromStringAndSize(packed,
3959 sizeof(struct in6_addr));
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003960#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003961 } else {
3962 PyErr_SetString(socket_error, "unknown address family");
3963 return NULL;
3964 }
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003965}
Thomas Wouters477c8d52006-05-27 19:21:47 +00003966
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003967PyDoc_STRVAR(inet_ntop_doc,
3968"inet_ntop(af, packed_ip) -> string formatted IP address\n\
3969\n\
3970Convert a packed IP address of the given family to string format.");
3971
3972static PyObject *
3973socket_inet_ntop(PyObject *self, PyObject *args)
3974{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003975 int af;
3976 char* packed;
3977 int len;
3978 const char* retval;
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003979#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003980 char ip[MAX(INET_ADDRSTRLEN, INET6_ADDRSTRLEN) + 1];
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003981#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003982 char ip[INET_ADDRSTRLEN + 1];
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003983#endif
Thomas Wouters477c8d52006-05-27 19:21:47 +00003984
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003985 /* Guarantee NUL-termination for PyUnicode_FromString() below */
3986 memset((void *) &ip[0], '\0', sizeof(ip));
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003987
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003988 if (!PyArg_ParseTuple(args, "iy#:inet_ntop", &af, &packed, &len)) {
3989 return NULL;
3990 }
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003991
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003992 if (af == AF_INET) {
3993 if (len != sizeof(struct in_addr)) {
3994 PyErr_SetString(PyExc_ValueError,
3995 "invalid length of packed IP address string");
3996 return NULL;
3997 }
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003998#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003999 } else if (af == AF_INET6) {
4000 if (len != sizeof(struct in6_addr)) {
4001 PyErr_SetString(PyExc_ValueError,
4002 "invalid length of packed IP address string");
4003 return NULL;
4004 }
Martin v. Löwisa94568a2003-05-10 07:36:56 +00004005#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004006 } else {
4007 PyErr_Format(PyExc_ValueError,
4008 "unknown address family %d", af);
4009 return NULL;
4010 }
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00004011
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004012 retval = inet_ntop(af, packed, ip, sizeof(ip));
4013 if (!retval) {
4014 PyErr_SetFromErrno(socket_error);
4015 return NULL;
4016 } else {
4017 return PyUnicode_FromString(retval);
4018 }
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00004019
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004020 /* NOTREACHED */
4021 PyErr_SetString(PyExc_RuntimeError, "invalid handling of inet_ntop");
4022 return NULL;
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00004023}
4024
4025#endif /* HAVE_INET_PTON */
4026
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004027/* Python interface to getaddrinfo(host, port). */
4028
4029/*ARGSUSED*/
4030static PyObject *
Giampaolo Rodolàccfb91c2010-08-17 15:30:23 +00004031socket_getaddrinfo(PyObject *self, PyObject *args, PyObject* kwargs)
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004032{
Giampaolo Rodolàccfb91c2010-08-17 15:30:23 +00004033 static char* kwnames[] = {"host", "port", "family", "type", "proto",
4034 "flags", 0};
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004035 struct addrinfo hints, *res;
4036 struct addrinfo *res0 = NULL;
4037 PyObject *hobj = NULL;
4038 PyObject *pobj = (PyObject *)NULL;
4039 char pbuf[30];
4040 char *hptr, *pptr;
4041 int family, socktype, protocol, flags;
4042 int error;
4043 PyObject *all = (PyObject *)NULL;
4044 PyObject *idna = NULL;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004045
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004046 family = socktype = protocol = flags = 0;
4047 family = AF_UNSPEC;
Giampaolo Rodolàccfb91c2010-08-17 15:30:23 +00004048 if (!PyArg_ParseTupleAndKeywords(args, kwargs, "OO|iiii:getaddrinfo",
4049 kwnames, &hobj, &pobj, &family, &socktype,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004050 &protocol, &flags)) {
4051 return NULL;
4052 }
4053 if (hobj == Py_None) {
4054 hptr = NULL;
4055 } else if (PyUnicode_Check(hobj)) {
4056 idna = PyObject_CallMethod(hobj, "encode", "s", "idna");
4057 if (!idna)
4058 return NULL;
4059 assert(PyBytes_Check(idna));
4060 hptr = PyBytes_AS_STRING(idna);
4061 } else if (PyBytes_Check(hobj)) {
4062 hptr = PyBytes_AsString(hobj);
4063 } else {
4064 PyErr_SetString(PyExc_TypeError,
4065 "getaddrinfo() argument 1 must be string or None");
4066 return NULL;
4067 }
4068 if (PyLong_CheckExact(pobj)) {
4069 long value = PyLong_AsLong(pobj);
4070 if (value == -1 && PyErr_Occurred())
4071 goto err;
4072 PyOS_snprintf(pbuf, sizeof(pbuf), "%ld", value);
4073 pptr = pbuf;
4074 } else if (PyUnicode_Check(pobj)) {
4075 pptr = _PyUnicode_AsString(pobj);
Alexander Belopolskye239d232010-12-08 23:31:48 +00004076 if (pptr == NULL)
4077 goto err;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004078 } else if (PyBytes_Check(pobj)) {
Alexander Belopolskye239d232010-12-08 23:31:48 +00004079 pptr = PyBytes_AS_STRING(pobj);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004080 } else if (pobj == Py_None) {
4081 pptr = (char *)NULL;
4082 } else {
4083 PyErr_SetString(socket_error, "Int or String expected");
4084 goto err;
4085 }
4086 memset(&hints, 0, sizeof(hints));
4087 hints.ai_family = family;
4088 hints.ai_socktype = socktype;
4089 hints.ai_protocol = protocol;
4090 hints.ai_flags = flags;
4091 Py_BEGIN_ALLOW_THREADS
4092 ACQUIRE_GETADDRINFO_LOCK
4093 error = getaddrinfo(hptr, pptr, &hints, &res0);
4094 Py_END_ALLOW_THREADS
4095 RELEASE_GETADDRINFO_LOCK /* see comment in setipaddr() */
4096 if (error) {
4097 set_gaierror(error);
4098 goto err;
4099 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004100
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004101 if ((all = PyList_New(0)) == NULL)
4102 goto err;
4103 for (res = res0; res; res = res->ai_next) {
4104 PyObject *single;
4105 PyObject *addr =
4106 makesockaddr(-1, res->ai_addr, res->ai_addrlen, protocol);
4107 if (addr == NULL)
4108 goto err;
4109 single = Py_BuildValue("iiisO", res->ai_family,
4110 res->ai_socktype, res->ai_protocol,
4111 res->ai_canonname ? res->ai_canonname : "",
4112 addr);
4113 Py_DECREF(addr);
4114 if (single == NULL)
4115 goto err;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004116
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004117 if (PyList_Append(all, single))
4118 goto err;
4119 Py_XDECREF(single);
4120 }
4121 Py_XDECREF(idna);
4122 if (res0)
4123 freeaddrinfo(res0);
4124 return all;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004125 err:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004126 Py_XDECREF(all);
4127 Py_XDECREF(idna);
4128 if (res0)
4129 freeaddrinfo(res0);
4130 return (PyObject *)NULL;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004131}
4132
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004133PyDoc_STRVAR(getaddrinfo_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00004134"getaddrinfo(host, port [, family, socktype, proto, flags])\n\
4135 -> list of (family, socktype, proto, canonname, sockaddr)\n\
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004136\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004137Resolve host and port into addrinfo struct.");
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004138
4139/* Python interface to getnameinfo(sa, flags). */
4140
4141/*ARGSUSED*/
4142static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004143socket_getnameinfo(PyObject *self, PyObject *args)
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004144{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004145 PyObject *sa = (PyObject *)NULL;
4146 int flags;
4147 char *hostp;
4148 int port, flowinfo, scope_id;
4149 char hbuf[NI_MAXHOST], pbuf[NI_MAXSERV];
4150 struct addrinfo hints, *res = NULL;
4151 int error;
4152 PyObject *ret = (PyObject *)NULL;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004153
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004154 flags = flowinfo = scope_id = 0;
4155 if (!PyArg_ParseTuple(args, "Oi:getnameinfo", &sa, &flags))
4156 return NULL;
4157 if (!PyTuple_Check(sa)) {
4158 PyErr_SetString(PyExc_TypeError,
4159 "getnameinfo() argument 1 must be a tuple");
4160 return NULL;
4161 }
4162 if (!PyArg_ParseTuple(sa, "si|ii",
4163 &hostp, &port, &flowinfo, &scope_id))
4164 return NULL;
4165 PyOS_snprintf(pbuf, sizeof(pbuf), "%d", port);
4166 memset(&hints, 0, sizeof(hints));
4167 hints.ai_family = AF_UNSPEC;
4168 hints.ai_socktype = SOCK_DGRAM; /* make numeric port happy */
Martin v. Löwis112c0f32010-08-25 07:38:15 +00004169 hints.ai_flags = AI_NUMERICHOST; /* don't do any name resolution */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004170 Py_BEGIN_ALLOW_THREADS
4171 ACQUIRE_GETADDRINFO_LOCK
4172 error = getaddrinfo(hostp, pbuf, &hints, &res);
4173 Py_END_ALLOW_THREADS
4174 RELEASE_GETADDRINFO_LOCK /* see comment in setipaddr() */
4175 if (error) {
4176 set_gaierror(error);
4177 goto fail;
4178 }
4179 if (res->ai_next) {
4180 PyErr_SetString(socket_error,
4181 "sockaddr resolved to multiple addresses");
4182 goto fail;
4183 }
4184 switch (res->ai_family) {
4185 case AF_INET:
4186 {
4187 if (PyTuple_GET_SIZE(sa) != 2) {
4188 PyErr_SetString(socket_error,
4189 "IPv4 sockaddr must be 2 tuple");
4190 goto fail;
4191 }
4192 break;
4193 }
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00004194#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004195 case AF_INET6:
4196 {
4197 struct sockaddr_in6 *sin6;
4198 sin6 = (struct sockaddr_in6 *)res->ai_addr;
4199 sin6->sin6_flowinfo = flowinfo;
4200 sin6->sin6_scope_id = scope_id;
4201 break;
4202 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004203#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004204 }
Antoine Pitrou19467d22010-08-17 19:33:30 +00004205 error = getnameinfo(res->ai_addr, (socklen_t) res->ai_addrlen,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004206 hbuf, sizeof(hbuf), pbuf, sizeof(pbuf), flags);
4207 if (error) {
4208 set_gaierror(error);
4209 goto fail;
4210 }
4211 ret = Py_BuildValue("ss", hbuf, pbuf);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004212
4213fail:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004214 if (res)
4215 freeaddrinfo(res);
4216 return ret;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004217}
4218
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004219PyDoc_STRVAR(getnameinfo_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00004220"getnameinfo(sockaddr, flags) --> (host, port)\n\
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004221\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004222Get host and port for a sockaddr.");
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004223
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004224
4225/* Python API to getting and setting the default timeout value. */
4226
4227static PyObject *
4228socket_getdefaulttimeout(PyObject *self)
4229{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004230 if (defaulttimeout < 0.0) {
4231 Py_INCREF(Py_None);
4232 return Py_None;
4233 }
4234 else
4235 return PyFloat_FromDouble(defaulttimeout);
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004236}
4237
4238PyDoc_STRVAR(getdefaulttimeout_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00004239"getdefaulttimeout() -> timeout\n\
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004240\n\
4241Returns the default timeout in floating seconds for new socket objects.\n\
4242A value of None indicates that new socket objects have no timeout.\n\
4243When the socket module is first imported, the default is None.");
4244
4245static PyObject *
4246socket_setdefaulttimeout(PyObject *self, PyObject *arg)
4247{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004248 double timeout;
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004249
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004250 if (arg == Py_None)
4251 timeout = -1.0;
4252 else {
4253 timeout = PyFloat_AsDouble(arg);
4254 if (timeout < 0.0) {
4255 if (!PyErr_Occurred())
4256 PyErr_SetString(PyExc_ValueError,
4257 "Timeout value out of range");
4258 return NULL;
4259 }
4260 }
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004261
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004262 defaulttimeout = timeout;
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004263
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004264 Py_INCREF(Py_None);
4265 return Py_None;
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004266}
4267
4268PyDoc_STRVAR(setdefaulttimeout_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00004269"setdefaulttimeout(timeout)\n\
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004270\n\
4271Set the default timeout in floating seconds for new socket objects.\n\
4272A value of None indicates that new socket objects have no timeout.\n\
4273When the socket module is first imported, the default is None.");
4274
4275
Gregory P. Smith5ed2e772011-05-15 00:26:45 -07004276#ifdef HAVE_IF_NAMEINDEX
4277/* Python API for getting interface indices and names */
4278
4279static PyObject *
4280socket_if_nameindex(PyObject *self, PyObject *arg)
4281{
4282 int i = 0;
4283 PyObject *list;
4284 struct if_nameindex *ni = if_nameindex();
4285
4286 if (ni == NULL) {
4287 PyErr_SetString(socket_error, "if_nameindex() returned NULL.");
4288 return NULL;
4289 }
4290
4291 list = PyList_New(0);
4292 if (list == NULL) {
4293 if_freenameindex(ni);
4294 return NULL;
4295 }
4296
4297 while (ni[i].if_index != 0 && i < INT_MAX) {
4298 PyObject *ni_tuple = Py_BuildValue(
4299 "Iy", ni[i].if_index, ni[i].if_name);
4300
4301 if (ni_tuple == NULL || PyList_Append(list, ni_tuple) == -1) {
4302 Py_XDECREF(ni_tuple);
4303 goto error;
4304 }
4305 Py_DECREF(ni_tuple);
4306
4307 ++i;
4308 }
4309
4310 if_freenameindex(ni);
4311 return list;
4312
4313error:
4314 Py_DECREF(list);
4315 if_freenameindex(ni);
4316 return NULL;
4317}
4318
4319PyDoc_STRVAR(if_nameindex_doc,
4320"if_nameindex()\n\
4321\n\
4322Returns a list of network interface information (index, name) tuples.");
4323
4324
4325PyObject*
4326socket_if_nametoindex(PyObject *self, PyObject *arg)
4327{
4328 char* ifname = PyBytes_AsString(arg);
4329 unsigned long index;
4330
4331 if (ifname == NULL)
4332 return NULL;
4333
4334 index = if_nametoindex(ifname);
4335 if (index == 0) {
4336 PyErr_SetString(socket_error, "no interface with this name");
4337 return NULL;
4338 }
4339
4340 return PyLong_FromUnsignedLong(index);
4341}
4342
4343PyDoc_STRVAR(if_nametoindex_doc,
4344"if_nametoindex(if_name)\n\
4345\n\
4346Returns the interface index corresponding to the interface name if_name.");
4347
4348
4349PyObject*
4350socket_if_indextoname(PyObject *self, PyObject *arg)
4351{
4352 unsigned long index = PyLong_AsUnsignedLongMask(arg);
4353 char name[IF_NAMESIZE + 1]; /* or use IFNAMSIZ ?*/
4354 char *ret = if_indextoname(index, &name[0]);
4355
4356 if (ret == NULL) {
4357 PyErr_SetString(socket_error, "no interface with this index");
4358 return NULL;
4359 }
4360
4361 return PyBytes_FromString(name);
4362}
4363
4364PyDoc_STRVAR(if_indextoname_doc,
4365"if_indextoname(if_index)\n\
4366\n\
4367Returns the interface name corresponding to the interface index if_index.");
4368
4369#endif /* HAVE_IF_NAMEINDEX */
4370
4371
Guido van Rossum30a685f1991-06-27 15:51:29 +00004372/* List of functions exported by this module. */
4373
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004374static PyMethodDef socket_methods[] = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004375 {"gethostbyname", socket_gethostbyname,
4376 METH_VARARGS, gethostbyname_doc},
4377 {"gethostbyname_ex", socket_gethostbyname_ex,
4378 METH_VARARGS, ghbn_ex_doc},
4379 {"gethostbyaddr", socket_gethostbyaddr,
4380 METH_VARARGS, gethostbyaddr_doc},
4381 {"gethostname", socket_gethostname,
4382 METH_NOARGS, gethostname_doc},
Antoine Pitrou061cfb52011-02-28 22:25:22 +00004383#ifdef HAVE_SETHOSTNAME
4384 {"sethostname", socket_sethostname,
4385 METH_VARARGS, sethostname_doc},
4386#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004387 {"getservbyname", socket_getservbyname,
4388 METH_VARARGS, getservbyname_doc},
4389 {"getservbyport", socket_getservbyport,
4390 METH_VARARGS, getservbyport_doc},
4391 {"getprotobyname", socket_getprotobyname,
4392 METH_VARARGS, getprotobyname_doc},
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00004393#ifndef NO_DUP
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004394 {"dup", socket_dup,
4395 METH_O, dup_doc},
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00004396#endif
Dave Cole331708b2004-08-09 04:51:41 +00004397#ifdef HAVE_SOCKETPAIR
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004398 {"socketpair", socket_socketpair,
4399 METH_VARARGS, socketpair_doc},
Dave Cole331708b2004-08-09 04:51:41 +00004400#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004401 {"ntohs", socket_ntohs,
4402 METH_VARARGS, ntohs_doc},
4403 {"ntohl", socket_ntohl,
4404 METH_O, ntohl_doc},
4405 {"htons", socket_htons,
4406 METH_VARARGS, htons_doc},
4407 {"htonl", socket_htonl,
4408 METH_O, htonl_doc},
4409 {"inet_aton", socket_inet_aton,
4410 METH_VARARGS, inet_aton_doc},
4411 {"inet_ntoa", socket_inet_ntoa,
4412 METH_VARARGS, inet_ntoa_doc},
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00004413#ifdef HAVE_INET_PTON
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004414 {"inet_pton", socket_inet_pton,
4415 METH_VARARGS, inet_pton_doc},
4416 {"inet_ntop", socket_inet_ntop,
4417 METH_VARARGS, inet_ntop_doc},
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00004418#endif
Giampaolo Rodolàccfb91c2010-08-17 15:30:23 +00004419 {"getaddrinfo", (PyCFunction)socket_getaddrinfo,
4420 METH_VARARGS | METH_KEYWORDS, getaddrinfo_doc},
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004421 {"getnameinfo", socket_getnameinfo,
4422 METH_VARARGS, getnameinfo_doc},
4423 {"getdefaulttimeout", (PyCFunction)socket_getdefaulttimeout,
4424 METH_NOARGS, getdefaulttimeout_doc},
4425 {"setdefaulttimeout", socket_setdefaulttimeout,
4426 METH_O, setdefaulttimeout_doc},
Gregory P. Smith5ed2e772011-05-15 00:26:45 -07004427#ifdef HAVE_IF_NAMEINDEX
4428 {"if_nameindex", socket_if_nameindex,
4429 METH_NOARGS, if_nameindex_doc},
4430 {"if_nametoindex", socket_if_nametoindex,
4431 METH_O, if_nametoindex_doc},
4432 {"if_indextoname", socket_if_indextoname,
4433 METH_O, if_indextoname_doc},
4434#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004435 {NULL, NULL} /* Sentinel */
Guido van Rossum6574b3e1991-06-25 21:36:08 +00004436};
4437
Guido van Rossum30a685f1991-06-27 15:51:29 +00004438
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004439#ifdef MS_WINDOWS
4440#define OS_INIT_DEFINED
4441
4442/* Additional initialization and cleanup for Windows */
Guido van Rossumbe32c891996-06-20 16:25:29 +00004443
4444static void
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004445os_cleanup(void)
Guido van Rossumbe32c891996-06-20 16:25:29 +00004446{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004447 WSACleanup();
Guido van Rossumbe32c891996-06-20 16:25:29 +00004448}
4449
4450static int
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004451os_init(void)
Guido van Rossumbe32c891996-06-20 16:25:29 +00004452{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004453 WSADATA WSAData;
4454 int ret;
4455 ret = WSAStartup(0x0101, &WSAData);
4456 switch (ret) {
4457 case 0: /* No error */
4458 Py_AtExit(os_cleanup);
4459 return 1; /* Success */
4460 case WSASYSNOTREADY:
4461 PyErr_SetString(PyExc_ImportError,
4462 "WSAStartup failed: network not ready");
4463 break;
4464 case WSAVERNOTSUPPORTED:
4465 case WSAEINVAL:
4466 PyErr_SetString(
4467 PyExc_ImportError,
4468 "WSAStartup failed: requested version not supported");
4469 break;
4470 default:
4471 PyErr_Format(PyExc_ImportError, "WSAStartup failed: error code %d", ret);
4472 break;
4473 }
4474 return 0; /* Failure */
Guido van Rossumbe32c891996-06-20 16:25:29 +00004475}
4476
Guido van Rossum8d665e61996-06-26 18:22:49 +00004477#endif /* MS_WINDOWS */
Guido van Rossumbe32c891996-06-20 16:25:29 +00004478
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004479
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004480#ifdef PYOS_OS2
4481#define OS_INIT_DEFINED
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004482
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004483/* Additional initialization for OS/2 */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004484
4485static int
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004486os_init(void)
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004487{
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004488#ifndef PYCC_GCC
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004489 int rc = sock_init();
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004490
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004491 if (rc == 0) {
4492 return 1; /* Success */
4493 }
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004494
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004495 PyErr_Format(PyExc_ImportError, "OS/2 TCP/IP Error# %d", sock_errno());
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004496
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004497 return 0; /* Failure */
Andrew MacIntyreba43e872002-03-03 03:03:52 +00004498#else
Ezio Melotti13925002011-03-16 11:05:33 +02004499 /* No need to initialize sockets with GCC/EMX */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004500 return 1; /* Success */
Andrew MacIntyreba43e872002-03-03 03:03:52 +00004501#endif
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004502}
4503
4504#endif /* PYOS_OS2 */
4505
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004506
4507#ifndef OS_INIT_DEFINED
4508static int
4509os_init(void)
4510{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004511 return 1; /* Success */
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004512}
4513#endif
4514
4515
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00004516/* C API table - always add new things to the end for binary
4517 compatibility. */
4518static
4519PySocketModule_APIObject PySocketModuleAPI =
4520{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004521 &sock_type,
Antoine Pitrouc4df7842010-12-03 19:59:41 +00004522 NULL,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004523 NULL
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00004524};
4525
Guido van Rossum3eede5a2002-06-07 02:08:35 +00004526
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004527/* Initialize the _socket module.
Guido van Rossum3eede5a2002-06-07 02:08:35 +00004528
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004529 This module is actually called "_socket", and there's a wrapper
Guido van Rossum7d0a8262007-05-21 23:13:11 +00004530 "socket.py" which implements some additional functionality.
4531 The import of "_socket" may fail with an ImportError exception if
4532 os-specific initialization fails. On Windows, this does WINSOCK
Ezio Melotti13925002011-03-16 11:05:33 +02004533 initialization. When WINSOCK is initialized successfully, a call to
Guido van Rossum7d0a8262007-05-21 23:13:11 +00004534 WSACleanup() is scheduled to be made at exit time.
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004535*/
Guido van Rossum3eede5a2002-06-07 02:08:35 +00004536
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004537PyDoc_STRVAR(socket_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00004538"Implementation module for socket operations.\n\
4539\n\
4540See the socket module for documentation.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00004541
Martin v. Löwis1a214512008-06-11 05:26:20 +00004542static struct PyModuleDef socketmodule = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004543 PyModuleDef_HEAD_INIT,
4544 PySocket_MODULE_NAME,
4545 socket_doc,
4546 -1,
4547 socket_methods,
4548 NULL,
4549 NULL,
4550 NULL,
4551 NULL
Martin v. Löwis1a214512008-06-11 05:26:20 +00004552};
4553
Mark Hammondfe51c6d2002-08-02 02:27:13 +00004554PyMODINIT_FUNC
Martin v. Löwis1a214512008-06-11 05:26:20 +00004555PyInit__socket(void)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00004556{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004557 PyObject *m, *has_ipv6;
Fred Drake4baedc12002-04-01 14:53:37 +00004558
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004559 if (!os_init())
4560 return NULL;
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004561
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004562 Py_TYPE(&sock_type) = &PyType_Type;
4563 m = PyModule_Create(&socketmodule);
4564 if (m == NULL)
4565 return NULL;
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004566
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004567 socket_error = PyErr_NewException("socket.error",
4568 PyExc_IOError, NULL);
4569 if (socket_error == NULL)
4570 return NULL;
4571 PySocketModuleAPI.error = socket_error;
4572 Py_INCREF(socket_error);
4573 PyModule_AddObject(m, "error", socket_error);
4574 socket_herror = PyErr_NewException("socket.herror",
4575 socket_error, NULL);
4576 if (socket_herror == NULL)
4577 return NULL;
4578 Py_INCREF(socket_herror);
4579 PyModule_AddObject(m, "herror", socket_herror);
4580 socket_gaierror = PyErr_NewException("socket.gaierror", socket_error,
4581 NULL);
4582 if (socket_gaierror == NULL)
4583 return NULL;
4584 Py_INCREF(socket_gaierror);
4585 PyModule_AddObject(m, "gaierror", socket_gaierror);
4586 socket_timeout = PyErr_NewException("socket.timeout",
4587 socket_error, NULL);
4588 if (socket_timeout == NULL)
4589 return NULL;
Antoine Pitrouc4df7842010-12-03 19:59:41 +00004590 PySocketModuleAPI.timeout_error = socket_timeout;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004591 Py_INCREF(socket_timeout);
4592 PyModule_AddObject(m, "timeout", socket_timeout);
4593 Py_INCREF((PyObject *)&sock_type);
4594 if (PyModule_AddObject(m, "SocketType",
4595 (PyObject *)&sock_type) != 0)
4596 return NULL;
4597 Py_INCREF((PyObject *)&sock_type);
4598 if (PyModule_AddObject(m, "socket",
4599 (PyObject *)&sock_type) != 0)
4600 return NULL;
Guido van Rossum09be4091999-08-09 14:40:40 +00004601
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00004602#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004603 has_ipv6 = Py_True;
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00004604#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004605 has_ipv6 = Py_False;
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00004606#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004607 Py_INCREF(has_ipv6);
4608 PyModule_AddObject(m, "has_ipv6", has_ipv6);
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00004609
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004610 /* Export C API */
4611 if (PyModule_AddObject(m, PySocket_CAPI_NAME,
4612 PyCapsule_New(&PySocketModuleAPI, PySocket_CAPSULE_NAME, NULL)
4613 ) != 0)
4614 return NULL;
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00004615
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004616 /* Address families (we only support AF_INET and AF_UNIX) */
Guido van Rossum09be4091999-08-09 14:40:40 +00004617#ifdef AF_UNSPEC
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004618 PyModule_AddIntConstant(m, "AF_UNSPEC", AF_UNSPEC);
Guido van Rossum09be4091999-08-09 14:40:40 +00004619#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004620 PyModule_AddIntConstant(m, "AF_INET", AF_INET);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004621#ifdef AF_INET6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004622 PyModule_AddIntConstant(m, "AF_INET6", AF_INET6);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004623#endif /* AF_INET6 */
Andrew MacIntyred12dfbb2004-04-04 07:13:49 +00004624#if defined(AF_UNIX)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004625 PyModule_AddIntConstant(m, "AF_UNIX", AF_UNIX);
Guido van Rossumb6775db1994-08-01 11:34:53 +00004626#endif /* AF_UNIX */
Guido van Rossum09be4091999-08-09 14:40:40 +00004627#ifdef AF_AX25
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004628 /* Amateur Radio AX.25 */
4629 PyModule_AddIntConstant(m, "AF_AX25", AF_AX25);
Guido van Rossum09be4091999-08-09 14:40:40 +00004630#endif
4631#ifdef AF_IPX
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004632 PyModule_AddIntConstant(m, "AF_IPX", AF_IPX); /* Novell IPX */
Guido van Rossum09be4091999-08-09 14:40:40 +00004633#endif
4634#ifdef AF_APPLETALK
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004635 /* Appletalk DDP */
4636 PyModule_AddIntConstant(m, "AF_APPLETALK", AF_APPLETALK);
Guido van Rossum09be4091999-08-09 14:40:40 +00004637#endif
4638#ifdef AF_NETROM
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004639 /* Amateur radio NetROM */
4640 PyModule_AddIntConstant(m, "AF_NETROM", AF_NETROM);
Guido van Rossum09be4091999-08-09 14:40:40 +00004641#endif
4642#ifdef AF_BRIDGE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004643 /* Multiprotocol bridge */
4644 PyModule_AddIntConstant(m, "AF_BRIDGE", AF_BRIDGE);
Guido van Rossum09be4091999-08-09 14:40:40 +00004645#endif
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004646#ifdef AF_ATMPVC
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004647 /* ATM PVCs */
4648 PyModule_AddIntConstant(m, "AF_ATMPVC", AF_ATMPVC);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004649#endif
Guido van Rossum09be4091999-08-09 14:40:40 +00004650#ifdef AF_AAL5
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004651 /* Reserved for Werner's ATM */
4652 PyModule_AddIntConstant(m, "AF_AAL5", AF_AAL5);
Guido van Rossum09be4091999-08-09 14:40:40 +00004653#endif
4654#ifdef AF_X25
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004655 /* Reserved for X.25 project */
4656 PyModule_AddIntConstant(m, "AF_X25", AF_X25);
Guido van Rossum09be4091999-08-09 14:40:40 +00004657#endif
4658#ifdef AF_INET6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004659 PyModule_AddIntConstant(m, "AF_INET6", AF_INET6); /* IP version 6 */
Guido van Rossum09be4091999-08-09 14:40:40 +00004660#endif
4661#ifdef AF_ROSE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004662 /* Amateur Radio X.25 PLP */
4663 PyModule_AddIntConstant(m, "AF_ROSE", AF_ROSE);
Guido van Rossum09be4091999-08-09 14:40:40 +00004664#endif
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004665#ifdef AF_DECnet
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004666 /* Reserved for DECnet project */
4667 PyModule_AddIntConstant(m, "AF_DECnet", AF_DECnet);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004668#endif
4669#ifdef AF_NETBEUI
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004670 /* Reserved for 802.2LLC project */
4671 PyModule_AddIntConstant(m, "AF_NETBEUI", AF_NETBEUI);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004672#endif
4673#ifdef AF_SECURITY
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004674 /* Security callback pseudo AF */
4675 PyModule_AddIntConstant(m, "AF_SECURITY", AF_SECURITY);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004676#endif
4677#ifdef AF_KEY
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004678 /* PF_KEY key management API */
4679 PyModule_AddIntConstant(m, "AF_KEY", AF_KEY);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004680#endif
4681#ifdef AF_NETLINK
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004682 /* */
4683 PyModule_AddIntConstant(m, "AF_NETLINK", AF_NETLINK);
4684 PyModule_AddIntConstant(m, "NETLINK_ROUTE", NETLINK_ROUTE);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004685#ifdef NETLINK_SKIP
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004686 PyModule_AddIntConstant(m, "NETLINK_SKIP", NETLINK_SKIP);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004687#endif
4688#ifdef NETLINK_W1
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004689 PyModule_AddIntConstant(m, "NETLINK_W1", NETLINK_W1);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004690#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004691 PyModule_AddIntConstant(m, "NETLINK_USERSOCK", NETLINK_USERSOCK);
4692 PyModule_AddIntConstant(m, "NETLINK_FIREWALL", NETLINK_FIREWALL);
Guido van Rossum668a94a2006-02-21 01:07:27 +00004693#ifdef NETLINK_TCPDIAG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004694 PyModule_AddIntConstant(m, "NETLINK_TCPDIAG", NETLINK_TCPDIAG);
Guido van Rossum668a94a2006-02-21 01:07:27 +00004695#endif
4696#ifdef NETLINK_NFLOG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004697 PyModule_AddIntConstant(m, "NETLINK_NFLOG", NETLINK_NFLOG);
Guido van Rossum668a94a2006-02-21 01:07:27 +00004698#endif
Neal Norwitz65851662006-01-16 04:31:40 +00004699#ifdef NETLINK_XFRM
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004700 PyModule_AddIntConstant(m, "NETLINK_XFRM", NETLINK_XFRM);
Neal Norwitz65851662006-01-16 04:31:40 +00004701#endif
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004702#ifdef NETLINK_ARPD
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004703 PyModule_AddIntConstant(m, "NETLINK_ARPD", NETLINK_ARPD);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004704#endif
4705#ifdef NETLINK_ROUTE6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004706 PyModule_AddIntConstant(m, "NETLINK_ROUTE6", NETLINK_ROUTE6);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004707#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004708 PyModule_AddIntConstant(m, "NETLINK_IP6_FW", NETLINK_IP6_FW);
Thomas Wouterscf297e42007-02-23 15:07:44 +00004709#ifdef NETLINK_DNRTMSG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004710 PyModule_AddIntConstant(m, "NETLINK_DNRTMSG", NETLINK_DNRTMSG);
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00004711#endif
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004712#ifdef NETLINK_TAPBASE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004713 PyModule_AddIntConstant(m, "NETLINK_TAPBASE", NETLINK_TAPBASE);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004714#endif
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004715#endif /* AF_NETLINK */
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004716#ifdef AF_ROUTE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004717 /* Alias to emulate 4.4BSD */
4718 PyModule_AddIntConstant(m, "AF_ROUTE", AF_ROUTE);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004719#endif
4720#ifdef AF_ASH
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004721 /* Ash */
4722 PyModule_AddIntConstant(m, "AF_ASH", AF_ASH);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004723#endif
4724#ifdef AF_ECONET
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004725 /* Acorn Econet */
4726 PyModule_AddIntConstant(m, "AF_ECONET", AF_ECONET);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004727#endif
4728#ifdef AF_ATMSVC
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004729 /* ATM SVCs */
4730 PyModule_AddIntConstant(m, "AF_ATMSVC", AF_ATMSVC);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004731#endif
4732#ifdef AF_SNA
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004733 /* Linux SNA Project (nutters!) */
4734 PyModule_AddIntConstant(m, "AF_SNA", AF_SNA);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004735#endif
4736#ifdef AF_IRDA
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004737 /* IRDA sockets */
4738 PyModule_AddIntConstant(m, "AF_IRDA", AF_IRDA);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004739#endif
4740#ifdef AF_PPPOX
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004741 /* PPPoX sockets */
4742 PyModule_AddIntConstant(m, "AF_PPPOX", AF_PPPOX);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004743#endif
4744#ifdef AF_WANPIPE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004745 /* Wanpipe API Sockets */
4746 PyModule_AddIntConstant(m, "AF_WANPIPE", AF_WANPIPE);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004747#endif
4748#ifdef AF_LLC
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004749 /* Linux LLC */
4750 PyModule_AddIntConstant(m, "AF_LLC", AF_LLC);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004751#endif
Martin v. Löwis12af0482004-01-31 12:34:17 +00004752
Hye-Shik Chang81268602004-02-02 06:05:24 +00004753#ifdef USE_BLUETOOTH
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004754 PyModule_AddIntConstant(m, "AF_BLUETOOTH", AF_BLUETOOTH);
4755 PyModule_AddIntConstant(m, "BTPROTO_L2CAP", BTPROTO_L2CAP);
4756 PyModule_AddIntConstant(m, "BTPROTO_HCI", BTPROTO_HCI);
4757 PyModule_AddIntConstant(m, "SOL_HCI", SOL_HCI);
Gregory P. Smith3e85dfd2010-10-17 00:43:10 +00004758#if !defined(__NetBSD__) && !defined(__DragonFly__)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004759 PyModule_AddIntConstant(m, "HCI_FILTER", HCI_FILTER);
Gregory P. Smith3e85dfd2010-10-17 00:43:10 +00004760#endif
Hye-Shik Chang81268602004-02-02 06:05:24 +00004761#if !defined(__FreeBSD__)
Gregory P. Smith3e85dfd2010-10-17 00:43:10 +00004762#if !defined(__NetBSD__) && !defined(__DragonFly__)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004763 PyModule_AddIntConstant(m, "HCI_TIME_STAMP", HCI_TIME_STAMP);
Gregory P. Smith3e85dfd2010-10-17 00:43:10 +00004764#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004765 PyModule_AddIntConstant(m, "HCI_DATA_DIR", HCI_DATA_DIR);
4766 PyModule_AddIntConstant(m, "BTPROTO_SCO", BTPROTO_SCO);
Hye-Shik Chang81268602004-02-02 06:05:24 +00004767#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004768 PyModule_AddIntConstant(m, "BTPROTO_RFCOMM", BTPROTO_RFCOMM);
4769 PyModule_AddStringConstant(m, "BDADDR_ANY", "00:00:00:00:00:00");
4770 PyModule_AddStringConstant(m, "BDADDR_LOCAL", "00:00:00:FF:FF:FF");
Martin v. Löwis12af0482004-01-31 12:34:17 +00004771#endif
4772
Antoine Pitroub156a462010-10-27 20:13:57 +00004773#ifdef AF_PACKET
4774 PyModule_AddIntMacro(m, AF_PACKET);
4775#endif
4776#ifdef PF_PACKET
4777 PyModule_AddIntMacro(m, PF_PACKET);
4778#endif
4779#ifdef PACKET_HOST
4780 PyModule_AddIntMacro(m, PACKET_HOST);
4781#endif
4782#ifdef PACKET_BROADCAST
4783 PyModule_AddIntMacro(m, PACKET_BROADCAST);
4784#endif
4785#ifdef PACKET_MULTICAST
4786 PyModule_AddIntMacro(m, PACKET_MULTICAST);
4787#endif
4788#ifdef PACKET_OTHERHOST
4789 PyModule_AddIntMacro(m, PACKET_OTHERHOST);
4790#endif
4791#ifdef PACKET_OUTGOING
4792 PyModule_AddIntMacro(m, PACKET_OUTGOING);
4793#endif
4794#ifdef PACKET_LOOPBACK
4795 PyModule_AddIntMacro(m, PACKET_LOOPBACK);
4796#endif
4797#ifdef PACKET_FASTROUTE
4798 PyModule_AddIntMacro(m, PACKET_FASTROUTE);
Guido van Rossum48a680c2001-03-02 06:34:14 +00004799#endif
Guido van Rossum09be4091999-08-09 14:40:40 +00004800
Christian Heimes043d6f62008-01-07 17:19:16 +00004801#ifdef HAVE_LINUX_TIPC_H
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004802 PyModule_AddIntConstant(m, "AF_TIPC", AF_TIPC);
Christian Heimes043d6f62008-01-07 17:19:16 +00004803
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004804 /* for addresses */
4805 PyModule_AddIntConstant(m, "TIPC_ADDR_NAMESEQ", TIPC_ADDR_NAMESEQ);
4806 PyModule_AddIntConstant(m, "TIPC_ADDR_NAME", TIPC_ADDR_NAME);
4807 PyModule_AddIntConstant(m, "TIPC_ADDR_ID", TIPC_ADDR_ID);
Christian Heimes043d6f62008-01-07 17:19:16 +00004808
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004809 PyModule_AddIntConstant(m, "TIPC_ZONE_SCOPE", TIPC_ZONE_SCOPE);
4810 PyModule_AddIntConstant(m, "TIPC_CLUSTER_SCOPE", TIPC_CLUSTER_SCOPE);
4811 PyModule_AddIntConstant(m, "TIPC_NODE_SCOPE", TIPC_NODE_SCOPE);
Christian Heimes043d6f62008-01-07 17:19:16 +00004812
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004813 /* for setsockopt() */
4814 PyModule_AddIntConstant(m, "SOL_TIPC", SOL_TIPC);
4815 PyModule_AddIntConstant(m, "TIPC_IMPORTANCE", TIPC_IMPORTANCE);
4816 PyModule_AddIntConstant(m, "TIPC_SRC_DROPPABLE", TIPC_SRC_DROPPABLE);
4817 PyModule_AddIntConstant(m, "TIPC_DEST_DROPPABLE",
4818 TIPC_DEST_DROPPABLE);
4819 PyModule_AddIntConstant(m, "TIPC_CONN_TIMEOUT", TIPC_CONN_TIMEOUT);
Christian Heimes043d6f62008-01-07 17:19:16 +00004820
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004821 PyModule_AddIntConstant(m, "TIPC_LOW_IMPORTANCE",
4822 TIPC_LOW_IMPORTANCE);
4823 PyModule_AddIntConstant(m, "TIPC_MEDIUM_IMPORTANCE",
4824 TIPC_MEDIUM_IMPORTANCE);
4825 PyModule_AddIntConstant(m, "TIPC_HIGH_IMPORTANCE",
4826 TIPC_HIGH_IMPORTANCE);
4827 PyModule_AddIntConstant(m, "TIPC_CRITICAL_IMPORTANCE",
4828 TIPC_CRITICAL_IMPORTANCE);
Christian Heimes043d6f62008-01-07 17:19:16 +00004829
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004830 /* for subscriptions */
4831 PyModule_AddIntConstant(m, "TIPC_SUB_PORTS", TIPC_SUB_PORTS);
4832 PyModule_AddIntConstant(m, "TIPC_SUB_SERVICE", TIPC_SUB_SERVICE);
Christian Heimes25bb7832008-01-11 16:17:00 +00004833#ifdef TIPC_SUB_CANCEL
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004834 /* doesn't seem to be available everywhere */
4835 PyModule_AddIntConstant(m, "TIPC_SUB_CANCEL", TIPC_SUB_CANCEL);
Christian Heimes25bb7832008-01-11 16:17:00 +00004836#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004837 PyModule_AddIntConstant(m, "TIPC_WAIT_FOREVER", TIPC_WAIT_FOREVER);
4838 PyModule_AddIntConstant(m, "TIPC_PUBLISHED", TIPC_PUBLISHED);
4839 PyModule_AddIntConstant(m, "TIPC_WITHDRAWN", TIPC_WITHDRAWN);
4840 PyModule_AddIntConstant(m, "TIPC_SUBSCR_TIMEOUT", TIPC_SUBSCR_TIMEOUT);
4841 PyModule_AddIntConstant(m, "TIPC_CFG_SRV", TIPC_CFG_SRV);
4842 PyModule_AddIntConstant(m, "TIPC_TOP_SRV", TIPC_TOP_SRV);
Christian Heimes043d6f62008-01-07 17:19:16 +00004843#endif
4844
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004845 /* Socket types */
4846 PyModule_AddIntConstant(m, "SOCK_STREAM", SOCK_STREAM);
4847 PyModule_AddIntConstant(m, "SOCK_DGRAM", SOCK_DGRAM);
Guido van Rossumbcc20741998-08-04 22:53:56 +00004848/* We have incomplete socket support. */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004849 PyModule_AddIntConstant(m, "SOCK_RAW", SOCK_RAW);
4850 PyModule_AddIntConstant(m, "SOCK_SEQPACKET", SOCK_SEQPACKET);
Martin v. Löwiscf8f47e2002-12-11 13:10:57 +00004851#if defined(SOCK_RDM)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004852 PyModule_AddIntConstant(m, "SOCK_RDM", SOCK_RDM);
Guido van Rossumbcc20741998-08-04 22:53:56 +00004853#endif
Antoine Pitroub1c54962010-10-14 15:05:38 +00004854#ifdef SOCK_CLOEXEC
4855 PyModule_AddIntConstant(m, "SOCK_CLOEXEC", SOCK_CLOEXEC);
4856#endif
4857#ifdef SOCK_NONBLOCK
4858 PyModule_AddIntConstant(m, "SOCK_NONBLOCK", SOCK_NONBLOCK);
4859#endif
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004860
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004861#ifdef SO_DEBUG
4862 PyModule_AddIntConstant(m, "SO_DEBUG", SO_DEBUG);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004863#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004864#ifdef SO_ACCEPTCONN
4865 PyModule_AddIntConstant(m, "SO_ACCEPTCONN", SO_ACCEPTCONN);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004866#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004867#ifdef SO_REUSEADDR
4868 PyModule_AddIntConstant(m, "SO_REUSEADDR", SO_REUSEADDR);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004869#endif
Andrew M. Kuchling42851ab2004-07-10 14:19:21 +00004870#ifdef SO_EXCLUSIVEADDRUSE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004871 PyModule_AddIntConstant(m, "SO_EXCLUSIVEADDRUSE", SO_EXCLUSIVEADDRUSE);
Andrew M. Kuchling42851ab2004-07-10 14:19:21 +00004872#endif
4873
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004874#ifdef SO_KEEPALIVE
4875 PyModule_AddIntConstant(m, "SO_KEEPALIVE", SO_KEEPALIVE);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004876#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004877#ifdef SO_DONTROUTE
4878 PyModule_AddIntConstant(m, "SO_DONTROUTE", SO_DONTROUTE);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004879#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004880#ifdef SO_BROADCAST
4881 PyModule_AddIntConstant(m, "SO_BROADCAST", SO_BROADCAST);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004882#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004883#ifdef SO_USELOOPBACK
4884 PyModule_AddIntConstant(m, "SO_USELOOPBACK", SO_USELOOPBACK);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004885#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004886#ifdef SO_LINGER
4887 PyModule_AddIntConstant(m, "SO_LINGER", SO_LINGER);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004888#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004889#ifdef SO_OOBINLINE
4890 PyModule_AddIntConstant(m, "SO_OOBINLINE", SO_OOBINLINE);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004891#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004892#ifdef SO_REUSEPORT
4893 PyModule_AddIntConstant(m, "SO_REUSEPORT", SO_REUSEPORT);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004894#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004895#ifdef SO_SNDBUF
4896 PyModule_AddIntConstant(m, "SO_SNDBUF", SO_SNDBUF);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004897#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004898#ifdef SO_RCVBUF
4899 PyModule_AddIntConstant(m, "SO_RCVBUF", SO_RCVBUF);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004900#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004901#ifdef SO_SNDLOWAT
4902 PyModule_AddIntConstant(m, "SO_SNDLOWAT", SO_SNDLOWAT);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004903#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004904#ifdef SO_RCVLOWAT
4905 PyModule_AddIntConstant(m, "SO_RCVLOWAT", SO_RCVLOWAT);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004906#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004907#ifdef SO_SNDTIMEO
4908 PyModule_AddIntConstant(m, "SO_SNDTIMEO", SO_SNDTIMEO);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004909#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004910#ifdef SO_RCVTIMEO
4911 PyModule_AddIntConstant(m, "SO_RCVTIMEO", SO_RCVTIMEO);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004912#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004913#ifdef SO_ERROR
4914 PyModule_AddIntConstant(m, "SO_ERROR", SO_ERROR);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004915#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004916#ifdef SO_TYPE
4917 PyModule_AddIntConstant(m, "SO_TYPE", SO_TYPE);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004918#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004919#ifdef SO_SETFIB
4920 PyModule_AddIntConstant(m, "SO_SETFIB", SO_SETFIB);
Larry Hastingsf0f37952010-04-02 11:47:10 +00004921#endif
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004922
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004923 /* Maximum number of connections for "listen" */
4924#ifdef SOMAXCONN
4925 PyModule_AddIntConstant(m, "SOMAXCONN", SOMAXCONN);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004926#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004927 PyModule_AddIntConstant(m, "SOMAXCONN", 5); /* Common value */
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004928#endif
4929
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004930 /* Flags for send, recv */
4931#ifdef MSG_OOB
4932 PyModule_AddIntConstant(m, "MSG_OOB", MSG_OOB);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004933#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004934#ifdef MSG_PEEK
4935 PyModule_AddIntConstant(m, "MSG_PEEK", MSG_PEEK);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004936#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004937#ifdef MSG_DONTROUTE
4938 PyModule_AddIntConstant(m, "MSG_DONTROUTE", MSG_DONTROUTE);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004939#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004940#ifdef MSG_DONTWAIT
4941 PyModule_AddIntConstant(m, "MSG_DONTWAIT", MSG_DONTWAIT);
Guido van Rossum2c8bcb82000-04-25 21:34:53 +00004942#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004943#ifdef MSG_EOR
4944 PyModule_AddIntConstant(m, "MSG_EOR", MSG_EOR);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004945#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004946#ifdef MSG_TRUNC
4947 PyModule_AddIntConstant(m, "MSG_TRUNC", MSG_TRUNC);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004948#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004949#ifdef MSG_CTRUNC
4950 PyModule_AddIntConstant(m, "MSG_CTRUNC", MSG_CTRUNC);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004951#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004952#ifdef MSG_WAITALL
4953 PyModule_AddIntConstant(m, "MSG_WAITALL", MSG_WAITALL);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004954#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004955#ifdef MSG_BTAG
4956 PyModule_AddIntConstant(m, "MSG_BTAG", MSG_BTAG);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004957#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004958#ifdef MSG_ETAG
4959 PyModule_AddIntConstant(m, "MSG_ETAG", MSG_ETAG);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004960#endif
4961
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004962 /* Protocol level and numbers, usable for [gs]etsockopt */
4963#ifdef SOL_SOCKET
4964 PyModule_AddIntConstant(m, "SOL_SOCKET", SOL_SOCKET);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004965#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004966#ifdef SOL_IP
4967 PyModule_AddIntConstant(m, "SOL_IP", SOL_IP);
Guido van Rossum09be4091999-08-09 14:40:40 +00004968#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004969 PyModule_AddIntConstant(m, "SOL_IP", 0);
Guido van Rossum09be4091999-08-09 14:40:40 +00004970#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004971#ifdef SOL_IPX
4972 PyModule_AddIntConstant(m, "SOL_IPX", SOL_IPX);
Guido van Rossum09be4091999-08-09 14:40:40 +00004973#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004974#ifdef SOL_AX25
4975 PyModule_AddIntConstant(m, "SOL_AX25", SOL_AX25);
Guido van Rossum09be4091999-08-09 14:40:40 +00004976#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004977#ifdef SOL_ATALK
4978 PyModule_AddIntConstant(m, "SOL_ATALK", SOL_ATALK);
Guido van Rossum09be4091999-08-09 14:40:40 +00004979#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004980#ifdef SOL_NETROM
4981 PyModule_AddIntConstant(m, "SOL_NETROM", SOL_NETROM);
Guido van Rossum09be4091999-08-09 14:40:40 +00004982#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004983#ifdef SOL_ROSE
4984 PyModule_AddIntConstant(m, "SOL_ROSE", SOL_ROSE);
Guido van Rossum09be4091999-08-09 14:40:40 +00004985#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004986#ifdef SOL_TCP
4987 PyModule_AddIntConstant(m, "SOL_TCP", SOL_TCP);
Guido van Rossum09be4091999-08-09 14:40:40 +00004988#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004989 PyModule_AddIntConstant(m, "SOL_TCP", 6);
Guido van Rossum09be4091999-08-09 14:40:40 +00004990#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004991#ifdef SOL_UDP
4992 PyModule_AddIntConstant(m, "SOL_UDP", SOL_UDP);
Guido van Rossum09be4091999-08-09 14:40:40 +00004993#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004994 PyModule_AddIntConstant(m, "SOL_UDP", 17);
Guido van Rossum09be4091999-08-09 14:40:40 +00004995#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004996#ifdef IPPROTO_IP
4997 PyModule_AddIntConstant(m, "IPPROTO_IP", IPPROTO_IP);
Guido van Rossum578de301998-05-28 20:18:18 +00004998#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004999 PyModule_AddIntConstant(m, "IPPROTO_IP", 0);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00005000#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005001#ifdef IPPROTO_HOPOPTS
5002 PyModule_AddIntConstant(m, "IPPROTO_HOPOPTS", IPPROTO_HOPOPTS);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00005003#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005004#ifdef IPPROTO_ICMP
5005 PyModule_AddIntConstant(m, "IPPROTO_ICMP", IPPROTO_ICMP);
Guido van Rossum578de301998-05-28 20:18:18 +00005006#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005007 PyModule_AddIntConstant(m, "IPPROTO_ICMP", 1);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00005008#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005009#ifdef IPPROTO_IGMP
5010 PyModule_AddIntConstant(m, "IPPROTO_IGMP", IPPROTO_IGMP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00005011#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005012#ifdef IPPROTO_GGP
5013 PyModule_AddIntConstant(m, "IPPROTO_GGP", IPPROTO_GGP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00005014#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005015#ifdef IPPROTO_IPV4
5016 PyModule_AddIntConstant(m, "IPPROTO_IPV4", IPPROTO_IPV4);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00005017#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005018#ifdef IPPROTO_IPV6
5019 PyModule_AddIntConstant(m, "IPPROTO_IPV6", IPPROTO_IPV6);
Martin v. Löwisa0f17342003-10-03 13:56:20 +00005020#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005021#ifdef IPPROTO_IPIP
5022 PyModule_AddIntConstant(m, "IPPROTO_IPIP", IPPROTO_IPIP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00005023#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005024#ifdef IPPROTO_TCP
5025 PyModule_AddIntConstant(m, "IPPROTO_TCP", IPPROTO_TCP);
Guido van Rossum578de301998-05-28 20:18:18 +00005026#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005027 PyModule_AddIntConstant(m, "IPPROTO_TCP", 6);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00005028#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005029#ifdef IPPROTO_EGP
5030 PyModule_AddIntConstant(m, "IPPROTO_EGP", IPPROTO_EGP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00005031#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005032#ifdef IPPROTO_PUP
5033 PyModule_AddIntConstant(m, "IPPROTO_PUP", IPPROTO_PUP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00005034#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005035#ifdef IPPROTO_UDP
5036 PyModule_AddIntConstant(m, "IPPROTO_UDP", IPPROTO_UDP);
Guido van Rossum578de301998-05-28 20:18:18 +00005037#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005038 PyModule_AddIntConstant(m, "IPPROTO_UDP", 17);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00005039#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005040#ifdef IPPROTO_IDP
5041 PyModule_AddIntConstant(m, "IPPROTO_IDP", IPPROTO_IDP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00005042#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005043#ifdef IPPROTO_HELLO
5044 PyModule_AddIntConstant(m, "IPPROTO_HELLO", IPPROTO_HELLO);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00005045#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005046#ifdef IPPROTO_ND
5047 PyModule_AddIntConstant(m, "IPPROTO_ND", IPPROTO_ND);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00005048#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005049#ifdef IPPROTO_TP
5050 PyModule_AddIntConstant(m, "IPPROTO_TP", IPPROTO_TP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00005051#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005052#ifdef IPPROTO_IPV6
5053 PyModule_AddIntConstant(m, "IPPROTO_IPV6", IPPROTO_IPV6);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00005054#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005055#ifdef IPPROTO_ROUTING
5056 PyModule_AddIntConstant(m, "IPPROTO_ROUTING", IPPROTO_ROUTING);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00005057#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005058#ifdef IPPROTO_FRAGMENT
5059 PyModule_AddIntConstant(m, "IPPROTO_FRAGMENT", IPPROTO_FRAGMENT);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00005060#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005061#ifdef IPPROTO_RSVP
5062 PyModule_AddIntConstant(m, "IPPROTO_RSVP", IPPROTO_RSVP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00005063#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005064#ifdef IPPROTO_GRE
5065 PyModule_AddIntConstant(m, "IPPROTO_GRE", IPPROTO_GRE);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00005066#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005067#ifdef IPPROTO_ESP
5068 PyModule_AddIntConstant(m, "IPPROTO_ESP", IPPROTO_ESP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00005069#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005070#ifdef IPPROTO_AH
5071 PyModule_AddIntConstant(m, "IPPROTO_AH", IPPROTO_AH);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00005072#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005073#ifdef IPPROTO_MOBILE
5074 PyModule_AddIntConstant(m, "IPPROTO_MOBILE", IPPROTO_MOBILE);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00005075#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005076#ifdef IPPROTO_ICMPV6
5077 PyModule_AddIntConstant(m, "IPPROTO_ICMPV6", IPPROTO_ICMPV6);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00005078#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005079#ifdef IPPROTO_NONE
5080 PyModule_AddIntConstant(m, "IPPROTO_NONE", IPPROTO_NONE);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00005081#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005082#ifdef IPPROTO_DSTOPTS
5083 PyModule_AddIntConstant(m, "IPPROTO_DSTOPTS", IPPROTO_DSTOPTS);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00005084#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005085#ifdef IPPROTO_XTP
5086 PyModule_AddIntConstant(m, "IPPROTO_XTP", IPPROTO_XTP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00005087#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005088#ifdef IPPROTO_EON
5089 PyModule_AddIntConstant(m, "IPPROTO_EON", IPPROTO_EON);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00005090#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005091#ifdef IPPROTO_PIM
5092 PyModule_AddIntConstant(m, "IPPROTO_PIM", IPPROTO_PIM);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00005093#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005094#ifdef IPPROTO_IPCOMP
5095 PyModule_AddIntConstant(m, "IPPROTO_IPCOMP", IPPROTO_IPCOMP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00005096#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005097#ifdef IPPROTO_VRRP
5098 PyModule_AddIntConstant(m, "IPPROTO_VRRP", IPPROTO_VRRP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00005099#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005100#ifdef IPPROTO_BIP
5101 PyModule_AddIntConstant(m, "IPPROTO_BIP", IPPROTO_BIP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00005102#endif
5103/**/
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005104#ifdef IPPROTO_RAW
5105 PyModule_AddIntConstant(m, "IPPROTO_RAW", IPPROTO_RAW);
Guido van Rossum578de301998-05-28 20:18:18 +00005106#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005107 PyModule_AddIntConstant(m, "IPPROTO_RAW", 255);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00005108#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005109#ifdef IPPROTO_MAX
5110 PyModule_AddIntConstant(m, "IPPROTO_MAX", IPPROTO_MAX);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00005111#endif
5112
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005113 /* Some port configuration */
5114#ifdef IPPORT_RESERVED
5115 PyModule_AddIntConstant(m, "IPPORT_RESERVED", IPPORT_RESERVED);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00005116#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005117 PyModule_AddIntConstant(m, "IPPORT_RESERVED", 1024);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00005118#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005119#ifdef IPPORT_USERRESERVED
5120 PyModule_AddIntConstant(m, "IPPORT_USERRESERVED", IPPORT_USERRESERVED);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00005121#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005122 PyModule_AddIntConstant(m, "IPPORT_USERRESERVED", 5000);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00005123#endif
5124
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005125 /* Some reserved IP v.4 addresses */
5126#ifdef INADDR_ANY
5127 PyModule_AddIntConstant(m, "INADDR_ANY", INADDR_ANY);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00005128#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005129 PyModule_AddIntConstant(m, "INADDR_ANY", 0x00000000);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00005130#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005131#ifdef INADDR_BROADCAST
5132 PyModule_AddIntConstant(m, "INADDR_BROADCAST", INADDR_BROADCAST);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00005133#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005134 PyModule_AddIntConstant(m, "INADDR_BROADCAST", 0xffffffff);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00005135#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005136#ifdef INADDR_LOOPBACK
5137 PyModule_AddIntConstant(m, "INADDR_LOOPBACK", INADDR_LOOPBACK);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00005138#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005139 PyModule_AddIntConstant(m, "INADDR_LOOPBACK", 0x7F000001);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00005140#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005141#ifdef INADDR_UNSPEC_GROUP
5142 PyModule_AddIntConstant(m, "INADDR_UNSPEC_GROUP", INADDR_UNSPEC_GROUP);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00005143#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005144 PyModule_AddIntConstant(m, "INADDR_UNSPEC_GROUP", 0xe0000000);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00005145#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005146#ifdef INADDR_ALLHOSTS_GROUP
5147 PyModule_AddIntConstant(m, "INADDR_ALLHOSTS_GROUP",
5148 INADDR_ALLHOSTS_GROUP);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00005149#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005150 PyModule_AddIntConstant(m, "INADDR_ALLHOSTS_GROUP", 0xe0000001);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00005151#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005152#ifdef INADDR_MAX_LOCAL_GROUP
5153 PyModule_AddIntConstant(m, "INADDR_MAX_LOCAL_GROUP",
5154 INADDR_MAX_LOCAL_GROUP);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00005155#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005156 PyModule_AddIntConstant(m, "INADDR_MAX_LOCAL_GROUP", 0xe00000ff);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00005157#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005158#ifdef INADDR_NONE
5159 PyModule_AddIntConstant(m, "INADDR_NONE", INADDR_NONE);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00005160#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005161 PyModule_AddIntConstant(m, "INADDR_NONE", 0xffffffff);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00005162#endif
5163
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005164 /* IPv4 [gs]etsockopt options */
5165#ifdef IP_OPTIONS
5166 PyModule_AddIntConstant(m, "IP_OPTIONS", IP_OPTIONS);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00005167#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005168#ifdef IP_HDRINCL
5169 PyModule_AddIntConstant(m, "IP_HDRINCL", IP_HDRINCL);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00005170#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005171#ifdef IP_TOS
5172 PyModule_AddIntConstant(m, "IP_TOS", IP_TOS);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00005173#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005174#ifdef IP_TTL
5175 PyModule_AddIntConstant(m, "IP_TTL", IP_TTL);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00005176#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005177#ifdef IP_RECVOPTS
5178 PyModule_AddIntConstant(m, "IP_RECVOPTS", IP_RECVOPTS);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00005179#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005180#ifdef IP_RECVRETOPTS
5181 PyModule_AddIntConstant(m, "IP_RECVRETOPTS", IP_RECVRETOPTS);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00005182#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005183#ifdef IP_RECVDSTADDR
5184 PyModule_AddIntConstant(m, "IP_RECVDSTADDR", IP_RECVDSTADDR);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00005185#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005186#ifdef IP_RETOPTS
5187 PyModule_AddIntConstant(m, "IP_RETOPTS", IP_RETOPTS);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00005188#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005189#ifdef IP_MULTICAST_IF
5190 PyModule_AddIntConstant(m, "IP_MULTICAST_IF", IP_MULTICAST_IF);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00005191#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005192#ifdef IP_MULTICAST_TTL
5193 PyModule_AddIntConstant(m, "IP_MULTICAST_TTL", IP_MULTICAST_TTL);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00005194#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005195#ifdef IP_MULTICAST_LOOP
5196 PyModule_AddIntConstant(m, "IP_MULTICAST_LOOP", IP_MULTICAST_LOOP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00005197#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005198#ifdef IP_ADD_MEMBERSHIP
5199 PyModule_AddIntConstant(m, "IP_ADD_MEMBERSHIP", IP_ADD_MEMBERSHIP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00005200#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005201#ifdef IP_DROP_MEMBERSHIP
5202 PyModule_AddIntConstant(m, "IP_DROP_MEMBERSHIP", IP_DROP_MEMBERSHIP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00005203#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005204#ifdef IP_DEFAULT_MULTICAST_TTL
5205 PyModule_AddIntConstant(m, "IP_DEFAULT_MULTICAST_TTL",
5206 IP_DEFAULT_MULTICAST_TTL);
Guido van Rossum09be4091999-08-09 14:40:40 +00005207#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005208#ifdef IP_DEFAULT_MULTICAST_LOOP
5209 PyModule_AddIntConstant(m, "IP_DEFAULT_MULTICAST_LOOP",
5210 IP_DEFAULT_MULTICAST_LOOP);
Guido van Rossum09be4091999-08-09 14:40:40 +00005211#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005212#ifdef IP_MAX_MEMBERSHIPS
5213 PyModule_AddIntConstant(m, "IP_MAX_MEMBERSHIPS", IP_MAX_MEMBERSHIPS);
Guido van Rossum09be4091999-08-09 14:40:40 +00005214#endif
5215
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005216 /* IPv6 [gs]etsockopt options, defined in RFC2553 */
5217#ifdef IPV6_JOIN_GROUP
5218 PyModule_AddIntConstant(m, "IPV6_JOIN_GROUP", IPV6_JOIN_GROUP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00005219#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005220#ifdef IPV6_LEAVE_GROUP
5221 PyModule_AddIntConstant(m, "IPV6_LEAVE_GROUP", IPV6_LEAVE_GROUP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00005222#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005223#ifdef IPV6_MULTICAST_HOPS
5224 PyModule_AddIntConstant(m, "IPV6_MULTICAST_HOPS", IPV6_MULTICAST_HOPS);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00005225#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005226#ifdef IPV6_MULTICAST_IF
5227 PyModule_AddIntConstant(m, "IPV6_MULTICAST_IF", IPV6_MULTICAST_IF);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00005228#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005229#ifdef IPV6_MULTICAST_LOOP
5230 PyModule_AddIntConstant(m, "IPV6_MULTICAST_LOOP", IPV6_MULTICAST_LOOP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00005231#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005232#ifdef IPV6_UNICAST_HOPS
5233 PyModule_AddIntConstant(m, "IPV6_UNICAST_HOPS", IPV6_UNICAST_HOPS);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00005234#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005235 /* Additional IPV6 socket options, defined in RFC 3493 */
Martin v. Löwisda91d022003-12-30 11:14:01 +00005236#ifdef IPV6_V6ONLY
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005237 PyModule_AddIntConstant(m, "IPV6_V6ONLY", IPV6_V6ONLY);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005238#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005239 /* Advanced IPV6 socket options, from RFC 3542 */
Martin v. Löwisda91d022003-12-30 11:14:01 +00005240#ifdef IPV6_CHECKSUM
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005241 PyModule_AddIntConstant(m, "IPV6_CHECKSUM", IPV6_CHECKSUM);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005242#endif
5243#ifdef IPV6_DONTFRAG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005244 PyModule_AddIntConstant(m, "IPV6_DONTFRAG", IPV6_DONTFRAG);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005245#endif
5246#ifdef IPV6_DSTOPTS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005247 PyModule_AddIntConstant(m, "IPV6_DSTOPTS", IPV6_DSTOPTS);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005248#endif
5249#ifdef IPV6_HOPLIMIT
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005250 PyModule_AddIntConstant(m, "IPV6_HOPLIMIT", IPV6_HOPLIMIT);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005251#endif
5252#ifdef IPV6_HOPOPTS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005253 PyModule_AddIntConstant(m, "IPV6_HOPOPTS", IPV6_HOPOPTS);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005254#endif
5255#ifdef IPV6_NEXTHOP
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005256 PyModule_AddIntConstant(m, "IPV6_NEXTHOP", IPV6_NEXTHOP);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005257#endif
5258#ifdef IPV6_PATHMTU
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005259 PyModule_AddIntConstant(m, "IPV6_PATHMTU", IPV6_PATHMTU);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005260#endif
5261#ifdef IPV6_PKTINFO
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005262 PyModule_AddIntConstant(m, "IPV6_PKTINFO", IPV6_PKTINFO);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005263#endif
5264#ifdef IPV6_RECVDSTOPTS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005265 PyModule_AddIntConstant(m, "IPV6_RECVDSTOPTS", IPV6_RECVDSTOPTS);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005266#endif
5267#ifdef IPV6_RECVHOPLIMIT
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005268 PyModule_AddIntConstant(m, "IPV6_RECVHOPLIMIT", IPV6_RECVHOPLIMIT);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005269#endif
5270#ifdef IPV6_RECVHOPOPTS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005271 PyModule_AddIntConstant(m, "IPV6_RECVHOPOPTS", IPV6_RECVHOPOPTS);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005272#endif
5273#ifdef IPV6_RECVPKTINFO
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005274 PyModule_AddIntConstant(m, "IPV6_RECVPKTINFO", IPV6_RECVPKTINFO);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005275#endif
5276#ifdef IPV6_RECVRTHDR
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005277 PyModule_AddIntConstant(m, "IPV6_RECVRTHDR", IPV6_RECVRTHDR);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005278#endif
5279#ifdef IPV6_RECVTCLASS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005280 PyModule_AddIntConstant(m, "IPV6_RECVTCLASS", IPV6_RECVTCLASS);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005281#endif
5282#ifdef IPV6_RTHDR
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005283 PyModule_AddIntConstant(m, "IPV6_RTHDR", IPV6_RTHDR);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005284#endif
5285#ifdef IPV6_RTHDRDSTOPTS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005286 PyModule_AddIntConstant(m, "IPV6_RTHDRDSTOPTS", IPV6_RTHDRDSTOPTS);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005287#endif
5288#ifdef IPV6_RTHDR_TYPE_0
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005289 PyModule_AddIntConstant(m, "IPV6_RTHDR_TYPE_0", IPV6_RTHDR_TYPE_0);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005290#endif
5291#ifdef IPV6_RECVPATHMTU
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005292 PyModule_AddIntConstant(m, "IPV6_RECVPATHMTU", IPV6_RECVPATHMTU);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005293#endif
5294#ifdef IPV6_TCLASS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005295 PyModule_AddIntConstant(m, "IPV6_TCLASS", IPV6_TCLASS);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005296#endif
5297#ifdef IPV6_USE_MIN_MTU
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005298 PyModule_AddIntConstant(m, "IPV6_USE_MIN_MTU", IPV6_USE_MIN_MTU);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005299#endif
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00005300
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005301 /* TCP options */
5302#ifdef TCP_NODELAY
5303 PyModule_AddIntConstant(m, "TCP_NODELAY", TCP_NODELAY);
Guido van Rossum09be4091999-08-09 14:40:40 +00005304#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005305#ifdef TCP_MAXSEG
5306 PyModule_AddIntConstant(m, "TCP_MAXSEG", TCP_MAXSEG);
Guido van Rossum09be4091999-08-09 14:40:40 +00005307#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005308#ifdef TCP_CORK
5309 PyModule_AddIntConstant(m, "TCP_CORK", TCP_CORK);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00005310#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005311#ifdef TCP_KEEPIDLE
5312 PyModule_AddIntConstant(m, "TCP_KEEPIDLE", TCP_KEEPIDLE);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00005313#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005314#ifdef TCP_KEEPINTVL
5315 PyModule_AddIntConstant(m, "TCP_KEEPINTVL", TCP_KEEPINTVL);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00005316#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005317#ifdef TCP_KEEPCNT
5318 PyModule_AddIntConstant(m, "TCP_KEEPCNT", TCP_KEEPCNT);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00005319#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005320#ifdef TCP_SYNCNT
5321 PyModule_AddIntConstant(m, "TCP_SYNCNT", TCP_SYNCNT);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00005322#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005323#ifdef TCP_LINGER2
5324 PyModule_AddIntConstant(m, "TCP_LINGER2", TCP_LINGER2);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00005325#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005326#ifdef TCP_DEFER_ACCEPT
5327 PyModule_AddIntConstant(m, "TCP_DEFER_ACCEPT", TCP_DEFER_ACCEPT);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00005328#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005329#ifdef TCP_WINDOW_CLAMP
5330 PyModule_AddIntConstant(m, "TCP_WINDOW_CLAMP", TCP_WINDOW_CLAMP);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00005331#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005332#ifdef TCP_INFO
5333 PyModule_AddIntConstant(m, "TCP_INFO", TCP_INFO);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00005334#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005335#ifdef TCP_QUICKACK
5336 PyModule_AddIntConstant(m, "TCP_QUICKACK", TCP_QUICKACK);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00005337#endif
5338
Guido van Rossum09be4091999-08-09 14:40:40 +00005339
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005340 /* IPX options */
5341#ifdef IPX_TYPE
5342 PyModule_AddIntConstant(m, "IPX_TYPE", IPX_TYPE);
Guido van Rossum09be4091999-08-09 14:40:40 +00005343#endif
Guido van Rossum4f199ea1998-04-09 20:56:35 +00005344
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005345 /* get{addr,name}info parameters */
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005346#ifdef EAI_ADDRFAMILY
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005347 PyModule_AddIntConstant(m, "EAI_ADDRFAMILY", EAI_ADDRFAMILY);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005348#endif
5349#ifdef EAI_AGAIN
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005350 PyModule_AddIntConstant(m, "EAI_AGAIN", EAI_AGAIN);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005351#endif
5352#ifdef EAI_BADFLAGS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005353 PyModule_AddIntConstant(m, "EAI_BADFLAGS", EAI_BADFLAGS);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005354#endif
5355#ifdef EAI_FAIL
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005356 PyModule_AddIntConstant(m, "EAI_FAIL", EAI_FAIL);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005357#endif
5358#ifdef EAI_FAMILY
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005359 PyModule_AddIntConstant(m, "EAI_FAMILY", EAI_FAMILY);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005360#endif
5361#ifdef EAI_MEMORY
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005362 PyModule_AddIntConstant(m, "EAI_MEMORY", EAI_MEMORY);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005363#endif
5364#ifdef EAI_NODATA
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005365 PyModule_AddIntConstant(m, "EAI_NODATA", EAI_NODATA);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005366#endif
5367#ifdef EAI_NONAME
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005368 PyModule_AddIntConstant(m, "EAI_NONAME", EAI_NONAME);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005369#endif
Martin v. Löwisda91d022003-12-30 11:14:01 +00005370#ifdef EAI_OVERFLOW
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005371 PyModule_AddIntConstant(m, "EAI_OVERFLOW", EAI_OVERFLOW);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005372#endif
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005373#ifdef EAI_SERVICE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005374 PyModule_AddIntConstant(m, "EAI_SERVICE", EAI_SERVICE);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005375#endif
5376#ifdef EAI_SOCKTYPE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005377 PyModule_AddIntConstant(m, "EAI_SOCKTYPE", EAI_SOCKTYPE);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005378#endif
5379#ifdef EAI_SYSTEM
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005380 PyModule_AddIntConstant(m, "EAI_SYSTEM", EAI_SYSTEM);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005381#endif
5382#ifdef EAI_BADHINTS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005383 PyModule_AddIntConstant(m, "EAI_BADHINTS", EAI_BADHINTS);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005384#endif
5385#ifdef EAI_PROTOCOL
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005386 PyModule_AddIntConstant(m, "EAI_PROTOCOL", EAI_PROTOCOL);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005387#endif
5388#ifdef EAI_MAX
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005389 PyModule_AddIntConstant(m, "EAI_MAX", EAI_MAX);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005390#endif
5391#ifdef AI_PASSIVE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005392 PyModule_AddIntConstant(m, "AI_PASSIVE", AI_PASSIVE);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005393#endif
5394#ifdef AI_CANONNAME
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005395 PyModule_AddIntConstant(m, "AI_CANONNAME", AI_CANONNAME);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005396#endif
5397#ifdef AI_NUMERICHOST
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005398 PyModule_AddIntConstant(m, "AI_NUMERICHOST", AI_NUMERICHOST);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005399#endif
Martin v. Löwisda91d022003-12-30 11:14:01 +00005400#ifdef AI_NUMERICSERV
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005401 PyModule_AddIntConstant(m, "AI_NUMERICSERV", AI_NUMERICSERV);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005402#endif
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005403#ifdef AI_MASK
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005404 PyModule_AddIntConstant(m, "AI_MASK", AI_MASK);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005405#endif
5406#ifdef AI_ALL
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005407 PyModule_AddIntConstant(m, "AI_ALL", AI_ALL);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005408#endif
5409#ifdef AI_V4MAPPED_CFG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005410 PyModule_AddIntConstant(m, "AI_V4MAPPED_CFG", AI_V4MAPPED_CFG);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005411#endif
5412#ifdef AI_ADDRCONFIG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005413 PyModule_AddIntConstant(m, "AI_ADDRCONFIG", AI_ADDRCONFIG);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005414#endif
5415#ifdef AI_V4MAPPED
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005416 PyModule_AddIntConstant(m, "AI_V4MAPPED", AI_V4MAPPED);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005417#endif
5418#ifdef AI_DEFAULT
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005419 PyModule_AddIntConstant(m, "AI_DEFAULT", AI_DEFAULT);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005420#endif
5421#ifdef NI_MAXHOST
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005422 PyModule_AddIntConstant(m, "NI_MAXHOST", NI_MAXHOST);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005423#endif
5424#ifdef NI_MAXSERV
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005425 PyModule_AddIntConstant(m, "NI_MAXSERV", NI_MAXSERV);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005426#endif
5427#ifdef NI_NOFQDN
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005428 PyModule_AddIntConstant(m, "NI_NOFQDN", NI_NOFQDN);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005429#endif
5430#ifdef NI_NUMERICHOST
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005431 PyModule_AddIntConstant(m, "NI_NUMERICHOST", NI_NUMERICHOST);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005432#endif
5433#ifdef NI_NAMEREQD
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005434 PyModule_AddIntConstant(m, "NI_NAMEREQD", NI_NAMEREQD);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005435#endif
5436#ifdef NI_NUMERICSERV
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005437 PyModule_AddIntConstant(m, "NI_NUMERICSERV", NI_NUMERICSERV);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005438#endif
5439#ifdef NI_DGRAM
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005440 PyModule_AddIntConstant(m, "NI_DGRAM", NI_DGRAM);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005441#endif
5442
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005443 /* shutdown() parameters */
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005444#ifdef SHUT_RD
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005445 PyModule_AddIntConstant(m, "SHUT_RD", SHUT_RD);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005446#elif defined(SD_RECEIVE)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005447 PyModule_AddIntConstant(m, "SHUT_RD", SD_RECEIVE);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005448#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005449 PyModule_AddIntConstant(m, "SHUT_RD", 0);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005450#endif
5451#ifdef SHUT_WR
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005452 PyModule_AddIntConstant(m, "SHUT_WR", SHUT_WR);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005453#elif defined(SD_SEND)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005454 PyModule_AddIntConstant(m, "SHUT_WR", SD_SEND);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005455#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005456 PyModule_AddIntConstant(m, "SHUT_WR", 1);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005457#endif
5458#ifdef SHUT_RDWR
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005459 PyModule_AddIntConstant(m, "SHUT_RDWR", SHUT_RDWR);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005460#elif defined(SD_BOTH)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005461 PyModule_AddIntConstant(m, "SHUT_RDWR", SD_BOTH);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005462#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005463 PyModule_AddIntConstant(m, "SHUT_RDWR", 2);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005464#endif
5465
Christian Heimesfaf2f632008-01-06 16:59:19 +00005466#ifdef SIO_RCVALL
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005467 {
5468 DWORD codes[] = {SIO_RCVALL, SIO_KEEPALIVE_VALS};
5469 const char *names[] = {"SIO_RCVALL", "SIO_KEEPALIVE_VALS"};
5470 int i;
5471 for(i = 0; i<sizeof(codes)/sizeof(*codes); ++i) {
5472 PyObject *tmp;
5473 tmp = PyLong_FromUnsignedLong(codes[i]);
5474 if (tmp == NULL)
5475 return NULL;
5476 PyModule_AddObject(m, names[i], tmp);
5477 }
5478 }
5479 PyModule_AddIntConstant(m, "RCVALL_OFF", RCVALL_OFF);
5480 PyModule_AddIntConstant(m, "RCVALL_ON", RCVALL_ON);
5481 PyModule_AddIntConstant(m, "RCVALL_SOCKETLEVELONLY", RCVALL_SOCKETLEVELONLY);
Amaury Forgeot d'Arc762681b2008-06-12 23:03:41 +00005482#ifdef RCVALL_IPLEVEL
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005483 PyModule_AddIntConstant(m, "RCVALL_IPLEVEL", RCVALL_IPLEVEL);
Amaury Forgeot d'Arc762681b2008-06-12 23:03:41 +00005484#endif
5485#ifdef RCVALL_MAX
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005486 PyModule_AddIntConstant(m, "RCVALL_MAX", RCVALL_MAX);
Amaury Forgeot d'Arc762681b2008-06-12 23:03:41 +00005487#endif
Christian Heimesfaf2f632008-01-06 16:59:19 +00005488#endif /* _MSTCPIP_ */
5489
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005490 /* Initialize gethostbyname lock */
Just van Rossum1040d2c2003-05-09 07:53:18 +00005491#if defined(USE_GETHOSTBYNAME_LOCK) || defined(USE_GETADDRINFO_LOCK)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005492 netdb_lock = PyThread_allocate_lock();
Guido van Rossum4f199ea1998-04-09 20:56:35 +00005493#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005494 return m;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00005495}
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005496
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00005497
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005498#ifndef HAVE_INET_PTON
Christian Heimes96e7b3d2007-11-20 06:51:17 +00005499#if !defined(NTDDI_VERSION) || (NTDDI_VERSION < NTDDI_LONGHORN)
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00005500
5501/* Simplistic emulation code for inet_pton that only works for IPv4 */
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00005502/* These are not exposed because they do not set errno properly */
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00005503
Guido van Rossum3eede5a2002-06-07 02:08:35 +00005504int
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00005505inet_pton(int af, const char *src, void *dst)
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005506{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005507 if (af == AF_INET) {
Benjamin Petersonf91df042009-02-13 02:50:59 +00005508#if (SIZEOF_INT != 4)
5509#error "Not sure if in_addr_t exists and int is not 32-bits."
5510#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005511 unsigned int packed_addr;
5512 packed_addr = inet_addr(src);
5513 if (packed_addr == INADDR_NONE)
5514 return 0;
5515 memcpy(dst, &packed_addr, 4);
5516 return 1;
5517 }
5518 /* Should set errno to EAFNOSUPPORT */
5519 return -1;
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005520}
5521
Martin v. Löwisc925b1532001-07-21 09:42:15 +00005522const char *
5523inet_ntop(int af, const void *src, char *dst, socklen_t size)
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005524{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005525 if (af == AF_INET) {
5526 struct in_addr packed_addr;
5527 if (size < 16)
5528 /* Should set errno to ENOSPC. */
5529 return NULL;
5530 memcpy(&packed_addr, src, sizeof(packed_addr));
5531 return strncpy(dst, inet_ntoa(packed_addr), size);
5532 }
5533 /* Should set errno to EAFNOSUPPORT */
5534 return NULL;
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005535}
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00005536
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005537#endif
Christian Heimesb6150692007-11-15 23:37:07 +00005538#endif