blob: 50ca91144834e8889930434f595e7e9f00675012 [file] [log] [blame]
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001/* Socket module */
2
3/*
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004
Guido van Rossum6574b3e1991-06-25 21:36:08 +00005This module provides an interface to Berkeley socket IPC.
6
7Limitations:
8
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00009- Only AF_INET, AF_INET6 and AF_UNIX address families are supported in a
Christian Heimes043d6f62008-01-07 17:19:16 +000010 portable manner, though AF_PACKET, AF_NETLINK and AF_TIPC are supported
11 under Linux.
Guido van Rossumc4fcfa32002-06-07 03:19:37 +000012- No read/write operations (use sendall/recv or makefile instead).
13- Additional restrictions apply on some non-Unix platforms (compensated
14 for by socket.py).
Guido van Rossum6574b3e1991-06-25 21:36:08 +000015
Guido van Rossum27e177d1995-03-16 15:43:47 +000016Module interface:
Guido van Rossum6574b3e1991-06-25 21:36:08 +000017
Guido van Rossum27e177d1995-03-16 15:43:47 +000018- socket.error: exception raised for socket specific errors
Martin v. Löwis2d8d4272001-07-21 18:05:31 +000019- socket.gaierror: exception raised for getaddrinfo/getnameinfo errors,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000020 a subclass of socket.error
Martin v. Löwis2d8d4272001-07-21 18:05:31 +000021- socket.herror: exception raised for gethostby* errors,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000022 a subclass of socket.error
Guido van Rossum30a685f1991-06-27 15:51:29 +000023- socket.gethostbyname(hostname) --> host IP address (string: 'dd.dd.dd.dd')
Guido van Rossum3bbc62e1995-01-02 19:30:30 +000024- socket.gethostbyaddr(IP address) --> (hostname, [alias, ...], [IP addr, ...])
Guido van Rossum27e177d1995-03-16 15:43:47 +000025- socket.gethostname() --> host name (string: 'spam' or 'spam.domain.com')
Guido van Rossum25405c71996-12-19 16:42:52 +000026- socket.getprotobyname(protocolname) --> protocol number
Barry Warsaw11b91a02004-06-28 00:50:43 +000027- socket.getservbyname(servicename[, protocolname]) --> port number
28- socket.getservbyport(portnumber[, protocolname]) --> service name
Guido van Rossum7d0a8262007-05-21 23:13:11 +000029- socket.socket([family[, type [, proto, fileno]]]) --> new socket object
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000030 (fileno specifies a pre-existing socket file descriptor)
Dave Cole331708b2004-08-09 04:51:41 +000031- socket.socketpair([family[, type [, proto]]]) --> (socket, socket)
Guido van Rossum006bf911996-06-12 04:04:55 +000032- socket.ntohs(16 bit value) --> new int object
33- socket.ntohl(32 bit value) --> new int object
34- socket.htons(16 bit value) --> new int object
35- socket.htonl(32 bit value) --> new int object
Martin v. Löwis2d8d4272001-07-21 18:05:31 +000036- socket.getaddrinfo(host, port [, family, socktype, proto, flags])
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000037 --> List of (family, socktype, proto, canonname, sockaddr)
Martin v. Löwis2d8d4272001-07-21 18:05:31 +000038- socket.getnameinfo(sockaddr, flags) --> (host, port)
Guido van Rossum27e177d1995-03-16 15:43:47 +000039- socket.AF_INET, socket.SOCK_STREAM, etc.: constants from <socket.h>
Guido van Rossum47dfa4a2003-04-25 05:48:32 +000040- socket.has_ipv6: boolean value indicating if IPv6 is supported
Guido van Rossum5c9eb211999-08-20 18:21:51 +000041- socket.inet_aton(IP address) -> 32-bit packed IP representation
42- socket.inet_ntoa(packed IP) -> IP address string
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +000043- socket.getdefaulttimeout() -> None | float
44- socket.setdefaulttimeout(None | float)
Guido van Rossum6574b3e1991-06-25 21:36:08 +000045- an Internet socket address is a pair (hostname, port)
46 where hostname can be anything recognized by gethostbyname()
47 (including the dd.dd.dd.dd notation) and port is in host byte order
48- where a hostname is returned, the dd.dd.dd.dd notation is used
Guido van Rossum27e177d1995-03-16 15:43:47 +000049- a UNIX domain socket address is a string specifying the pathname
Jeremy Hylton22308652001-02-02 03:23:09 +000050- an AF_PACKET socket address is a tuple containing a string
51 specifying the ethernet interface and an integer specifying
52 the Ethernet protocol number to be received. For example:
Jeremy Hyltondbfb6622001-02-02 19:55:17 +000053 ("eth0",0x1234). Optional 3rd,4th,5th elements in the tuple
Andrew M. Kuchlingb8e17172004-07-10 23:39:35 +000054 specify packet-type and ha-type/addr.
Christian Heimes043d6f62008-01-07 17:19:16 +000055- an AF_TIPC socket address is expressed as
56 (addr_type, v1, v2, v3 [, scope]); where addr_type can be one of:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000057 TIPC_ADDR_NAMESEQ, TIPC_ADDR_NAME, and TIPC_ADDR_ID;
Christian Heimes043d6f62008-01-07 17:19:16 +000058 and scope can be one of:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000059 TIPC_ZONE_SCOPE, TIPC_CLUSTER_SCOPE, and TIPC_NODE_SCOPE.
Christian Heimes043d6f62008-01-07 17:19:16 +000060 The meaning of v1, v2 and v3 depends on the value of addr_type:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000061 if addr_type is TIPC_ADDR_NAME:
62 v1 is the server type
63 v2 is the port identifier
64 v3 is ignored
65 if addr_type is TIPC_ADDR_NAMESEQ:
66 v1 is the server type
67 v2 is the lower port number
68 v3 is the upper port number
69 if addr_type is TIPC_ADDR_ID:
70 v1 is the node
71 v2 is the ref
72 v3 is ignored
Christian Heimes043d6f62008-01-07 17:19:16 +000073
Guido van Rossum6574b3e1991-06-25 21:36:08 +000074
Guido van Rossumc4fcfa32002-06-07 03:19:37 +000075Local naming conventions:
Guido van Rossum6574b3e1991-06-25 21:36:08 +000076
Guido van Rossumc4fcfa32002-06-07 03:19:37 +000077- names starting with sock_ are socket object methods
78- names starting with socket_ are module-level functions
79- names starting with PySocket are exported through socketmodule.h
Guido van Rossum30a685f1991-06-27 15:51:29 +000080
Guido van Rossum6574b3e1991-06-25 21:36:08 +000081*/
82
Thomas Wouters477c8d52006-05-27 19:21:47 +000083#ifdef __APPLE__
84 /*
85 * inet_aton is not available on OSX 10.3, yet we want to use a binary
86 * that was build on 10.4 or later to work on that release, weak linking
87 * comes to the rescue.
88 */
89# pragma weak inet_aton
90#endif
91
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000092#include "Python.h"
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000093#include "structmember.h"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000094
Guido van Rossum47dfa4a2003-04-25 05:48:32 +000095#undef MAX
96#define MAX(x, y) ((x) < (y) ? (y) : (x))
97
Guido van Rossumc4fcfa32002-06-07 03:19:37 +000098/* Socket object documentation */
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000099PyDoc_STRVAR(sock_doc,
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000100"socket([family[, type[, proto]]]) -> socket object\n\
101\n\
102Open a socket of the given type. The family argument specifies the\n\
103address family; it defaults to AF_INET. The type argument specifies\n\
104whether this is a stream (SOCK_STREAM, this is the default)\n\
105or datagram (SOCK_DGRAM) socket. The protocol argument defaults to 0,\n\
106specifying the default protocol. Keyword arguments are accepted.\n\
107\n\
108A socket object represents one endpoint of a network connection.\n\
109\n\
110Methods of socket objects (keyword arguments not allowed):\n\
111\n\
Guido van Rossum39eb8fa2007-11-16 01:24:05 +0000112_accept() -- accept connection, returning new socket fd and client address\n\
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000113bind(addr) -- bind the socket to a local address\n\
114close() -- close the socket\n\
115connect(addr) -- connect the socket to a remote address\n\
116connect_ex(addr) -- connect, return an error code instead of an exception\n\
Guido van Rossum39eb8fa2007-11-16 01:24:05 +0000117_dup() -- return a new socket fd duplicated from fileno()\n\
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000118fileno() -- return underlying file descriptor\n\
119getpeername() -- return remote address [*]\n\
120getsockname() -- return local address\n\
121getsockopt(level, optname[, buflen]) -- get socket options\n\
122gettimeout() -- return timeout or None\n\
123listen(n) -- start listening for incoming connections\n\
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000124recv(buflen[, flags]) -- receive data\n\
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000125recv_into(buffer[, nbytes[, flags]]) -- receive data (into a buffer)\n\
Thomas Wouters477c8d52006-05-27 19:21:47 +0000126recvfrom(buflen[, flags]) -- receive data and sender\'s address\n\
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000127recvfrom_into(buffer[, nbytes, [, flags])\n\
Thomas Wouters477c8d52006-05-27 19:21:47 +0000128 -- receive data and sender\'s address (into a buffer)\n\
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000129sendall(data[, flags]) -- send all data\n\
130send(data[, flags]) -- send data, may not send all of it\n\
131sendto(data[, flags], addr) -- send data to a given address\n\
132setblocking(0 | 1) -- set or clear the blocking I/O flag\n\
133setsockopt(level, optname, value) -- set socket options\n\
134settimeout(None | float) -- set or clear the timeout\n\
135shutdown(how) -- shut down traffic in one or both directions\n\
136\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000137 [*] not available on all platforms!");
Guido van Rossum3baaa131999-03-22 21:44:51 +0000138
Walter Dörwaldf0dfc7a2003-10-20 14:01:56 +0000139/* XXX This is a terrible mess of platform-dependent preprocessor hacks.
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000140 I hope some day someone can clean this up please... */
141
Guido van Rossum9376b741999-09-15 22:01:40 +0000142/* Hacks for gethostbyname_r(). On some non-Linux platforms, the configure
143 script doesn't get this right, so we hardcode some platform checks below.
144 On the other hand, not all Linux versions agree, so there the settings
145 computed by the configure script are needed! */
146
147#ifndef linux
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000148# undef HAVE_GETHOSTBYNAME_R_3_ARG
149# undef HAVE_GETHOSTBYNAME_R_5_ARG
150# undef HAVE_GETHOSTBYNAME_R_6_ARG
Guido van Rossum9376b741999-09-15 22:01:40 +0000151#endif
Guido van Rossume7de2061999-03-24 17:24:33 +0000152
Guido van Rossum7a122991999-04-13 04:07:32 +0000153#ifndef WITH_THREAD
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000154# undef HAVE_GETHOSTBYNAME_R
Guido van Rossum7a122991999-04-13 04:07:32 +0000155#endif
156
Guido van Rossume7de2061999-03-24 17:24:33 +0000157#ifdef HAVE_GETHOSTBYNAME_R
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000158# if defined(_AIX) || defined(__osf__)
159# define HAVE_GETHOSTBYNAME_R_3_ARG
160# elif defined(__sun) || defined(__sgi)
161# define HAVE_GETHOSTBYNAME_R_5_ARG
162# elif defined(linux)
Guido van Rossum9376b741999-09-15 22:01:40 +0000163/* Rely on the configure script */
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000164# else
165# undef HAVE_GETHOSTBYNAME_R
166# endif
Guido van Rossume7de2061999-03-24 17:24:33 +0000167#endif
168
Guido van Rossum3eede5a2002-06-07 02:08:35 +0000169#if !defined(HAVE_GETHOSTBYNAME_R) && defined(WITH_THREAD) && \
170 !defined(MS_WINDOWS)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000171# define USE_GETHOSTBYNAME_LOCK
Guido van Rossum3baaa131999-03-22 21:44:51 +0000172#endif
173
Hye-Shik Chang9ceebd52005-09-24 14:58:47 +0000174/* To use __FreeBSD_version */
175#ifdef HAVE_SYS_PARAM_H
176#include <sys/param.h>
177#endif
Just van Rossum1040d2c2003-05-09 07:53:18 +0000178/* On systems on which getaddrinfo() is believed to not be thread-safe,
Just van Rossum09aecd72003-05-09 08:03:44 +0000179 (this includes the getaddrinfo emulation) protect access with a lock. */
Hye-Shik Chang9ceebd52005-09-24 14:58:47 +0000180#if defined(WITH_THREAD) && (defined(__APPLE__) || \
181 (defined(__FreeBSD__) && __FreeBSD_version+0 < 503000) || \
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000182 defined(__OpenBSD__) || defined(__NetBSD__) || \
183 defined(__VMS) || !defined(HAVE_GETADDRINFO))
Just van Rossum1040d2c2003-05-09 07:53:18 +0000184#define USE_GETADDRINFO_LOCK
185#endif
186
187#ifdef USE_GETADDRINFO_LOCK
188#define ACQUIRE_GETADDRINFO_LOCK PyThread_acquire_lock(netdb_lock, 1);
189#define RELEASE_GETADDRINFO_LOCK PyThread_release_lock(netdb_lock);
190#else
191#define ACQUIRE_GETADDRINFO_LOCK
192#define RELEASE_GETADDRINFO_LOCK
193#endif
194
195#if defined(USE_GETHOSTBYNAME_LOCK) || defined(USE_GETADDRINFO_LOCK)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000196# include "pythread.h"
Guido van Rossum4f199ea1998-04-09 20:56:35 +0000197#endif
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000198
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000199#if defined(PYCC_VACPP)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000200# include <types.h>
201# include <io.h>
202# include <sys/ioctl.h>
203# include <utils.h>
204# include <ctype.h>
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000205#endif
206
Martin v. Löwis9e437302002-12-06 12:57:26 +0000207#if defined(__VMS)
Martin v. Löwis79acb9e2002-12-06 12:48:53 +0000208# include <ioctl.h>
209#endif
210
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000211#if defined(PYOS_OS2)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000212# define INCL_DOS
213# define INCL_DOSERRORS
214# define INCL_NOPMAPI
215# include <os2.h>
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000216#endif
217
Jeremy Hyltonfb509a32003-07-17 16:58:48 +0000218#if defined(__sgi) && _COMPILER_VERSION>700 && !_SGIAPI
Martin v. Löwis8eb92a02002-09-19 08:03:21 +0000219/* make sure that the reentrant (gethostbyaddr_r etc)
220 functions are declared correctly if compiling with
221 MIPSPro 7.x in ANSI C mode (default) */
Jeremy Hyltonfb509a32003-07-17 16:58:48 +0000222
Thomas Wouters477c8d52006-05-27 19:21:47 +0000223/* XXX Using _SGIAPI is the wrong thing,
Jeremy Hyltonfb509a32003-07-17 16:58:48 +0000224 but I don't know what the right thing is. */
Trent Mick8ea5bdf2004-09-13 17:48:41 +0000225#undef _SGIAPI /* to avoid warning */
Martin v. Löwis8eb92a02002-09-19 08:03:21 +0000226#define _SGIAPI 1
Jeremy Hyltonfb509a32003-07-17 16:58:48 +0000227
Trent Mick8ea5bdf2004-09-13 17:48:41 +0000228#undef _XOPEN_SOURCE
229#include <sys/socket.h>
230#include <sys/types.h>
231#include <netinet/in.h>
232#ifdef _SS_ALIGNSIZE
233#define HAVE_GETADDRINFO 1
234#define HAVE_GETNAMEINFO 1
235#endif
236
Jeremy Hyltonfb509a32003-07-17 16:58:48 +0000237#define HAVE_INET_PTON
238#include <netdb.h>
Martin v. Löwis8eb92a02002-09-19 08:03:21 +0000239#endif
240
Thomas Wouters477c8d52006-05-27 19:21:47 +0000241/* Irix 6.5 fails to define this variable at all. This is needed
242 for both GCC and SGI's compiler. I'd say that the SGI headers
Georg Brandldbd83392006-02-20 09:42:33 +0000243 are just busted. Same thing for Solaris. */
244#if (defined(__sgi) || defined(sun)) && !defined(INET_ADDRSTRLEN)
Anthony Baxterbab23cf2003-10-04 08:00:49 +0000245#define INET_ADDRSTRLEN 16
246#endif
247
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000248/* Generic includes */
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000249#ifdef HAVE_SYS_TYPES_H
Guido van Rossumb6775db1994-08-01 11:34:53 +0000250#include <sys/types.h>
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000251#endif
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000252
Marc-André Lemburg976ade62002-02-16 18:47:07 +0000253/* Generic socket object definitions and includes */
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000254#define PySocket_BUILDING_SOCKET
Marc-André Lemburgbb8b78b2002-02-16 18:44:52 +0000255#include "socketmodule.h"
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000256
257/* Addressing includes */
258
Guido van Rossum6f489d91996-06-28 20:15:15 +0000259#ifndef MS_WINDOWS
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000260
261/* Non-MS WINDOWS includes */
262# include <netdb.h>
Guido van Rossum5c9eb211999-08-20 18:21:51 +0000263
Guido van Rossum9376b741999-09-15 22:01:40 +0000264/* Headers needed for inet_ntoa() and inet_addr() */
Skip Montanaroeb33e5a2007-08-17 12:57:41 +0000265# if defined(PYOS_OS2) && defined(PYCC_VACPP)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000266# include <netdb.h>
Tim Peters603c6832001-11-05 02:45:59 +0000267typedef size_t socklen_t;
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000268# else
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000269# include <arpa/inet.h>
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000270# endif
Guido van Rossum5c9eb211999-08-20 18:21:51 +0000271
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000272# include <fcntl.h>
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000273
Guido van Rossume2ae77b2001-10-24 20:42:55 +0000274#else
Guido van Rossum48a680c2001-03-02 06:34:14 +0000275
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000276/* MS_WINDOWS includes */
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000277# ifdef HAVE_FCNTL_H
278# include <fcntl.h>
279# endif
Guido van Rossum48a680c2001-03-02 06:34:14 +0000280
Jeremy Hylton22308652001-02-02 03:23:09 +0000281#endif
282
Skip Montanaro7befb992004-02-10 16:50:21 +0000283#include <stddef.h>
Martin v. Löwisa45ecae2001-06-24 21:28:42 +0000284
285#ifndef offsetof
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000286# define offsetof(type, member) ((size_t)(&((type *)0)->member))
Martin v. Löwisa45ecae2001-06-24 21:28:42 +0000287#endif
288
Neal Norwitz39d22e52002-11-02 19:55:21 +0000289#ifndef O_NONBLOCK
290# define O_NONBLOCK O_NDELAY
Guido van Rossum3bbc62e1995-01-02 19:30:30 +0000291#endif
292
Trent Micka708d6e2004-09-07 17:48:26 +0000293/* include Python's addrinfo.h unless it causes trouble */
294#if defined(__sgi) && _COMPILER_VERSION>700 && defined(_SS_ALIGNSIZE)
295 /* Do not include addinfo.h on some newer IRIX versions.
296 * _SS_ALIGNSIZE is defined in sys/socket.h by 6.5.21,
297 * for example, but not by 6.5.10.
298 */
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000299#elif defined(_MSC_VER) && _MSC_VER>1201
Trent Micka708d6e2004-09-07 17:48:26 +0000300 /* Do not include addrinfo.h for MSVC7 or greater. 'addrinfo' and
301 * EAI_* constants are defined in (the already included) ws2tcpip.h.
302 */
303#else
304# include "addrinfo.h"
305#endif
Jason Tishlerc246cb72004-08-09 13:25:59 +0000306
Martin v. Löwisb9ab1592001-06-24 21:18:26 +0000307#ifndef HAVE_INET_PTON
Christian Heimes96e7b3d2007-11-20 06:51:17 +0000308#if !defined(NTDDI_VERSION) || (NTDDI_VERSION < NTDDI_LONGHORN)
Guido van Rossum3eede5a2002-06-07 02:08:35 +0000309int inet_pton(int af, const char *src, void *dst);
Martin v. Löwisc925b1532001-07-21 09:42:15 +0000310const char *inet_ntop(int af, const void *src, char *dst, socklen_t size);
Martin v. Löwisb9ab1592001-06-24 21:18:26 +0000311#endif
Christian Heimesb6150692007-11-15 23:37:07 +0000312#endif
Martin v. Löwisb9ab1592001-06-24 21:18:26 +0000313
Martin v. Löwisae26dc22001-08-03 10:02:29 +0000314#ifdef __APPLE__
315/* On OS X, getaddrinfo returns no error indication of lookup
316 failure, so we must use the emulation instead of the libinfo
317 implementation. Unfortunately, performing an autoconf test
318 for this bug would require DNS access for the machine performing
319 the configuration, which is not acceptable. Therefore, we
320 determine the bug just by checking for __APPLE__. If this bug
321 gets ever fixed, perhaps checking for sys/version.h would be
322 appropriate, which is 10/0 on the system with the bug. */
Jack Jansen84262fb2002-07-02 14:40:42 +0000323#ifndef HAVE_GETNAMEINFO
324/* This bug seems to be fixed in Jaguar. Ths easiest way I could
325 Find to check for Jaguar is that it has getnameinfo(), which
326 older releases don't have */
Martin v. Löwisae26dc22001-08-03 10:02:29 +0000327#undef HAVE_GETADDRINFO
Martin v. Löwisae26dc22001-08-03 10:02:29 +0000328#endif
Thomas Wouters477c8d52006-05-27 19:21:47 +0000329
330#ifdef HAVE_INET_ATON
331#define USE_INET_ATON_WEAKLINK
332#endif
333
Jack Jansen84262fb2002-07-02 14:40:42 +0000334#endif
Martin v. Löwisae26dc22001-08-03 10:02:29 +0000335
Martin v. Löwis01dfdb32001-06-23 16:30:13 +0000336/* I know this is a bad practice, but it is the easiest... */
Martin v. Löwisae26dc22001-08-03 10:02:29 +0000337#if !defined(HAVE_GETADDRINFO)
Martin v. Löwisfccac2e2003-05-01 05:20:46 +0000338/* avoid clashes with the C library definition of the symbol. */
339#define getaddrinfo fake_getaddrinfo
340#define gai_strerror fake_gai_strerror
341#define freeaddrinfo fake_freeaddrinfo
Martin v. Löwis01dfdb32001-06-23 16:30:13 +0000342#include "getaddrinfo.c"
343#endif
Martin v. Löwisae26dc22001-08-03 10:02:29 +0000344#if !defined(HAVE_GETNAMEINFO)
Martin v. Löwisfccac2e2003-05-01 05:20:46 +0000345#define getnameinfo fake_getnameinfo
Martin v. Löwis01dfdb32001-06-23 16:30:13 +0000346#include "getnameinfo.c"
347#endif
348
Guido van Rossum39eb8fa2007-11-16 01:24:05 +0000349#ifdef MS_WINDOWS
350/* On Windows a socket is really a handle not an fd */
351static SOCKET
352dup_socket(SOCKET handle)
353{
Daniel Stutzbach6c765282010-09-03 12:38:33 +0000354 WSAPROTOCOL_INFO info;
Guido van Rossum39eb8fa2007-11-16 01:24:05 +0000355
Daniel Stutzbach6c765282010-09-03 12:38:33 +0000356 if (WSADuplicateSocket(handle, GetCurrentProcessId(), &info))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000357 return INVALID_SOCKET;
Daniel Stutzbach6c765282010-09-03 12:38:33 +0000358
359 return WSASocket(FROM_PROTOCOL_INFO, FROM_PROTOCOL_INFO,
360 FROM_PROTOCOL_INFO, &info, 0, 0);
Guido van Rossum39eb8fa2007-11-16 01:24:05 +0000361}
Guido van Rossum2dd8ddd2000-04-21 20:33:00 +0000362#define SOCKETCLOSE closesocket
Guido van Rossum39eb8fa2007-11-16 01:24:05 +0000363#else
364/* On Unix we can use dup to duplicate the file descriptor of a socket*/
365#define dup_socket(fd) dup(fd)
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +0000366#endif
367
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000368#ifdef MS_WIN32
Guido van Rossum3eede5a2002-06-07 02:08:35 +0000369#define EAFNOSUPPORT WSAEAFNOSUPPORT
370#define snprintf _snprintf
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000371#endif
Fred Drakea04eaad2000-06-30 02:46:07 +0000372
Andrew MacIntyreba43e872002-03-03 03:03:52 +0000373#if defined(PYOS_OS2) && !defined(PYCC_GCC)
Guido van Rossum2dd8ddd2000-04-21 20:33:00 +0000374#define SOCKETCLOSE soclose
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000375#define NO_DUP /* Sockets are Not Actual File Handles under OS/2 */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000376#endif
377
Guido van Rossum2dd8ddd2000-04-21 20:33:00 +0000378#ifndef SOCKETCLOSE
379#define SOCKETCLOSE close
380#endif
381
Gregory P. Smith2501aca2010-10-13 03:39:47 +0000382#if (defined(HAVE_BLUETOOTH_H) || defined(HAVE_BLUETOOTH_BLUETOOTH_H)) && !defined(__NetBSD__)
Hye-Shik Chang81268602004-02-02 06:05:24 +0000383#define USE_BLUETOOTH 1
384#if defined(__FreeBSD__)
385#define BTPROTO_L2CAP BLUETOOTH_PROTO_L2CAP
386#define BTPROTO_RFCOMM BLUETOOTH_PROTO_RFCOMM
Thomas Wouterscf297e42007-02-23 15:07:44 +0000387#define BTPROTO_HCI BLUETOOTH_PROTO_HCI
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000388#define SOL_HCI SOL_HCI_RAW
389#define HCI_FILTER SO_HCI_RAW_FILTER
Hye-Shik Chang81268602004-02-02 06:05:24 +0000390#define sockaddr_l2 sockaddr_l2cap
391#define sockaddr_rc sockaddr_rfcomm
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000392#define hci_dev hci_node
Hye-Shik Chang81268602004-02-02 06:05:24 +0000393#define _BT_L2_MEMB(sa, memb) ((sa)->l2cap_##memb)
394#define _BT_RC_MEMB(sa, memb) ((sa)->rfcomm_##memb)
Thomas Wouterscf297e42007-02-23 15:07:44 +0000395#define _BT_HCI_MEMB(sa, memb) ((sa)->hci_##memb)
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000396#elif defined(__NetBSD__)
397#define sockaddr_l2 sockaddr_bt
398#define sockaddr_rc sockaddr_bt
Thomas Wouterscf297e42007-02-23 15:07:44 +0000399#define sockaddr_hci sockaddr_bt
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000400#define sockaddr_sco sockaddr_bt
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000401#define _BT_L2_MEMB(sa, memb) ((sa)->bt_##memb)
402#define _BT_RC_MEMB(sa, memb) ((sa)->bt_##memb)
Thomas Wouterscf297e42007-02-23 15:07:44 +0000403#define _BT_HCI_MEMB(sa, memb) ((sa)->bt_##memb)
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000404#define _BT_SCO_MEMB(sa, memb) ((sa)->bt_##memb)
Hye-Shik Chang81268602004-02-02 06:05:24 +0000405#else
Hye-Shik Chang81268602004-02-02 06:05:24 +0000406#define _BT_L2_MEMB(sa, memb) ((sa)->l2_##memb)
407#define _BT_RC_MEMB(sa, memb) ((sa)->rc_##memb)
Thomas Wouterscf297e42007-02-23 15:07:44 +0000408#define _BT_HCI_MEMB(sa, memb) ((sa)->hci_##memb)
Hye-Shik Chang81268602004-02-02 06:05:24 +0000409#define _BT_SCO_MEMB(sa, memb) ((sa)->sco_##memb)
410#endif
411#endif
412
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000413#ifdef __VMS
414/* TCP/IP Services for VMS uses a maximum send/recv buffer length */
415#define SEGMENT_SIZE (32 * 1024 -1)
416#endif
417
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000418#define SAS2SA(x) ((struct sockaddr *)(x))
Thomas Wouters89f507f2006-12-13 04:49:30 +0000419
Martin v. Löwise9416172003-05-03 10:12:45 +0000420/*
421 * Constants for getnameinfo()
422 */
423#if !defined(NI_MAXHOST)
424#define NI_MAXHOST 1025
425#endif
426#if !defined(NI_MAXSERV)
427#define NI_MAXSERV 32
428#endif
429
Guido van Rossum7d0a8262007-05-21 23:13:11 +0000430#ifndef INVALID_SOCKET /* MS defines this */
431#define INVALID_SOCKET (-1)
432#endif
433
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000434/* XXX There's a problem here: *static* functions are not supposed to have
435 a Py prefix (or use CapitalizedWords). Later... */
436
Guido van Rossum30a685f1991-06-27 15:51:29 +0000437/* Global variable holding the exception type for errors detected
438 by this module (but not argument type or memory errors, etc.). */
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000439static PyObject *socket_error;
440static PyObject *socket_herror;
441static PyObject *socket_gaierror;
Raymond Hettingeref7343c2003-06-29 03:08:05 +0000442static PyObject *socket_timeout;
Guido van Rossum30a685f1991-06-27 15:51:29 +0000443
Tim Peters643a7fc2002-02-17 04:13:21 +0000444/* A forward reference to the socket type object.
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000445 The sock_type variable contains pointers to various functions,
446 some of which call new_sockobject(), which uses sock_type, so
Tim Peters643a7fc2002-02-17 04:13:21 +0000447 there has to be a circular reference. */
Jeremy Hylton938ace62002-07-17 16:30:39 +0000448static PyTypeObject sock_type;
Guido van Rossum48a680c2001-03-02 06:34:14 +0000449
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000450#if defined(HAVE_POLL_H)
451#include <poll.h>
452#elif defined(HAVE_SYS_POLL_H)
453#include <sys/poll.h>
454#endif
455
Martin v. Löwisf84d1b92006-02-11 09:27:05 +0000456#ifdef Py_SOCKET_FD_CAN_BE_GE_FD_SETSIZE
457/* Platform can select file descriptors beyond FD_SETSIZE */
458#define IS_SELECTABLE(s) 1
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000459#elif defined(HAVE_POLL)
460/* Instead of select(), we'll use poll() since poll() works on any fd. */
461#define IS_SELECTABLE(s) 1
462/* Can we call select() with this socket without a buffer overrun? */
Martin v. Löwisf84d1b92006-02-11 09:27:05 +0000463#else
464/* POSIX says selecting file descriptors beyond FD_SETSIZE
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000465 has undefined behaviour. If there's no timeout left, we don't have to
466 call select, so it's a safe, little white lie. */
467#define IS_SELECTABLE(s) ((s)->sock_fd < FD_SETSIZE || s->sock_timeout <= 0.0)
Martin v. Löwisf84d1b92006-02-11 09:27:05 +0000468#endif
Neal Norwitz082b2df2006-02-07 07:04:46 +0000469
470static PyObject*
471select_error(void)
472{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000473 PyErr_SetString(socket_error, "unable to select on socket");
474 return NULL;
Neal Norwitz082b2df2006-02-07 07:04:46 +0000475}
476
Antoine Pitrou3e1fd272010-09-28 21:23:11 +0000477#ifdef MS_WINDOWS
Antoine Pitrouc8f01962010-09-28 22:03:27 +0000478#ifndef WSAEAGAIN
479#define WSAEAGAIN WSAEWOULDBLOCK
480#endif
Antoine Pitrou3e1fd272010-09-28 21:23:11 +0000481#define CHECK_ERRNO(expected) \
482 (WSAGetLastError() == WSA ## expected)
483#else
484#define CHECK_ERRNO(expected) \
485 (errno == expected)
486#endif
487
Guido van Rossum30a685f1991-06-27 15:51:29 +0000488/* Convenience function to raise an error according to errno
489 and return a NULL pointer from a function. */
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000490
Guido van Rossum73624e91994-10-10 17:59:00 +0000491static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000492set_error(void)
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000493{
Guido van Rossum8d665e61996-06-26 18:22:49 +0000494#ifdef MS_WINDOWS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000495 int err_no = WSAGetLastError();
496 /* PyErr_SetExcFromWindowsErr() invokes FormatMessage() which
497 recognizes the error codes used by both GetLastError() and
498 WSAGetLastError */
499 if (err_no)
500 return PyErr_SetExcFromWindowsErr(socket_error, err_no);
Guido van Rossum14ed0b21994-09-29 09:50:09 +0000501#endif
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000502
Andrew MacIntyreba43e872002-03-03 03:03:52 +0000503#if defined(PYOS_OS2) && !defined(PYCC_GCC)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000504 if (sock_errno() != NO_ERROR) {
505 APIRET rc;
506 ULONG msglen;
507 char outbuf[100];
508 int myerrorcode = sock_errno();
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000509
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000510 /* Retrieve socket-related error message from MPTN.MSG file */
511 rc = DosGetMessage(NULL, 0, outbuf, sizeof(outbuf),
512 myerrorcode - SOCBASEERR + 26,
513 "mptn.msg",
514 &msglen);
515 if (rc == NO_ERROR) {
516 PyObject *v;
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000517
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000518 /* OS/2 doesn't guarantee a terminator */
519 outbuf[msglen] = '\0';
520 if (strlen(outbuf) > 0) {
521 /* If non-empty msg, trim CRLF */
522 char *lastc = &outbuf[ strlen(outbuf)-1 ];
523 while (lastc > outbuf &&
524 isspace(Py_CHARMASK(*lastc))) {
525 /* Trim trailing whitespace (CRLF) */
526 *lastc-- = '\0';
527 }
528 }
529 v = Py_BuildValue("(is)", myerrorcode, outbuf);
530 if (v != NULL) {
531 PyErr_SetObject(socket_error, v);
532 Py_DECREF(v);
533 }
534 return NULL;
535 }
536 }
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000537#endif
538
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000539 return PyErr_SetFromErrno(socket_error);
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000540}
541
Guido van Rossum30a685f1991-06-27 15:51:29 +0000542
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000543static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000544set_herror(int h_error)
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000545{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000546 PyObject *v;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000547
548#ifdef HAVE_HSTRERROR
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000549 v = Py_BuildValue("(is)", h_error, (char *)hstrerror(h_error));
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000550#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000551 v = Py_BuildValue("(is)", h_error, "host not found");
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000552#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000553 if (v != NULL) {
554 PyErr_SetObject(socket_herror, v);
555 Py_DECREF(v);
556 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000557
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000558 return NULL;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000559}
560
561
562static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000563set_gaierror(int error)
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000564{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000565 PyObject *v;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000566
Martin v. Löwis272cb402002-03-01 08:31:07 +0000567#ifdef EAI_SYSTEM
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000568 /* EAI_SYSTEM is not available on Windows XP. */
569 if (error == EAI_SYSTEM)
570 return set_error();
Martin v. Löwis272cb402002-03-01 08:31:07 +0000571#endif
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000572
Martin v. Löwisf95dd0a2001-08-15 17:14:33 +0000573#ifdef HAVE_GAI_STRERROR
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000574 v = Py_BuildValue("(is)", error, gai_strerror(error));
Martin v. Löwisf95dd0a2001-08-15 17:14:33 +0000575#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000576 v = Py_BuildValue("(is)", error, "getaddrinfo failed");
Martin v. Löwisf95dd0a2001-08-15 17:14:33 +0000577#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000578 if (v != NULL) {
579 PyErr_SetObject(socket_gaierror, v);
580 Py_DECREF(v);
581 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000582
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000583 return NULL;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000584}
585
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000586#ifdef __VMS
587/* Function to send in segments */
588static int
589sendsegmented(int sock_fd, char *buf, int len, int flags)
590{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000591 int n = 0;
592 int remaining = len;
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000593
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000594 while (remaining > 0) {
595 unsigned int segment;
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000596
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000597 segment = (remaining >= SEGMENT_SIZE ? SEGMENT_SIZE : remaining);
598 n = send(sock_fd, buf, segment, flags);
599 if (n < 0) {
600 return n;
601 }
602 remaining -= segment;
603 buf += segment;
604 } /* end while */
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000605
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000606 return len;
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000607}
608#endif
609
Guido van Rossum3eede5a2002-06-07 02:08:35 +0000610/* Function to perform the setting of socket blocking mode
611 internally. block = (1 | 0). */
Guido van Rossum67f7a382002-06-06 21:08:16 +0000612static int
613internal_setblocking(PySocketSockObject *s, int block)
614{
Guido van Rossum67f7a382002-06-06 21:08:16 +0000615#ifndef MS_WINDOWS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000616 int delay_flag;
Guido van Rossum67f7a382002-06-06 21:08:16 +0000617#endif
Antoine Pitroub1c54962010-10-14 15:05:38 +0000618#ifdef SOCK_NONBLOCK
619 if (block)
620 s->sock_type &= (~SOCK_NONBLOCK);
621 else
622 s->sock_type |= SOCK_NONBLOCK;
623#endif
Guido van Rossum67f7a382002-06-06 21:08:16 +0000624
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000625 Py_BEGIN_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +0000626#ifndef MS_WINDOWS
627#if defined(PYOS_OS2) && !defined(PYCC_GCC)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000628 block = !block;
629 ioctl(s->sock_fd, FIONBIO, (caddr_t)&block, sizeof(block));
Martin v. Löwis79acb9e2002-12-06 12:48:53 +0000630#elif defined(__VMS)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000631 block = !block;
632 ioctl(s->sock_fd, FIONBIO, (unsigned int *)&block);
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000633#else /* !PYOS_OS2 && !__VMS */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000634 delay_flag = fcntl(s->sock_fd, F_GETFL, 0);
635 if (block)
636 delay_flag &= (~O_NONBLOCK);
637 else
638 delay_flag |= O_NONBLOCK;
639 fcntl(s->sock_fd, F_SETFL, delay_flag);
Guido van Rossum67f7a382002-06-06 21:08:16 +0000640#endif /* !PYOS_OS2 */
641#else /* MS_WINDOWS */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000642 block = !block;
643 ioctlsocket(s->sock_fd, FIONBIO, (u_long*)&block);
Guido van Rossum67f7a382002-06-06 21:08:16 +0000644#endif /* MS_WINDOWS */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000645 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +0000646
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000647 /* Since these don't return anything */
648 return 1;
Guido van Rossum67f7a382002-06-06 21:08:16 +0000649}
650
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000651/* Do a select()/poll() on the socket, if necessary (sock_timeout > 0).
Guido van Rossum11ba0942002-06-13 15:07:44 +0000652 The argument writing indicates the direction.
Raymond Hettingeref7343c2003-06-29 03:08:05 +0000653 This does not raise an exception; we'll let our caller do that
654 after they've reacquired the interpreter lock.
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000655 Returns 1 on timeout, -1 on error, 0 otherwise. */
Raymond Hettingeref7343c2003-06-29 03:08:05 +0000656static int
Antoine Pitrou3e1fd272010-09-28 21:23:11 +0000657internal_select_ex(PySocketSockObject *s, int writing, double interval)
Guido van Rossum67f7a382002-06-06 21:08:16 +0000658{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000659 int n;
Guido van Rossum11ba0942002-06-13 15:07:44 +0000660
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000661 /* Nothing to do unless we're in timeout mode (not non-blocking) */
662 if (s->sock_timeout <= 0.0)
663 return 0;
Guido van Rossum67f7a382002-06-06 21:08:16 +0000664
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000665 /* Guard against closed socket */
666 if (s->sock_fd < 0)
667 return 0;
Guido van Rossumad654902002-07-19 12:44:59 +0000668
Antoine Pitrou3e1fd272010-09-28 21:23:11 +0000669 /* Handling this condition here simplifies the select loops */
670 if (interval < 0.0)
671 return 1;
672
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000673 /* Prefer poll, if available, since you can poll() any fd
674 * which can't be done with select(). */
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000675#ifdef HAVE_POLL
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000676 {
677 struct pollfd pollfd;
678 int timeout;
Guido van Rossum67f7a382002-06-06 21:08:16 +0000679
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000680 pollfd.fd = s->sock_fd;
681 pollfd.events = writing ? POLLOUT : POLLIN;
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000682
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000683 /* s->sock_timeout is in seconds, timeout in ms */
Antoine Pitrou3e1fd272010-09-28 21:23:11 +0000684 timeout = (int)(interval * 1000 + 0.5);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000685 n = poll(&pollfd, 1, timeout);
686 }
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000687#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000688 {
689 /* Construct the arguments to select */
690 fd_set fds;
691 struct timeval tv;
Antoine Pitrou3e1fd272010-09-28 21:23:11 +0000692 tv.tv_sec = (int)interval;
693 tv.tv_usec = (int)((interval - tv.tv_sec) * 1e6);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000694 FD_ZERO(&fds);
695 FD_SET(s->sock_fd, &fds);
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000696
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000697 /* See if the socket is ready */
698 if (writing)
Antoine Pitrou19467d22010-08-17 19:33:30 +0000699 n = select(Py_SAFE_DOWNCAST(s->sock_fd+1, SOCKET_T, int),
700 NULL, &fds, NULL, &tv);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000701 else
Antoine Pitrou19467d22010-08-17 19:33:30 +0000702 n = select(Py_SAFE_DOWNCAST(s->sock_fd+1, SOCKET_T, int),
703 &fds, NULL, NULL, &tv);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000704 }
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000705#endif
Guido van Rossum39eb8fa2007-11-16 01:24:05 +0000706
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000707 if (n < 0)
708 return -1;
709 if (n == 0)
710 return 1;
711 return 0;
Guido van Rossum67f7a382002-06-06 21:08:16 +0000712}
713
Antoine Pitrou3e1fd272010-09-28 21:23:11 +0000714static int
715internal_select(PySocketSockObject *s, int writing)
716{
717 return internal_select_ex(s, writing, s->sock_timeout);
718}
719
720/*
721 Two macros for automatic retry of select() in case of false positives
722 (for example, select() could indicate a socket is ready for reading
723 but the data then discarded by the OS because of a wrong checksum).
724 Here is an example of use:
725
726 BEGIN_SELECT_LOOP(s)
727 Py_BEGIN_ALLOW_THREADS
728 timeout = internal_select_ex(s, 0, interval);
729 if (!timeout)
730 outlen = recv(s->sock_fd, cbuf, len, flags);
731 Py_END_ALLOW_THREADS
732 if (timeout == 1) {
733 PyErr_SetString(socket_timeout, "timed out");
734 return -1;
735 }
736 END_SELECT_LOOP(s)
737*/
738
739#define BEGIN_SELECT_LOOP(s) \
740 { \
741 _PyTime_timeval now, deadline = {0, 0}; \
742 double interval = s->sock_timeout; \
743 int has_timeout = s->sock_timeout > 0.0; \
744 if (has_timeout) { \
745 _PyTime_gettimeofday(&now); \
746 deadline = now; \
747 _PyTime_ADD_SECONDS(deadline, s->sock_timeout); \
748 } \
749 while (1) { \
750 errno = 0; \
751
752#define END_SELECT_LOOP(s) \
753 if (!has_timeout || \
754 (!CHECK_ERRNO(EWOULDBLOCK) && !CHECK_ERRNO(EAGAIN))) \
755 break; \
756 _PyTime_gettimeofday(&now); \
757 interval = _PyTime_INTERVAL(now, deadline); \
758 } \
759 } \
760
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000761/* Initialize a new socket object. */
762
Tim Petersa12b4cf2002-07-18 22:38:44 +0000763static double defaulttimeout = -1.0; /* Default timeout for new sockets */
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +0000764
Martin v. Löwis1a214512008-06-11 05:26:20 +0000765static void
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000766init_sockobject(PySocketSockObject *s,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000767 SOCKET_T fd, int family, int type, int proto)
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000768{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000769 s->sock_fd = fd;
770 s->sock_family = family;
771 s->sock_type = type;
772 s->sock_proto = proto;
Guido van Rossum67f7a382002-06-06 21:08:16 +0000773
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000774 s->errorhandler = &set_error;
Antoine Pitroub1c54962010-10-14 15:05:38 +0000775#ifdef SOCK_NONBLOCK
776 if (type & SOCK_NONBLOCK)
777 s->sock_timeout = 0.0;
778 else
779#endif
780 {
781 s->sock_timeout = defaulttimeout;
782 if (defaulttimeout >= 0.0)
783 internal_setblocking(s, 0);
784 }
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +0000785
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000786}
787
788
Guido van Rossum30a685f1991-06-27 15:51:29 +0000789/* Create a new socket object.
790 This just creates the object and initializes it.
791 If the creation fails, return NULL and set an exception (implicit
792 in NEWOBJ()). */
793
Guido van Rossum73624e91994-10-10 17:59:00 +0000794static PySocketSockObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000795new_sockobject(SOCKET_T fd, int family, int type, int proto)
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000796{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000797 PySocketSockObject *s;
798 s = (PySocketSockObject *)
799 PyType_GenericNew(&sock_type, NULL, NULL);
800 if (s != NULL)
801 init_sockobject(s, fd, family, type, proto);
802 return s;
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000803}
804
Guido van Rossum30a685f1991-06-27 15:51:29 +0000805
Guido van Rossum48a680c2001-03-02 06:34:14 +0000806/* Lock to allow python interpreter to continue, but only allow one
Just van Rossum1040d2c2003-05-09 07:53:18 +0000807 thread to be in gethostbyname or getaddrinfo */
808#if defined(USE_GETHOSTBYNAME_LOCK) || defined(USE_GETADDRINFO_LOCK)
809PyThread_type_lock netdb_lock;
Guido van Rossum4f199ea1998-04-09 20:56:35 +0000810#endif
811
812
Guido van Rossum30a685f1991-06-27 15:51:29 +0000813/* Convert a string specifying a host name or one of a few symbolic
814 names to a numeric IP address. This usually calls gethostbyname()
815 to do the work; the names "" and "<broadcast>" are special.
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000816 Return the length (IPv4 should be 4 bytes), or negative if
Guido van Rossum30a685f1991-06-27 15:51:29 +0000817 an error occurred; then an exception is raised. */
818
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000819static int
Martin v. Löwisddc6f472002-07-28 16:10:31 +0000820setipaddr(char *name, struct sockaddr *addr_ret, size_t addr_ret_size, int af)
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000821{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000822 struct addrinfo hints, *res;
823 int error;
824 int d1, d2, d3, d4;
825 char ch;
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000826
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000827 memset((void *) addr_ret, '\0', sizeof(*addr_ret));
828 if (name[0] == '\0') {
829 int siz;
830 memset(&hints, 0, sizeof(hints));
831 hints.ai_family = af;
832 hints.ai_socktype = SOCK_DGRAM; /*dummy*/
833 hints.ai_flags = AI_PASSIVE;
834 Py_BEGIN_ALLOW_THREADS
835 ACQUIRE_GETADDRINFO_LOCK
836 error = getaddrinfo(NULL, "0", &hints, &res);
837 Py_END_ALLOW_THREADS
838 /* We assume that those thread-unsafe getaddrinfo() versions
839 *are* safe regarding their return value, ie. that a
840 subsequent call to getaddrinfo() does not destroy the
841 outcome of the first call. */
842 RELEASE_GETADDRINFO_LOCK
843 if (error) {
844 set_gaierror(error);
845 return -1;
846 }
847 switch (res->ai_family) {
848 case AF_INET:
849 siz = 4;
850 break;
Martin v. Löwis44ddbde2001-12-02 10:15:37 +0000851#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000852 case AF_INET6:
853 siz = 16;
854 break;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000855#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000856 default:
857 freeaddrinfo(res);
858 PyErr_SetString(socket_error,
859 "unsupported address family");
860 return -1;
861 }
862 if (res->ai_next) {
863 freeaddrinfo(res);
864 PyErr_SetString(socket_error,
865 "wildcard resolved to multiple address");
866 return -1;
867 }
868 if (res->ai_addrlen < addr_ret_size)
869 addr_ret_size = res->ai_addrlen;
870 memcpy(addr_ret, res->ai_addr, addr_ret_size);
871 freeaddrinfo(res);
872 return siz;
873 }
874 if (name[0] == '<' && strcmp(name, "<broadcast>") == 0) {
875 struct sockaddr_in *sin;
876 if (af != AF_INET && af != AF_UNSPEC) {
877 PyErr_SetString(socket_error,
878 "address family mismatched");
879 return -1;
880 }
881 sin = (struct sockaddr_in *)addr_ret;
882 memset((void *) sin, '\0', sizeof(*sin));
883 sin->sin_family = AF_INET;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000884#ifdef HAVE_SOCKADDR_SA_LEN
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000885 sin->sin_len = sizeof(*sin);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000886#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000887 sin->sin_addr.s_addr = INADDR_BROADCAST;
888 return sizeof(sin->sin_addr);
889 }
890 if (sscanf(name, "%d.%d.%d.%d%c", &d1, &d2, &d3, &d4, &ch) == 4 &&
891 0 <= d1 && d1 <= 255 && 0 <= d2 && d2 <= 255 &&
892 0 <= d3 && d3 <= 255 && 0 <= d4 && d4 <= 255) {
893 struct sockaddr_in *sin;
894 sin = (struct sockaddr_in *)addr_ret;
895 sin->sin_addr.s_addr = htonl(
896 ((long) d1 << 24) | ((long) d2 << 16) |
897 ((long) d3 << 8) | ((long) d4 << 0));
898 sin->sin_family = AF_INET;
Anthony Baxter0e85f9d2003-05-02 15:40:46 +0000899#ifdef HAVE_SOCKADDR_SA_LEN
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000900 sin->sin_len = sizeof(*sin);
Anthony Baxter0e85f9d2003-05-02 15:40:46 +0000901#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000902 return 4;
903 }
904 memset(&hints, 0, sizeof(hints));
905 hints.ai_family = af;
906 Py_BEGIN_ALLOW_THREADS
907 ACQUIRE_GETADDRINFO_LOCK
908 error = getaddrinfo(name, NULL, &hints, &res);
Martin v. Löwis7c4b5fa2001-10-25 09:04:03 +0000909#if defined(__digital__) && defined(__unix__)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000910 if (error == EAI_NONAME && af == AF_UNSPEC) {
911 /* On Tru64 V5.1, numeric-to-addr conversion fails
912 if no address family is given. Assume IPv4 for now.*/
913 hints.ai_family = AF_INET;
914 error = getaddrinfo(name, NULL, &hints, &res);
915 }
Martin v. Löwis7c4b5fa2001-10-25 09:04:03 +0000916#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000917 Py_END_ALLOW_THREADS
918 RELEASE_GETADDRINFO_LOCK /* see comment in setipaddr() */
919 if (error) {
920 set_gaierror(error);
921 return -1;
922 }
923 if (res->ai_addrlen < addr_ret_size)
924 addr_ret_size = res->ai_addrlen;
925 memcpy((char *) addr_ret, res->ai_addr, addr_ret_size);
926 freeaddrinfo(res);
927 switch (addr_ret->sa_family) {
928 case AF_INET:
929 return 4;
Martin v. Löwis44ddbde2001-12-02 10:15:37 +0000930#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000931 case AF_INET6:
932 return 16;
Guido van Rossum955becc1999-03-22 20:14:53 +0000933#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000934 default:
935 PyErr_SetString(socket_error, "unknown address family");
936 return -1;
937 }
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000938}
939
Guido van Rossum30a685f1991-06-27 15:51:29 +0000940
Guido van Rossum30a685f1991-06-27 15:51:29 +0000941/* Create a string object representing an IP address.
942 This is always a string of the form 'dd.dd.dd.dd' (with variable
943 size numbers). */
944
Guido van Rossum73624e91994-10-10 17:59:00 +0000945static PyObject *
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000946makeipaddr(struct sockaddr *addr, int addrlen)
Guido van Rossum30a685f1991-06-27 15:51:29 +0000947{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000948 char buf[NI_MAXHOST];
949 int error;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000950
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000951 error = getnameinfo(addr, addrlen, buf, sizeof(buf), NULL, 0,
952 NI_NUMERICHOST);
953 if (error) {
954 set_gaierror(error);
955 return NULL;
956 }
957 return PyUnicode_FromString(buf);
Guido van Rossum30a685f1991-06-27 15:51:29 +0000958}
959
960
Martin v. Löwis558d9bf2004-06-03 09:24:42 +0000961#ifdef USE_BLUETOOTH
962/* Convert a string representation of a Bluetooth address into a numeric
963 address. Returns the length (6), or raises an exception and returns -1 if
964 an error occurred. */
965
966static int
967setbdaddr(char *name, bdaddr_t *bdaddr)
968{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000969 unsigned int b0, b1, b2, b3, b4, b5;
970 char ch;
971 int n;
Martin v. Löwis558d9bf2004-06-03 09:24:42 +0000972
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000973 n = sscanf(name, "%X:%X:%X:%X:%X:%X%c",
974 &b5, &b4, &b3, &b2, &b1, &b0, &ch);
975 if (n == 6 && (b0 | b1 | b2 | b3 | b4 | b5) < 256) {
976 bdaddr->b[0] = b0;
977 bdaddr->b[1] = b1;
978 bdaddr->b[2] = b2;
979 bdaddr->b[3] = b3;
980 bdaddr->b[4] = b4;
981 bdaddr->b[5] = b5;
982 return 6;
983 } else {
984 PyErr_SetString(socket_error, "bad bluetooth address");
985 return -1;
986 }
Martin v. Löwis558d9bf2004-06-03 09:24:42 +0000987}
988
989/* Create a string representation of the Bluetooth address. This is always a
990 string of the form 'XX:XX:XX:XX:XX:XX' where XX is a two digit hexadecimal
991 value (zero padded if necessary). */
992
993static PyObject *
994makebdaddr(bdaddr_t *bdaddr)
995{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000996 char buf[(6 * 2) + 5 + 1];
Martin v. Löwis558d9bf2004-06-03 09:24:42 +0000997
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000998 sprintf(buf, "%02X:%02X:%02X:%02X:%02X:%02X",
999 bdaddr->b[5], bdaddr->b[4], bdaddr->b[3],
1000 bdaddr->b[2], bdaddr->b[1], bdaddr->b[0]);
1001 return PyUnicode_FromString(buf);
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001002}
1003#endif
1004
1005
Guido van Rossum30a685f1991-06-27 15:51:29 +00001006/* Create an object representing the given socket address,
1007 suitable for passing it back to bind(), connect() etc.
1008 The family field of the sockaddr structure is inspected
1009 to determine what kind of address it really is. */
1010
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001011/*ARGSUSED*/
Guido van Rossum73624e91994-10-10 17:59:00 +00001012static PyObject *
Antoine Pitrou19467d22010-08-17 19:33:30 +00001013makesockaddr(SOCKET_T sockfd, struct sockaddr *addr, size_t addrlen, int proto)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001014{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001015 if (addrlen == 0) {
1016 /* No address -- may be recvfrom() from known socket */
1017 Py_INCREF(Py_None);
1018 return Py_None;
1019 }
Guido van Rossum25bec8c1992-08-05 19:00:45 +00001020
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001021 switch (addr->sa_family) {
Guido van Rossum30a685f1991-06-27 15:51:29 +00001022
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001023 case AF_INET:
1024 {
1025 struct sockaddr_in *a;
1026 PyObject *addrobj = makeipaddr(addr, sizeof(*a));
1027 PyObject *ret = NULL;
1028 if (addrobj) {
1029 a = (struct sockaddr_in *)addr;
1030 ret = Py_BuildValue("Oi", addrobj, ntohs(a->sin_port));
1031 Py_DECREF(addrobj);
1032 }
1033 return ret;
1034 }
Guido van Rossum30a685f1991-06-27 15:51:29 +00001035
Andrew MacIntyred12dfbb2004-04-04 07:13:49 +00001036#if defined(AF_UNIX)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001037 case AF_UNIX:
1038 {
1039 struct sockaddr_un *a = (struct sockaddr_un *) addr;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001040#ifdef linux
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001041 if (a->sun_path[0] == 0) { /* Linux abstract namespace */
1042 addrlen -= offsetof(struct sockaddr_un, sun_path);
1043 return PyBytes_FromStringAndSize(a->sun_path, addrlen);
1044 }
1045 else
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001046#endif /* linux */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001047 {
1048 /* regular NULL-terminated string */
1049 return PyUnicode_FromString(a->sun_path);
1050 }
1051 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00001052#endif /* AF_UNIX */
1053
Martin v. Löwis11017b12006-01-14 18:12:57 +00001054#if defined(AF_NETLINK)
1055 case AF_NETLINK:
1056 {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001057 struct sockaddr_nl *a = (struct sockaddr_nl *) addr;
1058 return Py_BuildValue("II", a->nl_pid, a->nl_groups);
Martin v. Löwis11017b12006-01-14 18:12:57 +00001059 }
1060#endif /* AF_NETLINK */
1061
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00001062#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001063 case AF_INET6:
1064 {
1065 struct sockaddr_in6 *a;
1066 PyObject *addrobj = makeipaddr(addr, sizeof(*a));
1067 PyObject *ret = NULL;
1068 if (addrobj) {
1069 a = (struct sockaddr_in6 *)addr;
1070 ret = Py_BuildValue("Oiii",
1071 addrobj,
1072 ntohs(a->sin6_port),
1073 a->sin6_flowinfo,
1074 a->sin6_scope_id);
1075 Py_DECREF(addrobj);
1076 }
1077 return ret;
1078 }
Jeremy Hylton22308652001-02-02 03:23:09 +00001079#endif
Guido van Rossum30a685f1991-06-27 15:51:29 +00001080
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001081#ifdef USE_BLUETOOTH
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001082 case AF_BLUETOOTH:
1083 switch (proto) {
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001084
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001085 case BTPROTO_L2CAP:
1086 {
1087 struct sockaddr_l2 *a = (struct sockaddr_l2 *) addr;
1088 PyObject *addrobj = makebdaddr(&_BT_L2_MEMB(a, bdaddr));
1089 PyObject *ret = NULL;
1090 if (addrobj) {
1091 ret = Py_BuildValue("Oi",
1092 addrobj,
1093 _BT_L2_MEMB(a, psm));
1094 Py_DECREF(addrobj);
1095 }
1096 return ret;
1097 }
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001098
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001099 case BTPROTO_RFCOMM:
1100 {
1101 struct sockaddr_rc *a = (struct sockaddr_rc *) addr;
1102 PyObject *addrobj = makebdaddr(&_BT_RC_MEMB(a, bdaddr));
1103 PyObject *ret = NULL;
1104 if (addrobj) {
1105 ret = Py_BuildValue("Oi",
1106 addrobj,
1107 _BT_RC_MEMB(a, channel));
1108 Py_DECREF(addrobj);
1109 }
1110 return ret;
1111 }
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001112
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001113 case BTPROTO_HCI:
1114 {
1115 struct sockaddr_hci *a = (struct sockaddr_hci *) addr;
1116 PyObject *ret = NULL;
1117 ret = Py_BuildValue("i", _BT_HCI_MEMB(a, dev));
1118 return ret;
1119 }
Thomas Wouterscf297e42007-02-23 15:07:44 +00001120
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001121#if !defined(__FreeBSD__)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001122 case BTPROTO_SCO:
1123 {
1124 struct sockaddr_sco *a = (struct sockaddr_sco *) addr;
1125 return makebdaddr(&_BT_SCO_MEMB(a, bdaddr));
1126 }
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001127#endif
1128
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001129 default:
1130 PyErr_SetString(PyExc_ValueError,
1131 "Unknown Bluetooth protocol");
1132 return NULL;
1133 }
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001134#endif
1135
Martin v. Löwis1ba3fd52001-08-10 20:29:40 +00001136#ifdef HAVE_NETPACKET_PACKET_H
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001137 case AF_PACKET:
1138 {
1139 struct sockaddr_ll *a = (struct sockaddr_ll *)addr;
1140 char *ifname = "";
1141 struct ifreq ifr;
1142 /* need to look up interface name give index */
1143 if (a->sll_ifindex) {
1144 ifr.ifr_ifindex = a->sll_ifindex;
1145 if (ioctl(sockfd, SIOCGIFNAME, &ifr) == 0)
1146 ifname = ifr.ifr_name;
1147 }
1148 return Py_BuildValue("shbhy#",
1149 ifname,
1150 ntohs(a->sll_protocol),
1151 a->sll_pkttype,
1152 a->sll_hatype,
1153 a->sll_addr,
1154 a->sll_halen);
1155 }
Jeremy Hylton22308652001-02-02 03:23:09 +00001156#endif
Guido van Rossum48a680c2001-03-02 06:34:14 +00001157
Christian Heimes043d6f62008-01-07 17:19:16 +00001158#ifdef HAVE_LINUX_TIPC_H
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001159 case AF_TIPC:
1160 {
1161 struct sockaddr_tipc *a = (struct sockaddr_tipc *) addr;
1162 if (a->addrtype == TIPC_ADDR_NAMESEQ) {
1163 return Py_BuildValue("IIIII",
1164 a->addrtype,
1165 a->addr.nameseq.type,
1166 a->addr.nameseq.lower,
1167 a->addr.nameseq.upper,
1168 a->scope);
1169 } else if (a->addrtype == TIPC_ADDR_NAME) {
1170 return Py_BuildValue("IIIII",
1171 a->addrtype,
1172 a->addr.name.name.type,
1173 a->addr.name.name.instance,
1174 a->addr.name.name.instance,
1175 a->scope);
1176 } else if (a->addrtype == TIPC_ADDR_ID) {
1177 return Py_BuildValue("IIIII",
1178 a->addrtype,
1179 a->addr.id.node,
1180 a->addr.id.ref,
1181 0,
1182 a->scope);
1183 } else {
1184 PyErr_SetString(PyExc_ValueError,
1185 "Invalid address type");
1186 return NULL;
1187 }
1188 }
Christian Heimes043d6f62008-01-07 17:19:16 +00001189#endif
1190
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001191 /* More cases here... */
Guido van Rossum30a685f1991-06-27 15:51:29 +00001192
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001193 default:
1194 /* If we don't know the address family, don't raise an
1195 exception -- return it as an (int, bytes) tuple. */
1196 return Py_BuildValue("iy#",
1197 addr->sa_family,
1198 addr->sa_data,
1199 sizeof(addr->sa_data));
Guido van Rossum25bec8c1992-08-05 19:00:45 +00001200
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001201 }
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001202}
1203
Guido van Rossum30a685f1991-06-27 15:51:29 +00001204
1205/* Parse a socket address argument according to the socket object's
1206 address family. Return 1 if the address was in the proper format,
1207 0 of not. The address is returned through addr_ret, its length
1208 through len_ret. */
1209
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001210static int
Guido van Rossum48a680c2001-03-02 06:34:14 +00001211getsockaddrarg(PySocketSockObject *s, PyObject *args,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001212 struct sockaddr *addr_ret, int *len_ret)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001213{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001214 switch (s->sock_family) {
Guido van Rossum30a685f1991-06-27 15:51:29 +00001215
Andrew MacIntyred12dfbb2004-04-04 07:13:49 +00001216#if defined(AF_UNIX)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001217 case AF_UNIX:
1218 {
1219 struct sockaddr_un* addr;
1220 char *path;
1221 int len;
1222 if (!PyArg_Parse(args, "s#", &path, &len))
1223 return 0;
Thomas Wouters89f507f2006-12-13 04:49:30 +00001224
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001225 addr = (struct sockaddr_un*)addr_ret;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001226#ifdef linux
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001227 if (len > 0 && path[0] == 0) {
1228 /* Linux abstract namespace extension */
1229 if (len > sizeof addr->sun_path) {
1230 PyErr_SetString(socket_error,
1231 "AF_UNIX path too long");
1232 return 0;
1233 }
1234 }
1235 else
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001236#endif /* linux */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001237 {
1238 /* regular NULL-terminated string */
1239 if (len >= sizeof addr->sun_path) {
1240 PyErr_SetString(socket_error,
1241 "AF_UNIX path too long");
1242 return 0;
1243 }
1244 addr->sun_path[len] = 0;
1245 }
1246 addr->sun_family = s->sock_family;
1247 memcpy(addr->sun_path, path, len);
Andrew MacIntyredaedf212004-04-11 12:03:57 +00001248#if defined(PYOS_OS2)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001249 *len_ret = sizeof(*addr);
Andrew MacIntyredaedf212004-04-11 12:03:57 +00001250#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001251 *len_ret = len + offsetof(struct sockaddr_un, sun_path);
Andrew MacIntyredaedf212004-04-11 12:03:57 +00001252#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001253 return 1;
1254 }
Guido van Rossumb6775db1994-08-01 11:34:53 +00001255#endif /* AF_UNIX */
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001256
Martin v. Löwis11017b12006-01-14 18:12:57 +00001257#if defined(AF_NETLINK)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001258 case AF_NETLINK:
1259 {
1260 struct sockaddr_nl* addr;
1261 int pid, groups;
1262 addr = (struct sockaddr_nl *)addr_ret;
1263 if (!PyTuple_Check(args)) {
1264 PyErr_Format(
1265 PyExc_TypeError,
1266 "getsockaddrarg: "
1267 "AF_NETLINK address must be tuple, not %.500s",
1268 Py_TYPE(args)->tp_name);
1269 return 0;
1270 }
1271 if (!PyArg_ParseTuple(args, "II:getsockaddrarg", &pid, &groups))
1272 return 0;
1273 addr->nl_family = AF_NETLINK;
1274 addr->nl_pid = pid;
1275 addr->nl_groups = groups;
1276 *len_ret = sizeof(*addr);
1277 return 1;
1278 }
Martin v. Löwis11017b12006-01-14 18:12:57 +00001279#endif
1280
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001281 case AF_INET:
1282 {
1283 struct sockaddr_in* addr;
1284 char *host;
1285 int port, result;
1286 if (!PyTuple_Check(args)) {
1287 PyErr_Format(
1288 PyExc_TypeError,
1289 "getsockaddrarg: "
1290 "AF_INET address must be tuple, not %.500s",
1291 Py_TYPE(args)->tp_name);
1292 return 0;
1293 }
1294 if (!PyArg_ParseTuple(args, "eti:getsockaddrarg",
1295 "idna", &host, &port))
1296 return 0;
1297 addr=(struct sockaddr_in*)addr_ret;
1298 result = setipaddr(host, (struct sockaddr *)addr,
1299 sizeof(*addr), AF_INET);
1300 PyMem_Free(host);
1301 if (result < 0)
1302 return 0;
1303 if (port < 0 || port > 0xffff) {
1304 PyErr_SetString(
1305 PyExc_OverflowError,
1306 "getsockaddrarg: port must be 0-65535.");
1307 return 0;
1308 }
1309 addr->sin_family = AF_INET;
1310 addr->sin_port = htons((short)port);
1311 *len_ret = sizeof *addr;
1312 return 1;
1313 }
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001314
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00001315#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001316 case AF_INET6:
1317 {
1318 struct sockaddr_in6* addr;
1319 char *host;
1320 int port, flowinfo, scope_id, result;
1321 flowinfo = scope_id = 0;
1322 if (!PyTuple_Check(args)) {
1323 PyErr_Format(
1324 PyExc_TypeError,
1325 "getsockaddrarg: "
1326 "AF_INET6 address must be tuple, not %.500s",
1327 Py_TYPE(args)->tp_name);
1328 return 0;
1329 }
1330 if (!PyArg_ParseTuple(args, "eti|ii",
1331 "idna", &host, &port, &flowinfo,
1332 &scope_id)) {
1333 return 0;
1334 }
1335 addr = (struct sockaddr_in6*)addr_ret;
1336 result = setipaddr(host, (struct sockaddr *)addr,
1337 sizeof(*addr), AF_INET6);
1338 PyMem_Free(host);
1339 if (result < 0)
1340 return 0;
1341 if (port < 0 || port > 0xffff) {
1342 PyErr_SetString(
1343 PyExc_OverflowError,
1344 "getsockaddrarg: port must be 0-65535.");
1345 return 0;
1346 }
1347 addr->sin6_family = s->sock_family;
1348 addr->sin6_port = htons((short)port);
1349 addr->sin6_flowinfo = flowinfo;
1350 addr->sin6_scope_id = scope_id;
1351 *len_ret = sizeof *addr;
1352 return 1;
1353 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00001354#endif
1355
Hye-Shik Chang81268602004-02-02 06:05:24 +00001356#ifdef USE_BLUETOOTH
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001357 case AF_BLUETOOTH:
1358 {
1359 switch (s->sock_proto) {
1360 case BTPROTO_L2CAP:
1361 {
1362 struct sockaddr_l2 *addr;
1363 char *straddr;
Martin v. Löwis12af0482004-01-31 12:34:17 +00001364
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001365 addr = (struct sockaddr_l2 *)addr_ret;
1366 memset(addr, 0, sizeof(struct sockaddr_l2));
1367 _BT_L2_MEMB(addr, family) = AF_BLUETOOTH;
1368 if (!PyArg_ParseTuple(args, "si", &straddr,
1369 &_BT_L2_MEMB(addr, psm))) {
1370 PyErr_SetString(socket_error, "getsockaddrarg: "
1371 "wrong format");
1372 return 0;
1373 }
1374 if (setbdaddr(straddr, &_BT_L2_MEMB(addr, bdaddr)) < 0)
1375 return 0;
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001376
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001377 *len_ret = sizeof *addr;
1378 return 1;
1379 }
1380 case BTPROTO_RFCOMM:
1381 {
1382 struct sockaddr_rc *addr;
1383 char *straddr;
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001384
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001385 addr = (struct sockaddr_rc *)addr_ret;
1386 _BT_RC_MEMB(addr, family) = AF_BLUETOOTH;
1387 if (!PyArg_ParseTuple(args, "si", &straddr,
1388 &_BT_RC_MEMB(addr, channel))) {
1389 PyErr_SetString(socket_error, "getsockaddrarg: "
1390 "wrong format");
1391 return 0;
1392 }
1393 if (setbdaddr(straddr, &_BT_RC_MEMB(addr, bdaddr)) < 0)
1394 return 0;
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001395
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001396 *len_ret = sizeof *addr;
1397 return 1;
1398 }
1399 case BTPROTO_HCI:
1400 {
1401 struct sockaddr_hci *addr = (struct sockaddr_hci *)addr_ret;
1402 _BT_HCI_MEMB(addr, family) = AF_BLUETOOTH;
1403 if (!PyArg_ParseTuple(args, "i", &_BT_HCI_MEMB(addr, dev))) {
1404 PyErr_SetString(socket_error, "getsockaddrarg: "
1405 "wrong format");
1406 return 0;
1407 }
1408 *len_ret = sizeof *addr;
1409 return 1;
1410 }
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001411#if !defined(__FreeBSD__)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001412 case BTPROTO_SCO:
1413 {
1414 struct sockaddr_sco *addr;
1415 char *straddr;
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001416
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001417 addr = (struct sockaddr_sco *)addr_ret;
1418 _BT_SCO_MEMB(addr, family) = AF_BLUETOOTH;
1419 if (!PyBytes_Check(args)) {
1420 PyErr_SetString(socket_error, "getsockaddrarg: "
1421 "wrong format");
1422 return 0;
1423 }
1424 straddr = PyBytes_AS_STRING(args);
1425 if (setbdaddr(straddr, &_BT_SCO_MEMB(addr, bdaddr)) < 0)
1426 return 0;
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001427
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001428 *len_ret = sizeof *addr;
1429 return 1;
1430 }
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001431#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001432 default:
1433 PyErr_SetString(socket_error, "getsockaddrarg: unknown Bluetooth protocol");
1434 return 0;
1435 }
1436 }
Martin v. Löwis12af0482004-01-31 12:34:17 +00001437#endif
1438
Martin v. Löwis1ba3fd52001-08-10 20:29:40 +00001439#ifdef HAVE_NETPACKET_PACKET_H
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001440 case AF_PACKET:
1441 {
1442 struct sockaddr_ll* addr;
1443 struct ifreq ifr;
1444 char *interfaceName;
1445 int protoNumber;
1446 int hatype = 0;
1447 int pkttype = 0;
1448 char *haddr = NULL;
1449 unsigned int halen = 0;
Guido van Rossum48a680c2001-03-02 06:34:14 +00001450
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001451 if (!PyTuple_Check(args)) {
1452 PyErr_Format(
1453 PyExc_TypeError,
1454 "getsockaddrarg: "
1455 "AF_PACKET address must be tuple, not %.500s",
1456 Py_TYPE(args)->tp_name);
1457 return 0;
1458 }
1459 if (!PyArg_ParseTuple(args, "si|iiy#", &interfaceName,
1460 &protoNumber, &pkttype, &hatype,
1461 &haddr, &halen))
1462 return 0;
1463 strncpy(ifr.ifr_name, interfaceName, sizeof(ifr.ifr_name));
1464 ifr.ifr_name[(sizeof(ifr.ifr_name))-1] = '\0';
1465 if (ioctl(s->sock_fd, SIOCGIFINDEX, &ifr) < 0) {
1466 s->errorhandler();
1467 return 0;
1468 }
1469 if (halen > 8) {
1470 PyErr_SetString(PyExc_ValueError,
1471 "Hardware address must be 8 bytes or less");
1472 return 0;
1473 }
1474 if (protoNumber < 0 || protoNumber > 0xffff) {
1475 PyErr_SetString(
1476 PyExc_OverflowError,
1477 "getsockaddrarg: protoNumber must be 0-65535.");
1478 return 0;
1479 }
1480 addr = (struct sockaddr_ll*)addr_ret;
1481 addr->sll_family = AF_PACKET;
1482 addr->sll_protocol = htons((short)protoNumber);
1483 addr->sll_ifindex = ifr.ifr_ifindex;
1484 addr->sll_pkttype = pkttype;
1485 addr->sll_hatype = hatype;
1486 if (halen != 0) {
1487 memcpy(&addr->sll_addr, haddr, halen);
1488 }
1489 addr->sll_halen = halen;
1490 *len_ret = sizeof *addr;
1491 return 1;
1492 }
Guido van Rossum48a680c2001-03-02 06:34:14 +00001493#endif
1494
Christian Heimes043d6f62008-01-07 17:19:16 +00001495#ifdef HAVE_LINUX_TIPC_H
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001496 case AF_TIPC:
1497 {
1498 unsigned int atype, v1, v2, v3;
1499 unsigned int scope = TIPC_CLUSTER_SCOPE;
1500 struct sockaddr_tipc *addr;
Christian Heimes043d6f62008-01-07 17:19:16 +00001501
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001502 if (!PyTuple_Check(args)) {
1503 PyErr_Format(
1504 PyExc_TypeError,
1505 "getsockaddrarg: "
1506 "AF_TIPC address must be tuple, not %.500s",
1507 Py_TYPE(args)->tp_name);
1508 return 0;
1509 }
Christian Heimes043d6f62008-01-07 17:19:16 +00001510
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001511 if (!PyArg_ParseTuple(args,
1512 "IIII|I;Invalid TIPC address format",
1513 &atype, &v1, &v2, &v3, &scope))
1514 return 0;
Christian Heimes043d6f62008-01-07 17:19:16 +00001515
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001516 addr = (struct sockaddr_tipc *) addr_ret;
1517 memset(addr, 0, sizeof(struct sockaddr_tipc));
Christian Heimes043d6f62008-01-07 17:19:16 +00001518
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001519 addr->family = AF_TIPC;
1520 addr->scope = scope;
1521 addr->addrtype = atype;
Christian Heimes043d6f62008-01-07 17:19:16 +00001522
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001523 if (atype == TIPC_ADDR_NAMESEQ) {
1524 addr->addr.nameseq.type = v1;
1525 addr->addr.nameseq.lower = v2;
1526 addr->addr.nameseq.upper = v3;
1527 } else if (atype == TIPC_ADDR_NAME) {
1528 addr->addr.name.name.type = v1;
1529 addr->addr.name.name.instance = v2;
1530 } else if (atype == TIPC_ADDR_ID) {
1531 addr->addr.id.node = v1;
1532 addr->addr.id.ref = v2;
1533 } else {
1534 /* Shouldn't happen */
1535 PyErr_SetString(PyExc_TypeError, "Invalid address type");
1536 return 0;
1537 }
Christian Heimes043d6f62008-01-07 17:19:16 +00001538
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001539 *len_ret = sizeof(*addr);
Christian Heimes043d6f62008-01-07 17:19:16 +00001540
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001541 return 1;
1542 }
Christian Heimes043d6f62008-01-07 17:19:16 +00001543#endif
1544
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001545 /* More cases here... */
Guido van Rossum30a685f1991-06-27 15:51:29 +00001546
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001547 default:
1548 PyErr_SetString(socket_error, "getsockaddrarg: bad family");
1549 return 0;
Guido van Rossum30a685f1991-06-27 15:51:29 +00001550
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001551 }
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001552}
1553
Guido van Rossum30a685f1991-06-27 15:51:29 +00001554
Guido van Rossum48a680c2001-03-02 06:34:14 +00001555/* Get the address length according to the socket object's address family.
Guido van Rossum710e1df1992-06-12 10:39:36 +00001556 Return 1 if the family is known, 0 otherwise. The length is returned
1557 through len_ret. */
1558
1559static int
Peter Schneider-Kamp618e25d2000-07-11 23:00:12 +00001560getsockaddrlen(PySocketSockObject *s, socklen_t *len_ret)
Guido van Rossum710e1df1992-06-12 10:39:36 +00001561{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001562 switch (s->sock_family) {
Guido van Rossum710e1df1992-06-12 10:39:36 +00001563
Andrew MacIntyred12dfbb2004-04-04 07:13:49 +00001564#if defined(AF_UNIX)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001565 case AF_UNIX:
1566 {
1567 *len_ret = sizeof (struct sockaddr_un);
1568 return 1;
1569 }
Guido van Rossumb6775db1994-08-01 11:34:53 +00001570#endif /* AF_UNIX */
Martin v. Löwis11017b12006-01-14 18:12:57 +00001571#if defined(AF_NETLINK)
1572 case AF_NETLINK:
1573 {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001574 *len_ret = sizeof (struct sockaddr_nl);
1575 return 1;
Martin v. Löwis11017b12006-01-14 18:12:57 +00001576 }
1577#endif
Guido van Rossum710e1df1992-06-12 10:39:36 +00001578
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001579 case AF_INET:
1580 {
1581 *len_ret = sizeof (struct sockaddr_in);
1582 return 1;
1583 }
Guido van Rossum710e1df1992-06-12 10:39:36 +00001584
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00001585#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001586 case AF_INET6:
1587 {
1588 *len_ret = sizeof (struct sockaddr_in6);
1589 return 1;
1590 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00001591#endif
1592
Hye-Shik Chang81268602004-02-02 06:05:24 +00001593#ifdef USE_BLUETOOTH
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001594 case AF_BLUETOOTH:
1595 {
1596 switch(s->sock_proto)
1597 {
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001598
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001599 case BTPROTO_L2CAP:
1600 *len_ret = sizeof (struct sockaddr_l2);
1601 return 1;
1602 case BTPROTO_RFCOMM:
1603 *len_ret = sizeof (struct sockaddr_rc);
1604 return 1;
1605 case BTPROTO_HCI:
1606 *len_ret = sizeof (struct sockaddr_hci);
1607 return 1;
Hye-Shik Chang81268602004-02-02 06:05:24 +00001608#if !defined(__FreeBSD__)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001609 case BTPROTO_SCO:
1610 *len_ret = sizeof (struct sockaddr_sco);
1611 return 1;
Hye-Shik Chang81268602004-02-02 06:05:24 +00001612#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001613 default:
1614 PyErr_SetString(socket_error, "getsockaddrlen: "
1615 "unknown BT protocol");
1616 return 0;
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001617
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001618 }
1619 }
Martin v. Löwis12af0482004-01-31 12:34:17 +00001620#endif
1621
Martin v. Löwis1ba3fd52001-08-10 20:29:40 +00001622#ifdef HAVE_NETPACKET_PACKET_H
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001623 case AF_PACKET:
1624 {
1625 *len_ret = sizeof (struct sockaddr_ll);
1626 return 1;
1627 }
Jeremy Hylton22308652001-02-02 03:23:09 +00001628#endif
Guido van Rossum48a680c2001-03-02 06:34:14 +00001629
Christian Heimes043d6f62008-01-07 17:19:16 +00001630#ifdef HAVE_LINUX_TIPC_H
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001631 case AF_TIPC:
1632 {
1633 *len_ret = sizeof (struct sockaddr_tipc);
1634 return 1;
1635 }
Christian Heimes043d6f62008-01-07 17:19:16 +00001636#endif
1637
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001638 /* More cases here... */
Guido van Rossum710e1df1992-06-12 10:39:36 +00001639
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001640 default:
1641 PyErr_SetString(socket_error, "getsockaddrlen: bad family");
1642 return 0;
Guido van Rossum710e1df1992-06-12 10:39:36 +00001643
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001644 }
Guido van Rossum710e1df1992-06-12 10:39:36 +00001645}
1646
1647
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00001648/* s._accept() -> (fd, address) */
Guido van Rossum30a685f1991-06-27 15:51:29 +00001649
Guido van Rossum73624e91994-10-10 17:59:00 +00001650static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001651sock_accept(PySocketSockObject *s)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001652{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001653 sock_addr_t addrbuf;
1654 SOCKET_T newfd = INVALID_SOCKET;
1655 socklen_t addrlen;
1656 PyObject *sock = NULL;
1657 PyObject *addr = NULL;
1658 PyObject *res = NULL;
1659 int timeout;
Antoine Pitroub1c54962010-10-14 15:05:38 +00001660#ifdef HAVE_ACCEPT4
1661 int flags = 0;
1662#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001663 if (!getsockaddrlen(s, &addrlen))
1664 return NULL;
1665 memset(&addrbuf, 0, addrlen);
Guido van Rossum67f7a382002-06-06 21:08:16 +00001666
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001667 if (!IS_SELECTABLE(s))
1668 return select_error();
Neal Norwitz082b2df2006-02-07 07:04:46 +00001669
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00001670 BEGIN_SELECT_LOOP(s)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001671 Py_BEGIN_ALLOW_THREADS
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00001672 timeout = internal_select_ex(s, 0, interval);
Antoine Pitroub1c54962010-10-14 15:05:38 +00001673 if (!timeout) {
1674#ifdef HAVE_ACCEPT4
1675 /* inherit socket flags and use accept4 call */
1676 flags = s->sock_type & (SOCK_CLOEXEC | SOCK_NONBLOCK);
1677 newfd = accept4(s->sock_fd, SAS2SA(&addrbuf), &addrlen, flags);
1678#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001679 newfd = accept(s->sock_fd, SAS2SA(&addrbuf), &addrlen);
Antoine Pitroub1c54962010-10-14 15:05:38 +00001680#endif /* HAVE_ACCEPT4 */
1681 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001682 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00001683
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001684 if (timeout == 1) {
1685 PyErr_SetString(socket_timeout, "timed out");
1686 return NULL;
1687 }
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00001688 END_SELECT_LOOP(s)
Raymond Hettingeref7343c2003-06-29 03:08:05 +00001689
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001690 if (newfd == INVALID_SOCKET)
1691 return s->errorhandler();
Barry Warsaw752300b1997-01-03 17:18:10 +00001692
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001693 sock = PyLong_FromSocket_t(newfd);
1694 if (sock == NULL) {
1695 SOCKETCLOSE(newfd);
1696 goto finally;
1697 }
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00001698
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001699 addr = makesockaddr(s->sock_fd, SAS2SA(&addrbuf),
1700 addrlen, s->sock_proto);
1701 if (addr == NULL)
1702 goto finally;
Barry Warsaw752300b1997-01-03 17:18:10 +00001703
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001704 res = PyTuple_Pack(2, sock, addr);
Barry Warsaw752300b1997-01-03 17:18:10 +00001705
Guido van Rossum67f7a382002-06-06 21:08:16 +00001706finally:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001707 Py_XDECREF(sock);
1708 Py_XDECREF(addr);
1709 return res;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001710}
1711
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001712PyDoc_STRVAR(accept_doc,
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00001713"_accept() -> (integer, address info)\n\
Guido van Rossum82a5c661998-07-07 20:45:43 +00001714\n\
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00001715Wait for an incoming connection. Return a new socket file descriptor\n\
1716representing the connection, and the address of the client.\n\
1717For IP sockets, the address info is a pair (hostaddr, port).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001718
Guido van Rossum11ba0942002-06-13 15:07:44 +00001719/* s.setblocking(flag) method. Argument:
1720 False -- non-blocking mode; same as settimeout(0)
1721 True -- blocking mode; same as settimeout(None)
1722*/
Guido van Rossume4485b01994-09-07 14:32:49 +00001723
Guido van Rossum73624e91994-10-10 17:59:00 +00001724static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001725sock_setblocking(PySocketSockObject *s, PyObject *arg)
Guido van Rossume4485b01994-09-07 14:32:49 +00001726{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001727 int block;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001728
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001729 block = PyLong_AsLong(arg);
1730 if (block == -1 && PyErr_Occurred())
1731 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001732
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001733 s->sock_timeout = block ? -1.0 : 0.0;
1734 internal_setblocking(s, block);
Guido van Rossume4485b01994-09-07 14:32:49 +00001735
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001736 Py_INCREF(Py_None);
1737 return Py_None;
Guido van Rossume4485b01994-09-07 14:32:49 +00001738}
Guido van Rossume4485b01994-09-07 14:32:49 +00001739
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001740PyDoc_STRVAR(setblocking_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001741"setblocking(flag)\n\
1742\n\
1743Set the socket to blocking (flag is true) or non-blocking (false).\n\
Guido van Rossum11ba0942002-06-13 15:07:44 +00001744setblocking(True) is equivalent to settimeout(None);\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001745setblocking(False) is equivalent to settimeout(0.0).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001746
Guido van Rossum11ba0942002-06-13 15:07:44 +00001747/* s.settimeout(timeout) method. Argument:
1748 None -- no timeout, blocking mode; same as setblocking(True)
1749 0.0 -- non-blocking mode; same as setblocking(False)
1750 > 0 -- timeout mode; operations time out after timeout seconds
1751 < 0 -- illegal; raises an exception
1752*/
Guido van Rossum67f7a382002-06-06 21:08:16 +00001753static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001754sock_settimeout(PySocketSockObject *s, PyObject *arg)
Guido van Rossum67f7a382002-06-06 21:08:16 +00001755{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001756 double timeout;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001757
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001758 if (arg == Py_None)
1759 timeout = -1.0;
1760 else {
1761 timeout = PyFloat_AsDouble(arg);
1762 if (timeout < 0.0) {
1763 if (!PyErr_Occurred())
1764 PyErr_SetString(PyExc_ValueError,
1765 "Timeout value out of range");
1766 return NULL;
1767 }
1768 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00001769
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001770 s->sock_timeout = timeout;
1771 internal_setblocking(s, timeout < 0.0);
Guido van Rossum67f7a382002-06-06 21:08:16 +00001772
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001773 Py_INCREF(Py_None);
1774 return Py_None;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001775}
1776
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001777PyDoc_STRVAR(settimeout_doc,
Guido van Rossum3eede5a2002-06-07 02:08:35 +00001778"settimeout(timeout)\n\
Guido van Rossum67f7a382002-06-06 21:08:16 +00001779\n\
Guido van Rossum11ba0942002-06-13 15:07:44 +00001780Set a timeout on socket operations. 'timeout' can be a float,\n\
1781giving in seconds, or None. Setting a timeout of None disables\n\
1782the timeout feature and is equivalent to setblocking(1).\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001783Setting a timeout of zero is the same as setblocking(0).");
Guido van Rossum67f7a382002-06-06 21:08:16 +00001784
Guido van Rossum3eede5a2002-06-07 02:08:35 +00001785/* s.gettimeout() method.
1786 Returns the timeout associated with a socket. */
Guido van Rossum67f7a382002-06-06 21:08:16 +00001787static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001788sock_gettimeout(PySocketSockObject *s)
Guido van Rossum67f7a382002-06-06 21:08:16 +00001789{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001790 if (s->sock_timeout < 0.0) {
1791 Py_INCREF(Py_None);
1792 return Py_None;
1793 }
1794 else
1795 return PyFloat_FromDouble(s->sock_timeout);
Guido van Rossum67f7a382002-06-06 21:08:16 +00001796}
1797
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001798PyDoc_STRVAR(gettimeout_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00001799"gettimeout() -> timeout\n\
Guido van Rossum67f7a382002-06-06 21:08:16 +00001800\n\
1801Returns the timeout in floating seconds associated with socket \n\
1802operations. A timeout of None indicates that timeouts on socket \n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001803operations are disabled.");
Guido van Rossume4485b01994-09-07 14:32:49 +00001804
Guido van Rossumaee08791992-09-08 09:05:33 +00001805/* s.setsockopt() method.
1806 With an integer third argument, sets an integer option.
1807 With a string third argument, sets an option from a buffer;
1808 use optional built-in module 'struct' to encode the string. */
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001809
Guido van Rossum73624e91994-10-10 17:59:00 +00001810static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001811sock_setsockopt(PySocketSockObject *s, PyObject *args)
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001812{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001813 int level;
1814 int optname;
1815 int res;
1816 char *buf;
1817 int buflen;
1818 int flag;
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001819
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001820 if (PyArg_ParseTuple(args, "iii:setsockopt",
1821 &level, &optname, &flag)) {
1822 buf = (char *) &flag;
1823 buflen = sizeof flag;
1824 }
1825 else {
1826 PyErr_Clear();
1827 if (!PyArg_ParseTuple(args, "iiy#:setsockopt",
1828 &level, &optname, &buf, &buflen))
1829 return NULL;
1830 }
1831 res = setsockopt(s->sock_fd, level, optname, (void *)buf, buflen);
1832 if (res < 0)
1833 return s->errorhandler();
1834 Py_INCREF(Py_None);
1835 return Py_None;
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001836}
1837
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001838PyDoc_STRVAR(setsockopt_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001839"setsockopt(level, option, value)\n\
1840\n\
1841Set a socket option. See the Unix manual for level and option.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001842The value argument can either be an integer or a string.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001843
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001844
Guido van Rossumaee08791992-09-08 09:05:33 +00001845/* s.getsockopt() method.
1846 With two arguments, retrieves an integer option.
1847 With a third integer argument, retrieves a string buffer of that size;
1848 use optional built-in module 'struct' to decode the string. */
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001849
Guido van Rossum73624e91994-10-10 17:59:00 +00001850static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001851sock_getsockopt(PySocketSockObject *s, PyObject *args)
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001852{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001853 int level;
1854 int optname;
1855 int res;
1856 PyObject *buf;
1857 socklen_t buflen = 0;
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001858
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001859 if (!PyArg_ParseTuple(args, "ii|i:getsockopt",
1860 &level, &optname, &buflen))
1861 return NULL;
Guido van Rossum48a680c2001-03-02 06:34:14 +00001862
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001863 if (buflen == 0) {
1864 int flag = 0;
1865 socklen_t flagsize = sizeof flag;
1866 res = getsockopt(s->sock_fd, level, optname,
1867 (void *)&flag, &flagsize);
1868 if (res < 0)
1869 return s->errorhandler();
1870 return PyLong_FromLong(flag);
1871 }
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001872#ifdef __VMS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001873 /* socklen_t is unsigned so no negative test is needed,
1874 test buflen == 0 is previously done */
1875 if (buflen > 1024) {
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001876#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001877 if (buflen <= 0 || buflen > 1024) {
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001878#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001879 PyErr_SetString(socket_error,
1880 "getsockopt buflen out of range");
1881 return NULL;
1882 }
1883 buf = PyBytes_FromStringAndSize((char *)NULL, buflen);
1884 if (buf == NULL)
1885 return NULL;
1886 res = getsockopt(s->sock_fd, level, optname,
1887 (void *)PyBytes_AS_STRING(buf), &buflen);
1888 if (res < 0) {
1889 Py_DECREF(buf);
1890 return s->errorhandler();
1891 }
1892 _PyBytes_Resize(&buf, buflen);
1893 return buf;
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001894}
1895
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001896PyDoc_STRVAR(getsockopt_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001897"getsockopt(level, option[, buffersize]) -> value\n\
1898\n\
1899Get a socket option. See the Unix manual for level and option.\n\
1900If a nonzero buffersize argument is given, the return value is a\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001901string of that length; otherwise it is an integer.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001902
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001903
Fred Drake728819a2000-07-01 03:40:12 +00001904/* s.bind(sockaddr) method */
Guido van Rossum30a685f1991-06-27 15:51:29 +00001905
Guido van Rossum73624e91994-10-10 17:59:00 +00001906static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001907sock_bind(PySocketSockObject *s, PyObject *addro)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001908{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001909 sock_addr_t addrbuf;
1910 int addrlen;
1911 int res;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00001912
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001913 if (!getsockaddrarg(s, addro, SAS2SA(&addrbuf), &addrlen))
1914 return NULL;
1915 Py_BEGIN_ALLOW_THREADS
1916 res = bind(s->sock_fd, SAS2SA(&addrbuf), addrlen);
1917 Py_END_ALLOW_THREADS
1918 if (res < 0)
1919 return s->errorhandler();
1920 Py_INCREF(Py_None);
1921 return Py_None;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001922}
1923
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001924PyDoc_STRVAR(bind_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001925"bind(address)\n\
1926\n\
1927Bind the socket to a local address. For IP sockets, the address is a\n\
Jeremy Hylton22308652001-02-02 03:23:09 +00001928pair (host, port); the host must refer to the local host. For raw packet\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001929sockets the address is a tuple (ifname, proto [,pkttype [,hatype]])");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001930
Guido van Rossum30a685f1991-06-27 15:51:29 +00001931
1932/* s.close() method.
1933 Set the file descriptor to -1 so operations tried subsequently
1934 will surely fail. */
1935
Guido van Rossum73624e91994-10-10 17:59:00 +00001936static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001937sock_close(PySocketSockObject *s)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001938{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001939 SOCKET_T fd;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00001940
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001941 if ((fd = s->sock_fd) != -1) {
1942 s->sock_fd = -1;
1943 Py_BEGIN_ALLOW_THREADS
1944 (void) SOCKETCLOSE(fd);
1945 Py_END_ALLOW_THREADS
1946 }
1947 Py_INCREF(Py_None);
1948 return Py_None;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001949}
1950
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001951PyDoc_STRVAR(close_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001952"close()\n\
1953\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001954Close the socket. It cannot be used after this call.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001955
Antoine Pitroue43f9d02010-08-08 23:24:50 +00001956static PyObject *
Antoine Pitrou6e451df2010-08-09 20:39:54 +00001957sock_detach(PySocketSockObject *s)
Antoine Pitroue43f9d02010-08-08 23:24:50 +00001958{
Antoine Pitrou6e451df2010-08-09 20:39:54 +00001959 SOCKET_T fd = s->sock_fd;
Antoine Pitroue43f9d02010-08-08 23:24:50 +00001960 s->sock_fd = -1;
Antoine Pitrou6e451df2010-08-09 20:39:54 +00001961 return PyLong_FromSocket_t(fd);
Antoine Pitroue43f9d02010-08-08 23:24:50 +00001962}
1963
Antoine Pitrou6e451df2010-08-09 20:39:54 +00001964PyDoc_STRVAR(detach_doc,
1965"detach()\n\
Antoine Pitroue43f9d02010-08-08 23:24:50 +00001966\n\
1967Close the socket object without closing the underlying file descriptor.\
1968The object cannot be used after this call, but the file descriptor\
Antoine Pitrou6e451df2010-08-09 20:39:54 +00001969can be reused for other purposes. The file descriptor is returned.");
Antoine Pitroue43f9d02010-08-08 23:24:50 +00001970
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001971static int
Raymond Hettingeref7343c2003-06-29 03:08:05 +00001972internal_connect(PySocketSockObject *s, struct sockaddr *addr, int addrlen,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001973 int *timeoutp)
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001974{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001975 int res, timeout;
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001976
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001977 timeout = 0;
1978 res = connect(s->sock_fd, addr, addrlen);
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001979
1980#ifdef MS_WINDOWS
1981
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001982 if (s->sock_timeout > 0.0) {
1983 if (res < 0 && WSAGetLastError() == WSAEWOULDBLOCK &&
1984 IS_SELECTABLE(s)) {
1985 /* This is a mess. Best solution: trust select */
1986 fd_set fds;
1987 fd_set fds_exc;
1988 struct timeval tv;
1989 tv.tv_sec = (int)s->sock_timeout;
1990 tv.tv_usec = (int)((s->sock_timeout - tv.tv_sec) * 1e6);
1991 FD_ZERO(&fds);
1992 FD_SET(s->sock_fd, &fds);
1993 FD_ZERO(&fds_exc);
1994 FD_SET(s->sock_fd, &fds_exc);
Antoine Pitrou19467d22010-08-17 19:33:30 +00001995 res = select(Py_SAFE_DOWNCAST(s->sock_fd+1, SOCKET_T, int),
1996 NULL, &fds, &fds_exc, &tv);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001997 if (res == 0) {
1998 res = WSAEWOULDBLOCK;
1999 timeout = 1;
2000 } else if (res > 0) {
2001 if (FD_ISSET(s->sock_fd, &fds))
2002 /* The socket is in the writable set - this
2003 means connected */
2004 res = 0;
2005 else {
2006 /* As per MS docs, we need to call getsockopt()
2007 to get the underlying error */
2008 int res_size = sizeof res;
2009 /* It must be in the exception set */
2010 assert(FD_ISSET(s->sock_fd, &fds_exc));
2011 if (0 == getsockopt(s->sock_fd, SOL_SOCKET, SO_ERROR,
2012 (char *)&res, &res_size))
2013 /* getsockopt also clears WSAGetLastError,
2014 so reset it back. */
2015 WSASetLastError(res);
2016 else
2017 res = WSAGetLastError();
2018 }
2019 }
2020 /* else if (res < 0) an error occurred */
2021 }
2022 }
Guido van Rossum7b8bac12002-06-13 16:07:04 +00002023
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002024 if (res < 0)
2025 res = WSAGetLastError();
Guido van Rossum7b8bac12002-06-13 16:07:04 +00002026
2027#else
2028
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002029 if (s->sock_timeout > 0.0) {
2030 if (res < 0 && errno == EINPROGRESS && IS_SELECTABLE(s)) {
2031 timeout = internal_select(s, 1);
2032 if (timeout == 0) {
2033 /* Bug #1019808: in case of an EINPROGRESS,
2034 use getsockopt(SO_ERROR) to get the real
2035 error. */
2036 socklen_t res_size = sizeof res;
2037 (void)getsockopt(s->sock_fd, SOL_SOCKET,
2038 SO_ERROR, &res, &res_size);
2039 if (res == EISCONN)
2040 res = 0;
2041 errno = res;
2042 }
2043 else if (timeout == -1) {
2044 res = errno; /* had error */
2045 }
2046 else
2047 res = EWOULDBLOCK; /* timed out */
2048 }
2049 }
Guido van Rossum7b8bac12002-06-13 16:07:04 +00002050
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002051 if (res < 0)
2052 res = errno;
Guido van Rossum7b8bac12002-06-13 16:07:04 +00002053
2054#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002055 *timeoutp = timeout;
Guido van Rossum7b8bac12002-06-13 16:07:04 +00002056
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002057 return res;
Guido van Rossum7b8bac12002-06-13 16:07:04 +00002058}
Guido van Rossum30a685f1991-06-27 15:51:29 +00002059
Fred Drake728819a2000-07-01 03:40:12 +00002060/* s.connect(sockaddr) method */
Guido van Rossum30a685f1991-06-27 15:51:29 +00002061
Guido van Rossum73624e91994-10-10 17:59:00 +00002062static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002063sock_connect(PySocketSockObject *s, PyObject *addro)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002064{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002065 sock_addr_t addrbuf;
2066 int addrlen;
2067 int res;
2068 int timeout;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00002069
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002070 if (!getsockaddrarg(s, addro, SAS2SA(&addrbuf), &addrlen))
2071 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002072
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002073 Py_BEGIN_ALLOW_THREADS
2074 res = internal_connect(s, SAS2SA(&addrbuf), addrlen, &timeout);
2075 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00002076
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002077 if (timeout == 1) {
2078 PyErr_SetString(socket_timeout, "timed out");
2079 return NULL;
2080 }
2081 if (res != 0)
2082 return s->errorhandler();
2083 Py_INCREF(Py_None);
2084 return Py_None;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002085}
2086
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002087PyDoc_STRVAR(connect_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002088"connect(address)\n\
2089\n\
2090Connect the socket to a remote address. For IP sockets, the address\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002091is a pair (host, port).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002092
Guido van Rossum30a685f1991-06-27 15:51:29 +00002093
Fred Drake728819a2000-07-01 03:40:12 +00002094/* s.connect_ex(sockaddr) method */
Guido van Rossumfc4255d1997-11-19 18:57:13 +00002095
2096static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002097sock_connect_ex(PySocketSockObject *s, PyObject *addro)
Guido van Rossumfc4255d1997-11-19 18:57:13 +00002098{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002099 sock_addr_t addrbuf;
2100 int addrlen;
2101 int res;
2102 int timeout;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00002103
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002104 if (!getsockaddrarg(s, addro, SAS2SA(&addrbuf), &addrlen))
2105 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002106
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002107 Py_BEGIN_ALLOW_THREADS
2108 res = internal_connect(s, SAS2SA(&addrbuf), addrlen, &timeout);
2109 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00002110
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002111 /* Signals are not errors (though they may raise exceptions). Adapted
2112 from PyErr_SetFromErrnoWithFilenameObject(). */
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002113#ifdef EINTR
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002114 if (res == EINTR && PyErr_CheckSignals())
2115 return NULL;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002116#endif
2117
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002118 return PyLong_FromLong((long) res);
Guido van Rossumfc4255d1997-11-19 18:57:13 +00002119}
2120
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002121PyDoc_STRVAR(connect_ex_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00002122"connect_ex(address) -> errno\n\
Guido van Rossum82a5c661998-07-07 20:45:43 +00002123\n\
2124This is like connect(address), but returns an error code (the errno value)\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002125instead of raising an exception when an error occurs.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002126
Guido van Rossumfc4255d1997-11-19 18:57:13 +00002127
Guido van Rossumed233a51992-06-23 09:07:03 +00002128/* s.fileno() method */
2129
Guido van Rossum73624e91994-10-10 17:59:00 +00002130static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002131sock_fileno(PySocketSockObject *s)
Guido van Rossumed233a51992-06-23 09:07:03 +00002132{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002133 return PyLong_FromSocket_t(s->sock_fd);
Guido van Rossumed233a51992-06-23 09:07:03 +00002134}
2135
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002136PyDoc_STRVAR(fileno_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002137"fileno() -> integer\n\
2138\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002139Return the integer file descriptor of the socket.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002140
Guido van Rossumed233a51992-06-23 09:07:03 +00002141
Guido van Rossumc89705d1992-11-26 08:54:07 +00002142/* s.getsockname() method */
2143
Guido van Rossum73624e91994-10-10 17:59:00 +00002144static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002145sock_getsockname(PySocketSockObject *s)
Guido van Rossumc89705d1992-11-26 08:54:07 +00002146{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002147 sock_addr_t addrbuf;
2148 int res;
2149 socklen_t addrlen;
Guido van Rossumff3ab422000-04-24 15:16:03 +00002150
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002151 if (!getsockaddrlen(s, &addrlen))
2152 return NULL;
2153 memset(&addrbuf, 0, addrlen);
2154 Py_BEGIN_ALLOW_THREADS
2155 res = getsockname(s->sock_fd, SAS2SA(&addrbuf), &addrlen);
2156 Py_END_ALLOW_THREADS
2157 if (res < 0)
2158 return s->errorhandler();
2159 return makesockaddr(s->sock_fd, SAS2SA(&addrbuf), addrlen,
2160 s->sock_proto);
Guido van Rossumc89705d1992-11-26 08:54:07 +00002161}
2162
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002163PyDoc_STRVAR(getsockname_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002164"getsockname() -> address info\n\
2165\n\
2166Return the address of the local endpoint. For IP sockets, the address\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002167info is a pair (hostaddr, port).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002168
Guido van Rossumc89705d1992-11-26 08:54:07 +00002169
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002170#ifdef HAVE_GETPEERNAME /* Cray APP doesn't have this :-( */
Guido van Rossumc89705d1992-11-26 08:54:07 +00002171/* s.getpeername() method */
2172
Guido van Rossum73624e91994-10-10 17:59:00 +00002173static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002174sock_getpeername(PySocketSockObject *s)
Guido van Rossumc89705d1992-11-26 08:54:07 +00002175{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002176 sock_addr_t addrbuf;
2177 int res;
2178 socklen_t addrlen;
Guido van Rossumff3ab422000-04-24 15:16:03 +00002179
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002180 if (!getsockaddrlen(s, &addrlen))
2181 return NULL;
2182 memset(&addrbuf, 0, addrlen);
2183 Py_BEGIN_ALLOW_THREADS
2184 res = getpeername(s->sock_fd, SAS2SA(&addrbuf), &addrlen);
2185 Py_END_ALLOW_THREADS
2186 if (res < 0)
2187 return s->errorhandler();
2188 return makesockaddr(s->sock_fd, SAS2SA(&addrbuf), addrlen,
2189 s->sock_proto);
Guido van Rossumc89705d1992-11-26 08:54:07 +00002190}
Guido van Rossum82a5c661998-07-07 20:45:43 +00002191
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002192PyDoc_STRVAR(getpeername_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002193"getpeername() -> address info\n\
2194\n\
2195Return the address of the remote endpoint. For IP sockets, the address\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002196info is a pair (hostaddr, port).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002197
Guido van Rossumb6775db1994-08-01 11:34:53 +00002198#endif /* HAVE_GETPEERNAME */
Guido van Rossumc89705d1992-11-26 08:54:07 +00002199
2200
Guido van Rossum30a685f1991-06-27 15:51:29 +00002201/* s.listen(n) method */
2202
Guido van Rossum73624e91994-10-10 17:59:00 +00002203static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002204sock_listen(PySocketSockObject *s, PyObject *arg)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002205{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002206 int backlog;
2207 int res;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00002208
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002209 backlog = PyLong_AsLong(arg);
2210 if (backlog == -1 && PyErr_Occurred())
2211 return NULL;
2212 Py_BEGIN_ALLOW_THREADS
2213 if (backlog < 1)
2214 backlog = 1;
2215 res = listen(s->sock_fd, backlog);
2216 Py_END_ALLOW_THREADS
2217 if (res < 0)
2218 return s->errorhandler();
2219 Py_INCREF(Py_None);
2220 return Py_None;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002221}
2222
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002223PyDoc_STRVAR(listen_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002224"listen(backlog)\n\
2225\n\
2226Enable a server to accept connections. The backlog argument must be at\n\
2227least 1; it specifies the number of unaccepted connection that the system\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002228will allow before refusing new connections.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002229
2230
Thomas Wouters477c8d52006-05-27 19:21:47 +00002231/*
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002232 * This is the guts of the recv() and recv_into() methods, which reads into a
Thomas Wouters902d6eb2007-01-09 23:18:33 +00002233 * char buffer. If you have any inc/dec ref to do to the objects that contain
Thomas Wouters477c8d52006-05-27 19:21:47 +00002234 * the buffer, do it in the caller. This function returns the number of bytes
2235 * succesfully read. If there was an error, it returns -1. Note that it is
2236 * also possible that we return a number of bytes smaller than the request
2237 * bytes.
2238 */
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002239
Antoine Pitrou19467d22010-08-17 19:33:30 +00002240static Py_ssize_t
2241sock_recv_guts(PySocketSockObject *s, char* cbuf, Py_ssize_t len, int flags)
Thomas Wouters477c8d52006-05-27 19:21:47 +00002242{
Antoine Pitrou19467d22010-08-17 19:33:30 +00002243 Py_ssize_t outlen = -1;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002244 int timeout;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002245#ifdef __VMS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002246 int remaining;
2247 char *read_buf;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002248#endif
2249
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002250 if (!IS_SELECTABLE(s)) {
2251 select_error();
2252 return -1;
2253 }
2254 if (len == 0) {
2255 /* If 0 bytes were requested, do nothing. */
2256 return 0;
2257 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002258
2259#ifndef __VMS
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002260 BEGIN_SELECT_LOOP(s)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002261 Py_BEGIN_ALLOW_THREADS
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002262 timeout = internal_select_ex(s, 0, interval);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002263 if (!timeout)
2264 outlen = recv(s->sock_fd, cbuf, len, flags);
2265 Py_END_ALLOW_THREADS
Thomas Wouters477c8d52006-05-27 19:21:47 +00002266
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002267 if (timeout == 1) {
2268 PyErr_SetString(socket_timeout, "timed out");
2269 return -1;
2270 }
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002271 END_SELECT_LOOP(s)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002272 if (outlen < 0) {
2273 /* Note: the call to errorhandler() ALWAYS indirectly returned
2274 NULL, so ignore its return value */
2275 s->errorhandler();
2276 return -1;
2277 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002278#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002279 read_buf = cbuf;
2280 remaining = len;
2281 while (remaining != 0) {
2282 unsigned int segment;
2283 int nread = -1;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002284
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002285 segment = remaining /SEGMENT_SIZE;
2286 if (segment != 0) {
2287 segment = SEGMENT_SIZE;
2288 }
2289 else {
2290 segment = remaining;
2291 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002292
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002293 BEGIN_SELECT_LOOP(s)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002294 Py_BEGIN_ALLOW_THREADS
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002295 timeout = internal_select_ex(s, 0, interval);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002296 if (!timeout)
2297 nread = recv(s->sock_fd, read_buf, segment, flags);
2298 Py_END_ALLOW_THREADS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002299 if (timeout == 1) {
2300 PyErr_SetString(socket_timeout, "timed out");
2301 return -1;
2302 }
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002303 END_SELECT_LOOP(s)
2304
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002305 if (nread < 0) {
2306 s->errorhandler();
2307 return -1;
2308 }
2309 if (nread != remaining) {
2310 read_buf += nread;
2311 break;
2312 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002313
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002314 remaining -= segment;
2315 read_buf += segment;
2316 }
2317 outlen = read_buf - cbuf;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002318#endif /* !__VMS */
2319
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002320 return outlen;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002321}
2322
Guido van Rossum48a680c2001-03-02 06:34:14 +00002323
Guido van Rossumeb6b33a1993-05-25 09:38:27 +00002324/* s.recv(nbytes [,flags]) method */
Guido van Rossum30a685f1991-06-27 15:51:29 +00002325
Guido van Rossum73624e91994-10-10 17:59:00 +00002326static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002327sock_recv(PySocketSockObject *s, PyObject *args)
Guido van Rossum30a685f1991-06-27 15:51:29 +00002328{
Antoine Pitrou19467d22010-08-17 19:33:30 +00002329 Py_ssize_t recvlen, outlen;
2330 int flags = 0;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002331 PyObject *buf;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002332
Antoine Pitrou19467d22010-08-17 19:33:30 +00002333 if (!PyArg_ParseTuple(args, "n|i:recv", &recvlen, &flags))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002334 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002335
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002336 if (recvlen < 0) {
2337 PyErr_SetString(PyExc_ValueError,
2338 "negative buffersize in recv");
2339 return NULL;
2340 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00002341
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002342 /* Allocate a new string. */
2343 buf = PyBytes_FromStringAndSize((char *) 0, recvlen);
2344 if (buf == NULL)
2345 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002346
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002347 /* Call the guts */
2348 outlen = sock_recv_guts(s, PyBytes_AS_STRING(buf), recvlen, flags);
2349 if (outlen < 0) {
2350 /* An error occurred, release the string and return an
2351 error. */
2352 Py_DECREF(buf);
2353 return NULL;
2354 }
2355 if (outlen != recvlen) {
2356 /* We did not read as many bytes as we anticipated, resize the
2357 string if possible and be successful. */
2358 _PyBytes_Resize(&buf, outlen);
2359 }
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00002360
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002361 return buf;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002362}
2363
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002364PyDoc_STRVAR(recv_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002365"recv(buffersize[, flags]) -> data\n\
2366\n\
2367Receive up to buffersize bytes from the socket. For the optional flags\n\
2368argument, see the Unix manual. When no data is available, block until\n\
2369at least one byte is available or until the remote end is closed. When\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002370the remote end is closed and all data is read, return the empty string.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002371
Guido van Rossum30a685f1991-06-27 15:51:29 +00002372
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002373/* s.recv_into(buffer, [nbytes [,flags]]) method */
Guido van Rossum30a685f1991-06-27 15:51:29 +00002374
Thomas Wouters477c8d52006-05-27 19:21:47 +00002375static PyObject*
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002376sock_recv_into(PySocketSockObject *s, PyObject *args, PyObject *kwds)
Thomas Wouters477c8d52006-05-27 19:21:47 +00002377{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002378 static char *kwlist[] = {"buffer", "nbytes", "flags", 0};
Thomas Wouters477c8d52006-05-27 19:21:47 +00002379
Antoine Pitrou19467d22010-08-17 19:33:30 +00002380 int flags = 0;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002381 Py_buffer pbuf;
2382 char *buf;
Antoine Pitrou19467d22010-08-17 19:33:30 +00002383 Py_ssize_t buflen, readlen, recvlen = 0;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002384
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002385 /* Get the buffer's memory */
Antoine Pitrou19467d22010-08-17 19:33:30 +00002386 if (!PyArg_ParseTupleAndKeywords(args, kwds, "w*|ni:recv_into", kwlist,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002387 &pbuf, &recvlen, &flags))
2388 return NULL;
2389 buf = pbuf.buf;
2390 buflen = pbuf.len;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002391
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002392 if (recvlen < 0) {
2393 PyBuffer_Release(&pbuf);
2394 PyErr_SetString(PyExc_ValueError,
2395 "negative buffersize in recv_into");
2396 return NULL;
2397 }
2398 if (recvlen == 0) {
2399 /* If nbytes was not specified, use the buffer's length */
2400 recvlen = buflen;
2401 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002402
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002403 /* Check if the buffer is large enough */
2404 if (buflen < recvlen) {
2405 PyBuffer_Release(&pbuf);
2406 PyErr_SetString(PyExc_ValueError,
2407 "buffer too small for requested bytes");
2408 return NULL;
2409 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002410
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002411 /* Call the guts */
2412 readlen = sock_recv_guts(s, buf, recvlen, flags);
2413 if (readlen < 0) {
2414 /* Return an error. */
2415 PyBuffer_Release(&pbuf);
2416 return NULL;
2417 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002418
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002419 PyBuffer_Release(&pbuf);
2420 /* Return the number of bytes read. Note that we do not do anything
2421 special here in the case that readlen < recvlen. */
2422 return PyLong_FromSsize_t(readlen);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002423}
2424
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002425PyDoc_STRVAR(recv_into_doc,
2426"recv_into(buffer, [nbytes[, flags]]) -> nbytes_read\n\
Thomas Wouters477c8d52006-05-27 19:21:47 +00002427\n\
2428A version of recv() that stores its data into a buffer rather than creating \n\
2429a new string. Receive up to buffersize bytes from the socket. If buffersize \n\
2430is not specified (or 0), receive up to the size available in the given buffer.\n\
2431\n\
2432See recv() for documentation about the flags.");
2433
2434
2435/*
Christian Heimes99170a52007-12-19 02:07:34 +00002436 * This is the guts of the recvfrom() and recvfrom_into() methods, which reads
2437 * into a char buffer. If you have any inc/def ref to do to the objects that
2438 * contain the buffer, do it in the caller. This function returns the number
2439 * of bytes succesfully read. If there was an error, it returns -1. Note
2440 * that it is also possible that we return a number of bytes smaller than the
2441 * request bytes.
Thomas Wouters477c8d52006-05-27 19:21:47 +00002442 *
2443 * 'addr' is a return value for the address object. Note that you must decref
2444 * it yourself.
2445 */
Antoine Pitrou19467d22010-08-17 19:33:30 +00002446static Py_ssize_t
2447sock_recvfrom_guts(PySocketSockObject *s, char* cbuf, Py_ssize_t len, int flags,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002448 PyObject** addr)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002449{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002450 sock_addr_t addrbuf;
2451 int timeout;
Antoine Pitrou19467d22010-08-17 19:33:30 +00002452 Py_ssize_t n = -1;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002453 socklen_t addrlen;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002454
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002455 *addr = NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002456
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002457 if (!getsockaddrlen(s, &addrlen))
2458 return -1;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002459
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002460 if (!IS_SELECTABLE(s)) {
2461 select_error();
2462 return -1;
2463 }
Neal Norwitz082b2df2006-02-07 07:04:46 +00002464
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002465 BEGIN_SELECT_LOOP(s)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002466 Py_BEGIN_ALLOW_THREADS
2467 memset(&addrbuf, 0, addrlen);
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002468 timeout = internal_select_ex(s, 0, interval);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002469 if (!timeout) {
Guido van Rossum8d665e61996-06-26 18:22:49 +00002470#ifndef MS_WINDOWS
Andrew MacIntyreba43e872002-03-03 03:03:52 +00002471#if defined(PYOS_OS2) && !defined(PYCC_GCC)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002472 n = recvfrom(s->sock_fd, cbuf, len, flags,
2473 SAS2SA(&addrbuf), &addrlen);
Guido van Rossum32c575d1997-12-02 20:37:32 +00002474#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002475 n = recvfrom(s->sock_fd, cbuf, len, flags,
2476 (void *) &addrbuf, &addrlen);
Guido van Rossum32c575d1997-12-02 20:37:32 +00002477#endif
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00002478#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002479 n = recvfrom(s->sock_fd, cbuf, len, flags,
2480 SAS2SA(&addrbuf), &addrlen);
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00002481#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002482 }
2483 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00002484
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002485 if (timeout == 1) {
2486 PyErr_SetString(socket_timeout, "timed out");
2487 return -1;
2488 }
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002489 END_SELECT_LOOP(s)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002490 if (n < 0) {
2491 s->errorhandler();
2492 return -1;
2493 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00002494
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002495 if (!(*addr = makesockaddr(s->sock_fd, SAS2SA(&addrbuf),
2496 addrlen, s->sock_proto)))
2497 return -1;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002498
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002499 return n;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002500}
2501
2502/* s.recvfrom(nbytes [,flags]) method */
2503
2504static PyObject *
2505sock_recvfrom(PySocketSockObject *s, PyObject *args)
2506{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002507 PyObject *buf = NULL;
2508 PyObject *addr = NULL;
2509 PyObject *ret = NULL;
Antoine Pitrou19467d22010-08-17 19:33:30 +00002510 int flags = 0;
2511 Py_ssize_t recvlen, outlen;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002512
Antoine Pitrou19467d22010-08-17 19:33:30 +00002513 if (!PyArg_ParseTuple(args, "n|i:recvfrom", &recvlen, &flags))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002514 return NULL;
Guido van Rossum48a680c2001-03-02 06:34:14 +00002515
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002516 if (recvlen < 0) {
2517 PyErr_SetString(PyExc_ValueError,
2518 "negative buffersize in recvfrom");
2519 return NULL;
2520 }
Guido van Rossumd8faa362007-04-27 19:54:29 +00002521
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002522 buf = PyBytes_FromStringAndSize((char *) 0, recvlen);
2523 if (buf == NULL)
2524 return NULL;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002525
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002526 outlen = sock_recvfrom_guts(s, PyBytes_AS_STRING(buf),
2527 recvlen, flags, &addr);
2528 if (outlen < 0) {
2529 goto finally;
2530 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002531
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002532 if (outlen != recvlen) {
2533 /* We did not read as many bytes as we anticipated, resize the
2534 string if possible and be succesful. */
2535 if (_PyBytes_Resize(&buf, outlen) < 0)
2536 /* Oopsy, not so succesful after all. */
2537 goto finally;
2538 }
Barry Warsaw752300b1997-01-03 17:18:10 +00002539
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002540 ret = PyTuple_Pack(2, buf, addr);
Guido van Rossum67f7a382002-06-06 21:08:16 +00002541
2542finally:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002543 Py_XDECREF(buf);
2544 Py_XDECREF(addr);
2545 return ret;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002546}
2547
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002548PyDoc_STRVAR(recvfrom_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002549"recvfrom(buffersize[, flags]) -> (data, address info)\n\
2550\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002551Like recv(buffersize, flags) but also return the sender's address info.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002552
Thomas Wouters477c8d52006-05-27 19:21:47 +00002553
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002554/* s.recvfrom_into(buffer[, nbytes [,flags]]) method */
Thomas Wouters477c8d52006-05-27 19:21:47 +00002555
2556static PyObject *
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002557sock_recvfrom_into(PySocketSockObject *s, PyObject *args, PyObject* kwds)
Thomas Wouters477c8d52006-05-27 19:21:47 +00002558{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002559 static char *kwlist[] = {"buffer", "nbytes", "flags", 0};
Thomas Wouters477c8d52006-05-27 19:21:47 +00002560
Antoine Pitrou19467d22010-08-17 19:33:30 +00002561 int flags = 0;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002562 Py_buffer pbuf;
2563 char *buf;
Antoine Pitrou19467d22010-08-17 19:33:30 +00002564 Py_ssize_t readlen, buflen, recvlen = 0;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002565
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002566 PyObject *addr = NULL;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002567
Antoine Pitrou19467d22010-08-17 19:33:30 +00002568 if (!PyArg_ParseTupleAndKeywords(args, kwds, "w*|ni:recvfrom_into",
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002569 kwlist, &pbuf,
2570 &recvlen, &flags))
2571 return NULL;
2572 buf = pbuf.buf;
2573 buflen = pbuf.len;
2574 assert(buf != 0 && buflen > 0);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002575
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002576 if (recvlen < 0) {
2577 PyBuffer_Release(&pbuf);
2578 PyErr_SetString(PyExc_ValueError,
2579 "negative buffersize in recvfrom_into");
2580 return NULL;
2581 }
2582 if (recvlen == 0) {
2583 /* If nbytes was not specified, use the buffer's length */
2584 recvlen = buflen;
2585 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002586
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002587 readlen = sock_recvfrom_guts(s, buf, recvlen, flags, &addr);
2588 if (readlen < 0) {
2589 PyBuffer_Release(&pbuf);
2590 /* Return an error */
2591 Py_XDECREF(addr);
2592 return NULL;
2593 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002594
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002595 PyBuffer_Release(&pbuf);
2596 /* Return the number of bytes read and the address. Note that we do
2597 not do anything special here in the case that readlen < recvlen. */
Antoine Pitrou19467d22010-08-17 19:33:30 +00002598 return Py_BuildValue("nN", readlen, addr);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002599}
2600
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002601PyDoc_STRVAR(recvfrom_into_doc,
2602"recvfrom_into(buffer[, nbytes[, flags]]) -> (nbytes, address info)\n\
Thomas Wouters477c8d52006-05-27 19:21:47 +00002603\n\
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002604Like recv_into(buffer[, nbytes[, flags]]) but also return the sender's address info.");
Thomas Wouters477c8d52006-05-27 19:21:47 +00002605
2606
Guido van Rossumeb6b33a1993-05-25 09:38:27 +00002607/* s.send(data [,flags]) method */
Guido van Rossum30a685f1991-06-27 15:51:29 +00002608
Guido van Rossum73624e91994-10-10 17:59:00 +00002609static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002610sock_send(PySocketSockObject *s, PyObject *args)
Guido van Rossum30a685f1991-06-27 15:51:29 +00002611{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002612 char *buf;
Antoine Pitrou19467d22010-08-17 19:33:30 +00002613 Py_ssize_t len, n = -1;
2614 int flags = 0, timeout;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002615 Py_buffer pbuf;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002616
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002617 if (!PyArg_ParseTuple(args, "y*|i:send", &pbuf, &flags))
2618 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002619
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002620 if (!IS_SELECTABLE(s)) {
2621 PyBuffer_Release(&pbuf);
2622 return select_error();
2623 }
2624 buf = pbuf.buf;
2625 len = pbuf.len;
Neal Norwitz082b2df2006-02-07 07:04:46 +00002626
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002627 BEGIN_SELECT_LOOP(s)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002628 Py_BEGIN_ALLOW_THREADS
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002629 timeout = internal_select_ex(s, 1, interval);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002630 if (!timeout)
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002631#ifdef __VMS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002632 n = sendsegmented(s->sock_fd, buf, len, flags);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002633#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002634 n = send(s->sock_fd, buf, len, flags);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002635#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002636 Py_END_ALLOW_THREADS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002637 if (timeout == 1) {
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002638 PyBuffer_Release(&pbuf);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002639 PyErr_SetString(socket_timeout, "timed out");
2640 return NULL;
2641 }
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002642 END_SELECT_LOOP(s)
2643
2644 PyBuffer_Release(&pbuf);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002645 if (n < 0)
2646 return s->errorhandler();
Antoine Pitrou19467d22010-08-17 19:33:30 +00002647 return PyLong_FromSsize_t(n);
Guido van Rossum30a685f1991-06-27 15:51:29 +00002648}
2649
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002650PyDoc_STRVAR(send_doc,
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002651"send(data[, flags]) -> count\n\
Guido van Rossum82a5c661998-07-07 20:45:43 +00002652\n\
2653Send a data string to the socket. For the optional flags\n\
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002654argument, see the Unix manual. Return the number of bytes\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002655sent; this may be less than len(data) if the network is busy.");
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002656
2657
2658/* s.sendall(data [,flags]) method */
2659
2660static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002661sock_sendall(PySocketSockObject *s, PyObject *args)
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002662{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002663 char *buf;
Antoine Pitrou19467d22010-08-17 19:33:30 +00002664 Py_ssize_t len, n = -1;
Antoine Pitrou6d7df632010-09-27 17:52:25 +00002665 int flags = 0, timeout, saved_errno;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002666 Py_buffer pbuf;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002667
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002668 if (!PyArg_ParseTuple(args, "y*|i:sendall", &pbuf, &flags))
2669 return NULL;
2670 buf = pbuf.buf;
2671 len = pbuf.len;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002672
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002673 if (!IS_SELECTABLE(s)) {
2674 PyBuffer_Release(&pbuf);
2675 return select_error();
2676 }
Neal Norwitz082b2df2006-02-07 07:04:46 +00002677
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002678 do {
Antoine Pitrou6d7df632010-09-27 17:52:25 +00002679 Py_BEGIN_ALLOW_THREADS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002680 timeout = internal_select(s, 1);
2681 n = -1;
Antoine Pitrou6d7df632010-09-27 17:52:25 +00002682 if (!timeout) {
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002683#ifdef __VMS
Antoine Pitrou6d7df632010-09-27 17:52:25 +00002684 n = sendsegmented(s->sock_fd, buf, len, flags);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002685#else
Antoine Pitrou6d7df632010-09-27 17:52:25 +00002686 n = send(s->sock_fd, buf, len, flags);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002687#endif
Antoine Pitrou6d7df632010-09-27 17:52:25 +00002688 }
2689 Py_END_ALLOW_THREADS
2690 if (timeout == 1) {
2691 PyBuffer_Release(&pbuf);
2692 PyErr_SetString(socket_timeout, "timed out");
2693 return NULL;
2694 }
2695 /* PyErr_CheckSignals() might change errno */
2696 saved_errno = errno;
2697 /* We must run our signal handlers before looping again.
2698 send() can return a successful partial write when it is
2699 interrupted, so we can't restrict ourselves to EINTR. */
2700 if (PyErr_CheckSignals()) {
2701 PyBuffer_Release(&pbuf);
2702 return NULL;
2703 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002704 if (n < 0) {
Antoine Pitrou6d7df632010-09-27 17:52:25 +00002705 /* If interrupted, try again */
2706 if (saved_errno == EINTR)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002707 continue;
Antoine Pitrou6d7df632010-09-27 17:52:25 +00002708 else
2709 break;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002710 }
2711 buf += n;
2712 len -= n;
2713 } while (len > 0);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002714 PyBuffer_Release(&pbuf);
Guido van Rossum67f7a382002-06-06 21:08:16 +00002715
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002716 if (n < 0)
2717 return s->errorhandler();
Guido van Rossum67f7a382002-06-06 21:08:16 +00002718
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002719 Py_INCREF(Py_None);
2720 return Py_None;
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002721}
2722
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002723PyDoc_STRVAR(sendall_doc,
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002724"sendall(data[, flags])\n\
2725\n\
2726Send a data string to the socket. For the optional flags\n\
2727argument, see the Unix manual. This calls send() repeatedly\n\
2728until all data is sent. If an error occurs, it's impossible\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002729to tell how much data has been sent.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002730
Guido van Rossum30a685f1991-06-27 15:51:29 +00002731
Guido van Rossumeb6b33a1993-05-25 09:38:27 +00002732/* s.sendto(data, [flags,] sockaddr) method */
Guido van Rossum30a685f1991-06-27 15:51:29 +00002733
Guido van Rossum73624e91994-10-10 17:59:00 +00002734static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002735sock_sendto(PySocketSockObject *s, PyObject *args)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002736{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002737 Py_buffer pbuf;
2738 PyObject *addro;
2739 char *buf;
2740 Py_ssize_t len;
2741 sock_addr_t addrbuf;
2742 int addrlen, n = -1, flags, timeout;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002743
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002744 flags = 0;
2745 if (!PyArg_ParseTuple(args, "y*O:sendto", &pbuf, &addro)) {
2746 PyErr_Clear();
2747 if (!PyArg_ParseTuple(args, "y*iO:sendto",
2748 &pbuf, &flags, &addro))
2749 return NULL;
2750 }
2751 buf = pbuf.buf;
2752 len = pbuf.len;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002753
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002754 if (!IS_SELECTABLE(s)) {
2755 PyBuffer_Release(&pbuf);
2756 return select_error();
2757 }
Neal Norwitz082b2df2006-02-07 07:04:46 +00002758
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002759 if (!getsockaddrarg(s, addro, SAS2SA(&addrbuf), &addrlen)) {
2760 PyBuffer_Release(&pbuf);
2761 return NULL;
2762 }
Thomas Wouters89f507f2006-12-13 04:49:30 +00002763
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002764 BEGIN_SELECT_LOOP(s)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002765 Py_BEGIN_ALLOW_THREADS
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002766 timeout = internal_select_ex(s, 1, interval);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002767 if (!timeout)
2768 n = sendto(s->sock_fd, buf, len, flags, SAS2SA(&addrbuf), addrlen);
2769 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00002770
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002771 if (timeout == 1) {
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002772 PyBuffer_Release(&pbuf);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002773 PyErr_SetString(socket_timeout, "timed out");
2774 return NULL;
2775 }
Antoine Pitrou3e1fd272010-09-28 21:23:11 +00002776 END_SELECT_LOOP(s)
2777 PyBuffer_Release(&pbuf);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002778 if (n < 0)
2779 return s->errorhandler();
Antoine Pitrou19467d22010-08-17 19:33:30 +00002780 return PyLong_FromSsize_t(n);
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002781}
2782
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002783PyDoc_STRVAR(sendto_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00002784"sendto(data[, flags], address) -> count\n\
Guido van Rossum82a5c661998-07-07 20:45:43 +00002785\n\
2786Like send(data, flags) but allows specifying the destination address.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002787For IP sockets, the address is a pair (hostaddr, port).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002788
Guido van Rossum30a685f1991-06-27 15:51:29 +00002789
2790/* s.shutdown(how) method */
2791
Guido van Rossum73624e91994-10-10 17:59:00 +00002792static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002793sock_shutdown(PySocketSockObject *s, PyObject *arg)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002794{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002795 int how;
2796 int res;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00002797
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002798 how = PyLong_AsLong(arg);
2799 if (how == -1 && PyErr_Occurred())
2800 return NULL;
2801 Py_BEGIN_ALLOW_THREADS
2802 res = shutdown(s->sock_fd, how);
2803 Py_END_ALLOW_THREADS
2804 if (res < 0)
2805 return s->errorhandler();
2806 Py_INCREF(Py_None);
2807 return Py_None;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002808}
2809
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002810PyDoc_STRVAR(shutdown_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002811"shutdown(flag)\n\
2812\n\
Martin v. Löwis94681fc2003-11-27 19:40:22 +00002813Shut down the reading side of the socket (flag == SHUT_RD), the writing side\n\
2814of the socket (flag == SHUT_WR), or both ends (flag == SHUT_RDWR).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002815
Amaury Forgeot d'Arc3d17a5c2008-06-13 01:09:34 +00002816#if defined(MS_WINDOWS) && defined(SIO_RCVALL)
Christian Heimesfaf2f632008-01-06 16:59:19 +00002817static PyObject*
2818sock_ioctl(PySocketSockObject *s, PyObject *arg)
2819{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002820 unsigned long cmd = SIO_RCVALL;
2821 PyObject *argO;
2822 DWORD recv;
Christian Heimesfaf2f632008-01-06 16:59:19 +00002823
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002824 if (!PyArg_ParseTuple(arg, "kO:ioctl", &cmd, &argO))
2825 return NULL;
Christian Heimesfaf2f632008-01-06 16:59:19 +00002826
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002827 switch (cmd) {
2828 case SIO_RCVALL: {
2829 unsigned int option = RCVALL_ON;
2830 if (!PyArg_ParseTuple(arg, "kI:ioctl", &cmd, &option))
2831 return NULL;
2832 if (WSAIoctl(s->sock_fd, cmd, &option, sizeof(option),
2833 NULL, 0, &recv, NULL, NULL) == SOCKET_ERROR) {
2834 return set_error();
2835 }
2836 return PyLong_FromUnsignedLong(recv); }
2837 case SIO_KEEPALIVE_VALS: {
2838 struct tcp_keepalive ka;
2839 if (!PyArg_ParseTuple(arg, "k(kkk):ioctl", &cmd,
2840 &ka.onoff, &ka.keepalivetime, &ka.keepaliveinterval))
2841 return NULL;
2842 if (WSAIoctl(s->sock_fd, cmd, &ka, sizeof(ka),
2843 NULL, 0, &recv, NULL, NULL) == SOCKET_ERROR) {
2844 return set_error();
2845 }
2846 return PyLong_FromUnsignedLong(recv); }
2847 default:
2848 PyErr_Format(PyExc_ValueError, "invalid ioctl command %d", cmd);
2849 return NULL;
2850 }
Christian Heimesfaf2f632008-01-06 16:59:19 +00002851}
2852PyDoc_STRVAR(sock_ioctl_doc,
2853"ioctl(cmd, option) -> long\n\
2854\n\
Kristján Valur Jónsson847ec752009-09-27 21:10:38 +00002855Control the socket with WSAIoctl syscall. Currently supported 'cmd' values are\n\
2856SIO_RCVALL: 'option' must be one of the socket.RCVALL_* constants.\n\
2857SIO_KEEPALIVE_VALS: 'option' is a tuple of (onoff, timeout, interval).");
Christian Heimesfaf2f632008-01-06 16:59:19 +00002858
2859#endif
Guido van Rossum30a685f1991-06-27 15:51:29 +00002860
2861/* List of methods for socket objects */
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002862
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002863static PyMethodDef sock_methods[] = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002864 {"_accept", (PyCFunction)sock_accept, METH_NOARGS,
2865 accept_doc},
2866 {"bind", (PyCFunction)sock_bind, METH_O,
2867 bind_doc},
2868 {"close", (PyCFunction)sock_close, METH_NOARGS,
2869 close_doc},
2870 {"connect", (PyCFunction)sock_connect, METH_O,
2871 connect_doc},
2872 {"connect_ex", (PyCFunction)sock_connect_ex, METH_O,
2873 connect_ex_doc},
Antoine Pitrou6e451df2010-08-09 20:39:54 +00002874 {"detach", (PyCFunction)sock_detach, METH_NOARGS,
2875 detach_doc},
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002876 {"fileno", (PyCFunction)sock_fileno, METH_NOARGS,
2877 fileno_doc},
Guido van Rossumb6775db1994-08-01 11:34:53 +00002878#ifdef HAVE_GETPEERNAME
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002879 {"getpeername", (PyCFunction)sock_getpeername,
2880 METH_NOARGS, getpeername_doc},
Guido van Rossum9575a441993-04-07 14:06:14 +00002881#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002882 {"getsockname", (PyCFunction)sock_getsockname,
2883 METH_NOARGS, getsockname_doc},
2884 {"getsockopt", (PyCFunction)sock_getsockopt, METH_VARARGS,
2885 getsockopt_doc},
Amaury Forgeot d'Arc3d17a5c2008-06-13 01:09:34 +00002886#if defined(MS_WINDOWS) && defined(SIO_RCVALL)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002887 {"ioctl", (PyCFunction)sock_ioctl, METH_VARARGS,
2888 sock_ioctl_doc},
Christian Heimesfaf2f632008-01-06 16:59:19 +00002889#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002890 {"listen", (PyCFunction)sock_listen, METH_O,
2891 listen_doc},
2892 {"recv", (PyCFunction)sock_recv, METH_VARARGS,
2893 recv_doc},
2894 {"recv_into", (PyCFunction)sock_recv_into, METH_VARARGS | METH_KEYWORDS,
2895 recv_into_doc},
2896 {"recvfrom", (PyCFunction)sock_recvfrom, METH_VARARGS,
2897 recvfrom_doc},
2898 {"recvfrom_into", (PyCFunction)sock_recvfrom_into, METH_VARARGS | METH_KEYWORDS,
2899 recvfrom_into_doc},
2900 {"send", (PyCFunction)sock_send, METH_VARARGS,
2901 send_doc},
2902 {"sendall", (PyCFunction)sock_sendall, METH_VARARGS,
2903 sendall_doc},
2904 {"sendto", (PyCFunction)sock_sendto, METH_VARARGS,
2905 sendto_doc},
2906 {"setblocking", (PyCFunction)sock_setblocking, METH_O,
2907 setblocking_doc},
2908 {"settimeout", (PyCFunction)sock_settimeout, METH_O,
2909 settimeout_doc},
2910 {"gettimeout", (PyCFunction)sock_gettimeout, METH_NOARGS,
2911 gettimeout_doc},
2912 {"setsockopt", (PyCFunction)sock_setsockopt, METH_VARARGS,
2913 setsockopt_doc},
2914 {"shutdown", (PyCFunction)sock_shutdown, METH_O,
2915 shutdown_doc},
2916 {NULL, NULL} /* sentinel */
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002917};
2918
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002919/* SockObject members */
2920static PyMemberDef sock_memberlist[] = {
2921 {"family", T_INT, offsetof(PySocketSockObject, sock_family), READONLY, "the socket family"},
2922 {"type", T_INT, offsetof(PySocketSockObject, sock_type), READONLY, "the socket type"},
2923 {"proto", T_INT, offsetof(PySocketSockObject, sock_proto), READONLY, "the socket protocol"},
2924 {"timeout", T_DOUBLE, offsetof(PySocketSockObject, sock_timeout), READONLY, "the socket timeout"},
2925 {0},
2926};
Guido van Rossum30a685f1991-06-27 15:51:29 +00002927
Guido van Rossum73624e91994-10-10 17:59:00 +00002928/* Deallocate a socket object in response to the last Py_DECREF().
Guido van Rossum30a685f1991-06-27 15:51:29 +00002929 First close the file description. */
2930
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002931static void
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002932sock_dealloc(PySocketSockObject *s)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002933{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002934 if (s->sock_fd != -1)
2935 (void) SOCKETCLOSE(s->sock_fd);
2936 Py_TYPE(s)->tp_free((PyObject *)s);
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002937}
2938
Guido van Rossum30a685f1991-06-27 15:51:29 +00002939
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00002940static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002941sock_repr(PySocketSockObject *s)
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00002942{
Fred Drakea04eaad2000-06-30 02:46:07 +00002943#if SIZEOF_SOCKET_T > SIZEOF_LONG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002944 if (s->sock_fd > LONG_MAX) {
2945 /* this can occur on Win64, and actually there is a special
2946 ugly printf formatter for decimal pointer length integer
2947 printing, only bother if necessary*/
2948 PyErr_SetString(PyExc_OverflowError,
2949 "no printf formatter to display "
2950 "the socket descriptor in decimal");
2951 return NULL;
2952 }
Fred Drakea04eaad2000-06-30 02:46:07 +00002953#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002954 return PyUnicode_FromFormat(
2955 "<socket object, fd=%ld, family=%d, type=%d, proto=%d>",
2956 (long)s->sock_fd, s->sock_family,
2957 s->sock_type,
2958 s->sock_proto);
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00002959}
2960
2961
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002962/* Create a new, uninitialized socket object. */
2963
2964static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002965sock_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002966{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002967 PyObject *new;
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002968
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002969 new = type->tp_alloc(type, 0);
2970 if (new != NULL) {
2971 ((PySocketSockObject *)new)->sock_fd = -1;
2972 ((PySocketSockObject *)new)->sock_timeout = -1.0;
2973 ((PySocketSockObject *)new)->errorhandler = &set_error;
2974 }
2975 return new;
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002976}
2977
2978
2979/* Initialize a new socket object. */
2980
2981/*ARGSUSED*/
2982static int
Andrew MacIntyre7aec4a22002-06-13 11:53:52 +00002983sock_initobj(PyObject *self, PyObject *args, PyObject *kwds)
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002984{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002985 PySocketSockObject *s = (PySocketSockObject *)self;
2986 PyObject *fdobj = NULL;
2987 SOCKET_T fd = INVALID_SOCKET;
2988 int family = AF_INET, type = SOCK_STREAM, proto = 0;
2989 static char *keywords[] = {"family", "type", "proto", "fileno", 0};
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002990
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002991 if (!PyArg_ParseTupleAndKeywords(args, kwds,
2992 "|iiiO:socket", keywords,
2993 &family, &type, &proto, &fdobj))
2994 return -1;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002995
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002996 if (fdobj != NULL && fdobj != Py_None) {
2997 fd = PyLong_AsSocket_t(fdobj);
2998 if (fd == (SOCKET_T)(-1) && PyErr_Occurred())
2999 return -1;
3000 if (fd == INVALID_SOCKET) {
3001 PyErr_SetString(PyExc_ValueError,
3002 "can't use invalid socket value");
3003 return -1;
3004 }
3005 }
3006 else {
3007 Py_BEGIN_ALLOW_THREADS
3008 fd = socket(family, type, proto);
3009 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00003010
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003011 if (fd == INVALID_SOCKET) {
3012 set_error();
3013 return -1;
3014 }
3015 }
3016 init_sockobject(s, fd, family, type, proto);
Guido van Rossum67f7a382002-06-06 21:08:16 +00003017
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003018 return 0;
Guido van Rossum67f7a382002-06-06 21:08:16 +00003019
Guido van Rossum384ca9c2001-10-27 22:20:47 +00003020}
3021
3022
Guido van Rossumb6775db1994-08-01 11:34:53 +00003023/* Type object for socket objects. */
Guido van Rossum30a685f1991-06-27 15:51:29 +00003024
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003025static PyTypeObject sock_type = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003026 PyVarObject_HEAD_INIT(0, 0) /* Must fill in type value later */
3027 "_socket.socket", /* tp_name */
3028 sizeof(PySocketSockObject), /* tp_basicsize */
3029 0, /* tp_itemsize */
3030 (destructor)sock_dealloc, /* tp_dealloc */
3031 0, /* tp_print */
3032 0, /* tp_getattr */
3033 0, /* tp_setattr */
3034 0, /* tp_reserved */
3035 (reprfunc)sock_repr, /* tp_repr */
3036 0, /* tp_as_number */
3037 0, /* tp_as_sequence */
3038 0, /* tp_as_mapping */
3039 0, /* tp_hash */
3040 0, /* tp_call */
3041 0, /* tp_str */
3042 PyObject_GenericGetAttr, /* tp_getattro */
3043 0, /* tp_setattro */
3044 0, /* tp_as_buffer */
3045 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE, /* tp_flags */
3046 sock_doc, /* tp_doc */
3047 0, /* tp_traverse */
3048 0, /* tp_clear */
3049 0, /* tp_richcompare */
3050 0, /* tp_weaklistoffset */
3051 0, /* tp_iter */
3052 0, /* tp_iternext */
3053 sock_methods, /* tp_methods */
3054 sock_memberlist, /* tp_members */
3055 0, /* tp_getset */
3056 0, /* tp_base */
3057 0, /* tp_dict */
3058 0, /* tp_descr_get */
3059 0, /* tp_descr_set */
3060 0, /* tp_dictoffset */
3061 sock_initobj, /* tp_init */
3062 PyType_GenericAlloc, /* tp_alloc */
3063 sock_new, /* tp_new */
3064 PyObject_Del, /* tp_free */
Guido van Rossum6574b3e1991-06-25 21:36:08 +00003065};
3066
Guido van Rossum30a685f1991-06-27 15:51:29 +00003067
Guido van Rossum81194471991-07-27 21:42:02 +00003068/* Python interface to gethostname(). */
3069
3070/*ARGSUSED*/
Guido van Rossum73624e91994-10-10 17:59:00 +00003071static PyObject *
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00003072socket_gethostname(PyObject *self, PyObject *unused)
Guido van Rossum81194471991-07-27 21:42:02 +00003073{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003074 char buf[1024];
3075 int res;
3076 Py_BEGIN_ALLOW_THREADS
3077 res = gethostname(buf, (int) sizeof buf - 1);
3078 Py_END_ALLOW_THREADS
3079 if (res < 0)
3080 return set_error();
3081 buf[sizeof buf - 1] = '\0';
3082 return PyUnicode_FromString(buf);
Guido van Rossum81194471991-07-27 21:42:02 +00003083}
Guido van Rossumff4949e1992-08-05 19:58:53 +00003084
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003085PyDoc_STRVAR(gethostname_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003086"gethostname() -> string\n\
3087\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003088Return the current host name.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003089
Guido van Rossumff4949e1992-08-05 19:58:53 +00003090
Guido van Rossum30a685f1991-06-27 15:51:29 +00003091/* Python interface to gethostbyname(name). */
3092
3093/*ARGSUSED*/
Guido van Rossum73624e91994-10-10 17:59:00 +00003094static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003095socket_gethostbyname(PyObject *self, PyObject *args)
Guido van Rossum30a685f1991-06-27 15:51:29 +00003096{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003097 char *name;
3098 sock_addr_t addrbuf;
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003099 PyObject *ret = NULL;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003100
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003101 if (!PyArg_ParseTuple(args, "et:gethostbyname", "idna", &name))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003102 return NULL;
3103 if (setipaddr(name, SAS2SA(&addrbuf), sizeof(addrbuf), AF_INET) < 0)
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003104 goto finally;
3105 ret = makeipaddr(SAS2SA(&addrbuf), sizeof(struct sockaddr_in));
3106finally:
3107 PyMem_Free(name);
3108 return ret;
Guido van Rossum30a685f1991-06-27 15:51:29 +00003109}
3110
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003111PyDoc_STRVAR(gethostbyname_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003112"gethostbyname(host) -> address\n\
3113\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003114Return the IP address (a string of the form '255.255.255.255') for a host.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003115
3116
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003117/* Convenience function common to gethostbyname_ex and gethostbyaddr */
3118
3119static PyObject *
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003120gethost_common(struct hostent *h, struct sockaddr *addr, int alen, int af)
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003121{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003122 char **pch;
3123 PyObject *rtn_tuple = (PyObject *)NULL;
3124 PyObject *name_list = (PyObject *)NULL;
3125 PyObject *addr_list = (PyObject *)NULL;
3126 PyObject *tmp;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003127
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003128 if (h == NULL) {
3129 /* Let's get real error message to return */
3130 set_herror(h_errno);
3131 return NULL;
3132 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00003133
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003134 if (h->h_addrtype != af) {
3135 /* Let's get real error message to return */
3136 PyErr_SetString(socket_error,
3137 (char *)strerror(EAFNOSUPPORT));
Christian Heimesada8c3b2008-03-18 18:26:33 +00003138
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003139 return NULL;
3140 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00003141
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003142 switch (af) {
Guido van Rossum67f7a382002-06-06 21:08:16 +00003143
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003144 case AF_INET:
3145 if (alen < sizeof(struct sockaddr_in))
3146 return NULL;
3147 break;
Guido van Rossum67f7a382002-06-06 21:08:16 +00003148
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00003149#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003150 case AF_INET6:
3151 if (alen < sizeof(struct sockaddr_in6))
3152 return NULL;
3153 break;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003154#endif
Guido van Rossum67f7a382002-06-06 21:08:16 +00003155
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003156 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00003157
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003158 if ((name_list = PyList_New(0)) == NULL)
3159 goto err;
Guido van Rossum67f7a382002-06-06 21:08:16 +00003160
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003161 if ((addr_list = PyList_New(0)) == NULL)
3162 goto err;
Guido van Rossum67f7a382002-06-06 21:08:16 +00003163
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003164 /* SF #1511317: h_aliases can be NULL */
3165 if (h->h_aliases) {
3166 for (pch = h->h_aliases; *pch != NULL; pch++) {
3167 int status;
3168 tmp = PyUnicode_FromString(*pch);
3169 if (tmp == NULL)
3170 goto err;
Guido van Rossum67f7a382002-06-06 21:08:16 +00003171
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003172 status = PyList_Append(name_list, tmp);
3173 Py_DECREF(tmp);
Guido van Rossum67f7a382002-06-06 21:08:16 +00003174
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003175 if (status)
3176 goto err;
3177 }
3178 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00003179
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003180 for (pch = h->h_addr_list; *pch != NULL; pch++) {
3181 int status;
Guido van Rossum67f7a382002-06-06 21:08:16 +00003182
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003183 switch (af) {
Guido van Rossum67f7a382002-06-06 21:08:16 +00003184
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003185 case AF_INET:
3186 {
3187 struct sockaddr_in sin;
3188 memset(&sin, 0, sizeof(sin));
3189 sin.sin_family = af;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003190#ifdef HAVE_SOCKADDR_SA_LEN
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003191 sin.sin_len = sizeof(sin);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003192#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003193 memcpy(&sin.sin_addr, *pch, sizeof(sin.sin_addr));
3194 tmp = makeipaddr((struct sockaddr *)&sin, sizeof(sin));
Guido van Rossum67f7a382002-06-06 21:08:16 +00003195
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003196 if (pch == h->h_addr_list && alen >= sizeof(sin))
3197 memcpy((char *) addr, &sin, sizeof(sin));
3198 break;
3199 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00003200
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00003201#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003202 case AF_INET6:
3203 {
3204 struct sockaddr_in6 sin6;
3205 memset(&sin6, 0, sizeof(sin6));
3206 sin6.sin6_family = af;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003207#ifdef HAVE_SOCKADDR_SA_LEN
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003208 sin6.sin6_len = sizeof(sin6);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003209#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003210 memcpy(&sin6.sin6_addr, *pch, sizeof(sin6.sin6_addr));
3211 tmp = makeipaddr((struct sockaddr *)&sin6,
3212 sizeof(sin6));
Guido van Rossum67f7a382002-06-06 21:08:16 +00003213
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003214 if (pch == h->h_addr_list && alen >= sizeof(sin6))
3215 memcpy((char *) addr, &sin6, sizeof(sin6));
3216 break;
3217 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003218#endif
Guido van Rossum67f7a382002-06-06 21:08:16 +00003219
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003220 default: /* can't happen */
3221 PyErr_SetString(socket_error,
3222 "unsupported address family");
3223 return NULL;
3224 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00003225
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003226 if (tmp == NULL)
3227 goto err;
Guido van Rossum67f7a382002-06-06 21:08:16 +00003228
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003229 status = PyList_Append(addr_list, tmp);
3230 Py_DECREF(tmp);
Guido van Rossum67f7a382002-06-06 21:08:16 +00003231
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003232 if (status)
3233 goto err;
3234 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00003235
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003236 rtn_tuple = Py_BuildValue("sOO", h->h_name, name_list, addr_list);
Guido van Rossum67f7a382002-06-06 21:08:16 +00003237
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003238 err:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003239 Py_XDECREF(name_list);
3240 Py_XDECREF(addr_list);
3241 return rtn_tuple;
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003242}
3243
3244
3245/* Python interface to gethostbyname_ex(name). */
3246
3247/*ARGSUSED*/
3248static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003249socket_gethostbyname_ex(PyObject *self, PyObject *args)
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003250{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003251 char *name;
3252 struct hostent *h;
Martin v. Löwisc16f3bd2003-05-03 09:14:54 +00003253#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003254 struct sockaddr_storage addr;
Martin v. Löwisc16f3bd2003-05-03 09:14:54 +00003255#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003256 struct sockaddr_in addr;
Martin v. Löwisc16f3bd2003-05-03 09:14:54 +00003257#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003258 struct sockaddr *sa;
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003259 PyObject *ret = NULL;
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003260#ifdef HAVE_GETHOSTBYNAME_R
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003261 struct hostent hp_allocated;
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003262#ifdef HAVE_GETHOSTBYNAME_R_3_ARG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003263 struct hostent_data data;
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003264#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003265 char buf[16384];
3266 int buf_len = (sizeof buf) - 1;
3267 int errnop;
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003268#endif
3269#if defined(HAVE_GETHOSTBYNAME_R_3_ARG) || defined(HAVE_GETHOSTBYNAME_R_6_ARG)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003270 int result;
Guido van Rossume9cd07b1999-03-15 21:40:14 +00003271#endif
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003272#endif /* HAVE_GETHOSTBYNAME_R */
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003273
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003274 if (!PyArg_ParseTuple(args, "et:gethostbyname_ex", "idna", &name))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003275 return NULL;
3276 if (setipaddr(name, (struct sockaddr *)&addr, sizeof(addr), AF_INET) < 0)
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003277 goto finally;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003278 Py_BEGIN_ALLOW_THREADS
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003279#ifdef HAVE_GETHOSTBYNAME_R
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003280#if defined(HAVE_GETHOSTBYNAME_R_6_ARG)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003281 result = gethostbyname_r(name, &hp_allocated, buf, buf_len,
3282 &h, &errnop);
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003283#elif defined(HAVE_GETHOSTBYNAME_R_5_ARG)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003284 h = gethostbyname_r(name, &hp_allocated, buf, buf_len, &errnop);
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003285#else /* HAVE_GETHOSTBYNAME_R_3_ARG */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003286 memset((void *) &data, '\0', sizeof(data));
3287 result = gethostbyname_r(name, &hp_allocated, &data);
3288 h = (result != 0) ? NULL : &hp_allocated;
Guido van Rossume9cd07b1999-03-15 21:40:14 +00003289#endif
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003290#else /* not HAVE_GETHOSTBYNAME_R */
Guido van Rossum3baaa131999-03-22 21:44:51 +00003291#ifdef USE_GETHOSTBYNAME_LOCK
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003292 PyThread_acquire_lock(netdb_lock, 1);
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003293#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003294 h = gethostbyname(name);
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003295#endif /* HAVE_GETHOSTBYNAME_R */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003296 Py_END_ALLOW_THREADS
3297 /* Some C libraries would require addr.__ss_family instead of
3298 addr.ss_family.
3299 Therefore, we cast the sockaddr_storage into sockaddr to
3300 access sa_family. */
3301 sa = (struct sockaddr*)&addr;
3302 ret = gethost_common(h, (struct sockaddr *)&addr, sizeof(addr),
3303 sa->sa_family);
Guido van Rossum3baaa131999-03-22 21:44:51 +00003304#ifdef USE_GETHOSTBYNAME_LOCK
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003305 PyThread_release_lock(netdb_lock);
Guido van Rossum955becc1999-03-22 20:14:53 +00003306#endif
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003307finally:
3308 PyMem_Free(name);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003309 return ret;
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003310}
3311
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003312PyDoc_STRVAR(ghbn_ex_doc,
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003313"gethostbyname_ex(host) -> (name, aliaslist, addresslist)\n\
3314\n\
3315Return the true host name, a list of aliases, and a list of IP addresses,\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003316for a host. The host argument is a string giving a host name or IP number.");
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003317
3318
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00003319/* Python interface to gethostbyaddr(IP). */
3320
3321/*ARGSUSED*/
3322static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003323socket_gethostbyaddr(PyObject *self, PyObject *args)
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00003324{
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00003325#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003326 struct sockaddr_storage addr;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003327#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003328 struct sockaddr_in addr;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003329#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003330 struct sockaddr *sa = (struct sockaddr *)&addr;
3331 char *ip_num;
3332 struct hostent *h;
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003333 PyObject *ret = NULL;
Guido van Rossum4f199ea1998-04-09 20:56:35 +00003334#ifdef HAVE_GETHOSTBYNAME_R
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003335 struct hostent hp_allocated;
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003336#ifdef HAVE_GETHOSTBYNAME_R_3_ARG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003337 struct hostent_data data;
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003338#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003339 /* glibcs up to 2.10 assume that the buf argument to
3340 gethostbyaddr_r is 8-byte aligned, which at least llvm-gcc
3341 does not ensure. The attribute below instructs the compiler
3342 to maintain this alignment. */
3343 char buf[16384] Py_ALIGNED(8);
3344 int buf_len = (sizeof buf) - 1;
3345 int errnop;
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003346#endif
3347#if defined(HAVE_GETHOSTBYNAME_R_3_ARG) || defined(HAVE_GETHOSTBYNAME_R_6_ARG)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003348 int result;
Guido van Rossume9cd07b1999-03-15 21:40:14 +00003349#endif
Guido van Rossum4f199ea1998-04-09 20:56:35 +00003350#endif /* HAVE_GETHOSTBYNAME_R */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003351 char *ap;
3352 int al;
3353 int af;
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00003354
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003355 if (!PyArg_ParseTuple(args, "et:gethostbyaddr", "idna", &ip_num))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003356 return NULL;
3357 af = AF_UNSPEC;
3358 if (setipaddr(ip_num, sa, sizeof(addr), af) < 0)
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003359 goto finally;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003360 af = sa->sa_family;
3361 ap = NULL;
3362 al = 0;
3363 switch (af) {
3364 case AF_INET:
3365 ap = (char *)&((struct sockaddr_in *)sa)->sin_addr;
3366 al = sizeof(((struct sockaddr_in *)sa)->sin_addr);
3367 break;
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00003368#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003369 case AF_INET6:
3370 ap = (char *)&((struct sockaddr_in6 *)sa)->sin6_addr;
3371 al = sizeof(((struct sockaddr_in6 *)sa)->sin6_addr);
3372 break;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003373#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003374 default:
3375 PyErr_SetString(socket_error, "unsupported address family");
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003376 goto finally;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003377 }
3378 Py_BEGIN_ALLOW_THREADS
Guido van Rossum4f199ea1998-04-09 20:56:35 +00003379#ifdef HAVE_GETHOSTBYNAME_R
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003380#if defined(HAVE_GETHOSTBYNAME_R_6_ARG)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003381 result = gethostbyaddr_r(ap, al, af,
3382 &hp_allocated, buf, buf_len,
3383 &h, &errnop);
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003384#elif defined(HAVE_GETHOSTBYNAME_R_5_ARG)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003385 h = gethostbyaddr_r(ap, al, af,
3386 &hp_allocated, buf, buf_len, &errnop);
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003387#else /* HAVE_GETHOSTBYNAME_R_3_ARG */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003388 memset((void *) &data, '\0', sizeof(data));
3389 result = gethostbyaddr_r(ap, al, af, &hp_allocated, &data);
3390 h = (result != 0) ? NULL : &hp_allocated;
Guido van Rossume9cd07b1999-03-15 21:40:14 +00003391#endif
Guido van Rossum4f199ea1998-04-09 20:56:35 +00003392#else /* not HAVE_GETHOSTBYNAME_R */
Guido van Rossum3baaa131999-03-22 21:44:51 +00003393#ifdef USE_GETHOSTBYNAME_LOCK
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003394 PyThread_acquire_lock(netdb_lock, 1);
Guido van Rossum4f199ea1998-04-09 20:56:35 +00003395#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003396 h = gethostbyaddr(ap, al, af);
Guido van Rossum4f199ea1998-04-09 20:56:35 +00003397#endif /* HAVE_GETHOSTBYNAME_R */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003398 Py_END_ALLOW_THREADS
3399 ret = gethost_common(h, (struct sockaddr *)&addr, sizeof(addr), af);
Guido van Rossum3baaa131999-03-22 21:44:51 +00003400#ifdef USE_GETHOSTBYNAME_LOCK
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003401 PyThread_release_lock(netdb_lock);
Guido van Rossum3baaa131999-03-22 21:44:51 +00003402#endif
Martin v. Löwisfc0275a2010-08-22 19:33:47 +00003403finally:
3404 PyMem_Free(ip_num);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003405 return ret;
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00003406}
3407
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003408PyDoc_STRVAR(gethostbyaddr_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003409"gethostbyaddr(host) -> (name, aliaslist, addresslist)\n\
3410\n\
3411Return the true host name, a list of aliases, and a list of IP addresses,\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003412for a host. The host argument is a string giving a host name or IP number.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003413
Guido van Rossum30a685f1991-06-27 15:51:29 +00003414
3415/* Python interface to getservbyname(name).
3416 This only returns the port number, since the other info is already
3417 known or not useful (like the list of aliases). */
3418
3419/*ARGSUSED*/
Guido van Rossum73624e91994-10-10 17:59:00 +00003420static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003421socket_getservbyname(PyObject *self, PyObject *args)
Guido van Rossum30a685f1991-06-27 15:51:29 +00003422{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003423 char *name, *proto=NULL;
3424 struct servent *sp;
3425 if (!PyArg_ParseTuple(args, "s|s:getservbyname", &name, &proto))
3426 return NULL;
3427 Py_BEGIN_ALLOW_THREADS
3428 sp = getservbyname(name, proto);
3429 Py_END_ALLOW_THREADS
3430 if (sp == NULL) {
3431 PyErr_SetString(socket_error, "service/proto not found");
3432 return NULL;
3433 }
3434 return PyLong_FromLong((long) ntohs(sp->s_port));
Guido van Rossum30a685f1991-06-27 15:51:29 +00003435}
3436
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003437PyDoc_STRVAR(getservbyname_doc,
Barry Warsaw11b91a02004-06-28 00:50:43 +00003438"getservbyname(servicename[, protocolname]) -> integer\n\
Guido van Rossum82a5c661998-07-07 20:45:43 +00003439\n\
3440Return a port number from a service name and protocol name.\n\
Barry Warsaw11b91a02004-06-28 00:50:43 +00003441The optional protocol name, if given, should be 'tcp' or 'udp',\n\
3442otherwise any protocol will match.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003443
Guido van Rossum30a685f1991-06-27 15:51:29 +00003444
Barry Warsaw11b91a02004-06-28 00:50:43 +00003445/* Python interface to getservbyport(port).
3446 This only returns the service name, since the other info is already
3447 known or not useful (like the list of aliases). */
3448
3449/*ARGSUSED*/
3450static PyObject *
3451socket_getservbyport(PyObject *self, PyObject *args)
3452{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003453 int port;
3454 char *proto=NULL;
3455 struct servent *sp;
3456 if (!PyArg_ParseTuple(args, "i|s:getservbyport", &port, &proto))
3457 return NULL;
3458 if (port < 0 || port > 0xffff) {
3459 PyErr_SetString(
3460 PyExc_OverflowError,
3461 "getservbyport: port must be 0-65535.");
3462 return NULL;
3463 }
3464 Py_BEGIN_ALLOW_THREADS
3465 sp = getservbyport(htons((short)port), proto);
3466 Py_END_ALLOW_THREADS
3467 if (sp == NULL) {
3468 PyErr_SetString(socket_error, "port/proto not found");
3469 return NULL;
3470 }
3471 return PyUnicode_FromString(sp->s_name);
Barry Warsaw11b91a02004-06-28 00:50:43 +00003472}
3473
3474PyDoc_STRVAR(getservbyport_doc,
3475"getservbyport(port[, protocolname]) -> string\n\
3476\n\
3477Return the service name from a port number and protocol name.\n\
3478The optional protocol name, if given, should be 'tcp' or 'udp',\n\
3479otherwise any protocol will match.");
3480
Guido van Rossum3901d851996-12-19 16:35:04 +00003481/* Python interface to getprotobyname(name).
3482 This only returns the protocol number, since the other info is
3483 already known or not useful (like the list of aliases). */
3484
3485/*ARGSUSED*/
3486static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003487socket_getprotobyname(PyObject *self, PyObject *args)
Guido van Rossum3901d851996-12-19 16:35:04 +00003488{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003489 char *name;
3490 struct protoent *sp;
3491 if (!PyArg_ParseTuple(args, "s:getprotobyname", &name))
3492 return NULL;
3493 Py_BEGIN_ALLOW_THREADS
3494 sp = getprotobyname(name);
3495 Py_END_ALLOW_THREADS
3496 if (sp == NULL) {
3497 PyErr_SetString(socket_error, "protocol not found");
3498 return NULL;
3499 }
3500 return PyLong_FromLong((long) sp->p_proto);
Guido van Rossum3901d851996-12-19 16:35:04 +00003501}
3502
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003503PyDoc_STRVAR(getprotobyname_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003504"getprotobyname(name) -> integer\n\
3505\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003506Return the protocol number for the named protocol. (Rarely used.)");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003507
Guido van Rossum3901d851996-12-19 16:35:04 +00003508
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00003509#ifndef NO_DUP
3510/* dup() function for socket fds */
3511
3512static PyObject *
3513socket_dup(PyObject *self, PyObject *fdobj)
3514{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003515 SOCKET_T fd, newfd;
3516 PyObject *newfdobj;
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00003517
3518
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003519 fd = PyLong_AsSocket_t(fdobj);
3520 if (fd == (SOCKET_T)(-1) && PyErr_Occurred())
3521 return NULL;
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00003522
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003523 newfd = dup_socket(fd);
3524 if (newfd == INVALID_SOCKET)
3525 return set_error();
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00003526
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003527 newfdobj = PyLong_FromSocket_t(newfd);
3528 if (newfdobj == NULL)
3529 SOCKETCLOSE(newfd);
3530 return newfdobj;
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00003531}
3532
3533PyDoc_STRVAR(dup_doc,
3534"dup(integer) -> integer\n\
3535\n\
3536Duplicate an integer socket file descriptor. This is like os.dup(), but for\n\
3537sockets; on some platforms os.dup() won't work for socket file descriptors.");
3538#endif
3539
3540
Dave Cole331708b2004-08-09 04:51:41 +00003541#ifdef HAVE_SOCKETPAIR
3542/* Create a pair of sockets using the socketpair() function.
Dave Cole07fda7e2004-08-23 05:16:23 +00003543 Arguments as for socket() except the default family is AF_UNIX if
Dave Colee8bbfe42004-08-26 00:51:16 +00003544 defined on the platform; otherwise, the default is AF_INET. */
Dave Cole331708b2004-08-09 04:51:41 +00003545
3546/*ARGSUSED*/
3547static PyObject *
3548socket_socketpair(PyObject *self, PyObject *args)
3549{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003550 PySocketSockObject *s0 = NULL, *s1 = NULL;
3551 SOCKET_T sv[2];
3552 int family, type = SOCK_STREAM, proto = 0;
3553 PyObject *res = NULL;
Dave Cole331708b2004-08-09 04:51:41 +00003554
3555#if defined(AF_UNIX)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003556 family = AF_UNIX;
Dave Cole331708b2004-08-09 04:51:41 +00003557#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003558 family = AF_INET;
Dave Cole331708b2004-08-09 04:51:41 +00003559#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003560 if (!PyArg_ParseTuple(args, "|iii:socketpair",
3561 &family, &type, &proto))
3562 return NULL;
3563 /* Create a pair of socket fds */
3564 if (socketpair(family, type, proto, sv) < 0)
3565 return set_error();
3566 s0 = new_sockobject(sv[0], family, type, proto);
3567 if (s0 == NULL)
3568 goto finally;
3569 s1 = new_sockobject(sv[1], family, type, proto);
3570 if (s1 == NULL)
3571 goto finally;
3572 res = PyTuple_Pack(2, s0, s1);
Dave Cole331708b2004-08-09 04:51:41 +00003573
3574finally:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003575 if (res == NULL) {
3576 if (s0 == NULL)
3577 SOCKETCLOSE(sv[0]);
3578 if (s1 == NULL)
3579 SOCKETCLOSE(sv[1]);
3580 }
3581 Py_XDECREF(s0);
3582 Py_XDECREF(s1);
3583 return res;
Dave Cole331708b2004-08-09 04:51:41 +00003584}
3585
3586PyDoc_STRVAR(socketpair_doc,
3587"socketpair([family[, type[, proto]]]) -> (socket object, socket object)\n\
3588\n\
3589Create a pair of socket objects from the sockets returned by the platform\n\
3590socketpair() function.\n\
Dave Cole07fda7e2004-08-23 05:16:23 +00003591The arguments are the same as for socket() except the default family is\n\
Dave Colee8bbfe42004-08-26 00:51:16 +00003592AF_UNIX if defined on the platform; otherwise, the default is AF_INET.");
Dave Cole331708b2004-08-09 04:51:41 +00003593
3594#endif /* HAVE_SOCKETPAIR */
3595
3596
Guido van Rossum006bf911996-06-12 04:04:55 +00003597static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003598socket_ntohs(PyObject *self, PyObject *args)
Guido van Rossum006bf911996-06-12 04:04:55 +00003599{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003600 int x1, x2;
Guido van Rossum006bf911996-06-12 04:04:55 +00003601
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003602 if (!PyArg_ParseTuple(args, "i:ntohs", &x1)) {
3603 return NULL;
3604 }
3605 if (x1 < 0) {
3606 PyErr_SetString(PyExc_OverflowError,
3607 "can't convert negative number to unsigned long");
3608 return NULL;
3609 }
3610 x2 = (unsigned int)ntohs((unsigned short)x1);
3611 return PyLong_FromLong(x2);
Guido van Rossum006bf911996-06-12 04:04:55 +00003612}
3613
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003614PyDoc_STRVAR(ntohs_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003615"ntohs(integer) -> integer\n\
3616\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003617Convert a 16-bit integer from network to host byte order.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003618
3619
Guido van Rossum006bf911996-06-12 04:04:55 +00003620static PyObject *
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003621socket_ntohl(PyObject *self, PyObject *arg)
Guido van Rossum006bf911996-06-12 04:04:55 +00003622{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003623 unsigned long x;
Guido van Rossum006bf911996-06-12 04:04:55 +00003624
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003625 if (PyLong_Check(arg)) {
3626 x = PyLong_AsUnsignedLong(arg);
3627 if (x == (unsigned long) -1 && PyErr_Occurred())
3628 return NULL;
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003629#if SIZEOF_LONG > 4
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003630 {
3631 unsigned long y;
3632 /* only want the trailing 32 bits */
3633 y = x & 0xFFFFFFFFUL;
3634 if (y ^ x)
3635 return PyErr_Format(PyExc_OverflowError,
3636 "long int larger than 32 bits");
3637 x = y;
3638 }
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003639#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003640 }
3641 else
3642 return PyErr_Format(PyExc_TypeError,
3643 "expected int/long, %s found",
3644 Py_TYPE(arg)->tp_name);
3645 if (x == (unsigned long) -1 && PyErr_Occurred())
3646 return NULL;
3647 return PyLong_FromUnsignedLong(ntohl(x));
Guido van Rossum006bf911996-06-12 04:04:55 +00003648}
3649
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003650PyDoc_STRVAR(ntohl_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003651"ntohl(integer) -> integer\n\
3652\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003653Convert a 32-bit integer from network to host byte order.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003654
3655
Guido van Rossum006bf911996-06-12 04:04:55 +00003656static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003657socket_htons(PyObject *self, PyObject *args)
Guido van Rossum006bf911996-06-12 04:04:55 +00003658{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003659 int x1, x2;
Guido van Rossum006bf911996-06-12 04:04:55 +00003660
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003661 if (!PyArg_ParseTuple(args, "i:htons", &x1)) {
3662 return NULL;
3663 }
3664 if (x1 < 0) {
3665 PyErr_SetString(PyExc_OverflowError,
3666 "can't convert negative number to unsigned long");
3667 return NULL;
3668 }
3669 x2 = (unsigned int)htons((unsigned short)x1);
3670 return PyLong_FromLong(x2);
Guido van Rossum006bf911996-06-12 04:04:55 +00003671}
3672
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003673PyDoc_STRVAR(htons_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003674"htons(integer) -> integer\n\
3675\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003676Convert a 16-bit integer from host to network byte order.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003677
3678
Guido van Rossum006bf911996-06-12 04:04:55 +00003679static PyObject *
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003680socket_htonl(PyObject *self, PyObject *arg)
Guido van Rossum006bf911996-06-12 04:04:55 +00003681{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003682 unsigned long x;
Guido van Rossum006bf911996-06-12 04:04:55 +00003683
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003684 if (PyLong_Check(arg)) {
3685 x = PyLong_AsUnsignedLong(arg);
3686 if (x == (unsigned long) -1 && PyErr_Occurred())
3687 return NULL;
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003688#if SIZEOF_LONG > 4
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003689 {
3690 unsigned long y;
3691 /* only want the trailing 32 bits */
3692 y = x & 0xFFFFFFFFUL;
3693 if (y ^ x)
3694 return PyErr_Format(PyExc_OverflowError,
3695 "long int larger than 32 bits");
3696 x = y;
3697 }
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003698#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003699 }
3700 else
3701 return PyErr_Format(PyExc_TypeError,
3702 "expected int/long, %s found",
3703 Py_TYPE(arg)->tp_name);
3704 return PyLong_FromUnsignedLong(htonl((unsigned long)x));
Guido van Rossum006bf911996-06-12 04:04:55 +00003705}
3706
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003707PyDoc_STRVAR(htonl_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003708"htonl(integer) -> integer\n\
3709\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003710Convert a 32-bit integer from host to network byte order.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003711
Guido van Rossum3eede5a2002-06-07 02:08:35 +00003712/* socket.inet_aton() and socket.inet_ntoa() functions. */
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003713
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003714PyDoc_STRVAR(inet_aton_doc,
Guido van Rossum7d0a8262007-05-21 23:13:11 +00003715"inet_aton(string) -> bytes giving packed 32-bit IP representation\n\
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003716\n\
Guido van Rossumc6a164b1999-08-20 19:11:27 +00003717Convert 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 +00003718binary format used in low-level network functions.");
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003719
3720static PyObject*
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003721socket_inet_aton(PyObject *self, PyObject *args)
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003722{
Guido van Rossuma2e48551999-09-09 15:42:59 +00003723#ifndef INADDR_NONE
3724#define INADDR_NONE (-1)
3725#endif
Neal Norwitz88f115b2003-02-13 02:15:42 +00003726#ifdef HAVE_INET_ATON
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003727 struct in_addr buf;
Thomas Wouters477c8d52006-05-27 19:21:47 +00003728#endif
3729
3730#if !defined(HAVE_INET_ATON) || defined(USE_INET_ATON_WEAKLINK)
Benjamin Petersonf91df042009-02-13 02:50:59 +00003731#if (SIZEOF_INT != 4)
3732#error "Not sure if in_addr_t exists and int is not 32-bits."
3733#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003734 /* Have to use inet_addr() instead */
3735 unsigned int packed_addr;
Tim Peters1df9fdd2003-02-13 03:13:40 +00003736#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003737 char *ip_addr;
Guido van Rossumc6a164b1999-08-20 19:11:27 +00003738
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003739 if (!PyArg_ParseTuple(args, "s:inet_aton", &ip_addr))
3740 return NULL;
Guido van Rossumad05cdf2003-02-12 23:08:22 +00003741
Tim Peters1df9fdd2003-02-13 03:13:40 +00003742
3743#ifdef HAVE_INET_ATON
Thomas Wouters477c8d52006-05-27 19:21:47 +00003744
3745#ifdef USE_INET_ATON_WEAKLINK
3746 if (inet_aton != NULL) {
3747#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003748 if (inet_aton(ip_addr, &buf))
3749 return PyBytes_FromStringAndSize((char *)(&buf),
3750 sizeof(buf));
Guido van Rossumad05cdf2003-02-12 23:08:22 +00003751
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003752 PyErr_SetString(socket_error,
3753 "illegal IP address string passed to inet_aton");
3754 return NULL;
Guido van Rossumad05cdf2003-02-12 23:08:22 +00003755
Thomas Wouters477c8d52006-05-27 19:21:47 +00003756#ifdef USE_INET_ATON_WEAKLINK
3757 } else {
3758#endif
3759
3760#endif
3761
3762#if !defined(HAVE_INET_ATON) || defined(USE_INET_ATON_WEAKLINK)
3763
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003764 /* special-case this address as inet_addr might return INADDR_NONE
3765 * for this */
3766 if (strcmp(ip_addr, "255.255.255.255") == 0) {
3767 packed_addr = 0xFFFFFFFF;
3768 } else {
Thomas Wouters477c8d52006-05-27 19:21:47 +00003769
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003770 packed_addr = inet_addr(ip_addr);
Guido van Rossumc6a164b1999-08-20 19:11:27 +00003771
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003772 if (packed_addr == INADDR_NONE) { /* invalid address */
3773 PyErr_SetString(socket_error,
3774 "illegal IP address string passed to inet_aton");
3775 return NULL;
3776 }
3777 }
3778 return PyBytes_FromStringAndSize((char *) &packed_addr,
3779 sizeof(packed_addr));
Thomas Wouters477c8d52006-05-27 19:21:47 +00003780
3781#ifdef USE_INET_ATON_WEAKLINK
3782 }
3783#endif
3784
Guido van Rossumad05cdf2003-02-12 23:08:22 +00003785#endif
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003786}
3787
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003788PyDoc_STRVAR(inet_ntoa_doc,
Fred Drakee0661342000-03-07 14:05:16 +00003789"inet_ntoa(packed_ip) -> ip_address_string\n\
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003790\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003791Convert an IP address from 32-bit packed binary format to string format");
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003792
3793static PyObject*
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003794socket_inet_ntoa(PyObject *self, PyObject *args)
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003795{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003796 char *packed_str;
3797 int addr_len;
3798 struct in_addr packed_addr;
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003799
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003800 if (!PyArg_ParseTuple(args, "y#:inet_ntoa", &packed_str, &addr_len)) {
3801 return NULL;
3802 }
Guido van Rossum48a680c2001-03-02 06:34:14 +00003803
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003804 if (addr_len != sizeof(packed_addr)) {
3805 PyErr_SetString(socket_error,
3806 "packed IP wrong length for inet_ntoa");
3807 return NULL;
3808 }
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003809
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003810 memcpy(&packed_addr, packed_str, addr_len);
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003811
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003812 return PyUnicode_FromString(inet_ntoa(packed_addr));
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003813}
Guido van Rossum82a5c661998-07-07 20:45:43 +00003814
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003815#ifdef HAVE_INET_PTON
3816
3817PyDoc_STRVAR(inet_pton_doc,
3818"inet_pton(af, ip) -> packed IP address string\n\
3819\n\
3820Convert an IP address from string format to a packed string suitable\n\
3821for use with low-level network functions.");
3822
3823static PyObject *
3824socket_inet_pton(PyObject *self, PyObject *args)
3825{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003826 int af;
3827 char* ip;
3828 int retval;
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003829#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003830 char packed[MAX(sizeof(struct in_addr), sizeof(struct in6_addr))];
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003831#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003832 char packed[sizeof(struct in_addr)];
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003833#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003834 if (!PyArg_ParseTuple(args, "is:inet_pton", &af, &ip)) {
3835 return NULL;
3836 }
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003837
Martin v. Löwis04697e82004-06-02 12:35:29 +00003838#if !defined(ENABLE_IPV6) && defined(AF_INET6)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003839 if(af == AF_INET6) {
3840 PyErr_SetString(socket_error,
3841 "can't use AF_INET6, IPv6 is disabled");
3842 return NULL;
3843 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00003844#endif
Martin v. Löwis10649092003-08-05 06:25:06 +00003845
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003846 retval = inet_pton(af, ip, packed);
3847 if (retval < 0) {
3848 PyErr_SetFromErrno(socket_error);
3849 return NULL;
3850 } else if (retval == 0) {
3851 PyErr_SetString(socket_error,
3852 "illegal IP address string passed to inet_pton");
3853 return NULL;
3854 } else if (af == AF_INET) {
3855 return PyBytes_FromStringAndSize(packed,
3856 sizeof(struct in_addr));
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003857#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003858 } else if (af == AF_INET6) {
3859 return PyBytes_FromStringAndSize(packed,
3860 sizeof(struct in6_addr));
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003861#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003862 } else {
3863 PyErr_SetString(socket_error, "unknown address family");
3864 return NULL;
3865 }
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003866}
Thomas Wouters477c8d52006-05-27 19:21:47 +00003867
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003868PyDoc_STRVAR(inet_ntop_doc,
3869"inet_ntop(af, packed_ip) -> string formatted IP address\n\
3870\n\
3871Convert a packed IP address of the given family to string format.");
3872
3873static PyObject *
3874socket_inet_ntop(PyObject *self, PyObject *args)
3875{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003876 int af;
3877 char* packed;
3878 int len;
3879 const char* retval;
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003880#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003881 char ip[MAX(INET_ADDRSTRLEN, INET6_ADDRSTRLEN) + 1];
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003882#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003883 char ip[INET_ADDRSTRLEN + 1];
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003884#endif
Thomas Wouters477c8d52006-05-27 19:21:47 +00003885
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003886 /* Guarantee NUL-termination for PyUnicode_FromString() below */
3887 memset((void *) &ip[0], '\0', sizeof(ip));
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003888
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003889 if (!PyArg_ParseTuple(args, "iy#:inet_ntop", &af, &packed, &len)) {
3890 return NULL;
3891 }
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003892
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003893 if (af == AF_INET) {
3894 if (len != sizeof(struct in_addr)) {
3895 PyErr_SetString(PyExc_ValueError,
3896 "invalid length of packed IP address string");
3897 return NULL;
3898 }
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003899#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003900 } else if (af == AF_INET6) {
3901 if (len != sizeof(struct in6_addr)) {
3902 PyErr_SetString(PyExc_ValueError,
3903 "invalid length of packed IP address string");
3904 return NULL;
3905 }
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003906#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003907 } else {
3908 PyErr_Format(PyExc_ValueError,
3909 "unknown address family %d", af);
3910 return NULL;
3911 }
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003912
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003913 retval = inet_ntop(af, packed, ip, sizeof(ip));
3914 if (!retval) {
3915 PyErr_SetFromErrno(socket_error);
3916 return NULL;
3917 } else {
3918 return PyUnicode_FromString(retval);
3919 }
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003920
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003921 /* NOTREACHED */
3922 PyErr_SetString(PyExc_RuntimeError, "invalid handling of inet_ntop");
3923 return NULL;
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003924}
3925
3926#endif /* HAVE_INET_PTON */
3927
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003928/* Python interface to getaddrinfo(host, port). */
3929
3930/*ARGSUSED*/
3931static PyObject *
Giampaolo Rodolàccfb91c2010-08-17 15:30:23 +00003932socket_getaddrinfo(PyObject *self, PyObject *args, PyObject* kwargs)
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003933{
Giampaolo Rodolàccfb91c2010-08-17 15:30:23 +00003934 static char* kwnames[] = {"host", "port", "family", "type", "proto",
3935 "flags", 0};
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003936 struct addrinfo hints, *res;
3937 struct addrinfo *res0 = NULL;
3938 PyObject *hobj = NULL;
3939 PyObject *pobj = (PyObject *)NULL;
3940 char pbuf[30];
3941 char *hptr, *pptr;
3942 int family, socktype, protocol, flags;
3943 int error;
3944 PyObject *all = (PyObject *)NULL;
3945 PyObject *idna = NULL;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003946
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003947 family = socktype = protocol = flags = 0;
3948 family = AF_UNSPEC;
Giampaolo Rodolàccfb91c2010-08-17 15:30:23 +00003949 if (!PyArg_ParseTupleAndKeywords(args, kwargs, "OO|iiii:getaddrinfo",
3950 kwnames, &hobj, &pobj, &family, &socktype,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003951 &protocol, &flags)) {
3952 return NULL;
3953 }
3954 if (hobj == Py_None) {
3955 hptr = NULL;
3956 } else if (PyUnicode_Check(hobj)) {
3957 idna = PyObject_CallMethod(hobj, "encode", "s", "idna");
3958 if (!idna)
3959 return NULL;
3960 assert(PyBytes_Check(idna));
3961 hptr = PyBytes_AS_STRING(idna);
3962 } else if (PyBytes_Check(hobj)) {
3963 hptr = PyBytes_AsString(hobj);
3964 } else {
3965 PyErr_SetString(PyExc_TypeError,
3966 "getaddrinfo() argument 1 must be string or None");
3967 return NULL;
3968 }
3969 if (PyLong_CheckExact(pobj)) {
3970 long value = PyLong_AsLong(pobj);
3971 if (value == -1 && PyErr_Occurred())
3972 goto err;
3973 PyOS_snprintf(pbuf, sizeof(pbuf), "%ld", value);
3974 pptr = pbuf;
3975 } else if (PyUnicode_Check(pobj)) {
3976 pptr = _PyUnicode_AsString(pobj);
3977 } else if (PyBytes_Check(pobj)) {
3978 pptr = PyBytes_AsString(pobj);
3979 } else if (pobj == Py_None) {
3980 pptr = (char *)NULL;
3981 } else {
3982 PyErr_SetString(socket_error, "Int or String expected");
3983 goto err;
3984 }
3985 memset(&hints, 0, sizeof(hints));
3986 hints.ai_family = family;
3987 hints.ai_socktype = socktype;
3988 hints.ai_protocol = protocol;
3989 hints.ai_flags = flags;
3990 Py_BEGIN_ALLOW_THREADS
3991 ACQUIRE_GETADDRINFO_LOCK
3992 error = getaddrinfo(hptr, pptr, &hints, &res0);
3993 Py_END_ALLOW_THREADS
3994 RELEASE_GETADDRINFO_LOCK /* see comment in setipaddr() */
3995 if (error) {
3996 set_gaierror(error);
3997 goto err;
3998 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003999
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004000 if ((all = PyList_New(0)) == NULL)
4001 goto err;
4002 for (res = res0; res; res = res->ai_next) {
4003 PyObject *single;
4004 PyObject *addr =
4005 makesockaddr(-1, res->ai_addr, res->ai_addrlen, protocol);
4006 if (addr == NULL)
4007 goto err;
4008 single = Py_BuildValue("iiisO", res->ai_family,
4009 res->ai_socktype, res->ai_protocol,
4010 res->ai_canonname ? res->ai_canonname : "",
4011 addr);
4012 Py_DECREF(addr);
4013 if (single == NULL)
4014 goto err;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004015
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004016 if (PyList_Append(all, single))
4017 goto err;
4018 Py_XDECREF(single);
4019 }
4020 Py_XDECREF(idna);
4021 if (res0)
4022 freeaddrinfo(res0);
4023 return all;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004024 err:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004025 Py_XDECREF(all);
4026 Py_XDECREF(idna);
4027 if (res0)
4028 freeaddrinfo(res0);
4029 return (PyObject *)NULL;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004030}
4031
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004032PyDoc_STRVAR(getaddrinfo_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00004033"getaddrinfo(host, port [, family, socktype, proto, flags])\n\
4034 -> list of (family, socktype, proto, canonname, sockaddr)\n\
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004035\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004036Resolve host and port into addrinfo struct.");
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004037
4038/* Python interface to getnameinfo(sa, flags). */
4039
4040/*ARGSUSED*/
4041static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004042socket_getnameinfo(PyObject *self, PyObject *args)
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004043{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004044 PyObject *sa = (PyObject *)NULL;
4045 int flags;
4046 char *hostp;
4047 int port, flowinfo, scope_id;
4048 char hbuf[NI_MAXHOST], pbuf[NI_MAXSERV];
4049 struct addrinfo hints, *res = NULL;
4050 int error;
4051 PyObject *ret = (PyObject *)NULL;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004052
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004053 flags = flowinfo = scope_id = 0;
4054 if (!PyArg_ParseTuple(args, "Oi:getnameinfo", &sa, &flags))
4055 return NULL;
4056 if (!PyTuple_Check(sa)) {
4057 PyErr_SetString(PyExc_TypeError,
4058 "getnameinfo() argument 1 must be a tuple");
4059 return NULL;
4060 }
4061 if (!PyArg_ParseTuple(sa, "si|ii",
4062 &hostp, &port, &flowinfo, &scope_id))
4063 return NULL;
4064 PyOS_snprintf(pbuf, sizeof(pbuf), "%d", port);
4065 memset(&hints, 0, sizeof(hints));
4066 hints.ai_family = AF_UNSPEC;
4067 hints.ai_socktype = SOCK_DGRAM; /* make numeric port happy */
Martin v. Löwis112c0f32010-08-25 07:38:15 +00004068 hints.ai_flags = AI_NUMERICHOST; /* don't do any name resolution */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004069 Py_BEGIN_ALLOW_THREADS
4070 ACQUIRE_GETADDRINFO_LOCK
4071 error = getaddrinfo(hostp, pbuf, &hints, &res);
4072 Py_END_ALLOW_THREADS
4073 RELEASE_GETADDRINFO_LOCK /* see comment in setipaddr() */
4074 if (error) {
4075 set_gaierror(error);
4076 goto fail;
4077 }
4078 if (res->ai_next) {
4079 PyErr_SetString(socket_error,
4080 "sockaddr resolved to multiple addresses");
4081 goto fail;
4082 }
4083 switch (res->ai_family) {
4084 case AF_INET:
4085 {
4086 if (PyTuple_GET_SIZE(sa) != 2) {
4087 PyErr_SetString(socket_error,
4088 "IPv4 sockaddr must be 2 tuple");
4089 goto fail;
4090 }
4091 break;
4092 }
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00004093#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004094 case AF_INET6:
4095 {
4096 struct sockaddr_in6 *sin6;
4097 sin6 = (struct sockaddr_in6 *)res->ai_addr;
4098 sin6->sin6_flowinfo = flowinfo;
4099 sin6->sin6_scope_id = scope_id;
4100 break;
4101 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004102#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004103 }
Antoine Pitrou19467d22010-08-17 19:33:30 +00004104 error = getnameinfo(res->ai_addr, (socklen_t) res->ai_addrlen,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004105 hbuf, sizeof(hbuf), pbuf, sizeof(pbuf), flags);
4106 if (error) {
4107 set_gaierror(error);
4108 goto fail;
4109 }
4110 ret = Py_BuildValue("ss", hbuf, pbuf);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004111
4112fail:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004113 if (res)
4114 freeaddrinfo(res);
4115 return ret;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004116}
4117
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004118PyDoc_STRVAR(getnameinfo_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00004119"getnameinfo(sockaddr, flags) --> (host, port)\n\
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004120\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004121Get host and port for a sockaddr.");
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004122
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004123
4124/* Python API to getting and setting the default timeout value. */
4125
4126static PyObject *
4127socket_getdefaulttimeout(PyObject *self)
4128{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004129 if (defaulttimeout < 0.0) {
4130 Py_INCREF(Py_None);
4131 return Py_None;
4132 }
4133 else
4134 return PyFloat_FromDouble(defaulttimeout);
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004135}
4136
4137PyDoc_STRVAR(getdefaulttimeout_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00004138"getdefaulttimeout() -> timeout\n\
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004139\n\
4140Returns the default timeout in floating seconds for new socket objects.\n\
4141A value of None indicates that new socket objects have no timeout.\n\
4142When the socket module is first imported, the default is None.");
4143
4144static PyObject *
4145socket_setdefaulttimeout(PyObject *self, PyObject *arg)
4146{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004147 double timeout;
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004148
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004149 if (arg == Py_None)
4150 timeout = -1.0;
4151 else {
4152 timeout = PyFloat_AsDouble(arg);
4153 if (timeout < 0.0) {
4154 if (!PyErr_Occurred())
4155 PyErr_SetString(PyExc_ValueError,
4156 "Timeout value out of range");
4157 return NULL;
4158 }
4159 }
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004160
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004161 defaulttimeout = timeout;
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004162
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004163 Py_INCREF(Py_None);
4164 return Py_None;
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004165}
4166
4167PyDoc_STRVAR(setdefaulttimeout_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00004168"setdefaulttimeout(timeout)\n\
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004169\n\
4170Set the default timeout in floating seconds for new socket objects.\n\
4171A value of None indicates that new socket objects have no timeout.\n\
4172When the socket module is first imported, the default is None.");
4173
4174
Guido van Rossum30a685f1991-06-27 15:51:29 +00004175/* List of functions exported by this module. */
4176
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004177static PyMethodDef socket_methods[] = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004178 {"gethostbyname", socket_gethostbyname,
4179 METH_VARARGS, gethostbyname_doc},
4180 {"gethostbyname_ex", socket_gethostbyname_ex,
4181 METH_VARARGS, ghbn_ex_doc},
4182 {"gethostbyaddr", socket_gethostbyaddr,
4183 METH_VARARGS, gethostbyaddr_doc},
4184 {"gethostname", socket_gethostname,
4185 METH_NOARGS, gethostname_doc},
4186 {"getservbyname", socket_getservbyname,
4187 METH_VARARGS, getservbyname_doc},
4188 {"getservbyport", socket_getservbyport,
4189 METH_VARARGS, getservbyport_doc},
4190 {"getprotobyname", socket_getprotobyname,
4191 METH_VARARGS, getprotobyname_doc},
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00004192#ifndef NO_DUP
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004193 {"dup", socket_dup,
4194 METH_O, dup_doc},
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00004195#endif
Dave Cole331708b2004-08-09 04:51:41 +00004196#ifdef HAVE_SOCKETPAIR
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004197 {"socketpair", socket_socketpair,
4198 METH_VARARGS, socketpair_doc},
Dave Cole331708b2004-08-09 04:51:41 +00004199#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004200 {"ntohs", socket_ntohs,
4201 METH_VARARGS, ntohs_doc},
4202 {"ntohl", socket_ntohl,
4203 METH_O, ntohl_doc},
4204 {"htons", socket_htons,
4205 METH_VARARGS, htons_doc},
4206 {"htonl", socket_htonl,
4207 METH_O, htonl_doc},
4208 {"inet_aton", socket_inet_aton,
4209 METH_VARARGS, inet_aton_doc},
4210 {"inet_ntoa", socket_inet_ntoa,
4211 METH_VARARGS, inet_ntoa_doc},
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00004212#ifdef HAVE_INET_PTON
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004213 {"inet_pton", socket_inet_pton,
4214 METH_VARARGS, inet_pton_doc},
4215 {"inet_ntop", socket_inet_ntop,
4216 METH_VARARGS, inet_ntop_doc},
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00004217#endif
Giampaolo Rodolàccfb91c2010-08-17 15:30:23 +00004218 {"getaddrinfo", (PyCFunction)socket_getaddrinfo,
4219 METH_VARARGS | METH_KEYWORDS, getaddrinfo_doc},
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004220 {"getnameinfo", socket_getnameinfo,
4221 METH_VARARGS, getnameinfo_doc},
4222 {"getdefaulttimeout", (PyCFunction)socket_getdefaulttimeout,
4223 METH_NOARGS, getdefaulttimeout_doc},
4224 {"setdefaulttimeout", socket_setdefaulttimeout,
4225 METH_O, setdefaulttimeout_doc},
4226 {NULL, NULL} /* Sentinel */
Guido van Rossum6574b3e1991-06-25 21:36:08 +00004227};
4228
Guido van Rossum30a685f1991-06-27 15:51:29 +00004229
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004230#ifdef MS_WINDOWS
4231#define OS_INIT_DEFINED
4232
4233/* Additional initialization and cleanup for Windows */
Guido van Rossumbe32c891996-06-20 16:25:29 +00004234
4235static void
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004236os_cleanup(void)
Guido van Rossumbe32c891996-06-20 16:25:29 +00004237{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004238 WSACleanup();
Guido van Rossumbe32c891996-06-20 16:25:29 +00004239}
4240
4241static int
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004242os_init(void)
Guido van Rossumbe32c891996-06-20 16:25:29 +00004243{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004244 WSADATA WSAData;
4245 int ret;
4246 ret = WSAStartup(0x0101, &WSAData);
4247 switch (ret) {
4248 case 0: /* No error */
4249 Py_AtExit(os_cleanup);
4250 return 1; /* Success */
4251 case WSASYSNOTREADY:
4252 PyErr_SetString(PyExc_ImportError,
4253 "WSAStartup failed: network not ready");
4254 break;
4255 case WSAVERNOTSUPPORTED:
4256 case WSAEINVAL:
4257 PyErr_SetString(
4258 PyExc_ImportError,
4259 "WSAStartup failed: requested version not supported");
4260 break;
4261 default:
4262 PyErr_Format(PyExc_ImportError, "WSAStartup failed: error code %d", ret);
4263 break;
4264 }
4265 return 0; /* Failure */
Guido van Rossumbe32c891996-06-20 16:25:29 +00004266}
4267
Guido van Rossum8d665e61996-06-26 18:22:49 +00004268#endif /* MS_WINDOWS */
Guido van Rossumbe32c891996-06-20 16:25:29 +00004269
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004270
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004271#ifdef PYOS_OS2
4272#define OS_INIT_DEFINED
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004273
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004274/* Additional initialization for OS/2 */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004275
4276static int
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004277os_init(void)
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004278{
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004279#ifndef PYCC_GCC
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004280 int rc = sock_init();
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004281
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004282 if (rc == 0) {
4283 return 1; /* Success */
4284 }
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004285
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004286 PyErr_Format(PyExc_ImportError, "OS/2 TCP/IP Error# %d", sock_errno());
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004287
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004288 return 0; /* Failure */
Andrew MacIntyreba43e872002-03-03 03:03:52 +00004289#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004290 /* No need to initialise sockets with GCC/EMX */
4291 return 1; /* Success */
Andrew MacIntyreba43e872002-03-03 03:03:52 +00004292#endif
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004293}
4294
4295#endif /* PYOS_OS2 */
4296
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004297
4298#ifndef OS_INIT_DEFINED
4299static int
4300os_init(void)
4301{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004302 return 1; /* Success */
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004303}
4304#endif
4305
4306
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00004307/* C API table - always add new things to the end for binary
4308 compatibility. */
4309static
4310PySocketModule_APIObject PySocketModuleAPI =
4311{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004312 &sock_type,
4313 NULL
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00004314};
4315
Guido van Rossum3eede5a2002-06-07 02:08:35 +00004316
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004317/* Initialize the _socket module.
Guido van Rossum3eede5a2002-06-07 02:08:35 +00004318
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004319 This module is actually called "_socket", and there's a wrapper
Guido van Rossum7d0a8262007-05-21 23:13:11 +00004320 "socket.py" which implements some additional functionality.
4321 The import of "_socket" may fail with an ImportError exception if
4322 os-specific initialization fails. On Windows, this does WINSOCK
4323 initialization. When WINSOCK is initialized succesfully, a call to
4324 WSACleanup() is scheduled to be made at exit time.
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004325*/
Guido van Rossum3eede5a2002-06-07 02:08:35 +00004326
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004327PyDoc_STRVAR(socket_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00004328"Implementation module for socket operations.\n\
4329\n\
4330See the socket module for documentation.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00004331
Martin v. Löwis1a214512008-06-11 05:26:20 +00004332static struct PyModuleDef socketmodule = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004333 PyModuleDef_HEAD_INIT,
4334 PySocket_MODULE_NAME,
4335 socket_doc,
4336 -1,
4337 socket_methods,
4338 NULL,
4339 NULL,
4340 NULL,
4341 NULL
Martin v. Löwis1a214512008-06-11 05:26:20 +00004342};
4343
Mark Hammondfe51c6d2002-08-02 02:27:13 +00004344PyMODINIT_FUNC
Martin v. Löwis1a214512008-06-11 05:26:20 +00004345PyInit__socket(void)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00004346{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004347 PyObject *m, *has_ipv6;
Fred Drake4baedc12002-04-01 14:53:37 +00004348
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004349 if (!os_init())
4350 return NULL;
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004351
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004352 Py_TYPE(&sock_type) = &PyType_Type;
4353 m = PyModule_Create(&socketmodule);
4354 if (m == NULL)
4355 return NULL;
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004356
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004357 socket_error = PyErr_NewException("socket.error",
4358 PyExc_IOError, NULL);
4359 if (socket_error == NULL)
4360 return NULL;
4361 PySocketModuleAPI.error = socket_error;
4362 Py_INCREF(socket_error);
4363 PyModule_AddObject(m, "error", socket_error);
4364 socket_herror = PyErr_NewException("socket.herror",
4365 socket_error, NULL);
4366 if (socket_herror == NULL)
4367 return NULL;
4368 Py_INCREF(socket_herror);
4369 PyModule_AddObject(m, "herror", socket_herror);
4370 socket_gaierror = PyErr_NewException("socket.gaierror", socket_error,
4371 NULL);
4372 if (socket_gaierror == NULL)
4373 return NULL;
4374 Py_INCREF(socket_gaierror);
4375 PyModule_AddObject(m, "gaierror", socket_gaierror);
4376 socket_timeout = PyErr_NewException("socket.timeout",
4377 socket_error, NULL);
4378 if (socket_timeout == NULL)
4379 return NULL;
4380 Py_INCREF(socket_timeout);
4381 PyModule_AddObject(m, "timeout", socket_timeout);
4382 Py_INCREF((PyObject *)&sock_type);
4383 if (PyModule_AddObject(m, "SocketType",
4384 (PyObject *)&sock_type) != 0)
4385 return NULL;
4386 Py_INCREF((PyObject *)&sock_type);
4387 if (PyModule_AddObject(m, "socket",
4388 (PyObject *)&sock_type) != 0)
4389 return NULL;
Guido van Rossum09be4091999-08-09 14:40:40 +00004390
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00004391#ifdef ENABLE_IPV6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004392 has_ipv6 = Py_True;
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00004393#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004394 has_ipv6 = Py_False;
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00004395#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004396 Py_INCREF(has_ipv6);
4397 PyModule_AddObject(m, "has_ipv6", has_ipv6);
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00004398
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004399 /* Export C API */
4400 if (PyModule_AddObject(m, PySocket_CAPI_NAME,
4401 PyCapsule_New(&PySocketModuleAPI, PySocket_CAPSULE_NAME, NULL)
4402 ) != 0)
4403 return NULL;
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00004404
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004405 /* Address families (we only support AF_INET and AF_UNIX) */
Guido van Rossum09be4091999-08-09 14:40:40 +00004406#ifdef AF_UNSPEC
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004407 PyModule_AddIntConstant(m, "AF_UNSPEC", AF_UNSPEC);
Guido van Rossum09be4091999-08-09 14:40:40 +00004408#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004409 PyModule_AddIntConstant(m, "AF_INET", AF_INET);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004410#ifdef AF_INET6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004411 PyModule_AddIntConstant(m, "AF_INET6", AF_INET6);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004412#endif /* AF_INET6 */
Andrew MacIntyred12dfbb2004-04-04 07:13:49 +00004413#if defined(AF_UNIX)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004414 PyModule_AddIntConstant(m, "AF_UNIX", AF_UNIX);
Guido van Rossumb6775db1994-08-01 11:34:53 +00004415#endif /* AF_UNIX */
Guido van Rossum09be4091999-08-09 14:40:40 +00004416#ifdef AF_AX25
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004417 /* Amateur Radio AX.25 */
4418 PyModule_AddIntConstant(m, "AF_AX25", AF_AX25);
Guido van Rossum09be4091999-08-09 14:40:40 +00004419#endif
4420#ifdef AF_IPX
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004421 PyModule_AddIntConstant(m, "AF_IPX", AF_IPX); /* Novell IPX */
Guido van Rossum09be4091999-08-09 14:40:40 +00004422#endif
4423#ifdef AF_APPLETALK
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004424 /* Appletalk DDP */
4425 PyModule_AddIntConstant(m, "AF_APPLETALK", AF_APPLETALK);
Guido van Rossum09be4091999-08-09 14:40:40 +00004426#endif
4427#ifdef AF_NETROM
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004428 /* Amateur radio NetROM */
4429 PyModule_AddIntConstant(m, "AF_NETROM", AF_NETROM);
Guido van Rossum09be4091999-08-09 14:40:40 +00004430#endif
4431#ifdef AF_BRIDGE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004432 /* Multiprotocol bridge */
4433 PyModule_AddIntConstant(m, "AF_BRIDGE", AF_BRIDGE);
Guido van Rossum09be4091999-08-09 14:40:40 +00004434#endif
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004435#ifdef AF_ATMPVC
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004436 /* ATM PVCs */
4437 PyModule_AddIntConstant(m, "AF_ATMPVC", AF_ATMPVC);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004438#endif
Guido van Rossum09be4091999-08-09 14:40:40 +00004439#ifdef AF_AAL5
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004440 /* Reserved for Werner's ATM */
4441 PyModule_AddIntConstant(m, "AF_AAL5", AF_AAL5);
Guido van Rossum09be4091999-08-09 14:40:40 +00004442#endif
4443#ifdef AF_X25
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004444 /* Reserved for X.25 project */
4445 PyModule_AddIntConstant(m, "AF_X25", AF_X25);
Guido van Rossum09be4091999-08-09 14:40:40 +00004446#endif
4447#ifdef AF_INET6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004448 PyModule_AddIntConstant(m, "AF_INET6", AF_INET6); /* IP version 6 */
Guido van Rossum09be4091999-08-09 14:40:40 +00004449#endif
4450#ifdef AF_ROSE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004451 /* Amateur Radio X.25 PLP */
4452 PyModule_AddIntConstant(m, "AF_ROSE", AF_ROSE);
Guido van Rossum09be4091999-08-09 14:40:40 +00004453#endif
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004454#ifdef AF_DECnet
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004455 /* Reserved for DECnet project */
4456 PyModule_AddIntConstant(m, "AF_DECnet", AF_DECnet);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004457#endif
4458#ifdef AF_NETBEUI
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004459 /* Reserved for 802.2LLC project */
4460 PyModule_AddIntConstant(m, "AF_NETBEUI", AF_NETBEUI);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004461#endif
4462#ifdef AF_SECURITY
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004463 /* Security callback pseudo AF */
4464 PyModule_AddIntConstant(m, "AF_SECURITY", AF_SECURITY);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004465#endif
4466#ifdef AF_KEY
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004467 /* PF_KEY key management API */
4468 PyModule_AddIntConstant(m, "AF_KEY", AF_KEY);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004469#endif
4470#ifdef AF_NETLINK
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004471 /* */
4472 PyModule_AddIntConstant(m, "AF_NETLINK", AF_NETLINK);
4473 PyModule_AddIntConstant(m, "NETLINK_ROUTE", NETLINK_ROUTE);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004474#ifdef NETLINK_SKIP
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004475 PyModule_AddIntConstant(m, "NETLINK_SKIP", NETLINK_SKIP);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004476#endif
4477#ifdef NETLINK_W1
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004478 PyModule_AddIntConstant(m, "NETLINK_W1", NETLINK_W1);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004479#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004480 PyModule_AddIntConstant(m, "NETLINK_USERSOCK", NETLINK_USERSOCK);
4481 PyModule_AddIntConstant(m, "NETLINK_FIREWALL", NETLINK_FIREWALL);
Guido van Rossum668a94a2006-02-21 01:07:27 +00004482#ifdef NETLINK_TCPDIAG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004483 PyModule_AddIntConstant(m, "NETLINK_TCPDIAG", NETLINK_TCPDIAG);
Guido van Rossum668a94a2006-02-21 01:07:27 +00004484#endif
4485#ifdef NETLINK_NFLOG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004486 PyModule_AddIntConstant(m, "NETLINK_NFLOG", NETLINK_NFLOG);
Guido van Rossum668a94a2006-02-21 01:07:27 +00004487#endif
Neal Norwitz65851662006-01-16 04:31:40 +00004488#ifdef NETLINK_XFRM
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004489 PyModule_AddIntConstant(m, "NETLINK_XFRM", NETLINK_XFRM);
Neal Norwitz65851662006-01-16 04:31:40 +00004490#endif
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004491#ifdef NETLINK_ARPD
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004492 PyModule_AddIntConstant(m, "NETLINK_ARPD", NETLINK_ARPD);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004493#endif
4494#ifdef NETLINK_ROUTE6
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004495 PyModule_AddIntConstant(m, "NETLINK_ROUTE6", NETLINK_ROUTE6);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004496#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004497 PyModule_AddIntConstant(m, "NETLINK_IP6_FW", NETLINK_IP6_FW);
Thomas Wouterscf297e42007-02-23 15:07:44 +00004498#ifdef NETLINK_DNRTMSG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004499 PyModule_AddIntConstant(m, "NETLINK_DNRTMSG", NETLINK_DNRTMSG);
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00004500#endif
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004501#ifdef NETLINK_TAPBASE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004502 PyModule_AddIntConstant(m, "NETLINK_TAPBASE", NETLINK_TAPBASE);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004503#endif
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004504#endif /* AF_NETLINK */
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004505#ifdef AF_ROUTE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004506 /* Alias to emulate 4.4BSD */
4507 PyModule_AddIntConstant(m, "AF_ROUTE", AF_ROUTE);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004508#endif
4509#ifdef AF_ASH
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004510 /* Ash */
4511 PyModule_AddIntConstant(m, "AF_ASH", AF_ASH);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004512#endif
4513#ifdef AF_ECONET
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004514 /* Acorn Econet */
4515 PyModule_AddIntConstant(m, "AF_ECONET", AF_ECONET);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004516#endif
4517#ifdef AF_ATMSVC
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004518 /* ATM SVCs */
4519 PyModule_AddIntConstant(m, "AF_ATMSVC", AF_ATMSVC);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004520#endif
4521#ifdef AF_SNA
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004522 /* Linux SNA Project (nutters!) */
4523 PyModule_AddIntConstant(m, "AF_SNA", AF_SNA);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004524#endif
4525#ifdef AF_IRDA
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004526 /* IRDA sockets */
4527 PyModule_AddIntConstant(m, "AF_IRDA", AF_IRDA);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004528#endif
4529#ifdef AF_PPPOX
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004530 /* PPPoX sockets */
4531 PyModule_AddIntConstant(m, "AF_PPPOX", AF_PPPOX);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004532#endif
4533#ifdef AF_WANPIPE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004534 /* Wanpipe API Sockets */
4535 PyModule_AddIntConstant(m, "AF_WANPIPE", AF_WANPIPE);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004536#endif
4537#ifdef AF_LLC
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004538 /* Linux LLC */
4539 PyModule_AddIntConstant(m, "AF_LLC", AF_LLC);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004540#endif
Martin v. Löwis12af0482004-01-31 12:34:17 +00004541
Hye-Shik Chang81268602004-02-02 06:05:24 +00004542#ifdef USE_BLUETOOTH
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004543 PyModule_AddIntConstant(m, "AF_BLUETOOTH", AF_BLUETOOTH);
4544 PyModule_AddIntConstant(m, "BTPROTO_L2CAP", BTPROTO_L2CAP);
4545 PyModule_AddIntConstant(m, "BTPROTO_HCI", BTPROTO_HCI);
4546 PyModule_AddIntConstant(m, "SOL_HCI", SOL_HCI);
4547 PyModule_AddIntConstant(m, "HCI_FILTER", HCI_FILTER);
Hye-Shik Chang81268602004-02-02 06:05:24 +00004548#if !defined(__FreeBSD__)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004549 PyModule_AddIntConstant(m, "HCI_TIME_STAMP", HCI_TIME_STAMP);
4550 PyModule_AddIntConstant(m, "HCI_DATA_DIR", HCI_DATA_DIR);
4551 PyModule_AddIntConstant(m, "BTPROTO_SCO", BTPROTO_SCO);
Hye-Shik Chang81268602004-02-02 06:05:24 +00004552#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004553 PyModule_AddIntConstant(m, "BTPROTO_RFCOMM", BTPROTO_RFCOMM);
4554 PyModule_AddStringConstant(m, "BDADDR_ANY", "00:00:00:00:00:00");
4555 PyModule_AddStringConstant(m, "BDADDR_LOCAL", "00:00:00:FF:FF:FF");
Martin v. Löwis12af0482004-01-31 12:34:17 +00004556#endif
4557
Martin v. Löwis1ba3fd52001-08-10 20:29:40 +00004558#ifdef HAVE_NETPACKET_PACKET_H
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004559 PyModule_AddIntConstant(m, "AF_PACKET", AF_PACKET);
4560 PyModule_AddIntConstant(m, "PF_PACKET", PF_PACKET);
4561 PyModule_AddIntConstant(m, "PACKET_HOST", PACKET_HOST);
4562 PyModule_AddIntConstant(m, "PACKET_BROADCAST", PACKET_BROADCAST);
4563 PyModule_AddIntConstant(m, "PACKET_MULTICAST", PACKET_MULTICAST);
4564 PyModule_AddIntConstant(m, "PACKET_OTHERHOST", PACKET_OTHERHOST);
4565 PyModule_AddIntConstant(m, "PACKET_OUTGOING", PACKET_OUTGOING);
4566 PyModule_AddIntConstant(m, "PACKET_LOOPBACK", PACKET_LOOPBACK);
4567 PyModule_AddIntConstant(m, "PACKET_FASTROUTE", PACKET_FASTROUTE);
Guido van Rossum48a680c2001-03-02 06:34:14 +00004568#endif
Guido van Rossum09be4091999-08-09 14:40:40 +00004569
Christian Heimes043d6f62008-01-07 17:19:16 +00004570#ifdef HAVE_LINUX_TIPC_H
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004571 PyModule_AddIntConstant(m, "AF_TIPC", AF_TIPC);
Christian Heimes043d6f62008-01-07 17:19:16 +00004572
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004573 /* for addresses */
4574 PyModule_AddIntConstant(m, "TIPC_ADDR_NAMESEQ", TIPC_ADDR_NAMESEQ);
4575 PyModule_AddIntConstant(m, "TIPC_ADDR_NAME", TIPC_ADDR_NAME);
4576 PyModule_AddIntConstant(m, "TIPC_ADDR_ID", TIPC_ADDR_ID);
Christian Heimes043d6f62008-01-07 17:19:16 +00004577
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004578 PyModule_AddIntConstant(m, "TIPC_ZONE_SCOPE", TIPC_ZONE_SCOPE);
4579 PyModule_AddIntConstant(m, "TIPC_CLUSTER_SCOPE", TIPC_CLUSTER_SCOPE);
4580 PyModule_AddIntConstant(m, "TIPC_NODE_SCOPE", TIPC_NODE_SCOPE);
Christian Heimes043d6f62008-01-07 17:19:16 +00004581
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004582 /* for setsockopt() */
4583 PyModule_AddIntConstant(m, "SOL_TIPC", SOL_TIPC);
4584 PyModule_AddIntConstant(m, "TIPC_IMPORTANCE", TIPC_IMPORTANCE);
4585 PyModule_AddIntConstant(m, "TIPC_SRC_DROPPABLE", TIPC_SRC_DROPPABLE);
4586 PyModule_AddIntConstant(m, "TIPC_DEST_DROPPABLE",
4587 TIPC_DEST_DROPPABLE);
4588 PyModule_AddIntConstant(m, "TIPC_CONN_TIMEOUT", TIPC_CONN_TIMEOUT);
Christian Heimes043d6f62008-01-07 17:19:16 +00004589
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004590 PyModule_AddIntConstant(m, "TIPC_LOW_IMPORTANCE",
4591 TIPC_LOW_IMPORTANCE);
4592 PyModule_AddIntConstant(m, "TIPC_MEDIUM_IMPORTANCE",
4593 TIPC_MEDIUM_IMPORTANCE);
4594 PyModule_AddIntConstant(m, "TIPC_HIGH_IMPORTANCE",
4595 TIPC_HIGH_IMPORTANCE);
4596 PyModule_AddIntConstant(m, "TIPC_CRITICAL_IMPORTANCE",
4597 TIPC_CRITICAL_IMPORTANCE);
Christian Heimes043d6f62008-01-07 17:19:16 +00004598
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004599 /* for subscriptions */
4600 PyModule_AddIntConstant(m, "TIPC_SUB_PORTS", TIPC_SUB_PORTS);
4601 PyModule_AddIntConstant(m, "TIPC_SUB_SERVICE", TIPC_SUB_SERVICE);
Christian Heimes25bb7832008-01-11 16:17:00 +00004602#ifdef TIPC_SUB_CANCEL
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004603 /* doesn't seem to be available everywhere */
4604 PyModule_AddIntConstant(m, "TIPC_SUB_CANCEL", TIPC_SUB_CANCEL);
Christian Heimes25bb7832008-01-11 16:17:00 +00004605#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004606 PyModule_AddIntConstant(m, "TIPC_WAIT_FOREVER", TIPC_WAIT_FOREVER);
4607 PyModule_AddIntConstant(m, "TIPC_PUBLISHED", TIPC_PUBLISHED);
4608 PyModule_AddIntConstant(m, "TIPC_WITHDRAWN", TIPC_WITHDRAWN);
4609 PyModule_AddIntConstant(m, "TIPC_SUBSCR_TIMEOUT", TIPC_SUBSCR_TIMEOUT);
4610 PyModule_AddIntConstant(m, "TIPC_CFG_SRV", TIPC_CFG_SRV);
4611 PyModule_AddIntConstant(m, "TIPC_TOP_SRV", TIPC_TOP_SRV);
Christian Heimes043d6f62008-01-07 17:19:16 +00004612#endif
4613
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004614 /* Socket types */
4615 PyModule_AddIntConstant(m, "SOCK_STREAM", SOCK_STREAM);
4616 PyModule_AddIntConstant(m, "SOCK_DGRAM", SOCK_DGRAM);
Guido van Rossumbcc20741998-08-04 22:53:56 +00004617/* We have incomplete socket support. */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004618 PyModule_AddIntConstant(m, "SOCK_RAW", SOCK_RAW);
4619 PyModule_AddIntConstant(m, "SOCK_SEQPACKET", SOCK_SEQPACKET);
Martin v. Löwiscf8f47e2002-12-11 13:10:57 +00004620#if defined(SOCK_RDM)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004621 PyModule_AddIntConstant(m, "SOCK_RDM", SOCK_RDM);
Guido van Rossumbcc20741998-08-04 22:53:56 +00004622#endif
Antoine Pitroub1c54962010-10-14 15:05:38 +00004623#ifdef SOCK_CLOEXEC
4624 PyModule_AddIntConstant(m, "SOCK_CLOEXEC", SOCK_CLOEXEC);
4625#endif
4626#ifdef SOCK_NONBLOCK
4627 PyModule_AddIntConstant(m, "SOCK_NONBLOCK", SOCK_NONBLOCK);
4628#endif
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004629
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004630#ifdef SO_DEBUG
4631 PyModule_AddIntConstant(m, "SO_DEBUG", SO_DEBUG);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004632#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004633#ifdef SO_ACCEPTCONN
4634 PyModule_AddIntConstant(m, "SO_ACCEPTCONN", SO_ACCEPTCONN);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004635#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004636#ifdef SO_REUSEADDR
4637 PyModule_AddIntConstant(m, "SO_REUSEADDR", SO_REUSEADDR);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004638#endif
Andrew M. Kuchling42851ab2004-07-10 14:19:21 +00004639#ifdef SO_EXCLUSIVEADDRUSE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004640 PyModule_AddIntConstant(m, "SO_EXCLUSIVEADDRUSE", SO_EXCLUSIVEADDRUSE);
Andrew M. Kuchling42851ab2004-07-10 14:19:21 +00004641#endif
4642
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004643#ifdef SO_KEEPALIVE
4644 PyModule_AddIntConstant(m, "SO_KEEPALIVE", SO_KEEPALIVE);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004645#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004646#ifdef SO_DONTROUTE
4647 PyModule_AddIntConstant(m, "SO_DONTROUTE", SO_DONTROUTE);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004648#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004649#ifdef SO_BROADCAST
4650 PyModule_AddIntConstant(m, "SO_BROADCAST", SO_BROADCAST);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004651#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004652#ifdef SO_USELOOPBACK
4653 PyModule_AddIntConstant(m, "SO_USELOOPBACK", SO_USELOOPBACK);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004654#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004655#ifdef SO_LINGER
4656 PyModule_AddIntConstant(m, "SO_LINGER", SO_LINGER);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004657#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004658#ifdef SO_OOBINLINE
4659 PyModule_AddIntConstant(m, "SO_OOBINLINE", SO_OOBINLINE);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004660#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004661#ifdef SO_REUSEPORT
4662 PyModule_AddIntConstant(m, "SO_REUSEPORT", SO_REUSEPORT);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004663#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004664#ifdef SO_SNDBUF
4665 PyModule_AddIntConstant(m, "SO_SNDBUF", SO_SNDBUF);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004666#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004667#ifdef SO_RCVBUF
4668 PyModule_AddIntConstant(m, "SO_RCVBUF", SO_RCVBUF);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004669#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004670#ifdef SO_SNDLOWAT
4671 PyModule_AddIntConstant(m, "SO_SNDLOWAT", SO_SNDLOWAT);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004672#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004673#ifdef SO_RCVLOWAT
4674 PyModule_AddIntConstant(m, "SO_RCVLOWAT", SO_RCVLOWAT);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004675#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004676#ifdef SO_SNDTIMEO
4677 PyModule_AddIntConstant(m, "SO_SNDTIMEO", SO_SNDTIMEO);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004678#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004679#ifdef SO_RCVTIMEO
4680 PyModule_AddIntConstant(m, "SO_RCVTIMEO", SO_RCVTIMEO);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004681#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004682#ifdef SO_ERROR
4683 PyModule_AddIntConstant(m, "SO_ERROR", SO_ERROR);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004684#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004685#ifdef SO_TYPE
4686 PyModule_AddIntConstant(m, "SO_TYPE", SO_TYPE);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004687#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004688#ifdef SO_SETFIB
4689 PyModule_AddIntConstant(m, "SO_SETFIB", SO_SETFIB);
Larry Hastingsf0f37952010-04-02 11:47:10 +00004690#endif
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004691
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004692 /* Maximum number of connections for "listen" */
4693#ifdef SOMAXCONN
4694 PyModule_AddIntConstant(m, "SOMAXCONN", SOMAXCONN);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004695#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004696 PyModule_AddIntConstant(m, "SOMAXCONN", 5); /* Common value */
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004697#endif
4698
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004699 /* Flags for send, recv */
4700#ifdef MSG_OOB
4701 PyModule_AddIntConstant(m, "MSG_OOB", MSG_OOB);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004702#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004703#ifdef MSG_PEEK
4704 PyModule_AddIntConstant(m, "MSG_PEEK", MSG_PEEK);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004705#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004706#ifdef MSG_DONTROUTE
4707 PyModule_AddIntConstant(m, "MSG_DONTROUTE", MSG_DONTROUTE);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004708#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004709#ifdef MSG_DONTWAIT
4710 PyModule_AddIntConstant(m, "MSG_DONTWAIT", MSG_DONTWAIT);
Guido van Rossum2c8bcb82000-04-25 21:34:53 +00004711#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004712#ifdef MSG_EOR
4713 PyModule_AddIntConstant(m, "MSG_EOR", MSG_EOR);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004714#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004715#ifdef MSG_TRUNC
4716 PyModule_AddIntConstant(m, "MSG_TRUNC", MSG_TRUNC);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004717#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004718#ifdef MSG_CTRUNC
4719 PyModule_AddIntConstant(m, "MSG_CTRUNC", MSG_CTRUNC);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004720#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004721#ifdef MSG_WAITALL
4722 PyModule_AddIntConstant(m, "MSG_WAITALL", MSG_WAITALL);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004723#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004724#ifdef MSG_BTAG
4725 PyModule_AddIntConstant(m, "MSG_BTAG", MSG_BTAG);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004726#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004727#ifdef MSG_ETAG
4728 PyModule_AddIntConstant(m, "MSG_ETAG", MSG_ETAG);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004729#endif
4730
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004731 /* Protocol level and numbers, usable for [gs]etsockopt */
4732#ifdef SOL_SOCKET
4733 PyModule_AddIntConstant(m, "SOL_SOCKET", SOL_SOCKET);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004734#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004735#ifdef SOL_IP
4736 PyModule_AddIntConstant(m, "SOL_IP", SOL_IP);
Guido van Rossum09be4091999-08-09 14:40:40 +00004737#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004738 PyModule_AddIntConstant(m, "SOL_IP", 0);
Guido van Rossum09be4091999-08-09 14:40:40 +00004739#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004740#ifdef SOL_IPX
4741 PyModule_AddIntConstant(m, "SOL_IPX", SOL_IPX);
Guido van Rossum09be4091999-08-09 14:40:40 +00004742#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004743#ifdef SOL_AX25
4744 PyModule_AddIntConstant(m, "SOL_AX25", SOL_AX25);
Guido van Rossum09be4091999-08-09 14:40:40 +00004745#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004746#ifdef SOL_ATALK
4747 PyModule_AddIntConstant(m, "SOL_ATALK", SOL_ATALK);
Guido van Rossum09be4091999-08-09 14:40:40 +00004748#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004749#ifdef SOL_NETROM
4750 PyModule_AddIntConstant(m, "SOL_NETROM", SOL_NETROM);
Guido van Rossum09be4091999-08-09 14:40:40 +00004751#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004752#ifdef SOL_ROSE
4753 PyModule_AddIntConstant(m, "SOL_ROSE", SOL_ROSE);
Guido van Rossum09be4091999-08-09 14:40:40 +00004754#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004755#ifdef SOL_TCP
4756 PyModule_AddIntConstant(m, "SOL_TCP", SOL_TCP);
Guido van Rossum09be4091999-08-09 14:40:40 +00004757#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004758 PyModule_AddIntConstant(m, "SOL_TCP", 6);
Guido van Rossum09be4091999-08-09 14:40:40 +00004759#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004760#ifdef SOL_UDP
4761 PyModule_AddIntConstant(m, "SOL_UDP", SOL_UDP);
Guido van Rossum09be4091999-08-09 14:40:40 +00004762#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004763 PyModule_AddIntConstant(m, "SOL_UDP", 17);
Guido van Rossum09be4091999-08-09 14:40:40 +00004764#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004765#ifdef IPPROTO_IP
4766 PyModule_AddIntConstant(m, "IPPROTO_IP", IPPROTO_IP);
Guido van Rossum578de301998-05-28 20:18:18 +00004767#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004768 PyModule_AddIntConstant(m, "IPPROTO_IP", 0);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004769#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004770#ifdef IPPROTO_HOPOPTS
4771 PyModule_AddIntConstant(m, "IPPROTO_HOPOPTS", IPPROTO_HOPOPTS);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004772#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004773#ifdef IPPROTO_ICMP
4774 PyModule_AddIntConstant(m, "IPPROTO_ICMP", IPPROTO_ICMP);
Guido van Rossum578de301998-05-28 20:18:18 +00004775#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004776 PyModule_AddIntConstant(m, "IPPROTO_ICMP", 1);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004777#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004778#ifdef IPPROTO_IGMP
4779 PyModule_AddIntConstant(m, "IPPROTO_IGMP", IPPROTO_IGMP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004780#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004781#ifdef IPPROTO_GGP
4782 PyModule_AddIntConstant(m, "IPPROTO_GGP", IPPROTO_GGP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004783#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004784#ifdef IPPROTO_IPV4
4785 PyModule_AddIntConstant(m, "IPPROTO_IPV4", IPPROTO_IPV4);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004786#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004787#ifdef IPPROTO_IPV6
4788 PyModule_AddIntConstant(m, "IPPROTO_IPV6", IPPROTO_IPV6);
Martin v. Löwisa0f17342003-10-03 13:56:20 +00004789#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004790#ifdef IPPROTO_IPIP
4791 PyModule_AddIntConstant(m, "IPPROTO_IPIP", IPPROTO_IPIP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004792#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004793#ifdef IPPROTO_TCP
4794 PyModule_AddIntConstant(m, "IPPROTO_TCP", IPPROTO_TCP);
Guido van Rossum578de301998-05-28 20:18:18 +00004795#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004796 PyModule_AddIntConstant(m, "IPPROTO_TCP", 6);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004797#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004798#ifdef IPPROTO_EGP
4799 PyModule_AddIntConstant(m, "IPPROTO_EGP", IPPROTO_EGP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004800#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004801#ifdef IPPROTO_PUP
4802 PyModule_AddIntConstant(m, "IPPROTO_PUP", IPPROTO_PUP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004803#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004804#ifdef IPPROTO_UDP
4805 PyModule_AddIntConstant(m, "IPPROTO_UDP", IPPROTO_UDP);
Guido van Rossum578de301998-05-28 20:18:18 +00004806#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004807 PyModule_AddIntConstant(m, "IPPROTO_UDP", 17);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004808#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004809#ifdef IPPROTO_IDP
4810 PyModule_AddIntConstant(m, "IPPROTO_IDP", IPPROTO_IDP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004811#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004812#ifdef IPPROTO_HELLO
4813 PyModule_AddIntConstant(m, "IPPROTO_HELLO", IPPROTO_HELLO);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004814#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004815#ifdef IPPROTO_ND
4816 PyModule_AddIntConstant(m, "IPPROTO_ND", IPPROTO_ND);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004817#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004818#ifdef IPPROTO_TP
4819 PyModule_AddIntConstant(m, "IPPROTO_TP", IPPROTO_TP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004820#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004821#ifdef IPPROTO_IPV6
4822 PyModule_AddIntConstant(m, "IPPROTO_IPV6", IPPROTO_IPV6);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004823#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004824#ifdef IPPROTO_ROUTING
4825 PyModule_AddIntConstant(m, "IPPROTO_ROUTING", IPPROTO_ROUTING);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004826#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004827#ifdef IPPROTO_FRAGMENT
4828 PyModule_AddIntConstant(m, "IPPROTO_FRAGMENT", IPPROTO_FRAGMENT);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004829#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004830#ifdef IPPROTO_RSVP
4831 PyModule_AddIntConstant(m, "IPPROTO_RSVP", IPPROTO_RSVP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004832#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004833#ifdef IPPROTO_GRE
4834 PyModule_AddIntConstant(m, "IPPROTO_GRE", IPPROTO_GRE);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004835#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004836#ifdef IPPROTO_ESP
4837 PyModule_AddIntConstant(m, "IPPROTO_ESP", IPPROTO_ESP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004838#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004839#ifdef IPPROTO_AH
4840 PyModule_AddIntConstant(m, "IPPROTO_AH", IPPROTO_AH);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004841#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004842#ifdef IPPROTO_MOBILE
4843 PyModule_AddIntConstant(m, "IPPROTO_MOBILE", IPPROTO_MOBILE);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004844#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004845#ifdef IPPROTO_ICMPV6
4846 PyModule_AddIntConstant(m, "IPPROTO_ICMPV6", IPPROTO_ICMPV6);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004847#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004848#ifdef IPPROTO_NONE
4849 PyModule_AddIntConstant(m, "IPPROTO_NONE", IPPROTO_NONE);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004850#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004851#ifdef IPPROTO_DSTOPTS
4852 PyModule_AddIntConstant(m, "IPPROTO_DSTOPTS", IPPROTO_DSTOPTS);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004853#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004854#ifdef IPPROTO_XTP
4855 PyModule_AddIntConstant(m, "IPPROTO_XTP", IPPROTO_XTP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004856#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004857#ifdef IPPROTO_EON
4858 PyModule_AddIntConstant(m, "IPPROTO_EON", IPPROTO_EON);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004859#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004860#ifdef IPPROTO_PIM
4861 PyModule_AddIntConstant(m, "IPPROTO_PIM", IPPROTO_PIM);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004862#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004863#ifdef IPPROTO_IPCOMP
4864 PyModule_AddIntConstant(m, "IPPROTO_IPCOMP", IPPROTO_IPCOMP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004865#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004866#ifdef IPPROTO_VRRP
4867 PyModule_AddIntConstant(m, "IPPROTO_VRRP", IPPROTO_VRRP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004868#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004869#ifdef IPPROTO_BIP
4870 PyModule_AddIntConstant(m, "IPPROTO_BIP", IPPROTO_BIP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004871#endif
4872/**/
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004873#ifdef IPPROTO_RAW
4874 PyModule_AddIntConstant(m, "IPPROTO_RAW", IPPROTO_RAW);
Guido van Rossum578de301998-05-28 20:18:18 +00004875#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004876 PyModule_AddIntConstant(m, "IPPROTO_RAW", 255);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004877#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004878#ifdef IPPROTO_MAX
4879 PyModule_AddIntConstant(m, "IPPROTO_MAX", IPPROTO_MAX);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004880#endif
4881
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004882 /* Some port configuration */
4883#ifdef IPPORT_RESERVED
4884 PyModule_AddIntConstant(m, "IPPORT_RESERVED", IPPORT_RESERVED);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004885#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004886 PyModule_AddIntConstant(m, "IPPORT_RESERVED", 1024);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004887#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004888#ifdef IPPORT_USERRESERVED
4889 PyModule_AddIntConstant(m, "IPPORT_USERRESERVED", IPPORT_USERRESERVED);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004890#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004891 PyModule_AddIntConstant(m, "IPPORT_USERRESERVED", 5000);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004892#endif
4893
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004894 /* Some reserved IP v.4 addresses */
4895#ifdef INADDR_ANY
4896 PyModule_AddIntConstant(m, "INADDR_ANY", INADDR_ANY);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004897#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004898 PyModule_AddIntConstant(m, "INADDR_ANY", 0x00000000);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004899#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004900#ifdef INADDR_BROADCAST
4901 PyModule_AddIntConstant(m, "INADDR_BROADCAST", INADDR_BROADCAST);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004902#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004903 PyModule_AddIntConstant(m, "INADDR_BROADCAST", 0xffffffff);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004904#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004905#ifdef INADDR_LOOPBACK
4906 PyModule_AddIntConstant(m, "INADDR_LOOPBACK", INADDR_LOOPBACK);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004907#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004908 PyModule_AddIntConstant(m, "INADDR_LOOPBACK", 0x7F000001);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004909#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004910#ifdef INADDR_UNSPEC_GROUP
4911 PyModule_AddIntConstant(m, "INADDR_UNSPEC_GROUP", INADDR_UNSPEC_GROUP);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004912#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004913 PyModule_AddIntConstant(m, "INADDR_UNSPEC_GROUP", 0xe0000000);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004914#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004915#ifdef INADDR_ALLHOSTS_GROUP
4916 PyModule_AddIntConstant(m, "INADDR_ALLHOSTS_GROUP",
4917 INADDR_ALLHOSTS_GROUP);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004918#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004919 PyModule_AddIntConstant(m, "INADDR_ALLHOSTS_GROUP", 0xe0000001);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004920#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004921#ifdef INADDR_MAX_LOCAL_GROUP
4922 PyModule_AddIntConstant(m, "INADDR_MAX_LOCAL_GROUP",
4923 INADDR_MAX_LOCAL_GROUP);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004924#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004925 PyModule_AddIntConstant(m, "INADDR_MAX_LOCAL_GROUP", 0xe00000ff);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004926#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004927#ifdef INADDR_NONE
4928 PyModule_AddIntConstant(m, "INADDR_NONE", INADDR_NONE);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004929#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004930 PyModule_AddIntConstant(m, "INADDR_NONE", 0xffffffff);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004931#endif
4932
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004933 /* IPv4 [gs]etsockopt options */
4934#ifdef IP_OPTIONS
4935 PyModule_AddIntConstant(m, "IP_OPTIONS", IP_OPTIONS);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004936#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004937#ifdef IP_HDRINCL
4938 PyModule_AddIntConstant(m, "IP_HDRINCL", IP_HDRINCL);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004939#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004940#ifdef IP_TOS
4941 PyModule_AddIntConstant(m, "IP_TOS", IP_TOS);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004942#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004943#ifdef IP_TTL
4944 PyModule_AddIntConstant(m, "IP_TTL", IP_TTL);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004945#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004946#ifdef IP_RECVOPTS
4947 PyModule_AddIntConstant(m, "IP_RECVOPTS", IP_RECVOPTS);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004948#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004949#ifdef IP_RECVRETOPTS
4950 PyModule_AddIntConstant(m, "IP_RECVRETOPTS", IP_RECVRETOPTS);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004951#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004952#ifdef IP_RECVDSTADDR
4953 PyModule_AddIntConstant(m, "IP_RECVDSTADDR", IP_RECVDSTADDR);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004954#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004955#ifdef IP_RETOPTS
4956 PyModule_AddIntConstant(m, "IP_RETOPTS", IP_RETOPTS);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004957#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004958#ifdef IP_MULTICAST_IF
4959 PyModule_AddIntConstant(m, "IP_MULTICAST_IF", IP_MULTICAST_IF);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004960#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004961#ifdef IP_MULTICAST_TTL
4962 PyModule_AddIntConstant(m, "IP_MULTICAST_TTL", IP_MULTICAST_TTL);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004963#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004964#ifdef IP_MULTICAST_LOOP
4965 PyModule_AddIntConstant(m, "IP_MULTICAST_LOOP", IP_MULTICAST_LOOP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004966#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004967#ifdef IP_ADD_MEMBERSHIP
4968 PyModule_AddIntConstant(m, "IP_ADD_MEMBERSHIP", IP_ADD_MEMBERSHIP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004969#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004970#ifdef IP_DROP_MEMBERSHIP
4971 PyModule_AddIntConstant(m, "IP_DROP_MEMBERSHIP", IP_DROP_MEMBERSHIP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004972#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004973#ifdef IP_DEFAULT_MULTICAST_TTL
4974 PyModule_AddIntConstant(m, "IP_DEFAULT_MULTICAST_TTL",
4975 IP_DEFAULT_MULTICAST_TTL);
Guido van Rossum09be4091999-08-09 14:40:40 +00004976#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004977#ifdef IP_DEFAULT_MULTICAST_LOOP
4978 PyModule_AddIntConstant(m, "IP_DEFAULT_MULTICAST_LOOP",
4979 IP_DEFAULT_MULTICAST_LOOP);
Guido van Rossum09be4091999-08-09 14:40:40 +00004980#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004981#ifdef IP_MAX_MEMBERSHIPS
4982 PyModule_AddIntConstant(m, "IP_MAX_MEMBERSHIPS", IP_MAX_MEMBERSHIPS);
Guido van Rossum09be4091999-08-09 14:40:40 +00004983#endif
4984
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004985 /* IPv6 [gs]etsockopt options, defined in RFC2553 */
4986#ifdef IPV6_JOIN_GROUP
4987 PyModule_AddIntConstant(m, "IPV6_JOIN_GROUP", IPV6_JOIN_GROUP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004988#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004989#ifdef IPV6_LEAVE_GROUP
4990 PyModule_AddIntConstant(m, "IPV6_LEAVE_GROUP", IPV6_LEAVE_GROUP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004991#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004992#ifdef IPV6_MULTICAST_HOPS
4993 PyModule_AddIntConstant(m, "IPV6_MULTICAST_HOPS", IPV6_MULTICAST_HOPS);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004994#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004995#ifdef IPV6_MULTICAST_IF
4996 PyModule_AddIntConstant(m, "IPV6_MULTICAST_IF", IPV6_MULTICAST_IF);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004997#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004998#ifdef IPV6_MULTICAST_LOOP
4999 PyModule_AddIntConstant(m, "IPV6_MULTICAST_LOOP", IPV6_MULTICAST_LOOP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00005000#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005001#ifdef IPV6_UNICAST_HOPS
5002 PyModule_AddIntConstant(m, "IPV6_UNICAST_HOPS", IPV6_UNICAST_HOPS);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00005003#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005004 /* Additional IPV6 socket options, defined in RFC 3493 */
Martin v. Löwisda91d022003-12-30 11:14:01 +00005005#ifdef IPV6_V6ONLY
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005006 PyModule_AddIntConstant(m, "IPV6_V6ONLY", IPV6_V6ONLY);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005007#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005008 /* Advanced IPV6 socket options, from RFC 3542 */
Martin v. Löwisda91d022003-12-30 11:14:01 +00005009#ifdef IPV6_CHECKSUM
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005010 PyModule_AddIntConstant(m, "IPV6_CHECKSUM", IPV6_CHECKSUM);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005011#endif
5012#ifdef IPV6_DONTFRAG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005013 PyModule_AddIntConstant(m, "IPV6_DONTFRAG", IPV6_DONTFRAG);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005014#endif
5015#ifdef IPV6_DSTOPTS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005016 PyModule_AddIntConstant(m, "IPV6_DSTOPTS", IPV6_DSTOPTS);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005017#endif
5018#ifdef IPV6_HOPLIMIT
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005019 PyModule_AddIntConstant(m, "IPV6_HOPLIMIT", IPV6_HOPLIMIT);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005020#endif
5021#ifdef IPV6_HOPOPTS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005022 PyModule_AddIntConstant(m, "IPV6_HOPOPTS", IPV6_HOPOPTS);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005023#endif
5024#ifdef IPV6_NEXTHOP
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005025 PyModule_AddIntConstant(m, "IPV6_NEXTHOP", IPV6_NEXTHOP);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005026#endif
5027#ifdef IPV6_PATHMTU
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005028 PyModule_AddIntConstant(m, "IPV6_PATHMTU", IPV6_PATHMTU);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005029#endif
5030#ifdef IPV6_PKTINFO
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005031 PyModule_AddIntConstant(m, "IPV6_PKTINFO", IPV6_PKTINFO);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005032#endif
5033#ifdef IPV6_RECVDSTOPTS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005034 PyModule_AddIntConstant(m, "IPV6_RECVDSTOPTS", IPV6_RECVDSTOPTS);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005035#endif
5036#ifdef IPV6_RECVHOPLIMIT
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005037 PyModule_AddIntConstant(m, "IPV6_RECVHOPLIMIT", IPV6_RECVHOPLIMIT);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005038#endif
5039#ifdef IPV6_RECVHOPOPTS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005040 PyModule_AddIntConstant(m, "IPV6_RECVHOPOPTS", IPV6_RECVHOPOPTS);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005041#endif
5042#ifdef IPV6_RECVPKTINFO
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005043 PyModule_AddIntConstant(m, "IPV6_RECVPKTINFO", IPV6_RECVPKTINFO);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005044#endif
5045#ifdef IPV6_RECVRTHDR
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005046 PyModule_AddIntConstant(m, "IPV6_RECVRTHDR", IPV6_RECVRTHDR);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005047#endif
5048#ifdef IPV6_RECVTCLASS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005049 PyModule_AddIntConstant(m, "IPV6_RECVTCLASS", IPV6_RECVTCLASS);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005050#endif
5051#ifdef IPV6_RTHDR
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005052 PyModule_AddIntConstant(m, "IPV6_RTHDR", IPV6_RTHDR);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005053#endif
5054#ifdef IPV6_RTHDRDSTOPTS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005055 PyModule_AddIntConstant(m, "IPV6_RTHDRDSTOPTS", IPV6_RTHDRDSTOPTS);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005056#endif
5057#ifdef IPV6_RTHDR_TYPE_0
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005058 PyModule_AddIntConstant(m, "IPV6_RTHDR_TYPE_0", IPV6_RTHDR_TYPE_0);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005059#endif
5060#ifdef IPV6_RECVPATHMTU
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005061 PyModule_AddIntConstant(m, "IPV6_RECVPATHMTU", IPV6_RECVPATHMTU);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005062#endif
5063#ifdef IPV6_TCLASS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005064 PyModule_AddIntConstant(m, "IPV6_TCLASS", IPV6_TCLASS);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005065#endif
5066#ifdef IPV6_USE_MIN_MTU
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005067 PyModule_AddIntConstant(m, "IPV6_USE_MIN_MTU", IPV6_USE_MIN_MTU);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005068#endif
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00005069
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005070 /* TCP options */
5071#ifdef TCP_NODELAY
5072 PyModule_AddIntConstant(m, "TCP_NODELAY", TCP_NODELAY);
Guido van Rossum09be4091999-08-09 14:40:40 +00005073#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005074#ifdef TCP_MAXSEG
5075 PyModule_AddIntConstant(m, "TCP_MAXSEG", TCP_MAXSEG);
Guido van Rossum09be4091999-08-09 14:40:40 +00005076#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005077#ifdef TCP_CORK
5078 PyModule_AddIntConstant(m, "TCP_CORK", TCP_CORK);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00005079#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005080#ifdef TCP_KEEPIDLE
5081 PyModule_AddIntConstant(m, "TCP_KEEPIDLE", TCP_KEEPIDLE);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00005082#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005083#ifdef TCP_KEEPINTVL
5084 PyModule_AddIntConstant(m, "TCP_KEEPINTVL", TCP_KEEPINTVL);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00005085#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005086#ifdef TCP_KEEPCNT
5087 PyModule_AddIntConstant(m, "TCP_KEEPCNT", TCP_KEEPCNT);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00005088#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005089#ifdef TCP_SYNCNT
5090 PyModule_AddIntConstant(m, "TCP_SYNCNT", TCP_SYNCNT);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00005091#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005092#ifdef TCP_LINGER2
5093 PyModule_AddIntConstant(m, "TCP_LINGER2", TCP_LINGER2);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00005094#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005095#ifdef TCP_DEFER_ACCEPT
5096 PyModule_AddIntConstant(m, "TCP_DEFER_ACCEPT", TCP_DEFER_ACCEPT);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00005097#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005098#ifdef TCP_WINDOW_CLAMP
5099 PyModule_AddIntConstant(m, "TCP_WINDOW_CLAMP", TCP_WINDOW_CLAMP);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00005100#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005101#ifdef TCP_INFO
5102 PyModule_AddIntConstant(m, "TCP_INFO", TCP_INFO);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00005103#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005104#ifdef TCP_QUICKACK
5105 PyModule_AddIntConstant(m, "TCP_QUICKACK", TCP_QUICKACK);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00005106#endif
5107
Guido van Rossum09be4091999-08-09 14:40:40 +00005108
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005109 /* IPX options */
5110#ifdef IPX_TYPE
5111 PyModule_AddIntConstant(m, "IPX_TYPE", IPX_TYPE);
Guido van Rossum09be4091999-08-09 14:40:40 +00005112#endif
Guido van Rossum4f199ea1998-04-09 20:56:35 +00005113
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005114 /* get{addr,name}info parameters */
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005115#ifdef EAI_ADDRFAMILY
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005116 PyModule_AddIntConstant(m, "EAI_ADDRFAMILY", EAI_ADDRFAMILY);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005117#endif
5118#ifdef EAI_AGAIN
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005119 PyModule_AddIntConstant(m, "EAI_AGAIN", EAI_AGAIN);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005120#endif
5121#ifdef EAI_BADFLAGS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005122 PyModule_AddIntConstant(m, "EAI_BADFLAGS", EAI_BADFLAGS);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005123#endif
5124#ifdef EAI_FAIL
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005125 PyModule_AddIntConstant(m, "EAI_FAIL", EAI_FAIL);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005126#endif
5127#ifdef EAI_FAMILY
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005128 PyModule_AddIntConstant(m, "EAI_FAMILY", EAI_FAMILY);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005129#endif
5130#ifdef EAI_MEMORY
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005131 PyModule_AddIntConstant(m, "EAI_MEMORY", EAI_MEMORY);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005132#endif
5133#ifdef EAI_NODATA
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005134 PyModule_AddIntConstant(m, "EAI_NODATA", EAI_NODATA);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005135#endif
5136#ifdef EAI_NONAME
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005137 PyModule_AddIntConstant(m, "EAI_NONAME", EAI_NONAME);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005138#endif
Martin v. Löwisda91d022003-12-30 11:14:01 +00005139#ifdef EAI_OVERFLOW
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005140 PyModule_AddIntConstant(m, "EAI_OVERFLOW", EAI_OVERFLOW);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005141#endif
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005142#ifdef EAI_SERVICE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005143 PyModule_AddIntConstant(m, "EAI_SERVICE", EAI_SERVICE);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005144#endif
5145#ifdef EAI_SOCKTYPE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005146 PyModule_AddIntConstant(m, "EAI_SOCKTYPE", EAI_SOCKTYPE);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005147#endif
5148#ifdef EAI_SYSTEM
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005149 PyModule_AddIntConstant(m, "EAI_SYSTEM", EAI_SYSTEM);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005150#endif
5151#ifdef EAI_BADHINTS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005152 PyModule_AddIntConstant(m, "EAI_BADHINTS", EAI_BADHINTS);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005153#endif
5154#ifdef EAI_PROTOCOL
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005155 PyModule_AddIntConstant(m, "EAI_PROTOCOL", EAI_PROTOCOL);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005156#endif
5157#ifdef EAI_MAX
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005158 PyModule_AddIntConstant(m, "EAI_MAX", EAI_MAX);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005159#endif
5160#ifdef AI_PASSIVE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005161 PyModule_AddIntConstant(m, "AI_PASSIVE", AI_PASSIVE);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005162#endif
5163#ifdef AI_CANONNAME
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005164 PyModule_AddIntConstant(m, "AI_CANONNAME", AI_CANONNAME);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005165#endif
5166#ifdef AI_NUMERICHOST
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005167 PyModule_AddIntConstant(m, "AI_NUMERICHOST", AI_NUMERICHOST);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005168#endif
Martin v. Löwisda91d022003-12-30 11:14:01 +00005169#ifdef AI_NUMERICSERV
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005170 PyModule_AddIntConstant(m, "AI_NUMERICSERV", AI_NUMERICSERV);
Martin v. Löwisda91d022003-12-30 11:14:01 +00005171#endif
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005172#ifdef AI_MASK
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005173 PyModule_AddIntConstant(m, "AI_MASK", AI_MASK);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005174#endif
5175#ifdef AI_ALL
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005176 PyModule_AddIntConstant(m, "AI_ALL", AI_ALL);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005177#endif
5178#ifdef AI_V4MAPPED_CFG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005179 PyModule_AddIntConstant(m, "AI_V4MAPPED_CFG", AI_V4MAPPED_CFG);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005180#endif
5181#ifdef AI_ADDRCONFIG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005182 PyModule_AddIntConstant(m, "AI_ADDRCONFIG", AI_ADDRCONFIG);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005183#endif
5184#ifdef AI_V4MAPPED
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005185 PyModule_AddIntConstant(m, "AI_V4MAPPED", AI_V4MAPPED);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005186#endif
5187#ifdef AI_DEFAULT
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005188 PyModule_AddIntConstant(m, "AI_DEFAULT", AI_DEFAULT);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005189#endif
5190#ifdef NI_MAXHOST
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005191 PyModule_AddIntConstant(m, "NI_MAXHOST", NI_MAXHOST);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005192#endif
5193#ifdef NI_MAXSERV
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005194 PyModule_AddIntConstant(m, "NI_MAXSERV", NI_MAXSERV);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005195#endif
5196#ifdef NI_NOFQDN
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005197 PyModule_AddIntConstant(m, "NI_NOFQDN", NI_NOFQDN);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005198#endif
5199#ifdef NI_NUMERICHOST
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005200 PyModule_AddIntConstant(m, "NI_NUMERICHOST", NI_NUMERICHOST);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005201#endif
5202#ifdef NI_NAMEREQD
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005203 PyModule_AddIntConstant(m, "NI_NAMEREQD", NI_NAMEREQD);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005204#endif
5205#ifdef NI_NUMERICSERV
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005206 PyModule_AddIntConstant(m, "NI_NUMERICSERV", NI_NUMERICSERV);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005207#endif
5208#ifdef NI_DGRAM
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005209 PyModule_AddIntConstant(m, "NI_DGRAM", NI_DGRAM);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005210#endif
5211
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005212 /* shutdown() parameters */
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005213#ifdef SHUT_RD
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005214 PyModule_AddIntConstant(m, "SHUT_RD", SHUT_RD);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005215#elif defined(SD_RECEIVE)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005216 PyModule_AddIntConstant(m, "SHUT_RD", SD_RECEIVE);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005217#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005218 PyModule_AddIntConstant(m, "SHUT_RD", 0);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005219#endif
5220#ifdef SHUT_WR
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005221 PyModule_AddIntConstant(m, "SHUT_WR", SHUT_WR);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005222#elif defined(SD_SEND)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005223 PyModule_AddIntConstant(m, "SHUT_WR", SD_SEND);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005224#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005225 PyModule_AddIntConstant(m, "SHUT_WR", 1);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005226#endif
5227#ifdef SHUT_RDWR
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005228 PyModule_AddIntConstant(m, "SHUT_RDWR", SHUT_RDWR);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005229#elif defined(SD_BOTH)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005230 PyModule_AddIntConstant(m, "SHUT_RDWR", SD_BOTH);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005231#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005232 PyModule_AddIntConstant(m, "SHUT_RDWR", 2);
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005233#endif
5234
Christian Heimesfaf2f632008-01-06 16:59:19 +00005235#ifdef SIO_RCVALL
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005236 {
5237 DWORD codes[] = {SIO_RCVALL, SIO_KEEPALIVE_VALS};
5238 const char *names[] = {"SIO_RCVALL", "SIO_KEEPALIVE_VALS"};
5239 int i;
5240 for(i = 0; i<sizeof(codes)/sizeof(*codes); ++i) {
5241 PyObject *tmp;
5242 tmp = PyLong_FromUnsignedLong(codes[i]);
5243 if (tmp == NULL)
5244 return NULL;
5245 PyModule_AddObject(m, names[i], tmp);
5246 }
5247 }
5248 PyModule_AddIntConstant(m, "RCVALL_OFF", RCVALL_OFF);
5249 PyModule_AddIntConstant(m, "RCVALL_ON", RCVALL_ON);
5250 PyModule_AddIntConstant(m, "RCVALL_SOCKETLEVELONLY", RCVALL_SOCKETLEVELONLY);
Amaury Forgeot d'Arc762681b2008-06-12 23:03:41 +00005251#ifdef RCVALL_IPLEVEL
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005252 PyModule_AddIntConstant(m, "RCVALL_IPLEVEL", RCVALL_IPLEVEL);
Amaury Forgeot d'Arc762681b2008-06-12 23:03:41 +00005253#endif
5254#ifdef RCVALL_MAX
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005255 PyModule_AddIntConstant(m, "RCVALL_MAX", RCVALL_MAX);
Amaury Forgeot d'Arc762681b2008-06-12 23:03:41 +00005256#endif
Christian Heimesfaf2f632008-01-06 16:59:19 +00005257#endif /* _MSTCPIP_ */
5258
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005259 /* Initialize gethostbyname lock */
Just van Rossum1040d2c2003-05-09 07:53:18 +00005260#if defined(USE_GETHOSTBYNAME_LOCK) || defined(USE_GETADDRINFO_LOCK)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005261 netdb_lock = PyThread_allocate_lock();
Guido van Rossum4f199ea1998-04-09 20:56:35 +00005262#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005263 return m;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00005264}
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005265
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00005266
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005267#ifndef HAVE_INET_PTON
Christian Heimes96e7b3d2007-11-20 06:51:17 +00005268#if !defined(NTDDI_VERSION) || (NTDDI_VERSION < NTDDI_LONGHORN)
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00005269
5270/* Simplistic emulation code for inet_pton that only works for IPv4 */
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00005271/* These are not exposed because they do not set errno properly */
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00005272
Guido van Rossum3eede5a2002-06-07 02:08:35 +00005273int
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00005274inet_pton(int af, const char *src, void *dst)
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005275{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005276 if (af == AF_INET) {
Benjamin Petersonf91df042009-02-13 02:50:59 +00005277#if (SIZEOF_INT != 4)
5278#error "Not sure if in_addr_t exists and int is not 32-bits."
5279#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005280 unsigned int packed_addr;
5281 packed_addr = inet_addr(src);
5282 if (packed_addr == INADDR_NONE)
5283 return 0;
5284 memcpy(dst, &packed_addr, 4);
5285 return 1;
5286 }
5287 /* Should set errno to EAFNOSUPPORT */
5288 return -1;
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005289}
5290
Martin v. Löwisc925b1532001-07-21 09:42:15 +00005291const char *
5292inet_ntop(int af, const void *src, char *dst, socklen_t size)
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005293{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00005294 if (af == AF_INET) {
5295 struct in_addr packed_addr;
5296 if (size < 16)
5297 /* Should set errno to ENOSPC. */
5298 return NULL;
5299 memcpy(&packed_addr, src, sizeof(packed_addr));
5300 return strncpy(dst, inet_ntoa(packed_addr), size);
5301 }
5302 /* Should set errno to EAFNOSUPPORT */
5303 return NULL;
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005304}
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00005305
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005306#endif
Christian Heimesb6150692007-11-15 23:37:07 +00005307#endif